git.m6w6.name Git - m6w6/ext-http/blob - http_deflatestream_object.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-2006, Michael Wallner <mike@php.net>            |
  10     +--------------------------------------------------------------------+
  11 */
  12
  13 /* $Id$ */
  14
  15
  16 #ifdef HAVE_CONFIG_H
  17 #       include "config.h"
  18 #endif
  19
  20 #define HTTP_WANT_ZLIB
  21 #include "php_http.h"
  22
  23 #if defined(ZEND_ENGINE_2) && defined(HTTP_HAVE_ZLIB)
  24
  25 #include "php_http_api.h"
  26 #include "php_http_encoding_api.h"
  27 #include "php_http_exception_object.h"
  28 #include "php_http_deflatestream_object.h"
  29
  30 #define HTTP_BEGIN_ARGS(method, ret_ref, req_args)      HTTP_BEGIN_ARGS_EX(HttpDeflateStream, method, ret_ref, req_args)
  31 #define HTTP_EMPTY_ARGS(method, ret_ref)                        HTTP_EMPTY_ARGS_EX(HttpDeflateStream, method, ret_ref)
  32 #define HTTP_DEFLATE_ME(method, visibility)                     PHP_ME(HttpDeflateStream, method, HTTP_ARGS(HttpDeflateStream, method), visibility)
  33
  34 HTTP_BEGIN_ARGS(__construct, 0, 0)
  35         HTTP_ARG_VAL(flags, 0)
  36 HTTP_END_ARGS;
  37
  38 HTTP_BEGIN_ARGS(update, 0, 1)
  39         HTTP_ARG_VAL(data, 0)
  40 HTTP_END_ARGS;
  41
  42 HTTP_BEGIN_ARGS(flush, 0, 0)
  43         HTTP_ARG_VAL(data, 0)
  44 HTTP_END_ARGS;
  45
  46 HTTP_BEGIN_ARGS(finish, 0, 0)
  47         HTTP_ARG_VAL(data, 0)
  48 HTTP_END_ARGS;
  49
  50 #define http_deflatestream_object_declare_default_properties() _http_deflatestream_object_declare_default_properties(TSRMLS_C)
  51 static inline void _http_deflatestream_object_declare_default_properties(TSRMLS_D);
  52
  53 zend_class_entry *http_deflatestream_object_ce;
  54 zend_function_entry http_deflatestream_object_fe[] = {
  55         HTTP_DEFLATE_ME(__construct, ZEND_ACC_PUBLIC|ZEND_ACC_CTOR)
  56         HTTP_DEFLATE_ME(update, ZEND_ACC_PUBLIC)
  57         HTTP_DEFLATE_ME(flush, ZEND_ACC_PUBLIC)
  58         HTTP_DEFLATE_ME(finish, ZEND_ACC_PUBLIC)
  59
  60         EMPTY_FUNCTION_ENTRY
  61 };
  62 static zend_object_handlers http_deflatestream_object_handlers;
  63
  64 PHP_MINIT_FUNCTION(http_deflatestream_object)
  65 {
  66         HTTP_REGISTER_CLASS_EX(HttpDeflateStream, http_deflatestream_object, NULL, 0);
  67         http_deflatestream_object_handlers.clone_obj = _http_deflatestream_object_clone_obj;
  68         return SUCCESS;
  69 }
  70
  71 zend_object_value _http_deflatestream_object_new(zend_class_entry *ce TSRMLS_DC)
  72 {
  73         return http_deflatestream_object_new_ex(ce, NULL, NULL);
  74 }
  75
  76 zend_object_value _http_deflatestream_object_new_ex(zend_class_entry *ce, http_encoding_stream *s, http_deflatestream_object **ptr TSRMLS_DC)
  77 {
  78         zend_object_value ov;
  79         http_deflatestream_object *o;
  80
  81         o = ecalloc(1, sizeof(http_deflatestream_object));
  82         o->zo.ce = ce;
  83
  84         if (ptr) {
  85                 *ptr = o;
  86         }
  87
  88         if (s) {
  89                 o->stream = s;
  90         }
  91
  92         ALLOC_HASHTABLE(OBJ_PROP(o));
  93         zend_hash_init(OBJ_PROP(o), 0, NULL, ZVAL_PTR_DTOR, 0);
  94
  95         ov.handle = putObject(http_deflatestream_object, o);
  96         ov.handlers = &http_deflatestream_object_handlers;
  97
  98         return ov;
  99 }
 100
 101 zend_object_value _http_deflatestream_object_clone_obj(zval *this_ptr TSRMLS_DC)
 102 {
 103         http_encoding_stream *s;
 104         getObject(http_deflatestream_object, obj);
 105
 106         s = ecalloc(1, sizeof(http_encoding_stream));
 107         s->flags = obj->stream->flags;
 108         deflateCopy(&s->stream, &obj->stream->stream);
 109         s->stream.opaque = phpstr_dup(s->stream.opaque);
 110
 111         return http_deflatestream_object_new_ex(Z_OBJCE_P(this_ptr), s, NULL);
 112 }
 113
 114 static inline void _http_deflatestream_object_declare_default_properties(TSRMLS_D)
 115 {
 116         zend_class_entry *ce = http_deflatestream_object_ce;
 117
 118 #ifndef WONKY
 119         DCL_CONST(long, "TYPE_GZIP", HTTP_DEFLATE_TYPE_GZIP);
 120         DCL_CONST(long, "TYPE_ZLIB", HTTP_DEFLATE_TYPE_ZLIB);
 121         DCL_CONST(long, "TYPE_RAW", HTTP_DEFLATE_TYPE_RAW);
 122         DCL_CONST(long, "LEVEL_DEF", HTTP_DEFLATE_LEVEL_DEF);
 123         DCL_CONST(long, "LEVEL_MIN", HTTP_DEFLATE_LEVEL_MIN);
 124         DCL_CONST(long, "LEVEL_MAX", HTTP_DEFLATE_LEVEL_MAX);
 125         DCL_CONST(long, "STRATEGY_DEF", HTTP_DEFLATE_STRATEGY_DEF);
 126         DCL_CONST(long, "STRATEGY_FILT", HTTP_DEFLATE_STRATEGY_FILT);
 127         DCL_CONST(long, "STRATEGY_HUFF", HTTP_DEFLATE_STRATEGY_HUFF);
 128         DCL_CONST(long, "STRATEGY_RLE", HTTP_DEFLATE_STRATEGY_RLE);
 129         DCL_CONST(long, "STRATEGY_FIXED", HTTP_DEFLATE_STRATEGY_FIXED);
 130 #endif
 131 }
 132
 133 void _http_deflatestream_object_free(zend_object *object TSRMLS_DC)
 134 {
 135         http_deflatestream_object *o = (http_deflatestream_object *) object;
 136
 137         if (OBJ_PROP(o)) {
 138                 zend_hash_destroy(OBJ_PROP(o));
 139                 FREE_HASHTABLE(OBJ_PROP(o));
 140         }
 141         if (o->stream) {
 142                 http_encoding_deflate_stream_free(&o->stream);
 143         }
 144         efree(o);
 145 }
 146
 147 /* {{{ proto void HttpDeflateStream::__construct([int flags = 0])
 148  *
 149  * Creates a new HttpDeflateStream object instance.
 150  *
 151  * Accepts an optional int parameter specifying how to initialize the deflate stream.
 152  */
 153 PHP_METHOD(HttpDeflateStream, __construct)
 154 {
 155         long flags = 0;
 156
 157         SET_EH_THROW_HTTP();
 158         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|l", &flags)) {
 159                 getObject(http_deflatestream_object, obj);
 160
 161                 if (!obj->stream) {
 162                         obj->stream = http_encoding_deflate_stream_init(NULL, flags);
 163                 } else {
 164                         http_error_ex(HE_WARNING, HTTP_E_ENCODING, "HttpDeflateStream cannot be initialized twice");
 165                 }
 166         }
 167         SET_EH_NORMAL();
 168 }
 169 /* }}} */
 170
 171 /* {{{ proto string HttpDeflateStream::update(string data)
 172  *
 173  * Passes more data through the deflate stream.
 174  *
 175  * Expects a string parameter containing (a part of) the data to deflate.
 176  *
 177  * Returns deflated data on success or FALSE on failure.
 178  */
 179 PHP_METHOD(HttpDeflateStream, update)
 180 {
 181         int data_len;
 182         size_t encoded_len = 0;
 183         char *data, *encoded = NULL;
 184         getObject(http_deflatestream_object, obj);
 185
 186         if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &data, &data_len)) {
 187                 RETURN_FALSE;
 188         }
 189
 190         if (!obj->stream && !(obj->stream = http_encoding_deflate_stream_init(NULL, 0))) {
 191                 RETURN_FALSE;
 192         }
 193
 194         if (SUCCESS == http_encoding_deflate_stream_update(obj->stream, data, data_len, &encoded, &encoded_len)) {
 195                 RETURN_STRINGL(encoded, encoded_len, 0);
 196         } else {
 197                 RETURN_FALSE;
 198         }
 199 }
 200 /* }}} */
 201
 202 /* {{{ proto string HttpDeflateStream::flush([string data])
 203  *
 204  * Flushes the deflate stream.
 205  *
 206  * Returns some deflated data as string on success or FALSE on failure.
 207  */
 208 PHP_METHOD(HttpDeflateStream, flush)
 209 {
 210         int data_len = 0;
 211         size_t updated_len = 0, encoded_len = 0;
 212         char *updated = NULL, *encoded = NULL, *data = NULL;
 213         getObject(http_deflatestream_object, obj);
 214
 215         if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|s", &data, &data_len)) {
 216                 RETURN_FALSE;
 217         }
 218
 219         if (!obj->stream && !(obj->stream = http_encoding_deflate_stream_init(NULL, 0))) {
 220                 RETURN_FALSE;
 221         }
 222
 223         if (data_len) {
 224                 if (SUCCESS != http_encoding_deflate_stream_update(obj->stream, data, data_len, &updated, &updated_len)) {
 225                         RETURN_FALSE;
 226                 }
 227         }
 228
 229         if (SUCCESS == http_encoding_deflate_stream_flush(obj->stream, &encoded, &encoded_len)) {
 230                 if (updated_len) {
 231                         updated = erealloc(updated, updated_len + encoded_len + 1);
 232                         updated[updated_len + encoded_len] = '\0';
 233                         memcpy(updated + updated_len, encoded, encoded_len);
 234                         STR_FREE(encoded);
 235                         updated_len += encoded_len;
 236                         RETURN_STRINGL(updated, updated_len, 0);
 237                 } else {
 238                         RETVAL_STRINGL(encoded, encoded_len, 0);
 239                 }
 240         } else {
 241                 RETVAL_FALSE;
 242         }
 243         STR_FREE(updated);
 244 }
 245 /* }}} */
 246
 247 /* {{{ proto string HttpDeflateStream::finish([string data])
 248  *
 249  * Finalizes the deflate stream.  The deflate stream can be reused after finalizing.
 250  *
 251  * Returns the final part of deflated data.
 252  */
 253 PHP_METHOD(HttpDeflateStream, finish)
 254 {
 255         int data_len = 0;
 256         size_t updated_len = 0, encoded_len = 0;
 257         char *updated = NULL, *encoded = NULL, *data = NULL;
 258         getObject(http_deflatestream_object, obj);
 259
 260         if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|s", &data, &data_len)) {
 261                 RETURN_FALSE;
 262         }
 263
 264         if (!obj->stream && !(obj->stream = http_encoding_deflate_stream_init(NULL, 0))) {
 265                 RETURN_FALSE;
 266         }
 267
 268         if (data_len) {
 269                 if (SUCCESS != http_encoding_deflate_stream_update(obj->stream, data, data_len, &updated, &updated_len)) {
 270                         RETURN_FALSE;
 271                 }
 272         }
 273
 274         if (SUCCESS == http_encoding_deflate_stream_finish(obj->stream, &encoded, &encoded_len)) {
 275                 if (updated_len) {
 276                         updated = erealloc(updated, updated_len + encoded_len + 1);
 277                         updated[updated_len + encoded_len] = '\0';
 278                         memcpy(updated + updated_len, encoded, encoded_len);
 279                         STR_FREE(encoded);
 280                         updated_len += encoded_len;
 281                         RETVAL_STRINGL(updated, updated_len, 0);
 282                 } else {
 283                         STR_FREE(updated);
 284                         RETVAL_STRINGL(encoded, encoded_len, 0);
 285                 }
 286         } else {
 287                 STR_FREE(updated);
 288                 RETVAL_FALSE;
 289         }
 290
 291         http_encoding_deflate_stream_dtor(obj->stream);
 292         http_encoding_deflate_stream_init(obj->stream, obj->stream->flags);
 293 }
 294 /* }}} */
 295
 296
 297 #endif /* ZEND_ENGINE_2 && HTTP_HAVE_ZLIB*/
 298
 299 /*
 300  * Local variables:
 301  * tab-width: 4
 302  * c-basic-offset: 4
 303  * End:
 304  * vim600: noet sw=4 ts=4 fdm=marker
 305  * vim<600: noet sw=4 ts=4
 306  */
 307