main.c
#include <stdio.h> #include <unistd.h> #include <stdlib.h> #include "epoll_http.h" int main(int argc, const char* argv[]) { if(argc < 3) { printf("eg: ./a.out port path\n"); exit(1); } // 端口 int port = atoi(argv[1]); // 修改进程的工作目录, 方便后续操作 int ret = chdir(argv[2]); if(ret == -1) { perror("chdir error"); exit(1); } // 启动epoll模型 epoll_run(port); return 0; }epoll_http.h
#ifndef _EPOLL_SERVER_H #define _EPOLL_SERVER_H int init_listen_fd(int port, int epfd); void epoll_run(int port); void do_accept(int lfd, int epfd); void do_read(int cfd, int epfd); int get_line(int sock, char *buf, int size); void disconnect(int cfd, int epfd); void http_request(const char* request, int cfd); void send_respond_head(int cfd, int no, const char* desp, const char* type, long len); void send_file(int cfd, const char* filename); void send_dir(int cfd, const char* dirname); void encode_str(char* to, int tosize, const char* from); void decode_str(char *to, char *from); const char *get_file_type(const char *name); #endifepoll_http.c
#include <stdio.h> #include <unistd.h> #include <stdlib.h> #include <sys/types.h> #include <string.h> #include <sys/epoll.h> #include <arpa/inet.h> #include <fcntl.h> #include <dirent.h> #include <sys/stat.h> #include <ctype.h> #include "epoll_http.h" #define MAXSIZE 2000 void epoll_run(int port) { // 创建一个epoll树的根节点 int epfd = epoll_create(MAXSIZE); if(epfd == -1) { perror("epoll_create error"); exit(1); } // 添加要监听的节点 // 先添加监听lfd int lfd = init_listen_fd(port, epfd); // 委托内核检测添加到树上的节点 struct epoll_event all[MAXSIZE]; while(1) { int ret = epoll_wait(epfd, all, MAXSIZE, -1); if(ret == -1) { perror("epoll_wait error"); exit(1); } // 遍历发生变化的节点 for(int i=0; i<ret; ++i) { // 只处理读事件, 其他事件默认不处理 struct epoll_event *pev = &all[i]; if(!(pev->events & EPOLLIN)) { // 不是读事件 continue; } if(pev->data.fd == lfd) { // 接受连接请求 do_accept(lfd, epfd); } else { // 读数据 do_read(pev->data.fd, epfd); } } } } // 读数据 void do_read(int cfd, int epfd) { // 将浏览器发过来的数据, 读到buf中 char line[1024] = {0}; // 读请求行 int len = get_line(cfd, line, sizeof(line)); if(len == 0) { printf("客户端断开了连接...\n"); // 关闭套接字, cfd从epoll上del disconnect(cfd, epfd); } else { printf("请求行数据: %s", line); printf("============= 请求头 ============\n"); // 还有数据没读完 // 继续读 while(len) { char buf[1024] = {0}; len = get_line(cfd, buf, sizeof(buf)); printf("-----: %s", buf); } printf("============= The End ============\n"); } // 请求行: get /xxx http/1.1 // 判断是不是get请求 if(strncasecmp("get", line, 3) == 0) { // 处理http请求 http_request(line, cfd); // 关闭套接字, cfd从epoll上del disconnect(cfd, epfd); } } // 断开连接的函数 void disconnect(int cfd, int epfd) { int ret = epoll_ctl(epfd, EPOLL_CTL_DEL, cfd, NULL); if(ret == -1) { perror("epoll_ctl del cfd error"); exit(1); } close(cfd); } // http请求处理 void http_request(const char* request, int cfd) { // 拆分http请求行 // get /xxx http/1.1 char method[12], path[1024], protocol[12]; sscanf(request, "%[^ ] %[^ ] %[^ ]", method, path, protocol); printf("method = %s, path = %s, protocol = %s\n", method, path, protocol); // 转码 将不能识别的中文乱码 - > 中文 // 解码 # 4 _ decode_str(path, path); // 处理path /xx // 去掉path中的/ char* file = path+1; // 如果没有指定访问的资源, 默认显示资源目录中的内容 if(strcmp(path, "/") == 0) { // file的值, 资源目录的当前位置 file = "./"; } // 获取文件属性 struct stat st; int ret = stat(file, &st); if(ret == -1) { // show 404 send_respond_head(cfd, 404, "File Not Found", ".html", -1); send_file(cfd, "404.html"); } // 判断是目录还是文件 // 如果是目录 if(S_ISDIR(st.st_mode)) { // 发送头信息 send_respond_head(cfd, 200, "OK", get_file_type(".html"), -1); // 发送目录信息 send_dir(cfd, file); } else if(S_ISREG(st.st_mode)) { // 文件 // 发送消息报头 send_respond_head(cfd, 200, "OK", get_file_type(file), st.st_size); // 发送文件内容 send_file(cfd, file); } } // 发送目录内容 void send_dir(int cfd, const char* dirname) { // 拼一个html页面<table></table> char buf[4094] = {0}; sprintf(buf, "<html><head><title>目录名: %s</title></head>", dirname); sprintf(buf+strlen(buf), "<body><h1>当前目录: %s</h1><table>", dirname); char enstr[1024] = {0}; char path[1024] = {0}; // 目录项二级指针 struct dirent** ptr; int num = scandir(dirname, &ptr, NULL, alphasort); // 遍历 for(int i=0; i<num; ++i) { char* name = ptr[i]->d_name; // 拼接文件的完整路径 sprintf(path, "%s/%s", dirname, name); printf("path = %s ===================\n", path); struct stat st; stat(path, &st); encode_str(enstr, sizeof(enstr), name); // 如果是文件 if(S_ISREG(st.st_mode)) { sprintf(buf+strlen(buf), "<tr><td><a href=\"%s\">%s</a></td><td>%ld</td></tr>", enstr, name, (long)st.st_size); } // 如果是目录 else if(S_ISDIR(st.st_mode)) { sprintf(buf+strlen(buf), "<tr><td><a href=\"%s/\">%s/</a></td><td>%ld</td></tr>", enstr, name, (long)st.st_size); } send(cfd, buf, strlen(buf), 0); memset(buf, 0, sizeof(buf)); // 字符串拼接 } sprintf(buf+strlen(buf), "</table></body></html>"); send(cfd, buf, strlen(buf), 0); printf("dir message send OK!!!!\n"); #if 0 // 打开目录 DIR* dir = opendir(dirname); if(dir == NULL) { perror("opendir error"); exit(1); } // 读目录 struct dirent* ptr = NULL; while( (ptr = readdir(dir)) != NULL ) { char* name = ptr->d_name; } closedir(dir); #endif } // 发送响应头 void send_respond_head(int cfd, int no, const char* desp, const char* type, long len) { char buf[1024] = {0}; // 状态行 sprintf(buf, "http/1.1 %d %s\r\n", no, desp); send(cfd, buf, strlen(buf), 0); // 消息报头 sprintf(buf, "Content-Type:%s\r\n", type); sprintf(buf+strlen(buf), "Content-Length:%ld\r\n", len); send(cfd, buf, strlen(buf), 0); // 空行 send(cfd, "\r\n", 2, 0); } // 发送文件 void send_file(int cfd, const char* filename) { // 打开文件 int fd = open(filename, O_RDONLY); if(fd == -1) { // show 404 return; } // 循环读文件 char buf[4096] = {0}; int len = 0; while( (len = read(fd, buf, sizeof(buf))) > 0 ) { // 发送读出的数据 send(cfd, buf, len, 0); } if(len == -1) { perror("read file error"); exit(1); } close(fd); } // 解析http请求消息的每一行内容 int get_line(int sock, char *buf, int size) { int i = 0; char c = '\0'; int n; while ((i < size - 1) && (c != '\n')) { n = recv(sock, &c, 1, 0); if (n > 0) { if (c == '\r') { n = recv(sock, &c, 1, MSG_PEEK); if ((n > 0) && (c == '\n')) { recv(sock, &c, 1, 0); } else { c = '\n'; } } buf[i] = c; i++; } else { c = '\n'; } } buf[i] = '\0'; return i; } // 接受新连接处理 void do_accept(int lfd, int epfd) { struct sockaddr_in client; socklen_t len = sizeof(client); int cfd = accept(lfd, (struct sockaddr*)&client, &len); if(cfd == -1) { perror("accept error"); exit(1); } // 打印客户端信息 char ip[64] = {0}; printf("New Client IP: %s, Port: %d, cfd = %d\n", inet_ntop(AF_INET, &client.sin_addr.s_addr, ip, sizeof(ip)), ntohs(client.sin_port), cfd); // 设置cfd为非阻塞 int flag = fcntl(cfd, F_GETFL); flag |= O_NONBLOCK; fcntl(cfd, F_SETFL, flag); // 得到的新节点挂到epoll树上 struct epoll_event ev; ev.data.fd = cfd; // 边沿非阻塞模式 ev.events = EPOLLIN | EPOLLET; int ret = epoll_ctl(epfd, EPOLL_CTL_ADD, cfd, &ev); if(ret == -1) { perror("epoll_ctl add cfd error"); exit(1); } } int init_listen_fd(int port, int epfd) { // 创建监听的套接字 int lfd = socket(AF_INET, SOCK_STREAM, 0); if(lfd == -1) { perror("socket error"); exit(1); } // lfd绑定本地IP和port struct sockaddr_in serv; memset(&serv, 0, sizeof(serv)); serv.sin_family = AF_INET; serv.sin_port = htons(port); serv.sin_addr.s_addr = htonl(INADDR_ANY); // 端口复用 int flag = 1; setsockopt(lfd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag)); int ret = bind(lfd, (struct sockaddr*)&serv, sizeof(serv)); if(ret == -1) { perror("bind error"); exit(1); } // 设置监听 ret = listen(lfd, 64); if(ret == -1) { perror("listen error"); exit(1); } // lfd添加到epoll树上 struct epoll_event ev; ev.events = EPOLLIN; ev.data.fd = lfd; ret = epoll_ctl(epfd, EPOLL_CTL_ADD, lfd, &ev); if(ret == -1) { perror("epoll_ctl add lfd error"); exit(1); } return lfd; } // 16进制数转化为10进制 int hexit(char c) { if (c >= '0' && c <= '9') return c - '0'; if (c >= 'a' && c <= 'f') return c - 'a' + 10; if (c >= 'A' && c <= 'F') return c - 'A' + 10; return 0; } /* * 这里的内容是处理 之类的东西！是"解码"过程。 * URL编码中的‘ ’(space) * ! '!' " '"' # '#' $ '$' * % '%' & '&' ' ''' ( '('...... * 相关知识html中的‘ ’(space)是 */ void encode_str(char* to, int tosize, const char* from) { int tolen; for (tolen = 0; *from != '\0' && tolen + 4 < tosize; ++from) { if (isalnum(*from) || strchr("/_.-~", *from) != (char*)0) { *to = *from; ++to; ++tolen; } else { sprintf(to, "%%x", (int) *from & 0xff); to += 3; tolen += 3; } } *to = '\0'; } void decode_str(char *to, char *from) { for ( ; *from != '\0'; ++to, ++from ) { if (from[0] == '%' && isxdigit(from[1]) && isxdigit(from[2])) { *to = hexit(from[1])*16 + hexit(from[2]); from += 2; } else { *to = *from; } } *to = '\0'; } // 通过文件名获取文件的类型 const char *get_file_type(const char *name) { char* dot; // 自右向左查找‘.’字符, 如不存在返回NULL dot = strrchr(name, '.'); if (dot == NULL) return "text/plain; charset=utf-8"; if (strcmp(dot, ".html") == 0 || strcmp(dot, ".htm") == 0) return "text/html; charset=utf-8"; if (strcmp(dot, ".jpg") == 0 || strcmp(dot, ".jpeg") == 0) return "image/jpeg"; if (strcmp(dot, ".gif") == 0) return "image/gif"; if (strcmp(dot, ".png") == 0) return "image/png"; if (strcmp(dot, ".css") == 0) return "text/css"; if (strcmp(dot, ".au") == 0) return "audio/basic"; if (strcmp( dot, ".wav" ) == 0) return "audio/wav"; if (strcmp(dot, ".avi") == 0) return "video/x-msvideo"; if (strcmp(dot, ".mov") == 0 || strcmp(dot, ".qt") == 0) return "video/quicktime"; if (strcmp(dot, ".mpeg") == 0 || strcmp(dot, ".mpe") == 0) return "video/mpeg"; if (strcmp(dot, ".vrml") == 0 || strcmp(dot, ".wrl") == 0) return "model/vrml"; if (strcmp(dot, ".midi") == 0 || strcmp(dot, ".mid") == 0) return "audio/midi"; if (strcmp(dot, ".mp3") == 0) return "audio/mpeg"; if (strcmp(dot, ".ogg") == 0) return "application/ogg"; if (strcmp(dot, ".pac") == 0) return "application/x-ns-proxy-autoconfig"; return "text/plain; charset=utf-8"; }main.c
#include <stdio.h> #include <sys/socket.h> #include <arpa/inet.h> #include <unistd.h> #include <stdlib.h> #include "libevent_http.h" int main(int argc, char **argv) { // 端口 int port = atoi(argv[1]); // 修改进程的工作目录, 方便后续操作 int ret = chdir(argv[2]); if (ret == -1) { perror("chdir error"); exit(1); } struct event_base *base = NULL; base = event_base_new(); if (base == NULL) { perror("event_base_new"); exit(1); } struct sockaddr_in addr; addr.sin_family = AF_INET; addr.sin_port = htons(port); addr.sin_addr.s_addr = htonl(INADDR_ANY); //创建监听的套接字,绑定,监听,以及接受客户端连接请求 struct evconnlistener *lis = evconnlistener_new_bind(base, listencb, base, LEV_OPT_CLOSE_ON_FREE | LEV_OPT_REUSEABLE, -1, (struct sockaddr *) &addr, sizeof(addr)); if (!lis) { perror("evconnlistener_new_bind"); exit(1); } struct event *signal_event = evsignal_new(base, SIGINT, signal_cb, (void *) base); if (!signal_event || event_add(signal_event, NULL) < 0) { fprintf(stderr, "不能创建一个信号事件"); exit(EXIT_FAILURE); } //调度 event_base_dispatch(base); //释放 event_free(signal_event); event_base_free(base); printf("done \n"); return 0; }libevent_http.h
#ifndef __LIBEV__ #define __LIBEV__ #include <event2/bufferevent.h> #include <event2/listener.h> void listencb(struct evconnlistener *listener, evutil_socket_t fd, struct sockaddr *cli, int len, void *ptr); void readcb(struct bufferevent *bev, void *ctx); void eventcb(struct bufferevent *bev, short what, void *ctx); void signal_cb(evutil_socket_t sig, short events, void *user_data); #endiflibevent_http.c
#include "libevent_http.h" #include <stdio.h> #include <event2/buffer.h> #include <string.h> #include <sys/stat.h> #include <stdlib.h> #include <ctype.h> #include <fcntl.h> #include <unistd.h> #include <dirent.h> // 通过文件名获取文件的类型 const char *get_file_type(const char *name) { char *dot; // 自右向左查找‘.’字符, 如不存在返回NULL dot = strrchr(name, '.'); if (dot == NULL) return "text/plain; charset=utf-8"; if (strcmp(dot, ".html") == 0 || strcmp(dot, ".htm") == 0) return "text/html; charset=utf-8"; if (strcmp(dot, ".jpg") == 0 || strcmp(dot, ".jpeg") == 0) return "image/jpeg"; if (strcmp(dot, ".gif") == 0) return "image/gif"; if (strcmp(dot, ".png") == 0) return "image/png"; if (strcmp(dot, ".css") == 0) return "text/css"; if (strcmp(dot, ".au") == 0) return "audio/basic"; if (strcmp(dot, ".wav") == 0) return "audio/wav"; if (strcmp(dot, ".avi") == 0) return "video/x-msvideo"; if (strcmp(dot, ".mov") == 0 || strcmp(dot, ".qt") == 0) return "video/quicktime"; if (strcmp(dot, ".mpeg") == 0 || strcmp(dot, ".mpe") == 0) return "video/mpeg"; if (strcmp(dot, ".vrml") == 0 || strcmp(dot, ".wrl") == 0) return "model/vrml"; if (strcmp(dot, ".midi") == 0 || strcmp(dot, ".mid") == 0) return "audio/midi"; if (strcmp(dot, ".mp3") == 0) return "audio/mpeg"; if (strcmp(dot, ".ogg") == 0) return "application/ogg"; if (strcmp(dot, ".pac") == 0) return "application/x-ns-proxy-autoconfig"; return "text/plain; charset=utf-8"; } // 16进制数转化为10进制 int hexit(char c) { if (c >= '0' && c <= '9') return c - '0'; if (c >= 'a' && c <= 'f') return c - 'a' + 10; if (c >= 'A' && c <= 'F') return c - 'A' + 10; return 0; } /* * 这里的内容是处理 之类的东西！是"解码"过程。 * URL编码中的‘ ’(space) * ! '!' " '"' # '#' $ '$' * % '%' & '&' ' ''' ( '('...... * 相关知识html中的‘ ’(space)是 */ void encode_str(char *to, int tosize, const char *from) { int tolen; for (tolen = 0; *from != '\0' && tolen + 4 < tosize; ++from) { if (isalnum(*from) || strchr("/_.-~", *from) != (char *) 0) { *to = *from; ++to; ++tolen; } else { sprintf(to, "%%x", (int) *from & 0xff); to += 3; tolen += 3; } } *to = '\0'; } void decode_str(char *to, char *from) { for (; *from != '\0'; ++to, ++from) { if (from[0] == '%' && isxdigit(from[1]) && isxdigit(from[2])) { *to = hexit(from[1]) * 16 + hexit(from[2]); from += 2; } else { *to = *from; } } *to = '\0'; } void listencb(struct evconnlistener *listener, evutil_socket_t fd, struct sockaddr *cli, int len, void *ptr) { struct event_base *base = ptr; struct bufferevent *bev = bufferevent_socket_new(base, fd, BEV_OPT_CLOSE_ON_FREE); //强制刷新缓冲区 bufferevent_flush(bev, EV_READ | EV_WRITE, BEV_NORMAL); //set callback bufferevent_setcb(bev, readcb, NULL, eventcb, NULL); bufferevent_enable(bev, EV_READ | EV_WRITE); } // 断开连接的函数 void disconnect(struct bufferevent *bev) { bufferevent_disable(bev, EV_READ | EV_WRITE); } void send_respond_head(struct bufferevent *bev, int no, const char *desp, const char *type, long len) { char buf[1024] = {0}; // 状态行 sprintf(buf, "http/1.1 %d %s\r\n", no, desp); bufferevent_write(bev, buf, strlen(buf)); // 消息报头 memset(buf, 0, sizeof buf); sprintf(buf, "Content-Type:%s\r\n", type); sprintf(buf + strlen(buf), "Content-Length:%ld\r\n", len); bufferevent_write(bev, buf, strlen(buf)); // 空行 bufferevent_write(bev, "\r\n", 2); } void send_file(struct bufferevent *bev, const char *path) { int fd = open(path, O_RDONLY); if (fd == -1) { //todo 404 perror("open err"); exit(1); } char buf[4096]; int ret = 0; while ((ret = read(fd, buf, sizeof buf)) > 0) { bufferevent_write(bev, buf, ret); } if (ret == -1) { perror("read err"); exit(1); } close(fd); } void send_dir(struct bufferevent *bev, const char *path) { char file[1024] = {0}; char buf[4096] = {0}; sprintf(buf, "<html><head><title>目录名: %s</title></head>", path); sprintf(buf + strlen(buf), "<body><h1>当前目录: %s</h1><table>", path); #if 1 DIR *dir = opendir(path); if (dir == NULL) { perror("opendir error"); exit(1); } // 读目录 char enstr[1024] = {0}; struct dirent *ptr = NULL; while ((ptr = readdir(dir)) != NULL) { char *name = ptr->d_name; // 拼接文件的完整路径 sprintf(file, "%s/%s", path, name); encode_str(enstr, sizeof(enstr), name); struct stat st; int ret = stat(file, &st); if (ret == -1) { perror("stat err"); exit(1); } // 如果是文件 if (S_ISREG(st.st_mode)) { sprintf(buf + strlen(buf), "<tr><td><a href=\"%s\">%s</a></td><td>%ld</td></tr>", enstr, name, (long) st.st_size); } // 如果是目录 else if (S_ISDIR(st.st_mode)) { sprintf(buf + strlen(buf), "<tr><td><a href=\"%s/\">%s/</a></td><td>%ld</td></tr>", enstr, name, (long) st.st_size); } bufferevent_write(bev, buf, strlen(buf)); memset(buf, 0, sizeof(buf)); } sprintf(buf + strlen(buf), "</table></body></html>"); bufferevent_write(bev, buf, strlen(buf)); closedir(dir); #endif } void http_get_request(const char *line, struct bufferevent *bev) { char method[12] = {0}; char res[1024] = {0}; sscanf(line, "%[^ ] %[^ ]", method, res); // 转码 将不能识别的中文乱码 - > 中文 // 解码 # 4 _ char destr[1024]; decode_str(destr, res); // 处理path /xx // 去掉path中的/ char *file = destr + 1; char fbuf[1024]; if (strcmp(res, "/") == 0) { file = "./"; //getcwd(fbuf, sizeof(fbuf)); //printf("root:[%s]\n", fbuf); // file = fbuf; } struct stat st; printf("file:[%s]\n", file); int ret = stat(file, &st); if (ret == -1) { //todo 404 perror("stat error"); exit(1); } if (S_ISDIR(st.st_mode)) { //dir // 发送头信息 send_respond_head(bev, 200, "OK", get_file_type(".html"), 100000); send_dir(bev, file); } else if (S_ISREG(st.st_mode)) { //reg file send_respond_head(bev, 200, "OK", get_file_type(file), st.st_size); send_file(bev, file); } } void readcb(struct bufferevent *bev, void *ctx) { struct evbuffer *evbuf = bufferevent_get_input(bev); char *line; size_t len; line = evbuffer_readln(evbuf, &len, EVBUFFER_EOL_CRLF); /* char* tmp = line; while(1){ tmp = evbuffer_readln(evbuf, &len, EVBUFFER_EOL_CRLF); if(tmp) break; printf("%s\n", tmp); } printf("line:[%s]", line); */ if (strncasecmp("get", line, 3) == 0) { http_get_request(line, bev); //disconnect(bev); } } void eventcb(struct bufferevent *bev, short what, void *ctx) { if (what & BEV_EVENT_EOF) { printf("连接关闭 \n"); } else if (what & BEV_EVENT_ERROR) { printf("发生一个连接错误 \n"); } bufferevent_free(bev); } void signal_cb(evutil_socket_t sig, short events, void *user_data) { struct event_base *base = user_data; struct timeval delay = {1, 0}; printf("延迟一秒关闭 \n"); event_base_loopexit(base, &delay); }