A massively spiffy yet delicately unobtrusive compression library. (grpc依赖)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1197 lines
34 KiB

14 years ago
/* gzio.c -- IO on .gz files
14 years ago
* Copyright (C) 1995-2010 Jean-loup Gailly.
14 years ago
* For conditions of distribution and use, see copyright notice in zlib.h
14 years ago
*
14 years ago
* Compile this file with -DNO_GZCOMPRESS to avoid the compression code.
14 years ago
*/
14 years ago
/* @(#) $Id$ */
14 years ago
14 years ago
#ifdef OLD_GZIO
14 years ago
#ifdef _LARGEFILE64_SOURCE
# ifndef _LARGEFILE_SOURCE
# define _LARGEFILE_SOURCE
# endif
# ifdef _FILE_OFFSET_BITS
# undef _FILE_OFFSET_BITS
# endif
#endif
14 years ago
#include "zutil.h"
14 years ago
#include <stdio.h>
14 years ago
14 years ago
#ifdef NO_DEFLATE /* for compatibility with old definition */
14 years ago
# define NO_GZCOMPRESS
#endif
14 years ago
#ifndef NO_DUMMY_DECL
14 years ago
struct internal_state {int dummy;}; /* for buggy compilers */
14 years ago
#endif
14 years ago
14 years ago
#ifndef Z_BUFSIZE
# ifdef MAXSEG_64K
# define Z_BUFSIZE 4096 /* minimize memory usage for 16-bit DOS */
# else
# define Z_BUFSIZE 16384
# endif
#endif
#ifndef Z_PRINTF_BUFSIZE
# define Z_PRINTF_BUFSIZE 4096
#endif
14 years ago
14 years ago
#if defined UNDER_CE && defined NO_ERRNO_H
# include <windows.h>
/* Map the Windows error number in ERROR to a locale-dependent error
message string and return a pointer to it. Typically, the values
for ERROR come from GetLastError.
The string pointed to shall not be modified by the application,
but may be overwritten by a subsequent call to strwinerror
The strwinerror function does not change the current setting
of GetLastError. */
local char *strwinerror (error)
DWORD error;
{
static char buf[1024];
wchar_t *msgbuf;
DWORD lasterr = GetLastError();
DWORD chars = FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM
14 years ago
| FORMAT_MESSAGE_ALLOCATE_BUFFER,
NULL,
error,
0, /* Default language */
(LPVOID)&msgbuf,
0,
NULL);
14 years ago
if (chars != 0) {
/* If there is an \r\n appended, zap it. */
if (chars >= 2
&& msgbuf[chars - 2] == '\r' && msgbuf[chars - 1] == '\n') {
chars -= 2;
msgbuf[chars] = 0;
}
if (chars > sizeof (buf) - 1) {
chars = sizeof (buf) - 1;
msgbuf[chars] = 0;
}
wcstombs(buf, msgbuf, chars + 1);
LocalFree(msgbuf);
}
else {
sprintf(buf, "unknown win32 error (%ld)", error);
}
SetLastError(lasterr);
return buf;
}
#endif /* UNDER_CE && NO_ERRNO_H */
14 years ago
#ifdef __MVS__
14 years ago
#pragma map (fdopen , "\174\174FDOPEN")
14 years ago
FILE *fdopen(int, const char *);
#endif
14 years ago
#ifndef STDC
extern voidp malloc OF((uInt size));
extern void free OF((voidpf ptr));
#endif
14 years ago
#ifdef NO_FSEEKO
# define FSEEK fseek
# define FTELL ftell
#else
# define FSEEK fseeko
# define FTELL ftello
#endif
14 years ago
#define ALLOC(size) malloc(size)
#define TRYFREE(p) {if (p) free(p);}
14 years ago
14 years ago
static int const gz_magic[2] = {0x1f, 0x8b}; /* gzip magic header */
14 years ago
/* gzip flag byte */
#define ASCII_FLAG 0x01 /* bit 0 set: file probably ascii text */
#define HEAD_CRC 0x02 /* bit 1 set: header CRC present */
#define EXTRA_FIELD 0x04 /* bit 2 set: extra field present */
#define ORIG_NAME 0x08 /* bit 3 set: original file name present */
#define COMMENT 0x10 /* bit 4 set: file comment present */
#define RESERVED 0xE0 /* bits 5..7: reserved */
typedef struct gz_stream {
z_stream stream;
14 years ago
int z_err; /* error code for last stream operation */
int z_eof; /* set if end of input file */
FILE *file; /* .gz file */
Byte *inbuf; /* input buffer */
Byte *outbuf; /* output buffer */
uLong crc; /* crc32 of uncompressed data */
char *msg; /* error message */
char *path; /* path name for debugging only */
int transparent; /* 1 if input file is not a .gz file */
char mode; /* 'w' or 'r' */
z_off64_t start; /* start of compressed data in file */
z_off64_t in; /* bytes into deflate or inflate */
z_off64_t out; /* bytes out of deflate or inflate */
int back; /* one character push-back */
int last; /* true if push-back is last character */
14 years ago
} gz_stream;
14 years ago
local gzFile gz_open OF((const char *, const char *, int, int));
local z_off64_t gz_seek OF((gzFile, z_off64_t, int, int));
local int do_flush OF((gzFile, int));
local int get_byte OF((gz_stream *));
local void check_header OF((gz_stream *));
local int destroy OF((gz_stream *));
local void putLong OF((FILE *, uLong));
local uLong getLong OF((gz_stream *));
14 years ago
/* ===========================================================================
Opens a gzip (.gz) file for reading or writing. The mode parameter
14 years ago
is as in fopen ("rb" or "wb"). The file is given either by file descriptor
14 years ago
or path name (if fd == -1).
14 years ago
gz_open returns NULL if the file could not be opened or if there was
14 years ago
insufficient memory to allocate the (de)compression state; errno
can be checked to distinguish the two cases (if errno is zero, the
zlib error is Z_MEM_ERROR).
*/
14 years ago
local gzFile gz_open (path, mode, fd, use64)
14 years ago
const char *path;
const char *mode;
14 years ago
int fd;
14 years ago
int use64;
14 years ago
{
int err;
14 years ago
int level = Z_DEFAULT_COMPRESSION; /* compression level */
14 years ago
int strategy = Z_DEFAULT_STRATEGY; /* compression strategy */
14 years ago
char *p = (char*)mode;
gz_stream *s;
char fmode[80]; /* copy of mode, without the compression level */
char *m = fmode;
if (!path || !mode) return Z_NULL;
14 years ago
14 years ago
s = (gz_stream *)ALLOC(sizeof(gz_stream));
14 years ago
if (!s) return Z_NULL;
s->stream.zalloc = (alloc_func)0;
s->stream.zfree = (free_func)0;
14 years ago
s->stream.opaque = (voidpf)0;
14 years ago
s->stream.next_in = s->inbuf = Z_NULL;
s->stream.next_out = s->outbuf = Z_NULL;
s->stream.avail_in = s->stream.avail_out = 0;
14 years ago
s->stream.state = Z_NULL;
14 years ago
s->file = NULL;
s->z_err = Z_OK;
s->z_eof = 0;
14 years ago
s->in = 0;
s->out = 0;
14 years ago
s->back = EOF;
14 years ago
s->crc = crc32(0L, Z_NULL, 0);
s->msg = NULL;
s->transparent = 0;
s->path = (char*)ALLOC(strlen(path)+1);
if (s->path == NULL) {
14 years ago
return destroy(s), (gzFile)Z_NULL;
14 years ago
}
strcpy(s->path, path); /* do this early for debugging */
s->mode = '\0';
do {
14 years ago
if (*p == 'r') s->mode = 'r';
14 years ago
if (*p == 'w' || *p == 'a') s->mode = 'w';
if (*p >= '0' && *p <= '9') {
14 years ago
level = *p - '0';
} else if (*p == 'f') {
strategy = Z_FILTERED;
} else if (*p == 'h') {
strategy = Z_HUFFMAN_ONLY;
} else if (*p == 'R') {
strategy = Z_RLE;
} else {
*m++ = *p; /* copy the mode */
}
14 years ago
} while (*p++ && m != fmode + sizeof(fmode));
14 years ago
if (s->mode == '\0') return destroy(s), (gzFile)Z_NULL;
14 years ago
14 years ago
if (s->mode == 'w') {
14 years ago
#ifdef NO_GZCOMPRESS
14 years ago
err = Z_STREAM_ERROR;
#else
14 years ago
err = deflateInit2(&(s->stream), level,
14 years ago
Z_DEFLATED, -MAX_WBITS, DEF_MEM_LEVEL, strategy);
14 years ago
/* windowBits is passed < 0 to suppress zlib header */
14 years ago
14 years ago
s->stream.next_out = s->outbuf = (Byte*)ALLOC(Z_BUFSIZE);
14 years ago
#endif
14 years ago
if (err != Z_OK || s->outbuf == Z_NULL) {
return destroy(s), (gzFile)Z_NULL;
}
14 years ago
} else {
14 years ago
s->stream.next_in = s->inbuf = (Byte*)ALLOC(Z_BUFSIZE);
14 years ago
14 years ago
err = inflateInit2(&(s->stream), -MAX_WBITS);
14 years ago
/* windowBits is passed < 0 to tell that there is no zlib header */
14 years ago
if (err != Z_OK || s->inbuf == Z_NULL) {
return destroy(s), (gzFile)Z_NULL;
}
14 years ago
}
s->stream.avail_out = Z_BUFSIZE;
14 years ago
zseterrno(0);
s->file = fd == -1 ?
(use64 ? F_OPEN64(path, fmode) : F_OPEN(path, fmode)) :
(FILE*)fdopen(fd, fmode);
14 years ago
if (s->file == NULL) {
14 years ago
return destroy(s), (gzFile)Z_NULL;
14 years ago
}
if (s->mode == 'w') {
14 years ago
/* Write a very simple .gz header:
*/
14 years ago
fprintf(s->file, "%c%c%c%c%c%c%c%c%c%c", gz_magic[0], gz_magic[1],
14 years ago
Z_DEFLATED, 0 /*flags*/, 0,0,0,0 /*time*/, level == 9 ? 2 :
(strategy >= Z_HUFFMAN_ONLY ||
(level != Z_DEFAULT_COMPRESSION && level < 2) ?
4 : 0) /*xflags*/, OS_CODE);
14 years ago
s->start = 10L;
14 years ago
/* We use 10L instead of ftell(s->file) to because ftell causes an
14 years ago
* fflush on some systems. This version of the library doesn't use
14 years ago
* start anyway in write mode, so this initialization is not
14 years ago
* necessary.
*/
14 years ago
} else {
14 years ago
check_header(s); /* skip the .gz header */
14 years ago
s->start = FTELL(s->file) - s->stream.avail_in;
14 years ago
}
14 years ago
14 years ago
return (gzFile)s;
}
/* ===========================================================================
Opens a gzip (.gz) file for reading or writing.
*/
14 years ago
gzFile ZEXPORT gzopen (path, mode)
14 years ago
const char *path;
const char *mode;
14 years ago
{
14 years ago
return gz_open (path, mode, -1, 0);
}
/* ===========================================================================
Opens a gzip (.gz) file for reading or writing for 64-bit offsets
*/
gzFile ZEXPORT gzopen64 (path, mode)
const char *path;
const char *mode;
{
return gz_open (path, mode, -1, 1);
14 years ago
}
/* ===========================================================================
14 years ago
Associate a gzFile with the file descriptor fd. fd is not dup'ed here
to mimic the behavio(u)r of fdopen.
14 years ago
*/
14 years ago
gzFile ZEXPORT gzdopen (fd, mode)
14 years ago
int fd;
14 years ago
const char *mode;
14 years ago
{
14 years ago
char name[46]; /* allow for up to 128-bit integers */
14 years ago
14 years ago
if (fd == -1) return (gzFile)Z_NULL;
14 years ago
sprintf(name, "<fd:%d>", fd); /* for debugging */
14 years ago
14 years ago
return gz_open (name, mode, fd, 0);
14 years ago
}
14 years ago
/* ===========================================================================
* Update the compression level and strategy
*/
14 years ago
int ZEXPORT gzsetparams (file, level, strategy)
14 years ago
gzFile file;
int level;
int strategy;
{
gz_stream *s = (gz_stream*)file;
if (s == NULL || s->mode != 'w') return Z_STREAM_ERROR;
/* Make room to allow flushing */
if (s->stream.avail_out == 0) {
14 years ago
s->stream.next_out = s->outbuf;
if (fwrite(s->outbuf, 1, Z_BUFSIZE, s->file) != Z_BUFSIZE) {
s->z_err = Z_ERRNO;
}
s->stream.avail_out = Z_BUFSIZE;
14 years ago
}
return deflateParams (&(s->stream), level, strategy);
}
14 years ago
/* ===========================================================================
Read a byte from a gz_stream; update next_in and avail_in. Return EOF
for end of file.
14 years ago
IN assertion: the stream s has been successfully opened for reading.
14 years ago
*/
local int get_byte(s)
gz_stream *s;
{
if (s->z_eof) return EOF;
if (s->stream.avail_in == 0) {
14 years ago
zseterrno(0);
14 years ago
s->stream.avail_in = (uInt)fread(s->inbuf, 1, Z_BUFSIZE, s->file);
14 years ago
if (s->stream.avail_in == 0) {
s->z_eof = 1;
if (ferror(s->file)) s->z_err = Z_ERRNO;
return EOF;
}
s->stream.next_in = s->inbuf;
14 years ago
}
s->stream.avail_in--;
return *(s->stream.next_in)++;
}
/* ===========================================================================
Check the gzip header of a gz_stream opened for reading. Set the stream
mode to transparent if the gzip magic header is not present; set s->err
to Z_DATA_ERROR if the magic header is present but the rest of the header
is incorrect.
14 years ago
IN assertion: the stream s has already been created successfully;
14 years ago
s->stream.avail_in is zero for the first time, but may be non-zero
for concatenated .gz files.
*/
local void check_header(s)
gz_stream *s;
{
int method; /* method byte */
int flags; /* flags byte */
uInt len;
int c;
14 years ago
/* Assure two bytes in the buffer so we can peek ahead -- handle case
where first byte of header is at the end of the buffer after the last
gzip segment */
len = s->stream.avail_in;
if (len < 2) {
if (len) s->inbuf[0] = s->stream.next_in[0];
14 years ago
zseterrno(0);
14 years ago
len = (uInt)fread(s->inbuf + len, 1, Z_BUFSIZE >> len, s->file);
14 years ago
if (len == 0) s->z_eof = 1;
14 years ago
if (len == 0 && ferror(s->file)) s->z_err = Z_ERRNO;
s->stream.avail_in += len;
s->stream.next_in = s->inbuf;
if (s->stream.avail_in < 2) {
s->transparent = s->stream.avail_in;
return;
}
14 years ago
}
14 years ago
/* Peek ahead to check the gzip magic header */
if (s->stream.next_in[0] != gz_magic[0] ||
14 years ago
s->stream.next_in[1] != gz_magic[1]) {
14 years ago
s->transparent = 1;
return;
}
s->stream.avail_in -= 2;
s->stream.next_in += 2;
/* Check the rest of the gzip header */
14 years ago
method = get_byte(s);
flags = get_byte(s);
if (method != Z_DEFLATED || (flags & RESERVED) != 0) {
14 years ago
s->z_err = Z_DATA_ERROR;
return;
14 years ago
}
/* Discard time, xflags and OS code: */
for (len = 0; len < 6; len++) (void)get_byte(s);
if ((flags & EXTRA_FIELD) != 0) { /* skip the extra field */
14 years ago
len = (uInt)get_byte(s);
len += ((uInt)get_byte(s))<<8;
/* len is garbage if EOF but the loop below will quit anyway */
while (len-- != 0 && get_byte(s) != EOF) ;
14 years ago
}
if ((flags & ORIG_NAME) != 0) { /* skip the original file name */
14 years ago
while ((c = get_byte(s)) != 0 && c != EOF) ;
14 years ago
}
if ((flags & COMMENT) != 0) { /* skip the .gz file comment */
14 years ago
while ((c = get_byte(s)) != 0 && c != EOF) ;
14 years ago
}
if ((flags & HEAD_CRC) != 0) { /* skip the header crc */
14 years ago
for (len = 0; len < 2; len++) (void)get_byte(s);
14 years ago
}
s->z_err = s->z_eof ? Z_DATA_ERROR : Z_OK;
}
/* ===========================================================================
* Cleanup then free the given gz_stream. Return a zlib error code.
Try freeing in the reverse order of allocations.
*/
local int destroy (s)
gz_stream *s;
{
int err = Z_OK;
if (!s) return Z_STREAM_ERROR;
TRYFREE(s->msg);
if (s->stream.state != NULL) {
14 years ago
if (s->mode == 'w') {
14 years ago
#ifdef NO_GZCOMPRESS
14 years ago
err = Z_STREAM_ERROR;
14 years ago
#else
14 years ago
err = deflateEnd(&(s->stream));
14 years ago
#endif
14 years ago
} else if (s->mode == 'r') {
err = inflateEnd(&(s->stream));
}
14 years ago
}
if (s->file != NULL && fclose(s->file)) {
14 years ago
#ifdef ESPIPE
14 years ago
if (zerrno() != ESPIPE) /* fclose is broken for pipes in HP/UX */
14 years ago
#endif
14 years ago
err = Z_ERRNO;
14 years ago
}
if (s->z_err < 0) err = s->z_err;
TRYFREE(s->inbuf);
TRYFREE(s->outbuf);
TRYFREE(s->path);
TRYFREE(s);
return err;
}
14 years ago
/* ===========================================================================
Reads the given number of uncompressed bytes from the compressed file.
gzread returns the number of bytes actually read (0 for end of file).
*/
14 years ago
int ZEXPORT gzread (file, buf, len)
14 years ago
gzFile file;
14 years ago
voidp buf;
14 years ago
unsigned len;
{
gz_stream *s = (gz_stream*)file;
14 years ago
Bytef *start = (Bytef*)buf; /* starting point for crc computation */
14 years ago
Byte *next_out; /* == stream.next_out but not forced far (for MSDOS) */
14 years ago
if (s == NULL || s->mode != 'r') return Z_STREAM_ERROR;
14 years ago
if (s->z_err == Z_DATA_ERROR || s->z_err == Z_ERRNO) return -1;
14 years ago
if (s->z_err == Z_STREAM_END) return 0; /* EOF */
14 years ago
14 years ago
next_out = (Byte*)buf;
s->stream.next_out = (Bytef*)buf;
14 years ago
s->stream.avail_out = len;
14 years ago
if (s->stream.avail_out && s->back != EOF) {
*next_out++ = s->back;
s->stream.next_out++;
s->stream.avail_out--;
s->back = EOF;
14 years ago
s->out++;
14 years ago
start++;
14 years ago
if (s->last) {
s->z_err = Z_STREAM_END;
return 1;
}
}
14 years ago
while (s->stream.avail_out != 0) {
14 years ago
if (s->transparent) {
/* Copy first the lookahead bytes: */
uInt n = s->stream.avail_in;
if (n > s->stream.avail_out) n = s->stream.avail_out;
if (n > 0) {
zmemcpy(s->stream.next_out, s->stream.next_in, n);
next_out += n;
s->stream.next_out = next_out;
s->stream.next_in += n;
s->stream.avail_out -= n;
s->stream.avail_in -= n;
}
14 years ago
if (s->stream.avail_out > 0 && !feof(s->file)) {
14 years ago
s->stream.avail_out -=
(uInt)fread(next_out, 1, s->stream.avail_out, s->file);
14 years ago
}
len -= s->stream.avail_out;
14 years ago
s->in += len;
s->out += len;
14 years ago
if (len == 0 && feof(s->file)) s->z_eof = 1;
14 years ago
return (int)len;
}
14 years ago
if (s->stream.avail_in == 0 && !s->z_eof) {
14 years ago
zseterrno(0);
14 years ago
s->stream.avail_in = (uInt)fread(s->inbuf, 1, Z_BUFSIZE, s->file);
14 years ago
if (s->stream.avail_in == 0) {
s->z_eof = 1;
14 years ago
if (ferror(s->file)) {
s->z_err = Z_ERRNO;
break;
}
14 years ago
}
s->stream.next_in = s->inbuf;
}
14 years ago
s->in += s->stream.avail_in;
s->out += s->stream.avail_out;
14 years ago
s->z_err = inflate(&(s->stream), Z_NO_FLUSH);
14 years ago
s->in -= s->stream.avail_in;
s->out -= s->stream.avail_out;
14 years ago
14 years ago
if (s->z_err == Z_STREAM_END) {
/* Check CRC and original size */
s->crc = crc32(s->crc, start, (uInt)(s->stream.next_out - start));
start = s->stream.next_out;
14 years ago
14 years ago
if (getLong(s) != s->crc) {
s->z_err = Z_DATA_ERROR;
} else {
(void)getLong(s);
14 years ago
/* The uncompressed length returned by above getlong() may be
* different from s->out in case of concatenated .gz files.
* Check for such files:
14 years ago
*/
check_header(s);
if (s->z_err == Z_OK) {
inflateReset(&(s->stream));
s->crc = crc32(0L, Z_NULL, 0);
}
}
}
if (s->z_err != Z_OK || s->z_eof) break;
14 years ago
}
14 years ago
s->crc = crc32(s->crc, start, (uInt)(s->stream.next_out - start));
14 years ago
if (len == s->stream.avail_out &&
(s->z_err == Z_DATA_ERROR || s->z_err == Z_ERRNO))
return -1;
14 years ago
return (int)(len - s->stream.avail_out);
14 years ago
}
14 years ago
/* ===========================================================================
Reads one byte from the compressed file. gzgetc returns this byte
or -1 in case of end of file or error.
*/
14 years ago
int ZEXPORT gzgetc(file)
14 years ago
gzFile file;
{
14 years ago
unsigned char c;
14 years ago
return gzread(file, &c, 1) == 1 ? c : -1;
}
14 years ago
/* ===========================================================================
Push one byte back onto the stream.
*/
int ZEXPORT gzungetc(c, file)
int c;
gzFile file;
{
gz_stream *s = (gz_stream*)file;
if (s == NULL || s->mode != 'r' || c == EOF || s->back != EOF) return EOF;
s->back = c;
14 years ago
s->out--;
14 years ago
s->last = (s->z_err == Z_STREAM_END);
14 years ago
if (s->last) s->z_err = Z_OK;
s->z_eof = 0;
14 years ago
return c;
}
14 years ago
/* ===========================================================================
Reads bytes from the compressed file until len-1 characters are
read, or a newline character is read and transferred to buf, or an
end-of-file condition is encountered. The string is then terminated
with a null character.
gzgets returns buf, or Z_NULL in case of error.
The current implementation is not optimized at all.
*/
char * ZEXPORT gzgets(file, buf, len)
gzFile file;
char *buf;
int len;
{
char *b = buf;
if (buf == Z_NULL || len <= 0) return Z_NULL;
while (--len > 0 && gzread(file, buf, 1) == 1 && *buf++ != '\n') ;
*buf = '\0';
return b == buf && len > 0 ? Z_NULL : b;
}
14 years ago
#ifndef NO_GZCOMPRESS
14 years ago
/* ===========================================================================
Writes the given number of uncompressed bytes into the compressed file.
gzwrite returns the number of bytes actually written (0 in case of error).
*/
14 years ago
int ZEXPORT gzwrite (file, buf, len)
14 years ago
gzFile file;
14 years ago
voidpc buf;
14 years ago
unsigned len;
{
gz_stream *s = (gz_stream*)file;
if (s == NULL || s->mode != 'w') return Z_STREAM_ERROR;
14 years ago
s->stream.next_in = (Bytef*)buf;
14 years ago
s->stream.avail_in = len;
while (s->stream.avail_in != 0) {
14 years ago
if (s->stream.avail_out == 0) {
14 years ago
14 years ago
s->stream.next_out = s->outbuf;
if (fwrite(s->outbuf, 1, Z_BUFSIZE, s->file) != Z_BUFSIZE) {
s->z_err = Z_ERRNO;
break;
}
s->stream.avail_out = Z_BUFSIZE;
}
14 years ago
s->in += s->stream.avail_in;
s->out += s->stream.avail_out;
14 years ago
s->z_err = deflate(&(s->stream), Z_NO_FLUSH);
14 years ago
s->in -= s->stream.avail_in;
s->out -= s->stream.avail_out;
14 years ago
if (s->z_err != Z_OK) break;
14 years ago
}
14 years ago
s->crc = crc32(s->crc, (const Bytef *)buf, len);
14 years ago
14 years ago
return (int)(len - s->stream.avail_in);
14 years ago
}
14 years ago
14 years ago
/* ===========================================================================
Converts, formats, and writes the args to the compressed file under
control of the format string, as in fprintf. gzprintf returns the number of
uncompressed bytes actually written (0 in case of error).
*/
#ifdef STDC
#include <stdarg.h>
14 years ago
int ZEXPORTVA gzprintf (gzFile file, const char *format, /* args */ ...)
14 years ago
{
14 years ago
char buf[Z_PRINTF_BUFSIZE];
14 years ago
va_list va;
int len;
14 years ago
buf[sizeof(buf) - 1] = 0;
14 years ago
va_start(va, format);
14 years ago
#ifdef NO_vsnprintf
# ifdef HAS_vsprintf_void
14 years ago
(void)vsprintf(buf, format, va);
14 years ago
va_end(va);
14 years ago
for (len = 0; len < sizeof(buf); len++)
if (buf[len] == 0) break;
14 years ago
# else
len = vsprintf(buf, format, va);
va_end(va);
# endif
#else
# ifdef HAS_vsnprintf_void
(void)vsnprintf(buf, sizeof(buf), format, va);
va_end(va);
len = strlen(buf);
# else
len = vsnprintf(buf, sizeof(buf), format, va);
va_end(va);
# endif
#endif
14 years ago
if (len <= 0 || len >= (int)sizeof(buf) || buf[sizeof(buf) - 1] != 0)
14 years ago
return 0;
14 years ago
return gzwrite(file, buf, (unsigned)len);
}
#else /* not ANSI C */
14 years ago
int ZEXPORTVA gzprintf (file, format, a1, a2, a3, a4, a5, a6, a7, a8, a9, a10,
14 years ago
a11, a12, a13, a14, a15, a16, a17, a18, a19, a20)
14 years ago
gzFile file;
const char *format;
int a1, a2, a3, a4, a5, a6, a7, a8, a9, a10,
14 years ago
a11, a12, a13, a14, a15, a16, a17, a18, a19, a20;
14 years ago
{
14 years ago
char buf[Z_PRINTF_BUFSIZE];
14 years ago
int len;
14 years ago
buf[sizeof(buf) - 1] = 0;
14 years ago
#ifdef NO_snprintf
# ifdef HAS_sprintf_void
14 years ago
sprintf(buf, format, a1, a2, a3, a4, a5, a6, a7, a8,
14 years ago
a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20);
14 years ago
for (len = 0; len < sizeof(buf); len++)
if (buf[len] == 0) break;
14 years ago
# else
len = sprintf(buf, format, a1, a2, a3, a4, a5, a6, a7, a8,
14 years ago
a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20);
14 years ago
# endif
#else
# ifdef HAS_snprintf_void
snprintf(buf, sizeof(buf), format, a1, a2, a3, a4, a5, a6, a7, a8,
14 years ago
a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20);
14 years ago
len = strlen(buf);
# else
len = snprintf(buf, sizeof(buf), format, a1, a2, a3, a4, a5, a6, a7, a8,
14 years ago
a9, a10, a11, a12, a13, a14, a15, a16, a17, a18, a19, a20);
14 years ago
# endif
#endif
14 years ago
if (len <= 0 || len >= sizeof(buf) || buf[sizeof(buf) - 1] != 0)
return 0;
14 years ago
return gzwrite(file, buf, len);
}
#endif
/* ===========================================================================
Writes c, converted to an unsigned char, into the compressed file.
gzputc returns the value that was written, or -1 in case of error.
*/
14 years ago
int ZEXPORT gzputc(file, c)
14 years ago
gzFile file;
int c;
{
14 years ago
unsigned char cc = (unsigned char) c; /* required for big endian systems */
return gzwrite(file, &cc, 1) == 1 ? (int)cc : -1;
14 years ago
}
14 years ago
/* ===========================================================================
Writes the given null-terminated string to the compressed file, excluding
the terminating null character.
gzputs returns the number of characters written, or -1 in case of error.
*/
int ZEXPORT gzputs(file, s)
gzFile file;
const char *s;
{
14 years ago
return gzwrite(file, (char*)s, (unsigned)strlen(s));
14 years ago
}
14 years ago
/* ===========================================================================
Flushes all pending output into the compressed file. The parameter
flush is as in the deflate() function.
*/
14 years ago
local int do_flush (file, flush)
14 years ago
gzFile file;
int flush;
{
uInt len;
int done = 0;
gz_stream *s = (gz_stream*)file;
if (s == NULL || s->mode != 'w') return Z_STREAM_ERROR;
s->stream.avail_in = 0; /* should be zero already anyway */
for (;;) {
14 years ago
len = Z_BUFSIZE - s->stream.avail_out;
if (len != 0) {
14 years ago
if ((uInt)fwrite(s->outbuf, 1, len, s->file) != len) {
14 years ago
s->z_err = Z_ERRNO;
return Z_ERRNO;
}
s->stream.next_out = s->outbuf;
s->stream.avail_out = Z_BUFSIZE;
}
if (done) break;
14 years ago
s->out += s->stream.avail_out;
14 years ago
s->z_err = deflate(&(s->stream), flush);
14 years ago
s->out -= s->stream.avail_out;
14 years ago
14 years ago
/* Ignore the second of two consecutive flushes: */
if (len == 0 && s->z_err == Z_BUF_ERROR) s->z_err = Z_OK;
14 years ago
14 years ago
/* deflate has finished flushing only when it hasn't used up
14 years ago
* all the available space in the output buffer:
14 years ago
*/
14 years ago
done = (s->stream.avail_out != 0 || s->z_err == Z_STREAM_END);
14 years ago
14 years ago
if (s->z_err != Z_OK && s->z_err != Z_STREAM_END) break;
14 years ago
}
14 years ago
return s->z_err == Z_STREAM_END ? Z_OK : s->z_err;
}
14 years ago
int ZEXPORT gzflush (file, flush)
14 years ago
gzFile file;
int flush;
{
gz_stream *s = (gz_stream*)file;
int err = do_flush (file, flush);
if (err) return err;
14 years ago
fflush(s->file);
14 years ago
return s->z_err == Z_STREAM_END ? Z_OK : s->z_err;
14 years ago
}
14 years ago
#endif /* NO_GZCOMPRESS */
14 years ago
/* ===========================================================================
Sets the starting position for the next gzread or gzwrite on the given
compressed file. The offset represents a number of bytes in the
gzseek returns the resulting offset location as measured in bytes from
the beginning of the uncompressed stream, or -1 in case of error.
SEEK_END is not implemented, returns error.
In this version of the library, gzseek can be extremely slow.
*/
14 years ago
local z_off64_t gz_seek (file, offset, whence, use64)
14 years ago
gzFile file;
14 years ago
z_off64_t offset;
14 years ago
int whence;
14 years ago
int use64;
14 years ago
{
gz_stream *s = (gz_stream*)file;
14 years ago
if (s == NULL || whence == SEEK_END ||
14 years ago
s->z_err == Z_ERRNO || s->z_err == Z_DATA_ERROR) {
return -1L;
14 years ago
}
14 years ago
14 years ago
if (s->mode == 'w') {
14 years ago
#ifdef NO_GZCOMPRESS
14 years ago
return -1L;
14 years ago
#else
14 years ago
if (whence == SEEK_SET) {
14 years ago
offset -= s->in;
14 years ago
}
if (offset < 0) return -1L;
/* At this point, offset is the number of zero bytes to write. */
if (s->inbuf == Z_NULL) {
s->inbuf = (Byte*)ALLOC(Z_BUFSIZE); /* for seeking */
14 years ago
if (s->inbuf == Z_NULL) return -1L;
14 years ago
zmemzero(s->inbuf, Z_BUFSIZE);
}
while (offset > 0) {
uInt size = Z_BUFSIZE;
if (offset < Z_BUFSIZE) size = (uInt)offset;
size = gzwrite(file, s->inbuf, size);
if (size == 0) return -1L;
offset -= size;
}
14 years ago
return s->in;
14 years ago
#endif
}
/* Rest of function is for reading only */
/* compute absolute position */
if (whence == SEEK_CUR) {
14 years ago
offset += s->out;
14 years ago
}
if (offset < 0) return -1L;
if (s->transparent) {
14 years ago
/* map to fseek */
14 years ago
s->back = EOF;
14 years ago
s->stream.avail_in = 0;
s->stream.next_in = s->inbuf;
14 years ago
#ifdef _LARGEFILE64_SOURCE
if ((use64 ? fseeko64(s->file, offset, SEEK_SET) :
FSEEK(s->file, offset, SEEK_SET)) < 0)
return -1L;
#else
if (FSEEK(s->file, offset, SEEK_SET) < 0) return -1L;
#endif
14 years ago
14 years ago
s->in = s->out = offset;
14 years ago
return offset;
14 years ago
}
/* For a negative seek, rewind and use positive seek */
14 years ago
if (offset >= s->out) {
offset -= s->out;
14 years ago
} else if (gzrewind(file) < 0) {
14 years ago
return -1L;
14 years ago
}
/* offset is now the number of bytes to skip. */
if (offset != 0 && s->outbuf == Z_NULL) {
14 years ago
s->outbuf = (Byte*)ALLOC(Z_BUFSIZE);
14 years ago
if (s->outbuf == Z_NULL) return -1L;
14 years ago
}
14 years ago
if (offset && s->back != EOF) {
14 years ago
s->back = EOF;
14 years ago
s->out++;
14 years ago
offset--;
if (s->last) s->z_err = Z_STREAM_END;
}
14 years ago
while (offset > 0) {
14 years ago
int size = Z_BUFSIZE;
if (offset < Z_BUFSIZE) size = (int)offset;
14 years ago
14 years ago
size = gzread(file, s->outbuf, (uInt)size);
if (size <= 0) return -1L;
offset -= size;
14 years ago
}
14 years ago
return s->out;
14 years ago
}
14 years ago
/* ===========================================================================
Define external functions gzseek() and gzseek64() using local gz_seek().
*/
z_off_t ZEXPORT gzseek (file, offset, whence)
gzFile file;
z_off_t offset;
int whence;
{
return (z_off_t)gz_seek(file, offset, whence, 0);
}
14 years ago
z_off64_t ZEXPORT gzseek64 (file, offset, whence)
14 years ago
gzFile file;
14 years ago
z_off64_t offset;
14 years ago
int whence;
{
14 years ago
#ifdef _LARGEFILE64_SOURCE
14 years ago
return gz_seek(file, offset, whence, 1);
#else
return gz_seek(file, offset, whence, 0);
#endif
14 years ago
}
14 years ago
14 years ago
/* ===========================================================================
14 years ago
Rewinds input file.
14 years ago
*/
14 years ago
int ZEXPORT gzrewind (file)
14 years ago
gzFile file;
{
gz_stream *s = (gz_stream*)file;
14 years ago
14 years ago
if (s == NULL || s->mode != 'r') return -1;
s->z_err = Z_OK;
s->z_eof = 0;
14 years ago
s->back = EOF;
14 years ago
s->stream.avail_in = 0;
s->stream.next_in = s->inbuf;
14 years ago
s->crc = crc32(0L, Z_NULL, 0);
14 years ago
if (!s->transparent) (void)inflateReset(&s->stream);
s->in = 0;
s->out = 0;
14 years ago
return FSEEK(s->file, s->start, SEEK_SET);
14 years ago
}
/* ===========================================================================
Returns the starting position for the next gzread or gzwrite on the
given compressed file. This position represents a number of bytes in the
uncompressed data stream.
*/
14 years ago
z_off_t ZEXPORT gztell (file)
14 years ago
gzFile file;
{
return gzseek(file, 0L, SEEK_CUR);
}
14 years ago
/* ===========================================================================
64-bit version
*/
14 years ago
z_off64_t ZEXPORT gztell64 (file)
14 years ago
gzFile file;
{
return gzseek64(file, 0L, SEEK_CUR);
}
14 years ago
/* ===========================================================================
Returns 1 when EOF has previously been detected reading the given
input stream, otherwise zero.
*/
14 years ago
int ZEXPORT gzeof (file)
14 years ago
gzFile file;
{
gz_stream *s = (gz_stream*)file;
14 years ago
14 years ago
/* With concatenated compressed files that can have embedded
* crc trailers, z_eof is no longer the only/best indicator of EOF
* on a gz_stream. Handle end-of-stream error explicitly here.
*/
if (s == NULL || s->mode != 'r') return 0;
if (s->z_eof) return 1;
return s->z_err == Z_STREAM_END;
14 years ago
}
14 years ago
14 years ago
/* ===========================================================================
Returns 1 if reading and doing so transparently, otherwise zero.
*/
int ZEXPORT gzdirect (file)
gzFile file;
{
gz_stream *s = (gz_stream*)file;
if (s == NULL || s->mode != 'r') return 0;
return s->transparent;
}
14 years ago
/* ===========================================================================
Outputs a long in LSB order to the given file
*/
local void putLong (file, x)
FILE *file;
uLong x;
{
int n;
for (n = 0; n < 4; n++) {
14 years ago
fputc((int)(x & 0xff), file);
x >>= 8;
14 years ago
}
}
/* ===========================================================================
14 years ago
Reads a long in LSB order from the given gz_stream. Sets z_err in case
of error.
14 years ago
*/
14 years ago
local uLong getLong (s)
gz_stream *s;
14 years ago
{
14 years ago
uLong x = (uLong)get_byte(s);
int c;
x += ((uLong)get_byte(s))<<8;
x += ((uLong)get_byte(s))<<16;
c = get_byte(s);
if (c == EOF) s->z_err = Z_DATA_ERROR;
x += ((uLong)c)<<24;
14 years ago
return x;
}
/* ===========================================================================
Flushes all pending output if necessary, closes the compressed file
and deallocates all the (de)compression state.
*/
14 years ago
int ZEXPORT gzclose (file)
14 years ago
gzFile file;
{
gz_stream *s = (gz_stream*)file;
if (s == NULL) return Z_STREAM_ERROR;
if (s->mode == 'w') {
14 years ago
#ifdef NO_GZCOMPRESS
14 years ago
return Z_STREAM_ERROR;
14 years ago
#else
14 years ago
if (do_flush (file, Z_FINISH) != Z_OK)
return destroy((gz_stream*)file);
14 years ago
14 years ago
putLong (s->file, s->crc);
14 years ago
putLong (s->file, (uLong)(s->in & 0xffffffff));
14 years ago
#endif
14 years ago
}
14 years ago
return destroy((gz_stream*)file);
14 years ago
}
14 years ago
#if defined UNDER_CE && defined NO_ERRNO_H
# define zstrerror(errnum) strwinerror((DWORD)errnum)
14 years ago
#else
14 years ago
# if defined (STDC)
# define zstrerror(errnum) strerror(errnum)
# else
# define zstrerror(errnum) ""
# endif
14 years ago
#endif
14 years ago
/* ===========================================================================
14 years ago
Returns the error message for the last error which occurred on the
14 years ago
given compressed file. errnum is set to zlib error number. If an
14 years ago
error occurred in the file system and not in the compression library,
14 years ago
errnum is set to Z_ERRNO and the application may consult errno
to get the exact error code.
*/
14 years ago
const char * ZEXPORT gzerror (file, errnum)
14 years ago
gzFile file;
int *errnum;
{
char *m;
gz_stream *s = (gz_stream*)file;
if (s == NULL) {
14 years ago
*errnum = Z_STREAM_ERROR;
14 years ago
return (const char*)ERR_MSG(Z_STREAM_ERROR);
14 years ago
}
*errnum = s->z_err;
14 years ago
if (*errnum == Z_OK) return (const char*)"";
14 years ago
14 years ago
m = (char*)(*errnum == Z_ERRNO ? zstrerror(zerrno()) : s->stream.msg);
14 years ago
14 years ago
if (m == NULL || *m == '\0') m = (char*)ERR_MSG(s->z_err);
14 years ago
TRYFREE(s->msg);
s->msg = (char*)ALLOC(strlen(s->path) + strlen(m) + 3);
14 years ago
if (s->msg == Z_NULL) return (const char*)ERR_MSG(Z_MEM_ERROR);
14 years ago
strcpy(s->msg, s->path);
strcat(s->msg, ": ");
strcat(s->msg, m);
14 years ago
return (const char*)s->msg;
14 years ago
}
14 years ago
/* ===========================================================================
Clear the error and end-of-file flags, and do the same for the real file.
*/
void ZEXPORT gzclearerr (file)
gzFile file;
{
gz_stream *s = (gz_stream*)file;
if (s == NULL) return;
if (s->z_err != Z_STREAM_END) s->z_err = Z_OK;
s->z_eof = 0;
clearerr(s->file);
}
14 years ago
/* new functions in gzlib, but only partially implemented here */
int ZEXPORT gzbuffer (file, size)
gzFile file;
unsigned size;
{
return -1;
}
z_off_t ZEXPORT gzoffset (file)
gzFile file;
{
return -1;
}
z_off64_t ZEXPORT gzoffset64 (file)
gzFile file;
{
return -1;
}
int ZEXPORT gzclose_r (file)
gzFile file;
{
return gzclose(file);
}
int ZEXPORT gzclose_w (file)
gzFile file;
{
return gzclose(file);
}
int gzio_old = 1;
#else /* !OLD_GZIO */
int gzio_old = 0;
#endif /* OLD_GZIO */