bbstreamer_file.c File Reference

#include "postgres_fe.h"
#include <unistd.h>
#include "bbstreamer.h"
#include "common/logging.h"
#include "common/file_perm.h"
#include "common/string.h"

Include dependency graph for bbstreamer_file.c:

Go to the source code of this file.

Data Structures
struct	bbstreamer_plain_writer
struct	bbstreamer_extractor

Typedefs
typedef struct bbstreamer_plain_writer	bbstreamer_plain_writer
typedef struct bbstreamer_extractor	bbstreamer_extractor

Functions
static void	bbstreamer_plain_writer_content (bbstreamer *streamer, bbstreamer_member *member, const char *data, int len, bbstreamer_archive_context context)
static void	bbstreamer_plain_writer_finalize (bbstreamer *streamer)
static void	bbstreamer_plain_writer_free (bbstreamer *streamer)
static void	bbstreamer_extractor_content (bbstreamer *streamer, bbstreamer_member *member, const char *data, int len, bbstreamer_archive_context context)
static void	bbstreamer_extractor_finalize (bbstreamer *streamer)
static void	bbstreamer_extractor_free (bbstreamer *streamer)
static void	extract_directory (const char *filename, mode_t mode)
static void	extract_link (const char *filename, const char *linktarget)
static FILE *	create_file_for_extract (const char *filename, mode_t mode)
bbstreamer *	bbstreamer_plain_writer_new (char *pathname, FILE *file)
bbstreamer *	bbstreamer_extractor_new (const char *basepath, const char *(*link_map)(const char *), void(*report_output_file)(const char *))
static bool	should_allow_existing_directory (const char *pathname)

Variables
const bbstreamer_ops	bbstreamer_plain_writer_ops
const bbstreamer_ops	bbstreamer_extractor_ops

Typedef Documentation

bbstreamer_extractor

typedef struct bbstreamer_extractor bbstreamer_extractor

bbstreamer_plain_writer

typedef struct bbstreamer_plain_writer bbstreamer_plain_writer

Function Documentation

bbstreamer_extractor_content()

static void bbstreamer_extractor_content ( bbstreamer *  streamer, bbstreamer_member *  member, const char *  data, int  len, bbstreamer_archive_context  context  )

static

Definition at line 203 of file bbstreamer_file.c.

{
    bbstreamer_extractor *mystreamer = (bbstreamer_extractor *) streamer;
    int         fnamelen;
 
    Assert(member != NULL || context == BBSTREAMER_ARCHIVE_TRAILER);
    Assert(context != BBSTREAMER_UNKNOWN);
 
    switch (context)
    {
        case BBSTREAMER_MEMBER_HEADER:
            Assert(mystreamer->file == NULL);
 
            /* Prepend basepath. */
            snprintf(mystreamer->filename, sizeof(mystreamer->filename),
                     "%s/%s", mystreamer->basepath, member->pathname);
 
            /* Remove any trailing slash. */
            fnamelen = strlen(mystreamer->filename);
            if (mystreamer->filename[fnamelen - 1] == '/')
                mystreamer->filename[fnamelen - 1] = '\0';
 
            /* Dispatch based on file type. */
            if (member->is_directory)
                extract_directory(mystreamer->filename, member->mode);
            else if (member->is_link)
            {
                const char *linktarget = member->linktarget;
 
                if (mystreamer->link_map)
                    linktarget = mystreamer->link_map(linktarget);
                extract_link(mystreamer->filename, linktarget);
            }
            else
                mystreamer->file =
                    create_file_for_extract(mystreamer->filename,
                                            member->mode);
 
            /* Report output file change. */
            if (mystreamer->report_output_file)
                mystreamer->report_output_file(mystreamer->filename);
            break;
 
        case BBSTREAMER_MEMBER_CONTENTS:
            if (mystreamer->file == NULL)
                break;
 
            errno = 0;
            if (len > 0 && fwrite(data, len, 1, mystreamer->file) != 1)
            {
                /* if write didn't set errno, assume problem is no disk space */
                if (errno == 0)
                    errno = ENOSPC;
                pg_fatal("could not write to file \"%s\": %m",
                         mystreamer->filename);
            }
            break;
 
        case BBSTREAMER_MEMBER_TRAILER:
            if (mystreamer->file == NULL)
                break;
            fclose(mystreamer->file);
            mystreamer->file = NULL;
            break;
 
        case BBSTREAMER_ARCHIVE_TRAILER:
            break;
 
        default:
            /* Shouldn't happen. */
            pg_fatal("unexpected state while extracting archive");
    }
}

References Assert(), bbstreamer_extractor::basepath, BBSTREAMER_ARCHIVE_TRAILER, BBSTREAMER_MEMBER_CONTENTS, BBSTREAMER_MEMBER_HEADER, BBSTREAMER_MEMBER_TRAILER, BBSTREAMER_UNKNOWN, create_file_for_extract(), data, extract_directory(), extract_link(), bbstreamer_extractor::file, bbstreamer_extractor::filename, bbstreamer_member::is_directory, bbstreamer_member::is_link, len, bbstreamer_extractor::link_map, bbstreamer_member::linktarget, bbstreamer_member::mode, bbstreamer_member::pathname, pg_fatal, bbstreamer_extractor::report_output_file, and snprintf.

bbstreamer_extractor_finalize()

static void bbstreamer_extractor_finalize ( bbstreamer *  streamer )

static

Definition at line 377 of file bbstreamer_file.c.

{
    bbstreamer_extractor *mystreamer PG_USED_FOR_ASSERTS_ONLY
    = (bbstreamer_extractor *) streamer;
 
    Assert(mystreamer->file == NULL);
}

References Assert(), and PG_USED_FOR_ASSERTS_ONLY.

bbstreamer_extractor_free()

static void bbstreamer_extractor_free ( bbstreamer *  streamer )

static

Definition at line 389 of file bbstreamer_file.c.

{
    bbstreamer_extractor *mystreamer = (bbstreamer_extractor *) streamer;
 
    pfree(mystreamer->basepath);
    pfree(mystreamer);
}

References bbstreamer_extractor::basepath, and pfree().

bbstreamer_extractor_new()

bbstreamer* bbstreamer_extractor_new	(	const char *	basepath,
		const char *(*)(const char *)	link_map,
		void(*)(const char *)	report_output_file
	)

Definition at line 183 of file bbstreamer_file.c.

{
    bbstreamer_extractor *streamer;
 
    streamer = palloc0(sizeof(bbstreamer_extractor));
    *((const bbstreamer_ops **) &streamer->base.bbs_ops) =
        &bbstreamer_extractor_ops;
    streamer->basepath = pstrdup(basepath);
    streamer->link_map = link_map;
    streamer->report_output_file = report_output_file;
 
    return &streamer->base;
}

References bbstreamer_extractor::base, bbstreamer_extractor::basepath, bbstreamer::bbs_ops, bbstreamer_extractor_ops, bbstreamer_extractor::link_map, palloc0(), pstrdup(), and bbstreamer_extractor::report_output_file.

Referenced by CreateBackupStreamer().

bbstreamer_plain_writer_content()

static void bbstreamer_plain_writer_content ( bbstreamer *  streamer, bbstreamer_member *  member, const char *  data, int  len, bbstreamer_archive_context  context  )

static

Definition at line 104 of file bbstreamer_file.c.

{
    bbstreamer_plain_writer *mystreamer;
 
    mystreamer = (bbstreamer_plain_writer *) streamer;
 
    if (len == 0)
        return;
 
    errno = 0;
    if (fwrite(data, len, 1, mystreamer->file) != 1)
    {
        /* if write didn't set errno, assume problem is no disk space */
        if (errno == 0)
            errno = ENOSPC;
        pg_fatal("could not write to file \"%s\": %m",
                 mystreamer->pathname);
    }
}

References data, bbstreamer_plain_writer::file, len, bbstreamer_plain_writer::pathname, and pg_fatal.

bbstreamer_plain_writer_finalize()

static void bbstreamer_plain_writer_finalize ( bbstreamer *  streamer )

static

Definition at line 131 of file bbstreamer_file.c.

{
    bbstreamer_plain_writer *mystreamer;
 
    mystreamer = (bbstreamer_plain_writer *) streamer;
 
    if (mystreamer->should_close_file && fclose(mystreamer->file) != 0)
        pg_fatal("could not close file \"%s\": %m",
                 mystreamer->pathname);
 
    mystreamer->file = NULL;
    mystreamer->should_close_file = false;
}

References bbstreamer_plain_writer::file, bbstreamer_plain_writer::pathname, pg_fatal, and bbstreamer_plain_writer::should_close_file.

bbstreamer_plain_writer_free()

static void bbstreamer_plain_writer_free ( bbstreamer *  streamer )

static

Definition at line 149 of file bbstreamer_file.c.

{
    bbstreamer_plain_writer *mystreamer;
 
    mystreamer = (bbstreamer_plain_writer *) streamer;
 
    Assert(!mystreamer->should_close_file);
    Assert(mystreamer->base.bbs_next == NULL);
 
    pfree(mystreamer->pathname);
    pfree(mystreamer);
}

References Assert(), bbstreamer_plain_writer::base, bbstreamer::bbs_next, bbstreamer_plain_writer::pathname, pfree(), and bbstreamer_plain_writer::should_close_file.

bbstreamer_plain_writer_new()

bbstreamer* bbstreamer_plain_writer_new	(	char *	pathname,
		FILE *	file
	)

Definition at line 78 of file bbstreamer_file.c.

{
    bbstreamer_plain_writer *streamer;
 
    streamer = palloc0(sizeof(bbstreamer_plain_writer));
    *((const bbstreamer_ops **) &streamer->base.bbs_ops) =
        &bbstreamer_plain_writer_ops;
 
    streamer->pathname = pstrdup(pathname);
    streamer->file = file;
 
    if (file == NULL)
    {
        streamer->file = fopen(pathname, "wb");
        if (streamer->file == NULL)
            pg_fatal("could not create file \"%s\": %m", pathname);
        streamer->should_close_file = true;
    }
 
    return &streamer->base;
}

References bbstreamer_plain_writer::base, bbstreamer::bbs_ops, bbstreamer_plain_writer_ops, bbstreamer_plain_writer::file, palloc0(), bbstreamer_plain_writer::pathname, pg_fatal, pstrdup(), and bbstreamer_plain_writer::should_close_file.

Referenced by CreateBackupStreamer().

create_file_for_extract()

static FILE * create_file_for_extract ( const char *  filename, mode_t  mode  )

static

Definition at line 354 of file bbstreamer_file.c.

{
    FILE       *file;
 
    file = fopen(filename, "wb");
    if (file == NULL)
        pg_fatal("could not create file \"%s\": %m", filename);
 
#ifndef WIN32
    if (chmod(filename, mode))
        pg_fatal("could not set permissions on file \"%s\": %m",
                 filename);
#endif
 
    return file;
}

References filename, mode, and pg_fatal.

Referenced by bbstreamer_extractor_content().

extract_directory()

static void extract_directory ( const char *  filename, mode_t  mode  )

static

Definition at line 316 of file bbstreamer_file.c.

{
    if (mkdir(filename, pg_dir_create_mode) != 0 &&
        (errno != EEXIST || !should_allow_existing_directory(filename)))
        pg_fatal("could not create directory \"%s\": %m",
                 filename);
 
#ifndef WIN32
    if (chmod(filename, mode))
        pg_fatal("could not set permissions on directory \"%s\": %m",
                 filename);
#endif
}

References filename, mkdir, mode, pg_dir_create_mode, pg_fatal, and should_allow_existing_directory().

Referenced by bbstreamer_extractor_content().

extract_link()

static void extract_link ( const char *  filename, const char *  linktarget  )

static

Definition at line 341 of file bbstreamer_file.c.

{
    if (symlink(linktarget, filename) != 0)
        pg_fatal("could not create symbolic link from \"%s\" to \"%s\": %m",
                 filename, linktarget);
}

References filename, pg_fatal, and symlink.

Referenced by bbstreamer_extractor_content().

should_allow_existing_directory()

static bool should_allow_existing_directory ( const char *  pathname )

static

Definition at line 292 of file bbstreamer_file.c.

{
    const char *filename = last_dir_separator(pathname) + 1;
 
    if (strcmp(filename, "pg_wal") == 0 ||
        strcmp(filename, "pg_xlog") == 0 ||
        strcmp(filename, "archive_status") == 0 ||
        strcmp(filename, "pg_tblspc") == 0)
        return true;
 
    if (strspn(filename, "0123456789") == strlen(filename))
    {
        const char *pg_tblspc = strstr(pathname, "/pg_tblspc/");
 
        return pg_tblspc != NULL && pg_tblspc + 11 == filename;
    }
 
    return false;
}

References filename, and last_dir_separator().

Referenced by extract_directory().

Variable Documentation

bbstreamer_extractor_ops

const bbstreamer_ops bbstreamer_extractor_ops

Initial value:

= {
    .content = bbstreamer_extractor_content,
    .finalize = bbstreamer_extractor_finalize,
    .free = bbstreamer_extractor_free
}

Definition at line 62 of file bbstreamer_file.c.

Referenced by bbstreamer_extractor_new().

bbstreamer_plain_writer_ops

const bbstreamer_ops bbstreamer_plain_writer_ops

Initial value:

= {
    .content = bbstreamer_plain_writer_content,
    .finalize = bbstreamer_plain_writer_finalize,
    .free = bbstreamer_plain_writer_free
}

Definition at line 46 of file bbstreamer_file.c.

Referenced by bbstreamer_plain_writer_new().