feat(lnm): better routing detection
ci/woodpecker/push/build Pipeline was successful
Details
ci/woodpecker/push/build Pipeline was successful
Details
parent
8ec667af3b
commit
ad243ea9f5
|
@ -33,6 +33,9 @@ typedef struct lnm_http_req {
|
||||||
struct phr_header arr[LNM_HTTP_MAX_REQ_HEADERS];
|
struct phr_header arr[LNM_HTTP_MAX_REQ_HEADERS];
|
||||||
size_t len;
|
size_t len;
|
||||||
} headers;
|
} headers;
|
||||||
|
struct {
|
||||||
|
uint64_t expected_len;
|
||||||
|
} body;
|
||||||
uint64_t content_length;
|
uint64_t content_length;
|
||||||
} lnm_http_req;
|
} lnm_http_req;
|
||||||
|
|
||||||
|
|
|
@ -7,8 +7,8 @@
|
||||||
#include "lnm/common.h"
|
#include "lnm/common.h"
|
||||||
#include "lnm/http/consts.h"
|
#include "lnm/http/consts.h"
|
||||||
|
|
||||||
typedef lnm_err (*data_fn)(size_t *written, char *buf, lnm_http_conn *conn,
|
typedef lnm_err (*data_fn)(uint64_t *written, char *buf, lnm_http_conn *conn,
|
||||||
size_t offset, size_t len);
|
uint64_t offset, uint64_t len);
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Linked list elements used to store the response headers
|
* Linked list elements used to store the response headers
|
||||||
|
|
|
@ -12,7 +12,7 @@
|
||||||
/* static const lnm_http_loop_state lnm_http_loop_state_first_req =
|
/* static const lnm_http_loop_state lnm_http_loop_state_first_req =
|
||||||
* lnm_http_loop_state_parse_req; */
|
* lnm_http_loop_state_parse_req; */
|
||||||
static const lnm_http_loop_state lnm_http_loop_state_first_res =
|
static const lnm_http_loop_state lnm_http_loop_state_first_res =
|
||||||
lnm_http_loop_state_write_status_line;
|
lnm_http_loop_state_add_headers;
|
||||||
|
|
||||||
void lnm_http_loop_process_parse_req(lnm_http_conn *conn) {
|
void lnm_http_loop_process_parse_req(lnm_http_conn *conn) {
|
||||||
lnm_http_loop_ctx *ctx = conn->ctx;
|
lnm_http_loop_ctx *ctx = conn->ctx;
|
||||||
|
@ -52,7 +52,7 @@ void lnm_http_loop_process_route(lnm_http_conn *conn) {
|
||||||
int match_level = 0;
|
int match_level = 0;
|
||||||
lnm_http_route *route;
|
lnm_http_route *route;
|
||||||
|
|
||||||
for (size_t i = 0; i < gctx->routes.len && match_level < 2; i++) {
|
for (size_t i = 0; i < gctx->routes.len && match_level < 3; i++) {
|
||||||
route = gctx->routes.arr[i];
|
route = gctx->routes.arr[i];
|
||||||
bool matched_path;
|
bool matched_path;
|
||||||
|
|
||||||
|
@ -69,20 +69,21 @@ void lnm_http_loop_process_route(lnm_http_conn *conn) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Remember the previous match levels
|
// Remember the previous match levels
|
||||||
int new_match_level = matched_path + (route->method == ctx->req.method);
|
int new_match_level = 2 * matched_path + (route->method == ctx->req.method);
|
||||||
match_level = match_level < new_match_level ? new_match_level : match_level;
|
match_level = match_level < new_match_level ? new_match_level : match_level;
|
||||||
}
|
}
|
||||||
|
|
||||||
switch (match_level) {
|
switch (match_level) {
|
||||||
case 0:
|
case 0:
|
||||||
|
case 1:
|
||||||
ctx->res.status = lnm_http_status_not_found;
|
ctx->res.status = lnm_http_status_not_found;
|
||||||
ctx->state = lnm_http_loop_state_first_res;
|
ctx->state = lnm_http_loop_state_first_res;
|
||||||
break;
|
break;
|
||||||
case 1:
|
case 2:
|
||||||
ctx->res.status = lnm_http_status_method_not_allowed;
|
ctx->res.status = lnm_http_status_method_not_allowed;
|
||||||
ctx->state = lnm_http_loop_state_first_res;
|
ctx->state = lnm_http_loop_state_first_res;
|
||||||
break;
|
break;
|
||||||
case 2:
|
case 3:
|
||||||
ctx->route = route;
|
ctx->route = route;
|
||||||
ctx->cur_step = route->step;
|
ctx->cur_step = route->step;
|
||||||
ctx->state = lnm_http_loop_state_parse_headers;
|
ctx->state = lnm_http_loop_state_parse_headers;
|
||||||
|
@ -138,7 +139,6 @@ void lnm_http_loop_state_process_add_headers(lnm_http_conn *conn) {
|
||||||
lnm_http_loop_ctx *ctx = conn->ctx;
|
lnm_http_loop_ctx *ctx = conn->ctx;
|
||||||
lnm_http_res *res = &ctx->res;
|
lnm_http_res *res = &ctx->res;
|
||||||
|
|
||||||
if (res->body.len > 0) {
|
|
||||||
uint64_t digits = lnm_digits(res->body.len);
|
uint64_t digits = lnm_digits(res->body.len);
|
||||||
char *buf = malloc(digits + 1);
|
char *buf = malloc(digits + 1);
|
||||||
|
|
||||||
|
@ -151,7 +151,6 @@ void lnm_http_loop_state_process_add_headers(lnm_http_conn *conn) {
|
||||||
sprintf(buf, "%lu", res->body.len);
|
sprintf(buf, "%lu", res->body.len);
|
||||||
lnm_http_res_add_header_len(res, lnm_http_header_content_length, buf,
|
lnm_http_res_add_header_len(res, lnm_http_header_content_length, buf,
|
||||||
digits, true);
|
digits, true);
|
||||||
}
|
|
||||||
|
|
||||||
ctx->state = lnm_http_loop_state_write_status_line;
|
ctx->state = lnm_http_loop_state_write_status_line;
|
||||||
}
|
}
|
||||||
|
|
|
@ -16,9 +16,11 @@ lnm_http_parse_err lnm_http_req_parse(lnm_http_req *req, char *buf,
|
||||||
|
|
||||||
int req_len = phr_parse_request(
|
int req_len = phr_parse_request(
|
||||||
buf, len, &method, &method_len, (const char **)&path, &path_len,
|
buf, len, &method, &method_len, (const char **)&path, &path_len,
|
||||||
&req->minor_version, req->headers.arr, &req->headers.len, 0);
|
&req->minor_version, req->headers.arr, &req->headers.len, req->len);
|
||||||
|
|
||||||
if (req_len == -1) {
|
if (req_len == -1) {
|
||||||
|
req->len = len;
|
||||||
|
|
||||||
return lnm_http_parse_err_invalid;
|
return lnm_http_parse_err_invalid;
|
||||||
} else if (req_len == -2) {
|
} else if (req_len == -2) {
|
||||||
return lnm_http_parse_err_incomplete;
|
return lnm_http_parse_err_incomplete;
|
||||||
|
|
Loading…
Reference in New Issue