/mandos/trunk

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

« back to all changes in this revision

Viewing changes to mandos

* mandos (transitional_dbus_metaclass): Renamed to
                                        "AlternateDBusNamesMetaclass".
                                        All users changed.  Also added
                                        comments, wrapped lines, and
                                        searching of all base classes
                                        for D-Bus attributes.  It now
                                        adds the *old* interface name,
                                        not the new one.
  (ClientDBus, MandosDBusService): Use the new interface names.

Show diffs side-by-side

added added

removed removed

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