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
|
#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;
}
|