mirror of
https://github.com/postgrespro/pg_probackup.git
synced 2024-12-01 09:51:43 +02:00
85 lines
1.9 KiB
C
85 lines
1.9 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* fetch.c
|
|
* Functions for fetching files from PostgreSQL data directory
|
|
*
|
|
* Portions Copyright (c) 1996-2016, PostgreSQL Global Development Group
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
|
|
#include "postgres_fe.h"
|
|
|
|
#include "catalog/catalog.h"
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <dirent.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
#include "pg_arman.h"
|
|
|
|
/*
|
|
* Read a file into memory. The file to be read is <datadir>/<path>.
|
|
* The file contents are returned in a malloc'd buffer, and *filesize
|
|
* is set to the length of the file.
|
|
*
|
|
* The returned buffer is always zero-terminated; the size of the returned
|
|
* buffer is actually *filesize + 1. That's handy when reading a text file.
|
|
* This function can be used to read binary files as well, you can just
|
|
* ignore the zero-terminator in that case.
|
|
*
|
|
*/
|
|
char *
|
|
slurpFile(const char *datadir, const char *path, size_t *filesize, bool safe)
|
|
{
|
|
int fd;
|
|
char *buffer;
|
|
struct stat statbuf;
|
|
char fullpath[MAXPGPATH];
|
|
int len;
|
|
snprintf(fullpath, sizeof(fullpath), "%s/%s", datadir, path);
|
|
|
|
if ((fd = open(fullpath, O_RDONLY | PG_BINARY, 0)) == -1)
|
|
{
|
|
if (safe)
|
|
return NULL;
|
|
else
|
|
elog(ERROR, "could not open file \"%s\" for reading: %s",
|
|
fullpath, strerror(errno));
|
|
}
|
|
|
|
if (fstat(fd, &statbuf) < 0)
|
|
{
|
|
if (safe)
|
|
return NULL;
|
|
else
|
|
elog(ERROR, "could not open file \"%s\" for reading: %s",
|
|
fullpath, strerror(errno));
|
|
}
|
|
|
|
len = statbuf.st_size;
|
|
|
|
buffer = pg_malloc(len + 1);
|
|
|
|
if (read(fd, buffer, len) != len)
|
|
{
|
|
if (safe)
|
|
return NULL;
|
|
else
|
|
elog(ERROR, "could not read file \"%s\": %s\n",
|
|
fullpath, strerror(errno));
|
|
}
|
|
|
|
close(fd);
|
|
|
|
/* Zero-terminate the buffer. */
|
|
buffer[len] = '\0';
|
|
|
|
if (filesize)
|
|
*filesize = len;
|
|
return buffer;
|
|
}
|