與TCP版的Echo Server類似,我們從Server Socket類中派生出Echo Server類。

C++代碼
  1. class UDPEchoServer: public UDPServerSock{  
  2. public:  
  3.     explicit UDPEchoServer(  
  4.                 unsigned short server_port,  
  5.                 int pre_buffer_size = 32);  
  6.     ~UDPEchoServer();  
  7.     bool handEcho();  
  8. };  

       我們依然讓handEcho()返回一個bool,true表示客戶端“正常”離開(這里沒用“斷開”這個詞是因為UDP是無連接的;另外,我們這里用recvfrom()返回小于0來表示客戶端其實是“非正常”的離開了,比如連接被重置。事實上,作為UDP服務器,根本不關心客戶端是在連還是已經離開),false表示客戶端發出指令要求服務器端關閉。

C++代碼
  1. UDPEchoServer::UDPEchoServer(  
  2.                 unsigned short server_port,  
  3.                 int pre_buffer_size):  
  4. UDPServerSock(server_port, pre_buffer_size)  
  5. {}  
  6.   
  7. UDPEchoServer::~UDPEchoServer()  
  8. {}  
  9.   
  10. bool UDPEchoServer::handEcho()  
  11. {  
  12.     const std::string SHUTDOWN_CMD = "/shutdown";  
  13.     while (UDPReceive() >= 0) {  
  14.         std::string cmd(preBuffer, SHUTDOWN_CMD.size());  
  15.         if (cmd == SHUTDOWN_CMD &&  
  16.             preReceivedLength == static_cast<int>(SHUTDOWN_CMD.size())) {  
  17.             return false;  
  18.         }  
  19.         std::cout    << "Client ( "  
  20.                     << inet_ntoa(lastfromSockAddr.sin_addr)  
  21.                     << " : "  
  22.                     << ntohs(lastfromSockAddr.sin_port)  
  23.                     << " ) sent a message."  
  24.                     << std::endl;  
  25.         UDPSetDest(lastfromSockAddr);  
  26.         UDPSendtoDest(preBuffer, preReceivedLength);  
  27.     }  
  28.     return true;  
  29. }  

       這里跟TCP有些細微的差別。在TCP中,recv()返回0表示連接正常斷開,而UDP中沒有連接和斷開的概念,recv()或者recvfrom()返回0表示收到一個0字節大小數據的數據報。另外,因為TCP是一對一連接的,所以一旦連接上,TCP服務器只能處理來自一個客戶端的echo請求(后面會講到多線程的使用,就可以讓TCP同時處理多個客戶端了);而UDP服務器則可以處理來自任何客戶端的echo請求,為了返回信息到正確的客戶端,我們的策略是,接收一個UDP數據包后,馬上刷新發送目標地址為上一次接收地址,然后再回發數據,所以這里每次多了一個重新指定發送目的地的函數。

       最后,主程序基本不需要改變:

C++代碼
  1. int main(int argc, char* argv[])  
  2. {  
  3.     const unsigned short DEFAULT_PORT = 5000;  
  4.     unsigned short server_port = DEFAULT_PORT;  
  5.     if (argc == 2 && atoi(argv[1]) > 0) {  
  6.         server_port = atoi(argv[1]);  
  7.     }  
  8.   
  9.     UDPEchoServer echo_server(server_port);  
  10.   
  11.     bool go_on = true;  
  12.     while (go_on){  
  13.         go_on = echo_server.handEcho();  
  14.     }  
  15.   
  16.     return 0;  
  17. }  
除非特別注明,雞啄米文章均為原創
轉載請標明本文地址:http://www.vkzldl.live/software/472.html
2015年11月30日
作者:雞啄米 分類:軟件開發 瀏覽: 評論:0