现在,我们用前面所构建的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。
本节源代码下载:
Linux:
http://www.163pan.com/files/c0x000g0o.html
win32:
http://www.163pan.com/files/c0x000g0q.html