From 5ee76cbbef52097723419949bf732aa91986bdaa Mon Sep 17 00:00:00 2001 From: John-Mark Gurney Date: Sat, 13 Jun 2020 22:02:31 -0700 Subject: [PATCH] fix issue where tcp default string wasn't supported.. --- ntunnel/__init__.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/ntunnel/__init__.py b/ntunnel/__init__.py index 3e51ff6..54de11b 100644 --- a/ntunnel/__init__.py +++ b/ntunnel/__init__.py @@ -227,11 +227,9 @@ def parsesockstr(sockstr): if '=' not in rem: if proto == 'unix' and rem[0] != '/': raise ValueError('bare path MUST start w/ a slash (/).') - - if proto == 'unix': + elif proto == 'unix': args = { 'path': rem } - - if proto == 'udp': + elif proto in ('tcp', 'udp'): h, p = rem.split(':') args = { 'host': h, 'port': p } else: @@ -557,6 +555,8 @@ class Tests_misc(unittest.TestCase): # function 'unix:/apath': ('unix', { 'path': '/apath' }), 'unix:path=apath': ('unix', { 'path': 'apath' }), + 'tcp:ahost:1234': ('tcp', { 'host': 'ahost', + 'port': 1234 }), 'tcp:host=apath': ('tcp', { 'host': 'apath' }), 'tcp:host=apath,port=5': ('tcp', { 'host': 'apath', 'port': 5 }),