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-2010, Michael Wallner <mike@php.net> |
10 +--------------------------------------------------------------------+
13 /* $Id: http_message_body.c 292841 2009-12-31 08:48:57Z mike $ */
18 #include <ext/standard/php_lcg.h>
19 #include <ext/standard/php_string.h>
21 #define BOUNDARY_OPEN(body) \
23 size_t size = php_http_message_body_size(body); \
25 php_stream_truncate_set_size(php_http_message_body_stream(body), size - lenof("--" PHP_HTTP_CRLF)); \
26 php_http_message_body_append(body, ZEND_STRL(PHP_HTTP_CRLF)); \
28 php_http_message_body_appendf(body, "--%s" PHP_HTTP_CRLF, php_http_message_body_boundary(body)); \
31 #define BOUNDARY_CLOSE(body) \
32 php_http_message_body_appendf(body, PHP_HTTP_CRLF "--%s--" PHP_HTTP_CRLF, php_http_message_body_boundary(body))
34 static STATUS
add_recursive_fields(php_http_message_body_t
*body
, const char *name
, zval
*value
);
35 static STATUS
add_recursive_files(php_http_message_body_t
*body
, const char *name
, zval
*value
);
37 PHP_HTTP_API php_http_message_body_t
*php_http_message_body_init(php_http_message_body_t
*body
, php_stream
*stream TSRMLS_DC
)
40 body
= emalloc(sizeof(php_http_message_body_t
));
42 memset(body
, 0, sizeof(*body
));
45 php_stream_auto_cleanup(stream
);
46 body
->stream_id
= php_stream_get_resource_id(stream
);
47 zend_list_addref(body
->stream_id
);
49 stream
= php_stream_temp_create(TEMP_STREAM_DEFAULT
, 0xffff);
50 php_stream_auto_cleanup(stream
);
51 body
->stream_id
= php_stream_get_resource_id(stream
);
53 TSRMLS_SET_CTX(body
->ts
);
58 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
)
63 TSRMLS_FETCH_FROM_CTX(from
->ts
);
65 if (dup_internal_stream_and_contents
) {
66 to
= php_http_message_body_init(to
, NULL TSRMLS_CC
);
67 php_http_message_body_to_stream(from
, php_http_message_body_stream(to
), 0, 0);
69 to
= php_http_message_body_init(to
, php_http_message_body_stream(from
) TSRMLS_CC
);
73 to
->boundary
= estrdup(from
->boundary
);
80 PHP_HTTP_API
void php_http_message_body_dtor(php_http_message_body_t
*body
)
82 TSRMLS_FETCH_FROM_CTX(body
->ts
);
83 /* NO FIXME: shows leakinfo in DEBUG mode */
84 zend_list_delete(body
->stream_id
);
85 STR_FREE(body
->boundary
);
88 PHP_HTTP_API
void php_http_message_body_free(php_http_message_body_t
**body
)
91 php_http_message_body_dtor(*body
);
97 PHP_HTTP_API
const php_stream_statbuf
*php_http_message_body_stat(php_http_message_body_t
*body
)
99 TSRMLS_FETCH_FROM_CTX(body
->ts
);
100 php_stream_stat(php_http_message_body_stream(body
), &body
->ssb
);
104 PHP_HTTP_API
const char *php_http_message_body_boundary(php_http_message_body_t
*body
)
106 if (!body
->boundary
) {
107 union { double dbl
; int num
[2]; } data
;
108 TSRMLS_FETCH_FROM_CTX(body
->ts
);
110 data
.dbl
= php_combined_lcg(TSRMLS_C
);
111 spprintf(&body
->boundary
, 0, "%x.%x", data
.num
[0], data
.num
[1]);
113 return body
->boundary
;
116 PHP_HTTP_API
char *php_http_message_body_etag(php_http_message_body_t
*body
)
118 TSRMLS_FETCH_FROM_CTX(body
->ts
);
119 const php_stream_statbuf
*ssb
= php_http_message_body_stat(body
);
121 /* real file or temp buffer ? */
122 if (body
->ssb
.sb
.st_mtime
) {
125 spprintf(&etag
, 0, "%lx-%lx-%lx", ssb
->sb
.st_ino
, ssb
->sb
.st_mtime
, ssb
->sb
.st_size
);
128 php_http_etag_t
*etag
= php_http_etag_init(PHP_HTTP_G
->env
.etag_mode TSRMLS_CC
);
130 php_http_message_body_to_callback(body
, (php_http_pass_callback_t
) php_http_etag_update
, etag
, 0, 0);
131 return php_http_etag_finish(etag
);
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
)
137 TSRMLS_FETCH_FROM_CTX(body
->ts
);
138 php_stream
*s
= php_http_message_body_stream(body
);
140 php_stream_seek(s
, offset
, SEEK_SET
);
144 *len
= php_stream_copy_to_mem(s
, buf
, forlen
, 0);
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
)
149 TSRMLS_FETCH_FROM_CTX(body
->ts
);
150 php_stream
*s
= php_http_message_body_stream(body
);
152 php_stream_seek(s
, offset
, SEEK_SET
);
156 php_stream_copy_to_stream_ex(s
, dst
, forlen
, NULL
);
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
)
161 TSRMLS_FETCH_FROM_CTX(body
->ts
);
162 php_stream
*s
= php_http_message_body_stream(body
);
163 char *buf
= emalloc(0x1000);
165 php_stream_seek(s
, offset
, SEEK_SET
);
170 while (!php_stream_eof(s
)) {
171 size_t read
= php_stream_read(s
, buf
, MIN(forlen
, 0x1000));
174 cb(cb_arg
, buf
, read
);
177 if (read
< MIN(forlen
, sizeof(buf
))) {
181 if (forlen
&& !(forlen
-= read
)) {
188 PHP_HTTP_API
size_t php_http_message_body_append(php_http_message_body_t
*body
, const char *buf
, size_t len
)
191 TSRMLS_FETCH_FROM_CTX(body
->ts
);
193 s
= php_http_message_body_stream(body
);
194 php_stream_seek(s
, 0, SEEK_END
);
195 return php_stream_write(s
, buf
, len
);
198 PHP_HTTP_API
size_t php_http_message_body_appendf(php_http_message_body_t
*body
, const char *fmt
, ...)
205 print_len
= vspprintf(&print_str
, 0, fmt
, argv
);
208 print_len
= php_http_message_body_append(body
, print_str
, print_len
);
214 PHP_HTTP_API STATUS
php_http_message_body_add(php_http_message_body_t
*body
, HashTable
*fields
, HashTable
*files
)
219 INIT_PZVAL_ARRAY(&tmp
, fields
);
220 if (SUCCESS
!= add_recursive_fields(body
, NULL
, &tmp
)) {
225 INIT_PZVAL_ARRAY(&tmp
, files
);
226 if (SUCCESS
!= add_recursive_files(body
, NULL
, &tmp
)) {
235 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
)
238 TSRMLS_FETCH_FROM_CTX(body
->ts
);
240 safe_name
= php_addslashes(estrdup(name
), strlen(name
), NULL
, 1 TSRMLS_CC
);
243 php_http_message_body_appendf(
245 "Content-Disposition: form-data; name=\"%s\"" PHP_HTTP_CRLF
248 php_http_message_body_append(body
, value_str
, value_len
);
249 BOUNDARY_CLOSE(body
);
255 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
)
258 char *path_dup
= estrdup(path
);
259 TSRMLS_FETCH_FROM_CTX(body
->ts
);
261 if ((in
= php_stream_open_wrapper(path_dup
, "r", REPORT_ERRORS
|USE_PATH
|STREAM_MUST_SEEK
, NULL
))) {
262 php_stream_statbuf ssb
= {{0}};
264 if (SUCCESS
== php_stream_stat(in
, &ssb
) && S_ISREG(ssb
.sb
.st_mode
)) {
265 char *safe_name
= php_addslashes(estrdup(name
), strlen(name
), NULL
, 1 TSRMLS_CC
);
268 php_http_message_body_appendf(
270 "Content-Disposition: attachment; name=\"%s\"; filename=\"%s\"" PHP_HTTP_CRLF
271 "Content-Type: %s" PHP_HTTP_CRLF
272 "Content-Length: %zu" PHP_HTTP_CRLF
274 safe_name
, basename(path_dup
), ctype
, ssb
.sb
.st_size
);
275 php_stream_copy_to_stream_ex(in
, php_http_message_body_stream(body
), PHP_STREAM_COPY_ALL
, NULL
);
276 BOUNDARY_CLOSE(body
);
280 php_stream_close(in
);
284 php_stream_close(in
);
285 php_http_error(HE_WARNING
, PHP_HTTP_E_MESSAGE_BODY
, "Not a valid regular file: %s", path
);
295 static inline char *format_key(uint type
, char *str
, ulong num
, const char *prefix
) {
296 char *new_key
= NULL
;
298 if (prefix
&& *prefix
) {
299 if (type
== HASH_KEY_IS_STRING
) {
300 spprintf(&new_key
, 0, "%s[%s]", prefix
, str
);
302 spprintf(&new_key
, 0, "%s[%lu]", prefix
, num
);
304 } else if (type
== HASH_KEY_IS_STRING
) {
305 new_key
= estrdup(str
);
307 spprintf(&new_key
, 0, "%lu", num
);
313 static STATUS
add_recursive_fields(php_http_message_body_t
*body
, const char *name
, zval
*value
)
315 if (Z_TYPE_P(value
) == IS_ARRAY
|| Z_TYPE_P(value
) == IS_OBJECT
) {
318 php_http_array_hashkey_t key
= php_http_array_hashkey_init(0);
319 TSRMLS_FETCH_FROM_CTX(body
->ts
);
321 if (!HASH_OF(value
)->nApplyCount
) {
322 ++HASH_OF(value
)->nApplyCount
;
323 FOREACH_KEYVAL(pos
, value
, key
, val
) {
324 char *str
= format_key(key
.type
, key
.str
, key
.num
, name
);
325 if (SUCCESS
!= add_recursive_fields(body
, str
, *val
)) {
327 HASH_OF(value
)->nApplyCount
--;
332 --HASH_OF(value
)->nApplyCount
;
335 zval
*cpy
= php_http_ztyp(IS_STRING
, value
);
336 php_http_message_body_add_field(body
, name
, Z_STRVAL_P(cpy
), Z_STRLEN_P(cpy
));
343 static STATUS
add_recursive_files(php_http_message_body_t
*body
, const char *name
, zval
*value
)
345 if (Z_TYPE_P(value
) == IS_ARRAY
|| Z_TYPE_P(value
) == IS_OBJECT
) {
346 zval
**zfile
, **zname
, **ztype
;
347 TSRMLS_FETCH_FROM_CTX(body
->ts
);
349 if ((SUCCESS
== zend_hash_find(HASH_OF(value
), ZEND_STRS("name"), (void *) &zname
))
350 && (SUCCESS
== zend_hash_find(HASH_OF(value
), ZEND_STRS("file"), (void *) &zfile
))
351 && (SUCCESS
== zend_hash_find(HASH_OF(value
), ZEND_STRS("type"), (void *) &ztype
))
353 zval
*zfc
= php_http_ztyp(IS_STRING
, *zfile
), *znc
= php_http_ztyp(IS_STRING
, *zname
), *ztc
= php_http_ztyp(IS_STRING
, *ztype
);
354 char *str
= format_key(HASH_KEY_IS_STRING
, Z_STRVAL_P(znc
), 0, name
);
355 STATUS ret
= php_http_message_body_add_file(body
, str
, Z_STRVAL_P(zfc
), Z_STRVAL_P(ztc
));
362 if (ret
!= SUCCESS
) {
368 php_http_array_hashkey_t key
= php_http_array_hashkey_init(0);
370 if (!HASH_OF(value
)->nApplyCount
) {
371 ++HASH_OF(value
)->nApplyCount
;
372 FOREACH_KEYVAL(pos
, value
, key
, val
) {
373 char *str
= format_key(key
.type
, key
.str
, key
.num
, name
);
374 if (SUCCESS
!= add_recursive_files(body
, str
, *val
)) {
376 --HASH_OF(value
)->nApplyCount
;
381 --HASH_OF(value
)->nApplyCount
;
385 TSRMLS_FETCH_FROM_CTX(body
->ts
);
386 php_http_error(HE_WARNING
, PHP_HTTP_E_MESSAGE_BODY
, "Unrecognized array format for message body file to add");
395 #define PHP_HTTP_BEGIN_ARGS(method, req_args) PHP_HTTP_BEGIN_ARGS_EX(HttpMessageBody, method, 0, req_args)
396 #define PHP_HTTP_EMPTY_ARGS(method) PHP_HTTP_EMPTY_ARGS_EX(HttpMessageBody, method, 0)
397 #define PHP_HTTP_MESSAGE_BODY_ME(method, visibility) PHP_ME(HttpMessageBody, method, PHP_HTTP_ARGS(HttpMessageBody, method), visibility)
399 PHP_HTTP_BEGIN_ARGS(__construct
, 0)
400 PHP_HTTP_ARG_VAL(stream
, 0)
403 PHP_HTTP_EMPTY_ARGS(__toString
);
405 PHP_HTTP_BEGIN_ARGS(toStream
, 1)
406 PHP_HTTP_ARG_VAL(stream
, 0)
409 PHP_HTTP_BEGIN_ARGS(toCallback
, 1)
410 PHP_HTTP_ARG_VAL(callback
, 0)
413 PHP_HTTP_BEGIN_ARGS(append
, 1)
414 PHP_HTTP_ARG_VAL(string
, 0)
417 PHP_HTTP_BEGIN_ARGS(add
, 0)
418 PHP_HTTP_ARG_VAL(fields
, 0)
419 PHP_HTTP_ARG_VAL(files
, 0)
423 zend_class_entry
*php_http_message_body_class_entry
;
424 zend_function_entry php_http_message_body_method_entry
[] = {
425 PHP_HTTP_MESSAGE_BODY_ME(__construct
, ZEND_ACC_PUBLIC
|ZEND_ACC_CTOR
)
426 PHP_HTTP_MESSAGE_BODY_ME(__toString
, ZEND_ACC_PUBLIC
)
427 PHP_MALIAS(HttpMessageBody
, toString
, __toString
, args_for_HttpMessageBody___toString
, ZEND_ACC_PUBLIC
)
428 PHP_HTTP_MESSAGE_BODY_ME(toStream
, ZEND_ACC_PUBLIC
)
429 PHP_HTTP_MESSAGE_BODY_ME(toCallback
, ZEND_ACC_PUBLIC
)
430 PHP_HTTP_MESSAGE_BODY_ME(append
, ZEND_ACC_PUBLIC
)
431 PHP_HTTP_MESSAGE_BODY_ME(add
, ZEND_ACC_PUBLIC
)
434 static zend_object_handlers php_http_message_body_object_handlers
;
436 PHP_MINIT_FUNCTION(http_message_body
)
438 PHP_HTTP_REGISTER_CLASS(http
\\Message
, Body
, http_message_body
, php_http_object_class_entry
, 0);
439 php_http_message_body_class_entry
->create_object
= php_http_message_body_object_new
;
440 memcpy(&php_http_message_body_object_handlers
, zend_get_std_object_handlers(), sizeof(zend_object_handlers
));
441 php_http_message_body_object_handlers
.clone_obj
= php_http_message_body_object_clone
;
446 zend_object_value
php_http_message_body_object_new(zend_class_entry
*ce TSRMLS_DC
)
448 return php_http_message_body_object_new_ex(ce
, NULL
, NULL TSRMLS_CC
);
451 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
)
453 zend_object_value ov
;
454 php_http_message_body_object_t
*o
;
456 o
= ecalloc(1, sizeof(php_http_message_body_object_t
));
457 zend_object_std_init((zend_object
*) o
, php_http_message_body_class_entry TSRMLS_CC
);
458 object_properties_init((zend_object
*) o
, ce
);
468 ov
.handle
= zend_objects_store_put((zend_object
*) o
, NULL
, php_http_message_body_object_free
, NULL TSRMLS_CC
);
469 ov
.handlers
= &php_http_message_body_object_handlers
;
474 zend_object_value
php_http_message_body_object_clone(zval
*object TSRMLS_DC
)
476 zend_object_value new_ov
;
477 php_http_message_body_object_t
*new_obj
= NULL
;
478 php_http_message_body_object_t
*old_obj
= zend_object_store_get_object(object TSRMLS_CC
);
480 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
);
481 zend_objects_clone_members(&new_obj
->zo
, new_ov
, &old_obj
->zo
, Z_OBJ_HANDLE_P(object
) TSRMLS_CC
);
486 void php_http_message_body_object_free(void *object TSRMLS_DC
)
488 php_http_message_body_object_t
*obj
= object
;
490 php_http_message_body_free(&obj
->body
);
492 zend_object_std_dtor((zend_object
*) obj TSRMLS_CC
);
496 PHP_METHOD(HttpMessageBody
, __construct
)
498 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
499 zval
*zstream
= NULL
;
502 with_error_handling(EH_THROW
, php_http_exception_class_entry
) {
503 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|r!", &zstream
)) {
505 php_stream_from_zval(stream
, &zstream
);
509 php_http_message_body_dtor(obj
->body
);
511 obj
->body
= php_http_message_body_init(obj
->body
, stream TSRMLS_CC
);
515 obj
->body
= php_http_message_body_init(NULL
, NULL TSRMLS_CC
);
518 } end_error_handling();
521 PHP_METHOD(HttpMessageBody
, __toString
)
523 if (SUCCESS
== zend_parse_parameters_none()) {
524 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
528 php_http_message_body_to_string(obj
->body
, &str
, &len
, 0, 0);
530 RETURN_STRINGL(str
, len
, 0);
533 RETURN_EMPTY_STRING();
536 PHP_METHOD(HttpMessageBody
, toStream
)
539 long offset
= 0, forlen
= 0;
541 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "r|ll", &zstream
, &offset
, &forlen
)) {
543 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
545 php_stream_from_zval(stream
, &zstream
);
546 php_http_message_body_to_stream(obj
->body
, stream
, offset
, forlen
);
554 zend_fcall_info
*fci
;
555 zend_fcall_info_cache
*fcc
;
561 static size_t pass(void *cb_arg
, const char *str
, size_t len
)
563 struct fcd
*fcd
= cb_arg
;
565 TSRMLS_FETCH_FROM_CTX(fcd
->ts
);
567 MAKE_STD_ZVAL(zdata
);
568 ZVAL_STRINGL(zdata
, str
, len
, 1);
569 if (SUCCESS
== zend_fcall_info_argn(fcd
->fci TSRMLS_CC
, 2, fcd
->fcz
, zdata
)) {
570 zend_fcall_info_call(fcd
->fci
, fcd
->fcc
, NULL
, NULL TSRMLS_CC
);
571 zend_fcall_info_args_clear(fcd
->fci
, 0);
573 zval_ptr_dtor(&zdata
);
577 PHP_METHOD(HttpMessageBody
, toCallback
)
580 long offset
= 0, forlen
= 0;
582 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "f|ll", &fcd
.fci
, &fcd
.fcc
, &offset
, &forlen
)) {
583 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
586 php_http_message_body_to_callback(obj
->body
, pass
, &fcd
, offset
, forlen
);
587 zval_ptr_dtor(&fcd
.fcz
);
593 PHP_METHOD(HttpMessageBody
, append
)
598 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "s", &str
, &len
)) {
599 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
601 RETURN_LONG(php_http_message_body_append(obj
->body
, str
, len
));
606 PHP_METHOD(HttpMessageBody
, add
)
608 HashTable
*fields
= NULL
, *files
= NULL
;
610 if (SUCCESS
== zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC
, "|hh", &fields
, &files
)) {
611 php_http_message_body_object_t
*obj
= zend_object_store_get_object(getThis() TSRMLS_CC
);
613 RETURN_SUCCESS(php_http_message_body_add(obj
->body
, fields
, files
));
622 * vim600: noet sw=4 ts=4 fdm=marker
623 * vim<600: noet sw=4 ts=4