Handle udp ports mapped (#534)
* Handle udp ports mapped * Fix lint error
This commit is contained in:
parent
e0ecbc3ab3
commit
0b512db693
@ -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}"
|
||||||
|
Loading…
Reference in New Issue
Block a user