diff --git a/configure b/configure index 16834c0..32ef31b 100755 --- a/configure +++ b/configure @@ -11,6 +11,7 @@ gmnisrv() { src/main.c \ src/mime.c \ src/server.c \ + src/serve.c \ src/tls.c \ src/url.c \ src/util.c diff --git a/include/server.h b/include/server.h index 96d9917..884f210 100644 --- a/include/server.h +++ b/include/server.h @@ -58,8 +58,19 @@ struct gmnisrv_server { bool run; }; +// server.c int server_init(struct gmnisrv_server *server, struct gmnisrv_config *conf); void server_run(struct gmnisrv_server *server); void server_finish(struct gmnisrv_server *server); +void disconnect_client(struct gmnisrv_server *server, + struct gmnisrv_client *client); + +// serve.c +void serve_request(struct gmnisrv_client *client); +bool request_validate(struct gmnisrv_client *client, char **path); +void client_submit_response(struct gmnisrv_client *client, + enum gemini_status status, const char *meta, int bodyfd); +void client_oom(struct gmnisrv_client *client); + #endif diff --git a/src/serve.c b/src/serve.c new file mode 100644 index 0000000..00d51b5 --- /dev/null +++ b/src/serve.c @@ -0,0 +1,128 @@ +#include +#include +#include +#include +#include +#include +#include "config.h" +#include "gemini.h" +#include "log.h" +#include "mime.h" +#include "server.h" +#include "url.h" + +void +client_submit_response(struct gmnisrv_client *client, + enum gemini_status status, const char *meta, int bodyfd) +{ + client->status = status; + client->meta = strdup(meta); + client->bodyfd = bodyfd; + client->pollfd->events = POLLOUT; +} + +void +client_oom(struct gmnisrv_client *client) +{ + const char *error = "Out of memory"; + client_submit_response(client, + GEMINI_STATUS_TEMPORARY_FAILURE, error, -1); +} + +void +serve_request(struct gmnisrv_client *client) +{ + struct gmnisrv_host *host = client->host; + assert(host); + assert(host->root); // TODO: reverse proxy support + + char path[PATH_MAX + 1]; + int n = snprintf(path, sizeof(path), "%s%s", host->root, client->path); + if ((size_t)n >= sizeof(path)) { + client_submit_response(client, GEMINI_STATUS_PERMANENT_FAILURE, + "Request path exceeds PATH_MAX", -1); + return; + } + + if (path[strlen(path) - 1] == '/') { + // TODO: Let user configure index file name? + strncat(path, "index.gmi", sizeof(path) - 1); + } + + int fd = open(path, O_RDONLY); + if (fd == -1) { + if (errno == ENOENT) { + client_submit_response(client, GEMINI_STATUS_NOT_FOUND, + "Not found", -1); + return; + } else { + client_error(&client->addr, "error opening %s: %s", + path, strerror(errno)); + client_submit_response(client, GEMINI_STATUS_PERMANENT_FAILURE, + "Internal server error", -1); + return; + } + } + + const char *meta = gmnisrv_mimetype_for_path(path); + client_submit_response(client, GEMINI_STATUS_SUCCESS, meta, fd); +} + +bool +request_validate(struct gmnisrv_client *client, char **path) +{ + struct Curl_URL *url = curl_url(); + if (!url) { + client_oom(client); + return false; + } + if (curl_url_set(url, CURLUPART_URL, client->buf, 0) != CURLUE_OK) { + const char *error = "Protocol error: invalid URL"; + client_submit_response(client, + GEMINI_STATUS_BAD_REQUEST, error, -1); + goto exit; + } + + char *part; + if (curl_url_get(url, CURLUPART_SCHEME, &part, 0) != CURLUE_OK) { + const char *error = "Protocol error: invalid URL (expected scheme)"; + client_submit_response(client, + GEMINI_STATUS_BAD_REQUEST, error, -1); + goto exit; + } else if (strcmp(part, "gemini") != 0) { + free(part); + const char *error = "Refusing proxy to non-gemini URL"; + client_submit_response(client, + GEMINI_STATUS_PROXY_REQUEST_REFUSED, error, -1); + goto exit; + } + free(part); + + if (curl_url_get(url, CURLUPART_HOST, &part, 0) != CURLUE_OK) { + const char *error = "Protocol error: invalid URL (expected host)"; + client_submit_response(client, + GEMINI_STATUS_BAD_REQUEST, error, -1); + goto exit; + } else if (strcmp(part, client->host->hostname) != 0) { + free(part); + const char *error = "Protocol error: hostname does not match SNI"; + client_submit_response(client, + GEMINI_STATUS_BAD_REQUEST, error, -1); + goto exit; + } + free(part); + + if (curl_url_get(url, CURLUPART_PATH, &part, 0) != CURLUE_OK) { + const char *error = "Protocol error: invalid URL (expected path)"; + client_submit_response(client, + GEMINI_STATUS_BAD_REQUEST, error, -1); + goto exit; + } + // NOTE: curl_url_set(..., CURLUPART_URL, ..., 0) will consoldate .. and + // . to prevent directory traversal without additional code. + *path = part; + +exit: + curl_url_cleanup(url); + return true; +} diff --git a/src/server.c b/src/server.c index 19326ac..e69e381 100644 --- a/src/server.c +++ b/src/server.c @@ -13,10 +13,8 @@ #include "config.h" #include "gemini.h" #include "log.h" -#include "mime.h" #include "server.h" #include "tls.h" -#include "url.h" int server_init(struct gmnisrv_server *server, struct gmnisrv_config *conf) @@ -174,7 +172,7 @@ timespec_diff(struct timespec *start, } } -static void +void disconnect_client(struct gmnisrv_server *server, struct gmnisrv_client *client) { if (client->status != GEMINI_STATUS_NONE) { @@ -233,122 +231,6 @@ client_init_ssl(struct gmnisrv_server *server, struct gmnisrv_client *client) return 0; } -static void -client_submit_response(struct gmnisrv_client *client, - enum gemini_status status, const char *meta, int bodyfd) -{ - client->status = status; - client->meta = strdup(meta); - client->bodyfd = bodyfd; - client->pollfd->events = POLLOUT; -} - -static void -client_oom(struct gmnisrv_client *client) -{ - const char *error = "Out of memory"; - client_submit_response(client, - GEMINI_STATUS_TEMPORARY_FAILURE, error, -1); -} - -static bool -request_validate(struct gmnisrv_client *client, char **path) -{ - struct Curl_URL *url = curl_url(); - if (!url) { - client_oom(client); - return false; - } - if (curl_url_set(url, CURLUPART_URL, client->buf, 0) != CURLUE_OK) { - const char *error = "Protocol error: invalid URL"; - client_submit_response(client, - GEMINI_STATUS_BAD_REQUEST, error, -1); - goto exit; - } - - char *part; - if (curl_url_get(url, CURLUPART_SCHEME, &part, 0) != CURLUE_OK) { - const char *error = "Protocol error: invalid URL (expected scheme)"; - client_submit_response(client, - GEMINI_STATUS_BAD_REQUEST, error, -1); - goto exit; - } else if (strcmp(part, "gemini") != 0) { - free(part); - const char *error = "Refusing proxy to non-gemini URL"; - client_submit_response(client, - GEMINI_STATUS_PROXY_REQUEST_REFUSED, error, -1); - goto exit; - } - free(part); - - if (curl_url_get(url, CURLUPART_HOST, &part, 0) != CURLUE_OK) { - const char *error = "Protocol error: invalid URL (expected host)"; - client_submit_response(client, - GEMINI_STATUS_BAD_REQUEST, error, -1); - goto exit; - } else if (strcmp(part, client->host->hostname) != 0) { - free(part); - const char *error = "Protocol error: hostname does not match SNI"; - client_submit_response(client, - GEMINI_STATUS_BAD_REQUEST, error, -1); - goto exit; - } - free(part); - - if (curl_url_get(url, CURLUPART_PATH, &part, 0) != CURLUE_OK) { - const char *error = "Protocol error: invalid URL (expected path)"; - client_submit_response(client, - GEMINI_STATUS_BAD_REQUEST, error, -1); - goto exit; - } - // NOTE: curl_url_set(..., CURLUPART_URL, ..., 0) will consoldate .. and - // . to prevent directory traversal without additional code. - *path = part; - -exit: - curl_url_cleanup(url); - return true; -} - -static void -serve_request(struct gmnisrv_client *client) -{ - struct gmnisrv_host *host = client->host; - assert(host); - assert(host->root); // TODO: reverse proxy support - - char path[PATH_MAX + 1]; - int n = snprintf(path, sizeof(path), "%s%s", host->root, client->path); - if ((size_t)n >= sizeof(path)) { - client_submit_response(client, GEMINI_STATUS_PERMANENT_FAILURE, - "Request path exceeds PATH_MAX", -1); - return; - } - - if (path[strlen(path) - 1] == '/') { - // TODO: Let user configure index file name? - strncat(path, "index.gmi", sizeof(path) - 1); - } - - int fd = open(path, O_RDONLY); - if (fd == -1) { - if (errno == ENOENT) { - client_submit_response(client, GEMINI_STATUS_NOT_FOUND, - "Not found", -1); - return; - } else { - client_error(&client->addr, "error opening %s: %s", - path, strerror(errno)); - client_submit_response(client, GEMINI_STATUS_PERMANENT_FAILURE, - "Internal server error", -1); - return; - } - } - - const char *meta = gmnisrv_mimetype_for_path(path); - client_submit_response(client, GEMINI_STATUS_SUCCESS, meta, fd); -} - static void client_readable(struct gmnisrv_server *server, struct gmnisrv_client *client) {