Chromium Code Reviews| Index: test/connect-vpn |
| diff --git a/test/connect-vpn b/test/connect-vpn |
| index 2055cc0bd7834bf7ff26df9200f3a35cba180d23..ead5bc8522d9e8f17fbe073b6b6c1ec0ad43dae9 100755 |
| --- a/test/connect-vpn |
| +++ b/test/connect-vpn |
| @@ -7,34 +7,41 @@ import optparse |
| import pprint, sys |
| import dbus, flimflam |
| +def show_usage(parser, vpn_type): |
| + parser.error("Incorrect number of parameters provided for %s" % vpn_type) |
| + |
| def main(argv): |
| - parser = optparse.OptionParser('%prog [options]... (VPN)\n\n' |
| - ' VPN := openvpn <vpn-name> ' |
| - '<remote-host-ip> <vpn-domain>\n ' |
| - ' <ca-file> <client-cert-file> ' |
| - '<client-key-file>') |
| - parser.add_option('--verbose', |
| - dest = 'verbose', |
| - action = 'store_true', |
| + parser = optparse.OptionParser( |
| + "%prog [options]... (VPN)\n" |
| + "\n" |
| + " VPN := openvpn <vpn-name> <remote-host-ip> <vpn-domain>\n" |
| + " <ca-file> <client-cert-file> <client-key-file>\n" |
| + " | l2tpipsec <vpn-name> <remote-host-ip> <vpn-domain>\n" |
| + " <ca-file> <client-cert-file> <client-key-file>\n" |
| + " <ipsec-psk> <username> <password>") |
| + parser.add_option("--verbose", |
| + dest = "verbose", |
| + action = "store_true", |
| default = False, |
| help = "Output diagnostic information during run.") |
| - parser.add_option('--complzo', |
| - dest = 'complzo', |
| - action = 'store_true', |
| + parser.add_option("--complzo", |
| + dest = "complzo", |
| + action = "store_true", |
| default = True, |
| - help = "Enables the OpenVPN option 'complzo' (default).") |
| - parser.add_option('--no-complzo', |
| - dest = 'complzo', |
| - action = 'store_false', |
| + help = ("Enables the OpenVPN option 'complzo' " |
| + "(default)")) |
| + parser.add_option("--no-complzo", |
| + dest = "complzo", |
| + action = "store_false", |
| help = "Disables the OpenVPN option 'complzo'.") |
| - parser.add_option('--remote-cert-tls', |
| - dest = 'remote_cert_tls', |
| - action = 'store', |
| - default = 'server', |
| - type = 'string', |
| - metavar = '(server | client | none)', |
| - help = "This is passed through to OpenVPN when " |
| - "not 'none'.") |
| + parser.add_option("--remote-cert-tls", |
| + dest = "remote_cert_tls", |
| + action = "store", |
| + default = "server", |
| + type = "string", |
| + metavar = "(server | client | none)", |
| + help = ("This is passed through to OpenVPN when " |
| + "not 'none'.")) |
| (options, args) = parser.parse_args(argv[1:]) |
| @@ -50,20 +57,32 @@ def main(argv): |
| params["OpenVPN.Cert"] = args[5] |
| params["OpenVPN.Key"] = args[6] |
| - if options.complzo: # 'complzo' can only be enabled. |
| - params["OpenVPN.CompLZO"] = 'true' |
| + if options.complzo: # "complzo" can only be enabled. |
|
thutt
2011/03/17 15:02:21
I think the single quotes on complzo here are ok,
|
| + params["OpenVPN.CompLZO"] = "true" |
| - if options.remote_cert_tls != 'server' and \ |
| - options.remote_cert_tls != 'client' and \ |
| - options.remote_cert_tls != 'none': |
| - print "\n--remote-cert-tls argument ('%s') " \ |
| - "is invalid.\n" % options.remote_cert_tls |
| + if options.remote_cert_tls != "server" and \ |
| + options.remote_cert_tls != "client" and \ |
| + options.remote_cert_tls != "none": |
| + print ("\n--remote-cert-tls argument ('%s') " |
|
thutt
2011/03/17 15:02:21
Would it be better to remove the space preceding '
|
| + "is invalid.\n" % options.remote_cert_tls) |
| sys.exit(1) |
| params["OpenVPN.RemoteCertTLS"] = options.remote_cert_tls |
| else: |
| - parser.error("Incorrect number of parameters " |
| - "provided for 'openvpn'") |
| + show_usage(parser, params["Type"]) |
|
thutt
2011/03/17 15:02:21
I look forward to seeing what affect this has on t
|
| + elif params["Type"] == "l2tpipsec": |
|
thutt
2011/03/17 15:02:21
Just noticed that it looks like your editor is put
|
| + if len(args) == 10: |
| + params["Name"] = args[1] |
| + params["Host"] = args[2] |
| + params["VPN.Domain"] = args[3] |
| + params["L2TPIPsec.CACert"] = args[4] |
| + params["L2TPIPsec.Cert"] = args[5] |
| + params["L2TPIPsec.Key"] = args[6] |
| + params["L2TPIPsec.PSK"] = args[7] |
| + params["L2TPIPsec.User"] = args[8] |
| + params["L2TPIPsec.Password"] = args[9] |
| + else: |
| + show_usage(parser, params["Type"]) |
| else: |
| print "Unknown VPN type: '%s'" % params["Type"] |
|
thutt
2011/03/17 15:02:21
Should this be replaced with a show_usage() like a
|
| sys.exit(1) |
| @@ -77,13 +96,13 @@ def main(argv): |
| flim = flimflam.FlimFlam(dbus.SystemBus()) |
| service = flim.GetVPNService(params) |
| - if options.verbose == 'true': |
| + if options.verbose == "true": |
| print "VPN is %s, connecting..." % service.object_path |
| - (success, diagnostics) = flim.ConnectService(service_type = 'vpn', |
| + (success, diagnostics) = flim.ConnectService(service_type = "vpn", |
| service = service) |
| if not success or options.verbose: |
| - print 'Success:', success |
| + print "Success:", success |
| pprint.pprint(diagnostics) |
| else: |
| parser.print_help() |