linux poll 使用简单例子

         可以监听多个客户端,代码比select简洁。直接上服务端代码,测试可以用telnet。 

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdlib.h>
#include <assert.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <poll.h>

#define MAX_FD_NUM 20
#define MAXLEN 1024

int main(int argc,char* argv[])
{
    printf("server start up\n");

    if(argc <= 2)
    {
        printf("usage:%s ip port\n",basename(argv[0]));
        return 1;
    }

    //IP地址
    const char* ip = argv[1];
    //端口号
    int port = atoi(argv[2]);
    //内核监听队列的最大长度(完全连接的socket)
    //int backlog = atoi(argv[3]);

    //创建socket	(TIP/IP协议族,流式socket)
    int server_sockfd = socket(PF_INET,SOCK_STREAM,0);

    //TCP/IP协议族的socket地址结构体
    struct sockaddr_in server_addr;
    bzero(&server_addr,sizeof(server_addr));
    server_addr.sin_family = AF_INET;			//TCP/IPv4的地址族
    inet_pton(AF_INET,ip,&server_addr.sin_addr);		//将IP地址字符串转换为二进制的整数并赋给addr.sin_server_addr
    server_addr.sin_port = htons(port);			//端口,host to net,将主机字节序(小端)转换为网络字节序(大端)

    //将文件描述符sock和socket地址关联,仅服务端需要,客户端自动绑定地址
    //注意需要强制转换为 struct sockaddr*
    int ret = bind(server_sockfd,(struct sockaddr*)&server_addr,sizeof(server_addr));
    assert(ret != -1);

    //监听
    ret = listen(server_sockfd,MAX_FD_NUM-1);
    assert(ret != -1);

    //等待客户端做些连接等相关工作
    sleep(3);

    //客户端地址信息
    struct sockaddr_in client_addr;
    socklen_t client_addr_len = sizeof(struct sockaddr_in);

    //poll fds
    struct pollfd pollfdArry[MAX_FD_NUM];
    for(int i=0;i<MAX_FD_NUM;++i)
    {
        pollfdArry[i].fd = -1;
    }

    //insert the server socket fd
    pollfdArry[0].fd = server_sockfd;
    pollfdArry[0].events = POLLIN;

    int cur_fd_num = 1;
    char buf[MAXLEN]={0};

    //
    while (1)
    {
        int nready = poll(pollfdArry,cur_fd_num,-1);

        //server socket fd
        if(pollfdArry[0].revents & POLLIN)
        {
            if(cur_fd_num > MAX_FD_NUM)
            {
                printf("socket num to much\n");
            }
            else
            {
                //接受连接,并将被接受的远端sock地址信息保存在第二个参数中
                //只是从监听队列中取出连接,即使客户端已经断开网络连接也会accept成功
                int client_sockfd = accept(server_sockfd,(struct sockaddr*)&client_addr,&client_addr_len);

                if(client_sockfd < 0)
                {
                   perror("accept");
                }
                else
                {
                    //inet_ntoa(struct addr_in) 将IP地址转换为字符串并返回
                    printf("accept client_addr %s\n",inet_ntoa(client_addr.sin_addr));

                    for(int i=0;i<MAX_FD_NUM;++i)
                    {
                        if(pollfdArry[i].fd == -1)
                        {
                            pollfdArry[i].fd = client_sockfd;
                            pollfdArry[i].events = POLLIN;
                            cur_fd_num++;
                            break;
                        }

                    }
                }

            }

            if(--nready <= 0)
            {
                continue;
            }
        }

        for(int i=0;i<MAX_FD_NUM;++i)
        {
            if(pollfdArry[i].fd < 0)
            {
                continue;
            }

            if(pollfdArry[i].revents & (POLLIN |POLLERR))
            {

                int n = recv(pollfdArry[i].fd,buf,MAXLEN,0);

                if(n < 0)
                {
                    if(ECONNRESET == errno)
                    {
                        close(pollfdArry[i].fd);
                        pollfdArry[i].fd = -1;
                        cur_fd_num--;
                    }
                    else
                    {
                        perror("recv");
                    }
                }
                else if(n == 0)
                {
                    close(pollfdArry[i].fd);
                    pollfdArry[i].fd = -1;
                    cur_fd_num--;
                }
                else
                {
                    printf("recived:%s\n",buf);
                }

            }

            if(--nready)
            {
                break;
            }
        }
    }


    //关闭连接,实际只是socket的引用-1,必须引用为0才会真正关闭
    for(int i=0;i<MAX_FD_NUM;++i)
    {
        if(pollfdArry[i].fd != -1)
        {
            close(pollfdArry[i].fd);
        }
    }

    return 0;
}

 

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值