（牛客网C++课程）Linux 高并发Web服务器项目实战（带定时检测代码）

#include "http_conn.h" http_conn::http_conn(){} http_conn::~http_conn(){} int http_conn::m_epoll_fd = -1; // 类中静态成员需要外部定义 int http_conn::m_user_cnt = 0; int http_conn::m_request_cnt = 0; sort_timer_lst http_conn::m_timer_lst; // locker http_conn::m_timer_lst_locker; // 网站的根目录 const char* doc_root = "/home/cyf/Linux/webserver/resources"; // 定义HTTP响应的一些状态信息 const char* ok_200_title = "OK"; const char* error_400_title = "Bad Request"; const char* error_400_form = "Your request has bad syntax or is inherently impossible to satisfy.\n"; const char* error_403_title = "Forbidden"; const char* error_403_form = "You do not have permission to get file from this server.\n"; const char* error_404_title = "Not Found"; const char* error_404_form = "The requested file was not found on this server.\n"; const char* error_500_title = "Internal Error"; const char* error_500_form = "There was an unusual problem serving the requested file.\n"; // 设置文件描述符为非阻塞 void set_nonblocking(int fd){ int flag = fcntl(fd, F_GETFL); flag |= O_NONBLOCK; fcntl(fd, F_SETFL, flag); } // 添加需要监听的文件描述符到epoll中 void addfd(int epoll_fd, int fd, bool one_shot, bool et){ epoll_event event; event.data.fd = fd; if(et){ event.events = EPOLLIN | EPOLLRDHUP | EPOLLET; // 对所有fd设置边沿触发，但是listen_fd不需要，可以另行判断处理 }else{ event.events = EPOLLIN | EPOLLRDHUP; // 默认水平触发 对端连接断开触发的epoll 事件包含 EPOLLIN | EPOLLRDHUP挂起，不用根据返回值判断，直接通过事件判断异常断开 } if(one_shot){ event.events |= EPOLLONESHOT; // 注册为 EPOLLONESHOT事件，防止同一个通信被不同的线程处理 } epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fd, &event); // 设置文件描述符为非阻塞（epoll ET模式） set_nonblocking(fd); } // 从epoll中删除文件描述符 void rmfd(int epoll_fd, int fd){ epoll_ctl(epoll_fd, EPOLL_CTL_DEL, fd, 0); close(fd); } // 在epoll中修改文件描述符，重置socket上EPOLLONESHOT事件，确保下次可读时，EPOLLIN 事件被触发 void modfd(int epoll_fd, int fd, int ev){ epoll_event event; event.data.fd = fd; event.events = ev | EPOLLONESHOT | EPOLLRDHUP; epoll_ctl(epoll_fd, EPOLL_CTL_MOD, fd, &event); } // 初始化新的连接 void http_conn::init(int sock_fd, const sockaddr_in& addr){ m_sock_fd = sock_fd; // 套接字 m_addr = addr; // 客户端地址 // 设置端口复用 int reuse = 1; setsockopt(sock_fd, SOL_SOCKET, SO_REUSEPORT, &reuse, sizeof(reuse)); // 添加sock_fd到epoll对象中 addfd(m_epoll_fd, sock_fd, true, ET); ++m_user_cnt; char ip[16] = ""; const char* str = inet_ntop(AF_INET, &addr.sin_addr.s_addr, ip, sizeof(ip)); EMlog(LOGLEVEL_INFO, "The No.%d user. sock_fd = %d, ip = %s.\n", m_user_cnt, sock_fd, str); init(); // 初始化其他信息，私有 // 创建定时器，设置其回调函数与超时时间，然后绑定定时器与用户数据，最后将定时器添加到链表timer_lst中 util_timer* new_timer = new util_timer; new_timer->user_data = this; time_t curr_time = time(NULL); new_timer->expire = curr_time + 3 * TIMESLOT; this->timer = new_timer; m_timer_lst.add_timer(new_timer); } // 初始化连接之外的其他信息 void http_conn::init(){ m_method = GET; m_url = 0; m_version = 0; m_linger = false; // 默认不保持连接 m_content_len = 0; m_host = 0; m_check_stat = CHECK_STATE_REQUESTLINE; // 初始化状态为正在解析请求首行 m_checked_idx = 0; // 初始化解析字符索引 m_line_start = 0; // 行的起始位置 m_rd_idx = 0; // 读取字符的位置 m_write_idx = 0; bytes_have_send = 0; bytes_to_send = 0; bzero(m_rd_buf, RD_BUF_SIZE); // 清空读缓存 bzero(m_write_buf, WD_BUF_SIZE); // 清空写缓存 bzero(m_real_file, FILENAME_LEN); // 清空文件路径 } // 关闭连接 void http_conn::conn_close(){ if(m_sock_fd != -1){ --m_user_cnt; // 客户端数量减一 EMlog(LOGLEVEL_INFO, "closing fd: %d, rest user num :%d\n", m_sock_fd, m_user_cnt); rmfd(m_epoll_fd, m_sock_fd); // 移除epoll检测,关闭套接字 m_sock_fd = -1; } } // 循环读取客户数据，直到无数据可读 或 关闭连接 bool http_conn::read(){ if(timer) { // 更新超时时间 time_t curr_time = time( NULL ); timer->expire = curr_time + 3 * TIMESLOT; m_timer_lst.adjust_timer( timer ); } if(m_rd_idx >= RD_BUF_SIZE) return false; // 超过缓冲区大小 int bytes_rd = 0; while(true){ // m_sock_fd已设置非阻塞 bytes_rd = recv(m_sock_fd, m_rd_buf + m_rd_idx, RD_BUF_SIZE - m_rd_idx, 0); // 第二个参数传递的是缓冲区中开始读入的地址偏移 if(bytes_rd == -1){ if(errno == EAGAIN || errno == EWOULDBLOCK){ break; // 非阻塞读取，没有数据了 } return false; // 读取错误，调用conn_close() }else if(bytes_rd == 0){ return false; // 对方关闭连接，调用conn_close() } m_rd_idx += bytes_rd; // 更新下一次读取位置 } ++m_request_cnt; EMlog(LOGLEVEL_INFO, "sock_fd = %d read done. request cnt = %d\n", m_sock_fd, m_request_cnt); // 全部读取完毕 return true; } // 主状态机 解析HTTP请求 http_conn::HTTP_CODE http_conn::process_read(){ LINE_STATUS line_stat = LINE_OK; HTTP_CODE ret = NO_REQUEST; char* text = 0; while((m_check_stat == CHECK_STATE_CONTENT && line_stat == LINE_OK) // 主状态机正在解析请求体，且从状态机OK，不需要一行一行解析 || (line_stat = parse_one_line()) == LINE_OK){ // 从状态机解析到一行数据 // 获取一行数据 text = get_line(); m_line_start = m_checked_idx; // 更新下一行的起始位置 EMlog(LOGLEVEL_DEBUG, ">>>>>> %s\n", text); switch(m_check_stat){ case CHECK_STATE_REQUESTLINE: { ret = parse_request_line(text); if(ret == BAD_REQUEST){ return BAD_REQUEST; } break; } case CHECK_STATE_HEADER: { ret = parse_request_headers(text); if(ret == BAD_REQUEST){ return BAD_REQUEST; }else if(ret == GET_REQUEST){ return do_request(); // 解析具体的请求信息 } break; } case CHECK_STATE_CONTENT: { ret = parse_request_content(text); if(ret == GET_REQUEST){ return do_request(); // 解析具体的请求信息 } line_stat = LINE_OPEN; // != GET_REQUEST break; } default: { return INTERNAL_ERROR; // 内部错误 } } } return NO_REQUEST; // 数据不完整 } // 解析请求首行，获得请求方法，目标URL，HTTP版本 http_conn::HTTP_CODE http_conn::parse_request_line(char* text){ // GET /index.html HTTP/1.1 m_url = strpbrk(text, " \t"); // 找到第一次出现空格或者\t的下标 if(!m_url) return BAD_REQUEST; *m_url = '\0'; // GET\0/index.html HTTP/1.1，此时text到\0结束，表示 GET\0 m_url++; // /index.h