1997-04-29 12:38:09 -03:00
/* zlibmodule.c -- gzip-compatible data compression */
1997-06-03 19:21:47 -03:00
# include "Python.h"
# include "zlib.h"
1997-04-29 12:38:09 -03:00
/* The following parameters are copied from zutil.h, version 0.95 */
# define DEFLATED 8
# if MAX_MEM_LEVEL >= 8
# define DEF_MEM_LEVEL 8
# else
# define DEF_MEM_LEVEL MAX_MEM_LEVEL
# endif
# define DEF_WBITS MAX_WBITS
/* The output buffer will be increased in chunks of ADDCHUNK bytes. */
1997-08-13 20:19:55 -03:00
# define DEFAULTALLOC 16*1024
1997-04-29 12:38:09 -03:00
# define PyInit_zlib initzlib
staticforward PyTypeObject Comptype ;
staticforward PyTypeObject Decomptype ;
static PyObject * ZlibError ;
typedef struct
{
PyObject_HEAD
z_stream zst ;
} compobject ;
1997-06-03 19:21:03 -03:00
static char compressobj__doc__ [ ] =
" compressobj() -- Return a compressor object. \n "
" compressobj(level) -- Return a compressor object, using the given compression level. \n "
;
static char decompressobj__doc__ [ ] =
" decompressobj() -- Return a decompressor object. \n "
" decompressobj(wbits) -- Return a decompressor object, setting the window buffer size to wbits. \n "
;
1997-04-29 12:38:09 -03:00
static compobject *
newcompobject ( type )
PyTypeObject * type ;
{
compobject * self ;
self = PyObject_NEW ( compobject , type ) ;
if ( self = = NULL )
return NULL ;
return self ;
}
1997-06-03 19:21:03 -03:00
static char compress__doc__ [ ] =
" compress(string) -- Compress string using the default compression level, "
" returning a string containing compressed data. \n "
" compress(string, level) -- Compress string, using the chosen compression "
" level (from 1 to 9). Return a string containing the compressed data. \n "
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_compress ( self , args )
PyObject * self ;
PyObject * args ;
{
PyObject * ReturnVal ;
Byte * input , * output ;
int length , level = Z_DEFAULT_COMPRESSION , err ;
z_stream zst ;
if ( ! PyArg_ParseTuple ( args , " s#|i " , & input , & length , & level ) )
return NULL ;
1997-09-04 20:39:23 -03:00
zst . avail_out = length + length / 1000 + 12 + 1 ;
1997-04-29 12:38:09 -03:00
output = ( Byte * ) malloc ( zst . avail_out ) ;
if ( output = = NULL )
{
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to compress data " ) ;
return NULL ;
}
1997-06-03 19:21:47 -03:00
zst . zalloc = ( alloc_func ) NULL ;
zst . zfree = ( free_func ) Z_NULL ;
1997-04-29 12:38:09 -03:00
zst . next_out = ( Byte * ) output ;
zst . next_in = ( Byte * ) input ;
zst . avail_in = length ;
err = deflateInit ( & zst , level ) ;
switch ( err )
{
case ( Z_OK ) :
break ;
case ( Z_MEM_ERROR ) :
PyErr_SetString ( PyExc_MemoryError ,
" Out of memory while compressing data " ) ;
free ( output ) ;
return NULL ;
break ;
case ( Z_STREAM_ERROR ) :
PyErr_SetString ( ZlibError ,
" Bad compression level " ) ;
free ( output ) ;
return NULL ;
break ;
default :
{
1997-09-04 20:39:23 -03:00
if ( zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while compressing data " ,
err ) ;
else
PyErr_Format ( ZlibError , " Error %i while compressing data: %.200s " ,
err , zst . msg ) ;
1997-04-29 12:38:09 -03:00
deflateEnd ( & zst ) ;
free ( output ) ;
return NULL ;
}
break ;
}
err = deflate ( & zst , Z_FINISH ) ;
switch ( err )
1997-09-04 20:39:23 -03:00
{
1997-04-29 12:38:09 -03:00
case ( Z_STREAM_END ) :
break ;
/* Are there other errors to be trapped here? */
default :
1997-09-04 20:39:23 -03:00
{
if ( zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while compressing data " ,
err ) ;
else
PyErr_Format ( ZlibError , " Error %i while compressing data: %.200s " ,
err , zst . msg ) ;
deflateEnd ( & zst ) ;
free ( output ) ;
return NULL ;
}
}
1997-04-29 12:38:09 -03:00
err = deflateEnd ( & zst ) ;
if ( err ! = Z_OK )
1997-09-04 20:39:23 -03:00
{
if ( zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while finishing compression " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while finishing compression: %.200s " ,
err , zst . msg ) ;
1997-04-29 12:38:09 -03:00
free ( output ) ;
return NULL ;
}
1997-06-03 19:21:47 -03:00
ReturnVal = PyString_FromStringAndSize ( ( char * ) output , zst . total_out ) ;
1997-04-29 12:38:09 -03:00
free ( output ) ;
return ReturnVal ;
}
1997-06-03 19:21:03 -03:00
static char decompress__doc__ [ ] =
1997-08-13 20:19:55 -03:00
" decompress(string) -- Decompress the data in string, returning a string containing the decompressed data. \n "
" decompress(string, wbits) -- Decompress the data in string with a window buffer size of wbits. \n "
" decompress(string, wbits, bufsize) -- Decompress the data in string with a window buffer size of wbits and an initial output buffer size of bufsize. \n "
1997-06-03 19:21:03 -03:00
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_decompress ( self , args )
PyObject * self ;
PyObject * args ;
{
1997-08-13 20:19:55 -03:00
PyObject * result_str ;
Byte * input ;
1997-04-29 12:38:09 -03:00
int length , err ;
1997-08-13 20:19:55 -03:00
int wsize = DEF_WBITS , r_strlen = DEFAULTALLOC ;
1997-04-29 12:38:09 -03:00
z_stream zst ;
1997-08-13 20:19:55 -03:00
if ( ! PyArg_ParseTuple ( args , " s#|ii " , & input , & length , & wsize , & r_strlen ) )
1997-04-29 12:38:09 -03:00
return NULL ;
1997-09-04 20:39:23 -03:00
if ( r_strlen < = 0 )
r_strlen = 1 ;
1997-04-29 12:38:09 -03:00
zst . avail_in = length ;
1997-08-13 20:19:55 -03:00
zst . avail_out = r_strlen ;
if ( ! ( result_str = PyString_FromStringAndSize ( NULL , r_strlen ) ) )
{
1997-04-29 12:38:09 -03:00
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to decompress data " ) ;
return NULL ;
1997-08-13 20:19:55 -03:00
}
1997-06-03 19:21:47 -03:00
zst . zalloc = ( alloc_func ) NULL ;
zst . zfree = ( free_func ) Z_NULL ;
1997-08-13 20:19:55 -03:00
zst . next_out = ( Byte * ) PyString_AsString ( result_str ) ;
1997-04-29 12:38:09 -03:00
zst . next_in = ( Byte * ) input ;
1997-08-13 20:19:55 -03:00
err = inflateInit2 ( & zst , wsize ) ;
1997-04-29 12:38:09 -03:00
switch ( err )
{
case ( Z_OK ) :
break ;
case ( Z_MEM_ERROR ) :
PyErr_SetString ( PyExc_MemoryError ,
" Out of memory while decompressing data " ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( result_str ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
default :
{
1997-09-04 20:39:23 -03:00
if ( zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i preparing to decompress data " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while preparing to decompress data: %.200s " ,
err , zst . msg ) ;
1997-04-29 12:38:09 -03:00
inflateEnd ( & zst ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( result_str ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
}
}
do
{
err = inflate ( & zst , Z_FINISH ) ;
switch ( err )
{
case ( Z_STREAM_END ) :
1997-08-13 20:19:55 -03:00
break ;
case ( Z_OK ) :
/* need more memory */
if ( _PyString_Resize ( & result_str , r_strlen < < 1 ) = = - 1 )
1997-04-29 12:38:09 -03:00
{
PyErr_SetString ( PyExc_MemoryError ,
" Out of memory while decompressing data " ) ;
inflateEnd ( & zst ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( result_str ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
}
1997-08-18 12:31:24 -03:00
zst . next_out = ( unsigned char * ) PyString_AsString ( result_str ) + r_strlen ;
1997-08-13 20:19:55 -03:00
zst . avail_out = r_strlen ;
r_strlen = r_strlen < < 1 ;
break ;
1997-04-29 12:38:09 -03:00
default :
{
1997-09-04 20:39:23 -03:00
if ( zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while decompressing data " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while decompressing data: %.200s " ,
err , zst . msg ) ;
1997-04-29 12:38:09 -03:00
inflateEnd ( & zst ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( result_str ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
}
}
} while ( err ! = Z_STREAM_END ) ;
err = inflateEnd ( & zst ) ;
if ( err ! = Z_OK )
1997-09-04 20:39:23 -03:00
{
if ( zst . msg = = Z_NULL )
PyErr_Format ( ZlibError ,
" Error %i while finishing data decompression " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while finishing data decompression: %.200s " ,
err , zst . msg ) ;
Py_DECREF ( result_str ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
}
1997-08-13 20:19:55 -03:00
_PyString_Resize ( & result_str , zst . total_out ) ;
return result_str ;
1997-04-29 12:38:09 -03:00
}
static PyObject *
PyZlib_compressobj ( selfptr , args )
PyObject * selfptr ;
PyObject * args ;
{
compobject * self ;
int level = Z_DEFAULT_COMPRESSION , method = DEFLATED ;
int wbits = MAX_WBITS , memLevel = DEF_MEM_LEVEL , strategy = 0 , err ;
1997-08-13 20:19:55 -03:00
if ( ! PyArg_ParseTuple ( args , " |iiiii " , & level , & method , & wbits ,
& memLevel , & strategy ) )
return NULL ;
1997-09-04 20:39:23 -03:00
self = newcompobject ( & Comptype ) ;
1997-04-29 12:38:09 -03:00
if ( self = = NULL ) return ( NULL ) ;
1997-09-04 20:39:23 -03:00
self - > zst . zalloc = ( alloc_func ) NULL ;
self - > zst . zfree = ( free_func ) Z_NULL ;
err = deflateInit2 ( & self - > zst , level , method , wbits , memLevel , strategy ) ;
1997-04-29 12:38:09 -03:00
switch ( err )
{
case ( Z_OK ) :
return ( PyObject * ) self ;
break ;
case ( Z_MEM_ERROR ) :
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory for compression object " ) ;
return NULL ;
break ;
case ( Z_STREAM_ERROR ) :
PyErr_SetString ( PyExc_ValueError ,
1997-09-04 20:39:23 -03:00
" Invalid initialization option " ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
break ;
default :
{
1997-09-04 20:39:23 -03:00
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError ,
" Error %i while creating compression object " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while creating compression object: %.200s " ,
err , self - > zst . msg ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
break ;
}
}
}
static PyObject *
PyZlib_decompressobj ( selfptr , args )
PyObject * selfptr ;
PyObject * args ;
{
int wbits = DEF_WBITS , err ;
compobject * self ;
if ( ! PyArg_ParseTuple ( args , " |i " , & wbits ) )
{
return NULL ;
}
self = newcompobject ( & Decomptype ) ;
if ( self = = NULL ) return ( NULL ) ;
1997-06-03 19:21:47 -03:00
self - > zst . zalloc = ( alloc_func ) NULL ;
self - > zst . zfree = ( free_func ) Z_NULL ;
1997-04-29 12:38:09 -03:00
err = inflateInit2 ( & self - > zst , wbits ) ;
switch ( err )
1997-09-04 20:39:23 -03:00
{
1997-04-29 12:38:09 -03:00
case ( Z_OK ) :
return ( PyObject * ) self ;
break ;
1997-09-04 20:39:23 -03:00
case ( Z_STREAM_ERROR ) :
PyErr_SetString ( PyExc_ValueError ,
" Invalid initialization option " ) ;
return NULL ;
break ;
1997-04-29 12:38:09 -03:00
case ( Z_MEM_ERROR ) :
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory for decompression object " ) ;
return NULL ;
break ;
default :
1997-09-04 20:39:23 -03:00
{
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError ,
" Error %i while creating decompression object " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while creating decompression object: %.200s " ,
err , self - > zst . msg ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
break ;
}
1997-09-04 20:39:23 -03:00
}
1997-04-29 12:38:09 -03:00
}
static void
Comp_dealloc ( self )
compobject * self ;
{
1997-09-04 20:39:23 -03:00
deflateEnd ( & self - > zst ) ;
PyMem_DEL ( self ) ;
1997-04-29 12:38:09 -03:00
}
static void
Decomp_dealloc ( self )
compobject * self ;
{
1997-09-04 20:39:23 -03:00
inflateEnd ( & self - > zst ) ;
PyMem_DEL ( self ) ;
1997-04-29 12:38:09 -03:00
}
1997-06-03 19:21:03 -03:00
static char comp_compress__doc__ [ ] =
" compress(data) -- Return a string containing a compressed version of the data. \n \n "
" After calling this function, some of the input data may still \n "
" be stored in internal buffers for later processing. \n "
" Call the flush() method to clear these buffers. "
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_objcompress ( self , args )
compobject * self ;
PyObject * args ;
{
1997-08-14 18:06:42 -03:00
int err = Z_OK , inplen ;
int length = DEFAULTALLOC ;
1997-04-29 12:38:09 -03:00
PyObject * RetVal ;
Byte * input ;
1997-09-04 20:39:23 -03:00
unsigned long start_total_out ;
1997-04-29 12:38:09 -03:00
if ( ! PyArg_ParseTuple ( args , " s# " , & input , & inplen ) )
1997-08-14 18:06:42 -03:00
return NULL ;
1997-09-04 20:39:23 -03:00
self - > zst . avail_in = inplen ;
self - > zst . next_in = input ;
1997-08-14 18:06:42 -03:00
if ( ! ( RetVal = PyString_FromStringAndSize ( NULL , length ) ) ) {
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to compress data " ) ;
return NULL ;
}
1997-09-04 20:39:23 -03:00
start_total_out = self - > zst . total_out ;
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) ;
1997-08-14 18:06:42 -03:00
self - > zst . avail_out = length ;
while ( self - > zst . avail_in ! = 0 & & err = = Z_OK )
{
err = deflate ( & ( self - > zst ) , Z_NO_FLUSH ) ;
if ( self - > zst . avail_out < = 0 ) {
if ( _PyString_Resize ( & RetVal , length < < 1 ) = = - 1 ) {
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to compress data " ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( RetVal ) ;
1997-08-14 18:06:42 -03:00
return NULL ;
}
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) + length ;
1997-08-14 18:06:42 -03:00
self - > zst . avail_out = length ;
length = length < < 1 ;
}
}
if ( err ! = Z_OK )
1997-09-04 20:39:23 -03:00
{
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while compressing " ,
err ) ;
else
PyErr_Format ( ZlibError , " Error %i while compressing: %.200s " ,
err , self - > zst . msg ) ;
Py_DECREF ( RetVal ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
}
1997-09-04 20:39:23 -03:00
_PyString_Resize ( & RetVal , self - > zst . total_out - start_total_out ) ;
1997-04-29 12:38:09 -03:00
return RetVal ;
}
1997-06-03 19:21:03 -03:00
static char decomp_decompress__doc__ [ ] =
" decompress(data) -- Return a string containing the decompressed version of the data. \n \n "
" After calling this function, some of the input data may still \n "
" be stored in internal buffers for later processing. \n "
" Call the flush() method to clear these buffers. "
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_objdecompress ( self , args )
compobject * self ;
PyObject * args ;
{
1997-08-13 20:19:55 -03:00
int length , err , inplen ;
1997-04-29 12:38:09 -03:00
PyObject * RetVal ;
Byte * input ;
1997-09-04 20:39:23 -03:00
unsigned long start_total_out ;
1997-04-29 12:38:09 -03:00
if ( ! PyArg_ParseTuple ( args , " s# " , & input , & inplen ) )
return NULL ;
1997-09-04 20:39:23 -03:00
start_total_out = self - > zst . total_out ;
1997-08-13 20:19:55 -03:00
RetVal = PyString_FromStringAndSize ( NULL , DEFAULTALLOC ) ;
1997-09-04 20:39:23 -03:00
self - > zst . avail_in = inplen ;
self - > zst . next_in = input ;
1997-08-13 20:19:55 -03:00
self - > zst . avail_out = length = DEFAULTALLOC ;
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) ;
1997-08-13 20:19:55 -03:00
err = Z_OK ;
while ( self - > zst . avail_in ! = 0 & & err = = Z_OK )
{
err = inflate ( & ( self - > zst ) , Z_NO_FLUSH ) ;
if ( err = = Z_OK & & self - > zst . avail_out < = 0 )
{
if ( _PyString_Resize ( & RetVal , length < < 1 ) = = - 1 )
{
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to compress data " ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( RetVal ) ;
1997-08-13 20:19:55 -03:00
return NULL ;
}
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) + length ;
1997-08-13 20:19:55 -03:00
self - > zst . avail_out = length ;
length = length < < 1 ;
}
}
1997-08-13 18:39:18 -03:00
1997-09-04 20:39:23 -03:00
if ( err ! = Z_OK & & err ! = Z_STREAM_END )
{
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while decompressing " ,
err ) ;
else
PyErr_Format ( ZlibError , " Error %i while decompressing: %.200s " ,
err , self - > zst . msg ) ;
Py_DECREF ( RetVal ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
}
1997-09-04 20:39:23 -03:00
_PyString_Resize ( & RetVal , self - > zst . total_out - start_total_out ) ;
1997-04-29 12:38:09 -03:00
return RetVal ;
}
1997-06-03 19:21:03 -03:00
static char comp_flush__doc__ [ ] =
" flush() -- Return a string containing any remaining compressed data. "
" The compressor object can no longer be used after this call. "
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_flush ( self , args )
compobject * self ;
PyObject * args ;
{
1997-08-14 18:06:42 -03:00
int length = DEFAULTALLOC , err = Z_OK ;
1997-04-29 12:38:09 -03:00
PyObject * RetVal ;
if ( ! PyArg_NoArgs ( args ) )
1997-08-14 18:06:42 -03:00
return NULL ;
self - > zst . avail_in = 0 ;
self - > zst . next_in = Z_NULL ;
if ( ! ( RetVal = PyString_FromStringAndSize ( NULL , length ) ) ) {
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to compress data " ) ;
return NULL ;
}
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) ;
1997-08-14 18:06:42 -03:00
self - > zst . avail_out = length ;
while ( err = = Z_OK )
{
err = deflate ( & ( self - > zst ) , Z_FINISH ) ;
if ( self - > zst . avail_out < = 0 ) {
if ( _PyString_Resize ( & RetVal , length < < 1 ) = = - 1 ) {
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to compress data " ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( RetVal ) ;
1997-08-14 18:06:42 -03:00
return NULL ;
}
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) + length ;
1997-08-14 18:06:42 -03:00
self - > zst . avail_out = length ;
length = length < < 1 ;
}
}
if ( err ! = Z_STREAM_END ) {
1997-09-04 20:39:23 -03:00
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while compressing " ,
err ) ;
else
PyErr_Format ( ZlibError , " Error %i while compressing: %.200s " ,
err , self - > zst . msg ) ;
Py_DECREF ( RetVal ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
1997-08-14 18:06:42 -03:00
}
1997-04-29 12:38:09 -03:00
err = deflateEnd ( & ( self - > zst ) ) ;
1997-08-14 18:06:42 -03:00
if ( err ! = Z_OK ) {
1997-09-04 20:39:23 -03:00
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while flushing compression object " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while flushing compression object: %.200s " ,
err , self - > zst . msg ) ;
Py_DECREF ( RetVal ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
1997-08-14 18:06:42 -03:00
}
_PyString_Resize ( & RetVal ,
( char * ) self - > zst . next_out - PyString_AsString ( RetVal ) ) ;
1997-04-29 12:38:09 -03:00
return RetVal ;
}
1997-06-03 19:21:03 -03:00
static char decomp_flush__doc__ [ ] =
" flush() -- Return a string containing any remaining decompressed data. "
" The decompressor object can no longer be used after this call. "
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_unflush ( self , args )
compobject * self ;
PyObject * args ;
{
int length = 0 , err ;
PyObject * RetVal ;
if ( ! PyArg_NoArgs ( args ) )
1997-08-13 20:19:55 -03:00
return NULL ;
1997-08-14 18:06:42 -03:00
if ( ! ( RetVal = PyString_FromStringAndSize ( NULL , DEFAULTALLOC ) ) )
{
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to decompress data " ) ;
return NULL ;
}
1997-04-29 12:38:09 -03:00
self - > zst . avail_in = 0 ;
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) ;
1997-08-13 20:19:55 -03:00
length = self - > zst . avail_out = DEFAULTALLOC ;
err = Z_OK ;
while ( err = = Z_OK )
{
err = inflate ( & ( self - > zst ) , Z_FINISH ) ;
if ( err = = Z_OK & & self - > zst . avail_out = = 0 )
{
if ( _PyString_Resize ( & RetVal , length < < 1 ) = = - 1 )
{
PyErr_SetString ( PyExc_MemoryError ,
" Can't allocate memory to decompress data " ) ;
1997-09-04 20:39:23 -03:00
Py_DECREF ( RetVal ) ;
1997-08-13 20:19:55 -03:00
return NULL ;
}
1997-08-18 12:31:24 -03:00
self - > zst . next_out = ( unsigned char * ) PyString_AsString ( RetVal ) + length ;
1997-08-13 20:19:55 -03:00
self - > zst . avail_out = length ;
length = length < < 1 ;
}
}
1997-04-29 12:38:09 -03:00
if ( err ! = Z_STREAM_END )
1997-09-04 20:39:23 -03:00
{
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError , " Error %i while decompressing " ,
err ) ;
else
PyErr_Format ( ZlibError , " Error %i while decompressing: %.200s " ,
err , self - > zst . msg ) ;
Py_DECREF ( RetVal ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
1997-09-04 20:39:23 -03:00
}
1997-04-29 12:38:09 -03:00
err = inflateEnd ( & ( self - > zst ) ) ;
if ( err ! = Z_OK )
1997-09-04 20:39:23 -03:00
{
if ( self - > zst . msg = = Z_NULL )
PyErr_Format ( ZlibError ,
" Error %i while flushing decompression object " ,
err ) ;
else
PyErr_Format ( ZlibError ,
" Error %i while flushing decompression object: %.200s " ,
err , self - > zst . msg ) ;
Py_DECREF ( RetVal ) ;
1997-04-29 12:38:09 -03:00
return NULL ;
1997-09-04 20:39:23 -03:00
}
1997-08-14 18:06:42 -03:00
_PyString_Resize ( & RetVal ,
( char * ) self - > zst . next_out - PyString_AsString ( RetVal ) ) ;
1997-04-29 12:38:09 -03:00
return RetVal ;
}
static PyMethodDef comp_methods [ ] =
{
1997-08-28 18:21:22 -03:00
{ " compress " , ( binaryfunc ) PyZlib_objcompress , 1 , comp_compress__doc__ } ,
{ " flush " , ( binaryfunc ) PyZlib_flush , 0 , comp_flush__doc__ } ,
1997-04-29 12:38:09 -03:00
{ NULL , NULL }
} ;
static PyMethodDef Decomp_methods [ ] =
{
1997-08-28 18:21:22 -03:00
{ " decompress " , ( binaryfunc ) PyZlib_objdecompress , 1 , decomp_decompress__doc__ } ,
{ " flush " , ( binaryfunc ) PyZlib_unflush , 0 , decomp_flush__doc__ } ,
1997-04-29 12:38:09 -03:00
{ NULL , NULL }
} ;
static PyObject *
Comp_getattr ( self , name )
compobject * self ;
char * name ;
{
return Py_FindMethod ( comp_methods , ( PyObject * ) self , name ) ;
}
static PyObject *
Decomp_getattr ( self , name )
compobject * self ;
char * name ;
{
return Py_FindMethod ( Decomp_methods , ( PyObject * ) self , name ) ;
}
1997-06-03 19:21:03 -03:00
static char adler32__doc__ [ ] =
" adler32(string) -- Compute an Adler-32 checksum of string, using "
" a default starting value, and returning an integer value. \n "
" adler32(string, value) -- Compute an Adler-32 checksum of string, using "
" the starting value provided, and returning an integer value \n "
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_adler32 ( self , args )
PyObject * self , * args ;
{
1997-09-04 20:39:23 -03:00
uLong adler32val = adler32 ( 0L , Z_NULL , 0 ) ;
Byte * buf ;
int len ;
1997-04-29 12:38:09 -03:00
1997-09-04 20:39:23 -03:00
if ( ! PyArg_ParseTuple ( args , " s#|l " , & buf , & len , & adler32val ) )
{
return NULL ;
}
adler32val = adler32 ( adler32val , buf , len ) ;
return PyInt_FromLong ( adler32val ) ;
1997-04-29 12:38:09 -03:00
}
1997-06-03 19:21:03 -03:00
static char crc32__doc__ [ ] =
" crc32(string) -- Compute a CRC-32 checksum of string, using "
" a default starting value, and returning an integer value. \n "
" crc32(string, value) -- Compute a CRC-32 checksum of string, using "
" the starting value provided, and returning an integer value. \n "
;
1997-04-29 12:38:09 -03:00
static PyObject *
PyZlib_crc32 ( self , args )
PyObject * self , * args ;
{
1997-09-04 20:39:23 -03:00
uLong crc32val = crc32 ( 0L , Z_NULL , 0 ) ;
Byte * buf ;
int len ;
if ( ! PyArg_ParseTuple ( args , " s#|l " , & buf , & len , & crc32val ) )
{
return NULL ;
}
crc32val = crc32 ( crc32val , buf , len ) ;
return PyInt_FromLong ( crc32val ) ;
1997-04-29 12:38:09 -03:00
}
static PyMethodDef zlib_methods [ ] =
{
1997-06-03 19:21:03 -03:00
{ " adler32 " , ( PyCFunction ) PyZlib_adler32 , 1 , adler32__doc__ } ,
{ " compress " , ( PyCFunction ) PyZlib_compress , 1 , compress__doc__ } ,
{ " compressobj " , ( PyCFunction ) PyZlib_compressobj , 1 , compressobj__doc__ } ,
{ " crc32 " , ( PyCFunction ) PyZlib_crc32 , 1 , crc32__doc__ } ,
{ " decompress " , ( PyCFunction ) PyZlib_decompress , 1 , decompress__doc__ } ,
{ " decompressobj " , ( PyCFunction ) PyZlib_decompressobj , 1 , decompressobj__doc__ } ,
1997-04-29 12:38:09 -03:00
{ NULL , NULL }
} ;
statichere PyTypeObject Comptype = {
PyObject_HEAD_INIT ( & PyType_Type )
0 ,
" Compress " ,
sizeof ( compobject ) ,
0 ,
( destructor ) Comp_dealloc , /*tp_dealloc*/
0 , /*tp_print*/
( getattrfunc ) Comp_getattr , /*tp_getattr*/
0 , /*tp_setattr*/
0 , /*tp_compare*/
0 , /*tp_repr*/
0 , /*tp_as_number*/
0 , /*tp_as_sequence*/
0 , /*tp_as_mapping*/
} ;
statichere PyTypeObject Decomptype = {
PyObject_HEAD_INIT ( & PyType_Type )
0 ,
" Decompress " ,
sizeof ( compobject ) ,
0 ,
( destructor ) Decomp_dealloc , /*tp_dealloc*/
0 , /*tp_print*/
( getattrfunc ) Decomp_getattr , /*tp_getattr*/
0 , /*tp_setattr*/
0 , /*tp_compare*/
0 , /*tp_repr*/
0 , /*tp_as_number*/
0 , /*tp_as_sequence*/
0 , /*tp_as_mapping*/
} ;
/* The following insint() routine was blatantly ripped off from
socketmodule . c */
/* Convenience routine to export an integer value.
For simplicity , errors ( which are unlikely anyway ) are ignored . */
static void
insint ( d , name , value )
PyObject * d ;
char * name ;
int value ;
{
PyObject * v = PyInt_FromLong ( ( long ) value ) ;
if ( v = = NULL ) {
/* Don't bother reporting this error */
PyErr_Clear ( ) ;
}
else {
PyDict_SetItemString ( d , name , v ) ;
Py_DECREF ( v ) ;
}
}
1997-06-03 19:21:03 -03:00
static char zlib_module_documentation [ ] =
" The functions in this module allow compression and decompression "
" using the zlib library, which is based on GNU zip. \n \n "
" adler32(string) -- Compute an Adler-32 checksum. \n "
" adler32(string, start) -- Compute an Adler-32 checksum using a given starting value. \n "
" compress(string) -- Compress a string. \n "
" compress(string, level) -- Compress a string with the given level of compression (1--9). \n "
" compressobj([level]) -- Return a compressor object. \n "
" crc32(string) -- Compute a CRC-32 checksum. \n "
" crc32(string, start) -- Compute a CRC-32 checksum using a given starting value. \n "
1997-08-13 20:19:55 -03:00
" decompress(string,[wbites],[bufsize]) -- Decompresses a compressed string. \n "
1997-06-03 19:21:03 -03:00
" decompressobj([wbits]) -- Return a decompressor object (wbits=window buffer size). \n \n "
" Compressor objects support compress() and flush() methods; decompressor \n "
" objects support decompress() and flush(). "
;
1997-04-29 12:38:09 -03:00
void
PyInit_zlib ( )
{
1997-09-04 20:39:23 -03:00
PyObject * m , * d , * ver ;
1997-06-03 19:21:03 -03:00
m = Py_InitModule4 ( " zlib " , zlib_methods ,
zlib_module_documentation ,
( PyObject * ) NULL , PYTHON_API_VERSION ) ;
1997-04-29 12:38:09 -03:00
d = PyModule_GetDict ( m ) ;
1997-10-01 01:29:29 -03:00
ZlibError = PyErr_NewException ( " zlib.error " , NULL , NULL ) ;
1997-04-29 12:38:09 -03:00
PyDict_SetItemString ( d , " error " , ZlibError ) ;
1997-09-04 20:39:23 -03:00
1997-04-29 12:38:09 -03:00
insint ( d , " MAX_WBITS " , MAX_WBITS ) ;
insint ( d , " DEFLATED " , DEFLATED ) ;
insint ( d , " DEF_MEM_LEVEL " , DEF_MEM_LEVEL ) ;
1997-09-04 20:39:23 -03:00
insint ( d , " Z_BEST_SPEED " , Z_BEST_SPEED ) ;
insint ( d , " Z_BEST_COMPRESSION " , Z_BEST_COMPRESSION ) ;
insint ( d , " Z_DEFAULT_COMPRESSION " , Z_DEFAULT_COMPRESSION ) ;
insint ( d , " Z_FILTERED " , Z_FILTERED ) ;
insint ( d , " Z_HUFFMAN_ONLY " , Z_HUFFMAN_ONLY ) ;
insint ( d , " Z_DEFAULT_STRATEGY " , Z_DEFAULT_STRATEGY ) ;
ver = PyString_FromString ( ZLIB_VERSION ) ;
PyDict_SetItemString ( d , " ZLIB_VERSION " , ver ) ;
1997-04-29 12:38:09 -03:00
}