src/http_header.c

   1 #include "first.h"
   2
   3 #include "http_header.h"
   4 #include "base.h"
   5 #include "array.h"
   6 #include "buffer.h"
   7
   8
   9 typedef struct keyvlenvalue {
  10     const int key;
  11     const unsigned int vlen;
  12     const char * const value;
  13 } keyvlenvalue;
  14
  15 /* Note: must be sorted by length */
  16 /* Note: must be kept in sync with http_header.h enum http_header_e */
  17 #define CONST_LEN_STR(x) (unsigned int)(sizeof(x)-1), (x)
  18 static const keyvlenvalue http_headers[] = {
  19   { HTTP_HEADER_HOST,                 CONST_LEN_STR("Host") }
  20  ,{ HTTP_HEADER_DATE,                 CONST_LEN_STR("Date") }
  21  ,{ HTTP_HEADER_ETAG,                 CONST_LEN_STR("ETag") }
  22  ,{ HTTP_HEADER_VARY,                 CONST_LEN_STR("Vary") }
  23  ,{ HTTP_HEADER_RANGE,                CONST_LEN_STR("Range") }
  24  ,{ HTTP_HEADER_COOKIE,               CONST_LEN_STR("Cookie") }
  25  ,{ HTTP_HEADER_EXPECT,               CONST_LEN_STR("Expect") }
  26  ,{ HTTP_HEADER_STATUS,               CONST_LEN_STR("Status") }
  27  ,{ HTTP_HEADER_SERVER,               CONST_LEN_STR("Server") }
  28  ,{ HTTP_HEADER_UPGRADE,              CONST_LEN_STR("Upgrade") }
  29  ,{ HTTP_HEADER_LOCATION,             CONST_LEN_STR("Location") }
  30  ,{ HTTP_HEADER_FORWARDED,            CONST_LEN_STR("Forwarded") }
  31  ,{ HTTP_HEADER_CONNECTION,           CONST_LEN_STR("Connection") }
  32  ,{ HTTP_HEADER_SET_COOKIE,           CONST_LEN_STR("Set-Cookie") }
  33  ,{ HTTP_HEADER_CONTENT_TYPE,         CONST_LEN_STR("Content-Type") }
  34  ,{ HTTP_HEADER_LAST_MODIFIED,        CONST_LEN_STR("Last-Modified") }
  35  ,{ HTTP_HEADER_AUTHORIZATION,        CONST_LEN_STR("Authorization") }
  36  ,{ HTTP_HEADER_IF_NONE_MATCH,        CONST_LEN_STR("If-None-Match") }
  37  ,{ HTTP_HEADER_CACHE_CONTROL,        CONST_LEN_STR("Cache-Control") }
  38  ,{ HTTP_HEADER_CONTENT_LENGTH,       CONST_LEN_STR("Content-Length") }
  39  ,{ HTTP_HEADER_ACCEPT_ENCODING,      CONST_LEN_STR("Accept-Encoding") }
  40  ,{ HTTP_HEADER_X_FORWARDED_FOR,      CONST_LEN_STR("X-Forwarded-For") }
  41  ,{ HTTP_HEADER_CONTENT_ENCODING,     CONST_LEN_STR("Content-Encoding") }
  42  ,{ HTTP_HEADER_CONTENT_LOCATION,     CONST_LEN_STR("Content-Location") }
  43  ,{ HTTP_HEADER_IF_MODIFIED_SINCE,    CONST_LEN_STR("If-Modified-Since") }
  44  ,{ HTTP_HEADER_TRANSFER_ENCODING,    CONST_LEN_STR("Transfer-Encoding") }
  45  ,{ HTTP_HEADER_X_FORWARDED_PROTO,    CONST_LEN_STR("X-Forwarded-Proto") }
  46  ,{ HTTP_HEADER_OTHER, 0, NULL }
  47 };
  48
  49 enum http_header_e http_header_hkey_get(const char *s, size_t slen) {
  50     const struct keyvlenvalue * const kv = http_headers;
  51     for (int i = 0; kv[i].vlen && slen >= kv[i].vlen; ++i) {
  52         if (slen == kv[i].vlen
  53             && buffer_eq_icase_ssn(s, kv[i].value, slen))
  54             return (enum http_header_e)kv[i].key;
  55     }
  56     return HTTP_HEADER_OTHER;
  57 }
  58
  59
  60 buffer * http_header_response_get(connection *con, enum http_header_e id, const char *k, size_t klen) {
  61     data_string * const ds =
  62       (id <= HTTP_HEADER_OTHER || (con->response.htags & id))
  63       ? (data_string *)array_get_element_klen(con->response.headers, k, klen)
  64       : NULL;
  65     return ds && !buffer_string_is_empty(ds->value) ? ds->value : NULL;
  66 }
  67
  68 void http_header_response_unset(connection *con, enum http_header_e id, const char *k, size_t klen) {
  69     if (id <= HTTP_HEADER_OTHER || (con->response.htags & id)) {
  70         if (id > HTTP_HEADER_OTHER) con->response.htags &= ~id;
  71         array_set_key_value(con->response.headers, k, klen, CONST_STR_LEN(""));
  72     }
  73 }
  74
  75 void http_header_response_set(connection *con, enum http_header_e id, const char *k, size_t klen, const char *v, size_t vlen) {
  76     /* set value, including setting blank value if 0 == vlen
  77      * (note: if 0 == vlen, header is still inserted with blank value,
  78      *  which is used to indicate a "removed" header)
  79      */
  80     if (id > HTTP_HEADER_OTHER)
  81         (vlen) ? (con->response.htags |= id) : (con->response.htags &= ~id);
  82     array_set_key_value(con->response.headers, k, klen, v, vlen);
  83 }
  84
  85 void http_header_response_append(connection *con, enum http_header_e id, const char *k, size_t klen, const char *v, size_t vlen) {
  86     if (vlen) {
  87         data_string *ds= (id <= HTTP_HEADER_OTHER || (con->response.htags & id))
  88           ? (data_string *)array_get_element_klen(con->response.headers,k,klen)
  89           : NULL;
  90         if (id > HTTP_HEADER_OTHER) con->response.htags |= id;
  91         if (NULL == ds) {
  92             array_insert_key_value(con->response.headers, k, klen, v, vlen);
  93         }
  94         else { /* append value */
  95             buffer *vb = ds->value;
  96             if (!buffer_string_is_empty(vb))
  97                 buffer_append_string_len(vb, CONST_STR_LEN(", "));
  98             buffer_append_string_len(vb, v, vlen);
  99         }
 100     }
 101 }
 102
 103 void http_header_response_insert(connection *con, enum http_header_e id, const char *k, size_t klen, const char *v, size_t vlen) {
 104     if (vlen) {
 105         data_string *ds= (id <= HTTP_HEADER_OTHER || (con->response.htags & id))
 106           ? (data_string *)array_get_element_klen(con->response.headers,k,klen)
 107           : NULL;
 108         if (id > HTTP_HEADER_OTHER) con->response.htags |= id;
 109         if (NULL == ds) {
 110             array_insert_key_value(con->response.headers, k, klen, v, vlen);
 111         }
 112         else { /* append value */
 113             buffer *vb = ds->value;
 114             if (!buffer_string_is_empty(vb)) {
 115                 buffer_append_string_len(vb, CONST_STR_LEN("\r\n"));
 116                 buffer_append_string_len(vb, k, klen);
 117                 buffer_append_string_len(vb, CONST_STR_LEN(": "));
 118             }
 119             buffer_append_string_len(vb, v, vlen);
 120         }
 121     }
 122 }
 123
 124
 125 buffer * http_header_request_get(connection *con, enum http_header_e id, const char *k, size_t klen) {
 126     data_string * const ds =
 127       (id <= HTTP_HEADER_OTHER || (con->request.htags & id))
 128       ? (data_string *)array_get_element_klen(con->request.headers, k, klen)
 129       : NULL;
 130     return ds && !buffer_string_is_empty(ds->value) ? ds->value : NULL;
 131 }
 132
 133 void http_header_request_unset(connection *con, enum http_header_e id, const char *k, size_t klen) {
 134     if (id <= HTTP_HEADER_OTHER || (con->request.htags & id)) {
 135         if (id > HTTP_HEADER_OTHER) con->request.htags &= ~id;
 136         array_set_key_value(con->request.headers, k, klen, CONST_STR_LEN(""));
 137     }
 138 }
 139
 140 void http_header_request_set(connection *con, enum http_header_e id, const char *k, size_t klen, const char *v, size_t vlen) {
 141     /* set value, including setting blank value if 0 == vlen
 142      * (note: if 0 == vlen, header is still inserted with blank value,
 143      *  which is used to indicate a "removed" header)
 144      */
 145     if (id > HTTP_HEADER_OTHER)
 146         (vlen) ? (con->request.htags |= id) : (con->request.htags &= ~id);
 147     array_set_key_value(con->request.headers, k, klen, v, vlen);
 148 }
 149
 150 void http_header_request_append(connection *con, enum http_header_e id, const char *k, size_t klen, const char *v, size_t vlen) {
 151     if (vlen) {
 152         data_string *ds = (id <= HTTP_HEADER_OTHER || (con->request.htags & id))
 153           ? (data_string *)array_get_element_klen(con->request.headers, k, klen)
 154           : NULL;
 155         if (id > HTTP_HEADER_OTHER) con->request.htags |= id;
 156         if (NULL == ds) {
 157             array_insert_key_value(con->request.headers, k, klen, v, vlen);
 158         }
 159         else { /* append value */
 160             buffer *vb = ds->value;
 161             if (!buffer_string_is_empty(vb))
 162                 buffer_append_string_len(vb, CONST_STR_LEN(", "));
 163             buffer_append_string_len(vb, v, vlen);
 164         }
 165     }
 166 }
 167
 168
 169 buffer * http_header_env_get(connection *con, const char *k, size_t klen) {
 170     data_string * const ds =
 171       (data_string *)array_get_element_klen(con->environment, k, klen);
 172     return ds && !buffer_string_is_empty(ds->value) ? ds->value : NULL;
 173 }
 174
 175 void http_header_env_set(connection *con, const char *k, size_t klen, const char *v, size_t vlen) {
 176     array_set_key_value(con->environment, k, klen, v, vlen);
 177 }
 178
 179 void http_header_env_append(connection *con, const char *k, size_t klen, const char *v, size_t vlen) {
 180     /*if (vlen)*/ /* skip check; permit env var w/ blank value to be appended */
 181     {
 182         buffer * const vb = http_header_env_get(con, k, klen);
 183         if (NULL == vb) {
 184             array_insert_key_value(con->environment, k, klen, v, vlen);
 185         }
 186         else if (vlen) { /* append value */
 187             buffer_append_string_len(vb, CONST_STR_LEN(", "));
 188             buffer_append_string_len(vb, v, vlen);
 189         }
 190     }
 191 }