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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
#include "server-connection.h"
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <string.h>
#include "util.h"
sds constructFilePath(const sds root, const char* file);
void sanitizeAddress(char* address);
sds* findVhost(char* address, sds** vhosts, const int vhostsc);
#define DISCONNECT printf("[%s@%d] Disconnected\n", client, fd_client); return;
void on_connection(const char* client, const int fd_client, sds **vhosts, const int vhostsc) {
printf("[%s@%d] Connected successfully!\n", client, fd_client);
/* Get address request */
char address[256];
memset(address, 0, 256);
read(fd_client, address, 256);
sanitizeAddress(address);
printf("[%s@%d] Requested %s\n", client, fd_client, address);
/* Is the username connected with any file path? */
const sds *vhost = findVhost(address, vhosts, vhostsc);
if (vhost == NULL) {
fprintf(stderr, "[%s@%d] Unknown username in address %s\n", client, fd_client, address);
printf("[%s@%d] Requested %s\n", client, fd_client, address);
DISCONNECT
}
/* Try to open the requested file or the error file */
sds filePath = constructFilePath(vhost[vh_path], strchr(address, '@') + 1);
int fd = -2;
/* Check if file is directory */
struct stat buf;
if (stat(filePath, &buf) == 0) {
if (S_ISDIR(buf.st_mode)) {
filePath = sdscat(filePath, "/index.md");
fd = 0;
}
else if (S_ISREG(buf.st_mode)) {
fd = 0;
}
else {
fprintf(stderr, "[%s@%d] %s is not a regular file!\n", client, fd_client, filePath);
}
if (fd == 0)
fd = open(filePath, O_RDONLY);
}
// Couldn't open file
if (fd < 0) {
fprintf(stderr, "[%s@%d] Error opening %s\n", client, fd_client, filePath);
sdsfree(filePath);
filePath = constructFilePath(vhost[vh_path], vhost[vh_error]);
if (sdslen(filePath) > 0)
fd = open(filePath, O_RDONLY);
}
// Couldn't open error file
if (fd < 0) {
fprintf(stderr, "[%s@%d] Error opening error file %s\n", client, fd_client, filePath);
sdsfree(filePath);
DISCONNECT
}
/* Send the file to the client */
printf("[%s@%d] Serving %s\n", client, fd_client, filePath);
sdsfree(filePath);
char buff[256];
memset(buff, 0, sizeof(buff));
while (read(fd, buff, 256)) {
write(fd_client, buff, strlen(buff));
memset(buff, 0, sizeof(buff));
}
/* Finalize */
close(fd);
printf("[%s@%d] Served!\n", client, fd_client);
DISCONNECT
}
sds constructFilePath(const sds root, const char* file) {
if (*root == '\0' && *file == '\0')
return sdsempty();
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 sanitizeAddress(char* address) {
/* Remove host and port */
char* startPath = strchr(address, '/');
if (startPath == NULL)
startPath = strchr(address, '\0');
char* startHost = strchr(address, '@');
shiftLeft(startHost + 1, address - startHost, startPath - startHost - 1);
/* Remove ../ */
for (char* prev = startHost+1, *i = startHost+1; i != NULL && *i != '\0';) {
if (i[1] == '.' && i[2] == '.' && i[3] == '/') {
shiftLeft(prev, strlen(prev), i - prev + 3);
i = prev;
}
else {
prev = i;
i = strchr(i+1, '/');
}
}
}
sds* findVhost(char* address, sds** vhosts, const int vhostsc) {
sds* vhost = NULL;
int usernameLen = strchr(address, '@') - address;
for (int i = 0; i < vhostsc; i++) {
if (strncmp(vhosts[i][vh_user], address, usernameLen) == 0) {
vhost = *vhosts + i;
break;
}
}
return vhost;
}
|
