/mandos/trunk

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

« back to all changes in this revision

Viewing changes to mandos

  • Committer: Teddy Hogeborn
  • Date: 2011-12-21 17:08:29 UTC
  • mto: This revision was merged to the branch mainline in revision 527.
  • Revision ID: teddy@recompile.se-20111221170829-gf93q0kejpgh3ur3
* debian/control (mandos-client/Depends): Added "initramfs-tools".

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