基于c++及linux网络编程的web服务器源码.zip

#include "requestData.h" #include "util.h" #include "epoll.h" #include <sys/epoll.h> #include <unistd.h> #include <sys/time.h> #include <unordered_map> #include <fcntl.h> #include <sys/stat.h> #include <sys/mman.h> #include <queue> #include <cstdlib> #include <opencv/cv.h> #include <opencv2/core/core.hpp> #include <opencv2/highgui/highgui.hpp> #include <opencv2/opencv.hpp> using namespace cv; #include <iostream> using namespace std; pthread_mutex_t MutexLockGuard::lock=PTHREAD_MUTEX_INITIALIZER; pthread_mutex_t MimeType::lock=PTHREAD_MUTEX_INITIALIZER; std::unordered_map<std::string,std::string> MimeType::mime; std::string MimeType::getMime(const std::string &suffix) { if (mime.size() == 0) { pthread_mutex_lock(&lock); if (mime.size() == 0) { mime[".html"] = "text/html"; mime[".avi"] = "video/x-msvideo"; mime[".bmp"] = "image/bmp"; mime[".c"] = "text/plain"; mime[".doc"] = "application/msword"; mime[".gif"] = "image/gif"; mime[".gz"] = "application/x-gzip"; mime[".htm"] = "text/html"; mime[".ico"] = "application/x-ico"; mime[".jpg"] = "image/jpeg"; mime[".png"] = "image/png"; mime[".txt"] = "text/plain"; mime[".mp3"] = "audio/mp3"; mime["default"] = "text/html"; } pthread_mutex_unlock(&lock); } if (mime.find(suffix) == mime.end()) return mime["default"]; else return mime[suffix]; } std::priority_queue<shared_ptr<mytimer>,std::deque<shared_ptr<mytimer>>,timerCmp> myTimerQueue; requestData::requestData(): now_read_pos(0), state(STATE_PARSE_URI), h_state(h_start), keep_alive(false), againTimes(0) { cout << "requestData()" << endl; } requestData::requestData(int _epollfd, int _fd, std::string _path): now_read_pos(0), state(STATE_PARSE_URI), h_state(h_start), keep_alive(false), againTimes(0), path(_path), fd(_fd), epollfd(_epollfd) { cout << "requestData()" << endl; } requestData::~requestData() { cout << "~requestData()" << endl; Epoll::epoll_del(fd,EPOLLIN | EPOLLET | EPOLLONESHOT); if(timer.lock()) { shared_ptr<mytimer> my_timer(timer.lock()); my_timer->clearReq(); timer.reset(); } close(fd); } void requestData::addTimer(shared_ptr<mytimer> mtimer) { timer=mtimer; } int requestData::getFd() { return fd; } void requestData::setFd(int _fd) { fd = _fd; } void requestData::reset() { againTimes = 0; content.clear(); file_name.clear(); path.clear(); now_read_pos = 0; state = STATE_PARSE_URI; h_state = h_start; headers.clear(); keep_alive = false; if(timer.lock()) { shared_ptr<mytimer> my_timer(timer.lock()); my_timer->clearReq(); timer.reset(); } } void requestData::seperateTimer() { if (timer.lock()) { shared_ptr<mytimer> my_timer(timer.lock()); my_timer->clearReq(); timer.reset(); } } void requestData::handleRequest() { char buff[MAX_BUFF]; bool isError = false; while (true) { int read_num = readn(fd, buff, MAX_BUFF); if (read_num < 0) { perror("1"); isError = true; break; } else if (read_num == 0) { // 有请求出现但是读不到数据，可能是Request Aborted，或者来自网络的数据没有达到等原因 perror("read_num == 0"); if (errno == EAGAIN) { if (againTimes > AGAIN_MAX_TIMES) isError = true; else ++againTimes; } else if (errno != 0) isError = true; break; } string now_read(buff, buff + read_num); content += now_read; if (state == STATE_PARSE_URI) { int flag = this->parse_URI(); if (flag == PARSE_URI_AGAIN) { break; } else if (flag == PARSE_URI_ERROR) { perror("2"); isError = true; break; } } if (state == STATE_PARSE_HEADERS) { int flag = this->parse_Headers(); if (flag == PARSE_HEADER_AGAIN) { break; } else if (flag == PARSE_HEADER_ERROR) { perror("3"); isError = true; break; } if(method == METHOD_POST) { state = STATE_RECV_BODY; } else { state = STATE_ANALYSIS; } } if (state == STATE_RECV_BODY) { int content_length = -1; if (headers.find("Content-length") != headers.end()) { content_length = stoi(headers["Content-length"]); } else { isError = true; break; } if (content.size() < content_length) continue; state = STATE_ANALYSIS; } if (state == STATE_ANALYSIS) { int flag = this->analysisRequest(); if (flag < 0) { isError = true; break; } else if (flag == ANALYSIS_SUCCESS) { state = STATE_FINISH; break; } else { isError = true; break; } } } if (isError) { //delete this; return; } // 加入epoll继续 if (state == STATE_FINISH) { if (keep_alive) { //printf("ok\n"); this->reset(); } else { //delete this; return; } } shared_ptr<mytimer> mtimer(new mytimer(shared_from_this(),500)); //shared_from_this：返回this指针 this->addTimer(mtimer); { MutexLockGuard lock; myTimerQueue.push(mtimer); } __uint32_t _epo_event= EPOLLIN | EPOLLET | EPOLLONESHOT; int ret=Epoll::epoll_mod(fd,shared_from_this(),_epo_event); if(ret<0) { return ; } } int requestData::parse_URI() { string &str = content; // 读到完整的请求行再开始解析请求 int pos = str.find('\r', now_read_pos); if (pos < 0) { return PARSE_URI_AGAIN; } // 去掉请求行所占的空间，节省空间 string request_line = str.substr(0, pos); if (str.size() > pos + 1) str = str.substr(pos + 1); else str.clear(); // Method pos = request_line.find("GET"); if (pos < 0) { pos = request_line.find("POST"); if (pos < 0) { return PARSE_URI_ERROR; } else { method = METHOD_POST; } } else { method = METHOD_GET; } //printf("method = %d\n", method); // filename pos = request_line.find("/", pos); if (pos < 0) { return PARSE_URI_ERROR; } else { int _pos = request_line.find(' ', pos); if (_pos < 0) return PARSE_URI_ERROR; else { if (_pos - pos > 1) { file_name = request_line.substr(pos + 1, _pos - pos - 1); int __pos = file_name.find('?'); if (__pos >= 0) { file_name = file_name.substr(0, __pos); } } else file_name = "index.html"; } pos = _pos; } //cout << "file_nam