diff options
Diffstat (limited to 'src/leap/bitmask')
| -rw-r--r-- | src/leap/bitmask/cli/mail.py | 21 | ||||
| -rw-r--r-- | src/leap/bitmask/core/configurable.py | 1 | ||||
| -rw-r--r-- | src/leap/bitmask/core/dispatcher.py | 11 | ||||
| -rw-r--r-- | src/leap/bitmask/core/mail_services.py | 11 | ||||
| -rw-r--r-- | src/leap/bitmask/core/service.py | 21 |
5 files changed, 52 insertions, 13 deletions
diff --git a/src/leap/bitmask/cli/mail.py b/src/leap/bitmask/cli/mail.py index 1624606..21b93c2 100644 --- a/src/leap/bitmask/cli/mail.py +++ b/src/leap/bitmask/cli/mail.py @@ -20,8 +20,6 @@ Bitmask Command Line interface: mail import argparse import sys -from colorama import Fore - from leap.bitmask.cli import command @@ -58,3 +56,22 @@ SUBCOMMANDS: self.data += ['status', uid] return self._send(command.print_status) + + def mixnet_status(self, raw_args): + parser = argparse.ArgumentParser( + description='Bitmask mixnet status', + prog='%s %s %s' % tuple(sys.argv[:3])) + parser.add_argument('-u', '--userid', default='', + help='uid to check the status of') + parser.add_argument('address', nargs=1, + help='the recipient address') + subargs = parser.parse_args(raw_args) + + userid = None + if subargs.userid: + userid = subargs.userid + else: + userid = self.cfg.get('bonafide', 'active', default=None) + self.data += ['mixnet_status', userid, subargs.address[0]] + + return self._send(command.default_dict_printer) diff --git a/src/leap/bitmask/core/configurable.py b/src/leap/bitmask/core/configurable.py index 726c541..262f030 100644 --- a/src/leap/bitmask/core/configurable.py +++ b/src/leap/bitmask/core/configurable.py @@ -51,6 +51,7 @@ zmq = True web = True onion = False websockets = False +mixnet = False """ __all__ = ["ConfigurableService", DEFAULT_BASEDIR, MissingConfigEntry] diff --git a/src/leap/bitmask/core/dispatcher.py b/src/leap/bitmask/core/dispatcher.py index 7a2d48a..519ec9d 100644 --- a/src/leap/bitmask/core/dispatcher.py +++ b/src/leap/bitmask/core/dispatcher.py @@ -267,6 +267,17 @@ class MailCmd(SubCommand): d = mail.get_token() return d + @register_method('dict') + def do_MIXNET_STATUS(self, mail, *parts, **kw): + try: + userid = parts[2] + address = parts[3] + except IndexError: + raise DispatchError( + 'wrong number of arguments: expected 2') + d = mail.do_mixnet_status(userid, address) + return d + class WebUICmd(SubCommand): diff --git a/src/leap/bitmask/core/mail_services.py b/src/leap/bitmask/core/mail_services.py index 20d6a34..be84b3c 100644 --- a/src/leap/bitmask/core/mail_services.py +++ b/src/leap/bitmask/core/mail_services.py @@ -469,13 +469,14 @@ class StandardMailService(service.MultiService, HookableService): # TODO factor out Mail Service to inside mail package. - def __init__(self, basedir): + def __init__(self, basedir, mixnet_enabled=False): self._basedir = basedir self._soledad_sessions = {} self._keymanager_sessions = {} self._sendmail_opts = {} self._service_tokens = {} self._active_user = None + self._mixnet_enabled = mixnet_enabled super(StandardMailService, self).__init__() self.initializeChildrenServices() @@ -603,6 +604,14 @@ class StandardMailService(service.MultiService, HookableService): } defer.returnValue(merge_status(childrenStatus)) + def do_mixnet_status(self, userid, address): + # XXX: for now there is no support in the provider + # we'll mock it if it's enabled + status = 'disabled' + if self._mixnet_enabled: + status = 'ok' + return {'status': status} + def get_token(self): active_user = self._active_user if not active_user: diff --git a/src/leap/bitmask/core/service.py b/src/leap/bitmask/core/service.py index 5b4f5f7..e823d82 100644 --- a/src/leap/bitmask/core/service.py +++ b/src/leap/bitmask/core/service.py @@ -83,9 +83,6 @@ class BitmaskBackend(configurable.ConfigurableService): # top of the global app token, this should be removed. self.tokens = {} - def enabled(service): - return self.get_config('services', service, False, boolean=True) - def with_manhole(): user = self.get_config('manhole', 'user', '') passwd = self.get_config('manhole', 'passwd', '') @@ -101,26 +98,29 @@ class BitmaskBackend(configurable.ConfigurableService): on_start(self.init_bonafide) on_start(self.init_sessions) - if enabled('mail'): + if self._enabled('mail'): on_start(self._init_mail_services) - if enabled('vpn'): + if self._enabled('vpn'): on_start(self._init_vpn) - if enabled('zmq'): + if self._enabled('zmq'): on_start(self._init_zmq) - if enabled('web'): - onion = enabled('onion') + if self._enabled('web'): + onion = self._enabled('onion') on_start(self._init_web, onion=onion) - if enabled('websockets'): + if self._enabled('websockets'): on_start(self._init_websockets) manholecfg = with_manhole() if manholecfg: on_start(self._init_manhole, manholecfg) + def _enabled(self, service): + return self.get_config('services', service, False, boolean=True) + def _touch_token_file(self): path = os.path.join(self.basedir, 'authtoken') with open(path, 'w') as f: @@ -203,7 +203,8 @@ class BitmaskBackend(configurable.ConfigurableService): def _init_mail(self): service = mail_services.StandardMailService - self._maybe_init_service('mail', service, self.basedir) + self._maybe_init_service('mail', service, self.basedir, + self._enabled('mixnet')) def _init_vpn(self): if HAS_VPN: |