Modified parsing for ipv6 support

This commit is contained in:
Algervivien 2011-06-27 09:45:49 +02:00
parent 8ae73adbb7
commit 8d71c243b1
1 changed files with 2 additions and 2 deletions

View File

@ -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)