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;
}