现在,我们用前面所构建的socket类,重新设计第一章中echo的服务器,然后设计客户端程序。
echo服务器的工作原理很简单:
1、接收客户端传来的信息;
2、将接收到的信息原封不动的返回给客户端。
可以看到我们所设计的TCPServerSock类具备了echo服务的所有数据成员,我们只需要添加一个具体的echo方法。因此,我们让设计的echo类从TCPServerSock类中派生出来。
C++代码//Filename AppSock.hpp
#ifndef APP_SOCK_HPP
#define APP_SOCK_HPP
#include "SockClass.hpp"
classTCPEchoServer:publicTCPServerSock{
public:
TCPEchoServer(
constTCPListenSock& listen_sock,
intpre_buffer_size = 32);
~TCPEchoServer();
boolhandEcho()const;
};
#endif //AppSock.hpp
将handEcho()设计成返回值为bool是出于以下考虑:因为服务器端通常是无限循环提供服务的,我们希望客户端能简单的对服务器端的控制,比如说进行关闭,这样就不用每次用Ctrl+c来关闭服务器端的程序。所以,handlEcho()返回true表示客户端正常断开,false表示服务器被要求终止。
C++代码//Filename: AppSock.cpp
#include
#include "AppSock.hpp"
TCPEchoServer::TCPEchoServer(constTCPListenSock& listen_sock,intpre_buffer_size):
TCPServerSock(listen_sock, pre_buffer_size)
{}
TCPEchoServer::~TCPEchoServer()
{}
boolTCPEchoServer::handEcho()const
{
conststd::string SHUTDOWN_CMD ="/shutdown";
while(TCPReceive() > 0) {
std::string cmd(preBuffer, SHUTDOWN_CMD.size());
if(cmd == SHUTDOWN_CMD && preReceivedLength == (int)SHUTDOWN_CMD.size()) {
returnfalse;
}
TCPSend(preBuffer, preReceivedLength);
}
returntrue;
}
我们为服务器指定一个关闭的的特殊字符串/shutdown,如果客户端传来这个字符串,服务器就会终止;其他字符串则会履行echo服务。
最后我们设计主程序:
C++代码//Filename: main.cpp
#include "SockClass.hpp"
#include "AppSock.hpp"
intmain(intargc,char* argv[])
{
constunsignedshortDEFAULT_PORT = 5000;
unsigned shortlisten_port = DEFAULT_PORT;
if(argc == 2 && atoi(argv[1]) > 0) {
listen_port = atoi(argv[1]);
}
TCPListenSock listen_sock(listen_port);
listen_sock.TCPListen();
boolgo_on =true;
while(go_on){
TCPEchoServer echo_server(listen_sock);
go_on = echo_server.handEcho();
}
return0;
}
主程序以第一个参数(argv[1])来指定服务器端口,如果不指定,则默认端口是5000。
除非特别注明,鸡啄米文章均为原创
2015年10月30日
作者:鸡啄米
浏览: