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-2011, Michael Wallner <mike@php.net> |
10 +--------------------------------------------------------------------+
16 #include <ext/standard/php_lcg.h>
17 #include <ext/standard/php_string.h>
19 #define BOUNDARY_OPEN(body) \
21 size_t size = php_http_message_body_size(body); \
23 php_stream_truncate_set_size(php_http_message_body_stream(body), size - lenof("--" PHP_HTTP_CRLF)); \
24 php_http_message_body_append(body, ZEND_STRL(PHP_HTTP_CRLF)); \
26 php_http_message_body_appendf(body, "--%s" PHP_HTTP_CRLF, php_http_message_body_boundary(body)); \
29 #define BOUNDARY_CLOSE(body) \
30 php_http_message_body_appendf(body, PHP_HTTP_CRLF "--%s--" PHP_HTTP_CRLF, php_http_message_body_boundary(body))
32 static STATUS
add_recursive_fields(php_http_message_body_t
*body
, const char *name
, zval
*value
);
33 static STATUS
add_recursive_files(php_http_message_body_t
*body
, const char *name
, zval
*value
);
35 PHP_HTTP_API php_http_message_body_t
*php_http_message_body_init(php_http_message_body_t
*body
, php_stream
*stream TSRMLS_DC
)
38 body
= emalloc(sizeof(php_http_message_body_t
));
40 memset(body
, 0, sizeof(*body
));
43 php_stream_auto_cleanup(stream
);
44 body
->stream_id
= php_stream_get_resource_id(stream
);
45 zend_list_addref(body
->stream_id
);
47 stream
= php_stream_temp_create(TEMP_STREAM_DEFAULT
, 0xffff);
48 php_stream_auto_cleanup(stream
);
49 body
->stream_id
= php_stream_get_resource_id(stream
);
51 TSRMLS_SET_CTX(body
->ts
);
56 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
)
61 TSRMLS_FETCH_FROM_CTX(from
->ts
);
63 if (dup_internal_stream_and_contents
) {
64 to
= php_http_message_body_init(to
, NULL TSRMLS_CC
);
65 php_http_message_body_to_stream(from
, php_http_message_body_stream(to
), 0, 0);
67 to
= php_http_message_body_init(to
, php_http_message_body_stream(from
) TSRMLS_CC
);
71 to
->boundary
= estrdup(from
->boundary
);
78 PHP_HTTP_API
void php_http_message_body_dtor(php_http_message_body_t
*body
)
80 TSRMLS_FETCH_FROM_CTX(body
->ts
);
81 /* NO FIXME: shows leakinfo in DEBUG mode */
82 zend_list_delete(body
->stream_id
);
83 STR_FREE(body
->boundary
);
86 PHP_HTTP_API
void php_http_message_body_free(php_http_message_body_t
**body
)
89 php_http_message_body_dtor(*body
);
95 PHP_HTTP_API
const php_stream_statbuf
*php_http_message_body_stat(php_http_message_body_t
*body
)
97 TSRMLS_FETCH_FROM_CTX(body
->ts
);
98 php_stream_stat(php_http_message_body_stream(body
), &body
->ssb
);
102 PHP_HTTP_API
const char *php_http_message_body_boundary(php_http_message_body_t
*body
)
104 if (!body
->boundary
) {
105 union { double dbl
; int num
[2]; } data
;
106 TSRMLS_FETCH_FROM_CTX(body
->ts
);
108 data
.dbl
= php_combined_lcg(TSRMLS_C
);
109 spprintf(&body
->boundary
, 0, "%x.%x", data
.num
[0], data
.num
[1]);
111 return body
->boundary
;
114 PHP_HTTP_API
char *php_http_message_body_etag(php_http_message_body_t
*body
)
116 TSRMLS_FETCH_FROM_CTX(body
->ts
);
117 const php_stream_statbuf
*ssb
= php_http_message_body_stat(body
);
119 /* real file or temp buffer ? */
120 if (body
->ssb
.sb
.st_mtime
) {
123 spprintf(&etag
, 0, "%lx-%lx-%lx", ssb
->sb
.st_ino
, ssb
->sb
.st_mtime
, ssb
->sb
.st_size
);
126 php_http_etag_t
*etag
= php_http_etag_init(PHP_HTTP_G
->env
.etag_mode TSRMLS_CC
);
128 php_http_message_body_to_callback(body
, (php_http_pass_callback_t
) php_http_etag_update
, etag
, 0, 0);
129 return php_http_etag_finish(etag
);
133 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
)
135 TSRMLS_FETCH_FROM_CTX(body
->ts
);
136 php_stream
*s
= php_http_message_body_stream(body
);
138 php_stream_seek(s
, offset
, SEEK_SET
);
142 *len
= php_stream_copy_to_mem(s
, buf
, forlen
, 0);
145 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
)
147 TSRMLS_FETCH_FROM_CTX(body
->ts
);
148 php_stream
*s
= php_http_message_body_stream(body
);
150 php_stream_seek(s
, offset
, SEEK_SET
);
154 php_stream_copy_to_stream_ex(s
, dst
, forlen
, NULL
);
157 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
)
159 TSRMLS_FETCH_FROM_CTX(body
->ts
);
160 php_stream
*s
= php_http_message_body_stream(body
);
161 char *buf
= emalloc(0x1000);
163 php_stream_seek(s
, offset
, SEEK_SET
);
168 while (!php_stream_eof(s
)) {
169 size_t read
= php_stream_read(s
, buf
, MIN(forlen
, 0x1000));
172 cb(cb_arg
, buf
, read
);
175 if (read
< MIN(forlen
, sizeof(buf
))) {
179 if (forlen
&& !(forlen
-= read
)) {
186 PHP_HTTP_API
size_t php_http_message_body_append(php_http_message_body_t
*body
, const char *buf
, size_t len
)
189 TSRMLS_FETCH_FROM_CTX(body
->ts
);
191 s
= php_http_message_body_stream(body
);
192 php_stream_seek(s
, 0, SEEK_END
);
193 return php_stream_write(s
, buf
, len
);
196 PHP_HTTP_API
size_t php_http_message_body_appendf(php_http_message_body_t
*body
, const char *fmt
, ...)
203 print_len
= vspprintf(&print_str
, 0, fmt
, argv
);
206 print_len
= php_http_message_body_append(body
, print_str
, print_len
);
212 PHP_HTTP_API STATUS
php_http_message_body_add(php_http_message_body_t
*body
, HashTable
*fields
, HashTable
*files
)
217 INIT_PZVAL_ARRAY(&tmp
, fields
);
218 if (SUCCESS
!= add_recursive_fields(body
, NULL
, &tmp
)) {
223 INIT_PZVAL_ARRAY(&tmp
, files
);
224 if (SUCCESS
!= add_recursive_files(body
, NULL
, &tmp
)) {
233 PHP_HTTP_API STATUS
php_http_message_body_add_field(php_http_message_body_t
*body
, const char *name
, const char *value_str
, size_t value_len
)
236 TSRMLS_FETCH_FROM_CTX(body
->ts
);
238 safe_name
= php_addslashes(estrdup(name
), strlen(name
), NULL
, 1 TSRMLS_CC
);
241 php_http_message_body_appendf(
243 "Content-Disposition: form-data; name=\"%s\"" PHP_HTTP_CRLF
246 php_http_message_body_append(body
, value_str
, value_len
);
247 BOUNDARY_CLOSE(body
);
253 PHP_HTTP_API STATUS
php_http_message_body_add_file(php_http_message_body_t
*body
, const char *name
, const char *path
, const char *ctype
)
256 char *path_dup
= estrdup(path
);
257 TSRMLS_FETCH_FROM_CTX(body
->ts
);
259 if ((in
= php_stream_open_wrapper(path_dup
, "r", REPORT_ERRORS
|USE_PATH
|STREAM_MUST_SEEK
, NULL
))) {
260 php_stream_statbuf ssb
= {{0}};
262 if (SUCCESS
== php_stream_stat(in
, &ssb
) && S_ISREG(ssb
.sb
.st_mode
)) {
263 char *safe_name
= php_addslashes(estrdup(name
), strlen(name
), NULL
, 1 TSRMLS_CC
);
266 php_http_message_body_appendf(
268 "Content-Disposition: attachment; name=\"%s\"; filename=\"%s\"" PHP_HTTP_CRLF
269 "Content-Type: %s" PHP_HTTP_CRLF
270 "Content-Length: %zu" PHP_HTTP_CRLF
272 safe_name
, basename(path_dup
), ctype
, ssb
.sb
.st_size
);
273 php_stream_copy_to_stream_ex(in
, php_http_message_body_stream(body
), PHP_STREAM_COPY_ALL
, NULL
);
274 BOUNDARY_CLOSE(body
);
278 php_stream_close(in
);
282 php_stream_close(in
);
283 php_http_error(HE_WARNING
, PHP_HTTP_E_MESSAGE_BODY
, "Not a valid regular file: %s", path
);
293 static inline char *format_key(uint type
, char *str
, ulong num
, const char *prefix
) {
294 char *new_key
= NULL
;
296 if (prefix
&& *prefix
) {
297 if (type
== HASH_KEY_IS_STRING
) {
298 spprintf(&new_key
, 0, "%s[%s]", prefix
, str
);
300 spprintf(&new_key
, 0, "%s[%lu]", prefix
, num
);
302 } else if (type
== HASH_KEY_IS_STRING
) {
303 new_key
= estrdup(str
);
305 spprintf(&new_key
, 0, "%lu", num
);
311 static STATUS
add_recursive_fields(php_http_message_body_t
*body
, const char *name
, zval
*value
)
313 if (Z_TYPE_P(value
) == IS_ARRAY
|| Z_TYPE_P(value
) == IS_OBJECT
) {
316 php_http_array_hashkey_t key
= php_http_array_hashkey_init(0);
317 TSRMLS_FETCH_FROM_CTX(body
->ts
);
319 if (!HASH_OF(value
)->nApplyCount
) {
320 ++HASH_OF(value
)->nApplyCount
;
321 FOREACH_KEYVAL(pos
, value
, key
, val
) {
322 char *str
= format_key(key
.type
, key
.str
, key
.num
, name
);
323 if (SUCCESS
!= add_recursive_fields(body
, str
, *val
)) {
325 HASH_OF(value
)->nApplyCount
--;
330 --HASH_OF(value
)->nApplyCount
;
333 zval
*cpy
= php_http_ztyp(IS_STRING
, value
);
334 php_http_message_body_add_field(body
, name
, Z_STRVAL_P(cpy
), Z_STRLEN_P(cpy
));
341 static STATUS
add_recursive_files(php_http_message_body_t
*body
, const char *name
, zval
*value
)
343 if (Z_TYPE_P(value
) == IS_ARRAY
|| Z_TYPE_P(value
) == IS_OBJECT
) {
344 zval
**zfile
, **zname
, **ztype
;
345 TSRMLS_FETCH_FROM_CTX(body
->ts
);
347 if ((SUCCESS
== zend_hash_find(HASH_OF(value
), ZEND_STRS("name"), (void *) &zname
))
348 && (SUCCESS
== zend_hash_find(HASH_OF(value
), ZEND_STRS("file"), (void *) &zfile
))
349 && (SUCCESS
== zend_hash_find(HASH_OF(value
), ZEND_STRS("type"), (void *) &ztype
))
351 zval
*zfc
= php_http_ztyp(IS_STRING
, *zfile
), *znc
= php_http_ztyp(IS_STRING
, *zname
), *ztc
= php_http_ztyp(IS_STRING
, *ztype
);
352 char *str
= format_key(HASH_KEY_IS_STRING
, Z_STRVAL_P(znc
), 0, name
);
353 STATUS ret
= php_http_message_body_add_file(body
, str
, Z_STRVAL_P(zfc
), Z_STRVAL_P(ztc
));
360 if (ret
!= SUCCESS
) {
366 php_http_array_hashkey_t key
= php_http_array_hashkey_init(0);
368 if (!HASH_OF(value
)->nApplyCount
) {
369 ++HASH_OF(value
)->nApplyCount
;
370 FOREACH_KEYVAL(pos
, value
, key
, val
) {
371 char *str
= format_key(key
.type
, key
.str
, key
.num
, name
);
372 if (SUCCESS
!= add_recursive_files(body
, str
, *val
)) {
374 --HASH_OF(value
)->nApplyCount
;
379 --HASH_OF(value
)->nApplyCount
;
383 TSRMLS_FETCH_FROM_CTX(body
->ts
);
384 php_http_error(HE_WARNING
, PHP_HTTP_E_MESSAGE_BODY
, "Unrecognized array format for message body file to add");
393 #define PHP_HTTP_BEGIN_ARGS(method, req_args) PHP_HTTP_BEGIN_ARGS_EX(HttpMessageBody, method, 0, req_args)
394 #define PHP_HTTP_EMPTY_ARGS(method) PHP_HTTP_EMPTY_ARGS_EX(HttpMessageBody, method, 0)
395 #define PHP_HTTP_MESSAGE_BODY_ME(method, visibility) PHP_ME(HttpMessageBody, method, PHP_HTTP_ARGS(HttpMessageBody, method), visibility)
397 PHP_HTTP_BEGIN_ARGS(__construct
, 0)
398 PHP_HTTP_ARG_VAL(stream
, 0)
401 PHP_HTTP_EMPTY_ARGS(__toString
);
403 PHP_HTTP_BEGIN_ARGS(toStream
, 1)
404 PHP_HTTP_ARG_VAL(stream
, 0)
407 PHP_HTTP_BEGIN_ARGS(toCallback
, 1)
408 PHP_HTTP_ARG_VAL(callback
, 0)
411 PHP_HTTP_BEGIN_ARGS(append
, 1)
412 PHP_HTTP_ARG_VAL(string
, 0)
415 PHP_HTTP_BEGIN_ARGS(add
, 0)
416 PHP_HTTP_ARG_VAL(fields
, 0)
417 PHP_HTTP_ARG_VAL(files
, 0)
421 zend_class_entry
*php_http_message_body_class_entry
;
422 zend_function_entry php_http_message_body_method_entry
[] = {
423 PHP_HTTP_MESSAGE_BODY_ME(__construct
, ZEND_ACC_PUBLIC
|ZEND_ACC_CTOR
)
424 PHP_HTTP_MESSAGE_BODY_ME(__toString
, ZEND_ACC_PUBLIC
)
425 PHP_MALIAS(HttpMessageBody
, toString
, __toString
, args_for_HttpMessageBody___toString
, ZEND_ACC_PUBLIC
)
426 PHP_HTTP_MESSAGE_BODY_ME(toStream
, ZEND_ACC_PUBLIC
)
427 PHP_HTTP_MESSAGE_BODY_ME(toCallback
, ZEND_ACC_PUBLIC
)
428 PHP_HTTP_MESSAGE_BODY_ME(append
, ZEND_ACC_PUBLIC
)
429 PHP_HTTP_MESSAGE_BODY_ME(add
, ZEND_ACC_PUBLIC
)
432 static zend_object_handlers php_http_message_body_object_handlers
;
434 PHP_MINIT_FUNCTION(http_message_body
)
436 PHP_HTTP_REGISTER_CLASS(http
\\Message
, Body
, http_message_body
, php_http_object_class_entry
, 0);
437 php_http_message_body_class_entry
->create_object
= php_http_message_body_object_new
;
438 memcpy(&php_http_message_body_object_handlers
, zend_get_std_object_handlers(), sizeof(zend_object_handlers
));
439 php_http_message_body_object_handlers
.clone_obj
= php_http_message_body_object_clone
;
444 zend_object_value
php_http_message_body_object_new(zend_class_entry
*ce TSRMLS_DC
)
446 return php_http_message_body_object_new_ex(ce
, NULL
, NULL TSRMLS_CC
);
449 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
)
451 zend_object_value ov
;
452 php_http_message_body_object_t
*o
;
454 o
= ecalloc(1, sizeof(php_http_message_body_object_t
));
455 zend_object_std_init((zend_object
*) o
, php_http_message_body_class_entry TSRMLS_CC
);
456 object_properties_init((zend_object
*) o
, ce
);
466 ov
.handle
= zend_objects_store_put((zend_object
*) o
, NULL
, php_http_message_body_object_free
, NULL TSRMLS_CC
);
467 ov
.handlers
= &php_http_message_body_object_handlers
;
472 zend_object_value
php_http_message_body_object_clone(zval
*object TSRMLS_DC
)
474 zend_object_value new_ov
;
475 php_http_message_body_object_t
*new_obj
= NULL
;
476 php_http_message_body_object_t
*old_obj
= zend_object_store_get_object(object TSRMLS_CC
);
478 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
);
479 zend_objects_clone_members(&new_obj
->zo
, new_ov
, &old_obj
->zo
, Z_OBJ_HANDLE_P(object
) TSRMLS_CC
);
484 void php_http_message_body_object_free(void *object TSRMLS_DC
)
486 php_http_message_body_object_t
*obj
= object
;
488 php_http_message_body_free(&obj
->body
);
490 zend_object_std_dtor((zend_object
*) obj TSRMLS_CC
);
494 PHP_METHOD(HttpMessageBody
, __construct
)
496 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
497 zval
*zstream
= NULL
;
500 with_error_handling(EH_THROW
, php_http_exception_class_entry
) {
501 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|r!", &zstream
)) {
503 php_stream_from_zval(stream
, &zstream
);
507 php_http_message_body_dtor(obj
->body
);
509 obj
->body
= php_http_message_body_init(obj
->body
, stream TSRMLS_CC
);
513 obj
->body
= php_http_message_body_init(NULL
, NULL TSRMLS_CC
);
516 } end_error_handling();
519 PHP_METHOD(HttpMessageBody
, __toString
)
521 if (SUCCESS
== zend_parse_parameters_none()) {
522 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
526 php_http_message_body_to_string(obj
->body
, &str
, &len
, 0, 0);
528 RETURN_STRINGL(str
, len
, 0);
531 RETURN_EMPTY_STRING();
534 PHP_METHOD(HttpMessageBody
, toStream
)
537 long offset
= 0, forlen
= 0;
539 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "r|ll", &zstream
, &offset
, &forlen
)) {
541 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
543 php_stream_from_zval(stream
, &zstream
);
544 php_http_message_body_to_stream(obj
->body
, stream
, offset
, forlen
);
552 zend_fcall_info
*fci
;
553 zend_fcall_info_cache
*fcc
;
559 static size_t pass(void *cb_arg
, const char *str
, size_t len
)
561 struct fcd
*fcd
= cb_arg
;
563 TSRMLS_FETCH_FROM_CTX(fcd
->ts
);
565 MAKE_STD_ZVAL(zdata
);
566 ZVAL_STRINGL(zdata
, str
, len
, 1);
567 if (SUCCESS
== zend_fcall_info_argn(fcd
->fci TSRMLS_CC
, 2, fcd
->fcz
, zdata
)) {
568 zend_fcall_info_call(fcd
->fci
, fcd
->fcc
, NULL
, NULL TSRMLS_CC
);
569 zend_fcall_info_args_clear(fcd
->fci
, 0);
571 zval_ptr_dtor(&zdata
);
575 PHP_METHOD(HttpMessageBody
, toCallback
)
578 long offset
= 0, forlen
= 0;
580 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "f|ll", &fcd
.fci
, &fcd
.fcc
, &offset
, &forlen
)) {
581 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
584 php_http_message_body_to_callback(obj
->body
, pass
, &fcd
, offset
, forlen
);
585 zval_ptr_dtor(&fcd
.fcz
);
591 PHP_METHOD(HttpMessageBody
, append
)
596 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "s", &str
, &len
)) {
597 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
599 RETURN_LONG(php_http_message_body_append(obj
->body
, str
, len
));
604 PHP_METHOD(HttpMessageBody
, add
)
606 HashTable
*fields
= NULL
, *files
= NULL
;
608 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|hh", &fields
, &files
)) {
609 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
611 RETURN_SUCCESS(php_http_message_body_add(obj
->body
, fields
, files
));
620 * vim600: noet sw=4 ts=4 fdm=marker
621 * vim<600: noet sw=4 ts=4