UDP 打洞示例 包含 服务器 客户端

客户端示例：

1. #include "Net.h"
2. #include "../p2pInfo.h"
3. int main()
4. {
5. CUdp  udp;
6. if (0!=udp.Open(16888))
7. {
8. printf("client udp open failed \n");
9. return -1;
10. }
11. P2P_CLIENT_INFO clientInfo;
12. memset(&clientInfo,0,sizeof(P2P_CLIENT_INFO));
13. clientInfo.nClientPort=16888;
14. clientInfo.nID=0;
15. sprintf(clientInfo.sClientIP,"%s","10.10.2.161");
16. P2P_CMD_HEAD cmdHead;
17. cmdHead.nCmd=P2P_CMD_REGISTER_REQ;
18. cmdHead.nPayloadLen=sizeof(P2P_CLIENT_INFO);
19. memcpy(cmdHead.sPayLoad,&clientInfo,sizeof(P2P_CLIENT_INFO));
20. long lDestIP=0;
21. CBSocket::ConvertStringToIP((int*)&lDestIP,"114.247.165.37");
22. int nRet=0;
23. nRet=udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN+cmdHead.nPayloadLen,lDestIP,P2P_SERVER_PORT);
24. printf("send register len %d \n",nRet);
25. nRet=udp.Recv();
26. if (nRet>0)
27. {
28. printf("recv data \n");
29. }
30. else
31. {
32. printf("recv P2P_CMD_REGISTER_ACK time out \n");
33. return 0;
34. }
35. P2P_CLIENT_INFO remoteClientInfo;
36. memset(&remoteClientInfo,0,sizeof(P2P_CLIENT_INFO));
37. P2P_CMD_HEAD recvHead;
38. memset(&recvHead,0,sizeof(P2P_CMD_HEAD));
39. while (1)
40. {
41. cmdHead.nCmd=P2P_CMD_COMUNICATION_REQ;
42. cmdHead.nValue=1;    // remote id
43. cmdHead.nPayloadLen=sizeof(P2P_CLIENT_INFO);
44. nRet=udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN+cmdHead.nPayloadLen,lDestIP,P2P_SERVER_PORT);
45. printf("send commuication len %d \n",nRet);
46. nRet=udp.Recv();
47. if (nRet>0)
48. {
49. memcpy(&recvHead,udp.GetBuffer(),nRet);
50. if (0==recvHead.nValue)
51. {
52. memcpy(&remoteClientInfo,recvHead.sPayLoad,sizeof(P2P_CLIENT_INFO));
53. printf("recv P2P_CMD_COMUNICATION_ACK remote ip:%s port:%d\n",
54. remoteClientInfo.sClientPublicIP,remoteClientInfo.nClientPublicPort);
55. break;
56. }
57. usleep(100*1000);
58. }
59. else
60. {
61. printf("recv P2P_CMD_COMUNICATION_ACK time out \n");
62. }
63. }
64. // send udp hole to remote ip
65. CBSocket::ConvertStringToIP((int*)&lDestIP,remoteClientInfo.sClientPublicIP);
66. cmdHead.nCmd=P2P_CMD_UDP_HOLE_REQ;
67. cmdHead.nPayloadLen=0;
68. udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN,lDestIP,remoteClientInfo.nClientPublicPort);
69. usleep(1000*1000);
70. udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN,lDestIP,remoteClientInfo.nClientPublicPort);
71. usleep(1000*1000);
72. udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN,lDestIP,remoteClientInfo.nClientPublicPort);
73. nRet=udp.Recv();
74. if (nRet>0)
75. {
76. memcpy(&recvHead,udp.GetBuffer(),nRet);
77. if (recvHead.nCmd==P2P_CMD_UDP_HOLE_ACK || recvHead.nCmd==P2P_CMD_UDP_HOLE_REQ)
78. {
79. printf("recv P2P_CMD_UDP_HOLE  udp hole success\n");
80. }
81. }
82. else
83. {
84. printf("P2P_CMD_UDP_HOLE recv out \n");
85. return 0;
86. }
87. return 0;
88. }

服务器端示例:

1. #include "Net.h"
2. #include "../p2pInfo.h"
3. #define MAX_CLIENT_COUNT 16
4. P2P_CLIENT_INFO* pClientList[MAX_CLIENT_COUNT];
5. void AddClient(P2P_CLIENT_INFO* pClientInfo)
6. {
7. for(int i=0;i<MAX_CLIENT_COUNT;i++)
8. {
9. if (!pClientList[i])
10. {
11. P2P_CLIENT_INFO* pClient=new P2P_CLIENT_INFO;
12. memcpy(pClient,pClientInfo,sizeof(P2P_CLIENT_INFO));
13. pClientList[i]=pClient;
14. return;
15. }
16. }
17. }
18. P2P_CLIENT_INFO* FindClient(int nID)
19. {
20. for(int i=0;i<MAX_CLIENT_COUNT;i++)
21. {
22. if (pClientList[i])
23. {
24. if (nID==pClientList[i]->nID)
25. {
26. return pClientList[i];
27. }
28. }
29. }
30. return NULL;
31. }
32. P2P_CLIENT_INFO* FindClient(long ip, unsigned short nPort)
33. {
34. for(int i=0;i<MAX_CLIENT_COUNT;i++)
35. {
36. if (pClientList[i])
37. {
38. int nIp=0;
39. CBSocket::ConvertStringToIP(&nIp,pClientList[i]->sClientPublicIP);
40. if (nPort==pClientList[i]->nClientPublicPort && ip==nIp)
41. {
42. return pClientList[i];
43. }
44. }
45. }
46. return NULL;
47. }
48. int main()
49. {
50. CUdp udp;
51. udp.Open(P2P_SERVER_PORT);
52. int nRet=0;
53. P2P_CMD_HEAD* pCmdHead=new P2P_CMD_HEAD;
54. P2P_CMD_HEAD* pSendCmdHead=new P2P_CMD_HEAD;
55. memset(pCmdHead,0,sizeof(P2P_CMD_HEAD));
56. memset(pSendCmdHead,0,sizeof(P2P_CMD_HEAD));
57. long lRemoteIP=0;
58. unsigned short nRemotePort=0;
59. int i=0;
60. for (i = 0; i <16; i++)
61. {
62. pClientList[i]=NULL;
63. }
64. while (1)
65. {
66. nRet=udp.Recv(lRemoteIP,nRemotePort);
67. if (nRet>0)
68. {
69. memcpy(pCmdHead,udp.GetBuffer(),nRet);
70. switch (pCmdHead->nCmd)
71. {
72. case P2P_CMD_REGISTER_REQ:
73. {
74. printf("recv register req \n");
75. P2P_CLIENT_INFO* pClient=(P2P_CLIENT_INFO*)pCmdHead->sPayLoad;
76. CBSocket::ConvertIPToString(lRemoteIP,pClient->sClientPublicIP);
77. pClient->nClientPublicPort=nRemotePort;
78. AddClient(pClient);
79. printf("recv command P2P_CMD_REGISTER_REQ from ip:%s port:%d\n",pClient->sClientPublicIP,pClient->nClientPublicPort);
80. pSendCmdHead->nCmd=P2P_CMD_REGISTER_ACK;
81. pSendCmdHead->nPayloadLen=sizeof(P2P_CLIENT_INFO);
82. memcpy(pSendCmdHead->sPayLoad,pClient,sizeof(P2P_CLIENT_INFO));
83. udp.Send((unsigned char *)pSendCmdHead,CMD_HEAD_LEN+pSendCmdHead->nPayloadLen,lRemoteIP,nRemotePort);
84. }
85. break;
86. case P2P_CMD_COMUNICATION_REQ:
87. {
88. printf("recv command P2P_CMD_COMUNICATION_REQ\n");
89. P2P_CLIENT_INFO* pTargetClientInfo=FindClient(pCmdHead->nValue);
90. if (!pTargetClientInfo)
91. {
92. printf("not find client info id:%d\n",pCmdHead->nValue);
93. continue;
94. }
95. pSendCmdHead->nCmd=P2P_CMD_COMUNICATION_ACK;
96. pSendCmdHead->nPayloadLen=sizeof(P2P_CLIENT_INFO);
97. memcpy(pSendCmdHead->sPayLoad,pTargetClientInfo,sizeof(P2P_CLIENT_INFO));
98. udp.Send((unsigned char *)pSendCmdHead,CMD_HEAD_LEN+pSendCmdHead->nPayloadLen,lRemoteIP,nRemotePort);
99. // send to remote client
100. int nIP=0;
101. P2P_CLIENT_INFO* pSelfClientInfo=FindClient(lRemoteIP,nRemotePort);
102. if (pSelfClientInfo)
103. {
104. printf("find self client info ip:%s\n",pSelfClientInfo->sClientPublicIP);
105. memcpy(pSendCmdHead->sPayLoad,pSelfClientInfo,sizeof(P2P_CLIENT_INFO));
106. CBSocket::ConvertStringToIP(&nIP,pTargetClientInfo->sClientPublicIP);
107. udp.Send((unsigned char *)pSendCmdHead,CMD_HEAD_LEN+pSendCmdHead->nPayloadLen,nIP,pTargetClientInfo->nClientPublicPort);
108. }
109. else
110. {
111. printf("not find self client info");
112. }
113. }
114. break;
115. default:
116. break;
117. }
118. }
119. else
120. {
121. printf("udp recv time out \n");
122. }
123. }
124. return 0;
125. }

示例下载地址:    http://download.csdn.net/detail/mtour/8119489