- gcc didn't like those changes
[m6w6/ext-http] / http_encoding_api.c
1 /*
2 +--------------------------------------------------------------------+
3 | PECL :: http |
4 +--------------------------------------------------------------------+
5 | Redistribution and use in source and binary forms, with or without |
6 | modification, are permitted provided that the conditions mentioned |
7 | in the accompanying LICENSE file are met. |
8 +--------------------------------------------------------------------+
9 | Copyright (c) 2004-2005, Michael Wallner <mike@php.net> |
10 +--------------------------------------------------------------------+
11 */
12
13 /* $Id$ */
14
15 #ifdef HAVE_CONFIG_H
16 # include "config.h"
17 #endif
18
19 #define HTTP_WANT_ZLIB
20 #include "php_http.h"
21
22 #include "php_http_api.h"
23 #include "php_http_encoding_api.h"
24 #include "php_http_send_api.h"
25 #include "php_http_headers_api.h"
26
27 ZEND_EXTERN_MODULE_GLOBALS(http);
28
29 #ifdef HTTP_HAVE_ZLIB
30 PHP_MINIT_FUNCTION(http_encoding)
31 {
32 HTTP_LONG_CONSTANT("HTTP_DEFLATE_LEVEL_DEF", HTTP_DEFLATE_LEVEL_DEF);
33 HTTP_LONG_CONSTANT("HTTP_DEFLATE_LEVEL_MIN", HTTP_DEFLATE_LEVEL_MIN);
34 HTTP_LONG_CONSTANT("HTTP_DEFLATE_LEVEL_MAX", HTTP_DEFLATE_LEVEL_MAX);
35 HTTP_LONG_CONSTANT("HTTP_DEFLATE_TYPE_ZLIB", HTTP_DEFLATE_TYPE_ZLIB);
36 HTTP_LONG_CONSTANT("HTTP_DEFLATE_TYPE_GZIP", HTTP_DEFLATE_TYPE_GZIP);
37 HTTP_LONG_CONSTANT("HTTP_DEFLATE_TYPE_RAW", HTTP_DEFLATE_TYPE_RAW);
38 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_DEF", HTTP_DEFLATE_STRATEGY_DEF);
39 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_FILT", HTTP_DEFLATE_STRATEGY_FILT);
40 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_HUFF", HTTP_DEFLATE_STRATEGY_HUFF);
41 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_RLE", HTTP_DEFLATE_STRATEGY_RLE);
42 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_FIXED", HTTP_DEFLATE_STRATEGY_FIXED);
43 return SUCCESS;
44 }
45 #endif
46
47 static inline int eol_match(char **line, int *eol_len)
48 {
49 char *ptr = *line;
50
51 while (0x20 == *ptr) ++ptr;
52
53 if (ptr == http_locate_eol(*line, eol_len)) {
54 *line = ptr;
55 return 1;
56 } else {
57 return 0;
58 }
59 }
60
61 /* {{{ char *http_encoding_dechunk(char *, size_t, char **, size_t *) */
62 PHP_HTTP_API const char *_http_encoding_dechunk(const char *encoded, size_t encoded_len, char **decoded, size_t *decoded_len TSRMLS_DC)
63 {
64 int eol_len = 0;
65 char *n_ptr = NULL;
66 const char *e_ptr = encoded;
67
68 *decoded_len = 0;
69 *decoded = ecalloc(1, encoded_len);
70
71 while ((encoded + encoded_len - e_ptr) > 0) {
72 ulong chunk_len = 0, rest;
73
74 chunk_len = strtoul(e_ptr, &n_ptr, 16);
75
76 /* we could not read in chunk size */
77 if (n_ptr == e_ptr) {
78 /*
79 * if this is the first turn and there doesn't seem to be a chunk
80 * size at the begining of the body, do not fail on apparently
81 * not encoded data and return a copy
82 */
83 if (e_ptr == encoded) {
84 http_error(HE_NOTICE, HTTP_E_ENCODING, "Data does not seem to be chunked encoded");
85 memcpy(*decoded, encoded, encoded_len);
86 *decoded_len = encoded_len;
87 return encoded + encoded_len;
88 } else {
89 efree(*decoded);
90 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Expected chunk size at pos %tu of %zu but got trash", n_ptr - encoded, encoded_len);
91 return NULL;
92 }
93 }
94
95 /* reached the end */
96 if (!chunk_len) {
97 /* move over '0' chunked encoding terminator */
98 while (*e_ptr == '0') ++e_ptr;
99 break;
100 }
101
102 /* there should be CRLF after the chunk size, but we'll ignore SP+ too */
103 if (*n_ptr && !eol_match(&n_ptr, &eol_len)) {
104 if (eol_len == 2) {
105 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Expected CRLF at pos %tu of %zu but got 0x%02X 0x%02X", n_ptr - encoded, encoded_len, *n_ptr, *(n_ptr + 1));
106 } else {
107 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Expected LF at pos %tu of %zu but got 0x%02X", n_ptr - encoded, encoded_len, *n_ptr);
108 }
109 }
110 n_ptr += eol_len;
111
112 /* chunk size pretends more data than we actually got, so it's probably a truncated message */
113 if (chunk_len > (rest = encoded + encoded_len - n_ptr)) {
114 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Truncated message: chunk size %lu exceeds remaining data size %lu at pos %tu of %zu", chunk_len, rest, n_ptr - encoded, encoded_len);
115 chunk_len = rest;
116 }
117
118 /* copy the chunk */
119 memcpy(*decoded + *decoded_len, n_ptr, chunk_len);
120 *decoded_len += chunk_len;
121
122 if (chunk_len == rest) {
123 e_ptr = n_ptr + chunk_len;
124 break;
125 } else {
126 /* advance to next chunk */
127 e_ptr = n_ptr + chunk_len + eol_len;
128 }
129 }
130
131 return e_ptr;
132 }
133 /* }}} */
134
135 #ifdef HTTP_HAVE_ZLIB
136
137 #define HTTP_DEFLATE_LEVEL_SET(flags, level) \
138 switch (flags & 0xf) \
139 { \
140 default: \
141 if ((flags & 0xf) < 10) { \
142 level = flags & 0xf; \
143 break; \
144 } \
145 case HTTP_DEFLATE_LEVEL_DEF: \
146 level = Z_DEFAULT_COMPRESSION; \
147 break; \
148 }
149
150 #define HTTP_DEFLATE_WBITS_SET(flags, wbits) \
151 switch (flags & 0xf0) \
152 { \
153 case HTTP_DEFLATE_TYPE_GZIP: \
154 wbits = HTTP_WINDOW_BITS_GZIP; \
155 break; \
156 case HTTP_DEFLATE_TYPE_RAW: \
157 wbits = HTTP_WINDOW_BITS_RAW; \
158 break; \
159 default: \
160 wbits = HTTP_WINDOW_BITS_ZLIB; \
161 break; \
162 }
163
164 #define HTTP_INFLATE_WBITS_SET(flags, wbits) \
165 if (flags & HTTP_INFLATE_TYPE_RAW) { \
166 wbits = HTTP_WINDOW_BITS_RAW; \
167 } else { \
168 wbits = HTTP_WINDOW_BITS_ANY; \
169 }
170
171 #define HTTP_DEFLATE_STRATEGY_SET(flags, strategy) \
172 switch (flags & 0xf00) \
173 { \
174 case HTTP_DEFLATE_STRATEGY_FILT: \
175 strategy = Z_FILTERED; \
176 break; \
177 case HTTP_DEFLATE_STRATEGY_HUFF: \
178 strategy = Z_HUFFMAN_ONLY; \
179 break; \
180 case HTTP_DEFLATE_STRATEGY_RLE: \
181 strategy = Z_RLE; \
182 break; \
183 case HTTP_DEFLATE_STRATEGY_FIXED: \
184 strategy = Z_FIXED; \
185 break; \
186 default: \
187 strategy = Z_DEFAULT_STRATEGY; \
188 break; \
189 }
190
191 #define HTTP_WINDOW_BITS_ZLIB 0x0000000f
192 #define HTTP_WINDOW_BITS_GZIP 0x0000001f
193 #define HTTP_WINDOW_BITS_ANY 0x0000002f
194 #define HTTP_WINDOW_BITS_RAW -0x000000f
195
196 PHP_HTTP_API STATUS _http_encoding_deflate(int flags, const char *data, size_t data_len, char **encoded, size_t *encoded_len TSRMLS_DC)
197 {
198 int status, level, wbits, strategy;
199 z_stream Z;
200
201 HTTP_DEFLATE_LEVEL_SET(flags, level);
202 HTTP_DEFLATE_WBITS_SET(flags, wbits);
203 HTTP_DEFLATE_STRATEGY_SET(flags, strategy);
204
205 memset(&Z, 0, sizeof(z_stream));
206 *encoded = NULL;
207 *encoded_len = 0;
208
209 status = deflateInit2(&Z, level, Z_DEFLATED, wbits, MAX_MEM_LEVEL, strategy);
210 if (Z_OK == status) {
211 *encoded_len = HTTP_ENCODING_BUFLEN(data_len);
212 *encoded = emalloc(*encoded_len);
213
214 Z.next_in = (Bytef *) data;
215 Z.next_out = (Bytef *) *encoded;
216 Z.avail_in = data_len;
217 Z.avail_out = *encoded_len;
218
219 status = deflate(&Z, Z_FINISH);
220 deflateEnd(&Z);
221
222 if (Z_STREAM_END == status) {
223 /* size buffer down to actual length */
224 *encoded = erealloc(*encoded, Z.total_out + 1);
225 (*encoded)[*encoded_len = Z.total_out] = '\0';
226 return SUCCESS;
227 } else {
228 STR_SET(*encoded, NULL);
229 *encoded_len = 0;
230 }
231 }
232
233 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Could not deflate data: %s (%s)", zError(status));
234 return FAILURE;
235 }
236
237 PHP_HTTP_API STATUS _http_encoding_inflate(const char *data, size_t data_len, char **decoded, size_t *decoded_len TSRMLS_DC)
238 {
239 int status, max = 0, wbits = HTTP_WINDOW_BITS_ANY;
240 z_stream Z;
241 phpstr buffer;
242
243 memset(&Z, 0, sizeof(z_stream));
244 *decoded = NULL;
245 *decoded_len = 0;
246
247 phpstr_init_ex(&buffer, data_len << 2, PHPSTR_INIT_PREALLOC);
248 buffer.size = data_len;
249
250 retry_inflate:
251 status = inflateInit2(&Z, wbits);
252 if (Z_OK == status) {
253 Z.next_in = (Bytef *) data;
254 Z.avail_in = data_len;
255
256 do {
257 phpstr_resize(&buffer, data_len << 2);
258
259 do {
260 Z.avail_out = (buffer.free -= Z.total_out - buffer.used);
261 Z.next_out = (Bytef *) buffer.data + (buffer.used = Z.total_out);
262 status = inflate(&Z, Z_NO_FLUSH);
263 } while (Z_OK == status);
264 } while (Z_BUF_ERROR == status && ++max < HTTP_ENCODING_MAXTRY);
265
266 if (Z_DATA_ERROR == status && HTTP_WINDOW_BITS_ANY == wbits) {
267 /* raw deflated data? */
268 inflateEnd(&Z);
269 wbits = HTTP_WINDOW_BITS_RAW;
270 goto retry_inflate;
271 }
272
273 inflateEnd(&Z);
274
275 if (Z_STREAM_END == status) {
276 *decoded_len = Z.total_out;
277 *decoded = erealloc(buffer.data, *decoded_len + 1);
278 (*decoded)[*decoded_len] = '\0';
279 return SUCCESS;
280 } else {
281 phpstr_dtor(&buffer);
282 }
283 }
284
285 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Could not inflate data: %s", zError(status));
286 return FAILURE;
287 }
288
289
290 PHP_HTTP_API http_encoding_stream *_http_encoding_deflate_stream_init(http_encoding_stream *s, int flags TSRMLS_DC)
291 {
292 int status, level, wbits, strategy, free_stream;
293
294 if ((free_stream = !s)) {
295 s = pemalloc(sizeof(http_encoding_stream), (flags & HTTP_ENCODING_STREAM_PERSISTENT));
296 }
297 memset(s, 0, sizeof(http_encoding_stream));
298 s->flags = flags;
299
300 HTTP_DEFLATE_LEVEL_SET(flags, level);
301 HTTP_DEFLATE_WBITS_SET(flags, wbits);
302 HTTP_DEFLATE_STRATEGY_SET(flags, strategy);
303
304 if (Z_OK == (status = deflateInit2(&s->stream, level, Z_DEFLATED, wbits, MAX_MEM_LEVEL, strategy))) {
305 int p = (flags & HTTP_ENCODING_STREAM_PERSISTENT) ? PHPSTR_INIT_PERSISTENT:0;
306
307 if ((s->stream.opaque = phpstr_init_ex(NULL, 0x8000, p))) {
308 return s;
309 }
310 deflateEnd(&s->stream);
311 status = Z_MEM_ERROR;
312 }
313
314 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to initialize deflate encoding stream: %s", zError(status));
315 if (free_stream) {
316 efree(s);
317 }
318 return NULL;
319 }
320
321 PHP_HTTP_API http_encoding_stream *_http_encoding_inflate_stream_init(http_encoding_stream *s, int flags TSRMLS_DC)
322 {
323 int status, wbits, free_stream;
324
325 if ((free_stream = !s)) {
326 s = emalloc(sizeof(http_encoding_stream));
327 }
328 memset(s, 0, sizeof(http_encoding_stream));
329 s->flags = flags;
330
331 HTTP_INFLATE_WBITS_SET(flags, wbits);
332
333 if (Z_OK == (status = inflateInit2(&s->stream, wbits))) {
334 int p = (flags & HTTP_ENCODING_STREAM_PERSISTENT) ? PHPSTR_INIT_PERSISTENT:0;
335
336 if ((s->stream.opaque = phpstr_init_ex(NULL, 0x8000, p))) {
337 return s;
338 }
339 inflateEnd(&s->stream);
340 status = Z_MEM_ERROR;
341 }
342
343 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to initialize inflate stream: %s", zError(status));
344 if (free_stream) {
345 efree(s);
346 }
347 return NULL;
348 }
349
350 PHP_HTTP_API STATUS _http_encoding_deflate_stream_update(http_encoding_stream *s, const char *data, size_t data_len, char **encoded, size_t *encoded_len TSRMLS_DC)
351 {
352 int status;
353
354 /* append input to our buffer */
355 phpstr_append(PHPSTR(s->stream.opaque), data, data_len);
356
357 s->stream.next_in = (Bytef *) PHPSTR_VAL(s->stream.opaque);
358 s->stream.avail_in = PHPSTR_LEN(s->stream.opaque);
359
360 /* deflate */
361 *encoded_len = HTTP_ENCODING_BUFLEN(data_len);
362 *encoded = emalloc(*encoded_len);
363 s->stream.avail_out = *encoded_len;
364 s->stream.next_out = (Bytef *) *encoded;
365
366 switch (status = deflate(&s->stream, Z_NO_FLUSH))
367 {
368 case Z_OK:
369 case Z_STREAM_END:
370 /* cut processed chunk off the buffer */
371 phpstr_cut(PHPSTR(s->stream.opaque), 0, data_len - s->stream.avail_in);
372
373 /* size buffer down to actual size */
374 *encoded_len -= s->stream.avail_out;
375 *encoded = erealloc(*encoded, *encoded_len + 1);
376 (*encoded)[*encoded_len] = '\0';
377 return SUCCESS;
378 break;
379 }
380
381 STR_SET(*encoded, NULL);
382 *encoded_len = 0;
383 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to update deflate stream: %s", zError(status));
384 return FAILURE;
385 }
386
387 PHP_HTTP_API STATUS _http_encoding_inflate_stream_update(http_encoding_stream *s, const char *data, size_t data_len, char **decoded, size_t *decoded_len TSRMLS_DC)
388 {
389 int status, max = 0;
390
391 /* append input to buffer */
392 phpstr_append(PHPSTR(s->stream.opaque), data, data_len);
393
394 /* for realloc() */
395 *decoded = NULL;
396 *decoded_len = data_len << 1;
397
398 /* inflate */
399 do {
400 *decoded_len <<= 1;
401 *decoded = erealloc(*decoded, *decoded_len);
402
403 retry_raw_inflate:
404 s->stream.next_in = (Bytef *) PHPSTR_VAL(s->stream.opaque);
405 s->stream.avail_in = PHPSTR_LEN(s->stream.opaque);
406
407 s->stream.next_out = (Bytef *) *decoded;
408 s->stream.avail_out = *decoded_len;
409
410 switch (status = inflate(&s->stream, Z_NO_FLUSH))
411 {
412 case Z_OK:
413 case Z_STREAM_END:
414 /* cut off */
415 phpstr_cut(PHPSTR(s->stream.opaque), 0, data_len - s->stream.avail_in);
416
417 /* size down */
418 *decoded_len -= s->stream.avail_out;
419 *decoded = erealloc(*decoded, *decoded_len + 1);
420 (*decoded)[*decoded_len] = '\0';
421 return SUCCESS;
422 break;
423
424 case Z_DATA_ERROR:
425 /* raw deflated data ? */
426 if (!(s->flags & HTTP_INFLATE_TYPE_RAW) && !s->stream.total_out) {
427 inflateEnd(&s->stream);
428 s->flags |= HTTP_INFLATE_TYPE_RAW;
429 inflateInit2(&s->stream, HTTP_WINDOW_BITS_RAW);
430 goto retry_raw_inflate;
431 }
432 break;
433 }
434 } while (Z_BUF_ERROR == status && ++max < HTTP_ENCODING_MAXTRY);
435
436 STR_SET(*decoded, NULL);
437 *decoded_len = 0;
438 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to update inflate stream: %s", zError(status));
439 return FAILURE;
440 }
441
442 PHP_HTTP_API STATUS _http_encoding_deflate_stream_flush(http_encoding_stream *s, char **encoded, size_t *encoded_len TSRMLS_DC)
443 {
444 int status;
445
446 *encoded_len = 0x800;
447 *encoded = emalloc(*encoded_len);
448
449 s->stream.avail_in = 0;
450 s->stream.next_in = NULL;
451 s->stream.avail_out = *encoded_len;
452 s->stream.next_out = (Bytef *) *encoded;
453
454 switch (status = deflate(&s->stream, Z_SYNC_FLUSH))
455 {
456 case Z_OK:
457 case Z_STREAM_END:
458 *encoded_len = 0x800 - s->stream.avail_out;
459 *encoded = erealloc(*encoded, *encoded_len + 1);
460 (*encoded)[*encoded_len] = '\0';
461 return SUCCESS;
462 break;
463 }
464
465 STR_SET(*encoded, NULL);
466 *encoded_len = 0;
467 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to flush deflate stream: %s", zError(status));
468 return FAILURE;
469 }
470
471 PHP_HTTP_API STATUS _http_encoding_inflate_stream_flush(http_encoding_stream *s, char **decoded, size_t *decoded_len TSRMLS_DC)
472 {
473 int status;
474
475 *decoded_len = 0x800;
476 *decoded = emalloc(*decoded_len);
477
478 s->stream.avail_in = 0;
479 s->stream.next_in = NULL;
480 s->stream.avail_out = *decoded_len;
481 s->stream.next_out = (Bytef *) *decoded;
482
483 switch (status = inflate(&s->stream, Z_SYNC_FLUSH))
484 {
485 case Z_OK:
486 case Z_STREAM_END:
487 *decoded_len = 0x800 - s->stream.avail_out;
488 *decoded = erealloc(*decoded, *decoded_len + 1);
489 (*decoded)[*decoded_len] = '\0';
490 return SUCCESS;
491 break;
492 }
493
494 STR_SET(*decoded, NULL);
495 *decoded_len = 0;
496 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to flush inflate stream: %s", zError(status));
497 return FAILURE;
498 }
499
500 PHP_HTTP_API STATUS _http_encoding_deflate_stream_finish(http_encoding_stream *s, char **encoded, size_t *encoded_len TSRMLS_DC)
501 {
502 int status;
503
504 *encoded_len = 0x800;
505 *encoded = emalloc(*encoded_len);
506
507 /* deflate remaining input */
508 s->stream.next_in = (Bytef *) PHPSTR_VAL(s->stream.opaque);
509 s->stream.avail_in = PHPSTR_LEN(s->stream.opaque);
510
511 s->stream.avail_out = *encoded_len;
512 s->stream.next_out = (Bytef *) *encoded;
513
514 do {
515 status = deflate(&s->stream, Z_FINISH);
516 } while (Z_OK == status);
517
518 if (Z_STREAM_END == status) {
519 /* cut processed intp off */
520 phpstr_cut(PHPSTR(s->stream.opaque), 0, PHPSTR_LEN(s->stream.opaque) - s->stream.avail_in);
521
522 /* size down */
523 *encoded_len -= s->stream.avail_out;
524 *encoded = erealloc(*encoded, *encoded_len + 1);
525 (*encoded)[*encoded_len] = '\0';
526 return SUCCESS;
527 }
528
529 STR_SET(*encoded, NULL);
530 *encoded_len = 0;
531 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to finish deflate stream: %s", zError(status));
532 return FAILURE;
533 }
534
535 PHP_HTTP_API STATUS _http_encoding_inflate_stream_finish(http_encoding_stream *s, char **decoded, size_t *decoded_len TSRMLS_DC)
536 {
537 int status;
538
539 *decoded_len = s->stream.avail_in << 2;
540 *decoded = emalloc(*decoded_len);
541
542 /* inflate remaining input */
543 s->stream.next_in = (Bytef *) PHPSTR_VAL(s->stream.opaque);
544 s->stream.avail_in = PHPSTR_LEN(s->stream.opaque);
545
546 s->stream.avail_out = *decoded_len;
547 s->stream.next_out = (Bytef *) *decoded;
548
549 if (Z_STREAM_END == (status = inflate(&s->stream, Z_FINISH))) {
550 /* cut processed input off */
551 phpstr_cut(PHPSTR(s->stream.opaque), 0, PHPSTR_LEN(s->stream.opaque) - s->stream.avail_in);
552
553 /* size down */
554 *decoded_len -= s->stream.avail_out;
555 *decoded = erealloc(*decoded, *decoded_len + 1);
556 (*decoded)[*decoded_len] = '\0';
557 return SUCCESS;
558 }
559
560 STR_SET(*decoded, NULL);
561 *decoded_len = 0;
562 http_error_ex(HE_WARNING, HTTP_E_ENCODING, "Failed to finish inflate stream: %s", zError(status));
563 return FAILURE;
564 }
565
566 PHP_HTTP_API void _http_encoding_deflate_stream_dtor(http_encoding_stream *s TSRMLS_DC)
567 {
568 if (s) {
569 if (s->stream.opaque) {
570 phpstr_free((phpstr **) &s->stream.opaque);
571 }
572 deflateEnd(&s->stream);
573 }
574 }
575
576 PHP_HTTP_API void _http_encoding_inflate_stream_dtor(http_encoding_stream *s TSRMLS_DC)
577 {
578 if (s) {
579 if (s->stream.opaque) {
580 phpstr_free((phpstr **) &s->stream.opaque);
581 }
582 inflateEnd(&s->stream);
583 }
584 }
585
586 PHP_HTTP_API void _http_encoding_deflate_stream_free(http_encoding_stream **s TSRMLS_DC)
587 {
588 if (s) {
589 http_encoding_deflate_stream_dtor(*s);
590 if (*s) {
591 pefree(*s, (*s)->flags & HTTP_ENCODING_STREAM_PERSISTENT);
592 }
593 *s = NULL;
594 }
595 }
596
597 PHP_HTTP_API void _http_encoding_inflate_stream_free(http_encoding_stream **s TSRMLS_DC)
598 {
599 if (s) {
600 http_encoding_inflate_stream_dtor(*s);
601 if (*s) {
602 pefree(*s, (*s)->flags & HTTP_ENCODING_STREAM_PERSISTENT);
603 }
604 *s = NULL;
605 }
606 }
607
608 static const char http_encoding_gzip_header[] = {
609 (const char) 0x1f, // fixed value
610 (const char) 0x8b, // fixed value
611 (const char) Z_DEFLATED, // compression algorithm
612 (const char) 0, // none of the possible flags defined by the GZIP "RFC"
613 (const char) 0, // MTIME
614 (const char) 0, // =*=
615 (const char) 0, // =*=
616 (const char) 0, // =*=
617 (const char) 0, // two possible flag values for 9 compression levels? o_O
618 #ifdef PHP_WIN32
619 (const char) 0x0b // OS_CODE
620 #else
621 (const char) 0x03 // OS_CODE
622 #endif
623 };
624
625 PHP_HTTP_API STATUS _http_encoding_gzencode_verify(const char *data, size_t data_len, const char **encoded, size_t *encoded_len, int error_level TSRMLS_DC)
626 {
627 size_t offset = sizeof(http_encoding_gzip_header);
628
629 if (data_len < offset) {
630 goto really_bad_gzip_header;
631 }
632
633 if (data[0] != (const char) 0x1F || data[1] != (const char) 0x8B) {
634 http_error_ex(error_level TSRMLS_CC, HTTP_E_ENCODING, "Unrecognized GZIP header start: 0x%02X 0x%02X", (int) data[0], (int) (data[1] & 0xFF));
635 return FAILURE;
636 }
637
638 if (data[2] != (const char) Z_DEFLATED) {
639 http_error_ex(error_level TSRMLS_CC, HTTP_E_ENCODING, "Unrecognized compression format (%d)", (int) (data[2] & 0xFF));
640 /* still try to decode */
641 }
642 if ((data[3] & 0x4) == 0x4) {
643 if (data_len < offset + 2) {
644 goto really_bad_gzip_header;
645 }
646 /* there are extra fields, the length follows the common header as 2 bytes LSB */
647 offset += (unsigned) ((data[offset] & 0xFF));
648 offset += 1;
649 offset += (unsigned) ((data[offset] & 0xFF) << 8);
650 offset += 1;
651 }
652 if ((data[3] & 0x8) == 0x8) {
653 if (data_len <= offset) {
654 goto really_bad_gzip_header;
655 }
656 /* there's a file name */
657 offset += strlen(&data[offset]) + 1 /*NUL*/;
658 }
659 if ((data[3] & 0x10) == 0x10) {
660 if (data_len <= offset) {
661 goto really_bad_gzip_header;
662 }
663 /* there's a comment */
664 offset += strlen(&data[offset]) + 1 /* NUL */;
665 }
666 if ((data[3] & 0x2) == 0x2) {
667 /* there's a CRC16 of the header */
668 offset += 2;
669 if (data_len <= offset) {
670 goto really_bad_gzip_header;
671 } else {
672 ulong crc, cmp;
673
674 cmp = (unsigned) ((data[offset-2] & 0xFF));
675 cmp += (unsigned) ((data[offset-1] & 0xFF) << 8);
676
677 crc = crc32(0L, Z_NULL, 0);
678 crc = crc32(crc, (const Bytef *) data, sizeof(http_encoding_gzip_header));
679
680 if (cmp != (crc & 0xFFFF)) {
681 http_error_ex(error_level TSRMLS_CC, HTTP_E_ENCODING, "GZIP headers CRC checksums so not match (%lu, %lu)", cmp, crc & 0xFFFF);
682 return FAILURE;
683 }
684 }
685 }
686
687 if (data_len < offset + 8) {
688 http_error(error_level TSRMLS_CC, HTTP_E_ENCODING, "Missing or truncated GZIP footer");
689 return FAILURE;
690 }
691
692 if (encoded) {
693 *encoded = data + offset;
694 }
695 if (encoded_len) {
696 *encoded_len = data_len - offset - 8 /* size of the assumed GZIP footer */;
697 }
698
699 return SUCCESS;
700
701 really_bad_gzip_header:
702 http_error(error_level TSRMLS_CC, HTTP_E_ENCODING, "Missing or truncated GZIP header");
703 return FAILURE;
704 }
705
706 PHP_HTTP_API STATUS _http_encoding_gzdecode_verify(const char *data, size_t data_len, const char *decoded, size_t decoded_len, int error_level TSRMLS_DC)
707 {
708 STATUS status = SUCCESS;
709 ulong len, cmp, crc;
710
711 crc = crc32(0L, Z_NULL, 0);
712 crc = crc32(crc, (const Bytef *) decoded, decoded_len);
713
714 cmp = (unsigned) ((data[data_len-8] & 0xFF));
715 cmp += (unsigned) ((data[data_len-7] & 0xFF) << 8);
716 cmp += (unsigned) ((data[data_len-6] & 0xFF) << 16);
717 cmp += (unsigned) ((data[data_len-5] & 0xFF) << 24);
718 len = (unsigned) ((data[data_len-4] & 0xFF));
719 len += (unsigned) ((data[data_len-3] & 0xFF) << 8);
720 len += (unsigned) ((data[data_len-2] & 0xFF) << 16);
721 len += (unsigned) ((data[data_len-1] & 0xFF) << 24);
722
723 if (cmp != crc) {
724 http_error_ex(error_level TSRMLS_CC, HTTP_E_ENCODING, "Could not verify data integrity: CRC checksums do not match (%lu, %lu)", cmp, crc);
725 status = FAILURE;
726 }
727 if (len != decoded_len) {
728 http_error_ex(error_level TSRMLS_CC, HTTP_E_ENCODING, "Could not verify data integrity: data sizes do not match (%lu, %lu)", len, decoded_len);
729 status = FAILURE;
730 }
731 return status;
732 }
733
734 #endif /* HTTP_HAVE_ZLIB */
735
736 PHP_HTTP_API zend_bool _http_encoding_response_start(size_t content_length TSRMLS_DC)
737 {
738 if ( php_ob_handler_used("ob_gzhandler" TSRMLS_CC) ||
739 php_ob_handler_used("zlib output compression" TSRMLS_CC)) {
740 HTTP_G(send).gzip_encoding = 0;
741 } else {
742 if (!HTTP_G(send).gzip_encoding) {
743 /* emit a content-length header */
744 if (content_length) {
745 char cl_header_str[128];
746 size_t cl_header_len;
747 cl_header_len = snprintf(cl_header_str, lenof(cl_header_str), "Content-Length: %zu", content_length);
748 http_send_header_string_ex(cl_header_str, cl_header_len, 1);
749 }
750 } else {
751 #ifndef HTTP_HAVE_ZLIB
752 HTTP_G(send).gzip_encoding = 0;
753 php_start_ob_buffer_named("ob_gzhandler", 0, 0 TSRMLS_CC);
754 #else
755 HashTable *selected;
756 zval zsupported;
757
758 INIT_PZVAL(&zsupported);
759 array_init(&zsupported);
760 add_next_index_stringl(&zsupported, "gzip", lenof("gzip"), 1);
761 add_next_index_stringl(&zsupported, "x-gzip", lenof("x-gzip"), 1);
762 add_next_index_stringl(&zsupported, "deflate", lenof("deflate"), 1);
763
764 HTTP_G(send).gzip_encoding = 0;
765
766 if ((selected = http_negotiate_encoding(&zsupported))) {
767 STATUS hs = FAILURE;
768 char *encoding = NULL;
769 ulong idx;
770
771 if (HASH_KEY_IS_STRING == zend_hash_get_current_key(selected, &encoding, &idx, 0) && encoding) {
772 if (!strcmp(encoding, "gzip") || !strcmp(encoding, "x-gzip")) {
773 if (SUCCESS == (hs = http_send_header_string("Content-Encoding: gzip"))) {
774 HTTP_G(send).gzip_encoding = HTTP_ENCODING_GZIP;
775 }
776 } else if (!strcmp(encoding, "deflate")) {
777 if (SUCCESS == (hs = http_send_header_string("Content-Encoding: deflate"))) {
778 HTTP_G(send).gzip_encoding = HTTP_ENCODING_DEFLATE;
779 }
780 }
781 if (SUCCESS == hs) {
782 http_send_header_string("Vary: Accept-Encoding");
783 }
784 }
785
786 zend_hash_destroy(selected);
787 FREE_HASHTABLE(selected);
788 }
789
790 zval_dtor(&zsupported);
791 return HTTP_G(send).gzip_encoding;
792 #endif
793 }
794 }
795 return 0;
796 }
797
798 /*
799 * Local variables:
800 * tab-width: 4
801 * c-basic-offset: 4
802 * End:
803 * vim600: noet sw=4 ts=4 fdm=marker
804 * vim<600: noet sw=4 ts=4
805 */
806