网络编程day5-(IO多路复用)

最新推荐文章于 2024-09-10 20:37:53 发布

Bear-ear

最新推荐文章于 2024-09-10 20:37:53 发布

阅读量43

点赞数

文章标签： c语言网络服务器

本文链接：https://blog.csdn.net/designesb/article/details/130116710

版权

该代码示例展示了如何使用select模型创建一个TCP服务器，监听指定端口并处理客户端连接。服务器端通过socket、bind、listen和accept处理连接请求，客户端则通过socket和connect与服务器建立连接。双方都使用select进行I/O多路复用，监听键盘输入和网络数据收发。

摘要由CSDN通过智能技术生成

TCP服务器（select）

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

#define ERR_MSG(msg)                          \
    do                                        \
    {                                         \
        fprintf(stderr, "line:%d", __LINE__); \
        perror(msg);                          \
    } while (0)

#define IP "192.168.0.156" // 本机IP地址
#define PORT 1562          // 自定义端口号1023~49151

char buf[128] = ""; // 用来存客户端发送的信息

int main(int argc, const char *argv[])
{
    // 创建套字节，在内核中创建两个缓冲区，用户空间可以接收这两个文件描述符
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("socket success __%d__\n", __LINE__);

    // 允许端口快速重启
    int reuse = 1;
    if ((setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse))) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速重启成功");

    // 填充地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;            // 必须填写AF_INET
    sin.sin_port = htons(PORT);          // 端口号1023~49151 要转为网络字节序
    sin.sin_addr.s_addr = inet_addr(IP); // 本机IP地址 要转为网络字节序

    // 将IP和端口绑定到套字节上
    if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success__%d__\n", __LINE__);

    // 将套节字设为监听状态，监听是否有客户端链接成功
    if (listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success__%d__\n", __LINE__);

    // 设置一个集合
    fd_set readfds, tmpfds; // vi -t fd_set

    //由于readfds中需要防止要监测的文件描述符，所以不能让他是随机值
    //所以需要将readfds清空
    FD_ZERO(&readfds);
    FD_ZERO(&tmpfds);

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

    int s_res = 0;
    char buf[128] = "";
    int newfd = -1;
    ssize_t res = 0;
    int maxfd = sfd;  //最大文件描述符

    struct sockaddr_in cin;          // 存储链接成功的客户端信息
    socklen_t addrlen = sizeof(cin); // 计算结构体信息大小

    struct sockaddr_in saveCin[1024-4]; //另存客户端地址信息，0，1，2，sfd不可能有对应的客户端


    while (1)
    {
        tmpfds = readfds;
        s_res = select(maxfd + 1, &tmpfds, NULL, NULL, NULL);
        if (s_res < 0)
        {
            ERR_MSG("select");
            return -1;
        }
        else if (0 == s_res)
        {
            printf("time out......\n");
            return -1;
        }

        // 能运行到当前位置，则代表有文件描述符准备就绪
        // 走触发事件的文件描述符对应的处理函数
        // 当集合中有文件描述符准备就绪了，集合中会只保留产生事件的文件描述符
        // 当0准备就绪，集合中会只保留0
        // 当sfd准备就绪，集合中会只保留sfd
        // 当o和sfd都准备就绪，集合中会保留o和sfd
        // 所以只需要判断集合中剩下哪个文件描述符，走对应函数即可
        for (int i = 0; i <= maxfd; i++)
        {
            if (!(FD_ISSET(i, &tmpfds))) // 如果不在集合中，则直接往后继续变量
            {
                continue;
            }

            // 能运行到当前位置，则说明i所代表的文件描述符在tmpfds中
            // 要判断i所代表的文件描述符需要走什么对应的函数

            if (0 == i)
            {
                printf("触发键盘输入事件>>>");
                fgets(buf, sizeof(buf), stdin);
                buf[strlen(buf) - 1] = 0;
                printf(":%s\n", buf);
            }
            else if (sfd == i)
            {
                printf("触发客户连接事件>>>");
                fflush(stdout);
                newfd = accept(sfd, (struct sockaddr *)&cin, &addrlen);
                if (newfd < 0)
                {
                    ERR_MSG("accept");
                    return -1;
                }
                printf("[%s:%d] newfd = %d 连接成功__%d__\n",
                       inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);

                saveCin[newfd - 4] = cin;

                // 将newfd添加到读集合中
                FD_SET(newfd, &readfds);

                // 更新maxfd
                maxfd = maxfd > newfd ? maxfd : newfd;
            }
            else
            {
                bzero(buf, sizeof(buf));
                // 接收
                res = recv(i, buf, sizeof(buf), 0);
                if (res < 0)
                {
                    ERR_MSG("recv");
                    return -1;
                }
                else if (0 == res)
                {
                    printf("[%s:%d] newfd=%d 客户端下线__%d__\n",
                           inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);

                    //关闭文件描述符
                    close(i);
                    //从集合中剔除该文件描述符
                    FD_CLR(i, &tmpfds);
                    
                    //更新maxfd
                    //从目前最大的文件描述符往小的判断，直到判断到在集合中
                    //那么该文件描述就是最大的
                    int j = 0;
                    for (j = maxfd; j >= 0; j--)
                    {
                        if (FD_ISSET(j, &readfds))
                        {
                            maxfd = j;
                            break;
                        }
                    }
                    if (j < 0) //若集合中没有要监测的文件描述符了，则让maxfd = -1
                        maxfd = -1;

                    continue;
                }
                printf("[%s:%d] newfd=%d: %s __%d__\n",
                       inet_ntoa(saveCin[i-4].sin_addr), ntohs(saveCin[i-4].sin_port), i, buf, __LINE__);
            }
        }
    }

    // 关闭文件
    close(sfd);
    close(newfd);
    return 0;
}

TCP客户端（select）

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

#define ERR_MSG(msg)                          \
    do                                        \
    {                                         \
        fprintf(stderr, "line:%d", __LINE__); \
        perror(msg);                          \
    } while (0);

#define IP "192.168.0.156" // 要连接的服务器ip
#define PORT 1562          // 端口

int main(int argc, const char *argv[])
{
    // 创建流式套接字
    int cfd = socket(AF_INET, SOCK_STREAM, 0);
    if (cfd < 0)
    {
        ERR_MSG("socket");
        return -1;
    }
    printf("cfd=%d\n", cfd);

    // 允许端口快速重用
    int reuse = 1;
    if (setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }

    printf("允许端口快速重用成功\n");

    // 绑定客户端的地址信息结构体,非必须绑定的
    // 如果不绑定,则有操作系统自动选择一个端口号以及ip绑定到嵌套字上

    // 填冲地址信息结构体
    // 真实的地址信息结构体根据地址族指定 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

    // 连接服务器

    if (connect(cfd, (struct sockaddr *)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("connect");
        return -1;
    }
    printf("connect success __%d__\n", __LINE__);

    //=========================================================
    // 设置一个集合
    fd_set readfds, tmpfds;

    // 清空数据，防止readfds和tempfds是随机值
    FD_ZERO(&readfds);
    FD_ZERO(&tmpfds);

    // 将需要的文件描述符添加到集合中
    FD_SET(0, &readfds);
    FD_SET(cfd, &readfds);

    struct sockaddr_in cin;          // 存储链接成功的客户端信息
    socklen_t addrlen = sizeof(cin); // 计算结构体信息大小
    //=========================================================

    char buf[128] = "";
    ssize_t res = 0;
    int s_res = -1;
    int maxfd = cfd;

    while (1)
    {
        s_res = select(maxfd + 1, &readfds, 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, &readfds))
        {
            bzero(buf, sizeof(buf));
            // 发送
            printf("请输入>>>\n");
            fgets(buf, sizeof(buf), stdin);
            buf[strlen(buf) - 1] = 0;

            if (send(cfd, buf, sizeof(buf), 0) < 0)
            {
                ERR_MSG("send");
                return -1;
            }
            printf("发送成功\n");
        }
        if (FD_ISSET(cfd, &readfds))
        {
            // 接收信息
            res = recv(cfd, buf, sizeof(buf), 0);
            if (res < 0)
            {
                ERR_MSG("recv");
                return -1;
            }
            else if (0 == res)
            {
                printf("cfd=%d 服务器下线 __%d__\n", cfd, __LINE__);
                break;
            }
            printf("cfd=%d : %s __%d__\n", cfd, buf, __LINE__);
        }
    }

    close(cfd);
    return 0;
}