2
 
# -*- mode: python; coding: utf-8 -*-
 
 
2
# -*- mode: python; after-save-hook: (lambda () (let ((command (if (fboundp 'file-local-name) (file-local-name (buffer-file-name)) (or (file-remote-p (buffer-file-name) 'localname) (buffer-file-name))))) (if (= (progn (if (get-buffer "*Test*") (kill-buffer "*Test*")) (process-file-shell-command (format "%s --check" (shell-quote-argument command)) nil "*Test*")) 0) (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w))) (progn (with-current-buffer "*Test*" (compilation-mode)) (display-buffer "*Test*" '(display-buffer-in-side-window)))))); coding: utf-8 -*-
 
4
4
# Mandos server - give out binary blobs to connecting clients.
 
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
9
# methods "add", "remove", "server_state_changed",
 
10
10
# "entry_group_state_changed", "cleanup", and "activate" in the
 
11
11
# "AvahiService" class, and some lines in "main".
 
13
13
# Everything else is
 
14
 
# Copyright © 2008-2016 Teddy Hogeborn
 
15
 
# Copyright © 2008-2016 Björn Påhlsson
 
17
 
# This program is free software: you can redistribute it and/or modify
 
18
 
# it under the terms of the GNU General Public License as published by
 
 
14
# Copyright © 2008-2019 Teddy Hogeborn
 
 
15
# Copyright © 2008-2019 Björn Påhlsson
 
 
17
# This file is part of Mandos.
 
 
19
# Mandos is free software: you can redistribute it and/or modify it
 
 
20
# under the terms of the GNU General Public License as published by
 
19
21
# the Free Software Foundation, either version 3 of the License, or
 
20
22
# (at your option) any later version.
 
22
 
#     This program is distributed in the hope that it will be useful,
 
23
 
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
 
 
24
#     Mandos is distributed in the hope that it will be useful, but
 
 
25
#     WITHOUT ANY WARRANTY; without even the implied warranty of
 
24
26
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
25
27
#     GNU General Public License for more details.
 
27
29
# You should have received a copy of the GNU General Public License
 
28
 
# along with this program.  If not, see
 
29
 
# <http://www.gnu.org/licenses/>.
 
 
30
# along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
 
31
32
# Contact the authors at <mandos@recompile.se>.
 
34
35
from __future__ import (division, absolute_import, print_function,
 
 
86
89
import xml.dom.minidom
 
 
92
if sys.version_info.major == 2:
 
 
95
# Show warnings by default
 
 
96
if not sys.warnoptions:
 
 
98
    warnings.simplefilter("default")
 
 
100
# Try to find the value of SO_BINDTODEVICE:
 
 
102
    # This is where SO_BINDTODEVICE is in Python 3.3 (or 3.4?) and
 
 
103
    # newer, and it is also the most natural place for it:
 
90
104
    SO_BINDTODEVICE = socket.SO_BINDTODEVICE
 
91
105
except AttributeError:
 
 
107
        # This is where SO_BINDTODEVICE was up to and including Python
 
93
109
        from IN import SO_BINDTODEVICE
 
94
110
    except ImportError:
 
95
 
        SO_BINDTODEVICE = None
 
 
111
        # In Python 2.7 it seems to have been removed entirely.
 
 
112
        # Try running the C preprocessor:
 
 
114
            cc = subprocess.Popen(["cc", "--language=c", "-E",
 
 
116
                                  stdin=subprocess.PIPE,
 
 
117
                                  stdout=subprocess.PIPE)
 
 
118
            stdout = cc.communicate(
 
 
119
                "#include <sys/socket.h>\nSO_BINDTODEVICE\n")[0]
 
 
120
            SO_BINDTODEVICE = int(stdout.splitlines()[-1])
 
 
121
        except (OSError, ValueError, IndexError):
 
 
123
            SO_BINDTODEVICE = None
 
97
125
if sys.version_info.major == 2:
 
 
128
if sys.version_info < (3, 2):
 
 
129
    configparser.Configparser = configparser.SafeConfigParser
 
101
132
stored_state_file = "clients.pickle"
 
103
134
logger = logging.getLogger()
 
 
135
logging.captureWarnings(True)   # Show warnings via the logging system
 
107
139
    if_nametoindex = ctypes.cdll.LoadLibrary(
 
108
140
        ctypes.util.find_library("c")).if_nametoindex
 
109
141
except (OSError, AttributeError):
 
111
143
    def if_nametoindex(interface):
 
112
144
        "Get an interface index the hard way, i.e. using fcntl()"
 
113
145
        SIOCGIFINDEX = 0x8933  # From /usr/include/linux/sockios.h
 
 
227
261
                                     '--passphrase-file',
 
229
263
                                    + self.gnupgargs,
 
230
 
                                    stdin = subprocess.PIPE,
 
231
 
                                    stdout = subprocess.PIPE,
 
232
 
                                    stderr = subprocess.PIPE)
 
233
 
            ciphertext, err = proc.communicate(input = data)
 
 
264
                                    stdin=subprocess.PIPE,
 
 
265
                                    stdout=subprocess.PIPE,
 
 
266
                                    stderr=subprocess.PIPE)
 
 
267
            ciphertext, err = proc.communicate(input=data)
 
234
268
        if proc.returncode != 0:
 
235
269
            raise PGPError(err)
 
236
270
        return ciphertext
 
238
272
    def decrypt(self, data, password):
 
239
273
        passphrase = self.password_encode(password)
 
240
274
        with tempfile.NamedTemporaryFile(
 
241
 
                dir = self.tempdir) as passfile:
 
 
275
                dir=self.tempdir) as passfile:
 
242
276
            passfile.write(passphrase)
 
244
278
            proc = subprocess.Popen([self.gpg, '--decrypt',
 
245
279
                                     '--passphrase-file',
 
247
281
                                    + self.gnupgargs,
 
248
 
                                    stdin = subprocess.PIPE,
 
249
 
                                    stdout = subprocess.PIPE,
 
250
 
                                    stderr = subprocess.PIPE)
 
251
 
            decrypted_plaintext, err = proc.communicate(input = data)
 
 
282
                                    stdin=subprocess.PIPE,
 
 
283
                                    stdout=subprocess.PIPE,
 
 
284
                                    stderr=subprocess.PIPE)
 
 
285
            decrypted_plaintext, err = proc.communicate(input=data)
 
252
286
        if proc.returncode != 0:
 
253
287
            raise PGPError(err)
 
254
288
        return decrypted_plaintext
 
256
291
# Pretend that we have an Avahi module
 
258
 
    """This isn't so much a class as it is a module-like namespace.
 
259
 
    It is instantiated once, and simulates having an Avahi module."""
 
260
 
    IF_UNSPEC = -1              # avahi-common/address.h
 
261
 
    PROTO_UNSPEC = -1           # avahi-common/address.h
 
262
 
    PROTO_INET = 0              # avahi-common/address.h
 
263
 
    PROTO_INET6 = 1             # avahi-common/address.h
 
 
293
    """This isn't so much a class as it is a module-like namespace."""
 
 
294
    IF_UNSPEC = -1               # avahi-common/address.h
 
 
295
    PROTO_UNSPEC = -1            # avahi-common/address.h
 
 
296
    PROTO_INET = 0               # avahi-common/address.h
 
 
297
    PROTO_INET6 = 1              # avahi-common/address.h
 
264
298
    DBUS_NAME = "org.freedesktop.Avahi"
 
265
299
    DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
 
266
300
    DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
 
267
301
    DBUS_PATH_SERVER = "/"
 
268
 
    def string_array_to_txt_array(self, t):
 
 
304
    def string_array_to_txt_array(t):
 
269
305
        return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
 
270
306
                           for s in t), signature="ay")
 
271
 
    ENTRY_GROUP_ESTABLISHED = 2 # avahi-common/defs.h
 
272
 
    ENTRY_GROUP_COLLISION = 3   # avahi-common/defs.h
 
273
 
    ENTRY_GROUP_FAILURE = 4     # avahi-common/defs.h
 
274
 
    SERVER_INVALID = 0          # avahi-common/defs.h
 
275
 
    SERVER_REGISTERING = 1      # avahi-common/defs.h
 
276
 
    SERVER_RUNNING = 2          # avahi-common/defs.h
 
277
 
    SERVER_COLLISION = 3        # avahi-common/defs.h
 
278
 
    SERVER_FAILURE = 4          # avahi-common/defs.h
 
 
307
    ENTRY_GROUP_ESTABLISHED = 2  # avahi-common/defs.h
 
 
308
    ENTRY_GROUP_COLLISION = 3    # avahi-common/defs.h
 
 
309
    ENTRY_GROUP_FAILURE = 4      # avahi-common/defs.h
 
 
310
    SERVER_INVALID = 0           # avahi-common/defs.h
 
 
311
    SERVER_REGISTERING = 1       # avahi-common/defs.h
 
 
312
    SERVER_RUNNING = 2           # avahi-common/defs.h
 
 
313
    SERVER_COLLISION = 3         # avahi-common/defs.h
 
 
314
    SERVER_FAILURE = 4           # avahi-common/defs.h
 
281
317
class AvahiError(Exception):
 
282
318
    def __init__(self, value, *args, **kwargs):
 
 
473
509
class AvahiServiceToSyslog(AvahiService):
 
474
510
    def rename(self, *args, **kwargs):
 
475
511
        """Add the new name to the syslog messages"""
 
476
 
        ret = AvahiService.rename(self, *args, **kwargs)
 
 
512
        ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
 
477
513
        syslogger.setFormatter(logging.Formatter(
 
478
514
            'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
 
479
515
            .format(self.name)))
 
482
519
# Pretend that we have a GnuTLS module
 
483
 
class GnuTLS(object):
 
484
 
    """This isn't so much a class as it is a module-like namespace.
 
485
 
    It is instantiated once, and simulates having a GnuTLS module."""
 
487
 
    _library = ctypes.cdll.LoadLibrary(
 
488
 
        ctypes.util.find_library("gnutls"))
 
489
 
    _need_version = b"3.3.0"
 
491
 
        # Need to use class name "GnuTLS" here, since this method is
 
492
 
        # called before the assignment to the "gnutls" global variable
 
494
 
        if GnuTLS.check_version(self._need_version) is None:
 
495
 
            raise GnuTLS.Error("Needs GnuTLS {} or later"
 
496
 
                               .format(self._need_version))
 
 
521
    """This isn't so much a class as it is a module-like namespace."""
 
 
523
    library = ctypes.util.find_library("gnutls")
 
 
525
        library = ctypes.util.find_library("gnutls-deb0")
 
 
526
    _library = ctypes.cdll.LoadLibrary(library)
 
498
529
    # Unless otherwise indicated, the constants and types below are
 
499
530
    # all from the gnutls/gnutls.h C header file.
 
503
534
    E_INTERRUPTED = -52
 
508
540
    CRD_CERTIFICATE = 1
 
509
541
    E_NO_CERTIFICATE_FOUND = -49
 
 
546
    KEYID_USE_SHA256 = 1        # gnutls/x509.h
 
510
547
    OPENPGP_FMT_RAW = 0         # gnutls/openpgp.h
 
513
550
    class session_int(ctypes.Structure):
 
515
552
    session_t = ctypes.POINTER(session_int)
 
516
554
    class certificate_credentials_st(ctypes.Structure):
 
518
556
    certificate_credentials_t = ctypes.POINTER(
 
519
557
        certificate_credentials_st)
 
520
558
    certificate_type_t = ctypes.c_int
 
521
560
    class datum_t(ctypes.Structure):
 
522
561
        _fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
 
523
562
                    ('size', ctypes.c_uint)]
 
524
564
    class openpgp_crt_int(ctypes.Structure):
 
526
566
    openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
 
527
 
    openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
 
 
567
    openpgp_crt_fmt_t = ctypes.c_int  # gnutls/openpgp.h
 
528
568
    log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
 
529
569
    credentials_type_t = ctypes.c_int
 
530
570
    transport_ptr_t = ctypes.c_void_p
 
531
571
    close_request_t = ctypes.c_int
 
534
574
    class Error(Exception):
 
535
 
        # We need to use the class name "GnuTLS" here, since this
 
536
 
        # exception might be raised from within GnuTLS.__init__,
 
537
 
        # which is called before the assignment to the "gnutls"
 
538
 
        # global variable has happened.
 
539
 
        def __init__(self, message = None, code = None, args=()):
 
 
575
        def __init__(self, message=None, code=None, args=()):
 
540
576
            # Default usage is by a message string, but if a return
 
541
577
            # code is passed, convert it to a string with
 
542
578
            # gnutls.strerror()
 
544
580
            if message is None and code is not None:
 
545
 
                message = GnuTLS.strerror(code)
 
546
 
            return super(GnuTLS.Error, self).__init__(
 
 
581
                message = gnutls.strerror(code)
 
 
582
            return super(gnutls.Error, self).__init__(
 
549
585
    class CertificateSecurityError(Error):
 
553
 
    class Credentials(object):
 
554
590
        def __init__(self):
 
555
591
            self._c_object = gnutls.certificate_credentials_t()
 
556
592
            gnutls.certificate_allocate_credentials(
 
557
593
                ctypes.byref(self._c_object))
 
558
594
            self.type = gnutls.CRD_CERTIFICATE
 
560
596
        def __del__(self):
 
561
597
            gnutls.certificate_free_credentials(self._c_object)
 
563
 
    class ClientSession(object):
 
564
 
        def __init__(self, socket, credentials = None):
 
 
600
        def __init__(self, socket, credentials=None):
 
565
601
            self._c_object = gnutls.session_t()
 
566
 
            gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)
 
 
602
            gnutls_flags = gnutls.CLIENT
 
 
603
            if gnutls.check_version(b"3.5.6"):
 
 
604
                gnutls_flags |= gnutls.NO_TICKETS
 
 
606
                gnutls_flags |= gnutls.ENABLE_RAWPK
 
 
607
            gnutls.init(ctypes.byref(self._c_object), gnutls_flags)
 
567
609
            gnutls.set_default_priority(self._c_object)
 
568
610
            gnutls.transport_set_ptr(self._c_object, socket.fileno())
 
569
611
            gnutls.handshake_set_private_extensions(self._c_object,
 
 
611
653
                return _error_code(result)
 
612
654
            result = func(*arguments)
 
615
657
    # Unless otherwise indicated, the function declarations below are
 
616
658
    # all from the gnutls/gnutls.h C header file.
 
619
661
    priority_set_direct = _library.gnutls_priority_set_direct
 
620
662
    priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
 
621
663
                                    ctypes.POINTER(ctypes.c_char_p)]
 
622
664
    priority_set_direct.restype = _error_code
 
624
666
    init = _library.gnutls_init
 
625
667
    init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
 
626
668
    init.restype = _error_code
 
628
670
    set_default_priority = _library.gnutls_set_default_priority
 
629
671
    set_default_priority.argtypes = [session_t]
 
630
672
    set_default_priority.restype = _error_code
 
632
674
    record_send = _library.gnutls_record_send
 
633
675
    record_send.argtypes = [session_t, ctypes.c_void_p,
 
635
677
    record_send.restype = ctypes.c_ssize_t
 
636
678
    record_send.errcheck = _retry_on_error
 
638
680
    certificate_allocate_credentials = (
 
639
681
        _library.gnutls_certificate_allocate_credentials)
 
640
682
    certificate_allocate_credentials.argtypes = [
 
641
683
        ctypes.POINTER(certificate_credentials_t)]
 
642
684
    certificate_allocate_credentials.restype = _error_code
 
644
686
    certificate_free_credentials = (
 
645
687
        _library.gnutls_certificate_free_credentials)
 
646
 
    certificate_free_credentials.argtypes = [certificate_credentials_t]
 
 
688
    certificate_free_credentials.argtypes = [
 
 
689
        certificate_credentials_t]
 
647
690
    certificate_free_credentials.restype = None
 
649
692
    handshake_set_private_extensions = (
 
650
693
        _library.gnutls_handshake_set_private_extensions)
 
651
694
    handshake_set_private_extensions.argtypes = [session_t,
 
653
696
    handshake_set_private_extensions.restype = None
 
655
698
    credentials_set = _library.gnutls_credentials_set
 
656
699
    credentials_set.argtypes = [session_t, credentials_type_t,
 
658
701
    credentials_set.restype = _error_code
 
660
703
    strerror = _library.gnutls_strerror
 
661
704
    strerror.argtypes = [ctypes.c_int]
 
662
705
    strerror.restype = ctypes.c_char_p
 
664
707
    certificate_type_get = _library.gnutls_certificate_type_get
 
665
708
    certificate_type_get.argtypes = [session_t]
 
666
709
    certificate_type_get.restype = _error_code
 
668
711
    certificate_get_peers = _library.gnutls_certificate_get_peers
 
669
712
    certificate_get_peers.argtypes = [session_t,
 
670
713
                                      ctypes.POINTER(ctypes.c_uint)]
 
671
714
    certificate_get_peers.restype = ctypes.POINTER(datum_t)
 
673
716
    global_set_log_level = _library.gnutls_global_set_log_level
 
674
717
    global_set_log_level.argtypes = [ctypes.c_int]
 
675
718
    global_set_log_level.restype = None
 
677
720
    global_set_log_function = _library.gnutls_global_set_log_function
 
678
721
    global_set_log_function.argtypes = [log_func]
 
679
722
    global_set_log_function.restype = None
 
681
724
    deinit = _library.gnutls_deinit
 
682
725
    deinit.argtypes = [session_t]
 
683
726
    deinit.restype = None
 
685
728
    handshake = _library.gnutls_handshake
 
686
729
    handshake.argtypes = [session_t]
 
687
730
    handshake.restype = _error_code
 
688
731
    handshake.errcheck = _retry_on_error
 
690
733
    transport_set_ptr = _library.gnutls_transport_set_ptr
 
691
734
    transport_set_ptr.argtypes = [session_t, transport_ptr_t]
 
692
735
    transport_set_ptr.restype = None
 
694
737
    bye = _library.gnutls_bye
 
695
738
    bye.argtypes = [session_t, close_request_t]
 
696
739
    bye.restype = _error_code
 
697
740
    bye.errcheck = _retry_on_error
 
699
742
    check_version = _library.gnutls_check_version
 
700
743
    check_version.argtypes = [ctypes.c_char_p]
 
701
744
    check_version.restype = ctypes.c_char_p
 
703
 
    # All the function declarations below are from gnutls/openpgp.h
 
705
 
    openpgp_crt_init = _library.gnutls_openpgp_crt_init
 
706
 
    openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
 
707
 
    openpgp_crt_init.restype = _error_code
 
709
 
    openpgp_crt_import = _library.gnutls_openpgp_crt_import
 
710
 
    openpgp_crt_import.argtypes = [openpgp_crt_t,
 
711
 
                                   ctypes.POINTER(datum_t),
 
713
 
    openpgp_crt_import.restype = _error_code
 
715
 
    openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
 
716
 
    openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
 
717
 
                                        ctypes.POINTER(ctypes.c_uint)]
 
718
 
    openpgp_crt_verify_self.restype = _error_code
 
720
 
    openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
 
721
 
    openpgp_crt_deinit.argtypes = [openpgp_crt_t]
 
722
 
    openpgp_crt_deinit.restype = None
 
724
 
    openpgp_crt_get_fingerprint = (
 
725
 
        _library.gnutls_openpgp_crt_get_fingerprint)
 
726
 
    openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
 
730
 
    openpgp_crt_get_fingerprint.restype = _error_code
 
 
746
    _need_version = b"3.3.0"
 
 
747
    if check_version(_need_version) is None:
 
 
748
        raise self.Error("Needs GnuTLS {} or later"
 
 
749
                         .format(_need_version))
 
 
751
    _tls_rawpk_version = b"3.6.6"
 
 
752
    has_rawpk = bool(check_version(_tls_rawpk_version))
 
 
756
        class pubkey_st(ctypes.Structure):
 
 
758
        pubkey_t = ctypes.POINTER(pubkey_st)
 
 
760
        x509_crt_fmt_t = ctypes.c_int
 
 
762
        # All the function declarations below are from gnutls/abstract.h
 
 
763
        pubkey_init = _library.gnutls_pubkey_init
 
 
764
        pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
 
 
765
        pubkey_init.restype = _error_code
 
 
767
        pubkey_import = _library.gnutls_pubkey_import
 
 
768
        pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
 
 
770
        pubkey_import.restype = _error_code
 
 
772
        pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
 
 
773
        pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
 
 
774
                                      ctypes.POINTER(ctypes.c_ubyte),
 
 
775
                                      ctypes.POINTER(ctypes.c_size_t)]
 
 
776
        pubkey_get_key_id.restype = _error_code
 
 
778
        pubkey_deinit = _library.gnutls_pubkey_deinit
 
 
779
        pubkey_deinit.argtypes = [pubkey_t]
 
 
780
        pubkey_deinit.restype = None
 
 
782
        # All the function declarations below are from gnutls/openpgp.h
 
 
784
        openpgp_crt_init = _library.gnutls_openpgp_crt_init
 
 
785
        openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
 
 
786
        openpgp_crt_init.restype = _error_code
 
 
788
        openpgp_crt_import = _library.gnutls_openpgp_crt_import
 
 
789
        openpgp_crt_import.argtypes = [openpgp_crt_t,
 
 
790
                                       ctypes.POINTER(datum_t),
 
 
792
        openpgp_crt_import.restype = _error_code
 
 
794
        openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
 
 
795
        openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
 
 
796
                                            ctypes.POINTER(ctypes.c_uint)]
 
 
797
        openpgp_crt_verify_self.restype = _error_code
 
 
799
        openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
 
 
800
        openpgp_crt_deinit.argtypes = [openpgp_crt_t]
 
 
801
        openpgp_crt_deinit.restype = None
 
 
803
        openpgp_crt_get_fingerprint = (
 
 
804
            _library.gnutls_openpgp_crt_get_fingerprint)
 
 
805
        openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
 
 
809
        openpgp_crt_get_fingerprint.restype = _error_code
 
 
811
    if check_version(b"3.6.4"):
 
 
812
        certificate_type_get2 = _library.gnutls_certificate_type_get2
 
 
813
        certificate_type_get2.argtypes = [session_t, ctypes.c_int]
 
 
814
        certificate_type_get2.restype = _error_code
 
732
816
    # Remove non-public functions
 
733
817
    del _error_code, _retry_on_error
 
734
 
# Create the global "gnutls" object, simulating a module
 
737
820
def call_pipe(connection,       # : multiprocessing.Connection
 
738
821
              func, *args, **kwargs):
 
739
822
    """This function is meant to be called by multiprocessing.Process
 
741
824
    This function runs func(*args, **kwargs), and writes the resulting
 
742
825
    return value on the provided multiprocessing.Connection.
 
744
827
    connection.send(func(*args, **kwargs))
 
745
828
    connection.close()
 
747
 
class Client(object):
 
748
832
    """A representation of a client host served by this server.
 
751
835
    approved:   bool(); 'None' if not yet approved/disapproved
 
752
836
    approval_delay: datetime.timedelta(); Time to wait for approval
 
753
837
    approval_duration: datetime.timedelta(); Duration of one approval
 
754
 
    checker:    subprocess.Popen(); a running checker process used
 
755
 
                                    to see if the client lives.
 
756
 
                                    'None' if no process is running.
 
 
838
    checker: multiprocessing.Process(); a running checker process used
 
 
839
             to see if the client lives. 'None' if no process is
 
757
841
    checker_callback_tag: a GLib event source tag, or None
 
758
842
    checker_command: string; External command which is run to check
 
759
843
                     if client lives.  %() expansions are done at
 
 
1428
1519
                         exc_info=error)
 
1429
1520
        return xmlstring
 
1432
1524
    dbus.OBJECT_MANAGER_IFACE
 
1433
1525
except AttributeError:
 
1434
1526
    dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
 
1436
1529
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
 
1437
1530
    """A D-Bus object with an ObjectManager.
 
1439
1532
    Classes inheriting from this exposes the standard
 
1440
1533
    GetManagedObjects call and the InterfacesAdded and
 
1441
1534
    InterfacesRemoved signals on the standard
 
1442
1535
    "org.freedesktop.DBus.ObjectManager" interface.
 
1444
1537
    Note: No signals are sent automatically; they must be sent
 
1447
1540
    @dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
 
1448
 
                         out_signature = "a{oa{sa{sv}}}")
 
 
1541
                         out_signature="a{oa{sa{sv}}}")
 
1449
1542
    def GetManagedObjects(self):
 
1450
1543
        """This function must be overridden"""
 
1451
1544
        raise NotImplementedError()
 
1453
1546
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
 
1454
 
                         signature = "oa{sa{sv}}")
 
 
1547
                         signature="oa{sa{sv}}")
 
1455
1548
    def InterfacesAdded(self, object_path, interfaces_and_properties):
 
1458
 
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
 
 
1551
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
 
1459
1552
    def InterfacesRemoved(self, object_path, interfaces):
 
1462
1555
    @dbus.service.method(dbus.INTROSPECTABLE_IFACE,
 
1463
 
                         out_signature = "s",
 
1464
 
                         path_keyword = 'object_path',
 
1465
 
                         connection_keyword = 'connection')
 
 
1557
                         path_keyword='object_path',
 
 
1558
                         connection_keyword='connection')
 
1466
1559
    def Introspect(self, object_path, connection):
 
1467
1560
        """Overloading of standard D-Bus method.
 
1469
1562
        Override return argument name of GetManagedObjects to be
 
1470
1563
        "objpath_interfaces_and_properties"
 
 
2152
2255
class ClientHandler(socketserver.BaseRequestHandler, object):
 
2153
2256
    """A class to handle client connections.
 
2155
2258
    Instantiated once for each connection to handle it.
 
2156
2259
    Note: This will run in its own forked process."""
 
2158
2261
    def handle(self):
 
2159
2262
        with contextlib.closing(self.server.child_pipe) as child_pipe:
 
2160
2263
            logger.info("TCP connection from: %s",
 
2161
2264
                        str(self.client_address))
 
2162
2265
            logger.debug("Pipe FD: %d",
 
2163
2266
                         self.server.child_pipe.fileno())
 
2165
2268
            session = gnutls.ClientSession(self.request)
 
2167
 
            #priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
2168
 
            #                      "+AES-256-CBC", "+SHA1",
 
2169
 
            #                      "+COMP-NULL", "+CTYPE-OPENPGP",
 
 
2270
            # priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
 
2271
            #                       "+AES-256-CBC", "+SHA1",
 
 
2272
            #                       "+COMP-NULL", "+CTYPE-OPENPGP",
 
2171
2274
            # Use a fallback default, since this MUST be set.
 
2172
2275
            priority = self.server.gnutls_priority
 
2173
2276
            if priority is None:
 
2174
2277
                priority = "NORMAL"
 
2175
 
            gnutls.priority_set_direct(session._c_object, priority,
 
 
2278
            gnutls.priority_set_direct(session._c_object,
 
 
2279
                                       priority.encode("utf-8"),
 
2178
2282
            # Start communication using the Mandos protocol
 
2179
2283
            # Get protocol number
 
2180
2284
            line = self.request.makefile().readline()
 
 
2345
 
class MultiprocessingMixIn(object):
 
 
2506
class MultiprocessingMixIn:
 
2346
2507
    """Like socketserver.ThreadingMixIn, but with multiprocessing"""
 
2348
2509
    def sub_process_main(self, request, address):
 
2350
2511
            self.finish_request(request, address)
 
2351
2512
        except Exception:
 
2352
2513
            self.handle_error(request, address)
 
2353
2514
        self.close_request(request)
 
2355
2516
    def process_request(self, request, address):
 
2356
2517
        """Start a new process to process the request."""
 
2357
 
        proc = multiprocessing.Process(target = self.sub_process_main,
 
2358
 
                                       args = (request, address))
 
 
2518
        proc = multiprocessing.Process(target=self.sub_process_main,
 
 
2519
                                       args=(request, address))
 
2363
 
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
 
 
2524
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
 
2364
2525
    """ adds a pipe to the MixIn """
 
2366
2527
    def process_request(self, request, client_address):
 
2367
2528
        """Overrides and wraps the original process_request().
 
2369
2530
        This function creates a new pipe in self.pipe
 
2371
2532
        parent_pipe, self.child_pipe = multiprocessing.Pipe()
 
2373
2534
        proc = MultiprocessingMixIn.process_request(self, request,
 
2374
2535
                                                    client_address)
 
2375
2536
        self.child_pipe.close()
 
2376
2537
        self.add_pipe(parent_pipe, proc)
 
2378
2539
    def add_pipe(self, parent_pipe, proc):
 
2379
2540
        """Dummy function; override as necessary"""
 
2380
2541
        raise NotImplementedError()
 
2383
2544
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
 
2384
 
                     socketserver.TCPServer, object):
 
 
2545
                     socketserver.TCPServer):
 
2385
2546
    """IPv6-capable TCP server.  Accepts 'None' as address and/or port
 
2388
2549
        enabled:        Boolean; whether this server is activated yet
 
2389
2550
        interface:      None or a network interface name (string)
 
2390
2551
        use_ipv6:       Boolean; to use IPv6 or not
 
2393
2554
    def __init__(self, server_address, RequestHandlerClass,
 
2394
2555
                 interface=None,
 
 
2428
2590
        # socket_wrapper(), if socketfd was set.
 
2429
2591
        socketserver.TCPServer.__init__(self, server_address,
 
2430
2592
                                        RequestHandlerClass)
 
2432
2594
    def server_bind(self):
 
2433
2595
        """This overrides the normal server_bind() function
 
2434
2596
        to bind to an interface if one was specified, and also NOT to
 
2435
2597
        bind to an address or port if they were not specified."""
 
 
2598
        global SO_BINDTODEVICE
 
2436
2599
        if self.interface is not None:
 
2437
2600
            if SO_BINDTODEVICE is None:
 
2438
 
                logger.error("SO_BINDTODEVICE does not exist;"
 
2439
 
                             " cannot bind to interface %s",
 
2443
 
                    self.socket.setsockopt(
 
2444
 
                        socket.SOL_SOCKET, SO_BINDTODEVICE,
 
2445
 
                        (self.interface + "\0").encode("utf-8"))
 
2446
 
                except socket.error as error:
 
2447
 
                    if error.errno == errno.EPERM:
 
2448
 
                        logger.error("No permission to bind to"
 
2449
 
                                     " interface %s", self.interface)
 
2450
 
                    elif error.errno == errno.ENOPROTOOPT:
 
2451
 
                        logger.error("SO_BINDTODEVICE not available;"
 
2452
 
                                     " cannot bind to interface %s",
 
2454
 
                    elif error.errno == errno.ENODEV:
 
2455
 
                        logger.error("Interface %s does not exist,"
 
2456
 
                                     " cannot bind", self.interface)
 
 
2601
                # Fall back to a hard-coded value which seems to be
 
 
2603
                logger.warning("SO_BINDTODEVICE not found, trying 25")
 
 
2604
                SO_BINDTODEVICE = 25
 
 
2606
                self.socket.setsockopt(
 
 
2607
                    socket.SOL_SOCKET, SO_BINDTODEVICE,
 
 
2608
                    (self.interface + "\0").encode("utf-8"))
 
 
2609
            except socket.error as error:
 
 
2610
                if error.errno == errno.EPERM:
 
 
2611
                    logger.error("No permission to bind to"
 
 
2612
                                 " interface %s", self.interface)
 
 
2613
                elif error.errno == errno.ENOPROTOOPT:
 
 
2614
                    logger.error("SO_BINDTODEVICE not available;"
 
 
2615
                                 " cannot bind to interface %s",
 
 
2617
                elif error.errno == errno.ENODEV:
 
 
2618
                    logger.error("Interface %s does not exist,"
 
 
2619
                                 " cannot bind", self.interface)
 
2459
2622
        # Only bind(2) the socket if we really need to.
 
2460
2623
        if self.server_address[0] or self.server_address[1]:
 
 
2624
            if self.server_address[1]:
 
 
2625
                self.allow_reuse_address = True
 
2461
2626
            if not self.server_address[0]:
 
2462
2627
                if self.address_family == socket.AF_INET6:
 
2463
 
                    any_address = "::" # in6addr_any
 
 
2628
                    any_address = "::"  # in6addr_any
 
2465
 
                    any_address = "0.0.0.0" # INADDR_ANY
 
 
2630
                    any_address = "0.0.0.0"  # INADDR_ANY
 
2466
2631
                self.server_address = (any_address,
 
2467
2632
                                       self.server_address[1])
 
2468
2633
            elif not self.server_address[1]:
 
 
2502
2667
        self.gnutls_priority = gnutls_priority
 
2503
2668
        IPv6_TCPServer.__init__(self, server_address,
 
2504
2669
                                RequestHandlerClass,
 
2505
 
                                interface = interface,
 
2506
 
                                use_ipv6 = use_ipv6,
 
2507
 
                                socketfd = socketfd)
 
 
2670
                                interface=interface,
 
2509
2674
    def server_activate(self):
 
2510
2675
        if self.enabled:
 
2511
2676
            return socketserver.TCPServer.server_activate(self)
 
2513
2678
    def enable(self):
 
2514
2679
        self.enabled = True
 
2516
2681
    def add_pipe(self, parent_pipe, proc):
 
2517
2682
        # Call "handle_ipc" for both data and EOF events
 
2518
2683
        GLib.io_add_watch(
 
2519
 
            parent_pipe.fileno(),
 
2520
 
            GLib.IO_IN | GLib.IO_HUP,
 
 
2684
            GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
 
2685
            GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
 
2521
2686
            functools.partial(self.handle_ipc,
 
2522
 
                              parent_pipe = parent_pipe,
 
 
2687
                              parent_pipe=parent_pipe,
 
2525
2690
    def handle_ipc(self, source, condition,
 
2526
2691
                   parent_pipe=None,
 
2528
2693
                   client_object=None):
 
2529
2694
        # error, or the other end of multiprocessing.Pipe has closed
 
2530
2695
        if condition & (GLib.IO_ERR | GLib.IO_HUP):
 
2531
2696
            # Wait for other process to exit
 
2535
2700
        # Read a request from the child
 
2536
2701
        request = parent_pipe.recv()
 
2537
2702
        command = request[0]
 
2539
2704
        if command == 'init':
 
2541
 
            address = request[2]
 
 
2705
            key_id = request[1].decode("ascii")
 
 
2706
            fpr = request[2].decode("ascii")
 
 
2707
            address = request[3]
 
2543
2709
            for c in self.clients.values():
 
2544
 
                if c.fingerprint == fpr:
 
 
2710
                if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
 
 
2712
                if key_id and c.key_id == key_id:
 
 
2715
                if fpr and c.fingerprint == fpr:
 
2548
 
                logger.info("Client not found for fingerprint: %s, ad"
 
2549
 
                            "dress: %s", fpr, address)
 
 
2719
                logger.info("Client not found for key ID: %s, address"
 
 
2720
                            ": %s", key_id or fpr, address)
 
2550
2721
                if self.use_dbus:
 
2551
2722
                    # Emit D-Bus signal
 
2552
 
                    mandos_dbus_service.ClientNotFound(fpr,
 
 
2723
                    mandos_dbus_service.ClientNotFound(key_id or fpr,
 
2554
2725
                parent_pipe.send(False)
 
2557
2728
            GLib.io_add_watch(
 
2558
 
                parent_pipe.fileno(),
 
2559
 
                GLib.IO_IN | GLib.IO_HUP,
 
 
2729
                GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
 
2730
                GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
 
2560
2731
                functools.partial(self.handle_ipc,
 
2561
 
                                  parent_pipe = parent_pipe,
 
2563
 
                                  client_object = client))
 
 
2732
                                  parent_pipe=parent_pipe,
 
 
2734
                                  client_object=client))
 
2564
2735
            parent_pipe.send(True)
 
2565
2736
            # remove the old hook in favor of the new above hook on
 
 
2583
2754
                parent_pipe.send((
 
2584
2755
                    'data', client_object.__getattribute__(attrname)))
 
2586
2757
        if command == 'setattr':
 
2587
2758
            attrname = request[1]
 
2588
2759
            value = request[2]
 
2589
2760
            setattr(client_object, attrname, value)
 
2594
2765
def rfc3339_duration_to_delta(duration):
 
2595
2766
    """Parse an RFC 3339 "duration" and return a datetime.timedelta
 
2597
 
    >>> rfc3339_duration_to_delta("P7D")
 
2598
 
    datetime.timedelta(7)
 
2599
 
    >>> rfc3339_duration_to_delta("PT60S")
 
2600
 
    datetime.timedelta(0, 60)
 
2601
 
    >>> rfc3339_duration_to_delta("PT60M")
 
2602
 
    datetime.timedelta(0, 3600)
 
2603
 
    >>> rfc3339_duration_to_delta("PT24H")
 
2604
 
    datetime.timedelta(1)
 
2605
 
    >>> rfc3339_duration_to_delta("P1W")
 
2606
 
    datetime.timedelta(7)
 
2607
 
    >>> rfc3339_duration_to_delta("PT5M30S")
 
2608
 
    datetime.timedelta(0, 330)
 
2609
 
    >>> rfc3339_duration_to_delta("P1DT3M20S")
 
2610
 
    datetime.timedelta(1, 200)
 
 
2768
    >>> rfc3339_duration_to_delta("P7D") == datetime.timedelta(7)
 
 
2770
    >>> rfc3339_duration_to_delta("PT60S") == datetime.timedelta(0, 60)
 
 
2772
    >>> rfc3339_duration_to_delta("PT60M") == datetime.timedelta(0, 3600)
 
 
2774
    >>> rfc3339_duration_to_delta("PT24H") == datetime.timedelta(1)
 
 
2776
    >>> rfc3339_duration_to_delta("P1W") == datetime.timedelta(7)
 
 
2778
    >>> rfc3339_duration_to_delta("PT5M30S") == datetime.timedelta(0, 330)
 
 
2780
    >>> rfc3339_duration_to_delta("P1DT3M20S") == datetime.timedelta(1, 200)
 
2613
2784
    # Parsing an RFC 3339 duration with regular expressions is not
 
2614
2785
    # possible - there would have to be multiple places for the same
 
2615
2786
    # values, like seconds.  The current code, while more esoteric, is
 
2616
2787
    # cleaner without depending on a parsing library.  If Python had a
 
2617
2788
    # built-in library for parsing we would use it, but we'd like to
 
2618
2789
    # avoid excessive use of external libraries.
 
2620
2791
    # New type for defining tokens, syntax, and semantics all-in-one
 
2621
2792
    Token = collections.namedtuple("Token", (
 
2622
2793
        "regexp",  # To match token; if "value" is not None, must have
 
 
2806
2980
    parser.add_argument("--no-zeroconf", action="store_false",
 
2807
2981
                        dest="zeroconf", help="Do not use Zeroconf",
 
2810
2984
    options = parser.parse_args()
 
2814
 
        fail_count, test_count = doctest.testmod()
 
2815
 
        sys.exit(os.EX_OK if fail_count == 0 else 1)
 
2817
2986
    # Default values for config file for server-global settings
 
2818
 
    server_defaults = { "interface": "",
 
2823
 
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
 
2824
 
                        ":+SIGN-DSA-SHA256",
 
2825
 
                        "servicename": "Mandos",
 
2831
 
                        "statedir": "/var/lib/mandos",
 
2832
 
                        "foreground": "False",
 
 
2987
    if gnutls.has_rawpk:
 
 
2988
        priority = ("SECURE128:!CTYPE-X.509:+CTYPE-RAWPK:!RSA"
 
 
2989
                    ":!VERS-ALL:+VERS-TLS1.3:%PROFILE_ULTRA")
 
 
2991
        priority = ("SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
 
 
2992
                    ":+SIGN-DSA-SHA256")
 
 
2993
    server_defaults = {"interface": "",
 
 
2997
                       "priority": priority,
 
 
2998
                       "servicename": "Mandos",
 
 
3004
                       "statedir": "/var/lib/mandos",
 
 
3005
                       "foreground": "False",
 
2836
3010
    # Parse config file for server-global settings
 
2837
 
    server_config = configparser.SafeConfigParser(server_defaults)
 
 
3011
    server_config = configparser.ConfigParser(server_defaults)
 
2838
3012
    del server_defaults
 
2839
3013
    server_config.read(os.path.join(options.configdir, "mandos.conf"))
 
2840
 
    # Convert the SafeConfigParser object to a dict
 
 
3014
    # Convert the ConfigParser object to a dict
 
2841
3015
    server_settings = server_config.defaults()
 
2842
3016
    # Use the appropriate methods on the non-string config options
 
2843
 
    for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
 
 
3017
    for option in ("debug", "use_dbus", "use_ipv6", "restore",
 
 
3018
                   "foreground", "zeroconf"):
 
2844
3019
        server_settings[option] = server_config.getboolean("DEFAULT",
 
2846
3021
    if server_settings["port"]:
 
 
3019
3194
        protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
 
3020
3195
        service = AvahiServiceToSyslog(
 
3021
 
            name = server_settings["servicename"],
 
3022
 
            servicetype = "_mandos._tcp",
 
3023
 
            protocol = protocol,
 
 
3196
            name=server_settings["servicename"],
 
 
3197
            servicetype="_mandos._tcp",
 
3025
3200
        if server_settings["interface"]:
 
3026
3201
            service.interface = if_nametoindex(
 
3027
3202
                server_settings["interface"].encode("utf-8"))
 
3029
3204
    global multiprocessing_manager
 
3030
3205
    multiprocessing_manager = multiprocessing.Manager()
 
3032
3207
    client_class = Client
 
3034
 
        client_class = functools.partial(ClientDBus, bus = bus)
 
 
3209
        client_class = functools.partial(ClientDBus, bus=bus)
 
3036
3211
    client_settings = Client.config_parser(client_config)
 
3037
3212
    old_client_settings = {}
 
3038
3213
    clients_data = {}
 
3040
3215
    # This is used to redirect stdout and stderr for checker processes
 
3042
 
    wnull = open(os.devnull, "w") # A writable /dev/null
 
 
3217
    wnull = open(os.devnull, "w")  # A writable /dev/null
 
3043
3218
    # Only used if server is running in foreground but not in debug
 
3045
3220
    if debug or not foreground:
 
3048
3223
    # Get client data and settings from last running state.
 
3049
3224
    if server_settings["restore"]:
 
3051
3226
            with open(stored_state_path, "rb") as stored_state:
 
3052
 
                if sys.version_info.major == 2:                
 
 
3227
                if sys.version_info.major == 2:
 
3053
3228
                    clients_data, old_client_settings = pickle.load(
 
3056
3231
                    bytes_clients_data, bytes_old_client_settings = (
 
3057
 
                        pickle.load(stored_state, encoding = "bytes"))
 
3058
 
                    ### Fix bytes to strings
 
 
3232
                        pickle.load(stored_state, encoding="bytes"))
 
 
3233
                    #   Fix bytes to strings
 
3061
 
                    clients_data = { (key.decode("utf-8")
 
3062
 
                                      if isinstance(key, bytes)
 
3065
 
                                     bytes_clients_data.items() }
 
 
3236
                    clients_data = {(key.decode("utf-8")
 
 
3237
                                     if isinstance(key, bytes)
 
 
3240
                                    bytes_clients_data.items()}
 
3066
3241
                    del bytes_clients_data
 
3067
3242
                    for key in clients_data:
 
3068
 
                        value = { (k.decode("utf-8")
 
3069
 
                                   if isinstance(k, bytes) else k): v
 
3071
 
                                  clients_data[key].items() }
 
 
3243
                        value = {(k.decode("utf-8")
 
 
3244
                                  if isinstance(k, bytes) else k): v
 
 
3246
                                 clients_data[key].items()}
 
3072
3247
                        clients_data[key] = value
 
3073
3248
                        # .client_structure
 
3074
3249
                        value["client_structure"] = [
 
3075
3250
                            (s.decode("utf-8")
 
3076
3251
                             if isinstance(s, bytes)
 
3077
3252
                             else s) for s in
 
3078
 
                            value["client_structure"] ]
 
 
3253
                            value["client_structure"]]
 
3079
3254
                        # .name & .host
 
3080
3255
                        for k in ("name", "host"):
 
3081
3256
                            if isinstance(value[k], bytes):
 
3082
3257
                                value[k] = value[k].decode("utf-8")
 
3083
 
                    ## old_client_settings
 
 
3258
                        if "key_id" not in value:
 
 
3259
                            value["key_id"] = ""
 
 
3260
                        elif "fingerprint" not in value:
 
 
3261
                            value["fingerprint"] = ""
 
 
3262
                    #  old_client_settings
 
3085
3264
                    old_client_settings = {
 
3086
3265
                        (key.decode("utf-8")
 
3087
3266
                         if isinstance(key, bytes)
 
3088
3267
                         else key): value
 
3089
3268
                        for key, value in
 
3090
 
                        bytes_old_client_settings.items() }
 
 
3269
                        bytes_old_client_settings.items()}
 
3091
3270
                    del bytes_old_client_settings
 
3093
3272
                    for value in old_client_settings.values():