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