使用select的客户端

最新推荐文章于 2023-11-22 21:21:48 发布

原创最新推荐文章于 2023-11-22 21:21:48 发布 · 179 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#服务器 #tcp

该代码示例展示了如何创建一个TCP客户端，利用select函数实现在连接到指定IP和端口后，能持续接收和发送数据。程序通过FD_SET设置文件描述符集，监控标准输入和socket，以便在有输入事件或服务器响应时进行相应操作。当发生错误、超时或服务器关闭连接时，程序会给出相应的处理。

作业需求：

使用select函数完成TCP客户端，实现可随时收发。

实现过程：

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <pthread.h>
#include <poll.h>

#define ERR_MSG(msg) do{\
    fprintf(stderr,"line:%d\n",__LINE__);\
    perror(msg);\
}while(0)
#define IP "192.168.8.18"
#define PORT 6666


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

    int sfd = socket(AF_INET,SOCK_STREAM,0);
    if(sfd<0)
    {
        ERR_MSG("socket");
        return -1;
    }

    struct sockaddr_in sin;
    sin.sin_family  = AF_INET;
    sin.sin_port    = htons(PORT);
    sin.sin_addr.s_addr = inet_addr(IP);


    if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
    {
        ERR_MSG("connect error");
        return -1;
    }
    fd_set readfds,tempfds;

    FD_ZERO(&readfds);

    FD_SET(0,&readfds);
    FD_SET(sfd,&readfds);

    printf("connect success\n");
    
    char buf[128] ="";
    ssize_t res =0;    
    int s_res=0;

    while(1)
    {
        tempfds=readfds;
        s_res = select(sfd+1,&tempfds,NULL,NULL,NULL);

        if(s_res <0)
        {
            ERR_MSG("poll");
            return -1;
        }
        else if(0==s_res)
        {
            printf("time out...\n");
            break;
        }
        if((FD_ISSET(0,&tempfds))!=0)
        {
            printf("键盘输入事件\n");

            bzero(buf,sizeof(buf));
            fgets(buf,sizeof(buf),stdin);
            buf[strlen(buf)-1]=0;

            if(send(sfd,buf,sizeof(buf),0)<0)
            {
                ERR_MSG("send");
                return -1;
            }

            printf("send success\n");
        }
        if((FD_ISSET(sfd,&tempfds))!=0)
        {
            printf("服务器交互事件\n");
            bzero(buf,sizeof(buf));

            res=recv(sfd,buf,sizeof(buf),0);
            if(res < 0)
            {
                ERR_MSG("recv");
                return -1;
            }
            else if(0==res)
            {
                fprintf(stderr,"服务器关闭\n");
                break;
            }
            printf(": %s\n",buf);
        }
    }
    

    close(sfd);
    return 0;
}

实验结果：