Handle udp ports mapped (#534)

* Handle udp ports mapped

* Fix lint error
This commit is contained in:
David Boreham 2023-09-19 13:27:34 -06:00 committed by GitHub
parent e0ecbc3ab3
commit 0b512db693
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -178,16 +178,18 @@ def _get_mapped_ports(stack: str, map_recipe: str):
ports_array = ports[service] ports_array = ports[service]
for x in range(0, len(ports_array)): for x in range(0, len(ports_array)):
orig_port = ports_array[x] orig_port = ports_array[x]
# Strip /udp suffix if present
bare_orig_port = orig_port.replace("/udp", "")
random_port = random.randint(20000, 50000) # Beware: we're relying on luck to not collide random_port = random.randint(20000, 50000) # Beware: we're relying on luck to not collide
if map_recipe == "any-variable-random": if map_recipe == "any-variable-random":
# This is the default so take no action # This is the default so take no action
pass pass
elif map_recipe == "localhost-same": elif map_recipe == "localhost-same":
# Replace instances of "- XX" with "- 127.0.0.1:XX" # Replace instances of "- XX" with "- 127.0.0.1:XX"
ports_array[x] = f"127.0.0.1:{orig_port}:{orig_port}" ports_array[x] = f"127.0.0.1:{bare_orig_port}:{orig_port}"
elif map_recipe == "any-same": elif map_recipe == "any-same":
# Replace instances of "- XX" with "- 0.0.0.0:XX" # Replace instances of "- XX" with "- 0.0.0.0:XX"
ports_array[x] = f"0.0.0.0:{orig_port}:{orig_port}" ports_array[x] = f"0.0.0.0:{bare_orig_port}:{orig_port}"
elif map_recipe == "localhost-fixed-random": elif map_recipe == "localhost-fixed-random":
# Replace instances of "- XX" with "- 127.0.0.1:<rnd>:XX" # Replace instances of "- XX" with "- 127.0.0.1:<rnd>:XX"
ports_array[x] = f"127.0.0.1:{random_port}:{orig_port}" ports_array[x] = f"127.0.0.1:{random_port}:{orig_port}"