git.m6w6.name Git - m6w6/ext-http/blob - php_http_header_parser.c

   1
   2 #include "php_http.h"
   3
   4 typedef struct php_http_header_parser_state_spec {
   5         php_http_header_parser_state_t state;
   6         unsigned need_data:1;
   7 } php_http_header_parser_state_spec_t;
   8
   9 static const php_http_header_parser_state_spec_t php_http_header_parser_states[] = {
  10                 {PHP_HTTP_HEADER_PARSER_STATE_START,            1},
  11                 {PHP_HTTP_HEADER_PARSER_STATE_KEY,                      1},
  12                 {PHP_HTTP_HEADER_PARSER_STATE_VALUE,            1},
  13                 {PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE,      0},
  14                 {PHP_HTTP_HEADER_PARSER_STATE_DONE,                     0}
  15 };
  16
  17
  18 PHP_HTTP_API php_http_header_parser_t *php_http_header_parser_init(php_http_header_parser_t *parser TSRMLS_DC)
  19 {
  20         if (!parser) {
  21                 parser = emalloc(sizeof(*parser));
  22         }
  23         memset(parser, 0, sizeof(*parser));
  24
  25         TSRMLS_SET_CTX(parser->ts);
  26
  27         return parser;
  28 }
  29
  30
  31 PHP_HTTP_API php_http_header_parser_state_t php_http_header_parser_state_push(php_http_header_parser_t *parser, unsigned argc, ...)
  32 {
  33         va_list va_args;
  34         unsigned i;
  35         php_http_header_parser_state_t state = 0;
  36
  37         va_start(va_args, argc);
  38         for (i = 0; i < argc; ++i) {
  39                 state = va_arg(va_args, php_http_header_parser_state_t);
  40                 zend_stack_push(&parser->stack, &state, sizeof(state));
  41         }
  42         va_end(va_args);
  43
  44         return state;
  45 }
  46
  47 PHP_HTTP_API php_http_header_parser_state_t php_http_header_parser_state_is(php_http_header_parser_t *parser)
  48 {
  49         php_http_header_parser_state_t *state;
  50
  51         if (SUCCESS == zend_stack_top(&parser->stack, (void *) &state)) {
  52                 return *state;
  53         }
  54         return PHP_HTTP_HEADER_PARSER_STATE_START;
  55 }
  56
  57 PHP_HTTP_API php_http_header_parser_state_t php_http_header_parser_state_pop(php_http_header_parser_t *parser)
  58 {
  59         php_http_header_parser_state_t state, *state_ptr;
  60         if (SUCCESS == zend_stack_top(&parser->stack, (void *) &state_ptr)) {
  61                 state = *state_ptr;
  62                 zend_stack_del_top(&parser->stack);
  63                 return state;
  64         }
  65         return PHP_HTTP_HEADER_PARSER_STATE_START;
  66 }
  67
  68 PHP_HTTP_API void php_http_header_parser_dtor(php_http_header_parser_t *parser)
  69 {
  70         zend_stack_destroy(&parser->stack);
  71         php_http_info_dtor(&parser->info);
  72         STR_FREE(parser->_key.str);
  73         STR_FREE(parser->_val.str);
  74 }
  75
  76 PHP_HTTP_API void php_http_header_parser_free(php_http_header_parser_t **parser)
  77 {
  78         if (*parser) {
  79                 php_http_header_parser_dtor(*parser);
  80                 efree(*parser);
  81                 *parser = NULL;
  82         }
  83 }
  84
  85 PHP_HTTP_API STATUS php_http_header_parser_parse(php_http_header_parser_t *parser, php_http_buffer_t *buffer, unsigned flags, HashTable *headers, php_http_info_callback_t callback_func, void *callback_arg)
  86 {
  87         TSRMLS_FETCH_FROM_CTX(parser->ts);
  88
  89         while (buffer->used || !php_http_header_parser_states[php_http_header_parser_state_is(parser)].need_data) {
  90 #if 0
  91                 const char *state[] = {"START", "KEY", "VALUE", "HEADER_DONE", "DONE"};
  92                 fprintf(stderr, "#HP: %s (%d)\n", php_http_header_parser_state_is(parser) < 0 ? "FAILURE" : state[php_http_header_parser_state_is(parser)], zend_hash_num_elements(headers));
  93                 _dpf(0, buffer->data, buffer->used);
  94 #endif
  95                 switch (php_http_header_parser_state_pop(parser)) {
  96                         case PHP_HTTP_HEADER_PARSER_STATE_FAILURE:
  97                                 return php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_FAILURE);
  98
  99                         case PHP_HTTP_HEADER_PARSER_STATE_START: {
 100                                 char *ptr = buffer->data;
 101
 102                                 while (ptr - buffer->data < buffer->used && PHP_HTTP_IS_CTYPE(space, *ptr)) {
 103                                         ++ptr;
 104                                 }
 105
 106                                 php_http_buffer_cut(buffer, 0, ptr - buffer->data);
 107                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_KEY);
 108                                 break;
 109                         }
 110
 111                         case PHP_HTTP_HEADER_PARSER_STATE_KEY: {
 112                                 const char *colon, *eol_str;
 113                                 int eol_len;
 114
 115                                 if (buffer->data == (eol_str = php_http_locate_bin_eol(buffer->data, buffer->used, &eol_len))) {
 116                                         /* end of headers */
 117                                         php_http_buffer_cut(buffer, 0, eol_len);
 118                                         php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_DONE);
 119                                 } else if (php_http_info_parse(&parser->info, php_http_buffer_fix(buffer)->data TSRMLS_CC)) {
 120                                         /* new message starting with request/response line */
 121                                         if (callback_func) {
 122                                                 callback_func(callback_arg, &headers, &parser->info TSRMLS_CC);
 123                                         }
 124                                         php_http_info_dtor(&parser->info);
 125                                         php_http_buffer_cut(buffer, 0, eol_str + eol_len - buffer->data);
 126                                         php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE);
 127                                 } else if ((colon = memchr(buffer->data, ':', buffer->used)) && (!eol_str || eol_str > colon)) {
 128                                         /* header: string */
 129                                         parser->_key.str = estrndup(buffer->data, parser->_key.len = colon - buffer->data);
 130                                         while (PHP_HTTP_IS_CTYPE(space, *++colon));
 131                                         php_http_buffer_cut(buffer, 0, colon - buffer->data);
 132                                         php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_VALUE);
 133                                 } else {
 134                                         /* neither reqeust/response line nor header: string */
 135                                         return php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_FAILURE);
 136                                 }
 137                                 break;
 138                         }
 139
 140                         case PHP_HTTP_HEADER_PARSER_STATE_VALUE: {
 141                                 const char *eol_str;
 142                                 int eol_len;
 143
 144                                 do {
 145                                         if ((eol_str = php_http_locate_bin_eol(buffer->data, buffer->used, &eol_len))) {
 146                                                 if (eol_str + eol_len - buffer->data < buffer->used) {
 147                                                         char nextline = *(eol_str + eol_len);
 148
 149                                                         if (nextline == '\t' || nextline == ' ') {
 150                                                                 php_http_buffer_cut(buffer, eol_str - buffer->data, eol_len);
 151                                                                 continue;
 152                                                         }
 153                                                 }
 154
 155                                                 parser->_val.str = estrndup(buffer->data, parser->_val.len = eol_str - buffer->data);
 156                                                 php_http_buffer_cut(buffer, 0, eol_str + eol_len - buffer->data);
 157                                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE);
 158                                         } else if (flags & PHP_HTTP_HEADER_PARSER_CLEANUP) {
 159                                                 if (buffer->used) {
 160                                                         parser->_val.str = estrndup(buffer->data, parser->_val.len = buffer->used);
 161                                                         php_http_buffer_reset(buffer);
 162                                                 }
 163                                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE);
 164                                         } else {
 165                                                 return php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_VALUE);
 166                                         }
 167                                 } while (0);
 168
 169                                 break;
 170                         }
 171
 172                         case PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE:
 173                                 if (parser->_key.str && parser->_val.str) {
 174                                         zval array, **exist;
 175
 176                                         INIT_PZVAL_ARRAY(&array, headers);
 177                                         php_http_pretty_key(parser->_key.str, parser->_key.len, 1, 1);
 178                                         if (SUCCESS == zend_hash_find(headers, parser->_key.str, parser->_key.len + 1, (void *) &exist)) {
 179                                                 convert_to_array(*exist);
 180                                                 add_next_index_stringl(*exist, parser->_val.str, parser->_val.len, 0);
 181                                         } else {
 182                                                 add_assoc_stringl_ex(&array, parser->_key.str, parser->_key.len + 1, parser->_val.str, parser->_val.len, 0);
 183                                         }
 184                                         parser->_val.str = NULL;
 185                                 }
 186
 187                                 STR_SET(parser->_key.str, NULL);
 188                                 STR_SET(parser->_val.str, NULL);
 189
 190                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_KEY);
 191                                 break;
 192
 193                         case PHP_HTTP_HEADER_PARSER_STATE_DONE:
 194                                 return PHP_HTTP_HEADER_PARSER_STATE_DONE;
 195                 }
 196         }
 197
 198         return php_http_header_parser_state_is(parser);
 199 }