tsdwl: little tcp server

1 files changed, 125 insertions, 0 deletions

diff --git a/Year_3/TSDWL/webserver/server.c b/Year_3/TSDWL/webserver/server.c
new file mode 100644
index 0000000..9bf1f7d
--- /dev/null
+++ b/Year_3/TSDWL/webserver/server.c
@@ -0,0 +1,125 @@
+#include <arpa/inet.h>
+#include <netinet/in.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <unistd.h>
+
+#define DEFAULT_WWW "./www"
+
+/* TODO: Make the parsing of headers */
+void
+read_headers(int* fd, char** headers, int* headers_num)
+{
+    char ch;
+    int flag_read; /* Used to check the return value of `read` function */
+    unsigned i;
+    char buffer[1024];
+
+    *headers_num = 0;
+
+    while (1) {
+        i = 0;
+
+        buffer[0] = '\0';
+        headers[*headers_num] = malloc(sizeof(char) * 1024);
+        while ((flag_read = read(*fd, &ch, 1)) != 0) {
+            if (flag_read == -1) {
+                perror("Error reading the headers");
+                exit(1);
+            }
+
+            *(buffer + i) = ch;
+            if (ch == '\n' && (i < 1 || buffer[i - 1] == '\r')) {
+                break;
+            }
+
+            i++;
+        }
+
+        buffer[i + 1] = '\0';
+        if (i < 2) {
+            break;
+        }
+
+        memcpy(headers[*headers_num], buffer, i + 1);
+        (*headers_num)++;
+    }
+}
+
+void
+get_client_ip(int fd, char* buffer)
+{
+    struct sockaddr_in addr;
+    socklen_t addr_size = sizeof(struct sockaddr_in);
+    getpeername(fd, (struct sockaddr*)&addr, &addr_size);
+    strcpy(buffer, inet_ntoa(addr.sin_addr));
+}
+
+int
+main(int argc, char* argv[])
+{
+    char* www_path;
+    char** headers;
+    int headers_num;
+    unsigned i;
+    int sockfd, clientfd;
+    struct sockaddr_in address;
+    char client_ip[20];
+    socklen_t address_len;
+
+    if (argc < 2) {
+        fprintf(stderr, "Usage: %s <port> [root_path]", argv[0]);
+        exit(1);
+    }
+
+    www_path = (argc >= 3) ? argv[2] : DEFAULT_WWW;
+
+    /* Max 24 headers for now on the stack */
+    headers = malloc(sizeof(char*) * 24);
+
+    if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) == -1) {
+        perror("Error creating a new socket");
+        exit(1);
+    }
+
+    address.sin_addr.s_addr = INADDR_ANY;
+    address.sin_port = htons(atoi(argv[1]));
+    address.sin_family = AF_INET;
+
+    address_len = sizeof(address);
+
+    if (bind(sockfd, (struct sockaddr*)&address, address_len) == -1) {
+        perror("Error binding the socket");
+        exit(1);
+    }
+
+    if (listen(sockfd, 1) == -1) {
+        perror("Unable to listen");
+        exit(1);
+    }
+
+    while (1) {
+        if ((clientfd = accept(sockfd, (struct sockaddr*)&address, &address_len)) == -1) {
+            perror("Error with client connection, maybe try again later");
+            continue;
+        }
+
+        get_client_ip(clientfd, client_ip);
+        printf("Connection from '%s':\n", client_ip);
+
+        read_headers(&clientfd, headers, &headers_num);
+
+        for (i = 0; i < headers_num; ++i) {
+            printf("%s", headers[i]);
+            free(headers + i);
+        }
+
+        printf("[Connection closed]\n\n");
+        close(clientfd);
+    }
+
+    close(sockfd);
+    return 0;
+}