mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-11-06 07:11:13 +02:00
fix: UDP port allocation race causing link self-loop (one-way links)
PortManager.get_free_udp_port had an unguarded find-then-add sequence. A link allocates both ends concurrently (asyncio.gather in UDPLink._prepare -> two POST /ports/udp) and FastAPI runs the sync route handler in a threadpool, so both threads could probe the same 'free' port before either recorded it — handing lport == rport to both ends. uBridge sets SO_REUSEADDR on UDP NIO sockets, so the double bind succeeds silently and the kernel delivers everything to the last-bound socket: one node starves, the other echoes to itself. Make every TCP/UDP allocate/reserve/release path atomic with an RLock, and rebuild _link_data in UDPLink._prepare so reset() commits the fresh port pair instead of re-sending the stale, already-released one. Regression tests: threaded barrier allocation never returns duplicates (red on the old code, UDP and TCP); reset() leaves exactly one mirrored NIO pair per side with lport != rport (red on the old code).
This commit is contained in:
7 files changed
+285
-89
No files matched your search
@@ -16,6 +16,7 @@
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import pytest
|
||||
import threading
|
||||
import uuid
|
||||
|
||||
from fastapi import HTTPException
|
||||
@@ -116,6 +117,82 @@ def test_release_udp_port():
|
||||
pm.reserve_udp_port(20000, project)
|
||||
|
||||
|
||||
def test_concurrent_udp_port_allocation_no_duplicates():
|
||||
"""
|
||||
Regression test for the link UDP self-loop bug (docs/bugs/link-udp-self-loop.md):
|
||||
both ends of a link are allocated concurrently on the controller
|
||||
(asyncio.gather -> two POST /ports/udp), and FastAPI runs the sync route
|
||||
handler in a threadpool. The find-then-add allocation must be atomic,
|
||||
otherwise both threads can probe and return the same "free" port —
|
||||
handing lport == rport to both ends, which makes every packet loop back
|
||||
to its sender (one-way link).
|
||||
"""
|
||||
|
||||
pm = PortManager()
|
||||
pm.udp_port_range = (50000, 50100)
|
||||
project = Project(project_id=str(uuid.uuid4()))
|
||||
|
||||
workers = 8
|
||||
rounds = 10
|
||||
barrier = threading.Barrier(workers)
|
||||
results = []
|
||||
results_lock = threading.Lock()
|
||||
|
||||
def worker():
|
||||
allocated = []
|
||||
for _ in range(rounds):
|
||||
# start each round together to maximize the collision window
|
||||
barrier.wait()
|
||||
allocated.append(pm.get_free_udp_port(project))
|
||||
with results_lock:
|
||||
results.extend(allocated)
|
||||
|
||||
threads = [threading.Thread(target=worker) for _ in range(workers)]
|
||||
for thread in threads:
|
||||
thread.start()
|
||||
for thread in threads:
|
||||
thread.join()
|
||||
|
||||
assert len(results) == workers * rounds
|
||||
assert len(set(results)) == len(results), "the same UDP port was handed to two callers"
|
||||
assert pm.udp_ports == set(results)
|
||||
|
||||
|
||||
def test_concurrent_tcp_port_allocation_no_duplicates():
|
||||
"""
|
||||
Same race class as the UDP self-loop bug, on the console/TCP side:
|
||||
concurrent get_free_tcp_port calls must never return the same port.
|
||||
"""
|
||||
|
||||
pm = PortManager()
|
||||
pm.console_port_range = (51000, 51100)
|
||||
project = Project(project_id=str(uuid.uuid4()))
|
||||
|
||||
workers = 8
|
||||
rounds = 10
|
||||
barrier = threading.Barrier(workers)
|
||||
results = []
|
||||
results_lock = threading.Lock()
|
||||
|
||||
def worker():
|
||||
allocated = []
|
||||
for _ in range(rounds):
|
||||
barrier.wait()
|
||||
allocated.append(pm.get_free_tcp_port(project))
|
||||
with results_lock:
|
||||
results.extend(allocated)
|
||||
|
||||
threads = [threading.Thread(target=worker) for _ in range(workers)]
|
||||
for thread in threads:
|
||||
thread.start()
|
||||
for thread in threads:
|
||||
thread.join()
|
||||
|
||||
assert len(results) == workers * rounds
|
||||
assert len(set(results)) == len(results), "the same TCP port was handed to two callers"
|
||||
assert pm.tcp_ports == set(results)
|
||||
|
||||
|
||||
def test_find_unused_port():
|
||||
|
||||
p = PortManager().find_unused_port(1000, 10000)
|
||||
|
||||
Reference in new issue
Block a user