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