git.m6w6.name Git - m6w6/ext-http/blob - php_http_message_body.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-2011, Michael Wallner <mike@php.net>            |
  10     +--------------------------------------------------------------------+
  11 */
  12
  13 #include "php_http_api.h"
  14
  15 #include <ext/standard/php_lcg.h>
  16
  17 #define BOUNDARY_OPEN(body) \
  18         do {\
  19                 size_t size = php_http_message_body_size(body); \
  20                 if (size) { \
  21                         php_stream_truncate_set_size(php_http_message_body_stream(body), size - lenof("--" PHP_HTTP_CRLF)); \
  22                         php_http_message_body_append(body, ZEND_STRL(PHP_HTTP_CRLF)); \
  23                 } else { \
  24                         php_http_message_body_appendf(body, "--%s" PHP_HTTP_CRLF, php_http_message_body_boundary(body)); \
  25                 } \
  26         } while(0)
  27 #define BOUNDARY_CLOSE(body) \
  28                 php_http_message_body_appendf(body, PHP_HTTP_CRLF "--%s--" PHP_HTTP_CRLF, php_http_message_body_boundary(body))
  29
  30 static STATUS add_recursive_fields(php_http_message_body_t *body, const char *name, zval *value);
  31 static STATUS add_recursive_files(php_http_message_body_t *body, const char *name, zval *value);
  32
  33 PHP_HTTP_API php_http_message_body_t *php_http_message_body_init(php_http_message_body_t *body, php_stream *stream TSRMLS_DC)
  34 {
  35         if (!body) {
  36                 body = emalloc(sizeof(php_http_message_body_t));
  37         }
  38         memset(body, 0, sizeof(*body));
  39
  40         if (stream) {
  41                 php_stream_auto_cleanup(stream);
  42                 body->stream_id = php_stream_get_resource_id(stream);
  43                 zend_list_addref(body->stream_id);
  44         } else {
  45                 stream = php_stream_temp_create(TEMP_STREAM_DEFAULT, 0xffff);
  46                 php_stream_auto_cleanup(stream);
  47                 body->stream_id = php_stream_get_resource_id(stream);
  48         }
  49         TSRMLS_SET_CTX(body->ts);
  50
  51         return body;
  52 }
  53
  54 PHP_HTTP_API php_http_message_body_t *php_http_message_body_copy(php_http_message_body_t *from, php_http_message_body_t *to, zend_bool dup_internal_stream_and_contents)
  55 {
  56         if (!from) {
  57                 return NULL;
  58         } else {
  59                 TSRMLS_FETCH_FROM_CTX(from->ts);
  60
  61                 if (dup_internal_stream_and_contents) {
  62                         to = php_http_message_body_init(to, NULL TSRMLS_CC);
  63                         php_http_message_body_to_stream(from, php_http_message_body_stream(to), 0, 0);
  64                 } else {
  65                         to = php_http_message_body_init(to, php_http_message_body_stream(from) TSRMLS_CC);
  66                 }
  67
  68                 if (from->boundary) {
  69                         to->boundary = estrdup(from->boundary);
  70                 }
  71
  72                 return to;
  73         }
  74 }
  75
  76 PHP_HTTP_API void php_http_message_body_dtor(php_http_message_body_t *body)
  77 {
  78         TSRMLS_FETCH_FROM_CTX(body->ts);
  79         /* NO FIXME: shows leakinfo in DEBUG mode */
  80         zend_list_delete(body->stream_id);
  81         STR_FREE(body->boundary);
  82 }
  83
  84 PHP_HTTP_API void php_http_message_body_free(php_http_message_body_t **body)
  85 {
  86         if (*body) {
  87                 php_http_message_body_dtor(*body);
  88                 efree(*body);
  89                 *body = NULL;
  90         }
  91 }
  92
  93 PHP_HTTP_API const php_stream_statbuf *php_http_message_body_stat(php_http_message_body_t *body)
  94 {
  95         TSRMLS_FETCH_FROM_CTX(body->ts);
  96         php_stream_stat(php_http_message_body_stream(body), &body->ssb);
  97         return &body->ssb;
  98 }
  99
 100 PHP_HTTP_API const char *php_http_message_body_boundary(php_http_message_body_t *body)
 101 {
 102         if (!body->boundary) {
 103                 union { double dbl; int num[2]; } data;
 104                 TSRMLS_FETCH_FROM_CTX(body->ts);
 105
 106                 data.dbl = php_combined_lcg(TSRMLS_C);
 107                 spprintf(&body->boundary, 0, "%x.%x", data.num[0], data.num[1]);
 108         }
 109         return body->boundary;
 110 }
 111
 112 PHP_HTTP_API char *php_http_message_body_etag(php_http_message_body_t *body)
 113 {
 114         const php_stream_statbuf *ssb = php_http_message_body_stat(body);
 115         TSRMLS_FETCH_FROM_CTX(body->ts);
 116
 117         /* real file or temp buffer ? */
 118         if (ssb && ssb->sb.st_mtime) {
 119                 char *etag;
 120
 121                 spprintf(&etag, 0, "%lx-%lx-%lx", ssb->sb.st_ino, ssb->sb.st_mtime, ssb->sb.st_size);
 122                 return etag;
 123         } else {
 124                 php_http_etag_t *etag = php_http_etag_init(PHP_HTTP_G->env.etag_mode TSRMLS_CC);
 125
 126                 if (etag) {
 127                         php_http_message_body_to_callback(body, (php_http_pass_callback_t) php_http_etag_update, etag, 0, 0);
 128                         return php_http_etag_finish(etag);
 129                 } else {
 130                         return NULL;
 131                 }
 132         }
 133 }
 134
 135 PHP_HTTP_API void php_http_message_body_to_string(php_http_message_body_t *body, char **buf, size_t *len, off_t offset, size_t forlen)
 136 {
 137         php_stream *s = php_http_message_body_stream(body);
 138         TSRMLS_FETCH_FROM_CTX(body->ts);
 139
 140         php_stream_seek(s, offset, SEEK_SET);
 141         if (!forlen) {
 142                 forlen = -1;
 143         }
 144         *len = php_stream_copy_to_mem(s, buf, forlen, 0);
 145 }
 146
 147 PHP_HTTP_API void php_http_message_body_to_stream(php_http_message_body_t *body, php_stream *dst, off_t offset, size_t forlen)
 148 {
 149         php_stream *s = php_http_message_body_stream(body);
 150         TSRMLS_FETCH_FROM_CTX(body->ts);
 151
 152         php_stream_seek(s, offset, SEEK_SET);
 153         if (!forlen) {
 154                 forlen = -1;
 155         }
 156         php_stream_copy_to_stream_ex(s, dst, forlen, NULL);
 157 }
 158
 159 PHP_HTTP_API void php_http_message_body_to_callback(php_http_message_body_t *body, php_http_pass_callback_t cb, void *cb_arg, off_t offset, size_t forlen)
 160 {
 161         php_stream *s = php_http_message_body_stream(body);
 162         char *buf = emalloc(0x1000);
 163         TSRMLS_FETCH_FROM_CTX(body->ts);
 164
 165         php_stream_seek(s, offset, SEEK_SET);
 166
 167         if (!forlen) {
 168                 forlen = -1;
 169         }
 170         while (!php_stream_eof(s)) {
 171                 size_t read = php_stream_read(s, buf, MIN(forlen, 0x1000));
 172
 173                 if (read) {
 174                         cb(cb_arg, buf, read);
 175                 }
 176
 177                 if (read < MIN(forlen, sizeof(buf))) {
 178                         break;
 179                 }
 180
 181                 if (forlen && !(forlen -= read)) {
 182                         break;
 183                 }
 184         }
 185         efree(buf);
 186 }
 187
 188 PHP_HTTP_API size_t php_http_message_body_append(php_http_message_body_t *body, const char *buf, size_t len)
 189 {
 190         php_stream *s;
 191         TSRMLS_FETCH_FROM_CTX(body->ts);
 192
 193         s = php_http_message_body_stream(body);
 194         php_stream_seek(s, 0, SEEK_END);
 195         return php_stream_write(s, buf, len);
 196 }
 197
 198 PHP_HTTP_API size_t php_http_message_body_appendf(php_http_message_body_t *body, const char *fmt, ...)
 199 {
 200         va_list argv;
 201         char *print_str;
 202         size_t print_len;
 203
 204         va_start(argv, fmt);
 205         print_len = vspprintf(&print_str, 0, fmt, argv);
 206         va_end(argv);
 207
 208         print_len = php_http_message_body_append(body, print_str, print_len);
 209         efree(print_str);
 210
 211         return print_len;
 212 }
 213
 214 PHP_HTTP_API STATUS php_http_message_body_add_form(php_http_message_body_t *body, HashTable *fields, HashTable *files)
 215 {
 216         zval tmp;
 217
 218         if (fields) {
 219                 INIT_PZVAL_ARRAY(&tmp, fields);
 220                 if (SUCCESS != add_recursive_fields(body, NULL, &tmp)) {
 221                         return FAILURE;
 222                 }
 223         }
 224         if (files) {
 225                 INIT_PZVAL_ARRAY(&tmp, files);
 226                 if (SUCCESS != add_recursive_files(body, NULL, &tmp)) {
 227                         return FAILURE;
 228                 }
 229         }
 230
 231         return SUCCESS;
 232 }
 233
 234 PHP_HTTP_API void php_http_message_body_add_part(php_http_message_body_t *body, php_http_message_t *part)
 235 {
 236         TSRMLS_FETCH_FROM_CTX(body->ts);
 237
 238         BOUNDARY_OPEN(body);
 239         php_http_message_to_callback(part, (php_http_pass_callback_t) php_http_message_body_append, body);
 240         BOUNDARY_CLOSE(body);
 241 }
 242
 243
 244 PHP_HTTP_API STATUS php_http_message_body_add_form_field(php_http_message_body_t *body, const char *name, const char *value_str, size_t value_len)
 245 {
 246         char *safe_name;
 247         TSRMLS_FETCH_FROM_CTX(body->ts);
 248
 249         safe_name = php_addslashes(estrdup(name), strlen(name), NULL, 1 TSRMLS_CC);
 250
 251         BOUNDARY_OPEN(body);
 252         php_http_message_body_appendf(
 253                 body,
 254                 "Content-Disposition: form-data; name=\"%s\"" PHP_HTTP_CRLF
 255                 "" PHP_HTTP_CRLF,
 256                 safe_name
 257         );
 258         php_http_message_body_append(body, value_str, value_len);
 259         BOUNDARY_CLOSE(body);
 260
 261         efree(safe_name);
 262         return SUCCESS;
 263 }
 264
 265 PHP_HTTP_API STATUS php_http_message_body_add_form_file(php_http_message_body_t *body, const char *name, const char *ctype, const char *path, php_stream *in)
 266 {
 267         char *safe_name, *path_dup = estrdup(path), *bname;
 268         size_t bname_len;
 269         TSRMLS_FETCH_FROM_CTX(body->ts);
 270
 271         safe_name = php_addslashes(estrdup(name), strlen(name), NULL, 1 TSRMLS_CC);
 272
 273         php_basename(path_dup, strlen(path_dup), NULL, 0, &bname, &bname_len TSRMLS_CC);
 274
 275         BOUNDARY_OPEN(body);
 276         php_http_message_body_appendf(
 277                 body,
 278                 "Content-Disposition: form-data; name=\"%s\"; filename=\"%s\"" PHP_HTTP_CRLF
 279                 "Content-Transfer-Encoding: binary" PHP_HTTP_CRLF
 280                 "Content-Type: %s" PHP_HTTP_CRLF
 281                 PHP_HTTP_CRLF,
 282                 safe_name, bname, ctype
 283         );
 284         php_stream_copy_to_stream_ex(in, php_http_message_body_stream(body), PHP_STREAM_COPY_ALL, NULL);
 285         BOUNDARY_CLOSE(body);
 286
 287         efree(safe_name);
 288         efree(path_dup);
 289         efree(bname);
 290
 291         return SUCCESS;
 292 }
 293
 294 static inline char *format_key(uint type, char *str, ulong num, const char *prefix) {
 295         char *new_key = NULL;
 296
 297         if (prefix && *prefix) {
 298                 if (type == HASH_KEY_IS_STRING) {
 299                         spprintf(&new_key, 0, "%s[%s]", prefix, str);
 300                 } else {
 301                         spprintf(&new_key, 0, "%s[%lu]", prefix, num);
 302                 }
 303         } else if (type == HASH_KEY_IS_STRING) {
 304                 new_key = estrdup(str);
 305         } else {
 306                 new_key = estrdup("");
 307         }
 308
 309         return new_key;
 310 }
 311
 312 static STATUS add_recursive_fields(php_http_message_body_t *body, const char *name, zval *value)
 313 {
 314         if (Z_TYPE_P(value) == IS_ARRAY || Z_TYPE_P(value) == IS_OBJECT) {
 315                 zval **val;
 316                 HashTable *ht;
 317                 HashPosition pos;
 318                 php_http_array_hashkey_t key = php_http_array_hashkey_init(0);
 319                 TSRMLS_FETCH_FROM_CTX(body->ts);
 320
 321                 ht = HASH_OF(value);
 322                 if (!ht->nApplyCount) {
 323                         ++ht->nApplyCount;
 324                         FOREACH_KEYVAL(pos, value, key, val) {
 325                                 char *str = format_key(key.type, key.str, key.num, name);
 326                                 if (SUCCESS != add_recursive_fields(body, str, *val)) {
 327                                         efree(str);
 328                                         ht->nApplyCount--;
 329                                         return FAILURE;
 330                                 }
 331                                 efree(str);
 332                         }
 333                         --ht->nApplyCount;
 334                 }
 335         } else {
 336                 zval *cpy = php_http_ztyp(IS_STRING, value);
 337                 php_http_message_body_add_form_field(body, name, Z_STRVAL_P(cpy), Z_STRLEN_P(cpy));
 338                 zval_ptr_dtor(&cpy);
 339         }
 340
 341         return SUCCESS;
 342 }
 343
 344 static STATUS add_recursive_files(php_http_message_body_t *body, const char *name, zval *value)
 345 {
 346         zval **zdata = NULL, **zfile, **zname, **ztype;
 347         HashTable *ht;
 348         TSRMLS_FETCH_FROM_CTX(body->ts);
 349
 350         if (Z_TYPE_P(value) != IS_ARRAY && Z_TYPE_P(value) != IS_OBJECT) {
 351                 php_http_error(HE_WARNING, PHP_HTTP_E_MESSAGE_BODY, "Expected array or object (name, type, file) for message body file to add");
 352                 return FAILURE;
 353         }
 354
 355         ht = HASH_OF(value);
 356
 357         if ((SUCCESS != zend_hash_find(ht, ZEND_STRS("name"), (void *) &zname))
 358         ||      (SUCCESS != zend_hash_find(ht, ZEND_STRS("type"), (void *) &ztype))
 359         ||      (SUCCESS != zend_hash_find(ht, ZEND_STRS("file"), (void *) &zfile))
 360         ) {
 361                 zval **val;
 362                 HashPosition pos;
 363                 php_http_array_hashkey_t key = php_http_array_hashkey_init(0);
 364
 365                 if (!ht->nApplyCount) {
 366                         ++ht->nApplyCount;
 367                         FOREACH_HASH_KEYVAL(pos, ht, key, val) {
 368                                 if (Z_TYPE_PP(val) == IS_ARRAY || Z_TYPE_PP(val) == IS_OBJECT) {
 369                                         char *str = format_key(key.type, key.str, key.num, name);
 370
 371                                         if (SUCCESS != add_recursive_files(body, str, *val)) {
 372                                                 efree(str);
 373                                                 --ht->nApplyCount;
 374                                                 return FAILURE;
 375                                         }
 376                                         efree(str);
 377                                 }
 378                         }
 379                         --ht->nApplyCount;
 380                 }
 381                 return SUCCESS;
 382         } else {
 383                 php_stream *stream;
 384                 zval *zfc = php_http_ztyp(IS_STRING, *zfile);
 385
 386                 if (SUCCESS == zend_hash_find(ht, ZEND_STRS("data"), (void *) &zdata)) {
 387                         if (Z_TYPE_PP(zdata) == IS_RESOURCE) {
 388                                 php_stream_from_zval_no_verify(stream, zdata);
 389                         } else {
 390                                 zval *tmp = php_http_ztyp(IS_STRING, *zdata);
 391
 392                                 stream = php_stream_memory_open(TEMP_STREAM_READONLY, Z_STRVAL_P(tmp), Z_STRLEN_P(tmp));
 393                                 zval_ptr_dtor(&tmp);
 394                         }
 395                 } else {
 396                         stream = php_stream_open_wrapper(Z_STRVAL_P(zfc), "r", REPORT_ERRORS|USE_PATH, NULL);
 397                 }
 398
 399                 if (!stream) {
 400                         zval_ptr_dtor(&zfc);
 401                         return FAILURE;
 402                 } else {
 403                         zval *znc = php_http_ztyp(IS_STRING, *zname), *ztc = php_http_ztyp(IS_STRING, *ztype);
 404                         char *key = format_key(HASH_KEY_IS_STRING, Z_STRVAL_P(znc), 0, name);
 405                         STATUS ret =  php_http_message_body_add_form_file(body, key, Z_STRVAL_P(ztc), Z_STRVAL_P(zfc), stream);
 406
 407                         efree(key);
 408                         zval_ptr_dtor(&znc);
 409                         zval_ptr_dtor(&ztc);
 410                         zval_ptr_dtor(&zfc);
 411                         if (!zdata || Z_TYPE_PP(zdata) != IS_RESOURCE) {
 412                                 php_stream_close(stream);
 413                         }
 414                         return ret;
 415                 }
 416
 417         }
 418 }
 419
 420 struct splitbody_arg {
 421         php_http_buffer_t buf;
 422         php_http_message_parser_t *parser;
 423         char *boundary_str;
 424         size_t boundary_len;
 425         size_t consumed;
 426 };
 427
 428 static size_t splitbody(void *opaque, char *buf, size_t len TSRMLS_DC)
 429 {
 430         struct splitbody_arg *arg = opaque;
 431         const char *boundary = NULL;
 432         size_t consumed = 0;
 433         int first_boundary;
 434
 435         do {
 436                 first_boundary = !(consumed || arg->consumed);
 437
 438                 if ((boundary = php_http_locate_str(buf, len, arg->boundary_str + first_boundary, arg->boundary_len - first_boundary))) {
 439                         size_t real_boundary_len = arg->boundary_len - 1, cut;
 440                         const char *real_boundary = boundary + !first_boundary;
 441                         int eol_len = 0;
 442
 443                         if (buf + len <= real_boundary + real_boundary_len) {
 444                                 /* if we just have enough data for the boundary, it's just a byte too less */
 445                                 arg->consumed += consumed;
 446                                 return consumed;
 447                         }
 448
 449                         if (!first_boundary) {
 450                                 /* this is not the first boundary, read rest of this message */
 451                                 php_http_buffer_append(&arg->buf, buf, real_boundary - buf);
 452                                 php_http_message_parser_parse(arg->parser, &arg->buf, 0, &arg->parser->message);
 453                         }
 454
 455                         /* move after the boundary */
 456                         cut = real_boundary - buf + real_boundary_len;
 457                         buf += cut;
 458                         len -= cut;
 459                         consumed += cut;
 460
 461                         if (buf == php_http_locate_bin_eol(buf, len, &eol_len)) {
 462                                 /* skip CRLF */
 463                                 buf += eol_len;
 464                                 len -= eol_len;
 465                                 consumed += eol_len;
 466
 467                                 if (!first_boundary) {
 468                                         /* advance messages */
 469                                         php_http_message_t *msg;
 470
 471                                         msg = php_http_message_init(NULL, 0 TSRMLS_CC);
 472                                         msg->parent = arg->parser->message;
 473                                         arg->parser->message = msg;
 474                                 }
 475                         } else {
 476                                 /* is this the last boundary? */
 477                                 if (*buf == '-') {
 478                                         /* ignore the rest */
 479                                         consumed += len;
 480                                         len = 0;
 481                                 } else {
 482                                         /* let this be garbage */
 483                                         php_http_error(HE_WARNING, PHP_HTTP_E_MESSAGE_BODY, "Malformed multipart boundary at pos %zu", consumed);
 484                                         return -1;
 485                                 }
 486                         }
 487                 }
 488         } while (boundary && len);
 489
 490         /* let there be room for the next boundary */
 491         if (len > arg->boundary_len) {
 492                 consumed += len - arg->boundary_len;
 493                 php_http_buffer_append(&arg->buf, buf, len - arg->boundary_len);
 494                 php_http_message_parser_parse(arg->parser, &arg->buf, 0, &arg->parser->message);
 495         }
 496
 497         arg->consumed += consumed;
 498         return consumed;
 499 }
 500
 501 PHP_HTTP_API php_http_message_t *php_http_message_body_split(php_http_message_body_t *body, const char *boundary)
 502 {
 503         php_stream *s = php_http_message_body_stream(body);
 504         php_http_buffer_t *tmp = NULL;
 505         php_http_message_t *msg = NULL;
 506         struct splitbody_arg arg;
 507         TSRMLS_FETCH_FROM_CTX(body->ts);
 508
 509         php_http_buffer_init(&arg.buf);
 510         arg.parser = php_http_message_parser_init(NULL TSRMLS_CC);
 511         arg.boundary_len = spprintf(&arg.boundary_str, 0, "\n--%s", boundary);
 512         arg.consumed = 0;
 513
 514         php_stream_rewind(s);
 515         while (!php_stream_eof(s)) {
 516                 php_http_buffer_passthru(&tmp, 0x1000, (php_http_buffer_pass_func_t) _php_stream_read, s, splitbody, &arg TSRMLS_CC);
 517         }
 518
 519         msg = arg.parser->message;
 520         arg.parser->message = NULL;
 521
 522         php_http_buffer_free(&tmp);
 523         php_http_message_parser_free(&arg.parser);
 524         php_http_buffer_dtor(&arg.buf);
 525         STR_FREE(arg.boundary_str);
 526
 527         return msg;
 528 }
 529
 530 /* PHP */
 531
 532 #define PHP_HTTP_BEGIN_ARGS(method, req_args)                   PHP_HTTP_BEGIN_ARGS_EX(HttpMessageBody, method, 0, req_args)
 533 #define PHP_HTTP_EMPTY_ARGS(method)                                             PHP_HTTP_EMPTY_ARGS_EX(HttpMessageBody, method, 0)
 534 #define PHP_HTTP_MESSAGE_BODY_ME(method, visibility)    PHP_ME(HttpMessageBody, method, PHP_HTTP_ARGS(HttpMessageBody, method), visibility)
 535
 536 PHP_HTTP_BEGIN_ARGS(__construct, 0)
 537         PHP_HTTP_ARG_VAL(stream, 0)
 538 PHP_HTTP_END_ARGS;
 539
 540 PHP_HTTP_EMPTY_ARGS(__toString);
 541 PHP_HTTP_BEGIN_ARGS(unserialize, 1)
 542         PHP_HTTP_ARG_VAL(serialized, 0)
 543 PHP_HTTP_END_ARGS;
 544
 545 PHP_HTTP_BEGIN_ARGS(toStream, 1)
 546         PHP_HTTP_ARG_VAL(stream, 0)
 547 PHP_HTTP_END_ARGS;
 548
 549 PHP_HTTP_BEGIN_ARGS(toCallback, 1)
 550         PHP_HTTP_ARG_VAL(callback, 0)
 551 PHP_HTTP_END_ARGS;
 552
 553 PHP_HTTP_EMPTY_ARGS(getResource);
 554
 555 PHP_HTTP_BEGIN_ARGS(append, 1)
 556         PHP_HTTP_ARG_VAL(string, 0)
 557 PHP_HTTP_END_ARGS;
 558
 559 PHP_HTTP_BEGIN_ARGS(addForm, 0)
 560         PHP_HTTP_ARG_ARR(fields, 1, 0)
 561         PHP_HTTP_ARG_ARR(files, 1, 0)
 562 PHP_HTTP_END_ARGS;
 563
 564 PHP_HTTP_BEGIN_ARGS(addPart, 1)
 565         PHP_HTTP_ARG_OBJ(http\\Message, message, 0)
 566 PHP_HTTP_END_ARGS;
 567
 568 PHP_HTTP_EMPTY_ARGS(etag);
 569
 570 PHP_HTTP_BEGIN_ARGS(stat, 0)
 571         PHP_HTTP_ARG_VAL(what, 0)
 572 PHP_HTTP_END_ARGS;
 573
 574 static zend_class_entry *php_http_message_body_class_entry;
 575
 576 zend_class_entry *php_http_message_body_get_class_entry(void)
 577 {
 578         return php_http_message_body_class_entry;
 579 }
 580
 581 static zend_function_entry php_http_message_body_method_entry[] = {
 582         PHP_HTTP_MESSAGE_BODY_ME(__construct, ZEND_ACC_PUBLIC|ZEND_ACC_CTOR)
 583         PHP_HTTP_MESSAGE_BODY_ME(__toString, ZEND_ACC_PUBLIC)
 584         PHP_MALIAS(HttpMessageBody, toString, __toString, args_for_HttpMessageBody___toString, ZEND_ACC_PUBLIC)
 585         PHP_MALIAS(HttpMessageBody, serialize, __toString, args_for_HttpMessageBody___toString, ZEND_ACC_PUBLIC)
 586         PHP_HTTP_MESSAGE_BODY_ME(unserialize, ZEND_ACC_PUBLIC)
 587         PHP_HTTP_MESSAGE_BODY_ME(toStream, ZEND_ACC_PUBLIC)
 588         PHP_HTTP_MESSAGE_BODY_ME(toCallback, ZEND_ACC_PUBLIC)
 589         PHP_HTTP_MESSAGE_BODY_ME(getResource, ZEND_ACC_PUBLIC)
 590         PHP_HTTP_MESSAGE_BODY_ME(append, ZEND_ACC_PUBLIC)
 591         PHP_HTTP_MESSAGE_BODY_ME(addForm, ZEND_ACC_PUBLIC)
 592         PHP_HTTP_MESSAGE_BODY_ME(addPart, ZEND_ACC_PUBLIC)
 593         PHP_HTTP_MESSAGE_BODY_ME(etag, ZEND_ACC_PUBLIC)
 594         PHP_HTTP_MESSAGE_BODY_ME(stat, ZEND_ACC_PUBLIC)
 595         EMPTY_FUNCTION_ENTRY
 596 };
 597 static zend_object_handlers php_http_message_body_object_handlers;
 598
 599 PHP_MINIT_FUNCTION(http_message_body)
 600 {
 601         PHP_HTTP_REGISTER_CLASS(http\\Message, Body, http_message_body, php_http_object_get_class_entry(), 0);
 602         php_http_message_body_class_entry->create_object = php_http_message_body_object_new;
 603         memcpy(&php_http_message_body_object_handlers, zend_get_std_object_handlers(), sizeof(zend_object_handlers));
 604         php_http_message_body_object_handlers.clone_obj = php_http_message_body_object_clone;
 605         zend_class_implements(php_http_message_body_class_entry TSRMLS_CC, 1, zend_ce_serializable);
 606         return SUCCESS;
 607 }
 608
 609 zend_object_value php_http_message_body_object_new(zend_class_entry *ce TSRMLS_DC)
 610 {
 611         return php_http_message_body_object_new_ex(ce, NULL, NULL TSRMLS_CC);
 612 }
 613
 614 zend_object_value php_http_message_body_object_new_ex(zend_class_entry *ce, php_http_message_body_t *body, php_http_message_body_object_t **ptr TSRMLS_DC)
 615 {
 616         zend_object_value ov;
 617         php_http_message_body_object_t *o;
 618
 619         o = ecalloc(1, sizeof(php_http_message_body_object_t));
 620         zend_object_std_init((zend_object *) o, php_http_message_body_class_entry TSRMLS_CC);
 621         object_properties_init((zend_object *) o, ce);
 622
 623         if (ptr) {
 624                 *ptr = o;
 625         }
 626
 627         if (body) {
 628                 o->body = body;
 629         }
 630
 631         ov.handle = zend_objects_store_put((zend_object *) o, NULL, php_http_message_body_object_free, NULL TSRMLS_CC);
 632         ov.handlers = &php_http_message_body_object_handlers;
 633
 634         return ov;
 635 }
 636
 637 zend_object_value php_http_message_body_object_clone(zval *object TSRMLS_DC)
 638 {
 639         zend_object_value new_ov;
 640         php_http_message_body_object_t *new_obj = NULL;
 641         php_http_message_body_object_t *old_obj = zend_object_store_get_object(object TSRMLS_CC);
 642
 643         new_ov = php_http_message_body_object_new_ex(old_obj->zo.ce, php_http_message_body_copy(old_obj->body, NULL, 1), &new_obj TSRMLS_CC);
 644         zend_objects_clone_members(&new_obj->zo, new_ov, &old_obj->zo, Z_OBJ_HANDLE_P(object) TSRMLS_CC);
 645
 646         return new_ov;
 647 }
 648
 649 void php_http_message_body_object_free(void *object TSRMLS_DC)
 650 {
 651         php_http_message_body_object_t *obj = object;
 652
 653         php_http_message_body_free(&obj->body);
 654
 655         zend_object_std_dtor((zend_object *) obj TSRMLS_CC);
 656         efree(obj);
 657 }
 658
 659 PHP_METHOD(HttpMessageBody, __construct)
 660 {
 661         php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 662         zval *zstream = NULL;
 663         php_stream *stream;
 664
 665         with_error_handling(EH_THROW, php_http_exception_get_class_entry()) {
 666                 if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|r!", &zstream)) {
 667                         if (zstream) {
 668                                 php_stream_from_zval(stream, &zstream);
 669
 670                                 if (stream) {
 671                                         if (obj->body) {
 672                                                 php_http_message_body_dtor(obj->body);
 673                                         }
 674                                         obj->body = php_http_message_body_init(obj->body, stream TSRMLS_CC);
 675                                 }
 676                         }
 677                         if (!obj->body) {
 678                                 obj->body = php_http_message_body_init(NULL, NULL TSRMLS_CC);
 679                         }
 680                 }
 681         } end_error_handling();
 682 }
 683
 684 PHP_METHOD(HttpMessageBody, __toString)
 685 {
 686         if (SUCCESS == zend_parse_parameters_none()) {
 687                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 688                 char *str;
 689                 size_t len;
 690
 691                 php_http_message_body_to_string(obj->body, &str, &len, 0, 0);
 692                 if (str) {
 693                         RETURN_STRINGL(str, len, 0);
 694                 }
 695         }
 696         RETURN_EMPTY_STRING();
 697 }
 698
 699 PHP_METHOD(HttpMessageBody, unserialize)
 700 {
 701         char *us_str;
 702         int us_len;
 703
 704         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &us_str, &us_len)) {
 705                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 706                 php_stream *s = php_stream_memory_open(0, us_str, us_len);
 707
 708                 obj->body = php_http_message_body_init(NULL, s TSRMLS_CC);
 709         }
 710 }
 711
 712 PHP_METHOD(HttpMessageBody, toStream)
 713 {
 714         zval *zstream;
 715         long offset = 0, forlen = 0;
 716
 717         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "r|ll", &zstream, &offset, &forlen)) {
 718                 php_stream *stream;
 719                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 720
 721                 php_stream_from_zval(stream, &zstream);
 722                 php_http_message_body_to_stream(obj->body, stream, offset, forlen);
 723                 RETURN_TRUE;
 724         }
 725         RETURN_FALSE;
 726 }
 727
 728
 729 PHP_METHOD(HttpMessageBody, toCallback)
 730 {
 731         php_http_pass_fcall_arg_t fcd;
 732         long offset = 0, forlen = 0;
 733
 734         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "f|ll", &fcd.fci, &fcd.fcc, &offset, &forlen)) {
 735                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 736
 737                 fcd.fcz = getThis();
 738                 Z_ADDREF_P(fcd.fcz);
 739                 TSRMLS_SET_CTX(fcd.ts);
 740
 741                 php_http_message_body_to_callback(obj->body, php_http_pass_fcall_callback, &fcd, offset, forlen);
 742                 zend_fcall_info_args_clear(&fcd.fci, 1);
 743
 744                 zval_ptr_dtor(&fcd.fcz);
 745                 RETURN_TRUE;
 746         }
 747         RETURN_FALSE;
 748 }
 749
 750 PHP_METHOD(HttpMessageBody, getResource)
 751 {
 752         if (SUCCESS == zend_parse_parameters_none()) {
 753                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 754
 755                 zend_list_addref(obj->body->stream_id);
 756                 RETVAL_RESOURCE(obj->body->stream_id);
 757         }
 758 }
 759
 760 PHP_METHOD(HttpMessageBody, append)
 761 {
 762         char *str;
 763         int len;
 764
 765         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "s", &str, &len)) {
 766                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 767
 768                 RETURN_LONG(php_http_message_body_append(obj->body, str, len));
 769         }
 770         RETURN_FALSE;
 771 }
 772
 773 PHP_METHOD(HttpMessageBody, addForm)
 774 {
 775         HashTable *fields = NULL, *files = NULL;
 776
 777         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|h!h!", &fields, &files)) {
 778                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 779
 780                 RETURN_SUCCESS(php_http_message_body_add_form(obj->body, fields, files));
 781         }
 782         RETURN_FALSE;
 783 }
 784
 785 PHP_METHOD(HttpMessageBody, addPart)
 786 {
 787         zval *zobj;
 788
 789         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "O", &zobj, php_http_message_get_class_entry())) {
 790                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 791                 php_http_message_object_t *mobj = zend_object_store_get_object(zobj TSRMLS_CC);
 792
 793                 php_http_message_body_add_part(obj->body, mobj->message);
 794                 RETURN_TRUE;
 795         }
 796         RETURN_FALSE;
 797 }
 798
 799 PHP_METHOD(HttpMessageBody, etag)
 800 {
 801         if (SUCCESS == zend_parse_parameters_none()) {
 802                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 803                 char *etag = php_http_message_body_etag(obj->body);
 804
 805                 if (etag) {
 806                         RETURN_STRING(etag, 0);
 807                 }
 808         }
 809         RETURN_FALSE;
 810 }
 811
 812 PHP_METHOD(HttpMessageBody, stat)
 813 {
 814         char *field_str = NULL;
 815         int field_len = 0;
 816
 817         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "|s", &field_str, &field_len)) {
 818                 php_http_message_body_object_t *obj = zend_object_store_get_object(getThis() TSRMLS_CC);
 819                 const php_stream_statbuf *sb = php_http_message_body_stat(obj->body);
 820
 821                 if (sb) {
 822                         if (field_str && field_len) {
 823                                         switch (*field_str) {
 824                                                 case 's':
 825                                                 case 'S':
 826                                                         RETURN_LONG(sb->sb.st_size);
 827                                                         break;
 828                                                 case 'a':
 829                                                 case 'A':
 830                                                         RETURN_LONG(sb->sb.st_atime);
 831                                                         break;
 832                                                 case 'm':
 833                                                 case 'M':
 834                                                         RETURN_LONG(sb->sb.st_mtime);
 835                                                         break;
 836                                                 case 'c':
 837                                                 case 'C':
 838                                                         RETURN_LONG(sb->sb.st_ctime);
 839                                                         break;
 840                                                 default:
 841                                                         php_http_error(HE_WARNING, PHP_HTTP_E_MESSAGE_BODY, "unknown stat field: '%s' (should be one of [s]ize, [a]time, [m]time or [c]time)", field_str);
 842                                                         break;
 843                                         }
 844                         } else {
 845                                 array_init(return_value);
 846                                 add_assoc_long_ex(return_value, ZEND_STRS("size"), sb->sb.st_size);
 847                                 add_assoc_long_ex(return_value, ZEND_STRS("atime"), sb->sb.st_atime);
 848                                 add_assoc_long_ex(return_value, ZEND_STRS("mtime"), sb->sb.st_mtime);
 849                                 add_assoc_long_ex(return_value, ZEND_STRS("ctime"), sb->sb.st_ctime);
 850                                 return;
 851                         }
 852                 }
 853         }
 854         RETURN_FALSE;
 855 }
 856
 857 /*
 858  * Local variables:
 859  * tab-width: 4
 860  * c-basic-offset: 4
 861  * End:
 862  * vim600: noet sw=4 ts=4 fdm=marker
 863  * vim<600: noet sw=4 ts=4
 864  */