bbstreamer_gzip.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_gzip.c:

Go to the source code of this file.

Functions
bbstreamer *	bbstreamer_gzip_writer_new (char *pathname, FILE *file, pg_compress_specification *compress)
bbstreamer *	bbstreamer_gzip_decompressor_new (bbstreamer *next)

Function Documentation

bbstreamer_gzip_decompressor_new()

bbstreamer* bbstreamer_gzip_decompressor_new

(

bbstreamer *

next

)

Definition at line 214 of file bbstreamer_gzip.c.

{
#ifdef HAVE_LIBZ
    bbstreamer_gzip_decompressor *streamer;
    z_stream   *zs;
 
    Assert(next != NULL);
 
    streamer = palloc0(sizeof(bbstreamer_gzip_decompressor));
    *((const bbstreamer_ops **) &streamer->base.bbs_ops) =
        &bbstreamer_gzip_decompressor_ops;
 
    streamer->base.bbs_next = next;
    initStringInfo(&streamer->base.bbs_buffer);
 
    /* Initialize internal stream state for decompression */
    zs = &streamer->zstream;
    zs->zalloc = gzip_palloc;
    zs->zfree = gzip_pfree;
    zs->next_out = (uint8 *) streamer->base.bbs_buffer.data;
    zs->avail_out = streamer->base.bbs_buffer.maxlen;
 
    /*
     * Data compression was initialized using deflateInit2 to request a gzip
     * header. Similarly, we are using inflateInit2 to initialize data
     * decompression.
     *
     * Per the documentation for inflateInit2, the second argument is
     * "windowBits" and its value must be greater than or equal to the value
     * provided while compressing the data, so we are using the maximum
     * possible value for safety.
     */
    if (inflateInit2(zs, 15 + 16) != Z_OK)
        pg_fatal("could not initialize compression library");
 
    return &streamer->base;
#else
    pg_fatal("this build does not support gzip compression");
    return NULL;                /* keep compiler quiet */
#endif
}

References Assert(), _z_stream::avail_out, if(), initStringInfo(), next, _z_stream::next_out, palloc0(), and pg_fatal.

Referenced by CreateBackupStreamer().

bbstreamer_gzip_writer_new()

bbstreamer* bbstreamer_gzip_writer_new	(	char *	pathname,
		FILE *	file,
		pg_compress_specification *	compress
	)

Definition at line 79 of file bbstreamer_gzip.c.

{
#ifdef HAVE_LIBZ
    bbstreamer_gzip_writer *streamer;
 
    streamer = palloc0(sizeof(bbstreamer_gzip_writer));
    *((const bbstreamer_ops **) &streamer->base.bbs_ops) =
        &bbstreamer_gzip_writer_ops;
 
    streamer->pathname = pstrdup(pathname);
 
    if (file == NULL)
    {
        streamer->gzfile = gzopen(pathname, "wb");
        if (streamer->gzfile == NULL)
            pg_fatal("could not create compressed file \"%s\": %m",
                     pathname);
    }
    else
    {
        int         fd = dup(fileno(file));
 
        if (fd < 0)
            pg_fatal("could not duplicate stdout: %m");
 
        streamer->gzfile = gzdopen(fd, "wb");
        if (streamer->gzfile == NULL)
            pg_fatal("could not open output file: %m");
    }
 
    if ((compress->options & PG_COMPRESSION_OPTION_LEVEL) != 0 &&
        gzsetparams(streamer->gzfile, compress->level,
                    Z_DEFAULT_STRATEGY) != Z_OK)
        pg_fatal("could not set compression level %d: %s",
                 compress->level, get_gz_error(streamer->gzfile));
 
    return &streamer->base;
#else
    pg_fatal("this build does not support gzip compression");
    return NULL;                /* keep compiler quiet */
#endif
}

References fd(), pg_compress_specification::level, pg_compress_specification::options, palloc0(), PG_COMPRESSION_OPTION_LEVEL, pg_fatal, and pstrdup().

Referenced by CreateBackupStreamer().