summaryrefslogtreecommitdiff
path: root/Year_3/TSDWL/webserver/server.c
diff options
context:
space:
mode:
Diffstat (limited to 'Year_3/TSDWL/webserver/server.c')
-rw-r--r--Year_3/TSDWL/webserver/server.c125
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;
+}