ZBLOG

while读取数据的循环里面,你错误处理后break不就退出这个线程了嘛?那这个交互也就没有了,程序就没有用了,如何把错误处理恢复到正常状态?

你提出的观点非常正确。在读取数据时,如果遇到错误并直接 break 出去,确实会导致线程终止,从而停止与 TUN 设备的交互。为了处理错误并尝试恢复正常状态,可以考虑以下几种方法:

  1. 使用循环重试机制:在读取失败时,等待一段时间后再进行重试,而不是直接退出。

  2. 记录错误类型并采取适当行动:根据不同类型的错误(例如可恢复的错误和不可恢复的错误)来决定是继续尝试还是退出。

下面是修改后的示例代码,以实现更好的错误处理和恢复机制:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <pthread.h>
#include <errno.h>
#include <linux/if_tun.h>

// 定义缓冲区大小
#define BUF_SIZE 2048

// 函数用于设置 TUN 接口
int tun_alloc(char *dev) {
    struct ifreq ifr;
    int fd, err;

    // 打开 /dev/net/tun 设备
    fd = open("/dev/net/tun", O_RDWR);
    if (fd < 0) {
        perror("Opening /dev/net/tun");
        return fd;
    }

    memset(&ifr, 0, sizeof(ifr));
    
    // 设置为 TUN 接口
    ifr.ifr_flags = IFF_TUN | IFF_NO_PI; 

    // 如果指定了 dev 名称,则设置它
    if (*dev && (strlen(dev) > IFNAMSIZ)) {
        close(fd);
        errno = EINVAL;
        return -1;
    }
    
    strncpy(ifr.ifr_name, dev, IFNAMSIZ);

    // 创建 TUN 接口
    err = ioctl(fd, TUNSETIFF, (void *) &ifr);
    if (err < 0) {
        perror("ioctl(TUNSETIFF)");
        close(fd);
        return err;
    }

    return fd; // 返回文件描述符
}

// 读取数据的线程函数
void *read_tun(void *arg) {
    int tun_fd = *(int *)arg;
    char buffer[BUF_SIZE];

    while (1) {
        int nread = read(tun_fd, buffer, sizeof(buffer));

        if (nread < 0) {
            // 根据 errno 判断具体的错误情况
            if (errno == EINTR) {
                // 被信号中断,继续下一次读取尝试
                continue;
            } else if (errno == EAGAIN || errno == EWOULDBLOCK) {
                // 非阻塞模式下无数据可读,稍等再尝试
                usleep(100000); // 等待100毫秒再重试(可调整)
                continue;
            } else {
                perror("Read error");
                break; // 对于其他致命错误,退出循环
            }
        } else if (nread == 0) {
            printf("EOF reached\n");
            break; 
        } else {
            // 在此处理读取到的数据,例如打印出来或者转发等
            printf("Read %d bytes\n", nread);
            // 此处可以根据需要处理数据...
        }
        
        // 可以添加其他逻辑,比如睡眠或检查退出条件等。
   }

   return NULL;
}

int main() {
   char *dev = "tun0"; 

   int tun_fd = tun_alloc(dev);

   if (tun_fd < 0) {
       fprintf(stderr, "Error creating TUN interface\n");
       exit(1);
   }

   pthread_t read_thread;

   // 创建线程来读取 TUN 数据包
   if(pthread_create(&read_thread, NULL, read_tun, &tun_fd)) {
       perror("Failed to create thread");
       close(tun_fd);
       exit(1);
   }

   pthread_join(read_thread, NULL); // 等待读线程结束

   close(tun_fd); // 清理资源

   return 0;
}

修改要点:

  • 使用 continue 来跳过当前迭代,继续等待下一个 read 调用。
  • 针对特定的可恢复性错误(如 EINTR, EAGAIN),做出适当反应。比如,对于非阻塞模式下没有数据可读,可以选择稍微休眠一下,然后再次尝试。
  • 对于无法恢复的致命错误,我们依然会选择退出。

这种方式使得程序在面对临时问题时能够自我修复,而不至于完全停止运行。


内容由零声教学AI助手提供,问题来源于学员提问

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:https://sdn.0voice.com/?id=7504

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?