Linux高并发web服务器开发 【转载】 include

 1 void http_respond_head(int cfd, char* type)
 2 {
 3     char buf[1024];
 4     // 状态行
 5     sprintf(buf, "http/1.1 200 OK
");
 6     write(cfd, buf, strlen(buf));
 7     // 消息报头
 8     sprintf(buf, "Content-Type: %s
", type);
 9     write(cfd, buf, strlen(buf));
10     
11     // 空行
12     write(cfd, "
", 2);
13 }
14 
15 void main()
16 {
17     // 修改进程的工作目录
18     chdir(path);
19     // 创建监听的套接字
20     int lfd = socket(af_inet, sock_stream, 0);
21     // 绑定
22     struct sockaddr_in serv;
23     serv.family = af_inet;
24     serv.port = htons(8989);
25     bind(lfd, &serv, len);
26     // 监听
27     listen();
28     
29     int cfd = accept();
30     // 读数据
31     read(cfd, buf, sizeof(buf));
32     // 先将buf中的请求行拿出来
33     // GET /hello.c http/1.1
34     char method[12], path[1024], protocol[12];
35     // 得到文件名
36     char* file = path+1;
37     // 打开文件
38     int fdd = open(file, O_RDONLY);
39     int len = 0;
40     http_respond_head(cfd, "text/plain");
41     // 循环读数据
42     while( (len=read(fdd, buf, sizeof(buf))) > 0)
43     {
44         // 数据发送给浏览器
45         write(fdd, buf, len));
46     }
47 }

  1 #include <stdio.h>
  2 #include <unistd.h>
  3 #include <stdlib.h>
  4 #include <sys/types.h>
  5 #include <string.h>
  6 #include <sys/epoll.h>
  7 #include <arpa/inet.h>
  8 #include <fcntl.h>
  9 #include <dirent.h>
 10 #include <sys/stat.h>
 11 #include <ctype.h>
 12 #include "epoll_server.h"
 13 
 14 #define MAXSIZE 2000
 15 
 16 void epoll_run(int port)
 17 {
 18     // 创建一个epoll树的根节点
 19     int epfd = epoll_create(MAXSIZE);
 20     if(epfd == -1)
 21     {
 22         perror("epoll_create error");
 23         exit(1);
 24     }
 25 
 26     // 添加要监听的节点
 27     // 先添加监听lfd
 28     int lfd = init_listen_fd(port, epfd);
 29 
 30     // 委托内核检测添加到树上的节点
 31     struct epoll_event all[MAXSIZE];
 32     while(1)
 33     {
 34         int ret = epoll_wait(epfd, all, MAXSIZE, -1);
 35         if(ret == -1)
 36         {
 37             perror("epoll_wait error");
 38             exit(1);
 39         }
 40 
 41         // 遍历发生变化的节点
 42         for(int i=0; i<ret; ++i)
 43         {
 44             // 只处理读事件, 其他事件默认不处理
 45             struct epoll_event *pev = &all[i];
 46             if(!(pev->events & EPOLLIN))
 47             {
 48                 // 不是读事件
 49                 continue;
 50             }
 51 
 52             if(pev->data.fd == lfd)
 53             {
 54                 // 接受连接请求
 55                 do_accept(lfd, epfd);
 56             }
 57             else
 58             {
 59                 // 读数据
 60                 do_read(pev->data.fd, epfd);
 61             }
 62         }
 63     }
 64 }
 65 
 66 // 读数据
 67 void do_read(int cfd, int epfd)
 68 {
 69     // 将浏览器发过来的数据, 读到buf中 
 70     char line[1024] = {0};
 71     // 读请求行
 72     int len = get_line(cfd, line, sizeof(line));
 73     if(len == 0)
 74     {
 75         printf("客户端断开了连接...
");
 76         // 关闭套接字, cfd从epoll上del
 77         disconnect(cfd, epfd);         
 78     }
 79     else
 80     {
 81         printf("请求行数据: %s", line);
 82         printf("============= 请求头 ============
");
 83         // 还有数据没读完
 84         // 继续读
 85         while(len)
 86         {
 87             char buf[1024] = {0};
 88             len = get_line(cfd, buf, sizeof(buf));
 89             printf("-----: %s", buf);
 90         }
 91         printf("============= The End ============
");
 92     }
 93 
 94     // 请求行: get /xxx http/1.1
 95     // 判断是不是get请求
 96     if(strncasecmp("get", line, 3) == 0)
 97     {
 98         // 处理http请求
 99         http_request(line, cfd);
100         // 关闭套接字, cfd从epoll上del
101         disconnect(cfd, epfd);         
102     }
103 }
104 
105 // 断开连接的函数
106 void disconnect(int cfd, int epfd)
107 {
108     int ret = epoll_ctl(epfd, EPOLL_CTL_DEL, cfd, NULL);
109     if(ret == -1)
110     {
111         perror("epoll_ctl del cfd error");
112         exit(1);
113     }
114     close(cfd);
115 }
116 
117 // http请求处理
118 void http_request(const char* request, int cfd)
119 {
120     // 拆分http请求行
121     // get /xxx http/1.1
122     char method[12], path[1024], protocol[12];
123     sscanf(request, "%[^ ] %[^ ] %[^ ]", method, path, protocol);
124 
125     printf("method = %s, path = %s, protocol = %s
", method, path, protocol);
126 
127     // 转码 将不能识别的中文乱码 - > 中文
128     // 解码 %23 %34 %5f
129     decode_str(path, path);
130         // 处理path  /xx
131         // 去掉path中的/
132         char* file = path+1;
133     // 如果没有指定访问的资源, 默认显示资源目录中的内容
134     if(strcmp(path, "/") == 0)
135     {
136         // file的值, 资源目录的当前位置
137         file = "./";
138     }
139 
140     // 获取文件属性
141     struct stat st;
142     int ret = stat(file, &st);
143     if(ret == -1)
144     {
145         // show 404
146         send_respond_head(cfd, 404, "File Not Found", ".html", -1);
147         send_file(cfd, "404.html");
148     }
149 
150     // 判断是目录还是文件
151     // 如果是目录
152     if(S_ISDIR(st.st_mode))
153     {
154         // 发送头信息
155         send_respond_head(cfd, 200, "OK", get_file_type(".html"), -1);
156         // 发送目录信息
157         send_dir(cfd, file);
158     }
159     else if(S_ISREG(st.st_mode))
160     {
161         // 文件
162         // 发送消息报头
163         send_respond_head(cfd, 200, "OK", get_file_type(file), st.st_size);
164         // 发送文件内容
165         send_file(cfd, file);
166     }
167 }
168 
169 // 发送目录内容
170 void send_dir(int cfd, const char* dirname)
171 {
172     // 拼一个html页面<table></table>
173     char buf[4094] = {0};
174 
175     sprintf(buf, "<html><head><title>目录名: %s</title></head>", dirname);
176     sprintf(buf+strlen(buf), "<body><h1>当前目录: %s</h1><table>", dirname);
177 
178     char enstr[1024] = {0};
179     char path[1024] = {0};
180     // 目录项二级指针
181     struct dirent** ptr;
182     int num = scandir(dirname, &ptr, NULL, alphasort);
183     // 遍历
184     for(int i=0; i<num; ++i)
185     {
186         char* name = ptr[i]->d_name;
187 
188         // 拼接文件的完整路径
189         sprintf(path, "%s/%s", dirname, name);
190         printf("path = %s ===================
", path);
191         struct stat st;
192         stat(path, &st);
193 
194         encode_str(enstr, sizeof(enstr), name);
195         // 如果是文件
196         if(S_ISREG(st.st_mode))
197         {
198             sprintf(buf+strlen(buf), 
199                     "<tr><td><a href="%s">%s</a></td><td>%ld</td></tr>",
200                     enstr, name, (long)st.st_size);
201         }
202         // 如果是目录
203         else if(S_ISDIR(st.st_mode))
204         {
205             sprintf(buf+strlen(buf), 
206                     "<tr><td><a href="%s/">%s/</a></td><td>%ld</td></tr>",
207                     enstr, name, (long)st.st_size);
208         }
209         send(cfd, buf, strlen(buf), 0);
210         memset(buf, 0, sizeof(buf));
211         // 字符串拼接
212     }
213 
214     sprintf(buf+strlen(buf), "</table></body></html>");
215     send(cfd, buf, strlen(buf), 0);
216 
217     printf("dir message send OK!!!!
");
218 #if 0
219     // 打开目录
220     DIR* dir = opendir(dirname);
221     if(dir == NULL)
222     {
223         perror("opendir error");
224         exit(1);
225     }
226 
227     // 读目录
228     struct dirent* ptr = NULL;
229     while( (ptr = readdir(dir)) != NULL )
230     {
231         char* name = ptr->d_name;
232     }
233     closedir(dir);
234 #endif
235 }
236 
237 // 发送响应头
238 void send_respond_head(int cfd, int no, const char* desp, const char* type, long len)
239 {
240     char buf[1024] = {0};
241     // 状态行
242     sprintf(buf, "http/1.1 %d %s
", no, desp);
243     send(cfd, buf, strlen(buf), 0);
244     // 消息报头
245     sprintf(buf, "Content-Type:%s
", type);
246     sprintf(buf+strlen(buf), "Content-Length:%ld
", len);
247     send(cfd, buf, strlen(buf), 0);
248     // 空行
249     send(cfd, "
", 2, 0);
250 }
251 
252 // 发送文件
253 void send_file(int cfd, const char* filename)
254 {
255     // 打开文件
256     int fd = open(filename, O_RDONLY);
257     if(fd == -1)
258     {
259         // show 404
260         return;
261     }
262 
263     // 循环读文件
264     char buf[4096] = {0};
265     int len = 0;
266     while( (len = read(fd, buf, sizeof(buf))) > 0 )
267     {
268         // 发送读出的数据
269         send(cfd, buf, len, 0);
270     }
271     if(len == -1)
272     {
273         perror("read file error");
274         exit(1);
275     }
276 
277     close(fd);
278 }
279 
280 // 解析http请求消息的每一行内容
281 int get_line(int sock, char *buf, int size)
282 {
283     int i = 0;
284     char c = '