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