/mandos/trunk

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

« back to all changes in this revision

Viewing changes to mandos

  • Committer: Teddy Hogeborn
  • Date: 2011-09-26 19:36:18 UTC
  • mfrom: (24.1.184 mandos)
  • Revision ID: teddy@fukt.bsnet.se-20110926193618-vtj5c9hena1maixx
Merge from Björn

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