2 +--------------------------------------------------------------------+
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-2010, Michael Wallner <mike@php.net> |
10 +--------------------------------------------------------------------+
15 #define HTTP_WANT_ZLIB
18 #include "php_http_api.h"
19 #include "php_http_encoding_api.h"
20 #include "php_http_send_api.h"
21 #include "php_http_headers_api.h"
25 PHP_MINIT_FUNCTION(http_encoding
)
27 HTTP_LONG_CONSTANT("HTTP_DEFLATE_LEVEL_DEF", HTTP_DEFLATE_LEVEL_DEF
);
28 HTTP_LONG_CONSTANT("HTTP_DEFLATE_LEVEL_MIN", HTTP_DEFLATE_LEVEL_MIN
);
29 HTTP_LONG_CONSTANT("HTTP_DEFLATE_LEVEL_MAX", HTTP_DEFLATE_LEVEL_MAX
);
30 HTTP_LONG_CONSTANT("HTTP_DEFLATE_TYPE_ZLIB", HTTP_DEFLATE_TYPE_ZLIB
);
31 HTTP_LONG_CONSTANT("HTTP_DEFLATE_TYPE_GZIP", HTTP_DEFLATE_TYPE_GZIP
);
32 HTTP_LONG_CONSTANT("HTTP_DEFLATE_TYPE_RAW", HTTP_DEFLATE_TYPE_RAW
);
33 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_DEF", HTTP_DEFLATE_STRATEGY_DEF
);
34 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_FILT", HTTP_DEFLATE_STRATEGY_FILT
);
35 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_HUFF", HTTP_DEFLATE_STRATEGY_HUFF
);
36 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_RLE", HTTP_DEFLATE_STRATEGY_RLE
);
37 HTTP_LONG_CONSTANT("HTTP_DEFLATE_STRATEGY_FIXED", HTTP_DEFLATE_STRATEGY_FIXED
);
39 HTTP_LONG_CONSTANT("HTTP_ENCODING_STREAM_FLUSH_NONE", HTTP_ENCODING_STREAM_FLUSH_NONE
);
40 HTTP_LONG_CONSTANT("HTTP_ENCODING_STREAM_FLUSH_SYNC", HTTP_ENCODING_STREAM_FLUSH_SYNC
);
41 HTTP_LONG_CONSTANT("HTTP_ENCODING_STREAM_FLUSH_FULL", HTTP_ENCODING_STREAM_FLUSH_FULL
);
46 PHP_RINIT_FUNCTION(http_encoding
)
48 if (HTTP_G
->send
.inflate
.start_auto
) {
49 php_ob_set_internal_handler(_http_ob_inflatehandler
, HTTP_INFLATE_BUFFER_SIZE
, "http inflate", 0 TSRMLS_CC
);
51 if (HTTP_G
->send
.deflate
.start_auto
) {
52 php_ob_set_internal_handler(_http_ob_deflatehandler
, HTTP_DEFLATE_BUFFER_SIZE
, "http deflate", 0 TSRMLS_CC
);
57 PHP_RSHUTDOWN_FUNCTION(http_encoding
)
59 if (HTTP_G
->send
.deflate
.stream
) {
60 http_encoding_deflate_stream_free((http_encoding_stream
**) &HTTP_G
->send
.deflate
.stream
);
62 if (HTTP_G
->send
.inflate
.stream
) {
63 http_encoding_inflate_stream_free((http_encoding_stream
**) &HTTP_G
->send
.inflate
.stream
);
70 /* {{{ eol_match(char **, int *) */
71 static inline int eol_match(char **line
, int *eol_len
)
75 while (' ' == *ptr
) ++ptr
;
77 if (ptr
== http_locate_eol(*line
, eol_len
)) {
86 /* {{{ char *http_encoding_dechunk(char *, size_t, char **, size_t *) */
87 PHP_HTTP_API
const char *_http_encoding_dechunk(const char *encoded
, size_t encoded_len
, char **decoded
, size_t *decoded_len TSRMLS_DC
)
91 const char *e_ptr
= encoded
;
94 *decoded
= ecalloc(1, encoded_len
);
96 while ((encoded
+ encoded_len
- e_ptr
) > 0) {
97 ulong chunk_len
= 0, rest
;
99 chunk_len
= strtoul(e_ptr
, &n_ptr
, 16);
101 /* we could not read in chunk size */
102 if (n_ptr
== e_ptr
) {
104 * if this is the first turn and there doesn't seem to be a chunk
105 * size at the begining of the body, do not fail on apparently
106 * not encoded data and return a copy
108 if (e_ptr
== encoded
) {
109 http_error(HE_NOTICE
, HTTP_E_ENCODING
, "Data does not seem to be chunked encoded");
110 memcpy(*decoded
, encoded
, encoded_len
);
111 *decoded_len
= encoded_len
;
112 return encoded
+ encoded_len
;
115 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Expected chunk size at pos %tu of %zu but got trash", n_ptr
- encoded
, encoded_len
);
120 /* reached the end */
122 /* move over '0' chunked encoding terminator */
123 while (*e_ptr
== '0') ++e_ptr
;
127 /* there should be CRLF after the chunk size, but we'll ignore SP+ too */
128 if (*n_ptr
&& !eol_match(&n_ptr
, &eol_len
)) {
130 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));
132 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
);
137 /* chunk size pretends more data than we actually got, so it's probably a truncated message */
138 if (chunk_len
> (rest
= encoded
+ encoded_len
- n_ptr
)) {
139 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
);
144 memcpy(*decoded
+ *decoded_len
, n_ptr
, chunk_len
);
145 *decoded_len
+= chunk_len
;
147 if (chunk_len
== rest
) {
148 e_ptr
= n_ptr
+ chunk_len
;
151 /* advance to next chunk */
152 e_ptr
= n_ptr
+ chunk_len
+ eol_len
;
160 /* {{{ int http_encoding_response_start(size_t) */
161 PHP_HTTP_API
int _http_encoding_response_start(size_t content_length
, zend_bool ignore_http_ohandler TSRMLS_DC
)
163 int response
= HTTP_G
->send
.deflate
.response
;
164 int ohandler
= php_ob_handler_used("ob_gzhandler" TSRMLS_CC
) || php_ob_handler_used("zlib output compression" TSRMLS_CC
);
166 if (!ohandler
&& !ignore_http_ohandler
) {
167 ohandler
= php_ob_handler_used("ob_deflatehandler" TSRMLS_CC
) || php_ob_handler_used("http deflate" TSRMLS_CC
);
170 if (response
&& !ohandler
) {
171 #ifdef HTTP_HAVE_ZLIB
175 HTTP_G
->send
.deflate
.encoding
= 0;
177 INIT_PZVAL(&zsupported
);
178 array_init(&zsupported
);
179 add_next_index_stringl(&zsupported
, "gzip", lenof("gzip"), 1);
180 add_next_index_stringl(&zsupported
, "x-gzip", lenof("x-gzip"), 1);
181 add_next_index_stringl(&zsupported
, "deflate", lenof("deflate"), 1);
183 if ((selected
= http_negotiate_encoding(&zsupported
))) {
185 char *encoding
= NULL
;
188 if (HASH_KEY_IS_STRING
== zend_hash_get_current_key(selected
, &encoding
, &idx
, 0) && encoding
) {
189 if (!strcmp(encoding
, "gzip") || !strcmp(encoding
, "x-gzip")) {
190 if (SUCCESS
== (hs
= http_send_header_string("Content-Encoding: gzip"))) {
191 HTTP_G
->send
.deflate
.encoding
= HTTP_ENCODING_GZIP
;
193 } else if (!strcmp(encoding
, "deflate")) {
194 if (SUCCESS
== (hs
= http_send_header_string("Content-Encoding: deflate"))) {
195 HTTP_G
->send
.deflate
.encoding
= HTTP_ENCODING_DEFLATE
;
199 http_send_header_string("Vary: Accept-Encoding");
203 zend_hash_destroy(selected
);
204 FREE_HASHTABLE(selected
);
207 zval_dtor(&zsupported
);
209 HTTP_G
->send
.deflate
.encoding
= 0;
210 php_start_ob_buffer_named("ob_gzhandler", 0, 0 TSRMLS_CC
);
211 #endif /* HTTP_HAVE_ZLIB */
212 } else if (content_length
&& !ohandler
) {
213 /* emit a content-length header */
216 phpstr_init(&header
);
217 phpstr_appendf(&header
, "Content-Length: %zu", content_length
);
219 http_send_header_string_ex(PHPSTR_VAL(&header
), PHPSTR_LEN(&header
), 1);
220 phpstr_dtor(&header
);
222 HTTP_G
->send
.deflate
.encoding
= 0;
225 return HTTP_G
->send
.deflate
.encoding
;
229 #ifdef HTTP_HAVE_ZLIB
231 /* {{{ inline int http_inflate_rounds */
232 static inline int http_inflate_rounds(z_stream
*Z
, int flush
, char **buf
, size_t *len
)
234 int status
= 0, round
= 0;
240 phpstr_init_ex(&buffer
, Z
->avail_in
, PHPSTR_INIT_PREALLOC
);
243 if (PHPSTR_NOMEM
== phpstr_resize_ex(&buffer
, buffer
.size
, 0, 1)) {
244 status
= Z_MEM_ERROR
;
246 Z
->avail_out
= buffer
.free
;
247 Z
->next_out
= (Bytef
*) buffer
.data
+ buffer
.used
;
249 fprintf(stderr
, "\n%3d: %3d PRIOR: size=%7lu,\tfree=%7lu,\tused=%7lu,\tavail_in=%7lu,\tavail_out=%7lu\n", round
, status
, buffer
.size
, buffer
.free
, buffer
.used
, Z
->avail_in
, Z
->avail_out
);
251 status
= inflate(Z
, flush
);
253 buffer
.used
+= buffer
.free
- Z
->avail_out
;
254 buffer
.free
= Z
->avail_out
;
256 fprintf(stderr
, "%3d: %3d AFTER: size=%7lu,\tfree=%7lu,\tused=%7lu,\tavail_in=%7lu,\tavail_out=%7lu\n", round
, status
, buffer
.size
, buffer
.free
, buffer
.used
, Z
->avail_in
, Z
->avail_out
);
258 HTTP_INFLATE_BUFFER_SIZE_ALIGN(buffer
.size
);
260 } while ((Z_BUF_ERROR
== status
|| (Z_OK
== status
&& Z
->avail_in
)) && ++round
< HTTP_INFLATE_ROUNDS
);
262 if (status
== Z_OK
|| status
== Z_STREAM_END
) {
263 phpstr_shrink(&buffer
);
268 phpstr_dtor(&buffer
);
275 /* {{{ STATUS http_encoding_deflate(int, char *, size_t, char **, size_t *) */
276 PHP_HTTP_API STATUS
_http_encoding_deflate(int flags
, const char *data
, size_t data_len
, char **encoded
, size_t *encoded_len ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
278 int status
, level
, wbits
, strategy
;
281 HTTP_DEFLATE_LEVEL_SET(flags
, level
);
282 HTTP_DEFLATE_WBITS_SET(flags
, wbits
);
283 HTTP_DEFLATE_STRATEGY_SET(flags
, strategy
);
285 memset(&Z
, 0, sizeof(z_stream
));
289 status
= deflateInit2(&Z
, level
, Z_DEFLATED
, wbits
, MAX_MEM_LEVEL
, strategy
);
290 if (Z_OK
== status
) {
291 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE_GUESS(data_len
);
292 *encoded
= emalloc_rel(*encoded_len
);
294 Z
.next_in
= (Bytef
*) data
;
295 Z
.next_out
= (Bytef
*) *encoded
;
296 Z
.avail_in
= data_len
;
297 Z
.avail_out
= *encoded_len
;
299 status
= deflate(&Z
, Z_FINISH
);
302 if (Z_STREAM_END
== status
) {
303 /* size buffer down to actual length */
304 *encoded
= erealloc_rel(*encoded
, Z
.total_out
+ 1);
305 (*encoded
)[*encoded_len
= Z
.total_out
] = '\0';
308 STR_SET(*encoded
, NULL
);
313 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Could not deflate data: %s", zError(status
));
318 /* {{{ STATUS http_encoding_inflate(char *, size_t, char **, size_t) */
319 PHP_HTTP_API STATUS
_http_encoding_inflate(const char *data
, size_t data_len
, char **decoded
, size_t *decoded_len ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
322 int status
, wbits
= HTTP_WINDOW_BITS_ANY
;
324 memset(&Z
, 0, sizeof(z_stream
));
327 status
= inflateInit2(&Z
, wbits
);
328 if (Z_OK
== status
) {
329 Z
.next_in
= (Bytef
*) data
;
330 Z
.avail_in
= data_len
;
332 switch (status
= http_inflate_rounds(&Z
, Z_NO_FLUSH
, decoded
, decoded_len
)) {
338 status
= Z_DATA_ERROR
;
342 /* raw deflated data? */
343 if (HTTP_WINDOW_BITS_ANY
== wbits
) {
345 wbits
= HTTP_WINDOW_BITS_RAW
;
346 goto retry_raw_inflate
;
351 if (decoded_len
&& *decoded
) {
356 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Could not inflate data: %s", zError(status
));
361 /* {{{ http_encoding_stream *_http_encoding_deflate_stream_init(http_encoding_stream *, int) */
362 PHP_HTTP_API http_encoding_stream
*_http_encoding_deflate_stream_init(http_encoding_stream
*s
, int flags ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
364 int status
, level
, wbits
, strategy
, free_stream
;
366 if ((free_stream
= !s
)) {
367 s
= pemalloc_rel(sizeof(http_encoding_stream
), (flags
& HTTP_ENCODING_STREAM_PERSISTENT
));
369 memset(s
, 0, sizeof(http_encoding_stream
));
372 HTTP_DEFLATE_LEVEL_SET(flags
, level
);
373 HTTP_DEFLATE_WBITS_SET(flags
, wbits
);
374 HTTP_DEFLATE_STRATEGY_SET(flags
, strategy
);
376 if (Z_OK
== (status
= deflateInit2(&s
->stream
, level
, Z_DEFLATED
, wbits
, MAX_MEM_LEVEL
, strategy
))) {
377 int p
= (flags
& HTTP_ENCODING_STREAM_PERSISTENT
) ? PHPSTR_INIT_PERSISTENT
:0;
379 if ((s
->stream
.opaque
= phpstr_init_ex(NULL
, HTTP_DEFLATE_BUFFER_SIZE
, p
))) {
382 deflateEnd(&s
->stream
);
383 status
= Z_MEM_ERROR
;
386 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to initialize deflate encoding stream: %s", zError(status
));
394 /* {{{ http_encoding_stream *http_encoding_inflate_stream_init(http_encoding_stream *, int) */
395 PHP_HTTP_API http_encoding_stream
*_http_encoding_inflate_stream_init(http_encoding_stream
*s
, int flags ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
397 int status
, wbits
, free_stream
;
399 if ((free_stream
= !s
)) {
400 s
= pemalloc_rel(sizeof(http_encoding_stream
), (flags
& HTTP_ENCODING_STREAM_PERSISTENT
));
402 memset(s
, 0, sizeof(http_encoding_stream
));
405 HTTP_INFLATE_WBITS_SET(flags
, wbits
);
407 if (Z_OK
== (status
= inflateInit2(&s
->stream
, wbits
))) {
408 int p
= (flags
& HTTP_ENCODING_STREAM_PERSISTENT
) ? PHPSTR_INIT_PERSISTENT
:0;
410 if ((s
->stream
.opaque
= phpstr_init_ex(NULL
, HTTP_DEFLATE_BUFFER_SIZE
, p
))) {
413 inflateEnd(&s
->stream
);
414 status
= Z_MEM_ERROR
;
417 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to initialize inflate stream: %s", zError(status
));
425 /* {{{ STATUS http_encoding_deflate_stream_update(http_encoding_stream *, char *, size_t, char **, size_t *) */
426 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 ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
430 /* append input to our buffer */
431 phpstr_append(PHPSTR(s
->stream
.opaque
), data
, data_len
);
433 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
434 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
437 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE_GUESS(data_len
);
438 *encoded
= emalloc_rel(*encoded_len
);
439 s
->stream
.avail_out
= *encoded_len
;
440 s
->stream
.next_out
= (Bytef
*) *encoded
;
442 switch (status
= deflate(&s
->stream
, HTTP_ENCODING_STREAM_FLUSH_FLAG(s
->flags
))) {
445 /* cut processed chunk off the buffer */
446 if (s
->stream
.avail_in
) {
447 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
449 phpstr_reset(PHPSTR(s
->stream
.opaque
));
452 /* size buffer down to actual size */
453 *encoded_len
-= s
->stream
.avail_out
;
454 *encoded
= erealloc_rel(*encoded
, *encoded_len
+ 1);
455 (*encoded
)[*encoded_len
] = '\0';
459 STR_SET(*encoded
, NULL
);
461 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to update deflate stream: %s", zError(status
));
466 /* {{{ STATUS http_encoding_inflate_stream_update(http_encoding_stream *, char *, size_t, char **, size_t *) */
467 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 ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
471 /* append input to buffer */
472 phpstr_append(PHPSTR(s
->stream
.opaque
), data
, data_len
);
475 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
476 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
478 switch (status
= http_inflate_rounds(&s
->stream
, HTTP_ENCODING_STREAM_FLUSH_FLAG(s
->flags
), decoded
, decoded_len
)) {
482 if (s
->stream
.avail_in
) {
483 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
485 phpstr_reset(PHPSTR(s
->stream
.opaque
));
490 /* raw deflated data ? */
491 if (!(s
->flags
& HTTP_INFLATE_TYPE_RAW
) && !s
->stream
.total_out
) {
492 inflateEnd(&s
->stream
);
493 s
->flags
|= HTTP_INFLATE_TYPE_RAW
;
494 inflateInit2(&s
->stream
, HTTP_WINDOW_BITS_RAW
);
495 goto retry_raw_inflate
;
499 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to update inflate stream: %s", zError(status
));
504 /* {{{ STATUS http_encoding_deflate_stream_flush(http_encoding_stream *, char **, size_t *) */
505 PHP_HTTP_API STATUS
_http_encoding_deflate_stream_flush(http_encoding_stream
*s
, char **encoded
, size_t *encoded_len ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
509 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE
;
510 *encoded
= emalloc_rel(*encoded_len
);
512 s
->stream
.avail_in
= 0;
513 s
->stream
.next_in
= NULL
;
514 s
->stream
.avail_out
= *encoded_len
;
515 s
->stream
.next_out
= (Bytef
*) *encoded
;
517 switch (status
= deflate(&s
->stream
, Z_FULL_FLUSH
)) {
520 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE
- s
->stream
.avail_out
;
521 *encoded
= erealloc_rel(*encoded
, *encoded_len
+ 1);
522 (*encoded
)[*encoded_len
] = '\0';
526 STR_SET(*encoded
, NULL
);
528 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to flush deflate stream: %s", zError(status
));
533 /* {{{ STATUS http_encoding_inflate_straem_flush(http_encoding_stream *, char **, size_t *) */
534 PHP_HTTP_API STATUS
_http_encoding_inflate_stream_flush(http_encoding_stream
*s
, char **decoded
, size_t *decoded_len ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
537 *decoded
= estrndup("", *decoded_len
= 0);
542 /* {{{ STATUS http_encoding_deflate_stream_finish(http_encoding_stream *, char **, size_t *) */
543 PHP_HTTP_API STATUS
_http_encoding_deflate_stream_finish(http_encoding_stream
*s
, char **encoded
, size_t *encoded_len ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
547 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE
;
548 *encoded
= emalloc_rel(*encoded_len
);
550 /* deflate remaining input */
551 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
552 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
554 s
->stream
.avail_out
= *encoded_len
;
555 s
->stream
.next_out
= (Bytef
*) *encoded
;
558 status
= deflate(&s
->stream
, Z_FINISH
);
559 } while (Z_OK
== status
);
561 if (Z_STREAM_END
== status
) {
562 /* cut processed intp off */
563 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
566 *encoded_len
-= s
->stream
.avail_out
;
567 *encoded
= erealloc_rel(*encoded
, *encoded_len
+ 1);
568 (*encoded
)[*encoded_len
] = '\0';
572 STR_SET(*encoded
, NULL
);
574 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to finish deflate stream: %s", zError(status
));
579 /* {{{ STATUS http_encoding_inflate_stream_finish(http_encoding_stream *, char **, size_t *) */
580 PHP_HTTP_API STATUS
_http_encoding_inflate_stream_finish(http_encoding_stream
*s
, char **decoded
, size_t *decoded_len ZEND_FILE_LINE_DC ZEND_FILE_LINE_ORIG_DC TSRMLS_DC
)
584 if (!PHPSTR_LEN(s
->stream
.opaque
)) {
590 *decoded_len
= (PHPSTR_LEN(s
->stream
.opaque
) + 1) * HTTP_INFLATE_ROUNDS
;
591 *decoded
= emalloc_rel(*decoded_len
);
593 /* inflate remaining input */
594 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
595 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
597 s
->stream
.avail_out
= *decoded_len
;
598 s
->stream
.next_out
= (Bytef
*) *decoded
;
600 if (Z_STREAM_END
== (status
= inflate(&s
->stream
, Z_FINISH
))) {
601 /* cut processed input off */
602 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
605 *decoded_len
-= s
->stream
.avail_out
;
606 *decoded
= erealloc_rel(*decoded
, *decoded_len
+ 1);
607 (*decoded
)[*decoded_len
] = '\0';
611 STR_SET(*decoded
, NULL
);
613 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to finish inflate stream: %s", zError(status
));
618 /* {{{ void http_encoding_deflate_stream_dtor(http_encoding_stream *) */
619 PHP_HTTP_API
void _http_encoding_deflate_stream_dtor(http_encoding_stream
*s TSRMLS_DC
)
622 if (s
->stream
.opaque
) {
623 phpstr_free((phpstr
**) &s
->stream
.opaque
);
625 deflateEnd(&s
->stream
);
630 /* {{{ void http_encoding_inflate_stream_dtor(http_encoding_stream *) */
631 PHP_HTTP_API
void _http_encoding_inflate_stream_dtor(http_encoding_stream
*s TSRMLS_DC
)
634 if (s
->stream
.opaque
) {
635 phpstr_free((phpstr
**) &s
->stream
.opaque
);
637 inflateEnd(&s
->stream
);
642 /* {{{ void http_encoding_deflate_stream_free(http_encoding_stream **) */
643 PHP_HTTP_API
void _http_encoding_deflate_stream_free(http_encoding_stream
**s TSRMLS_DC
)
646 http_encoding_deflate_stream_dtor(*s
);
648 pefree(*s
, (*s
)->flags
& HTTP_ENCODING_STREAM_PERSISTENT
);
655 /* {{{ void http_encoding_inflate_stream_free(http_encoding_stream **) */
656 PHP_HTTP_API
void _http_encoding_inflate_stream_free(http_encoding_stream
**s TSRMLS_DC
)
659 http_encoding_inflate_stream_dtor(*s
);
661 pefree(*s
, (*s
)->flags
& HTTP_ENCODING_STREAM_PERSISTENT
);
668 /* {{{ void http_ob_deflatehandler(char *, uint, char **, uint *, int) */
669 void _http_ob_deflatehandler(char *output
, uint output_len
, char **handled_output
, uint
*handled_output_len
, int mode TSRMLS_DC
)
673 *handled_output
= NULL
;
674 *handled_output_len
= 0;
676 if (mode
& PHP_OUTPUT_HANDLER_START
) {
679 if (HTTP_G
->send
.deflate
.stream
) {
680 zend_error(E_ERROR
, "ob_deflatehandler() can only be used once");
684 HTTP_G
->send
.deflate
.response
= 1;
685 encoding
= http_encoding_response_start(0, 1);
686 HTTP_G
->send
.deflate
.response
= 0;
689 case HTTP_ENCODING_GZIP
:
690 flags
= HTTP_DEFLATE_TYPE_GZIP
;
693 case HTTP_ENCODING_DEFLATE
:
694 flags
= HTTP_DEFLATE_TYPE_ZLIB
;
698 goto deflate_passthru_plain
;
701 flags
|= (HTTP_G
->send
.deflate
.start_flags
&~ 0xf0);
702 HTTP_G
->send
.deflate
.stream
= http_encoding_deflate_stream_init(NULL
, flags
);
705 if (HTTP_G
->send
.deflate
.stream
) {
709 http_encoding_deflate_stream_update((http_encoding_stream
*) HTTP_G
->send
.deflate
.stream
, output
, output_len
, handled_output
, &tmp_len
);
710 *handled_output_len
= tmp_len
;
713 if (mode
& PHP_OUTPUT_HANDLER_END
) {
714 char *remaining
= NULL
;
715 size_t remaining_len
= 0;
717 http_encoding_deflate_stream_finish((http_encoding_stream
*) HTTP_G
->send
.deflate
.stream
, &remaining
, &remaining_len
);
718 http_encoding_deflate_stream_free((http_encoding_stream
**) &HTTP_G
->send
.deflate
.stream
);
720 *handled_output
= erealloc(*handled_output
, *handled_output_len
+ remaining_len
+ 1);
721 memcpy(*handled_output
+ *handled_output_len
, remaining
, remaining_len
);
722 (*handled_output
)[*handled_output_len
+= remaining_len
] = '\0';
727 deflate_passthru_plain
:
728 *handled_output
= estrndup(output
, *handled_output_len
= output_len
);
733 /* {{{ void http_ob_inflatehandler(char *, uint, char **, uint *, int) */
734 void _http_ob_inflatehandler(char *output
, uint output_len
, char **handled_output
, uint
*handled_output_len
, int mode TSRMLS_DC
)
736 *handled_output
= NULL
;
737 *handled_output_len
= 0;
739 if (mode
& PHP_OUTPUT_HANDLER_START
) {
740 if (HTTP_G
->send
.inflate
.stream
) {
741 zend_error(E_ERROR
, "ob_inflatehandler() can only be used once");
744 HTTP_G
->send
.inflate
.stream
= http_encoding_inflate_stream_init(NULL
, (HTTP_G
->send
.inflate
.start_flags
&~ 0xf0));
747 if (HTTP_G
->send
.inflate
.stream
) {
751 http_encoding_inflate_stream_update((http_encoding_stream
*) HTTP_G
->send
.inflate
.stream
, output
, output_len
, handled_output
, &tmp_len
);
752 *handled_output_len
= tmp_len
;
755 if (mode
& PHP_OUTPUT_HANDLER_END
) {
756 char *remaining
= NULL
;
757 size_t remaining_len
= 0;
759 http_encoding_inflate_stream_finish((http_encoding_stream
*) HTTP_G
->send
.inflate
.stream
, &remaining
, &remaining_len
);
760 http_encoding_inflate_stream_free((http_encoding_stream
**) &HTTP_G
->send
.inflate
.stream
);
762 *handled_output
= erealloc(*handled_output
, *handled_output_len
+ remaining_len
+ 1);
763 memcpy(*handled_output
+ *handled_output_len
, remaining
, remaining_len
);
764 (*handled_output
)[*handled_output_len
+= remaining_len
] = '\0';
769 *handled_output
= estrndup(output
, *handled_output_len
= output_len
);
774 #endif /* HTTP_HAVE_ZLIB */
781 * vim600: noet sw=4 ts=4 fdm=marker
782 * vim<600: noet sw=4 ts=4