summaryrefslogtreecommitdiff
path: root/docker/utils
diff options
context:
space:
mode:
authorMark Adams <mark@markadams.me>2015-08-19 21:54:24 -0500
committerMark Adams <mark@markadams.me>2015-08-19 22:01:44 -0500
commitb7aa1d76ebf4704527e73f417a201e0d18becbdc (patch)
treea4e8dffb15d08134cb61e313e1ef995a3e50447e /docker/utils
parenta6064e356e1c50f501c18a6b6ba075b88d6565a4 (diff)
downloaddocker-py-b7aa1d76ebf4704527e73f417a201e0d18becbdc.tar.gz
Fixed #726 issue where split_port was checking `len(None)`
- Fixed #726 by adding a check for None in split_port - Also switched to using a _raise_invalid_port() function to replace multiple slightly differing ValueErrors for invalid port
Diffstat (limited to 'docker/utils')
-rw-r--r--docker/utils/ports/ports.py20
1 files changed, 14 insertions, 6 deletions
diff --git a/docker/utils/ports/ports.py b/docker/utils/ports/ports.py
index 6a0a862..326ef94 100644
--- a/docker/utils/ports/ports.py
+++ b/docker/utils/ports/ports.py
@@ -1,5 +1,4 @@
-
def add_port_mapping(port_bindings, internal_port, external):
if internal_port in port_bindings:
port_bindings[internal_port].append(external)
@@ -33,9 +32,8 @@ def to_port_range(port):
if "/" in port:
parts = port.split("/")
if len(parts) != 2:
- raise ValueError('Invalid port "%s", should be '
- '[[remote_ip:]remote_port[-remote_port]:]'
- 'port[/protocol]' % port)
+ _raise_invalid_port(port)
+
port, protocol = parts
protocol = "/" + protocol
@@ -52,11 +50,17 @@ def to_port_range(port):
'port or startport-endport' % port)
+def _raise_invalid_port(port):
+ raise ValueError('Invalid port "%s", should be '
+ '[[remote_ip:]remote_port[-remote_port]:]'
+ 'port[/protocol]' % port)
+
+
def split_port(port):
parts = str(port).split(':')
+
if not 1 <= len(parts) <= 3:
- raise ValueError('Invalid port "%s", should be '
- '[[remote_ip:]remote_port:]port[/protocol]' % port)
+ _raise_invalid_port(port)
if len(parts) == 1:
internal_port, = parts
@@ -66,6 +70,10 @@ def split_port(port):
internal_range = to_port_range(internal_port)
external_range = to_port_range(external_port)
+
+ if internal_range is None or external_range is None:
+ _raise_invalid_port(port)
+
if len(internal_range) != len(external_range):
raise ValueError('Port ranges don\'t match in length')