Modified parsing for ipv6 support
This commit is contained in:
+2
-2
@@ -276,7 +276,7 @@ if __name__ == '__main__':
|
||||
|
||||
# Parse host:port and convert ports to numbers
|
||||
if args[0].count(':') > 0:
|
||||
opts.listen_host, opts.listen_port = args[0].split(':')
|
||||
opts.listen_host, opts.listen_port = ":".join(args[0].split(':')[:len(args[0].split(':')) - 1]), args[0].split(':')[len(args[0].split(':')) - 1]
|
||||
else:
|
||||
opts.listen_host, opts.listen_port = '', args[0]
|
||||
try: opts.listen_port = int(opts.listen_port)
|
||||
@@ -287,7 +287,7 @@ if __name__ == '__main__':
|
||||
opts.target_port = None
|
||||
else:
|
||||
if args[1].count(':') > 0:
|
||||
opts.target_host, opts.target_port = args[1].split(':')
|
||||
opts.target_host, opts.target_port = ":".join(args[1].split(':')[:len(args[1].split(':')) - 1]), args[1].split(':')[len(args[1].split(':')) - 1]
|
||||
else:
|
||||
parser.error("Error parsing target")
|
||||
try: opts.target_port = int(opts.target_port)
|
||||
|
||||
Reference in New Issue
Block a user