Workflow项目中获取TCP连接对端地址与端口的方法解析

在基于sogou/workflow项目进行网络编程时，开发者经常需要获取TCP连接的对端地址和端口信息。本文将详细介绍在非HTTP协议场景下如何通过WFNetworkTask获取这些关键网络信息。

核心接口：get_peer_addr()

workflow项目提供了一个简洁高效的接口get_peer_addr()来获取TCP连接的对端信息。这个接口返回的是一个标准的网络地址结构，包含了对端的IP地址和端口号。

使用方法

当使用WFNetworkTask处理TCP连接时，可以通过以下方式获取连接信息：

1. 在任务处理器中，通过任务的get_peer_addr()方法获取地址信息
2. 该方法返回的是sockaddr结构体指针
3. 可以进一步转换为sockaddr_in或sockaddr_in6结构体来解析具体信息

代码示例

void process_task(WFNetworkTask *task) {
    struct sockaddr *peer_addr = task->get_peer_addr();
    if (peer_addr->sa_family == AF_INET) {
        struct sockaddr_in *addr_in = (struct sockaddr_in *)peer_addr;
        char ip[INET_ADDRSTRLEN];
        inet_ntop(AF_INET, &addr_in->sin_addr, ip, sizeof(ip));
        unsigned short port = ntohs(addr_in->sin_port);
        // 使用ip和port...
    } else if (peer_addr->sa_family == AF_INET6) {
        // 处理IPv6地址...
    }
}

注意事项

1. 地址结构体需要根据地址族(AF_INET或AF_INET6)进行正确的类型转换
2. 端口号需要使用ntohs()函数进行字节序转换
3. IP地址需要使用inet_ntop()等函数转换为可读格式
4. 对于IPv6地址，需要使用对应的结构体和处理方式

实际应用场景

获取对端地址和端口信息在以下场景中特别有用：

1. 连接日志记录：记录客户端连接信息用于审计和分析
2. 访问控制：基于IP地址实现白名单/黑名单功能
3. 连接管理：识别和管理特定来源的连接
4. 统计监控：按来源IP进行流量统计和分析

通过workflow提供的这个简单接口，开发者可以轻松实现上述功能，而无需关心底层网络细节，大大提高了开发效率。