git.m6w6.name Git - m6w6/ext-http/blob - src/php_http_header.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-2014, Michael Wallner <mike@php.net>            |
  10     +--------------------------------------------------------------------+
  11 */
  12
  13 #include "php_http_api.h"
  14
  15 ZEND_RESULT_CODE php_http_header_parse(const char *header, size_t length, HashTable *headers, php_http_info_callback_t callback_func, void **callback_data)
  16 {
  17         php_http_header_parser_t ctx;
  18         php_http_buffer_t buf;
  19         php_http_header_parser_state_t rs;
  20
  21         if (!php_http_buffer_from_string_ex(&buf, header, length)) {
  22                 php_error_docref(NULL, E_WARNING, "Could not allocate buffer");
  23                 return FAILURE;
  24         }
  25
  26         if (!php_http_header_parser_init(&ctx)) {
  27                 php_http_buffer_dtor(&buf);
  28                 php_error_docref(NULL, E_WARNING, "Could not initialize header parser");
  29                 return FAILURE;
  30         }
  31
  32         rs = php_http_header_parser_parse(&ctx, &buf, PHP_HTTP_HEADER_PARSER_CLEANUP, headers, callback_func, callback_data);
  33         php_http_header_parser_dtor(&ctx);
  34         php_http_buffer_dtor(&buf);
  35
  36         return rs == PHP_HTTP_HEADER_PARSER_STATE_FAILURE ? FAILURE : SUCCESS;
  37 }
  38
  39 void php_http_header_to_callback(HashTable *headers, zend_bool crlf, php_http_pass_format_callback_t cb, void *cb_arg)
  40 {
  41         php_http_arrkey_t key;
  42         zval *header;
  43
  44         ZEND_HASH_FOREACH_KEY_VAL(headers, key.h, key.key, header)
  45         {
  46                 if (key.key) {
  47                         php_http_header_to_callback_ex(key.key->val, header, crlf, cb, cb_arg);
  48                 }
  49         }
  50         ZEND_HASH_FOREACH_END();
  51 /*
  52 <<<<<<< HEAD
  53         php_http_arrkey_t key;
  54         zval *header, *single_header;
  55
  56         ZEND_HASH_FOREACH_KEY_VAL(headers, key.h, key.key, header)
  57         {
  58                 if (key.key) {
  59                         if (zend_string_equals_literal(key.key, "Set-Cookie") && Z_TYPE_P(header) == IS_ARRAY) {
  60                                 ZEND_HASH_FOREACH_VAL(Z_ARRVAL_P(header), single_header)
  61                                 {
  62                                         if (Z_TYPE_P(single_header) == IS_ARRAY) {
  63                                                 php_http_cookie_list_t *cookie = php_http_cookie_list_from_struct(NULL, single_header);
  64
  65                                                 if (cookie) {
  66                                                         char *buf;
  67                                                         size_t len;
  68
  69                                                         php_http_cookie_list_to_string(cookie, &buf, &len);
  70                                                         cb(cb_arg, crlf ? "Set-Cookie: %s" PHP_HTTP_CRLF : "Set-Cookie: %s", buf);
  71                                                         php_http_cookie_list_free(&cookie);
  72                                                         efree(buf);
  73                                                 }
  74                                         } else {
  75                                                 zend_string *zs = php_http_header_value_to_string(single_header);
  76
  77                                                 cb(cb_arg, crlf ? "Set-Cookie: %s" PHP_HTTP_CRLF : "Set-Cookie: %s", zs->val);
  78                                                 zend_string_release(zs);
  79                                         }
  80                                 }
  81                                 ZEND_HASH_FOREACH_END();
  82                         } else {
  83                                 zend_string *zs = php_http_header_value_to_string(header);
  84
  85                                 cb(cb_arg, crlf ? "%s: %s" PHP_HTTP_CRLF : "%s: %s", key.key->val, zs->val);
  86                                 zend_string_release(zs);
  87                         }
  88 =======
  89 >>>>>>> 343738ad56eb70017704fdac57cf0d74da3d0f2e
  90 */
  91 }
  92
  93 void php_http_header_to_string(php_http_buffer_t *str, HashTable *headers)
  94 {
  95         php_http_header_to_callback(headers, 1, (php_http_pass_format_callback_t) php_http_buffer_appendf, str);
  96 }
  97
  98 void php_http_header_to_callback_ex(const char *key, zval *val, zend_bool crlf, php_http_pass_format_callback_t cb, void *cb_arg)
  99 {
 100         zval *aval;
 101         zend_string *str;
 102
 103         ZVAL_DEREF(val);
 104         switch (Z_TYPE_P(val)) {
 105         case IS_ARRAY:
 106                 ZEND_HASH_FOREACH_VAL(Z_ARRVAL_P(val), aval)
 107                 {
 108                         php_http_header_to_callback_ex(key, aval, crlf, cb, cb_arg);
 109                 }
 110                 ZEND_HASH_FOREACH_END();
 111                 break;
 112
 113         case IS_TRUE:
 114                 cb(cb_arg, "%s: true%s", key, crlf ? PHP_HTTP_CRLF:"");
 115                 break;
 116
 117         case IS_FALSE:
 118                 cb(cb_arg, "%s: false%s", key, crlf ? PHP_HTTP_CRLF:"");
 119                 break;
 120
 121         default:
 122                 str = zval_get_string(val);
 123                 cb(cb_arg, "%s: %s%s", key, str->val, crlf ? PHP_HTTP_CRLF:"");
 124                 zend_string_release(str);
 125                 break;
 126         }
 127 }
 128
 129 void php_http_header_to_string_ex(php_http_buffer_t *str, const char *key, zval *val)
 130 {
 131         php_http_header_to_callback_ex(key, val, 1, (php_http_pass_format_callback_t) php_http_buffer_appendf, str);
 132 }
 133
 134 zend_string *php_http_header_value_array_to_string(zval *header)
 135 {
 136         zval *val;
 137         php_http_buffer_t str;
 138
 139         php_http_buffer_init(&str);
 140         ZEND_HASH_FOREACH_VAL(Z_ARRVAL_P(header), val)
 141         {
 142                 zend_string *zs = php_http_header_value_to_string(val);
 143
 144                 php_http_buffer_appendf(&str, str.used ? ", %s":"%s", zs->val);
 145                 zend_string_release(zs);
 146         }
 147         ZEND_HASH_FOREACH_END();
 148         php_http_buffer_fix(&str);
 149
 150         return php_http_cs2zs(str.data, str.used);
 151 }
 152
 153 zend_string *php_http_header_value_to_string(zval *header)
 154 {
 155         switch (Z_TYPE_P(header)) {
 156         case IS_TRUE:
 157                 return zend_string_init(ZEND_STRL("true"), 0);
 158         case IS_FALSE:
 159                 return zend_string_init(ZEND_STRL("false"), 0);
 160         case IS_ARRAY:
 161                 return php_http_header_value_array_to_string(header);
 162         default:
 163                 return zval_get_string(header);
 164         }
 165 }
 166
 167 static zend_class_entry *php_http_header_class_entry;
 168 zend_class_entry *php_http_header_get_class_entry(void)
 169 {
 170         return php_http_header_class_entry;
 171 }
 172
 173 ZEND_BEGIN_ARG_INFO_EX(ai_HttpHeader___construct, 0, 0, 0)
 174         ZEND_ARG_INFO(0, name)
 175         ZEND_ARG_INFO(0, value)
 176 ZEND_END_ARG_INFO();
 177 PHP_METHOD(HttpHeader, __construct)
 178 {
 179         char *name_str = NULL, *value_str = NULL;
 180         size_t name_len = 0, value_len = 0;
 181
 182         php_http_expect(SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS(), "|s!s!", &name_str, &name_len, &value_str, &value_len), invalid_arg, return);
 183
 184         if (name_str && name_len) {
 185                 char *pretty_str = estrndup(name_str, name_len);
 186                 zend_update_property_stringl(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("name"), php_http_pretty_key(pretty_str, name_len, 1, 1), name_len);
 187                 efree(pretty_str);
 188         }
 189         if (value_str && value_len) {
 190                 zend_update_property_stringl(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("value"), value_str, value_len);
 191         }
 192 }
 193
 194 ZEND_BEGIN_ARG_INFO_EX(ai_HttpHeader_serialize, 0, 0, 0)
 195 ZEND_END_ARG_INFO();
 196 PHP_METHOD(HttpHeader, serialize)
 197 {
 198         if (SUCCESS == zend_parse_parameters_none()) {
 199                 php_http_buffer_t buf;
 200                 zend_string *zs;
 201                 zval name_tmp, value_tmp;
 202
 203                 php_http_buffer_init(&buf);
 204                 zs = zval_get_string(zend_read_property(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("name"), 0, &name_tmp));
 205                 php_http_buffer_appendz(&buf, zs);
 206                 zend_string_release(zs);
 207
 208                 zs = zval_get_string(zend_read_property(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("value"), 0, &value_tmp));
 209                 if (zs->len) {
 210                         php_http_buffer_appends(&buf, ": ");
 211                         php_http_buffer_appendz(&buf, zs);
 212                 } else {
 213                         php_http_buffer_appends(&buf, ":");
 214                 }
 215                 zend_string_release(zs);
 216
 217                 RETURN_STR(php_http_cs2zs(buf.data, buf.used));
 218         }
 219         RETURN_EMPTY_STRING();
 220 }
 221
 222 ZEND_BEGIN_ARG_INFO_EX(ai_HttpHeader_unserialize, 0, 0, 1)
 223         ZEND_ARG_INFO(0, serialized)
 224 ZEND_END_ARG_INFO();
 225 PHP_METHOD(HttpHeader, unserialize)
 226 {
 227         char *serialized_str;
 228         size_t serialized_len;
 229
 230         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS(), "s", &serialized_str, &serialized_len)) {
 231                 HashTable ht;
 232
 233                 zend_hash_init(&ht, 1, NULL, ZVAL_PTR_DTOR, 0);
 234                 if (SUCCESS == php_http_header_parse(serialized_str, serialized_len, &ht, NULL, NULL)) {
 235                         if (zend_hash_num_elements(&ht)) {
 236                                 zend_string *zs, *key;
 237                                 zend_ulong idx;
 238
 239                                 zend_hash_internal_pointer_reset(&ht);
 240                                 switch (zend_hash_get_current_key(&ht, &key, &idx)) {
 241                                         case HASH_KEY_IS_STRING:
 242                                                 zend_update_property_str(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("name"), key);
 243                                                 break;
 244                                         case HASH_KEY_IS_LONG:
 245                                                 zend_update_property_long(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("name"), idx);
 246                                                 break;
 247                                         default:
 248                                                 break;
 249                                 }
 250                                 zs = zval_get_string(zend_hash_get_current_data(&ht));
 251                                 zend_update_property_str(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("value"), zs);
 252                                 zend_string_release(zs);
 253                         }
 254                 }
 255                 zend_hash_destroy(&ht);
 256         }
 257
 258 }
 259
 260 ZEND_BEGIN_ARG_INFO_EX(ai_HttpHeader_match, 0, 0, 1)
 261         ZEND_ARG_INFO(0, value)
 262         ZEND_ARG_INFO(0, flags)
 263 ZEND_END_ARG_INFO();
 264 PHP_METHOD(HttpHeader, match)
 265 {
 266         char *val_str = NULL;
 267         size_t val_len = 0;
 268         zend_long flags = PHP_HTTP_MATCH_LOOSE;
 269         zend_string *zs;
 270         zval value_tmp;
 271
 272         if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "|sl", &val_str, &val_len, &flags)) {
 273                 return;
 274         }
 275
 276         zs = zval_get_string(zend_read_property(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("value"), 0, &value_tmp));
 277         RETVAL_BOOL(php_http_match(zs->val, val_str, flags));
 278         zend_string_release(zs);
 279 }
 280
 281 ZEND_BEGIN_ARG_INFO_EX(ai_HttpHeader_negotiate, 0, 0, 1)
 282         ZEND_ARG_INFO(0, supported)
 283         ZEND_ARG_INFO(1, result)
 284 ZEND_END_ARG_INFO();
 285 PHP_METHOD(HttpHeader, negotiate)
 286 {
 287         HashTable *supported, *rs;
 288         zval name_tmp, value_tmp, *rs_array = NULL;
 289         zend_string *zs;
 290         char *sep_str = NULL;
 291         size_t sep_len = 0;
 292
 293         if (SUCCESS != zend_parse_parameters(ZEND_NUM_ARGS(), "H|z", &supported, &rs_array)) {
 294                 return;
 295         }
 296         if (rs_array) {
 297                 ZVAL_DEREF(rs_array);
 298                 zval_dtor(rs_array);
 299                 array_init(rs_array);
 300         }
 301
 302         zs = zval_get_string(zend_read_property(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("name"), 0, &name_tmp));
 303         if (zend_string_equals_literal(zs, "Accept")) {
 304                 sep_str = "/";
 305                 sep_len = 1;
 306         } else if (zend_string_equals_literal(zs, "Accept-Language")) {
 307                 sep_str = "-";
 308                 sep_len = 1;
 309         }
 310         zend_string_release(zs);
 311
 312         zs = zval_get_string(zend_read_property(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("value"), 0, &value_tmp));
 313         if ((rs = php_http_negotiate(zs->val, zs->len, supported, sep_str, sep_len))) {
 314                 PHP_HTTP_DO_NEGOTIATE_HANDLE_RESULT(rs, supported, rs_array);
 315         } else {
 316                 PHP_HTTP_DO_NEGOTIATE_HANDLE_DEFAULT(supported, rs_array);
 317         }
 318         zend_string_release(zs);
 319 }
 320
 321 ZEND_BEGIN_ARG_INFO_EX(ai_HttpHeader_getParams, 0, 0, 0)
 322         ZEND_ARG_INFO(0, param_sep)
 323         ZEND_ARG_INFO(0, arg_sep)
 324         ZEND_ARG_INFO(0, val_sep)
 325         ZEND_ARG_INFO(0, flags)
 326 ZEND_END_ARG_INFO();
 327 PHP_METHOD(HttpHeader, getParams)
 328 {
 329         zval value_tmp, zctor, zparams_obj, *zargs = NULL;
 330
 331         ZVAL_STRINGL(&zctor, "__construct", lenof("__construct"));
 332
 333         object_init_ex(&zparams_obj, php_http_params_get_class_entry());
 334
 335         zargs = (zval *) ecalloc(ZEND_NUM_ARGS()+1, sizeof(zval));
 336         ZVAL_COPY_VALUE(&zargs[0], zend_read_property(php_http_header_class_entry, Z_OBJ_P(ZEND_THIS), ZEND_STRL("value"), 0, &value_tmp));
 337         if (ZEND_NUM_ARGS()) {
 338                 zend_get_parameters_array(ZEND_NUM_ARGS(), ZEND_NUM_ARGS(), &zargs[1]);
 339         }
 340
 341         if (SUCCESS == call_user_function(NULL, &zparams_obj, &zctor, return_value, ZEND_NUM_ARGS()+1, zargs)) {
 342                 RETVAL_ZVAL(&zparams_obj, 0, 1);
 343         }
 344
 345         zval_ptr_dtor(&zctor);
 346         if (zargs) {
 347                 efree(zargs);
 348         }
 349 }
 350
 351 ZEND_BEGIN_ARG_INFO_EX(ai_HttpHeader_parse, 0, 0, 1)
 352         ZEND_ARG_INFO(0, string)
 353         ZEND_ARG_INFO(0, header_class)
 354 ZEND_END_ARG_INFO();
 355 PHP_METHOD(HttpHeader, parse)
 356 {
 357         char *header_str;
 358         size_t header_len;
 359         zend_class_entry *ce = NULL;
 360
 361         if (SUCCESS == zend_parse_parameters(ZEND_NUM_ARGS(), "s|C", &header_str, &header_len, &ce)) {
 362                 array_init(return_value);
 363
 364                 if (SUCCESS != php_http_header_parse(header_str, header_len, Z_ARRVAL_P(return_value), NULL, NULL)) {
 365                         zval_dtor(return_value);
 366                         RETURN_FALSE;
 367                 } else {
 368                         if (ce && instanceof_function(ce, php_http_header_class_entry)) {
 369                                 php_http_arrkey_t key;
 370                                 zval *val;
 371
 372                                 ZEND_HASH_FOREACH_KEY_VAL(Z_ARRVAL_P(return_value), key.h, key.key, val)
 373                                 {
 374                                         zval zkey, zho;
 375
 376                                         if (key.key) {
 377                                                 ZVAL_STR_COPY(&zkey, key.key);
 378                                         } else {
 379                                                 ZVAL_LONG(&zkey, key.h);
 380                                         }
 381
 382                                         object_init_ex(&zho, ce);
 383                                         Z_TRY_ADDREF_P(val);
 384                                         zend_call_method_with_2_params(Z_OBJ(zho), ce, NULL, "__construct", NULL, &zkey, val);
 385                                         zval_ptr_dtor(val);
 386                                         zval_ptr_dtor(&zkey);
 387
 388                                         if (key.key) {
 389                                                 add_assoc_zval_ex(return_value, key.key->val, key.key->len, &zho);
 390                                         } else {
 391                                                 add_index_zval(return_value, key.h, &zho);
 392                                         }
 393                                 }
 394                                 ZEND_HASH_FOREACH_END();
 395                         }
 396                 }
 397         }
 398 }
 399
 400 static zend_function_entry php_http_header_methods[] = {
 401         PHP_ME(HttpHeader, __construct,   ai_HttpHeader___construct, ZEND_ACC_PUBLIC)
 402         PHP_ME(HttpHeader, serialize,     ai_HttpHeader_serialize, ZEND_ACC_PUBLIC)
 403         ZEND_MALIAS(HttpHeader, __toString, serialize, ai_HttpHeader_serialize, ZEND_ACC_PUBLIC)
 404         ZEND_MALIAS(HttpHeader, toString, serialize, ai_HttpHeader_serialize, ZEND_ACC_PUBLIC)
 405         PHP_ME(HttpHeader, unserialize,   ai_HttpHeader_unserialize, ZEND_ACC_PUBLIC)
 406         PHP_ME(HttpHeader, match,         ai_HttpHeader_match, ZEND_ACC_PUBLIC)
 407         PHP_ME(HttpHeader, negotiate,     ai_HttpHeader_negotiate, ZEND_ACC_PUBLIC)
 408         PHP_ME(HttpHeader, getParams,     ai_HttpHeader_getParams, ZEND_ACC_PUBLIC)
 409         PHP_ME(HttpHeader, parse,         ai_HttpHeader_parse, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
 410         EMPTY_FUNCTION_ENTRY
 411 };
 412
 413 PHP_MINIT_FUNCTION(http_header)
 414 {
 415         zend_class_entry ce = {0};
 416
 417         INIT_NS_CLASS_ENTRY(ce, "http", "Header", php_http_header_methods);
 418         php_http_header_class_entry = zend_register_internal_class(&ce);
 419         zend_class_implements(php_http_header_class_entry, 1, zend_ce_serializable);
 420         zend_declare_class_constant_long(php_http_header_class_entry, ZEND_STRL("MATCH_LOOSE"), PHP_HTTP_MATCH_LOOSE);
 421         zend_declare_class_constant_long(php_http_header_class_entry, ZEND_STRL("MATCH_CASE"), PHP_HTTP_MATCH_CASE);
 422         zend_declare_class_constant_long(php_http_header_class_entry, ZEND_STRL("MATCH_WORD"), PHP_HTTP_MATCH_WORD);
 423         zend_declare_class_constant_long(php_http_header_class_entry, ZEND_STRL("MATCH_FULL"), PHP_HTTP_MATCH_FULL);
 424         zend_declare_class_constant_long(php_http_header_class_entry, ZEND_STRL("MATCH_STRICT"), PHP_HTTP_MATCH_STRICT);
 425         zend_declare_property_null(php_http_header_class_entry, ZEND_STRL("name"), ZEND_ACC_PUBLIC);
 426         zend_declare_property_null(php_http_header_class_entry, ZEND_STRL("value"), ZEND_ACC_PUBLIC);
 427
 428         return SUCCESS;
 429 }
 430
 431 /*
 432  * Local variables:
 433  * tab-width: 4
 434  * c-basic-offset: 4
 435  * End:
 436  * vim600: noet sw=4 ts=4 fdm=marker
 437  * vim<600: noet sw=4 ts=4
 438  */
 439