socket 编程入门教程（一）TCP server 端：8、本章的完整源代码

// Filename: TcpServerClass.hpp

#ifndef TCPSERVERCLASS_HPP_INCLUDED
#define  TCPSERVERCLASS_HPP_INCLUDED

#include  < unistd.h >
#include  < iostream >
#include  < sys / socket.h >
#include  < arpa / inet.h >

class  TcpServer
{
private :
     int  listenSock;
     int  communicationSock;
    sockaddr_in servAddr;
    sockaddr_in clntAddr;
public :
    TcpServer( int  listen_port);
     bool  isAccept();
     void  handleEcho();
};


#endif   //  TCPSERVERCLASS_HPP_INCLUDED

 

// Filename: TcpServerClass.cpp

#include  " TcpServerClass.hpp "

TcpServer::TcpServer( int  listen_port)
{
     if  ( (listenSock  =  socket(PF_INET, SOCK_STREAM, IPPROTO_TCP))  <   0  ) {
         throw   " socket() failed " ;
    }

    memset( & servAddr,  0 ,  sizeof (servAddr));
    servAddr.sin_family  =  AF_INET;
    servAddr.sin_addr.s_addr  =  htonl(INADDR_ANY);
    servAddr.sin_port  =  htons(listen_port);

     if  ( bind(listenSock, (sockaddr * ) & servAddr,  sizeof (servAddr))  <   0  ) {
         throw   " bind() failed " ;
    }

     if  ( listen(listenSock,  10 )  <   0  ) {
         throw   " listen() failed " ;
    }
}

bool  TcpServer::isAccept()
{
    unsigned  int  clntAddrLen  =   sizeof (clntAddr);

     if  ( (communicationSock  =  accept(listenSock, (sockaddr * ) & clntAddr,  & clntAddrLen))  <   0  ) {
         return   false ;
    }  else  {
        std::cout  <<   " Client(IP:  "   <<  inet_ntoa(clntAddr.sin_addr)  <<   " ) connected.\n " ;
         return   true ;
    }
}

void  TcpServer::handleEcho()
{
     const   int  BUFFERSIZE  =   32 ;
     char  buffer[BUFFERSIZE];
     int  recvMsgSize;
     bool  goon  =   true ;

     while  ( goon  ==   true  ) {
         if  ( (recvMsgSize  =  recv(communicationSock, buffer, BUFFERSIZE,  0 ))  <   0  ) {
             throw   " recv() failed " ;
        }  else   if  ( recvMsgSize  ==   0  ) {
            goon  =   false ;
        }  else  {
             if  ( send(communicationSock, buffer, recvMsgSize,  0 )  !=  recvMsgSize ) {
                 throw   " send() failed " ;
            }
        }
    }

    close(communicationSock);
}

演示程序：

// Filename: main.cpp
// Tcp Server C++ style, single work

#include  < iostream >
#include  " TcpServerClass.hpp "

int  echo_server( int  argc,  char *  argv[]);

int  main( int  argc,  char *  argv[])
{
     int  mainRtn  =   0 ;
     try  {
        mainRtn  =  echo_server(argc, argv);
    }
     catch  (  const   char *  s ) {
        perror(s);
        exit(EXIT_FAILURE);
    }

     return  mainRtn;
}

int  echo_server( int  argc,  char *  argv[])
{
     int  port;
     if  ( argc  ==   2  ) {
        port  =  atoi(argv[ 1 ]);
    }  else  {
        port  =   5000 ;
    }

    TcpServer myServ(port);

     while  (  true  ) {
         if  ( myServ.isAccept()  ==   true  ) {
            myServ.handleEcho();
        }
    }

     return   0 ;
}