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
|
#include "server.h"
#include "common.h"
#include <assert.h>
#include <arpa/inet.h>
#include <errno.h>
#include <fcntl.h>
#include <netdb.h>
#include <netinet/in.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/select.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
server svr; // server
// You don't need to know how the following codes are working
void server_init(unsigned short port, int maxfd) {
struct sockaddr_in servaddr;
int val, flags;
gethostname(svr.hostname, sizeof(svr.hostname));
svr.port = port;
svr.listen_fd = socket(AF_INET, SOCK_STREAM, 0);
if (svr.listen_fd < 0)
e_err_exit("socket");
memset(&servaddr, 0, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(port);
val = 1;
if (setsockopt(svr.listen_fd, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val)) < 0)
e_err_exit("setsockopt");
if (bind(svr.listen_fd, (struct sockaddr*)&servaddr, sizeof(servaddr)) < 0)
e_err_exit("bind");
if (listen(svr.listen_fd, 1024) < 0)
e_err_exit("listen");
strcpy(svr.accept_hdr, "ACCEPT\n");
strcpy(svr.reject_hdr, "REJECT\n");
FD_ZERO(&svr.readfds);
FD_ZERO(&svr.writefds);
FD_ZERO(&svr.exceptfds);
FD_SET(svr.listen_fd, &svr.readfds);
flags = fcntl(svr.listen_fd, F_GETFL);
if(flags < 0){
perror("Warning: F_GETFL");
return;
}
fcntl(svr.listen_fd, F_SETFL, flags | O_NONBLOCK);
svr.file_table = ftab_create (maxfd);
}
void request_init(request* reqP) {
reqP->conn_fd = -1;
reqP->buf_set = 0;
reqP->buf_len = 0;
reqP->filename = NULL;
reqP->file_fd = -1;
reqP->file_info = NULL;
reqP->active = false;
reqP->header_done = false;
reqP->header_accept = false;
}
void request_free(request* reqP, server* svr) {
if (reqP->filename != NULL) {
free(reqP->filename);
reqP->filename = NULL;
}
shutdown (reqP->conn_fd, SHUT_RDWR);
close (reqP->conn_fd);
if (reqP->file_info != NULL)
fnode_unref (reqP->file_info, svr->file_table);
if (svr != NULL) {
FD_CLR (reqP->conn_fd, &(svr->readfds));
FD_CLR (reqP->conn_fd, &(svr->writefds));
FD_CLR (reqP->conn_fd, &(svr->exceptfds));
}
request_init(reqP);
}
int request_read(request* reqP) {
int r;
char buf[SVR_BUFFER_SIZE];
// Read in request from client
r = read(reqP->conn_fd, buf, sizeof(buf));
if (r < 0) return -1;
if (r == 0) return 0;
if (!reqP->header_done) {
char* p1 = strstr(buf, "\r\n");
int newline_len = 2;
// be careful that in Windows, line ends with \r\n
if (p1 == NULL) {
p1 = strchr(buf, '\n');
newline_len = 1;
assert (p1 != NULL);
}
size_t len = p1 - buf + 1;
reqP->filename = (char*)e_malloc(len);
memmove(reqP->filename, buf, len - 1);
reqP->filename[len - 1] = '\0';
p1 += newline_len;
reqP->buf_len = r - (p1 - buf);
memmove(reqP->buf, p1, reqP->buf_len);
reqP->header_done = true;
} else {
reqP->buf_len = r;
memmove(reqP->buf, buf, r);
}
return 1;
}
|
