IO模型用select实现并发

2023-12-26 19:33:16

客户端

代码
#include <func.h>
#define PORT 6666
#define IP "192.168.124.42"

int main(int argc, const char *argv[]) 
{
    //流式套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);

    if(sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    //允许端口快速被复用---监测端口号是否真的有进程在占用,如果没有,则快速复
    int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速复用成功\n");

    //绑定字节的客户端地址 可不写
    //若客户端没有绑定,则操作系统会自动绑定本机IP和随机端口(49152~65535)

    需要手动填充连接服务器对应的地址信息(服务器的地址信息)

    //绑定服务器
    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;     //必须AF_INET  
    sin.sin_port        = htons(PORT); //服务器绑定的端口号 
    sin.sin_addr.s_addr = inet_addr(IP);//服务器绑定的IO
    if(connect(sfd,(struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }

    fd_set readfds, tempfds;
    FD_ZERO(&readfds);

    //填充
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);

    int maxfd = sfd;  //最大文件描述符
    char arr[128] = "";
    ssize_t a = 0;
    while(1)
    {
        tempfds = readfds;
        int s_ser = select(maxfd+1, &tempfds, NULL,NULL,NULL);
        if(s_ser <0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if(s_ser == 0)
        {
            printf("连接超时\n");
            return 0;
        }
        for(int i = 0; i<= maxfd; i++)
        {
            if(FD_ISSET(i, &tempfds) == 0)
            {
                continue;
            }
            if(i == 0)
            {   //发送给服务器
                printf("发送信息给服务端\n");
                fgets(arr, sizeof(arr), stdin);
                if(send(sfd, arr, sizeof(arr),0)< 0 )
                {
                    ERR_MSG("send");
                    return -1;
                }
                printf("发送成功\n");
            }
            if(i = sfd)
            {
                //收到服务器信息
                bzero(arr, sizeof(arr));
                a = recv(i, arr, sizeof(arr), 0);
                if(a < 0 )
                {
                    ERR_MSG("recv");
                    return -1;
                }
                if(a == 0)
                {
                    printf("对端关闭发送结束\n");
                    break;
                }
                printf("接收成功\n");
                printf("服务器信息:%s\n", arr);
            }
        }
    }
    close(sfd);
    return 0;
}
现象

服务器

#include <func.h>
#define PORT 6666
#define IP "192.168.124.42"
int main(int argc, const char *argv[])
{
    //创建流式套接  
    int sfd = socket(AF_INET, SOCK_STREAM, 0);

    if(sfd < 0)
    {
        ERR_MSG("sfd");
        return -1;
    }
    printf("socket create succes sfd=%d __%d__\n", sfd, __LINE__);
 //允许端口快速被复用---监测端口号是否真的有进程在占用,如果没有,则快速复用
    int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速复用成功\n");

    //定义并填充地址信息结构体,给服务器绑定
    //真实的地址信息结构体根据地址族指定 AF_INET :man ip(7)
    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;
    sin.sin_port        = htons(PORT);      //端口号的网络字节序 : 1024-49151
    sin.sin_addr.s_addr = inet_addr(IP);    //本机号的网络字节序列


    //绑定服务器自身的地址信息----必须绑定
    if(bind(sfd,(struct sockaddr*) &sin, sizeof(sin)) <0)
    {
        ERR_MSG("bind");
        return -1;
    }
    //将套接字设置为被动监听状态
    if(listen(sfd, 128) <0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen succress __%d__\n", __LINE__);

        //创建一个集合
    //sf_set 是个数据结构  清空
    fd_set readfds, tempfds;
    FD_ZERO(&readfds);

    //加入文件描述符 到数组进行监听
    FD_SET(0, &readfds);
    FD_SET(sfd, &readfds);

    int maxfd = sfd; //最大的文件描述符
    ssize_t a = 0;
    ssize_t b = 0;
    char arr[128] = "";
    char crr[128] = "";
    struct sockaddr_in cin; //接收客户端的地址信息

    socklen_t addrlen = sizeof(cin);

    while(1)
    {
        tempfds = readfds;
        int s_ser = select(maxfd+1, &tempfds, NULL,NULL,NULL);
        if(s_ser<0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if(s_ser == 0)
        {
            printf("无数据超时\n");
            return -1;
        }
        for(int i = 0; i <= maxfd; i++)
        {
            if(FD_ISSET(i, &tempfds) == 0)
            {
                continue;
            }
            if(i == 0)
            {
                printf("键盘输入响应\n");
                printf("请输入----->");
                fgets(arr, sizeof(arr),stdin);
                printf("arr= %s\n", arr);
            }
            else if(i == sfd)
            {
                //获取连接成功的客户端信息,生成一个新的套接字
                //accept(sfd, NULL, NULL);  不接收客户端信息
                int newfd = accept(sfd,(struct sockaddr *)&cin, &addrlen);
                if(newfd < 0)
                {
                    ERR_MSG("accept");
                    return -1;
                }
                printf("newfd= %d 客户端连接成功 __%d__\n",newfd,__LINE__);
                printf("端口 = %s用户端%d\n",inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
                FD_SET(newfd, &readfds);
                //更新下标
                maxfd = maxfd > newfd? maxfd:newfd;
            }
            else
            {
                printf("接受数据段响应\n");
                bzero(crr, sizeof(crr));
                a = recv(i, crr, sizeof(crr), 0);
                if(a < 0 )
                {
                    ERR_MSG("recv");
                    return -1;
                }
                if(a == 0)
                {
                    printf("对端关闭发送结束\n");
                    close(i);
                    FD_CLR(i, &readfds);                                                                                                        
                    while(FD_ISSET(maxfd,&readfds) == 0&& maxfd-->0);
                    printf("端口 = %s用户端%d\n",inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
                    break;
                }
                printf("对端成功接收\n");
                printf("端口 = %s用户端%d\n",inet_ntoa(cin.sin_addr), ntohs(cin.sin_port));
                printf("客户端信息:%s\n", arr);
                //向客户端发送数据
                printf("请输入需要发送的数据\n");
                strcat(crr, "<_>");
                b = send(i, crr, sizeof(crr), 0);
                printf("发送成功");

            }

        }
    }

    //关闭文件描述符
    if(close(sfd) < 0)
    {
        ERR_MSG("close");
        return -1;
    }

    return 0;
}
现象

文章来源:https://blog.csdn.net/weixin_64965464/article/details/135228696
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。