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

客户端示例：

[cpp] view plain copy







#include "Net.h"

#include "../p2pInfo.h"

int main()

{

CUdp udp;

if (0!=udp.Open(16888))

{

printf("client udp open failed \n");

return -1;

}

P2P_CLIENT_INFO clientInfo;

memset(&clientInfo,0,sizeof(P2P_CLIENT_INFO));

clientInfo.nClientPort=16888;

clientInfo.nID=0;

sprintf(clientInfo.sClientIP,"%s","10.10.2.161");

P2P_CMD_HEAD cmdHead;

cmdHead.nCmd=P2P_CMD_REGISTER_REQ;

cmdHead.nPayloadLen=sizeof(P2P_CLIENT_INFO);

memcpy(cmdHead.sPayLoad,&clientInfo,sizeof(P2P_CLIENT_INFO));

long lDestIP=0;

CBSocket::ConvertStringToIP((int*)&lDestIP,"114.247.165.37");

int nRet=0;

nRet=udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN+cmdHead.nPayloadLen,lDestIP,P2P_SERVER_PORT);

printf("send register len %d \n",nRet);

nRet=udp.Recv();

if (nRet>0)

{

printf("recv data \n");

}

else

{

printf("recv P2P_CMD_REGISTER_ACK time out \n");

return 0;

}

P2P_CLIENT_INFO remoteClientInfo;

memset(&remoteClientInfo,0,sizeof(P2P_CLIENT_INFO));

P2P_CMD_HEAD recvHead;

memset(&recvHead,0,sizeof(P2P_CMD_HEAD));

while (1)

{

cmdHead.nCmd=P2P_CMD_COMUNICATION_REQ;

cmdHead.nValue=1; // remote id

cmdHead.nPayloadLen=sizeof(P2P_CLIENT_INFO);

nRet=udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN+cmdHead.nPayloadLen,lDestIP,P2P_SERVER_PORT);

printf("send commuication len %d \n",nRet);

nRet=udp.Recv();

if (nRet>0)

{

memcpy(&recvHead,udp.GetBuffer(),nRet);

if (0==recvHead.nValue)

{

memcpy(&remoteClientInfo,recvHead.sPayLoad,sizeof(P2P_CLIENT_INFO));

printf("recv P2P_CMD_COMUNICATION_ACK remote ip:%s port:%d\n",

remoteClientInfo.sClientPublicIP,remoteClientInfo.nClientPublicPort);

break;

}

usleep(100*1000);

}

else

{

printf("recv P2P_CMD_COMUNICATION_ACK time out \n");

}

}

// send udp hole to remote ip

CBSocket::ConvertStringToIP((int*)&lDestIP,remoteClientInfo.sClientPublicIP);

cmdHead.nCmd=P2P_CMD_UDP_HOLE_REQ;

cmdHead.nPayloadLen=0;

udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN,lDestIP,remoteClientInfo.nClientPublicPort);

usleep(1000*1000);

udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN,lDestIP,remoteClientInfo.nClientPublicPort);

usleep(1000*1000);

udp.Send((unsigned char *)&cmdHead,CMD_HEAD_LEN,lDestIP,remoteClientInfo.nClientPublicPort);

nRet=udp.Recv();

if (nRet>0)

{

memcpy(&recvHead,udp.GetBuffer(),nRet);

if (recvHead.nCmd==P2P_CMD_UDP_HOLE_ACK || recvHead.nCmd==P2P_CMD_UDP_HOLE_REQ)

{

printf("recv P2P_CMD_UDP_HOLE udp hole success\n");

}

}

else

{

printf("P2P_CMD_UDP_HOLE recv out \n");

return 0;

}

return 0;

}

服务器端示例:

[cpp] view plain copy







#include "Net.h"

#include "../p2pInfo.h"

#define MAX_CLIENT_COUNT 16

P2P_CLIENT_INFO* pClientList[MAX_CLIENT_COUNT];

void AddClient(P2P_CLIENT_INFO* pClientInfo)

{

for(int i=0;i<MAX_CLIENT_COUNT;i++)

{

if (!pClientList[i])

{

P2P_CLIENT_INFO* pClient=new P2P_CLIENT_INFO;

memcpy(pClient,pClientInfo,sizeof(P2P_CLIENT_INFO));

pClientList[i]=pClient;

return;

}

}

}

P2P_CLIENT_INFO* FindClient(int nID)

{

for(int i=0;i<MAX_CLIENT_COUNT;i++)

{

if (pClientList[i])

{

if (nID==pClientList[i]->nID)

{

return pClientList[i];

}

}

}

return NULL;

}

P2P_CLIENT_INFO* FindClient(long ip, unsigned short nPort)

{

for(int i=0;i<MAX_CLIENT_COUNT;i++)

{

if (pClientList[i])

{

int nIp=0;

CBSocket::ConvertStringToIP(&nIp,pClientList[i]->sClientPublicIP);

if (nPort==pClientList[i]->nClientPublicPort && ip==nIp)

{

return pClientList[i];

}

}

}

return NULL;

}

int main()

{

CUdp udp;

udp.Open(P2P_SERVER_PORT);

int nRet=0;

P2P_CMD_HEAD* pCmdHead=new P2P_CMD_HEAD;

P2P_CMD_HEAD* pSendCmdHead=new P2P_CMD_HEAD;

memset(pCmdHead,0,sizeof(P2P_CMD_HEAD));

memset(pSendCmdHead,0,sizeof(P2P_CMD_HEAD));

long lRemoteIP=0;

unsigned short nRemotePort=0;

int i=0;

for (i = 0; i <16; i++)

{

pClientList[i]=NULL;

}

while (1)

{

nRet=udp.Recv(lRemoteIP,nRemotePort);

if (nRet>0)

{

memcpy(pCmdHead,udp.GetBuffer(),nRet);

switch (pCmdHead->nCmd)

{

case P2P_CMD_REGISTER_REQ:

{

printf("recv register req \n");

P2P_CLIENT_INFO* pClient=(P2P_CLIENT_INFO*)pCmdHead->sPayLoad;

CBSocket::ConvertIPToString(lRemoteIP,pClient->sClientPublicIP);

pClient->nClientPublicPort=nRemotePort;

AddClient(pClient);

printf("recv command P2P_CMD_REGISTER_REQ from ip:%s port:%d\n",pClient->sClientPublicIP,pClient->nClientPublicPort);

pSendCmdHead->nCmd=P2P_CMD_REGISTER_ACK;

pSendCmdHead->nPayloadLen=sizeof(P2P_CLIENT_INFO);

memcpy(pSendCmdHead->sPayLoad,pClient,sizeof(P2P_CLIENT_INFO));

udp.Send((unsigned char *)pSendCmdHead,CMD_HEAD_LEN+pSendCmdHead->nPayloadLen,lRemoteIP,nRemotePort);

}

break;

case P2P_CMD_COMUNICATION_REQ:

{

printf("recv command P2P_CMD_COMUNICATION_REQ\n");

P2P_CLIENT_INFO* pTargetClientInfo=FindClient(pCmdHead->nValue);

if (!pTargetClientInfo)

{

printf("not find client info id:%d\n",pCmdHead->nValue);

continue;

}

pSendCmdHead->nCmd=P2P_CMD_COMUNICATION_ACK;

pSendCmdHead->nPayloadLen=sizeof(P2P_CLIENT_INFO);

memcpy(pSendCmdHead->sPayLoad,pTargetClientInfo,sizeof(P2P_CLIENT_INFO));

udp.Send((unsigned char *)pSendCmdHead,CMD_HEAD_LEN+pSendCmdHead->nPayloadLen,lRemoteIP,nRemotePort);

// send to remote client

int nIP=0;

P2P_CLIENT_INFO* pSelfClientInfo=FindClient(lRemoteIP,nRemotePort);

if (pSelfClientInfo)

{

printf("find self client info ip:%s\n",pSelfClientInfo->sClientPublicIP);

memcpy(pSendCmdHead->sPayLoad,pSelfClientInfo,sizeof(P2P_CLIENT_INFO));

CBSocket::ConvertStringToIP(&nIP,pTargetClientInfo->sClientPublicIP);

udp.Send((unsigned char *)pSendCmdHead,CMD_HEAD_LEN+pSendCmdHead->nPayloadLen,nIP,pTargetClientInfo->nClientPublicPort);

}

else

{

printf("not find self client info");

}

}

break;

default:

break;

}

}

else

{

printf("udp recv time out \n");

}

}

return 0;

}

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