#include "access/xlogdefs.h"
#include "fe_utils/astreamer.h"

Include dependency graph for pg_waldump.h:

This graph shows which files directly or indirectly include this file:

Data Structures
struct	XLogDumpPrivate

Typedefs
typedef struct XLogDumpPrivate	XLogDumpPrivate

Functions
int	open_file_in_directory (const char directory, const char fname)

void	init_archive_reader (XLogDumpPrivate *privateInfo, pg_compress_algorithm compression)

void	free_archive_reader (XLogDumpPrivate *privateInfo)

int	read_archive_wal_page (XLogDumpPrivate privateInfo, XLogRecPtr targetPagePtr, size_t count, char readBuff)

void	free_archive_wal_entry (const char fname, XLogDumpPrivate privateInfo)

Variables
char *	TmpWalSegDir

Typedef Documentation

◆ XLogDumpPrivate

typedef struct XLogDumpPrivate XLogDumpPrivate

Function Documentation

◆ free_archive_reader()

void free_archive_reader ( XLogDumpPrivate * privateInfo )

extern

Definition at line 253 of file archive_waldump.c.

{
    /*
     * NB: Normally, astreamer_finalize() is called before astreamer_free() to
     * flush any remaining buffered data or to ensure the end of the tar
     * archive is reached.  read_archive_file() may have done so.  However,
     * when decoding WAL we can stop once we hit the end LSN, so we may never
     * have read all of the input file.  In that case any remaining buffered
     * data or unread portion of the archive can be safely ignored.
     */
    astreamer_free(privateInfo->archive_streamer);
 
    /* Free any remaining hash table entries and their buffers. */
    if (privateInfo->archive_wal_htab != NULL)
    {
        ArchivedWAL_iterator iter;
        ArchivedWALFile *entry;
 
        ArchivedWAL_start_iterate(privateInfo->archive_wal_htab, &iter);
        while ((entry = ArchivedWAL_iterate(privateInfo->archive_wal_htab,
                                            &iter)) != NULL)
        {
            if (entry->buf != NULL)
                destroyStringInfo(entry->buf);
        }
        ArchivedWAL_destroy(privateInfo->archive_wal_htab);
        privateInfo->archive_wal_htab = NULL;
    }
 
    /* Free the reusable read buffer. */
    if (privateInfo->archive_read_buf != NULL)
    {
        pg_free(privateInfo->archive_read_buf);
        privateInfo->archive_read_buf = NULL;
    }
 
    /* Close the file. */
    if (close(privateInfo->archive_fd) != 0)
        pg_log_error("could not close file \"%s\": %m",
                     privateInfo->archive_name);
}

References XLogDumpPrivate::archive_fd, XLogDumpPrivate::archive_name, XLogDumpPrivate::archive_read_buf, XLogDumpPrivate::archive_streamer, XLogDumpPrivate::archive_wal_htab, astreamer_free(), ArchivedWALFile::buf, close, destroyStringInfo(), fb(), pg_free(), and pg_log_error.

Referenced by main().

◆ free_archive_wal_entry()

void free_archive_wal_entry	(	const char *	fname,
		XLogDumpPrivate *	privateInfo
	)

extern

Definition at line 405 of file archive_waldump.c.

{
    ArchivedWALFile *entry;
    const char *oldfname;
 
    entry = ArchivedWAL_lookup(privateInfo->archive_wal_htab, fname);
 
    if (entry == NULL)
        return;
 
    /* Destroy the buffer */
    destroyStringInfo(entry->buf);
    entry->buf = NULL;
 
    /* Remove temporary file if any */
    if (entry->spilled)
    {
        char        fpath[MAXPGPATH];
 
        snprintf(fpath, MAXPGPATH, "%s/%s", TmpWalSegDir, fname);
 
        if (unlink(fpath) == 0)
            pg_log_debug("removed file \"%s\"", fpath);
    }
 
    /* Clear cur_file if it points to the entry being freed */
    if (privateInfo->cur_file == entry)
        privateInfo->cur_file = NULL;
 
    /*
     * ArchivedWAL_delete_item may cause other hash table entries to move.
     * Therefore, if cur_file isn't NULL now, we have to be prepared to look
     * that entry up again after the deletion.  Fortunately, the entry's fname
     * string won't move.
     */
    oldfname = privateInfo->cur_file ? privateInfo->cur_file->fname : NULL;
 
    ArchivedWAL_delete_item(privateInfo->archive_wal_htab, entry);
 
    if (oldfname)
    {
        privateInfo->cur_file = ArchivedWAL_lookup(privateInfo->archive_wal_htab,
                                                   oldfname);
        /* ... it had better still be there */
        Assert(privateInfo->cur_file != NULL);
    }
}

References XLogDumpPrivate::archive_wal_htab, Assert, ArchivedWALFile::buf, XLogDumpPrivate::cur_file, destroyStringInfo(), fb(), ArchivedWALFile::fname, MAXPGPATH, pg_log_debug, snprintf, ArchivedWALFile::spilled, and TmpWalSegDir.

Referenced by init_archive_reader(), and TarWALDumpReadPage().

◆ init_archive_reader()

void init_archive_reader	(	XLogDumpPrivate *	privateInfo,
		pg_compress_algorithm	compression
	)

extern

Definition at line 129 of file archive_waldump.c.

{
    int         fd;
    astreamer  *streamer;
    ArchivedWALFile *entry = NULL;
    XLogLongPageHeader longhdr;
    ArchivedWAL_iterator iter;
 
    /* Open tar archive and store its file descriptor */
    fd = open_file_in_directory(privateInfo->archive_dir,
                                privateInfo->archive_name);
 
    if (fd < 0)
        pg_fatal("could not open file \"%s\"", privateInfo->archive_name);
 
    privateInfo->archive_fd = fd;
    privateInfo->archive_fd_eof = false;
 
    streamer = astreamer_waldump_new(privateInfo);
 
    /* We must first parse the tar archive. */
    streamer = astreamer_tar_parser_new(streamer);
 
    /* If the archive is compressed, decompress before parsing. */
    if (compression == PG_COMPRESSION_GZIP)
        streamer = astreamer_gzip_decompressor_new(streamer);
    else if (compression == PG_COMPRESSION_LZ4)
        streamer = astreamer_lz4_decompressor_new(streamer);
    else if (compression == PG_COMPRESSION_ZSTD)
        streamer = astreamer_zstd_decompressor_new(streamer);
 
    privateInfo->archive_streamer = streamer;
 
    /*
     * Allocate a buffer for reading the archive file to begin content
     * decoding.
     */
    privateInfo->archive_read_buf = pg_malloc(READ_CHUNK_SIZE);
    privateInfo->archive_read_buf_size = READ_CHUNK_SIZE;
 
    /*
     * Hash table storing WAL entries read from the archive with an arbitrary
     * initial size.
     */
    privateInfo->archive_wal_htab = ArchivedWAL_create(8, NULL);
 
    /*
     * Read until we have at least one WAL segment with enough data to extract
     * the WAL segment size from the long page header.
     *
     * We must not rely on cur_file here, because it can become NULL if a
     * member trailer is processed during a read_archive_file() call. Instead,
     * scan the hash table after each read to find any entry with sufficient
     * data.
     */
    while (entry == NULL)
    {
        if (!read_archive_file(privateInfo))
            pg_fatal("could not find WAL in archive \"%s\"",
                     privateInfo->archive_name);
 
        ArchivedWAL_start_iterate(privateInfo->archive_wal_htab, &iter);
        while ((entry = ArchivedWAL_iterate(privateInfo->archive_wal_htab,
                                            &iter)) != NULL)
        {
            if (entry->read_len >= sizeof(XLogLongPageHeaderData))
                break;
        }
    }
 
    /* Extract the WAL segment size from the long page header */
    longhdr = (XLogLongPageHeader) entry->buf->data;
 
    if (!IsValidWalSegSize(longhdr->xlp_seg_size))
    {
        pg_log_error(ngettext("invalid WAL segment size in WAL file from archive \"%s\" (%d byte)",
                              "invalid WAL segment size in WAL file from archive \"%s\" (%d bytes)",
                              longhdr->xlp_seg_size),
                     privateInfo->archive_name, longhdr->xlp_seg_size);
        pg_log_error_detail("The WAL segment size must be a power of two between 1 MB and 1 GB.");
        exit(1);
    }
 
    privateInfo->segsize = longhdr->xlp_seg_size;
 
    /*
     * With the WAL segment size available, we can now initialize the
     * dependent start and end segment numbers.
     */
    Assert(!XLogRecPtrIsInvalid(privateInfo->startptr));
    XLByteToSeg(privateInfo->startptr, privateInfo->start_segno,
                privateInfo->segsize);
 
    if (!XLogRecPtrIsInvalid(privateInfo->endptr))
        XLByteToSeg(privateInfo->endptr, privateInfo->end_segno,
                    privateInfo->segsize);
 
    /*
     * Now that we have initialized the filtering parameters (start_segno and
     * end_segno), we can discard any already-loaded WAL hash table entries
     * for segments we don't actually need.  Subsequent WAL will be filtered
     * automatically by the archive streamer using the updated start_segno and
     * end_segno values.
     */
    ArchivedWAL_start_iterate(privateInfo->archive_wal_htab, &iter);
    while ((entry = ArchivedWAL_iterate(privateInfo->archive_wal_htab,
                                        &iter)) != NULL)
    {
        XLogSegNo   segno;
        TimeLineID  timeline;
 
        XLogFromFileName(entry->fname, &timeline, &segno, privateInfo->segsize);
        if (privateInfo->timeline != timeline ||
            privateInfo->start_segno > segno ||
            privateInfo->end_segno < segno)
            free_archive_wal_entry(entry->fname, privateInfo);
    }
}

Referenced by main().

◆ open_file_in_directory()

int open_file_in_directory	(	const char *	directory,
		const char *	fname
	)

extern

Definition at line 182 of file pg_waldump.c.

{
    int         fd = -1;
    char        fpath[MAXPGPATH];
 
    Assert(directory != NULL);
 
    snprintf(fpath, MAXPGPATH, "%s/%s", directory, fname);
    fd = open(fpath, O_RDONLY | PG_BINARY, 0);
 
    if (fd < 0 && errno != ENOENT)
        pg_fatal("could not open file \"%s\": %m", fname);
    return fd;
}

References Assert, directory, fb(), fd(), MAXPGPATH, PG_BINARY, pg_fatal, and snprintf.

Referenced by init_archive_reader(), main(), search_directory(), TarWALDumpReadPage(), and WALDumpOpenSegment().

◆ read_archive_wal_page()

int read_archive_wal_page	(	XLogDumpPrivate *	privateInfo,
		XLogRecPtr	targetPagePtr,
		size_t	count,
		char *	readBuff
	)

extern

Definition at line 301 of file archive_waldump.c.

{
    char       *p = readBuff;
    size_t      nbytes = count;
    XLogRecPtr  recptr = targetPagePtr;
    int         segsize = privateInfo->segsize;
    XLogSegNo   segno;
    char        fname[MAXFNAMELEN];
    ArchivedWALFile *entry;
 
    /* Identify the segment and locate its entry in the archive hash */
    XLByteToSeg(targetPagePtr, segno, segsize);
    XLogFileName(fname, privateInfo->timeline, segno, segsize);
    entry = get_archive_wal_entry(fname, privateInfo);
    Assert(!entry->spilled);
 
    while (nbytes > 0)
    {
        char       *buf = entry->buf->data;
        int         bufLen = entry->buf->len;
        XLogRecPtr  endPtr;
        XLogRecPtr  startPtr;
 
        /*
         * Calculate the LSN range currently residing in the buffer.
         *
         * read_len tracks total bytes received for this segment, so endPtr is
         * the LSN just past the last buffered byte, and startPtr is the LSN
         * of the first buffered byte.
         */
        XLogSegNoOffsetToRecPtr(segno, entry->read_len, segsize, endPtr);
        startPtr = endPtr - bufLen;
 
        /*
         * Copy the requested WAL record if it exists in the buffer.
         */
        if (bufLen > 0 && startPtr <= recptr && recptr < endPtr)
        {
            int         copyBytes;
            int         offset = recptr - startPtr;
 
            /*
             * Given startPtr <= recptr < endPtr and a total buffer size
             * 'bufLen', the offset (recptr - startPtr) will always be less
             * than 'bufLen'.
             */
            Assert(offset < bufLen);
 
            copyBytes = Min(nbytes, bufLen - offset);
            memcpy(p, buf + offset, copyBytes);
 
            /* Update state for read */
            recptr += copyBytes;
            nbytes -= copyBytes;
            p += copyBytes;
        }
        else
        {
            /*
             * We evidently need to fetch more data.  Raise an error if the
             * archive streamer has moved past our segment (meaning the WAL
             * file in the archive is shorter than expected) or if reading the
             * archive reached EOF.
             */
            if (privateInfo->cur_file != entry)
                pg_fatal("WAL segment \"%s\" in archive \"%s\" is too short: read %zu of %zu bytes",
                         fname, privateInfo->archive_name,
                         (count - nbytes), count);
            if (!read_archive_file(privateInfo))
                pg_fatal("unexpected end of archive \"%s\" while reading \"%s\": read %zu of %zu bytes",
                         privateInfo->archive_name, fname,
                         (count - nbytes), count);
 
            /*
             * Loading more data may have moved hash table entries, so we must
             * re-look-up the one we are reading from.
             */
            entry = ArchivedWAL_lookup(privateInfo->archive_wal_htab, fname);
            /* ... it had better still be there */
            Assert(entry != NULL);
        }
    }
 
    /*
     * Should have successfully read all the requested bytes or reported a
     * failure before this point.
     */
    Assert(nbytes == 0);
 
    /*
     * Return count unchanged; the caller expects this convention, matching
     * the routine that reads WAL pages from physical files.
     */
    return count;
}

References XLogDumpPrivate::archive_name, XLogDumpPrivate::archive_wal_htab, Assert, buf, ArchivedWALFile::buf, XLogDumpPrivate::cur_file, StringInfoData::data, fb(), get_archive_wal_entry(), StringInfoData::len, MAXFNAMELEN, memcpy(), Min, pg_fatal, read_archive_file(), ArchivedWALFile::read_len, XLogDumpPrivate::segsize, ArchivedWALFile::spilled, XLogDumpPrivate::timeline, XLByteToSeg, XLogFileName(), and XLogSegNoOffsetToRecPtr.

Referenced by TarWALDumpReadPage().

Variable Documentation

◆ TmpWalSegDir

char* TmpWalSegDir

extern

Definition at line 32 of file archive_waldump.c.

Referenced by cleanup_tmpwal_dir_atexit(), free_archive_wal_entry(), get_archive_wal_entry(), perform_tmp_write(), prepare_tmp_write(), setup_tmpwal_dir(), and TarWALDumpReadPage().

Data Structures

Typedefs

Functions

Variables

Typedef Documentation

◆ XLogDumpPrivate

Function Documentation

◆ free_archive_reader()

◆ free_archive_wal_entry()

◆ init_archive_reader()

◆ open_file_in_directory()

◆ read_archive_wal_page()

Variable Documentation

◆ TmpWalSegDir