基于linux TCP的select回射服务器

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <errno.h>
#include <sys/select.h>
#include <unistd.h>
#include <stdlib.h>
#include <unordered_map>
#include <string>
#include <iostream>
using namespace std;

//int->fd
//string->ip&port
unordered_map<int, string> clients_addr;

int main(int argc, char const *argv[])
{

    int listenfd = socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, IPPROTO_TCP );
    struct sockaddr_in serv;
    serv.sin_family = AF_INET;
    serv.sin_port = htons(5000);
    serv.sin_addr.s_addr = inet_addr("127.0.0.1");

    int on = 1;
    int connfd;
    if (setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on)) < 0)
    {
        perror("setsockopt");
        exit(EXIT_FAILURE);
    }

    if (bind(listenfd, (struct sockaddr *)&serv, sizeof(serv)) < 0)
    {
        perror("bind");
        exit(EXIT_FAILURE);
    }

    if (listen(listenfd, 3) < 0)
    {
        perror("listen");
        exit(EXIT_FAILURE);
    }

    sockaddr_in peer_addr;
    socklen_t peer_addr_len = sizeof(peer_addr);
    int maxfd = listenfd;

    fd_set fdall, fdread;
    FD_ZERO(&fdall);
    FD_SET(listenfd, &fdall);
    // printf("listen fd = %d\n",listenfd);
    while (1)
    {
        fdread = fdall;
        int ret = select(maxfd + 1, &fdread, NULL, NULL, NULL);
        if (ret == -1)
        {
            perror("select");
            exit(EXIT_FAILURE);
        }

        if (ret == 0)
        {
            continue;
        }

        for (int i = 0; i <= maxfd; i++)
        {
            if (FD_ISSET(i, &fdread))
            {
                if (i == listenfd)
                {
                    connfd = accept4(listenfd, (struct sockaddr *)&peer_addr, &peer_addr_len, SOCK_CLOEXEC | SOCK_NONBLOCK);
                    char ip_str_buf[1024];
                    sprintf(ip_str_buf, "%s:%d", inet_ntoa(peer_addr.sin_addr), ntohs(peer_addr.sin_port));

                    clients_addr[connfd] = ip_str_buf;
                    cout << clients_addr[connfd] << "-->" << "login" << endl;

                    FD_SET(connfd, &fdall);
                    if (maxfd < connfd)
                    {
                        maxfd = connfd;
                    }
                }
                else
                {
                    char msg_buf[1024] = {0x00};
                    ret = recv(i, msg_buf, sizeof(msg_buf), 0);
                    if (ret == -1)
                    {
                        perror("recv");
                        exit(EXIT_FAILURE);
                    }

                    if (ret == 0)
                    {
                        close(i);
                        FD_CLR(i, &fdall);
                        cout << clients_addr[i] << "-->" << "log out" << endl;
                    }
                    else
                    {
                        cout << clients_addr[i] << "-->" << msg_buf << endl;
                        ret = send(i, msg_buf, sizeof(msg_buf), 0);
                    }

                }
            }

        }

    }

    return 0;
}

需要注意的地方：

（1）在winsock中，select函数第一个参数被忽略了。而linux上实际上该参数是指定查询的文件描述符区间。故要设为最大文件描述符+1。

（2）为什么需要两个文件描述符集合？因为fdread在select之后，它的值被改变了。所以需要fdall保存它之前的值。而winsock不会改变fdread的值。

（3）没有给出客户端代码。客户端测试可用telnet或netcat。

（4）使用了C++11中的容器unordered_map(即哈希表)来存储ip地址。

使用方法：

telnet 127.0.0.1 5000

xxxxx…

还可使用netcat:

netcat 127.0.0.1 5000 

xxxxx…

建议使用netcat命令，因为netcat对udp也能支持（-u参数），将来写UDP服务端的时候，也可用其进行测试。而telnet只支持TCP。