C语言高手参考手册：网络编程高级话题与技术细节-CSDN博客

本文链接：https://blog.csdn.net/suifengme/article/details/141271776

在这里插入图片描述

在上一篇文章中，我们介绍了基本的网络编程概念和操作。本文将深入探讨网络编程的一些高级话题和技术细节，包括错误处理、非阻塞I/O、多路复用（select/poll/epoll）、套接字选项以及安全编程等。

1. 错误处理

1.1 错误码

在处理网络编程中的错误时，通常需要检查函数的返回值，并利用 errno 获取具体的错误原因。

1.2 示例代码

#include <sys/socket.h>
#include <errno.h>
#include <stdio.h>

int main() {
    int sockfd = socket(AF_INET, SOCK_STREAM, 0);
    if (sockfd == -1) {
        perror("Failed to create socket");
        return EXIT_FAILURE;
    }

    // 其他操作...

    // 错误处理
    if (some_function() == -1) {
        perror("Some function failed");
        return EXIT_FAILURE;
    }

    // 关闭Socket
    if (close(sockfd) == -1) {
        perror("Failed to close socket");
        return EXIT_FAILURE;
    }

    return 0;
}

2. 非阻塞I/O

2.1 概念

非阻塞I/O允许程序在没有数据可读或写时立即返回，而不是阻塞等待。

2.2 设置非阻塞I/O

#include <fcntl.h>

void set_nonblocking(int sockfd) {
    int flags = fcntl(sockfd, F_GETFL, 0);
    if (flags == -1) {
        perror("Failed to get flags");
        exit(EXIT_FAILURE);
    }
    flags |= O_NONBLOCK;
    if (fcntl(sockfd, F_SETFL, flags) == -1) {
        perror("Failed to set non-blocking mode");
        exit(EXIT_FAILURE);
    }
}

3. 多路复用

3.1 select函数

select() 是最早的多路复用函数，用于检测一组文件描述符的状态。

3.2 poll函数

poll() 是 select() 的改进版本，没有文件描述符限制，并提供了更多的灵活性。

3.3 epoll函数

epoll() 是 Linux 中最高效的多路复用机制，适用于大量文件描述符的情况。

3.4 示例代码

使用select

#include <sys/time.h>
#include <sys/select.h>
#include <unistd.h>

void handle_events_with_select(int sockfd) {
    fd_set readfds;
    struct timeval timeout;

    FD_ZERO(&readfds);
    FD_SET(sockfd, &readfds);

    timeout.tv_sec = 1;
    timeout.tv_usec = 0;

    int ret = select(sockfd + 1, &readfds, NULL, NULL, &timeout);
    if (ret == -1) {
        perror("Select failed");
        exit(EXIT_FAILURE);
    }
    if (ret > 0 && FD_ISSET(sockfd, &readfds)) {
        // 数据可读
    }
}

使用poll

#include <sys/poll.h>

void handle_events_with_poll(int sockfd) {
    struct pollfd pfd;
    pfd.fd = sockfd;
    pfd.events = POLLIN;

    int ret = poll(&pfd, 1, 1000); // 超时1秒
    if (ret == -1) {
        perror("Poll failed");
        exit(EXIT_FAILURE);
    }
    if (ret > 0 && pfd.revents & POLLIN) {
        // 数据可读
    }
}

使用epoll

#include <sys/epoll.h>

void handle_events_with_epoll(int sockfd) {
    int epfd = epoll_create1(0);
    if (epfd == -1) {
        perror("Failed to create epoll file descriptor");
        exit(EXIT_FAILURE);
    }

    struct epoll_event ev;
    ev.events = EPOLLIN;
    ev.data.fd = sockfd;
    if (epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &ev) == -1) {
        perror("Failed to add file descriptor to epoll");
        exit(EXIT_FAILURE);
    }

    struct epoll_event events[10];
    int num_events = epoll_wait(epfd, events, 10, 1000); // 超时1秒
    if (num_events == -1) {
        perror("Epoll wait failed");
        exit(EXIT_FAILURE);
    }
    for (int i = 0; i < num_events; i++) {
        if (events[i].events & EPOLLIN) {
            // 数据可读
        }
    }

    if (epoll_ctl(epfd, EPOLL_CTL_DEL, sockfd, NULL) == -1) {
        perror("Failed to delete file descriptor from epoll");
        exit(EXIT_FAILURE);
    }

    if (close(epfd) == -1) {
        perror("Failed to close epoll file descriptor");
        exit(EXIT_FAILURE);
    }
}

4. 套接字选项

4.1 SO_REUSEADDR

允许在服务器重启后快速重用地址。

4.2 SO_RCVBUF/SO_SNDBUF

设置接收缓冲区和发送缓冲区的大小。

4.3 示例代码

#include <sys/socket.h>

void set_socket_options(int sockfd) {
    int optval = 1;

    // 设置SO_REUSEADDR选项
    if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)) == -1) {
        perror("Failed to set SO_REUSEADDR option");
        exit(EXIT_FAILURE);
    }

    // 设置接收缓冲区大小
    int rcvbuf_size = 1024 * 1024; // 1MB
    if (setsockopt(sockfd, SOL_SOCKET, SO_RCVBUF, &rcvbuf_size, sizeof(rcvbuf_size)) == -1) {
        perror("Failed to set SO_RCVBUF option");
        exit(EXIT_FAILURE);
    }

    // 设置发送缓冲区大小
    int sndbuf_size = 1024 * 1024; // 1MB
    if (setsockopt(sockfd, SOL_SOCKET, SO_SNDBUF, &sndbuf_size, sizeof(sndbuf_size)) == -1) {
        perror("Failed to set SO_SNDBUF option");
        exit(EXIT_FAILURE);
    }
}