#include "postgres_fe.h"
#include <dirent.h>
#include <sys/stat.h>
#include "common/file_utils.h"
#include "compress_io.h"
#include "parallel.h"
#include "pg_backup_utils.h"

Include dependency graph for pg_backup_directory.c:

Data Structures
struct	lclContext

struct	lclTocEntry

Functions
static void	_ArchiveEntry (ArchiveHandle AH, TocEntry te)

static void	_StartData (ArchiveHandle AH, TocEntry te)

static void	_EndData (ArchiveHandle AH, TocEntry te)

static void	_WriteData (ArchiveHandle AH, const void data, size_t dLen)

static int	_WriteByte (ArchiveHandle *AH, const int i)

static int	_ReadByte (ArchiveHandle *)

static void	_WriteBuf (ArchiveHandle AH, const void buf, size_t len)

static void	_ReadBuf (ArchiveHandle AH, void buf, size_t len)

static void	_CloseArchive (ArchiveHandle *AH)

static void	_ReopenArchive (ArchiveHandle *AH)

static void	_PrintTocData (ArchiveHandle AH, TocEntry te)

static void	_WriteExtraToc (ArchiveHandle AH, TocEntry te)

static void	_ReadExtraToc (ArchiveHandle AH, TocEntry te)

static void	_PrintExtraToc (ArchiveHandle AH, TocEntry te)

static void	_StartBlobs (ArchiveHandle AH, TocEntry te)

static void	_StartBlob (ArchiveHandle AH, TocEntry te, Oid oid)

static void	_EndBlob (ArchiveHandle AH, TocEntry te, Oid oid)

static void	_EndBlobs (ArchiveHandle AH, TocEntry te)

static void	_LoadBlobs (ArchiveHandle *AH)

static void	_PrepParallelRestore (ArchiveHandle *AH)

static void	_Clone (ArchiveHandle *AH)

static void	_DeClone (ArchiveHandle *AH)

static int	_WorkerJobRestoreDirectory (ArchiveHandle AH, TocEntry te)

static int	_WorkerJobDumpDirectory (ArchiveHandle AH, TocEntry te)

static void	setFilePath (ArchiveHandle AH, char buf, const char *relativeFilename)

void	InitArchiveFmt_Directory (ArchiveHandle *AH)

static void	_PrintFileData (ArchiveHandle AH, char filename)

Function Documentation

◆ _ArchiveEntry()

static void _ArchiveEntry	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 233 of file pg_backup_directory.c.

References _tocEntry::dataDumper, _tocEntry::desc, _tocEntry::dumpId, lclTocEntry::filename, fn(), _tocEntry::formatData, MAXPGPATH, pg_malloc0(), pg_strdup(), and snprintf.

Referenced by InitArchiveFmt_Directory().

 {
     lclTocEntry *tctx;
     char        fn[MAXPGPATH];
 
     tctx = (lclTocEntry *) pg_malloc0(sizeof(lclTocEntry));
     if (strcmp(te->desc, "BLOBS") == 0)
         tctx->filename = pg_strdup("blobs.toc");
     else if (te->dataDumper)
     {
         snprintf(fn, MAXPGPATH, "%d.dat", te->dumpId);
         tctx->filename = pg_strdup(fn);
     }
     else
         tctx->filename = NULL;
 
     te->formatData = (void *) tctx;
 }

◆ _Clone()

static void _Clone ( ArchiveHandle * AH )

static

Definition at line 794 of file pg_backup_directory.c.

References _archiveHandle::formatData, and pg_malloc().

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     AH->formatData = (lclContext *) pg_malloc(sizeof(lclContext));
     memcpy(AH->formatData, ctx, sizeof(lclContext));
     ctx = (lclContext *) AH->formatData;
 
     /*
      * Note: we do not make a local lo_buf because we expect at most one BLOBS
      * entry per archive, so no parallelism is possible.  Likewise,
      * TOC-entry-local state isn't an issue because any one TOC entry is
      * touched by just one worker child.
      */
 
     /*
      * We also don't copy the ParallelState pointer (pstate), only the leader
      * process ever writes to it.
      */
 }

◆ _CloseArchive()

static void _CloseArchive ( ArchiveHandle * AH )

static

Definition at line 568 of file pg_backup_directory.c.

References archDirectory, archModeWrite, archTar, cfclose(), cfopen_write(), lclContext::dataFH, lclContext::directory, _archiveHandle::dosync, fatal, _archiveHandle::FH, _archiveHandle::format, _archiveHandle::formatData, fsync_dir_recurse(), MAXPGPATH, _archiveHandle::mode, ParallelBackupEnd(), ParallelBackupStart(), PG_BINARY_W, lclContext::pstate, setFilePath(), WriteDataChunks(), WriteHead(), and WriteToc().

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     if (AH->mode == archModeWrite)
     {
         cfp        *tocFH;
         char        fname[MAXPGPATH];
 
         setFilePath(AH, fname, "toc.dat");
 
         /* this will actually fork the processes for a parallel backup */
         ctx->pstate = ParallelBackupStart(AH);
 
         /* The TOC is always created uncompressed */
         tocFH = cfopen_write(fname, PG_BINARY_W, 0);
         if (tocFH == NULL)
             fatal("could not open output file \"%s\": %m", fname);
         ctx->dataFH = tocFH;
 
         /*
          * Write 'tar' in the format field of the toc.dat file. The directory
          * is compatible with 'tar', so there's no point having a different
          * format code for it.
          */
         AH->format = archTar;
         WriteHead(AH);
         AH->format = archDirectory;
         WriteToc(AH);
         if (cfclose(tocFH) != 0)
             fatal("could not close TOC file: %m");
         WriteDataChunks(AH, ctx->pstate);
 
         ParallelBackupEnd(AH, ctx->pstate);
 
         /*
          * In directory mode, there is no need to sync all the entries
          * individually. Just recurse once through all the files generated.
          */
         if (AH->dosync)
             fsync_dir_recurse(ctx->directory);
     }
     AH->FH = NULL;
 }

◆ _DeClone()

static void _DeClone ( ArchiveHandle * AH )

static

Definition at line 816 of file pg_backup_directory.c.

References _archiveHandle::formatData, and free.

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     free(ctx);
 }

◆ _EndBlob()

static void _EndBlob	(	ArchiveHandle *	AH,
		TocEntry *	te,
		Oid	oid
	)

static

Definition at line 676 of file pg_backup_directory.c.

References lclContext::blobsTocFH, buf, cfclose(), cfwrite(), lclContext::dataFH, fatal, _archiveHandle::formatData, and snprintf.

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
     char        buf[50];
     int         len;
 
     /* Close the BLOB data file itself */
     cfclose(ctx->dataFH);
     ctx->dataFH = NULL;
 
     /* register the blob in blobs.toc */
     len = snprintf(buf, sizeof(buf), "%u blob_%u.dat\n", oid, oid);
     if (cfwrite(buf, len, ctx->blobsTocFH) != len)
         fatal("could not write to blobs TOC file");
 }

◆ _EndBlobs()

static void _EndBlobs	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 698 of file pg_backup_directory.c.

References lclContext::blobsTocFH, cfclose(), and _archiveHandle::formatData.

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     cfclose(ctx->blobsTocFH);
     ctx->blobsTocFH = NULL;
 }

◆ _EndData()

static void _EndData	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 367 of file pg_backup_directory.c.

References cfclose(), lclContext::dataFH, and _archiveHandle::formatData.

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     /* Close the file */
     cfclose(ctx->dataFH);
 
     ctx->dataFH = NULL;
 }

◆ _LoadBlobs()

static void _LoadBlobs ( ArchiveHandle * AH )

static

Definition at line 432 of file pg_backup_directory.c.

References _PrintFileData(), lclContext::blobsTocFH, cfclose(), cfeof(), cfgets(), cfopen_read(), lclContext::directory, _restoreOptions::dropSchema, EndRestoreBlob(), EndRestoreBlobs(), fatal, _archiveHandle::formatData, MAXPGPATH, PG_BINARY_R, _archiveHandle::public, Archive::ropt, setFilePath(), snprintf, StartRestoreBlob(), and StartRestoreBlobs().

Referenced by _PrintTocData().

 {
     Oid         oid;
     lclContext *ctx = (lclContext *) AH->formatData;
     char        fname[MAXPGPATH];
     char        line[MAXPGPATH];
 
     StartRestoreBlobs(AH);
 
     setFilePath(AH, fname, "blobs.toc");
 
     ctx->blobsTocFH = cfopen_read(fname, PG_BINARY_R);
 
     if (ctx->blobsTocFH == NULL)
         fatal("could not open large object TOC file \"%s\" for input: %m",
               fname);
 
     /* Read the blobs TOC file line-by-line, and process each blob */
     while ((cfgets(ctx->blobsTocFH, line, MAXPGPATH)) != NULL)
     {
         char        fname[MAXPGPATH];
         char        path[MAXPGPATH];
 
         /* Can't overflow because line and fname are the same length. */
         if (sscanf(line, "%u %s\n", &oid, fname) != 2)
             fatal("invalid line in large object TOC file \"%s\": \"%s\"",
                   fname, line);
 
         StartRestoreBlob(AH, oid, AH->public.ropt->dropSchema);
         snprintf(path, MAXPGPATH, "%s/%s", ctx->directory, fname);
         _PrintFileData(AH, path);
         EndRestoreBlob(AH, oid);
     }
     if (!cfeof(ctx->blobsTocFH))
         fatal("error reading large object TOC file \"%s\"",
               fname);
 
     if (cfclose(ctx->blobsTocFH) != 0)
         fatal("could not close large object TOC file \"%s\": %m",
               fname);
 
     ctx->blobsTocFH = NULL;
 
     EndRestoreBlobs(AH);
 }

◆ _PrepParallelRestore()

static void _PrepParallelRestore ( ArchiveHandle * AH )

static

Definition at line 739 of file pg_backup_directory.c.

References _tocEntry::dataLength, _tocEntry::desc, lclTocEntry::filename, _tocEntry::formatData, MAXPGPATH, _tocEntry::next, REQ_DATA, _tocEntry::reqs, setFilePath(), stat, strlcat(), and _archiveHandle::toc.

Referenced by InitArchiveFmt_Directory().

 {
     TocEntry   *te;
 
     for (te = AH->toc->next; te != AH->toc; te = te->next)
     {
         lclTocEntry *tctx = (lclTocEntry *) te->formatData;
         char        fname[MAXPGPATH];
         struct stat st;
 
         /*
          * A dumpable object has set tctx->filename, any other object has not.
          * (see _ArchiveEntry).
          */
         if (tctx->filename == NULL)
             continue;
 
         /* We may ignore items not due to be restored */
         if ((te->reqs & REQ_DATA) == 0)
             continue;
 
         /*
          * Stat the file and, if successful, put its size in dataLength.  When
          * using compression, the physical file size might not be a very good
          * guide to the amount of work involved in restoring the file, but we
          * only need an approximate indicator of that.
          */
         setFilePath(AH, fname, tctx->filename);
 
         if (stat(fname, &st) == 0)
             te->dataLength = st.st_size;
         else
         {
             /* It might be compressed */
             strlcat(fname, ".gz", sizeof(fname));
             if (stat(fname, &st) == 0)
                 te->dataLength = st.st_size;
         }
 
         /*
          * If this is the BLOBS entry, what we stat'd was blobs.toc, which
          * most likely is a lot smaller than the actual blob data.  We don't
          * have a cheap way to estimate how much smaller, but fortunately it
          * doesn't matter too much as long as we get the blobs processed
          * reasonably early.  Arbitrarily scale up by a factor of 1K.
          */
         if (strcmp(te->desc, "BLOBS") == 0)
             te->dataLength *= 1024;
     }
 }

◆ _PrintExtraToc()

static void _PrintExtraToc	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 304 of file pg_backup_directory.c.

References ahprintf(), lclTocEntry::filename, _tocEntry::formatData, _archiveHandle::public, and Archive::verbose.

Referenced by InitArchiveFmt_Directory().

 {
     lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
     if (AH->public.verbose && tctx->filename)
         ahprintf(AH, "-- File: %s\n", tctx->filename);
 }

◆ _PrintFileData()

static void _PrintFileData	(	ArchiveHandle *	AH,
		char *	filename
	)

static

Definition at line 381 of file pg_backup_directory.c.

References ahwrite(), buf, cfclose(), cfopen_read(), cfread(), fatal, free, PG_BINARY_R, pg_malloc(), and ZLIB_OUT_SIZE.

Referenced by _LoadBlobs(), and _PrintTocData().

 {
     size_t      cnt;
     char       *buf;
     size_t      buflen;
     cfp        *cfp;
 
     if (!filename)
         return;
 
     cfp = cfopen_read(filename, PG_BINARY_R);
 
     if (!cfp)
         fatal("could not open input file \"%s\": %m", filename);
 
     buf = pg_malloc(ZLIB_OUT_SIZE);
     buflen = ZLIB_OUT_SIZE;
 
     while ((cnt = cfread(buf, buflen, cfp)))
     {
         ahwrite(buf, 1, cnt, AH);
     }
 
     free(buf);
     if (cfclose(cfp) != 0)
         fatal("could not close data file \"%s\": %m", filename);
 }

◆ _PrintTocData()

static void _PrintTocData	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 413 of file pg_backup_directory.c.

References _LoadBlobs(), _PrintFileData(), _tocEntry::desc, lclTocEntry::filename, _tocEntry::formatData, MAXPGPATH, and setFilePath().

Referenced by InitArchiveFmt_Directory().

 {
     lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
     if (!tctx->filename)
         return;
 
     if (strcmp(te->desc, "BLOBS") == 0)
         _LoadBlobs(AH);
     else
     {
         char        fname[MAXPGPATH];
 
         setFilePath(AH, fname, tctx->filename);
         _PrintFileData(AH, fname);
     }
 }

◆ _ReadBuf()

static void _ReadBuf	(	ArchiveHandle *	AH,
		void *	buf,
		size_t	len
	)

static

Definition at line 543 of file pg_backup_directory.c.

References cfread(), lclContext::dataFH, fatal, and _archiveHandle::formatData.

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     /*
      * If there was an I/O error, we already exited in cfread(), so here we
      * exit on short reads.
      */
     if (cfread(buf, len, ctx->dataFH) != len)
         fatal("could not read from input file: end of file");
 }

◆ _ReadByte()

static int _ReadByte ( ArchiveHandle * AH )

static

Definition at line 510 of file pg_backup_directory.c.

References cfgetc(), lclContext::dataFH, and _archiveHandle::formatData.

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     return cfgetc(ctx->dataFH);
 }

◆ _ReadExtraToc()

static void _ReadExtraToc	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 281 of file pg_backup_directory.c.

References lclTocEntry::filename, _tocEntry::formatData, free, pg_malloc0(), and ReadStr().

Referenced by InitArchiveFmt_Directory().

 {
     lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
     if (tctx == NULL)
     {
         tctx = (lclTocEntry *) pg_malloc0(sizeof(lclTocEntry));
         te->formatData = (void *) tctx;
     }
 
     tctx->filename = ReadStr(AH);
     if (strlen(tctx->filename) == 0)
     {
         free(tctx->filename);
         tctx->filename = NULL;
     }
 }

◆ _ReopenArchive()

static void _ReopenArchive ( ArchiveHandle * AH )

static

Definition at line 617 of file pg_backup_directory.c.

Referenced by InitArchiveFmt_Directory().

 {
     /*
      * Our TOC is in memory, our data files are opened by each child anyway as
      * they are separate. We support reopening the archive by just doing
      * nothing.
      */
 }

◆ _StartBlob()

static void _StartBlob	(	ArchiveHandle *	AH,
		TocEntry *	te,
		Oid	oid
	)

static

Definition at line 657 of file pg_backup_directory.c.

References cfopen_write(), _archiveHandle::compression, lclContext::dataFH, lclContext::directory, fatal, _archiveHandle::formatData, MAXPGPATH, PG_BINARY_W, and snprintf.

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
     char        fname[MAXPGPATH];
 
     snprintf(fname, MAXPGPATH, "%s/blob_%u.dat", ctx->directory, oid);
 
     ctx->dataFH = cfopen_write(fname, PG_BINARY_W, AH->compression);
 
     if (ctx->dataFH == NULL)
         fatal("could not open output file \"%s\": %m", fname);
 }

◆ _StartBlobs()

static void _StartBlobs	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 638 of file pg_backup_directory.c.

References lclContext::blobsTocFH, cfopen_write(), fatal, _archiveHandle::formatData, MAXPGPATH, and setFilePath().

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
     char        fname[MAXPGPATH];
 
     setFilePath(AH, fname, "blobs.toc");
 
     /* The blob TOC file is never compressed */
     ctx->blobsTocFH = cfopen_write(fname, "ab", 0);
     if (ctx->blobsTocFH == NULL)
         fatal("could not open output file \"%s\": %m", fname);
 }

◆ _StartData()

static void _StartData	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 322 of file pg_backup_directory.c.

References cfopen_write(), _archiveHandle::compression, fatal, lclTocEntry::filename, _archiveHandle::formatData, _tocEntry::formatData, MAXPGPATH, PG_BINARY_W, and setFilePath().

Referenced by InitArchiveFmt_Directory().

 {
     lclTocEntry *tctx = (lclTocEntry *) te->formatData;
     lclContext *ctx = (lclContext *) AH->formatData;
     char        fname[MAXPGPATH];
 
     setFilePath(AH, fname, tctx->filename);
 
     ctx->dataFH = cfopen_write(fname, PG_BINARY_W, AH->compression);
     if (ctx->dataFH == NULL)
         fatal("could not open output file \"%s\": %m", fname);
 }

◆ _WorkerJobDumpDirectory()

static int _WorkerJobDumpDirectory	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 828 of file pg_backup_directory.c.

References WriteDataChunksForTocEntry().

Referenced by InitArchiveFmt_Directory().

 {
     /*
      * This function returns void. We either fail and die horribly or
      * succeed... A failure will be detected by the parent when the child dies
      * unexpectedly.
      */
     WriteDataChunksForTocEntry(AH, te);
 
     return 0;
 }

◆ _WorkerJobRestoreDirectory()

static int _WorkerJobRestoreDirectory	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 845 of file pg_backup_directory.c.

References parallel_restore().

Referenced by InitArchiveFmt_Directory().

 {
     return parallel_restore(AH, te);
 }

◆ _WriteBuf()

static void _WriteBuf	(	ArchiveHandle *	AH,
		const void *	buf,
		size_t	len
	)

static

Definition at line 522 of file pg_backup_directory.c.

References cfwrite(), lclContext::dataFH, fatal, _archiveHandle::formatData, and get_cfp_error().

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     errno = 0;
     if (cfwrite(buf, len, ctx->dataFH) != len)
     {
         /* if write didn't set errno, assume problem is no disk space */
         if (errno == 0)
             errno = ENOSPC;
         fatal("could not write to output file: %s",
               get_cfp_error(ctx->dataFH));
     }
 }

◆ _WriteByte()

static int _WriteByte	(	ArchiveHandle *	AH,
		const int	i
	)

static

Definition at line 485 of file pg_backup_directory.c.

References cfwrite(), lclContext::dataFH, fatal, _archiveHandle::formatData, and get_cfp_error().

Referenced by InitArchiveFmt_Directory().

 {
     unsigned char c = (unsigned char) i;
     lclContext *ctx = (lclContext *) AH->formatData;
 
     errno = 0;
     if (cfwrite(&c, 1, ctx->dataFH) != 1)
     {
         /* if write didn't set errno, assume problem is no disk space */
         if (errno == 0)
             errno = ENOSPC;
         fatal("could not write to output file: %s",
               get_cfp_error(ctx->dataFH));
     }
 
     return 1;
 }

◆ _WriteData()

static void _WriteData	(	ArchiveHandle *	AH,
		const void *	data,
		size_t	dLen
	)

static

Definition at line 345 of file pg_backup_directory.c.

References cfwrite(), lclContext::dataFH, fatal, _archiveHandle::formatData, and get_cfp_error().

Referenced by InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
 
     errno = 0;
     if (dLen > 0 && cfwrite(data, dLen, ctx->dataFH) != dLen)
     {
         /* if write didn't set errno, assume problem is no disk space */
         if (errno == 0)
             errno = ENOSPC;
         fatal("could not write to output file: %s",
               get_cfp_error(ctx->dataFH));
     }
 }

◆ _WriteExtraToc()

static void _WriteExtraToc	(	ArchiveHandle *	AH,
		TocEntry *	te
	)

static

Definition at line 260 of file pg_backup_directory.c.

References lclTocEntry::filename, _tocEntry::formatData, and WriteStr().

Referenced by InitArchiveFmt_Directory().

 {
     lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
     /*
      * A dumpable object has set tctx->filename, any other object has not.
      * (see _ArchiveEntry).
      */
     if (tctx->filename)
         WriteStr(AH, tctx->filename);
     else
         WriteStr(AH, "");
 }

◆ InitArchiveFmt_Directory()

void InitArchiveFmt_Directory ( ArchiveHandle * AH )

Definition at line 108 of file pg_backup_directory.c.

References _ArchiveEntry(), _Clone(), _CloseArchive(), _DeClone(), _EndBlob(), _EndBlobs(), _EndData(), _PrepParallelRestore(), _PrintExtraToc(), _PrintTocData(), _ReadBuf(), _ReadByte(), _ReadExtraToc(), _ReopenArchive(), _StartBlob(), _StartBlobs(), _StartData(), _WorkerJobDumpDirectory(), _WorkerJobRestoreDirectory(), _WriteBuf(), _WriteByte(), _WriteData(), _WriteExtraToc(), archDirectory, _archiveHandle::ArchiveEntryPtr, archModeWrite, archTar, lclContext::blobsTocFH, cfclose(), cfopen_read(), _archiveHandle::ClonePtr, closedir(), _archiveHandle::ClosePtr, dirent::d_name, lclContext::dataFH, _archiveHandle::DeClonePtr, lclContext::directory, _archiveHandle::EndBlobPtr, _archiveHandle::EndBlobsPtr, _archiveHandle::EndDataPtr, fatal, _archiveHandle::format, _archiveHandle::formatData, _archiveHandle::fSpec, _archiveHandle::lo_buf, _archiveHandle::lo_buf_size, LOBBUFSIZE, MAXPGPATH, mkdir, _archiveHandle::mode, opendir(), PG_BINARY_R, pg_malloc(), pg_malloc0(), _archiveHandle::PrepParallelRestorePtr, _archiveHandle::PrintExtraTocPtr, _archiveHandle::PrintTocDataPtr, _archiveHandle::ReadBufPtr, _archiveHandle::ReadBytePtr, readdir(), _archiveHandle::ReadExtraTocPtr, ReadHead(), ReadToc(), _archiveHandle::ReopenPtr, S_ISDIR, setFilePath(), _archiveHandle::StartBlobPtr, _archiveHandle::StartBlobsPtr, _archiveHandle::StartDataPtr, stat, _archiveHandle::WorkerJobDumpPtr, _archiveHandle::WorkerJobRestorePtr, _archiveHandle::WriteBufPtr, _archiveHandle::WriteBytePtr, _archiveHandle::WriteDataPtr, and _archiveHandle::WriteExtraTocPtr.

Referenced by _allocAH().

 {
     lclContext *ctx;
 
     /* Assuming static functions, this can be copied for each format. */
     AH->ArchiveEntryPtr = _ArchiveEntry;
     AH->StartDataPtr = _StartData;
     AH->WriteDataPtr = _WriteData;
     AH->EndDataPtr = _EndData;
     AH->WriteBytePtr = _WriteByte;
     AH->ReadBytePtr = _ReadByte;
     AH->WriteBufPtr = _WriteBuf;
     AH->ReadBufPtr = _ReadBuf;
     AH->ClosePtr = _CloseArchive;
     AH->ReopenPtr = _ReopenArchive;
     AH->PrintTocDataPtr = _PrintTocData;
     AH->ReadExtraTocPtr = _ReadExtraToc;
     AH->WriteExtraTocPtr = _WriteExtraToc;
     AH->PrintExtraTocPtr = _PrintExtraToc;
 
     AH->StartBlobsPtr = _StartBlobs;
     AH->StartBlobPtr = _StartBlob;
     AH->EndBlobPtr = _EndBlob;
     AH->EndBlobsPtr = _EndBlobs;
 
     AH->PrepParallelRestorePtr = _PrepParallelRestore;
     AH->ClonePtr = _Clone;
     AH->DeClonePtr = _DeClone;
 
     AH->WorkerJobRestorePtr = _WorkerJobRestoreDirectory;
     AH->WorkerJobDumpPtr = _WorkerJobDumpDirectory;
 
     /* Set up our private context */
     ctx = (lclContext *) pg_malloc0(sizeof(lclContext));
     AH->formatData = (void *) ctx;
 
     ctx->dataFH = NULL;
     ctx->blobsTocFH = NULL;
 
     /* Initialize LO buffering */
     AH->lo_buf_size = LOBBUFSIZE;
     AH->lo_buf = (void *) pg_malloc(LOBBUFSIZE);
 
     /*
      * Now open the TOC file
      */
 
     if (!AH->fSpec || strcmp(AH->fSpec, "") == 0)
         fatal("no output directory specified");
 
     ctx->directory = AH->fSpec;
 
     if (AH->mode == archModeWrite)
     {
         struct stat st;
         bool        is_empty = false;
 
         /* we accept an empty existing directory */
         if (stat(ctx->directory, &st) == 0 && S_ISDIR(st.st_mode))
         {
             DIR        *dir = opendir(ctx->directory);
 
             if (dir)
             {
                 struct dirent *d;
 
                 is_empty = true;
                 while (errno = 0, (d = readdir(dir)))
                 {
                     if (strcmp(d->d_name, ".") != 0 && strcmp(d->d_name, "..") != 0)
                     {
                         is_empty = false;
                         break;
                     }
                 }
 
                 if (errno)
                     fatal("could not read directory \"%s\": %m",
                           ctx->directory);
 
                 if (closedir(dir))
                     fatal("could not close directory \"%s\": %m",
                           ctx->directory);
             }
         }
 
         if (!is_empty && mkdir(ctx->directory, 0700) < 0)
             fatal("could not create directory \"%s\": %m",
                   ctx->directory);
     }
     else
     {                           /* Read Mode */
         char        fname[MAXPGPATH];
         cfp        *tocFH;
 
         setFilePath(AH, fname, "toc.dat");
 
         tocFH = cfopen_read(fname, PG_BINARY_R);
         if (tocFH == NULL)
             fatal("could not open input file \"%s\": %m", fname);
 
         ctx->dataFH = tocFH;
 
         /*
          * The TOC of a directory format dump shares the format code of the
          * tar format.
          */
         AH->format = archTar;
         ReadHead(AH);
         AH->format = archDirectory;
         ReadToc(AH);
 
         /* Nothing else in the file, so close it again... */
         if (cfclose(tocFH) != 0)
             fatal("could not close TOC file: %m");
         ctx->dataFH = NULL;
     }
 }

◆ setFilePath()

static void setFilePath	(	ArchiveHandle *	AH,
		char *	buf,
		const char *	relativeFilename
	)

static

Definition at line 713 of file pg_backup_directory.c.

References lclContext::directory, fatal, _archiveHandle::formatData, and MAXPGPATH.

Referenced by _CloseArchive(), _LoadBlobs(), _PrepParallelRestore(), _PrintTocData(), _StartBlobs(), _StartData(), and InitArchiveFmt_Directory().

 {
     lclContext *ctx = (lclContext *) AH->formatData;
     char       *dname;
 
     dname = ctx->directory;
 
     if (strlen(dname) + 1 + strlen(relativeFilename) + 1 > MAXPGPATH)
         fatal("file name too long: \"%s\"", dname);
 
     strcpy(buf, dname);
     strcat(buf, "/");
     strcat(buf, relativeFilename);
 }

Data Structures

Functions

Function Documentation

◆ _ArchiveEntry()

◆ _Clone()

◆ _CloseArchive()

◆ _DeClone()

◆ _EndBlob()

◆ _EndBlobs()

◆ _EndData()

◆ _LoadBlobs()

◆ _PrepParallelRestore()

◆ _PrintExtraToc()

◆ _PrintFileData()

◆ _PrintTocData()

◆ _ReadBuf()

◆ _ReadByte()

◆ _ReadExtraToc()

◆ _ReopenArchive()

◆ _StartBlob()

◆ _StartBlobs()

◆ _StartData()

◆ _WorkerJobDumpDirectory()

◆ _WorkerJobRestoreDirectory()

◆ _WriteBuf()

◆ _WriteByte()

◆ _WriteData()

◆ _WriteExtraToc()

◆ InitArchiveFmt_Directory()

◆ setFilePath()