修复rtsp组播遍历迭代器导致崩溃的bug

2023-08-26 11:34:22 +08:00 · 2023-08-26 11:34:22 +08:00 · cb4ab21548
parent f3f4b49332
commit cb4ab21548
1 changed files with 3 additions and 1 deletions
--- a/src/Rtsp/UDPServer.cpp
+++ b/src/Rtsp/UDPServer.cpp
@ -82,10 +82,12 @@ void UDPServer::onRecv(int interleaved, const Buffer::Ptr &buf, struct sockaddr*
        return;
    }
    auto &ref = it0->second;
-    for (auto it1 = ref.begin(); it1 != ref.end(); ++it1) {
+    for (auto it1 = ref.begin(); it1 != ref.end();) {
        auto &func = it1->second;
        if (!func(interleaved, buf, peer_addr)) {
            it1 = ref.erase(it1);
+        } else {
+            ++it1;
        }
    }
    if (ref.size() == 0) {