Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 1 | /*[clinic input] |
| 2 | preserve |
| 3 | [clinic start generated code]*/ |
| 4 | |
| 5 | PyDoc_STRVAR(zlib_compress__doc__, |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 6 | "compress($module, /, data, level=Z_DEFAULT_COMPRESSION)\n" |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 7 | "--\n" |
| 8 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 9 | "Returns a bytes object containing compressed data.\n" |
| 10 | "\n" |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 11 | " data\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 12 | " Binary data to be compressed.\n" |
| 13 | " level\n" |
Martin Panter | b0cb42d | 2016-02-10 10:45:54 +0000 | [diff] [blame] | 14 | " Compression level, in 0-9 or -1."); |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 15 | |
| 16 | #define ZLIB_COMPRESS_METHODDEF \ |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 17 | {"compress", (PyCFunction)zlib_compress, METH_VARARGS|METH_KEYWORDS, zlib_compress__doc__}, |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 18 | |
| 19 | static PyObject * |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 20 | zlib_compress_impl(PyModuleDef *module, Py_buffer *data, int level); |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 21 | |
| 22 | static PyObject * |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 23 | zlib_compress(PyModuleDef *module, PyObject *args, PyObject *kwargs) |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 24 | { |
| 25 | PyObject *return_value = NULL; |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 26 | static char *_keywords[] = {"data", "level", NULL}; |
| 27 | Py_buffer data = {NULL, NULL}; |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 28 | int level = Z_DEFAULT_COMPRESSION; |
| 29 | |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 30 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "y*|i:compress", _keywords, |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 31 | &data, &level)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 32 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 33 | } |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 34 | return_value = zlib_compress_impl(module, &data, level); |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 35 | |
| 36 | exit: |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 37 | /* Cleanup for data */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 38 | if (data.obj) { |
Martin Panter | 1fe0d13 | 2016-02-10 10:06:36 +0000 | [diff] [blame] | 39 | PyBuffer_Release(&data); |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 40 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 41 | |
| 42 | return return_value; |
| 43 | } |
| 44 | |
| 45 | PyDoc_STRVAR(zlib_decompress__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 46 | "decompress($module, data, wbits=MAX_WBITS, bufsize=DEF_BUF_SIZE, /)\n" |
| 47 | "--\n" |
| 48 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 49 | "Returns a bytes object containing the uncompressed data.\n" |
| 50 | "\n" |
| 51 | " data\n" |
| 52 | " Compressed data.\n" |
| 53 | " wbits\n" |
Martin Panter | 0fdf41d | 2016-05-27 07:32:11 +0000 | [diff] [blame] | 54 | " The window buffer size and container format.\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 55 | " bufsize\n" |
| 56 | " The initial output buffer size."); |
| 57 | |
| 58 | #define ZLIB_DECOMPRESS_METHODDEF \ |
| 59 | {"decompress", (PyCFunction)zlib_decompress, METH_VARARGS, zlib_decompress__doc__}, |
| 60 | |
| 61 | static PyObject * |
Larry Hastings | 89964c4 | 2015-04-14 18:07:59 -0400 | [diff] [blame] | 62 | zlib_decompress_impl(PyModuleDef *module, Py_buffer *data, int wbits, |
| 63 | unsigned int bufsize); |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 64 | |
| 65 | static PyObject * |
| 66 | zlib_decompress(PyModuleDef *module, PyObject *args) |
| 67 | { |
| 68 | PyObject *return_value = NULL; |
| 69 | Py_buffer data = {NULL, NULL}; |
| 70 | int wbits = MAX_WBITS; |
| 71 | unsigned int bufsize = DEF_BUF_SIZE; |
| 72 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 73 | if (!PyArg_ParseTuple(args, "y*|iO&:decompress", |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 74 | &data, &wbits, capped_uint_converter, &bufsize)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 75 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 76 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 77 | return_value = zlib_decompress_impl(module, &data, wbits, bufsize); |
| 78 | |
| 79 | exit: |
| 80 | /* Cleanup for data */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 81 | if (data.obj) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 82 | PyBuffer_Release(&data); |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 83 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 84 | |
| 85 | return return_value; |
| 86 | } |
| 87 | |
| 88 | PyDoc_STRVAR(zlib_compressobj__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 89 | "compressobj($module, /, level=Z_DEFAULT_COMPRESSION, method=DEFLATED,\n" |
| 90 | " wbits=MAX_WBITS, memLevel=DEF_MEM_LEVEL,\n" |
| 91 | " strategy=Z_DEFAULT_STRATEGY, zdict=None)\n" |
| 92 | "--\n" |
| 93 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 94 | "Return a compressor object.\n" |
| 95 | "\n" |
| 96 | " level\n" |
Martin Panter | 567d513 | 2016-02-03 07:06:33 +0000 | [diff] [blame] | 97 | " The compression level (an integer in the range 0-9 or -1; default is\n" |
| 98 | " currently equivalent to 6). Higher compression levels are slower,\n" |
| 99 | " but produce smaller results.\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 100 | " method\n" |
| 101 | " The compression algorithm. If given, this must be DEFLATED.\n" |
| 102 | " wbits\n" |
Martin Panter | 0fdf41d | 2016-05-27 07:32:11 +0000 | [diff] [blame] | 103 | " +9 to +15: The base-two logarithm of the window size. Include a zlib\n" |
| 104 | " container.\n" |
| 105 | " -9 to -15: Generate a raw stream.\n" |
| 106 | " +25 to +31: Include a gzip container.\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 107 | " memLevel\n" |
| 108 | " Controls the amount of memory used for internal compression state.\n" |
| 109 | " Valid values range from 1 to 9. Higher values result in higher memory\n" |
| 110 | " usage, faster compression, and smaller output.\n" |
| 111 | " strategy\n" |
| 112 | " Used to tune the compression algorithm. Possible values are\n" |
| 113 | " Z_DEFAULT_STRATEGY, Z_FILTERED, and Z_HUFFMAN_ONLY.\n" |
| 114 | " zdict\n" |
| 115 | " The predefined compression dictionary - a sequence of bytes\n" |
| 116 | " containing subsequences that are likely to occur in the input data."); |
| 117 | |
| 118 | #define ZLIB_COMPRESSOBJ_METHODDEF \ |
| 119 | {"compressobj", (PyCFunction)zlib_compressobj, METH_VARARGS|METH_KEYWORDS, zlib_compressobj__doc__}, |
| 120 | |
| 121 | static PyObject * |
Larry Hastings | 89964c4 | 2015-04-14 18:07:59 -0400 | [diff] [blame] | 122 | zlib_compressobj_impl(PyModuleDef *module, int level, int method, int wbits, |
| 123 | int memLevel, int strategy, Py_buffer *zdict); |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 124 | |
| 125 | static PyObject * |
| 126 | zlib_compressobj(PyModuleDef *module, PyObject *args, PyObject *kwargs) |
| 127 | { |
| 128 | PyObject *return_value = NULL; |
| 129 | static char *_keywords[] = {"level", "method", "wbits", "memLevel", "strategy", "zdict", NULL}; |
| 130 | int level = Z_DEFAULT_COMPRESSION; |
| 131 | int method = DEFLATED; |
| 132 | int wbits = MAX_WBITS; |
| 133 | int memLevel = DEF_MEM_LEVEL; |
| 134 | int strategy = Z_DEFAULT_STRATEGY; |
| 135 | Py_buffer zdict = {NULL, NULL}; |
| 136 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 137 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|iiiiiy*:compressobj", _keywords, |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 138 | &level, &method, &wbits, &memLevel, &strategy, &zdict)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 139 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 140 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 141 | return_value = zlib_compressobj_impl(module, level, method, wbits, memLevel, strategy, &zdict); |
| 142 | |
| 143 | exit: |
| 144 | /* Cleanup for zdict */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 145 | if (zdict.obj) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 146 | PyBuffer_Release(&zdict); |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 147 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 148 | |
| 149 | return return_value; |
| 150 | } |
| 151 | |
| 152 | PyDoc_STRVAR(zlib_decompressobj__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 153 | "decompressobj($module, /, wbits=MAX_WBITS, zdict=b\'\')\n" |
| 154 | "--\n" |
| 155 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 156 | "Return a decompressor object.\n" |
| 157 | "\n" |
| 158 | " wbits\n" |
Martin Panter | 0fdf41d | 2016-05-27 07:32:11 +0000 | [diff] [blame] | 159 | " The window buffer size and container format.\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 160 | " zdict\n" |
| 161 | " The predefined compression dictionary. This must be the same\n" |
| 162 | " dictionary as used by the compressor that produced the input data."); |
| 163 | |
| 164 | #define ZLIB_DECOMPRESSOBJ_METHODDEF \ |
| 165 | {"decompressobj", (PyCFunction)zlib_decompressobj, METH_VARARGS|METH_KEYWORDS, zlib_decompressobj__doc__}, |
| 166 | |
| 167 | static PyObject * |
| 168 | zlib_decompressobj_impl(PyModuleDef *module, int wbits, PyObject *zdict); |
| 169 | |
| 170 | static PyObject * |
| 171 | zlib_decompressobj(PyModuleDef *module, PyObject *args, PyObject *kwargs) |
| 172 | { |
| 173 | PyObject *return_value = NULL; |
| 174 | static char *_keywords[] = {"wbits", "zdict", NULL}; |
| 175 | int wbits = MAX_WBITS; |
| 176 | PyObject *zdict = NULL; |
| 177 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 178 | if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|iO:decompressobj", _keywords, |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 179 | &wbits, &zdict)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 180 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 181 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 182 | return_value = zlib_decompressobj_impl(module, wbits, zdict); |
| 183 | |
| 184 | exit: |
| 185 | return return_value; |
| 186 | } |
| 187 | |
| 188 | PyDoc_STRVAR(zlib_Compress_compress__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 189 | "compress($self, data, /)\n" |
| 190 | "--\n" |
| 191 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 192 | "Returns a bytes object containing compressed data.\n" |
| 193 | "\n" |
| 194 | " data\n" |
| 195 | " Binary data to be compressed.\n" |
| 196 | "\n" |
| 197 | "After calling this function, some of the input data may still\n" |
| 198 | "be stored in internal buffers for later processing.\n" |
| 199 | "Call the flush() method to clear these buffers."); |
| 200 | |
| 201 | #define ZLIB_COMPRESS_COMPRESS_METHODDEF \ |
Serhiy Storchaka | 92e8af6 | 2015-04-04 00:12:11 +0300 | [diff] [blame] | 202 | {"compress", (PyCFunction)zlib_Compress_compress, METH_O, zlib_Compress_compress__doc__}, |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 203 | |
| 204 | static PyObject * |
| 205 | zlib_Compress_compress_impl(compobject *self, Py_buffer *data); |
| 206 | |
| 207 | static PyObject * |
Serhiy Storchaka | 92e8af6 | 2015-04-04 00:12:11 +0300 | [diff] [blame] | 208 | zlib_Compress_compress(compobject *self, PyObject *arg) |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 209 | { |
| 210 | PyObject *return_value = NULL; |
| 211 | Py_buffer data = {NULL, NULL}; |
| 212 | |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 213 | if (!PyArg_Parse(arg, "y*:compress", &data)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 214 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 215 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 216 | return_value = zlib_Compress_compress_impl(self, &data); |
| 217 | |
| 218 | exit: |
| 219 | /* Cleanup for data */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 220 | if (data.obj) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 221 | PyBuffer_Release(&data); |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 222 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 223 | |
| 224 | return return_value; |
| 225 | } |
| 226 | |
| 227 | PyDoc_STRVAR(zlib_Decompress_decompress__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 228 | "decompress($self, data, max_length=0, /)\n" |
| 229 | "--\n" |
| 230 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 231 | "Return a bytes object containing the decompressed version of the data.\n" |
| 232 | "\n" |
| 233 | " data\n" |
| 234 | " The binary data to decompress.\n" |
| 235 | " max_length\n" |
| 236 | " The maximum allowable length of the decompressed data.\n" |
| 237 | " Unconsumed input data will be stored in\n" |
| 238 | " the unconsumed_tail attribute.\n" |
| 239 | "\n" |
| 240 | "After calling this function, some of the input data may still be stored in\n" |
| 241 | "internal buffers for later processing.\n" |
| 242 | "Call the flush() method to clear these buffers."); |
| 243 | |
| 244 | #define ZLIB_DECOMPRESS_DECOMPRESS_METHODDEF \ |
| 245 | {"decompress", (PyCFunction)zlib_Decompress_decompress, METH_VARARGS, zlib_Decompress_decompress__doc__}, |
| 246 | |
| 247 | static PyObject * |
Larry Hastings | 89964c4 | 2015-04-14 18:07:59 -0400 | [diff] [blame] | 248 | zlib_Decompress_decompress_impl(compobject *self, Py_buffer *data, |
| 249 | unsigned int max_length); |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 250 | |
| 251 | static PyObject * |
| 252 | zlib_Decompress_decompress(compobject *self, PyObject *args) |
| 253 | { |
| 254 | PyObject *return_value = NULL; |
| 255 | Py_buffer data = {NULL, NULL}; |
| 256 | unsigned int max_length = 0; |
| 257 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 258 | if (!PyArg_ParseTuple(args, "y*|O&:decompress", |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 259 | &data, capped_uint_converter, &max_length)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 260 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 261 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 262 | return_value = zlib_Decompress_decompress_impl(self, &data, max_length); |
| 263 | |
| 264 | exit: |
| 265 | /* Cleanup for data */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 266 | if (data.obj) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 267 | PyBuffer_Release(&data); |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 268 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 269 | |
| 270 | return return_value; |
| 271 | } |
| 272 | |
| 273 | PyDoc_STRVAR(zlib_Compress_flush__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 274 | "flush($self, mode=zlib.Z_FINISH, /)\n" |
| 275 | "--\n" |
| 276 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 277 | "Return a bytes object containing any remaining compressed data.\n" |
| 278 | "\n" |
| 279 | " mode\n" |
| 280 | " One of the constants Z_SYNC_FLUSH, Z_FULL_FLUSH, Z_FINISH.\n" |
| 281 | " If mode == Z_FINISH, the compressor object can no longer be\n" |
| 282 | " used after calling the flush() method. Otherwise, more data\n" |
| 283 | " can still be compressed."); |
| 284 | |
| 285 | #define ZLIB_COMPRESS_FLUSH_METHODDEF \ |
| 286 | {"flush", (PyCFunction)zlib_Compress_flush, METH_VARARGS, zlib_Compress_flush__doc__}, |
| 287 | |
| 288 | static PyObject * |
| 289 | zlib_Compress_flush_impl(compobject *self, int mode); |
| 290 | |
| 291 | static PyObject * |
| 292 | zlib_Compress_flush(compobject *self, PyObject *args) |
| 293 | { |
| 294 | PyObject *return_value = NULL; |
| 295 | int mode = Z_FINISH; |
| 296 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 297 | if (!PyArg_ParseTuple(args, "|i:flush", |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 298 | &mode)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 299 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 300 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 301 | return_value = zlib_Compress_flush_impl(self, mode); |
| 302 | |
| 303 | exit: |
| 304 | return return_value; |
| 305 | } |
| 306 | |
Larry Hastings | 7726ac9 | 2014-01-31 22:03:12 -0800 | [diff] [blame] | 307 | #if defined(HAVE_ZLIB_COPY) |
| 308 | |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 309 | PyDoc_STRVAR(zlib_Compress_copy__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 310 | "copy($self, /)\n" |
| 311 | "--\n" |
| 312 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 313 | "Return a copy of the compression object."); |
| 314 | |
| 315 | #define ZLIB_COMPRESS_COPY_METHODDEF \ |
| 316 | {"copy", (PyCFunction)zlib_Compress_copy, METH_NOARGS, zlib_Compress_copy__doc__}, |
| 317 | |
| 318 | static PyObject * |
| 319 | zlib_Compress_copy_impl(compobject *self); |
| 320 | |
| 321 | static PyObject * |
| 322 | zlib_Compress_copy(compobject *self, PyObject *Py_UNUSED(ignored)) |
| 323 | { |
| 324 | return zlib_Compress_copy_impl(self); |
| 325 | } |
| 326 | |
Larry Hastings | 7726ac9 | 2014-01-31 22:03:12 -0800 | [diff] [blame] | 327 | #endif /* defined(HAVE_ZLIB_COPY) */ |
| 328 | |
Larry Hastings | 7726ac9 | 2014-01-31 22:03:12 -0800 | [diff] [blame] | 329 | #if defined(HAVE_ZLIB_COPY) |
| 330 | |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 331 | PyDoc_STRVAR(zlib_Decompress_copy__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 332 | "copy($self, /)\n" |
| 333 | "--\n" |
| 334 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 335 | "Return a copy of the decompression object."); |
| 336 | |
| 337 | #define ZLIB_DECOMPRESS_COPY_METHODDEF \ |
| 338 | {"copy", (PyCFunction)zlib_Decompress_copy, METH_NOARGS, zlib_Decompress_copy__doc__}, |
| 339 | |
| 340 | static PyObject * |
| 341 | zlib_Decompress_copy_impl(compobject *self); |
| 342 | |
| 343 | static PyObject * |
| 344 | zlib_Decompress_copy(compobject *self, PyObject *Py_UNUSED(ignored)) |
| 345 | { |
| 346 | return zlib_Decompress_copy_impl(self); |
| 347 | } |
| 348 | |
Larry Hastings | 7726ac9 | 2014-01-31 22:03:12 -0800 | [diff] [blame] | 349 | #endif /* defined(HAVE_ZLIB_COPY) */ |
| 350 | |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 351 | PyDoc_STRVAR(zlib_Decompress_flush__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 352 | "flush($self, length=zlib.DEF_BUF_SIZE, /)\n" |
| 353 | "--\n" |
| 354 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 355 | "Return a bytes object containing any remaining decompressed data.\n" |
| 356 | "\n" |
| 357 | " length\n" |
| 358 | " the initial size of the output buffer."); |
| 359 | |
| 360 | #define ZLIB_DECOMPRESS_FLUSH_METHODDEF \ |
| 361 | {"flush", (PyCFunction)zlib_Decompress_flush, METH_VARARGS, zlib_Decompress_flush__doc__}, |
| 362 | |
| 363 | static PyObject * |
| 364 | zlib_Decompress_flush_impl(compobject *self, unsigned int length); |
| 365 | |
| 366 | static PyObject * |
| 367 | zlib_Decompress_flush(compobject *self, PyObject *args) |
| 368 | { |
| 369 | PyObject *return_value = NULL; |
| 370 | unsigned int length = DEF_BUF_SIZE; |
| 371 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 372 | if (!PyArg_ParseTuple(args, "|O&:flush", |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 373 | capped_uint_converter, &length)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 374 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 375 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 376 | return_value = zlib_Decompress_flush_impl(self, length); |
| 377 | |
| 378 | exit: |
| 379 | return return_value; |
| 380 | } |
| 381 | |
| 382 | PyDoc_STRVAR(zlib_adler32__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 383 | "adler32($module, data, value=1, /)\n" |
| 384 | "--\n" |
| 385 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 386 | "Compute an Adler-32 checksum of data.\n" |
| 387 | "\n" |
| 388 | " value\n" |
| 389 | " Starting value of the checksum.\n" |
| 390 | "\n" |
| 391 | "The returned checksum is an integer."); |
| 392 | |
| 393 | #define ZLIB_ADLER32_METHODDEF \ |
| 394 | {"adler32", (PyCFunction)zlib_adler32, METH_VARARGS, zlib_adler32__doc__}, |
| 395 | |
| 396 | static PyObject * |
| 397 | zlib_adler32_impl(PyModuleDef *module, Py_buffer *data, unsigned int value); |
| 398 | |
| 399 | static PyObject * |
| 400 | zlib_adler32(PyModuleDef *module, PyObject *args) |
| 401 | { |
| 402 | PyObject *return_value = NULL; |
| 403 | Py_buffer data = {NULL, NULL}; |
| 404 | unsigned int value = 1; |
| 405 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 406 | if (!PyArg_ParseTuple(args, "y*|I:adler32", |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 407 | &data, &value)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 408 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 409 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 410 | return_value = zlib_adler32_impl(module, &data, value); |
| 411 | |
| 412 | exit: |
| 413 | /* Cleanup for data */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 414 | if (data.obj) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 415 | PyBuffer_Release(&data); |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 416 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 417 | |
| 418 | return return_value; |
| 419 | } |
| 420 | |
| 421 | PyDoc_STRVAR(zlib_crc32__doc__, |
Larry Hastings | 2623c8c | 2014-02-08 22:15:29 -0800 | [diff] [blame] | 422 | "crc32($module, data, value=0, /)\n" |
| 423 | "--\n" |
| 424 | "\n" |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 425 | "Compute a CRC-32 checksum of data.\n" |
| 426 | "\n" |
| 427 | " value\n" |
| 428 | " Starting value of the checksum.\n" |
| 429 | "\n" |
| 430 | "The returned checksum is an integer."); |
| 431 | |
| 432 | #define ZLIB_CRC32_METHODDEF \ |
| 433 | {"crc32", (PyCFunction)zlib_crc32, METH_VARARGS, zlib_crc32__doc__}, |
| 434 | |
| 435 | static PyObject * |
| 436 | zlib_crc32_impl(PyModuleDef *module, Py_buffer *data, unsigned int value); |
| 437 | |
| 438 | static PyObject * |
| 439 | zlib_crc32(PyModuleDef *module, PyObject *args) |
| 440 | { |
| 441 | PyObject *return_value = NULL; |
| 442 | Py_buffer data = {NULL, NULL}; |
| 443 | unsigned int value = 0; |
| 444 | |
Serhiy Storchaka | 247789c | 2015-04-24 00:40:51 +0300 | [diff] [blame] | 445 | if (!PyArg_ParseTuple(args, "y*|I:crc32", |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 446 | &data, &value)) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 447 | goto exit; |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 448 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 449 | return_value = zlib_crc32_impl(module, &data, value); |
| 450 | |
| 451 | exit: |
| 452 | /* Cleanup for data */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 453 | if (data.obj) { |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 454 | PyBuffer_Release(&data); |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 455 | } |
Serhiy Storchaka | 2c5ddbe | 2014-01-27 00:03:31 +0200 | [diff] [blame] | 456 | |
| 457 | return return_value; |
| 458 | } |
Larry Hastings | 0759f84 | 2015-04-03 13:09:02 -0700 | [diff] [blame] | 459 | |
| 460 | #ifndef ZLIB_COMPRESS_COPY_METHODDEF |
| 461 | #define ZLIB_COMPRESS_COPY_METHODDEF |
| 462 | #endif /* !defined(ZLIB_COMPRESS_COPY_METHODDEF) */ |
Serhiy Storchaka | 5dee655 | 2016-06-09 16:16:06 +0300 | [diff] [blame] | 463 | /*[clinic end generated code: output=9bd8a093baa653b2 input=a9049054013a1b77]*/ |