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