1月3号作业

题目：搭建select的客户端

代码：

#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.1.27"     //填你们自己的本机IP ifconfig查看
#define PORT 6666             //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地址会改变

    //绑定服务器的IP和端口
    if(connect(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }
    printf("connect success\n");

    //创建集合
    fd_set readfds, tempfds;

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

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

    int s_res = 0;

    //从已经完成连接的队列头中获取一个客户端的信息，生成一个新的文件描述符
    //生成的新的文件描述符才是与客户端通信的文件描述符
    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;
        }

        if(FD_ISSET(0,&tempfds))
        {
            //触发键盘输入事件
            printf("触发键盘输入事件\n");
            bzero(buf, sizeof(buf));
            res = scanf("%s",buf);
            while(getchar()!=10);
            if(res != 1)
            {
                fprintf(stderr, "请输入正确格式:string\n");
                continue;
            }
            if(send(sfd, buf, sizeof(buf), 0) < 0)
            {
                ERR_MSG("send");
                return -1;
            }
            printf("发送成功\n");
        }
        if(FD_ISSET(sfd,&tempfds))
        {
            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");
                //将i所代表的文件描述符关闭
                close(sfd);
                //将i所代表的文件描述符从集合中剔除
                FD_CLR(sfd, &readfds);
            }
            else
            {
                printf(": %s\n",buf);
            }

        }
    }
    close(sfd);
    return 0;
}

运行结果：