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_CC)
  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-%p: %s (%d) %.*s…\n", parser,
  93                                 php_http_header_parser_state_is(parser)<0?"FAILURE":state[php_http_header_parser_state_is(parser)],
  94                                                 zend_hash_num_elements(headers), MIN(16,buffer->used), buffer->data);
  95 #endif
  96                 switch (php_http_header_parser_state_pop(parser)) {
  97                         case PHP_HTTP_HEADER_PARSER_STATE_FAILURE:
  98                                 return php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_FAILURE);
  99
 100                         case PHP_HTTP_HEADER_PARSER_STATE_START: {
 101                                 char *ptr = buffer->data;
 102
 103                                 while (ptr - buffer->data < buffer->used && PHP_HTTP_IS_CTYPE(space, *ptr)) {
 104                                         ++ptr;
 105                                 }
 106
 107                                 php_http_buffer_cut(buffer, 0, ptr - buffer->data);
 108                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_KEY);
 109                                 break;
 110                         }
 111
 112                         case PHP_HTTP_HEADER_PARSER_STATE_KEY: {
 113                                 const char *colon, *eol_str;
 114                                 int eol_len;
 115
 116                                 if (buffer->data == (eol_str = php_http_locate_bin_eol(buffer->data, buffer->used, &eol_len))) {
 117                                         /* end of headers */
 118                                         php_http_buffer_cut(buffer, 0, eol_len);
 119                                         php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_DONE);
 120                                 } else if (php_http_info_parse(&parser->info, php_http_buffer_fix(buffer)->data TSRMLS_CC)) {
 121                                         /* new message starting with request/response line */
 122                                         if (callback_func) {
 123                                                 callback_func(callback_arg, &headers, &parser->info TSRMLS_CC);
 124                                         }
 125                                         php_http_info_dtor(&parser->info);
 126                                         php_http_buffer_cut(buffer, 0, eol_str + eol_len - buffer->data);
 127                                         php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE);
 128                                 } else if ((colon = memchr(buffer->data, ':', buffer->used)) && (!eol_str || eol_str > colon)) {
 129                                         /* header: string */
 130                                         parser->_key.str = estrndup(buffer->data, parser->_key.len = colon - buffer->data);
 131                                         while (PHP_HTTP_IS_CTYPE(space, *++colon));
 132                                         php_http_buffer_cut(buffer, 0, colon - buffer->data);
 133                                         php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_VALUE);
 134                                 } else {
 135                                         /* neither reqeust/response line nor header: string */
 136                                         return php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_FAILURE);
 137                                 }
 138                                 break;
 139                         }
 140
 141                         case PHP_HTTP_HEADER_PARSER_STATE_VALUE: {
 142                                 const char *eol_str;
 143                                 int eol_len;
 144
 145                                 do {
 146                                         if ((eol_str = php_http_locate_bin_eol(buffer->data, buffer->used, &eol_len))) {
 147                                                 if (eol_str + eol_len - buffer->data < buffer->used) {
 148                                                         char nextline = *(eol_str + eol_len);
 149
 150                                                         if (nextline == '\t' || nextline == ' ') {
 151                                                                 php_http_buffer_cut(buffer, eol_str - buffer->data, eol_len);
 152                                                                 continue;
 153                                                         }
 154                                                 }
 155
 156                                                 parser->_val.str = estrndup(buffer->data, parser->_val.len = eol_str - buffer->data);
 157                                                 php_http_buffer_cut(buffer, 0, eol_str + eol_len - buffer->data);
 158                                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE);
 159                                         } else if (flags & PHP_HTTP_HEADER_PARSER_CLEANUP) {
 160                                                 if (buffer->used) {
 161                                                         parser->_val.str = estrndup(buffer->data, parser->_val.len = buffer->used);
 162                                                         php_http_buffer_reset(buffer);
 163                                                 }
 164                                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE);
 165                                         } else {
 166                                                 return php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_VALUE);
 167                                         }
 168                                 } while (0);
 169
 170                                 break;
 171                         }
 172
 173                         case PHP_HTTP_HEADER_PARSER_STATE_HEADER_DONE:
 174                                 if (parser->_key.str && parser->_val.str) {
 175                                         zval array, **exist;
 176
 177                                         INIT_PZVAL_ARRAY(&array, headers);
 178                                         php_http_pretty_key(parser->_key.str, parser->_key.len, 1, 1);
 179                                         if (SUCCESS == zend_hash_find(headers, parser->_key.str, parser->_key.len + 1, (void *) &exist)) {
 180                                                 convert_to_array(*exist);
 181                                                 add_next_index_stringl(&array, parser->_val.str, parser->_val.len, 0);
 182                                         } else {
 183                                                 add_assoc_stringl_ex(&array, parser->_key.str, parser->_key.len + 1, parser->_val.str, parser->_val.len, 0);
 184                                         }
 185                                         parser->_val.str = NULL;
 186                                 }
 187
 188                                 STR_SET(parser->_key.str, NULL);
 189                                 STR_SET(parser->_val.str, NULL);
 190
 191                                 php_http_header_parser_state_push(parser, 1, PHP_HTTP_HEADER_PARSER_STATE_KEY);
 192                                 break;
 193
 194                         case PHP_HTTP_HEADER_PARSER_STATE_DONE:
 195                                 return PHP_HTTP_HEADER_PARSER_STATE_DONE;
 196                 }
 197         }
 198
 199         return php_http_header_parser_state_is(parser);
 200 }