TCP（select函数模型）

yayyy4

已于 2022-08-16 20:34:36 修改

阅读量615

点赞数

文章标签： tcp/ip 网络协议网络

于 2022-08-16 20:26:16 首次发布

本文链接：https://blog.csdn.net/yayyy4/article/details/126373704

版权

客户端代码

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

#define MSG_ERR(msg) do{fprintf(stderr,"_%d_",__LINE__);perror(msg);return -1;}while(0)

#define PORT 8888
#define IP "192.168.31.21"

int main()
{
        int sfd = socket(AF_INET,SOCK_STREAM,0);
        if(sfd < 0)
        {
                MSG_ERR("socket");
        }

        int reuse = 1;
        if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0)
        {
                MSG_ERR("setsockopt");
        }

        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)
        {
                MSG_ERR("connect");
        }
        printf("connect success\n");

        fd_set readfds,tempfds;

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

        char buf[128] = "";
        int maxfd = sfd;
        int select_res = 0;
        int i = 0;
        int res = 0;

        while(1)
        {
                tempfds = readfds;
                select_res = select(maxfd+1,&tempfds,NULL,NULL,NULL);
                if(select_res < 0)
                {
                        MSG_ERR("select");
                }
                else if(0 == select_res)
                {
                        fprintf(stderr,"select time out\n");
                        break;
                }
                if(FD_ISSET(0,&tempfds))
                {
                        bzero(buf,sizeof(buf));
                        fgets(buf,sizeof(buf),stdin);
                        buf[strlen(buf)-1] = 0;

                        if(send(sfd,buf,sizeof(buf),0) < 0)
                        {
                                MSG_ERR("send");
                        }
                        printf("send message success\n");
                }
                if(FD_ISSET(sfd,&tempfds))
                {
                        bzero(buf,sizeof(buf));

                        res = recv(sfd,buf,sizeof(buf),0);
                        if(res < 0)
                        {
                                MSG_ERR("recv");
                        }
                        else if(res == 0)
                        {
                                printf("server is off_line\n");
                                break;
                        }
                        printf("%d : %s\n",sfd,buf);
                }
        }
        close(sfd);
        return 0;
}

服务器代码

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

#define MSG_ERR(msg) do{fprintf(stderr,"_%d_",__LINE__);perror(msg);return -1;}while(0)

#define PORT 8888
#define IP "192.168.31.21"

int update_maxfd(int maxfd,fd_set readfds);

int main()
{
        int sfd = socket(AF_INET,SOCK_STREAM,0);
        if(sfd < 0)
        {
                MSG_ERR("socket");
        }

        int reuse = 1;
        if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0)
        {
                MSG_ERR("setsockopt");
        }

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

        if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
        {
                MSG_ERR("bind");
        }

        printf("bind success\n");
        if(listen(sfd,10) < 0)
        {
                MSG_ERR("listen");
        }
        printf("listen success\n");

        struct sockaddr_in cin;
        socklen_t addrlen = sizeof(cin);

        fd_set readfds,tempfds;
        FD_ZERO(&readfds);
        FD_ZERO(&tempfds);

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

        int maxfd = sfd;
        int select_res = 0;
        char buf[128] = "";
        int newfd = -1;
        int i = 0;
        ssize_t res = 0;
        struct sockaddr_in save_addr[1024-4];
        while(1)
        {
                tempfds = readfds;
                struct timeval tm;
                tm.tv_sec = 5;
                tm.tv_usec = 0;
                select_res = select(maxfd+1,&tempfds,NULL,NULL,&tm);
                if(select_res < 0)
                {
                        MSG_ERR("select");
                }
                else if(0 == select_res)
                {
                        fprintf(stderr,"select time out\n");
                        break;
                }

                for(i=0;i<=maxfd;i++)
                {
                        if(FD_ISSET(i,&tempfds) == 0)
                        {
                                continue;
                        }
                        if(0 == i)
                        {
                                printf("click enter\n");
                                fflush(stdout);

                                bzero(buf,sizeof(buf));
                                int sendfd;
                                int res = scanf("%d %s",&sendfd,buf);
                                while(getchar() != 10);
                                if(res != 2)
                                {
                                        fprintf(stderr,"enter the correct content\n");
                                        continue;
                                }
                                if(FD_ISSET(sendfd,&readfds) == 0)
                                {
                                        fprintf(stderr,"sendfd=%d fd not in the clooection\n",sendfd);
                                        continue;
                                }
                                if(send(sendfd,buf,sizeof(buf),0) < 0)
                                {
                                        MSG_ERR("send");
                                        return -1;
                                }
                                printf("send message success\n");
                        }
                        else if(sfd == i)
                        {
                                printf("click client\n");
                                fflush(stdout);
                                newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
                                if(newfd < 0)
                                {
                                        MSG_ERR("accept");
                                }

                                printf("[%s : %d] newfd = %d\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
                                save_addr[newfd-4] = cin;
                                FD_SET(newfd,&readfds);
                                maxfd = maxfd>newfd?maxfd:newfd;
                        }
                        else
                        {
                                bzero(buf,sizeof(buf));
                                res = recv(i,buf,sizeof(buf),0);
                                if(res < 0)
                                {
                                        MSG_ERR("recv");
                                }
                                else if(0 == res)
                                {
                                        printf("[%s : %d] newfd = %d client off-line\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i);
                                        close(i);
                                        FD_CLR(i,&readfds);
                                        maxfd = update_maxfd(maxfd,readfds);
                                }
                                else
                                {
                                        printf("[%s : %d] newfd = %d : %s\n",inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,buf);
                                }

                        }
                }
        }
}

int update_maxfd(int maxfd,fd_set readfds)
{
        int i = maxfd;
        for( ;i>0;i--)
        {
                if(FD_ISSET(i,&readfds))
                {
                        return i;
                }
        }
        return 0;
}