very simple work withs html based on file path

1 files changed, 158 insertions, 0 deletions

diff --git a/src/main.c b/src/main.c
new file mode 100644
index 0000000..78a1069
--- /dev/null
+++ b/src/main.c
@@ -0,0 +1,158 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <errno.h>
+#include <signal.h>
+
+#include "log.h"
+#include "server.h"
+
+#define BUF_CAP 80000
+#define PORT "8079"
+
+#define SEND_BUF_ADD(str) do {                               \
+        memcpy(&send_buf[*send_buf_sz], str, strlen(str));   \
+        *send_buf_sz += strlen(str);                         \
+    } while(0)
+
+#define SEND_BUF_ADD_LINE(str) do {                  \
+        SEND_BUF_ADD(str);                           \
+        SEND_BUF_ADD("\r\n");                        \
+    } while(0)
+
+static int try_file(char *req_path, FILE **fp)
+{
+    char file_path[1024];
+    sprintf(file_path, "%s%s", FILES,
+            (strlen(req_path) == 1) ? "/index.html" : req_path);
+
+    *fp = fopen(file_path, "r");
+    if(!(*fp)) {
+        err("fopen: file %s: %s", file_path, strerror(errno));
+        return 1;
+    }
+    info("fopen: file %s was opened", file_path);
+    return 0;
+}
+
+static int on_get(char *send_buf, ssize_t *send_buf_sz, char *req_path)
+{
+    SEND_BUF_ADD("HTTP/1.1 ");
+
+    FILE *fp;
+    if(try_file(req_path, &fp) != 0) {
+        if(errno == 2) { // 2 is 'No such file or directory'
+            SEND_BUF_ADD_LINE("404");
+            if(try_file("/404.html", &fp) != 0) return 1;
+        } else {
+            SEND_BUF_ADD_LINE("500");
+            if(try_file("/505.html", &fp) != 0) return 1;
+        }
+    } else {
+        SEND_BUF_ADD_LINE("200");
+    }
+
+    SEND_BUF_ADD_LINE("Server: potato");
+    // SEND_BUF_ADD_LINE("Content-Type: image/x-icon");
+    SEND_BUF_ADD("\r\n");
+
+    // fseek(fp, 0, SEEK_END);
+    // size_t file_sz = ftell(fp);
+    // rewind(fp);
+
+    // fread(&send_buf[*send_buf_sz], sizeof(char), file_sz, fp);
+    // *send_buf_sz += file_sz;
+
+    char line[BUF_CAP];
+    while(fgets(line, sizeof(line), fp) != NULL)
+    {
+        if(line[strlen(line)-1] == '\n') line[strlen(line)-1] = '\0';
+        SEND_BUF_ADD_LINE(line);
+    }
+
+    fclose(fp);
+    return 0;
+}
+
+static int handle_connection(sock_t *conn)
+{
+    char recv_buf[BUF_CAP] = {0};
+    ssize_t recv_buf_len = server_recv(conn, recv_buf, BUF_CAP);
+
+    if(recv_buf_len == 0) {
+        return 0;
+    } else if(recv_buf_len < 0) {
+        err("server_recv: %s", strerror(errno));
+        return 1;
+    }
+
+    /* HANDLE REQUEST */
+
+    char send_buf[BUF_CAP];
+    ssize_t send_buf_sz = 0;
+
+    char recv_buf_cpy[BUF_CAP];
+    memcpy(recv_buf_cpy, recv_buf, recv_buf_len);
+
+    char *req_method = strtok(recv_buf_cpy, " ");
+    if(strcmp(req_method, "GET") == 0) {
+        if(on_get(send_buf, &send_buf_sz, strtok(NULL, " "))) {
+            err("on_get: failed");
+            return 1;
+        }
+    } else {
+        err("request method %s has not been implemented", req_method);
+        return 1;
+    }
+
+    // TODO: handle when the whole message is not sent
+    ssize_t sent_sz = server_send(conn, send_buf, send_buf_sz);
+    if(sent_sz < 0) {
+        err("server_send: %s", strerror(errno));
+        return 1;
+    }
+
+    info("server_send: sent %ld out of %ld bytes", sent_sz, send_buf_sz);
+    return 0;
+}
+
+int main(void)
+{
+    signal(SIGCHLD, SIG_IGN);  // now i dont have to wait()
+
+    sock_t *sock = server_sock_create();
+
+    if(server_start(PORT, sock) != 0) {
+        err("server_init: failed");
+        return 1;
+    }
+
+    for(int i = 0; ; i++) {
+        sock_t *conn = server_sock_create();
+        if(server_accept(sock, conn) != 0) {
+            err("server_accept: failed");
+        }
+
+        info("server: got connection with %s", server_connection_ip(conn));
+
+        pid_t p = fork();
+        if(p == 0) {
+            server_sock_close(sock);
+            if(handle_connection(conn) != 0) {
+                err("handle_connection: failed");
+            } else {
+                info("server: got disconnected with %s", server_connection_ip(conn));
+            }
+            server_sock_close(conn);
+            return 0;
+        } else if (p < 0){
+            err("fork: %s", strerror(errno));
+            return 1;
+        }
+        server_sock_close(conn);
+    }
+
+    server_sock_close(sock);
+    return 0;
+}