1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
/* The server recieves connections and passes files to the clients
*/
#include <server-connection.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <fcntl.h>
#include <sys/wait.h>
#include <string.h>
#include <sds/sds.h>
#include <util.h>
sds constructFilePath(const sds root, const char* file) {
sds path = sdsdup(root);
if (root[sdslen(root)-1] != '/' && file[0] != '/')
path = sdscat(path, "/");
path = sdscat(path, file);
if (file[strlen(file)-1] == '/')
path = sdscat(path, "index.md");
return path;
}
void freeVhosts(sds **vhosts, int argc) {
for (int i = 1; i < argc; i++) {
sdsfreesplitres(vhosts[i-1], 3);
}
free(vhosts);
}
int main(int argc, char* argv[]) {
/*
* Get hosts
*/
sds **vhosts = malloc((argc - 1) * sizeof(sds*));
for (int i = 1, temp = 0; i < argc; i++) {
vhosts[i-1] = sdssplitlen(argv[i], strlen(argv[i]), ",", 1, &temp);
}
/*
* Create socket for accepting connections
*/
int fd_socket;
herr(fd_socket = socket(AF_INET, SOCK_STREAM, 0), "socket");
struct sockaddr_in sa_socket = {
.sin_family = AF_INET,
.sin_port = inet_atop("8080"),
};
herr(inet_aton("127.0.0.1", &sa_socket.sin_addr.s_addr), "inet_aton");
// Reuse address when in TIME_WAIT state, after listening socket was closed
// https://stackoverflow.com/a/10651048/12036073
// https://superuser.com/questions/173535/what-are-close-wait-and-time-wait-states#comment951880_173543
int true = 1;
herr(setsockopt(fd_socket, SOL_SOCKET, SO_REUSEADDR, &true, sizeof(int)), "setsockopt");
herr(bind(fd_socket, (struct sockaddr*)&sa_socket, sizeof(struct sockaddr_in)), "bind");
herr(listen(fd_socket, 50), "listen");
/*
* Accept connection on the socket
*/
struct sockaddr_in sa_client;
socklen_t sa_client_size = sizeof(struct sockaddr_in);
int fd_client;
int count = 0;
while (count < 5) {
herr(fd_client = accept(fd_socket, (struct sockaddr*)&sa_client, &sa_client_size), "accept");
char* strAddr = inet_ntoa(sa_client.sin_addr);
count++;
int fp = fork();
if (fp == 0) {
close(fd_socket);
on_connection(strAddr, fd_client, vhosts, argc - 1);
close(fd_client);
freeVhosts(vhosts, argc);
return 0;
}
close(fd_client);
}
close(fd_socket);
printf("Exiting");
while(wait(NULL) > 0);
freeVhosts(vhosts, argc);
}
|
