diff --git a/Makefile b/Makefile index a27e95b..0eaafed 100644 --- a/Makefile +++ b/Makefile @@ -47,7 +47,7 @@ bin/worker/%.o: src/worker/%.c bin/sesimos: bin/server.o bin/logger.o bin/cache_handler.o bin/async.o bin/workers.o \ bin/worker/request_handler.o bin/worker/tcp_acceptor.o bin/worker/tcp_closer.o bin/worker/responder.o \ - bin/worker/fastcgi_handler.o \ + bin/worker/fastcgi_handler.o bin/worker/local_handler.o bin/worker/proxy_handler.o \ bin/lib/compress.o bin/lib/config.o bin/lib/fastcgi.o bin/lib/geoip.o \ bin/lib/http.o bin/lib/http_static.o bin/lib/proxy.o bin/lib/sock.o bin/lib/uri.o \ bin/lib/utils.o bin/lib/websocket.o bin/lib/mpmc.o @@ -75,6 +75,10 @@ bin/worker/fastcgi_handler.o: src/worker/func.h bin/worker/responder.o: src/worker/func.h +bin/worker/local_handler.o: src/worker/func.h + +bin/worker/proxy_handler.o: src/worker/func.h + bin/lib/compress.o: src/lib/compress.h bin/lib/config.o: src/lib/config.h src/lib/utils.h src/lib/uri.h src/logger.h diff --git a/src/lib/fastcgi.c b/src/lib/fastcgi.c index 46c2e8d..5ec062e 100644 --- a/src/lib/fastcgi.c +++ b/src/lib/fastcgi.c @@ -132,8 +132,8 @@ int fastcgi_init(fastcgi_cnx_t *conn, int mode, unsigned int client_num, unsigne addr = (struct sockaddr_in6 *) &addr_storage; sprintf(buf0, "%i", addr->sin6_port); param_ptr = fastcgi_add_param(param_ptr, "REMOTE_PORT", buf0); - param_ptr = fastcgi_add_param(param_ptr, "REMOTE_ADDR", conn->ctx->addr); - param_ptr = fastcgi_add_param(param_ptr, "REMOTE_HOST", conn->ctx->host[0] != 0 ? conn->ctx->host : conn->ctx->addr); + param_ptr = fastcgi_add_param(param_ptr, "REMOTE_ADDR", conn->r_addr); + param_ptr = fastcgi_add_param(param_ptr, "REMOTE_HOST", conn->r_host != NULL ? conn->r_host : conn->r_addr); //param_ptr = fastcgi_add_param(param_ptr, "REMOTE_IDENT", ""); //param_ptr = fastcgi_add_param(param_ptr, "REMOTE_USER", ""); diff --git a/src/lib/fastcgi.h b/src/lib/fastcgi.h index 592e10f..b6d741c 100644 --- a/src/lib/fastcgi.h +++ b/src/lib/fastcgi.h @@ -12,7 +12,6 @@ #include "include/fastcgi.h" #include "http.h" #include "uri.h" -#include "../server.h" #define FASTCGI_CHUNKED 1 #define FASTCGI_COMPRESS_GZ 2 @@ -37,7 +36,8 @@ typedef struct { const char *webroot; unsigned short out_len; unsigned short out_off; - client_ctx_t *ctx; + char *r_addr; + char *r_host; } fastcgi_cnx_t; char *fastcgi_add_param(char *buf, const char *key, const char *value); diff --git a/src/server.h b/src/server.h index ed504fe..b453439 100644 --- a/src/server.h +++ b/src/server.h @@ -13,6 +13,7 @@ #include "lib/http.h" #include "lib/uri.h" #include "lib/config.h" +#include "lib/fastcgi.h" #include #include @@ -36,7 +37,7 @@ typedef struct { char *addr, *s_addr; unsigned char in_use: 1, s_keep_alive:1, c_keep_alive:1; char cc[3], host[256]; - char req_host[256]; + char req_host[256], err_msg[256]; char log_prefix[512]; char _c_addr[INET6_ADDRSTRLEN + 1], _s_addr[INET6_ADDRSTRLEN + 1]; struct timespec begin, end; @@ -44,10 +45,13 @@ typedef struct { http_res res; http_uri uri; http_status_ctx status; + http_status custom_status; int use_fastcgi, use_proxy; host_config_t *conf; FILE *file; long content_length; + fastcgi_cnx_t fcgi_cnx; + char msg_buf[8192], msg_content[1024]; } client_ctx_t; extern volatile sig_atomic_t server_alive; diff --git a/src/worker/fastcgi_handler.c b/src/worker/fastcgi_handler.c index 5918ce3..f858e61 100644 --- a/src/worker/fastcgi_handler.c +++ b/src/worker/fastcgi_handler.c @@ -1,6 +1,6 @@ /** * sesimos - secure, simple, modern web server - * @brief TCP closer + * @brief FastCGI handler * @file src/worker/fastcgi_handler.c * @author Lorenz Stechauner * @date 2022-12-28 @@ -9,8 +9,136 @@ #include "func.h" #include "../logger.h" #include "../lib/utils.h" +#include "../lib/compress.h" +#include "../workers.h" + +#include + +static int fastcgi_handler(client_ctx_t *ctx); void fastcgi_handler_func(client_ctx_t *ctx) { logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix); // TODO + fastcgi_handler(ctx); + respond(ctx); +} + +static int fastcgi_handler(client_ctx_t *ctx) { + http_res *res = &ctx->res; + http_req *req = &ctx->req; + http_uri *uri = &ctx->uri; + sock *client = &ctx->socket; + fastcgi_cnx_t *fcgi_cnx = &ctx->fcgi_cnx; + char *err_msg = ctx->err_msg; + + char buf[1024]; + + int mode, ret; + if (strcmp(uri->filename + strlen(uri->filename) - 4, ".ncr") == 0) { + mode = FASTCGI_SESIMOS; + } else if (strcmp(uri->filename + strlen(uri->filename) - 4, ".php") == 0) { + mode = FASTCGI_PHP; + } else { + res->status = http_get_status(500); + error("Invalid FastCGI extension: %s", uri->filename); + return 0; + } + + struct stat statbuf; + stat(uri->filename, &statbuf); + char *last_modified = http_format_date(statbuf.st_mtime, buf, sizeof(buf)); + http_add_header_field(&res->hdr, "Last-Modified", last_modified); + + res->status = http_get_status(200); + if (fastcgi_init(fcgi_cnx, mode, 0 /* TODO */, ctx->req_num, client, req, uri) != 0) { + res->status = http_get_status(503); + sprintf(err_msg, "Unable to communicate with FastCGI socket."); + return 0; + } + + const char *client_content_length = http_get_header_field(&req->hdr, "Content-Length"); + const char *client_transfer_encoding = http_get_header_field(&req->hdr, "Transfer-Encoding"); + if (client_content_length != NULL) { + unsigned long client_content_len = strtoul(client_content_length, NULL, 10); + ret = fastcgi_receive(fcgi_cnx, client, client_content_len); + } else if (client_transfer_encoding != NULL && strstr(client_transfer_encoding, "chunked") != NULL) { + ret = fastcgi_receive_chunked(fcgi_cnx, client); + } else { + ret = 0; + } + if (ret != 0) { + if (ret < 0) { + return -1; + } else { + sprintf(err_msg, "Unable to communicate with FastCGI socket."); + } + res->status = http_get_status(502); + return 0; + } + fastcgi_close_stdin(fcgi_cnx); + + ret = fastcgi_header(fcgi_cnx, res, err_msg); + if (ret != 0) { + return (ret < 0) ? -1 : 0; + } + + const char *status_hdr = http_get_header_field(&res->hdr, "Status"); + if (status_hdr != NULL) { + int status_code = (int) strtoul(status_hdr, NULL, 10); + res->status = http_get_status(status_code); + http_remove_header_field(&res->hdr, "Status", HTTP_REMOVE_ALL); + if (res->status == NULL && status_code >= 100 && status_code <= 999) { + ctx->custom_status.code = status_code; + strcpy(ctx->custom_status.type, ""); + strcpy(ctx->custom_status.msg, status_hdr + 4); + res->status = &ctx->custom_status; + } else if (res->status == NULL) { + res->status = http_get_status(500); + sprintf(err_msg, "The status_hdr code was set to an invalid or unknown value."); + return 0; + } + } + + const char *content_length_f = http_get_header_field(&res->hdr, "Content-Length"); + ctx->content_length = (content_length_f == NULL) ? -1 : strtol(content_length_f, NULL, 10); + + const char *content_type = http_get_header_field(&res->hdr, "Content-Type"); + const char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding"); + if (content_encoding == NULL && + content_type != NULL && + strncmp(content_type, "text/html", 9) == 0 && + ctx->content_length != -1 && + ctx->content_length <= sizeof(ctx->msg_content) - 1) + { + fastcgi_dump(fcgi_cnx, ctx->msg_content, sizeof(ctx->msg_content)); + return 0; + } + + ctx->use_fastcgi = 1; + + if (ctx->content_length != -1 && ctx->content_length < 1024000) { + ctx->use_fastcgi |= FASTCGI_COMPRESS_HOLD; + } + + ctx->content_length = -1; + + int http_comp = http_get_compression(req, res); + if (http_comp & COMPRESS) { + if (http_comp & COMPRESS_BR) { + ctx->use_fastcgi |= FASTCGI_COMPRESS_BR; + sprintf(buf, "br"); + } else if (http_comp & COMPRESS_GZ) { + ctx->use_fastcgi |= FASTCGI_COMPRESS_GZ; + sprintf(buf, "gzip"); + } + http_add_header_field(&res->hdr, "Vary", "Accept-Encoding"); + http_add_header_field(&res->hdr, "Content-Encoding", buf); + http_remove_header_field(&res->hdr, "Content-Length", HTTP_REMOVE_ALL); + } + + if (http_get_header_field(&res->hdr, "Content-Length") == NULL) { + http_add_header_field(&res->hdr, "Transfer-Encoding", "chunked"); + } + + return 0; } diff --git a/src/worker/func.h b/src/worker/func.h index b1e2729..26601b5 100644 --- a/src/worker/func.h +++ b/src/worker/func.h @@ -19,6 +19,10 @@ void request_handler_func(client_ctx_t *ctx); void responder_func(client_ctx_t *ctx); +void local_handler_func(client_ctx_t *ctx); + void fastcgi_handler_func(client_ctx_t *ctx); +void proxy_handler_func(client_ctx_t *ctx); + #endif //SESIMOS_FUNC_H diff --git a/src/worker/local_handler.c b/src/worker/local_handler.c new file mode 100644 index 0000000..97b4d4d --- /dev/null +++ b/src/worker/local_handler.c @@ -0,0 +1,203 @@ +/** + * sesimos - secure, simple, modern web server + * @brief Local filesystem handler + * @file src/worker/local_handler.c + * @author Lorenz Stechauner + * @date 2022-12-29 + */ + +#include "func.h" +#include "../logger.h" +#include "../lib/utils.h" +#include "../lib/compress.h" +#include "../workers.h" + +#include + +static int local_handler(client_ctx_t *ctx); + +void local_handler_func(client_ctx_t *ctx) { + logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix); + + switch (local_handler(ctx)) { + case 0: respond(ctx); break; + case 1: fastcgi_handle(ctx); break; + default: tcp_close(ctx); break; + } +} + +static int local_handler(client_ctx_t *ctx) { + http_res *res = &ctx->res; + http_req *req = &ctx->req; + http_uri *uri = &ctx->uri; + char *err_msg = ctx->err_msg; + + char buf1[1024], buf2[1024]; + int accept_if_modified_since = 0; + + if (strcmp(req->method, "TRACE") == 0) { + res->status = http_get_status(200); + http_add_header_field(&res->hdr, "Content-Type", "message/http"); + + ctx->content_length = snprintf(ctx->msg_buf, sizeof(ctx->msg_buf) - ctx->content_length, "%s %s HTTP/%s\r\n", req->method, req->uri, req->version); + for (int i = 0; i < req->hdr.field_num; i++) { + const http_field *f = &req->hdr.fields[i]; + ctx->content_length += snprintf(ctx->msg_buf + ctx->content_length, sizeof(ctx->msg_buf) - ctx->content_length, "%s: %s\r\n", http_field_get_name(f), http_field_get_value(f)); + } + + return 0; + } + + if (strncmp(uri->req_path, "/.well-known/", 13) == 0) { + http_add_header_field(&res->hdr, "Access-Control-Allow-Origin", "*"); + } + + if (strncmp(uri->req_path, "/.well-known/", 13) != 0 && strstr(uri->path, "/.") != NULL) { + res->status = http_get_status(403); + sprintf(err_msg, "Parts of this URI are hidden."); + return 0; + } else if (uri->filename == NULL && (int) uri->is_static && (int) uri->is_dir && strlen(uri->pathinfo) == 0) { + res->status = http_get_status(403); + sprintf(err_msg, "It is not allowed to list the contents of this directory."); + return 0; + } else if (uri->filename == NULL && (int) !uri->is_static && (int) uri->is_dir && strlen(uri->pathinfo) == 0) { + // TODO list directory contents + res->status = http_get_status(501); + sprintf(err_msg, "Listing contents of an directory is currently not implemented."); + return 0; + } else if (uri->filename == NULL || (strlen(uri->pathinfo) > 0 && (int) uri->is_static)) { + res->status = http_get_status(404); + return 0; + } else if (strlen(uri->pathinfo) != 0 && ctx->conf->local.dir_mode != URI_DIR_MODE_INFO) { + res->status = http_get_status(404); + return 0; + } + + if (uri->is_static) { + res->status = http_get_status(200); + http_add_header_field(&res->hdr, "Accept-Ranges", "bytes"); + if (strcmp(req->method, "GET") != 0 && strcmp(req->method, "HEAD") != 0) { + res->status = http_get_status(405); + return 0; + } + + if (http_get_header_field(&req->hdr, "Content-Length") != NULL || http_get_header_field(&req->hdr, "Transfer-Encoding") != NULL) { + res->status = http_get_status(400); + sprintf(err_msg, "A GET request must not contain a payload"); + return 0; + } + + cache_init_uri(ctx->conf->cache, uri); + + const char *last_modified = http_format_date(uri->meta->stat.st_mtime, buf1, sizeof(buf1)); + http_add_header_field(&res->hdr, "Last-Modified", last_modified); + sprintf(buf2, "%s; charset=%s", uri->meta->type, uri->meta->charset); + http_add_header_field(&res->hdr, "Content-Type", buf2); + + + const char *accept_encoding = http_get_header_field(&req->hdr, "Accept-Encoding"); + int enc = 0; + if (accept_encoding != NULL) { + if (uri->meta->filename_comp_br[0] != 0 && strstr(accept_encoding, "br") != NULL) { + ctx->file = fopen(uri->meta->filename_comp_br, "rb"); + if (ctx->file == NULL) { + cache_mark_dirty(ctx->conf->cache, uri->filename); + } else { + http_add_header_field(&res->hdr, "Content-Encoding", "br"); + enc = COMPRESS_BR; + } + } else if (uri->meta->filename_comp_gz[0] != 0 && strstr(accept_encoding, "gzip") != NULL) { + ctx->file = fopen(uri->meta->filename_comp_gz, "rb"); + if (ctx->file == NULL) { + cache_mark_dirty(ctx->conf->cache, uri->filename); + } else { + http_add_header_field(&res->hdr, "Content-Encoding", "gzip"); + enc = COMPRESS_GZ; + } + } + if (enc != 0) { + http_add_header_field(&res->hdr, "Vary", "Accept-Encoding"); + } + } + + if (uri->meta->etag[0] != 0) { + if (enc) { + sprintf(buf1, "%s-%s", uri->meta->etag, (enc & COMPRESS_BR) ? "br" : (enc & COMPRESS_GZ) ? "gzip" : ""); + http_add_header_field(&res->hdr, "ETag", buf1); + } else { + http_add_header_field(&res->hdr, "ETag", uri->meta->etag); + } + } + + if (strncmp(uri->meta->type, "text/", 5) == 0) { + http_add_header_field(&res->hdr, "Cache-Control", "public, max-age=3600"); + } else { + http_add_header_field(&res->hdr, "Cache-Control", "public, max-age=86400"); + } + + const char *if_modified_since = http_get_header_field(&req->hdr, "If-Modified-Since"); + const char *if_none_match = http_get_header_field(&req->hdr, "If-None-Match"); + if ((if_none_match != NULL && strstr(if_none_match, uri->meta->etag) == NULL) || + (accept_if_modified_since && if_modified_since != NULL && strcmp(if_modified_since, last_modified) == 0)) + { + res->status = http_get_status(304); + return 0; + } + + const char *range = http_get_header_field(&req->hdr, "Range"); + if (range != NULL) { + if (strlen(range) <= 6 || strncmp(range, "bytes=", 6) != 0) { + res->status = http_get_status(416); + http_remove_header_field(&res->hdr, "Content-Type", HTTP_REMOVE_ALL); + http_remove_header_field(&res->hdr, "Last-Modified", HTTP_REMOVE_ALL); + http_remove_header_field(&res->hdr, "ETag", HTTP_REMOVE_ALL); + http_remove_header_field(&res->hdr, "Cache-Control", HTTP_REMOVE_ALL); + return 0; + } + range += 6; + char *ptr = strchr(range, '-'); + if (ptr == NULL) { + res->status = http_get_status(416); + return 0; + } + ctx->file = fopen(uri->filename, "rb"); + fseek(ctx->file, 0, SEEK_END); + unsigned long file_len = ftell(ctx->file); + fseek(ctx->file, 0, SEEK_SET); + if (file_len == 0) { + ctx->content_length = 0; + return 0; + } + long num1 = 0; + long num2 = (long) file_len - 1; + + if (ptr != range) num1 = (long) strtoul(range, NULL, 10); + if (ptr[1] != 0) num2 = (long) strtoul(ptr + 1, NULL, 10); + + if (num1 >= file_len || num2 >= file_len || num1 > num2) { + res->status = http_get_status(416); + return 0; + } + sprintf(buf1, "bytes %li-%li/%li", num1, num2, file_len); + http_add_header_field(&res->hdr, "Content-Range", buf1); + + res->status = http_get_status(206); + fseek(ctx->file, num1, SEEK_SET); + ctx->content_length = num2 - num1 + 1; + + return 0; + } + + if (ctx->file == NULL) { + ctx->file = fopen(uri->filename, "rb"); + } + + fseek(ctx->file, 0, SEEK_END); + ctx->content_length = ftell(ctx->file); + fseek(ctx->file, 0, SEEK_SET); + } else { + return 1; + } + + return 0; +} diff --git a/src/worker/proxy_handler.c b/src/worker/proxy_handler.c new file mode 100644 index 0000000..24c1d4f --- /dev/null +++ b/src/worker/proxy_handler.c @@ -0,0 +1,105 @@ +/** + * sesimos - secure, simple, modern web server + * @brief Proxy handler + * @file src/worker/proxy_handler.c + * @author Lorenz Stechauner + * @date 2022-12-29 + */ + +#include "func.h" +#include "../logger.h" +#include "../lib/utils.h" +#include "../lib/proxy.h" +#include "../lib/websocket.h" +#include "../workers.h" + +#include + +static int proxy_handler(client_ctx_t *ctx); + +void proxy_handler_func(client_ctx_t *ctx) { + logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, ctx->req_host, CLR_STR, ctx->log_prefix); + + proxy_handler(ctx); + respond(ctx); +} + +static int proxy_handler(client_ctx_t *ctx) { + http_res *res = &ctx->res; + http_req *req = &ctx->req; + http_uri *uri = &ctx->uri; + http_status_ctx *status = &ctx->status; + sock *client = &ctx->socket; + char *err_msg = ctx->err_msg; + + char buf[1024]; + + info("Reverse proxy for " BLD_STR "%s:%i" CLR_STR, ctx->conf->proxy.hostname, ctx->conf->proxy.port); + http_remove_header_field(&res->hdr, "Date", HTTP_REMOVE_ALL); + http_remove_header_field(&res->hdr, "Server", HTTP_REMOVE_ALL); + + int ret = proxy_init(req, res, status, ctx->conf, client, ctx, &ctx->custom_status, err_msg); + ctx->use_proxy = (ret == 0); + + if (res->status->code == 101) { + const char *connection = http_get_header_field(&res->hdr, "Connection"); + const char *upgrade = http_get_header_field(&res->hdr, "Upgrade"); + if (connection != NULL && upgrade != NULL && + (strstr(connection, "upgrade") != NULL || strstr(connection, "Upgrade") != NULL) && + strcmp(upgrade, "websocket") == 0) + { + const char *ws_accept = http_get_header_field(&res->hdr, "Sec-WebSocket-Accept"); + if (ws_calc_accept_key(status->ws_key, buf) == 0) { + ctx->use_proxy = (strcmp(buf, ws_accept) == 0) ? 2 : 1; + } + } else { + status->status = 101; + status->origin = INTERNAL; + res->status = http_get_status(501); + } + } + + // Let 300 be formatted by origin server + if (ctx->use_proxy && res->status->code >= 301 && res->status->code < 600) { + const char *content_type = http_get_header_field(&res->hdr, "Content-Type"); + const char *content_length_f = http_get_header_field(&res->hdr, "Content-Length"); + const char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding"); + if (content_encoding == NULL && content_type != NULL && content_length_f != NULL && strncmp(content_type, "text/html", 9) == 0) { + long content_len = strtol(content_length_f, NULL, 10); + if (content_len <= sizeof(ctx->msg_content) - 1) { + if (status->status != 101) { + status->status = res->status->code; + status->origin = res->status->code >= 400 ? SERVER : NONE; + } + ctx->use_proxy = 0; + proxy_dump(ctx->msg_content, content_len); + } + } + } + + /* + char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding"); + if (use_proxy && content_encoding == NULL) { + int http_comp = http_get_compression(&req, &res); + if (http_comp & COMPRESS_BR) { + use_proxy |= PROXY_COMPRESS_BR; + } else if (http_comp & COMPRESS_GZ) { + use_proxy |= PROXY_COMPRESS_GZ; + } + } + + char *transfer_encoding = http_get_header_field(&res->hdr, "Transfer-Encoding"); + int chunked = transfer_encoding != NULL && strcmp(transfer_encoding, "chunked") == 0; + http_remove_header_field(&res->hdr, "Transfer-Encoding", HTTP_REMOVE_ALL); + ret = sprintf(buf, "%s%s%s", + (use_proxy & PROXY_COMPRESS_BR) ? "br" : + ((use_proxy & PROXY_COMPRESS_GZ) ? "gzip" : ""), + ((use_proxy & PROXY_COMPRESS) && chunked) ? ", " : "", + chunked ? "chunked" : ""); + if (ret > 0) { + http_add_header_field(&res->hdr, "Transfer-Encoding", buf); + } + */ + + return 0; +} diff --git a/src/worker/request_handler.c b/src/worker/request_handler.c index 5926561..92cf771 100644 --- a/src/worker/request_handler.c +++ b/src/worker/request_handler.c @@ -11,12 +11,7 @@ #include "../workers.h" #include "../lib/mpmc.h" #include "../logger.h" - #include "../lib/utils.h" -#include "../lib/proxy.h" -#include "../lib/fastcgi.h" -#include "../lib/compress.h" -#include "../lib/websocket.h" #include #include @@ -27,33 +22,25 @@ static int request_handler(client_ctx_t *ctx); void request_handler_func(client_ctx_t *ctx) { logger_set_prefix("[%*s]%s", INET6_ADDRSTRLEN, ctx->s_addr, ctx->log_prefix); - if (request_handler(ctx) == 0) { - respond(ctx); - } else { - tcp_close(ctx); + switch (request_handler(ctx)) { + case 0: respond(ctx); break; + case 1: local_handle(ctx); break; + case 2: proxy_handle(ctx); break; + default: tcp_close(ctx); break; } } static int request_handler(client_ctx_t *ctx) { sock *client = &ctx->socket; + char *err_msg = ctx->err_msg; + long ret; - char buf0[1024], buf1[1024]; - char msg_buf[8192], err_msg[256]; - char msg_content[1024]; - const char *host_ptr, *hdr_connection; - msg_buf[0] = 0; err_msg[0] = 0; - msg_content[0] = 0; - int accept_if_modified_since = 0; ctx->use_fastcgi = 0; ctx->use_proxy = 0; - int p_len; - - fastcgi_cnx_t fcgi_cnx = {.socket = 0, .req_id = 0, .ctx = ctx}; - http_status custom_status; http_res *res = &ctx->res; res->status = http_get_status(501); @@ -66,6 +53,11 @@ static int request_handler(client_ctx_t *ctx) { status->origin = NONE; status->ws_key = NULL; + ctx->fcgi_cnx.socket = 0; + ctx->fcgi_cnx.req_id = 0; + ctx->fcgi_cnx.r_addr = ctx->addr; + ctx->fcgi_cnx.r_host = (ctx->host[0] != 0) ? ctx->host : NULL; + clock_gettime(CLOCK_MONOTONIC, &ctx->begin); //ret = sock_poll_read(&client, NULL, NULL, 1, NULL, NULL, CLIENT_TIMEOUT * 1000); @@ -102,9 +94,9 @@ static int request_handler(client_ctx_t *ctx) { return 0; } - hdr_connection = http_get_header_field(&req->hdr, "Connection"); + const char *hdr_connection = http_get_header_field(&req->hdr, "Connection"); ctx->c_keep_alive = (hdr_connection != NULL && (strstr(hdr_connection, "keep-alive") != NULL || strstr(hdr_connection, "Keep-Alive") != NULL)); - host_ptr = http_get_header_field(&req->hdr, "Host"); + const char *host_ptr = http_get_header_field(&req->hdr, "Host"); if (host_ptr != NULL && strlen(host_ptr) > 255) { ctx->req_host[0] = 0; res->status = http_get_status(400); @@ -162,7 +154,7 @@ static int request_handler(client_ctx_t *ctx) { res->status = http_get_status(308); size = url_encode(uri->uri, strlen(uri->uri), buf0, sizeof(buf0)); if (change_proto) { - p_len = snprintf(buf1, sizeof(buf1), "https://%s%s", ctx->req_host, buf0); + int p_len = snprintf(buf1, sizeof(buf1), "https://%s%s", ctx->req_host, buf0); if (p_len < 0 || p_len >= sizeof(buf1)) { res->status = http_get_status(500); error("Header field 'Location' too long"); @@ -182,341 +174,9 @@ static int request_handler(client_ctx_t *ctx) { } if (ctx->conf->type == CONFIG_TYPE_LOCAL) { - if (strcmp(req->method, "TRACE") == 0) { - res->status = http_get_status(200); - http_add_header_field(&res->hdr, "Content-Type", "message/http"); - - ctx->content_length = snprintf(msg_buf, sizeof(msg_buf) - ctx->content_length, "%s %s HTTP/%s\r\n", req->method, req->uri, req->version); - for (int i = 0; i < req->hdr.field_num; i++) { - const http_field *f = &req->hdr.fields[i]; - ctx->content_length += snprintf(msg_buf + ctx->content_length, sizeof(msg_buf) - ctx->content_length, "%s: %s\r\n", http_field_get_name(f), http_field_get_value(f)); - } - - return 0; - } - - if (strncmp(uri->req_path, "/.well-known/", 13) == 0) { - http_add_header_field(&res->hdr, "Access-Control-Allow-Origin", "*"); - } - - if (strncmp(uri->req_path, "/.well-known/", 13) != 0 && strstr(uri->path, "/.") != NULL) { - res->status = http_get_status(403); - sprintf(err_msg, "Parts of this URI are hidden."); - return 0; - } else if (uri->filename == NULL && (int) uri->is_static && (int) uri->is_dir && strlen(uri->pathinfo) == 0) { - res->status = http_get_status(403); - sprintf(err_msg, "It is not allowed to list the contents of this directory."); - return 0; - } else if (uri->filename == NULL && (int) !uri->is_static && (int) uri->is_dir && strlen(uri->pathinfo) == 0) { - // TODO list directory contents - res->status = http_get_status(501); - sprintf(err_msg, "Listing contents of an directory is currently not implemented."); - return 0; - } else if (uri->filename == NULL || (strlen(uri->pathinfo) > 0 && (int) uri->is_static)) { - res->status = http_get_status(404); - return 0; - } else if (strlen(uri->pathinfo) != 0 && ctx->conf->local.dir_mode != URI_DIR_MODE_INFO) { - res->status = http_get_status(404); - return 0; - } - - if (uri->is_static) { - res->status = http_get_status(200); - http_add_header_field(&res->hdr, "Accept-Ranges", "bytes"); - if (strcmp(req->method, "GET") != 0 && strcmp(req->method, "HEAD") != 0) { - res->status = http_get_status(405); - return 0; - } - - if (http_get_header_field(&req->hdr, "Content-Length") != NULL || http_get_header_field(&req->hdr, "Transfer-Encoding") != NULL) { - res->status = http_get_status(400); - sprintf(err_msg, "A GET request must not contain a payload"); - return 0; - } - - cache_init_uri(ctx->conf->cache, uri); - - const char *last_modified = http_format_date(uri->meta->stat.st_mtime, buf0, sizeof(buf0)); - http_add_header_field(&res->hdr, "Last-Modified", last_modified); - sprintf(buf1, "%s; charset=%s", uri->meta->type, uri->meta->charset); - http_add_header_field(&res->hdr, "Content-Type", buf1); - - - const char *accept_encoding = http_get_header_field(&req->hdr, "Accept-Encoding"); - int enc = 0; - if (accept_encoding != NULL) { - if (uri->meta->filename_comp_br[0] != 0 && strstr(accept_encoding, "br") != NULL) { - ctx->file = fopen(uri->meta->filename_comp_br, "rb"); - if (ctx->file == NULL) { - cache_mark_dirty(ctx->conf->cache, uri->filename); - } else { - http_add_header_field(&res->hdr, "Content-Encoding", "br"); - enc = COMPRESS_BR; - } - } else if (uri->meta->filename_comp_gz[0] != 0 && strstr(accept_encoding, "gzip") != NULL) { - ctx->file = fopen(uri->meta->filename_comp_gz, "rb"); - if (ctx->file == NULL) { - cache_mark_dirty(ctx->conf->cache, uri->filename); - } else { - http_add_header_field(&res->hdr, "Content-Encoding", "gzip"); - enc = COMPRESS_GZ; - } - } - if (enc != 0) { - http_add_header_field(&res->hdr, "Vary", "Accept-Encoding"); - } - } - - if (uri->meta->etag[0] != 0) { - if (enc) { - sprintf(buf0, "%s-%s", uri->meta->etag, (enc & COMPRESS_BR) ? "br" : (enc & COMPRESS_GZ) ? "gzip" : ""); - http_add_header_field(&res->hdr, "ETag", buf0); - } else { - http_add_header_field(&res->hdr, "ETag", uri->meta->etag); - } - } - - if (strncmp(uri->meta->type, "text/", 5) == 0) { - http_add_header_field(&res->hdr, "Cache-Control", "public, max-age=3600"); - } else { - http_add_header_field(&res->hdr, "Cache-Control", "public, max-age=86400"); - } - - const char *if_modified_since = http_get_header_field(&req->hdr, "If-Modified-Since"); - const char *if_none_match = http_get_header_field(&req->hdr, "If-None-Match"); - if ((if_none_match != NULL && strstr(if_none_match, uri->meta->etag) == NULL) || - (accept_if_modified_since && if_modified_since != NULL && strcmp(if_modified_since, last_modified) == 0)) - { - res->status = http_get_status(304); - return 0; - } - - const char *range = http_get_header_field(&req->hdr, "Range"); - if (range != NULL) { - if (strlen(range) <= 6 || strncmp(range, "bytes=", 6) != 0) { - res->status = http_get_status(416); - http_remove_header_field(&res->hdr, "Content-Type", HTTP_REMOVE_ALL); - http_remove_header_field(&res->hdr, "Last-Modified", HTTP_REMOVE_ALL); - http_remove_header_field(&res->hdr, "ETag", HTTP_REMOVE_ALL); - http_remove_header_field(&res->hdr, "Cache-Control", HTTP_REMOVE_ALL); - return 0; - } - range += 6; - char *ptr = strchr(range, '-'); - if (ptr == NULL) { - res->status = http_get_status(416); - return 0; - } - ctx->file = fopen(uri->filename, "rb"); - fseek(ctx->file, 0, SEEK_END); - unsigned long file_len = ftell(ctx->file); - fseek(ctx->file, 0, SEEK_SET); - if (file_len == 0) { - ctx->content_length = 0; - return 0; - } - long num1 = 0; - long num2 = (long) file_len - 1; - - if (ptr != range) num1 = (long) strtoul(range, NULL, 10); - if (ptr[1] != 0) num2 = (long) strtoul(ptr + 1, NULL, 10); - - if (num1 >= file_len || num2 >= file_len || num1 > num2) { - res->status = http_get_status(416); - return 0; - } - sprintf(buf0, "bytes %li-%li/%li", num1, num2, file_len); - http_add_header_field(&res->hdr, "Content-Range", buf0); - - res->status = http_get_status(206); - fseek(ctx->file, num1, SEEK_SET); - ctx->content_length = num2 - num1 + 1; - - return 0; - } - - if (ctx->file == NULL) { - ctx->file = fopen(uri->filename, "rb"); - } - - fseek(ctx->file, 0, SEEK_END); - ctx->content_length = ftell(ctx->file); - fseek(ctx->file, 0, SEEK_SET); - } else { - int mode; - if (strcmp(uri->filename + strlen(uri->filename) - 4, ".ncr") == 0) { - mode = FASTCGI_SESIMOS; - } else if (strcmp(uri->filename + strlen(uri->filename) - 4, ".php") == 0) { - mode = FASTCGI_PHP; - } else { - res->status = http_get_status(500); - error("Invalid FastCGI extension: %s", uri->filename); - return 0; - } - - struct stat statbuf; - stat(uri->filename, &statbuf); - char *last_modified = http_format_date(statbuf.st_mtime, buf0, sizeof(buf0)); - http_add_header_field(&res->hdr, "Last-Modified", last_modified); - - res->status = http_get_status(200); - if (fastcgi_init(&fcgi_cnx, mode, 0 /* TODO */, ctx->req_num, client, req, uri) != 0) { - res->status = http_get_status(503); - sprintf(err_msg, "Unable to communicate with FastCGI socket."); - return 0; - } - - const char *client_content_length = http_get_header_field(&req->hdr, "Content-Length"); - const char *client_transfer_encoding = http_get_header_field(&req->hdr, "Transfer-Encoding"); - if (client_content_length != NULL) { - unsigned long client_content_len = strtoul(client_content_length, NULL, 10); - ret = fastcgi_receive(&fcgi_cnx, client, client_content_len); - } else if (client_transfer_encoding != NULL && strstr(client_transfer_encoding, "chunked") != NULL) { - ret = fastcgi_receive_chunked(&fcgi_cnx, client); - } else { - ret = 0; - } - if (ret != 0) { - if (ret < 0) { - return -1; - } else { - sprintf(err_msg, "Unable to communicate with FastCGI socket."); - } - res->status = http_get_status(502); - return 0; - } - fastcgi_close_stdin(&fcgi_cnx); - - ret = fastcgi_header(&fcgi_cnx, res, err_msg); - if (ret != 0) { - return (ret < 0) ? -1 : 0; - } - - const char *status_hdr = http_get_header_field(&res->hdr, "Status"); - if (status_hdr != NULL) { - int status_code = (int) strtoul(status_hdr, NULL, 10); - res->status = http_get_status(status_code); - http_remove_header_field(&res->hdr, "Status", HTTP_REMOVE_ALL); - if (res->status == NULL && status_code >= 100 && status_code <= 999) { - custom_status.code = status_code; - strcpy(custom_status.type, ""); - strcpy(custom_status.msg, status_hdr + 4); - res->status = &custom_status; - } else if (res->status == NULL) { - res->status = http_get_status(500); - sprintf(err_msg, "The status_hdr code was set to an invalid or unknown value."); - return 0; - } - } - - const char *content_length_f = http_get_header_field(&res->hdr, "Content-Length"); - ctx->content_length = (content_length_f == NULL) ? -1 : strtol(content_length_f, NULL, 10); - - const char *content_type = http_get_header_field(&res->hdr, "Content-Type"); - const char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding"); - if (content_encoding == NULL && - content_type != NULL && - strncmp(content_type, "text/html", 9) == 0 && - ctx->content_length != -1 && - ctx->content_length <= sizeof(msg_content) - 1) - { - fastcgi_dump(&fcgi_cnx, msg_content, sizeof(msg_content)); - return 0; - } - - ctx->use_fastcgi = 1; - - if (ctx->content_length != -1 && ctx->content_length < 1024000) { - ctx->use_fastcgi |= FASTCGI_COMPRESS_HOLD; - } - - ctx->content_length = -1; - - int http_comp = http_get_compression(req, res); - if (http_comp & COMPRESS) { - if (http_comp & COMPRESS_BR) { - ctx->use_fastcgi |= FASTCGI_COMPRESS_BR; - sprintf(buf0, "br"); - } else if (http_comp & COMPRESS_GZ) { - ctx->use_fastcgi |= FASTCGI_COMPRESS_GZ; - sprintf(buf0, "gzip"); - } - http_add_header_field(&res->hdr, "Vary", "Accept-Encoding"); - http_add_header_field(&res->hdr, "Content-Encoding", buf0); - http_remove_header_field(&res->hdr, "Content-Length", HTTP_REMOVE_ALL); - } - - if (http_get_header_field(&res->hdr, "Content-Length") == NULL) { - http_add_header_field(&res->hdr, "Transfer-Encoding", "chunked"); - } - } + return 1; } else if (ctx->conf->type == CONFIG_TYPE_REVERSE_PROXY) { - info("Reverse proxy for " BLD_STR "%s:%i" CLR_STR, ctx->conf->proxy.hostname, ctx->conf->proxy.port); - http_remove_header_field(&res->hdr, "Date", HTTP_REMOVE_ALL); - http_remove_header_field(&res->hdr, "Server", HTTP_REMOVE_ALL); - - ret = proxy_init(req, res, status, ctx->conf, client, ctx, &custom_status, err_msg); - ctx->use_proxy = (ret == 0); - - if (res->status->code == 101) { - const char *connection = http_get_header_field(&res->hdr, "Connection"); - const char *upgrade = http_get_header_field(&res->hdr, "Upgrade"); - if (connection != NULL && upgrade != NULL && - (strstr(connection, "upgrade") != NULL || strstr(connection, "Upgrade") != NULL) && - strcmp(upgrade, "websocket") == 0) - { - const char *ws_accept = http_get_header_field(&res->hdr, "Sec-WebSocket-Accept"); - if (ws_calc_accept_key(status->ws_key, buf0) == 0) { - ctx->use_proxy = (strcmp(buf0, ws_accept) == 0) ? 2 : 1; - } - } else { - status->status = 101; - status->origin = INTERNAL; - res->status = http_get_status(501); - } - } - - // Let 300 be formatted by origin server - if (ctx->use_proxy && res->status->code >= 301 && res->status->code < 600) { - const char *content_type = http_get_header_field(&res->hdr, "Content-Type"); - const char *content_length_f = http_get_header_field(&res->hdr, "Content-Length"); - const char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding"); - if (content_encoding == NULL && content_type != NULL && content_length_f != NULL && strncmp(content_type, "text/html", 9) == 0) { - long content_len = strtol(content_length_f, NULL, 10); - if (content_len <= sizeof(msg_content) - 1) { - if (status->status != 101) { - status->status = res->status->code; - status->origin = res->status->code >= 400 ? SERVER : NONE; - } - ctx->use_proxy = 0; - proxy_dump(msg_content, content_len); - } - } - } - - /* - char *content_encoding = http_get_header_field(&res->hdr, "Content-Encoding"); - if (use_proxy && content_encoding == NULL) { - int http_comp = http_get_compression(&req, &res); - if (http_comp & COMPRESS_BR) { - use_proxy |= PROXY_COMPRESS_BR; - } else if (http_comp & COMPRESS_GZ) { - use_proxy |= PROXY_COMPRESS_GZ; - } - } - - char *transfer_encoding = http_get_header_field(&res->hdr, "Transfer-Encoding"); - int chunked = transfer_encoding != NULL && strcmp(transfer_encoding, "chunked") == 0; - http_remove_header_field(&res->hdr, "Transfer-Encoding", HTTP_REMOVE_ALL); - ret = sprintf(buf0, "%s%s%s", - (use_proxy & PROXY_COMPRESS_BR) ? "br" : - ((use_proxy & PROXY_COMPRESS_GZ) ? "gzip" : ""), - ((use_proxy & PROXY_COMPRESS) && chunked) ? ", " : "", - chunked ? "chunked" : ""); - if (ret > 0) { - http_add_header_field(&res->hdr, "Transfer-Encoding", buf0); - } - */ + return 2; } else { error("Unknown host type: %i", ctx->conf->type); res->status = http_get_status(501); diff --git a/src/worker/responder.c b/src/worker/responder.c index 694eea1..3c53bdd 100644 --- a/src/worker/responder.c +++ b/src/worker/responder.c @@ -36,23 +36,17 @@ void responder_func(client_ctx_t *ctx) { } static int responder(client_ctx_t *ctx) { - sock *client = &ctx->socket; - long ret = 0; - - char buf0[1024]; - char msg_buf[8192], msg_pre_buf_1[4096], msg_pre_buf_2[4096], err_msg[256]; - char msg_content[1024]; - char buffer[CHUNK_SIZE]; - - msg_buf[0] = 0; - err_msg[0] = 0; - msg_content[0] = 0; - - fastcgi_cnx_t fcgi_cnx = {.socket = 0, .req_id = 0, .ctx = ctx}; - http_req *req = &ctx->req; http_res *res = &ctx->res; - http_status_ctx status = {.status = 0, .origin = NONE, .ws_key = NULL}; + sock *client = &ctx->socket; + http_status_ctx *status = &ctx->status; + fastcgi_cnx_t *fcgi_cnx = &ctx->fcgi_cnx; + char *err_msg = ctx->err_msg; + + long ret = 0; + char buf0[1024]; + char msg_pre_buf_1[4096], msg_pre_buf_2[4096]; + char buffer[CHUNK_SIZE]; if (!ctx->use_proxy) { if (ctx->conf != NULL && ctx->conf->type == CONFIG_TYPE_LOCAL && ctx->uri.is_static && res->status->code == 405) { @@ -76,45 +70,45 @@ static int responder(client_ctx_t *ctx) { const http_doc_info *info = http_get_status_info(res->status); const http_status_msg *http_msg = http_get_error_msg(res->status); - if (msg_content[0] == 0) { + if (ctx->msg_content[0] == 0) { if (res->status->code >= 300 && res->status->code < 400) { const char *location = http_get_header_field(&res->hdr, "Location"); if (location != NULL) { - snprintf(msg_content, sizeof(msg_content), "\n", location, location); + snprintf(ctx->msg_content, sizeof(ctx->msg_content), "\n", location, location); } } - } else if (strncmp(msg_content, "", 15) == 0 || strncmp(msg_content, "msg_content, "", 15) == 0 || strncmp(ctx->msg_content, "msg_content[0] = 0; // TODO let relevant information pass? } char *proxy_doc = ""; if (ctx->conf != NULL && ctx->conf->type == CONFIG_TYPE_REVERSE_PROXY) { - const http_status *status_hdr = http_get_status(status.status); + const http_status *status_hdr = http_get_status(status->status); char stat_str[8]; - sprintf(stat_str, "%03i", status.status); + sprintf(stat_str, "%03i", status->status); sprintf(msg_pre_buf_2, http_proxy_document, " success", - (status.origin == CLIENT_REQ) ? " error" : " success", - (status.origin == INTERNAL) ? " error" : " success", - (status.origin == SERVER_REQ) ? " error" : (status.status == 0 ? "" : " success"), - (status.origin == CLIENT_RES) ? " error" : " success", - (status.origin == SERVER) ? " error" : (status.status == 0 ? "" : " success"), - (status.origin == SERVER_RES) ? " error" : (status.status == 0 ? "" : " success"), - (status.origin == INTERNAL) ? " error" : " success", - (status.origin == INTERNAL || status.origin == SERVER) ? " error" : " success", + (status->origin == CLIENT_REQ) ? " error" : " success", + (status->origin == INTERNAL) ? " error" : " success", + (status->origin == SERVER_REQ) ? " error" : (status->status == 0 ? "" : " success"), + (status->origin == CLIENT_RES) ? " error" : " success", + (status->origin == SERVER) ? " error" : (status->status == 0 ? "" : " success"), + (status->origin == SERVER_RES) ? " error" : (status->status == 0 ? "" : " success"), + (status->origin == INTERNAL) ? " error" : " success", + (status->origin == INTERNAL || status->origin == SERVER) ? " error" : " success", res->status->code, res->status->msg, - (status.status == 0) ? "???" : stat_str, + (status->status == 0) ? "???" : stat_str, (status_hdr != NULL) ? status_hdr->msg : "", ctx->req_host); proxy_doc = msg_pre_buf_2; } sprintf(msg_pre_buf_1, info->doc, res->status->code, res->status->msg, http_msg != NULL ? http_msg->msg : "", err_msg[0] != 0 ? err_msg : ""); - ctx->content_length = snprintf(msg_buf, sizeof(msg_buf), http_default_document, res->status->code, + ctx->content_length = snprintf(ctx->msg_buf, sizeof(ctx->msg_buf), http_default_document, res->status->code, res->status->msg, msg_pre_buf_1, info->mode, info->icon, info->color, ctx->req_host, - proxy_doc, msg_content[0] != 0 ? msg_content : ""); + proxy_doc, ctx->msg_content[0] != 0 ? ctx->msg_content : ""); } if (ctx->content_length >= 0) { sprintf(buf0, "%li", ctx->content_length); @@ -163,8 +157,8 @@ static int responder(client_ctx_t *ctx) { // default response unsigned long snd_len = 0; unsigned long len; - if (msg_buf[0] != 0) { - ret = sock_send(client, msg_buf, ctx->content_length, 0); + if (ctx->msg_buf[0] != 0) { + ret = sock_send(client, ctx->msg_buf, ctx->content_length, 0); if (ret <= 0) { error("Unable to send: %s", sock_strerror(client)); } @@ -187,7 +181,7 @@ static int responder(client_ctx_t *ctx) { int chunked = (transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL); int flags = (chunked ? FASTCGI_CHUNKED : 0) | (ctx->use_fastcgi & (FASTCGI_COMPRESS | FASTCGI_COMPRESS_HOLD)); - ret = fastcgi_send(&fcgi_cnx, client, flags); + ret = fastcgi_send(fcgi_cnx, client, flags); } else if (ctx->use_proxy) { const char *transfer_encoding = http_get_header_field(&res->hdr, "Transfer-Encoding"); int chunked = transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL; @@ -217,9 +211,9 @@ static int responder(client_ctx_t *ctx) { info("Transfer complete: %s", format_duration(micros, buf0)); uri_free(&ctx->uri); - if (fcgi_cnx.socket != 0) { - close(fcgi_cnx.socket); - fcgi_cnx.socket = 0; + if (fcgi_cnx->socket != 0) { + close(fcgi_cnx->socket); + fcgi_cnx->socket = 0; } http_free_req(req); http_free_res(res); diff --git a/src/workers.c b/src/workers.c index 8a9b4ea..d83f468 100644 --- a/src/workers.c +++ b/src/workers.c @@ -11,19 +11,25 @@ #include "worker/func.h" -static mpmc_t tcp_acceptor_ctx, tcp_closer_ctx, request_handler_ctx, responder_ctx, fastcgi_handler_cxt; +static mpmc_t tcp_acceptor_ctx, tcp_closer_ctx, request_handler_ctx, responder_ctx, + local_handler_ctx, fastcgi_handler_cxt, proxy_handler_ctx; int workers_init(void) { mpmc_init(&tcp_acceptor_ctx, 8, 64, (void (*)(void *)) tcp_acceptor_func, "tcp_a"); mpmc_init(&tcp_closer_ctx, 8, 64, (void (*)(void *)) tcp_closer_func, "tcp_c"); mpmc_init(&request_handler_ctx, 16, 64, (void (*)(void *)) request_handler_func, "req"); mpmc_init(&responder_ctx, 16, 64, (void (*)(void *)) responder_func, "res"); + mpmc_init(&local_handler_ctx, 16, 64, (void (*)(void *)) local_handler_func, "local"); mpmc_init(&fastcgi_handler_cxt, 16, 64, (void (*)(void *)) fastcgi_handler_func, "fcgi"); + mpmc_init(&proxy_handler_ctx, 16, 64, (void (*)(void *)) proxy_handler_func, "proxy"); return -1; } void workers_stop(void) { mpmc_stop(&tcp_acceptor_ctx); + mpmc_stop(&local_handler_ctx); + mpmc_stop(&fastcgi_handler_cxt); + mpmc_stop(&proxy_handler_ctx); mpmc_stop(&request_handler_ctx); mpmc_stop(&responder_ctx); mpmc_stop(&tcp_closer_ctx); @@ -31,6 +37,9 @@ void workers_stop(void) { void workers_destroy(void) { mpmc_destroy(&tcp_acceptor_ctx); + mpmc_destroy(&local_handler_ctx); + mpmc_destroy(&fastcgi_handler_cxt); + mpmc_destroy(&proxy_handler_ctx); mpmc_destroy(&request_handler_ctx); mpmc_destroy(&responder_ctx); mpmc_destroy(&tcp_closer_ctx); @@ -52,7 +61,15 @@ int respond(client_ctx_t *ctx) { return mpmc_queue(&responder_ctx, ctx); } +int local_handle(client_ctx_t *ctx) { + return mpmc_queue(&local_handler_ctx, ctx); +} + int fastcgi_handle(client_ctx_t *ctx) { return mpmc_queue(&fastcgi_handler_cxt, ctx); } +int proxy_handle(client_ctx_t *ctx) { + return mpmc_queue(&proxy_handler_ctx, ctx); +} + diff --git a/src/workers.h b/src/workers.h index a355e55..f380cfa 100644 --- a/src/workers.h +++ b/src/workers.h @@ -25,6 +25,10 @@ int handle_request(client_ctx_t *ctx); int respond(client_ctx_t *ctx); +int local_handle(client_ctx_t *ctx); + int fastcgi_handle(client_ctx_t *ctx); +int proxy_handle(client_ctx_t *ctx); + #endif //SESIMOS_WORKERS_H