Delete client.h/.c
This commit is contained in:
8
Makefile
8
Makefile
@ -45,7 +45,7 @@ bin/lib/%.o: src/lib/%.c
|
|||||||
bin/worker/%.o: src/worker/%.c
|
bin/worker/%.o: src/worker/%.c
|
||||||
$(CC) -c -o $@ $(CFLAGS) $<
|
$(CC) -c -o $@ $(CFLAGS) $<
|
||||||
|
|
||||||
bin/sesimos: bin/server.o bin/client.o bin/logger.o bin/cache_handler.o bin/async.o \
|
bin/sesimos: bin/server.o bin/logger.o bin/cache_handler.o bin/async.o \
|
||||||
bin/worker/request_handler.o bin/worker/tcp_acceptor.o bin/worker/tcp_closer.o \
|
bin/worker/request_handler.o bin/worker/tcp_acceptor.o bin/worker/tcp_closer.o \
|
||||||
bin/lib/compress.o bin/lib/config.o bin/lib/fastcgi.o bin/lib/geoip.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/http.o bin/lib/http_static.o bin/lib/proxy.o bin/lib/sock.o bin/lib/uri.o \
|
||||||
@ -53,13 +53,9 @@ bin/sesimos: bin/server.o bin/client.o bin/logger.o bin/cache_handler.o bin/asyn
|
|||||||
$(CC) -o $@ $^ $(CFLAGS) $(LDFLAGS)
|
$(CC) -o $@ $^ $(CFLAGS) $(LDFLAGS)
|
||||||
|
|
||||||
|
|
||||||
bin/server.o: src/server.h src/defs.h src/client.h src/cache_handler.h src/lib/config.h src/lib/sock.h \
|
bin/server.o: src/server.h src/defs.h src/cache_handler.h src/lib/config.h src/lib/sock.h \
|
||||||
src/lib/proxy.h src/lib/geoip.h src/lib/utils.h src/logger.h
|
src/lib/proxy.h src/lib/geoip.h src/lib/utils.h src/logger.h
|
||||||
|
|
||||||
bin/client.o: src/client.h src/defs.h src/server.h src/lib/utils.h src/lib/config.h src/lib/sock.h \
|
|
||||||
src/lib/http.h src/lib/proxy.h src/lib/fastcgi.h src/cache_handler.h src/lib/geoip.h src/lib/compress.h \
|
|
||||||
src/lib/websocket.h src/logger.h
|
|
||||||
|
|
||||||
bin/logger.o: src/logger.h
|
bin/logger.o: src/logger.h
|
||||||
|
|
||||||
bin/cache_handler.o: src/cache_handler.h src/lib/utils.h src/lib/uri.h src/lib/compress.h src/logger.h
|
bin/cache_handler.o: src/cache_handler.h src/lib/utils.h src/lib/uri.h src/lib/compress.h src/logger.h
|
||||||
|
|||||||
844
src/client.c
844
src/client.c
@ -1,844 +0,0 @@
|
|||||||
/**
|
|
||||||
* sesimos - secure, simple, modern web server
|
|
||||||
* @brief Client connection and request handlers
|
|
||||||
* @file src/client.c
|
|
||||||
* @author Lorenz Stechauner
|
|
||||||
* @date 2020-12-03
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "defs.h"
|
|
||||||
#include "client.h"
|
|
||||||
#include "server.h"
|
|
||||||
#include "logger.h"
|
|
||||||
|
|
||||||
#include "lib/utils.h"
|
|
||||||
#include "lib/config.h"
|
|
||||||
#include "lib/sock.h"
|
|
||||||
#include "lib/http.h"
|
|
||||||
#include "lib/proxy.h"
|
|
||||||
#include "lib/fastcgi.h"
|
|
||||||
#include "cache_handler.h"
|
|
||||||
#include "lib/geoip.h"
|
|
||||||
#include "lib/compress.h"
|
|
||||||
#include "lib/websocket.h"
|
|
||||||
|
|
||||||
#include <string.h>
|
|
||||||
#include <errno.h>
|
|
||||||
#include <unistd.h>
|
|
||||||
#include <openssl/ssl.h>
|
|
||||||
#include <openssl/err.h>
|
|
||||||
#include <signal.h>
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
|
|
||||||
|
|
||||||
static const char *color_table[] = {"\x1B[31m", "\x1B[32m", "\x1B[33m", "\x1B[34m", "\x1B[35m", "\x1B[36m"};
|
|
||||||
|
|
||||||
host_config_t *get_host_config(const char *host) {
|
|
||||||
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
|
||||||
host_config_t *hc = &config.hosts[i];
|
|
||||||
if (hc->type == CONFIG_TYPE_UNSET) break;
|
|
||||||
if (strcmp(hc->name, host) == 0) return hc;
|
|
||||||
if (hc->name[0] == '*' && hc->name[1] == '.') {
|
|
||||||
const char *pos = strstr(host, hc->name + 1);
|
|
||||||
if (pos != NULL && strlen(pos) == strlen(hc->name + 1)) return hc;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
/*
|
|
||||||
void client_terminate(int _) {
|
|
||||||
server_keep_alive = 0;
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
int client_request_handler(client_ctx_t *cctx) {
|
|
||||||
sock *client = &cctx->socket;
|
|
||||||
struct timespec begin, end;
|
|
||||||
long ret;
|
|
||||||
|
|
||||||
char buf0[1024], buf1[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];
|
|
||||||
char host[256];
|
|
||||||
const char *host_ptr, *hdr_connection;
|
|
||||||
char log_req_prefix[512];
|
|
||||||
|
|
||||||
msg_buf[0] = 0;
|
|
||||||
err_msg[0] = 0;
|
|
||||||
msg_content[0] = 0;
|
|
||||||
|
|
||||||
host_config_t *conf = NULL;
|
|
||||||
FILE *file = NULL;
|
|
||||||
|
|
||||||
long content_length = 0;
|
|
||||||
int accept_if_modified_since = 0;
|
|
||||||
int use_fastcgi = 0;
|
|
||||||
int use_proxy = 0;
|
|
||||||
int p_len;
|
|
||||||
|
|
||||||
fastcgi_conn fcgi_conn = {.socket = 0, .req_id = 0, .ctx = cctx};
|
|
||||||
http_status custom_status;
|
|
||||||
|
|
||||||
http_res res = {.version = "1.1", .status = http_get_status(501), .hdr.field_num = 0, .hdr.last_field_num = -1};
|
|
||||||
http_status_ctx ctx = {.status = 0, .origin = NONE, .ws_key = NULL};
|
|
||||||
|
|
||||||
clock_gettime(CLOCK_MONOTONIC, &begin);
|
|
||||||
|
|
||||||
ret = sock_poll_read(&client, NULL, NULL, 1, NULL, NULL, CLIENT_TIMEOUT * 1000);
|
|
||||||
|
|
||||||
http_add_header_field(&res.hdr, "Date", http_get_date(buf0, sizeof(buf0)));
|
|
||||||
http_add_header_field(&res.hdr, "Server", SERVER_STR);
|
|
||||||
if (ret <= 0) {
|
|
||||||
if (errno != 0) return 1;
|
|
||||||
|
|
||||||
cctx->c_keep_alive = 0;
|
|
||||||
res.status = http_get_status(408);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
clock_gettime(CLOCK_MONOTONIC, &begin);
|
|
||||||
|
|
||||||
http_req req;
|
|
||||||
ret = http_receive_request(client, &req);
|
|
||||||
if (ret != 0) {
|
|
||||||
cctx->c_keep_alive = 0;
|
|
||||||
if (ret < 0) {
|
|
||||||
goto abort;
|
|
||||||
} else if (ret == 1) {
|
|
||||||
sprintf(err_msg, "Unable to parse http header: Invalid header format.");
|
|
||||||
} else if (ret == 2) {
|
|
||||||
sprintf(err_msg, "Unable to parse http header: Invalid method.");
|
|
||||||
} else if (ret == 3) {
|
|
||||||
sprintf(err_msg, "Unable to parse http header: Invalid version.");
|
|
||||||
} else if (ret == 4) {
|
|
||||||
sprintf(err_msg, "Unable to parse http header: Header contains illegal characters.");
|
|
||||||
} else if (ret == 5) {
|
|
||||||
sprintf(err_msg, "Unable to parse http header: End of header not found.");
|
|
||||||
}
|
|
||||||
res.status = http_get_status(400);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
hdr_connection = http_get_header_field(&req.hdr, "Connection");
|
|
||||||
cctx->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");
|
|
||||||
if (host_ptr != NULL && strlen(host_ptr) > 255) {
|
|
||||||
host[0] = 0;
|
|
||||||
res.status = http_get_status(400);
|
|
||||||
sprintf(err_msg, "Host header field is too long.");
|
|
||||||
goto respond;
|
|
||||||
} else if (host_ptr == NULL || strchr(host_ptr, '/') != NULL) {
|
|
||||||
if (strchr(cctx->addr, ':') == NULL) {
|
|
||||||
strcpy(host, cctx->addr);
|
|
||||||
} else {
|
|
||||||
sprintf(host, "[%s]", cctx->addr);
|
|
||||||
}
|
|
||||||
res.status = http_get_status(400);
|
|
||||||
sprintf(err_msg, "The client provided no or an invalid Host header field.");
|
|
||||||
goto respond;
|
|
||||||
} else {
|
|
||||||
strcpy(host, host_ptr);
|
|
||||||
}
|
|
||||||
|
|
||||||
logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, host, CLR_STR, cctx->log_prefix);
|
|
||||||
info(BLD_STR "%s %s", req.method, req.uri);
|
|
||||||
|
|
||||||
conf = get_host_config(host);
|
|
||||||
if (conf == NULL) {
|
|
||||||
info("Unknown host, redirecting to default");
|
|
||||||
res.status = http_get_status(307);
|
|
||||||
sprintf(buf0, "https://%s%s", DEFAULT_HOST, req.uri);
|
|
||||||
http_add_header_field(&res.hdr, "Location", buf0);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
http_uri uri;
|
|
||||||
unsigned char dir_mode = (conf->type == CONFIG_TYPE_LOCAL ? conf->local.dir_mode : URI_DIR_MODE_NO_VALIDATION);
|
|
||||||
ret = uri_init(&uri, conf->local.webroot, req.uri, dir_mode);
|
|
||||||
if (ret != 0) {
|
|
||||||
if (ret == 1) {
|
|
||||||
sprintf(err_msg, "Invalid URI: has to start with slash.");
|
|
||||||
res.status = http_get_status(400);
|
|
||||||
} else if (ret == 2) {
|
|
||||||
sprintf(err_msg, "Invalid URI: contains relative path change (/../).");
|
|
||||||
res.status = http_get_status(400);
|
|
||||||
} else if (ret == 3) {
|
|
||||||
sprintf(err_msg, "The specified webroot directory does not exist.");
|
|
||||||
res.status = http_get_status(404);
|
|
||||||
} else {
|
|
||||||
res.status = http_get_status(500);
|
|
||||||
}
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (dir_mode != URI_DIR_MODE_NO_VALIDATION) {
|
|
||||||
ssize_t size = sizeof(buf0);
|
|
||||||
url_decode(req.uri, buf0, &size);
|
|
||||||
int change_proto = strncmp(uri.uri, "/.well-known/", 13) != 0 && !client->enc;
|
|
||||||
if (strcmp(uri.uri, buf0) != 0 || change_proto) {
|
|
||||||
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", host, buf0);
|
|
||||||
if (p_len < 0 || p_len >= sizeof(buf1)) {
|
|
||||||
res.status = http_get_status(500);
|
|
||||||
error("Header field 'Location' too long");
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
http_add_header_field(&res.hdr, "Location", buf1);
|
|
||||||
} else {
|
|
||||||
http_add_header_field(&res.hdr, "Location", buf0);
|
|
||||||
}
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
} else if (!client->enc) {
|
|
||||||
res.status = http_get_status(308);
|
|
||||||
sprintf(buf0, "https://%s%s", host, req.uri);
|
|
||||||
http_add_header_field(&res.hdr, "Location", buf0);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (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");
|
|
||||||
|
|
||||||
content_length = snprintf(msg_buf, sizeof(msg_buf) - 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];
|
|
||||||
content_length += snprintf(msg_buf + content_length, sizeof(msg_buf) - content_length, "%s: %s\r\n", http_field_get_name(f), http_field_get_value(f));
|
|
||||||
}
|
|
||||||
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
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.");
|
|
||||||
goto respond;
|
|
||||||
} 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.");
|
|
||||||
goto respond;
|
|
||||||
} 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.");
|
|
||||||
goto respond;
|
|
||||||
} else if (uri.filename == NULL || (strlen(uri.pathinfo) > 0 && (int) uri.is_static)) {
|
|
||||||
res.status = http_get_status(404);
|
|
||||||
goto respond;
|
|
||||||
} else if (strlen(uri.pathinfo) != 0 && conf->local.dir_mode != URI_DIR_MODE_INFO) {
|
|
||||||
res.status = http_get_status(404);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
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);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
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");
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
cache_init_uri(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) {
|
|
||||||
file = fopen(uri.meta->filename_comp_br, "rb");
|
|
||||||
if (file == NULL) {
|
|
||||||
cache_mark_dirty(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) {
|
|
||||||
file = fopen(uri.meta->filename_comp_gz, "rb");
|
|
||||||
if (file == NULL) {
|
|
||||||
cache_mark_dirty(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);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
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);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
range += 6;
|
|
||||||
char *ptr = strchr(range, '-');
|
|
||||||
if (ptr == NULL) {
|
|
||||||
res.status = http_get_status(416);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
file = fopen(uri.filename, "rb");
|
|
||||||
fseek(file, 0, SEEK_END);
|
|
||||||
unsigned long file_len = ftell(file);
|
|
||||||
fseek(file, 0, SEEK_SET);
|
|
||||||
if (file_len == 0) {
|
|
||||||
content_length = 0;
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
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);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
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(file, num1, SEEK_SET);
|
|
||||||
content_length = num2 - num1 + 1;
|
|
||||||
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (file == NULL) {
|
|
||||||
file = fopen(uri.filename, "rb");
|
|
||||||
}
|
|
||||||
|
|
||||||
fseek(file, 0, SEEK_END);
|
|
||||||
content_length = ftell(file);
|
|
||||||
fseek(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);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
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_conn, mode, 0 /* TODO */, cctx->req_num, client, &req, &uri) != 0) {
|
|
||||||
res.status = http_get_status(503);
|
|
||||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
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_conn, client, client_content_len);
|
|
||||||
} else if (client_transfer_encoding != NULL && strstr(client_transfer_encoding, "chunked") != NULL) {
|
|
||||||
ret = fastcgi_receive_chunked(&fcgi_conn, client);
|
|
||||||
} else {
|
|
||||||
ret = 0;
|
|
||||||
}
|
|
||||||
if (ret != 0) {
|
|
||||||
if (ret < 0) {
|
|
||||||
goto abort;
|
|
||||||
} else {
|
|
||||||
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
|
||||||
}
|
|
||||||
res.status = http_get_status(502);
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
fastcgi_close_stdin(&fcgi_conn);
|
|
||||||
|
|
||||||
ret = fastcgi_header(&fcgi_conn, &res, err_msg);
|
|
||||||
if (ret != 0) {
|
|
||||||
if (ret < 0) goto abort;
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
const char *status = http_get_header_field(&res.hdr, "Status");
|
|
||||||
if (status != NULL) {
|
|
||||||
int status_code = (int) strtoul(status, 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 + 4);
|
|
||||||
res.status = &custom_status;
|
|
||||||
} else if (res.status == NULL) {
|
|
||||||
res.status = http_get_status(500);
|
|
||||||
sprintf(err_msg, "The status code was set to an invalid or unknown value.");
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const char *content_length_f = http_get_header_field(&res.hdr, "Content-Length");
|
|
||||||
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 &&
|
|
||||||
content_length != -1 &&
|
|
||||||
content_length <= sizeof(msg_content) - 1)
|
|
||||||
{
|
|
||||||
fastcgi_dump(&fcgi_conn, msg_content, sizeof(msg_content));
|
|
||||||
goto respond;
|
|
||||||
}
|
|
||||||
|
|
||||||
use_fastcgi = 1;
|
|
||||||
|
|
||||||
if (content_length != -1 && content_length < 1024000) {
|
|
||||||
use_fastcgi |= FASTCGI_COMPRESS_HOLD;
|
|
||||||
}
|
|
||||||
|
|
||||||
content_length = -1;
|
|
||||||
|
|
||||||
int http_comp = http_get_compression(&req, &res);
|
|
||||||
if (http_comp & COMPRESS) {
|
|
||||||
if (http_comp & COMPRESS_BR) {
|
|
||||||
use_fastcgi |= FASTCGI_COMPRESS_BR;
|
|
||||||
sprintf(buf0, "br");
|
|
||||||
} else if (http_comp & COMPRESS_GZ) {
|
|
||||||
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");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if (conf->type == CONFIG_TYPE_REVERSE_PROXY) {
|
|
||||||
info("Reverse proxy for " BLD_STR "%s:%i" CLR_STR, conf->proxy.hostname, 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, &ctx, conf, client, cctx, &custom_status, err_msg);
|
|
||||||
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(ctx.ws_key, buf0) == 0) {
|
|
||||||
use_proxy = (strcmp(buf0, ws_accept) == 0) ? 2 : 1;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
ctx.status = 101;
|
|
||||||
ctx.origin = INTERNAL;
|
|
||||||
res.status = http_get_status(501);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Let 300 be formatted by origin server
|
|
||||||
if (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 (ctx.status != 101) {
|
|
||||||
ctx.status = res.status->code;
|
|
||||||
ctx.origin = res.status->code >= 400 ? SERVER : NONE;
|
|
||||||
}
|
|
||||||
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);
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
} else {
|
|
||||||
error("Unknown host type: %i", conf->type);
|
|
||||||
res.status = http_get_status(501);
|
|
||||||
}
|
|
||||||
|
|
||||||
respond:
|
|
||||||
if (!use_proxy) {
|
|
||||||
if (conf != NULL && conf->type == CONFIG_TYPE_LOCAL && uri.is_static && res.status->code == 405) {
|
|
||||||
http_add_header_field(&res.hdr, "Allow", "GET, HEAD, TRACE");
|
|
||||||
}
|
|
||||||
if (http_get_header_field(&res.hdr, "Accept-Ranges") == NULL) {
|
|
||||||
http_add_header_field(&res.hdr, "Accept-Ranges", "none");
|
|
||||||
}
|
|
||||||
if (!use_fastcgi && file == NULL) {
|
|
||||||
http_remove_header_field(&res.hdr, "Date", HTTP_REMOVE_ALL);
|
|
||||||
http_remove_header_field(&res.hdr, "Server", HTTP_REMOVE_ALL);
|
|
||||||
http_remove_header_field(&res.hdr, "Cache-Control", HTTP_REMOVE_ALL);
|
|
||||||
http_remove_header_field(&res.hdr, "Content-Type", HTTP_REMOVE_ALL);
|
|
||||||
http_remove_header_field(&res.hdr, "Content-Encoding", HTTP_REMOVE_ALL);
|
|
||||||
http_add_header_field(&res.hdr, "Date", http_get_date(buf0, sizeof(buf0)));
|
|
||||||
http_add_header_field(&res.hdr, "Server", SERVER_STR);
|
|
||||||
http_add_header_field(&res.hdr, "Cache-Control", "no-cache");
|
|
||||||
http_add_header_field(&res.hdr, "Content-Type", "text/html; charset=UTF-8");
|
|
||||||
|
|
||||||
// TODO list Locations on 3xx Redirects
|
|
||||||
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 (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), "<ul>\n\t<li><a href=\"%s\">%s</a></li>\n</ul>\n", location, location);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else if (strncmp(msg_content, "<!DOCTYPE html>", 15) == 0 || strncmp(msg_content, "<html", 5) == 0) {
|
|
||||||
msg_content[0] = 0;
|
|
||||||
// TODO let relevant information pass?
|
|
||||||
}
|
|
||||||
|
|
||||||
char *proxy_doc = "";
|
|
||||||
if (conf != NULL && conf->type == CONFIG_TYPE_REVERSE_PROXY) {
|
|
||||||
const http_status *status = http_get_status(ctx.status);
|
|
||||||
char stat_str[8];
|
|
||||||
sprintf(stat_str, "%03i", ctx.status);
|
|
||||||
sprintf(msg_pre_buf_2, http_proxy_document,
|
|
||||||
" success",
|
|
||||||
(ctx.origin == CLIENT_REQ) ? " error" : " success",
|
|
||||||
(ctx.origin == INTERNAL) ? " error" : " success",
|
|
||||||
(ctx.origin == SERVER_REQ) ? " error" : (ctx.status == 0 ? "" : " success"),
|
|
||||||
(ctx.origin == CLIENT_RES) ? " error" : " success",
|
|
||||||
(ctx.origin == SERVER) ? " error" : (ctx.status == 0 ? "" : " success"),
|
|
||||||
(ctx.origin == SERVER_RES) ? " error" : (ctx.status == 0 ? "" : " success"),
|
|
||||||
(ctx.origin == INTERNAL) ? " error" : " success",
|
|
||||||
(ctx.origin == INTERNAL || ctx.origin == SERVER) ? " error" : " success",
|
|
||||||
res.status->code,
|
|
||||||
res.status->msg,
|
|
||||||
(ctx.status == 0) ? "???" : stat_str,
|
|
||||||
(status != NULL) ? status->msg : "",
|
|
||||||
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 : "");
|
|
||||||
content_length = snprintf(msg_buf, sizeof(msg_buf), http_default_document, res.status->code,
|
|
||||||
res.status->msg, msg_pre_buf_1, info->mode, info->icon, info->color, host,
|
|
||||||
proxy_doc, msg_content[0] != 0 ? msg_content : "");
|
|
||||||
}
|
|
||||||
if (content_length >= 0) {
|
|
||||||
sprintf(buf0, "%li", content_length);
|
|
||||||
http_remove_header_field(&res.hdr, "Content-Length", HTTP_REMOVE_ALL);
|
|
||||||
http_add_header_field(&res.hdr, "Content-Length", buf0);
|
|
||||||
} else if (http_get_header_field(&res.hdr, "Transfer-Encoding") == NULL) {
|
|
||||||
cctx->s_keep_alive = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int close_proxy = 0;
|
|
||||||
if (use_proxy != 2) {
|
|
||||||
const char *conn = http_get_header_field(&res.hdr, "Connection");
|
|
||||||
close_proxy = (conn == NULL || (strstr(conn, "keep-alive") == NULL && strstr(conn, "Keep-Alive") == NULL));
|
|
||||||
http_remove_header_field(&res.hdr, "Connection", HTTP_REMOVE_ALL);
|
|
||||||
http_remove_header_field(&res.hdr, "Keep-Alive", HTTP_REMOVE_ALL);
|
|
||||||
if (cctx->s_keep_alive && cctx->c_keep_alive) {
|
|
||||||
http_add_header_field(&res.hdr, "Connection", "keep-alive");
|
|
||||||
sprintf(buf0, "timeout=%i, max=%i", CLIENT_TIMEOUT, REQ_PER_CONNECTION);
|
|
||||||
http_add_header_field(&res.hdr, "Keep-Alive", buf0);
|
|
||||||
} else {
|
|
||||||
http_add_header_field(&res.hdr, "Connection", "close");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
http_send_response(client, &res);
|
|
||||||
clock_gettime(CLOCK_MONOTONIC, &end);
|
|
||||||
const char *location = http_get_header_field(&res.hdr, "Location");
|
|
||||||
unsigned long micros = (end.tv_nsec - begin.tv_nsec) / 1000 + (end.tv_sec - begin.tv_sec) * 1000000;
|
|
||||||
info("%s%s%03i %s%s%s (%s)%s", http_get_status_color(res.status), use_proxy ? "-> " : "", res.status->code,
|
|
||||||
res.status->msg, location != NULL ? " -> " : "", location != NULL ? location : "",
|
|
||||||
format_duration(micros, buf0), CLR_STR);
|
|
||||||
|
|
||||||
// TODO access/error log file
|
|
||||||
|
|
||||||
if (use_proxy == 2) {
|
|
||||||
// WebSocket
|
|
||||||
info("Upgrading connection to WebSocket connection");
|
|
||||||
ret = ws_handle_connection(client, &proxy);
|
|
||||||
if (ret != 0) {
|
|
||||||
cctx->c_keep_alive = 0;
|
|
||||||
close_proxy = 1;
|
|
||||||
}
|
|
||||||
info("WebSocket connection closed");
|
|
||||||
} else if (strcmp(req.method, "HEAD") != 0) {
|
|
||||||
// default response
|
|
||||||
unsigned long snd_len = 0;
|
|
||||||
unsigned long len;
|
|
||||||
if (msg_buf[0] != 0) {
|
|
||||||
ret = sock_send(client, msg_buf, content_length, 0);
|
|
||||||
if (ret <= 0) {
|
|
||||||
error("Unable to send: %s", sock_strerror(client));
|
|
||||||
}
|
|
||||||
snd_len += ret;
|
|
||||||
} else if (file != NULL) {
|
|
||||||
while (snd_len < content_length) {
|
|
||||||
len = fread(buffer, 1, CHUNK_SIZE, file);
|
|
||||||
if (snd_len + len > content_length) {
|
|
||||||
len = content_length - snd_len;
|
|
||||||
}
|
|
||||||
ret = sock_send(client, buffer, len, feof(file) ? 0 : MSG_MORE);
|
|
||||||
if (ret <= 0) {
|
|
||||||
error("Unable to send: %s", sock_strerror(client));
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
snd_len += ret;
|
|
||||||
}
|
|
||||||
} else if (use_fastcgi) {
|
|
||||||
const char *transfer_encoding = http_get_header_field(&res.hdr, "Transfer-Encoding");
|
|
||||||
int chunked = (transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL);
|
|
||||||
|
|
||||||
int flags = (chunked ? FASTCGI_CHUNKED : 0) | (use_fastcgi & (FASTCGI_COMPRESS | FASTCGI_COMPRESS_HOLD));
|
|
||||||
ret = fastcgi_send(&fcgi_conn, client, flags);
|
|
||||||
} else if (use_proxy) {
|
|
||||||
const char *transfer_encoding = http_get_header_field(&res.hdr, "Transfer-Encoding");
|
|
||||||
int chunked = transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL;
|
|
||||||
|
|
||||||
const char *content_len = http_get_header_field(&res.hdr, "Content-Length");
|
|
||||||
unsigned long len_to_send = 0;
|
|
||||||
if (content_len != NULL) {
|
|
||||||
len_to_send = strtol(content_len, NULL, 10);
|
|
||||||
}
|
|
||||||
|
|
||||||
int flags = (chunked ? PROXY_CHUNKED : 0) | (use_proxy & PROXY_COMPRESS);
|
|
||||||
ret = proxy_send(client, len_to_send, flags);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (ret < 0) {
|
|
||||||
cctx->c_keep_alive = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (close_proxy && proxy.socket != 0) {
|
|
||||||
info(BLUE_STR "Closing proxy connection");
|
|
||||||
sock_close(&proxy);
|
|
||||||
}
|
|
||||||
|
|
||||||
clock_gettime(CLOCK_MONOTONIC, &end);
|
|
||||||
micros = (end.tv_nsec - begin.tv_nsec) / 1000 + (end.tv_sec - begin.tv_sec) * 1000000;
|
|
||||||
info("Transfer complete: %s", format_duration(micros, buf0));
|
|
||||||
|
|
||||||
uri_free(&uri);
|
|
||||||
abort:
|
|
||||||
if (fcgi_conn.socket != 0) {
|
|
||||||
shutdown(fcgi_conn.socket, SHUT_RDWR);
|
|
||||||
close(fcgi_conn.socket);
|
|
||||||
fcgi_conn.socket = 0;
|
|
||||||
}
|
|
||||||
http_free_req(&req);
|
|
||||||
http_free_res(&res);
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
int client_connection_handler(client_ctx_t *ctx, unsigned long client_num, const char *restrict log_conn_prefix, const char *restrict log_client_prefix) {
|
|
||||||
struct timespec begin, end;
|
|
||||||
int ret;
|
|
||||||
char buf[1024];
|
|
||||||
sock *client = &ctx->socket;
|
|
||||||
|
|
||||||
clock_gettime(CLOCK_MONOTONIC, &begin);
|
|
||||||
|
|
||||||
if (config.dns_server[0] != 0) {
|
|
||||||
sprintf(buf, "dig @%s +short +time=1 -x %s", config.dns_server, ctx->addr);
|
|
||||||
FILE *dig = popen(buf, "r");
|
|
||||||
if (dig == NULL) {
|
|
||||||
error("Unable to start dig: %s", strerror(errno));
|
|
||||||
goto dig_err;
|
|
||||||
}
|
|
||||||
unsigned long read = fread(buf, 1, sizeof(buf), dig);
|
|
||||||
ret = pclose(dig);
|
|
||||||
if (ret != 0) {
|
|
||||||
error("Dig terminated with exit code %i", ret);
|
|
||||||
goto dig_err;
|
|
||||||
}
|
|
||||||
char *ptr = memchr(buf, '\n', read);
|
|
||||||
if (ptr == buf || ptr == NULL) {
|
|
||||||
goto dig_err;
|
|
||||||
}
|
|
||||||
ptr[-1] = 0;
|
|
||||||
strncpy(ctx->host, buf, sizeof(ctx->host));
|
|
||||||
} else {
|
|
||||||
dig_err:
|
|
||||||
ctx->host[0] = 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx->cc[0] = 0;
|
|
||||||
geoip_lookup_country(&client->addr.sock, ctx->cc);
|
|
||||||
|
|
||||||
info("Connection accepted from %s %s%s%s[%s]", ctx->addr, ctx->host[0] != 0 ? "(" : "",
|
|
||||||
ctx->host[0] != 0 ? ctx->host : "", ctx->host[0] != 0 ? ") " : "",
|
|
||||||
ctx->cc[0] != 0 ? ctx->cc : "N/A");
|
|
||||||
|
|
||||||
struct timeval client_timeout = {.tv_sec = CLIENT_TIMEOUT, .tv_usec = 0};
|
|
||||||
if (setsockopt(client->socket, SOL_SOCKET, SO_RCVTIMEO, &client_timeout, sizeof(client_timeout)) == -1 ||
|
|
||||||
setsockopt(client->socket, SOL_SOCKET, SO_SNDTIMEO, &client_timeout, sizeof(client_timeout)) == -1)
|
|
||||||
{
|
|
||||||
error("Unable to set timeout for socket");
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (client->enc) {
|
|
||||||
client->ssl = SSL_new(client->ctx);
|
|
||||||
SSL_set_fd(client->ssl, client->socket);
|
|
||||||
SSL_set_accept_state(client->ssl);
|
|
||||||
|
|
||||||
ret = SSL_accept(client->ssl);
|
|
||||||
client->_last_ret = ret;
|
|
||||||
client->_errno = errno;
|
|
||||||
client->_ssl_error = ERR_get_error();
|
|
||||||
if (ret <= 0) {
|
|
||||||
error("Unable to perform handshake: %s", sock_strerror(client));
|
|
||||||
ret = -1;
|
|
||||||
goto close;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
int req_num = 0;
|
|
||||||
ctx->s_keep_alive = 1;
|
|
||||||
ctx->c_keep_alive = 1;
|
|
||||||
while (ctx->c_keep_alive && ctx->s_keep_alive && req_num < REQ_PER_CONNECTION) {
|
|
||||||
ret = client_request_handler(ctx);
|
|
||||||
logger_set_prefix(log_conn_prefix);
|
|
||||||
}
|
|
||||||
|
|
||||||
close:
|
|
||||||
sock_close(client);
|
|
||||||
|
|
||||||
if (proxy.socket != 0) {
|
|
||||||
info(BLUE_STR "Closing proxy connection");
|
|
||||||
sock_close(&proxy);
|
|
||||||
}
|
|
||||||
|
|
||||||
clock_gettime(CLOCK_MONOTONIC, &end);
|
|
||||||
unsigned long micros = (end.tv_nsec - begin.tv_nsec) / 1000 + (end.tv_sec - begin.tv_sec) * 1000000;
|
|
||||||
|
|
||||||
info("Connection closed (%s)", format_duration(micros, buf));
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
void *client_handler(client_ctx_t *ctx) {
|
|
||||||
struct sockaddr_in6 *server_addr;
|
|
||||||
struct sockaddr_storage server_addr_storage;
|
|
||||||
char log_client_prefix[256], log_conn_prefix[512];
|
|
||||||
|
|
||||||
logger_set_name("client");
|
|
||||||
|
|
||||||
//signal(SIGINT, client_terminate);
|
|
||||||
//signal(SIGTERM, client_terminate);
|
|
||||||
|
|
||||||
inet_ntop(ctx->socket.addr.ipv6.sin6_family, &ctx->socket.addr.ipv6.sin6_addr, ctx->_c_addr, sizeof(ctx->_c_addr));
|
|
||||||
if (strncmp(ctx->_c_addr, "::ffff:", 7) == 0) {
|
|
||||||
ctx->addr = ctx->_c_addr + 7;
|
|
||||||
} else {
|
|
||||||
ctx->addr = ctx->_c_addr;
|
|
||||||
}
|
|
||||||
|
|
||||||
socklen_t len = sizeof(server_addr_storage);
|
|
||||||
getsockname(ctx->socket.socket, (struct sockaddr *) &server_addr_storage, &len);
|
|
||||||
server_addr = (struct sockaddr_in6 *) &server_addr_storage;
|
|
||||||
inet_ntop(server_addr->sin6_family, (void *) &server_addr->sin6_addr, ctx->_s_addr, sizeof(ctx->_s_addr));
|
|
||||||
if (strncmp(ctx->_s_addr, "::ffff:", 7) == 0) {
|
|
||||||
ctx->s_addr = ctx->_s_addr + 7;
|
|
||||||
} else {
|
|
||||||
ctx->s_addr = ctx->_s_addr;
|
|
||||||
}
|
|
||||||
|
|
||||||
sprintf(log_client_prefix, "[%s%4i%s]%s[%*s][%5i]%s", (int) ctx->socket.enc ? HTTPS_STR : HTTP_STR,
|
|
||||||
ntohs(server_addr->sin6_port), CLR_STR, color_table[0], INET6_ADDRSTRLEN, ctx->addr,
|
|
||||||
ntohs(ctx->socket.addr.ipv6.sin6_port), CLR_STR);
|
|
||||||
|
|
||||||
sprintf(log_conn_prefix, "[%*s]%s", INET6_ADDRSTRLEN, ctx->s_addr, log_client_prefix);
|
|
||||||
logger_set_prefix(log_conn_prefix);
|
|
||||||
|
|
||||||
info("Started thread");
|
|
||||||
|
|
||||||
client_connection_handler(ctx, 0, log_conn_prefix, log_client_prefix);
|
|
||||||
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
32
src/client.h
32
src/client.h
@ -1,32 +0,0 @@
|
|||||||
/**
|
|
||||||
* sesimos - secure, simple, modern web server
|
|
||||||
* @brief Client connection and request handlers (header file)
|
|
||||||
* @file src/client.h
|
|
||||||
* @author Lorenz Stechauner
|
|
||||||
* @date 2022-08-16
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef SESIMOS_CLIENT_H
|
|
||||||
#define SESIMOS_CLIENT_H
|
|
||||||
|
|
||||||
#include "lib/config.h"
|
|
||||||
#include "lib/sock.h"
|
|
||||||
|
|
||||||
#include <arpa/inet.h>
|
|
||||||
|
|
||||||
typedef struct {
|
|
||||||
sock socket;
|
|
||||||
int req_num;
|
|
||||||
char *addr, *s_addr;
|
|
||||||
unsigned char in_use: 1, s_keep_alive:1, c_keep_alive:1;
|
|
||||||
char cc[3], host[256];
|
|
||||||
char log_prefix[512];
|
|
||||||
char _c_addr[INET6_ADDRSTRLEN + 1], _s_addr[INET6_ADDRSTRLEN + 1];
|
|
||||||
struct timespec begin, end;
|
|
||||||
} client_ctx_t;
|
|
||||||
|
|
||||||
host_config_t *get_host_config(const char *host);
|
|
||||||
|
|
||||||
int client_request_handler(client_ctx_t *cctx);
|
|
||||||
|
|
||||||
#endif //SESIMOS_CLIENT_H
|
|
||||||
@ -208,3 +208,16 @@ int config_load(const char *filename) {
|
|||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
host_config_t *get_host_config(const char *host) {
|
||||||
|
for (int i = 0; i < CONFIG_MAX_HOST_CONFIG; i++) {
|
||||||
|
host_config_t *hc = &config.hosts[i];
|
||||||
|
if (hc->type == CONFIG_TYPE_UNSET) break;
|
||||||
|
if (strcmp(hc->name, host) == 0) return hc;
|
||||||
|
if (hc->name[0] == '*' && hc->name[1] == '.') {
|
||||||
|
const char *pos = strstr(host, hc->name + 1);
|
||||||
|
if (pos != NULL && strlen(pos) == strlen(hc->name + 1)) return hc;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|||||||
@ -60,4 +60,6 @@ extern config_t config;
|
|||||||
|
|
||||||
int config_load(const char *filename);
|
int config_load(const char *filename);
|
||||||
|
|
||||||
|
host_config_t *get_host_config(const char *host);
|
||||||
|
|
||||||
#endif //SESIMOS_CONFIG_H
|
#endif //SESIMOS_CONFIG_H
|
||||||
|
|||||||
@ -12,7 +12,7 @@
|
|||||||
#include "include/fastcgi.h"
|
#include "include/fastcgi.h"
|
||||||
#include "http.h"
|
#include "http.h"
|
||||||
#include "uri.h"
|
#include "uri.h"
|
||||||
#include "../client.h"
|
#include "../server.h"
|
||||||
|
|
||||||
#define FASTCGI_CHUNKED 1
|
#define FASTCGI_CHUNKED 1
|
||||||
#define FASTCGI_COMPRESS_GZ 2
|
#define FASTCGI_COMPRESS_GZ 2
|
||||||
|
|||||||
@ -20,7 +20,7 @@
|
|||||||
|
|
||||||
#include "http.h"
|
#include "http.h"
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
#include "../client.h"
|
#include "../server.h"
|
||||||
|
|
||||||
extern sock proxy;
|
extern sock proxy;
|
||||||
|
|
||||||
|
|||||||
@ -8,7 +8,6 @@
|
|||||||
|
|
||||||
#include "defs.h"
|
#include "defs.h"
|
||||||
#include "server.h"
|
#include "server.h"
|
||||||
#include "client.h"
|
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "async.h"
|
#include "async.h"
|
||||||
#include "worker/tcp_acceptor.h"
|
#include "worker/tcp_acceptor.h"
|
||||||
@ -44,6 +43,8 @@ static SSL_CTX *contexts[CONFIG_MAX_CERT_CONFIG];
|
|||||||
|
|
||||||
static client_ctx_t clients[MAX_CLIENTS]; // TODO dynamic
|
static client_ctx_t clients[MAX_CLIENTS]; // TODO dynamic
|
||||||
|
|
||||||
|
static const char *color_table[] = {"\x1B[31m", "\x1B[32m", "\x1B[33m", "\x1B[34m", "\x1B[35m", "\x1B[36m"};
|
||||||
|
|
||||||
static int clean() {
|
static int clean() {
|
||||||
remove("/var/sesimos/server/cache");
|
remove("/var/sesimos/server/cache");
|
||||||
rmdir("/var/sesimos/server/");
|
rmdir("/var/sesimos/server/");
|
||||||
|
|||||||
15
src/server.h
15
src/server.h
@ -9,9 +9,12 @@
|
|||||||
#ifndef SESIMOS_SERVER_H
|
#ifndef SESIMOS_SERVER_H
|
||||||
#define SESIMOS_SERVER_H
|
#define SESIMOS_SERVER_H
|
||||||
|
|
||||||
|
#include "lib/sock.h"
|
||||||
|
|
||||||
#include <sys/time.h>
|
#include <sys/time.h>
|
||||||
#include <maxminddb.h>
|
#include <maxminddb.h>
|
||||||
#include <signal.h>
|
#include <signal.h>
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
|
||||||
#define NUM_SOCKETS 2
|
#define NUM_SOCKETS 2
|
||||||
#define LISTEN_BACKLOG 16
|
#define LISTEN_BACKLOG 16
|
||||||
@ -24,6 +27,18 @@
|
|||||||
#define CNX_HANDLER_WORKERS 8
|
#define CNX_HANDLER_WORKERS 8
|
||||||
#define REQ_HANDLER_WORKERS 16
|
#define REQ_HANDLER_WORKERS 16
|
||||||
|
|
||||||
|
typedef struct {
|
||||||
|
sock socket;
|
||||||
|
int req_num;
|
||||||
|
char *addr, *s_addr;
|
||||||
|
unsigned char in_use: 1, s_keep_alive:1, c_keep_alive:1;
|
||||||
|
char cc[3], host[256];
|
||||||
|
char log_prefix[512];
|
||||||
|
char _c_addr[INET6_ADDRSTRLEN + 1], _s_addr[INET6_ADDRSTRLEN + 1];
|
||||||
|
struct timespec begin, end;
|
||||||
|
} client_ctx_t;
|
||||||
|
|
||||||
extern volatile sig_atomic_t server_alive;
|
extern volatile sig_atomic_t server_alive;
|
||||||
|
|
||||||
|
|
||||||
#endif //SESIMOS_SERVER_H
|
#endif //SESIMOS_SERVER_H
|
||||||
|
|||||||
@ -1,15 +1,40 @@
|
|||||||
|
/**
|
||||||
|
* sesimos - secure, simple, modern web server
|
||||||
|
* @brief Client request handler
|
||||||
|
* @file src/worker/request_handler.c
|
||||||
|
* @author Lorenz Stechauner
|
||||||
|
* @date 2022-12-28
|
||||||
|
*/
|
||||||
|
|
||||||
#include "request_handler.h"
|
#include "request_handler.h"
|
||||||
#include "../logger.h"
|
|
||||||
#include "../lib/mpmc.h"
|
#include "../lib/mpmc.h"
|
||||||
#include "../lib/utils.h"
|
|
||||||
#include "tcp_closer.h"
|
#include "tcp_closer.h"
|
||||||
#include "../async.h"
|
#include "../async.h"
|
||||||
|
|
||||||
|
#include "../defs.h"
|
||||||
#include "../server.h"
|
#include "../server.h"
|
||||||
|
#include "../logger.h"
|
||||||
|
|
||||||
|
#include "../lib/utils.h"
|
||||||
|
#include "../lib/config.h"
|
||||||
|
#include "../lib/sock.h"
|
||||||
|
#include "../lib/http.h"
|
||||||
|
#include "../lib/proxy.h"
|
||||||
|
#include "../lib/fastcgi.h"
|
||||||
|
#include "../cache_handler.h"
|
||||||
|
#include "../lib/compress.h"
|
||||||
|
#include "../lib/websocket.h"
|
||||||
|
|
||||||
|
#include <string.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
#include <openssl/err.h>
|
||||||
|
#include <arpa/inet.h>
|
||||||
|
|
||||||
static mpmc_t mpmc_ctx;
|
static mpmc_t mpmc_ctx;
|
||||||
|
|
||||||
static void request_handler_func(client_ctx_t *ctx);
|
static void request_handler_func(client_ctx_t *ctx);
|
||||||
|
static void request_handler(client_ctx_t *ctx);
|
||||||
|
|
||||||
int request_handler_init(int n_workers, int buf_size) {
|
int request_handler_init(int n_workers, int buf_size) {
|
||||||
return mpmc_init(&mpmc_ctx, n_workers, buf_size, (void (*)(void *)) request_handler_func, "req");
|
return mpmc_init(&mpmc_ctx, n_workers, buf_size, (void (*)(void *)) request_handler_func, "req");
|
||||||
@ -28,7 +53,7 @@ void request_handler_destroy(void) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
static void request_handler_func(client_ctx_t *ctx) {
|
static void request_handler_func(client_ctx_t *ctx) {
|
||||||
client_request_handler(ctx);
|
request_handler(ctx);
|
||||||
|
|
||||||
if (ctx->c_keep_alive && ctx->s_keep_alive && ctx->req_num < REQ_PER_CONNECTION) {
|
if (ctx->c_keep_alive && ctx->s_keep_alive && ctx->req_num < REQ_PER_CONNECTION) {
|
||||||
async(ctx->socket.socket, POLLIN, 0, (void (*)(void *)) handle_request, ctx, (void (*)(void *)) tcp_close, ctx);
|
async(ctx->socket.socket, POLLIN, 0, (void (*)(void *)) handle_request, ctx, (void (*)(void *)) tcp_close, ctx);
|
||||||
@ -37,3 +62,666 @@ static void request_handler_func(client_ctx_t *ctx) {
|
|||||||
tcp_close(ctx);
|
tcp_close(ctx);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void request_handler(client_ctx_t *cctx) {
|
||||||
|
sock *client = &cctx->socket;
|
||||||
|
struct timespec begin, end;
|
||||||
|
long ret;
|
||||||
|
|
||||||
|
char buf0[1024], buf1[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];
|
||||||
|
char host[256];
|
||||||
|
const char *host_ptr, *hdr_connection;
|
||||||
|
char log_req_prefix[512];
|
||||||
|
|
||||||
|
msg_buf[0] = 0;
|
||||||
|
err_msg[0] = 0;
|
||||||
|
msg_content[0] = 0;
|
||||||
|
|
||||||
|
host_config_t *conf = NULL;
|
||||||
|
FILE *file = NULL;
|
||||||
|
|
||||||
|
long content_length = 0;
|
||||||
|
int accept_if_modified_since = 0;
|
||||||
|
int use_fastcgi = 0;
|
||||||
|
int use_proxy = 0;
|
||||||
|
int p_len;
|
||||||
|
|
||||||
|
fastcgi_conn fcgi_conn = {.socket = 0, .req_id = 0, .ctx = cctx};
|
||||||
|
http_status custom_status;
|
||||||
|
|
||||||
|
http_res res = {.version = "1.1", .status = http_get_status(501), .hdr.field_num = 0, .hdr.last_field_num = -1};
|
||||||
|
http_status_ctx ctx = {.status = 0, .origin = NONE, .ws_key = NULL};
|
||||||
|
|
||||||
|
clock_gettime(CLOCK_MONOTONIC, &begin);
|
||||||
|
|
||||||
|
//ret = sock_poll_read(&client, NULL, NULL, 1, NULL, NULL, CLIENT_TIMEOUT * 1000);
|
||||||
|
|
||||||
|
http_add_header_field(&res.hdr, "Date", http_get_date(buf0, sizeof(buf0)));
|
||||||
|
http_add_header_field(&res.hdr, "Server", SERVER_STR);
|
||||||
|
/*if (ret <= 0) {
|
||||||
|
if (errno != 0) return;
|
||||||
|
|
||||||
|
cctx->c_keep_alive = 0;
|
||||||
|
res.status = http_get_status(408);
|
||||||
|
goto respond;
|
||||||
|
}*/
|
||||||
|
//clock_gettime(CLOCK_MONOTONIC, &begin);
|
||||||
|
|
||||||
|
http_req req;
|
||||||
|
ret = http_receive_request(client, &req);
|
||||||
|
if (ret != 0) {
|
||||||
|
cctx->c_keep_alive = 0;
|
||||||
|
if (ret < 0) {
|
||||||
|
goto abort;
|
||||||
|
} else if (ret == 1) {
|
||||||
|
sprintf(err_msg, "Unable to parse http header: Invalid header format.");
|
||||||
|
} else if (ret == 2) {
|
||||||
|
sprintf(err_msg, "Unable to parse http header: Invalid method.");
|
||||||
|
} else if (ret == 3) {
|
||||||
|
sprintf(err_msg, "Unable to parse http header: Invalid version.");
|
||||||
|
} else if (ret == 4) {
|
||||||
|
sprintf(err_msg, "Unable to parse http header: Header contains illegal characters.");
|
||||||
|
} else if (ret == 5) {
|
||||||
|
sprintf(err_msg, "Unable to parse http header: End of header not found.");
|
||||||
|
}
|
||||||
|
res.status = http_get_status(400);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
hdr_connection = http_get_header_field(&req.hdr, "Connection");
|
||||||
|
cctx->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");
|
||||||
|
if (host_ptr != NULL && strlen(host_ptr) > 255) {
|
||||||
|
host[0] = 0;
|
||||||
|
res.status = http_get_status(400);
|
||||||
|
sprintf(err_msg, "Host header field is too long.");
|
||||||
|
goto respond;
|
||||||
|
} else if (host_ptr == NULL || strchr(host_ptr, '/') != NULL) {
|
||||||
|
if (strchr(cctx->addr, ':') == NULL) {
|
||||||
|
strcpy(host, cctx->addr);
|
||||||
|
} else {
|
||||||
|
sprintf(host, "[%s]", cctx->addr);
|
||||||
|
}
|
||||||
|
res.status = http_get_status(400);
|
||||||
|
sprintf(err_msg, "The client provided no or an invalid Host header field.");
|
||||||
|
goto respond;
|
||||||
|
} else {
|
||||||
|
strcpy(host, host_ptr);
|
||||||
|
}
|
||||||
|
|
||||||
|
logger_set_prefix("[%s%*s%s]%s", BLD_STR, INET6_ADDRSTRLEN, host, CLR_STR, cctx->log_prefix);
|
||||||
|
info(BLD_STR "%s %s", req.method, req.uri);
|
||||||
|
|
||||||
|
conf = get_host_config(host);
|
||||||
|
if (conf == NULL) {
|
||||||
|
info("Unknown host, redirecting to default");
|
||||||
|
res.status = http_get_status(307);
|
||||||
|
sprintf(buf0, "https://%s%s", DEFAULT_HOST, req.uri);
|
||||||
|
http_add_header_field(&res.hdr, "Location", buf0);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
http_uri uri;
|
||||||
|
unsigned char dir_mode = (conf->type == CONFIG_TYPE_LOCAL ? conf->local.dir_mode : URI_DIR_MODE_NO_VALIDATION);
|
||||||
|
ret = uri_init(&uri, conf->local.webroot, req.uri, dir_mode);
|
||||||
|
if (ret != 0) {
|
||||||
|
if (ret == 1) {
|
||||||
|
sprintf(err_msg, "Invalid URI: has to start with slash.");
|
||||||
|
res.status = http_get_status(400);
|
||||||
|
} else if (ret == 2) {
|
||||||
|
sprintf(err_msg, "Invalid URI: contains relative path change (/../).");
|
||||||
|
res.status = http_get_status(400);
|
||||||
|
} else if (ret == 3) {
|
||||||
|
sprintf(err_msg, "The specified webroot directory does not exist.");
|
||||||
|
res.status = http_get_status(404);
|
||||||
|
} else {
|
||||||
|
res.status = http_get_status(500);
|
||||||
|
}
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dir_mode != URI_DIR_MODE_NO_VALIDATION) {
|
||||||
|
ssize_t size = sizeof(buf0);
|
||||||
|
url_decode(req.uri, buf0, &size);
|
||||||
|
int change_proto = strncmp(uri.uri, "/.well-known/", 13) != 0 && !client->enc;
|
||||||
|
if (strcmp(uri.uri, buf0) != 0 || change_proto) {
|
||||||
|
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", host, buf0);
|
||||||
|
if (p_len < 0 || p_len >= sizeof(buf1)) {
|
||||||
|
res.status = http_get_status(500);
|
||||||
|
error("Header field 'Location' too long");
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
http_add_header_field(&res.hdr, "Location", buf1);
|
||||||
|
} else {
|
||||||
|
http_add_header_field(&res.hdr, "Location", buf0);
|
||||||
|
}
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
} else if (!client->enc) {
|
||||||
|
res.status = http_get_status(308);
|
||||||
|
sprintf(buf0, "https://%s%s", host, req.uri);
|
||||||
|
http_add_header_field(&res.hdr, "Location", buf0);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (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");
|
||||||
|
|
||||||
|
content_length = snprintf(msg_buf, sizeof(msg_buf) - 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];
|
||||||
|
content_length += snprintf(msg_buf + content_length, sizeof(msg_buf) - content_length, "%s: %s\r\n", http_field_get_name(f), http_field_get_value(f));
|
||||||
|
}
|
||||||
|
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
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.");
|
||||||
|
goto respond;
|
||||||
|
} 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.");
|
||||||
|
goto respond;
|
||||||
|
} 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.");
|
||||||
|
goto respond;
|
||||||
|
} else if (uri.filename == NULL || (strlen(uri.pathinfo) > 0 && (int) uri.is_static)) {
|
||||||
|
res.status = http_get_status(404);
|
||||||
|
goto respond;
|
||||||
|
} else if (strlen(uri.pathinfo) != 0 && conf->local.dir_mode != URI_DIR_MODE_INFO) {
|
||||||
|
res.status = http_get_status(404);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
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);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
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");
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
cache_init_uri(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) {
|
||||||
|
file = fopen(uri.meta->filename_comp_br, "rb");
|
||||||
|
if (file == NULL) {
|
||||||
|
cache_mark_dirty(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) {
|
||||||
|
file = fopen(uri.meta->filename_comp_gz, "rb");
|
||||||
|
if (file == NULL) {
|
||||||
|
cache_mark_dirty(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);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
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);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
range += 6;
|
||||||
|
char *ptr = strchr(range, '-');
|
||||||
|
if (ptr == NULL) {
|
||||||
|
res.status = http_get_status(416);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
file = fopen(uri.filename, "rb");
|
||||||
|
fseek(file, 0, SEEK_END);
|
||||||
|
unsigned long file_len = ftell(file);
|
||||||
|
fseek(file, 0, SEEK_SET);
|
||||||
|
if (file_len == 0) {
|
||||||
|
content_length = 0;
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
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);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
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(file, num1, SEEK_SET);
|
||||||
|
content_length = num2 - num1 + 1;
|
||||||
|
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (file == NULL) {
|
||||||
|
file = fopen(uri.filename, "rb");
|
||||||
|
}
|
||||||
|
|
||||||
|
fseek(file, 0, SEEK_END);
|
||||||
|
content_length = ftell(file);
|
||||||
|
fseek(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);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
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_conn, mode, 0 /* TODO */, cctx->req_num, client, &req, &uri) != 0) {
|
||||||
|
res.status = http_get_status(503);
|
||||||
|
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
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_conn, client, client_content_len);
|
||||||
|
} else if (client_transfer_encoding != NULL && strstr(client_transfer_encoding, "chunked") != NULL) {
|
||||||
|
ret = fastcgi_receive_chunked(&fcgi_conn, client);
|
||||||
|
} else {
|
||||||
|
ret = 0;
|
||||||
|
}
|
||||||
|
if (ret != 0) {
|
||||||
|
if (ret < 0) {
|
||||||
|
goto abort;
|
||||||
|
} else {
|
||||||
|
sprintf(err_msg, "Unable to communicate with FastCGI socket.");
|
||||||
|
}
|
||||||
|
res.status = http_get_status(502);
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
fastcgi_close_stdin(&fcgi_conn);
|
||||||
|
|
||||||
|
ret = fastcgi_header(&fcgi_conn, &res, err_msg);
|
||||||
|
if (ret != 0) {
|
||||||
|
if (ret < 0) goto abort;
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *status = http_get_header_field(&res.hdr, "Status");
|
||||||
|
if (status != NULL) {
|
||||||
|
int status_code = (int) strtoul(status, 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 + 4);
|
||||||
|
res.status = &custom_status;
|
||||||
|
} else if (res.status == NULL) {
|
||||||
|
res.status = http_get_status(500);
|
||||||
|
sprintf(err_msg, "The status code was set to an invalid or unknown value.");
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *content_length_f = http_get_header_field(&res.hdr, "Content-Length");
|
||||||
|
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 &&
|
||||||
|
content_length != -1 &&
|
||||||
|
content_length <= sizeof(msg_content) - 1)
|
||||||
|
{
|
||||||
|
fastcgi_dump(&fcgi_conn, msg_content, sizeof(msg_content));
|
||||||
|
goto respond;
|
||||||
|
}
|
||||||
|
|
||||||
|
use_fastcgi = 1;
|
||||||
|
|
||||||
|
if (content_length != -1 && content_length < 1024000) {
|
||||||
|
use_fastcgi |= FASTCGI_COMPRESS_HOLD;
|
||||||
|
}
|
||||||
|
|
||||||
|
content_length = -1;
|
||||||
|
|
||||||
|
int http_comp = http_get_compression(&req, &res);
|
||||||
|
if (http_comp & COMPRESS) {
|
||||||
|
if (http_comp & COMPRESS_BR) {
|
||||||
|
use_fastcgi |= FASTCGI_COMPRESS_BR;
|
||||||
|
sprintf(buf0, "br");
|
||||||
|
} else if (http_comp & COMPRESS_GZ) {
|
||||||
|
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");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if (conf->type == CONFIG_TYPE_REVERSE_PROXY) {
|
||||||
|
info("Reverse proxy for " BLD_STR "%s:%i" CLR_STR, conf->proxy.hostname, 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, &ctx, conf, client, cctx, &custom_status, err_msg);
|
||||||
|
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(ctx.ws_key, buf0) == 0) {
|
||||||
|
use_proxy = (strcmp(buf0, ws_accept) == 0) ? 2 : 1;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
ctx.status = 101;
|
||||||
|
ctx.origin = INTERNAL;
|
||||||
|
res.status = http_get_status(501);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Let 300 be formatted by origin server
|
||||||
|
if (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 (ctx.status != 101) {
|
||||||
|
ctx.status = res.status->code;
|
||||||
|
ctx.origin = res.status->code >= 400 ? SERVER : NONE;
|
||||||
|
}
|
||||||
|
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);
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
} else {
|
||||||
|
error("Unknown host type: %i", conf->type);
|
||||||
|
res.status = http_get_status(501);
|
||||||
|
}
|
||||||
|
|
||||||
|
respond:
|
||||||
|
if (!use_proxy) {
|
||||||
|
if (conf != NULL && conf->type == CONFIG_TYPE_LOCAL && uri.is_static && res.status->code == 405) {
|
||||||
|
http_add_header_field(&res.hdr, "Allow", "GET, HEAD, TRACE");
|
||||||
|
}
|
||||||
|
if (http_get_header_field(&res.hdr, "Accept-Ranges") == NULL) {
|
||||||
|
http_add_header_field(&res.hdr, "Accept-Ranges", "none");
|
||||||
|
}
|
||||||
|
if (!use_fastcgi && file == NULL) {
|
||||||
|
http_remove_header_field(&res.hdr, "Date", HTTP_REMOVE_ALL);
|
||||||
|
http_remove_header_field(&res.hdr, "Server", HTTP_REMOVE_ALL);
|
||||||
|
http_remove_header_field(&res.hdr, "Cache-Control", HTTP_REMOVE_ALL);
|
||||||
|
http_remove_header_field(&res.hdr, "Content-Type", HTTP_REMOVE_ALL);
|
||||||
|
http_remove_header_field(&res.hdr, "Content-Encoding", HTTP_REMOVE_ALL);
|
||||||
|
http_add_header_field(&res.hdr, "Date", http_get_date(buf0, sizeof(buf0)));
|
||||||
|
http_add_header_field(&res.hdr, "Server", SERVER_STR);
|
||||||
|
http_add_header_field(&res.hdr, "Cache-Control", "no-cache");
|
||||||
|
http_add_header_field(&res.hdr, "Content-Type", "text/html; charset=UTF-8");
|
||||||
|
|
||||||
|
// TODO list Locations on 3xx Redirects
|
||||||
|
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 (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), "<ul>\n\t<li><a href=\"%s\">%s</a></li>\n</ul>\n", location, location);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else if (strncmp(msg_content, "<!DOCTYPE html>", 15) == 0 || strncmp(msg_content, "<html", 5) == 0) {
|
||||||
|
msg_content[0] = 0;
|
||||||
|
// TODO let relevant information pass?
|
||||||
|
}
|
||||||
|
|
||||||
|
char *proxy_doc = "";
|
||||||
|
if (conf != NULL && conf->type == CONFIG_TYPE_REVERSE_PROXY) {
|
||||||
|
const http_status *status = http_get_status(ctx.status);
|
||||||
|
char stat_str[8];
|
||||||
|
sprintf(stat_str, "%03i", ctx.status);
|
||||||
|
sprintf(msg_pre_buf_2, http_proxy_document,
|
||||||
|
" success",
|
||||||
|
(ctx.origin == CLIENT_REQ) ? " error" : " success",
|
||||||
|
(ctx.origin == INTERNAL) ? " error" : " success",
|
||||||
|
(ctx.origin == SERVER_REQ) ? " error" : (ctx.status == 0 ? "" : " success"),
|
||||||
|
(ctx.origin == CLIENT_RES) ? " error" : " success",
|
||||||
|
(ctx.origin == SERVER) ? " error" : (ctx.status == 0 ? "" : " success"),
|
||||||
|
(ctx.origin == SERVER_RES) ? " error" : (ctx.status == 0 ? "" : " success"),
|
||||||
|
(ctx.origin == INTERNAL) ? " error" : " success",
|
||||||
|
(ctx.origin == INTERNAL || ctx.origin == SERVER) ? " error" : " success",
|
||||||
|
res.status->code,
|
||||||
|
res.status->msg,
|
||||||
|
(ctx.status == 0) ? "???" : stat_str,
|
||||||
|
(status != NULL) ? status->msg : "",
|
||||||
|
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 : "");
|
||||||
|
content_length = snprintf(msg_buf, sizeof(msg_buf), http_default_document, res.status->code,
|
||||||
|
res.status->msg, msg_pre_buf_1, info->mode, info->icon, info->color, host,
|
||||||
|
proxy_doc, msg_content[0] != 0 ? msg_content : "");
|
||||||
|
}
|
||||||
|
if (content_length >= 0) {
|
||||||
|
sprintf(buf0, "%li", content_length);
|
||||||
|
http_remove_header_field(&res.hdr, "Content-Length", HTTP_REMOVE_ALL);
|
||||||
|
http_add_header_field(&res.hdr, "Content-Length", buf0);
|
||||||
|
} else if (http_get_header_field(&res.hdr, "Transfer-Encoding") == NULL) {
|
||||||
|
cctx->s_keep_alive = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int close_proxy = 0;
|
||||||
|
if (use_proxy != 2) {
|
||||||
|
const char *conn = http_get_header_field(&res.hdr, "Connection");
|
||||||
|
close_proxy = (conn == NULL || (strstr(conn, "keep-alive") == NULL && strstr(conn, "Keep-Alive") == NULL));
|
||||||
|
http_remove_header_field(&res.hdr, "Connection", HTTP_REMOVE_ALL);
|
||||||
|
http_remove_header_field(&res.hdr, "Keep-Alive", HTTP_REMOVE_ALL);
|
||||||
|
if (cctx->s_keep_alive && cctx->c_keep_alive) {
|
||||||
|
http_add_header_field(&res.hdr, "Connection", "keep-alive");
|
||||||
|
sprintf(buf0, "timeout=%i, max=%i", CLIENT_TIMEOUT, REQ_PER_CONNECTION);
|
||||||
|
http_add_header_field(&res.hdr, "Keep-Alive", buf0);
|
||||||
|
} else {
|
||||||
|
http_add_header_field(&res.hdr, "Connection", "close");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
http_send_response(client, &res);
|
||||||
|
clock_gettime(CLOCK_MONOTONIC, &end);
|
||||||
|
const char *location = http_get_header_field(&res.hdr, "Location");
|
||||||
|
unsigned long micros = (end.tv_nsec - begin.tv_nsec) / 1000 + (end.tv_sec - begin.tv_sec) * 1000000;
|
||||||
|
info("%s%s%03i %s%s%s (%s)%s", http_get_status_color(res.status), use_proxy ? "-> " : "", res.status->code,
|
||||||
|
res.status->msg, location != NULL ? " -> " : "", location != NULL ? location : "",
|
||||||
|
format_duration(micros, buf0), CLR_STR);
|
||||||
|
|
||||||
|
// TODO access/error log file
|
||||||
|
|
||||||
|
if (use_proxy == 2) {
|
||||||
|
// WebSocket
|
||||||
|
info("Upgrading connection to WebSocket connection");
|
||||||
|
ret = ws_handle_connection(client, &proxy);
|
||||||
|
if (ret != 0) {
|
||||||
|
cctx->c_keep_alive = 0;
|
||||||
|
close_proxy = 1;
|
||||||
|
}
|
||||||
|
info("WebSocket connection closed");
|
||||||
|
} else if (strcmp(req.method, "HEAD") != 0) {
|
||||||
|
// default response
|
||||||
|
unsigned long snd_len = 0;
|
||||||
|
unsigned long len;
|
||||||
|
if (msg_buf[0] != 0) {
|
||||||
|
ret = sock_send(client, msg_buf, content_length, 0);
|
||||||
|
if (ret <= 0) {
|
||||||
|
error("Unable to send: %s", sock_strerror(client));
|
||||||
|
}
|
||||||
|
snd_len += ret;
|
||||||
|
} else if (file != NULL) {
|
||||||
|
while (snd_len < content_length) {
|
||||||
|
len = fread(buffer, 1, CHUNK_SIZE, file);
|
||||||
|
if (snd_len + len > content_length) {
|
||||||
|
len = content_length - snd_len;
|
||||||
|
}
|
||||||
|
ret = sock_send(client, buffer, len, feof(file) ? 0 : MSG_MORE);
|
||||||
|
if (ret <= 0) {
|
||||||
|
error("Unable to send: %s", sock_strerror(client));
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
snd_len += ret;
|
||||||
|
}
|
||||||
|
} else if (use_fastcgi) {
|
||||||
|
const char *transfer_encoding = http_get_header_field(&res.hdr, "Transfer-Encoding");
|
||||||
|
int chunked = (transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL);
|
||||||
|
|
||||||
|
int flags = (chunked ? FASTCGI_CHUNKED : 0) | (use_fastcgi & (FASTCGI_COMPRESS | FASTCGI_COMPRESS_HOLD));
|
||||||
|
ret = fastcgi_send(&fcgi_conn, client, flags);
|
||||||
|
} else if (use_proxy) {
|
||||||
|
const char *transfer_encoding = http_get_header_field(&res.hdr, "Transfer-Encoding");
|
||||||
|
int chunked = transfer_encoding != NULL && strstr(transfer_encoding, "chunked") != NULL;
|
||||||
|
|
||||||
|
const char *content_len = http_get_header_field(&res.hdr, "Content-Length");
|
||||||
|
unsigned long len_to_send = 0;
|
||||||
|
if (content_len != NULL) {
|
||||||
|
len_to_send = strtol(content_len, NULL, 10);
|
||||||
|
}
|
||||||
|
|
||||||
|
int flags = (chunked ? PROXY_CHUNKED : 0) | (use_proxy & PROXY_COMPRESS);
|
||||||
|
ret = proxy_send(client, len_to_send, flags);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ret < 0) {
|
||||||
|
cctx->c_keep_alive = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (close_proxy && proxy.socket != 0) {
|
||||||
|
info(BLUE_STR "Closing proxy connection");
|
||||||
|
sock_close(&proxy);
|
||||||
|
}
|
||||||
|
|
||||||
|
clock_gettime(CLOCK_MONOTONIC, &end);
|
||||||
|
micros = (end.tv_nsec - begin.tv_nsec) / 1000 + (end.tv_sec - begin.tv_sec) * 1000000;
|
||||||
|
info("Transfer complete: %s", format_duration(micros, buf0));
|
||||||
|
|
||||||
|
uri_free(&uri);
|
||||||
|
abort:
|
||||||
|
if (fcgi_conn.socket != 0) {
|
||||||
|
shutdown(fcgi_conn.socket, SHUT_RDWR);
|
||||||
|
close(fcgi_conn.socket);
|
||||||
|
fcgi_conn.socket = 0;
|
||||||
|
}
|
||||||
|
http_free_req(&req);
|
||||||
|
http_free_res(&res);
|
||||||
|
}
|
||||||
|
|||||||
@ -1,8 +1,15 @@
|
|||||||
|
/**
|
||||||
|
* sesimos - secure, simple, modern web server
|
||||||
|
* @brief Client request handler (header file)
|
||||||
|
* @file src/worker/request_handler.h
|
||||||
|
* @author Lorenz Stechauner
|
||||||
|
* @date 2022-12-28
|
||||||
|
*/
|
||||||
|
|
||||||
#ifndef SESIMOS_REQUEST_HANDLER_H
|
#ifndef SESIMOS_REQUEST_HANDLER_H
|
||||||
#define SESIMOS_REQUEST_HANDLER_H
|
#define SESIMOS_REQUEST_HANDLER_H
|
||||||
|
|
||||||
#include "../client.h"
|
#include "../server.h"
|
||||||
|
|
||||||
int request_handler_init(int n_workers, int buf_size);
|
int request_handler_init(int n_workers, int buf_size);
|
||||||
|
|
||||||
|
|||||||
@ -1,13 +1,21 @@
|
|||||||
|
/**
|
||||||
|
* sesimos - secure, simple, modern web server
|
||||||
|
* @brief TCP acceptor
|
||||||
|
* @file src/worker/tcp_acceptor.c
|
||||||
|
* @author Lorenz Stechauner
|
||||||
|
* @date 2022-12-28
|
||||||
|
*/
|
||||||
|
|
||||||
#include "tcp_acceptor.h"
|
#include "../server.h"
|
||||||
|
#include "../async.h"
|
||||||
#include "../logger.h"
|
#include "../logger.h"
|
||||||
#include "../lib/mpmc.h"
|
#include "../lib/mpmc.h"
|
||||||
#include "../lib/utils.h"
|
#include "../lib/utils.h"
|
||||||
#include "../server.h"
|
|
||||||
#include "../lib/geoip.h"
|
#include "../lib/geoip.h"
|
||||||
#include "../async.h"
|
#include "../lib/config.h"
|
||||||
#include "tcp_closer.h"
|
#include "tcp_closer.h"
|
||||||
#include "request_handler.h"
|
#include "request_handler.h"
|
||||||
|
#include "tcp_acceptor.h"
|
||||||
|
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
#include <errno.h>
|
#include <errno.h>
|
||||||
|
|||||||
@ -1,8 +1,15 @@
|
|||||||
|
/**
|
||||||
|
* sesimos - secure, simple, modern web server
|
||||||
|
* @brief TCP acceptor (header file)
|
||||||
|
* @file src/worker/tcp_acceptor.h
|
||||||
|
* @author Lorenz Stechauner
|
||||||
|
* @date 2022-12-28
|
||||||
|
*/
|
||||||
|
|
||||||
#ifndef SESIMOS_TCP_ACCEPTOR_H
|
#ifndef SESIMOS_TCP_ACCEPTOR_H
|
||||||
#define SESIMOS_TCP_ACCEPTOR_H
|
#define SESIMOS_TCP_ACCEPTOR_H
|
||||||
|
|
||||||
#include "../client.h"
|
#include "../server.h"
|
||||||
|
|
||||||
int tcp_acceptor_init(int n_workers, int buf_size);
|
int tcp_acceptor_init(int n_workers, int buf_size);
|
||||||
|
|
||||||
|
|||||||
@ -1,3 +1,10 @@
|
|||||||
|
/**
|
||||||
|
* sesimos - secure, simple, modern web server
|
||||||
|
* @brief TCP closer
|
||||||
|
* @file src/worker/tcp_closer.c
|
||||||
|
* @author Lorenz Stechauner
|
||||||
|
* @date 2022-12-28
|
||||||
|
*/
|
||||||
|
|
||||||
#include "tcp_closer.h"
|
#include "tcp_closer.h"
|
||||||
#include "../logger.h"
|
#include "../logger.h"
|
||||||
|
|||||||
@ -1,8 +1,15 @@
|
|||||||
|
/**
|
||||||
|
* sesimos - secure, simple, modern web server
|
||||||
|
* @brief TCP closer (header file)
|
||||||
|
* @file src/worker/tcp_closer.h
|
||||||
|
* @author Lorenz Stechauner
|
||||||
|
* @date 2022-12-28
|
||||||
|
*/
|
||||||
|
|
||||||
#ifndef SESIMOS_TCP_CLOSER_H
|
#ifndef SESIMOS_TCP_CLOSER_H
|
||||||
#define SESIMOS_TCP_CLOSER_H
|
#define SESIMOS_TCP_CLOSER_H
|
||||||
|
|
||||||
#include "../client.h"
|
#include "../server.h"
|
||||||
|
|
||||||
int tcp_closer_init(int n_workers, int buf_size);
|
int tcp_closer_init(int n_workers, int buf_size);
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user