pg_backup_tar.c File Reference

#include "postgres_fe.h"
#include <sys/stat.h>
#include <ctype.h>
#include <limits.h>
#include <unistd.h>
#include "common/file_utils.h"
#include "fe_utils/string_utils.h"
#include "pg_backup_archiver.h"
#include "pg_backup_tar.h"
#include "pg_backup_utils.h"
#include "pgtar.h"

Include dependency graph for pg_backup_tar.c:

Go to the source code of this file.

Data Structures
struct	TAR_MEMBER
struct	lclContext
struct	lclTocEntry

Macros
#define	K_STD_BUF_SIZE   1024

Functions
static void	_ArchiveEntry (ArchiveHandle *AH, TocEntry *te)
static void	_StartData (ArchiveHandle *AH, TocEntry *te)
static void	_WriteData (ArchiveHandle *AH, const void *data, size_t dLen)
static void	_EndData (ArchiveHandle *AH, TocEntry *te)
static int	_WriteByte (ArchiveHandle *AH, const int i)
static int	_ReadByte (ArchiveHandle *AH)
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	_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	_StartLOs (ArchiveHandle *AH, TocEntry *te)
static void	_StartLO (ArchiveHandle *AH, TocEntry *te, Oid oid)
static void	_EndLO (ArchiveHandle *AH, TocEntry *te, Oid oid)
static void	_EndLOs (ArchiveHandle *AH, TocEntry *te)
static void	_LoadLOs (ArchiveHandle *AH, TocEntry *te)
static TAR_MEMBER *	tarOpen (ArchiveHandle *AH, const char *filename, char mode)
static void	tarClose (ArchiveHandle *AH, TAR_MEMBER *th)
static int	tarPrintf (TAR_MEMBER *th, const char *fmt,...) pg_attribute_printf(2
static int static void	_tarAddFile (ArchiveHandle *AH, TAR_MEMBER *th)
static TAR_MEMBER *	_tarPositionTo (ArchiveHandle *AH, const char *filename)
static size_t	tarRead (void *buf, size_t len, TAR_MEMBER *th)
static size_t	tarWrite (const void *buf, size_t len, TAR_MEMBER *th)
static void	_tarWriteHeader (TAR_MEMBER *th)
static int	_tarGetHeader (ArchiveHandle *AH, TAR_MEMBER *th)
static size_t	_tarReadRaw (ArchiveHandle *AH, void *buf, size_t len, TAR_MEMBER *th, FILE *fh)
static size_t	_scriptOut (ArchiveHandle *AH, const void *buf, size_t len)
void	InitArchiveFmt_Tar (ArchiveHandle *AH)
static void	_PrintFileData (ArchiveHandle *AH, char *filename)
bool	isValidTarHeader (char *header)

Macro Definition Documentation

K_STD_BUF_SIZE

#define K_STD_BUF_SIZE   1024

Definition at line 63 of file pg_backup_tar.c.

Function Documentation

_ArchiveEntry()

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

static

Definition at line 234 of file pg_backup_tar.c.

{
    lclTocEntry *ctx;
    char        fn[K_STD_BUF_SIZE];
 
    ctx = pg_malloc0_object(lclTocEntry);
    if (te->dataDumper != NULL)
    {
        snprintf(fn, sizeof(fn), "%d.dat", te->dumpId);
        ctx->filename = pg_strdup(fn);
    }
    else
    {
        ctx->filename = NULL;
        ctx->TH = NULL;
    }
    te->formatData = (void *) ctx;
}

References _tocEntry::dataDumper, _tocEntry::dumpId, lclTocEntry::filename, fn(), _tocEntry::formatData, K_STD_BUF_SIZE, pg_malloc0_object, pg_strdup(), snprintf, and lclTocEntry::TH.

Referenced by InitArchiveFmt_Tar().

_CloseArchive()

static void _CloseArchive ( ArchiveHandle *  AH )

static

Definition at line 769 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    TAR_MEMBER *th;
    RestoreOptions *ropt;
    RestoreOptions *savRopt;
    DumpOptions *savDopt;
    int         savVerbose,
                i;
 
    if (AH->mode == archModeWrite)
    {
        /*
         * Write the Header & TOC to the archive FIRST
         */
        th = tarOpen(AH, "toc.dat", 'w');
        ctx->FH = th;
        WriteHead(AH);
        WriteToc(AH);
        tarClose(AH, th);       /* Not needed any more */
 
        /*
         * Now send the data (tables & LOs)
         */
        WriteDataChunks(AH, NULL);
 
        /*
         * Now this format wants to append a script which does a full restore
         * if the files have been extracted.
         */
        th = tarOpen(AH, "restore.sql", 'w');
 
        tarPrintf(th, "--\n"
                  "-- NOTE:\n"
                  "--\n"
                  "-- File paths need to be edited. Search for $$PATH$$ and\n"
                  "-- replace it with the path to the directory containing\n"
                  "-- the extracted data files.\n"
                  "--\n");
 
        AH->CustomOutPtr = _scriptOut;
 
        ctx->isSpecialScript = 1;
        ctx->scriptTH = th;
 
        ropt = NewRestoreOptions();
        memcpy(ropt, AH->public.ropt, sizeof(RestoreOptions));
        ropt->filename = NULL;
        ropt->dropSchema = 1;
        ropt->superuser = NULL;
        ropt->suppressDumpWarnings = true;
 
        savDopt = AH->public.dopt;
        savRopt = AH->public.ropt;
 
        SetArchiveOptions((Archive *) AH, NULL, ropt);
 
        savVerbose = AH->public.verbose;
        AH->public.verbose = 0;
 
        RestoreArchive((Archive *) AH);
 
        SetArchiveOptions((Archive *) AH, savDopt, savRopt);
 
        AH->public.verbose = savVerbose;
 
        tarClose(AH, th);
 
        ctx->isSpecialScript = 0;
 
        /*
         * EOF marker for tar files is two blocks of NULLs.
         */
        for (i = 0; i < TAR_BLOCK_SIZE * 2; i++)
        {
            if (fputc(0, ctx->tarFH) == EOF)
                WRITE_ERROR_EXIT;
        }
 
        /* Sync the output file if one is defined */
        if (AH->dosync && AH->fSpec)
            (void) fsync_fname(AH->fSpec, false);
    }
 
    AH->FH = NULL;
}

References _scriptOut(), archModeWrite, _archiveHandle::CustomOutPtr, Archive::dopt, _archiveHandle::FH, lclContext::FH, _archiveHandle::formatData, fsync_fname(), i, if(), lclContext::isSpecialScript, _archiveHandle::mode, NewRestoreOptions(), _archiveHandle::public, RestoreArchive(), Archive::ropt, lclContext::scriptTH, SetArchiveOptions(), TAR_BLOCK_SIZE, tarClose(), lclContext::tarFH, tarOpen(), tarPrintf(), Archive::verbose, WRITE_ERROR_EXIT, WriteDataChunks(), WriteHead(), and WriteToc().

Referenced by InitArchiveFmt_Tar().

_EndData()

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

static

Definition at line 549 of file pg_backup_tar.c.

{
    lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
    /* Close the file */
    tarClose(AH, tctx->TH);
    tctx->TH = NULL;
}

References _tocEntry::formatData, tarClose(), and lclTocEntry::TH.

Referenced by InitArchiveFmt_Tar().

_EndLO()

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

static

Definition at line 922 of file pg_backup_tar.c.

{
    lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
    tarClose(AH, tctx->TH);
}

References _tocEntry::formatData, tarClose(), and lclTocEntry::TH.

Referenced by InitArchiveFmt_Tar().

_EndLOs()

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

static

Definition at line 936 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
 
    /* Write out a fake zero OID to mark end-of-LOs. */
    /* WriteInt(AH, 0); */
 
    tarClose(AH, ctx->loToc);
}

References _archiveHandle::formatData, lclContext::loToc, and tarClose().

Referenced by InitArchiveFmt_Tar().

_LoadLOs()

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

static

Definition at line 643 of file pg_backup_tar.c.

{
    Oid         oid;
    lclContext *ctx = (lclContext *) AH->formatData;
    TAR_MEMBER *th;
    size_t      cnt;
    bool        foundLO = false;
    char        buf[4096];
 
    StartRestoreLOs(AH);
 
    /*
     * The blobs_NNN.toc or blobs.toc file is fairly useless to us because it
     * will appear only after the associated blob_NNN.dat files.  For archive
     * versions >= 16 we can look at the BLOBS entry's te->tag to discover the
     * OID of the first blob we want to restore, and then search forward to
     * find the appropriate blob_<oid>.dat file.  For older versions we rely
     * on the knowledge that there was only one BLOBS entry and just search
     * for the first blob_<oid>.dat file.  Once we find the first blob file to
     * restore, restore all blobs until we reach the blobs[_NNN].toc file.
     */
    if (AH->version >= K_VERS_1_16)
    {
        /* We rely on atooid to not complain about nnnn..nnnn tags */
        oid = atooid(te->tag);
        snprintf(buf, sizeof(buf), "blob_%u.dat", oid);
        th = tarOpen(AH, buf, 'r'); /* Advance to first desired file */
    }
    else
        th = tarOpen(AH, NULL, 'r');    /* Open next file */
 
    while (th != NULL)
    {
        ctx->FH = th;
 
        if (strncmp(th->targetFile, "blob_", 5) == 0)
        {
            oid = atooid(&th->targetFile[5]);
            if (oid != 0)
            {
                pg_log_info("restoring large object with OID %u", oid);
 
                StartRestoreLO(AH, oid, AH->public.ropt->dropSchema);
 
                while ((cnt = tarRead(buf, 4095, th)) > 0)
                {
                    buf[cnt] = '\0';
                    ahwrite(buf, 1, cnt, AH);
                }
                EndRestoreLO(AH, oid);
                foundLO = true;
            }
            tarClose(AH, th);
        }
        else
        {
            tarClose(AH, th);
 
            /*
             * Once we have found the first LO, stop at the first non-LO entry
             * (which will be 'blobs[_NNN].toc').  This coding would eat all
             * the rest of the archive if there are no LOs ... but this
             * function shouldn't be called at all in that case.
             */
            if (foundLO)
                break;
        }
 
        th = tarOpen(AH, NULL, 'r');
    }
    EndRestoreLOs(AH);
}

References ahwrite(), atooid, buf, _restoreOptions::dropSchema, EndRestoreLO(), EndRestoreLOs(), lclContext::FH, _archiveHandle::formatData, K_VERS_1_16, pg_log_info, _archiveHandle::public, Archive::ropt, snprintf, StartRestoreLO(), StartRestoreLOs(), _tocEntry::tag, tarClose(), TAR_MEMBER::targetFile, tarOpen(), tarRead(), and _archiveHandle::version.

Referenced by _PrintTocData().

_PrintExtraToc()

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

static

Definition at line 285 of file pg_backup_tar.c.

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

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

Referenced by InitArchiveFmt_Tar().

_PrintFileData()

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

static

Definition at line 562 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    char        buf[4096];
    size_t      cnt;
    TAR_MEMBER *th;
 
    if (!filename)
        return;
 
    th = tarOpen(AH, filename, 'r');
    ctx->FH = th;
 
    while ((cnt = tarRead(buf, 4095, th)) > 0)
    {
        buf[cnt] = '\0';
        ahwrite(buf, 1, cnt, AH);
    }
 
    tarClose(AH, th);
}

References ahwrite(), buf, lclContext::FH, filename, _archiveHandle::formatData, if(), tarClose(), tarOpen(), and tarRead().

Referenced by _PrintTocData().

_PrintTocData()

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

static

Definition at line 589 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    lclTocEntry *tctx = (lclTocEntry *) te->formatData;
    int         pos1;
 
    if (!tctx->filename)
        return;
 
    /*
     * If we're writing the special restore.sql script, emit a suitable
     * command to include each table's data from the corresponding file.
     *
     * In the COPY case this is a bit klugy because the regular COPY command
     * was already printed before we get control.
     */
    if (ctx->isSpecialScript)
    {
        if (te->copyStmt)
        {
            /* Abort the COPY FROM stdin */
            ahprintf(AH, "\\.\n");
 
            /*
             * The COPY statement should look like "COPY ... FROM stdin;\n",
             * see dumpTableData().
             */
            pos1 = (int) strlen(te->copyStmt) - 13;
            if (pos1 < 6 || strncmp(te->copyStmt, "COPY ", 5) != 0 ||
                strcmp(te->copyStmt + pos1, " FROM stdin;\n") != 0)
                pg_fatal("unexpected COPY statement syntax: \"%s\"",
                         te->copyStmt);
 
            /* Emit all but the FROM part ... */
            ahwrite(te->copyStmt, 1, pos1, AH);
            /* ... and insert modified FROM */
            ahprintf(AH, " FROM '$$PATH$$/%s';\n\n", tctx->filename);
        }
        else
        {
            /* --inserts mode, no worries, just include the data file */
            ahprintf(AH, "\\i $$PATH$$/%s\n\n", tctx->filename);
        }
 
        return;
    }
 
    if (strcmp(te->desc, "BLOBS") == 0)
        _LoadLOs(AH, te);
    else
        _PrintFileData(AH, tctx->filename);
}

References _LoadLOs(), _PrintFileData(), ahprintf(), ahwrite(), _tocEntry::copyStmt, _tocEntry::desc, _archiveHandle::formatData, _tocEntry::formatData, lclContext::isSpecialScript, and pg_fatal.

Referenced by InitArchiveFmt_Tar().

_ReadBuf()

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

static

Definition at line 757 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
 
    if (tarRead(buf, len, ctx->FH) != len)
        /* We already would have exited for errors on reads, must be EOF */
        pg_fatal("could not read from input file: end of file");
 
    ctx->filePos += len;
}

References buf, lclContext::FH, lclContext::filePos, _archiveHandle::formatData, if(), len, pg_fatal, and tarRead().

Referenced by InitArchiveFmt_Tar().

_ReadByte()

static int _ReadByte ( ArchiveHandle *  AH )

static

Definition at line 731 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    size_t      res;
    unsigned char c;
 
    res = tarRead(&c, 1, ctx->FH);
    if (res != 1)
        /* We already would have exited for errors on reads, must be EOF */
        pg_fatal("could not read from input file: end of file");
    ctx->filePos += 1;
    return c;
}

References lclContext::FH, lclContext::filePos, _archiveHandle::formatData, pg_fatal, res, and tarRead().

Referenced by InitArchiveFmt_Tar().

_ReadExtraToc()

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

static

Definition at line 265 of file pg_backup_tar.c.

{
    lclTocEntry *ctx = (lclTocEntry *) te->formatData;
 
    if (ctx == NULL)
    {
        ctx = pg_malloc0_object(lclTocEntry);
        te->formatData = (void *) ctx;
    }
 
    ctx->filename = ReadStr(AH);
    if (strlen(ctx->filename) == 0)
    {
        free(ctx->filename);
        ctx->filename = NULL;
    }
    ctx->TH = NULL;
}

References lclTocEntry::filename, _tocEntry::formatData, free, if(), pg_malloc0_object, ReadStr(), and lclTocEntry::TH.

Referenced by InitArchiveFmt_Tar().

_scriptOut()

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

static

Definition at line 857 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
 
    return tarWrite(buf, len, ctx->scriptTH);
}

References buf, _archiveHandle::formatData, len, lclContext::scriptTH, and tarWrite().

Referenced by _CloseArchive().

_StartData()

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

static

Definition at line 294 of file pg_backup_tar.c.

{
    lclTocEntry *tctx = (lclTocEntry *) te->formatData;
 
    tctx->TH = tarOpen(AH, tctx->filename, 'w');
}

References lclTocEntry::filename, _tocEntry::formatData, tarOpen(), and lclTocEntry::TH.

Referenced by InitArchiveFmt_Tar().

_StartLO()

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

static

Definition at line 896 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    lclTocEntry *tctx = (lclTocEntry *) te->formatData;
    char        fname[255];
 
    if (oid == 0)
        pg_fatal("invalid OID for large object (%u)", oid);
 
    if (AH->compression_spec.algorithm != PG_COMPRESSION_NONE)
        pg_fatal("compression is not supported by tar archive format");
 
    sprintf(fname, "blob_%u.dat", oid);
 
    tarPrintf(ctx->loToc, "%u %s\n", oid, fname);
 
    tctx->TH = tarOpen(AH, fname, 'w');
}

References pg_compress_specification::algorithm, _archiveHandle::compression_spec, _archiveHandle::formatData, _tocEntry::formatData, lclContext::loToc, PG_COMPRESSION_NONE, pg_fatal, sprintf, tarOpen(), and tarPrintf().

Referenced by InitArchiveFmt_Tar().

_StartLOs()

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

static

Definition at line 879 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    char        fname[K_STD_BUF_SIZE];
 
    sprintf(fname, "blobs_%d.toc", te->dumpId);
    ctx->loToc = tarOpen(AH, fname, 'w');
}

References _tocEntry::dumpId, _archiveHandle::formatData, K_STD_BUF_SIZE, lclContext::loToc, sprintf, and tarOpen().

Referenced by InitArchiveFmt_Tar().

_tarAddFile()

static void _tarAddFile ( ArchiveHandle *  AH, TAR_MEMBER *  th  )

static

Definition at line 1014 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    FILE       *tmp = th->tmpFH;    /* Grab it for convenience */
    char        buf[32768];
    size_t      cnt;
    pgoff_t     len = 0;
    size_t      res;
    size_t      i,
                pad;
 
    /*
     * Find file len & go back to start.
     */
    if (fseeko(tmp, 0, SEEK_END) != 0)
        pg_fatal("error during file seek: %m");
    th->fileLen = ftello(tmp);
    if (th->fileLen < 0)
        pg_fatal("could not determine seek position in archive file: %m");
    if (fseeko(tmp, 0, SEEK_SET) != 0)
        pg_fatal("error during file seek: %m");
 
    _tarWriteHeader(th);
 
    while ((cnt = fread(buf, 1, sizeof(buf), tmp)) > 0)
    {
        if ((res = fwrite(buf, 1, cnt, th->tarFH)) != cnt)
            WRITE_ERROR_EXIT;
        len += res;
    }
    if (!feof(tmp))
        READ_ERROR_EXIT(tmp);
 
    if (fclose(tmp) != 0)       /* This *should* delete it... */
        pg_fatal("could not close temporary file: %m");
 
    if (len != th->fileLen)
        pg_fatal("actual file length (%lld) does not match expected (%lld)",
                 (long long) len, (long long) th->fileLen);
 
    pad = tarPaddingBytesRequired(len);
    for (i = 0; i < pad; i++)
    {
        if (fputc('\0', th->tarFH) == EOF)
            WRITE_ERROR_EXIT;
    }
 
    ctx->tarFHpos += len + pad;
}

References _tarWriteHeader(), buf, TAR_MEMBER::fileLen, _archiveHandle::formatData, fseeko, ftello, i, if(), len, pg_fatal, pgoff_t, READ_ERROR_EXIT, res, TAR_MEMBER::tarFH, lclContext::tarFHpos, tarPaddingBytesRequired(), TAR_MEMBER::tmpFH, and WRITE_ERROR_EXIT.

Referenced by tarClose().

_tarGetHeader()

static int _tarGetHeader ( ArchiveHandle *  AH, TAR_MEMBER *  th  )

static

Definition at line 1140 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    char        h[TAR_BLOCK_SIZE];
    char        tag[100 + 1];
    int         sum,
                chk;
    pgoff_t     len;
    pgoff_t     hPos;
    bool        gotBlock = false;
 
    while (!gotBlock)
    {
        /* Save the pos for reporting purposes */
        hPos = ctx->tarFHpos;
 
        /* Read the next tar block, return EOF, exit if short */
        len = _tarReadRaw(AH, h, TAR_BLOCK_SIZE, NULL, ctx->tarFH);
        if (len == 0)           /* EOF */
            return 0;
 
        if (len != TAR_BLOCK_SIZE)
            pg_fatal(ngettext("incomplete tar header found (%lu byte)",
                              "incomplete tar header found (%lu bytes)",
                              len),
                     (unsigned long) len);
 
        /* Calc checksum */
        chk = tarChecksum(h);
        sum = read_tar_number(&h[TAR_OFFSET_CHECKSUM], 8);
 
        /*
         * If the checksum failed, see if it is a null block. If so, silently
         * continue to the next block.
         */
        if (chk == sum)
            gotBlock = true;
        else
        {
            int         i;
 
            for (i = 0; i < TAR_BLOCK_SIZE; i++)
            {
                if (h[i] != 0)
                {
                    gotBlock = true;
                    break;
                }
            }
        }
    }
 
    /* Name field is 100 bytes, might not be null-terminated */
    strlcpy(tag, &h[TAR_OFFSET_NAME], 100 + 1);
 
    len = read_tar_number(&h[TAR_OFFSET_SIZE], 12);
 
    pg_log_debug("TOC Entry %s at %llu (length %llu, checksum %d)",
                 tag, (unsigned long long) hPos, (unsigned long long) len, sum);
 
    if (chk != sum)
        pg_fatal("corrupt tar header found in %s (expected %d, computed %d) file position %llu",
                 tag, sum, chk, (unsigned long long) ftello(ctx->tarFH));
 
    th->targetFile = pg_strdup(tag);
    th->fileLen = len;
 
    return 1;
}

References _tarReadRaw(), TAR_MEMBER::fileLen, _archiveHandle::formatData, ftello, i, len, ngettext, pg_fatal, pg_log_debug, pg_strdup(), pgoff_t, read_tar_number(), strlcpy(), TAR_BLOCK_SIZE, TAR_OFFSET_CHECKSUM, TAR_OFFSET_NAME, TAR_OFFSET_SIZE, tarChecksum(), lclContext::tarFH, lclContext::tarFHpos, TAR_MEMBER::targetFile, and while().

Referenced by _tarPositionTo().

_tarPositionTo()

static TAR_MEMBER * _tarPositionTo ( ArchiveHandle *  AH, const char *  filename  )

static

Definition at line 1066 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    TAR_MEMBER *th = pg_malloc0_object(TAR_MEMBER);
    char        c;
    char        header[TAR_BLOCK_SIZE];
    size_t      i,
                len,
                blks;
    int         id;
 
    th->AH = AH;
 
    /* Go to end of current file, if any */
    if (ctx->tarFHpos != 0)
    {
        pg_log_debug("moving from position %lld to next member at file position %lld",
                     (long long) ctx->tarFHpos, (long long) ctx->tarNextMember);
 
        while (ctx->tarFHpos < ctx->tarNextMember)
            _tarReadRaw(AH, &c, 1, NULL, ctx->tarFH);
    }
 
    pg_log_debug("now at file position %lld", (long long) ctx->tarFHpos);
 
    /* We are at the start of the file, or at the next member */
 
    /* Get the header */
    if (!_tarGetHeader(AH, th))
    {
        if (filename)
            pg_fatal("could not find header for file \"%s\" in tar archive", filename);
        else
        {
            /*
             * We're just scanning the archive for the next file, so return
             * null
             */
            free(th);
            return NULL;
        }
    }
 
    while (filename != NULL && strcmp(th->targetFile, filename) != 0)
    {
        pg_log_debug("skipping tar member %s", th->targetFile);
 
        id = atoi(th->targetFile);
        if ((TocIDRequired(AH, id) & REQ_DATA) != 0)
            pg_fatal("restoring data out of order is not supported in this archive format: "
                     "\"%s\" is required, but comes before \"%s\" in the archive file.",
                     th->targetFile, filename);
 
        /* Header doesn't match, so read to next header */
        len = th->fileLen;
        len += tarPaddingBytesRequired(th->fileLen);
        blks = len / TAR_BLOCK_SIZE;    /* # of tar blocks */
 
        for (i = 0; i < blks; i++)
            _tarReadRaw(AH, &header[0], TAR_BLOCK_SIZE, NULL, ctx->tarFH);
 
        if (!_tarGetHeader(AH, th))
            pg_fatal("could not find header for file \"%s\" in tar archive", filename);
    }
 
    ctx->tarNextMember = ctx->tarFHpos + th->fileLen
        + tarPaddingBytesRequired(th->fileLen);
    th->pos = 0;
 
    return th;
}

References _tarGetHeader(), _tarReadRaw(), TAR_MEMBER::AH, TAR_MEMBER::fileLen, filename, _archiveHandle::formatData, free, i, len, pg_fatal, pg_log_debug, pg_malloc0_object, TAR_MEMBER::pos, REQ_DATA, TAR_BLOCK_SIZE, lclContext::tarFH, lclContext::tarFHpos, TAR_MEMBER::targetFile, lclContext::tarNextMember, tarPaddingBytesRequired(), and TocIDRequired().

Referenced by tarOpen().

_tarReadRaw()

static size_t _tarReadRaw ( ArchiveHandle *  AH, void *  buf, size_t  len, TAR_MEMBER *  th, FILE *  fh  )

static

Definition at line 462 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    size_t      avail;
    size_t      used = 0;
    size_t      res = 0;
 
    Assert(th || fh);
 
    avail = AH->lookaheadLen - AH->lookaheadPos;
    if (avail > 0)
    {
        /* We have some lookahead bytes to use */
        if (avail >= len)       /* Just use the lookahead buffer */
            used = len;
        else
            used = avail;
 
        /* Copy, and adjust buffer pos */
        memcpy(buf, AH->lookahead + AH->lookaheadPos, used);
        AH->lookaheadPos += used;
 
        /* Adjust required length */
        len -= used;
    }
 
    /* Read the file if len > 0 */
    if (len > 0)
    {
        if (fh)
        {
            res = fread(&((char *) buf)[used], 1, len, fh);
            if (res != len && !feof(fh))
                READ_ERROR_EXIT(fh);
        }
        else if (th)
        {
            res = fread(&((char *) buf)[used], 1, len, th->nFH);
            if (res != len && !feof(th->nFH))
                READ_ERROR_EXIT(th->nFH);
        }
    }
 
    ctx->tarFHpos += res + used;
 
    return (res + used);
}

References Assert, buf, _archiveHandle::formatData, len, _archiveHandle::lookahead, _archiveHandle::lookaheadLen, _archiveHandle::lookaheadPos, TAR_MEMBER::nFH, READ_ERROR_EXIT, res, and lclContext::tarFHpos.

Referenced by _tarGetHeader(), _tarPositionTo(), and tarRead().

_tarWriteHeader()

static void _tarWriteHeader ( TAR_MEMBER *  th )

static

Definition at line 1212 of file pg_backup_tar.c.

{
    char        h[TAR_BLOCK_SIZE];
 
    tarCreateHeader(h, th->targetFile, NULL, th->fileLen,
                    0600, 04000, 02000, time(NULL));
 
    /* Now write the completed header. */
    if (fwrite(h, 1, TAR_BLOCK_SIZE, th->tarFH) != TAR_BLOCK_SIZE)
        WRITE_ERROR_EXIT;
}

References TAR_MEMBER::fileLen, TAR_BLOCK_SIZE, tarCreateHeader(), TAR_MEMBER::tarFH, TAR_MEMBER::targetFile, and WRITE_ERROR_EXIT.

Referenced by _tarAddFile().

_WriteBuf()

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

static

Definition at line 746 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
 
    if (tarWrite(buf, len, ctx->FH) != len)
        WRITE_ERROR_EXIT;
 
    ctx->filePos += len;
}

References buf, lclContext::FH, lclContext::filePos, _archiveHandle::formatData, if(), len, tarWrite(), and WRITE_ERROR_EXIT.

Referenced by InitArchiveFmt_Tar().

_WriteByte()

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

static

Definition at line 718 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    char        b = i;          /* Avoid endian problems */
 
    if (tarWrite(&b, 1, ctx->FH) != 1)
        WRITE_ERROR_EXIT;
 
    ctx->filePos += 1;
    return 1;
}

References b, lclContext::FH, lclContext::filePos, _archiveHandle::formatData, i, if(), tarWrite(), and WRITE_ERROR_EXIT.

Referenced by InitArchiveFmt_Tar().

_WriteData()

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

static

Definition at line 540 of file pg_backup_tar.c.

{
    lclTocEntry *tctx = (lclTocEntry *) AH->currToc->formatData;
 
    if (tarWrite(data, dLen, tctx->TH) != dLen)
        WRITE_ERROR_EXIT;
}

References _archiveHandle::currToc, data, _tocEntry::formatData, if(), tarWrite(), lclTocEntry::TH, and WRITE_ERROR_EXIT.

Referenced by InitArchiveFmt_Tar().

_WriteExtraToc()

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

static

Definition at line 254 of file pg_backup_tar.c.

{
    lclTocEntry *ctx = (lclTocEntry *) te->formatData;
 
    if (ctx->filename)
        WriteStr(AH, ctx->filename);
    else
        WriteStr(AH, "");
}

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

Referenced by InitArchiveFmt_Tar().

InitArchiveFmt_Tar()

void InitArchiveFmt_Tar

(

ArchiveHandle *

AH

)

Definition at line 121 of file pg_backup_tar.c.

{
    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 = NULL;
    AH->PrintTocDataPtr = _PrintTocData;
    AH->ReadExtraTocPtr = _ReadExtraToc;
    AH->WriteExtraTocPtr = _WriteExtraToc;
    AH->PrintExtraTocPtr = _PrintExtraToc;
 
    AH->StartLOsPtr = _StartLOs;
    AH->StartLOPtr = _StartLO;
    AH->EndLOPtr = _EndLO;
    AH->EndLOsPtr = _EndLOs;
    AH->ClonePtr = NULL;
    AH->DeClonePtr = NULL;
 
    AH->WorkerJobDumpPtr = NULL;
    AH->WorkerJobRestorePtr = NULL;
 
    /*
     * Set up some special context used in compressing data.
     */
    ctx = pg_malloc0_object(lclContext);
    AH->formatData = (void *) ctx;
    ctx->filePos = 0;
    ctx->isSpecialScript = 0;
 
    /*
     * Now open the tar file, and load the TOC if we're in read mode.
     */
    if (AH->mode == archModeWrite)
    {
        if (AH->fSpec && strcmp(AH->fSpec, "") != 0)
        {
            ctx->tarFH = fopen(AH->fSpec, PG_BINARY_W);
            if (ctx->tarFH == NULL)
                pg_fatal("could not open TOC file \"%s\" for output: %m",
                         AH->fSpec);
        }
        else
        {
            ctx->tarFH = stdout;
            if (ctx->tarFH == NULL)
                pg_fatal("could not open TOC file for output: %m");
        }
 
        ctx->tarFHpos = 0;
 
        /*
         * Make unbuffered since we will dup() it, and the buffers screw each
         * other
         */
        /* setvbuf(ctx->tarFH, NULL, _IONBF, 0); */
 
        ctx->hasSeek = checkSeek(ctx->tarFH);
 
        /*
         * We don't support compression because reading the files back is not
         * possible since gzdopen uses buffered IO which totally screws file
         * positioning.
         */
        if (AH->compression_spec.algorithm != PG_COMPRESSION_NONE)
            pg_fatal("compression is not supported by tar archive format");
    }
    else
    {                           /* Read Mode */
        if (AH->fSpec && strcmp(AH->fSpec, "") != 0)
        {
            ctx->tarFH = fopen(AH->fSpec, PG_BINARY_R);
            if (ctx->tarFH == NULL)
                pg_fatal("could not open TOC file \"%s\" for input: %m",
                         AH->fSpec);
        }
        else
        {
            ctx->tarFH = stdin;
            if (ctx->tarFH == NULL)
                pg_fatal("could not open TOC file for input: %m");
        }
 
        /*
         * Make unbuffered since we will dup() it, and the buffers screw each
         * other
         */
        /* setvbuf(ctx->tarFH, NULL, _IONBF, 0); */
 
        ctx->tarFHpos = 0;
 
        ctx->hasSeek = checkSeek(ctx->tarFH);
 
        ctx->FH = (void *) tarOpen(AH, "toc.dat", 'r');
        ReadHead(AH);
        ReadToc(AH);
        tarClose(AH, ctx->FH);  /* Nothing else in the file... */
    }
}

References _ArchiveEntry(), _CloseArchive(), _EndData(), _EndLO(), _EndLOs(), _PrintExtraToc(), _PrintTocData(), _ReadBuf(), _ReadByte(), _ReadExtraToc(), _StartData(), _StartLO(), _StartLOs(), _WriteBuf(), _WriteByte(), _WriteData(), _WriteExtraToc(), archModeWrite, checkSeek(), lclContext::FH, lclContext::filePos, lclContext::hasSeek, lclContext::isSpecialScript, PG_BINARY_R, PG_BINARY_W, PG_COMPRESSION_NONE, pg_fatal, pg_malloc0_object, ReadHead(), ReadToc(), generate_unaccent_rules::stdout, tarClose(), lclContext::tarFH, lclContext::tarFHpos, and tarOpen().

Referenced by _allocAH().

isValidTarHeader()

bool isValidTarHeader

(

char *

header

)

Definition at line 988 of file pg_backup_tar.c.

{
    int         sum;
    int         chk = tarChecksum(header);
 
    sum = read_tar_number(&header[TAR_OFFSET_CHECKSUM], 8);
 
    if (sum != chk)
        return false;
 
    /* POSIX tar format */
    if (memcmp(&header[TAR_OFFSET_MAGIC], "ustar\0", 6) == 0 &&
        memcmp(&header[TAR_OFFSET_VERSION], "00", 2) == 0)
        return true;
    /* GNU tar format */
    if (memcmp(&header[TAR_OFFSET_MAGIC], "ustar  \0", 8) == 0)
        return true;
    /* not-quite-POSIX format written by pre-9.3 pg_dump */
    if (memcmp(&header[TAR_OFFSET_MAGIC], "ustar00\0", 8) == 0)
        return true;
 
    return false;
}

References read_tar_number(), TAR_OFFSET_CHECKSUM, TAR_OFFSET_MAGIC, TAR_OFFSET_VERSION, and tarChecksum().

Referenced by _discoverArchiveFormat().

tarClose()

static void tarClose ( ArchiveHandle *  AH, TAR_MEMBER *  th  )

static

Definition at line 398 of file pg_backup_tar.c.

{
    if (AH->compression_spec.algorithm != PG_COMPRESSION_NONE)
        pg_fatal("compression is not supported by tar archive format");
 
    if (th->mode == 'w')
        _tarAddFile(AH, th);    /* This will close the temp file */
 
    /*
     * else Nothing to do for normal read since we don't dup() normal file
     * handle, and we don't use temp files.
     */
 
    free(th->targetFile);
 
    th->nFH = NULL;
}

References _tarAddFile(), pg_compress_specification::algorithm, _archiveHandle::compression_spec, free, TAR_MEMBER::mode, TAR_MEMBER::nFH, PG_COMPRESSION_NONE, pg_fatal, and TAR_MEMBER::targetFile.

Referenced by _CloseArchive(), _EndData(), _EndLO(), _EndLOs(), _LoadLOs(), _PrintFileData(), and InitArchiveFmt_Tar().

tarOpen()

static TAR_MEMBER * tarOpen ( ArchiveHandle *  AH, const char *  filename, char  mode  )

static

Definition at line 302 of file pg_backup_tar.c.

{
    lclContext *ctx = (lclContext *) AH->formatData;
    TAR_MEMBER *tm;
 
    if (mode == 'r')
    {
        tm = _tarPositionTo(AH, filename);
        if (!tm)                /* Not found */
        {
            if (filename)
            {
                /*
                 * Couldn't find the requested file. Future: do SEEK(0) and
                 * retry.
                 */
                pg_fatal("could not find file \"%s\" in archive", filename);
            }
            else
            {
                /* Any file OK, none left, so return NULL */
                return NULL;
            }
        }
 
        if (AH->compression_spec.algorithm == PG_COMPRESSION_NONE)
            tm->nFH = ctx->tarFH;
        else
            pg_fatal("compression is not supported by tar archive format");
    }
    else
    {
        int         old_umask;
 
        tm = pg_malloc0_object(TAR_MEMBER);
 
        /*
         * POSIX does not require, but permits, tmpfile() to restrict file
         * permissions.  Given an OS crash after we write data, the filesystem
         * might retain the data but forget tmpfile()'s unlink().  If so, the
         * file mode protects confidentiality of the data written.
         */
        old_umask = umask(S_IRWXG | S_IRWXO);
 
#ifndef WIN32
        tm->tmpFH = tmpfile();
#else
 
        /*
         * On WIN32, tmpfile() generates a filename in the root directory,
         * which requires administrative permissions on certain systems. Loop
         * until we find a unique file name we can create.
         */
        while (1)
        {
            char       *name;
            int         fd;
 
            name = _tempnam(NULL, "pg_temp_");
            if (name == NULL)
                break;
            fd = open(name, O_RDWR | O_CREAT | O_EXCL | O_BINARY |
                      O_TEMPORARY, S_IRUSR | S_IWUSR);
            free(name);
 
            if (fd != -1)       /* created a file */
            {
                tm->tmpFH = fdopen(fd, "w+b");
                break;
            }
            else if (errno != EEXIST)   /* failure other than file exists */
                break;
        }
#endif
 
        if (tm->tmpFH == NULL)
            pg_fatal("could not generate temporary file name: %m");
 
        umask(old_umask);
 
        if (AH->compression_spec.algorithm == PG_COMPRESSION_NONE)
            tm->nFH = tm->tmpFH;
        else
            pg_fatal("compression is not supported by tar archive format");
 
        tm->AH = AH;
        tm->targetFile = pg_strdup(filename);
    }
 
    tm->mode = mode;
    tm->tarFH = ctx->tarFH;
 
    return tm;
}

References _tarPositionTo(), pg_compress_specification::algorithm, _archiveHandle::compression_spec, fd(), filename, _archiveHandle::formatData, free, if(), mode, name, PG_COMPRESSION_NONE, pg_fatal, pg_malloc0_object, pg_strdup(), S_IRUSR, S_IRWXG, S_IRWXO, S_IWUSR, lclContext::tarFH, and tm.

Referenced by _CloseArchive(), _LoadLOs(), _PrintFileData(), _StartData(), _StartLO(), _StartLOs(), and InitArchiveFmt_Tar().

tarPrintf()

static int tarPrintf ( TAR_MEMBER *  th, const char *  fmt,   ...  )

static

Definition at line 954 of file pg_backup_tar.c.

{
    int         save_errno = errno;
    char       *p;
    size_t      len = 128;      /* initial assumption about buffer size */
    size_t      cnt;
 
    for (;;)
    {
        va_list     args;
 
        /* Allocate work buffer. */
        p = (char *) pg_malloc(len);
 
        /* Try to format the data. */
        errno = save_errno;
        va_start(args, fmt);
        cnt = pvsnprintf(p, len, fmt, args);
        va_end(args);
 
        if (cnt < len)
            break;              /* success */
 
        /* Release buffer and loop around to try again with larger len. */
        free(p);
        len = cnt;
    }
 
    cnt = tarWrite(p, cnt, th);
    free(p);
    return (int) cnt;
}

References generate_unaccent_rules::args, fmt, free, len, pg_malloc(), pvsnprintf(), tarWrite(), va_end(), and va_start().

Referenced by _CloseArchive(), and _StartLO().

tarRead()

static size_t tarRead ( void *  buf, size_t  len, TAR_MEMBER *  th  )

static

Definition at line 511 of file pg_backup_tar.c.

{
    size_t      res;
 
    if (th->pos + len > th->fileLen)
        len = th->fileLen - th->pos;
 
    if (len <= 0)
        return 0;
 
    res = _tarReadRaw(th->AH, buf, len, th, NULL);
 
    th->pos += res;
 
    return res;
}

References _tarReadRaw(), TAR_MEMBER::AH, buf, TAR_MEMBER::fileLen, len, TAR_MEMBER::pos, and res.

Referenced by _LoadLOs(), _PrintFileData(), _ReadBuf(), and _ReadByte().

tarWrite()

static size_t tarWrite ( const void *  buf, size_t  len, TAR_MEMBER *  th  )

static

Definition at line 529 of file pg_backup_tar.c.

{
    size_t      res;
 
    res = fwrite(buf, 1, len, th->nFH);
 
    th->pos += res;
    return res;
}

References buf, len, TAR_MEMBER::nFH, TAR_MEMBER::pos, and res.

Referenced by _scriptOut(), _WriteBuf(), _WriteByte(), _WriteData(), and tarPrintf().