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 <sys/select.h>
#include <sys/time.h>

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

#define IP "192.168.8.90"     //填你们自己的本机IP ifconfig查看
#define PORT 43210             //1024~49151

int main(int argc, const char *argv[])
{
    //创建流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }

    //允许端口快速重用--->允许程序退出后，端口号能被新的进程快速覆盖
    //必须放在bind之前
    int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) <0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    //    printf("允许端口快速重用成功\n");


    //填充地址信息结构体
    //根据地址族的不同，地址信息结构体不一致，AF_INET:man 7 ip
    struct sockaddr_in sin;
    sin.sin_family         = AF_INET;         //必须填AF_INET;
    sin.sin_port         = htons(PORT);     //端口号的网络字节序，1024~49151
    sin.sin_addr.s_addr = inet_addr(IP);     //本机IP地址的网络字节序，ifconfig 
    //更换网络环境后，IP地址会改变

    //连接服务器
    if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
    {
        ERR_MSG("continue");
        return -1;
    }
    printf("connect success\n");

    //创建集合
    fd_set readfds;
    fd_set tempfds;

    FD_ZERO(&readfds);     //清空文件描述符集合
    FD_ZERO(&tempfds);


    //将需要的文件描述符放入到读集合中
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);

    int s_res = 0;
    struct sockaddr_in cin;     //存储获取到的客户端的地址信息
    socklen_t addrlen = sizeof(cin);     //真实地址信息结构体的大小

    //从已经完成连接的队列头中获取一个客户端的信息，生成一个新的文件描述符
    //生成的新的文件描述符才是与客户端通信的文件描述符
    char buf[128] = "";
    ssize_t res = 0;


    while(1)
    {
        tempfds = readfds;
        //让内核监测集合
        s_res = select(sfd+1, &tempfds, NULL, NULL, NULL);
        if(s_res < 0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if(0 == s_res)
        {
            printf("time out...\n");
            break;
        }

        //若代码能运行到当前位置，则代表集合中有文件描述符准备就绪
        //集合中会只剩下准备就绪的文件描述符
        //例如0号准备就绪，则集合中会只剩下0；
        //sfd准备就绪，则集合中会只剩下sfd;
        //0和sfd均准备就绪，则集合中会剩下0和sfd;
        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("发送成功\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)
            {
                printf("服务器断开\n");
                break;
            }
            else
            {
                printf("[%s:%d]: %s\n",inet_ntoa(cin.sin_addr), ntohs(cin.sin_port),buf);
            }
        }
    }
    //关闭套接字
    close(sfd);
    return 0;
}

运行结果