c++实现的Mini Web Server，暂时当ftp使用

废话不多说，先上效果图：

启动web server：

 

浏览器截图：

 

 

 

 

下载使用(暂时只有Windows版，Linux版有时间的话补上)：miniweb.rar

 

 

 

下面直接贴出主要实现代码：

Http类主要负责接收、解析客户端请求，然后通过WebAction类响应并发送相应的数据

Http.h

#ifndef _HTTP_H #define _HTTP_H #include <WinSock2.h> #include <iostream> #include "zfqstring.h" class Http { public: Http(SOCKET m_socket, zfqstring id, zfqstring pwd); virtual ~Http(void); public: int Send(char* data, UINT len); int Recv(char* data, UINT len); int RecvByLine(char* data, UINT len); BOOL HttpRequestResponse(); BOOL Parse(char* data); BOOL Query(); protected: // zfqstring m_hostaddr; unsigned short m_hostport; // zfqstring m_id; zfqstring m_pwd; // zfqstring m_loginid; zfqstring m_loginpwd; // zfqstring m_postdata; zfqstring m_page; DWORD m_method; DWORD m_contentlength; // SOCKET m_socket; }; #endif

 

Http.cpp

#include "StdAfx.h" #include "Http.h" #include "Constant.h" #include "Base64.h" #include "FtpWebAction.h" Http::Http(SOCKET m_socket, zfqstring id, zfqstring pwd) { this->m_socket = m_socket; m_id = id; m_pwd = pwd; } Http::~Http(void) { if(m_socket) closesocket(m_socket); } int Http::Send(char* data, UINT len ) { long already_time = 0; struct fd_set fdset; struct timeval timeout; timeout.tv_sec = SELECT_WAIT_TIME; timeout.tv_usec = 0; size_t nSend = 0; size_t tLen = 0; while(1) { FD_ZERO(&fdset); FD_SET(m_socket, &fdset); int ret = select(m_socket + 1, NULL, &fdset, NULL, &timeout); switch(ret) { case 1: already_time = 0; tLen = send(m_socket, data, len - nSend, 0); nSend += tLen; if(nSend == len) return SOCKET_SENDRECV_OK; break; case 0: already_time += SELECT_WAIT_TIME; if(already_time >= MAX_WAIT_TIME_FOR_SENDRECV) { closesocket(m_socket); return SOCKET_SENDRECV_TIMEOUT; } break; default: return SOCKET_SENDRECV_ERROR; } } } int Http::Recv(char* data, UINT len ) { long already_time = 0; struct fd_set fdset; struct timeval timeout; timeout.tv_sec = SELECT_WAIT_TIME; timeout.tv_usec = 0; size_t nRecv = 0; size_t tLen = 0; while(1) { FD_ZERO(&fdset); FD_SET(m_socket, &fdset); int ret = select(m_socket + 1, &fdset, NULL, NULL, &timeout); switch(ret) { case 1: already_time = 0; tLen = recv(m_socket, data + nRecv, len - nRecv, 0); if(tLen <= 0) return nRecv; nRecv += tLen; if(nRecv == len) return nRecv; break; case 0: already_time += SELECT_WAIT_TIME; if(already_time >= MAX_WAIT_TIME_FOR_SENDRECV) { closesocket(m_socket); return SOCKET_SENDRECV_TIMEOUT; } break; default: return SOCKET_SENDRECV_ERROR; } } } int Http::RecvByLine(char* data, UINT len) { long already_time = 0; struct fd_set fdset; struct timeval timeout; timeout.tv_sec = SELECT_WAIT_TIME; timeout.tv_usec = 0; size_t nRecv = 0; size_t tLen = 0; while(1) { if(nRecv >= len - 1) { data[nRecv] = '/0'; char tempbuff[2]; while(1) { recv(m_socket, tempbuff, 1, 0); if(tempbuff[0] == '/n') break; } return SOCKET_SENDRECV_TOOLONG; } // FD_ZERO(&fdset); FD_SET(m_socket, &fdset); int ret = select(m_socket + 1, &fdset, NULL, NULL, &timeout); switch(ret) { case 1: already_time = 0; tLen = recv(m_socket, data + nRecv, 1, 0); if(tLen <= 0) { data[nRecv] = '/0'; return SOCKET_SENDRECV_ERROR; } nRecv += tLen; if(data[nRecv - 1] == '/n') goto L1; case 0: already_time += SELECT_WAIT_TIME; if(already_time >= MAX_WAIT_TIME_FOR_SENDRECV) { closesocket(m_socket); return SOCKET_SENDRECV_TIMEOUT; } break; default: return SOCKET_SENDRECV_ERROR; } } L1: if(data[nRecv - 2] == '/r') { data[nRecv - 2] = '/0'; } else if(data[nRecv - 1] == '/n') { data[nRecv - 1] = '/0'; } return SOCKET_SENDRECV_OK; } BOOL Http::Parse( char* data ) { if(strncmp(data, "GET ", 4) == 0) { m_method = HTTP_METHOD_GET; m_page = data; m_page.Cutout("GET /", " "); m_page.Cutout(NULL, "?"); m_postdata = data; m_postdata.Cutout("?", " "); m_postdata += "&"; } else if(strncmp(data, "POST ", 5) == 0) { m_method = HTTP_METHOD_POST; m_page = data; m_page.Cutout("POST /", " "); } else if(strncmp(data, "PUT ", 4) == 0) { m_method = HTTP_METHOD_PUT; m_page = data; m_page.Cutout("PUT /", " "); } else if(strncmp(data, "Content-Length:", 15) == 0) { sscanf(data, "Content-Length: %d", &m_contentlength); } else if(strncmp(data, "Host:", 5) == 0) { zfqstring tempstr; tempstr = data; tempstr.Cutout("Host: ", NULL); m_hostaddr = tempstr; m_hostaddr.Cutout(NULL, ":"); tempstr.Cutout(":", NULL); m_hostport = atoi(tempstr.GetString()); } else if(strncmp(data, "Authorization:", 14) == 0) { zfqstring tempstr; tempstr = data; tempstr.Cutout("Authorization: Basic ", NULL); tempstr = Base64::Decode(tempstr); m_loginid = tempstr; m_loginpwd = tempstr; m_loginid.Cutout(NULL, ":"); m_loginpwd.Cutout(":", NULL); } else if(strcmp(data, "") == 0) { if(m_method == HTTP_METHOD_POST) { char* post_data = (char*)malloc(m_contentlength + 1); memset(post_data, 0, m_contentlength + 1); Recv(post_data, m_contentlength); m_postdata += post_data; free(post_data); } HttpRequestResponse(); return FALSE; } return TRUE; } BOOL Http::Query() { char command[1024]; memset(command, 0, 1024); int ret = RecvByLine(command, 1024 - 1); if(ret == SOCKET_SENDRECV_OK) { if(!Parse(command)) { return FALSE; } } else if(ret == SOCKET_SENDRECV_ERROR) { return FALSE; } return TRUE; } BOOL Http::HttpRequestResponse() { WebAction* webaction = new FtpWebAction(m_socket, m_id, m_pwd, m_hostaddr, m_hostport, m_postdata, m_page, m_loginid, m_loginpwd); webaction->Response(); delete webaction; return TRUE; }

 

HttpSession类主要负责服务器端轮询，并借助于Http完成对客户端的响应

HttpSession.h

#ifndef _HTTPSESSION_H #define _HTTPSESSION_H #include "Http.h" class HttpSession { public: HttpSession(SOCKET socket, zfqstring id, zfqstring pwd); virtual ~HttpSession(void); public: void LoopResponse(); protected: Http* m_http; }; #endif

HttpSession.cpp

#include "StdAfx.h" #include "HttpSession.h" HttpSession::HttpSession(SOCKET socket, zfqstring id, zfqstring pwd) { m_http = new Http(socket, id, pwd); } HttpSession::~HttpSession(void) { if(NULL != m_http) delete m_http; m_http = NULL; } void HttpSession::LoopResponse() { if(m_http) { while(1) { if(!m_http->Query()) break; } } }

WebAction类是服务器和客户端进行cgi交互的基类

WebAction.h

#ifndef _WEBACTION_H #define _WEBACTION_H #include "zfqstring.h" #include "Constant.h" #include <WinSock2.h> class WebAction { public: WebAction(); virtual ~WebAction(void); public: virtual const char* Body(){return m_body.GetString();} virtual void Response(); virtual int Send(const char* data, size_t len); protected: // zfqstring m_hostaddr; unsigned short m_hostport; // zfqstring m_id; zfqstring m_pwd; // zfqstring m_loginid; zfqstring m_loginpwd; // zfqstring m_postdata; zfqstring m_page; // zfqstring m_body; // SOCKET m_socket; }; #endif

WebAction.cpp

#include "StdAfx.h" #include "WebAction.h" WebAction::WebAction() { } WebAction::~WebAction(void) { } int WebAction::Send(const char* data, size_t len ) { long already_time = 0; struct fd_set fdset; struct timeval timeout; timeout.tv_sec = SELECT_WAIT_TIME; timeout.tv_usec = 0; size_t nSend = 0; size_t tLen = 0; while(1) { FD_ZERO(&fdset); FD_SET(m_socket, &fdset); int ret = select(m_socket + 1, NULL, &fdset, NULL, &timeout); switch(ret) { case 1: already_time = 0; tLen = send(m_socket, data, len - nSend, 0); nSend += tLen; if(nSend == len) return SOCKET_SENDRECV_OK; break; case 0: already_time += SELECT_WAIT_TIME; if(already_time >= MAX_WAIT_TIME_FOR_SENDRECV) { closesocket(m_socket); return SOCKET_SENDRECV_TIMEOUT; } break; default: return SOCKET_SENDRECV_ERROR; } } } void WebAction::Response() { }

FtpWebAction类继承自WebAction，专门用于传输文件