ext/unicorn/http11/http11_parser.rl

   1 /**
   2  * Copyright (c) 2005 Zed A. Shaw
   3  * You can redistribute it and/or modify it under the same terms as Ruby.
   4  */
   5 #include "http11_parser.h"
   6 #include <assert.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9
  10 /*
  11  * capitalizes all lower-case ASCII characters,
  12  * converts dashes to underscores.
  13  */
  14 static void snake_upcase_char(char *c)
  15 {
  16   if (*c >= 'a' && *c <= 'z')
  17     *c &= ~0x20;
  18   else if (*c == '-')
  19     *c = '_';
  20 }
  21
  22 static void downcase_char(char *c)
  23 {
  24   if (*c >= 'A' && *c <= 'Z')
  25     *c |= 0x20;
  26 }
  27
  28 #define LEN(AT, FPC) (FPC - buffer - parser->AT)
  29 #define MARK(M,FPC) (parser->M = (FPC) - buffer)
  30 #define PTR_TO(F) (buffer + parser->F)
  31
  32 /** Machine **/
  33
  34 %%{
  35   machine http_parser;
  36
  37   action mark {MARK(mark, fpc); }
  38
  39   action start_field { MARK(field_start, fpc); }
  40   action snake_upcase_field { snake_upcase_char((char *)fpc); }
  41   action downcase_char { downcase_char((char *)fpc); }
  42   action write_field {
  43     parser->field_len = LEN(field_start, fpc);
  44   }
  45
  46   action start_value { MARK(mark, fpc); }
  47   action write_value {
  48     if(parser->http_field != NULL) {
  49       parser->http_field(parser->data, PTR_TO(field_start), parser->field_len, PTR_TO(mark), LEN(mark, fpc));
  50     }
  51   }
  52   action request_method {
  53     if(parser->request_method != NULL)
  54       parser->request_method(parser->data, PTR_TO(mark), LEN(mark, fpc));
  55   }
  56   action request_uri {
  57     if(parser->request_uri != NULL)
  58       parser->request_uri(parser->data, PTR_TO(mark), LEN(mark, fpc));
  59   }
  60   action fragment {
  61     if(parser->fragment != NULL)
  62       parser->fragment(parser->data, PTR_TO(mark), LEN(mark, fpc));
  63   }
  64
  65   action start_query {MARK(query_start, fpc); }
  66   action query_string {
  67     if(parser->query_string != NULL)
  68       parser->query_string(parser->data, PTR_TO(query_start), LEN(query_start, fpc));
  69   }
  70
  71   action http_version {
  72     if(parser->http_version != NULL)
  73       parser->http_version(parser->data, PTR_TO(mark), LEN(mark, fpc));
  74   }
  75
  76   action request_path {
  77     if(parser->request_path != NULL)
  78       parser->request_path(parser->data, PTR_TO(mark), LEN(mark,fpc));
  79   }
  80
  81   action done {
  82     parser->body_start = fpc - buffer + 1;
  83     if(parser->header_done != NULL)
  84       parser->header_done(parser->data, fpc + 1, pe - fpc - 1);
  85     fbreak;
  86   }
  87
  88   include http_parser_common "http11_parser_common.rl";
  89 }%%
  90
  91 /** Data **/
  92 %% write data;
  93
  94 int http_parser_init(http_parser *parser)  {
  95   int cs = 0;
  96   %% write init;
  97   parser->cs = cs;
  98   parser->body_start = 0;
  99   parser->mark = 0;
 100   parser->nread = 0;
 101   parser->field_len = 0;
 102   parser->field_start = 0;
 103
 104   return(1);
 105 }
 106
 107 /** exec **/
 108 size_t http_parser_execute(http_parser *parser, const char *buffer, size_t len)  {
 109   const char *p, *pe;
 110   int cs = parser->cs;
 111   size_t off = parser->nread;
 112
 113   assert(off <= len && "offset past end of buffer");
 114
 115   p = buffer+off;
 116   pe = buffer+len;
 117
 118   assert(*pe == '\0' && "pointer does not end on NUL");
 119   assert(pe - p == len - off && "pointers aren't same distance");
 120
 121   %% write exec;
 122
 123   if (!http_parser_has_error(parser))
 124     parser->cs = cs;
 125   parser->nread += p - (buffer + off);
 126
 127   assert(p <= pe && "buffer overflow after parsing execute");
 128   assert(parser->nread <= len && "nread longer than length");
 129   assert(parser->body_start <= len && "body starts after buffer end");
 130   assert(parser->mark < len && "mark is after buffer end");
 131   assert(parser->field_len <= len && "field has length longer than whole buffer");
 132   assert(parser->field_start < len && "field starts after buffer end");
 133
 134   return(parser->nread);
 135 }
 136
 137 int http_parser_finish(http_parser *parser)
 138 {
 139   if (http_parser_has_error(parser) ) {
 140     return -1;
 141   } else if (http_parser_is_finished(parser) ) {
 142     return 1;
 143   } else {
 144     return 0;
 145   }
 146 }
 147
 148 int http_parser_has_error(http_parser *parser) {
 149   return parser->cs == http_parser_error;
 150 }
 151
 152 int http_parser_is_finished(http_parser *parser) {
 153   return parser->cs == http_parser_first_final;
 154 }