/* $Id: server.c 21 2024-09-14 12:39:39Z nishi $ */ #define SOURCE #include "tw_server.h" #include "tw_ssl.h" #include "tw_config.h" #include "tw_http.h" #include "tw_module.h" #include "tw_version.h" #include #include #include #include #include #include #include #include #ifdef __MINGW32__ #include #include #else #include #include #include #include #include #endif extern struct tw_config config; extern char tw_server[]; fd_set fdset; int sockcount = 0; SOCKADDR addresses[MAX_PORTS]; int sockets[MAX_PORTS]; void close_socket(int sock) { #ifdef __MINGW32__ closesocket(sock); #else close(sock); #endif } int tw_server_init(void) { int i; #ifdef __MINGW32__ WSADATA wsa; WSAStartup(MAKEWORD(2, 0), &wsa); #endif for(i = 0; config.ports[i] != -1; i++) ; sockcount = i; for(i = 0; config.ports[i] != -1; i++) { #ifdef NO_IPV6 int sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); #else int sock = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP); #endif #ifdef __MINGW32__ if(sock == INVALID_SOCKET) #else if(sock < 0) #endif { cm_log("Server", "Socket creation failure"); return 1; } int yes = 1; if(setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, (void*)&yes, sizeof(yes)) < 0) { close_socket(sock); cm_log("Server", "setsockopt failure (reuseaddr)"); return 1; } if(setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, (void*)&yes, sizeof(yes)) < 0) { close_socket(sock); cm_log("Server", "setsockopt failure (nodelay)"); return 1; } #ifndef NO_IPV6 int no = 0; if(setsockopt(sock, IPPROTO_IPV6, IPV6_V6ONLY, (void*)&no, sizeof(no)) < 0) { close_socket(sock); cm_log("Server", "setsockopt failure (IPv6)"); return 1; } #endif memset(&addresses[i], 0, sizeof(addresses[i])); #ifdef NO_IPV6 addresses[i].sin_family = AF_INET; addresses[i].sin_addr.s_addr = INADDR_ANY; addresses[i].sin_port = htons(config.ports[i]); #else addresses[i].sin6_family = AF_INET6; addresses[i].sin6_addr = in6addr_any; addresses[i].sin6_port = htons(config.ports[i]); #endif if(bind(sock, (struct sockaddr*)&addresses[i], sizeof(addresses[i])) < 0) { close_socket(sock); cm_log("Server", "Bind failure"); return 1; } if(listen(sock, 128) < 0) { close_socket(sock); cm_log("Server", "Listen failure"); return 1; } sockets[i] = sock; } return 0; } size_t tw_read(SSL* ssl, int s, void* data, size_t len) { if(ssl == NULL) { return recv(s, data, len, 0); } else { return SSL_read(ssl, data, len); } } size_t tw_write(SSL* ssl, int s, void* data, size_t len) { if(ssl == NULL) { return send(s, data, len, 0); } else { return SSL_write(ssl, data, len); } } #define ERROR_HTML \ "\n" \ "\n" \ " \n" \ " %s\n" \ " \n" \ " \n" \ "

%s

\n" \ "

\n" \ " ", \ address, \ "\n" \ " \n" \ "\n" void tw_process_page(SSL* ssl, int sock, const char* status, const char* type, FILE* f, const unsigned char* doc, size_t size) { char construct[512]; sprintf(construct, "%llu", (unsigned long long)size); tw_write(ssl, sock, "HTTP/1.1 ", 9); tw_write(ssl, sock, (char*)status, strlen(status)); tw_write(ssl, sock, "\r\n", 2); tw_write(ssl, sock, "Content-Type: ", 7 + 5 + 2); tw_write(ssl, sock, (char*)type, strlen(type)); tw_write(ssl, sock, "\r\n", 2); tw_write(ssl, sock, "Server: ", 6 + 2); tw_write(ssl, sock, tw_server, strlen(tw_server)); tw_write(ssl, sock, "\r\n", 2); tw_write(ssl, sock, "Content-Length: ", 7 + 7 + 2); tw_write(ssl, sock, construct, strlen(construct)); tw_write(ssl, sock, "\r\n", 2); tw_write(ssl, sock, "\r\n", 2); size_t incr = 0; while(1) { if(f != NULL){ char buffer[128]; fread(buffer, size < 128 ? size : 128, 1, f); tw_write(ssl, sock, buffer, size < 128 ? size : 128); }else{ tw_write(ssl, sock, (unsigned char*)doc + incr, size < 128 ? size : 128); } incr += 128; if(size <= 128) break; size -= 128; } } const char* tw_http_status(int code) { if(code == 200) { return "200 OK"; } else if(code == 400) { return "400 Bad Request"; } else if(code == 401) { return "401 Unauthorized"; } else if(code == 403) { return "403 Forbidden"; } else if(code == 404) { return "404 Not Found"; } else { return "400 Bad Request"; } } char* tw_http_default_error(int code, char* name, int port) { char address[1024]; sprintf(address, "

%s Server at %s Port %d

", tw_server, name, port); char* st = cm_strdup(tw_http_status(code)); char* st2; int i; for(i = 0; st[i] != 0; i++) { if(st[i] == ' ') { st2 = cm_strdup(st + i + 1); break; } } char* buffer = malloc(4096); char* str = cm_strcat3(ERROR_HTML); sprintf(buffer, str, st, st2); free(str); free(st); return buffer; } void tw_http_error(SSL* ssl, int sock, int error, char* name, int port) { char* str = tw_http_default_error(error, name, port); tw_process_page(ssl, sock, tw_http_status(error), "text/html", NULL, str, strlen(str)); free(str); } void addstring(char** str, const char* add, ...) { int i; char cbuf[2]; cbuf[1] = 0; va_list va; va_start(va, add); for(i = 0; add[i] != 0; i++) { cbuf[0] = add[i]; if(add[i] == '%') { i++; if(add[i] == 's') { char* tmp = *str; *str = cm_strcat(tmp, va_arg(va, const char*)); free(tmp); } else if(add[i] == 'h') { char* h = cm_html_escape(va_arg(va, const char*)); char* tmp = *str; *str = cm_strcat(tmp, h); free(tmp); free(h); } else if(add[i] == 'l') { char* h = cm_url_escape(va_arg(va, const char*)); char* tmp = *str; *str = cm_strcat(tmp, h); free(tmp); free(h); } else if(add[i] == 'd') { int n = va_arg(va, int); char* h = malloc(512); sprintf(h, "%d", n); char* tmp = *str; *str = cm_strcat(tmp, h); free(tmp); free(h); } else if(add[i] == '%') { char* tmp = *str; *str = cm_strcat(tmp, "%"); free(tmp); } } else { char* tmp = *str; *str = cm_strcat(tmp, cbuf); free(tmp); } } } #ifdef __MINGW32__ struct pass_entry { int sock; int port; bool ssl; SOCKADDR addr; }; unsigned int WINAPI tw_server_pass(void* ptr) { int sock = ((struct pass_entry*)ptr)->sock; bool ssl = ((struct pass_entry*)ptr)->ssl; int port = ((struct pass_entry*)ptr)->port; SOCKADDR addr = ((struct pass_entry*)ptr)->addr; free(ptr); #else void tw_server_pass(int sock, bool ssl, int port, SOCKADDR addr) { #endif char* name = config.hostname; SSL_CTX* ctx = NULL; SSL* s = NULL; bool sslworks = false; if(ssl) { ctx = tw_create_ssl_ctx(port); s = SSL_new(ctx); SSL_set_fd(s, sock); if(SSL_accept(s) <= 0) goto cleanup; sslworks = true; } struct tw_http_request req; struct tw_http_response res; struct tw_tool tools; res._processed = false; tw_init_tools(&tools); int ret = tw_http_parse(s, sock, &req); if(ret == 0) { char* vhost = cm_strdup(config.hostname); int i; for(i = 0; req.headers[i] != NULL; i += 2){ if(cm_strcaseequ(req.headers[i], "Host")){ free(vhost); vhost = req.headers[i + 1]; break; } } cm_log("Server", "Host is %s", vhost); int port = s == NULL ? 80 : 443; char* host = cm_strdup(vhost); for(i = 0; vhost[i] != 0; i++){ if(vhost[i] == ':'){ host[i] = 0; port = atoi(host + i + 1); break; } } cm_log("Server", "Hostname is `%s', port is `%d'", host, port); struct tw_config_entry* vhost_entry = tw_vhost_match(host, port); for(i = 0; i < config.module_count; i++) { tw_mod_request_t mod_req = (tw_mod_request_t)tw_module_symbol(config.modules[i], "mod_request"); if(mod_req != NULL) { int ret = mod_req(&tools, &req, &res); int co = ret & 0xff; if(co == _TW_MODULE_PASS) continue; if(co == _TW_MODULE_STOP) { res._processed = true; break; } if(co == _TW_MODULE_ERROR) { tw_http_error(s, sock, (ret & 0xffff00) >> 8, name, port); break; } } } if(!res._processed) { cm_log("Server", "Document root is %s", vhost_entry->root == NULL ? "not set" : vhost_entry->root); char* path = cm_strcat(vhost_entry->root == NULL ? "" : vhost_entry->root, req.path); cm_log("Server", "Filesystem path is %s", path); struct stat st; if(stat(path, &st) == 0){ if(!tw_permission_allowed(path, addr, req, vhost_entry)){ tw_http_error(s, sock, 403, name, port); }else if(S_ISDIR(st.st_mode)){ char* str = malloc(1); str[0] = 0; char** items = cm_scandir(path); addstring(&str, "\n"); addstring(&str, "\n"); addstring(&str, " \n"); addstring(&str, " \n"); addstring(&str, " Index of %h\n", req.path); addstring(&str, " \n"); addstring(&str, " \n"); addstring(&str, "

Index of %h

\n", req.path); addstring(&str, "

\n"); addstring(&str, " \n"); addstring(&str, " \n"); addstring(&str, " \n"); addstring(&str, " \n"); addstring(&str, " \n"); if(items != NULL){ for(i = 0; items[i] != NULL; i++){ addstring(&str, "\n"); addstring(&str, " \n"); addstring(&str, " \n", items[i], items[i]); addstring(&str, "\n"); } } addstring(&str, "

	Filename
	%h

\n"); addstring(&str, "

%s Server at %s Port %d

\n", tw_server, name, port); addstring(&str, " \n"); addstring(&str, "\n"); tw_process_page(s, sock, tw_http_status(200), "text/html", NULL, str, strlen(str)); free(str); }else{ char* mime = "application/octet-stream"; bool set = false; char* ext = NULL; for(i = strlen(req.path) - 1; i >= 0; i--){ if(req.path[i] == '.'){ ext = cm_strdup(req.path + i); break; } } for(i = 0; i < vhost_entry->mime_count; i++){ if(strcmp(vhost_entry->mimes[i].ext, "all") == 0 || (ext != NULL && strcmp(vhost_entry->mimes[i].ext, ext) == 0)){ mime = vhost_entry->mimes[i].mime; set = true; } } if(!set){ for(i = 0; i < config.root.mime_count; i++){ if(strcmp(config.root.mimes[i].ext, "all") == 0 || (ext != NULL && strcmp(config.root.mimes[i].ext, ext) == 0)){ mime = config.root.mimes[i].mime; set = true; } } } if(ext != NULL) free(ext); FILE* f = fopen(path, "rb"); tw_process_page(s, sock, tw_http_status(200), mime, f, NULL, st.st_size); fclose(f); } }else{ tw_http_error(s, sock, 404, name, port); } free(path); } free(vhost); free(host); } else { tw_http_error(s, sock, 400, name, port); } cleanup: if(sslworks) { SSL_shutdown(s); } SSL_free(s); close_socket(sock); #ifdef __MINGW32__ _endthreadex(0); #endif } void tw_server_loop(void) { struct timeval tv; while(1) { FD_ZERO(&fdset); int i; for(i = 0; i < sockcount; i++) { FD_SET(sockets[i], &fdset); } tv.tv_sec = 1; tv.tv_usec = 0; int ret = select(FD_SETSIZE, &fdset, NULL, NULL, &tv); if(ret == -1) { break; } else if(ret > 0) { /* connection */ int i; for(i = 0; i < sockcount; i++) { if(FD_ISSET(sockets[i], &fdset)) { SOCKADDR claddr; int clen = sizeof(claddr); int sock = accept(sockets[i], (struct sockaddr*)&claddr, &clen); cm_log("Server", "New connection accepted"); #ifdef __MINGW32__ HANDLE thread; struct pass_entry* e = malloc(sizeof(*e)); e->sock = sock; e->ssl = config.ports[i] & (1ULL << 32); e->port = config.ports[i]; e->addr = claddr; thread = (HANDLE)_beginthreadex(NULL, 0, tw_server_pass, e, 0, NULL); #else pid_t pid = fork(); if(pid == 0) { tw_server_pass(sock, config.ports[i] & (1ULL << 32), config.ports[i], claddr); _exit(0); } else { close_socket(sock); } #endif } } } } }