echo客户端的工作原理也很简单:
1、向服务器端发送一个字符串;
2、接收服务器的返回信息(如果是echo服务器就会返回发送出去的字符串本身)。
3、在标准输出中回显服务器返回的信息。
与ehco服务器类似,我们的echo客户端类也可以从TCPClientSock中派生出来:
- //Filename AppSock.hpp
- #ifndef APP_SOCK_HPP
- #define APP_SOCK_HPP
- #include <string>
- #include "SockClass.hpp"
- class TCPEchoClient: public TCPClientSock{
- public:
- TCPEchoClient(
- const char* server_IP,
- unsigned short server_port,
- int pre_buffer_size = 32);
- ~TCPEchoClient();
- bool doEcho(const std::string& echo_message) const;
- };
- #endif //AppSock.hpp
我们的doEcho()接收一个C++风格的字符串(std::string),将返回值设计成bool是出于以下考虑:我们希望与服务器断开连接的信息能反馈到主程序中,并且在断开连接后终止echo客户端的程序。所以,返回true表示仍然与服务器保持连接,否则则已经断开(或者异常)。
- #include "AppSock.hpp"
- TCPEchoClient::TCPEchoClient(
- const char* server_IP,
- unsigned short server_port,
- int pre_buffer_size):
- TCPClientSock(server_IP, server_port, pre_buffer_size)
- {}
- TCPEchoClient::~TCPEchoClient()
- {}
- bool TCPEchoClient::doEcho(const std::string& echo_message) const
- {
- if (TCPSend(echo_message.data(), echo_message.size()) < 0) {
- return false;
- }
- size_t total_received_length = 0;
- while (total_received_length < echo_message.size()) {
- if (TCPReceive() <= 0) {
- return false;
- }
- std::cout.write(preBuffer, preReceivedLength);
- total_received_length += preReceivedLength;
- }
- std::cout << std::endl;
- return true;
- }
因为我们是先发再收,我们接收前是知道应该收到多少字节的信息的。由于TCP协议对边缘的无保障,我们应该假定是是最不利的情况,也就是一次recv()不能接收完我们需要的数据,所以,如果接收到的字节数小于我们的预期,就再次接收,直到跟我们发送的字符串长度一样。虽然事实上在这种小数据的传输中很难遇到以上所描述的那种情况,但是在网络程序的设计中,应该坚持这样一个基本假设:你永远不知道远程的主机会出什么状况,所以永远以最坏的可能性来设计程序。
最后是主程序。主程序在标准输入中阻塞等待用于echo的信息,为了避免无限循环,我们也设计一个可以关闭服务器端的命令/exit。这样,输入/exit或者服务器断开都可以导致客户端终止。
- #include "SockClass.hpp"
- #include "AppSock.hpp"
- 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();
- TCPEchoClient echo_client(argv[1], server_port);
- std::string msg;
- bool go_on = true;
- while (msg != "/exit" && go_on == true) {
- std::cout << "Echo: ";
- std::getline(std::cin, msg);
- go_on = echo_client.doEcho(msg);
- }
- return 0;
- }
发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。