IO多路复用 select TCP客户端

最新推荐文章于 2024-07-31 00:00:00 发布

天赐ytc

最新推荐文章于 2024-07-31 00:00:00 发布

阅读量76

点赞数

文章标签： tcp/ip 网络服务器

本文链接：https://blog.csdn.net/m0_49415349/article/details/129073286

版权

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

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

#define IP "192.168.0.144"   //宏定义服务端私网
#define POINT 8888           //宏定义服务端端口

char buf[32] = "";
int main(int argc, char const *argv[])
{
    //创建流式套接字
    int sockfd = socket(AF_INET,SOCK_STREAM,0);
    if(-1 == sockfd){                  //如果创建失败
        ERR_MSG("socket");             //输出错误
        return -1;
    }
    printf("Client created socket success\n");   //创建套接字成功输出提示
    //绑定客户端的IP和端口号
    
    
    //连接服务器
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;              //必须填AF_INET
    addr.sin_port = htons(POINT);           //服务器绑定的端口号
    addr.sin_addr.s_addr = inet_addr(IP);   //服务器绑定的IP
    socklen_t addrlen = sizeof(addr);

    if(connect(sockfd,(struct sockaddr*)&addr,sizeof(addr)) < 0){
        ERR_MSG("send");             //输出错误
        return -1;
    }
    printf("connect server success\n");

    fd_set readfds, tempfds;
    FD_ZERO(&readfds);

	FD_SET(0, &readfds);
    FD_SET(sockfd,&readfds);
    int maxfd = sockfd;

    //发送数据给服务器
    while (1)  //循环接收、打印服务端发来的消息和发送客户端消息
        {
            tempfds = readfds;
            int retsel = select(maxfd+1,&tempfds,NULL,NULL,NULL);
            if(-1 == retsel){
                ERR_MSG("select");
                return -1;
            }else if(0 == retsel){
                printf("time out...\n");
                return -1;
            }
            if(FD_ISSET(0,&tempfds)){
                //发送信息给服务端
                bzero(buf,sizeof(buf));   //清空字符串
                // printf("请输入发送给服务端的信息:");
                fgets(buf,sizeof(buf),stdin);
                buf[strlen(buf)-1] = 0;
                
                ssize_t resend = send(sockfd,buf,sizeof(buf),0);  //发送buf中的信息给服务端
                if(-1 == resend){                  //如果发送失败
                    ERR_MSG("send");             //输出错误
                    return -1;
                }
                printf("client send success\n");   //发送成功输出提示
            }
            if(FD_ISSET(sockfd,&tempfds)){
                bzero(buf,sizeof(buf));    //清空字符串，接收服务端发来的消息
                ssize_t crecv = recv(sockfd,buf,sizeof(buf),0);
                if(-1 == crecv){                  
                    ERR_MSG("recv");              //接收失败输出错误
                    return -1;
                }else if(0 == crecv){
                    printf("服务端关闭\n");
                    break;
                }
                printf("Client recv [%s: %d]: %s success\n",inet_ntoa(addr.sin_addr),ntohs(addr.sin_port),buf);
                //打印服务端ip、端口号和发来的消息
            }
            
        }
        close(sockfd);   //客户端退出的话，关闭为这个客户端收发消息生成的文件描述符
    //关闭文件描述符
    return 0;
}