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-2007, 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 */
214 char cl_header_str
[128];
215 size_t cl_header_len
;
216 cl_header_len
= snprintf(cl_header_str
, sizeof(cl_header_str
), "Content-Length: %zu", content_length
);
217 http_send_header_string_ex(cl_header_str
, cl_header_len
, 1);
219 HTTP_G
->send
.deflate
.encoding
= 0;
222 return HTTP_G
->send
.deflate
.encoding
;
226 #ifdef HTTP_HAVE_ZLIB
228 /* {{{ inline int http_inflate_rounds */
229 static inline int http_inflate_rounds(z_stream
*Z
, int flush
, char **buf
, size_t *len
)
231 int status
= 0, round
= 0;
237 phpstr_init_ex(&buffer
, Z
->avail_in
, PHPSTR_INIT_PREALLOC
);
240 if (PHPSTR_NOMEM
== phpstr_resize_ex(&buffer
, buffer
.size
, 0, 1)) {
241 status
= Z_MEM_ERROR
;
243 Z
->avail_out
= buffer
.free
;
244 Z
->next_out
= (Bytef
*) buffer
.data
+ buffer
.used
;
246 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
);
248 status
= inflate(Z
, flush
);
250 buffer
.used
+= buffer
.free
- Z
->avail_out
;
251 buffer
.free
= Z
->avail_out
;
253 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
);
255 HTTP_INFLATE_BUFFER_SIZE_ALIGN(buffer
.size
);
257 } while ((Z_BUF_ERROR
== status
|| (Z_OK
== status
&& Z
->avail_in
)) && ++round
< HTTP_INFLATE_ROUNDS
);
259 if (status
== Z_OK
|| status
== Z_STREAM_END
) {
260 phpstr_shrink(&buffer
);
265 phpstr_dtor(&buffer
);
272 /* {{{ STATUS http_encoding_deflate(int, char *, size_t, char **, size_t *) */
273 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
)
275 int status
, level
, wbits
, strategy
;
278 HTTP_DEFLATE_LEVEL_SET(flags
, level
);
279 HTTP_DEFLATE_WBITS_SET(flags
, wbits
);
280 HTTP_DEFLATE_STRATEGY_SET(flags
, strategy
);
282 memset(&Z
, 0, sizeof(z_stream
));
286 status
= deflateInit2(&Z
, level
, Z_DEFLATED
, wbits
, MAX_MEM_LEVEL
, strategy
);
287 if (Z_OK
== status
) {
288 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE_GUESS(data_len
);
289 *encoded
= emalloc_rel(*encoded_len
);
291 Z
.next_in
= (Bytef
*) data
;
292 Z
.next_out
= (Bytef
*) *encoded
;
293 Z
.avail_in
= data_len
;
294 Z
.avail_out
= *encoded_len
;
296 status
= deflate(&Z
, Z_FINISH
);
299 if (Z_STREAM_END
== status
) {
300 /* size buffer down to actual length */
301 *encoded
= erealloc_rel(*encoded
, Z
.total_out
+ 1);
302 (*encoded
)[*encoded_len
= Z
.total_out
] = '\0';
305 STR_SET(*encoded
, NULL
);
310 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Could not deflate data: %s", zError(status
));
315 /* {{{ STATUS http_encoding_inflate(char *, size_t, char **, size_t) */
316 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
)
319 int status
, wbits
= HTTP_WINDOW_BITS_ANY
;
321 memset(&Z
, 0, sizeof(z_stream
));
324 status
= inflateInit2(&Z
, wbits
);
325 if (Z_OK
== status
) {
326 Z
.next_in
= (Bytef
*) data
;
327 Z
.avail_in
= data_len
;
329 switch (status
= http_inflate_rounds(&Z
, Z_NO_FLUSH
, decoded
, decoded_len
)) {
336 /* raw deflated data? */
337 if (HTTP_WINDOW_BITS_ANY
== wbits
) {
339 wbits
= HTTP_WINDOW_BITS_RAW
;
340 goto retry_raw_inflate
;
346 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Could not inflate data: %s", zError(status
));
351 /* {{{ http_encoding_stream *_http_encoding_deflate_stream_init(http_encoding_stream *, int) */
352 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
)
354 int status
, level
, wbits
, strategy
, free_stream
;
356 if ((free_stream
= !s
)) {
357 s
= pemalloc_rel(sizeof(http_encoding_stream
), (flags
& HTTP_ENCODING_STREAM_PERSISTENT
));
359 memset(s
, 0, sizeof(http_encoding_stream
));
362 HTTP_DEFLATE_LEVEL_SET(flags
, level
);
363 HTTP_DEFLATE_WBITS_SET(flags
, wbits
);
364 HTTP_DEFLATE_STRATEGY_SET(flags
, strategy
);
366 if (Z_OK
== (status
= deflateInit2(&s
->stream
, level
, Z_DEFLATED
, wbits
, MAX_MEM_LEVEL
, strategy
))) {
367 int p
= (flags
& HTTP_ENCODING_STREAM_PERSISTENT
) ? PHPSTR_INIT_PERSISTENT
:0;
369 if ((s
->stream
.opaque
= phpstr_init_ex(NULL
, HTTP_DEFLATE_BUFFER_SIZE
, p
))) {
372 deflateEnd(&s
->stream
);
373 status
= Z_MEM_ERROR
;
376 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to initialize deflate encoding stream: %s", zError(status
));
384 /* {{{ http_encoding_stream *http_encoding_inflate_stream_init(http_encoding_stream *, int) */
385 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
)
387 int status
, wbits
, free_stream
;
389 if ((free_stream
= !s
)) {
390 s
= pemalloc_rel(sizeof(http_encoding_stream
), (flags
& HTTP_ENCODING_STREAM_PERSISTENT
));
392 memset(s
, 0, sizeof(http_encoding_stream
));
395 HTTP_INFLATE_WBITS_SET(flags
, wbits
);
397 if (Z_OK
== (status
= inflateInit2(&s
->stream
, wbits
))) {
398 int p
= (flags
& HTTP_ENCODING_STREAM_PERSISTENT
) ? PHPSTR_INIT_PERSISTENT
:0;
400 if ((s
->stream
.opaque
= phpstr_init_ex(NULL
, HTTP_DEFLATE_BUFFER_SIZE
, p
))) {
403 inflateEnd(&s
->stream
);
404 status
= Z_MEM_ERROR
;
407 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to initialize inflate stream: %s", zError(status
));
415 /* {{{ STATUS http_encoding_deflate_stream_update(http_encoding_stream *, char *, size_t, char **, size_t *) */
416 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
)
420 /* append input to our buffer */
421 phpstr_append(PHPSTR(s
->stream
.opaque
), data
, data_len
);
423 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
424 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
427 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE_GUESS(data_len
);
428 *encoded
= emalloc_rel(*encoded_len
);
429 s
->stream
.avail_out
= *encoded_len
;
430 s
->stream
.next_out
= (Bytef
*) *encoded
;
432 switch (status
= deflate(&s
->stream
, HTTP_ENCODING_STREAM_FLUSH_FLAG(s
->flags
))) {
435 /* cut processed chunk off the buffer */
436 if (s
->stream
.avail_in
) {
437 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
439 phpstr_reset(PHPSTR(s
->stream
.opaque
));
442 /* size buffer down to actual size */
443 *encoded_len
-= s
->stream
.avail_out
;
444 *encoded
= erealloc_rel(*encoded
, *encoded_len
+ 1);
445 (*encoded
)[*encoded_len
] = '\0';
449 STR_SET(*encoded
, NULL
);
451 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to update deflate stream: %s", zError(status
));
456 /* {{{ STATUS http_encoding_inflate_stream_update(http_encoding_stream *, char *, size_t, char **, size_t *) */
457 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
)
461 /* append input to buffer */
462 phpstr_append(PHPSTR(s
->stream
.opaque
), data
, data_len
);
465 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
466 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
468 switch (status
= http_inflate_rounds(&s
->stream
, HTTP_ENCODING_STREAM_FLUSH_FLAG(s
->flags
), decoded
, decoded_len
)) {
472 if (s
->stream
.avail_in
) {
473 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
475 phpstr_reset(PHPSTR(s
->stream
.opaque
));
480 /* raw deflated data ? */
481 if (!(s
->flags
& HTTP_INFLATE_TYPE_RAW
) && !s
->stream
.total_out
) {
482 inflateEnd(&s
->stream
);
483 s
->flags
|= HTTP_INFLATE_TYPE_RAW
;
484 inflateInit2(&s
->stream
, HTTP_WINDOW_BITS_RAW
);
485 goto retry_raw_inflate
;
489 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to update inflate stream: %s", zError(status
));
494 /* {{{ STATUS http_encoding_deflate_stream_flush(http_encoding_stream *, char **, size_t *) */
495 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
)
499 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE
;
500 *encoded
= emalloc_rel(*encoded_len
);
502 s
->stream
.avail_in
= 0;
503 s
->stream
.next_in
= NULL
;
504 s
->stream
.avail_out
= *encoded_len
;
505 s
->stream
.next_out
= (Bytef
*) *encoded
;
507 switch (status
= deflate(&s
->stream
, Z_FULL_FLUSH
)) {
510 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE
- s
->stream
.avail_out
;
511 *encoded
= erealloc_rel(*encoded
, *encoded_len
+ 1);
512 (*encoded
)[*encoded_len
] = '\0';
516 STR_SET(*encoded
, NULL
);
518 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to flush deflate stream: %s", zError(status
));
523 /* {{{ STATUS http_encoding_inflate_straem_flush(http_encoding_stream *, char **, size_t *) */
524 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
)
527 *decoded
= estrndup("", *decoded_len
= 0);
532 /* {{{ STATUS http_encoding_deflate_stream_finish(http_encoding_stream *, char **, size_t *) */
533 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
)
537 *encoded_len
= HTTP_DEFLATE_BUFFER_SIZE
;
538 *encoded
= emalloc_rel(*encoded_len
);
540 /* deflate remaining input */
541 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
542 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
544 s
->stream
.avail_out
= *encoded_len
;
545 s
->stream
.next_out
= (Bytef
*) *encoded
;
548 status
= deflate(&s
->stream
, Z_FINISH
);
549 } while (Z_OK
== status
);
551 if (Z_STREAM_END
== status
) {
552 /* cut processed intp off */
553 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
556 *encoded_len
-= s
->stream
.avail_out
;
557 *encoded
= erealloc_rel(*encoded
, *encoded_len
+ 1);
558 (*encoded
)[*encoded_len
] = '\0';
562 STR_SET(*encoded
, NULL
);
564 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to finish deflate stream: %s", zError(status
));
569 /* {{{ STATUS http_encoding_inflate_stream_finish(http_encoding_stream *, char **, size_t *) */
570 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
)
574 if (!PHPSTR_LEN(s
->stream
.opaque
)) {
580 *decoded_len
= (PHPSTR_LEN(s
->stream
.opaque
) + 1) * HTTP_INFLATE_ROUNDS
;
581 *decoded
= emalloc_rel(*decoded_len
);
583 /* inflate remaining input */
584 s
->stream
.next_in
= (Bytef
*) PHPSTR_VAL(s
->stream
.opaque
);
585 s
->stream
.avail_in
= PHPSTR_LEN(s
->stream
.opaque
);
587 s
->stream
.avail_out
= *decoded_len
;
588 s
->stream
.next_out
= (Bytef
*) *decoded
;
590 if (Z_STREAM_END
== (status
= inflate(&s
->stream
, Z_FINISH
))) {
591 /* cut processed input off */
592 phpstr_cut(PHPSTR(s
->stream
.opaque
), 0, PHPSTR_LEN(s
->stream
.opaque
) - s
->stream
.avail_in
);
595 *decoded_len
-= s
->stream
.avail_out
;
596 *decoded
= erealloc_rel(*decoded
, *decoded_len
+ 1);
597 (*decoded
)[*decoded_len
] = '\0';
601 STR_SET(*decoded
, NULL
);
603 http_error_ex(HE_WARNING
, HTTP_E_ENCODING
, "Failed to finish inflate stream: %s", zError(status
));
608 /* {{{ void http_encoding_deflate_stream_dtor(http_encoding_stream *) */
609 PHP_HTTP_API
void _http_encoding_deflate_stream_dtor(http_encoding_stream
*s TSRMLS_DC
)
612 if (s
->stream
.opaque
) {
613 phpstr_free((phpstr
**) &s
->stream
.opaque
);
615 deflateEnd(&s
->stream
);
620 /* {{{ void http_encoding_inflate_stream_dtor(http_encoding_stream *) */
621 PHP_HTTP_API
void _http_encoding_inflate_stream_dtor(http_encoding_stream
*s TSRMLS_DC
)
624 if (s
->stream
.opaque
) {
625 phpstr_free((phpstr
**) &s
->stream
.opaque
);
627 inflateEnd(&s
->stream
);
632 /* {{{ void http_encoding_deflate_stream_free(http_encoding_stream **) */
633 PHP_HTTP_API
void _http_encoding_deflate_stream_free(http_encoding_stream
**s TSRMLS_DC
)
636 http_encoding_deflate_stream_dtor(*s
);
638 pefree(*s
, (*s
)->flags
& HTTP_ENCODING_STREAM_PERSISTENT
);
645 /* {{{ void http_encoding_inflate_stream_free(http_encoding_stream **) */
646 PHP_HTTP_API
void _http_encoding_inflate_stream_free(http_encoding_stream
**s TSRMLS_DC
)
649 http_encoding_inflate_stream_dtor(*s
);
651 pefree(*s
, (*s
)->flags
& HTTP_ENCODING_STREAM_PERSISTENT
);
658 /* {{{ void http_ob_deflatehandler(char *, uint, char **, uint *, int) */
659 void _http_ob_deflatehandler(char *output
, uint output_len
, char **handled_output
, uint
*handled_output_len
, int mode TSRMLS_DC
)
663 *handled_output
= NULL
;
664 *handled_output_len
= 0;
666 if (mode
& PHP_OUTPUT_HANDLER_START
) {
669 if (HTTP_G
->send
.deflate
.stream
) {
670 zend_error(E_ERROR
, "ob_deflatehandler() can only be used once");
674 HTTP_G
->send
.deflate
.response
= 1;
675 encoding
= http_encoding_response_start(0, 1);
676 HTTP_G
->send
.deflate
.response
= 0;
679 case HTTP_ENCODING_GZIP
:
680 flags
= HTTP_DEFLATE_TYPE_GZIP
;
683 case HTTP_ENCODING_DEFLATE
:
684 flags
= HTTP_DEFLATE_TYPE_ZLIB
;
688 goto deflate_passthru_plain
;
691 flags
|= (HTTP_G
->send
.deflate
.start_flags
&~ 0xf0);
692 HTTP_G
->send
.deflate
.stream
= http_encoding_deflate_stream_init(NULL
, flags
);
695 if (HTTP_G
->send
.deflate
.stream
) {
699 http_encoding_deflate_stream_update((http_encoding_stream
*) HTTP_G
->send
.deflate
.stream
, output
, output_len
, handled_output
, &tmp_len
);
700 *handled_output_len
= tmp_len
;
703 if (mode
& PHP_OUTPUT_HANDLER_END
) {
704 char *remaining
= NULL
;
705 size_t remaining_len
= 0;
707 http_encoding_deflate_stream_finish((http_encoding_stream
*) HTTP_G
->send
.deflate
.stream
, &remaining
, &remaining_len
);
708 http_encoding_deflate_stream_free((http_encoding_stream
**) &HTTP_G
->send
.deflate
.stream
);
710 *handled_output
= erealloc(*handled_output
, *handled_output_len
+ remaining_len
+ 1);
711 memcpy(*handled_output
+ *handled_output_len
, remaining
, remaining_len
);
712 (*handled_output
)[*handled_output_len
+= remaining_len
] = '\0';
717 deflate_passthru_plain
:
718 *handled_output
= estrndup(output
, *handled_output_len
= output_len
);
723 /* {{{ void http_ob_inflatehandler(char *, uint, char **, uint *, int) */
724 void _http_ob_inflatehandler(char *output
, uint output_len
, char **handled_output
, uint
*handled_output_len
, int mode TSRMLS_DC
)
726 *handled_output
= NULL
;
727 *handled_output_len
= 0;
729 if (mode
& PHP_OUTPUT_HANDLER_START
) {
730 if (HTTP_G
->send
.inflate
.stream
) {
731 zend_error(E_ERROR
, "ob_inflatehandler() can only be used once");
734 HTTP_G
->send
.inflate
.stream
= http_encoding_inflate_stream_init(NULL
, (HTTP_G
->send
.inflate
.start_flags
&~ 0xf0));
737 if (HTTP_G
->send
.inflate
.stream
) {
741 http_encoding_inflate_stream_update((http_encoding_stream
*) HTTP_G
->send
.inflate
.stream
, output
, output_len
, handled_output
, &tmp_len
);
742 *handled_output_len
= tmp_len
;
745 if (mode
& PHP_OUTPUT_HANDLER_END
) {
746 char *remaining
= NULL
;
747 size_t remaining_len
= 0;
749 http_encoding_inflate_stream_finish((http_encoding_stream
*) HTTP_G
->send
.inflate
.stream
, &remaining
, &remaining_len
);
750 http_encoding_inflate_stream_free((http_encoding_stream
**) &HTTP_G
->send
.inflate
.stream
);
752 *handled_output
= erealloc(*handled_output
, *handled_output_len
+ remaining_len
+ 1);
753 memcpy(*handled_output
+ *handled_output_len
, remaining
, remaining_len
);
754 (*handled_output
)[*handled_output_len
+= remaining_len
] = '\0';
759 *handled_output
= estrndup(output
, *handled_output_len
= output_len
);
764 #endif /* HTTP_HAVE_ZLIB */
771 * vim600: noet sw=4 ts=4 fdm=marker
772 * vim<600: noet sw=4 ts=4