/mandos/release

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

« back to all changes in this revision

Viewing changes to mandos

* mandos: Break long lines.

Show diffs side-by-side

added added

removed removed

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