TCP客户端select应用函数

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

//打印错误信息的宏函数
#define ERR_MSG(msg) do{\
    fprintf(stderr,"__%d__\n",__LINE__);\
    perror(msg);\
}while(0)
//宏定义IP地址
#define IP "192.168.31.147"
//宏定义端口号pott
#define PORT 8888
char buf[128]="";
//线程函数
void *callback(void* arg)
{
    pthread_detach(pthread_self());//自己回收自己
    while(1){
        ssize_t res=0;
        int fd=*(int*)arg;
        res=recv(fd,&buf,sizeof(buf),0);
        if(res<0)
        {
            perror("recv");
            break;
        }
        else if(res==0)
        {

            printf("server is off_line\n");
            break;
        }
        printf("%ld:%s\n",res,buf);
    }
    pthread_exit(NULL);

}
int main(int argc, const char *argv[])
{
    //创建流式套接字
    int sfd=socket(AF_INET,SOCK_STREAM,0);//返回文件描述符
    if(sfd<0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("create socket success\n");
    //填充信息结构体信息
    struct sockaddr_in sock;
    //主机端口字节序转网络端口字节序
    sock.sin_port=htons(PORT);//返回值是网络端口字节序
    //主机源地址IP转换成网络字节序IP地址
    sock.sin_addr.s_addr=inet_addr(IP);//返回网络IP字节序
    sock.sin_family =AF_INET;
    //创建的这个结构体必须要IPV4地址类型
    //绑定本机IP和端口号到bind,填充文件描述符
    //这里完成后网络地址就绑定完成了
    if(connect(sfd,(struct sockaddr*)&sock,sizeof(sock))<0)
    {
        ERR_MSG("connect");
        return -1;
    }
    //创造俩个读集合
    fd_set readfd1;
    //先清空元素为0
    FD_ZERO(&readfd1);
    //将文件描述符添加进入数组
    FD_SET(0,&readfd1);
    FD_SET(sfd,&readfd1);
    int select_res;
    //创建线程接收信号
    pthread_t tid;//储存pid号的变量
    if(pthread_create(&tid,NULL,callback,(void*)&sfd)!=0)
    {
        ERR_MSG("ptnread_create");
        return -1;
    }
    while(1)
    {
        //接收select返回值,确定有没有生成等待就绪的文件描述符
        //这里使用的是哪个结构体，那个结构体内部所有文件描述符除了就绪的都会被清理
        select_res=select(sfd+1,&readfd1,NULL,NULL,NULL);
        if(select_res<0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if(select_res==0)
        {
            printf("time out\n ");
            return -1;
        }
        else{
            bzero(buf,sizeof(buf));
            printf("请输入>>>>");
            fgets(buf,sizeof(buf),stdin);
            buf[strlen(buf)-1]='\0';
            if(send(sfd,&buf,sizeof(buf),0)<0)
            {
                ERR_MSG("send");
                return -1;
            }
            printf("success send\n");
        }
    }

    close(sfd);
    return 0;
}