|
10 | 10 | // Purpose: 代理服务协议 |
11 | 11 | // History: |
12 | 12 | *********************************************************************/ |
13 | | -void CALLBACK XEngine_Proxy_CBRecv(XHANDLE xhToken, XNETHANDLE xhClient, XSOCKET hSocket, ENUM_XCLIENT_SOCKET_EVENTS enTCPClientEvents, LPCXSTR lpszMsgBuffer, int nMsgLen, XPVOID lParam) |
14 | | -{ |
15 | | - SESSION_FORWARD st_ProxyInfo = {}; |
16 | | - if (!ModuleSession_Proxy_GetForToken(xhClient, &st_ProxyInfo)) |
17 | | - { |
18 | | - XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_ERROR, _X("Proxy客户端:%lld,获取转发地址信息失败,原始地址:%s,目标地址:%s,错误码:%lX"), xhClient, st_ProxyInfo.tszSrcAddr, st_ProxyInfo.tszDstAddr, ModuleSession_GetLastError()); |
19 | | - return; |
20 | | - } |
21 | | - XEngine_Network_Send(st_ProxyInfo.tszSrcAddr, lpszMsgBuffer, nMsgLen, XENGINE_CLIENT_NETTYPE_PROXY); |
22 | | - XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_DEBUG, _X("Proxy客户端:%lld,转发数据成功,原始地址:%s,目标地址:%s,大小:%d"), xhClient, st_ProxyInfo.tszSrcAddr, st_ProxyInfo.tszDstAddr, nMsgLen); |
23 | | -} |
24 | | - |
25 | 13 | bool XEngine_Proxy_Connect(LPCXSTR lpszClientAddr) |
26 | 14 | { |
27 | | - int nPort = 0; |
| 15 | + int nSrcPort = 0; |
| 16 | + int nDstPort = 0; |
28 | 17 | XNETHANDLE xhClient = 0; |
29 | 18 | XCHAR tszIPAddr[128] = {}; |
| 19 | + XCHAR tszSrcIPAddr[128] = {}; |
| 20 | + XCHAR tszDstIPAddr[128] = {}; |
30 | 21 |
|
31 | | - _tcsxcpy(tszIPAddr, st_ServiceConfig.st_XProxy.tszIPAddr); |
32 | | - APIAddr_IPAddr_SegAddr(tszIPAddr, &nPort); |
33 | | - if (!XClient_TCPSelect_InsertEx(xhProxyClient, &xhClient, tszIPAddr, nPort, false)) |
| 22 | + _tcsxcpy(tszSrcIPAddr, lpszClientAddr); |
| 23 | + APIAddr_IPAddr_SegAddr(tszSrcIPAddr, &nSrcPort); |
| 24 | + |
| 25 | + bool bFound = false; |
| 26 | + auto stl_ListIterator = st_ServiceConfig.st_XProxy.pStl_ListIPAddr->begin(); |
| 27 | + for (; stl_ListIterator != st_ServiceConfig.st_XProxy.pStl_ListIPAddr->end(); stl_ListIterator++) |
34 | 28 | { |
35 | | - XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_ERROR, _X("Proxy客户端:%s,连接到转发代理服务器失败,服务地址:%s,错误码:%lX"), lpszClientAddr, st_ServiceConfig.st_XProxy.tszIPAddr, XClient_GetLastError()); |
| 29 | + XCHAR tszTmpIPAddr[128] = {}; |
| 30 | + _stxscanf(stl_ListIterator->c_str(), _X("%[^-]-%s"), tszTmpIPAddr, tszDstIPAddr); |
| 31 | + if (APIAddr_IPAddr_IsIPV4Addr(tszTmpIPAddr)) |
| 32 | + { |
| 33 | + //ip |
| 34 | + if (0 == _tcsxnicmp(tszSrcIPAddr, tszTmpIPAddr, _tcsxlen(tszSrcIPAddr))) |
| 35 | + { |
| 36 | + bFound = true; |
| 37 | + _tcsxcpy(tszIPAddr, tszDstIPAddr); |
| 38 | + APIAddr_IPAddr_SegAddr(tszDstIPAddr, &nDstPort); |
| 39 | + break; |
| 40 | + } |
| 41 | + } |
| 42 | + else |
| 43 | + { |
| 44 | + //port |
| 45 | + if (_ttxoi(tszTmpIPAddr) == nSrcPort) |
| 46 | + { |
| 47 | + bFound = true; |
| 48 | + _tcsxcpy(tszIPAddr, tszDstIPAddr); |
| 49 | + APIAddr_IPAddr_SegAddr(tszDstIPAddr, &nDstPort); |
| 50 | + break; |
| 51 | + } |
| 52 | + } |
| 53 | + } |
| 54 | + |
| 55 | + if (!bFound) |
| 56 | + { |
| 57 | + _tcsxcpy(tszDstIPAddr, st_ServiceConfig.st_XProxy.tszIPAddr); |
| 58 | + APIAddr_IPAddr_SegAddr(tszDstIPAddr, &nDstPort); |
| 59 | + XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_INFO, _X("Proxy客户端:%s,代理转未命中,使用默认地址:%s:%d"), lpszClientAddr, tszDstIPAddr, nDstPort); |
| 60 | + } |
| 61 | + if (!XClient_TCPSelect_InsertEx(xhProxyClient, &xhClient, tszDstIPAddr, nDstPort, false)) |
| 62 | + { |
| 63 | + XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_ERROR, _X("Proxy客户端:%s,连接到转发代理服务器失败,服务地址:%s:%d,错误码:%lX"), lpszClientAddr, tszDstIPAddr, nDstPort, XClient_GetLastError()); |
36 | 64 | return false; |
37 | 65 | } |
38 | 66 | SocketOpt_HeartBeat_InsertAddrEx(xhProxyHeart, lpszClientAddr); |
39 | | - ModuleSession_Proxy_Insert(lpszClientAddr, st_ServiceConfig.st_XProxy.tszIPAddr, xhClient); |
40 | | - XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_INFO, _X("Proxy客户端:%s,连接到服务器:%s 成功"), lpszClientAddr, st_ServiceConfig.st_XProxy.tszIPAddr); |
| 67 | + ModuleSession_Proxy_Insert(lpszClientAddr, tszIPAddr, xhClient); |
| 68 | + XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_INFO, _X("Proxy客户端:%s,连接到服务器:%s:%d 成功"), lpszClientAddr, tszDstIPAddr, nDstPort); |
41 | 69 | return true; |
42 | 70 | } |
| 71 | +void CALLBACK XEngine_Proxy_CBRecv(XHANDLE xhToken, XNETHANDLE xhClient, XSOCKET hSocket, ENUM_XCLIENT_SOCKET_EVENTS enTCPClientEvents, LPCXSTR lpszMsgBuffer, int nMsgLen, XPVOID lParam) |
| 72 | +{ |
| 73 | + SESSION_FORWARD st_ProxyInfo = {}; |
| 74 | + if (!ModuleSession_Proxy_GetForToken(xhClient, &st_ProxyInfo)) |
| 75 | + { |
| 76 | + XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_ERROR, _X("Proxy客户端:%lld,获取转发地址信息失败,原始地址:%s,目标地址:%s,错误码:%lX"), xhClient, st_ProxyInfo.tszSrcAddr, st_ProxyInfo.tszDstAddr, ModuleSession_GetLastError()); |
| 77 | + return; |
| 78 | + } |
| 79 | + XEngine_Network_Send(st_ProxyInfo.tszSrcAddr, lpszMsgBuffer, nMsgLen, XENGINE_CLIENT_NETTYPE_PROXY); |
| 80 | + XLOG_PRINT(xhLog, XENGINE_HELPCOMPONENTS_XLOG_IN_LOGLEVEL_DEBUG, _X("Proxy客户端:%lld,转发数据成功,原始地址:%s,目标地址:%s,大小:%d"), xhClient, st_ProxyInfo.tszSrcAddr, st_ProxyInfo.tszDstAddr, nMsgLen); |
| 81 | +} |
43 | 82 | bool XEngine_Proxy_Recvmsg(LPCXSTR lpszClientAddr, LPCXSTR lpszMsgBuffer, int nMsgLen) |
44 | 83 | { |
45 | 84 | SESSION_FORWARD st_ProxyInfo = {}; |
|
0 commit comments