xref: /DragonOS/user/apps/http_server/main.c (revision 660a04cef803fd73e9b294b30a96421b021a4b9b) 
1*660a04ceSlogin #include <arpa/inet.h>
2*660a04ceSlogin #include <fcntl.h>
3*660a04ceSlogin #include <stdio.h>
4*660a04ceSlogin #include <stdlib.h>
5*660a04ceSlogin #include <string.h>
6*660a04ceSlogin #include <sys/socket.h>
7*660a04ceSlogin #include <sys/stat.h>
8*660a04ceSlogin #include <unistd.h>
9*660a04ceSlogin 
10*660a04ceSlogin #define PORT 12580
11*660a04ceSlogin #define MAX_REQUEST_SIZE 1500
12*660a04ceSlogin #define MAX_RESPONSE_SIZE 1500
13*660a04ceSlogin // 网页根目录
14*660a04ceSlogin #define WEB_ROOT "/wwwroot/First-WebPage-On-DragonOS"
15*660a04ceSlogin #define EXIT_CODE 1
16*660a04ceSlogin #define min(a, b) ((a) < (b) ? (a) : (b))
17*660a04ceSlogin 
18*660a04ceSlogin #define DEFAULT_PAGE "/index.html"
19*660a04ceSlogin 
20*660a04ceSlogin int security_check(char *path)
21*660a04ceSlogin {
22*660a04ceSlogin     // 检查路径是否包含 ..
23*660a04ceSlogin     if (strstr(path, ".."))
24*660a04ceSlogin     {
25*660a04ceSlogin         return 0;
26*660a04ceSlogin     }
27*660a04ceSlogin     return 1;
28*660a04ceSlogin }
29*660a04ceSlogin 
30*660a04ceSlogin ssize_t send_response(int sockfd, char *response)
31*660a04ceSlogin {
32*660a04ceSlogin     return write(sockfd, response, strlen(response));
33*660a04ceSlogin }
34*660a04ceSlogin 
35*660a04ceSlogin void send_header(int sockfd, int content_length, char *path)
36*660a04ceSlogin {
37*660a04ceSlogin     char buffer[MAX_RESPONSE_SIZE];
38*660a04ceSlogin     // 获取文件类型
39*660a04ceSlogin     char *content_type;
40*660a04ceSlogin     if (strstr(path, ".html"))
41*660a04ceSlogin     {
42*660a04ceSlogin         content_type = "text/html";
43*660a04ceSlogin     }
44*660a04ceSlogin     else if (strstr(path, ".css"))
45*660a04ceSlogin     {
46*660a04ceSlogin         content_type = "text/css";
47*660a04ceSlogin     }
48*660a04ceSlogin     else if (strstr(path, ".js"))
49*660a04ceSlogin     {
50*660a04ceSlogin         content_type = "application/javascript";
51*660a04ceSlogin     }
52*660a04ceSlogin     else if (strstr(path, ".png"))
53*660a04ceSlogin     {
54*660a04ceSlogin         content_type = "image/png";
55*660a04ceSlogin     }
56*660a04ceSlogin     else if (strstr(path, ".jpg"))
57*660a04ceSlogin     {
58*660a04ceSlogin         content_type = "image/jpeg";
59*660a04ceSlogin     }
60*660a04ceSlogin     else if (strstr(path, ".gif"))
61*660a04ceSlogin     {
62*660a04ceSlogin         content_type = "image/gif";
63*660a04ceSlogin     }
64*660a04ceSlogin     else
65*660a04ceSlogin     {
66*660a04ceSlogin         content_type = "text/plain;charset=utf-8";
67*660a04ceSlogin     }
68*660a04ceSlogin     sprintf(buffer, "HTTP/1.1 200 OK\nContent-Type: %s\nContent-Length: %d\n\n", content_type, content_length);
69*660a04ceSlogin     send_response(sockfd, buffer);
70*660a04ceSlogin }
71*660a04ceSlogin 
72*660a04ceSlogin void send_file(int sockfd, char *path)
73*660a04ceSlogin {
74*660a04ceSlogin     printf("send_file: path: %s\n", path);
75*660a04ceSlogin 
76*660a04ceSlogin     int fd = open(path, 0);
77*660a04ceSlogin     if (fd == -1)
78*660a04ceSlogin     {
79*660a04ceSlogin         send_response(
80*660a04ceSlogin             sockfd,
81*660a04ceSlogin             "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>");
82*660a04ceSlogin         return;
83*660a04ceSlogin     }
84*660a04ceSlogin 
85*660a04ceSlogin     int content_length = lseek(fd, 0, SEEK_END);
86*660a04ceSlogin     int remaining = content_length;
87*660a04ceSlogin     printf("send_file: content_length: %d\n", content_length);
88*660a04ceSlogin     lseek(fd, 0, SEEK_SET);
89*660a04ceSlogin     send_header(sockfd, content_length, path);
90*660a04ceSlogin 
91*660a04ceSlogin     char buffer[1048576];
92*660a04ceSlogin     int readSize;
93*660a04ceSlogin     while (remaining)
94*660a04ceSlogin     {
95*660a04ceSlogin         // 由于磁盘IO耗时较长，所以每次读取1MB，然后再分批发送
96*660a04ceSlogin         int to_read = min(1048576, remaining);
97*660a04ceSlogin         readSize = read(fd, &buffer, to_read);
98*660a04ceSlogin 
99*660a04ceSlogin         remaining -= readSize;
100*660a04ceSlogin         void *p = buffer;
101*660a04ceSlogin         while (readSize > 0)
102*660a04ceSlogin         {
103*660a04ceSlogin             int wsize = write(sockfd, p, min(readSize, MAX_RESPONSE_SIZE));
104*660a04ceSlogin             if (wsize <= 0)
105*660a04ceSlogin             {
106*660a04ceSlogin                 printf("send_file failed: wsize: %d\n", wsize);
107*660a04ceSlogin                 close(fd);
108*660a04ceSlogin                 return;
109*660a04ceSlogin             }
110*660a04ceSlogin             p += wsize;
111*660a04ceSlogin             readSize -= wsize;
112*660a04ceSlogin         }
113*660a04ceSlogin     }
114*660a04ceSlogin 
115*660a04ceSlogin     close(fd);
116*660a04ceSlogin }
117*660a04ceSlogin 
118*660a04ceSlogin void handle_request(int sockfd, char *request)
119*660a04ceSlogin {
120*660a04ceSlogin     char *method, *url, *http_version;
121*660a04ceSlogin     char path[MAX_REQUEST_SIZE];
122*660a04ceSlogin 
123*660a04ceSlogin     method = strtok(request, " ");
124*660a04ceSlogin     url = strtok(NULL, " ");
125*660a04ceSlogin     http_version = strtok(NULL, "\r\n");
126*660a04ceSlogin 
127*660a04ceSlogin     printf("handle_request: method: %s, url: %s, http_version: %s\n", method, url, http_version);
128*660a04ceSlogin     // 检查空指针等异常情况
129*660a04ceSlogin     if (method == NULL || url == NULL || http_version == NULL)
130*660a04ceSlogin     {
131*660a04ceSlogin         send_response(sockfd, "HTTP/1.1 400 Bad Request\nContent-Type: text/html\n\n<html><body><h1>400 Bad "
132*660a04ceSlogin                               "Request</h1><p>DragonOS Http Server</p></body></html>");
133*660a04ceSlogin         return;
134*660a04ceSlogin     }
135*660a04ceSlogin     // 检查url是否为空
136*660a04ceSlogin     if (strlen(url) == 0)
137*660a04ceSlogin     {
138*660a04ceSlogin         send_response(sockfd, "HTTP/1.1 400 Bad Request\nContent-Type: text/html\n\n<html><body><h1>400 Bad "
139*660a04ceSlogin                               "Request</h1><p>DragonOS Http Server</p></body></html>");
140*660a04ceSlogin         return;
141*660a04ceSlogin     }
142*660a04ceSlogin     int default_page = 0;
143*660a04ceSlogin     if (url[strlen(url) - 1] == '/')
144*660a04ceSlogin     {
145*660a04ceSlogin         default_page = 1;
146*660a04ceSlogin     }
147*660a04ceSlogin 
148*660a04ceSlogin     if (strcmp(method, "GET") == 0)
149*660a04ceSlogin     {
150*660a04ceSlogin         if (default_page)
151*660a04ceSlogin         {
152*660a04ceSlogin             sprintf(path, "%s%s%s", WEB_ROOT, url, DEFAULT_PAGE);
153*660a04ceSlogin         }
154*660a04ceSlogin         else
155*660a04ceSlogin         {
156*660a04ceSlogin             sprintf(path, "%s%s", WEB_ROOT, url);
157*660a04ceSlogin         }
158*660a04ceSlogin         if (!security_check(path))
159*660a04ceSlogin         {
160*660a04ceSlogin             send_response(
161*660a04ceSlogin                 sockfd,
162*660a04ceSlogin                 "HTTP/1.1 403 Forbidden\nContent-Type: text/html\n\n<html><body><h1>403 Forbidden</h1><p>DragonOS Http Server</p></body></html>");
163*660a04ceSlogin             return;
164*660a04ceSlogin         }
165*660a04ceSlogin         send_file(sockfd, path);
166*660a04ceSlogin     }
167*660a04ceSlogin     else
168*660a04ceSlogin     {
169*660a04ceSlogin         send_response(sockfd, "HTTP/1.1 501 Not Implemented\nContent-Type: text/html\n\n<html><body><h1>501 Not "
170*660a04ceSlogin                               "Implemented</h1><p>DragonOS Http Server</p></body></html>");
171*660a04ceSlogin     }
172*660a04ceSlogin }
173*660a04ceSlogin 
174*660a04ceSlogin int main(int argc, char const *argv[])
175*660a04ceSlogin {
176*660a04ceSlogin     int server_fd, new_socket, valread;
177*660a04ceSlogin     struct sockaddr_in address;
178*660a04ceSlogin     int addrlen = sizeof(address);
179*660a04ceSlogin     char buffer[MAX_REQUEST_SIZE] = {0};
180*660a04ceSlogin     int opt = 1;
181*660a04ceSlogin 
182*660a04ceSlogin     // 创建socket
183*660a04ceSlogin     if ((server_fd = socket(AF_INET, SOCK_STREAM, 0)) == 0)
184*660a04ceSlogin     {
185*660a04ceSlogin         perror("socket failed");
186*660a04ceSlogin         exit(EXIT_CODE);
187*660a04ceSlogin     }
188*660a04ceSlogin 
189*660a04ceSlogin     // 设置socket选项，允许地址重用
190*660a04ceSlogin     // if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &opt, sizeof(opt)))
191*660a04ceSlogin     // {
192*660a04ceSlogin     //     perror("setsockopt failed");
193*660a04ceSlogin     //     exit(EXIT_CODE);
194*660a04ceSlogin     // }
195*660a04ceSlogin 
196*660a04ceSlogin     // 设置地址和端口
197*660a04ceSlogin     address.sin_family = AF_INET;
198*660a04ceSlogin     address.sin_addr.s_addr = INADDR_ANY;
199*660a04ceSlogin     address.sin_port = htons(PORT);
200*660a04ceSlogin 
201*660a04ceSlogin     // 把socket绑定到地址和端口上
202*660a04ceSlogin     if (bind(server_fd, (struct sockaddr *)&address, sizeof(address)) < 0)
203*660a04ceSlogin     {
204*660a04ceSlogin         perror("bind failed");
205*660a04ceSlogin         exit(EXIT_CODE);
206*660a04ceSlogin     }
207*660a04ceSlogin 
208*660a04ceSlogin     // 监听socket
209*660a04ceSlogin     if (listen(server_fd, 3) < 0)
210*660a04ceSlogin     {
211*660a04ceSlogin         perror("listen failed");
212*660a04ceSlogin         exit(EXIT_CODE);
213*660a04ceSlogin     }
214*660a04ceSlogin 
215*660a04ceSlogin     while (1)
216*660a04ceSlogin     {
217*660a04ceSlogin         printf("Waiting for a client...\n");
218*660a04ceSlogin 
219*660a04ceSlogin         // 等待并接受客户端连接
220*660a04ceSlogin         if ((new_socket = accept(server_fd, (struct sockaddr *)&address, (socklen_t *)&addrlen)) < 0)
221*660a04ceSlogin         {
222*660a04ceSlogin             perror("accept failed");
223*660a04ceSlogin             exit(EXIT_CODE);
224*660a04ceSlogin         }
225*660a04ceSlogin 
226*660a04ceSlogin         // 接收客户端消息
227*660a04ceSlogin         valread = read(new_socket, buffer, MAX_REQUEST_SIZE);
228*660a04ceSlogin         printf("%s\n", buffer);
229*660a04ceSlogin 
230*660a04ceSlogin         // 处理请求
231*660a04ceSlogin         handle_request(new_socket, buffer);
232*660a04ceSlogin 
233*660a04ceSlogin         // 关闭客户端连接
234*660a04ceSlogin         close(new_socket);
235*660a04ceSlogin     }
236*660a04ceSlogin 
237*660a04ceSlogin     return 0;
238*660a04ceSlogin }