与TCP版的Echo Server类似,我们从Server Socket类中派生出Echo Server类。
- class UDPEchoServer: public UDPServerSock{
- public:
- explicit UDPEchoServer(
- unsigned short server_port,
- int pre_buffer_size = 32);
- ~UDPEchoServer();
- bool handEcho();
- };
我们依然让handEcho()返回一个bool,true表示客户端“正常”离开(这里没用“断开”这个词是因为UDP是无连接的;另外,我们这里用recvfrom()返回小于0来表示客户端其实是“非正常”的离开了,比如连接被重置。事实上,作为UDP服务器,根本不关心客户端是在连还是已经离开),false表示客户端发出指令要求服务器端关闭。
- UDPEchoServer::UDPEchoServer(
- unsigned short server_port,
- int pre_buffer_size):
- UDPServerSock(server_port, pre_buffer_size)
- {}
- UDPEchoServer::~UDPEchoServer()
- {}
- bool UDPEchoServer::handEcho()
- {
- const std::string SHUTDOWN_CMD = "/shutdown";
- while (UDPReceive() >= 0) {
- std::string cmd(preBuffer, SHUTDOWN_CMD.size());
- if (cmd == SHUTDOWN_CMD &&
- preReceivedLength == static_cast<int>(SHUTDOWN_CMD.size())) {
- return false;
- }
- std::cout << "Client ( "
- << inet_ntoa(lastfromSockAddr.sin_addr)
- << " : "
- << ntohs(lastfromSockAddr.sin_port)
- << " ) sent a message."
- << std::endl;
- UDPSetDest(lastfromSockAddr);
- UDPSendtoDest(preBuffer, preReceivedLength);
- }
- return true;
- }
这里跟TCP有些细微的差别。在TCP中,recv()返回0表示连接正常断开,而UDP中没有连接和断开的概念,recv()或者recvfrom()返回0表示收到一个0字节大小数据的数据报。另外,因为TCP是一对一连接的,所以一旦连接上,TCP服务器只能处理来自一个客户端的echo请求(后面会讲到多线程的使用,就可以让TCP同时处理多个客户端了);而UDP服务器则可以处理来自任何客户端的echo请求,为了返回信息到正确的客户端,我们的策略是,接收一个UDP数据包后,马上刷新发送目标地址为上一次接收地址,然后再回发数据,所以这里每次多了一个重新指定发送目的地的函数。
最后,主程序基本不需要改变:
- int main(int argc, char* argv[])
- {
- const unsigned short DEFAULT_PORT = 5000;
- unsigned short server_port = DEFAULT_PORT;
- if (argc == 2 && atoi(argv[1]) > 0) {
- server_port = atoi(argv[1]);
- }
- UDPEchoServer echo_server(server_port);
- bool go_on = true;
- while (go_on){
- go_on = echo_server.handEcho();
- }
- return 0;
- }
发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。