X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-http;a=blobdiff_plain;f=src%2Fphp_http_message.c;h=56499bb8158d69db9f3e1f14c3d0cd849786fcea;hp=45f584f6094e5fff1c1561cf119660f15dcabe05;hb=b51f6b6c64d28742ef630f430c53954e286b72b1;hpb=cc8acf603d067ea4ad52e676e43f03701728eb9d

diff --git a/src/php_http_message.c b/src/php_http_message.c
index 45f584f..56499bb 100644
--- a/src/php_http_message.c
+++ b/src/php_http_message.c
@@ -70,6 +70,7 @@ php_http_message_t *php_http_message_init_env(php_http_message_t *message, php_h
 			if ((sval = php_http_env_get_server_var(ZEND_STRL("REQUEST_URI"), 1))) {
 				message->http.info.request.url = php_http_url_parse(Z_STRVAL_P(sval), Z_STRLEN_P(sval), ~0);
 			}
+
 			php_http_env_get_request_headers(&message->hdrs);
 			break;
 
@@ -81,11 +82,12 @@ php_http_message_t *php_http_message_init_env(php_http_message_t *message, php_h
 				}
 				message->http.info.response.status = estrdup(php_http_env_get_response_status_for_code(message->http.info.response.code));
 			}
-			
+
 			php_http_env_get_response_headers(&message->hdrs);
 			if (php_output_get_level()) {
 				if (php_output_get_status() & PHP_OUTPUT_SENT) {
 					php_error_docref(NULL, E_WARNING, "Could not fetch response body, output has already been sent at %s:%d", php_output_get_start_filename(), php_output_get_start_lineno());
+
 					goto error;
 				} else if (SUCCESS != php_output_get_contents(&tval)) {
 					php_error_docref(NULL, E_WARNING, "Could not fetch response body");
@@ -409,7 +411,7 @@ void php_http_message_serialize(php_http_message_t *message, char **string, size
 php_http_message_t *php_http_message_reverse(php_http_message_t *msg)
 {
 	size_t i, c = php_http_message_count(msg);
-	
+
 	if (c > 1) {
 		php_http_message_t *tmp = msg, **arr;
 
@@ -452,23 +454,22 @@ php_http_message_t *php_http_message_copy_ex(php_http_message_t *from, php_http_
 {
 	php_http_message_t *temp, *copy = NULL;
 	php_http_info_t info;
-	
+
 	if (from) {
 		info.type = from->type;
 		info.http = from->http;
-		
+
 		copy = temp = php_http_message_init(to, 0, php_http_message_body_copy(from->body, NULL));
 		php_http_message_set_info(temp, &info);
 		zend_hash_copy(&temp->hdrs, &from->hdrs, (copy_ctor_func_t) zval_add_ref);
-	
+
 		if (parents) while (from->parent) {
 			info.type = from->parent->type;
 			info.http = from->parent->http;
-		
+
 			temp->parent = php_http_message_init(NULL, 0, php_http_message_body_copy(from->parent->body, NULL));
 			php_http_message_set_info(temp->parent, &info);
 			array_copy(&from->parent->hdrs, &temp->parent->hdrs);
-		
 
 			temp = temp->parent;
 			from = from->parent;
@@ -944,7 +945,7 @@ static HashTable *php_http_message_object_get_debug_info(zval *object, int *is_t
 
 	PHP_HTTP_MESSAGE_OBJECT_INIT(obj);
 	*is_temp = 0;
-	
+
 #define UPDATE_PROP(name_str, action_with_tmp) \
 	do { \
 		zend_property_info *pi; \
@@ -1246,16 +1247,27 @@ static PHP_METHOD(HttpMessage, addHeader)
 	if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS(), "sz", &name_str, &name_len, &zvalue)) {
 		php_http_message_object_t *obj = PHP_HTTP_OBJ(NULL, getThis());
 		char *name = php_http_pretty_key(estrndup(name_str, name_len), name_len, 1, 1);
-		zval *header;
+		zend_string *hstr, *vstr = php_http_header_value_to_string(zvalue);
+		zval tmp, *header;
 
 		PHP_HTTP_MESSAGE_OBJECT_INIT(obj);
 
-		Z_TRY_ADDREF_P(zvalue);
-		if ((header = php_http_message_header(obj->message, name, name_len))) {
+		if ((name_len != lenof("Set-Cookie") && strcmp(name, "Set-Cookie"))
+		&&	(hstr = php_http_message_header_string(obj->message, name, name_len))) {
+			char *hdr_str;
+			size_t hdr_len = spprintf(&hdr_str, 0, "%s, %s", hstr->val, vstr->val);
+
+			ZVAL_STR(&tmp, php_http_cs2zs(hdr_str, hdr_len));
+			zend_symtable_str_update(&obj->message->hdrs, name, name_len, &tmp);
+			zend_string_release(hstr);
+			zend_string_release(vstr);
+		} else if ((header = php_http_message_header(obj->message, name, name_len))) {
 			convert_to_array(header);
-			zend_hash_next_index_insert(Z_ARRVAL_P(header), zvalue);
+			ZVAL_STR(&tmp, vstr);
+			zend_hash_next_index_insert(Z_ARRVAL_P(header), &tmp);
 		} else {
-			zend_symtable_str_update(&obj->message->hdrs, name, name_len, zvalue);
+			ZVAL_STR(&tmp, vstr);
+			zend_symtable_str_update(&obj->message->hdrs, name, name_len, &tmp);
 		}
 		efree(name);
 	}