WebRTC-RS项目中UDP套接字泄漏问题的分析与解决

问题背景

在WebRTC-RS项目中，开发者发现当关闭PeerConnection时，UDP套接字资源未能正确释放，导致系统资源泄漏。这个问题在RTSP转WebRTC的代理应用中尤为明显，每次会话结束后都会留下未关闭的套接字。

问题现象

通过系统监控工具可以观察到：

1. 调用PeerConnection.close().await时出现警告日志
2. ICE状态变更日志显示连接状态已变为Closed
3. 但通过lsof命令检查发现UDP套接字仍然存在
4. Tokio控制台显示有2个任务处于"lost their waker"状态

根本原因分析

经过深入排查，发现资源泄漏主要由以下几个因素导致：

1. 候选对象引用计数未清零：Arc<CandidateBase>的引用计数保持在5，无法被自动回收
2. AgentConn结构体未完全清理：
   • AgentConn::checklist未在关闭时清空
   • AgentConn.selected_pair也未在关闭时清空
3. mDNS查询任务未终止：mDNS查询任务在连接关闭后仍然运行
4. DTLS连接任务卡住：DTLS连接的任务在特定条件下会无限期挂起

解决方案

针对上述问题，可以采取以下修复措施：

1. 完善AgentConn清理逻辑：

   • 在关闭操作中显式清空checklist
   • 同时清空selected_pair

2. 优化mDNS处理：

   • 实现mDNS查询任务的主动终止机制
   • 在连接关闭时发送终止信号

3. 修复DTLS连接任务：

   • 检查DTLS状态机实现
   • 确保所有可能的执行路径都能正常退出

4. 引用循环处理：

   • 检查所有Arc使用场景
   • 必要时使用Weak引用来打破潜在的循环引用

经验总结

1. 资源管理：在Rust异步编程中，特别是使用Arc等共享所有权机制时，需要特别注意资源的释放
2. 任务监控：Tokio控制台等工具对于发现挂起任务非常有帮助
3. 引用计数调试：使用Arc::strong_count进行调试是定位资源泄漏的有效手段
4. 关闭顺序：复杂的网络协议栈实现需要注意各层组件的关闭顺序

最佳实践建议

1. 实现全面的连接关闭流程检查表
2. 为关键资源添加析构日志
3. 定期进行资源泄漏测试
4. 考虑实现连接健康检查机制

这个问题展示了在实现复杂网络协议时资源管理的重要性，特别是在异步环境下。通过系统性的分析和修复，可以构建更加健壮的WebRTC实现。