feiq/feiqlib/tcpserver.cpp

#include "tcpserver.h"
#include <sys/socket.h>
#include <stdio.h>
#include <string.h>
#include <arpa/inet.h>
#include <thread>
#include <unistd.h>

using namespace std;

TcpServer::TcpServer()
{

}

bool TcpServer::start(int port)
{
    if (mStarted)
        return true;

    mSocket = socket(AF_INET, SOCK_STREAM, 0);
    if (mSocket == -1)
    {
        perror("create tcp socket failed: %s");
        return false;
    }

    int val = 1;
    setsockopt(mSocket, SOL_SOCKET, SO_REUSEADDR, &val, sizeof(val));

    sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = htonl(INADDR_ANY);
    addr.sin_port = htons(port);
    int ret = ::bind(mSocket, (sockaddr*)&addr, sizeof(addr));
    if (ret == -1)
    {
        perror("bind failed");
        return false;
    }

    ret = listen(mSocket, 5);
    if (ret == -1)
    {
        perror("listen failed");
        return false;
    }

    mStarted=true;
    thread thd(&TcpServer::keepAccept, this);
    thd.detach();

    return true;
}

void TcpServer::whenNewClient(TcpServer::ClientHandler onClientConnected)
{
    mClientHandler = onClientConnected;
}

void TcpServer::stop()
{
    mStarted = false;
    close(mSocket);
}

void TcpServer::keepAccept()
{
    while (mStarted) {
        sockaddr_in addr;
        socklen_t len = sizeof(addr);
        int ret = ::accept(mSocket, (sockaddr*)&addr, &len);
        if (ret < 0)
        {
            perror("failed to accept");
            break;
        }

        if (mClientHandler)
        {
            mClientHandler(ret);
        }
    }
}