feat: improve request header api functions
parent
f5c0db7733
commit
fbba7a5a14
|
@ -14,7 +14,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_iheader {
|
||||||
struct {
|
struct {
|
||||||
size_t o;
|
size_t o;
|
||||||
size_t len;
|
size_t len;
|
||||||
|
@ -23,6 +27,20 @@ typedef struct lnm_http_req_header {
|
||||||
size_t o;
|
size_t o;
|
||||||
size_t len;
|
size_t len;
|
||||||
} value;
|
} value;
|
||||||
|
} lnm_http_req_iheader;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Represents an actual header value, with offsets already resolved.
|
||||||
|
*/
|
||||||
|
typedef struct lnm_http_req_header {
|
||||||
|
struct {
|
||||||
|
const char *s;
|
||||||
|
size_t len;
|
||||||
|
} name;
|
||||||
|
struct {
|
||||||
|
const char *s;
|
||||||
|
size_t len;
|
||||||
|
} value;
|
||||||
} lnm_http_req_header;
|
} lnm_http_req_header;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -46,7 +64,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_iheader arr[LNM_HTTP_MAX_REQ_HEADERS];
|
||||||
size_t len;
|
size_t len;
|
||||||
} headers;
|
} headers;
|
||||||
struct {
|
struct {
|
||||||
|
@ -82,32 +100,30 @@ 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.
|
* 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_header *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_header *out, lnm_http_req *req,
|
||||||
lnm_http_req *req, const char *name);
|
const char *name);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Retrieve a named key segment from the matched route.
|
* Retrieve a named key segment from the matched route.
|
||||||
|
|
|
@ -79,11 +79,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_header 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;
|
||||||
|
|
|
@ -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_header 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;
|
||||||
|
|
|
@ -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_header *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_header *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_iheader *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;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue