Compare commits

..

3 Commits

Author SHA1 Message Date
Jef Roosens 4a0df8db6b
wip 2024-02-14 10:35:17 +01:00
Jef Roosens 5b612127b2
feat: rename some structs 2024-02-12 13:40:56 +01:00
Jef Roosens 321e9c0b80
feat: improve request header api functions 2024-02-12 13:32:14 +01:00
6 changed files with 117 additions and 32 deletions

View File

@ -86,4 +86,17 @@ uint64_t lnm_atoi(const char *s, size_t len);
*/ */
uint64_t lnm_digits(uint64_t num); uint64_t lnm_digits(uint64_t num);
/**
* Find the first case-insensitive occurence of s2 in s1.
*
* @param out pointer to store start of match in
* @param s1 pointer to string to look in
* @param s1_len length of s1
* @param s2 string to search for in s1
* @param s2_len length of s2
* @return true if found, false otherwise
*/
bool lnm_stristr(const char **out, const char *s1, size_t s1_len,
const char *s2, size_t s2_len);
#endif #endif

View File

@ -13,7 +13,11 @@
#define LNM_HTTP_MAX_REQ_HEADERS 32 #define LNM_HTTP_MAX_REQ_HEADERS 32
#define LNM_HTTP_MAX_REGEX_GROUPS 4 #define LNM_HTTP_MAX_REGEX_GROUPS 4
typedef struct lnm_http_req_header { /**
* Internal representation of a header in a request, defined using offsets
* relative to the full buffer.
*/
typedef struct lnm_http_req_ihdr {
struct { struct {
size_t o; size_t o;
size_t len; size_t len;
@ -22,7 +26,7 @@ typedef struct lnm_http_req_header {
size_t o; size_t o;
size_t len; size_t len;
} value; } value;
} lnm_http_req_header; } lnm_http_req_ihhr;
/** /**
* Represents the parsed HTTP request * Represents the parsed HTTP request
@ -45,7 +49,7 @@ typedef struct lnm_http_req {
size_t len; size_t len;
} query; } query;
struct { struct {
lnm_http_req_header arr[LNM_HTTP_MAX_REQ_HEADERS]; lnm_http_req_ihhr arr[LNM_HTTP_MAX_REQ_HEADERS];
size_t len; size_t len;
} headers; } headers;
struct { struct {
@ -81,31 +85,71 @@ lnm_http_parse_err lnm_http_req_parse(lnm_http_req *req, char *buf, size_t len);
void lnm_http_req_reset(lnm_http_req *req); void lnm_http_req_reset(lnm_http_req *req);
/** /**
* Retrieve a specific header from the request. * Represents an actual header value, with offsets already resolved.
*/
typedef struct lnm_http_req_hdr {
struct {
const char *s;
size_t len;
} name;
struct {
const char *s;
size_t len;
} value;
} lnm_http_req_hdr;
/**
* Retrieve a known type header from the request.
* *
* Pointers retrieved from this function should never be used between step * Pointers retrieved from this function should never be used between step
* functions; simply request the header again if you need to. * functions; simply request the header again if you need to.
* *
* @param out where to write pointer to header value * @param out where to store pointer to struct representing header
* @param out_len where to store length of out value
* @param req request to look for header in * @param req request to look for header in
* @param type type of header to look for * @param type type of header to look for
*/ */
lnm_err lnm_http_req_header_get(const char **out, size_t *out_len, lnm_err lnm_http_req_header_get(lnm_http_req_hdr *out, lnm_http_req *req,
lnm_http_req *req, lnm_http_header type); lnm_http_header type);
/** /**
* Retrieve a specific header from the request by specifying its name. * Retrieve a header from the request using a case-insensitive name.
* *
* Pointers retrieved from this function should never be used between step * Pointers retrieved from this function should never be used between step
* functions; simply request the header again if you need to. * functions; simply request the header again if you need to.
* *
* @param out where to write pointer to header value * @param out where to store pointer to struct representing header
* @param out_len where to store length of out value
* @param req request to look for header in * @param req request to look for header in
* @param name name of the header; matches case-insensitive * @param name name of the header; matches case-insensitive
*/ */
lnm_err lnm_http_req_header_get_s(const char **out, size_t *out_len, lnm_err lnm_http_req_header_get_s(lnm_http_req_hdr *out, lnm_http_req *req,
lnm_http_req *req, const char *name); const char *name);
/**
* Represents a parameter in an HTTP header.
*/
typedef struct lnm_http_req_hdr_param {
struct {
const char *s;
size_t len;
} key;
struct {
const char *s;
size_t len;
} value;
} lnm_http_req_hdr_param;
/**
* Retrieve a specific key-value parameter from a header.
*
* Pointers retrieved from this function should never be used between step
* functions; simply request the header again if you need to.
*
* @param out output struct
* @param header header to look in
* @param key name of the parameter to return from the header
*/
lnm_err lnm_http_req_header_param(lnm_http_req_hdr_param *param,
const lnm_http_req_hdr *header,
const char *key);
#endif #endif

View File

@ -113,11 +113,10 @@ void lnm_http_loop_process_parse_headers(lnm_http_conn *conn) {
lnm_http_loop_ctx *ctx = conn->ctx; lnm_http_loop_ctx *ctx = conn->ctx;
lnm_http_req *req = &ctx->req; lnm_http_req *req = &ctx->req;
const char *value; lnm_http_req_hdr header;
size_t value_len; if (lnm_http_req_header_get(&header, req, lnm_http_header_content_length) ==
if (lnm_http_req_header_get(&value, &value_len, req, lnm_err_ok) {
lnm_http_header_content_length) == lnm_err_ok) { req->body.expected_len = lnm_atoi(header.value.s, header.value.len);
req->body.expected_len = lnm_atoi(value, value_len);
} }
ctx->state = lnm_http_loop_state_steps; ctx->state = lnm_http_loop_state_steps;

View File

@ -30,13 +30,13 @@ lnm_http_step_err lnm_http_loop_step_auth(lnm_http_conn *conn) {
// If there's no API key, requests are always authorized // If there's no API key, requests are always authorized
bool authorized = ctx->g->api_key == NULL; bool authorized = ctx->g->api_key == NULL;
const char *value; lnm_http_req_hdr header;
size_t value_len;
if (!authorized && lnm_http_req_header_get_s(&value, &value_len, &ctx->req, if (!authorized && lnm_http_req_header_get_s(&header, &ctx->req,
"X-Api-Key") == lnm_err_ok) { "X-Api-Key") == lnm_err_ok) {
authorized = (value_len == strlen(ctx->g->api_key)) && authorized =
(memcmp(value, ctx->g->api_key, value_len) == 0); (header.value.len == strlen(ctx->g->api_key)) &&
(memcmp(header.value.s, ctx->g->api_key, header.value.len) == 0);
} }
ctx->res.status = authorized ? ctx->res.status : lnm_http_status_unauthorized; ctx->res.status = authorized ? ctx->res.status : lnm_http_status_unauthorized;

View File

@ -88,23 +88,24 @@ void lnm_http_req_reset(lnm_http_req *req) {
memset(req, 0, sizeof(lnm_http_req)); memset(req, 0, sizeof(lnm_http_req));
} }
lnm_err lnm_http_req_header_get(const char **out, size_t *out_len, lnm_err lnm_http_req_header_get(lnm_http_req_hdr *out, lnm_http_req *req,
lnm_http_req *req, lnm_http_header type) { lnm_http_header type) {
return lnm_http_req_header_get_s(out, out_len, req, return lnm_http_req_header_get_s(out, req, lnm_http_header_names[type]);
lnm_http_header_names[type]);
} }
lnm_err lnm_http_req_header_get_s(const char **out, size_t *out_len, lnm_err lnm_http_req_header_get_s(lnm_http_req_hdr *out, lnm_http_req *req,
lnm_http_req *req, const char *name) { const char *name) {
size_t name_len = strlen(name); size_t name_len = strlen(name);
for (size_t i = 0; i < req->headers.len; i++) { for (size_t i = 0; i < req->headers.len; i++) {
const lnm_http_req_header *header = &req->headers.arr[i]; const lnm_http_req_ihhr *header = &req->headers.arr[i];
if (lnm_strnieq(req->buf.s + header->name.o, header->name.len, name, if (lnm_strnieq(req->buf.s + header->name.o, header->name.len, name,
name_len)) { name_len)) {
*out = req->buf.s + header->value.o; out->name.s = req->buf.s + header->name.o;
*out_len = header->value.len; out->name.len = header->name.len;
out->value.s = req->buf.s + header->value.o;
out->value.len = header->value.len;
return lnm_err_ok; return lnm_err_ok;
} }
@ -112,3 +113,15 @@ lnm_err lnm_http_req_header_get_s(const char **out, size_t *out_len,
return lnm_err_not_found; return lnm_err_not_found;
} }
lnm_err lnm_http_req_header_param(lnm_http_req_hdr_param *param,
const lnm_http_req_hdr *header,
const char *key) {
size_t key_len = strlen(key);
const char *s;
if (lnm_stristr(&s, header->name.s, header->name.len - 1, key, key_len) && (header->name.s[key_len] == '=')) {
// Value runs till end of header or next semi-colon
}
}

View File

@ -55,3 +55,19 @@ uint64_t lnm_digits(uint64_t num) {
return digits; return digits;
} }
bool lnm_stristr(const char **out, const char *s1, size_t s1_len,
const char *s2, size_t s2_len) {
while (s1_len >= s2_len) {
if (lnm_strnieq(s1, s1_len, s2, s2_len)) {
*out = s1;
return true;
}
s1_len--;
s1++;
}
return false;
}