/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-ctl (main): Bug fix: Handle empty strings correctly.

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-2012 Teddy Hogeborn
15
 
# Copyright © 2008-2012 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
34
34
from __future__ import (division, absolute_import, print_function,
35
35
                        unicode_literals)
36
36
 
37
 
from future_builtins import *
38
 
 
39
37
import SocketServer as socketserver
40
38
import socket
41
39
import argparse
65
63
import cPickle as pickle
66
64
import multiprocessing
67
65
import types
68
 
import binascii
69
 
import tempfile
70
 
import itertools
71
66
 
72
67
import dbus
73
68
import dbus.service
78
73
import ctypes.util
79
74
import xml.dom.minidom
80
75
import inspect
81
 
import GnuPGInterface
82
76
 
83
77
try:
84
78
    SO_BINDTODEVICE = socket.SO_BINDTODEVICE
88
82
    except ImportError:
89
83
        SO_BINDTODEVICE = None
90
84
 
91
 
version = "1.5.4"
92
 
stored_state_file = "clients.pickle"
93
 
 
94
 
logger = logging.getLogger()
 
85
 
 
86
version = "1.4.0"
 
87
 
 
88
#logger = logging.getLogger('mandos')
 
89
logger = logging.Logger('mandos')
95
90
syslogger = (logging.handlers.SysLogHandler
96
91
             (facility = logging.handlers.SysLogHandler.LOG_DAEMON,
97
92
              address = str("/dev/log")))
98
 
 
99
 
try:
100
 
    if_nametoindex = (ctypes.cdll.LoadLibrary
101
 
                      (ctypes.util.find_library("c"))
102
 
                      .if_nametoindex)
103
 
except (OSError, AttributeError):
104
 
    def if_nametoindex(interface):
105
 
        "Get an interface index the hard way, i.e. using fcntl()"
106
 
        SIOCGIFINDEX = 0x8933  # From /usr/include/linux/sockios.h
107
 
        with contextlib.closing(socket.socket()) as s:
108
 
            ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
109
 
                                struct.pack(str("16s16x"),
110
 
                                            interface))
111
 
        interface_index = struct.unpack(str("I"),
112
 
                                        ifreq[16:20])[0]
113
 
        return interface_index
114
 
 
115
 
 
116
 
def initlogger(debug, level=logging.WARNING):
117
 
    """init logger and add loglevel"""
118
 
    
119
 
    syslogger.setFormatter(logging.Formatter
120
 
                           ('Mandos [%(process)d]: %(levelname)s:'
121
 
                            ' %(message)s'))
122
 
    logger.addHandler(syslogger)
123
 
    
124
 
    if debug:
125
 
        console = logging.StreamHandler()
126
 
        console.setFormatter(logging.Formatter('%(asctime)s %(name)s'
127
 
                                               ' [%(process)d]:'
128
 
                                               ' %(levelname)s:'
129
 
                                               ' %(message)s'))
130
 
        logger.addHandler(console)
131
 
    logger.setLevel(level)
132
 
 
133
 
 
134
 
class PGPError(Exception):
135
 
    """Exception if encryption/decryption fails"""
136
 
    pass
137
 
 
138
 
 
139
 
class PGPEngine(object):
140
 
    """A simple class for OpenPGP symmetric encryption & decryption"""
141
 
    def __init__(self):
142
 
        self.gnupg = GnuPGInterface.GnuPG()
143
 
        self.tempdir = tempfile.mkdtemp(prefix="mandos-")
144
 
        self.gnupg = GnuPGInterface.GnuPG()
145
 
        self.gnupg.options.meta_interactive = False
146
 
        self.gnupg.options.homedir = self.tempdir
147
 
        self.gnupg.options.extra_args.extend(['--force-mdc',
148
 
                                              '--quiet',
149
 
                                              '--no-use-agent'])
150
 
    
151
 
    def __enter__(self):
152
 
        return self
153
 
    
154
 
    def __exit__(self, exc_type, exc_value, traceback):
155
 
        self._cleanup()
156
 
        return False
157
 
    
158
 
    def __del__(self):
159
 
        self._cleanup()
160
 
    
161
 
    def _cleanup(self):
162
 
        if self.tempdir is not None:
163
 
            # Delete contents of tempdir
164
 
            for root, dirs, files in os.walk(self.tempdir,
165
 
                                             topdown = False):
166
 
                for filename in files:
167
 
                    os.remove(os.path.join(root, filename))
168
 
                for dirname in dirs:
169
 
                    os.rmdir(os.path.join(root, dirname))
170
 
            # Remove tempdir
171
 
            os.rmdir(self.tempdir)
172
 
            self.tempdir = None
173
 
    
174
 
    def password_encode(self, password):
175
 
        # Passphrase can not be empty and can not contain newlines or
176
 
        # NUL bytes.  So we prefix it and hex encode it.
177
 
        return b"mandos" + binascii.hexlify(password)
178
 
    
179
 
    def encrypt(self, data, password):
180
 
        self.gnupg.passphrase = self.password_encode(password)
181
 
        with open(os.devnull, "w") as devnull:
182
 
            try:
183
 
                proc = self.gnupg.run(['--symmetric'],
184
 
                                      create_fhs=['stdin', 'stdout'],
185
 
                                      attach_fhs={'stderr': devnull})
186
 
                with contextlib.closing(proc.handles['stdin']) as f:
187
 
                    f.write(data)
188
 
                with contextlib.closing(proc.handles['stdout']) as f:
189
 
                    ciphertext = f.read()
190
 
                proc.wait()
191
 
            except IOError as e:
192
 
                raise PGPError(e)
193
 
        self.gnupg.passphrase = None
194
 
        return ciphertext
195
 
    
196
 
    def decrypt(self, data, password):
197
 
        self.gnupg.passphrase = self.password_encode(password)
198
 
        with open(os.devnull, "w") as devnull:
199
 
            try:
200
 
                proc = self.gnupg.run(['--decrypt'],
201
 
                                      create_fhs=['stdin', 'stdout'],
202
 
                                      attach_fhs={'stderr': devnull})
203
 
                with contextlib.closing(proc.handles['stdin']) as f:
204
 
                    f.write(data)
205
 
                with contextlib.closing(proc.handles['stdout']) as f:
206
 
                    decrypted_plaintext = f.read()
207
 
                proc.wait()
208
 
            except IOError as e:
209
 
                raise PGPError(e)
210
 
        self.gnupg.passphrase = None
211
 
        return decrypted_plaintext
212
 
 
 
93
syslogger.setFormatter(logging.Formatter
 
94
                       ('Mandos [%(process)d]: %(levelname)s:'
 
95
                        ' %(message)s'))
 
96
logger.addHandler(syslogger)
 
97
 
 
98
console = logging.StreamHandler()
 
99
console.setFormatter(logging.Formatter('%(name)s [%(process)d]:'
 
100
                                       ' %(levelname)s:'
 
101
                                       ' %(message)s'))
 
102
logger.addHandler(console)
213
103
 
214
104
class AvahiError(Exception):
215
105
    def __init__(self, value, *args, **kwargs):
245
135
    server: D-Bus Server
246
136
    bus: dbus.SystemBus()
247
137
    """
248
 
    
249
138
    def __init__(self, interface = avahi.IF_UNSPEC, name = None,
250
139
                 servicetype = None, port = None, TXT = None,
251
140
                 domain = "", host = "", max_renames = 32768,
264
153
        self.server = None
265
154
        self.bus = bus
266
155
        self.entry_group_state_changed_match = None
267
 
    
268
156
    def rename(self):
269
157
        """Derived from the Avahi example code"""
270
158
        if self.rename_count >= self.max_renames:
276
164
                            .GetAlternativeServiceName(self.name))
277
165
        logger.info("Changing Zeroconf service name to %r ...",
278
166
                    self.name)
 
167
        syslogger.setFormatter(logging.Formatter
 
168
                               ('Mandos (%s) [%%(process)d]:'
 
169
                                ' %%(levelname)s: %%(message)s'
 
170
                                % self.name))
279
171
        self.remove()
280
172
        try:
281
173
            self.add()
282
174
        except dbus.exceptions.DBusException as error:
283
 
            logger.critical("D-Bus Exception", exc_info=error)
 
175
            logger.critical("DBusException: %s", error)
284
176
            self.cleanup()
285
177
            os._exit(1)
286
178
        self.rename_count += 1
287
 
    
288
179
    def remove(self):
289
180
        """Derived from the Avahi example code"""
290
181
        if self.entry_group_state_changed_match is not None:
292
183
            self.entry_group_state_changed_match = None
293
184
        if self.group is not None:
294
185
            self.group.Reset()
295
 
    
296
186
    def add(self):
297
187
        """Derived from the Avahi example code"""
298
188
        self.remove()
303
193
                avahi.DBUS_INTERFACE_ENTRY_GROUP)
304
194
        self.entry_group_state_changed_match = (
305
195
            self.group.connect_to_signal(
306
 
                'StateChanged', self.entry_group_state_changed))
 
196
                'StateChanged', self .entry_group_state_changed))
307
197
        logger.debug("Adding Zeroconf service '%s' of type '%s' ...",
308
198
                     self.name, self.type)
309
199
        self.group.AddService(
315
205
            dbus.UInt16(self.port),
316
206
            avahi.string_array_to_txt_array(self.TXT))
317
207
        self.group.Commit()
318
 
    
319
208
    def entry_group_state_changed(self, state, error):
320
209
        """Derived from the Avahi example code"""
321
210
        logger.debug("Avahi entry group state change: %i", state)
328
217
        elif state == avahi.ENTRY_GROUP_FAILURE:
329
218
            logger.critical("Avahi: Error in group state changed %s",
330
219
                            unicode(error))
331
 
            raise AvahiGroupError("State changed: {0!s}"
332
 
                                  .format(error))
333
 
    
 
220
            raise AvahiGroupError("State changed: %s"
 
221
                                  % unicode(error))
334
222
    def cleanup(self):
335
223
        """Derived from the Avahi example code"""
336
224
        if self.group is not None:
337
225
            try:
338
226
                self.group.Free()
339
227
            except (dbus.exceptions.UnknownMethodException,
340
 
                    dbus.exceptions.DBusException):
 
228
                    dbus.exceptions.DBusException) as e:
341
229
                pass
342
230
            self.group = None
343
231
        self.remove()
344
 
    
345
232
    def server_state_changed(self, state, error=None):
346
233
        """Derived from the Avahi example code"""
347
234
        logger.debug("Avahi server state change: %i", state)
366
253
                logger.debug("Unknown state: %r", state)
367
254
            else:
368
255
                logger.debug("Unknown state: %r: %r", state, error)
369
 
    
370
256
    def activate(self):
371
257
        """Derived from the Avahi example code"""
372
258
        if self.server is None:
380
266
        self.server_state_changed(self.server.GetState())
381
267
 
382
268
 
383
 
class AvahiServiceToSyslog(AvahiService):
384
 
    def rename(self):
385
 
        """Add the new name to the syslog messages"""
386
 
        ret = AvahiService.rename(self)
387
 
        syslogger.setFormatter(logging.Formatter
388
 
                               ('Mandos ({0}) [%(process)d]:'
389
 
                                ' %(levelname)s: %(message)s'
390
 
                                .format(self.name)))
391
 
        return ret
392
 
 
393
 
 
394
 
def timedelta_to_milliseconds(td):
 
269
def _timedelta_to_milliseconds(td):
395
270
    "Convert a datetime.timedelta() to milliseconds"
396
271
    return ((td.days * 24 * 60 * 60 * 1000)
397
272
            + (td.seconds * 1000)
398
273
            + (td.microseconds // 1000))
399
 
 
400
 
 
 
274
        
401
275
class Client(object):
402
276
    """A representation of a client host served by this server.
403
277
    
404
278
    Attributes:
405
 
    approved:   bool(); 'None' if not yet approved/disapproved
 
279
    _approved:   bool(); 'None' if not yet approved/disapproved
406
280
    approval_delay: datetime.timedelta(); Time to wait for approval
407
281
    approval_duration: datetime.timedelta(); Duration of one approval
408
282
    checker:    subprocess.Popen(); a running checker process used
415
289
                     instance %(name)s can be used in the command.
416
290
    checker_initiator_tag: a gobject event source tag, or None
417
291
    created:    datetime.datetime(); (UTC) object creation
418
 
    client_structure: Object describing what attributes a client has
419
 
                      and is used for storing the client at exit
420
292
    current_checker_command: string; current running checker_command
 
293
    disable_hook:  If set, called by disable() as disable_hook(self)
421
294
    disable_initiator_tag: a gobject event source tag, or None
422
295
    enabled:    bool()
423
296
    fingerprint: string (40 or 32 hexadecimal digits); used to
426
299
    interval:   datetime.timedelta(); How often to start a new checker
427
300
    last_approval_request: datetime.datetime(); (UTC) or None
428
301
    last_checked_ok: datetime.datetime(); (UTC) or None
429
 
    last_checker_status: integer between 0 and 255 reflecting exit
430
 
                         status of last checker. -1 reflects crashed
431
 
                         checker, -2 means no checker completed yet.
432
 
    last_enabled: datetime.datetime(); (UTC) or None
 
302
    last_enabled: datetime.datetime(); (UTC)
433
303
    name:       string; from the config file, used in log messages and
434
304
                        D-Bus identifiers
435
305
    secret:     bytestring; sent verbatim (over TLS) to client
436
306
    timeout:    datetime.timedelta(); How long from last_checked_ok
437
307
                                      until this client is disabled
438
 
    extended_timeout:   extra long timeout when secret has been sent
 
308
    extended_timeout:   extra long timeout when password has been sent
439
309
    runtime_expansions: Allowed attributes for runtime expansion.
440
310
    expires:    datetime.datetime(); time (UTC) when a client will be
441
311
                disabled, or None
442
312
    """
443
313
    
444
314
    runtime_expansions = ("approval_delay", "approval_duration",
445
 
                          "created", "enabled", "expires",
446
 
                          "fingerprint", "host", "interval",
447
 
                          "last_approval_request", "last_checked_ok",
 
315
                          "created", "enabled", "fingerprint",
 
316
                          "host", "interval", "last_checked_ok",
448
317
                          "last_enabled", "name", "timeout")
449
 
    client_defaults = { "timeout": "5m",
450
 
                        "extended_timeout": "15m",
451
 
                        "interval": "2m",
452
 
                        "checker": "fping -q -- %%(host)s",
453
 
                        "host": "",
454
 
                        "approval_delay": "0s",
455
 
                        "approval_duration": "1s",
456
 
                        "approved_by_default": "True",
457
 
                        "enabled": "True",
458
 
                        }
459
318
    
460
319
    def timeout_milliseconds(self):
461
320
        "Return the 'timeout' attribute in milliseconds"
462
 
        return timedelta_to_milliseconds(self.timeout)
 
321
        return _timedelta_to_milliseconds(self.timeout)
463
322
    
464
323
    def extended_timeout_milliseconds(self):
465
324
        "Return the 'extended_timeout' attribute in milliseconds"
466
 
        return timedelta_to_milliseconds(self.extended_timeout)
 
325
        return _timedelta_to_milliseconds(self.extended_timeout)
467
326
    
468
327
    def interval_milliseconds(self):
469
328
        "Return the 'interval' attribute in milliseconds"
470
 
        return timedelta_to_milliseconds(self.interval)
 
329
        return _timedelta_to_milliseconds(self.interval)
471
330
    
472
331
    def approval_delay_milliseconds(self):
473
 
        return timedelta_to_milliseconds(self.approval_delay)
474
 
    
475
 
    @staticmethod
476
 
    def config_parser(config):
477
 
        """Construct a new dict of client settings of this form:
478
 
        { client_name: {setting_name: value, ...}, ...}
479
 
        with exceptions for any special settings as defined above.
480
 
        NOTE: Must be a pure function. Must return the same result
481
 
        value given the same arguments.
482
 
        """
483
 
        settings = {}
484
 
        for client_name in config.sections():
485
 
            section = dict(config.items(client_name))
486
 
            client = settings[client_name] = {}
487
 
            
488
 
            client["host"] = section["host"]
489
 
            # Reformat values from string types to Python types
490
 
            client["approved_by_default"] = config.getboolean(
491
 
                client_name, "approved_by_default")
492
 
            client["enabled"] = config.getboolean(client_name,
493
 
                                                  "enabled")
494
 
            
495
 
            client["fingerprint"] = (section["fingerprint"].upper()
496
 
                                     .replace(" ", ""))
497
 
            if "secret" in section:
498
 
                client["secret"] = section["secret"].decode("base64")
499
 
            elif "secfile" in section:
500
 
                with open(os.path.expanduser(os.path.expandvars
501
 
                                             (section["secfile"])),
502
 
                          "rb") as secfile:
503
 
                    client["secret"] = secfile.read()
504
 
            else:
505
 
                raise TypeError("No secret or secfile for section {0}"
506
 
                                .format(section))
507
 
            client["timeout"] = string_to_delta(section["timeout"])
508
 
            client["extended_timeout"] = string_to_delta(
509
 
                section["extended_timeout"])
510
 
            client["interval"] = string_to_delta(section["interval"])
511
 
            client["approval_delay"] = string_to_delta(
512
 
                section["approval_delay"])
513
 
            client["approval_duration"] = string_to_delta(
514
 
                section["approval_duration"])
515
 
            client["checker_command"] = section["checker"]
516
 
            client["last_approval_request"] = None
517
 
            client["last_checked_ok"] = None
518
 
            client["last_checker_status"] = -2
519
 
        
520
 
        return settings
521
 
    
522
 
    def __init__(self, settings, name = None):
 
332
        return _timedelta_to_milliseconds(self.approval_delay)
 
333
    
 
334
    def __init__(self, name = None, disable_hook=None, config=None):
 
335
        """Note: the 'checker' key in 'config' sets the
 
336
        'checker_command' attribute and *not* the 'checker'
 
337
        attribute."""
523
338
        self.name = name
524
 
        # adding all client settings
525
 
        for setting, value in settings.iteritems():
526
 
            setattr(self, setting, value)
527
 
        
528
 
        if self.enabled:
529
 
            if not hasattr(self, "last_enabled"):
530
 
                self.last_enabled = datetime.datetime.utcnow()
531
 
            if not hasattr(self, "expires"):
532
 
                self.expires = (datetime.datetime.utcnow()
533
 
                                + self.timeout)
534
 
        else:
535
 
            self.last_enabled = None
536
 
            self.expires = None
537
 
        
 
339
        if config is None:
 
340
            config = {}
538
341
        logger.debug("Creating client %r", self.name)
539
342
        # Uppercase and remove spaces from fingerprint for later
540
343
        # comparison purposes with return value from the fingerprint()
541
344
        # function
 
345
        self.fingerprint = (config["fingerprint"].upper()
 
346
                            .replace(" ", ""))
542
347
        logger.debug("  Fingerprint: %s", self.fingerprint)
543
 
        self.created = settings.get("created",
544
 
                                    datetime.datetime.utcnow())
545
 
        
546
 
        # attributes specific for this server instance
 
348
        if "secret" in config:
 
349
            self.secret = config["secret"].decode("base64")
 
350
        elif "secfile" in config:
 
351
            with open(os.path.expanduser(os.path.expandvars
 
352
                                         (config["secfile"])),
 
353
                      "rb") as secfile:
 
354
                self.secret = secfile.read()
 
355
        else:
 
356
            raise TypeError("No secret or secfile for client %s"
 
357
                            % self.name)
 
358
        self.host = config.get("host", "")
 
359
        self.created = datetime.datetime.utcnow()
 
360
        self.enabled = False
 
361
        self.last_approval_request = None
 
362
        self.last_enabled = None
 
363
        self.last_checked_ok = None
 
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"])
 
368
        self.disable_hook = disable_hook
547
369
        self.checker = None
548
370
        self.checker_initiator_tag = None
549
371
        self.disable_initiator_tag = None
 
372
        self.expires = None
550
373
        self.checker_callback_tag = None
 
374
        self.checker_command = config["checker"]
551
375
        self.current_checker_command = None
552
 
        self.approved = None
 
376
        self.last_connect = None
 
377
        self._approved = None
 
378
        self.approved_by_default = config.get("approved_by_default",
 
379
                                              True)
553
380
        self.approvals_pending = 0
 
381
        self.approval_delay = string_to_delta(
 
382
            config["approval_delay"])
 
383
        self.approval_duration = string_to_delta(
 
384
            config["approval_duration"])
554
385
        self.changedstate = (multiprocessing_manager
555
386
                             .Condition(multiprocessing_manager
556
387
                                        .Lock()))
557
 
        self.client_structure = [attr for attr in
558
 
                                 self.__dict__.iterkeys()
559
 
                                 if not attr.startswith("_")]
560
 
        self.client_structure.append("client_structure")
561
 
        
562
 
        for name, t in inspect.getmembers(type(self),
563
 
                                          lambda obj:
564
 
                                              isinstance(obj,
565
 
                                                         property)):
566
 
            if not name.startswith("_"):
567
 
                self.client_structure.append(name)
568
388
    
569
 
    # Send notice to process children that client state has changed
570
389
    def send_changedstate(self):
571
 
        with self.changedstate:
572
 
            self.changedstate.notify_all()
 
390
        self.changedstate.acquire()
 
391
        self.changedstate.notify_all()
 
392
        self.changedstate.release()
573
393
    
574
394
    def enable(self):
575
395
        """Start this client's checker and timeout hooks"""
576
396
        if getattr(self, "enabled", False):
577
397
            # Already enabled
578
398
            return
 
399
        self.send_changedstate()
 
400
        # Schedule a new checker to be started an 'interval' from now,
 
401
        # and every interval from then on.
 
402
        self.checker_initiator_tag = (gobject.timeout_add
 
403
                                      (self.interval_milliseconds(),
 
404
                                       self.start_checker))
 
405
        # Schedule a disable() when 'timeout' has passed
579
406
        self.expires = datetime.datetime.utcnow() + self.timeout
 
407
        self.disable_initiator_tag = (gobject.timeout_add
 
408
                                   (self.timeout_milliseconds(),
 
409
                                    self.disable))
580
410
        self.enabled = True
581
411
        self.last_enabled = datetime.datetime.utcnow()
582
 
        self.init_checker()
583
 
        self.send_changedstate()
 
412
        # Also start a new checker *right now*.
 
413
        self.start_checker()
584
414
    
585
415
    def disable(self, quiet=True):
586
416
        """Disable this client."""
587
417
        if not getattr(self, "enabled", False):
588
418
            return False
589
419
        if not quiet:
 
420
            self.send_changedstate()
 
421
        if not quiet:
590
422
            logger.info("Disabling client %s", self.name)
591
 
        if getattr(self, "disable_initiator_tag", None) is not None:
 
423
        if getattr(self, "disable_initiator_tag", False):
592
424
            gobject.source_remove(self.disable_initiator_tag)
593
425
            self.disable_initiator_tag = None
594
426
        self.expires = None
595
 
        if getattr(self, "checker_initiator_tag", None) is not None:
 
427
        if getattr(self, "checker_initiator_tag", False):
596
428
            gobject.source_remove(self.checker_initiator_tag)
597
429
            self.checker_initiator_tag = None
598
430
        self.stop_checker()
 
431
        if self.disable_hook:
 
432
            self.disable_hook(self)
599
433
        self.enabled = False
600
 
        if not quiet:
601
 
            self.send_changedstate()
602
434
        # Do not run this again if called by a gobject.timeout_add
603
435
        return False
604
436
    
605
437
    def __del__(self):
 
438
        self.disable_hook = None
606
439
        self.disable()
607
440
    
608
 
    def init_checker(self):
609
 
        # Schedule a new checker to be started an 'interval' from now,
610
 
        # and every interval from then on.
611
 
        if self.checker_initiator_tag is not None:
612
 
            gobject.source_remove(self.checker_initiator_tag)
613
 
        self.checker_initiator_tag = (gobject.timeout_add
614
 
                                      (self.interval_milliseconds(),
615
 
                                       self.start_checker))
616
 
        # Schedule a disable() when 'timeout' has passed
617
 
        if self.disable_initiator_tag is not None:
618
 
            gobject.source_remove(self.disable_initiator_tag)
619
 
        self.disable_initiator_tag = (gobject.timeout_add
620
 
                                   (self.timeout_milliseconds(),
621
 
                                    self.disable))
622
 
        # Also start a new checker *right now*.
623
 
        self.start_checker()
624
 
    
625
441
    def checker_callback(self, pid, condition, command):
626
442
        """The checker has completed, so take appropriate actions."""
627
443
        self.checker_callback_tag = None
628
444
        self.checker = None
629
445
        if os.WIFEXITED(condition):
630
 
            self.last_checker_status = os.WEXITSTATUS(condition)
631
 
            if self.last_checker_status == 0:
 
446
            exitstatus = os.WEXITSTATUS(condition)
 
447
            if exitstatus == 0:
632
448
                logger.info("Checker for %(name)s succeeded",
633
449
                            vars(self))
634
450
                self.checked_ok()
636
452
                logger.info("Checker for %(name)s failed",
637
453
                            vars(self))
638
454
        else:
639
 
            self.last_checker_status = -1
640
455
            logger.warning("Checker for %(name)s crashed?",
641
456
                           vars(self))
642
457
    
643
 
    def checked_ok(self):
644
 
        """Assert that the client has been seen, alive and well."""
645
 
        self.last_checked_ok = datetime.datetime.utcnow()
646
 
        self.last_checker_status = 0
647
 
        self.bump_timeout()
648
 
    
649
 
    def bump_timeout(self, timeout=None):
650
 
        """Bump up the timeout for this client."""
 
458
    def checked_ok(self, timeout=None):
 
459
        """Bump up the timeout for this client.
 
460
        
 
461
        This should only be called when the client has been seen,
 
462
        alive and well.
 
463
        """
651
464
        if timeout is None:
652
465
            timeout = self.timeout
653
 
        if self.disable_initiator_tag is not None:
654
 
            gobject.source_remove(self.disable_initiator_tag)
655
 
            self.disable_initiator_tag = None
656
 
        if getattr(self, "enabled", False):
657
 
            self.disable_initiator_tag = (gobject.timeout_add
658
 
                                          (timedelta_to_milliseconds
659
 
                                           (timeout), self.disable))
660
 
            self.expires = datetime.datetime.utcnow() + timeout
 
466
        self.last_checked_ok = datetime.datetime.utcnow()
 
467
        gobject.source_remove(self.disable_initiator_tag)
 
468
        self.disable_initiator_tag = (gobject.timeout_add
 
469
                                      (_timedelta_to_milliseconds
 
470
                                       (timeout), self.disable))
 
471
        self.expires = datetime.datetime.utcnow() + timeout
661
472
    
662
473
    def need_approval(self):
663
474
        self.last_approval_request = datetime.datetime.utcnow()
668
479
        If a checker already exists, leave it running and do
669
480
        nothing."""
670
481
        # The reason for not killing a running checker is that if we
671
 
        # did that, and if a checker (for some reason) started running
672
 
        # slowly and taking more than 'interval' time, then the client
673
 
        # would inevitably timeout, since no checker would get a
674
 
        # chance to run to completion.  If we instead leave running
 
482
        # did that, then if a checker (for some reason) started
 
483
        # running slowly and taking more than 'interval' time, the
 
484
        # client would inevitably timeout, since no checker would get
 
485
        # a chance to run to completion.  If we instead leave running
675
486
        # checkers alone, the checker would have to take more time
676
487
        # than 'timeout' for the client to be disabled, which is as it
677
488
        # should be.
691
502
                                      self.current_checker_command)
692
503
        # Start a new checker if needed
693
504
        if self.checker is None:
694
 
            # Escape attributes for the shell
695
 
            escaped_attrs = dict(
696
 
                (attr, re.escape(unicode(getattr(self, attr))))
697
 
                for attr in
698
 
                self.runtime_expansions)
699
505
            try:
700
 
                command = self.checker_command % escaped_attrs
701
 
            except TypeError as error:
702
 
                logger.error('Could not format string "%s"',
703
 
                             self.checker_command, exc_info=error)
704
 
                return True # Try again later
 
506
                # In case checker_command has exactly one % operator
 
507
                command = self.checker_command % self.host
 
508
            except TypeError:
 
509
                # Escape attributes for the shell
 
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
                
 
518
                try:
 
519
                    command = self.checker_command % escaped_attrs
 
520
                except TypeError as error:
 
521
                    logger.error('Could not format string "%s":'
 
522
                                 ' %s', self.checker_command, error)
 
523
                    return True # Try again later
705
524
            self.current_checker_command = command
706
525
            try:
707
526
                logger.info("Starting checker %r for %s",
713
532
                self.checker = subprocess.Popen(command,
714
533
                                                close_fds=True,
715
534
                                                shell=True, cwd="/")
 
535
                self.checker_callback_tag = (gobject.child_watch_add
 
536
                                             (self.checker.pid,
 
537
                                              self.checker_callback,
 
538
                                              data=command))
 
539
                # The checker may have completed before the gobject
 
540
                # watch was added.  Check for this.
 
541
                pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
 
542
                if pid:
 
543
                    gobject.source_remove(self.checker_callback_tag)
 
544
                    self.checker_callback(pid, status, command)
716
545
            except OSError as error:
717
 
                logger.error("Failed to start subprocess",
718
 
                             exc_info=error)
719
 
            self.checker_callback_tag = (gobject.child_watch_add
720
 
                                         (self.checker.pid,
721
 
                                          self.checker_callback,
722
 
                                          data=command))
723
 
            # The checker may have completed before the gobject
724
 
            # watch was added.  Check for this.
725
 
            pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
726
 
            if pid:
727
 
                gobject.source_remove(self.checker_callback_tag)
728
 
                self.checker_callback(pid, status, command)
 
546
                logger.error("Failed to start subprocess: %s",
 
547
                             error)
729
548
        # Re-run this periodically if run by gobject.timeout_add
730
549
        return True
731
550
    
738
557
            return
739
558
        logger.debug("Stopping checker for %(name)s", vars(self))
740
559
        try:
741
 
            self.checker.terminate()
 
560
            os.kill(self.checker.pid, signal.SIGTERM)
742
561
            #time.sleep(0.5)
743
562
            #if self.checker.poll() is None:
744
 
            #    self.checker.kill()
 
563
            #    os.kill(self.checker.pid, signal.SIGKILL)
745
564
        except OSError as error:
746
565
            if error.errno != errno.ESRCH: # No such process
747
566
                raise
764
583
    # "Set" method, so we fail early here:
765
584
    if byte_arrays and signature != "ay":
766
585
        raise ValueError("Byte arrays not supported for non-'ay'"
767
 
                         " signature {0!r}".format(signature))
 
586
                         " signature %r" % signature)
768
587
    def decorator(func):
769
588
        func._dbus_is_property = True
770
589
        func._dbus_interface = dbus_interface
778
597
    return decorator
779
598
 
780
599
 
781
 
def dbus_interface_annotations(dbus_interface):
782
 
    """Decorator for marking functions returning interface annotations
783
 
    
784
 
    Usage:
785
 
    
786
 
    @dbus_interface_annotations("org.example.Interface")
787
 
    def _foo(self):  # Function name does not matter
788
 
        return {"org.freedesktop.DBus.Deprecated": "true",
789
 
                "org.freedesktop.DBus.Property.EmitsChangedSignal":
790
 
                    "false"}
791
 
    """
792
 
    def decorator(func):
793
 
        func._dbus_is_interface = True
794
 
        func._dbus_interface = dbus_interface
795
 
        func._dbus_name = dbus_interface
796
 
        return func
797
 
    return decorator
798
 
 
799
 
 
800
 
def dbus_annotations(annotations):
801
 
    """Decorator to annotate D-Bus methods, signals or properties
802
 
    Usage:
803
 
    
804
 
    @dbus_service_property("org.example.Interface", signature="b",
805
 
                           access="r")
806
 
    @dbus_annotations({{"org.freedesktop.DBus.Deprecated": "true",
807
 
                        "org.freedesktop.DBus.Property."
808
 
                        "EmitsChangedSignal": "false"})
809
 
    def Property_dbus_property(self):
810
 
        return dbus.Boolean(False)
811
 
    """
812
 
    def decorator(func):
813
 
        func._dbus_annotations = annotations
814
 
        return func
815
 
    return decorator
816
 
 
817
 
 
818
600
class DBusPropertyException(dbus.exceptions.DBusException):
819
601
    """A base class for D-Bus property-related exceptions
820
602
    """
843
625
    """
844
626
    
845
627
    @staticmethod
846
 
    def _is_dbus_thing(thing):
847
 
        """Returns a function testing if an attribute is a D-Bus thing
848
 
        
849
 
        If called like _is_dbus_thing("method") it returns a function
850
 
        suitable for use as predicate to inspect.getmembers().
851
 
        """
852
 
        return lambda obj: getattr(obj, "_dbus_is_{0}".format(thing),
853
 
                                   False)
 
628
    def _is_dbus_property(obj):
 
629
        return getattr(obj, "_dbus_is_property", False)
854
630
    
855
 
    def _get_all_dbus_things(self, thing):
 
631
    def _get_all_dbus_properties(self):
856
632
        """Returns a generator of (name, attribute) pairs
857
633
        """
858
 
        return ((getattr(athing.__get__(self), "_dbus_name",
859
 
                         name),
860
 
                 athing.__get__(self))
 
634
        return ((prop.__get__(self)._dbus_name, prop.__get__(self))
861
635
                for cls in self.__class__.__mro__
862
 
                for name, athing in
863
 
                inspect.getmembers(cls,
864
 
                                   self._is_dbus_thing(thing)))
 
636
                for name, prop in
 
637
                inspect.getmembers(cls, self._is_dbus_property))
865
638
    
866
639
    def _get_dbus_property(self, interface_name, property_name):
867
640
        """Returns a bound method if one exists which is a D-Bus
869
642
        """
870
643
        for cls in  self.__class__.__mro__:
871
644
            for name, value in (inspect.getmembers
872
 
                                (cls,
873
 
                                 self._is_dbus_thing("property"))):
 
645
                                (cls, self._is_dbus_property)):
874
646
                if (value._dbus_name == property_name
875
647
                    and value._dbus_interface == interface_name):
876
648
                    return value.__get__(self)
905
677
            # signatures other than "ay".
906
678
            if prop._dbus_signature != "ay":
907
679
                raise ValueError
908
 
            value = dbus.ByteArray(b''.join(chr(byte)
909
 
                                            for byte in value))
 
680
            value = dbus.ByteArray(''.join(unichr(byte)
 
681
                                           for byte in value))
910
682
        prop(value)
911
683
    
912
684
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="s",
917
689
        
918
690
        Note: Will not include properties with access="write".
919
691
        """
920
 
        properties = {}
921
 
        for name, prop in self._get_all_dbus_things("property"):
 
692
        all = {}
 
693
        for name, prop in self._get_all_dbus_properties():
922
694
            if (interface_name
923
695
                and interface_name != prop._dbus_interface):
924
696
                # Interface non-empty but did not match
928
700
                continue
929
701
            value = prop()
930
702
            if not hasattr(value, "variant_level"):
931
 
                properties[name] = value
 
703
                all[name] = value
932
704
                continue
933
 
            properties[name] = type(value)(value, variant_level=
934
 
                                           value.variant_level+1)
935
 
        return dbus.Dictionary(properties, signature="sv")
 
705
            all[name] = type(value)(value, variant_level=
 
706
                                    value.variant_level+1)
 
707
        return dbus.Dictionary(all, signature="sv")
936
708
    
937
709
    @dbus.service.method(dbus.INTROSPECTABLE_IFACE,
938
710
                         out_signature="s",
939
711
                         path_keyword='object_path',
940
712
                         connection_keyword='connection')
941
713
    def Introspect(self, object_path, connection):
942
 
        """Overloading of standard D-Bus method.
943
 
        
944
 
        Inserts property tags and interface annotation tags.
 
714
        """Standard D-Bus method, overloaded to insert property tags.
945
715
        """
946
716
        xmlstring = dbus.service.Object.Introspect(self, object_path,
947
717
                                                   connection)
954
724
                e.setAttribute("access", prop._dbus_access)
955
725
                return e
956
726
            for if_tag in document.getElementsByTagName("interface"):
957
 
                # Add property tags
958
727
                for tag in (make_tag(document, name, prop)
959
728
                            for name, prop
960
 
                            in self._get_all_dbus_things("property")
 
729
                            in self._get_all_dbus_properties()
961
730
                            if prop._dbus_interface
962
731
                            == if_tag.getAttribute("name")):
963
732
                    if_tag.appendChild(tag)
964
 
                # Add annotation tags
965
 
                for typ in ("method", "signal", "property"):
966
 
                    for tag in if_tag.getElementsByTagName(typ):
967
 
                        annots = dict()
968
 
                        for name, prop in (self.
969
 
                                           _get_all_dbus_things(typ)):
970
 
                            if (name == tag.getAttribute("name")
971
 
                                and prop._dbus_interface
972
 
                                == if_tag.getAttribute("name")):
973
 
                                annots.update(getattr
974
 
                                              (prop,
975
 
                                               "_dbus_annotations",
976
 
                                               {}))
977
 
                        for name, value in annots.iteritems():
978
 
                            ann_tag = document.createElement(
979
 
                                "annotation")
980
 
                            ann_tag.setAttribute("name", name)
981
 
                            ann_tag.setAttribute("value", value)
982
 
                            tag.appendChild(ann_tag)
983
 
                # Add interface annotation tags
984
 
                for annotation, value in dict(
985
 
                    itertools.chain.from_iterable(
986
 
                        annotations().iteritems()
987
 
                        for name, annotations in
988
 
                        self._get_all_dbus_things("interface")
989
 
                        if name == if_tag.getAttribute("name")
990
 
                        )).iteritems():
991
 
                    ann_tag = document.createElement("annotation")
992
 
                    ann_tag.setAttribute("name", annotation)
993
 
                    ann_tag.setAttribute("value", value)
994
 
                    if_tag.appendChild(ann_tag)
995
733
                # Add the names to the return values for the
996
734
                # "org.freedesktop.DBus.Properties" methods
997
735
                if (if_tag.getAttribute("name")
1012
750
        except (AttributeError, xml.dom.DOMException,
1013
751
                xml.parsers.expat.ExpatError) as error:
1014
752
            logger.error("Failed to override Introspection method",
1015
 
                         exc_info=error)
 
753
                         error)
1016
754
        return xmlstring
1017
755
 
1018
756
 
1019
 
def datetime_to_dbus(dt, variant_level=0):
 
757
def datetime_to_dbus (dt, variant_level=0):
1020
758
    """Convert a UTC datetime.datetime() to a D-Bus type."""
1021
759
    if dt is None:
1022
760
        return dbus.String("", variant_level = variant_level)
1023
761
    return dbus.String(dt.isoformat(),
1024
762
                       variant_level=variant_level)
1025
763
 
1026
 
 
1027
 
def alternate_dbus_interfaces(alt_interface_names, deprecate=True):
1028
 
    """A class decorator; applied to a subclass of
1029
 
    dbus.service.Object, it will add alternate D-Bus attributes with
1030
 
    interface names according to the "alt_interface_names" mapping.
1031
 
    Usage:
1032
 
    
1033
 
    @alternate_dbus_interfaces({"org.example.Interface":
1034
 
                                    "net.example.AlternateInterface"})
1035
 
    class SampleDBusObject(dbus.service.Object):
1036
 
        @dbus.service.method("org.example.Interface")
1037
 
        def SampleDBusMethod():
1038
 
            pass
1039
 
    
1040
 
    The above "SampleDBusMethod" on "SampleDBusObject" will be
1041
 
    reachable via two interfaces: "org.example.Interface" and
1042
 
    "net.example.AlternateInterface", the latter of which will have
1043
 
    its D-Bus annotation "org.freedesktop.DBus.Deprecated" set to
1044
 
    "true", unless "deprecate" is passed with a False value.
1045
 
    
1046
 
    This works for methods and signals, and also for D-Bus properties
1047
 
    (from DBusObjectWithProperties) and interfaces (from the
1048
 
    dbus_interface_annotations decorator).
 
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.
1049
768
    """
1050
 
    def wrapper(cls):
1051
 
        for orig_interface_name, alt_interface_name in (
1052
 
            alt_interface_names.iteritems()):
1053
 
            attr = {}
1054
 
            interface_names = set()
1055
 
            # Go though all attributes of the class
1056
 
            for attrname, attribute in inspect.getmembers(cls):
 
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):
1057
776
                # Ignore non-D-Bus attributes, and D-Bus attributes
1058
777
                # with the wrong interface name
1059
778
                if (not hasattr(attribute, "_dbus_interface")
1060
779
                    or not attribute._dbus_interface
1061
 
                    .startswith(orig_interface_name)):
 
780
                    .startswith("se.recompile.Mandos")):
1062
781
                    continue
1063
782
                # Create an alternate D-Bus interface name based on
1064
783
                # the current name
1065
784
                alt_interface = (attribute._dbus_interface
1066
 
                                 .replace(orig_interface_name,
1067
 
                                          alt_interface_name))
1068
 
                interface_names.add(alt_interface)
 
785
                                 .replace("se.recompile.Mandos",
 
786
                                          "se.bsnet.fukt.Mandos"))
1069
787
                # Is this a D-Bus signal?
1070
788
                if getattr(attribute, "_dbus_is_signal", False):
1071
789
                    # Extract the original non-method function by
1086
804
                                nonmethod_func.func_name,
1087
805
                                nonmethod_func.func_defaults,
1088
806
                                nonmethod_func.func_closure)))
1089
 
                    # Copy annotations, if any
1090
 
                    try:
1091
 
                        new_function._dbus_annotations = (
1092
 
                            dict(attribute._dbus_annotations))
1093
 
                    except AttributeError:
1094
 
                        pass
1095
807
                    # Define a creator of a function to call both the
1096
 
                    # original and alternate functions, so both the
1097
 
                    # original and alternate signals gets sent when
1098
 
                    # the function is called
 
808
                    # old and new functions, so both the old and new
 
809
                    # signals gets sent when the function is called
1099
810
                    def fixscope(func1, func2):
1100
811
                        """This function is a scope container to pass
1101
812
                        func1 and func2 to the "call_both" function
1108
819
                        return call_both
1109
820
                    # Create the "call_both" function and add it to
1110
821
                    # the class
1111
 
                    attr[attrname] = fixscope(attribute, new_function)
 
822
                    attr[attrname] = fixscope(attribute,
 
823
                                              new_function)
1112
824
                # Is this a D-Bus method?
1113
825
                elif getattr(attribute, "_dbus_is_method", False):
1114
826
                    # Create a new, but exactly alike, function
1125
837
                                        attribute.func_name,
1126
838
                                        attribute.func_defaults,
1127
839
                                        attribute.func_closure)))
1128
 
                    # Copy annotations, if any
1129
 
                    try:
1130
 
                        attr[attrname]._dbus_annotations = (
1131
 
                            dict(attribute._dbus_annotations))
1132
 
                    except AttributeError:
1133
 
                        pass
1134
840
                # Is this a D-Bus property?
1135
841
                elif getattr(attribute, "_dbus_is_property", False):
1136
842
                    # Create a new, but exactly alike, function
1150
856
                                        attribute.func_name,
1151
857
                                        attribute.func_defaults,
1152
858
                                        attribute.func_closure)))
1153
 
                    # Copy annotations, if any
1154
 
                    try:
1155
 
                        attr[attrname]._dbus_annotations = (
1156
 
                            dict(attribute._dbus_annotations))
1157
 
                    except AttributeError:
1158
 
                        pass
1159
 
                # Is this a D-Bus interface?
1160
 
                elif getattr(attribute, "_dbus_is_interface", False):
1161
 
                    # Create a new, but exactly alike, function
1162
 
                    # object.  Decorate it to be a new D-Bus interface
1163
 
                    # with the alternate D-Bus interface name.  Add it
1164
 
                    # to the class.
1165
 
                    attr[attrname] = (dbus_interface_annotations
1166
 
                                      (alt_interface)
1167
 
                                      (types.FunctionType
1168
 
                                       (attribute.func_code,
1169
 
                                        attribute.func_globals,
1170
 
                                        attribute.func_name,
1171
 
                                        attribute.func_defaults,
1172
 
                                        attribute.func_closure)))
1173
 
            if deprecate:
1174
 
                # Deprecate all alternate interfaces
1175
 
                iname="_AlternateDBusNames_interface_annotation{0}"
1176
 
                for interface_name in interface_names:
1177
 
                    @dbus_interface_annotations(interface_name)
1178
 
                    def func(self):
1179
 
                        return { "org.freedesktop.DBus.Deprecated":
1180
 
                                     "true" }
1181
 
                    # Find an unused name
1182
 
                    for aname in (iname.format(i)
1183
 
                                  for i in itertools.count()):
1184
 
                        if aname not in attr:
1185
 
                            attr[aname] = func
1186
 
                            break
1187
 
            if interface_names:
1188
 
                # Replace the class with a new subclass of it with
1189
 
                # methods, signals, etc. as created above.
1190
 
                cls = type(b"{0}Alternate".format(cls.__name__),
1191
 
                           (cls,), attr)
1192
 
        return cls
1193
 
    return wrapper
1194
 
 
1195
 
 
1196
 
@alternate_dbus_interfaces({"se.recompile.Mandos":
1197
 
                                "se.bsnet.fukt.Mandos"})
 
859
        return type.__new__(mcs, name, bases, attr)
 
860
 
1198
861
class ClientDBus(Client, DBusObjectWithProperties):
1199
862
    """A Client class using D-Bus
1200
863
    
1209
872
    # dbus.service.Object doesn't use super(), so we can't either.
1210
873
    
1211
874
    def __init__(self, bus = None, *args, **kwargs):
 
875
        self._approvals_pending = 0
1212
876
        self.bus = bus
1213
877
        Client.__init__(self, *args, **kwargs)
1214
878
        # Only now, when this client is initialized, can it show up on
1220
884
                                 ("/clients/" + client_object_name))
1221
885
        DBusObjectWithProperties.__init__(self, self.bus,
1222
886
                                          self.dbus_object_path)
1223
 
    
 
887
        
1224
888
    def notifychangeproperty(transform_func,
1225
889
                             dbus_name, type_func=lambda x: x,
1226
890
                             variant_level=1):
1227
891
        """ Modify a variable so that it's a property which announces
1228
892
        its changes to DBus.
1229
 
        
 
893
 
1230
894
        transform_fun: Function that takes a value and a variant_level
1231
895
                       and transforms it to a D-Bus type.
1232
896
        dbus_name: D-Bus name of the variable
1249
913
        
1250
914
        return property(lambda self: getattr(self, attrname), setter)
1251
915
    
 
916
    
1252
917
    expires = notifychangeproperty(datetime_to_dbus, "Expires")
1253
918
    approvals_pending = notifychangeproperty(dbus.Boolean,
1254
919
                                             "ApprovalPending",
1261
926
                                       checker is not None)
1262
927
    last_checked_ok = notifychangeproperty(datetime_to_dbus,
1263
928
                                           "LastCheckedOK")
1264
 
    last_checker_status = notifychangeproperty(dbus.Int16,
1265
 
                                               "LastCheckerStatus")
1266
929
    last_approval_request = notifychangeproperty(
1267
930
        datetime_to_dbus, "LastApprovalRequest")
1268
931
    approved_by_default = notifychangeproperty(dbus.Boolean,
1269
932
                                               "ApprovedByDefault")
1270
 
    approval_delay = notifychangeproperty(dbus.UInt64,
 
933
    approval_delay = notifychangeproperty(dbus.UInt16,
1271
934
                                          "ApprovalDelay",
1272
935
                                          type_func =
1273
 
                                          timedelta_to_milliseconds)
 
936
                                          _timedelta_to_milliseconds)
1274
937
    approval_duration = notifychangeproperty(
1275
 
        dbus.UInt64, "ApprovalDuration",
1276
 
        type_func = timedelta_to_milliseconds)
 
938
        dbus.UInt16, "ApprovalDuration",
 
939
        type_func = _timedelta_to_milliseconds)
1277
940
    host = notifychangeproperty(dbus.String, "Host")
1278
 
    timeout = notifychangeproperty(dbus.UInt64, "Timeout",
 
941
    timeout = notifychangeproperty(dbus.UInt16, "Timeout",
1279
942
                                   type_func =
1280
 
                                   timedelta_to_milliseconds)
 
943
                                   _timedelta_to_milliseconds)
1281
944
    extended_timeout = notifychangeproperty(
1282
 
        dbus.UInt64, "ExtendedTimeout",
1283
 
        type_func = timedelta_to_milliseconds)
1284
 
    interval = notifychangeproperty(dbus.UInt64,
 
945
        dbus.UInt16, "ExtendedTimeout",
 
946
        type_func = _timedelta_to_milliseconds)
 
947
    interval = notifychangeproperty(dbus.UInt16,
1285
948
                                    "Interval",
1286
949
                                    type_func =
1287
 
                                    timedelta_to_milliseconds)
 
950
                                    _timedelta_to_milliseconds)
1288
951
    checker_command = notifychangeproperty(dbus.String, "Checker")
1289
952
    
1290
953
    del notifychangeproperty
1332
995
        return r
1333
996
    
1334
997
    def _reset_approved(self):
1335
 
        self.approved = None
 
998
        self._approved = None
1336
999
        return False
1337
1000
    
1338
1001
    def approve(self, value=True):
1339
 
        self.approved = value
1340
 
        gobject.timeout_add(timedelta_to_milliseconds
 
1002
        self.send_changedstate()
 
1003
        self._approved = value
 
1004
        gobject.timeout_add(_timedelta_to_milliseconds
1341
1005
                            (self.approval_duration),
1342
1006
                            self._reset_approved)
1343
 
        self.send_changedstate()
 
1007
    
1344
1008
    
1345
1009
    ## D-Bus methods, signals & properties
1346
1010
    _interface = "se.recompile.Mandos.Client"
1347
1011
    
1348
 
    ## Interfaces
1349
 
    
1350
 
    @dbus_interface_annotations(_interface)
1351
 
    def _foo(self):
1352
 
        return { "org.freedesktop.DBus.Property.EmitsChangedSignal":
1353
 
                     "false"}
1354
 
    
1355
1012
    ## Signals
1356
1013
    
1357
1014
    # CheckerCompleted - signal
1456
1113
                           access="readwrite")
1457
1114
    def ApprovalDuration_dbus_property(self, value=None):
1458
1115
        if value is None:       # get
1459
 
            return dbus.UInt64(timedelta_to_milliseconds(
 
1116
            return dbus.UInt64(_timedelta_to_milliseconds(
1460
1117
                    self.approval_duration))
1461
1118
        self.approval_duration = datetime.timedelta(0, 0, 0, value)
1462
1119
    
1476
1133
    def Host_dbus_property(self, value=None):
1477
1134
        if value is None:       # get
1478
1135
            return dbus.String(self.host)
1479
 
        self.host = unicode(value)
 
1136
        self.host = value
1480
1137
    
1481
1138
    # Created - property
1482
1139
    @dbus_service_property(_interface, signature="s", access="read")
1483
1140
    def Created_dbus_property(self):
1484
 
        return datetime_to_dbus(self.created)
 
1141
        return dbus.String(datetime_to_dbus(self.created))
1485
1142
    
1486
1143
    # LastEnabled - property
1487
1144
    @dbus_service_property(_interface, signature="s", access="read")
1508
1165
            return
1509
1166
        return datetime_to_dbus(self.last_checked_ok)
1510
1167
    
1511
 
    # LastCheckerStatus - property
1512
 
    @dbus_service_property(_interface, signature="n",
1513
 
                           access="read")
1514
 
    def LastCheckerStatus_dbus_property(self):
1515
 
        return dbus.Int16(self.last_checker_status)
1516
 
    
1517
1168
    # Expires - property
1518
1169
    @dbus_service_property(_interface, signature="s", access="read")
1519
1170
    def Expires_dbus_property(self):
1530
1181
    def Timeout_dbus_property(self, value=None):
1531
1182
        if value is None:       # get
1532
1183
            return dbus.UInt64(self.timeout_milliseconds())
1533
 
        old_timeout = self.timeout
1534
1184
        self.timeout = datetime.timedelta(0, 0, 0, value)
1535
 
        # Reschedule disabling
1536
 
        if self.enabled:
1537
 
            now = datetime.datetime.utcnow()
1538
 
            self.expires += self.timeout - old_timeout
1539
 
            if self.expires <= now:
1540
 
                # The timeout has passed
1541
 
                self.disable()
1542
 
            else:
1543
 
                if (getattr(self, "disable_initiator_tag", None)
1544
 
                    is None):
1545
 
                    return
1546
 
                gobject.source_remove(self.disable_initiator_tag)
1547
 
                self.disable_initiator_tag = (
1548
 
                    gobject.timeout_add(
1549
 
                        timedelta_to_milliseconds(self.expires - now),
1550
 
                        self.disable))
 
1185
        if getattr(self, "disable_initiator_tag", None) is None:
 
1186
            return
 
1187
        # Reschedule timeout
 
1188
        gobject.source_remove(self.disable_initiator_tag)
 
1189
        self.disable_initiator_tag = None
 
1190
        self.expires = None
 
1191
        time_to_die = _timedelta_to_milliseconds((self
 
1192
                                                  .last_checked_ok
 
1193
                                                  + self.timeout)
 
1194
                                                 - datetime.datetime
 
1195
                                                 .utcnow())
 
1196
        if time_to_die <= 0:
 
1197
            # The timeout has passed
 
1198
            self.disable()
 
1199
        else:
 
1200
            self.expires = (datetime.datetime.utcnow()
 
1201
                            + datetime.timedelta(milliseconds =
 
1202
                                                 time_to_die))
 
1203
            self.disable_initiator_tag = (gobject.timeout_add
 
1204
                                          (time_to_die, self.disable))
1551
1205
    
1552
1206
    # ExtendedTimeout - property
1553
1207
    @dbus_service_property(_interface, signature="t",
1566
1220
        self.interval = datetime.timedelta(0, 0, 0, value)
1567
1221
        if getattr(self, "checker_initiator_tag", None) is None:
1568
1222
            return
1569
 
        if self.enabled:
1570
 
            # Reschedule checker run
1571
 
            gobject.source_remove(self.checker_initiator_tag)
1572
 
            self.checker_initiator_tag = (gobject.timeout_add
1573
 
                                          (value, self.start_checker))
1574
 
            self.start_checker()    # Start one now, too
 
1223
        # Reschedule checker run
 
1224
        gobject.source_remove(self.checker_initiator_tag)
 
1225
        self.checker_initiator_tag = (gobject.timeout_add
 
1226
                                      (value, self.start_checker))
 
1227
        self.start_checker()    # Start one now, too
1575
1228
    
1576
1229
    # Checker - property
1577
1230
    @dbus_service_property(_interface, signature="s",
1579
1232
    def Checker_dbus_property(self, value=None):
1580
1233
        if value is None:       # get
1581
1234
            return dbus.String(self.checker_command)
1582
 
        self.checker_command = unicode(value)
 
1235
        self.checker_command = value
1583
1236
    
1584
1237
    # CheckerRunning - property
1585
1238
    @dbus_service_property(_interface, signature="b",
1614
1267
            raise KeyError()
1615
1268
    
1616
1269
    def __getattribute__(self, name):
1617
 
        if name == '_pipe':
 
1270
        if(name == '_pipe'):
1618
1271
            return super(ProxyClient, self).__getattribute__(name)
1619
1272
        self._pipe.send(('getattr', name))
1620
1273
        data = self._pipe.recv()
1627
1280
            return func
1628
1281
    
1629
1282
    def __setattr__(self, name, value):
1630
 
        if name == '_pipe':
 
1283
        if(name == '_pipe'):
1631
1284
            return super(ProxyClient, self).__setattr__(name, value)
1632
1285
        self._pipe.send(('setattr', name, value))
1633
1286
 
 
1287
class ClientDBusTransitional(ClientDBus):
 
1288
    __metaclass__ = AlternateDBusNamesMetaclass
1634
1289
 
1635
1290
class ClientHandler(socketserver.BaseRequestHandler, object):
1636
1291
    """A class to handle client connections.
1719
1374
                            client.Rejected("Disabled")
1720
1375
                        return
1721
1376
                    
1722
 
                    if client.approved or not client.approval_delay:
 
1377
                    if client._approved or not client.approval_delay:
1723
1378
                        #We are approved or approval is disabled
1724
1379
                        break
1725
 
                    elif client.approved is None:
 
1380
                    elif client._approved is None:
1726
1381
                        logger.info("Client %s needs approval",
1727
1382
                                    client.name)
1728
1383
                        if self.server.use_dbus:
1741
1396
                    #wait until timeout or approved
1742
1397
                    time = datetime.datetime.now()
1743
1398
                    client.changedstate.acquire()
1744
 
                    client.changedstate.wait(
1745
 
                        float(timedelta_to_milliseconds(delay)
1746
 
                              / 1000))
 
1399
                    (client.changedstate.wait
 
1400
                     (float(client._timedelta_to_milliseconds(delay)
 
1401
                            / 1000)))
1747
1402
                    client.changedstate.release()
1748
1403
                    time2 = datetime.datetime.now()
1749
1404
                    if (time2 - time) >= delay:
1765
1420
                    try:
1766
1421
                        sent = session.send(client.secret[sent_size:])
1767
1422
                    except gnutls.errors.GNUTLSError as error:
1768
 
                        logger.warning("gnutls send failed",
1769
 
                                       exc_info=error)
 
1423
                        logger.warning("gnutls send failed")
1770
1424
                        return
1771
1425
                    logger.debug("Sent: %d, remaining: %d",
1772
1426
                                 sent, len(client.secret)
1775
1429
                
1776
1430
                logger.info("Sending secret to %s", client.name)
1777
1431
                # bump the timeout using extended_timeout
1778
 
                client.bump_timeout(client.extended_timeout)
 
1432
                client.checked_ok(client.extended_timeout)
1779
1433
                if self.server.use_dbus:
1780
1434
                    # Emit D-Bus signal
1781
1435
                    client.GotSecret()
1786
1440
                try:
1787
1441
                    session.bye()
1788
1442
                except gnutls.errors.GNUTLSError as error:
1789
 
                    logger.warning("GnuTLS bye failed",
1790
 
                                   exc_info=error)
 
1443
                    logger.warning("GnuTLS bye failed")
1791
1444
    
1792
1445
    @staticmethod
1793
1446
    def peer_certificate(session):
1849
1502
        # Convert the buffer to a Python bytestring
1850
1503
        fpr = ctypes.string_at(buf, buf_len.value)
1851
1504
        # Convert the bytestring to hexadecimal notation
1852
 
        hex_fpr = binascii.hexlify(fpr).upper()
 
1505
        hex_fpr = ''.join("%02X" % ord(char) for char in fpr)
1853
1506
        return hex_fpr
1854
1507
 
1855
1508
 
1858
1511
    def sub_process_main(self, request, address):
1859
1512
        try:
1860
1513
            self.finish_request(request, address)
1861
 
        except Exception:
 
1514
        except:
1862
1515
            self.handle_error(request, address)
1863
1516
        self.close_request(request)
1864
1517
    
1865
1518
    def process_request(self, request, address):
1866
1519
        """Start a new process to process the request."""
1867
1520
        proc = multiprocessing.Process(target = self.sub_process_main,
1868
 
                                       args = (request, address))
 
1521
                                       args = (request,
 
1522
                                               address))
1869
1523
        proc.start()
1870
1524
        return proc
1871
1525
 
1921
1575
                                           str(self.interface
1922
1576
                                               + '\0'))
1923
1577
                except socket.error as error:
1924
 
                    if error.errno == errno.EPERM:
 
1578
                    if error[0] == errno.EPERM:
1925
1579
                        logger.error("No permission to"
1926
1580
                                     " bind to interface %s",
1927
1581
                                     self.interface)
1928
 
                    elif error.errno == errno.ENOPROTOOPT:
 
1582
                    elif error[0] == errno.ENOPROTOOPT:
1929
1583
                        logger.error("SO_BINDTODEVICE not available;"
1930
1584
                                     " cannot bind to interface %s",
1931
1585
                                     self.interface)
1932
 
                    elif error.errno == errno.ENODEV:
1933
 
                        logger.error("Interface %s does not"
1934
 
                                     " exist, cannot bind",
1935
 
                                     self.interface)
1936
1586
                    else:
1937
1587
                        raise
1938
1588
        # Only bind(2) the socket if we really need to.
1972
1622
        self.enabled = False
1973
1623
        self.clients = clients
1974
1624
        if self.clients is None:
1975
 
            self.clients = {}
 
1625
            self.clients = set()
1976
1626
        self.use_dbus = use_dbus
1977
1627
        self.gnutls_priority = gnutls_priority
1978
1628
        IPv6_TCPServer.__init__(self, server_address,
1997
1647
    
1998
1648
    def handle_ipc(self, source, condition, parent_pipe=None,
1999
1649
                   proc = None, client_object=None):
 
1650
        condition_names = {
 
1651
            gobject.IO_IN: "IN",   # There is data to read.
 
1652
            gobject.IO_OUT: "OUT", # Data can be written (without
 
1653
                                    # blocking).
 
1654
            gobject.IO_PRI: "PRI", # There is urgent data to read.
 
1655
            gobject.IO_ERR: "ERR", # Error condition.
 
1656
            gobject.IO_HUP: "HUP"  # Hung up (the connection has been
 
1657
                                    # broken, usually for pipes and
 
1658
                                    # sockets).
 
1659
            }
 
1660
        conditions_string = ' | '.join(name
 
1661
                                       for cond, name in
 
1662
                                       condition_names.iteritems()
 
1663
                                       if cond & condition)
2000
1664
        # error, or the other end of multiprocessing.Pipe has closed
2001
 
        if condition & (gobject.IO_ERR | gobject.IO_HUP):
 
1665
        if condition & (gobject.IO_ERR | condition & gobject.IO_HUP):
2002
1666
            # Wait for other process to exit
2003
1667
            proc.join()
2004
1668
            return False
2011
1675
            fpr = request[1]
2012
1676
            address = request[2]
2013
1677
            
2014
 
            for c in self.clients.itervalues():
 
1678
            for c in self.clients:
2015
1679
                if c.fingerprint == fpr:
2016
1680
                    client = c
2017
1681
                    break
2094
1758
            elif suffix == "w":
2095
1759
                delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
2096
1760
            else:
2097
 
                raise ValueError("Unknown suffix {0!r}"
2098
 
                                 .format(suffix))
 
1761
                raise ValueError("Unknown suffix %r" % suffix)
2099
1762
        except (ValueError, IndexError) as e:
2100
1763
            raise ValueError(*(e.args))
2101
1764
        timevalue += delta
2102
1765
    return timevalue
2103
1766
 
2104
1767
 
 
1768
def if_nametoindex(interface):
 
1769
    """Call the C function if_nametoindex(), or equivalent
 
1770
    
 
1771
    Note: This function cannot accept a unicode string."""
 
1772
    global if_nametoindex
 
1773
    try:
 
1774
        if_nametoindex = (ctypes.cdll.LoadLibrary
 
1775
                          (ctypes.util.find_library("c"))
 
1776
                          .if_nametoindex)
 
1777
    except (OSError, AttributeError):
 
1778
        logger.warning("Doing if_nametoindex the hard way")
 
1779
        def if_nametoindex(interface):
 
1780
            "Get an interface index the hard way, i.e. using fcntl()"
 
1781
            SIOCGIFINDEX = 0x8933  # From /usr/include/linux/sockios.h
 
1782
            with contextlib.closing(socket.socket()) as s:
 
1783
                ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
 
1784
                                    struct.pack(str("16s16x"),
 
1785
                                                interface))
 
1786
            interface_index = struct.unpack(str("I"),
 
1787
                                            ifreq[16:20])[0]
 
1788
            return interface_index
 
1789
    return if_nametoindex(interface)
 
1790
 
 
1791
 
2105
1792
def daemon(nochdir = False, noclose = False):
2106
1793
    """See daemon(3).  Standard BSD Unix function.
2107
1794
    
2115
1802
        sys.exit()
2116
1803
    if not noclose:
2117
1804
        # Close all standard open file descriptors
2118
 
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
 
1805
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
2119
1806
        if not stat.S_ISCHR(os.fstat(null).st_mode):
2120
1807
            raise OSError(errno.ENODEV,
2121
 
                          "{0} not a character device"
2122
 
                          .format(os.devnull))
 
1808
                          "%s not a character device"
 
1809
                          % os.path.devnull)
2123
1810
        os.dup2(null, sys.stdin.fileno())
2124
1811
        os.dup2(null, sys.stdout.fileno())
2125
1812
        os.dup2(null, sys.stderr.fileno())
2134
1821
    
2135
1822
    parser = argparse.ArgumentParser()
2136
1823
    parser.add_argument("-v", "--version", action="version",
2137
 
                        version = "%(prog)s {0}".format(version),
 
1824
                        version = "%%(prog)s %s" % version,
2138
1825
                        help="show version number and exit")
2139
1826
    parser.add_argument("-i", "--interface", metavar="IF",
2140
1827
                        help="Bind to interface IF")
2162
1849
                        " system bus interface")
2163
1850
    parser.add_argument("--no-ipv6", action="store_false",
2164
1851
                        dest="use_ipv6", help="Do not use IPv6")
2165
 
    parser.add_argument("--no-restore", action="store_false",
2166
 
                        dest="restore", help="Do not restore stored"
2167
 
                        " state")
2168
 
    parser.add_argument("--statedir", metavar="DIR",
2169
 
                        help="Directory to save/restore state in")
2170
 
    
2171
1852
    options = parser.parse_args()
2172
1853
    
2173
1854
    if options.check:
2186
1867
                        "use_dbus": "True",
2187
1868
                        "use_ipv6": "True",
2188
1869
                        "debuglevel": "",
2189
 
                        "restore": "True",
2190
 
                        "statedir": "/var/lib/mandos"
2191
1870
                        }
2192
1871
    
2193
1872
    # Parse config file for server-global settings
2210
1889
    # options, if set.
2211
1890
    for option in ("interface", "address", "port", "debug",
2212
1891
                   "priority", "servicename", "configdir",
2213
 
                   "use_dbus", "use_ipv6", "debuglevel", "restore",
2214
 
                   "statedir"):
 
1892
                   "use_dbus", "use_ipv6", "debuglevel"):
2215
1893
        value = getattr(options, option)
2216
1894
        if value is not None:
2217
1895
            server_settings[option] = value
2229
1907
    debuglevel = server_settings["debuglevel"]
2230
1908
    use_dbus = server_settings["use_dbus"]
2231
1909
    use_ipv6 = server_settings["use_ipv6"]
2232
 
    stored_state_path = os.path.join(server_settings["statedir"],
2233
 
                                     stored_state_file)
2234
 
    
2235
 
    if debug:
2236
 
        initlogger(debug, logging.DEBUG)
2237
 
    else:
2238
 
        if not debuglevel:
2239
 
            initlogger(debug)
2240
 
        else:
2241
 
            level = getattr(logging, debuglevel.upper())
2242
 
            initlogger(debug, level)
2243
1910
    
2244
1911
    if server_settings["servicename"] != "Mandos":
2245
1912
        syslogger.setFormatter(logging.Formatter
2246
 
                               ('Mandos ({0}) [%(process)d]:'
2247
 
                                ' %(levelname)s: %(message)s'
2248
 
                                .format(server_settings
2249
 
                                        ["servicename"])))
 
1913
                               ('Mandos (%s) [%%(process)d]:'
 
1914
                                ' %%(levelname)s: %%(message)s'
 
1915
                                % server_settings["servicename"]))
2250
1916
    
2251
1917
    # Parse config file with clients
2252
 
    client_config = configparser.SafeConfigParser(Client
2253
 
                                                  .client_defaults)
 
1918
    client_defaults = { "timeout": "5m",
 
1919
                        "extended_timeout": "15m",
 
1920
                        "interval": "2m",
 
1921
                        "checker": "fping -q -- %%(host)s",
 
1922
                        "host": "",
 
1923
                        "approval_delay": "0s",
 
1924
                        "approval_duration": "1s",
 
1925
                        }
 
1926
    client_config = configparser.SafeConfigParser(client_defaults)
2254
1927
    client_config.read(os.path.join(server_settings["configdir"],
2255
1928
                                    "clients.conf"))
2256
1929
    
2270
1943
        pidfilename = "/var/run/mandos.pid"
2271
1944
        try:
2272
1945
            pidfile = open(pidfilename, "w")
2273
 
        except IOError as e:
2274
 
            logger.error("Could not open file %r", pidfilename,
2275
 
                         exc_info=e)
 
1946
        except IOError:
 
1947
            logger.error("Could not open file %r", pidfilename)
2276
1948
    
2277
 
    for name in ("_mandos", "mandos", "nobody"):
 
1949
    try:
 
1950
        uid = pwd.getpwnam("_mandos").pw_uid
 
1951
        gid = pwd.getpwnam("_mandos").pw_gid
 
1952
    except KeyError:
2278
1953
        try:
2279
 
            uid = pwd.getpwnam(name).pw_uid
2280
 
            gid = pwd.getpwnam(name).pw_gid
2281
 
            break
 
1954
            uid = pwd.getpwnam("mandos").pw_uid
 
1955
            gid = pwd.getpwnam("mandos").pw_gid
2282
1956
        except KeyError:
2283
 
            continue
2284
 
    else:
2285
 
        uid = 65534
2286
 
        gid = 65534
 
1957
            try:
 
1958
                uid = pwd.getpwnam("nobody").pw_uid
 
1959
                gid = pwd.getpwnam("nobody").pw_gid
 
1960
            except KeyError:
 
1961
                uid = 65534
 
1962
                gid = 65534
2287
1963
    try:
2288
1964
        os.setgid(gid)
2289
1965
        os.setuid(uid)
2290
1966
    except OSError as error:
2291
 
        if error.errno != errno.EPERM:
 
1967
        if error[0] != errno.EPERM:
2292
1968
            raise error
2293
1969
    
 
1970
    if not debug and not debuglevel:
 
1971
        syslogger.setLevel(logging.WARNING)
 
1972
        console.setLevel(logging.WARNING)
 
1973
    if debuglevel:
 
1974
        level = getattr(logging, debuglevel.upper())
 
1975
        syslogger.setLevel(level)
 
1976
        console.setLevel(level)
 
1977
    
2294
1978
    if debug:
2295
1979
        # Enable all possible GnuTLS debugging
2296
1980
        
2306
1990
         .gnutls_global_set_log_function(debug_gnutls))
2307
1991
        
2308
1992
        # Redirect stdin so all checkers get /dev/null
2309
 
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
 
1993
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
2310
1994
        os.dup2(null, sys.stdin.fileno())
2311
1995
        if null > 2:
2312
1996
            os.close(null)
 
1997
    else:
 
1998
        # No console logging
 
1999
        logger.removeHandler(console)
2313
2000
    
2314
2001
    # Need to fork before connecting to D-Bus
2315
2002
    if not debug:
2316
2003
        # Close all input and output, do double fork, etc.
2317
2004
        daemon()
2318
2005
    
2319
 
    gobject.threads_init()
2320
 
    
2321
2006
    global main_loop
2322
2007
    # From the Avahi example code
2323
 
    DBusGMainLoop(set_as_default=True)
 
2008
    DBusGMainLoop(set_as_default=True )
2324
2009
    main_loop = gobject.MainLoop()
2325
2010
    bus = dbus.SystemBus()
2326
2011
    # End of Avahi example code
2332
2017
                            ("se.bsnet.fukt.Mandos", bus,
2333
2018
                             do_not_queue=True))
2334
2019
        except dbus.exceptions.NameExistsException as e:
2335
 
            logger.error("Disabling D-Bus:", exc_info=e)
 
2020
            logger.error(unicode(e) + ", disabling D-Bus")
2336
2021
            use_dbus = False
2337
2022
            server_settings["use_dbus"] = False
2338
2023
            tcp_server.use_dbus = False
2339
2024
    protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
2340
 
    service = AvahiServiceToSyslog(name =
2341
 
                                   server_settings["servicename"],
2342
 
                                   servicetype = "_mandos._tcp",
2343
 
                                   protocol = protocol, bus = bus)
 
2025
    service = AvahiService(name = server_settings["servicename"],
 
2026
                           servicetype = "_mandos._tcp",
 
2027
                           protocol = protocol, bus = bus)
2344
2028
    if server_settings["interface"]:
2345
2029
        service.interface = (if_nametoindex
2346
2030
                             (str(server_settings["interface"])))
2350
2034
    
2351
2035
    client_class = Client
2352
2036
    if use_dbus:
2353
 
        client_class = functools.partial(ClientDBus, bus = bus)
2354
 
    
2355
 
    client_settings = Client.config_parser(client_config)
2356
 
    old_client_settings = {}
2357
 
    clients_data = {}
2358
 
    
2359
 
    # Get client data and settings from last running state.
2360
 
    if server_settings["restore"]:
2361
 
        try:
2362
 
            with open(stored_state_path, "rb") as stored_state:
2363
 
                clients_data, old_client_settings = (pickle.load
2364
 
                                                     (stored_state))
2365
 
            os.remove(stored_state_path)
2366
 
        except IOError as e:
2367
 
            if e.errno == errno.ENOENT:
2368
 
                logger.warning("Could not load persistent state: {0}"
2369
 
                                .format(os.strerror(e.errno)))
2370
 
            else:
2371
 
                logger.critical("Could not load persistent state:",
2372
 
                                exc_info=e)
2373
 
                raise
2374
 
        except EOFError as e:
2375
 
            logger.warning("Could not load persistent state: "
2376
 
                           "EOFError:", exc_info=e)
2377
 
    
2378
 
    with PGPEngine() as pgp:
2379
 
        for client_name, client in clients_data.iteritems():
2380
 
            # Decide which value to use after restoring saved state.
2381
 
            # We have three different values: Old config file,
2382
 
            # new config file, and saved state.
2383
 
            # New config value takes precedence if it differs from old
2384
 
            # config value, otherwise use saved state.
2385
 
            for name, value in client_settings[client_name].items():
2386
 
                try:
2387
 
                    # For each value in new config, check if it
2388
 
                    # differs from the old config value (Except for
2389
 
                    # the "secret" attribute)
2390
 
                    if (name != "secret" and
2391
 
                        value != old_client_settings[client_name]
2392
 
                        [name]):
2393
 
                        client[name] = value
2394
 
                except KeyError:
2395
 
                    pass
2396
 
            
2397
 
            # Clients who has passed its expire date can still be
2398
 
            # enabled if its last checker was successful.  Clients
2399
 
            # whose checker succeeded before we stored its state is
2400
 
            # assumed to have successfully run all checkers during
2401
 
            # downtime.
2402
 
            if client["enabled"]:
2403
 
                if datetime.datetime.utcnow() >= client["expires"]:
2404
 
                    if not client["last_checked_ok"]:
2405
 
                        logger.warning(
2406
 
                            "disabling client {0} - Client never "
2407
 
                            "performed a successful checker"
2408
 
                            .format(client_name))
2409
 
                        client["enabled"] = False
2410
 
                    elif client["last_checker_status"] != 0:
2411
 
                        logger.warning(
2412
 
                            "disabling client {0} - Client "
2413
 
                            "last checker failed with error code {1}"
2414
 
                            .format(client_name,
2415
 
                                    client["last_checker_status"]))
2416
 
                        client["enabled"] = False
2417
 
                    else:
2418
 
                        client["expires"] = (datetime.datetime
2419
 
                                             .utcnow()
2420
 
                                             + client["timeout"])
2421
 
                        logger.debug("Last checker succeeded,"
2422
 
                                     " keeping {0} enabled"
2423
 
                                     .format(client_name))
 
2037
        client_class = functools.partial(ClientDBusTransitional,
 
2038
                                         bus = bus)
 
2039
    def client_config_items(config, section):
 
2040
        special_settings = {
 
2041
            "approved_by_default":
 
2042
                lambda: config.getboolean(section,
 
2043
                                          "approved_by_default"),
 
2044
            }
 
2045
        for name, value in config.items(section):
2424
2046
            try:
2425
 
                client["secret"] = (
2426
 
                    pgp.decrypt(client["encrypted_secret"],
2427
 
                                client_settings[client_name]
2428
 
                                ["secret"]))
2429
 
            except PGPError:
2430
 
                # If decryption fails, we use secret from new settings
2431
 
                logger.debug("Failed to decrypt {0} old secret"
2432
 
                             .format(client_name))
2433
 
                client["secret"] = (
2434
 
                    client_settings[client_name]["secret"])
2435
 
    
2436
 
    # Add/remove clients based on new changes made to config
2437
 
    for client_name in (set(old_client_settings)
2438
 
                        - set(client_settings)):
2439
 
        del clients_data[client_name]
2440
 
    for client_name in (set(client_settings)
2441
 
                        - set(old_client_settings)):
2442
 
        clients_data[client_name] = client_settings[client_name]
2443
 
    
2444
 
    # Create all client objects
2445
 
    for client_name, client in clients_data.iteritems():
2446
 
        tcp_server.clients[client_name] = client_class(
2447
 
            name = client_name, settings = client)
2448
 
    
 
2047
                yield (name, special_settings[name]())
 
2048
            except KeyError:
 
2049
                yield (name, value)
 
2050
    
 
2051
    tcp_server.clients.update(set(
 
2052
            client_class(name = section,
 
2053
                         config= dict(client_config_items(
 
2054
                        client_config, section)))
 
2055
            for section in client_config.sections()))
2449
2056
    if not tcp_server.clients:
2450
2057
        logger.warning("No clients defined")
2451
 
    
 
2058
        
2452
2059
    if not debug:
2453
2060
        try:
2454
2061
            with pidfile:
2462
2069
            # "pidfile" was never created
2463
2070
            pass
2464
2071
        del pidfilename
 
2072
        
 
2073
        signal.signal(signal.SIGINT, signal.SIG_IGN)
2465
2074
    
2466
2075
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
2467
2076
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
2468
2077
    
2469
2078
    if use_dbus:
2470
 
        @alternate_dbus_interfaces({"se.recompile.Mandos":
2471
 
                                        "se.bsnet.fukt.Mandos"})
2472
 
        class MandosDBusService(DBusObjectWithProperties):
 
2079
        class MandosDBusService(dbus.service.Object):
2473
2080
            """A D-Bus proxy object"""
2474
2081
            def __init__(self):
2475
2082
                dbus.service.Object.__init__(self, bus, "/")
2476
2083
            _interface = "se.recompile.Mandos"
2477
2084
            
2478
 
            @dbus_interface_annotations(_interface)
2479
 
            def _foo(self):
2480
 
                return { "org.freedesktop.DBus.Property"
2481
 
                         ".EmitsChangedSignal":
2482
 
                             "false"}
2483
 
            
2484
2085
            @dbus.service.signal(_interface, signature="o")
2485
2086
            def ClientAdded(self, objpath):
2486
2087
                "D-Bus signal"
2500
2101
            def GetAllClients(self):
2501
2102
                "D-Bus method"
2502
2103
                return dbus.Array(c.dbus_object_path
2503
 
                                  for c in
2504
 
                                  tcp_server.clients.itervalues())
 
2104
                                  for c in tcp_server.clients)
2505
2105
            
2506
2106
            @dbus.service.method(_interface,
2507
2107
                                 out_signature="a{oa{sv}}")
2509
2109
                "D-Bus method"
2510
2110
                return dbus.Dictionary(
2511
2111
                    ((c.dbus_object_path, c.GetAll(""))
2512
 
                     for c in tcp_server.clients.itervalues()),
 
2112
                     for c in tcp_server.clients),
2513
2113
                    signature="oa{sv}")
2514
2114
            
2515
2115
            @dbus.service.method(_interface, in_signature="o")
2516
2116
            def RemoveClient(self, object_path):
2517
2117
                "D-Bus method"
2518
 
                for c in tcp_server.clients.itervalues():
 
2118
                for c in tcp_server.clients:
2519
2119
                    if c.dbus_object_path == object_path:
2520
 
                        del tcp_server.clients[c.name]
 
2120
                        tcp_server.clients.remove(c)
2521
2121
                        c.remove_from_connection()
2522
2122
                        # Don't signal anything except ClientRemoved
2523
2123
                        c.disable(quiet=True)
2528
2128
            
2529
2129
            del _interface
2530
2130
        
2531
 
        mandos_dbus_service = MandosDBusService()
 
2131
        class MandosDBusServiceTransitional(MandosDBusService):
 
2132
            __metaclass__ = AlternateDBusNamesMetaclass
 
2133
        mandos_dbus_service = MandosDBusServiceTransitional()
2532
2134
    
2533
2135
    def cleanup():
2534
2136
        "Cleanup function; run on exit"
2535
2137
        service.cleanup()
2536
2138
        
2537
2139
        multiprocessing.active_children()
2538
 
        if not (tcp_server.clients or client_settings):
2539
 
            return
2540
 
        
2541
 
        # Store client before exiting. Secrets are encrypted with key
2542
 
        # based on what config file has. If config file is
2543
 
        # removed/edited, old secret will thus be unrecovable.
2544
 
        clients = {}
2545
 
        with PGPEngine() as pgp:
2546
 
            for client in tcp_server.clients.itervalues():
2547
 
                key = client_settings[client.name]["secret"]
2548
 
                client.encrypted_secret = pgp.encrypt(client.secret,
2549
 
                                                      key)
2550
 
                client_dict = {}
2551
 
                
2552
 
                # A list of attributes that can not be pickled
2553
 
                # + secret.
2554
 
                exclude = set(("bus", "changedstate", "secret",
2555
 
                               "checker"))
2556
 
                for name, typ in (inspect.getmembers
2557
 
                                  (dbus.service.Object)):
2558
 
                    exclude.add(name)
2559
 
                
2560
 
                client_dict["encrypted_secret"] = (client
2561
 
                                                   .encrypted_secret)
2562
 
                for attr in client.client_structure:
2563
 
                    if attr not in exclude:
2564
 
                        client_dict[attr] = getattr(client, attr)
2565
 
                
2566
 
                clients[client.name] = client_dict
2567
 
                del client_settings[client.name]["secret"]
2568
 
        
2569
 
        try:
2570
 
            with (tempfile.NamedTemporaryFile
2571
 
                  (mode='wb', suffix=".pickle", prefix='clients-',
2572
 
                   dir=os.path.dirname(stored_state_path),
2573
 
                   delete=False)) as stored_state:
2574
 
                pickle.dump((clients, client_settings), stored_state)
2575
 
                tempname=stored_state.name
2576
 
            os.rename(tempname, stored_state_path)
2577
 
        except (IOError, OSError) as e:
2578
 
            if not debug:
2579
 
                try:
2580
 
                    os.remove(tempname)
2581
 
                except NameError:
2582
 
                    pass
2583
 
            if e.errno in (errno.ENOENT, errno.EACCES, errno.EEXIST):
2584
 
                logger.warning("Could not save persistent state: {0}"
2585
 
                               .format(os.strerror(e.errno)))
2586
 
            else:
2587
 
                logger.warning("Could not save persistent state:",
2588
 
                               exc_info=e)
2589
 
                raise e
2590
 
        
2591
 
        # Delete all clients, and settings from config
2592
2140
        while tcp_server.clients:
2593
 
            name, client = tcp_server.clients.popitem()
 
2141
            client = tcp_server.clients.pop()
2594
2142
            if use_dbus:
2595
2143
                client.remove_from_connection()
 
2144
            client.disable_hook = None
2596
2145
            # Don't signal anything except ClientRemoved
2597
2146
            client.disable(quiet=True)
2598
2147
            if use_dbus:
2600
2149
                mandos_dbus_service.ClientRemoved(client
2601
2150
                                                  .dbus_object_path,
2602
2151
                                                  client.name)
2603
 
        client_settings.clear()
2604
2152
    
2605
2153
    atexit.register(cleanup)
2606
2154
    
2607
 
    for client in tcp_server.clients.itervalues():
 
2155
    for client in tcp_server.clients:
2608
2156
        if use_dbus:
2609
2157
            # Emit D-Bus signal
2610
2158
            mandos_dbus_service.ClientAdded(client.dbus_object_path)
2611
 
        # Need to initiate checking of clients
2612
 
        if client.enabled:
2613
 
            client.init_checker()
 
2159
        client.enable()
2614
2160
    
2615
2161
    tcp_server.enable()
2616
2162
    tcp_server.server_activate()
2619
2165
    service.port = tcp_server.socket.getsockname()[1]
2620
2166
    if use_ipv6:
2621
2167
        logger.info("Now listening on address %r, port %d,"
2622
 
                    " flowinfo %d, scope_id %d",
2623
 
                    *tcp_server.socket.getsockname())
 
2168
                    " flowinfo %d, scope_id %d"
 
2169
                    % tcp_server.socket.getsockname())
2624
2170
    else:                       # IPv4
2625
 
        logger.info("Now listening on address %r, port %d",
2626
 
                    *tcp_server.socket.getsockname())
 
2171
        logger.info("Now listening on address %r, port %d"
 
2172
                    % tcp_server.socket.getsockname())
2627
2173
    
2628
2174
    #service.interface = tcp_server.socket.getsockname()[3]
2629
2175
    
2632
2178
        try:
2633
2179
            service.activate()
2634
2180
        except dbus.exceptions.DBusException as error:
2635
 
            logger.critical("D-Bus Exception", exc_info=error)
 
2181
            logger.critical("DBusException: %s", error)
2636
2182
            cleanup()
2637
2183
            sys.exit(1)
2638
2184
        # End of Avahi example code
2645
2191
        logger.debug("Starting main loop")
2646
2192
        main_loop.run()
2647
2193
    except AvahiError as error:
2648
 
        logger.critical("Avahi Error", exc_info=error)
 
2194
        logger.critical("AvahiError: %s", error)
2649
2195
        cleanup()
2650
2196
        sys.exit(1)
2651
2197
    except KeyboardInterrupt:
2656
2202
    # Must run before the D-Bus bus name gets deregistered
2657
2203
    cleanup()
2658
2204
 
 
2205
 
2659
2206
if __name__ == '__main__':
2660
2207
    main()