| OLD | NEW |
| 1 /* gzread.c -- zlib functions for reading gzip files | 1 /* gzread.c -- zlib functions for reading gzip files |
| 2 * Copyright (C) 2004, 2005, 2010 Mark Adler | 2 * Copyright (C) 2004, 2005, 2010, 2011, 2012, 2013 Mark Adler |
| 3 * For conditions of distribution and use, see copyright notice in zlib.h | 3 * For conditions of distribution and use, see copyright notice in zlib.h |
| 4 */ | 4 */ |
| 5 | 5 |
| 6 #include "gzguts.h" | 6 #include "gzguts.h" |
| 7 | 7 |
| 8 /* Local functions */ | 8 /* Local functions */ |
| 9 local int gz_load OF((gz_statep, unsigned char *, unsigned, unsigned *)); | 9 local int gz_load OF((gz_statep, unsigned char *, unsigned, unsigned *)); |
| 10 local int gz_avail OF((gz_statep)); | 10 local int gz_avail OF((gz_statep)); |
| 11 local int gz_next4 OF((gz_statep, unsigned long *)); | 11 local int gz_look OF((gz_statep)); |
| 12 local int gz_head OF((gz_statep)); | |
| 13 local int gz_decomp OF((gz_statep)); | 12 local int gz_decomp OF((gz_statep)); |
| 14 local int gz_make OF((gz_statep)); | 13 local int gz_fetch OF((gz_statep)); |
| 15 local int gz_skip OF((gz_statep, z_off64_t)); | 14 local int gz_skip OF((gz_statep, z_off64_t)); |
| 16 | 15 |
| 17 /* Use read() to load a buffer -- return -1 on error, otherwise 0. Read from | 16 /* Use read() to load a buffer -- return -1 on error, otherwise 0. Read from |
| 18 state->fd, and update state->eof, state->err, and state->msg as appropriate. | 17 state->fd, and update state->eof, state->err, and state->msg as appropriate. |
| 19 This function needs to loop on read(), since read() is not guaranteed to | 18 This function needs to loop on read(), since read() is not guaranteed to |
| 20 read the number of bytes requested, depending on the type of descriptor. */ | 19 read the number of bytes requested, depending on the type of descriptor. */ |
| 21 local int gz_load(state, buf, len, have) | 20 local int gz_load(state, buf, len, have) |
| 22 gz_statep state; | 21 gz_statep state; |
| 23 unsigned char *buf; | 22 unsigned char *buf; |
| 24 unsigned len; | 23 unsigned len; |
| (...skipping 14 matching lines...) Expand all Loading... |
| 39 } | 38 } |
| 40 if (ret == 0) | 39 if (ret == 0) |
| 41 state->eof = 1; | 40 state->eof = 1; |
| 42 return 0; | 41 return 0; |
| 43 } | 42 } |
| 44 | 43 |
| 45 /* Load up input buffer and set eof flag if last data loaded -- return -1 on | 44 /* Load up input buffer and set eof flag if last data loaded -- return -1 on |
| 46 error, 0 otherwise. Note that the eof flag is set when the end of the input | 45 error, 0 otherwise. Note that the eof flag is set when the end of the input |
| 47 file is reached, even though there may be unused data in the buffer. Once | 46 file is reached, even though there may be unused data in the buffer. Once |
| 48 that data has been used, no more attempts will be made to read the file. | 47 that data has been used, no more attempts will be made to read the file. |
| 49 gz_avail() assumes that strm->avail_in == 0. */ | 48 If strm->avail_in != 0, then the current data is moved to the beginning of |
| 49 the input buffer, and then the remainder of the buffer is loaded with the |
| 50 available data from the input file. */ |
| 50 local int gz_avail(state) | 51 local int gz_avail(state) |
| 51 gz_statep state; | 52 gz_statep state; |
| 52 { | 53 { |
| 54 unsigned got; |
| 53 z_streamp strm = &(state->strm); | 55 z_streamp strm = &(state->strm); |
| 54 | 56 |
| 55 if (state->err != Z_OK) | 57 if (state->err != Z_OK && state->err != Z_BUF_ERROR) |
| 56 return -1; | 58 return -1; |
| 57 if (state->eof == 0) { | 59 if (state->eof == 0) { |
| 58 if (gz_load(state, state->in, state->size, | 60 if (strm->avail_in) { /* copy what's there to the start */ |
| 59 (unsigned *)&(strm->avail_in)) == -1) | 61 unsigned char *p = state->in; |
| 62 unsigned const char *q = strm->next_in; |
| 63 unsigned n = strm->avail_in; |
| 64 do { |
| 65 *p++ = *q++; |
| 66 } while (--n); |
| 67 } |
| 68 if (gz_load(state, state->in + strm->avail_in, |
| 69 state->size - strm->avail_in, &got) == -1) |
| 60 return -1; | 70 return -1; |
| 71 strm->avail_in += got; |
| 61 strm->next_in = state->in; | 72 strm->next_in = state->in; |
| 62 } | 73 } |
| 63 return 0; | 74 return 0; |
| 64 } | 75 } |
| 65 | 76 |
| 66 /* Get next byte from input, or -1 if end or error. */ | 77 /* Look for gzip header, set up for inflate or copy. state->x.have must be 0. |
| 67 #define NEXT() ((strm->avail_in == 0 && gz_avail(state) == -1) ? -1 : \ | |
| 68 (strm->avail_in == 0 ? -1 : \ | |
| 69 (strm->avail_in--, *(strm->next_in)++))) | |
| 70 | |
| 71 /* Get a four-byte little-endian integer and return 0 on success and the value | |
| 72 in *ret. Otherwise -1 is returned and *ret is not modified. */ | |
| 73 local int gz_next4(state, ret) | |
| 74 gz_statep state; | |
| 75 unsigned long *ret; | |
| 76 { | |
| 77 int ch; | |
| 78 unsigned long val; | |
| 79 z_streamp strm = &(state->strm); | |
| 80 | |
| 81 val = NEXT(); | |
| 82 val += (unsigned)NEXT() << 8; | |
| 83 val += (unsigned long)NEXT() << 16; | |
| 84 ch = NEXT(); | |
| 85 if (ch == -1) | |
| 86 return -1; | |
| 87 val += (unsigned long)ch << 24; | |
| 88 *ret = val; | |
| 89 return 0; | |
| 90 } | |
| 91 | |
| 92 /* Look for gzip header, set up for inflate or copy. state->have must be zero. | |
| 93 If this is the first time in, allocate required memory. state->how will be | 78 If this is the first time in, allocate required memory. state->how will be |
| 94 left unchanged if there is no more input data available, will be set to COPY | 79 left unchanged if there is no more input data available, will be set to COPY |
| 95 if there is no gzip header and direct copying will be performed, or it will | 80 if there is no gzip header and direct copying will be performed, or it will |
| 96 be set to GZIP for decompression, and the gzip header will be skipped so | 81 be set to GZIP for decompression. If direct copying, then leftover input |
| 97 that the next available input data is the raw deflate stream. If direct | 82 data from the input buffer will be copied to the output buffer. In that |
| 98 copying, then leftover input data from the input buffer will be copied to | 83 case, all further file reads will be directly to either the output buffer or |
| 99 the output buffer. In that case, all further file reads will be directly to | 84 a user buffer. If decompressing, the inflate state will be initialized. |
| 100 either the output buffer or a user buffer. If decompressing, the inflate | 85 gz_look() will return 0 on success or -1 on failure. */ |
| 101 state and the check value will be initialized. gz_head() will return 0 on | 86 local int gz_look(state) |
| 102 success or -1 on failure. Failures may include read errors or gzip header | |
| 103 errors. */ | |
| 104 local int gz_head(state) | |
| 105 gz_statep state; | 87 gz_statep state; |
| 106 { | 88 { |
| 107 z_streamp strm = &(state->strm); | 89 z_streamp strm = &(state->strm); |
| 108 int flags; | |
| 109 unsigned len; | |
| 110 | 90 |
| 111 /* allocate read buffers and inflate memory */ | 91 /* allocate read buffers and inflate memory */ |
| 112 if (state->size == 0) { | 92 if (state->size == 0) { |
| 113 /* allocate buffers */ | 93 /* allocate buffers */ |
| 114 state->in = malloc(state->want); | 94 state->in = (unsigned char *)malloc(state->want); |
| 115 state->out = malloc(state->want << 1); | 95 state->out = (unsigned char *)malloc(state->want << 1); |
| 116 if (state->in == NULL || state->out == NULL) { | 96 if (state->in == NULL || state->out == NULL) { |
| 117 if (state->out != NULL) | 97 if (state->out != NULL) |
| 118 free(state->out); | 98 free(state->out); |
| 119 if (state->in != NULL) | 99 if (state->in != NULL) |
| 120 free(state->in); | 100 free(state->in); |
| 121 gz_error(state, Z_MEM_ERROR, "out of memory"); | 101 gz_error(state, Z_MEM_ERROR, "out of memory"); |
| 122 return -1; | 102 return -1; |
| 123 } | 103 } |
| 124 state->size = state->want; | 104 state->size = state->want; |
| 125 | 105 |
| 126 /* allocate inflate memory */ | 106 /* allocate inflate memory */ |
| 127 state->strm.zalloc = Z_NULL; | 107 state->strm.zalloc = Z_NULL; |
| 128 state->strm.zfree = Z_NULL; | 108 state->strm.zfree = Z_NULL; |
| 129 state->strm.opaque = Z_NULL; | 109 state->strm.opaque = Z_NULL; |
| 130 state->strm.avail_in = 0; | 110 state->strm.avail_in = 0; |
| 131 state->strm.next_in = Z_NULL; | 111 state->strm.next_in = Z_NULL; |
| 132 if (inflateInit2(&(state->strm), -15) != Z_OK) { /* raw inflate */ | 112 if (inflateInit2(&(state->strm), 15 + 16) != Z_OK) { /* gunzip */ |
| 133 free(state->out); | 113 free(state->out); |
| 134 free(state->in); | 114 free(state->in); |
| 135 state->size = 0; | 115 state->size = 0; |
| 136 gz_error(state, Z_MEM_ERROR, "out of memory"); | 116 gz_error(state, Z_MEM_ERROR, "out of memory"); |
| 137 return -1; | 117 return -1; |
| 138 } | 118 } |
| 139 } | 119 } |
| 140 | 120 |
| 141 /* get some data in the input buffer */ | 121 /* get at least the magic bytes in the input buffer */ |
| 142 if (strm->avail_in == 0) { | 122 if (strm->avail_in < 2) { |
| 143 if (gz_avail(state) == -1) | 123 if (gz_avail(state) == -1) |
| 144 return -1; | 124 return -1; |
| 145 if (strm->avail_in == 0) | 125 if (strm->avail_in == 0) |
| 146 return 0; | 126 return 0; |
| 147 } | 127 } |
| 148 | 128 |
| 149 /* look for the gzip magic header bytes 31 and 139 */ | 129 /* look for gzip magic bytes -- if there, do gzip decoding (note: there is |
| 150 if (strm->next_in[0] == 31) { | 130 a logical dilemma here when considering the case of a partially written |
| 151 strm->avail_in--; | 131 gzip file, to wit, if a single 31 byte is written, then we cannot tell |
| 152 strm->next_in++; | 132 whether this is a single-byte file, or just a partially written gzip |
| 153 if (strm->avail_in == 0 && gz_avail(state) == -1) | 133 file -- for here we assume that if a gzip file is being written, then |
| 154 return -1; | 134 the header will be written in a single operation, so that reading a |
| 155 if (strm->avail_in && strm->next_in[0] == 139) { | 135 single byte is sufficient indication that it is not a gzip file) */ |
| 156 /* we have a gzip header, woo hoo! */ | 136 if (strm->avail_in > 1 && |
| 157 strm->avail_in--; | 137 strm->next_in[0] == 31 && strm->next_in[1] == 139) { |
| 158 strm->next_in++; | 138 inflateReset(strm); |
| 159 | 139 state->how = GZIP; |
| 160 /* skip rest of header */ | 140 state->direct = 0; |
| 161 if (NEXT() != 8) { /* compression method */ | 141 return 0; |
| 162 gz_error(state, Z_DATA_ERROR, "unknown compression method"); | |
| 163 return -1; | |
| 164 } | |
| 165 flags = NEXT(); | |
| 166 if (flags & 0xe0) { /* reserved flag bits */ | |
| 167 gz_error(state, Z_DATA_ERROR, "unknown header flags set"); | |
| 168 return -1; | |
| 169 } | |
| 170 NEXT(); /* modification time */ | |
| 171 NEXT(); | |
| 172 NEXT(); | |
| 173 NEXT(); | |
| 174 NEXT(); /* extra flags */ | |
| 175 NEXT(); /* operating system */ | |
| 176 if (flags & 4) { /* extra field */ | |
| 177 len = (unsigned)NEXT(); | |
| 178 len += (unsigned)NEXT() << 8; | |
| 179 while (len--) | |
| 180 if (NEXT() < 0) | |
| 181 break; | |
| 182 } | |
| 183 if (flags & 8) /* file name */ | |
| 184 while (NEXT() > 0) | |
| 185 ; | |
| 186 if (flags & 16) /* comment */ | |
| 187 while (NEXT() > 0) | |
| 188 ; | |
| 189 if (flags & 2) { /* header crc */ | |
| 190 NEXT(); | |
| 191 NEXT(); | |
| 192 } | |
| 193 /* an unexpected end of file is not checked for here -- it will be | |
| 194 noticed on the first request for uncompressed data */ | |
| 195 | |
| 196 /* set up for decompression */ | |
| 197 inflateReset(strm); | |
| 198 strm->adler = crc32(0L, Z_NULL, 0); | |
| 199 state->how = GZIP; | |
| 200 state->direct = 0; | |
| 201 return 0; | |
| 202 } | |
| 203 else { | |
| 204 /* not a gzip file -- save first byte (31) and fall to raw i/o */ | |
| 205 state->out[0] = 31; | |
| 206 state->have = 1; | |
| 207 } | |
| 208 } | 142 } |
| 209 | 143 |
| 210 /* doing raw i/o, save start of raw data for seeking, copy any leftover | 144 /* no gzip header -- if we were decoding gzip before, then this is trailing |
| 211 input to output -- this assumes that the output buffer is larger than | 145 garbage. Ignore the trailing garbage and finish. */ |
| 212 the input buffer, which also assures space for gzungetc() */ | 146 if (state->direct == 0) { |
| 213 state->raw = state->pos; | 147 strm->avail_in = 0; |
| 214 state->next = state->out; | 148 state->eof = 1; |
| 149 state->x.have = 0; |
| 150 return 0; |
| 151 } |
| 152 |
| 153 /* doing raw i/o, copy any leftover input to output -- this assumes that |
| 154 the output buffer is larger than the input buffer, which also assures |
| 155 space for gzungetc() */ |
| 156 state->x.next = state->out; |
| 215 if (strm->avail_in) { | 157 if (strm->avail_in) { |
| 216 memcpy(state->next + state->have, strm->next_in, strm->avail_in); | 158 memcpy(state->x.next, strm->next_in, strm->avail_in); |
| 217 state->have += strm->avail_in; | 159 state->x.have = strm->avail_in; |
| 218 strm->avail_in = 0; | 160 strm->avail_in = 0; |
| 219 } | 161 } |
| 220 state->how = COPY; | 162 state->how = COPY; |
| 221 state->direct = 1; | 163 state->direct = 1; |
| 222 return 0; | 164 return 0; |
| 223 } | 165 } |
| 224 | 166 |
| 225 /* Decompress from input to the provided next_out and avail_out in the state. | 167 /* Decompress from input to the provided next_out and avail_out in the state. |
| 226 If the end of the compressed data is reached, then verify the gzip trailer | 168 On return, state->x.have and state->x.next point to the just decompressed |
| 227 check value and length (modulo 2^32). state->have and state->next are set | 169 data. If the gzip stream completes, state->how is reset to LOOK to look for |
| 228 to point to the just decompressed data, and the crc is updated. If the | 170 the next gzip stream or raw data, once state->x.have is depleted. Returns 0 |
| 229 trailer is verified, state->how is reset to LOOK to look for the next gzip | 171 on success, -1 on failure. */ |
| 230 stream or raw data, once state->have is depleted. Returns 0 on success, -1 | |
| 231 on failure. Failures may include invalid compressed data or a failed gzip | |
| 232 trailer verification. */ | |
| 233 local int gz_decomp(state) | 172 local int gz_decomp(state) |
| 234 gz_statep state; | 173 gz_statep state; |
| 235 { | 174 { |
| 236 int ret; | 175 int ret = Z_OK; |
| 237 unsigned had; | 176 unsigned had; |
| 238 unsigned long crc, len; | |
| 239 z_streamp strm = &(state->strm); | 177 z_streamp strm = &(state->strm); |
| 240 | 178 |
| 241 /* fill output buffer up to end of deflate stream */ | 179 /* fill output buffer up to end of deflate stream */ |
| 242 had = strm->avail_out; | 180 had = strm->avail_out; |
| 243 do { | 181 do { |
| 244 /* get more input for inflate() */ | 182 /* get more input for inflate() */ |
| 245 if (strm->avail_in == 0 && gz_avail(state) == -1) | 183 if (strm->avail_in == 0 && gz_avail(state) == -1) |
| 246 return -1; | 184 return -1; |
| 247 if (strm->avail_in == 0) { | 185 if (strm->avail_in == 0) { |
| 248 gz_error(state, Z_DATA_ERROR, "unexpected end of file"); | 186 gz_error(state, Z_BUF_ERROR, "unexpected end of file"); |
| 249 return -1; | 187 break; |
| 250 } | 188 } |
| 251 | 189 |
| 252 /* decompress and handle errors */ | 190 /* decompress and handle errors */ |
| 253 ret = inflate(strm, Z_NO_FLUSH); | 191 ret = inflate(strm, Z_NO_FLUSH); |
| 254 if (ret == Z_STREAM_ERROR || ret == Z_NEED_DICT) { | 192 if (ret == Z_STREAM_ERROR || ret == Z_NEED_DICT) { |
| 255 gz_error(state, Z_STREAM_ERROR, | 193 gz_error(state, Z_STREAM_ERROR, |
| 256 "internal error: inflate stream corrupt"); | 194 "internal error: inflate stream corrupt"); |
| 257 return -1; | 195 return -1; |
| 258 } | 196 } |
| 259 if (ret == Z_MEM_ERROR) { | 197 if (ret == Z_MEM_ERROR) { |
| 260 gz_error(state, Z_MEM_ERROR, "out of memory"); | 198 gz_error(state, Z_MEM_ERROR, "out of memory"); |
| 261 return -1; | 199 return -1; |
| 262 } | 200 } |
| 263 if (ret == Z_DATA_ERROR) { /* deflate stream invalid */ | 201 if (ret == Z_DATA_ERROR) { /* deflate stream invalid */ |
| 264 gz_error(state, Z_DATA_ERROR, | 202 gz_error(state, Z_DATA_ERROR, |
| 265 strm->msg == NULL ? "compressed data error" : strm->msg); | 203 strm->msg == NULL ? "compressed data error" : strm->msg); |
| 266 return -1; | 204 return -1; |
| 267 } | 205 } |
| 268 } while (strm->avail_out && ret != Z_STREAM_END); | 206 } while (strm->avail_out && ret != Z_STREAM_END); |
| 269 | 207 |
| 270 /* update available output and crc check value */ | 208 /* update available output */ |
| 271 state->have = had - strm->avail_out; | 209 state->x.have = had - strm->avail_out; |
| 272 state->next = strm->next_out - state->have; | 210 state->x.next = strm->next_out - state->x.have; |
| 273 strm->adler = crc32(strm->adler, state->next, state->have); | |
| 274 | 211 |
| 275 /* check gzip trailer if at end of deflate stream */ | 212 /* if the gzip stream completed successfully, look for another */ |
| 276 if (ret == Z_STREAM_END) { | 213 if (ret == Z_STREAM_END) |
| 277 if (gz_next4(state, &crc) == -1 || gz_next4(state, &len) == -1) { | 214 state->how = LOOK; |
| 278 gz_error(state, Z_DATA_ERROR, "unexpected end of file"); | |
| 279 return -1; | |
| 280 } | |
| 281 if (crc != strm->adler) { | |
| 282 gz_error(state, Z_DATA_ERROR, "incorrect data check"); | |
| 283 return -1; | |
| 284 } | |
| 285 if (len != (strm->total_out & 0xffffffffL)) { | |
| 286 gz_error(state, Z_DATA_ERROR, "incorrect length check"); | |
| 287 return -1; | |
| 288 } | |
| 289 state->how = LOOK; /* ready for next stream, once have is 0 (leave | |
| 290 state->direct unchanged to remember how) */ | |
| 291 } | |
| 292 | 215 |
| 293 /* good decompression */ | 216 /* good decompression */ |
| 294 return 0; | 217 return 0; |
| 295 } | 218 } |
| 296 | 219 |
| 297 /* Make data and put in the output buffer. Assumes that state->have == 0. | 220 /* Fetch data and put it in the output buffer. Assumes state->x.have is 0. |
| 298 Data is either copied from the input file or decompressed from the input | 221 Data is either copied from the input file or decompressed from the input |
| 299 file depending on state->how. If state->how is LOOK, then a gzip header is | 222 file depending on state->how. If state->how is LOOK, then a gzip header is |
| 300 looked for (and skipped if found) to determine wither to copy or decompress. | 223 looked for to determine whether to copy or decompress. Returns -1 on error, |
| 301 Returns -1 on error, otherwise 0. gz_make() will leave state->have as COPY | 224 otherwise 0. gz_fetch() will leave state->how as COPY or GZIP unless the |
| 302 or GZIP unless the end of the input file has been reached and all data has | 225 end of the input file has been reached and all data has been processed. */ |
| 303 been processed. */ | 226 local int gz_fetch(state) |
| 304 local int gz_make(state) | |
| 305 gz_statep state; | 227 gz_statep state; |
| 306 { | 228 { |
| 307 z_streamp strm = &(state->strm); | 229 z_streamp strm = &(state->strm); |
| 308 | 230 |
| 309 if (state->how == LOOK) { /* look for gzip header */ | 231 do { |
| 310 if (gz_head(state) == -1) | 232 switch(state->how) { |
| 311 return -1; | 233 case LOOK: /* -> LOOK, COPY (only if never GZIP), or GZIP */ |
| 312 if (state->have) /* got some data from gz_head() */ | 234 if (gz_look(state) == -1) |
| 235 return -1; |
| 236 if (state->how == LOOK) |
| 237 return 0; |
| 238 break; |
| 239 case COPY: /* -> COPY */ |
| 240 if (gz_load(state, state->out, state->size << 1, &(state->x.have)) |
| 241 == -1) |
| 242 return -1; |
| 243 state->x.next = state->out; |
| 313 return 0; | 244 return 0; |
| 314 } | 245 case GZIP: /* -> GZIP or LOOK (if end of gzip stream) */ |
| 315 if (state->how == COPY) { /* straight copy */ | 246 strm->avail_out = state->size << 1; |
| 316 if (gz_load(state, state->out, state->size << 1, &(state->have)) == -1) | 247 strm->next_out = state->out; |
| 317 return -1; | 248 if (gz_decomp(state) == -1) |
| 318 state->next = state->out; | 249 return -1; |
| 319 } | 250 } |
| 320 else if (state->how == GZIP) { /* decompress */ | 251 } while (state->x.have == 0 && (!state->eof || strm->avail_in)); |
| 321 strm->avail_out = state->size << 1; | |
| 322 strm->next_out = state->out; | |
| 323 if (gz_decomp(state) == -1) | |
| 324 return -1; | |
| 325 } | |
| 326 return 0; | 252 return 0; |
| 327 } | 253 } |
| 328 | 254 |
| 329 /* Skip len uncompressed bytes of output. Return -1 on error, 0 on success. */ | 255 /* Skip len uncompressed bytes of output. Return -1 on error, 0 on success. */ |
| 330 local int gz_skip(state, len) | 256 local int gz_skip(state, len) |
| 331 gz_statep state; | 257 gz_statep state; |
| 332 z_off64_t len; | 258 z_off64_t len; |
| 333 { | 259 { |
| 334 unsigned n; | 260 unsigned n; |
| 335 | 261 |
| 336 /* skip over len bytes or reach end-of-file, whichever comes first */ | 262 /* skip over len bytes or reach end-of-file, whichever comes first */ |
| 337 while (len) | 263 while (len) |
| 338 /* skip over whatever is in output buffer */ | 264 /* skip over whatever is in output buffer */ |
| 339 if (state->have) { | 265 if (state->x.have) { |
| 340 n = GT_OFF(state->have) || (z_off64_t)state->have > len ? | 266 n = GT_OFF(state->x.have) || (z_off64_t)state->x.have > len ? |
| 341 (unsigned)len : state->have; | 267 (unsigned)len : state->x.have; |
| 342 state->have -= n; | 268 state->x.have -= n; |
| 343 state->next += n; | 269 state->x.next += n; |
| 344 state->pos += n; | 270 state->x.pos += n; |
| 345 len -= n; | 271 len -= n; |
| 346 } | 272 } |
| 347 | 273 |
| 348 /* output buffer empty -- return if we're at the end of the input */ | 274 /* output buffer empty -- return if we're at the end of the input */ |
| 349 else if (state->eof && state->strm.avail_in == 0) | 275 else if (state->eof && state->strm.avail_in == 0) |
| 350 break; | 276 break; |
| 351 | 277 |
| 352 /* need more data to skip -- load up output buffer */ | 278 /* need more data to skip -- load up output buffer */ |
| 353 else { | 279 else { |
| 354 /* get more output, looking for header if required */ | 280 /* get more output, looking for header if required */ |
| 355 if (gz_make(state) == -1) | 281 if (gz_fetch(state) == -1) |
| 356 return -1; | 282 return -1; |
| 357 } | 283 } |
| 358 return 0; | 284 return 0; |
| 359 } | 285 } |
| 360 | 286 |
| 361 /* -- see zlib.h -- */ | 287 /* -- see zlib.h -- */ |
| 362 int ZEXPORT gzread(file, buf, len) | 288 int ZEXPORT gzread(file, buf, len) |
| 363 gzFile file; | 289 gzFile file; |
| 364 voidp buf; | 290 voidp buf; |
| 365 unsigned len; | 291 unsigned len; |
| 366 { | 292 { |
| 367 unsigned got, n; | 293 unsigned got, n; |
| 368 gz_statep state; | 294 gz_statep state; |
| 369 z_streamp strm; | 295 z_streamp strm; |
| 370 | 296 |
| 371 /* get internal structure */ | 297 /* get internal structure */ |
| 372 if (file == NULL) | 298 if (file == NULL) |
| 373 return -1; | 299 return -1; |
| 374 state = (gz_statep)file; | 300 state = (gz_statep)file; |
| 375 strm = &(state->strm); | 301 strm = &(state->strm); |
| 376 | 302 |
| 377 /* check that we're reading and that there's no error */ | 303 /* check that we're reading and that there's no (serious) error */ |
| 378 if (state->mode != GZ_READ || state->err != Z_OK) | 304 if (state->mode != GZ_READ || |
| 305 (state->err != Z_OK && state->err != Z_BUF_ERROR)) |
| 379 return -1; | 306 return -1; |
| 380 | 307 |
| 381 /* since an int is returned, make sure len fits in one, otherwise return | 308 /* since an int is returned, make sure len fits in one, otherwise return |
| 382 with an error (this avoids the flaw in the interface) */ | 309 with an error (this avoids the flaw in the interface) */ |
| 383 if ((int)len < 0) { | 310 if ((int)len < 0) { |
| 384 gz_error(state, Z_BUF_ERROR, "requested length does not fit in int"); | 311 gz_error(state, Z_DATA_ERROR, "requested length does not fit in int"); |
| 385 return -1; | 312 return -1; |
| 386 } | 313 } |
| 387 | 314 |
| 388 /* if len is zero, avoid unnecessary operations */ | 315 /* if len is zero, avoid unnecessary operations */ |
| 389 if (len == 0) | 316 if (len == 0) |
| 390 return 0; | 317 return 0; |
| 391 | 318 |
| 392 /* process a skip request */ | 319 /* process a skip request */ |
| 393 if (state->seek) { | 320 if (state->seek) { |
| 394 state->seek = 0; | 321 state->seek = 0; |
| 395 if (gz_skip(state, state->skip) == -1) | 322 if (gz_skip(state, state->skip) == -1) |
| 396 return -1; | 323 return -1; |
| 397 } | 324 } |
| 398 | 325 |
| 399 /* get len bytes to buf, or less than len if at the end */ | 326 /* get len bytes to buf, or less than len if at the end */ |
| 400 got = 0; | 327 got = 0; |
| 401 do { | 328 do { |
| 402 /* first just try copying data from the output buffer */ | 329 /* first just try copying data from the output buffer */ |
| 403 if (state->have) { | 330 if (state->x.have) { |
| 404 n = state->have > len ? len : state->have; | 331 n = state->x.have > len ? len : state->x.have; |
| 405 memcpy(buf, state->next, n); | 332 memcpy(buf, state->x.next, n); |
| 406 state->next += n; | 333 state->x.next += n; |
| 407 state->have -= n; | 334 state->x.have -= n; |
| 408 } | 335 } |
| 409 | 336 |
| 410 /* output buffer empty -- return if we're at the end of the input */ | 337 /* output buffer empty -- return if we're at the end of the input */ |
| 411 else if (state->eof && strm->avail_in == 0) | 338 else if (state->eof && strm->avail_in == 0) { |
| 339 state->past = 1; /* tried to read past end */ |
| 412 break; | 340 break; |
| 341 } |
| 413 | 342 |
| 414 /* need output data -- for small len or new stream load up our output | 343 /* need output data -- for small len or new stream load up our output |
| 415 buffer */ | 344 buffer */ |
| 416 else if (state->how == LOOK || len < (state->size << 1)) { | 345 else if (state->how == LOOK || len < (state->size << 1)) { |
| 417 /* get more output, looking for header if required */ | 346 /* get more output, looking for header if required */ |
| 418 if (gz_make(state) == -1) | 347 if (gz_fetch(state) == -1) |
| 419 return -1; | 348 return -1; |
| 420 continue; /* no progress yet -- go back to memcpy() above */ | 349 continue; /* no progress yet -- go back to copy above */ |
| 421 /* the copy above assures that we will leave with space in the | 350 /* the copy above assures that we will leave with space in the |
| 422 output buffer, allowing at least one gzungetc() to succeed */ | 351 output buffer, allowing at least one gzungetc() to succeed */ |
| 423 } | 352 } |
| 424 | 353 |
| 425 /* large len -- read directly into user buffer */ | 354 /* large len -- read directly into user buffer */ |
| 426 else if (state->how == COPY) { /* read directly */ | 355 else if (state->how == COPY) { /* read directly */ |
| 427 if (gz_load(state, buf, len, &n) == -1) | 356 if (gz_load(state, (unsigned char *)buf, len, &n) == -1) |
| 428 return -1; | 357 return -1; |
| 429 } | 358 } |
| 430 | 359 |
| 431 /* large len -- decompress directly into user buffer */ | 360 /* large len -- decompress directly into user buffer */ |
| 432 else { /* state->how == GZIP */ | 361 else { /* state->how == GZIP */ |
| 433 strm->avail_out = len; | 362 strm->avail_out = len; |
| 434 strm->next_out = buf; | 363 strm->next_out = (unsigned char *)buf; |
| 435 if (gz_decomp(state) == -1) | 364 if (gz_decomp(state) == -1) |
| 436 return -1; | 365 return -1; |
| 437 n = state->have; | 366 n = state->x.have; |
| 438 state->have = 0; | 367 state->x.have = 0; |
| 439 } | 368 } |
| 440 | 369 |
| 441 /* update progress */ | 370 /* update progress */ |
| 442 len -= n; | 371 len -= n; |
| 443 buf = (char *)buf + n; | 372 buf = (char *)buf + n; |
| 444 got += n; | 373 got += n; |
| 445 state->pos += n; | 374 state->x.pos += n; |
| 446 } while (len); | 375 } while (len); |
| 447 | 376 |
| 448 /* return number of bytes read into user buffer (will fit in int) */ | 377 /* return number of bytes read into user buffer (will fit in int) */ |
| 449 return (int)got; | 378 return (int)got; |
| 450 } | 379 } |
| 451 | 380 |
| 452 /* -- see zlib.h -- */ | 381 /* -- see zlib.h -- */ |
| 382 #ifdef Z_PREFIX_SET |
| 383 # undef z_gzgetc |
| 384 #else |
| 385 # undef gzgetc |
| 386 #endif |
| 453 int ZEXPORT gzgetc(file) | 387 int ZEXPORT gzgetc(file) |
| 454 gzFile file; | 388 gzFile file; |
| 455 { | 389 { |
| 456 int ret; | 390 int ret; |
| 457 unsigned char buf[1]; | 391 unsigned char buf[1]; |
| 458 gz_statep state; | 392 gz_statep state; |
| 459 | 393 |
| 460 /* get internal structure */ | 394 /* get internal structure */ |
| 461 if (file == NULL) | 395 if (file == NULL) |
| 462 return -1; | 396 return -1; |
| 463 state = (gz_statep)file; | 397 state = (gz_statep)file; |
| 464 | 398 |
| 465 /* check that we're reading and that there's no error */ | 399 /* check that we're reading and that there's no (serious) error */ |
| 466 if (state->mode != GZ_READ || state->err != Z_OK) | 400 if (state->mode != GZ_READ || |
| 401 (state->err != Z_OK && state->err != Z_BUF_ERROR)) |
| 467 return -1; | 402 return -1; |
| 468 | 403 |
| 469 /* try output buffer (no need to check for skip request) */ | 404 /* try output buffer (no need to check for skip request) */ |
| 470 if (state->have) { | 405 if (state->x.have) { |
| 471 state->have--; | 406 state->x.have--; |
| 472 state->pos++; | 407 state->x.pos++; |
| 473 return *(state->next)++; | 408 return *(state->x.next)++; |
| 474 } | 409 } |
| 475 | 410 |
| 476 /* nothing there -- try gzread() */ | 411 /* nothing there -- try gzread() */ |
| 477 ret = gzread(file, buf, 1); | 412 ret = gzread(file, buf, 1); |
| 478 return ret < 1 ? -1 : buf[0]; | 413 return ret < 1 ? -1 : buf[0]; |
| 479 } | 414 } |
| 480 | 415 |
| 416 int ZEXPORT gzgetc_(file) |
| 417 gzFile file; |
| 418 { |
| 419 return gzgetc(file); |
| 420 } |
| 421 |
| 481 /* -- see zlib.h -- */ | 422 /* -- see zlib.h -- */ |
| 482 int ZEXPORT gzungetc(c, file) | 423 int ZEXPORT gzungetc(c, file) |
| 483 int c; | 424 int c; |
| 484 gzFile file; | 425 gzFile file; |
| 485 { | 426 { |
| 486 gz_statep state; | 427 gz_statep state; |
| 487 | 428 |
| 488 /* get internal structure */ | 429 /* get internal structure */ |
| 489 if (file == NULL) | 430 if (file == NULL) |
| 490 return -1; | 431 return -1; |
| 491 state = (gz_statep)file; | 432 state = (gz_statep)file; |
| 492 | 433 |
| 493 /* check that we're reading and that there's no error */ | 434 /* check that we're reading and that there's no (serious) error */ |
| 494 if (state->mode != GZ_READ || state->err != Z_OK) | 435 if (state->mode != GZ_READ || |
| 436 (state->err != Z_OK && state->err != Z_BUF_ERROR)) |
| 495 return -1; | 437 return -1; |
| 496 | 438 |
| 497 /* process a skip request */ | 439 /* process a skip request */ |
| 498 if (state->seek) { | 440 if (state->seek) { |
| 499 state->seek = 0; | 441 state->seek = 0; |
| 500 if (gz_skip(state, state->skip) == -1) | 442 if (gz_skip(state, state->skip) == -1) |
| 501 return -1; | 443 return -1; |
| 502 } | 444 } |
| 503 | 445 |
| 504 /* can't push EOF */ | 446 /* can't push EOF */ |
| 505 if (c < 0) | 447 if (c < 0) |
| 506 return -1; | 448 return -1; |
| 507 | 449 |
| 508 /* if output buffer empty, put byte at end (allows more pushing) */ | 450 /* if output buffer empty, put byte at end (allows more pushing) */ |
| 509 if (state->have == 0) { | 451 if (state->x.have == 0) { |
| 510 state->have = 1; | 452 state->x.have = 1; |
| 511 state->next = state->out + (state->size << 1) - 1; | 453 state->x.next = state->out + (state->size << 1) - 1; |
| 512 state->next[0] = c; | 454 state->x.next[0] = c; |
| 513 state->pos--; | 455 state->x.pos--; |
| 456 state->past = 0; |
| 514 return c; | 457 return c; |
| 515 } | 458 } |
| 516 | 459 |
| 517 /* if no room, give up (must have already done a gzungetc()) */ | 460 /* if no room, give up (must have already done a gzungetc()) */ |
| 518 if (state->have == (state->size << 1)) { | 461 if (state->x.have == (state->size << 1)) { |
| 519 gz_error(state, Z_BUF_ERROR, "out of room to push characters"); | 462 gz_error(state, Z_DATA_ERROR, "out of room to push characters"); |
| 520 return -1; | 463 return -1; |
| 521 } | 464 } |
| 522 | 465 |
| 523 /* slide output data if needed and insert byte before existing data */ | 466 /* slide output data if needed and insert byte before existing data */ |
| 524 if (state->next == state->out) { | 467 if (state->x.next == state->out) { |
| 525 unsigned char *src = state->out + state->have; | 468 unsigned char *src = state->out + state->x.have; |
| 526 unsigned char *dest = state->out + (state->size << 1); | 469 unsigned char *dest = state->out + (state->size << 1); |
| 527 while (src > state->out) | 470 while (src > state->out) |
| 528 *--dest = *--src; | 471 *--dest = *--src; |
| 529 state->next = dest; | 472 state->x.next = dest; |
| 530 } | 473 } |
| 531 state->have++; | 474 state->x.have++; |
| 532 state->next--; | 475 state->x.next--; |
| 533 state->next[0] = c; | 476 state->x.next[0] = c; |
| 534 state->pos--; | 477 state->x.pos--; |
| 478 state->past = 0; |
| 535 return c; | 479 return c; |
| 536 } | 480 } |
| 537 | 481 |
| 538 /* -- see zlib.h -- */ | 482 /* -- see zlib.h -- */ |
| 539 char * ZEXPORT gzgets(file, buf, len) | 483 char * ZEXPORT gzgets(file, buf, len) |
| 540 gzFile file; | 484 gzFile file; |
| 541 char *buf; | 485 char *buf; |
| 542 int len; | 486 int len; |
| 543 { | 487 { |
| 544 unsigned left, n; | 488 unsigned left, n; |
| 545 char *str; | 489 char *str; |
| 546 unsigned char *eol; | 490 unsigned char *eol; |
| 547 gz_statep state; | 491 gz_statep state; |
| 548 | 492 |
| 549 /* check parameters and get internal structure */ | 493 /* check parameters and get internal structure */ |
| 550 if (file == NULL || buf == NULL || len < 1) | 494 if (file == NULL || buf == NULL || len < 1) |
| 551 return NULL; | 495 return NULL; |
| 552 state = (gz_statep)file; | 496 state = (gz_statep)file; |
| 553 | 497 |
| 554 /* check that we're reading and that there's no error */ | 498 /* check that we're reading and that there's no (serious) error */ |
| 555 if (state->mode != GZ_READ || state->err != Z_OK) | 499 if (state->mode != GZ_READ || |
| 500 (state->err != Z_OK && state->err != Z_BUF_ERROR)) |
| 556 return NULL; | 501 return NULL; |
| 557 | 502 |
| 558 /* process a skip request */ | 503 /* process a skip request */ |
| 559 if (state->seek) { | 504 if (state->seek) { |
| 560 state->seek = 0; | 505 state->seek = 0; |
| 561 if (gz_skip(state, state->skip) == -1) | 506 if (gz_skip(state, state->skip) == -1) |
| 562 return NULL; | 507 return NULL; |
| 563 } | 508 } |
| 564 | 509 |
| 565 /* copy output bytes up to new line or len - 1, whichever comes first -- | 510 /* copy output bytes up to new line or len - 1, whichever comes first -- |
| 566 append a terminating zero to the string (we don't check for a zero in | 511 append a terminating zero to the string (we don't check for a zero in |
| 567 the contents, let the user worry about that) */ | 512 the contents, let the user worry about that) */ |
| 568 str = buf; | 513 str = buf; |
| 569 left = (unsigned)len - 1; | 514 left = (unsigned)len - 1; |
| 570 if (left) do { | 515 if (left) do { |
| 571 /* assure that something is in the output buffer */ | 516 /* assure that something is in the output buffer */ |
| 572 if (state->have == 0) { | 517 if (state->x.have == 0 && gz_fetch(state) == -1) |
| 573 if (gz_make(state) == -1) | 518 return NULL; /* error */ |
| 574 return NULL; /* error */ | 519 if (state->x.have == 0) { /* end of file */ |
| 575 if (state->have == 0) { /* end of file */ | 520 state->past = 1; /* read past end */ |
| 576 if (buf == str) /* got bupkus */ | 521 break; /* return what we have */ |
| 577 return NULL; | |
| 578 break; /* got something -- return it */ | |
| 579 } | |
| 580 } | 522 } |
| 581 | 523 |
| 582 /* look for end-of-line in current output buffer */ | 524 /* look for end-of-line in current output buffer */ |
| 583 n = state->have > left ? left : state->have; | 525 n = state->x.have > left ? left : state->x.have; |
| 584 eol = memchr(state->next, '\n', n); | 526 eol = (unsigned char *)memchr(state->x.next, '\n', n); |
| 585 if (eol != NULL) | 527 if (eol != NULL) |
| 586 n = (unsigned)(eol - state->next) + 1; | 528 n = (unsigned)(eol - state->x.next) + 1; |
| 587 | 529 |
| 588 /* copy through end-of-line, or remainder if not found */ | 530 /* copy through end-of-line, or remainder if not found */ |
| 589 memcpy(buf, state->next, n); | 531 memcpy(buf, state->x.next, n); |
| 590 state->have -= n; | 532 state->x.have -= n; |
| 591 state->next += n; | 533 state->x.next += n; |
| 592 state->pos += n; | 534 state->x.pos += n; |
| 593 left -= n; | 535 left -= n; |
| 594 buf += n; | 536 buf += n; |
| 595 } while (left && eol == NULL); | 537 } while (left && eol == NULL); |
| 596 | 538 |
| 597 /* found end-of-line or out of space -- terminate string and return it */ | 539 /* return terminated string, or if nothing, end of file */ |
| 540 if (buf == str) |
| 541 return NULL; |
| 598 buf[0] = 0; | 542 buf[0] = 0; |
| 599 return str; | 543 return str; |
| 600 } | 544 } |
| 601 | 545 |
| 602 /* -- see zlib.h -- */ | 546 /* -- see zlib.h -- */ |
| 603 int ZEXPORT gzdirect(file) | 547 int ZEXPORT gzdirect(file) |
| 604 gzFile file; | 548 gzFile file; |
| 605 { | 549 { |
| 606 gz_statep state; | 550 gz_statep state; |
| 607 | 551 |
| 608 /* get internal structure */ | 552 /* get internal structure */ |
| 609 if (file == NULL) | 553 if (file == NULL) |
| 610 return 0; | 554 return 0; |
| 611 state = (gz_statep)file; | 555 state = (gz_statep)file; |
| 612 | 556 |
| 613 /* check that we're reading */ | |
| 614 if (state->mode != GZ_READ) | |
| 615 return 0; | |
| 616 | |
| 617 /* if the state is not known, but we can find out, then do so (this is | 557 /* if the state is not known, but we can find out, then do so (this is |
| 618 mainly for right after a gzopen() or gzdopen()) */ | 558 mainly for right after a gzopen() or gzdopen()) */ |
| 619 if (state->how == LOOK && state->have == 0) | 559 if (state->mode == GZ_READ && state->how == LOOK && state->x.have == 0) |
| 620 (void)gz_head(state); | 560 (void)gz_look(state); |
| 621 | 561 |
| 622 /* return 1 if reading direct, 0 if decompressing a gzip stream */ | 562 /* return 1 if transparent, 0 if processing a gzip stream */ |
| 623 return state->direct; | 563 return state->direct; |
| 624 } | 564 } |
| 625 | 565 |
| 626 /* -- see zlib.h -- */ | 566 /* -- see zlib.h -- */ |
| 627 int ZEXPORT gzclose_r(file) | 567 int ZEXPORT gzclose_r(file) |
| 628 gzFile file; | 568 gzFile file; |
| 629 { | 569 { |
| 630 int ret; | 570 int ret, err; |
| 631 gz_statep state; | 571 gz_statep state; |
| 632 | 572 |
| 633 /* get internal structure */ | 573 /* get internal structure */ |
| 634 if (file == NULL) | 574 if (file == NULL) |
| 635 return Z_STREAM_ERROR; | 575 return Z_STREAM_ERROR; |
| 636 state = (gz_statep)file; | 576 state = (gz_statep)file; |
| 637 | 577 |
| 638 /* check that we're reading */ | 578 /* check that we're reading */ |
| 639 if (state->mode != GZ_READ) | 579 if (state->mode != GZ_READ) |
| 640 return Z_STREAM_ERROR; | 580 return Z_STREAM_ERROR; |
| 641 | 581 |
| 642 /* free memory and close file */ | 582 /* free memory and close file */ |
| 643 if (state->size) { | 583 if (state->size) { |
| 644 inflateEnd(&(state->strm)); | 584 inflateEnd(&(state->strm)); |
| 645 free(state->out); | 585 free(state->out); |
| 646 free(state->in); | 586 free(state->in); |
| 647 } | 587 } |
| 588 err = state->err == Z_BUF_ERROR ? Z_BUF_ERROR : Z_OK; |
| 648 gz_error(state, Z_OK, NULL); | 589 gz_error(state, Z_OK, NULL); |
| 649 free(state->path); | 590 free(state->path); |
| 650 ret = close(state->fd); | 591 ret = close(state->fd); |
| 651 free(state); | 592 free(state); |
| 652 return ret ? Z_ERRNO : Z_OK; | 593 return ret ? Z_ERRNO : err; |
| 653 } | 594 } |
| OLD | NEW |