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