/*********************************************************************************************************************************** Test Archive Push Command ***********************************************************************************************************************************/ #include "common/io/bufferRead.h" #include "common/io/bufferWrite.h" #include "common/io/handleRead.h" #include "common/io/handleWrite.h" #include "common/time.h" #include "postgres/version.h" #include "storage/posix/storage.h" #include "common/harnessConfig.h" #include "common/harnessFork.h" #include "common/harnessInfo.h" /*********************************************************************************************************************************** Test Run ***********************************************************************************************************************************/ void testRun(void) { FUNCTION_HARNESS_VOID(); // Create default storage object for testing Storage *storageTest = storagePosixNew( strNew(testPath()), STORAGE_MODE_FILE_DEFAULT, STORAGE_MODE_PATH_DEFAULT, true, NULL); // Start a protocol server to test the protocol directly Buffer *serverWrite = bufNew(8192); IoWrite *serverWriteIo = ioBufferWriteNew(serverWrite); ioWriteOpen(serverWriteIo); ProtocolServer *server = protocolServerNew( strNew("test"), strNew("test"), ioBufferReadNew(bufNew(0)), serverWriteIo); bufUsedSet(serverWrite, 0); // ***************************************************************************************************************************** if (testBegin("archivePushReadyList(), archivePushProcessList(), and archivePushDrop()")) { StringList *argList = strLstNew(); strLstAddZ(argList, "--stanza=db"); strLstAdd(argList, strNewFmt("--pg1-path=%s/db", testPath())); strLstAdd(argList, strNewFmt("--spool-path=%s/spool", testPath())); harnessCfgLoad(cfgCmdArchivePushAsync, argList); storagePathCreateNP(storagePgWrite(), strNew("pg_wal/archive_status")); storagePathCreateNP(storageTest, strNew("spool/archive/db/out")); // Create ok files to indicate WAL that has already been archived storagePutNP( storageNewWriteNP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000001.ok")), NULL); storagePutNP( storageNewWriteNP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000003.ok")), NULL); storagePutNP( storageNewWriteNP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000004.ok")), NULL); storagePutNP( storageNewWriteNP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000005.error")), NULL); storagePutNP( storageNewWriteNP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000006.error")), NULL); storagePutNP( storageNewWriteNP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/global.error")), NULL); // Create ready files for wal that still needs to be archived storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/archive_status/000000010000000100000002.ready")), NULL); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/archive_status/000000010000000100000003.ready")), NULL); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/archive_status/000000010000000100000005.ready")), NULL); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/archive_status/000000010000000100000006.ready")), NULL); TEST_RESULT_STR( strPtr(strLstJoin(archivePushProcessList(strNewFmt("%s/db/pg_wal", testPath())), "|")), "000000010000000100000002|000000010000000100000005|000000010000000100000006", "ready list"); TEST_RESULT_STR( strPtr(strLstJoin(strLstSort(storageListNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT)), sortOrderAsc), "|")), "000000010000000100000003.ok", "remaining status list"); // Test drop // ------------------------------------------------------------------------------------------------------------------------- StringList *argListDrop = strLstDup(argList); strLstAdd(argListDrop, strNewFmt("--archive-push-queue-max=%zu", (size_t)1024 * 1024 * 1024)); harnessCfgLoad(cfgCmdArchivePushAsync, argListDrop); // Write the files that we claim are in pg_wal Buffer *walBuffer = bufNew((size_t)16 * 1024 * 1024); bufUsedSet(walBuffer, bufSize(walBuffer)); memset(bufPtr(walBuffer), 0, bufSize(walBuffer)); pgWalTestToBuffer((PgWal){.version = PG_VERSION_10, .systemId = 0xFACEFACEFACEFACE}, walBuffer); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000002")), walBuffer); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000003")), walBuffer); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000005")), walBuffer); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000006")), walBuffer); // Queue max is high enough that no WAL will be dropped TEST_RESULT_BOOL( archivePushDrop(strNew("pg_wal"), archivePushProcessList(strNewFmt("%s/db/pg_wal", testPath()))), false, "wal is not dropped"); // Now set queue max low enough that WAL will be dropped argListDrop = strLstDup(argList); strLstAdd(argListDrop, strNewFmt("--archive-push-queue-max=%zu", (size_t)16 * 1024 * 1024 * 2)); harnessCfgLoad(cfgCmdArchivePushAsync, argListDrop); TEST_RESULT_BOOL( archivePushDrop(strNew("pg_wal"), archivePushProcessList(strNewFmt("%s/db/pg_wal", testPath()))), true, "wal is dropped"); } // ***************************************************************************************************************************** if (testBegin("archivePushCheck()")) { StringList *argList = strLstNew(); strLstAddZ(argList, "--stanza=test"); strLstAdd(argList, strNewFmt("--pg1-path=%s/pg", testPath())); strLstAdd(argList, strNewFmt("--repo1-path=%s/repo", testPath())); harnessCfgLoad(cfgCmdArchivePush, argList); // Check mismatched pg_control and archive.info // ------------------------------------------------------------------------------------------------------------------------- storagePutNP( storageNewWriteNP(storageTest, strNew("pg/" PG_PATH_GLOBAL "/" PG_FILE_PGCONTROL)), pgControlTestToBuffer((PgControl){.version = PG_VERSION_96, .systemId = 0xFACEFACEFACEFACE})); // Create incorrect archive info storagePutNP( storageNewWriteNP(storageTest, strNew("repo/archive/test/archive.info")), harnessInfoChecksumZ( "[db]\n" "db-id=1\n" "\n" "[db:history]\n" "1={\"db-id\":5555555555555555555,\"db-version\":\"9.4\"}\n")); TEST_ERROR( archivePushCheck(cipherTypeNone, NULL), ArchiveMismatchError, "PostgreSQL version 9.6, system-id 18072658121562454734 do not match stanza version 9.4, system-id 5555555555555555555" "\nHINT: are you archiving to the correct stanza?"); // Fix the version storagePutNP( storageNewWriteNP(storageTest, strNew("repo/archive/test/archive.info")), harnessInfoChecksumZ( "[db]\n" "db-id=1\n" "\n" "[db:history]\n" "1={\"db-id\":5555555555555555555,\"db-version\":\"9.6\"}\n")); TEST_ERROR( archivePushCheck(cipherTypeNone, NULL), ArchiveMismatchError, "PostgreSQL version 9.6, system-id 18072658121562454734 do not match stanza version 9.6, system-id 5555555555555555555" "\nHINT: are you archiving to the correct stanza?"); // Fix archive info storagePutNP( storageNewWriteNP(storageTest, strNew("repo/archive/test/archive.info")), harnessInfoChecksumZ( "[db]\n" "db-id=1\n" "\n" "[db:history]\n" "1={\"db-id\":18072658121562454734,\"db-version\":\"9.6\"}\n")); ArchivePushCheckResult result = {0}; TEST_ASSIGN(result, archivePushCheck(cipherTypeNone, NULL), "get archive check result"); TEST_RESULT_UINT(result.pgVersion, PG_VERSION_96, "check pg version"); TEST_RESULT_UINT(result.pgSystemId, 0xFACEFACEFACEFACE, "check pg system id"); TEST_RESULT_UINT(result.pgWalSegmentSize, 16 * 1024 * 1024, "check wal segment size"); TEST_RESULT_STR(strPtr(result.archiveId), "9.6-1", "check archive id"); TEST_RESULT_STR(strPtr(result.archiveCipherPass), NULL, "check archive cipher pass (not set in this test)"); } // ***************************************************************************************************************************** if (testBegin("Synchronous cmdArchivePush(), archivePushFile() and archivePushProtocol()")) { StringList *argList = strLstNew(); strLstAddZ(argList, "--stanza=test"); harnessCfgLoad(cfgCmdArchivePush, argList); TEST_ERROR(cmdArchivePush(), ParamRequiredError, "WAL segment to push required"); // ------------------------------------------------------------------------------------------------------------------------- StringList *argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "pg_wal/000000010000000100000001"); harnessCfgLoad(cfgCmdArchivePush, argListTemp); TEST_ERROR( cmdArchivePush(), OptionRequiredError, "option 'pg1-path' must be specified when relative wal paths are used" "\nHINT: is %f passed to archive-push instead of %p?" "\nHINT: PostgreSQL may pass relative paths even with %p depending on the environment."); // Create pg_control and archive.info // ------------------------------------------------------------------------------------------------------------------------- strLstAdd(argList, strNewFmt("--pg1-path=%s/pg", testPath())); strLstAdd(argList, strNewFmt("--repo1-path=%s/repo", testPath())); argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "pg_wal/000000010000000100000001"); harnessCfgLoad(cfgCmdArchivePush, argListTemp); storagePutNP( storageNewWriteNP(storageTest, strNew("pg/" PG_PATH_GLOBAL "/" PG_FILE_PGCONTROL)), pgControlTestToBuffer((PgControl){.version = PG_VERSION_11, .systemId = 0xFACEFACEFACEFACE})); storagePutNP( storageNewWriteNP(storageTest, strNew("repo/archive/test/archive.info")), harnessInfoChecksumZ( "[db]\n" "db-id=1\n" "\n" "[db:history]\n" "1={\"db-id\":18072658121562454734,\"db-version\":\"11\"}\n")); // Generate WAL with incorrect headers and try to push them // ------------------------------------------------------------------------------------------------------------------------- Buffer *walBuffer1 = bufNew((size_t)16 * 1024 * 1024); bufUsedSet(walBuffer1, bufSize(walBuffer1)); memset(bufPtr(walBuffer1), 0, bufSize(walBuffer1)); pgWalTestToBuffer((PgWal){.version = PG_VERSION_10, .systemId = 0xFACEFACEFACEFACE}, walBuffer1); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000001")), walBuffer1); TEST_ERROR( cmdArchivePush(), ArchiveMismatchError, strPtr( strNewFmt( "WAL file '%s/pg/pg_wal/000000010000000100000001' version 10, system-id 18072658121562454734 do not match" " stanza version 11, system-id 18072658121562454734", testPath()))); memset(bufPtr(walBuffer1), 0, bufSize(walBuffer1)); pgWalTestToBuffer((PgWal){.version = PG_VERSION_11, .systemId = 0xECAFECAFECAFECAF}, walBuffer1); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000001")), walBuffer1); TEST_ERROR( cmdArchivePush(), ArchiveMismatchError, strPtr( strNewFmt( "WAL file '%s/pg/pg_wal/000000010000000100000001' version 11, system-id 17055110554209741999 do not match" " stanza version 11, system-id 18072658121562454734", testPath()))); // Generate valid WAL and push them // ------------------------------------------------------------------------------------------------------------------------- memset(bufPtr(walBuffer1), 0, bufSize(walBuffer1)); pgWalTestToBuffer((PgWal){.version = PG_VERSION_11, .systemId = 0xFACEFACEFACEFACE}, walBuffer1); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000001")), walBuffer1); TEST_RESULT_VOID(cmdArchivePush(), "push the WAL segment"); harnessLogResult("P00 INFO: pushed WAL file '000000010000000100000001' to the archive"); TEST_RESULT_BOOL( storageExistsNP( storageTest, strNewFmt( "repo/archive/test/11-1/0000000100000001/000000010000000100000001-%s.gz", TEST_64BIT() ? "3e5ecd22712f319b2420d5b901fd29f4f6be2336" : "6903dce7e3cd64ba9a6134056405eaeb8dedcd37")), true, "check repo for WAL file"); TEST_RESULT_VOID(cmdArchivePush(), "push the WAL segment again"); harnessLogResult( "P00 WARN: WAL file '000000010000000100000001' already exists in the archive with the same checksum\n" " HINT: this is valid in some recovery scenarios but may also indicate a problem.\n" "P00 INFO: pushed WAL file '000000010000000100000001' to the archive"); // Now create a new WAL buffer with a different checksum to test checksum errors Buffer *walBuffer2 = bufNew((size_t)16 * 1024 * 1024); bufUsedSet(walBuffer2, bufSize(walBuffer2)); memset(bufPtr(walBuffer2), 0xFF, bufSize(walBuffer2)); pgWalTestToBuffer((PgWal){.version = PG_VERSION_11, .systemId = 0xFACEFACEFACEFACE}, walBuffer2); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000001")), walBuffer2); TEST_ERROR(cmdArchivePush(), ArchiveDuplicateError, "WAL file '000000010000000100000001' already exists in the archive"); // Save it to a new file instead argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "pg_wal/000000010000000100000002"); harnessCfgLoad(cfgCmdArchivePush, argListTemp); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/000000010000000100000002")), walBuffer2); TEST_RESULT_VOID(cmdArchivePush(), "push the WAL segment"); harnessLogResult("P00 INFO: pushed WAL file '000000010000000100000002' to the archive"); TEST_RESULT_BOOL( storageExistsNP( storageTest, strNewFmt( "repo/archive/test/11-1/0000000100000001/000000010000000100000002-%s.gz", TEST_64BIT() ? "edad2f5a9d8a03ee3c09e8ce92c771e0d20232f5" : "e7c81f5513e0c6e3f19b9dbfc450019165994dda")), true, "check repo for WAL file"); // Push a history file // ------------------------------------------------------------------------------------------------------------------------- argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "pg_wal/00000001.history"); harnessCfgLoad(cfgCmdArchivePush, argListTemp); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/00000001.history")), BUFSTRDEF("FAKEHISTORY")); TEST_RESULT_VOID(cmdArchivePush(), "push a history file"); harnessLogResult("P00 INFO: pushed WAL file '00000001.history' to the archive"); TEST_RESULT_BOOL( storageExistsNP(storageTest, strNew("repo/archive/test/11-1/00000001.history")), true, "check repo for history file"); // Check drop functionality // ------------------------------------------------------------------------------------------------------------------------- storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/archive_status/000000010000000100000001.ready")), NULL); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_wal/archive_status/000000010000000100000002.ready")), NULL); argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "--archive-push-queue-max=16m"); strLstAddZ(argListTemp, "pg_wal/000000010000000100000002"); harnessCfgLoad(cfgCmdArchivePush, argListTemp); TEST_RESULT_VOID(cmdArchivePush(), "drop WAL file"); harnessLogResult("P00 WARN: dropped WAL file '000000010000000100000002' because archive queue exceeded 16MB"); argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "--archive-push-queue-max=1GB"); strLstAddZ(argListTemp, "pg_wal/000000010000000100000002"); harnessCfgLoad(cfgCmdArchivePush, argListTemp); TEST_RESULT_VOID(cmdArchivePush(), "push WAL file again"); harnessLogResult( "P00 WARN: WAL file '000000010000000100000002' already exists in the archive with the same checksum\n" " HINT: this is valid in some recovery scenarios but may also indicate a problem.\n" "P00 INFO: pushed WAL file '000000010000000100000002' to the archive"); // Check protocol function directly // ------------------------------------------------------------------------------------------------------------------------- VariantList *paramList = varLstNew(); varLstAdd(paramList, varNewStr(strNewFmt("%s/pg/pg_wal/000000010000000100000002", testPath()))); varLstAdd(paramList, varNewStrZ("11-1")); varLstAdd(paramList, varNewUInt64(PG_VERSION_11)); varLstAdd(paramList, varNewUInt64(0xFACEFACEFACEFACE)); varLstAdd(paramList, varNewStrZ("000000010000000100000002")); varLstAdd(paramList, varNewUInt64(cipherTypeNone)); varLstAdd(paramList, NULL); varLstAdd(paramList, varNewBool(false)); varLstAdd(paramList, varNewInt(6)); TEST_RESULT_BOOL( archivePushProtocol(PROTOCOL_COMMAND_ARCHIVE_PUSH_STR, paramList, server), true, "protocol archive put"); TEST_RESULT_STR( strPtr(strNewBuf(serverWrite)), "{\"out\":\"WAL file '000000010000000100000002' already exists in the archive with the same checksum" "\\nHINT: this is valid in some recovery scenarios but may also indicate a problem.\"}\n", "check result"); bufUsedSet(serverWrite, 0); // Check invalid protocol function // ------------------------------------------------------------------------------------------------------------------------- TEST_RESULT_BOOL(archivePushProtocol(strNew(BOGUS_STR), paramList, server), false, "invalid function"); // Create a new encrypted repo to test encryption // ------------------------------------------------------------------------------------------------------------------------- storagePathRemoveP(storageTest, strNew("repo"), .errorOnMissing = true, .recurse = true); StorageWrite *infoWrite = storageNewWriteNP(storageTest, strNew("repo/archive/test/archive.info")); ioFilterGroupAdd( ioWriteFilterGroup(storageWriteIo(infoWrite)), cipherBlockNew(cipherModeEncrypt, cipherTypeAes256Cbc, BUFSTRDEF("badpassphrase"), NULL)); storagePutNP( infoWrite, harnessInfoChecksumZ( "[cipher]\n" "cipher-pass=\"badsubpassphrase\"\n" "\n" "[db]\n" "db-id=1\n" "\n" "[db:history]\n" "1={\"db-id\":18072658121562454734,\"db-version\":\"11\"}")); // Push encrypted WAL segment argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "pg_wal/000000010000000100000002"); strLstAddZ(argListTemp, "--repo1-cipher-type=aes-256-cbc"); strLstAddZ(argListTemp, "--no-compress"); setenv("PGBACKREST_REPO1_CIPHER_PASS", "badpassphrase", true); harnessCfgLoad(cfgCmdArchivePush, argListTemp); unsetenv("PGBACKREST_REPO1_CIPHER_PASS"); TEST_RESULT_VOID(cmdArchivePush(), "push the WAL segment"); harnessLogResult("P00 INFO: pushed WAL file '000000010000000100000002' to the archive"); TEST_RESULT_BOOL( storageExistsNP( storageTest, strNewFmt( "repo/archive/test/11-1/0000000100000001/000000010000000100000002-%s", TEST_64BIT() ? "edad2f5a9d8a03ee3c09e8ce92c771e0d20232f5" : "e7c81f5513e0c6e3f19b9dbfc450019165994dda")), true, "check repo for WAL file"); } // ***************************************************************************************************************************** if (testBegin("Asynchronous cmdArchivePush() and cmdArchivePushAsync()")) { harnessLogLevelSet(logLevelDetail); // Call with a bogus exe name so the async process will error out and we can make sure timeouts work // ------------------------------------------------------------------------------------------------------------------------- StringList *argList = strLstNew(); strLstAddZ(argList, "pgbackrest-bogus"); strLstAddZ(argList, "--stanza=test"); strLstAddZ(argList, "--archive-async"); strLstAddZ(argList, "--archive-timeout=1"); strLstAdd(argList, strNewFmt("--lock-path=%s/lock", testPath())); strLstAdd(argList, strNewFmt("--spool-path=%s/spool", testPath())); strLstAdd(argList, strNewFmt("--pg1-path=%s/pg", testPath())); strLstAdd(argList, strNewFmt("--repo1-path=%s/repo", testPath())); strLstAddZ(argList, "archive-push"); strLstAddZ(argList, "pg_wal/bogus"); harnessCfgLoadRaw(strLstSize(argList), strLstPtr(argList)); TEST_ERROR( cmdArchivePush(), ArchiveTimeoutError, "unable to push WAL file 'bogus' to the archive asynchronously after 1 second(s)"); // Create pg_control and archive.info // ------------------------------------------------------------------------------------------------------------------------- argList = strLstNew(); strLstAddZ(argList, "--stanza=test"); strLstAddZ(argList, "--archive-async"); strLstAddZ(argList, "--no-compress"); strLstAdd(argList, strNewFmt("--spool-path=%s/spool", testPath())); strLstAdd(argList, strNewFmt("--pg1-path=%s/pg", testPath())); strLstAdd(argList, strNewFmt("--repo1-path=%s/repo", testPath())); strLstAddZ(argList, "--log-subprocess"); storagePutNP( storageNewWriteNP(storageTest, strNew("pg/" PG_PATH_GLOBAL "/" PG_FILE_PGCONTROL)), pgControlTestToBuffer((PgControl){.version = PG_VERSION_94, .systemId = 0xAAAABBBBCCCCDDDD})); storagePutNP( storageNewWriteNP(storageTest, strNew("repo/archive/test/archive.info")), harnessInfoChecksumZ( "[db]\n" "db-id=1\n" "\n" "[db:history]\n" "1={\"db-id\":12297848147757817309,\"db-version\":\"9.4\"}\n")); // Write out an error file that will be ignored on the first pass, then the async process will write a new one // ------------------------------------------------------------------------------------------------------------------------- StringList *argListTemp = strLstDup(argList); strLstAdd(argListTemp, strNewFmt("%s/pg/pg_xlog/000000010000000100000001", testPath())); harnessCfgLoad(cfgCmdArchivePush, argListTemp); storagePathCreateNP(storagePgWrite(), strNew("pg_xlog/archive_status")); storagePutNP( storageNewWriteNP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000001.error")), BUFSTRDEF("25\nbogus error")); TEST_ERROR(cmdArchivePush(), AssertError, "no WAL files to process"); storageRemoveP(storageSpoolWrite(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/global.error"), .errorOnMissing = true); // Acquire a lock so the async process will not be able to run -- this will result in a timeout // ------------------------------------------------------------------------------------------------------------------------- argListTemp = strLstDup(argList); strLstAdd(argListTemp, strNewFmt("%s/pg/pg_xlog/000000010000000100000001", testPath())); strLstAddZ(argListTemp, "--archive-timeout=1"); harnessCfgLoad(cfgCmdArchivePush, argListTemp); HARNESS_FORK_BEGIN() { HARNESS_FORK_CHILD_BEGIN(0, true) { IoRead *read = ioHandleReadNew(strNew("child read"), HARNESS_FORK_CHILD_READ(), 2000); ioReadOpen(read); IoWrite *write = ioHandleWriteNew(strNew("child write"), HARNESS_FORK_CHILD_WRITE()); ioWriteOpen(write); lockAcquire(cfgOptionStr(cfgOptLockPath), cfgOptionStr(cfgOptStanza), cfgLockType(), 30000, true); // Let the parent know the lock has been acquired and wait for the parent to allow lock release ioWriteStrLine(write, strNew("")); ioWriteFlush(write); ioReadLine(read); lockRelease(true); } HARNESS_FORK_CHILD_END(); HARNESS_FORK_PARENT_BEGIN() { IoRead *read = ioHandleReadNew(strNew("parent read"), HARNESS_FORK_PARENT_READ_PROCESS(0), 2000); ioReadOpen(read); IoWrite *write = ioHandleWriteNew(strNew("parent write"), HARNESS_FORK_PARENT_WRITE_PROCESS(0)); ioWriteOpen(write); // Wait for the child to acquire the lock ioReadLine(read); TEST_ERROR( cmdArchivePush(), ArchiveTimeoutError, "unable to push WAL file '000000010000000100000001' to the archive asynchronously after 1 second(s)"); // Notify the child to release the lock ioWriteLine(write, bufNew(0)); ioWriteFlush(write); } HARNESS_FORK_PARENT_END(); } HARNESS_FORK_END(); // Actually push a WAL file // ------------------------------------------------------------------------------------------------------------------------- argListTemp = strLstDup(argList); strLstAdd(argListTemp, strNewFmt("%s/pg/pg_xlog/000000010000000100000001", testPath())); harnessCfgLoad(cfgCmdArchivePush, argListTemp); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_xlog/archive_status/000000010000000100000001.ready")), NULL); Buffer *walBuffer1 = bufNew((size_t)16 * 1024 * 1024); bufUsedSet(walBuffer1, bufSize(walBuffer1)); memset(bufPtr(walBuffer1), 0xFF, bufSize(walBuffer1)); pgWalTestToBuffer((PgWal){.version = PG_VERSION_94, .systemId = 0xAAAABBBBCCCCDDDD}, walBuffer1); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_xlog/000000010000000100000001")), walBuffer1); TEST_RESULT_VOID(cmdArchivePush(), "push the WAL segment"); harnessLogResult("P00 INFO: pushed WAL file '000000010000000100000001' to the archive asynchronously"); TEST_RESULT_BOOL( storageExistsNP( storageTest, strNewFmt( "repo/archive/test/9.4-1/0000000100000001/000000010000000100000001-%s", TEST_64BIT() ? "f81d63dd5e258cd607534f3531bbd71442797e37" : "02d228126281e8e102b35a2737e45a0527946296")), true, "check repo for WAL file"); // Direct tests of the async function // ------------------------------------------------------------------------------------------------------------------------- argList = strLstNew(); strLstAddZ(argList, "--stanza=test"); strLstAddZ(argList, "--no-compress"); strLstAdd(argList, strNewFmt("--spool-path=%s/spool", testPath())); strLstAdd(argList, strNewFmt("--pg1-path=%s/pg", testPath())); strLstAdd(argList, strNewFmt("--repo1-path=%s/repo", testPath())); strLstAddZ(argList, "--log-subprocess"); harnessCfgLoad(cfgCmdArchivePushAsync, argList); TEST_ERROR(cmdArchivePushAsync(), ParamRequiredError, "WAL path to push required"); // Check that global.error is created // ------------------------------------------------------------------------------------------------------------------------- // Remove data from prior tests storagePathRemoveP(storageSpoolWrite(), STORAGE_SPOOL_ARCHIVE_OUT_STR, .recurse = true); storagePathCreateNP(storageSpoolWrite(), STORAGE_SPOOL_ARCHIVE_OUT_STR); storagePathRemoveP(storageRepoWrite(), strNew(STORAGE_REPO_ARCHIVE "/9.4-1"), .recurse = true); storagePathCreateNP(storageRepoWrite(), strNew(STORAGE_REPO_ARCHIVE "/9.4-1")); storagePathRemoveP(storagePgWrite(), strNew("pg_xlog/archive_status"), .recurse = true); storagePathCreateNP(storagePgWrite(), strNew("pg_xlog/archive_status")); strLstAdd(argList, strNewFmt("%s/pg/pg_xlog", testPath())); harnessCfgLoad(cfgCmdArchivePushAsync, argList); TEST_ERROR(cmdArchivePushAsync(), AssertError, "no WAL files to process"); TEST_RESULT_STR( strPtr(strNewBuf(storageGetNP(storageNewReadNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/global.error"))))), "25\nno WAL files to process", "check global.error"); TEST_RESULT_STR( strPtr(strLstJoin(strLstSort(storageListNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT)), sortOrderAsc), "|")), "global.error", "check status files"); // Push WAL // ------------------------------------------------------------------------------------------------------------------------- // Recreate ready file for WAL 1 storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_xlog/archive_status/000000010000000100000001.ready")), NULL); // Create a ready file for WAL 2 but don't create the segment yet -- this will test the file error storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_xlog/archive_status/000000010000000100000002.ready")), NULL); TEST_RESULT_VOID(cmdArchivePushAsync(), "push WAL segments"); harnessLogResult( strPtr( strNewFmt( "P00 INFO: push 2 WAL file(s) to archive: 000000010000000100000001...000000010000000100000002\n" "P01 DETAIL: pushed WAL file '000000010000000100000001' to the archive\n" "P01 WARN: could not push WAL file '000000010000000100000002' to the archive (will be retried): " "[55] raised from local-1 protocol: " STORAGE_ERROR_READ_MISSING, strPtr(strNewFmt("%s/pg/pg_xlog/000000010000000100000002", testPath()))))); TEST_RESULT_BOOL( storageExistsNP( storageTest, strNewFmt( "repo/archive/test/9.4-1/0000000100000001/000000010000000100000001-%s", TEST_64BIT() ? "f81d63dd5e258cd607534f3531bbd71442797e37" : "02d228126281e8e102b35a2737e45a0527946296")), true, "check repo for WAL 1 file"); TEST_RESULT_STR( strPtr(strLstJoin(strLstSort(storageListNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT)), sortOrderAsc), "|")), "000000010000000100000001.ok|000000010000000100000002.error", "check status files"); // Create WAL 2 segment Buffer *walBuffer2 = bufNew((size_t)16 * 1024 * 1024); bufUsedSet(walBuffer2, bufSize(walBuffer2)); memset(bufPtr(walBuffer2), 0x0C, bufSize(walBuffer2)); pgWalTestToBuffer((PgWal){.version = PG_VERSION_94, .systemId = 0xAAAABBBBCCCCDDDD}, walBuffer2); storagePutNP(storageNewWriteNP(storagePgWrite(), strNew("pg_xlog/000000010000000100000002")), walBuffer2); argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "--archive-push-queue-max=1gb"); harnessCfgLoad(cfgCmdArchivePushAsync, argListTemp); TEST_RESULT_VOID(cmdArchivePushAsync(), "push WAL segments"); harnessLogResult( "P00 INFO: push 1 WAL file(s) to archive: 000000010000000100000002\n" "P01 DETAIL: pushed WAL file '000000010000000100000002' to the archive"); TEST_RESULT_BOOL( storageExistsNP( storageTest, strNewFmt( "repo/archive/test/9.4-1/0000000100000001/000000010000000100000002-%s", TEST_64BIT() ? "0aea6fa5d53500ce548b84a86bc3a29ae77fa048" : "408822a89ef44ef6740e785743bf1b870d8024a2")), true, "check repo for WAL 2 file"); TEST_RESULT_STR( strPtr(strLstJoin(strLstSort(storageListNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT)), sortOrderAsc), "|")), "000000010000000100000001.ok|000000010000000100000002.ok", "check status files"); // Check that drop functionality works // ------------------------------------------------------------------------------------------------------------------------- // Remove status files storagePathRemoveP(storageSpoolWrite(), STORAGE_SPOOL_ARCHIVE_OUT_STR, .recurse = true); storagePathCreateNP(storageSpoolWrite(), STORAGE_SPOOL_ARCHIVE_OUT_STR); argListTemp = strLstDup(argList); strLstAddZ(argListTemp, "--archive-push-queue-max=16m"); harnessCfgLoad(cfgCmdArchivePushAsync, argListTemp); TEST_RESULT_VOID(cmdArchivePushAsync(), "push WAL segments"); harnessLogResult( "P00 INFO: push 2 WAL file(s) to archive: 000000010000000100000001...000000010000000100000002\n" "P00 WARN: dropped WAL file '000000010000000100000001' because archive queue exceeded 16MB\n" "P00 WARN: dropped WAL file '000000010000000100000002' because archive queue exceeded 16MB"); TEST_RESULT_STR( strPtr( strNewBuf( storageGetNP( storageNewReadNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000001.ok"))))), "0\ndropped WAL file '000000010000000100000001' because archive queue exceeded 16MB", "check WAL 1 warning"); TEST_RESULT_STR( strPtr( strNewBuf( storageGetNP( storageNewReadNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT "/000000010000000100000002.ok"))))), "0\ndropped WAL file '000000010000000100000002' because archive queue exceeded 16MB", "check WAL 2 warning"); TEST_RESULT_STR( strPtr(strLstJoin(strLstSort(storageListNP(storageSpool(), strNew(STORAGE_SPOOL_ARCHIVE_OUT)), sortOrderAsc), "|")), "000000010000000100000001.ok|000000010000000100000002.ok", "check status files"); } FUNCTION_HARNESS_RESULT_VOID(); }