现在,我们用前面所构建的socket类,重新设计第一章中echo的服务器,然后设计客户端程序。
echo服务器的工作原理很简单:
1、接收客户端传来的信息;
2、将接收到的信息原封不动的返回给客户端。
可以看到我们所设计的TCPServerSock类具备了echo服务的所有数据成员,我们只需要添加一个具体的echo方法。因此,我们让设计的echo类从TCPServerSock类中派生出来。
- //Filename AppSock.hpp
- #ifndef APP_SOCK_HPP
- #define APP_SOCK_HPP
- #include "SockClass.hpp"
- class TCPEchoServer: public TCPServerSock{
- public:
- TCPEchoServer(
- const TCPListenSock& listen_sock,
- int pre_buffer_size = 32);
- ~TCPEchoServer();
- bool handEcho() const;
- };
- #endif //AppSock.hpp
将handEcho()设计成返回值为bool是出于以下考虑:因为服务器端通常是无限循环提供服务的,我们希望客户端能简单的对服务器端的控制,比如说进行关闭,这样就不用每次用Ctrl+c来关闭服务器端的程序。所以,handlEcho()返回true表示客户端正常断开,false表示服务器被要求终止。
- //Filename: AppSock.cpp
- #include <string>
- #include "AppSock.hpp"
- TCPEchoServer::TCPEchoServer(const TCPListenSock& listen_sock, int pre_buffer_size):
- TCPServerSock(listen_sock, pre_buffer_size)
- {}
- TCPEchoServer::~TCPEchoServer()
- {}
- bool TCPEchoServer::handEcho() const
- {
- const std::string SHUTDOWN_CMD = "/shutdown";
- while (TCPReceive() > 0) {
- std::string cmd(preBuffer, SHUTDOWN_CMD.size());
- if (cmd == SHUTDOWN_CMD && preReceivedLength == (int)SHUTDOWN_CMD.size()) {
- return false;
- }
- TCPSend(preBuffer, preReceivedLength);
- }
- return true;
- }
我们为服务器指定一个关闭的的特殊字符串/shutdown,如果客户端传来这个字符串,服务器就会终止;其他字符串则会履行echo服务。
最后我们设计主程序:
- //Filename: main.cpp
- #include "SockClass.hpp"
- #include "AppSock.hpp"
- int main(int argc, char* argv[])
- {
- const unsigned short DEFAULT_PORT = 5000;
- unsigned short listen_port = DEFAULT_PORT;
- if (argc == 2 && atoi(argv[1]) > 0) {
- listen_port = atoi(argv[1]);
- }
- TCPListenSock listen_sock(listen_port);
- listen_sock.TCPListen();
- bool go_on = true;
- while (go_on){
- TCPEchoServer echo_server(listen_sock);
- go_on = echo_server.handEcho();
- }
- return 0;
- }
主程序以第一个参数(argv[1])来指定服务器端口,如果不指定,则默认端口是5000。
发表评论:
◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。