unp_day05

Shelton_z

已于 2023-04-25 07:19:44 修改

阅读量110

点赞数

文章标签：网络服务器

于 2023-04-25 06:42:52 首次发布

本文链接：https://blog.csdn.net/Shelton_z/article/details/130356132

版权

作业1

题目

TCP服务器 select 实现

代码

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

#define IP "192.168.1.74"
#define PORT 8888

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

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");

    // 填充服务器的地址信息结构体
    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) {
        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, tempfds;

    // 清空
    FD_ZERO(&readfds);

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

    int maxfd = sfd;  // 存储集合中最大的文件描述符

    struct sockaddr_in saveCin[1024 - 4];  // 另存连接成功的客户端的地址信息
    struct sockaddr_in cin;                // 存储连接成功的客户端地址信息
    socklen_t addrlen = sizeof(cin);
    int newfd = -1;
    char buf[128] = "";
    ssize_t res = 0;
    int s_res = 0;

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

        for (int i = 0; i <= maxfd; i++) {
            // 判断当前文件描述符是否在读集合中
            if (FD_ISSET(i, &tempfds) == 0)
                continue;

            // 键盘输入事件
            if (0 == i) {
                printf("触发键盘输入事件>>> ");
                fflush(stdout);

                int sndfd;

                res = scanf("%d %s", &sndfd, buf);
                while (getchar() != '\n')
                    ;
                if (res != 2) {
                    fprintf(stderr, "请输入正确格式：fd string\n");
                    continue;
                }

                // 若输入的sndfd非法，或者根本没有对应的客户端，则不发送
                if (sndfd <= sfd || sndfd >= 1024 || !FD_ISSET(sndfd, &readfds)) {
                    fprintf(stderr, "sndfd=%d 是非法的文件描述符\n", sndfd);
                    continue;
                }

                if (send(sndfd, buf, sizeof(buf), 0) < 0) {
                    ERR_MSG("send");
                    return -1;
                }
                printf("send to %d success\n", sndfd);

            } 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添加到readfds集合中
                FD_SET(newfd, &readfds);

                // 更新maxfd
                maxfd = maxfd > newfd ? maxfd : newfd;

            } else {
                fprintf(stderr, "触发客户端交互事件>>> ");
                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(saveCin[i - 4].sin_addr),
                           ntohs(saveCin[i - 4].sin_port), i, __LINE__);

                    // 关闭文件描述符
                    close(i);

                    // 将该文件描述符从集合中剔除
                    FD_CLR(i, &readfds);

                    // 更新maxfd：从大到小遍历集合中最大的文件描述符，更新给maxfd
                    int j = maxfd;
                    for (; j >= 0; j--) {
                        if (FD_ISSET(j, &readfds)) {
                            break;
                        }
                    }
                    maxfd = j;

                    // 当前客户端下线后，继续遍历后面的文件描述符是否产生事件，
                    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__);

                // 发送
                strcat(buf, "*_*");
                if (send(i, buf, sizeof(buf), 0) < 0) {
                    ERR_MSG("send");
                    return -1;
                }
                printf("发送成功\n");
            }
        }
    }

    // 关闭文件描述符
    close(sfd);
    return 0;
}

作业2

题目

TCP客户端 poll 实现

代码

#include <arpa/inet.h>
#include <netinet/in.h>  // sockaddr_in申明
#include <poll.h>
#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>

#define IP "192.168.31.65"  // 本机IP
#define PORT 8888           // 本机端口
#define MAXIMUN_LENGTH 128

#define ERR_MSG(msg)                            \
    {                                           \
        perror(msg);                            \
        printf("error in line %d\n", __LINE__); \
    }

int main(int argc, char const *argv[])
{
    // 1. 创建套接字文件
    int cfd = socket(AF_INET, SOCK_STREAM, 0);

    // 2. 连接服务器ip、端口
    struct sockaddr_in addr_server;
    addr_server.sin_family = AF_INET;             // IPV4协议
    addr_server.sin_addr.s_addr = inet_addr(IP);  // IP地址（网络字节序）
    addr_server.sin_port = htons(PORT);           // 端口（网络字节序）
    connect(cfd, (struct sockaddr *)&addr_server, sizeof(addr_server));
    printf("connect success\n");

    // 3. recv、send消息收发
    struct pollfd fds[2];
    fds[0].fd = 0;
    fds[0].events = POLLIN;

    fds[1].fd = 0;
    fds[1].events = POLLIN;

    char buf[1024] = {0};
    ssize_t res = 0;
    int p_res = -1;
    while (1) {
        p_res = poll(fds, 2, -1);
        if (p_res < 0) {
            ERR_MSG("poll");
            return -1;
        } else if (p_res == 0) {
            printf("time out...\n");
            break;
        }

        //
        if (fds[0].revents & POLLIN) {
            bzero(buf, sizeof(buf));

            res = recv(cfd, buf, sizeof(buf), 0);

            if (res < 0) {
                ERR_MSG("poll");
                return -1;
            } else if (0 == res) {
                printf("服务器下线\n");
                break;
            }
            printf("%s__%d\n", buf, __LINE__);
        }
    }

    // 6. close 关闭套接字文件
    close(cfd);

    return 0;
}

作业3

题目

TCP客户端 select 实现

结果
在这里插入图片描述
代码

#include <arpa/inet.h>
#include <netinet/in.h>  // sockaddr_in申明
#include <poll.h>
#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>

#define IP "192.168.1.74"  // 本机IP
#define PORT 8888          // 本机端口
#define MAXIMUN_LENGTH 128

#define ERR_MSG(msg)                            \
    {                                           \
        perror(msg);                            \
        printf("error in line %d\n", __LINE__); \
    }

int main(int argc, char const *argv[])
{
    // 1. 创建
    int cfd = socket(AF_INET, SOCK_STREAM, 0);

    // 2. 连接
    struct sockaddr_in addr_server;
    addr_server.sin_family = AF_INET;             // IPV4协议
    addr_server.sin_addr.s_addr = inet_addr(IP);  // IP地址（网络字节序）
    addr_server.sin_port = htons(PORT);           // 端口（网络字节序）

    if (connect(cfd, (struct sockaddr *)&addr_server, sizeof(addr_server)) < 0) {
        ERR_MSG("connect");
        return -1;
    }
    printf("连接成功\n");

    // 3. recv、send消息收发
    // 创建读集合
    fd_set readfds, tempfds;
    // 清空
    FD_ZERO(&readfds);
    // 将需要的文件描述符添加到集合中
    FD_SET(0, &readfds);
    FD_SET(cfd, &readfds);
    int maxfd = cfd;  // 存储集合中最大的文件描述符

    struct sockaddr_in cin;  // 存储连接成功的服务端地址信息
    socklen_t addrlen = sizeof(cin);
    char buf[128] = "";
    ssize_t res = 0;
    int s_res = 0;

    while (1) {
        tempfds = readfds;

        s_res = select(maxfd + 1, &tempfds, NULL, NULL, NULL);

        // 错误处理
        if (s_res < 0) {
            ERR_MSG("select");
            return -1;
        } else if (0 == s_res) {
            printf("time out...\n");
            break;
        }
        printf("__%d__\n", __LINE__);

        if (FD_ISSET(cfd, &tempfds)) {  // 接收

            printf("服务器交互事件>>> ");
            fflush(stdout);

            bzero(buf, sizeof(buf));
            res = recv(cfd, buf, sizeof(buf), 0);

            if (res < 0) {
                ERR_MSG("recv");
                return -1;
            } else if (0 == res) {
                printf("连接结束\n");
                break;
            }
            printf("%s__%d\n", buf, __LINE__);
        }

        if (FD_ISSET(STDIN_FILENO, &tempfds)) {  // 键盘输入

            printf("键盘输入事件>>> ");
            fflush(stdout);

            bzero(buf, sizeof(buf));
            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");
        }
    }

    // 关闭
    close(cfd);

    return 0;
}