c++ socket http服务端

待我称王封你为后i 2023-10-07 20:15 17阅读 0赞

> 非c++程序员，因要写一个dll注入工具，需要交互，写了一个socket实现的http服务端。
> 
> 由于非c++科班，代码可能会有错误，仅供参考。

#include <map>
    #include<string>
    //soket头文件
    #include<winsock2.h>  
    #include <Ws2tcpip.h> 
    //链接库文件
    #pragma comment(lib,"ws2_32.lib")
    
    using namespace std;
    
    #define PORT 9999//定义服务器端口号
    
    //声明函数
    DWORD WINAPI http_server(LPVOID lpParam);
    void response(SOCKET connect_socket, string html);
    string getURI(string str);
    string getBody(string str);
    map<string, wstring> getParams(string str);
    int hex2num(string str);
    wstring URLDecode(string str);
    wstring ANSI2UTF8(string text);
    
    
    int main()
    {
    	//F1查看函数api
    	/*HANDLE	hThread = CreateThread(NULL, 0, http_server, NULL, 0, NULL);
    	if (hThread != 0)
    	{
    		CloseHandle(hThread);
    	}*/
    	http_server(NULL);
    	return 0;
    }
    
    
    DWORD WINAPI http_server(LPVOID lpParam)
    {
    	//参考官网例子：https://learn.microsoft.com/zh-cn/windows/win32/winsock/complete-server-code?source=recommendations
    	SOCKET listen_socket;//监听socket
    	SOCKET connect_socket;//连接socket
    	WORD ver = MAKEWORD(2, 2);//2.2版本的Socket
    	WSADATA dat;
    	int ret;
    	if (WSAStartup(ver, &dat))
    	{
    		OutputDebugString(L"WSAStartup失败!\n");
    		return 1;
    	}
    
    	//AF_INET：地址族，SOCK_STREAM：连接类型，IPPROTO_TCP：协议类型
    	listen_socket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    	if (listen_socket == INVALID_SOCKET)
    	{
    		WSACleanup();
    		return 1;
    	}
    
    	//第1个参数：秒，第2个参数微秒
    	struct timeval timeout = { 30,0 };
    	//设置接收超时
    	ret = setsockopt(listen_socket, SOL_SOCKET, SO_RCVTIMEO, (char*)&timeout, sizeof(timeval));
    	if (ret == -1) {
    		WSACleanup();
    		return 1;
    	}
    	//设置发送超时
    	ret = setsockopt(listen_socket, SOL_SOCKET, SO_SNDTIMEO, (char*)&timeout, sizeof(timeval));
    	if (ret == -1) {
    		WSACleanup();
    		return 1;
    	}
    
    
    	//创建结构体 sockaddr_in对象储存自身信息
    	struct sockaddr_in sever_address;
    	memset(&sever_address, 0, sizeof(sever_address));
    	sever_address.sin_family = AF_INET;//设置服务器地址家族
    	//sever_address.sin_addr.s_addr = htonl(INADDR_ANY);//设置服务器IP地址
    	//sever_address.sin_addr.s_addr = inet_addr("127.0.0.1");//设置服务器IP地址
    	inet_pton(AF_INET, "127.0.0.1", &sever_address.sin_addr.s_addr);//vs2013版本以上使用新的函数转换IP地址
    	sever_address.sin_port = htons(PORT);//设置服务器端口号
    	//把名字和套接字绑定
    	ret = bind(listen_socket, (sockaddr*)&sever_address, sizeof(sever_address));
    	if (ret == SOCKET_ERROR)
    	{
    		closesocket(listen_socket);
    		WSACleanup();
    		return 1;
    	}
    
    	//listen函数在一般在调用bind之后调用，accept之前调用
    	ret = listen(listen_socket, 1);//1表示是未经过处理的连接请求队列可以容纳的最大数目
    	if (ret == SOCKET_ERROR)
    	{
    		closesocket(listen_socket);
    		WSACleanup();
    		return 1;
    	}
    	while (1)
    	{
    		OutputDebugString(L"等待客户端连接\n");
    		//创建结构体 sockaddr_in对象储存客户端信息
    		struct sockaddr_in client_address;
    		int len = sizeof(client_address);
    		memset(&client_address, 0, len);
    
    		//从监控队列取出一个socket，取出connect_socket后
    		//接受客户端连接，返回一个新的socket来和客户端通信
    		connect_socket = accept(listen_socket, (sockaddr*)&client_address, &len);
    		// 取出connect_socket后应该放入线程池执行，循环继续执行到accept()等待新的连接进来
    		if (connect_socket == INVALID_SOCKET) {
    			closesocket(connect_socket);
    			break;
    		}
    		char buf[1];
    		string buffer;
    		while (1) //循环接收发送的内容
    		{
    			//memset(&buf, 0, sizeof(buf));
    			//阻塞函数,等待接受内容
    			//如果发送方发送了多次信息，接收方没来得及进行recv，则数据堆积在输入缓冲区中，取数据的时候会都取出来。
    			//换句话说，recv并不能判断数据包的结束位置。这里简单使用超时来处理。
    			ret = recv(connect_socket, buf, sizeof(buf), 0);//每次接受的数据大小。结束返回-1
    			if (ret > 0)
    			{
    				buffer.push_back(buf[0]);
    			}
    			else break;
    		}
    		wstring out_buffer = ANSI2UTF8(buffer);
    		OutputDebugString(out_buffer.c_str());
    
    
    		string method = buffer.substr(0, buffer.find(" "));
    		string uri = getURI(buffer);
    		string body = getBody(buffer);
    
    		map<string, wstring> params = getParams(body);
    		string send_body = "成功";
    		if (uri.compare("/api/sendtxtmsg") == 0) {
    			send_body = "发送文本消息成功";
    			response(connect_socket, send_body);
    		}
    		else if (uri.compare("/api/sendpic") == 0) {
    			send_body = "发送图片消息成功";
    			response(connect_socket, send_body);
    		}
    		else if (uri.compare("/api/sendattatch") == 0) {
    			send_body = "发送文件消息成功";
    			response(connect_socket, send_body);
    		}
    		else {
    			send_body = "404";
    			response(connect_socket, send_body);
    		}
    		closesocket(connect_socket);
    
    	}
    	closesocket(listen_socket);
    	WSACleanup();
    	ret = getchar();
    	//Sleep(10*1000);//1000表示睡眠1000毫秒（及一秒）
    	return 0;
    }
    
    
    //响应数据到客户端
    void response(SOCKET connect_socket, string body) {
    	char head[] = "HTTP/1.1 200 OK\r\nContent-Type: text/html;charset=GB2312\r\n\r\n";
    	send(connect_socket, head, strlen(head), 0);
    	//不论是客户还是服务器应用程序都用send函数来向TCP连接的另一端发送数据
    	send(connect_socket, body.c_str(), strlen(body.c_str()), 0);
    }
    //截取请求地址
    string getURI(string str) {
    	int posStart = str.find(" ") + 1;
    	int posEnd = str.find(" ", posStart);
    	string uri = str.substr(posStart, posEnd - posStart);
    	posStart = uri.find("?");
    	if (posStart > 0) {
    		uri = uri.substr(0, posStart);
    	}
    	return uri;
    }
    //截取请求体
    string getBody(string str) {
    	if (str.empty()) return str;
    	int posStart = str.find("\r\n\r\n") + 4;
    	int size = str.size();
    	string body = str.substr(posStart, size - posStart);
    	return body;
    }
    
    //解析请求参数
    map<string, wstring> getParams(string str) {
    	int pos;
    	map<string, wstring> params;
    	string param, key, val;
    	wstring wval;
    	str += "&";//扩展字符串以方便判断
    	int size = str.size();
    	for (int i = 0; i < size; i++)
    	{
    		pos = str.find("&", i);
    		if (pos > 0)
    		{
    			param = str.substr(i, pos - i);
    			i = pos;
    			pos = param.find("=");
    			key = param.substr(0, pos);
    			val = param.substr(pos + 1, param.size() - pos - 1);
    			wval = URLDecode(val);
    			params[key] = wval;
    
    		}
    	}
    	return params;
    }
    
    //16进制转10进制
    int hex2num(string str)
    {
    	char c;
    	int num;
    	int total = 0;
    	int size = str.size();
    	for (int i = 0; i < size; i++)
    	{
    		c = str[i];
    		switch (c)
    		{
    		case 'A': num = 10; break;
    		case 'B': num = 11; break;
    		case 'C': num = 12; break;
    		case 'D': num = 13; break;
    		case 'E': num = 14; break;
    		case 'F': num = 15; break;
    		default:  num = c - '0'; break;
    		}
    		total += (num * (int)pow(16, size - 1 - i));//FC = 15*16^1 + 12*16^0
    	}
    	return total;
    }
    
    // url解码
    wstring URLDecode(string str)
    {
    	char c;
    	int num;
    	int size = str.size();
    	string result;
    	for (int i = 0; i < size; i++) {
    		c = str[i];
    		switch (c) {
    		case '+':
    			result.push_back(' ');
    			break;
    		case '%':
    			// 每3个字符串%xy将产生一个字节
    			if (i + 2 < size) {
    				num = hex2num(str.substr(i + 1, 2));
    				result.push_back((char)num);
    				i += 2;
    			}
    			break;
    		default:
    			result.push_back(c);
    			break;
    		}
    	}
    	wstring wresult = ANSI2UTF8(result);
    	return wresult;
    }
    
    //ANSI转UTF8
    wstring ANSI2UTF8(string text) {
    	/*
    	MultiByteToWideChar 参数详解
    	1、CodePage：编码类型
    	2、dwFlags：UTF-8的dwFlags必须设置为0
    	3、lpMultiByteStr：要转换的字符串。注意c_str()函数：将字符串的内容转换为以 null 结尾的 C 样式字符串。
    	4、cbMultiByte：要转换的字符串的大小（以字节为单位）。 或者，如果字符串以 null 结尾，则可以将此参数设置为 -1。
    	5、lpWideCharStr：接收转换后的字符串
    	6、cchWideChar：指示的缓冲区的大小（以字符为单位）。 如果此值为 0，则该函数将返回所需的缓冲区大小（以字符为单位，包括任何终止 null 字符）并且不使用 lpWideCharStr 缓冲区。
    	*/
    	//获取转为UTF8多字节后需要的缓冲区大小
    	int len = MultiByteToWideChar(CP_UTF8, 0, text.c_str(), -1, NULL, 0);
    	//创建多字节缓冲区
    	WCHAR* buffer = new WCHAR[len];
    	MultiByteToWideChar(CP_UTF8, 0, text.c_str(), -1, buffer, len);
    	buffer[len - 1] = '\0';
    	wstring wstr(buffer);
    	delete[] buffer;
    	return wstr;
    }

浏览器访问 http://localhost:9999/api/sendattatch