同样的,我们将UDP版的doEcho()也设计成返回bool:true表示循环继续;false表示关闭客户端。
- class UDPEchoClient: public UDPClientSock{
- public:
- explicit UDPEchoClient(
- int pre_buffer_size = 32);
- ~UDPEchoClient();
- bool doEcho(const std::string& echo_message);
- };
我们依然使用C++字符串。
- UDPEchoClient::UDPEchoClient(
- int pre_buffer_size):
- UDPClientSock(pre_buffer_size)
- {}
- UDPEchoClient::~UDPEchoClient()
- {}
- bool UDPEchoClient::doEcho(const std::string& echo_message)
- {
- if ( UDPSendtoDest(echo_message.data(), echo_message.size()) < 0) {
- return false;
- }
- if (echo_message == "/shutdown") {
- return false;
- }
- if (UDPReceive() < 0) {
- return false;
- }
- std::cout.write(preBuffer, preReceivedLength);
- std::cout << std::endl;
- return true;
- }
当echo_message为“空”的时候,即输入直接回车,是一个"",用C风格来说,即时'\0',从C++来说,是const char[1],其C++风格的长度echo_message.size()为0,这时候就会发送一个“0长度”的UDP数据包。
另外,我们小心设计了关闭服务器的请求,发送/shutdown后,客户端会自动返回false,表示会关闭,不再等待来自服务器的recvfrom()。否则,服务器已经关闭,recvfrom()则会一直阻塞。
- int main(int argc, char* argv[])
- {
- unsigned short server_port = 5000;
- if (argc == 3 && atoi(argv[2]) > 0) {
- server_port = atoi(argv[2]);
- }
- WinsockAPI winsockInfo;
- winsockInfo.showVersion();
- UDPEchoClient echo_client;
- echo_client.UDPSetDest(argv[1], server_port);
- std::string msg;
- bool go_on = true;
- while (msg != "/exit" && go_on){
- std::cout << "Echo: ";
- std::getline(std::cin, msg);
- go_on = echo_client.doEcho(msg);
- }
- return 0;
- }
主程序中,如果使用/exit,会先发送给服务器,然后再关闭。
发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。