/mandos/release

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

« back to all changes in this revision

Viewing changes to server.py

merge +
mandosclient
        Added a adjustbuffer function.

Show diffs side-by-side

added added

removed removed

Lines of Context:
server.py
1
1
#!/usr/bin/python
 
2
# -*- mode: python; coding: utf-8 -*-
 
3
# 
 
4
# Mandos server - give out binary blobs to connecting clients.
 
5
# 
 
6
# This program is partly derived from an example program for an Avahi
 
7
# service publisher, downloaded from
 
8
# <http://avahi.org/wiki/PythonPublishExample>.  This includes the
 
9
# following functions: "AvahiService.add", "AvahiService.remove",
 
10
# "server_state_changed", "entry_group_state_changed", and some lines
 
11
# in "main".
 
12
# 
 
13
# Everything else is
 
14
# Copyright © 2007-2008 Teddy Hogeborn & Björn Påhlsson
 
15
# 
 
16
# This program is free software: you can redistribute it and/or modify
 
17
# it under the terms of the GNU General Public License as published by
 
18
# the Free Software Foundation, either version 3 of the License, or
 
19
# (at your option) any later version.
 
20
#
 
21
#     This program is distributed in the hope that it will be useful,
 
22
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
 
23
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
24
#     GNU General Public License for more details.
 
25
# 
 
26
# 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
2
33
 
3
34
import SocketServer
4
35
import socket
9
40
import gnutls.crypto
10
41
import gnutls.connection
11
42
import gnutls.errors
 
43
import gnutls.library.functions
 
44
import gnutls.library.constants
 
45
import gnutls.library.types
12
46
import ConfigParser
13
47
import sys
 
48
import re
 
49
import os
 
50
import signal
 
51
from sets import Set
 
52
import subprocess
 
53
import atexit
 
54
import stat
 
55
import logging
 
56
import logging.handlers
 
57
 
 
58
import dbus
 
59
import gobject
 
60
import avahi
 
61
from dbus.mainloop.glib import DBusGMainLoop
 
62
import ctypes
 
63
 
 
64
# Brief description of the operation of this program:
 
65
# 
 
66
# This server announces itself as a Zeroconf service.  Connecting
 
67
# clients use the TLS protocol, with the unusual quirk that this
 
68
# server program acts as a TLS "client" while a connecting client acts
 
69
# as a TLS "server".  The client (acting as a TLS "server") must
 
70
# supply an OpenPGP certificate, and the fingerprint of this
 
71
# certificate is used by this server to look up (in a list read from a
 
72
# file at start time) which binary blob to give the client.  No other
 
73
# authentication or authorization is done by this server.
 
74
 
 
75
 
 
76
logger = logging.Logger('mandos')
 
77
syslogger = logging.handlers.SysLogHandler\
 
78
            (facility = logging.handlers.SysLogHandler.LOG_DAEMON)
 
79
syslogger.setFormatter(logging.Formatter\
 
80
                        ('%(levelname)s: %(message)s'))
 
81
logger.addHandler(syslogger)
 
82
del syslogger
 
83
 
 
84
 
 
85
class AvahiError(Exception):
 
86
    def __init__(self, value):
 
87
        self.value = value
 
88
    def __str__(self):
 
89
        return repr(self.value)
 
90
 
 
91
class AvahiServiceError(AvahiError):
 
92
    pass
 
93
 
 
94
class AvahiGroupError(AvahiError):
 
95
    pass
 
96
 
 
97
 
 
98
class AvahiService(object):
 
99
    """
 
100
    interface: integer; avahi.IF_UNSPEC or an interface index.
 
101
               Used to optionally bind to the specified interface.
 
102
    name = string; Example: "Mandos"
 
103
    type = string; Example: "_mandos._tcp".
 
104
                   See <http://www.dns-sd.org/ServiceTypes.html>
 
105
    port = integer; what port to announce
 
106
    TXT = list of strings; TXT record for the service
 
107
    domain = string; Domain to publish on, default to .local if empty.
 
108
    host = string; Host to publish records for, default to localhost
 
109
                   if empty.
 
110
    max_renames = integer; maximum number of renames
 
111
    rename_count = integer; counter so we only rename after collisions
 
112
                   a sensible number of times
 
113
    """
 
114
    def __init__(self, interface = avahi.IF_UNSPEC, name = None,
 
115
                 type = None, port = None, TXT = None, domain = "",
 
116
                 host = "", max_renames = 12):
 
117
        """An Avahi (Zeroconf) service. """
 
118
        self.interface = interface
 
119
        self.name = name
 
120
        self.type = type
 
121
        self.port = port
 
122
        if TXT is None:
 
123
            self.TXT = []
 
124
        else:
 
125
            self.TXT = TXT
 
126
        self.domain = domain
 
127
        self.host = host
 
128
        self.rename_count = 0
 
129
    def rename(self):
 
130
        """Derived from the Avahi example code"""
 
131
        if self.rename_count >= self.max_renames:
 
132
            logger.critical(u"No suitable service name found after %i"
 
133
                            u" retries, exiting.", rename_count)
 
134
            raise AvahiServiceError("Too many renames")
 
135
        name = server.GetAlternativeServiceName(name)
 
136
        logger.notice(u"Changing name to %r ...", name)
 
137
        self.remove()
 
138
        self.add()
 
139
        self.rename_count += 1
 
140
    def remove(self):
 
141
        """Derived from the Avahi example code"""
 
142
        if group is not None:
 
143
            group.Reset()
 
144
    def add(self):
 
145
        """Derived from the Avahi example code"""
 
146
        global group
 
147
        if group is None:
 
148
            group = dbus.Interface\
 
149
                    (bus.get_object(avahi.DBUS_NAME,
 
150
                                    server.EntryGroupNew()),
 
151
                     avahi.DBUS_INTERFACE_ENTRY_GROUP)
 
152
            group.connect_to_signal('StateChanged',
 
153
                                    entry_group_state_changed)
 
154
        logger.debug(u"Adding service '%s' of type '%s' ...",
 
155
                     service.name, service.type)
 
156
        group.AddService(
 
157
                self.interface,         # interface
 
158
                avahi.PROTO_INET6,      # protocol
 
159
                dbus.UInt32(0),         # flags
 
160
                self.name, self.type,
 
161
                self.domain, self.host,
 
162
                dbus.UInt16(self.port),
 
163
                avahi.string_array_to_txt_array(self.TXT))
 
164
        group.Commit()
 
165
 
 
166
# From the Avahi example code:
 
167
group = None                            # our entry group
 
168
# End of Avahi example code
14
169
 
15
170
 
16
171
class Client(object):
17
 
    def __init__(self, name=None, options=None, dn=None,
18
 
                 password=None, passfile=None, fqdn=None,
19
 
                 timeout=None, interval=-1):
 
172
    """A representation of a client host served by this server.
 
173
    Attributes:
 
174
    name:      string; from the config file, used in log messages
 
175
    fingerprint: string (40 or 32 hexadecimal digits); used to
 
176
                 uniquely identify the client
 
177
    secret:    bytestring; sent verbatim (over TLS) to client
 
178
    fqdn:      string (FQDN); available for use by the checker command
 
179
    created:   datetime.datetime(); object creation, not client host
 
180
    last_checked_ok: datetime.datetime() or None if not yet checked OK
 
181
    timeout:   datetime.timedelta(); How long from last_checked_ok
 
182
                                     until this client is invalid
 
183
    interval:  datetime.timedelta(); How often to start a new checker
 
184
    stop_hook: If set, called by stop() as stop_hook(self)
 
185
    checker:   subprocess.Popen(); a running checker process used
 
186
                                   to see if the client lives.
 
187
                                   'None' if no process is running.
 
188
    checker_initiator_tag: a gobject event source tag, or None
 
189
    stop_initiator_tag:    - '' -
 
190
    checker_callback_tag:  - '' -
 
191
    checker_command: string; External command which is run to check if
 
192
                     client lives.  %() expansions are done at
 
193
                     runtime with vars(self) as dict, so that for
 
194
                     instance %(name)s can be used in the command.
 
195
    Private attibutes:
 
196
    _timeout: Real variable for 'timeout'
 
197
    _interval: Real variable for 'interval'
 
198
    _timeout_milliseconds: Used when calling gobject.timeout_add()
 
199
    _interval_milliseconds: - '' -
 
200
    """
 
201
    def _set_timeout(self, timeout):
 
202
        "Setter function for 'timeout' attribute"
 
203
        self._timeout = timeout
 
204
        self._timeout_milliseconds = ((self.timeout.days
 
205
                                       * 24 * 60 * 60 * 1000)
 
206
                                      + (self.timeout.seconds * 1000)
 
207
                                      + (self.timeout.microseconds
 
208
                                         // 1000))
 
209
    timeout = property(lambda self: self._timeout,
 
210
                       _set_timeout)
 
211
    del _set_timeout
 
212
    def _set_interval(self, interval):
 
213
        "Setter function for 'interval' attribute"
 
214
        self._interval = interval
 
215
        self._interval_milliseconds = ((self.interval.days
 
216
                                        * 24 * 60 * 60 * 1000)
 
217
                                       + (self.interval.seconds
 
218
                                          * 1000)
 
219
                                       + (self.interval.microseconds
 
220
                                          // 1000))
 
221
    interval = property(lambda self: self._interval,
 
222
                        _set_interval)
 
223
    del _set_interval
 
224
    def __init__(self, name=None, stop_hook=None, fingerprint=None,
 
225
                 secret=None, secfile=None, fqdn=None, timeout=None,
 
226
                 interval=-1, checker=None):
 
227
        """Note: the 'checker' argument sets the 'checker_command'
 
228
        attribute and not the 'checker' attribute.."""
20
229
        self.name = name
21
 
        self.dn = dn
22
 
        if password:
23
 
            self.password = password
24
 
        elif passfile:
25
 
            self.password = open(passfile).readall()
 
230
        logger.debug(u"Creating client %r", self.name)
 
231
        # Uppercase and remove spaces from fingerprint
 
232
        # for later comparison purposes with return value of
 
233
        # the fingerprint() function
 
234
        self.fingerprint = fingerprint.upper().replace(u" ", u"")
 
235
        logger.debug(u"  Fingerprint: %s", self.fingerprint)
 
236
        if secret:
 
237
            self.secret = secret.decode(u"base64")
 
238
        elif secfile:
 
239
            sf = open(secfile)
 
240
            self.secret = sf.read()
 
241
            sf.close()
26
242
        else:
27
 
            print "No Password or Passfile in client config file"
28
 
            # raise RuntimeError XXX
29
 
            self.password = "gazonk"
 
243
            raise TypeError(u"No secret or secfile for client %s"
 
244
                            % self.name)
30
245
        self.fqdn = fqdn
31
246
        self.created = datetime.datetime.now()
32
 
        self.last_seen = None
33
 
        if timeout is None:
34
 
            timeout = options.timeout
35
 
        self.timeout = timeout
36
 
        if interval == -1:
37
 
            interval = options.interval
38
 
        self.interval = interval
39
 
        self.next_check = datetime.datetime.now()
40
 
 
41
 
 
42
 
class server_metaclass(type):
43
 
    "Common behavior for the UDP and TCP server classes"
44
 
    def __new__(cls, name, bases, attrs):
45
 
        attrs["address_family"] = socket.AF_INET6
46
 
        attrs["allow_reuse_address"] = True
47
 
        def server_bind(self):
48
 
            if self.options.interface:
49
 
                if not hasattr(socket, "SO_BINDTODEVICE"):
50
 
                    # From /usr/include/asm-i486/socket.h
51
 
                    socket.SO_BINDTODEVICE = 25
 
247
        self.last_checked_ok = None
 
248
        self.timeout = string_to_delta(timeout)
 
249
        self.interval = string_to_delta(interval)
 
250
        self.stop_hook = stop_hook
 
251
        self.checker = None
 
252
        self.checker_initiator_tag = None
 
253
        self.stop_initiator_tag = None
 
254
        self.checker_callback_tag = None
 
255
        self.check_command = checker
 
256
    def start(self):
 
257
        """Start this client's checker and timeout hooks"""
 
258
        # Schedule a new checker to be started an 'interval' from now,
 
259
        # and every interval from then on.
 
260
        self.checker_initiator_tag = gobject.timeout_add\
 
261
                                     (self._interval_milliseconds,
 
262
                                      self.start_checker)
 
263
        # Also start a new checker *right now*.
 
264
        self.start_checker()
 
265
        # Schedule a stop() when 'timeout' has passed
 
266
        self.stop_initiator_tag = gobject.timeout_add\
 
267
                                  (self._timeout_milliseconds,
 
268
                                   self.stop)
 
269
    def stop(self):
 
270
        """Stop this client.
 
271
        The possibility that a client might be restarted is left open,
 
272
        but not currently used."""
 
273
        # If this client doesn't have a secret, it is already stopped.
 
274
        if self.secret:
 
275
            logger.debug(u"Stopping client %s", self.name)
 
276
            self.secret = None
 
277
        else:
 
278
            return False
 
279
        if getattr(self, "stop_initiator_tag", False):
 
280
            gobject.source_remove(self.stop_initiator_tag)
 
281
            self.stop_initiator_tag = None
 
282
        if getattr(self, "checker_initiator_tag", False):
 
283
            gobject.source_remove(self.checker_initiator_tag)
 
284
            self.checker_initiator_tag = None
 
285
        self.stop_checker()
 
286
        if self.stop_hook:
 
287
            self.stop_hook(self)
 
288
        # Do not run this again if called by a gobject.timeout_add
 
289
        return False
 
290
    def __del__(self):
 
291
        self.stop_hook = None
 
292
        self.stop()
 
293
    def checker_callback(self, pid, condition):
 
294
        """The checker has completed, so take appropriate actions."""
 
295
        now = datetime.datetime.now()
 
296
        self.checker_callback_tag = None
 
297
        self.checker = None
 
298
        if os.WIFEXITED(condition) \
 
299
               and (os.WEXITSTATUS(condition) == 0):
 
300
            logger.debug(u"Checker for %(name)s succeeded",
 
301
                         vars(self))
 
302
            self.last_checked_ok = now
 
303
            gobject.source_remove(self.stop_initiator_tag)
 
304
            self.stop_initiator_tag = gobject.timeout_add\
 
305
                                      (self._timeout_milliseconds,
 
306
                                       self.stop)
 
307
        elif not os.WIFEXITED(condition):
 
308
            logger.warning(u"Checker for %(name)s crashed?",
 
309
                           vars(self))
 
310
        else:
 
311
            logger.debug(u"Checker for %(name)s failed",
 
312
                         vars(self))
 
313
    def start_checker(self):
 
314
        """Start a new checker subprocess if one is not running.
 
315
        If a checker already exists, leave it running and do
 
316
        nothing."""
 
317
        # The reason for not killing a running checker is that if we
 
318
        # did that, then if a checker (for some reason) started
 
319
        # running slowly and taking more than 'interval' time, the
 
320
        # client would inevitably timeout, since no checker would get
 
321
        # a chance to run to completion.  If we instead leave running
 
322
        # checkers alone, the checker would have to take more time
 
323
        # than 'timeout' for the client to be declared invalid, which
 
324
        # is as it should be.
 
325
        if self.checker is None:
 
326
            try:
 
327
                # In case check_command has exactly one % operator
 
328
                command = self.check_command % self.fqdn
 
329
            except TypeError:
 
330
                # Escape attributes for the shell
 
331
                escaped_attrs = dict((key, re.escape(str(val)))
 
332
                                     for key, val in
 
333
                                     vars(self).iteritems())
52
334
                try:
53
 
                    self.socket.setsockopt(socket.SOL_SOCKET,
54
 
                                           socket.SO_BINDTODEVICE,
55
 
                                           self.options.interface)
56
 
                except socket.error, error:
57
 
                    if error[0] == errno.EPERM:
58
 
                        print "Warning: No permission to bind to interface", \
59
 
                              self.options.interface
60
 
                    else:
61
 
                        raise error
62
 
            return super(type(self), self).server_bind()
63
 
        attrs["server_bind"] = server_bind
64
 
        def init(self, *args, **kwargs):
65
 
            if "options" in kwargs:
66
 
                self.options = kwargs["options"]
67
 
                del kwargs["options"]
68
 
            if "clients" in kwargs:
69
 
                self.clients = kwargs["clients"]
70
 
                del kwargs["clients"]
71
 
            if "credentials" in kwargs:
72
 
                self.credentials = kwargs["credentials"]
73
 
                del kwargs["credentials"]
74
 
            return super(type(self), self).__init__(*args, **kwargs)
75
 
        attrs["__init__"] = init
76
 
        return type.__new__(cls, name, bases, attrs)
77
 
 
78
 
 
79
 
class udp_handler(SocketServer.DatagramRequestHandler, object):
80
 
    def handle(self):
81
 
        self.wfile.write("Polo")
82
 
        print "UDP request answered"
83
 
 
84
 
 
85
 
class IPv6_UDPServer(SocketServer.UDPServer, object):
86
 
    __metaclass__ = server_metaclass
87
 
    def verify_request(self, request, client_address):
88
 
        print "UDP request came"
89
 
        return request[0] == "Marco"
 
335
                    command = self.check_command % escaped_attrs
 
336
                except TypeError, error:
 
337
                    logger.error(u'Could not format string "%s":'
 
338
                                 u' %s', self.check_command, error)
 
339
                    return True # Try again later
 
340
            try:
 
341
                logger.debug(u"Starting checker %r for %s",
 
342
                             command, self.name)
 
343
                self.checker = subprocess.Popen(command,
 
344
                                                close_fds=True,
 
345
                                                shell=True, cwd="/")
 
346
                self.checker_callback_tag = gobject.child_watch_add\
 
347
                                            (self.checker.pid,
 
348
                                             self.checker_callback)
 
349
            except subprocess.OSError, error:
 
350
                logger.error(u"Failed to start subprocess: %s",
 
351
                             error)
 
352
        # Re-run this periodically if run by gobject.timeout_add
 
353
        return True
 
354
    def stop_checker(self):
 
355
        """Force the checker process, if any, to stop."""
 
356
        if self.checker_callback_tag:
 
357
            gobject.source_remove(self.checker_callback_tag)
 
358
            self.checker_callback_tag = None
 
359
        if getattr(self, "checker", None) is None:
 
360
            return
 
361
        logger.debug("Stopping checker for %(name)s", vars(self))
 
362
        try:
 
363
            os.kill(self.checker.pid, signal.SIGTERM)
 
364
            #os.sleep(0.5)
 
365
            #if self.checker.poll() is None:
 
366
            #    os.kill(self.checker.pid, signal.SIGKILL)
 
367
        except OSError, error:
 
368
            if error.errno != errno.ESRCH: # No such process
 
369
                raise
 
370
        self.checker = None
 
371
    def still_valid(self):
 
372
        """Has the timeout not yet passed for this client?"""
 
373
        now = datetime.datetime.now()
 
374
        if self.last_checked_ok is None:
 
375
            return now < (self.created + self.timeout)
 
376
        else:
 
377
            return now < (self.last_checked_ok + self.timeout)
 
378
 
 
379
 
 
380
def peer_certificate(session):
 
381
    "Return the peer's OpenPGP certificate as a bytestring"
 
382
    # If not an OpenPGP certificate...
 
383
    if gnutls.library.functions.gnutls_certificate_type_get\
 
384
            (session._c_object) \
 
385
           != gnutls.library.constants.GNUTLS_CRT_OPENPGP:
 
386
        # ...do the normal thing
 
387
        return session.peer_certificate
 
388
    list_size = ctypes.c_uint()
 
389
    cert_list = gnutls.library.functions.gnutls_certificate_get_peers\
 
390
        (session._c_object, ctypes.byref(list_size))
 
391
    if list_size.value == 0:
 
392
        return None
 
393
    cert = cert_list[0]
 
394
    return ctypes.string_at(cert.data, cert.size)
 
395
 
 
396
 
 
397
def fingerprint(openpgp):
 
398
    "Convert an OpenPGP bytestring to a hexdigit fingerprint string"
 
399
    # New empty GnuTLS certificate
 
400
    crt = gnutls.library.types.gnutls_openpgp_crt_t()
 
401
    gnutls.library.functions.gnutls_openpgp_crt_init\
 
402
        (ctypes.byref(crt))
 
403
    # New GnuTLS "datum" with the OpenPGP public key
 
404
    datum = gnutls.library.types.gnutls_datum_t\
 
405
        (ctypes.cast(ctypes.c_char_p(openpgp),
 
406
                     ctypes.POINTER(ctypes.c_ubyte)),
 
407
         ctypes.c_uint(len(openpgp)))
 
408
    # Import the OpenPGP public key into the certificate
 
409
    ret = gnutls.library.functions.gnutls_openpgp_crt_import\
 
410
        (crt,
 
411
         ctypes.byref(datum),
 
412
         gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
 
413
    # New buffer for the fingerprint
 
414
    buffer = ctypes.create_string_buffer(20)
 
415
    buffer_length = ctypes.c_size_t()
 
416
    # Get the fingerprint from the certificate into the buffer
 
417
    gnutls.library.functions.gnutls_openpgp_crt_get_fingerprint\
 
418
        (crt, ctypes.byref(buffer), ctypes.byref(buffer_length))
 
419
    # Deinit the certificate
 
420
    gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
 
421
    # Convert the buffer to a Python bytestring
 
422
    fpr = ctypes.string_at(buffer, buffer_length.value)
 
423
    # Convert the bytestring to hexadecimal notation
 
424
    hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
 
425
    return hex_fpr
90
426
 
91
427
 
92
428
class tcp_handler(SocketServer.BaseRequestHandler, object):
 
429
    """A TCP request handler class.
 
430
    Instantiated by IPv6_TCPServer for each request to handle it.
 
431
    Note: This will run in its own forked process."""
 
432
    
93
433
    def handle(self):
94
 
        print "TCP request came"
95
 
        print "Request:", self.request
96
 
        print "Client Address:", self.client_address
97
 
        print "Server:", self.server
98
 
        session = gnutls.connection.ServerSession(self.request,
99
 
                                                  self.server.credentials)
100
 
        session.handshake()
101
 
        if session.peer_certificate:
102
 
            print "DN:", session.peer_certificate.subject
103
 
        try:
104
 
            session.verify_peer()
105
 
        except gnutls.errors.CertificateError, error:
106
 
            print "Verify failed", error
107
 
            session.bye()
108
 
            return
109
 
        try:
110
 
            session.send(dict((client.dn, client.password)
111
 
                              for client in self.server.clients)
112
 
                         [session.peer_certificate.subject])
113
 
        except KeyError:
114
 
            session.send("gazonk")
115
 
            # Log maybe? XXX
 
434
        logger.debug(u"TCP connection from: %s",
 
435
                     unicode(self.client_address))
 
436
 
 
437
        line = self.request.makefile().readline()
 
438
        logger.debug(u"Protocol version: %r", line)
 
439
        try:
 
440
            if int(line.strip().split()[0]) > 1:
 
441
                raise RuntimeError
 
442
        except (ValueError, IndexError, RuntimeError), error:
 
443
            logger.error(u"Unknown protocol version: %s", error)
 
444
            return
 
445
        
 
446
        session = gnutls.connection.ClientSession\
 
447
                  (self.request, gnutls.connection.X509Credentials())
 
448
        # Note: gnutls.connection.X509Credentials is really a generic
 
449
        # GnuTLS certificate credentials object so long as no X.509
 
450
        # keys are added to it.  Therefore, we can use it here despite
 
451
        # using OpenPGP certificates.
 
452
        
 
453
        #priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
 
454
        #                "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
 
455
        #                "+DHE-DSS"))
 
456
        priority = "NORMAL"             # Fallback default, since this
 
457
                                        # MUST be set.
 
458
        if self.server.settings["priority"]:
 
459
            priority = self.server.settings["priority"]
 
460
        gnutls.library.functions.gnutls_priority_set_direct\
 
461
            (session._c_object, priority, None);
 
462
        
 
463
        try:
 
464
            session.handshake()
 
465
        except gnutls.errors.GNUTLSError, error:
 
466
            logger.debug(u"Handshake failed: %s", error)
 
467
            # Do not run session.bye() here: the session is not
 
468
            # established.  Just abandon the request.
 
469
            return
 
470
        try:
 
471
            fpr = fingerprint(peer_certificate(session))
 
472
        except (TypeError, gnutls.errors.GNUTLSError), error:
 
473
            logger.debug(u"Bad certificate: %s", error)
 
474
            session.bye()
 
475
            return
 
476
        logger.debug(u"Fingerprint: %s", fpr)
 
477
        client = None
 
478
        for c in self.server.clients:
 
479
            if c.fingerprint == fpr:
 
480
                client = c
 
481
                break
 
482
        if not client:
 
483
            logger.debug(u"Client not found for fingerprint: %s", fpr)
 
484
            session.bye()
 
485
            return
 
486
        # Have to check if client.still_valid(), since it is possible
 
487
        # that the client timed out while establishing the GnuTLS
 
488
        # session.
 
489
        if not client.still_valid():
 
490
            logger.debug(u"Client %(name)s is invalid", vars(client))
 
491
            session.bye()
 
492
            return
 
493
        sent_size = 0
 
494
        while sent_size < len(client.secret):
 
495
            sent = session.send(client.secret[sent_size:])
 
496
            logger.debug(u"Sent: %d, remaining: %d",
 
497
                         sent, len(client.secret)
 
498
                         - (sent_size + sent))
 
499
            sent_size += sent
116
500
        session.bye()
117
501
 
118
502
 
119
503
class IPv6_TCPServer(SocketServer.ForkingTCPServer, object):
120
 
    __metaclass__ = server_metaclass
121
 
    request_queue_size = 1024
122
 
 
123
 
 
124
 
in6addr_any = "::"
 
504
    """IPv6 TCP server.  Accepts 'None' as address and/or port.
 
505
    Attributes:
 
506
        settings:       Server settings
 
507
        clients:        Set() of Client objects
 
508
    """
 
509
    address_family = socket.AF_INET6
 
510
    def __init__(self, *args, **kwargs):
 
511
        if "settings" in kwargs:
 
512
            self.settings = kwargs["settings"]
 
513
            del kwargs["settings"]
 
514
        if "clients" in kwargs:
 
515
            self.clients = kwargs["clients"]
 
516
            del kwargs["clients"]
 
517
        return super(type(self), self).__init__(*args, **kwargs)
 
518
    def server_bind(self):
 
519
        """This overrides the normal server_bind() function
 
520
        to bind to an interface if one was specified, and also NOT to
 
521
        bind to an address or port if they were not specified."""
 
522
        if self.settings["interface"]:
 
523
            # 25 is from /usr/include/asm-i486/socket.h
 
524
            SO_BINDTODEVICE = getattr(socket, "SO_BINDTODEVICE", 25)
 
525
            try:
 
526
                self.socket.setsockopt(socket.SOL_SOCKET,
 
527
                                       SO_BINDTODEVICE,
 
528
                                       self.settings["interface"])
 
529
            except socket.error, error:
 
530
                if error[0] == errno.EPERM:
 
531
                    logger.warning(u"No permission to"
 
532
                                   u" bind to interface %s",
 
533
                                   self.settings["interface"])
 
534
                else:
 
535
                    raise error
 
536
        # Only bind(2) the socket if we really need to.
 
537
        if self.server_address[0] or self.server_address[1]:
 
538
            if not self.server_address[0]:
 
539
                in6addr_any = "::"
 
540
                self.server_address = (in6addr_any,
 
541
                                       self.server_address[1])
 
542
            elif self.server_address[1] is None:
 
543
                self.server_address = (self.server_address[0],
 
544
                                       0)
 
545
            return super(type(self), self).server_bind()
 
546
 
125
547
 
126
548
def string_to_delta(interval):
127
549
    """Parse a string and return a datetime.timedelta
157
579
    return delta
158
580
 
159
581
 
 
582
def server_state_changed(state):
 
583
    """Derived from the Avahi example code"""
 
584
    if state == avahi.SERVER_COLLISION:
 
585
        logger.warning(u"Server name collision")
 
586
        service.remove()
 
587
    elif state == avahi.SERVER_RUNNING:
 
588
        service.add()
 
589
 
 
590
 
 
591
def entry_group_state_changed(state, error):
 
592
    """Derived from the Avahi example code"""
 
593
    logger.debug(u"state change: %i", state)
 
594
    
 
595
    if state == avahi.ENTRY_GROUP_ESTABLISHED:
 
596
        logger.debug(u"Service established.")
 
597
    elif state == avahi.ENTRY_GROUP_COLLISION:
 
598
        logger.warning(u"Service name collision.")
 
599
        service.rename()
 
600
    elif state == avahi.ENTRY_GROUP_FAILURE:
 
601
        logger.critical(u"Error in group state changed %s",
 
602
                        unicode(error))
 
603
        raise AvahiGroupError("State changed: %s", str(error))
 
604
 
 
605
def if_nametoindex(interface, _func=[None]):
 
606
    """Call the C function if_nametoindex(), or equivalent"""
 
607
    if _func[0] is not None:
 
608
        return _func[0](interface)
 
609
    try:
 
610
        if "ctypes.util" not in sys.modules:
 
611
            import ctypes.util
 
612
        while True:
 
613
            try:
 
614
                libc = ctypes.cdll.LoadLibrary\
 
615
                       (ctypes.util.find_library("c"))
 
616
                _func[0] = libc.if_nametoindex
 
617
                return _func[0](interface)
 
618
            except IOError, e:
 
619
                if e != errno.EINTR:
 
620
                    raise
 
621
    except (OSError, AttributeError):
 
622
        if "struct" not in sys.modules:
 
623
            import struct
 
624
        if "fcntl" not in sys.modules:
 
625
            import fcntl
 
626
        def the_hard_way(interface):
 
627
            "Get an interface index the hard way, i.e. using fcntl()"
 
628
            SIOCGIFINDEX = 0x8933  # From /usr/include/linux/sockios.h
 
629
            s = socket.socket()
 
630
            ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
 
631
                                struct.pack("16s16x", interface))
 
632
            s.close()
 
633
            interface_index = struct.unpack("I", ifreq[16:20])[0]
 
634
            return interface_index
 
635
        _func[0] = the_hard_way
 
636
        return _func[0](interface)
 
637
 
 
638
 
 
639
def daemon(nochdir, noclose):
 
640
    """See daemon(3).  Standard BSD Unix function.
 
641
    This should really exist as os.daemon, but it doesn't (yet)."""
 
642
    if os.fork():
 
643
        sys.exit()
 
644
    os.setsid()
 
645
    if not nochdir:
 
646
        os.chdir("/")
 
647
    if not noclose:
 
648
        # Close all standard open file descriptors
 
649
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
 
650
        if not stat.S_ISCHR(os.fstat(null).st_mode):
 
651
            raise OSError(errno.ENODEV,
 
652
                          "/dev/null not a character device")
 
653
        os.dup2(null, sys.stdin.fileno())
 
654
        os.dup2(null, sys.stdout.fileno())
 
655
        os.dup2(null, sys.stderr.fileno())
 
656
        if null > 2:
 
657
            os.close(null)
 
658
 
 
659
 
160
660
def main():
 
661
    global main_loop_started
 
662
    main_loop_started = False
 
663
    
161
664
    parser = OptionParser()
162
665
    parser.add_option("-i", "--interface", type="string",
163
 
                      default="eth0", metavar="IF",
164
 
                      help="Interface to bind to")
165
 
    parser.add_option("--cert", type="string", default="cert.pem",
166
 
                      metavar="FILE",
167
 
                      help="Public key certificate to use")
168
 
    parser.add_option("--key", type="string", default="key.pem",
169
 
                      metavar="FILE",
170
 
                      help="Private key to use")
171
 
    parser.add_option("--ca", type="string", default="ca.pem",
172
 
                      metavar="FILE",
173
 
                      help="Certificate Authority certificate to use")
174
 
    parser.add_option("--crl", type="string", default="crl.pem",
175
 
                      metavar="FILE",
176
 
                      help="Certificate Revokation List to use")
177
 
    parser.add_option("-p", "--port", type="int", default=49001,
 
666
                      metavar="IF", help="Bind to interface IF")
 
667
    parser.add_option("-a", "--address", type="string",
 
668
                      help="Address to listen for requests on")
 
669
    parser.add_option("-p", "--port", type="int",
178
670
                      help="Port number to receive requests on")
179
 
    parser.add_option("--dh", type="int", metavar="BITS",
180
 
                      help="DH group to use")
181
 
    parser.add_option("-t", "--timeout", type="string", # Parsed later
182
 
                      default="15m",
183
 
                      help="Amount of downtime allowed for clients")
184
 
    parser.add_option("--interval", type="string", # Parsed later
185
 
                      default="5m",
186
 
                      help="How often to check that a client is up")
187
671
    parser.add_option("--check", action="store_true", default=False,
188
672
                      help="Run self-test")
 
673
    parser.add_option("--debug", action="store_true", default=False,
 
674
                      help="Debug mode; run in foreground and log to"
 
675
                      " terminal")
 
676
    parser.add_option("--priority", type="string", help="GnuTLS"
 
677
                      " priority string (see GnuTLS documentation)")
 
678
    parser.add_option("--servicename", type="string", metavar="NAME",
 
679
                      help="Zeroconf service name")
 
680
    parser.add_option("--configdir", type="string",
 
681
                      default="/etc/mandos", metavar="DIR",
 
682
                      help="Directory to search for configuration"
 
683
                      " files")
189
684
    (options, args) = parser.parse_args()
190
 
 
 
685
    
191
686
    if options.check:
192
687
        import doctest
193
688
        doctest.testmod()
194
689
        sys.exit()
195
690
    
196
 
    # Parse the time arguments
197
 
    try:
198
 
        options.timeout = string_to_delta(options.timeout)
199
 
    except ValueError:
200
 
        parser.error("option --timeout: Unparseable time")
201
 
    
202
 
    try:
203
 
        options.interval = string_to_delta(options.interval)
204
 
    except ValueError:
205
 
        parser.error("option --interval: Unparseable time")
206
 
    
207
 
    cert = gnutls.crypto.X509Certificate(open(options.cert).read())
208
 
    key = gnutls.crypto.X509PrivateKey(open(options.key).read())
209
 
    ca = gnutls.crypto.X509Certificate(open(options.ca).read())
210
 
    crl = gnutls.crypto.X509CRL(open(options.crl).read())
211
 
    cred = gnutls.connection.X509Credentials(cert, key, [ca], [crl])
212
 
    
213
 
    # Parse config file
214
 
    defaults = {}
215
 
    client_config_object = ConfigParser.SafeConfigParser(defaults)
216
 
    client_config_object.read("mandos-clients.conf")
217
 
    clients = [Client(name=section, options=options,
218
 
                      **(dict(client_config_object.items(section))))
219
 
               for section in client_config_object.sections()]
220
 
    
221
 
    udp_server = IPv6_UDPServer((in6addr_any, options.port),
222
 
                                udp_handler,
223
 
                                options=options)
224
 
    
225
 
    tcp_server = IPv6_TCPServer((in6addr_any, options.port),
 
691
    # Default values for config file for server-global settings
 
692
    server_defaults = { "interface": "",
 
693
                        "address": "",
 
694
                        "port": "",
 
695
                        "debug": "False",
 
696
                        "priority":
 
697
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
 
698
                        "servicename": "Mandos",
 
699
                        }
 
700
    
 
701
    # Parse config file for server-global settings
 
702
    server_config = ConfigParser.SafeConfigParser(server_defaults)
 
703
    del server_defaults
 
704
    server_config.read(os.path.join(options.configdir, "server.conf"))
 
705
    server_section = "server"
 
706
    # Convert the SafeConfigParser object to a dict
 
707
    server_settings = dict(server_config.items(server_section))
 
708
    # Use getboolean on the boolean config option
 
709
    server_settings["debug"] = server_config.getboolean\
 
710
                               (server_section, "debug")
 
711
    del server_config
 
712
    
 
713
    # Override the settings from the config file with command line
 
714
    # options, if set.
 
715
    for option in ("interface", "address", "port", "debug",
 
716
                   "priority", "servicename", "configdir"):
 
717
        value = getattr(options, option)
 
718
        if value is not None:
 
719
            server_settings[option] = value
 
720
    del options
 
721
    # Now we have our good server settings in "server_settings"
 
722
    
 
723
    # Parse config file with clients
 
724
    client_defaults = { "timeout": "1h",
 
725
                        "interval": "5m",
 
726
                        "checker": "fping -q -- %%(fqdn)s",
 
727
                        }
 
728
    client_config = ConfigParser.SafeConfigParser(client_defaults)
 
729
    client_config.read(os.path.join(server_settings["configdir"],
 
730
                                    "clients.conf"))
 
731
    
 
732
    global service
 
733
    service = AvahiService(name = server_settings["servicename"],
 
734
                           type = "_mandos._tcp", );
 
735
    if server_settings["interface"]:
 
736
        service.interface = if_nametoindex(server_settings["interface"])
 
737
    
 
738
    global main_loop
 
739
    global bus
 
740
    global server
 
741
    # From the Avahi example code
 
742
    DBusGMainLoop(set_as_default=True )
 
743
    main_loop = gobject.MainLoop()
 
744
    bus = dbus.SystemBus()
 
745
    server = dbus.Interface(
 
746
            bus.get_object( avahi.DBUS_NAME, avahi.DBUS_PATH_SERVER ),
 
747
            avahi.DBUS_INTERFACE_SERVER )
 
748
    # End of Avahi example code
 
749
    
 
750
    debug = server_settings["debug"]
 
751
    
 
752
    if debug:
 
753
        console = logging.StreamHandler()
 
754
        # console.setLevel(logging.DEBUG)
 
755
        console.setFormatter(logging.Formatter\
 
756
                             ('%(levelname)s: %(message)s'))
 
757
        logger.addHandler(console)
 
758
        del console
 
759
    
 
760
    clients = Set()
 
761
    def remove_from_clients(client):
 
762
        clients.remove(client)
 
763
        if not clients:
 
764
            logger.debug(u"No clients left, exiting")
 
765
            sys.exit()
 
766
    
 
767
    clients.update(Set(Client(name=section,
 
768
                              stop_hook = remove_from_clients,
 
769
                              **(dict(client_config\
 
770
                                      .items(section))))
 
771
                       for section in client_config.sections()))
 
772
    
 
773
    if not debug:
 
774
        daemon(False, False)
 
775
    
 
776
    def cleanup():
 
777
        "Cleanup function; run on exit"
 
778
        global group
 
779
        # From the Avahi example code
 
780
        if not group is None:
 
781
            group.Free()
 
782
            group = None
 
783
        # End of Avahi example code
 
784
        
 
785
        while clients:
 
786
            client = clients.pop()
 
787
            client.stop_hook = None
 
788
            client.stop()
 
789
    
 
790
    atexit.register(cleanup)
 
791
    
 
792
    if not debug:
 
793
        signal.signal(signal.SIGINT, signal.SIG_IGN)
 
794
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
 
795
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
 
796
    
 
797
    for client in clients:
 
798
        client.start()
 
799
    
 
800
    tcp_server = IPv6_TCPServer((server_settings["address"],
 
801
                                 server_settings["port"]),
226
802
                                tcp_handler,
227
 
                                options=options,
228
 
                                clients=clients,
229
 
                                credentials=cred)
230
 
    
231
 
    while True:
232
 
        in_, out, err = select.select((udp_server,
233
 
                                       tcp_server), (), ())
234
 
        for server in in_:
235
 
            server.handle_request()
236
 
 
237
 
 
238
 
if __name__ == "__main__":
 
803
                                settings=server_settings,
 
804
                                clients=clients)
 
805
    # Find out what port we got
 
806
    service.port = tcp_server.socket.getsockname()[1]
 
807
    logger.debug(u"Now listening on address %r, port %d, flowinfo %d,"
 
808
                 u" scope_id %d" % tcp_server.socket.getsockname())
 
809
    
 
810
    #service.interface = tcp_server.socket.getsockname()[3]
 
811
    
 
812
    try:
 
813
        # From the Avahi example code
 
814
        server.connect_to_signal("StateChanged", server_state_changed)
 
815
        try:
 
816
            server_state_changed(server.GetState())
 
817
        except dbus.exceptions.DBusException, error:
 
818
            logger.critical(u"DBusException: %s", error)
 
819
            sys.exit(1)
 
820
        # End of Avahi example code
 
821
        
 
822
        gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
 
823
                             lambda *args, **kwargs:
 
824
                             tcp_server.handle_request\
 
825
                             (*args[2:], **kwargs) or True)
 
826
        
 
827
        logger.debug("Starting main loop")
 
828
        main_loop_started = True
 
829
        main_loop.run()
 
830
    except AvahiError, error:
 
831
        logger.critical(u"AvahiError: %s" + unicode(error))
 
832
        sys.exit(1)
 
833
    except KeyboardInterrupt:
 
834
        if debug:
 
835
            print
 
836
 
 
837
if __name__ == '__main__':
239
838
    main()
240