From 6756e922e9fe5e8f678d3ccaefadb27c23f400ae Mon Sep 17 00:00:00 2001 From: "Kali Kaneko (leap communications)" Date: Tue, 18 Apr 2017 22:30:41 +0200 Subject: [bug] fail if tried to stop vpn when not running --- src/leap/bitmask/cli/bitmask_cli.py | 5 +++-- src/leap/bitmask/vpn/privilege.py | 3 +-- src/leap/bitmask/vpn/service.py | 2 ++ 3 files changed, 6 insertions(+), 4 deletions(-) (limited to 'src/leap') diff --git a/src/leap/bitmask/cli/bitmask_cli.py b/src/leap/bitmask/cli/bitmask_cli.py index 839b7f5c..d34a7768 100755 --- a/src/leap/bitmask/cli/bitmask_cli.py +++ b/src/leap/bitmask/cli/bitmask_cli.py @@ -161,9 +161,10 @@ def execute(): try: yield cli.execute(args) - if 'start' in sys.argv or 'restart' in sys.argv: + cmdline = ' '.join(sys.argv) + if 'ctl start' in cmdline or 'ctl restart' in cmdline: command.default_dict_printer({'start': 'ok'}) - except Exception, e: + except Exception as e: print(Fore.RED + "ERROR: " + Fore.RESET + "%s" % str(e)) if not hasattr(e, 'expected'): diff --git a/src/leap/bitmask/vpn/privilege.py b/src/leap/bitmask/vpn/privilege.py index 3c66c7a5..135acbb8 100644 --- a/src/leap/bitmask/vpn/privilege.py +++ b/src/leap/bitmask/vpn/privilege.py @@ -51,11 +51,10 @@ def uninstall_helpers(): class NoPolkitAuthAgentAvailable(Exception): message = 'No polkit authentication agent available. Please run one.' - pass class NoPkexecAvailable(Exception): - pass + message = 'Could not find pkexec in the system' def is_missing_policy_permissions(): diff --git a/src/leap/bitmask/vpn/service.py b/src/leap/bitmask/vpn/service.py index eefccad1..2766b8c0 100644 --- a/src/leap/bitmask/vpn/service.py +++ b/src/leap/bitmask/vpn/service.py @@ -88,6 +88,8 @@ class VPNService(HookableService): self._vpn.stop() self._started = False return {'result': 'stopped'} + else: + raise Exception('VPN was not running') def do_status(self): status = { -- cgit v1.2.3