1660a04ceSlogin #include <arpa/inet.h>
2660a04ceSlogin #include <fcntl.h>
3660a04ceSlogin #include <stdio.h>
4660a04ceSlogin #include <stdlib.h>
5660a04ceSlogin #include <string.h>
6660a04ceSlogin #include <sys/socket.h>
7660a04ceSlogin #include <sys/stat.h>
8660a04ceSlogin #include <unistd.h>
9660a04ceSlogin
10660a04ceSlogin #define PORT 12580
11660a04ceSlogin #define MAX_REQUEST_SIZE 1500
12660a04ceSlogin #define MAX_RESPONSE_SIZE 1500
13660a04ceSlogin // 网页根目录
14c719ddc6SSaga1718 #define WEB_ROOT "/var/www/html/"
15660a04ceSlogin #define EXIT_CODE 1
16660a04ceSlogin #define min(a, b) ((a) < (b) ? (a) : (b))
17660a04ceSlogin
18660a04ceSlogin #define DEFAULT_PAGE "/index.html"
19660a04ceSlogin
security_check(char * path)20660a04ceSlogin int security_check(char *path)
21660a04ceSlogin {
22660a04ceSlogin // 检查路径是否包含 ..
23660a04ceSlogin if (strstr(path, ".."))
24660a04ceSlogin {
25660a04ceSlogin return 0;
26660a04ceSlogin }
27660a04ceSlogin return 1;
28660a04ceSlogin }
29660a04ceSlogin
send_response(int sockfd,char * response)30660a04ceSlogin ssize_t send_response(int sockfd, char *response)
31660a04ceSlogin {
32660a04ceSlogin return write(sockfd, response, strlen(response));
33660a04ceSlogin }
34660a04ceSlogin
send_header(int sockfd,int content_length,char * path)35660a04ceSlogin void send_header(int sockfd, int content_length, char *path)
36660a04ceSlogin {
37660a04ceSlogin char buffer[MAX_RESPONSE_SIZE];
38660a04ceSlogin // 获取文件类型
39660a04ceSlogin char *content_type;
40660a04ceSlogin if (strstr(path, ".html"))
41660a04ceSlogin {
42660a04ceSlogin content_type = "text/html";
43660a04ceSlogin }
44660a04ceSlogin else if (strstr(path, ".css"))
45660a04ceSlogin {
46660a04ceSlogin content_type = "text/css";
47660a04ceSlogin }
48660a04ceSlogin else if (strstr(path, ".js"))
49660a04ceSlogin {
50660a04ceSlogin content_type = "application/javascript";
51660a04ceSlogin }
52660a04ceSlogin else if (strstr(path, ".png"))
53660a04ceSlogin {
54660a04ceSlogin content_type = "image/png";
55660a04ceSlogin }
56660a04ceSlogin else if (strstr(path, ".jpg"))
57660a04ceSlogin {
58660a04ceSlogin content_type = "image/jpeg";
59660a04ceSlogin }
60660a04ceSlogin else if (strstr(path, ".gif"))
61660a04ceSlogin {
62660a04ceSlogin content_type = "image/gif";
63660a04ceSlogin }
64660a04ceSlogin else
65660a04ceSlogin {
66660a04ceSlogin content_type = "text/plain;charset=utf-8";
67660a04ceSlogin }
68660a04ceSlogin sprintf(buffer, "HTTP/1.1 200 OK\nContent-Type: %s\nContent-Length: %d\n\n", content_type, content_length);
69660a04ceSlogin send_response(sockfd, buffer);
70660a04ceSlogin }
71660a04ceSlogin
send_file(int sockfd,char * path)72660a04ceSlogin void send_file(int sockfd, char *path)
73660a04ceSlogin {
74660a04ceSlogin printf("send_file: path: %s\n", path);
75660a04ceSlogin
76660a04ceSlogin int fd = open(path, 0);
77660a04ceSlogin if (fd == -1)
78660a04ceSlogin {
79660a04ceSlogin send_response(
80660a04ceSlogin sockfd,
81660a04ceSlogin "HTTP/1.1 404 Not Found\nContent-Type: text/html\n\n<html><body><h1>404 Not Found</h1><p>DragonOS Http Server</p></body></html>");
82660a04ceSlogin return;
83660a04ceSlogin }
84660a04ceSlogin
85660a04ceSlogin int content_length = lseek(fd, 0, SEEK_END);
86660a04ceSlogin int remaining = content_length;
87660a04ceSlogin printf("send_file: content_length: %d\n", content_length);
88660a04ceSlogin lseek(fd, 0, SEEK_SET);
89660a04ceSlogin send_header(sockfd, content_length, path);
90660a04ceSlogin
91660a04ceSlogin char buffer[1048576];
92660a04ceSlogin int readSize;
93660a04ceSlogin while (remaining)
94660a04ceSlogin {
95660a04ceSlogin // 由于磁盘IO耗时较长，所以每次读取1MB，然后再分批发送
96660a04ceSlogin int to_read = min(1048576, remaining);
97660a04ceSlogin readSize = read(fd, &buffer, to_read);
98660a04ceSlogin
99660a04ceSlogin remaining -= readSize;
100660a04ceSlogin void *p = buffer;
101660a04ceSlogin while (readSize > 0)
102660a04ceSlogin {
103660a04ceSlogin int wsize = write(sockfd, p, min(readSize, MAX_RESPONSE_SIZE));
104660a04ceSlogin if (wsize <= 0)
105660a04ceSlogin {
106660a04ceSlogin printf("send_file failed: wsize: %d\n", wsize);
107660a04ceSlogin close(fd);
108660a04ceSlogin return;
109660a04ceSlogin }
110660a04ceSlogin p += wsize;
111660a04ceSlogin readSize -= wsize;
112660a04ceSlogin }
113660a04ceSlogin }
114660a04ceSlogin
115660a04ceSlogin close(fd);
116660a04ceSlogin }
117660a04ceSlogin
handle_request(int sockfd,char * request)118660a04ceSlogin void handle_request(int sockfd, char *request)
119660a04ceSlogin {
120660a04ceSlogin char *method, *url, *http_version;
121660a04ceSlogin char path[MAX_REQUEST_SIZE];
122660a04ceSlogin
123660a04ceSlogin method = strtok(request, " ");
124660a04ceSlogin url = strtok(NULL, " ");
125660a04ceSlogin http_version = strtok(NULL, "\r\n");
126660a04ceSlogin
127660a04ceSlogin printf("handle_request: method: %s, url: %s, http_version: %s\n", method, url, http_version);
128660a04ceSlogin // 检查空指针等异常情况
129660a04ceSlogin if (method == NULL || url == NULL || http_version == NULL)
130660a04ceSlogin {
131660a04ceSlogin send_response(sockfd, "HTTP/1.1 400 Bad Request\nContent-Type: text/html\n\n<html><body><h1>400 Bad "
132660a04ceSlogin "Request</h1><p>DragonOS Http Server</p></body></html>");
133660a04ceSlogin return;
134660a04ceSlogin }
135660a04ceSlogin // 检查url是否为空
136660a04ceSlogin if (strlen(url) == 0)
137660a04ceSlogin {
138660a04ceSlogin send_response(sockfd, "HTTP/1.1 400 Bad Request\nContent-Type: text/html\n\n<html><body><h1>400 Bad "
139660a04ceSlogin "Request</h1><p>DragonOS Http Server</p></body></html>");
140660a04ceSlogin return;
141660a04ceSlogin }
142660a04ceSlogin int default_page = 0;
143660a04ceSlogin if (url[strlen(url) - 1] == '/')
144660a04ceSlogin {
145660a04ceSlogin default_page = 1;
146660a04ceSlogin }
147660a04ceSlogin
148660a04ceSlogin if (strcmp(method, "GET") == 0)
149660a04ceSlogin {
150660a04ceSlogin if (default_page)
151660a04ceSlogin {
152660a04ceSlogin sprintf(path, "%s%s%s", WEB_ROOT, url, DEFAULT_PAGE);
153660a04ceSlogin }
154660a04ceSlogin else
155660a04ceSlogin {
156660a04ceSlogin sprintf(path, "%s%s", WEB_ROOT, url);
157660a04ceSlogin }
158660a04ceSlogin if (!security_check(path))
159660a04ceSlogin {
160660a04ceSlogin send_response(
161660a04ceSlogin sockfd,
162660a04ceSlogin "HTTP/1.1 403 Forbidden\nContent-Type: text/html\n\n<html><body><h1>403 Forbidden</h1><p>DragonOS Http Server</p></body></html>");
163660a04ceSlogin return;
164660a04ceSlogin }
165660a04ceSlogin send_file(sockfd, path);
166660a04ceSlogin }
167660a04ceSlogin else
168660a04ceSlogin {
169660a04ceSlogin send_response(sockfd, "HTTP/1.1 501 Not Implemented\nContent-Type: text/html\n\n<html><body><h1>501 Not "
170660a04ceSlogin "Implemented</h1><p>DragonOS Http Server</p></body></html>");
171660a04ceSlogin }
172660a04ceSlogin }
173660a04ceSlogin
main(int argc,char const * argv[])174660a04ceSlogin int main(int argc, char const *argv[])
175660a04ceSlogin {
176660a04ceSlogin int server_fd, new_socket, valread;
177660a04ceSlogin struct sockaddr_in address;
178660a04ceSlogin int addrlen = sizeof(address);
179660a04ceSlogin char buffer[MAX_REQUEST_SIZE] = {0};
180660a04ceSlogin int opt = 1;
181660a04ceSlogin
182660a04ceSlogin // 创建socket
183660a04ceSlogin if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0)
184660a04ceSlogin {
185660a04ceSlogin perror("socket failed");
186660a04ceSlogin exit(EXIT_CODE);
187660a04ceSlogin }
188660a04ceSlogin
189660a04ceSlogin // 设置socket选项，允许地址重用
190660a04ceSlogin // if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt)))
191660a04ceSlogin // {
192660a04ceSlogin // perror("setsockopt failed");
193660a04ceSlogin // exit(EXIT_CODE);
194660a04ceSlogin // }
195660a04ceSlogin
196660a04ceSlogin // 设置地址和端口
197660a04ceSlogin address.sin_family = AF_INET;
198660a04ceSlogin address.sin_addr.s_addr = INADDR_ANY;
199660a04ceSlogin address.sin_port = htons(PORT);
200660a04ceSlogin
201660a04ceSlogin // 把socket绑定到地址和端口上
202660a04ceSlogin if (bind(server_fd, (struct sockaddr *)&address, sizeof(address)) < 0)
203660a04ceSlogin {
204660a04ceSlogin perror("bind failed");
205660a04ceSlogin exit(EXIT_CODE);
206660a04ceSlogin }
207660a04ceSlogin
208660a04ceSlogin // 监听socket
209660a04ceSlogin if (listen(server_fd, 3) < 0)
210660a04ceSlogin {
211660a04ceSlogin perror("listen failed");
212660a04ceSlogin exit(EXIT_CODE);
213660a04ceSlogin }
214660a04ceSlogin
215660a04ceSlogin while (1)
216660a04ceSlogin {
217660a04ceSlogin printf("Waiting for a client...\n");
218660a04ceSlogin
219660a04ceSlogin // 等待并接受客户端连接
220660a04ceSlogin if ((new_socket = accept(server_fd, (struct sockaddr *)&address, (socklen_t *)&addrlen)) < 0)
221660a04ceSlogin {
222660a04ceSlogin perror("accept failed");
223660a04ceSlogin exit(EXIT_CODE);
224660a04ceSlogin }
225660a04ceSlogin
226660a04ceSlogin // 接收客户端消息
227660a04ceSlogin valread = read(new_socket, buffer, MAX_REQUEST_SIZE);
228660a04ceSlogin printf("%s\n", buffer);
229660a04ceSlogin
230660a04ceSlogin // 处理请求
231660a04ceSlogin handle_request(new_socket, buffer);
232660a04ceSlogin
233660a04ceSlogin // 关闭客户端连接
234660a04ceSlogin close(new_socket);
235660a04ceSlogin }
236*37cef00bSSamuel Dai // 关闭tcp socket
237*37cef00bSSamuel Dai close(server_fd);
238660a04ceSlogin
239660a04ceSlogin return 0;
240660a04ceSlogin }