/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release

« back to all changes in this revision

Viewing changes to mandos

  • Committer: Teddy Hogeborn
  • Date: 2011-03-08 19:12:56 UTC
  • mfrom: (237.7.20 trunk)
  • Revision ID: teddy@fukt.bsnet.se-20110308191256-mjcukty9on3z0yhd
Merge from trunk.

Show diffs side-by-side

added added

removed removed

Lines of Context:
11
11
# "AvahiService" class, and some lines in "main".
12
12
13
13
# Everything else is
14
 
# Copyright © 2008,2009 Teddy Hogeborn
15
 
# Copyright © 2008,2009 Björn Påhlsson
 
14
# Copyright © 2008-2011 Teddy Hogeborn
 
15
# Copyright © 2008-2011 Björn Påhlsson
16
16
17
17
# This program is free software: you can redistribute it and/or modify
18
18
# it under the terms of the GNU General Public License as published by
31
31
# Contact the authors at <mandos@fukt.bsnet.se>.
32
32
33
33
 
34
 
from __future__ import division, with_statement, absolute_import
 
34
from __future__ import (division, absolute_import, print_function,
 
35
                        unicode_literals)
35
36
 
36
37
import SocketServer as socketserver
37
38
import socket
60
61
import fcntl
61
62
import functools
62
63
import cPickle as pickle
 
64
import multiprocessing
63
65
 
64
66
import dbus
65
67
import dbus.service
80
82
        SO_BINDTODEVICE = None
81
83
 
82
84
 
83
 
version = "1.0.14"
 
85
version = "1.2.3"
84
86
 
85
 
logger = logging.Logger(u'mandos')
 
87
#logger = logging.getLogger('mandos')
 
88
logger = logging.Logger('mandos')
86
89
syslogger = (logging.handlers.SysLogHandler
87
90
             (facility = logging.handlers.SysLogHandler.LOG_DAEMON,
88
 
              address = "/dev/log"))
 
91
              address = str("/dev/log")))
89
92
syslogger.setFormatter(logging.Formatter
90
 
                       (u'Mandos [%(process)d]: %(levelname)s:'
91
 
                        u' %(message)s'))
 
93
                       ('Mandos [%(process)d]: %(levelname)s:'
 
94
                        ' %(message)s'))
92
95
logger.addHandler(syslogger)
93
96
 
94
97
console = logging.StreamHandler()
95
 
console.setFormatter(logging.Formatter(u'%(name)s [%(process)d]:'
96
 
                                       u' %(levelname)s:'
97
 
                                       u' %(message)s'))
 
98
console.setFormatter(logging.Formatter('%(name)s [%(process)d]:'
 
99
                                       ' %(levelname)s:'
 
100
                                       ' %(message)s'))
98
101
logger.addHandler(console)
99
102
 
100
103
class AvahiError(Exception):
117
120
    Attributes:
118
121
    interface: integer; avahi.IF_UNSPEC or an interface index.
119
122
               Used to optionally bind to the specified interface.
120
 
    name: string; Example: u'Mandos'
121
 
    type: string; Example: u'_mandos._tcp'.
 
123
    name: string; Example: 'Mandos'
 
124
    type: string; Example: '_mandos._tcp'.
122
125
                  See <http://www.dns-sd.org/ServiceTypes.html>
123
126
    port: integer; what port to announce
124
127
    TXT: list of strings; TXT record for the service
133
136
    """
134
137
    def __init__(self, interface = avahi.IF_UNSPEC, name = None,
135
138
                 servicetype = None, port = None, TXT = None,
136
 
                 domain = u"", host = u"", max_renames = 32768,
 
139
                 domain = "", host = "", max_renames = 32768,
137
140
                 protocol = avahi.PROTO_UNSPEC, bus = None):
138
141
        self.interface = interface
139
142
        self.name = name
151
154
    def rename(self):
152
155
        """Derived from the Avahi example code"""
153
156
        if self.rename_count >= self.max_renames:
154
 
            logger.critical(u"No suitable Zeroconf service name found"
155
 
                            u" after %i retries, exiting.",
 
157
            logger.critical("No suitable Zeroconf service name found"
 
158
                            " after %i retries, exiting.",
156
159
                            self.rename_count)
157
 
            raise AvahiServiceError(u"Too many renames")
158
 
        self.name = self.server.GetAlternativeServiceName(self.name)
159
 
        logger.info(u"Changing Zeroconf service name to %r ...",
160
 
                    unicode(self.name))
 
160
            raise AvahiServiceError("Too many renames")
 
161
        self.name = unicode(self.server.GetAlternativeServiceName(self.name))
 
162
        logger.info("Changing Zeroconf service name to %r ...",
 
163
                    self.name)
161
164
        syslogger.setFormatter(logging.Formatter
162
 
                               (u'Mandos (%s) [%%(process)d]:'
163
 
                                u' %%(levelname)s: %%(message)s'
 
165
                               ('Mandos (%s) [%%(process)d]:'
 
166
                                ' %%(levelname)s: %%(message)s'
164
167
                                % self.name))
165
168
        self.remove()
166
 
        self.add()
 
169
        try:
 
170
            self.add()
 
171
        except dbus.exceptions.DBusException, error:
 
172
            logger.critical("DBusException: %s", error)
 
173
            self.cleanup()
 
174
            os._exit(1)
167
175
        self.rename_count += 1
168
176
    def remove(self):
169
177
        """Derived from the Avahi example code"""
179
187
            self.group.connect_to_signal('StateChanged',
180
188
                                         self
181
189
                                         .entry_group_state_changed)
182
 
        logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
 
190
        logger.debug("Adding Zeroconf service '%s' of type '%s' ...",
183
191
                     self.name, self.type)
184
192
        self.group.AddService(
185
193
            self.interface,
192
200
        self.group.Commit()
193
201
    def entry_group_state_changed(self, state, error):
194
202
        """Derived from the Avahi example code"""
195
 
        logger.debug(u"Avahi state change: %i", state)
 
203
        logger.debug("Avahi entry group state change: %i", state)
196
204
        
197
205
        if state == avahi.ENTRY_GROUP_ESTABLISHED:
198
 
            logger.debug(u"Zeroconf service established.")
 
206
            logger.debug("Zeroconf service established.")
199
207
        elif state == avahi.ENTRY_GROUP_COLLISION:
200
 
            logger.warning(u"Zeroconf service name collision.")
 
208
            logger.info("Zeroconf service name collision.")
201
209
            self.rename()
202
210
        elif state == avahi.ENTRY_GROUP_FAILURE:
203
 
            logger.critical(u"Avahi: Error in group state changed %s",
 
211
            logger.critical("Avahi: Error in group state changed %s",
204
212
                            unicode(error))
205
 
            raise AvahiGroupError(u"State changed: %s"
 
213
            raise AvahiGroupError("State changed: %s"
206
214
                                  % unicode(error))
207
215
    def cleanup(self):
208
216
        """Derived from the Avahi example code"""
211
219
            self.group = None
212
220
    def server_state_changed(self, state):
213
221
        """Derived from the Avahi example code"""
 
222
        logger.debug("Avahi server state change: %i", state)
214
223
        if state == avahi.SERVER_COLLISION:
215
 
            logger.error(u"Zeroconf server name collision")
 
224
            logger.error("Zeroconf server name collision")
216
225
            self.remove()
217
226
        elif state == avahi.SERVER_RUNNING:
218
227
            self.add()
223
232
                self.bus.get_object(avahi.DBUS_NAME,
224
233
                                    avahi.DBUS_PATH_SERVER),
225
234
                avahi.DBUS_INTERFACE_SERVER)
226
 
        self.server.connect_to_signal(u"StateChanged",
 
235
        self.server.connect_to_signal("StateChanged",
227
236
                                 self.server_state_changed)
228
237
        self.server_state_changed(self.server.GetState())
229
238
 
232
241
    """A representation of a client host served by this server.
233
242
    
234
243
    Attributes:
235
 
    name:       string; from the config file, used in log messages and
236
 
                        D-Bus identifiers
 
244
    _approved:   bool(); 'None' if not yet approved/disapproved
 
245
    approval_delay: datetime.timedelta(); Time to wait for approval
 
246
    approval_duration: datetime.timedelta(); Duration of one approval
 
247
    checker:    subprocess.Popen(); a running checker process used
 
248
                                    to see if the client lives.
 
249
                                    'None' if no process is running.
 
250
    checker_callback_tag: a gobject event source tag, or None
 
251
    checker_command: string; External command which is run to check
 
252
                     if client lives.  %() expansions are done at
 
253
                     runtime with vars(self) as dict, so that for
 
254
                     instance %(name)s can be used in the command.
 
255
    checker_initiator_tag: a gobject event source tag, or None
 
256
    created:    datetime.datetime(); (UTC) object creation
 
257
    current_checker_command: string; current running checker_command
 
258
    disable_hook:  If set, called by disable() as disable_hook(self)
 
259
    disable_initiator_tag: a gobject event source tag, or None
 
260
    enabled:    bool()
237
261
    fingerprint: string (40 or 32 hexadecimal digits); used to
238
262
                 uniquely identify the client
239
 
    secret:     bytestring; sent verbatim (over TLS) to client
240
263
    host:       string; available for use by the checker command
241
 
    created:    datetime.datetime(); (UTC) object creation
 
264
    interval:   datetime.timedelta(); How often to start a new checker
 
265
    last_approval_request: datetime.datetime(); (UTC) or None
 
266
    last_checked_ok: datetime.datetime(); (UTC) or None
242
267
    last_enabled: datetime.datetime(); (UTC)
243
 
    enabled:    bool()
244
 
    last_checked_ok: datetime.datetime(); (UTC) or None
 
268
    name:       string; from the config file, used in log messages and
 
269
                        D-Bus identifiers
 
270
    secret:     bytestring; sent verbatim (over TLS) to client
245
271
    timeout:    datetime.timedelta(); How long from last_checked_ok
246
272
                                      until this client is disabled
247
 
    interval:   datetime.timedelta(); How often to start a new checker
248
 
    disable_hook:  If set, called by disable() as disable_hook(self)
249
 
    checker:    subprocess.Popen(); a running checker process used
250
 
                                    to see if the client lives.
251
 
                                    'None' if no process is running.
252
 
    checker_initiator_tag: a gobject event source tag, or None
253
 
    disable_initiator_tag: - '' -
254
 
    checker_callback_tag:  - '' -
255
 
    checker_command: string; External command which is run to check if
256
 
                     client lives.  %() expansions are done at
257
 
                     runtime with vars(self) as dict, so that for
258
 
                     instance %(name)s can be used in the command.
259
 
    current_checker_command: string; current running checker_command
 
273
    runtime_expansions: Allowed attributes for runtime expansion.
260
274
    """
261
275
    
 
276
    runtime_expansions = ("approval_delay", "approval_duration",
 
277
                          "created", "enabled", "fingerprint",
 
278
                          "host", "interval", "last_checked_ok",
 
279
                          "last_enabled", "name", "timeout")
 
280
    
262
281
    @staticmethod
263
282
    def _timedelta_to_milliseconds(td):
264
283
        "Convert a datetime.timedelta() to milliseconds"
273
292
    def interval_milliseconds(self):
274
293
        "Return the 'interval' attribute in milliseconds"
275
294
        return self._timedelta_to_milliseconds(self.interval)
 
295
 
 
296
    def approval_delay_milliseconds(self):
 
297
        return self._timedelta_to_milliseconds(self.approval_delay)
276
298
    
277
299
    def __init__(self, name = None, disable_hook=None, config=None):
278
300
        """Note: the 'checker' key in 'config' sets the
281
303
        self.name = name
282
304
        if config is None:
283
305
            config = {}
284
 
        logger.debug(u"Creating client %r", self.name)
 
306
        logger.debug("Creating client %r", self.name)
285
307
        # Uppercase and remove spaces from fingerprint for later
286
308
        # comparison purposes with return value from the fingerprint()
287
309
        # function
288
 
        self.fingerprint = (config[u"fingerprint"].upper()
289
 
                            .replace(u" ", u""))
290
 
        logger.debug(u"  Fingerprint: %s", self.fingerprint)
291
 
        if u"secret" in config:
292
 
            self.secret = config[u"secret"].decode(u"base64")
293
 
        elif u"secfile" in config:
 
310
        self.fingerprint = (config["fingerprint"].upper()
 
311
                            .replace(" ", ""))
 
312
        logger.debug("  Fingerprint: %s", self.fingerprint)
 
313
        if "secret" in config:
 
314
            self.secret = config["secret"].decode("base64")
 
315
        elif "secfile" in config:
294
316
            with open(os.path.expanduser(os.path.expandvars
295
 
                                         (config[u"secfile"])),
 
317
                                         (config["secfile"])),
296
318
                      "rb") as secfile:
297
319
                self.secret = secfile.read()
298
320
        else:
299
 
            raise TypeError(u"No secret or secfile for client %s"
 
321
            raise TypeError("No secret or secfile for client %s"
300
322
                            % self.name)
301
 
        self.host = config.get(u"host", u"")
 
323
        self.host = config.get("host", "")
302
324
        self.created = datetime.datetime.utcnow()
303
325
        self.enabled = False
 
326
        self.last_approval_request = None
304
327
        self.last_enabled = None
305
328
        self.last_checked_ok = None
306
 
        self.timeout = string_to_delta(config[u"timeout"])
307
 
        self.interval = string_to_delta(config[u"interval"])
 
329
        self.timeout = string_to_delta(config["timeout"])
 
330
        self.interval = string_to_delta(config["interval"])
308
331
        self.disable_hook = disable_hook
309
332
        self.checker = None
310
333
        self.checker_initiator_tag = None
311
334
        self.disable_initiator_tag = None
312
335
        self.checker_callback_tag = None
313
 
        self.checker_command = config[u"checker"]
 
336
        self.checker_command = config["checker"]
314
337
        self.current_checker_command = None
315
338
        self.last_connect = None
 
339
        self._approved = None
 
340
        self.approved_by_default = config.get("approved_by_default",
 
341
                                              True)
 
342
        self.approvals_pending = 0
 
343
        self.approval_delay = string_to_delta(
 
344
            config["approval_delay"])
 
345
        self.approval_duration = string_to_delta(
 
346
            config["approval_duration"])
 
347
        self.changedstate = multiprocessing_manager.Condition(multiprocessing_manager.Lock())
316
348
    
 
349
    def send_changedstate(self):
 
350
        self.changedstate.acquire()
 
351
        self.changedstate.notify_all()
 
352
        self.changedstate.release()
 
353
        
317
354
    def enable(self):
318
355
        """Start this client's checker and timeout hooks"""
319
 
        if getattr(self, u"enabled", False):
 
356
        if getattr(self, "enabled", False):
320
357
            # Already enabled
321
358
            return
 
359
        self.send_changedstate()
322
360
        self.last_enabled = datetime.datetime.utcnow()
323
361
        # Schedule a new checker to be started an 'interval' from now,
324
362
        # and every interval from then on.
338
376
        if not getattr(self, "enabled", False):
339
377
            return False
340
378
        if not quiet:
341
 
            logger.info(u"Disabling client %s", self.name)
342
 
        if getattr(self, u"disable_initiator_tag", False):
 
379
            self.send_changedstate()
 
380
        if not quiet:
 
381
            logger.info("Disabling client %s", self.name)
 
382
        if getattr(self, "disable_initiator_tag", False):
343
383
            gobject.source_remove(self.disable_initiator_tag)
344
384
            self.disable_initiator_tag = None
345
 
        if getattr(self, u"checker_initiator_tag", False):
 
385
        if getattr(self, "checker_initiator_tag", False):
346
386
            gobject.source_remove(self.checker_initiator_tag)
347
387
            self.checker_initiator_tag = None
348
388
        self.stop_checker()
363
403
        if os.WIFEXITED(condition):
364
404
            exitstatus = os.WEXITSTATUS(condition)
365
405
            if exitstatus == 0:
366
 
                logger.info(u"Checker for %(name)s succeeded",
 
406
                logger.info("Checker for %(name)s succeeded",
367
407
                            vars(self))
368
408
                self.checked_ok()
369
409
            else:
370
 
                logger.info(u"Checker for %(name)s failed",
 
410
                logger.info("Checker for %(name)s failed",
371
411
                            vars(self))
372
412
        else:
373
 
            logger.warning(u"Checker for %(name)s crashed?",
 
413
            logger.warning("Checker for %(name)s crashed?",
374
414
                           vars(self))
375
415
    
376
416
    def checked_ok(self):
385
425
                                      (self.timeout_milliseconds(),
386
426
                                       self.disable))
387
427
    
 
428
    def need_approval(self):
 
429
        self.last_approval_request = datetime.datetime.utcnow()
 
430
    
388
431
    def start_checker(self):
389
432
        """Start a new checker subprocess if one is not running.
390
433
        
408
451
                raise error
409
452
        else:
410
453
            if pid:
411
 
                logger.warning(u"Checker was a zombie")
 
454
                logger.warning("Checker was a zombie")
412
455
                gobject.source_remove(self.checker_callback_tag)
413
456
                self.checker_callback(pid, status,
414
457
                                      self.current_checker_command)
419
462
                command = self.checker_command % self.host
420
463
            except TypeError:
421
464
                # Escape attributes for the shell
422
 
                escaped_attrs = dict((key,
423
 
                                      re.escape(unicode(str(val),
424
 
                                                        errors=
425
 
                                                        u'replace')))
426
 
                                     for key, val in
427
 
                                     vars(self).iteritems())
 
465
                escaped_attrs = dict(
 
466
                    (attr,
 
467
                     re.escape(unicode(str(getattr(self, attr, "")),
 
468
                                       errors=
 
469
                                       'replace')))
 
470
                    for attr in
 
471
                    self.runtime_expansions)
 
472
 
428
473
                try:
429
474
                    command = self.checker_command % escaped_attrs
430
475
                except TypeError, error:
431
 
                    logger.error(u'Could not format string "%s":'
432
 
                                 u' %s', self.checker_command, error)
 
476
                    logger.error('Could not format string "%s":'
 
477
                                 ' %s', self.checker_command, error)
433
478
                    return True # Try again later
434
479
            self.current_checker_command = command
435
480
            try:
436
 
                logger.info(u"Starting checker %r for %s",
 
481
                logger.info("Starting checker %r for %s",
437
482
                            command, self.name)
438
483
                # We don't need to redirect stdout and stderr, since
439
484
                # in normal mode, that is already done by daemon(),
441
486
                # always replaced by /dev/null.)
442
487
                self.checker = subprocess.Popen(command,
443
488
                                                close_fds=True,
444
 
                                                shell=True, cwd=u"/")
 
489
                                                shell=True, cwd="/")
445
490
                self.checker_callback_tag = (gobject.child_watch_add
446
491
                                             (self.checker.pid,
447
492
                                              self.checker_callback,
453
498
                    gobject.source_remove(self.checker_callback_tag)
454
499
                    self.checker_callback(pid, status, command)
455
500
            except OSError, error:
456
 
                logger.error(u"Failed to start subprocess: %s",
 
501
                logger.error("Failed to start subprocess: %s",
457
502
                             error)
458
503
        # Re-run this periodically if run by gobject.timeout_add
459
504
        return True
463
508
        if self.checker_callback_tag:
464
509
            gobject.source_remove(self.checker_callback_tag)
465
510
            self.checker_callback_tag = None
466
 
        if getattr(self, u"checker", None) is None:
 
511
        if getattr(self, "checker", None) is None:
467
512
            return
468
 
        logger.debug(u"Stopping checker for %(name)s", vars(self))
 
513
        logger.debug("Stopping checker for %(name)s", vars(self))
469
514
        try:
470
515
            os.kill(self.checker.pid, signal.SIGTERM)
471
516
            #time.sleep(0.5)
476
521
                raise
477
522
        self.checker = None
478
523
 
479
 
 
480
 
def dbus_service_property(dbus_interface, signature=u"v",
481
 
                          access=u"readwrite", byte_arrays=False):
 
524
def dbus_service_property(dbus_interface, signature="v",
 
525
                          access="readwrite", byte_arrays=False):
482
526
    """Decorators for marking methods of a DBusObjectWithProperties to
483
527
    become properties on the D-Bus.
484
528
    
491
535
    """
492
536
    # Encoding deeply encoded byte arrays is not supported yet by the
493
537
    # "Set" method, so we fail early here:
494
 
    if byte_arrays and signature != u"ay":
495
 
        raise ValueError(u"Byte arrays not supported for non-'ay'"
496
 
                         u" signature %r" % signature)
 
538
    if byte_arrays and signature != "ay":
 
539
        raise ValueError("Byte arrays not supported for non-'ay'"
 
540
                         " signature %r" % signature)
497
541
    def decorator(func):
498
542
        func._dbus_is_property = True
499
543
        func._dbus_interface = dbus_interface
500
544
        func._dbus_signature = signature
501
545
        func._dbus_access = access
502
546
        func._dbus_name = func.__name__
503
 
        if func._dbus_name.endswith(u"_dbus_property"):
 
547
        if func._dbus_name.endswith("_dbus_property"):
504
548
            func._dbus_name = func._dbus_name[:-14]
505
 
        func._dbus_get_args_options = {u'byte_arrays': byte_arrays }
 
549
        func._dbus_get_args_options = {'byte_arrays': byte_arrays }
506
550
        return func
507
551
    return decorator
508
552
 
536
580
    
537
581
    @staticmethod
538
582
    def _is_dbus_property(obj):
539
 
        return getattr(obj, u"_dbus_is_property", False)
 
583
        return getattr(obj, "_dbus_is_property", False)
540
584
    
541
585
    def _get_all_dbus_properties(self):
542
586
        """Returns a generator of (name, attribute) pairs
550
594
        property with the specified name and interface.
551
595
        """
552
596
        for name in (property_name,
553
 
                     property_name + u"_dbus_property"):
 
597
                     property_name + "_dbus_property"):
554
598
            prop = getattr(self, name, None)
555
599
            if (prop is None
556
600
                or not self._is_dbus_property(prop)
560
604
                continue
561
605
            return prop
562
606
        # No such property
563
 
        raise DBusPropertyNotFound(self.dbus_object_path + u":"
564
 
                                   + interface_name + u"."
 
607
        raise DBusPropertyNotFound(self.dbus_object_path + ":"
 
608
                                   + interface_name + "."
565
609
                                   + property_name)
566
610
    
567
 
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"ss",
568
 
                         out_signature=u"v")
 
611
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="ss",
 
612
                         out_signature="v")
569
613
    def Get(self, interface_name, property_name):
570
614
        """Standard D-Bus property Get() method, see D-Bus standard.
571
615
        """
572
616
        prop = self._get_dbus_property(interface_name, property_name)
573
 
        if prop._dbus_access == u"write":
 
617
        if prop._dbus_access == "write":
574
618
            raise DBusPropertyAccessException(property_name)
575
619
        value = prop()
576
 
        if not hasattr(value, u"variant_level"):
 
620
        if not hasattr(value, "variant_level"):
577
621
            return value
578
622
        return type(value)(value, variant_level=value.variant_level+1)
579
623
    
580
 
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"ssv")
 
624
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="ssv")
581
625
    def Set(self, interface_name, property_name, value):
582
626
        """Standard D-Bus property Set() method, see D-Bus standard.
583
627
        """
584
628
        prop = self._get_dbus_property(interface_name, property_name)
585
 
        if prop._dbus_access == u"read":
 
629
        if prop._dbus_access == "read":
586
630
            raise DBusPropertyAccessException(property_name)
587
 
        if prop._dbus_get_args_options[u"byte_arrays"]:
 
631
        if prop._dbus_get_args_options["byte_arrays"]:
588
632
            # The byte_arrays option is not supported yet on
589
633
            # signatures other than "ay".
590
 
            if prop._dbus_signature != u"ay":
 
634
            if prop._dbus_signature != "ay":
591
635
                raise ValueError
592
636
            value = dbus.ByteArray(''.join(unichr(byte)
593
637
                                           for byte in value))
594
638
        prop(value)
595
639
    
596
 
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"s",
597
 
                         out_signature=u"a{sv}")
 
640
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="s",
 
641
                         out_signature="a{sv}")
598
642
    def GetAll(self, interface_name):
599
643
        """Standard D-Bus property GetAll() method, see D-Bus
600
644
        standard.
608
652
                # Interface non-empty but did not match
609
653
                continue
610
654
            # Ignore write-only properties
611
 
            if prop._dbus_access == u"write":
 
655
            if prop._dbus_access == "write":
612
656
                continue
613
657
            value = prop()
614
 
            if not hasattr(value, u"variant_level"):
 
658
            if not hasattr(value, "variant_level"):
615
659
                all[name] = value
616
660
                continue
617
661
            all[name] = type(value)(value, variant_level=
618
662
                                    value.variant_level+1)
619
 
        return dbus.Dictionary(all, signature=u"sv")
 
663
        return dbus.Dictionary(all, signature="sv")
620
664
    
621
665
    @dbus.service.method(dbus.INTROSPECTABLE_IFACE,
622
 
                         out_signature=u"s",
 
666
                         out_signature="s",
623
667
                         path_keyword='object_path',
624
668
                         connection_keyword='connection')
625
669
    def Introspect(self, object_path, connection):
630
674
        try:
631
675
            document = xml.dom.minidom.parseString(xmlstring)
632
676
            def make_tag(document, name, prop):
633
 
                e = document.createElement(u"property")
634
 
                e.setAttribute(u"name", name)
635
 
                e.setAttribute(u"type", prop._dbus_signature)
636
 
                e.setAttribute(u"access", prop._dbus_access)
 
677
                e = document.createElement("property")
 
678
                e.setAttribute("name", name)
 
679
                e.setAttribute("type", prop._dbus_signature)
 
680
                e.setAttribute("access", prop._dbus_access)
637
681
                return e
638
 
            for if_tag in document.getElementsByTagName(u"interface"):
 
682
            for if_tag in document.getElementsByTagName("interface"):
639
683
                for tag in (make_tag(document, name, prop)
640
684
                            for name, prop
641
685
                            in self._get_all_dbus_properties()
642
686
                            if prop._dbus_interface
643
 
                            == if_tag.getAttribute(u"name")):
 
687
                            == if_tag.getAttribute("name")):
644
688
                    if_tag.appendChild(tag)
645
689
                # Add the names to the return values for the
646
690
                # "org.freedesktop.DBus.Properties" methods
647
 
                if (if_tag.getAttribute(u"name")
648
 
                    == u"org.freedesktop.DBus.Properties"):
649
 
                    for cn in if_tag.getElementsByTagName(u"method"):
650
 
                        if cn.getAttribute(u"name") == u"Get":
651
 
                            for arg in cn.getElementsByTagName(u"arg"):
652
 
                                if (arg.getAttribute(u"direction")
653
 
                                    == u"out"):
654
 
                                    arg.setAttribute(u"name", u"value")
655
 
                        elif cn.getAttribute(u"name") == u"GetAll":
656
 
                            for arg in cn.getElementsByTagName(u"arg"):
657
 
                                if (arg.getAttribute(u"direction")
658
 
                                    == u"out"):
659
 
                                    arg.setAttribute(u"name", u"props")
660
 
            xmlstring = document.toxml(u"utf-8")
 
691
                if (if_tag.getAttribute("name")
 
692
                    == "org.freedesktop.DBus.Properties"):
 
693
                    for cn in if_tag.getElementsByTagName("method"):
 
694
                        if cn.getAttribute("name") == "Get":
 
695
                            for arg in cn.getElementsByTagName("arg"):
 
696
                                if (arg.getAttribute("direction")
 
697
                                    == "out"):
 
698
                                    arg.setAttribute("name", "value")
 
699
                        elif cn.getAttribute("name") == "GetAll":
 
700
                            for arg in cn.getElementsByTagName("arg"):
 
701
                                if (arg.getAttribute("direction")
 
702
                                    == "out"):
 
703
                                    arg.setAttribute("name", "props")
 
704
            xmlstring = document.toxml("utf-8")
661
705
            document.unlink()
662
706
        except (AttributeError, xml.dom.DOMException,
663
707
                xml.parsers.expat.ExpatError), error:
664
 
            logger.error(u"Failed to override Introspection method",
 
708
            logger.error("Failed to override Introspection method",
665
709
                         error)
666
710
        return xmlstring
667
711
 
673
717
    dbus_object_path: dbus.ObjectPath
674
718
    bus: dbus.SystemBus()
675
719
    """
 
720
    
 
721
    runtime_expansions = (Client.runtime_expansions
 
722
                          + ("dbus_object_path",))
 
723
    
676
724
    # dbus.service.Object doesn't use super(), so we can't either.
677
725
    
678
726
    def __init__(self, bus = None, *args, **kwargs):
 
727
        self._approvals_pending = 0
679
728
        self.bus = bus
680
729
        Client.__init__(self, *args, **kwargs)
681
730
        # Only now, when this client is initialized, can it show up on
682
731
        # the D-Bus
 
732
        client_object_name = unicode(self.name).translate(
 
733
            {ord("."): ord("_"),
 
734
             ord("-"): ord("_")})
683
735
        self.dbus_object_path = (dbus.ObjectPath
684
 
                                 (u"/clients/"
685
 
                                  + self.name.replace(u".", u"_")))
 
736
                                 ("/clients/" + client_object_name))
686
737
        DBusObjectWithProperties.__init__(self, self.bus,
687
738
                                          self.dbus_object_path)
 
739
        
 
740
    def _get_approvals_pending(self):
 
741
        return self._approvals_pending
 
742
    def _set_approvals_pending(self, value):
 
743
        old_value = self._approvals_pending
 
744
        self._approvals_pending = value
 
745
        bval = bool(value)
 
746
        if (hasattr(self, "dbus_object_path")
 
747
            and bval is not bool(old_value)):
 
748
            dbus_bool = dbus.Boolean(bval, variant_level=1)
 
749
            self.PropertyChanged(dbus.String("ApprovalPending"),
 
750
                                 dbus_bool)
 
751
 
 
752
    approvals_pending = property(_get_approvals_pending,
 
753
                                 _set_approvals_pending)
 
754
    del _get_approvals_pending, _set_approvals_pending
688
755
    
689
756
    @staticmethod
690
757
    def _datetime_to_dbus(dt, variant_level=0):
693
760
                           variant_level=variant_level)
694
761
    
695
762
    def enable(self):
696
 
        oldstate = getattr(self, u"enabled", False)
 
763
        oldstate = getattr(self, "enabled", False)
697
764
        r = Client.enable(self)
698
765
        if oldstate != self.enabled:
699
766
            # Emit D-Bus signals
700
 
            self.PropertyChanged(dbus.String(u"enabled"),
 
767
            self.PropertyChanged(dbus.String("Enabled"),
701
768
                                 dbus.Boolean(True, variant_level=1))
702
769
            self.PropertyChanged(
703
 
                dbus.String(u"last_enabled"),
 
770
                dbus.String("LastEnabled"),
704
771
                self._datetime_to_dbus(self.last_enabled,
705
772
                                       variant_level=1))
706
773
        return r
707
774
    
708
775
    def disable(self, quiet = False):
709
 
        oldstate = getattr(self, u"enabled", False)
 
776
        oldstate = getattr(self, "enabled", False)
710
777
        r = Client.disable(self, quiet=quiet)
711
778
        if not quiet and oldstate != self.enabled:
712
779
            # Emit D-Bus signal
713
 
            self.PropertyChanged(dbus.String(u"enabled"),
 
780
            self.PropertyChanged(dbus.String("Enabled"),
714
781
                                 dbus.Boolean(False, variant_level=1))
715
782
        return r
716
783
    
719
786
            self.remove_from_connection()
720
787
        except LookupError:
721
788
            pass
722
 
        if hasattr(DBusObjectWithProperties, u"__del__"):
 
789
        if hasattr(DBusObjectWithProperties, "__del__"):
723
790
            DBusObjectWithProperties.__del__(self, *args, **kwargs)
724
791
        Client.__del__(self, *args, **kwargs)
725
792
    
728
795
        self.checker_callback_tag = None
729
796
        self.checker = None
730
797
        # Emit D-Bus signal
731
 
        self.PropertyChanged(dbus.String(u"checker_running"),
 
798
        self.PropertyChanged(dbus.String("CheckerRunning"),
732
799
                             dbus.Boolean(False, variant_level=1))
733
800
        if os.WIFEXITED(condition):
734
801
            exitstatus = os.WEXITSTATUS(condition)
749
816
        r = Client.checked_ok(self, *args, **kwargs)
750
817
        # Emit D-Bus signal
751
818
        self.PropertyChanged(
752
 
            dbus.String(u"last_checked_ok"),
 
819
            dbus.String("LastCheckedOK"),
753
820
            (self._datetime_to_dbus(self.last_checked_ok,
754
821
                                    variant_level=1)))
755
822
        return r
756
823
    
 
824
    def need_approval(self, *args, **kwargs):
 
825
        r = Client.need_approval(self, *args, **kwargs)
 
826
        # Emit D-Bus signal
 
827
        self.PropertyChanged(
 
828
            dbus.String("LastApprovalRequest"),
 
829
            (self._datetime_to_dbus(self.last_approval_request,
 
830
                                    variant_level=1)))
 
831
        return r
 
832
    
757
833
    def start_checker(self, *args, **kwargs):
758
834
        old_checker = self.checker
759
835
        if self.checker is not None:
767
843
            # Emit D-Bus signal
768
844
            self.CheckerStarted(self.current_checker_command)
769
845
            self.PropertyChanged(
770
 
                dbus.String(u"checker_running"),
 
846
                dbus.String("CheckerRunning"),
771
847
                dbus.Boolean(True, variant_level=1))
772
848
        return r
773
849
    
774
850
    def stop_checker(self, *args, **kwargs):
775
 
        old_checker = getattr(self, u"checker", None)
 
851
        old_checker = getattr(self, "checker", None)
776
852
        r = Client.stop_checker(self, *args, **kwargs)
777
853
        if (old_checker is not None
778
 
            and getattr(self, u"checker", None) is None):
779
 
            self.PropertyChanged(dbus.String(u"checker_running"),
 
854
            and getattr(self, "checker", None) is None):
 
855
            self.PropertyChanged(dbus.String("CheckerRunning"),
780
856
                                 dbus.Boolean(False, variant_level=1))
781
857
        return r
 
858
 
 
859
    def _reset_approved(self):
 
860
        self._approved = None
 
861
        return False
 
862
    
 
863
    def approve(self, value=True):
 
864
        self.send_changedstate()
 
865
        self._approved = value
 
866
        gobject.timeout_add(self._timedelta_to_milliseconds
 
867
                            (self.approval_duration),
 
868
                            self._reset_approved)
 
869
    
782
870
    
783
871
    ## D-Bus methods, signals & properties
784
 
    _interface = u"se.bsnet.fukt.Mandos.Client"
 
872
    _interface = "se.bsnet.fukt.Mandos.Client"
785
873
    
786
874
    ## Signals
787
875
    
788
876
    # CheckerCompleted - signal
789
 
    @dbus.service.signal(_interface, signature=u"nxs")
 
877
    @dbus.service.signal(_interface, signature="nxs")
790
878
    def CheckerCompleted(self, exitcode, waitstatus, command):
791
879
        "D-Bus signal"
792
880
        pass
793
881
    
794
882
    # CheckerStarted - signal
795
 
    @dbus.service.signal(_interface, signature=u"s")
 
883
    @dbus.service.signal(_interface, signature="s")
796
884
    def CheckerStarted(self, command):
797
885
        "D-Bus signal"
798
886
        pass
799
887
    
800
888
    # PropertyChanged - signal
801
 
    @dbus.service.signal(_interface, signature=u"sv")
 
889
    @dbus.service.signal(_interface, signature="sv")
802
890
    def PropertyChanged(self, property, value):
803
891
        "D-Bus signal"
804
892
        pass
806
894
    # GotSecret - signal
807
895
    @dbus.service.signal(_interface)
808
896
    def GotSecret(self):
809
 
        "D-Bus signal"
 
897
        """D-Bus signal
 
898
        Is sent after a successful transfer of secret from the Mandos
 
899
        server to mandos-client
 
900
        """
810
901
        pass
811
902
    
812
903
    # Rejected - signal
813
 
    @dbus.service.signal(_interface)
814
 
    def Rejected(self):
 
904
    @dbus.service.signal(_interface, signature="s")
 
905
    def Rejected(self, reason):
815
906
        "D-Bus signal"
816
907
        pass
817
908
    
 
909
    # NeedApproval - signal
 
910
    @dbus.service.signal(_interface, signature="tb")
 
911
    def NeedApproval(self, timeout, default):
 
912
        "D-Bus signal"
 
913
        return self.need_approval()
 
914
    
818
915
    ## Methods
819
916
    
 
917
    # Approve - method
 
918
    @dbus.service.method(_interface, in_signature="b")
 
919
    def Approve(self, value):
 
920
        self.approve(value)
 
921
    
820
922
    # CheckedOK - method
821
923
    @dbus.service.method(_interface)
822
924
    def CheckedOK(self):
847
949
    
848
950
    ## Properties
849
951
    
850
 
    # name - property
851
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
852
 
    def name_dbus_property(self):
 
952
    # ApprovalPending - property
 
953
    @dbus_service_property(_interface, signature="b", access="read")
 
954
    def ApprovalPending_dbus_property(self):
 
955
        return dbus.Boolean(bool(self.approvals_pending))
 
956
    
 
957
    # ApprovedByDefault - property
 
958
    @dbus_service_property(_interface, signature="b",
 
959
                           access="readwrite")
 
960
    def ApprovedByDefault_dbus_property(self, value=None):
 
961
        if value is None:       # get
 
962
            return dbus.Boolean(self.approved_by_default)
 
963
        self.approved_by_default = bool(value)
 
964
        # Emit D-Bus signal
 
965
        self.PropertyChanged(dbus.String("ApprovedByDefault"),
 
966
                             dbus.Boolean(value, variant_level=1))
 
967
    
 
968
    # ApprovalDelay - property
 
969
    @dbus_service_property(_interface, signature="t",
 
970
                           access="readwrite")
 
971
    def ApprovalDelay_dbus_property(self, value=None):
 
972
        if value is None:       # get
 
973
            return dbus.UInt64(self.approval_delay_milliseconds())
 
974
        self.approval_delay = datetime.timedelta(0, 0, 0, value)
 
975
        # Emit D-Bus signal
 
976
        self.PropertyChanged(dbus.String("ApprovalDelay"),
 
977
                             dbus.UInt64(value, variant_level=1))
 
978
    
 
979
    # ApprovalDuration - property
 
980
    @dbus_service_property(_interface, signature="t",
 
981
                           access="readwrite")
 
982
    def ApprovalDuration_dbus_property(self, value=None):
 
983
        if value is None:       # get
 
984
            return dbus.UInt64(self._timedelta_to_milliseconds(
 
985
                    self.approval_duration))
 
986
        self.approval_duration = datetime.timedelta(0, 0, 0, value)
 
987
        # Emit D-Bus signal
 
988
        self.PropertyChanged(dbus.String("ApprovalDuration"),
 
989
                             dbus.UInt64(value, variant_level=1))
 
990
    
 
991
    # Name - property
 
992
    @dbus_service_property(_interface, signature="s", access="read")
 
993
    def Name_dbus_property(self):
853
994
        return dbus.String(self.name)
854
995
    
855
 
    # fingerprint - property
856
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
857
 
    def fingerprint_dbus_property(self):
 
996
    # Fingerprint - property
 
997
    @dbus_service_property(_interface, signature="s", access="read")
 
998
    def Fingerprint_dbus_property(self):
858
999
        return dbus.String(self.fingerprint)
859
1000
    
860
 
    # host - property
861
 
    @dbus_service_property(_interface, signature=u"s",
862
 
                           access=u"readwrite")
863
 
    def host_dbus_property(self, value=None):
 
1001
    # Host - property
 
1002
    @dbus_service_property(_interface, signature="s",
 
1003
                           access="readwrite")
 
1004
    def Host_dbus_property(self, value=None):
864
1005
        if value is None:       # get
865
1006
            return dbus.String(self.host)
866
1007
        self.host = value
867
1008
        # Emit D-Bus signal
868
 
        self.PropertyChanged(dbus.String(u"host"),
 
1009
        self.PropertyChanged(dbus.String("Host"),
869
1010
                             dbus.String(value, variant_level=1))
870
1011
    
871
 
    # created - property
872
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
873
 
    def created_dbus_property(self):
 
1012
    # Created - property
 
1013
    @dbus_service_property(_interface, signature="s", access="read")
 
1014
    def Created_dbus_property(self):
874
1015
        return dbus.String(self._datetime_to_dbus(self.created))
875
1016
    
876
 
    # last_enabled - property
877
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
878
 
    def last_enabled_dbus_property(self):
 
1017
    # LastEnabled - property
 
1018
    @dbus_service_property(_interface, signature="s", access="read")
 
1019
    def LastEnabled_dbus_property(self):
879
1020
        if self.last_enabled is None:
880
 
            return dbus.String(u"")
 
1021
            return dbus.String("")
881
1022
        return dbus.String(self._datetime_to_dbus(self.last_enabled))
882
1023
    
883
 
    # enabled - property
884
 
    @dbus_service_property(_interface, signature=u"b",
885
 
                           access=u"readwrite")
886
 
    def enabled_dbus_property(self, value=None):
 
1024
    # Enabled - property
 
1025
    @dbus_service_property(_interface, signature="b",
 
1026
                           access="readwrite")
 
1027
    def Enabled_dbus_property(self, value=None):
887
1028
        if value is None:       # get
888
1029
            return dbus.Boolean(self.enabled)
889
1030
        if value:
891
1032
        else:
892
1033
            self.disable()
893
1034
    
894
 
    # last_checked_ok - property
895
 
    @dbus_service_property(_interface, signature=u"s",
896
 
                           access=u"readwrite")
897
 
    def last_checked_ok_dbus_property(self, value=None):
 
1035
    # LastCheckedOK - property
 
1036
    @dbus_service_property(_interface, signature="s",
 
1037
                           access="readwrite")
 
1038
    def LastCheckedOK_dbus_property(self, value=None):
898
1039
        if value is not None:
899
1040
            self.checked_ok()
900
1041
            return
901
1042
        if self.last_checked_ok is None:
902
 
            return dbus.String(u"")
 
1043
            return dbus.String("")
903
1044
        return dbus.String(self._datetime_to_dbus(self
904
1045
                                                  .last_checked_ok))
905
1046
    
906
 
    # timeout - property
907
 
    @dbus_service_property(_interface, signature=u"t",
908
 
                           access=u"readwrite")
909
 
    def timeout_dbus_property(self, value=None):
 
1047
    # LastApprovalRequest - property
 
1048
    @dbus_service_property(_interface, signature="s", access="read")
 
1049
    def LastApprovalRequest_dbus_property(self):
 
1050
        if self.last_approval_request is None:
 
1051
            return dbus.String("")
 
1052
        return dbus.String(self.
 
1053
                           _datetime_to_dbus(self
 
1054
                                             .last_approval_request))
 
1055
    
 
1056
    # Timeout - property
 
1057
    @dbus_service_property(_interface, signature="t",
 
1058
                           access="readwrite")
 
1059
    def Timeout_dbus_property(self, value=None):
910
1060
        if value is None:       # get
911
1061
            return dbus.UInt64(self.timeout_milliseconds())
912
1062
        self.timeout = datetime.timedelta(0, 0, 0, value)
913
1063
        # Emit D-Bus signal
914
 
        self.PropertyChanged(dbus.String(u"timeout"),
 
1064
        self.PropertyChanged(dbus.String("Timeout"),
915
1065
                             dbus.UInt64(value, variant_level=1))
916
 
        if getattr(self, u"disable_initiator_tag", None) is None:
 
1066
        if getattr(self, "disable_initiator_tag", None) is None:
917
1067
            return
918
1068
        # Reschedule timeout
919
1069
        gobject.source_remove(self.disable_initiator_tag)
931
1081
            self.disable_initiator_tag = (gobject.timeout_add
932
1082
                                          (time_to_die, self.disable))
933
1083
    
934
 
    # interval - property
935
 
    @dbus_service_property(_interface, signature=u"t",
936
 
                           access=u"readwrite")
937
 
    def interval_dbus_property(self, value=None):
 
1084
    # Interval - property
 
1085
    @dbus_service_property(_interface, signature="t",
 
1086
                           access="readwrite")
 
1087
    def Interval_dbus_property(self, value=None):
938
1088
        if value is None:       # get
939
1089
            return dbus.UInt64(self.interval_milliseconds())
940
1090
        self.interval = datetime.timedelta(0, 0, 0, value)
941
1091
        # Emit D-Bus signal
942
 
        self.PropertyChanged(dbus.String(u"interval"),
 
1092
        self.PropertyChanged(dbus.String("Interval"),
943
1093
                             dbus.UInt64(value, variant_level=1))
944
 
        if getattr(self, u"checker_initiator_tag", None) is None:
 
1094
        if getattr(self, "checker_initiator_tag", None) is None:
945
1095
            return
946
1096
        # Reschedule checker run
947
1097
        gobject.source_remove(self.checker_initiator_tag)
949
1099
                                      (value, self.start_checker))
950
1100
        self.start_checker()    # Start one now, too
951
1101
 
952
 
    # checker - property
953
 
    @dbus_service_property(_interface, signature=u"s",
954
 
                           access=u"readwrite")
955
 
    def checker_dbus_property(self, value=None):
 
1102
    # Checker - property
 
1103
    @dbus_service_property(_interface, signature="s",
 
1104
                           access="readwrite")
 
1105
    def Checker_dbus_property(self, value=None):
956
1106
        if value is None:       # get
957
1107
            return dbus.String(self.checker_command)
958
1108
        self.checker_command = value
959
1109
        # Emit D-Bus signal
960
 
        self.PropertyChanged(dbus.String(u"checker"),
 
1110
        self.PropertyChanged(dbus.String("Checker"),
961
1111
                             dbus.String(self.checker_command,
962
1112
                                         variant_level=1))
963
1113
    
964
 
    # checker_running - property
965
 
    @dbus_service_property(_interface, signature=u"b",
966
 
                           access=u"readwrite")
967
 
    def checker_running_dbus_property(self, value=None):
 
1114
    # CheckerRunning - property
 
1115
    @dbus_service_property(_interface, signature="b",
 
1116
                           access="readwrite")
 
1117
    def CheckerRunning_dbus_property(self, value=None):
968
1118
        if value is None:       # get
969
1119
            return dbus.Boolean(self.checker is not None)
970
1120
        if value:
972
1122
        else:
973
1123
            self.stop_checker()
974
1124
    
975
 
    # object_path - property
976
 
    @dbus_service_property(_interface, signature=u"o", access=u"read")
977
 
    def object_path_dbus_property(self):
 
1125
    # ObjectPath - property
 
1126
    @dbus_service_property(_interface, signature="o", access="read")
 
1127
    def ObjectPath_dbus_property(self):
978
1128
        return self.dbus_object_path # is already a dbus.ObjectPath
979
1129
    
980
 
    # secret = property
981
 
    @dbus_service_property(_interface, signature=u"ay",
982
 
                           access=u"write", byte_arrays=True)
983
 
    def secret_dbus_property(self, value):
 
1130
    # Secret = property
 
1131
    @dbus_service_property(_interface, signature="ay",
 
1132
                           access="write", byte_arrays=True)
 
1133
    def Secret_dbus_property(self, value):
984
1134
        self.secret = str(value)
985
1135
    
986
1136
    del _interface
987
1137
 
988
1138
 
 
1139
class ProxyClient(object):
 
1140
    def __init__(self, child_pipe, fpr, address):
 
1141
        self._pipe = child_pipe
 
1142
        self._pipe.send(('init', fpr, address))
 
1143
        if not self._pipe.recv():
 
1144
            raise KeyError()
 
1145
 
 
1146
    def __getattribute__(self, name):
 
1147
        if(name == '_pipe'):
 
1148
            return super(ProxyClient, self).__getattribute__(name)
 
1149
        self._pipe.send(('getattr', name))
 
1150
        data = self._pipe.recv()
 
1151
        if data[0] == 'data':
 
1152
            return data[1]
 
1153
        if data[0] == 'function':
 
1154
            def func(*args, **kwargs):
 
1155
                self._pipe.send(('funcall', name, args, kwargs))
 
1156
                return self._pipe.recv()[1]
 
1157
            return func
 
1158
 
 
1159
    def __setattr__(self, name, value):
 
1160
        if(name == '_pipe'):
 
1161
            return super(ProxyClient, self).__setattr__(name, value)
 
1162
        self._pipe.send(('setattr', name, value))
 
1163
 
 
1164
 
989
1165
class ClientHandler(socketserver.BaseRequestHandler, object):
990
1166
    """A class to handle client connections.
991
1167
    
993
1169
    Note: This will run in its own forked process."""
994
1170
    
995
1171
    def handle(self):
996
 
        logger.info(u"TCP connection from: %s",
997
 
                    unicode(self.client_address))
998
 
        logger.debug(u"IPC Pipe FD: %d",
999
 
                     self.server.child_pipe[1].fileno())
1000
 
        # Open IPC pipe to parent process
1001
 
        with contextlib.nested(self.server.child_pipe[1],
1002
 
                               self.server.parent_pipe[0]
1003
 
                               ) as (ipc, ipc_return):
 
1172
        with contextlib.closing(self.server.child_pipe) as child_pipe:
 
1173
            logger.info("TCP connection from: %s",
 
1174
                        unicode(self.client_address))
 
1175
            logger.debug("Pipe FD: %d",
 
1176
                         self.server.child_pipe.fileno())
 
1177
 
1004
1178
            session = (gnutls.connection
1005
1179
                       .ClientSession(self.request,
1006
1180
                                      gnutls.connection
1007
1181
                                      .X509Credentials()))
1008
 
            
1009
 
            line = self.request.makefile().readline()
1010
 
            logger.debug(u"Protocol version: %r", line)
1011
 
            try:
1012
 
                if int(line.strip().split()[0]) > 1:
1013
 
                    raise RuntimeError
1014
 
            except (ValueError, IndexError, RuntimeError), error:
1015
 
                logger.error(u"Unknown protocol version: %s", error)
1016
 
                return
1017
 
            
 
1182
 
1018
1183
            # Note: gnutls.connection.X509Credentials is really a
1019
1184
            # generic GnuTLS certificate credentials object so long as
1020
1185
            # no X.509 keys are added to it.  Therefore, we can use it
1021
1186
            # here despite using OpenPGP certificates.
1022
 
            
1023
 
            #priority = u':'.join((u"NONE", u"+VERS-TLS1.1",
1024
 
            #                      u"+AES-256-CBC", u"+SHA1",
1025
 
            #                      u"+COMP-NULL", u"+CTYPE-OPENPGP",
1026
 
            #                      u"+DHE-DSS"))
 
1187
 
 
1188
            #priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
1189
            #                      "+AES-256-CBC", "+SHA1",
 
1190
            #                      "+COMP-NULL", "+CTYPE-OPENPGP",
 
1191
            #                      "+DHE-DSS"))
1027
1192
            # Use a fallback default, since this MUST be set.
1028
1193
            priority = self.server.gnutls_priority
1029
1194
            if priority is None:
1030
 
                priority = u"NORMAL"
 
1195
                priority = "NORMAL"
1031
1196
            (gnutls.library.functions
1032
1197
             .gnutls_priority_set_direct(session._c_object,
1033
1198
                                         priority, None))
1034
 
            
 
1199
 
 
1200
            # Start communication using the Mandos protocol
 
1201
            # Get protocol number
 
1202
            line = self.request.makefile().readline()
 
1203
            logger.debug("Protocol version: %r", line)
 
1204
            try:
 
1205
                if int(line.strip().split()[0]) > 1:
 
1206
                    raise RuntimeError
 
1207
            except (ValueError, IndexError, RuntimeError), error:
 
1208
                logger.error("Unknown protocol version: %s", error)
 
1209
                return
 
1210
 
 
1211
            # Start GnuTLS connection
1035
1212
            try:
1036
1213
                session.handshake()
1037
1214
            except gnutls.errors.GNUTLSError, error:
1038
 
                logger.warning(u"Handshake failed: %s", error)
 
1215
                logger.warning("Handshake failed: %s", error)
1039
1216
                # Do not run session.bye() here: the session is not
1040
1217
                # established.  Just abandon the request.
1041
1218
                return
1042
 
            logger.debug(u"Handshake succeeded")
 
1219
            logger.debug("Handshake succeeded")
 
1220
 
 
1221
            approval_required = False
1043
1222
            try:
1044
1223
                try:
1045
1224
                    fpr = self.fingerprint(self.peer_certificate
1046
1225
                                           (session))
1047
1226
                except (TypeError, gnutls.errors.GNUTLSError), error:
1048
 
                    logger.warning(u"Bad certificate: %s", error)
 
1227
                    logger.warning("Bad certificate: %s", error)
1049
1228
                    return
1050
 
                logger.debug(u"Fingerprint: %s", fpr)
 
1229
                logger.debug("Fingerprint: %s", fpr)
1051
1230
 
1052
 
                for c in self.server.clients:
1053
 
                    if c.fingerprint == fpr:
1054
 
                        client = c
 
1231
                try:
 
1232
                    client = ProxyClient(child_pipe, fpr,
 
1233
                                         self.client_address)
 
1234
                except KeyError:
 
1235
                    return
 
1236
                
 
1237
                if client.approval_delay:
 
1238
                    delay = client.approval_delay
 
1239
                    client.approvals_pending += 1
 
1240
                    approval_required = True
 
1241
                
 
1242
                while True:
 
1243
                    if not client.enabled:
 
1244
                        logger.warning("Client %s is disabled",
 
1245
                                       client.name)
 
1246
                        if self.server.use_dbus:
 
1247
                            # Emit D-Bus signal
 
1248
                            client.Rejected("Disabled")                    
 
1249
                        return
 
1250
                    
 
1251
                    if client._approved or not client.approval_delay:
 
1252
                        #We are approved or approval is disabled
1055
1253
                        break
1056
 
                else:
1057
 
                    ipc.write(u"NOTFOUND %s %s\n"
1058
 
                              % (fpr, unicode(self.client_address)))
1059
 
                    return
1060
 
                # Have to check if client.enabled, since it is
1061
 
                # possible that the client was disabled since the
1062
 
                # GnuTLS session was established.
1063
 
                ipc.write(u"GETATTR enabled %s\n" % fpr)
1064
 
                enabled = pickle.load(ipc_return)
1065
 
                if not enabled:
1066
 
                    ipc.write(u"DISABLED %s\n" % client.name)
1067
 
                    return
1068
 
                # Send "NEED_APPROVAL" here and hang waiting
1069
 
                # for response?  Leave timeout to parent process?
1070
 
                ipc.write(u"SENDING %s\n" % client.name)
 
1254
                    elif client._approved is None:
 
1255
                        logger.info("Client %s needs approval",
 
1256
                                    client.name)
 
1257
                        if self.server.use_dbus:
 
1258
                            # Emit D-Bus signal
 
1259
                            client.NeedApproval(
 
1260
                                client.approval_delay_milliseconds(),
 
1261
                                client.approved_by_default)
 
1262
                    else:
 
1263
                        logger.warning("Client %s was not approved",
 
1264
                                       client.name)
 
1265
                        if self.server.use_dbus:
 
1266
                            # Emit D-Bus signal
 
1267
                            client.Rejected("Denied")
 
1268
                        return
 
1269
                    
 
1270
                    #wait until timeout or approved
 
1271
                    #x = float(client._timedelta_to_milliseconds(delay))
 
1272
                    time = datetime.datetime.now()
 
1273
                    client.changedstate.acquire()
 
1274
                    client.changedstate.wait(float(client._timedelta_to_milliseconds(delay) / 1000))
 
1275
                    client.changedstate.release()
 
1276
                    time2 = datetime.datetime.now()
 
1277
                    if (time2 - time) >= delay:
 
1278
                        if not client.approved_by_default:
 
1279
                            logger.warning("Client %s timed out while"
 
1280
                                           " waiting for approval",
 
1281
                                           client.name)
 
1282
                            if self.server.use_dbus:
 
1283
                                # Emit D-Bus signal
 
1284
                                client.Rejected("Approval timed out")
 
1285
                            return
 
1286
                        else:
 
1287
                            break
 
1288
                    else:
 
1289
                        delay -= time2 - time
 
1290
                
1071
1291
                sent_size = 0
1072
1292
                while sent_size < len(client.secret):
1073
 
                    sent = session.send(client.secret[sent_size:])
1074
 
                    logger.debug(u"Sent: %d, remaining: %d",
 
1293
                    try:
 
1294
                        sent = session.send(client.secret[sent_size:])
 
1295
                    except (gnutls.errors.GNUTLSError), error:
 
1296
                        logger.warning("gnutls send failed")
 
1297
                        return
 
1298
                    logger.debug("Sent: %d, remaining: %d",
1075
1299
                                 sent, len(client.secret)
1076
1300
                                 - (sent_size + sent))
1077
1301
                    sent_size += sent
 
1302
 
 
1303
                logger.info("Sending secret to %s", client.name)
 
1304
                # bump the timeout as if seen
 
1305
                client.checked_ok()
 
1306
                if self.server.use_dbus:
 
1307
                    # Emit D-Bus signal
 
1308
                    client.GotSecret()
 
1309
            
1078
1310
            finally:
1079
 
                session.bye()
 
1311
                if approval_required:
 
1312
                    client.approvals_pending -= 1
 
1313
                try:
 
1314
                    session.bye()
 
1315
                except (gnutls.errors.GNUTLSError), error:
 
1316
                    logger.warning("GnuTLS bye failed")
1080
1317
    
1081
1318
    @staticmethod
1082
1319
    def peer_certificate(session):
1092
1329
                     .gnutls_certificate_get_peers
1093
1330
                     (session._c_object, ctypes.byref(list_size)))
1094
1331
        if not bool(cert_list) and list_size.value != 0:
1095
 
            raise gnutls.errors.GNUTLSError(u"error getting peer"
1096
 
                                            u" certificate")
 
1332
            raise gnutls.errors.GNUTLSError("error getting peer"
 
1333
                                            " certificate")
1097
1334
        if list_size.value == 0:
1098
1335
            return None
1099
1336
        cert = cert_list[0]
1125
1362
        if crtverify.value != 0:
1126
1363
            gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
1127
1364
            raise (gnutls.errors.CertificateSecurityError
1128
 
                   (u"Verify failed"))
 
1365
                   ("Verify failed"))
1129
1366
        # New buffer for the fingerprint
1130
1367
        buf = ctypes.create_string_buffer(20)
1131
1368
        buf_len = ctypes.c_size_t()
1138
1375
        # Convert the buffer to a Python bytestring
1139
1376
        fpr = ctypes.string_at(buf, buf_len.value)
1140
1377
        # Convert the bytestring to hexadecimal notation
1141
 
        hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
 
1378
        hex_fpr = ''.join("%02X" % ord(char) for char in fpr)
1142
1379
        return hex_fpr
1143
1380
 
1144
1381
 
1145
 
class ForkingMixInWithPipes(socketserver.ForkingMixIn, object):
1146
 
    """Like socketserver.ForkingMixIn, but also pass a pipe pair."""
 
1382
class MultiprocessingMixIn(object):
 
1383
    """Like socketserver.ThreadingMixIn, but with multiprocessing"""
 
1384
    def sub_process_main(self, request, address):
 
1385
        try:
 
1386
            self.finish_request(request, address)
 
1387
        except:
 
1388
            self.handle_error(request, address)
 
1389
        self.close_request(request)
 
1390
            
 
1391
    def process_request(self, request, address):
 
1392
        """Start a new process to process the request."""
 
1393
        multiprocessing.Process(target = self.sub_process_main,
 
1394
                                args = (request, address)).start()
 
1395
 
 
1396
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
 
1397
    """ adds a pipe to the MixIn """
1147
1398
    def process_request(self, request, client_address):
1148
1399
        """Overrides and wraps the original process_request().
1149
1400
        
1150
1401
        This function creates a new pipe in self.pipe
1151
1402
        """
1152
 
        # Child writes to child_pipe
1153
 
        self.child_pipe = map(os.fdopen, os.pipe(), u"rw", (1, 0))
1154
 
        # Parent writes to parent_pipe
1155
 
        self.parent_pipe = map(os.fdopen, os.pipe(), u"rw", (1, 0))
1156
 
        super(ForkingMixInWithPipes,
 
1403
        parent_pipe, self.child_pipe = multiprocessing.Pipe()
 
1404
 
 
1405
        super(MultiprocessingMixInWithPipe,
1157
1406
              self).process_request(request, client_address)
1158
 
        # Close unused ends for parent
1159
 
        self.parent_pipe[0].close() # close read end
1160
 
        self.child_pipe[1].close()  # close write end
1161
 
        self.add_pipe_fds(self.child_pipe[0], self.parent_pipe[1])
1162
 
    def add_pipe_fds(self, child_pipe_fd, parent_pipe_fd):
 
1407
        self.child_pipe.close()
 
1408
        self.add_pipe(parent_pipe)
 
1409
 
 
1410
    def add_pipe(self, parent_pipe):
1163
1411
        """Dummy function; override as necessary"""
1164
 
        child_pipe_fd.close()
1165
 
        parent_pipe_fd.close()
1166
 
 
1167
 
 
1168
 
class IPv6_TCPServer(ForkingMixInWithPipes,
 
1412
        raise NotImplementedError
 
1413
 
 
1414
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
1169
1415
                     socketserver.TCPServer, object):
1170
1416
    """IPv6-capable TCP server.  Accepts 'None' as address and/or port
1171
1417
    
1187
1433
        bind to an address or port if they were not specified."""
1188
1434
        if self.interface is not None:
1189
1435
            if SO_BINDTODEVICE is None:
1190
 
                logger.error(u"SO_BINDTODEVICE does not exist;"
1191
 
                             u" cannot bind to interface %s",
 
1436
                logger.error("SO_BINDTODEVICE does not exist;"
 
1437
                             " cannot bind to interface %s",
1192
1438
                             self.interface)
1193
1439
            else:
1194
1440
                try:
1195
1441
                    self.socket.setsockopt(socket.SOL_SOCKET,
1196
1442
                                           SO_BINDTODEVICE,
1197
1443
                                           str(self.interface
1198
 
                                               + u'\0'))
 
1444
                                               + '\0'))
1199
1445
                except socket.error, error:
1200
1446
                    if error[0] == errno.EPERM:
1201
 
                        logger.error(u"No permission to"
1202
 
                                     u" bind to interface %s",
 
1447
                        logger.error("No permission to"
 
1448
                                     " bind to interface %s",
1203
1449
                                     self.interface)
1204
1450
                    elif error[0] == errno.ENOPROTOOPT:
1205
 
                        logger.error(u"SO_BINDTODEVICE not available;"
1206
 
                                     u" cannot bind to interface %s",
 
1451
                        logger.error("SO_BINDTODEVICE not available;"
 
1452
                                     " cannot bind to interface %s",
1207
1453
                                     self.interface)
1208
1454
                    else:
1209
1455
                        raise
1211
1457
        if self.server_address[0] or self.server_address[1]:
1212
1458
            if not self.server_address[0]:
1213
1459
                if self.address_family == socket.AF_INET6:
1214
 
                    any_address = u"::" # in6addr_any
 
1460
                    any_address = "::" # in6addr_any
1215
1461
                else:
1216
1462
                    any_address = socket.INADDR_ANY
1217
1463
                self.server_address = (any_address,
1256
1502
            return socketserver.TCPServer.server_activate(self)
1257
1503
    def enable(self):
1258
1504
        self.enabled = True
1259
 
    def add_pipe_fds(self, child_pipe_fd, parent_pipe_fd):
 
1505
    def add_pipe(self, parent_pipe):
1260
1506
        # Call "handle_ipc" for both data and EOF events
1261
 
        gobject.io_add_watch(child_pipe_fd.fileno(),
 
1507
        gobject.io_add_watch(parent_pipe.fileno(),
1262
1508
                             gobject.IO_IN | gobject.IO_HUP,
1263
1509
                             functools.partial(self.handle_ipc,
1264
 
                                               reply = parent_pipe_fd,
1265
 
                                               sender= child_pipe_fd))
1266
 
    def handle_ipc(self, source, condition, reply=None, sender=None):
 
1510
                                               parent_pipe = parent_pipe))
 
1511
        
 
1512
    def handle_ipc(self, source, condition, parent_pipe=None,
 
1513
                   client_object=None):
1267
1514
        condition_names = {
1268
 
            gobject.IO_IN: u"IN",   # There is data to read.
1269
 
            gobject.IO_OUT: u"OUT", # Data can be written (without
 
1515
            gobject.IO_IN: "IN",   # There is data to read.
 
1516
            gobject.IO_OUT: "OUT", # Data can be written (without
1270
1517
                                    # blocking).
1271
 
            gobject.IO_PRI: u"PRI", # There is urgent data to read.
1272
 
            gobject.IO_ERR: u"ERR", # Error condition.
1273
 
            gobject.IO_HUP: u"HUP"  # Hung up (the connection has been
 
1518
            gobject.IO_PRI: "PRI", # There is urgent data to read.
 
1519
            gobject.IO_ERR: "ERR", # Error condition.
 
1520
            gobject.IO_HUP: "HUP"  # Hung up (the connection has been
1274
1521
                                    # broken, usually for pipes and
1275
1522
                                    # sockets).
1276
1523
            }
1278
1525
                                       for cond, name in
1279
1526
                                       condition_names.iteritems()
1280
1527
                                       if cond & condition)
1281
 
        logger.debug(u"Handling IPC: FD = %d, condition = %s", source,
1282
 
                     conditions_string)
1283
 
        
1284
 
        # Read a line from the file object
1285
 
        cmdline = sender.readline()
1286
 
        if not cmdline:             # Empty line means end of file
1287
 
            # close the IPC pipes
1288
 
            sender.close()
1289
 
            reply.close()
1290
 
            
1291
 
            # Stop calling this function
1292
 
            return False
1293
 
        
1294
 
        logger.debug(u"IPC command: %r", cmdline)
1295
 
        
1296
 
        # Parse and act on command
1297
 
        cmd, args = cmdline.rstrip(u"\r\n").split(None, 1)
1298
 
        
1299
 
        if cmd == u"NOTFOUND":
1300
 
            fpr, address = args.split(None, 1)
1301
 
            logger.warning(u"Client not found for fingerprint: %s, ad"
1302
 
                           u"dress: %s", fpr, address)
1303
 
            if self.use_dbus:
1304
 
                # Emit D-Bus signal
1305
 
                mandos_dbus_service.ClientNotFound(fpr, address)
1306
 
        elif cmd == u"DISABLED":
1307
 
            for client in self.clients:
1308
 
                if client.name == args:
1309
 
                    logger.warning(u"Client %s is disabled", args)
1310
 
                    if self.use_dbus:
1311
 
                        # Emit D-Bus signal
1312
 
                        client.Rejected()
1313
 
                    break
1314
 
            else:
1315
 
                logger.error(u"Unknown client %s is disabled", args)
1316
 
        elif cmd == u"SENDING":
1317
 
            for client in self.clients:
1318
 
                if client.name == args:
1319
 
                    logger.info(u"Sending secret to %s", client.name)
1320
 
                    client.checked_ok()
1321
 
                    if self.use_dbus:
1322
 
                        # Emit D-Bus signal
1323
 
                        client.GotSecret()
1324
 
                    break
1325
 
            else:
1326
 
                logger.error(u"Sending secret to unknown client %s",
1327
 
                             args)
1328
 
        elif cmd == u"GETATTR":
1329
 
            attr_name, fpr = args.split(None, 1)
1330
 
            for client in self.clients:
1331
 
                if client.fingerprint == fpr:
1332
 
                    attr_value = getattr(client, attr_name, None)
1333
 
                    logger.debug("IPC reply: %r", attr_value)
1334
 
                    pickle.dump(attr_value, reply)
1335
 
                    break
1336
 
            else:
1337
 
                logger.error(u"Client %s on address %s requesting "
1338
 
                             u"attribute %s not found", fpr, address,
1339
 
                             attr_name)
1340
 
                pickle.dump(None, reply)
1341
 
        else:
1342
 
            logger.error(u"Unknown IPC command: %r", cmdline)
1343
 
        
1344
 
        # Keep calling this function
 
1528
        # error or the other end of multiprocessing.Pipe has closed
 
1529
        if condition & (gobject.IO_ERR | condition & gobject.IO_HUP):
 
1530
            return False
 
1531
        
 
1532
        # Read a request from the child
 
1533
        request = parent_pipe.recv()
 
1534
        command = request[0]
 
1535
        
 
1536
        if command == 'init':
 
1537
            fpr = request[1]
 
1538
            address = request[2]
 
1539
            
 
1540
            for c in self.clients:
 
1541
                if c.fingerprint == fpr:
 
1542
                    client = c
 
1543
                    break
 
1544
            else:
 
1545
                logger.warning("Client not found for fingerprint: %s, ad"
 
1546
                               "dress: %s", fpr, address)
 
1547
                if self.use_dbus:
 
1548
                    # Emit D-Bus signal
 
1549
                    mandos_dbus_service.ClientNotFound(fpr, address[0])
 
1550
                parent_pipe.send(False)
 
1551
                return False
 
1552
            
 
1553
            gobject.io_add_watch(parent_pipe.fileno(),
 
1554
                                 gobject.IO_IN | gobject.IO_HUP,
 
1555
                                 functools.partial(self.handle_ipc,
 
1556
                                                   parent_pipe = parent_pipe,
 
1557
                                                   client_object = client))
 
1558
            parent_pipe.send(True)
 
1559
            # remove the old hook in favor of the new above hook on same fileno
 
1560
            return False
 
1561
        if command == 'funcall':
 
1562
            funcname = request[1]
 
1563
            args = request[2]
 
1564
            kwargs = request[3]
 
1565
            
 
1566
            parent_pipe.send(('data', getattr(client_object, funcname)(*args, **kwargs)))
 
1567
 
 
1568
        if command == 'getattr':
 
1569
            attrname = request[1]
 
1570
            if callable(client_object.__getattribute__(attrname)):
 
1571
                parent_pipe.send(('function',))
 
1572
            else:
 
1573
                parent_pipe.send(('data', client_object.__getattribute__(attrname)))
 
1574
        
 
1575
        if command == 'setattr':
 
1576
            attrname = request[1]
 
1577
            value = request[2]
 
1578
            setattr(client_object, attrname, value)
 
1579
 
1345
1580
        return True
1346
1581
 
1347
1582
 
1348
1583
def string_to_delta(interval):
1349
1584
    """Parse a string and return a datetime.timedelta
1350
1585
    
1351
 
    >>> string_to_delta(u'7d')
 
1586
    >>> string_to_delta('7d')
1352
1587
    datetime.timedelta(7)
1353
 
    >>> string_to_delta(u'60s')
 
1588
    >>> string_to_delta('60s')
1354
1589
    datetime.timedelta(0, 60)
1355
 
    >>> string_to_delta(u'60m')
 
1590
    >>> string_to_delta('60m')
1356
1591
    datetime.timedelta(0, 3600)
1357
 
    >>> string_to_delta(u'24h')
 
1592
    >>> string_to_delta('24h')
1358
1593
    datetime.timedelta(1)
1359
 
    >>> string_to_delta(u'1w')
 
1594
    >>> string_to_delta('1w')
1360
1595
    datetime.timedelta(7)
1361
 
    >>> string_to_delta(u'5m 30s')
 
1596
    >>> string_to_delta('5m 30s')
1362
1597
    datetime.timedelta(0, 330)
1363
1598
    """
1364
1599
    timevalue = datetime.timedelta(0)
1366
1601
        try:
1367
1602
            suffix = unicode(s[-1])
1368
1603
            value = int(s[:-1])
1369
 
            if suffix == u"d":
 
1604
            if suffix == "d":
1370
1605
                delta = datetime.timedelta(value)
1371
 
            elif suffix == u"s":
 
1606
            elif suffix == "s":
1372
1607
                delta = datetime.timedelta(0, value)
1373
 
            elif suffix == u"m":
 
1608
            elif suffix == "m":
1374
1609
                delta = datetime.timedelta(0, 0, 0, 0, value)
1375
 
            elif suffix == u"h":
 
1610
            elif suffix == "h":
1376
1611
                delta = datetime.timedelta(0, 0, 0, 0, 0, value)
1377
 
            elif suffix == u"w":
 
1612
            elif suffix == "w":
1378
1613
                delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
1379
1614
            else:
1380
 
                raise ValueError(u"Unknown suffix %r" % suffix)
 
1615
                raise ValueError("Unknown suffix %r" % suffix)
1381
1616
        except (ValueError, IndexError), e:
1382
 
            raise ValueError(e.message)
 
1617
            raise ValueError(*(e.args))
1383
1618
        timevalue += delta
1384
1619
    return timevalue
1385
1620
 
1391
1626
    global if_nametoindex
1392
1627
    try:
1393
1628
        if_nametoindex = (ctypes.cdll.LoadLibrary
1394
 
                          (ctypes.util.find_library(u"c"))
 
1629
                          (ctypes.util.find_library("c"))
1395
1630
                          .if_nametoindex)
1396
1631
    except (OSError, AttributeError):
1397
 
        logger.warning(u"Doing if_nametoindex the hard way")
 
1632
        logger.warning("Doing if_nametoindex the hard way")
1398
1633
        def if_nametoindex(interface):
1399
1634
            "Get an interface index the hard way, i.e. using fcntl()"
1400
1635
            SIOCGIFINDEX = 0x8933  # From /usr/include/linux/sockios.h
1401
1636
            with contextlib.closing(socket.socket()) as s:
1402
1637
                ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
1403
 
                                    struct.pack(str(u"16s16x"),
 
1638
                                    struct.pack(str("16s16x"),
1404
1639
                                                interface))
1405
 
            interface_index = struct.unpack(str(u"I"),
 
1640
            interface_index = struct.unpack(str("I"),
1406
1641
                                            ifreq[16:20])[0]
1407
1642
            return interface_index
1408
1643
    return if_nametoindex(interface)
1416
1651
        sys.exit()
1417
1652
    os.setsid()
1418
1653
    if not nochdir:
1419
 
        os.chdir(u"/")
 
1654
        os.chdir("/")
1420
1655
    if os.fork():
1421
1656
        sys.exit()
1422
1657
    if not noclose:
1424
1659
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
1425
1660
        if not stat.S_ISCHR(os.fstat(null).st_mode):
1426
1661
            raise OSError(errno.ENODEV,
1427
 
                          u"%s not a character device"
 
1662
                          "%s not a character device"
1428
1663
                          % os.path.devnull)
1429
1664
        os.dup2(null, sys.stdin.fileno())
1430
1665
        os.dup2(null, sys.stdout.fileno())
1439
1674
    # Parsing of options, both command line and config file
1440
1675
    
1441
1676
    parser = optparse.OptionParser(version = "%%prog %s" % version)
1442
 
    parser.add_option("-i", u"--interface", type=u"string",
1443
 
                      metavar="IF", help=u"Bind to interface IF")
1444
 
    parser.add_option("-a", u"--address", type=u"string",
1445
 
                      help=u"Address to listen for requests on")
1446
 
    parser.add_option("-p", u"--port", type=u"int",
1447
 
                      help=u"Port number to receive requests on")
1448
 
    parser.add_option("--check", action=u"store_true",
1449
 
                      help=u"Run self-test")
1450
 
    parser.add_option("--debug", action=u"store_true",
1451
 
                      help=u"Debug mode; run in foreground and log to"
1452
 
                      u" terminal")
1453
 
    parser.add_option("--priority", type=u"string", help=u"GnuTLS"
1454
 
                      u" priority string (see GnuTLS documentation)")
1455
 
    parser.add_option("--servicename", type=u"string",
1456
 
                      metavar=u"NAME", help=u"Zeroconf service name")
1457
 
    parser.add_option("--configdir", type=u"string",
1458
 
                      default=u"/etc/mandos", metavar=u"DIR",
1459
 
                      help=u"Directory to search for configuration"
1460
 
                      u" files")
1461
 
    parser.add_option("--no-dbus", action=u"store_false",
1462
 
                      dest=u"use_dbus", help=u"Do not provide D-Bus"
1463
 
                      u" system bus interface")
1464
 
    parser.add_option("--no-ipv6", action=u"store_false",
1465
 
                      dest=u"use_ipv6", help=u"Do not use IPv6")
 
1677
    parser.add_option("-i", "--interface", type="string",
 
1678
                      metavar="IF", help="Bind to interface IF")
 
1679
    parser.add_option("-a", "--address", type="string",
 
1680
                      help="Address to listen for requests on")
 
1681
    parser.add_option("-p", "--port", type="int",
 
1682
                      help="Port number to receive requests on")
 
1683
    parser.add_option("--check", action="store_true",
 
1684
                      help="Run self-test")
 
1685
    parser.add_option("--debug", action="store_true",
 
1686
                      help="Debug mode; run in foreground and log to"
 
1687
                      " terminal")
 
1688
    parser.add_option("--debuglevel", type="string", metavar="LEVEL",
 
1689
                      help="Debug level for stdout output")
 
1690
    parser.add_option("--priority", type="string", help="GnuTLS"
 
1691
                      " priority string (see GnuTLS documentation)")
 
1692
    parser.add_option("--servicename", type="string",
 
1693
                      metavar="NAME", help="Zeroconf service name")
 
1694
    parser.add_option("--configdir", type="string",
 
1695
                      default="/etc/mandos", metavar="DIR",
 
1696
                      help="Directory to search for configuration"
 
1697
                      " files")
 
1698
    parser.add_option("--no-dbus", action="store_false",
 
1699
                      dest="use_dbus", help="Do not provide D-Bus"
 
1700
                      " system bus interface")
 
1701
    parser.add_option("--no-ipv6", action="store_false",
 
1702
                      dest="use_ipv6", help="Do not use IPv6")
1466
1703
    options = parser.parse_args()[0]
1467
1704
    
1468
1705
    if options.check:
1471
1708
        sys.exit()
1472
1709
    
1473
1710
    # Default values for config file for server-global settings
1474
 
    server_defaults = { u"interface": u"",
1475
 
                        u"address": u"",
1476
 
                        u"port": u"",
1477
 
                        u"debug": u"False",
1478
 
                        u"priority":
1479
 
                        u"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
1480
 
                        u"servicename": u"Mandos",
1481
 
                        u"use_dbus": u"True",
1482
 
                        u"use_ipv6": u"True",
 
1711
    server_defaults = { "interface": "",
 
1712
                        "address": "",
 
1713
                        "port": "",
 
1714
                        "debug": "False",
 
1715
                        "priority":
 
1716
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
 
1717
                        "servicename": "Mandos",
 
1718
                        "use_dbus": "True",
 
1719
                        "use_ipv6": "True",
 
1720
                        "debuglevel": "",
1483
1721
                        }
1484
1722
    
1485
1723
    # Parse config file for server-global settings
1486
1724
    server_config = configparser.SafeConfigParser(server_defaults)
1487
1725
    del server_defaults
1488
1726
    server_config.read(os.path.join(options.configdir,
1489
 
                                    u"mandos.conf"))
 
1727
                                    "mandos.conf"))
1490
1728
    # Convert the SafeConfigParser object to a dict
1491
1729
    server_settings = server_config.defaults()
1492
1730
    # Use the appropriate methods on the non-string config options
1493
 
    for option in (u"debug", u"use_dbus", u"use_ipv6"):
1494
 
        server_settings[option] = server_config.getboolean(u"DEFAULT",
 
1731
    for option in ("debug", "use_dbus", "use_ipv6"):
 
1732
        server_settings[option] = server_config.getboolean("DEFAULT",
1495
1733
                                                           option)
1496
1734
    if server_settings["port"]:
1497
 
        server_settings["port"] = server_config.getint(u"DEFAULT",
1498
 
                                                       u"port")
 
1735
        server_settings["port"] = server_config.getint("DEFAULT",
 
1736
                                                       "port")
1499
1737
    del server_config
1500
1738
    
1501
1739
    # Override the settings from the config file with command line
1502
1740
    # options, if set.
1503
 
    for option in (u"interface", u"address", u"port", u"debug",
1504
 
                   u"priority", u"servicename", u"configdir",
1505
 
                   u"use_dbus", u"use_ipv6"):
 
1741
    for option in ("interface", "address", "port", "debug",
 
1742
                   "priority", "servicename", "configdir",
 
1743
                   "use_dbus", "use_ipv6", "debuglevel"):
1506
1744
        value = getattr(options, option)
1507
1745
        if value is not None:
1508
1746
            server_settings[option] = value
1516
1754
    ##################################################################
1517
1755
    
1518
1756
    # For convenience
1519
 
    debug = server_settings[u"debug"]
1520
 
    use_dbus = server_settings[u"use_dbus"]
1521
 
    use_ipv6 = server_settings[u"use_ipv6"]
1522
 
    
1523
 
    if not debug:
1524
 
        syslogger.setLevel(logging.WARNING)
1525
 
        console.setLevel(logging.WARNING)
1526
 
    
1527
 
    if server_settings[u"servicename"] != u"Mandos":
 
1757
    debug = server_settings["debug"]
 
1758
    debuglevel = server_settings["debuglevel"]
 
1759
    use_dbus = server_settings["use_dbus"]
 
1760
    use_ipv6 = server_settings["use_ipv6"]
 
1761
 
 
1762
    if server_settings["servicename"] != "Mandos":
1528
1763
        syslogger.setFormatter(logging.Formatter
1529
 
                               (u'Mandos (%s) [%%(process)d]:'
1530
 
                                u' %%(levelname)s: %%(message)s'
1531
 
                                % server_settings[u"servicename"]))
 
1764
                               ('Mandos (%s) [%%(process)d]:'
 
1765
                                ' %%(levelname)s: %%(message)s'
 
1766
                                % server_settings["servicename"]))
1532
1767
    
1533
1768
    # Parse config file with clients
1534
 
    client_defaults = { u"timeout": u"1h",
1535
 
                        u"interval": u"5m",
1536
 
                        u"checker": u"fping -q -- %%(host)s",
1537
 
                        u"host": u"",
 
1769
    client_defaults = { "timeout": "1h",
 
1770
                        "interval": "5m",
 
1771
                        "checker": "fping -q -- %%(host)s",
 
1772
                        "host": "",
 
1773
                        "approval_delay": "0s",
 
1774
                        "approval_duration": "1s",
1538
1775
                        }
1539
1776
    client_config = configparser.SafeConfigParser(client_defaults)
1540
 
    client_config.read(os.path.join(server_settings[u"configdir"],
1541
 
                                    u"clients.conf"))
 
1777
    client_config.read(os.path.join(server_settings["configdir"],
 
1778
                                    "clients.conf"))
1542
1779
    
1543
1780
    global mandos_dbus_service
1544
1781
    mandos_dbus_service = None
1545
1782
    
1546
 
    tcp_server = MandosServer((server_settings[u"address"],
1547
 
                               server_settings[u"port"]),
 
1783
    tcp_server = MandosServer((server_settings["address"],
 
1784
                               server_settings["port"]),
1548
1785
                              ClientHandler,
1549
 
                              interface=server_settings[u"interface"],
 
1786
                              interface=(server_settings["interface"]
 
1787
                                         or None),
1550
1788
                              use_ipv6=use_ipv6,
1551
1789
                              gnutls_priority=
1552
 
                              server_settings[u"priority"],
 
1790
                              server_settings["priority"],
1553
1791
                              use_dbus=use_dbus)
1554
 
    pidfilename = u"/var/run/mandos.pid"
1555
 
    try:
1556
 
        pidfile = open(pidfilename, u"w")
1557
 
    except IOError:
1558
 
        logger.error(u"Could not open file %r", pidfilename)
 
1792
    if not debug:
 
1793
        pidfilename = "/var/run/mandos.pid"
 
1794
        try:
 
1795
            pidfile = open(pidfilename, "w")
 
1796
        except IOError:
 
1797
            logger.error("Could not open file %r", pidfilename)
1559
1798
    
1560
1799
    try:
1561
 
        uid = pwd.getpwnam(u"_mandos").pw_uid
1562
 
        gid = pwd.getpwnam(u"_mandos").pw_gid
 
1800
        uid = pwd.getpwnam("_mandos").pw_uid
 
1801
        gid = pwd.getpwnam("_mandos").pw_gid
1563
1802
    except KeyError:
1564
1803
        try:
1565
 
            uid = pwd.getpwnam(u"mandos").pw_uid
1566
 
            gid = pwd.getpwnam(u"mandos").pw_gid
 
1804
            uid = pwd.getpwnam("mandos").pw_uid
 
1805
            gid = pwd.getpwnam("mandos").pw_gid
1567
1806
        except KeyError:
1568
1807
            try:
1569
 
                uid = pwd.getpwnam(u"nobody").pw_uid
1570
 
                gid = pwd.getpwnam(u"nobody").pw_gid
 
1808
                uid = pwd.getpwnam("nobody").pw_uid
 
1809
                gid = pwd.getpwnam("nobody").pw_gid
1571
1810
            except KeyError:
1572
1811
                uid = 65534
1573
1812
                gid = 65534
1578
1817
        if error[0] != errno.EPERM:
1579
1818
            raise error
1580
1819
    
1581
 
    # Enable all possible GnuTLS debugging
 
1820
    if not debug and not debuglevel:
 
1821
        syslogger.setLevel(logging.WARNING)
 
1822
        console.setLevel(logging.WARNING)
 
1823
    if debuglevel:
 
1824
        level = getattr(logging, debuglevel.upper())
 
1825
        syslogger.setLevel(level)
 
1826
        console.setLevel(level)
 
1827
 
1582
1828
    if debug:
 
1829
        # Enable all possible GnuTLS debugging
 
1830
        
1583
1831
        # "Use a log level over 10 to enable all debugging options."
1584
1832
        # - GnuTLS manual
1585
1833
        gnutls.library.functions.gnutls_global_set_log_level(11)
1586
1834
        
1587
1835
        @gnutls.library.types.gnutls_log_func
1588
1836
        def debug_gnutls(level, string):
1589
 
            logger.debug(u"GnuTLS: %s", string[:-1])
 
1837
            logger.debug("GnuTLS: %s", string[:-1])
1590
1838
        
1591
1839
        (gnutls.library.functions
1592
1840
         .gnutls_global_set_log_function(debug_gnutls))
 
1841
        
 
1842
        # Redirect stdin so all checkers get /dev/null
 
1843
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
 
1844
        os.dup2(null, sys.stdin.fileno())
 
1845
        if null > 2:
 
1846
            os.close(null)
 
1847
    else:
 
1848
        # No console logging
 
1849
        logger.removeHandler(console)
 
1850
    
 
1851
    # Need to fork before connecting to D-Bus
 
1852
    if not debug:
 
1853
        # Close all input and output, do double fork, etc.
 
1854
        daemon()
1593
1855
    
1594
1856
    global main_loop
1595
1857
    # From the Avahi example code
1599
1861
    # End of Avahi example code
1600
1862
    if use_dbus:
1601
1863
        try:
1602
 
            bus_name = dbus.service.BusName(u"se.bsnet.fukt.Mandos",
 
1864
            bus_name = dbus.service.BusName("se.bsnet.fukt.Mandos",
1603
1865
                                            bus, do_not_queue=True)
1604
1866
        except dbus.exceptions.NameExistsException, e:
1605
 
            logger.error(unicode(e) + u", disabling D-Bus")
 
1867
            logger.error(unicode(e) + ", disabling D-Bus")
1606
1868
            use_dbus = False
1607
 
            server_settings[u"use_dbus"] = False
 
1869
            server_settings["use_dbus"] = False
1608
1870
            tcp_server.use_dbus = False
1609
1871
    protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
1610
 
    service = AvahiService(name = server_settings[u"servicename"],
1611
 
                           servicetype = u"_mandos._tcp",
 
1872
    service = AvahiService(name = server_settings["servicename"],
 
1873
                           servicetype = "_mandos._tcp",
1612
1874
                           protocol = protocol, bus = bus)
1613
1875
    if server_settings["interface"]:
1614
1876
        service.interface = (if_nametoindex
1615
 
                             (str(server_settings[u"interface"])))
 
1877
                             (str(server_settings["interface"])))
 
1878
    
 
1879
    global multiprocessing_manager
 
1880
    multiprocessing_manager = multiprocessing.Manager()
1616
1881
    
1617
1882
    client_class = Client
1618
1883
    if use_dbus:
1619
1884
        client_class = functools.partial(ClientDBus, bus = bus)
 
1885
    def client_config_items(config, section):
 
1886
        special_settings = {
 
1887
            "approved_by_default":
 
1888
                lambda: config.getboolean(section,
 
1889
                                          "approved_by_default"),
 
1890
            }
 
1891
        for name, value in config.items(section):
 
1892
            try:
 
1893
                yield (name, special_settings[name]())
 
1894
            except KeyError:
 
1895
                yield (name, value)
 
1896
    
1620
1897
    tcp_server.clients.update(set(
1621
1898
            client_class(name = section,
1622
 
                         config= dict(client_config.items(section)))
 
1899
                         config= dict(client_config_items(
 
1900
                        client_config, section)))
1623
1901
            for section in client_config.sections()))
1624
1902
    if not tcp_server.clients:
1625
 
        logger.warning(u"No clients defined")
1626
 
    
1627
 
    if debug:
1628
 
        # Redirect stdin so all checkers get /dev/null
1629
 
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
1630
 
        os.dup2(null, sys.stdin.fileno())
1631
 
        if null > 2:
1632
 
            os.close(null)
1633
 
    else:
1634
 
        # No console logging
1635
 
        logger.removeHandler(console)
1636
 
        # Close all input and output, do double fork, etc.
1637
 
        daemon()
1638
 
    
1639
 
    try:
1640
 
        with pidfile:
1641
 
            pid = os.getpid()
1642
 
            pidfile.write(str(pid) + "\n")
1643
 
        del pidfile
1644
 
    except IOError:
1645
 
        logger.error(u"Could not write to file %r with PID %d",
1646
 
                     pidfilename, pid)
1647
 
    except NameError:
1648
 
        # "pidfile" was never created
1649
 
        pass
1650
 
    del pidfilename
1651
 
    
 
1903
        logger.warning("No clients defined")
 
1904
        
1652
1905
    if not debug:
 
1906
        try:
 
1907
            with pidfile:
 
1908
                pid = os.getpid()
 
1909
                pidfile.write(str(pid) + "\n".encode("utf-8"))
 
1910
            del pidfile
 
1911
        except IOError:
 
1912
            logger.error("Could not write to file %r with PID %d",
 
1913
                         pidfilename, pid)
 
1914
        except NameError:
 
1915
            # "pidfile" was never created
 
1916
            pass
 
1917
        del pidfilename
 
1918
        
1653
1919
        signal.signal(signal.SIGINT, signal.SIG_IGN)
 
1920
 
1654
1921
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
1655
1922
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
1656
1923
    
1658
1925
        class MandosDBusService(dbus.service.Object):
1659
1926
            """A D-Bus proxy object"""
1660
1927
            def __init__(self):
1661
 
                dbus.service.Object.__init__(self, bus, u"/")
1662
 
            _interface = u"se.bsnet.fukt.Mandos"
 
1928
                dbus.service.Object.__init__(self, bus, "/")
 
1929
            _interface = "se.bsnet.fukt.Mandos"
1663
1930
            
1664
 
            @dbus.service.signal(_interface, signature=u"o")
 
1931
            @dbus.service.signal(_interface, signature="o")
1665
1932
            def ClientAdded(self, objpath):
1666
1933
                "D-Bus signal"
1667
1934
                pass
1668
1935
            
1669
 
            @dbus.service.signal(_interface, signature=u"ss")
 
1936
            @dbus.service.signal(_interface, signature="ss")
1670
1937
            def ClientNotFound(self, fingerprint, address):
1671
1938
                "D-Bus signal"
1672
1939
                pass
1673
1940
            
1674
 
            @dbus.service.signal(_interface, signature=u"os")
 
1941
            @dbus.service.signal(_interface, signature="os")
1675
1942
            def ClientRemoved(self, objpath, name):
1676
1943
                "D-Bus signal"
1677
1944
                pass
1678
1945
            
1679
 
            @dbus.service.method(_interface, out_signature=u"ao")
 
1946
            @dbus.service.method(_interface, out_signature="ao")
1680
1947
            def GetAllClients(self):
1681
1948
                "D-Bus method"
1682
1949
                return dbus.Array(c.dbus_object_path
1683
1950
                                  for c in tcp_server.clients)
1684
1951
            
1685
1952
            @dbus.service.method(_interface,
1686
 
                                 out_signature=u"a{oa{sv}}")
 
1953
                                 out_signature="a{oa{sv}}")
1687
1954
            def GetAllClientsWithProperties(self):
1688
1955
                "D-Bus method"
1689
1956
                return dbus.Dictionary(
1690
 
                    ((c.dbus_object_path, c.GetAll(u""))
 
1957
                    ((c.dbus_object_path, c.GetAll(""))
1691
1958
                     for c in tcp_server.clients),
1692
 
                    signature=u"oa{sv}")
 
1959
                    signature="oa{sv}")
1693
1960
            
1694
 
            @dbus.service.method(_interface, in_signature=u"o")
 
1961
            @dbus.service.method(_interface, in_signature="o")
1695
1962
            def RemoveClient(self, object_path):
1696
1963
                "D-Bus method"
1697
1964
                for c in tcp_server.clients:
1739
2006
    # Find out what port we got
1740
2007
    service.port = tcp_server.socket.getsockname()[1]
1741
2008
    if use_ipv6:
1742
 
        logger.info(u"Now listening on address %r, port %d,"
 
2009
        logger.info("Now listening on address %r, port %d,"
1743
2010
                    " flowinfo %d, scope_id %d"
1744
2011
                    % tcp_server.socket.getsockname())
1745
2012
    else:                       # IPv4
1746
 
        logger.info(u"Now listening on address %r, port %d"
 
2013
        logger.info("Now listening on address %r, port %d"
1747
2014
                    % tcp_server.socket.getsockname())
1748
2015
    
1749
2016
    #service.interface = tcp_server.socket.getsockname()[3]
1753
2020
        try:
1754
2021
            service.activate()
1755
2022
        except dbus.exceptions.DBusException, error:
1756
 
            logger.critical(u"DBusException: %s", error)
 
2023
            logger.critical("DBusException: %s", error)
1757
2024
            cleanup()
1758
2025
            sys.exit(1)
1759
2026
        # End of Avahi example code
1763
2030
                             (tcp_server.handle_request
1764
2031
                              (*args[2:], **kwargs) or True))
1765
2032
        
1766
 
        logger.debug(u"Starting main loop")
 
2033
        logger.debug("Starting main loop")
1767
2034
        main_loop.run()
1768
2035
    except AvahiError, error:
1769
 
        logger.critical(u"AvahiError: %s", error)
 
2036
        logger.critical("AvahiError: %s", error)
1770
2037
        cleanup()
1771
2038
        sys.exit(1)
1772
2039
    except KeyboardInterrupt:
1773
2040
        if debug:
1774
 
            print >> sys.stderr
1775
 
        logger.debug(u"Server received KeyboardInterrupt")
1776
 
    logger.debug(u"Server exiting")
 
2041
            print("", file=sys.stderr)
 
2042
        logger.debug("Server received KeyboardInterrupt")
 
2043
    logger.debug("Server exiting")
1777
2044
    # Must run before the D-Bus bus name gets deregistered
1778
2045
    cleanup()
1779
2046