NAT網絡穿透

首先先介紹一些基本概念：
    NAT(Network Address Translators)，網絡地址轉換：網絡地址轉換是在IP地址日益缺少的狀況下產生的，它的主要目的就是爲了可以地址重用。NAT分爲兩大類，基本的NAT和NAPT(Network Address/Port Translator)。
    最開始NAT是運行在路由器上的一個功能模塊。
   
    最早提出的是基本的NAT，它的產生基於以下事實：一個私有網絡（域）中的節點中只有不多的節點須要與外網鏈接（呵呵，這是在上世紀90年代中期提出的）。那麼這個子網中其實只有少數的節點須要全球惟一的IP地址，其餘的節點的IP地址應該是能夠重用的。
    所以，基本的NAT實現的功能很簡單，在子網內使用一個保留的IP子網段，這些IP對外是不可見的。子網內只有少數一些IP地址能夠對應到真正全球惟一的 IP地址。若是這些節點須要訪問外部網絡，那麼基本NAT就負責將這個節點的子網內IP轉化爲一個全球惟一的IP而後發送出去。(基本的NAT會改變IP 包中的原IP地址，可是不會改變IP包中的端口)
    關於基本的NAT能夠參看RFC 1631
   
    另一種NAT叫作NAPT，從名稱上咱們也能夠看得出，NAPT不但會改變通過這個NAT設備的IP數據報的IP地址，還會改變IP數據報的 TCP/UDP端口。基本NAT的設備可能咱們見的很少（呵呵，我沒有見到過），NAPT纔是咱們真正討論的主角。                                Server S1                        
                                        18.181.0.31:1235                         
                                                 |
          ^  Session 1 (A-S1)   ^     | 
          |  18.181.0.31:1235    |      |  
          v 155.99.25.11:62000 v     |    
                                                 |
                                              NAT
                                           155.99.25.11
                                                 |
          ^  Session 1 (A-S1)  ^       | 
          |  18.181.0.31:1235  |        | 
          v   10.0.0.1:1234    v         | 
                                                 |
                                            Client A
                                           10.0.0.1:1234
    有一個私有網絡10.*.*.*，Client A是其中的一臺計算機，這個網絡的網關（一個NAT設備）的外網IP是155.99.25.11(應該還有一個內網的IP地址，好比 10.0.0.10)。若是Client A中的某個進程（這個進程建立了一個UDP Socket,這個Socket綁定1234端口）想訪問外網主機18.181.0.31的1235端口，那麼當數據包經過NAT時會發生什麼事情呢？
    首先NAT會改變這個數據包的原IP地址，改成155.99.25.11。接着NAT會爲這個傳輸建立一個Session（Session是一個抽象的概念，若是是TCP，也許Session是由一個SYN包開始，以一個FIN包結束。而UDP呢，以這個IP的這個端口的第一個UDP開始，結束呢，呵呵，也許是幾分鐘，也許是幾小時，這要看具體的實現了）而且給這個Session分配一個端口，好比62000，而後改變這個數據包的源端口爲62000。因此原本是（10.0.0.1:1234->18.181.0.31:1235）的數據包到了互聯網上變爲了（155.99.25.11:62000->18.181.0.31:1235）。
    一旦NAT建立了一個Session後，NAT會記住62000端口對應的是10.0.0.1的1234端口，之後從18.181.0.31發送到 62000端口的數據會被NAT自動的轉發到10.0.0.1上。（注意：這裏是說18.181.0.31發送到62000端口的數據會被轉發，其餘的 IP發送到這個端口的數據將被NAT拋棄）這樣Client A就與Server S1創建以了一個鏈接。

    呵呵，上面的基礎知識可能不少人都知道了，那麼下面是關鍵的部分了。
    看看下面的狀況：
    Server S1                                     Server S2
 18.181.0.31:1235                              138.76.29.7:1235
        |                                                       |
        |                                                       |
        +----------------------+----------------------+
                                        |
   ^  Session 1 (A-S1)  ^    |      ^  Session 2 (A-S2)  ^
   |  18.181.0.31:1235  |      |      |  138.76.29.7:1235  |
   v 155.99.25.11:62000 v   |      v 155.99.25.11:62000 v
                                       |
                                 Cone NAT
                               155.99.25.11
                                         |
   ^  Session 1 (A-S1)  ^      |      ^  Session 2 (A-S2)  ^
   |  18.181.0.31:1235  |       |      |  138.76.29.7:1235  |
   v   10.0.0.1:1234     v       |      v   10.0.0.1:1234    v
                                        |
                                   Client A
                                10.0.0.1:1234
    接上面的例子，若是Client A的原來那個Socket(綁定了1234端口的那個UDP Socket)又接着向另一個Server S2發送了一個UDP包，那麼這個UDP包在經過NAT時會怎麼樣呢？
    這時可能會有兩種狀況發生，一種是NAT再次建立一個Session，而且再次爲這個Session分配一個端口號（好比：62001）。另一種是 NAT再次建立一個Session，可是不會新分配一個端口號，而是用原來分配的端口號62000。前一種NAT叫作Symmetric NAT，後一種叫作Cone NAT。咱們指望咱們的NAT是第二種，呵呵，若是你的NAT恰好是第一種，那麼極可能會有不少P2P軟件失靈。（能夠慶幸的是，如今絕大多數的NAT屬於後者，即Cone NAT）
  
    好了，咱們看到，經過NAT,子網內的計算機向外連結是很容易的（NAT至關於透明的，子網內的和外網的計算機不用知道NAT的狀況）。
    可是若是外部的計算機想訪問子網內的計算機就比較困難了（而這正是P2P所須要的）。
    那麼咱們若是想從外部發送一個數據報給內網的計算機有什麼辦法呢？首先，咱們必須在內網的NAT上打上一個「洞」（也就是前面咱們說的在NAT上創建一個 Session），這個洞不能由外部來打，只能由內網內的主機來打。並且這個洞是有方向的，好比從內部某臺主機（好比：192.168.0.10）向外部的某個IP(好比：219.237.60.1)發送一個UDP包，那麼就在這個內網的NAT設備上打了一個方向爲219.237.60.1的「洞」，（這就是稱爲UDP Hole Punching的技術）之後219.237.60.1就能夠經過這個洞與內網的192.168.0.10聯繫了。（可是其餘的IP不能利用這個洞）。

呵呵，如今該輪到咱們的正題P2P了。有了上面的理論，實現兩個內網的主機通信就差最後一步了：那就是雞生蛋仍是蛋生雞的問題了，兩邊都沒法主動發出鏈接請求，誰也不知道誰的公網地址，那咱們如何來打這個洞呢？咱們須要一箇中間人來聯繫這兩個內網主機。
    如今咱們來看看一個P2P軟件的流程，如下爲例：

                    Server S （219.237.60.1）
                          |
                          |
   +----------------------+----------------------+
   |                                                     |
 NAT A (外網IP:202.187.45.3)             NAT B (外網IP:187.34.1.56)
   |         (內網IP:192.168.0.1)                |        (內網IP:192.168.0.1)
   |                                                     |
Client A  (192.168.0.20:4000)             Client B (192.168.0.10:40000)

    首先，Client A登陸服務器，NAT A爲此次的Session分配了一個端口60000，那麼Server S收到的Client A的地址是202.187.45.3:60000，這就是Client A的外網地址了。一樣，Client B登陸Server S，NAT B給這次Session分配的端口是40000，那麼Server S收到的B的地址是187.34.1.56:40000。
    此時，Client A與Client B均可以與Server S通訊了。若是Client A此時想直接發送信息給Client B，那麼他能夠從Server S那兒得到B的公網地址187.34.1.56:40000，是否是Client A向這個地址發送信息Client B就能收到了呢？答案是不行，由於若是這樣發送信息，NAT B會將這個信息丟棄（由於這樣的信息是不請自來的，爲了安全，大多數NAT都會執行丟棄動做）。如今咱們須要的是在NAT B上打一個方向爲202.187.45.3（即Client A的外網地址）的洞，那麼Client A發送到187.34.1.56:40000的信息,Client B就能收到了。這個打洞命令由誰來發呢，呵呵，固然是Server S。
    總結一下這個過程：若是Client A想向Client B發送信息，那麼Client A發送命令給Server S，請求Server S命令Client B向Client A方向打洞。呵呵，是否是很繞口，不過不要緊，想想就很清楚了，況且還有源代碼呢（侯老師說過：在源代碼面前沒有祕密 8）），而後Client A就能夠經過Client B的外網地址與Client B通訊了。
   
    注意：以上過程只適合於Cone NAT的狀況，若是是Symmetric NAT，那麼當Client B向Client A打洞的端口已經從新分配了，Client B將沒法知道這個端口（若是Symmetric NAT的端口是順序分配的，那麼咱們或許能夠猜想這個端口號，但是因爲可能致使失敗的因素太多，咱們不推薦這種猜想端口的方法）。
   
    下面是一個模擬P2P聊天的過程的源代碼，過程很簡單，P2PServer運行在一個擁有公網IP的計算機上，P2PClient運行在兩個不一樣的NAT 後（注意，若是兩個客戶端運行在一個NAT後，本程序極可能不能運行正常，這取決於你的NAT是否支持loopback translation，詳見http://midcom-p2p.sourceforge.net/draft-ford-midcom-p2p-01.txt，固然，此問題能夠經過雙方先嚐試鏈接對方的內網IP來解決，可是這個代碼只是爲了驗證原理，並無處理這些問題），後登陸的計算機能夠得到先登陸計算機的用戶名，後登陸的計算機經過send username message的格式來發送消息。若是發送成功，說明你已取得了直接與對方鏈接的成功。
    程序如今支持三個命令：send , getu , exit
   
    send格式：send username message
    功能：發送信息給username
   
    getu格式：getu
    功能：得到當前服務器用戶列表
   
    exit格式：exit
    功能：註銷與服務器的鏈接（服務器不會自動監測客戶是否吊線）
       
    代碼很短，相信很容易懂，若是有什麼問題，能夠給我發郵件zhouhuis22@sina.com  或者在CSDN上發送短消息。同時，歡迎轉發此文，但但願保留做者版權8-）。
   
    最後感謝CSDN網友 PiggyXP 和 Seilfer的測試幫助

P2PServer.c
/* P2P 程序服務端
 *
 * 文件名：P2PServer.c
 *
 * 日期：2004-5-21
 *
 * 做者：shootingstars(zhouhuis22@sina.com)
 *
 */
#pragma comment(lib, "ws2_32.lib")

#include "windows.h"
#include "..\proto.h"
#include "..\Exception.h"

UserList ClientList;

void InitWinSock()
{
 WSADATA wsaData;

 if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
 {
  printf("Windows sockets 2.2 startup");
  throw Exception("");
 }
 else{
  printf("Using %s (Status: %s)\n",
   wsaData.szDescription, wsaData.szSystemStatus);
  printf("with API versions %d.%d to %d.%d\n\n",
   LOBYTE(wsaData.wVersion), HIBYTE(wsaData.wVersion),
   LOBYTE(wsaData.wHighVersion), HIBYTE(wsaData.wHighVersion));
 
 }
}

SOCKET mksock(int type)
{
 SOCKET sock = socket(AF_INET, type, 0);
 if (sock < 0)
 {
        printf("create socket error");
  throw Exception("");
 }
 return sock;
}

stUserListNode GetUser(char *username)
{
 for(UserList::iterator UserIterator=ClientList.begin();
      UserIterator!=ClientList.end();
       ++UserIterator)
 {
  if( strcmp( ((*UserIterator)->userName), username) == 0 )
   return *(*UserIterator);
 }
 throw Exception("not find this user");
}

int main(int argc, char* argv[])
{
 try{
  InitWinSock();
 
  SOCKET PrimaryUDP;
  PrimaryUDP = mksock(SOCK_DGRAM);

  sockaddr_in local;
  local.sin_family=AF_INET;
  local.sin_port= htons(SERVER_PORT);
  local.sin_addr.s_addr = htonl(INADDR_ANY);
  int nResult=bind(PrimaryUDP,(sockaddr*)&local,sizeof(sockaddr));
  if(nResult==SOCKET_ERROR)
   throw Exception("bind error");

  sockaddr_in sender;
  stMessage recvbuf;
  memset(&recvbuf,0,sizeof(stMessage));

  // 開始主循環.
  // 主循環負責下面幾件事情:
  // 一:讀取客戶端登錄和登出消息,記錄客戶列表
  // 二:轉發客戶p2p請求
  for(;;)
  {
   int dwSender = sizeof(sender);
   int ret = recvfrom(PrimaryUDP, (char *)&recvbuf, sizeof(stMessage), 0, (sockaddr *)&sender, &dwSender);
   if(ret <= 0)
   {
    printf("recv error");
    continue;
   }
   else
   {
    int messageType = recvbuf.iMessageType;
    switch(messageType){
    case LOGIN:
     {
      //  將這個用戶的信息記錄到用戶列表中
      printf("has a user login : %s\n", recvbuf.message.loginmember.userName);
      stUserListNode *currentuser = new stUserListNode();
      strcpy(currentuser->userName, recvbuf.message.loginmember.userName);
      currentuser->ip = ntohl(sender.sin_addr.S_un.S_addr);
      currentuser->port = ntohs(sender.sin_port);
     
      ClientList.push_back(currentuser);

 

 

      // 發送已經登錄的客戶信息
      int nodecount = (int)ClientList.size();
      sendto(PrimaryUDP, (const char*)&nodecount, sizeof(int), 0, (const sockaddr*)&sender, sizeof(sender));
      for(UserList::iterator UserIterator=ClientList.begin();
        UserIterator!=ClientList.end();
        ++UserIterator)
      {
       sendto(PrimaryUDP, (const char*)(*UserIterator), sizeof(stUserListNode), 0, (const sockaddr*)&sender, sizeof(sender));
      }

      break;
     }
    case LOGOUT:
     {
      // 將此客戶信息刪除
      printf("has a user logout : %s\n", recvbuf.message.logoutmember.userName);
      UserList::iterator removeiterator = NULL;
      for(UserList::iterator UserIterator=ClientList.begin();
       UserIterator!=ClientList.end();
       ++UserIterator)
      {
       if( strcmp( ((*UserIterator)->userName), recvbuf.message.logoutmember.userName) == 0 )
       {
        removeiterator = UserIterator;
        break;
       }
      }
      if(removeiterator != NULL)
       ClientList.remove(*removeiterator);
      break;
     }
    case P2PTRANS:
     {
      // 某個客戶但願服務端向另一個客戶發送一個打洞消息
      printf("%s wants to p2p %s\n",inet_ntoa(sender.sin_addr),recvbuf.message.translatemessage.userName);
      stUserListNode node = GetUser(recvbuf.message.translatemessage.userName);
      sockaddr_in remote;
      remote.sin_family=AF_INET;
      remote.sin_port= htons(node.port);
      remote.sin_addr.s_addr = htonl(node.ip);

      in_addr tmp;
      tmp.S_un.S_addr = htonl(node.ip);
      printf("the address is %s,and port is %d\n",inet_ntoa(tmp), node.port);

      stP2PMessage transMessage;
      transMessage.iMessageType = P2PSOMEONEWANTTOCALLYOU;
      transMessage.iStringLen = ntohl(sender.sin_addr.S_un.S_addr);
      transMessage.Port = ntohs(sender.sin_port);
                       
      sendto(PrimaryUDP,(const char*)&transMessage, sizeof(transMessage), 0, (const sockaddr *)&remote, sizeof(remote));

      break;
     }
   
    case GETALLUSER:
     {
      int command = GETALLUSER;
      sendto(PrimaryUDP, (const char*)&command, sizeof(int), 0, (const sockaddr*)&sender, sizeof(sender));

      int nodecount = (int)ClientList.size();
      sendto(PrimaryUDP, (const char*)&nodecount, sizeof(int), 0, (const sockaddr*)&sender, sizeof(sender));

      for(UserList::iterator UserIterator=ClientList.begin();
        UserIterator!=ClientList.end();
        ++UserIterator)
      {
       sendto(PrimaryUDP, (const char*)(*UserIterator), sizeof(stUserListNode), 0, (const sockaddr*)&sender, sizeof(sender));
      }
      break;
     }
    }
   }
  }

 }
 catch(Exception &e)
 {
  printf(e.GetMessage());
  return 1;
 }

 return 0;
}

#pragma comment(lib,"ws2_32.lib")

#include "windows.h"
#include "..\proto.h"
#include "..\Exception.h"
#include <iostream>
using namespace std;

UserList ClientList;

#define COMMANDMAXC 256
#define MAXRETRY    5

SOCKET PrimaryUDP;
char UserName[10];
char ServerIP[20];

bool RecvedACK;

void InitWinSock()
{
 WSADATA wsaData;

 if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
 {
  printf("Windows sockets 2.2 startup");
  throw Exception("");
 }
 else{
  printf("Using %s (Status: %s)\n",
   wsaData.szDescription, wsaData.szSystemStatus);
  printf("with API versions %d.%d to %d.%d\n\n",
   LOBYTE(wsaData.wVersion), HIBYTE(wsaData.wVersion),
   LOBYTE(wsaData.wHighVersion), HIBYTE(wsaData.wHighVersion));
 }
}

SOCKET mksock(int type)
{
 SOCKET sock = socket(AF_INET, type, 0);
 if (sock < 0)
 {
        printf("create socket error");
  throw Exception("");
 }
 return sock;
}

stUserListNode GetUser(char *username)
{
 for(UserList::iterator UserIterator=ClientList.begin();
      UserIterator!=ClientList.end();
       ++UserIterator)
 {
  if( strcmp( ((*UserIterator)->userName), username) == 0 )
   return *(*UserIterator);
 }
 throw Exception("not find this user");
}

void BindSock(SOCKET sock)
{
 sockaddr_in sin;
 sin.sin_addr.S_un.S_addr = INADDR_ANY;
 sin.sin_family = AF_INET;
 sin.sin_port = 0;
 
 if (bind(sock, (struct sockaddr*)&sin, sizeof(sin)) < 0)
  throw Exception("bind error");
}

void ConnectToServer(SOCKET sock,char *username, char *serverip)
{
 sockaddr_in remote;
 remote.sin_addr.S_un.S_addr = inet_addr(serverip);
 remote.sin_family = AF_INET;
 remote.sin_port = htons(SERVER_PORT);
 
 stMessage sendbuf;
 sendbuf.iMessageType = LOGIN;
 strncpy(sendbuf.message.loginmember.userName, username, 10);

 sendto(sock, (const char*)&sendbuf, sizeof(sendbuf), 0, (const sockaddr*)&remote,sizeof(remote));

 int usercount;
 int fromlen = sizeof(remote);
 int iread = recvfrom(sock, (char *)&usercount, sizeof(int), 0, (sockaddr *)&remote, &fromlen);
 if(iread<=0)
 {
  throw Exception("Login error\n");
 }

 // 登陸到服務端後，接收服務端發來的已經登陸的用戶的信息
 cout<<"Have "<<usercount<<" users logined server:"<<endl;
 for(int i = 0;i<usercount;i++)
 {
  stUserListNode *node = new stUserListNode;
  recvfrom(sock, (char*)node, sizeof(stUserListNode), 0, (sockaddr *)&remote, &fromlen);
  ClientList.push_back(node);
  cout<<"Username:"<<node->userName<<endl;
  in_addr tmp;
  tmp.S_un.S_addr = htonl(node->ip);
  cout<<"UserIP:"<<inet_ntoa(tmp)<<endl;
  cout<<"UserPort:"<<node->port<<endl;
  cout<<""<<endl;
 }
}

void OutputUsage()
{
 cout<<"You can input you command:\n"
  <<"Command Type:\"send\",\"exit\",\"getu\"\n"
  <<"Example : send Username Message\n"
  <<"          exit\n"
  <<"          getu\n"
  <<endl;
}

 

/* 這是主要的函數：發送一個消息給某個用戶(C)
 *流程：直接向某個用戶的外網IP發送消息，若是此前沒有聯繫過
 *      那麼此消息將沒法發送，發送端等待超時。
 *      超時後，發送端將發送一個請求信息到服務端，
 *      要求服務端發送給客戶C一個請求，請求C給本機發送打洞消息
 *      以上流程將重複MAXRETRY次
 */
bool SendMessageTo(char *UserName, char *Message)
{
 char realmessage[256];
 unsigned int UserIP;
 unsigned short UserPort;
 bool FindUser = false;
 for(UserList::iterator UserIterator=ClientList.begin();
      UserIterator!=ClientList.end();
      ++UserIterator)
 {
  if( strcmp( ((*UserIterator)->userName), UserName) == 0 )
  {
   UserIP = (*UserIterator)->ip;
   UserPort = (*UserIterator)->port;
   FindUser = true;
  }
 }

 if(!FindUser)
  return false;

 strcpy(realmessage, Message);

for(int i=0;i<MAXRETRY;i++)
 {
  RecvedACK = false;

  sockaddr_in remote;
  remote.sin_addr.S_un.S_addr = htonl(UserIP);
  remote.sin_family = AF_INET;
  remote.sin_port = htons(UserPort);
  stP2PMessage MessageHead;
  MessageHead.iMessageType = P2PMESSAGE;
  MessageHead.iStringLen = (int)strlen(realmessage)+1;
  int isend = sendto(PrimaryUDP, (const char *)&MessageHead, sizeof(MessageHead), 0, (const sockaddr*)&remote, sizeof(remote));
  isend = sendto(PrimaryUDP, (const char *)&realmessage, MessageHead.iStringLen, 0, (const sockaddr*)&remote, sizeof(remote));
 
  // 等待接收線程將此標記修改
  for(int j=0;j<10;j++)
  {
   if(RecvedACK)
    return true;
   else
    Sleep(300);
  }

  // 沒有接收到目標主機的迴應，認爲目標主機的端口映射沒有
  // 打開，那麼發送請求信息給服務器，要服務器告訴目標主機
  // 打開映射端口（UDP打洞）
  sockaddr_in server;
  server.sin_addr.S_un.S_addr = inet_addr(ServerIP);
  server.sin_family = AF_INET;
  server.sin_port = htons(SERVER_PORT);
 
  stMessage transMessage;
  transMessage.iMessageType = P2PTRANS;
  strcpy(transMessage.message.translatemessage.userName, UserName);

  sendto(PrimaryUDP, (const char*)&transMessage, sizeof(transMessage), 0, (const sockaddr*)&server, sizeof(server));
  Sleep(100);// 等待對方先發送信息。
 }
 return false;
}

// 解析命令，暫時只有exit和send命令
// 新增getu命令，獲取當前服務器的全部用戶
void ParseCommand(char * CommandLine)
{
 if(strlen(CommandLine)<4)
  return;
 char Command[10];
 strncpy(Command, CommandLine, 4);
 Command[4]='\0';

 if(strcmp(Command,"exit")==0)
 {
  stMessage sendbuf;
  sendbuf.iMessageType = LOGOUT;
  strncpy(sendbuf.message.logoutmember.userName, UserName, 10);
  sockaddr_in server;
  server.sin_addr.S_un.S_addr = inet_addr(ServerIP);
  server.sin_family = AF_INET;
  server.sin_port = htons(SERVER_PORT);

  sendto(PrimaryUDP,(const char*)&sendbuf, sizeof(sendbuf), 0, (const sockaddr *)&server, sizeof(server));
  shutdown(PrimaryUDP, 2);
  closesocket(PrimaryUDP);
  exit(0);
 }
 else if(strcmp(Command,"send")==0)
 {
  char sendname[20];
  char message[COMMANDMAXC];
  int i;
  for(i=5;;i++)
  {
   if(CommandLine[i]!=' ')
    sendname[i-5]=CommandLine[i];
   else
   {
    sendname[i-5]='\0';
    break;
   }
  }
  strcpy(message, &(CommandLine[i+1]));
  if(SendMessageTo(sendname, message))
   printf("Send OK!\n");
  else
   printf("Send Failure!\n");
 }
 else if(strcmp(Command,"getu")==0)
 {
  int command = GETALLUSER;
  sockaddr_in server;
  server.sin_addr.S_un.S_addr = inet_addr(ServerIP);
  server.sin_family = AF_INET;
  server.sin_port = htons(SERVER_PORT);

  sendto(PrimaryUDP,(const char*)&command, sizeof(command), 0, (const sockaddr *)&server, sizeof(server));
 }
}

// 接受消息線程
DWORD WINAPI RecvThreadProc(LPVOID lpParameter)
{
 sockaddr_in remote;
 int sinlen = sizeof(remote);
 stP2PMessage recvbuf;
 for(;;)
 {
  int iread = recvfrom(PrimaryUDP, (char *)&recvbuf, sizeof(recvbuf), 0, (sockaddr *)&remote, &sinlen);
  if(iread<=0)
  {
   printf("recv error\n");
   continue;
  }
  switch(recvbuf.iMessageType)
  {
  case P2PMESSAGE:
   {
    // 接收到P2P的消息
    char *comemessage= new char[recvbuf.iStringLen];
    int iread1 = recvfrom(PrimaryUDP, comemessage, 256, 0, (sockaddr *)&remote, &sinlen);
    comemessage[iread1-1] = '\0';
    if(iread1<=0)
     throw Exception("Recv Message Error\n");
    else
    {
     printf("Recv a Message:%s\n",comemessage);
    
     stP2PMessage sendbuf;
     sendbuf.iMessageType = P2PMESSAGEACK;
     sendto(PrimaryUDP, (const char*)&sendbuf, sizeof(sendbuf), 0, (const sockaddr*)&remote, sizeof(remote));
    }

    delete []comemessage;
    break;

   }
  case P2PSOMEONEWANTTOCALLYOU:
   {
    // 接收到打洞命令，向指定的IP地址打洞
    printf("Recv p2someonewanttocallyou data\n");
    sockaddr_in remote;
    remote.sin_addr.S_un.S_addr = htonl(recvbuf.iStringLen);
    remote.sin_family = AF_INET;
    remote.sin_port = htons(recvbuf.Port);

    // UDP hole punching
    stP2PMessage message;
    message.iMessageType = P2PTRASH;
    sendto(PrimaryUDP, (const char *)&message, sizeof(message), 0, (const sockaddr*)&remote, sizeof(remote));
               
    break;
   }
  case P2PMESSAGEACK:
   {
    // 發送消息的應答
    RecvedACK = true;
    break;
   }
  case P2PTRASH:
   {
    // 對方發送的打洞消息，忽略掉。
    //do nothing ...
    printf("Recv p2ptrash data\n");
    break;
   }
  case GETALLUSER:
   {
    int usercount;
    int fromlen = sizeof(remote);
    int iread = recvfrom(PrimaryUDP, (char *)&usercount, sizeof(int), 0, (sockaddr *)&remote, &fromlen);
    if(iread<=0)
    {
     throw Exception("Login error\n");
    }
   
    ClientList.clear();

    cout<<"Have "<<usercount<<" users logined server:"<<endl;
    for(int i = 0;i<usercount;i++)
    {
     stUserListNode *node = new stUserListNode;
     recvfrom(PrimaryUDP, (char*)node, sizeof(stUserListNode), 0, (sockaddr *)&remote, &fromlen);
     ClientList.push_back(node);
     cout<<"Username:"<<node->userName<<endl;
     in_addr tmp;
     tmp.S_un.S_addr = htonl(node->ip);
     cout<<"UserIP:"<<inet_ntoa(tmp)<<endl;
     cout<<"UserPort:"<<node->port<<endl;
     cout<<""<<endl;
    }
    break;
   }
  }
 }
}
int main(int argc, char* argv[])
{
 try
 {
  InitWinSock();
 
  PrimaryUDP = mksock(SOCK_DGRAM);
  BindSock(PrimaryUDP);

  cout<<"Please input server ip:";
  cin>>ServerIP;

  cout<<"Please input your name:";
  cin>>UserName;

  ConnectToServer(PrimaryUDP, UserName, ServerIP);

  HANDLE threadhandle = CreateThread(NULL, 0, RecvThreadProc, NULL, NULL, NULL);
  CloseHandle(threadhandle);
  OutputUsage();

  for(;;)
  {
   char Command[COMMANDMAXC];
   gets(Command);
   ParseCommand(Command);
  }
 }
 catch(Exception &e)
 {
  printf(e.GetMessage());
  return 1;
 }
 return 0;
}

#ifndef __HZH_Exception__
#define __HZH_Exception__

#define EXCEPTION_MESSAGE_MAXLEN 256
#include "string.h"

class Exception
{
private:
 char m_ExceptionMessage[EXCEPTION_MESSAGE_MAXLEN];
public:
 Exception(char *msg)
 {
  strncpy(m_ExceptionMessage, msg, EXCEPTION_MESSAGE_MAXLEN);
 }

 char *GetMessage()
 {
  return m_ExceptionMessage;
 }
};

#endif

#pragma once
#include <list>

// 定義iMessageType的值
#define LOGIN 1
#define LOGOUT 2
#define P2PTRANS 3
#define GETALLUSER  4

// 服務器端口
#define SERVER_PORT 2280

// Client登陸時向服務器發送的消息
struct stLoginMessage
{
 char userName[10];
 char password[10];
};

// Client註銷時發送的消息
struct stLogoutMessage
{
 char userName[10];
};

// Client向服務器請求另一個Client(userName)向本身方向發送UDP打洞消息
struct stP2PTranslate
{
 char userName[10];
};

// Client向服務器發送的消息格式
struct stMessage
{
 int iMessageType;
 union _message
 {
  stLoginMessage loginmember;
  stLogoutMessage logoutmember;
  stP2PTranslate translatemessage;
 }message;
};

// 客戶節點信息
struct stUserListNode
{
 char userName[10];
 unsigned int ip;
 unsigned short port;
};

// Server向Client發送的消息
struct stServerToClient
{
 int iMessageType;
 union _message
 {
  stUserListNode user;
 }message;

};

//======================================
// 下面的協議用於客戶端之間的通訊
//======================================
#define P2PMESSAGE 100               // 發送消息
#define P2PMESSAGEACK 101            // 收到消息的應答
#define P2PSOMEONEWANTTOCALLYOU 102  // 服務器向客戶端發送的消息
                                     // 但願此客戶端發送一個UDP打洞包
#define P2PTRASH        103          // 客戶端發送的打洞包，接收端應該忽略此消息

// 客戶端之間發送消息格式 struct stP2PMessage {  int iMessageType;  int iStringLen;         // or IP address  unsigned short Port; };