开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 模拟服务器 > 查看源码
sockspacket.cpp
资源名称：SwordOnline.rar　[点击查看]
上传用户：dzyhzl
上传日期：2019-04-29
资源大小：56270k
文件大小：21k
源码类别：

模拟服务器

开发平台：

C/C++

sockspacket.cpp：源码内容
  2. ////////////////////////////////////////////////////////////////////////////////
  3. // SocksPacket.cpp: implementation of the CSocksPacket class.
  4. ////////////////////////////////////////////////////////////////////////////////
  5. #include "stdafx.h"
  6. #include "SocksPacket.h"
  8. //////////////////////////////////////////////////////////////////////
  9. // 构造函数
  10. //////////////////////////////////////////////////////////////////////
  11. CSocksPacket::CSocksPacket( CBufSocket* pBufSocket )
  12. {
  13. m_pBufSocket = pBufSocket;
  14. m_nRepPacketType = 0;
  15. }
  17. //////////////////////////////////////////////////////////////////////
  18. // 析构函数
  19. //////////////////////////////////////////////////////////////////////
  20. CSocksPacket::~CSocksPacket()
  21. {
  23. }
  25. ////////////////////////////////////////////////////////////////////////////////
  26. // 函数名：BOOL Pack(
  27. // OUT char *pszDest,
  28. // PSOCKSREQPACKET pPacket,
  29. // int nPacketType ) 
  30. // 用  途：将数据包打包成字符串
  31. // 对全局变量的影响：无
  32. // 参  数：
  33. // pszDest     : 字符串指针,用于保存打包数据(必须预先分配足够的内存空间)
  34. // pPacket     : 数据包指针(待打包的数据包)
  35. // nPacketType : 包类型
  36. // 返回值：BOOL
  37. ////////////////////////////////////////////////////////////////////////////////
  38. BOOL CSocksPacket::Pack(OUT char *pszDest,PSOCKSREQPACKET pPacket,int nPacketType)
  39. {
  41. int n,nIndex=0;
  42. WORD  wData;
  43. DWORD dwData;
  45. switch( nPacketType )
  46. {
  47. case PACKET_SOCKS4REQ:
  48. pszDest[nIndex++] = (char)pPacket->socks4Req.byVer;
  49. pszDest[nIndex++] = (char)pPacket->socks4Req.byCmd;
  51. // 端口
  52. wData = htons(pPacket->socks4Req.wDestPort);
  53. pszDest[nIndex++] = (char)( wData & 0x00ff );
  54. pszDest[nIndex++] = (char)( (wData>>8) & 0x00ff);
  56. // IP地址(不需要再进行htonl转换,由inet_addr出来的IP地址已经是网络字节序)
  57. dwData = pPacket->socks4Req.dwDestIP;
  58. pszDest[nIndex++] = (char)( (dwData    ) &0x000000ff); 
  59. pszDest[nIndex++] = (char)( (dwData>>8 ) &0x000000ff); 
  60. pszDest[nIndex++] = (char)( (dwData>>16) &0x000000ff); 
  61. pszDest[nIndex++] = (char)( (dwData>>24) &0x000000ff); 
  63. // UserID
  64. for( n = 0;n < (int)strlen(pPacket->socks4Req.pszUserID) ; n++ )
  65. pszDest[nIndex++] = (pPacket->socks4Req.pszUserID)[n];
  67. // NULL BYTE
  68. pszDest[nIndex++] = (char)pPacket->socks4Req.byNULL;
  69. pszDest[nIndex] = '';
  70. break;
  71. case PACKET_SOCKS4AREQ:
  72. pszDest[nIndex++] = (char)pPacket->socks4aReq.byVer;
  73. pszDest[nIndex++] = (char)pPacket->socks4aReq.byCmd;
  75. // 端口
  76. wData = htons(pPacket->socks4aReq.wDestPort);
  77. pszDest[nIndex++] = (char)( wData & 0x00ff );
  78. pszDest[nIndex++] = (char)( (wData>>8) & 0x00ff);
  80. // IP地址(不需要再进行htonl转换,由inet_addr出来的IP地址已经是网络字节序)
  81. dwData = pPacket->socks4aReq.dwDestIP;
  82. pszDest[nIndex++] = (char)( (dwData    ) &0x000000ff); 
  83. pszDest[nIndex++] = (char)( (dwData>>8 ) &0x000000ff); 
  84. pszDest[nIndex++] = (char)( (dwData>>16) &0x000000ff); 
  85. pszDest[nIndex++] = (char)( (dwData>>24) &0x000000ff); 
  87. // UserID
  88. for( n = 0;n < (int)strlen(pPacket->socks4aReq.pszUserID) ; n++ )
  89. pszDest[nIndex++] = (pPacket->socks4aReq.pszUserID)[n];
  91. // NULL1 BYTE
  92. pszDest[nIndex++] = (char)pPacket->socks4aReq.byNULL1;
  94. // DestHostName
  95. for( n = 0;n < (int)strlen(pPacket->socks4aReq.pszDestHostName); n++ )
  96. pszDest[nIndex++] = (pPacket->socks4aReq.pszDestHostName)[n];
  98. // NULL2 BYTE
  99. pszDest[nIndex++] = (char)pPacket->socks4aReq.byNULL2;
  100. pszDest[nIndex] = '';
  101. break;
  102. case PACKET_SOCKS5AUTHREQ:
  103. pszDest[nIndex++] = (char)pPacket->socks5AuthReq.byVer;
  104. pszDest[nIndex++] = (char)pPacket->socks5AuthReq.byNAuth;
  105. for( n = 0; n < pPacket->socks5AuthReq.byNAuth;n++ )
  106. pszDest[nIndex++] = (pPacket->socks5AuthReq.pszAuth)[n];
  107. pszDest[nIndex] = '';
  108. break;
  109. case PACKET_SOCKS5AUTHPASSWDREQ:
  110. pszDest[nIndex++] = (char)pPacket->socks5AuthPasswdReq.byVer;
  111. pszDest[nIndex++] = (char)pPacket->socks5AuthPasswdReq.byUserNameLen;
  113. for( n = 0 ; n < (int)strlen( pPacket->socks5AuthPasswdReq.pszUserName ); n++ )
  114. pszDest[nIndex++] = (pPacket->socks5AuthPasswdReq.pszUserName)[n];
  116. pszDest[nIndex++] = (char)pPacket->socks5AuthPasswdReq.byPasswdLen;
  118. for( n = 0; n < (int)strlen(pPacket->socks5AuthPasswdReq.pszPasswd); n++ )
  119. pszDest[nIndex++] = (pPacket->socks5AuthPasswdReq.pszPasswd)[n];
  120. pszDest[nIndex] = '';
  121. break;
  122. case PACKET_SOCKS5REQ:
  123. pszDest[nIndex++] = (char)pPacket->socks5Req.byVer;
  124. pszDest[nIndex++] = (char)pPacket->socks5Req.byCmd;
  125. pszDest[nIndex++] = (char)pPacket->socks5Req.byRsv;
  126. pszDest[nIndex++] = (char)pPacket->socks5Req.byAtyp;
  128. int nAddrLen;
  129. switch(pPacket->socks5Req.byAtyp)
  130. {
  131. case ATYP_IPV4ADDR:
  132. nAddrLen = 4;
  133. /*
  134. 如果目标地址IP = "123.222.111.222"
  135. 则传进来的参数socks5Req.pszDestAddr的形式应该是这样的:
  136. DWORD dwIP = inet_addr("123.222.111.222");
  137. char*  szAddr = new char[5];
  138. szAddr[0] = (char)( dwIP      & 0x000000ff);
  139. szAddr[1] = (char)((dwIP>>8 ) & 0x000000ff);
  140. szAddr[2] = (char)((dwIP>>16) & 0x000000ff);
  141. szAddr[3] = (char)((dwIp>>24) & 0x000000ff);
  142. szAddr[4] = '';
  144. socks5Req.pszDestAddr = szAddr;
  145. */
  146. break;
  147. case ATYP_IPV6ADDR:
  148. nAddrLen = 16;
  149. break;
  150. case ATYP_HOSTNAME:
  151. nAddrLen =  1;
  152. nAddrLen += (BYTE) (pPacket->socks5Req.pszDestAddr[0]);
  153. /*
  154. 如果目标地址DM = "www.yahoo.com"
  155. 则传进来的参数socks5Req.pszDestAddr的形式应该是这样的:
  156. int nLen = strlen("www.yahoo.com");
  157. char* szAddr = new char[nLen+2];
  158. szAddr[0] = (char)nLen; // 第一个字节保存域名的长度
  159. for( int i =0;i<nLen;i++) // 从第二个字节开始才是真正的域名
  160. szAddr[i+1] = DM[i];
  161. szAddr[nLen+1] = '';
  163. socks5Req.pszDestAddr = szAddr;
  164. */
  165. break;
  166. default:
  167. nAddrLen = (int)strlen(pPacket->socks5Req.pszDestAddr);
  168. break;
  169. }
  171. for( n = 0; n <nAddrLen; n++ )
  172. pszDest[nIndex++] = (pPacket->socks5Req.pszDestAddr)[n];
  174. wData = htons(pPacket->socks5Req.wDestPort);
  175. pszDest[nIndex++] = (char)( wData & 0x00ff );
  176. pszDest[nIndex++] = (char)( (wData>>8) & 0x00ff);
  177. pszDest[nIndex] = '';
  178. break;
  179. }
  181. //打包成功
  182. return TRUE;
  183. }
  185. ////////////////////////////////////////////////////////////////////////////////
  186. // 函数名：int PacketSize(
  187. // PSOCKSREQPACKET pPacket,
  188. // int nPacketType ) 
  189. // 用  途：计算包的尺寸
  190. // 对全局变量的影响：无
  191. // 参  数：
  192. // pPacket     : 数据包指针
  193. // nPacketType : PACKET类型
  194. // 返回值：int，数据包长度
  195. ////////////////////////////////////////////////////////////////////////////////
  196. int CSocksPacket::PacketSize(PSOCKSREQPACKET pPacket,int nPacketType)
  197. {
  198. int nLen = 0;
  199. switch( nPacketType )
  200. {
  201. case PACKET_SOCKS4REQ:
  202. nLen = 9 + strlen(pPacket->socks4Req.pszUserID);
  203. break;
  204. case PACKET_SOCKS4AREQ:
  205. nLen = 10 + strlen(pPacket->socks4aReq.pszUserID) + strlen(pPacket->socks4aReq.pszDestHostName);
  206. break;
  207. case PACKET_SOCKS5AUTHREQ:
  208. nLen = 2 + (BYTE)pPacket->socks5AuthReq.byNAuth;
  209. break;
  210. case PACKET_SOCKS5AUTHPASSWDREQ:
  211. nLen = 3 + strlen(pPacket->socks5AuthPasswdReq.pszUserName) + strlen(pPacket->socks5AuthPasswdReq.pszPasswd);
  212. break;
  213. case PACKET_SOCKS5REQ:
  214. nLen = 6;
  215. switch( pPacket->socks5Req.byAtyp )
  216. {
  217. case ATYP_IPV4ADDR:
  218. nLen += 4;
  219. break;
  220. case ATYP_IPV6ADDR:
  221. nLen += 16;
  222. break;
  223. case ATYP_HOSTNAME:
  224. nLen += ((BYTE)pPacket->socks5Req.pszDestAddr[0]);
  225. nLen += 1;
  226. break;
  227. }
  228. break;
  229. }
  230. return nLen;
  231. }
  234. ////////////////////////////////////////////////////////////////////////////////
  235. // 函数名：BOOL SendPacket(
  236. // PSOCKSREQPACKET pPacket,
  237. // int nPacketType,
  238. // DWORD dwTimeout/* = SOP_SEND_TIMEOUT*/ ) 
  239. // 用  途：发送包
  240. // 对全局变量的影响：无
  241. // 参  数：
  242. // pPacket     : 数据包指针(发送数据包)
  243. // nPacketType : 数据包类型
  244. // dwTimeout   :  发送超时(缺省为SOP_SEND_TIMEOUT)
  245. // 返回值：BOOL
  246. ////////////////////////////////////////////////////////////////////////////////
  247. BOOL CSocksPacket::SendPacket(PSOCKSREQPACKET pPacket,int nPacketType,DWORD dwTimeout/* = SOP_SEND_TIMEOUT*/)
  248. {
  249. // 计算包大小
  250. int nSize = PacketSize(pPacket,nPacketType);
  251. char* pSendBuf = (char *)malloc(nSize+1);
  253. if(pSendBuf == NULL)
  254. return FALSE;
  256. int nSend;
  257. BOOL bRet = FALSE;
  259. // 打包并发送
  260. if( Pack(pSendBuf,pPacket,nPacketType) )
  261. {
  262. nSend = m_pBufSocket->Send(pSendBuf,nSize,dwTimeout);
  263. bRet  = (nSend == nSize)? TRUE:FALSE;
  264. }
  266. free(pSendBuf);
  267. return (bRet);
  268. }
  270. ////////////////////////////////////////////////////////////////////////////////
  271. // 函数名：BOOL SendSocks4Req(
  272. // BYTE byCmd,
  273. // WORD wDestPort,
  274. // DWORD dwDestIP,
  275. // const char * pszUserID /* = NULL */,
  276. // DWORD dwTimeout /* = SOP_SEND_TIMEOUT */ ) 
  277. // 用  途：发送Socks4请求
  278. // 对全局变量的影响：无
  279. // 参  数：
  280. // byCmd     : 命令
  281. // wDestPort : 目标端口
  282. // dwDestIP  : 目标IP
  283. // pszUserID : 用户ID
  284. // dwTimeout : 超时
  285. // 返回值：BOOL
  286. ////////////////////////////////////////////////////////////////////////////////
  287. BOOL CSocksPacket::SendSocks4Req(BYTE byCmd,WORD wDestPort,DWORD dwDestIP,const char * pszUserID /* = NULL */,DWORD dwTimeout /* = SOP_SEND_TIMEOUT */)
  288. {
  289. SOCKSREQPACKET pack;
  290. pack.socks4Req.byVer = VER_SOCKS4;
  291. pack.socks4Req.byCmd = byCmd;
  292. pack.socks4Req.wDestPort = wDestPort;
  293. pack.socks4Req.dwDestIP = dwDestIP;
  294. pack.socks4Req.pszUserID = pszUserID;
  295. pack.socks4Req.byNULL = BYTE_NULL;
  297. return SendPacket(&pack,PACKET_SOCKS4REQ,dwTimeout);
  298. }
  301. ////////////////////////////////////////////////////////////////////////////////
  302. // 函数名：BOOL SendSocks4aReq(
  303. // BYTE byCmd,
  304. // WORD wDestPort,
  305. // const char* pszDestHostName,
  306. // const char * pszUserID /* = NULL */,
  307. // DWORD dwTimeout /* = SOP_SEND_TIMEOUT */ ) 
  308. // 用  途：发送Socks4a请求
  309. // 对全局变量的影响：无
  310. // 参  数：
  311. // byCmd           : 命令
  312. // wDestPort       : 端口
  313. // pszDestHostName : 主机名
  314. // pszUserID       : 用户ID
  315. // dwTimeout       : 超时
  316. // 返回值：BOOL
  317. ////////////////////////////////////////////////////////////////////////////////
  318. BOOL  CSocksPacket::SendSocks4aReq(BYTE byCmd,WORD wDestPort,const char* pszDestHostName,const char * pszUserID /* = NULL */,DWORD dwTimeout /* = SOP_SEND_TIMEOUT */)
  319. {
  320. SOCKSREQPACKET pack;
  321. pack.socks4aReq.byVer = VER_SOCKS4A;
  322. pack.socks4aReq.byCmd = byCmd;
  323. pack.socks4aReq.wDestPort = wDestPort;
  324. pack.socks4aReq.dwDestIP = UNKNOWN_IP;
  325. pack.socks4aReq.pszUserID = pszUserID;
  326. pack.socks4aReq.byNULL1 = BYTE_NULL;
  327. pack.socks4aReq.pszDestHostName = pszDestHostName;
  328. pack.socks4aReq.byNULL2 = BYTE_NULL;
  330. return SendPacket(&pack,PACKET_SOCKS4AREQ,dwTimeout);
  331. }
  333. ////////////////////////////////////////////////////////////////////////////////
  334. // 函数名：BOOL SendSocks5AuthReq(
  335. // BYTE byNAuth,
  336. // const char* pszAuth,
  337. // DWORD dwTimeout /* = SOP_SEND_TIMEOUT */ ) 
  338. // 用  途：发送SOCKS5认证请求 
  339. // 对全局变量的影响：无
  340. // 参  数：
  341. // byNAuth   : 认证方式数量
  342. // pszAuth   : 认证方式
  343. // dwTimeout : 超时
  344. // 返回值：BOOL
  345. ////////////////////////////////////////////////////////////////////////////////
  346. BOOL  CSocksPacket::SendSocks5AuthReq(BYTE byNAuth,const char* pszAuth,DWORD dwTimeout /* = SOP_SEND_TIMEOUT */)
  347. {
  348. SOCKSREQPACKET pack;
  349. pack.socks5AuthReq.byVer = VER_SOCKS5;
  350. pack.socks5AuthReq.byNAuth = byNAuth;
  351. pack.socks5AuthReq.pszAuth = pszAuth;
  353. return SendPacket(&pack,PACKET_SOCKS5AUTHREQ,dwTimeout);
  354. }
  357. ////////////////////////////////////////////////////////////////////////////////
  358. // 函数名：BOOL SendSocks5AuthPasswdReq(
  359. // const char* pszUserName,
  360. // const char * pszPassword,
  361. // DWORD dwTimeout /* = SOP_SEND_TIMEOUT */ ) 
  362. // 用  途：发送SOCKS5 USERNAME/PASSWORD验证请求
  363. // 对全局变量的影响：无
  364. // 参  数：
  365. // pszUserName : 用户名
  366. // pszPassword : 口令
  367. // dwTimeout   : 超时
  368. // 返回值：BOOL
  369. ////////////////////////////////////////////////////////////////////////////////
  370. BOOL  CSocksPacket::SendSocks5AuthPasswdReq(const char* pszUserName,const char * pszPassword,DWORD dwTimeout /* = SOP_SEND_TIMEOUT */)
  371. {
  372. SOCKSREQPACKET pack;
  373. pack.socks5AuthPasswdReq.byVer = 0x01;
  374. pack.socks5AuthPasswdReq.byUserNameLen = (BYTE)( (pszUserName == NULL)?0:strlen(pszUserName) );
  375. pack.socks5AuthPasswdReq.pszUserName = pszUserName;
  376. pack.socks5AuthPasswdReq.byPasswdLen = (BYTE)( (pszPassword == NULL)?0:strlen(pszPassword) );
  377. pack.socks5AuthPasswdReq.pszPasswd = pszPassword;
  379. return SendPacket(&pack,PACKET_SOCKS5AUTHPASSWDREQ,dwTimeout);
  380. }
  383. ////////////////////////////////////////////////////////////////////////////////
  384. // 函数名：BOOL SendSocks5Req(
  385. // BYTE byCmd,
  386. // BYTE byAtyp,
  387. // const char * pszDestAddr,
  388. // WORD wDestPort,
  389. // DWORD dwTimeout /* = SOP_SEND_TIMEOUT */ ) 
  390. // 用  途：发送SOCKS5请求
  391. // 对全局变量的影响：无
  392. // 参  数：
  393. // byCmd       : 命令
  394. // byAtyp      : 地址类型
  395. // ATYP_IPV4ADDR pszDestAddr = IPV4_Address (4 Octects)
  396. // ATYP_IPV6ADDR pszDestAddr = IPV6_Address (16 Octects) 
  397. // ATYP_HOSTNAME pszDestAddr = [1BYTE:(HostName)Length]+[HostName]
  398. // pszDestAddr : 目标地址
  399. // wDestPort   : 目标端口
  400. // dwTimeout   : 超时
  401. // 返回值：BOOL
  402. ////////////////////////////////////////////////////////////////////////////////
  403. BOOL  CSocksPacket::SendSocks5Req(BYTE byCmd,BYTE byAtyp,const char * pszDestAddr,WORD wDestPort,DWORD dwTimeout /* = SOP_SEND_TIMEOUT */)
  404. {
  405. SOCKSREQPACKET pack;
  406. pack.socks5Req.byVer = VER_SOCKS5;
  407. pack.socks5Req.byCmd = byCmd;
  408. pack.socks5Req.byRsv = BYTE_NULL;
  409. pack.socks5Req.byAtyp = byAtyp;
  410. pack.socks5Req.pszDestAddr = pszDestAddr;
  411. pack.socks5Req.wDestPort = wDestPort;
  412. return SendPacket(&pack,PACKET_SOCKS5REQ,dwTimeout);
  413. }
  416. ////////////////////////////////////////////////////////////////////////////////
  417. // 函数名：BOOL IsSocksOK(
  418. // PSOCKSREPPACKET pPacket,
  419. // int nPacketType ) 
  420. // 用  途：检查SOCKS的返回值是否正确
  421. // 对全局变量的影响：无
  422. // 参  数：
  423. // pPacket     : 数据包
  424. // nPacketType : 数据包类型
  425. // 返回值：BOOL
  426. ////////////////////////////////////////////////////////////////////////////////
  427. BOOL CSocksPacket::IsSocksOK( PSOCKSREPPACKET pPacket,int nPacketType )
  428. {
  429. switch ( nPacketType )
  430. {
  431. case PACKET_SOCKS4REP:
  432. return ( pPacket->socks4Rep.byRep == REP4_SUCCESS );
  433. break;
  434. case PACKET_SOCKS4AREP:
  435. return ( pPacket->socks4aRep.byRep == REP4_SUCCESS );
  436. break;
  437. case PACKET_SOCKS5AUTHREP:
  438. return ( pPacket->socks5AuthRep.byAuth == AUTH_NONE  ||
  439.  pPacket->socks5AuthRep.byAuth == AUTH_GSSAPI ||
  440.  pPacket->socks5AuthRep.byAuth == AUTH_PASSWD ||
  441.  pPacket->socks5AuthRep.byAuth == AUTH_CHAP ) ;
  442. break;
  443. case PACKET_SOCKS5AUTHPASSWDREP:
  444. return ( pPacket->socks5AuthPasswdRep.byStatus == AUTH_OK );
  445. break;
  446. case PACKET_SOCKS5REP:
  447. return ( pPacket->socks5Rep.byRep == REP5_SUCCESS );
  448. break;
  449. default:
  450. return FALSE;
  451. break;
  452. }
  453. }
  455. ////////////////////////////////////////////////////////////////////////////////
  456. // 函数名：BOOL RecvPacket(
  457. // OUT PSOCKSREPPACKET pPacket,
  458. // int nPacketType,
  459. // DWORD dwTimeout /*= SOP_RECV_TIMEOUT*/ ) 
  460. // 用  途：接收并解包
  461. // 对全局变量的影响：无
  462. // 参  数：
  463. // pPacket     : 数据包指针(用来保存接收数据包)
  464. // nPacketType : 包类型
  465. // dwTimeout   : 接收超时(缺省为SOP_RECV_TIMEOUT)
  466. // 返回值：BOOL
  467. ////////////////////////////////////////////////////////////////////////////////
  468. BOOL CSocksPacket::RecvPacket( OUT PSOCKSREPPACKET pPacket,int nPacketType,DWORD dwTimeout /*= SOP_RECV_TIMEOUT*/)
  469. {
  470. int nRecv,n,nCount;
  471. WORD wData = 0;
  472. DWORD dwData = 0;
  473. char* pszData = NULL;
  475. // 保存类型
  476. m_nRepPacketType = nPacketType;
  478. //接收数据
  479. switch( nPacketType )
  480. {
  481. case PACKET_SOCKS4REP:
  482. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  483. if( nRecv == INVALID_CHAR )
  484. return FALSE;
  485. pPacket->socks4Rep.byVer = (BYTE)nRecv;
  487. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  488. if( nRecv == INVALID_CHAR )
  489. return FALSE;
  490. pPacket->socks4Rep.byRep = (BYTE)nRecv;
  492. wData = 0;
  493. for( n = 0; n < 2; n++)
  494. {
  495. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  496. if( nRecv == INVALID_CHAR )
  497. return FALSE;
  499. wData |= ( ((WORD)nRecv) << (8*n) );
  500. }
  501. pPacket->socks4Rep.wDestPort =ntohs(wData);
  503. dwData = 0;
  504. for( n = 0; n < 4; n++)
  505. {
  506. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  507. if( nRecv == INVALID_CHAR )
  508. return FALSE;
  510. dwData |= ( ((DWORD)nRecv)<<(8*n) );
  511. }
  512. pPacket->socks4Rep.dwDestIP = ntohl(dwData);
  513. break;
  514. case PACKET_SOCKS4AREP:
  515. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  516. if( nRecv == INVALID_CHAR )
  517. return FALSE;
  518. pPacket->socks4aRep.byVer = (BYTE)nRecv;
  520. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  521. if( nRecv == INVALID_CHAR )
  522. return FALSE;
  523. pPacket->socks4aRep.byRep = (BYTE)nRecv;
  525. wData = 0;
  526. for( n = 0; n < 2; n++)
  527. {
  528. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  529. if( nRecv == INVALID_CHAR )
  530. return FALSE;
  532. wData |= ( ((WORD)nRecv) << (8*(n)) );
  533. }
  534. pPacket->socks4aRep.wDestPort = ntohs(wData);
  536. dwData = 0;
  537. for( n = 0; n < 4; n++)
  538. {
  539. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  540. if( nRecv == INVALID_CHAR )
  541. return FALSE;
  543. dwData |= ( ((DWORD)nRecv)<<(8*(n)) );
  544. }
  545. pPacket->socks4aRep.dwDestIP = ntohl(dwData);
  546. break;
  547. case PACKET_SOCKS5AUTHREP:
  548. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  549. if( nRecv == INVALID_CHAR )
  550. return FALSE;
  551. pPacket->socks5AuthRep.byVer = (BYTE)nRecv;
  553. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  554. if( nRecv == INVALID_CHAR )
  555. return FALSE;
  557. pPacket->socks5AuthRep.byAuth = (BYTE)nRecv;
  558. break;
  559. case PACKET_SOCKS5AUTHPASSWDREP:
  560. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  561. if( nRecv == INVALID_CHAR )
  562. return FALSE;
  563. pPacket->socks5AuthPasswdRep.byVer = (BYTE)nRecv;
  565. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  566. if( nRecv == INVALID_CHAR )
  567. return FALSE;
  569. pPacket->socks5AuthPasswdRep.byStatus = (BYTE)nRecv;
  570. break;
  571. case PACKET_SOCKS5REP:
  572. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  573. if( nRecv == INVALID_CHAR )
  574. return FALSE;
  575. pPacket->socks5Rep.byVer = (BYTE)nRecv;
  577. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  578. if( nRecv == INVALID_CHAR )
  579. return FALSE;
  580. pPacket->socks5Rep.byRep = (BYTE)nRecv;
  582. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  583. if( nRecv == INVALID_CHAR )
  584. return FALSE;
  585. pPacket->socks5Rep.byRsv = (BYTE)nRecv;
  587. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  588. if( nRecv == INVALID_CHAR )
  589. return FALSE;
  590. pPacket->socks5Rep.byAtyp = (BYTE)nRecv;
  592. if( pPacket->socks5Rep.byRep != REP5_SUCCESS )
  593. {
  594. pPacket->socks5Rep.pszBindAddr = NULL;
  595. pPacket->socks5Rep.wBindPort   = WORD_NULL;
  596. break;
  597. }
  599. switch( pPacket->socks5Rep.byAtyp )
  600. {
  601. case ATYP_IPV4ADDR:
  602. pszData = (char *)malloc(5);
  603. if(pszData == NULL)
  604. return FALSE;
  605. for( n = 0; n < 4; n++)
  606. {
  607. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  608. if( nRecv == INVALID_CHAR )
  609. return FALSE;
  611. pszData[n] = (char)nRecv;
  612. }
  613. pszData[n] = '';
  614. pPacket->socks5Rep.pszBindAddr = pszData;
  615. break;
  616. case ATYP_IPV6ADDR:
  618. pszData = (char *)malloc(17);
  619. if(pszData == NULL)
  620. return FALSE;
  621. for( n = 0; n < 16; n++)
  622. {
  623. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  624. if( nRecv == INVALID_CHAR )
  625. return FALSE;
  627. pszData[n] = (char)nRecv;
  628. }
  629. pszData[n] = '';
  630. pPacket->socks5Rep.pszBindAddr = pszData;
  631. break;
  632. case ATYP_HOSTNAME:
  633. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  634. if( nRecv == INVALID_CHAR )
  635. return FALSE;
  637. nCount = nRecv;
  638. pszData = (char *)malloc(nCount+1 );
  639. if(pszData == NULL)
  640. return FALSE;
  641. for( n = 0; n < nCount; n++)
  642. {
  643. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  644. if( nRecv == INVALID_CHAR )
  645. return FALSE;
  647. pszData[n] = (char)nRecv;
  648. }
  649. pszData[n] = '';
  650. pPacket->socks5Rep.pszBindAddr = pszData;
  651. break;
  652. default:
  653. pPacket->socks5Rep.pszBindAddr = NULL;
  654. break;
  655. }
  657. wData = 0;
  658. for( n = 0; n < 2; n++)
  659. {
  660. nRecv = m_pBufSocket->BSDGetChar(dwTimeout);
  661. if( nRecv == INVALID_CHAR )
  662. return FALSE;
  664. wData |= ( ((WORD)nRecv) << (8*n) );
  665. }
  666. pPacket->socks5Rep.wBindPort = ntohs(wData);
  667. break;
  668. default:
  669. return FALSE;
  670. break;
  671. }
  673. return TRUE;
  674. }
  677. ////////////////////////////////////////////////////////////////////////////////
  678. // 函数名：BOOL RecvPacket(
  679. // int nPacketType,
  680. // DWORD dwTimeout /*= SOP_RECV_TIMEOUT*/ ) 
  681. // 用  途：接收并解包
  682. // 对全局变量的影响：无
  683. // 参  数：
  684. // nPacketType : 包类型
  685. // dwTimeout   : 接收超时(缺省为SOP_RECV_TIMEOUT)
  686. // 返回值：BOOL
  687. ////////////////////////////////////////////////////////////////////////////////
  688. BOOL CSocksPacket::RecvPacket( int nPacketType,DWORD dwTimeout /*= SOP_RECV_TIMEOUT*/)
  689. {
  690. ZeroMemory( &m_stRepPacket,sizeof(SOCKSREPPACKET) );
  691. return RecvPacket( &m_stRepPacket,nPacketType,dwTimeout );
  692. }
  694. ////////////////////////////////////////////////////////////////////////////////
  695. // 函数名：BOOL IsSocksOK()
  696. // 用  途：检查SOCKS的返回值是否正确
  697. // 对全局变量的影响：无
  698. // 参  数：无
  699. // 返回值：BOOL
  700. ////////////////////////////////////////////////////////////////////////////////
  701. BOOL CSocksPacket::IsSocksOK( )
  702. {
  703. return IsSocksOK(&m_stRepPacket,m_nRepPacketType);
  704. }