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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
P2PClient.cpp
资源名称：P2PCSHARP.rar　[点击查看]
上传用户：zhangjjyh
上传日期：2021-11-11
资源大小：1251k
文件大小：11k
源码类别：

P2P编程

开发平台：

Objective-C

P2PClient.cpp：源码内容
  1. //////////////////////////////////////////////////////
  2. // p2pclient.cpp文件
  5. #include <winsock2.h>
  6. #include <stdio.h>
  7. #include "p2pclient.h"
  8. #pragma comment(lib, "WS2_32") // 链接到WS2_32.lib
  10. CP2PClient::CP2PClient()
  11. {
  12. m_bLogin = FALSE;
  13. m_hThread = NULL;
  14. m_s = INVALID_SOCKET;
  16. memset(&m_ol, 0, sizeof(m_ol));
  17. m_ol.hEvent = ::WSACreateEvent();
  19. ::InitializeCriticalSection(&m_PeerListLock);
  21. // 初始化WS2_32.dll
  22. WSADATA wsaData;
  23. WORD sockVersion = MAKEWORD(2, 2);
  24. ::WSAStartup(sockVersion, &wsaData);
  25. }
  27. CP2PClient::~CP2PClient()
  28. {
  29. Logout();
  31. // 通知接收线程退出
  32. if(m_hThread != NULL)
  33. {
  34. m_bThreadExit = TRUE;
  35. ::WSASetEvent(m_ol.hEvent);
  36. ::WaitForSingleObject(m_hThread, 300);
  37. ::CloseHandle(m_hThread);
  38. }
  40. if(m_s != INVALID_SOCKET)
  41. ::closesocket(m_s);
  43. ::WSACloseEvent(m_ol.hEvent);
  45. ::DeleteCriticalSection(&m_PeerListLock);
  46. ::WSACleanup();
  47. }
  49. BOOL CP2PClient::Init(USHORT usLocalPort)
  50. {
  51. if(m_s != INVALID_SOCKET)
  52. return FALSE;
  54. // 创建用于P2P通信的UDP套节字，进行绑定
  55. m_s = ::WSASocket(AF_INET, 
  56. SOCK_DGRAM , IPPROTO_UDP, NULL, 0, WSA_FLAG_OVERLAPPED);
  58. sockaddr_in localAddr = { 0 };
  59. localAddr.sin_family = AF_INET;
  60. localAddr.sin_port = htons(usLocalPort);
  61. localAddr.sin_addr.S_un.S_addr = INADDR_ANY;
  62. if(::bind(m_s, (LPSOCKADDR)&localAddr, sizeof(localAddr)) == SOCKET_ERROR)
  63. {
  64. ::closesocket(m_s);
  65. m_s = INVALID_SOCKET;
  66. return FALSE;
  67. }
  68. if(usLocalPort == 0)
  69. {
  70. int nLen = sizeof(localAddr);
  71. ::getsockname(m_s, (sockaddr*)&localAddr, &nLen);
  72. usLocalPort = ntohs(localAddr.sin_port);
  73. }
  75. // 获取本地机器的IP地址，得到当前用户的私有终端
  76. char szHost[256];
  77. ::gethostname(szHost, 256);
  78. hostent *pHost = ::gethostbyname(szHost);
  80. memset(&m_LocalPeer, 0, sizeof(m_LocalPeer));
  81. for(int i=0; i<MAX_ADDR_NUMBER - 1; i++)
  82. {
  83. char *p = pHost->h_addr_list[i];
  84. if(p == NULL)
  85. break;
  87. memcpy(&m_LocalPeer.addr[i].dwIp, &p, pHost->h_length);
  88. m_LocalPeer.addr[i].nPort = usLocalPort;
  89. m_LocalPeer.AddrNum ++;
  90. }
  92. // 创建接收服务线程
  93. m_bThreadExit = FALSE;
  94. m_hThread = ::CreateThread(NULL, 0, RecvThreadProc, this, 0, NULL);
  96. return TRUE;
  97. }
  99. BOOL CP2PClient::Login(char *pszUserName, char *pszServerIp)
  100. {
  101. if(m_bLogin || strlen(pszUserName) > MAX_USERNAME - 1)
  102. return FALSE;
  104. // 保存参数
  105. m_dwServerIp = ::inet_addr(pszServerIp);
  106. strncpy(m_LocalPeer.szUserName, pszUserName, strlen(pszUserName));
  109. // 服务器名称
  110. sockaddr_in serverAddr = { 0 };
  111. serverAddr.sin_family = AF_INET;
  112. serverAddr.sin_addr.S_un.S_addr = m_dwServerIp; 
  113. serverAddr.sin_port = htons(SERVER_PORT);
  115. // 向服务发送本用户信息
  116. CP2PMessage logMsg;
  117. logMsg.nMessageType = USERLOGIN;
  118. memcpy(&logMsg.peer, &m_LocalPeer, sizeof(PEER_INFO)); 
  120. for(int i=0; i<MAX_TRY_NUMBER; i++)
  121. {
  122. ::sendto(m_s, (char*)&logMsg, sizeof(logMsg), 0, (sockaddr*)&serverAddr, sizeof(serverAddr));
  123. for(int j=0; j<10; j++)
  124. {
  125. if(m_bLogin)
  126. return TRUE;
  127. ::Sleep(300);
  128. }
  129. }
  130. return FALSE;
  131. }
  133. void CP2PClient::Logout()
  134. {
  135. if(m_bLogin)
  136. {
  137. // 告诉服务器，我们要离开了
  138. CP2PMessage logMsg;
  139. logMsg.nMessageType = USERLOGOUT;
  140. memcpy(&logMsg.peer, &m_LocalPeer, sizeof(PEER_INFO)); 
  142. sockaddr_in serverAddr = { 0 };
  143. serverAddr.sin_family = AF_INET;
  144. serverAddr.sin_addr.S_un.S_addr = m_dwServerIp; 
  145. serverAddr.sin_port = htons(SERVER_PORT);
  147. ::sendto(m_s, (char*)&logMsg, sizeof(logMsg), 0, (sockaddr*)&serverAddr, sizeof(serverAddr));
  148. m_bLogin = FALSE;
  149. }
  150. }
  152. BOOL CP2PClient::SendText(char *pszUserName, char *pszText, int nTextLen)
  153. {
  154. if(!m_bLogin || strlen(pszUserName) > MAX_USERNAME - 1 
  155. || nTextLen > MAX_PACKET_SIZE - sizeof(CP2PMessage))
  156. return FALSE;
  158. // 构建封包
  159. char sendBuf[MAX_PACKET_SIZE];
  160. CP2PMessage *pMsg = (CP2PMessage*)sendBuf;
  161. pMsg->nMessageType = P2PMESSAGE;
  162. memcpy(&pMsg->peer, &m_LocalPeer, sizeof(m_LocalPeer));
  163. memcpy((pMsg + 1), pszText, nTextLen);
  165. m_bMessageACK = FALSE;
  166. for(int i=0; i<MAX_TRY_NUMBER; i++)
  167. {
  168. PEER_INFO *pInfo = m_PeerList.GetAPeer(pszUserName);
  169. if(pInfo == NULL)
  170. return FALSE;
  172. // 如果对方P2P地址不为0，就试图以它为目的地址发送数据，
  173. // 如果发送失败，则认为此P2P地址无效
  174. if(pInfo->p2pAddr.dwIp != 0) 
  175. {
  176. sockaddr_in peerAddr = { 0 };
  177. peerAddr.sin_family = AF_INET;
  178. peerAddr.sin_addr.S_un.S_addr = pInfo->p2pAddr.dwIp;
  179. peerAddr.sin_port = htons(pInfo->p2pAddr.nPort);
  181. ::sendto(m_s, sendBuf, 
  182. nTextLen + sizeof(CP2PMessage), 0, (sockaddr*)&peerAddr, sizeof(peerAddr));
  184. for(int j=0; j<10; j++)
  185. {
  186. if( m_bMessageACK)
  187. return TRUE;
  188. ::Sleep(300);
  189. }
  190. }
  192. // 请求打洞，并且重新设置P2P地址
  193. pInfo->p2pAddr.dwIp = 0;
  195. // 构建封包
  196. char tmpBuf[sizeof(CP2PMessage) + MAX_USERNAME];
  197. CP2PMessage *p = (CP2PMessage *)tmpBuf;
  198. p->nMessageType = P2PCONNECT;
  199. memcpy(&p->peer, &m_LocalPeer, sizeof(m_LocalPeer));
  200. memcpy((char*)(p + 1), pszUserName, strlen(pszUserName) + 1);
  202. // 首先直接发向目标，
  203. sockaddr_in peerAddr = { 0 };
  204. peerAddr.sin_family = AF_INET;
  205. for(int j=0; j<pInfo->AddrNum; j++)
  206. {
  207. peerAddr.sin_addr.S_un.S_addr = pInfo->addr[j].dwIp;
  208. peerAddr.sin_port = htons(pInfo->addr[j].nPort);
  209. ::sendto(m_s, tmpBuf, sizeof(CP2PMessage), 0, (sockaddr*)&peerAddr, sizeof(peerAddr));
  210. }
  212. // 然后通过服务器转发，请求对方向自己打洞
  213. sockaddr_in serverAddr = { 0 };
  214. serverAddr.sin_family = AF_INET;
  215. serverAddr.sin_addr.S_un.S_addr = m_dwServerIp; 
  216. serverAddr.sin_port = htons(SERVER_PORT);
  217. ::sendto(m_s, tmpBuf, 
  218. sizeof(CP2PMessage) + MAX_USERNAME, 0, (sockaddr*)&serverAddr, sizeof(serverAddr));
  220. // 等待对方的P2PCONNECTACK消息
  221. for(j=0; j<10; j++)
  222. {
  223. if(pInfo->p2pAddr.dwIp != 0)
  224. break;
  225. ::Sleep(300);
  226. }
  227. }
  228. return 0;
  229. }
  231. BOOL CP2PClient::GetUserList()
  232. {
  233. // 服务器地址
  234. sockaddr_in serverAddr = { 0 };
  235. serverAddr.sin_family = AF_INET;
  236. serverAddr.sin_addr.S_un.S_addr = m_dwServerIp; 
  237. serverAddr.sin_port = htons(SERVER_PORT);
  238. // 构建封包
  239. CP2PMessage msgList;
  240. msgList.nMessageType = GETUSERLIST;
  241. memcpy(&msgList.peer, &m_LocalPeer, sizeof(m_LocalPeer));
  242. // 删除所有节点
  243. ::EnterCriticalSection(&m_PeerListLock);
  244. m_PeerList.DeleteAllPeers();
  245. ::LeaveCriticalSection(&m_PeerListLock);
  247. // 发送GETUSERLIST请求，等待列表发送完成
  248. m_bUserlistCmp = FALSE;
  249. int nUserCount = 0;
  250. for(int i=0; i<MAX_TRY_NUMBER; i++)
  251. {
  252. ::sendto(m_s, (char*)&msgList, 
  253. sizeof(msgList), 0, (sockaddr*)&serverAddr, sizeof(serverAddr));
  254. do
  255. {
  256. nUserCount = m_PeerList.m_nCurrentSize;
  257. for(int j=0; j<10; j++)
  258. {
  259. if(m_bUserlistCmp)
  260. return TRUE;
  261. ::Sleep(300);
  262. }
  263. }while(m_PeerList.m_nCurrentSize > nUserCount);
  264. }
  265. return FALSE;
  266. }
  268. DWORD WINAPI CP2PClient::RecvThreadProc(LPVOID lpParam)
  269. {
  270. CP2PClient *pThis = (CP2PClient *)lpParam;
  271. char buff[MAX_PACKET_SIZE];
  272. sockaddr_in remoteAddr;
  273. int nAddrLen = sizeof(remoteAddr);
  274. WSABUF wsaBuf;
  275. wsaBuf.buf = buff;
  276. wsaBuf.len = MAX_PACKET_SIZE;
  278. // 接收处理到来的消息
  279. while(TRUE)
  280. {
  281. DWORD dwRecv, dwFlags = 0;
  282. int nRet = ::WSARecvFrom(pThis->m_s, &wsaBuf, 
  283. 1, &dwRecv, &dwFlags, (sockaddr*)&remoteAddr, &nAddrLen, &pThis->m_ol, NULL);
  284. if(nRet == SOCKET_ERROR && ::WSAGetLastError() == WSA_IO_PENDING)
  285. {
  286. ::WSAGetOverlappedResult(pThis->m_s, &pThis->m_ol, &dwRecv, TRUE, &dwFlags);
  287. }
  288. // 首先查看是否要退出
  289. if(pThis->m_bThreadExit)
  290. break;
  291. // 调用HandleIO函数来处理这个消息
  292. pThis->HandleIO(buff, dwRecv, (sockaddr *)&remoteAddr, nAddrLen);
  293. }
  294. return 0;
  295. }
  297. void CP2PClient::HandleIO(char *pBuf, int nBufLen, sockaddr *addr, int nAddrLen)
  298. {
  299. CP2PMessage *pMsg = (CP2PMessage*)pBuf;
  300. if(nBufLen < sizeof(CP2PMessage))
  301. return;
  303. switch(pMsg->nMessageType)
  304. {
  305. case USERLOGACK: // 接收到服务器发来的登陆确认
  306. {
  307. memcpy(&m_LocalPeer, &pMsg->peer, sizeof(PEER_INFO));
  308. m_bLogin = TRUE;
  309. }
  310. break;
  311. case P2PMESSAGE: // 有一个节点向我们发送消息
  312. {
  313. int nDataLen = nBufLen - sizeof(CP2PMessage);
  314. if(nDataLen > 0)
  315. {
  316. // 发送确认消息
  317. CP2PMessage ackMsg;
  318. ackMsg.nMessageType = P2PMESSAGEACK;
  319. memcpy(&ackMsg.peer, &m_LocalPeer, sizeof(PEER_INFO));
  320. ::sendto(m_s, (char*)&ackMsg, sizeof(ackMsg), 0, addr, nAddrLen);
  322. OnRecv(pMsg->peer.szUserName, (char*)(pMsg + 1), nDataLen);
  323. }
  324. }
  325. break;
  326. case P2PMESSAGEACK: // 收到消息的应答
  327. {
  328. m_bMessageACK = TRUE;
  329. }
  330. break;
  332. case P2PCONNECT: // 一个节点请求建立P2P连接（打洞），可能是服务器发来的，也可能是其它节点发来的
  333. {
  334. CP2PMessage ackMsg;
  335. ackMsg.nMessageType = P2PCONNECTACK;
  336. memcpy(&ackMsg.peer, &m_LocalPeer, sizeof(PEER_INFO));
  338. if(((sockaddr_in*)addr)->sin_addr.S_un.S_addr != m_dwServerIp) // 节点发来的消息
  339. {
  340. ::EnterCriticalSection(&m_PeerListLock);
  341. PEER_INFO *pInfo = m_PeerList.GetAPeer(pMsg->peer.szUserName);
  342. if(pInfo != NULL)
  343. {
  344. if(pInfo->p2pAddr.dwIp == 0)
  345. {
  346. pInfo->p2pAddr.dwIp = ((sockaddr_in*)addr)->sin_addr.S_un.S_addr;
  347. pInfo->p2pAddr.nPort = ntohs(((sockaddr_in*)addr)->sin_port);
  349. printf(" Set P2P address for %s -> %s:%ld n", pInfo->szUserName, 
  350. ::inet_ntoa(((sockaddr_in*)addr)->sin_addr), ntohs(((sockaddr_in*)addr)->sin_port));
  351. }
  352. }
  353. ::LeaveCriticalSection(&m_PeerListLock);
  355. ::sendto(m_s, (char*)&ackMsg, sizeof(ackMsg), 0, addr, nAddrLen);
  356. }
  357. else // 服务器转发的消息
  358. {
  359. // 向节点的所有终端发送打洞消息
  360. sockaddr_in peerAddr = { 0 };
  361. peerAddr.sin_family = AF_INET;
  362. for(int i=0; i<pMsg->peer.AddrNum; i++)
  363. {
  364. peerAddr.sin_addr.S_un.S_addr = pMsg->peer.addr[i].dwIp;
  365. peerAddr.sin_port = htons(pMsg->peer.addr[i].nPort);
  366. ::sendto(m_s, (char*)&ackMsg, sizeof(ackMsg), 0, (sockaddr*)&peerAddr, sizeof(peerAddr));
  367. }
  368. }
  369. }
  370. break;
  372. case P2PCONNECTACK: // 接收到节点的打洞消息，在这里设置它的P2P通信地址
  373. {
  374. ::EnterCriticalSection(&m_PeerListLock);
  375. PEER_INFO *pInfo = m_PeerList.GetAPeer(pMsg->peer.szUserName);
  376. if(pInfo != NULL)
  377. {
  378. if(pInfo->p2pAddr.dwIp == 0)
  379. {
  380. pInfo->p2pAddr.dwIp = ((sockaddr_in*)addr)->sin_addr.S_un.S_addr;
  381. pInfo->p2pAddr.nPort = ntohs(((sockaddr_in*)addr)->sin_port);
  383. printf(" Set P2P address for %s -> %s:%ld n", pInfo->szUserName, 
  384. ::inet_ntoa(((sockaddr_in*)addr)->sin_addr), ntohs(((sockaddr_in*)addr)->sin_port));
  385. }
  387. }
  388. ::LeaveCriticalSection(&m_PeerListLock);
  390. }
  391. break;
  393. case USERACTIVEQUERY: // 服务器询问是否存活
  394. {
  395. CP2PMessage ackMsg;
  396. ackMsg.nMessageType = USERACTIVEQUERYACK;
  397. memcpy(&ackMsg.peer, &m_LocalPeer, sizeof(PEER_INFO));
  398. ::sendto(m_s, (char*)&ackMsg, sizeof(ackMsg), 0, addr, nAddrLen);
  399. }
  400. break;
  401. case GETUSERLIST: // 服务器发送的用户列表
  402. {
  403. // 首先清除此用户的P2P地址，再将用户信息保存到本地用户列表中
  404. pMsg->peer.p2pAddr.dwIp = 0;
  405. ::EnterCriticalSection(&m_PeerListLock);
  406. m_PeerList.AddAPeer(&pMsg->peer);
  407. ::LeaveCriticalSection(&m_PeerListLock);
  408. }
  409. break;
  410. case USERLISTCMP: // 用户列表传输结束
  411. {
  412. m_bUserlistCmp = TRUE;
  413. }
  414. break;
  415. }
  416. }