compress_none.c File Reference

#include "postgres_fe.h"
#include <unistd.h>
#include "compress_none.h"
#include "pg_backup_utils.h"

Include dependency graph for compress_none.c:

Go to the source code of this file.

Data Structures
struct	NoneCompressorState

Typedefs
typedef struct NoneCompressorState	NoneCompressorState

Functions
static void	ReadDataFromArchiveNone (ArchiveHandle *AH, CompressorState *cs)
static void	WriteDataToArchiveNone (ArchiveHandle *AH, CompressorState *cs, const void *data, size_t dLen)
static void	EndCompressorNone (ArchiveHandle *AH, CompressorState *cs)
void	InitCompressorNone (CompressorState *cs, const pg_compress_specification compression_spec)
static size_t	read_none (void *ptr, size_t size, CompressFileHandle *CFH)
static void	write_none (const void *ptr, size_t size, CompressFileHandle *CFH)
static const char *	get_error_none (CompressFileHandle *CFH)
static char *	gets_none (char *ptr, int size, CompressFileHandle *CFH)
static int	getc_none (CompressFileHandle *CFH)
static bool	close_none (CompressFileHandle *CFH)
static bool	eof_none (CompressFileHandle *CFH)
static bool	open_none (const char *path, int fd, const char *mode, CompressFileHandle *CFH)
static bool	open_write_none (const char *path, const char *mode, CompressFileHandle *CFH)
void	InitCompressFileHandleNone (CompressFileHandle *CFH, const pg_compress_specification compression_spec)

Typedef Documentation

NoneCompressorState

typedef struct NoneCompressorState NoneCompressorState

Function Documentation

close_none()

static bool close_none ( CompressFileHandle *  CFH )

static

Definition at line 204 of file compress_none.c.

{
    FILE       *fp = (FILE *) CFH->private_data;
    int         ret = 0;
 
    CFH->private_data = NULL;
 
    if (fp)
    {
        errno = 0;
        ret = fclose(fp);
        if (ret != 0)
            pg_log_error("could not close file: %m");
    }
 
    return ret == 0;
}

References fb(), and pg_log_error.

Referenced by InitCompressFileHandleNone().

EndCompressorNone()

static void EndCompressorNone ( ArchiveHandle *  AH, CompressorState *  cs  )

static

Definition at line 89 of file compress_none.c.

{
    NoneCompressorState *nonecs = (NoneCompressorState *) cs->private_data;
 
    if (nonecs)
    {
        /* Dump buffer if nonempty */
        if (nonecs->bufdata > 0)
            cs->writeF(AH, nonecs->buffer, nonecs->bufdata);
        /* Free working state */
        pg_free(nonecs->buffer);
        pg_free(nonecs);
        cs->private_data = NULL;
    }
}

References fb(), pg_free(), CompressorState::private_data, and CompressorState::writeF.

Referenced by InitCompressorNone().

eof_none()

static bool eof_none ( CompressFileHandle *  CFH )

static

Definition at line 223 of file compress_none.c.

{
    return feof((FILE *) CFH->private_data) != 0;
}

References fb().

Referenced by InitCompressFileHandleNone().

get_error_none()

static const char * get_error_none ( CompressFileHandle *  CFH )

static

Definition at line 174 of file compress_none.c.

{
    return strerror(errno);
}

References fb(), and strerror.

Referenced by InitCompressFileHandleNone().

getc_none()

static int getc_none ( CompressFileHandle *  CFH )

static

Definition at line 186 of file compress_none.c.

{
    FILE       *fp = (FILE *) CFH->private_data;
    int         ret;
 
    ret = fgetc(fp);
    if (ret == EOF)
    {
        if (!feof(fp))
            pg_fatal("could not read from input file: %m");
        else
            pg_fatal("could not read from input file: end of file");
    }
 
    return ret;
}

References fb(), and pg_fatal.

Referenced by InitCompressFileHandleNone().

gets_none()

static char * gets_none ( char *  ptr, int  size, CompressFileHandle *  CFH  )

static

Definition at line 180 of file compress_none.c.

{
    return fgets(ptr, size, (FILE *) CFH->private_data);
}

References fb().

Referenced by InitCompressFileHandleNone().

InitCompressFileHandleNone()

void InitCompressFileHandleNone	(	CompressFileHandle *	CFH,
		const pg_compress_specification	compression_spec
	)

Definition at line 261 of file compress_none.c.

{
    CFH->open_func = open_none;
    CFH->open_write_func = open_write_none;
    CFH->read_func = read_none;
    CFH->write_func = write_none;
    CFH->gets_func = gets_none;
    CFH->getc_func = getc_none;
    CFH->close_func = close_none;
    CFH->eof_func = eof_none;
    CFH->get_error_func = get_error_none;
 
    CFH->private_data = NULL;
}

References close_none(), eof_none(), fb(), get_error_none(), getc_none(), gets_none(), open_none(), open_write_none(), read_none(), and write_none().

Referenced by InitCompressFileHandle().

InitCompressorNone()

void InitCompressorNone	(	CompressorState *	cs,
		const pg_compress_specification	compression_spec
	)

Definition at line 110 of file compress_none.c.

{
    cs->readData = ReadDataFromArchiveNone;
    cs->writeData = WriteDataToArchiveNone;
    cs->end = EndCompressorNone;
 
    cs->compression_spec = compression_spec;
 
    /*
     * If the caller has defined a write function, prepare the necessary
     * buffer.
     */
    if (cs->writeF)
    {
        NoneCompressorState *nonecs;
 
        nonecs = (NoneCompressorState *) pg_malloc(sizeof(NoneCompressorState));
        nonecs->buflen = DEFAULT_IO_BUFFER_SIZE;
        nonecs->buffer = pg_malloc(nonecs->buflen);
        nonecs->bufdata = 0;
 
        cs->private_data = nonecs;
    }
}

References NoneCompressorState::buflen, CompressorState::compression_spec, DEFAULT_IO_BUFFER_SIZE, CompressorState::end, EndCompressorNone(), fb(), pg_malloc(), CompressorState::private_data, CompressorState::readData, ReadDataFromArchiveNone(), CompressorState::writeData, WriteDataToArchiveNone(), and CompressorState::writeF.

Referenced by AllocateCompressor().

open_none()

static bool open_none ( const char *  path, int  fd, const char *  mode, CompressFileHandle *  CFH  )

static

Definition at line 229 of file compress_none.c.

{
    Assert(CFH->private_data == NULL);
 
    if (fd >= 0)
        CFH->private_data = fdopen(dup(fd), mode);
    else
        CFH->private_data = fopen(path, mode);
 
    if (CFH->private_data == NULL)
        return false;
 
    return true;
}

References Assert, fb(), fd(), and mode.

Referenced by InitCompressFileHandleNone().

open_write_none()

static bool open_write_none ( const char *  path, const char *  mode, CompressFileHandle *  CFH  )

static

Definition at line 245 of file compress_none.c.

{
    Assert(CFH->private_data == NULL);
 
    CFH->private_data = fopen(path, mode);
    if (CFH->private_data == NULL)
        return false;
 
    return true;
}

References Assert, fb(), and mode.

Referenced by InitCompressFileHandleNone().

read_none()

static size_t read_none ( void *  ptr, size_t  size, CompressFileHandle *  CFH  )

static

Definition at line 147 of file compress_none.c.

{
    FILE       *fp = (FILE *) CFH->private_data;
    size_t      ret;
 
    ret = fread(ptr, 1, size, fp);
    if (ferror(fp))
        pg_fatal("could not read from input file: %m");
 
    return ret;
}

References fb(), and pg_fatal.

Referenced by InitCompressFileHandleNone().

ReadDataFromArchiveNone()

static void ReadDataFromArchiveNone ( ArchiveHandle *  AH, CompressorState *  cs  )

static

Definition at line 42 of file compress_none.c.

{
    size_t      cnt;
    char       *buf;
    size_t      buflen;
 
    buflen = DEFAULT_IO_BUFFER_SIZE;
    buf = pg_malloc(buflen);
 
    while ((cnt = cs->readF(AH, &buf, &buflen)))
    {
        ahwrite(buf, 1, cnt, AH);
    }
 
    free(buf);
}

References ahwrite(), buf, DEFAULT_IO_BUFFER_SIZE, free, pg_malloc(), and CompressorState::readF.

Referenced by InitCompressorNone().

write_none()

static void write_none ( const void *  ptr, size_t  size, CompressFileHandle *  CFH  )

static

Definition at line 160 of file compress_none.c.

{
    size_t      ret;
 
    errno = 0;
    ret = fwrite(ptr, 1, size, (FILE *) CFH->private_data);
    if (ret != size)
    {
        errno = (errno) ? errno : ENOSPC;
        pg_fatal("could not write to file: %m");
    }
}

References fb(), and pg_fatal.

Referenced by InitCompressFileHandleNone().

WriteDataToArchiveNone()

static void WriteDataToArchiveNone ( ArchiveHandle *  AH, CompressorState *  cs, const void *  data, size_t  dLen  )

static

Definition at line 61 of file compress_none.c.

{
    NoneCompressorState *nonecs = (NoneCompressorState *) cs->private_data;
    size_t      remaining = dLen;
 
    while (remaining > 0)
    {
        size_t      chunk;
 
        /* Dump buffer if full */
        if (nonecs->bufdata >= nonecs->buflen)
        {
            cs->writeF(AH, nonecs->buffer, nonecs->bufdata);
            nonecs->bufdata = 0;
        }
        /* And fill it */
        chunk = nonecs->buflen - nonecs->bufdata;
        if (chunk > remaining)
            chunk = remaining;
        memcpy(nonecs->buffer + nonecs->bufdata, data, chunk);
        nonecs->bufdata += chunk;
        data = ((const char *) data) + chunk;
        remaining -= chunk;
    }
}

References data, fb(), CompressorState::private_data, remaining, and CompressorState::writeF.

Referenced by InitCompressorNone().