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 -*-
 
 
2
# -*- mode: python; 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-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
 
 
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
 
21
19
# the Free Software Foundation, either version 3 of the License, or
 
22
20
# (at your option) any later version.
 
24
 
#     Mandos is distributed in the hope that it will be useful, but
 
25
 
#     WITHOUT ANY WARRANTY; without even the implied warranty of
 
 
22
#     This program is distributed in the hope that it will be useful,
 
 
23
#     but WITHOUT ANY WARRANTY; without even the implied warranty of
 
26
24
#     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
27
25
#     GNU General Public License for more details.
 
29
27
# You should have received a copy of the GNU General Public License
 
30
 
# along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
 
 
28
# along with this program.  If not, see
 
 
29
# <http://www.gnu.org/licenses/>.
 
32
31
# Contact the authors at <mandos@recompile.se>.
 
35
34
from __future__ import (division, absolute_import, print_function,
 
 
89
86
import xml.dom.minidom
 
92
 
if sys.version_info.major == 2:
 
96
 
# Show warnings by default
 
97
 
if not sys.warnoptions:
 
99
 
    warnings.simplefilter("default")
 
101
 
# Try to find the value of SO_BINDTODEVICE:
 
103
 
    # This is where SO_BINDTODEVICE is in Python 3.3 (or 3.4?) and
 
104
 
    # newer, and it is also the most natural place for it:
 
105
90
    SO_BINDTODEVICE = socket.SO_BINDTODEVICE
 
106
91
except AttributeError:
 
108
 
        # This is where SO_BINDTODEVICE was up to and including Python
 
110
93
        from IN import SO_BINDTODEVICE
 
111
94
    except ImportError:
 
112
 
        # In Python 2.7 it seems to have been removed entirely.
 
113
 
        # Try running the C preprocessor:
 
115
 
            cc = subprocess.Popen(["cc", "--language=c", "-E",
 
117
 
                                  stdin=subprocess.PIPE,
 
118
 
                                  stdout=subprocess.PIPE)
 
119
 
            stdout = cc.communicate(
 
120
 
                "#include <sys/socket.h>\nSO_BINDTODEVICE\n")[0]
 
121
 
            SO_BINDTODEVICE = int(stdout.splitlines()[-1])
 
122
 
        except (OSError, ValueError, IndexError):
 
124
 
            SO_BINDTODEVICE = None
 
126
 
if sys.version_info < (3, 2):
 
127
 
    configparser.Configparser = configparser.SafeConfigParser
 
 
95
        SO_BINDTODEVICE = None
 
 
97
if sys.version_info.major == 2:
 
130
101
stored_state_file = "clients.pickle"
 
132
103
logger = logging.getLogger()
 
133
 
logging.captureWarnings(True)   # Show warnings via the logging system
 
137
107
    if_nametoindex = ctypes.cdll.LoadLibrary(
 
138
108
        ctypes.util.find_library("c")).if_nametoindex
 
139
109
except (OSError, AttributeError):
 
141
111
    def if_nametoindex(interface):
 
142
112
        "Get an interface index the hard way, i.e. using fcntl()"
 
143
113
        SIOCGIFINDEX = 0x8933  # From /usr/include/linux/sockios.h
 
 
259
227
                                     '--passphrase-file',
 
261
229
                                    + self.gnupgargs,
 
262
 
                                    stdin=subprocess.PIPE,
 
263
 
                                    stdout=subprocess.PIPE,
 
264
 
                                    stderr=subprocess.PIPE)
 
265
 
            ciphertext, err = proc.communicate(input=data)
 
 
230
                                    stdin = subprocess.PIPE,
 
 
231
                                    stdout = subprocess.PIPE,
 
 
232
                                    stderr = subprocess.PIPE)
 
 
233
            ciphertext, err = proc.communicate(input = data)
 
266
234
        if proc.returncode != 0:
 
267
235
            raise PGPError(err)
 
268
236
        return ciphertext
 
270
238
    def decrypt(self, data, password):
 
271
239
        passphrase = self.password_encode(password)
 
272
240
        with tempfile.NamedTemporaryFile(
 
273
 
                dir=self.tempdir) as passfile:
 
 
241
                dir = self.tempdir) as passfile:
 
274
242
            passfile.write(passphrase)
 
276
244
            proc = subprocess.Popen([self.gpg, '--decrypt',
 
277
245
                                     '--passphrase-file',
 
279
247
                                    + self.gnupgargs,
 
280
 
                                    stdin=subprocess.PIPE,
 
281
 
                                    stdout=subprocess.PIPE,
 
282
 
                                    stderr=subprocess.PIPE)
 
283
 
            decrypted_plaintext, err = proc.communicate(input=data)
 
 
248
                                    stdin = subprocess.PIPE,
 
 
249
                                    stdout = subprocess.PIPE,
 
 
250
                                    stderr = subprocess.PIPE)
 
 
251
            decrypted_plaintext, err = proc.communicate(input = data)
 
284
252
        if proc.returncode != 0:
 
285
253
            raise PGPError(err)
 
286
254
        return decrypted_plaintext
 
289
256
# Pretend that we have an Avahi module
 
291
 
    """This isn't so much a class as it is a module-like namespace."""
 
292
 
    IF_UNSPEC = -1               # avahi-common/address.h
 
293
 
    PROTO_UNSPEC = -1            # avahi-common/address.h
 
294
 
    PROTO_INET = 0               # avahi-common/address.h
 
295
 
    PROTO_INET6 = 1              # avahi-common/address.h
 
 
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
 
296
264
    DBUS_NAME = "org.freedesktop.Avahi"
 
297
265
    DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
 
298
266
    DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
 
299
267
    DBUS_PATH_SERVER = "/"
 
302
 
    def string_array_to_txt_array(t):
 
 
268
    def string_array_to_txt_array(self, t):
 
303
269
        return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
 
304
270
                           for s in t), signature="ay")
 
305
 
    ENTRY_GROUP_ESTABLISHED = 2  # avahi-common/defs.h
 
306
 
    ENTRY_GROUP_COLLISION = 3    # avahi-common/defs.h
 
307
 
    ENTRY_GROUP_FAILURE = 4      # avahi-common/defs.h
 
308
 
    SERVER_INVALID = 0           # avahi-common/defs.h
 
309
 
    SERVER_REGISTERING = 1       # avahi-common/defs.h
 
310
 
    SERVER_RUNNING = 2           # avahi-common/defs.h
 
311
 
    SERVER_COLLISION = 3         # avahi-common/defs.h
 
312
 
    SERVER_FAILURE = 4           # avahi-common/defs.h
 
 
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
 
315
281
class AvahiError(Exception):
 
316
282
    def __init__(self, value, *args, **kwargs):
 
 
507
473
class AvahiServiceToSyslog(AvahiService):
 
508
474
    def rename(self, *args, **kwargs):
 
509
475
        """Add the new name to the syslog messages"""
 
510
 
        ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
 
 
476
        ret = AvahiService.rename(self, *args, **kwargs)
 
511
477
        syslogger.setFormatter(logging.Formatter(
 
512
478
            'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
 
513
479
            .format(self.name)))
 
517
482
# Pretend that we have a GnuTLS module
 
519
 
    """This isn't so much a class as it is a module-like namespace."""
 
521
 
    library = ctypes.util.find_library("gnutls")
 
523
 
        library = ctypes.util.find_library("gnutls-deb0")
 
524
 
    _library = ctypes.cdll.LoadLibrary(library)
 
 
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))
 
527
498
    # Unless otherwise indicated, the constants and types below are
 
528
499
    # all from the gnutls/gnutls.h C header file.
 
532
503
    E_INTERRUPTED = -52
 
538
508
    CRD_CERTIFICATE = 1
 
539
509
    E_NO_CERTIFICATE_FOUND = -49
 
544
 
    KEYID_USE_SHA256 = 1        # gnutls/x509.h
 
545
510
    OPENPGP_FMT_RAW = 0         # gnutls/openpgp.h
 
548
513
    class session_int(ctypes.Structure):
 
550
515
    session_t = ctypes.POINTER(session_int)
 
552
516
    class certificate_credentials_st(ctypes.Structure):
 
554
518
    certificate_credentials_t = ctypes.POINTER(
 
555
519
        certificate_credentials_st)
 
556
520
    certificate_type_t = ctypes.c_int
 
558
521
    class datum_t(ctypes.Structure):
 
559
522
        _fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
 
560
523
                    ('size', ctypes.c_uint)]
 
562
524
    class openpgp_crt_int(ctypes.Structure):
 
564
526
    openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
 
565
 
    openpgp_crt_fmt_t = ctypes.c_int  # gnutls/openpgp.h
 
 
527
    openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
 
566
528
    log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
 
567
529
    credentials_type_t = ctypes.c_int
 
568
530
    transport_ptr_t = ctypes.c_void_p
 
569
531
    close_request_t = ctypes.c_int
 
572
534
    class Error(Exception):
 
573
 
        def __init__(self, message=None, code=None, args=()):
 
 
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=()):
 
574
540
            # Default usage is by a message string, but if a return
 
575
541
            # code is passed, convert it to a string with
 
576
542
            # gnutls.strerror()
 
578
544
            if message is None and code is not None:
 
579
 
                message = gnutls.strerror(code)
 
580
 
            return super(gnutls.Error, self).__init__(
 
 
545
                message = GnuTLS.strerror(code)
 
 
546
            return super(GnuTLS.Error, self).__init__(
 
583
549
    class CertificateSecurityError(Error):
 
 
553
    class Credentials(object):
 
588
554
        def __init__(self):
 
589
555
            self._c_object = gnutls.certificate_credentials_t()
 
590
556
            gnutls.certificate_allocate_credentials(
 
591
557
                ctypes.byref(self._c_object))
 
592
558
            self.type = gnutls.CRD_CERTIFICATE
 
594
560
        def __del__(self):
 
595
561
            gnutls.certificate_free_credentials(self._c_object)
 
598
 
        def __init__(self, socket, credentials=None):
 
 
563
    class ClientSession(object):
 
 
564
        def __init__(self, socket, credentials = None):
 
599
565
            self._c_object = gnutls.session_t()
 
600
 
            gnutls_flags = gnutls.CLIENT
 
601
 
            if gnutls.check_version(b"3.5.6"):
 
602
 
                gnutls_flags |= gnutls.NO_TICKETS
 
604
 
                gnutls_flags |= gnutls.ENABLE_RAWPK
 
605
 
            gnutls.init(ctypes.byref(self._c_object), gnutls_flags)
 
 
566
            gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)
 
607
567
            gnutls.set_default_priority(self._c_object)
 
608
568
            gnutls.transport_set_ptr(self._c_object, socket.fileno())
 
609
569
            gnutls.handshake_set_private_extensions(self._c_object,
 
 
651
611
                return _error_code(result)
 
652
612
            result = func(*arguments)
 
655
615
    # Unless otherwise indicated, the function declarations below are
 
656
616
    # all from the gnutls/gnutls.h C header file.
 
659
619
    priority_set_direct = _library.gnutls_priority_set_direct
 
660
620
    priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
 
661
621
                                    ctypes.POINTER(ctypes.c_char_p)]
 
662
622
    priority_set_direct.restype = _error_code
 
664
624
    init = _library.gnutls_init
 
665
625
    init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
 
666
626
    init.restype = _error_code
 
668
628
    set_default_priority = _library.gnutls_set_default_priority
 
669
629
    set_default_priority.argtypes = [session_t]
 
670
630
    set_default_priority.restype = _error_code
 
672
632
    record_send = _library.gnutls_record_send
 
673
633
    record_send.argtypes = [session_t, ctypes.c_void_p,
 
675
635
    record_send.restype = ctypes.c_ssize_t
 
676
636
    record_send.errcheck = _retry_on_error
 
678
638
    certificate_allocate_credentials = (
 
679
639
        _library.gnutls_certificate_allocate_credentials)
 
680
640
    certificate_allocate_credentials.argtypes = [
 
681
641
        ctypes.POINTER(certificate_credentials_t)]
 
682
642
    certificate_allocate_credentials.restype = _error_code
 
684
644
    certificate_free_credentials = (
 
685
645
        _library.gnutls_certificate_free_credentials)
 
686
 
    certificate_free_credentials.argtypes = [
 
687
 
        certificate_credentials_t]
 
 
646
    certificate_free_credentials.argtypes = [certificate_credentials_t]
 
688
647
    certificate_free_credentials.restype = None
 
690
649
    handshake_set_private_extensions = (
 
691
650
        _library.gnutls_handshake_set_private_extensions)
 
692
651
    handshake_set_private_extensions.argtypes = [session_t,
 
694
653
    handshake_set_private_extensions.restype = None
 
696
655
    credentials_set = _library.gnutls_credentials_set
 
697
656
    credentials_set.argtypes = [session_t, credentials_type_t,
 
699
658
    credentials_set.restype = _error_code
 
701
660
    strerror = _library.gnutls_strerror
 
702
661
    strerror.argtypes = [ctypes.c_int]
 
703
662
    strerror.restype = ctypes.c_char_p
 
705
664
    certificate_type_get = _library.gnutls_certificate_type_get
 
706
665
    certificate_type_get.argtypes = [session_t]
 
707
666
    certificate_type_get.restype = _error_code
 
709
668
    certificate_get_peers = _library.gnutls_certificate_get_peers
 
710
669
    certificate_get_peers.argtypes = [session_t,
 
711
670
                                      ctypes.POINTER(ctypes.c_uint)]
 
712
671
    certificate_get_peers.restype = ctypes.POINTER(datum_t)
 
714
673
    global_set_log_level = _library.gnutls_global_set_log_level
 
715
674
    global_set_log_level.argtypes = [ctypes.c_int]
 
716
675
    global_set_log_level.restype = None
 
718
677
    global_set_log_function = _library.gnutls_global_set_log_function
 
719
678
    global_set_log_function.argtypes = [log_func]
 
720
679
    global_set_log_function.restype = None
 
722
681
    deinit = _library.gnutls_deinit
 
723
682
    deinit.argtypes = [session_t]
 
724
683
    deinit.restype = None
 
726
685
    handshake = _library.gnutls_handshake
 
727
686
    handshake.argtypes = [session_t]
 
728
687
    handshake.restype = _error_code
 
729
688
    handshake.errcheck = _retry_on_error
 
731
690
    transport_set_ptr = _library.gnutls_transport_set_ptr
 
732
691
    transport_set_ptr.argtypes = [session_t, transport_ptr_t]
 
733
692
    transport_set_ptr.restype = None
 
735
694
    bye = _library.gnutls_bye
 
736
695
    bye.argtypes = [session_t, close_request_t]
 
737
696
    bye.restype = _error_code
 
738
697
    bye.errcheck = _retry_on_error
 
740
699
    check_version = _library.gnutls_check_version
 
741
700
    check_version.argtypes = [ctypes.c_char_p]
 
742
701
    check_version.restype = ctypes.c_char_p
 
744
 
    _need_version = b"3.3.0"
 
745
 
    if check_version(_need_version) is None:
 
746
 
        raise self.Error("Needs GnuTLS {} or later"
 
747
 
                         .format(_need_version))
 
749
 
    _tls_rawpk_version = b"3.6.6"
 
750
 
    has_rawpk = bool(check_version(_tls_rawpk_version))
 
754
 
        class pubkey_st(ctypes.Structure):
 
756
 
        pubkey_t = ctypes.POINTER(pubkey_st)
 
758
 
        x509_crt_fmt_t = ctypes.c_int
 
760
 
        # All the function declarations below are from gnutls/abstract.h
 
761
 
        pubkey_init = _library.gnutls_pubkey_init
 
762
 
        pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
 
763
 
        pubkey_init.restype = _error_code
 
765
 
        pubkey_import = _library.gnutls_pubkey_import
 
766
 
        pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
 
768
 
        pubkey_import.restype = _error_code
 
770
 
        pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
 
771
 
        pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
 
772
 
                                      ctypes.POINTER(ctypes.c_ubyte),
 
773
 
                                      ctypes.POINTER(ctypes.c_size_t)]
 
774
 
        pubkey_get_key_id.restype = _error_code
 
776
 
        pubkey_deinit = _library.gnutls_pubkey_deinit
 
777
 
        pubkey_deinit.argtypes = [pubkey_t]
 
778
 
        pubkey_deinit.restype = None
 
780
 
        # All the function declarations below are from gnutls/openpgp.h
 
782
 
        openpgp_crt_init = _library.gnutls_openpgp_crt_init
 
783
 
        openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
 
784
 
        openpgp_crt_init.restype = _error_code
 
786
 
        openpgp_crt_import = _library.gnutls_openpgp_crt_import
 
787
 
        openpgp_crt_import.argtypes = [openpgp_crt_t,
 
788
 
                                       ctypes.POINTER(datum_t),
 
790
 
        openpgp_crt_import.restype = _error_code
 
792
 
        openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
 
793
 
        openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
 
794
 
                                            ctypes.POINTER(ctypes.c_uint)]
 
795
 
        openpgp_crt_verify_self.restype = _error_code
 
797
 
        openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
 
798
 
        openpgp_crt_deinit.argtypes = [openpgp_crt_t]
 
799
 
        openpgp_crt_deinit.restype = None
 
801
 
        openpgp_crt_get_fingerprint = (
 
802
 
            _library.gnutls_openpgp_crt_get_fingerprint)
 
803
 
        openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
 
807
 
        openpgp_crt_get_fingerprint.restype = _error_code
 
809
 
    if check_version(b"3.6.4"):
 
810
 
        certificate_type_get2 = _library.gnutls_certificate_type_get2
 
811
 
        certificate_type_get2.argtypes = [session_t, ctypes.c_int]
 
812
 
        certificate_type_get2.restype = _error_code
 
 
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
 
814
732
    # Remove non-public functions
 
815
733
    del _error_code, _retry_on_error
 
 
734
# Create the global "gnutls" object, simulating a module
 
818
737
def call_pipe(connection,       # : multiprocessing.Connection
 
819
738
              func, *args, **kwargs):
 
820
739
    """This function is meant to be called by multiprocessing.Process
 
822
741
    This function runs func(*args, **kwargs), and writes the resulting
 
823
742
    return value on the provided multiprocessing.Connection.
 
825
744
    connection.send(func(*args, **kwargs))
 
826
745
    connection.close()
 
 
747
class Client(object):
 
830
748
    """A representation of a client host served by this server.
 
833
751
    approved:   bool(); 'None' if not yet approved/disapproved
 
834
752
    approval_delay: datetime.timedelta(); Time to wait for approval
 
835
753
    approval_duration: datetime.timedelta(); Duration of one approval
 
836
 
    checker: multiprocessing.Process(); a running checker process used
 
837
 
             to see if the client lives. 'None' if no process is
 
 
754
    checker:    subprocess.Popen(); a running checker process used
 
 
755
                                    to see if the client lives.
 
 
756
                                    'None' if no process is running.
 
839
757
    checker_callback_tag: a GLib event source tag, or None
 
840
758
    checker_command: string; External command which is run to check
 
841
759
                     if client lives.  %() expansions are done at
 
 
1517
1428
                         exc_info=error)
 
1518
1429
        return xmlstring
 
1522
1432
    dbus.OBJECT_MANAGER_IFACE
 
1523
1433
except AttributeError:
 
1524
1434
    dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
 
1527
1436
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
 
1528
1437
    """A D-Bus object with an ObjectManager.
 
1530
1439
    Classes inheriting from this exposes the standard
 
1531
1440
    GetManagedObjects call and the InterfacesAdded and
 
1532
1441
    InterfacesRemoved signals on the standard
 
1533
1442
    "org.freedesktop.DBus.ObjectManager" interface.
 
1535
1444
    Note: No signals are sent automatically; they must be sent
 
1538
1447
    @dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
 
1539
 
                         out_signature="a{oa{sa{sv}}}")
 
 
1448
                         out_signature = "a{oa{sa{sv}}}")
 
1540
1449
    def GetManagedObjects(self):
 
1541
1450
        """This function must be overridden"""
 
1542
1451
        raise NotImplementedError()
 
1544
1453
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
 
1545
 
                         signature="oa{sa{sv}}")
 
 
1454
                         signature = "oa{sa{sv}}")
 
1546
1455
    def InterfacesAdded(self, object_path, interfaces_and_properties):
 
1549
 
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
 
 
1458
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
 
1550
1459
    def InterfacesRemoved(self, object_path, interfaces):
 
1553
1462
    @dbus.service.method(dbus.INTROSPECTABLE_IFACE,
 
1555
 
                         path_keyword='object_path',
 
1556
 
                         connection_keyword='connection')
 
 
1463
                         out_signature = "s",
 
 
1464
                         path_keyword = 'object_path',
 
 
1465
                         connection_keyword = 'connection')
 
1557
1466
    def Introspect(self, object_path, connection):
 
1558
1467
        """Overloading of standard D-Bus method.
 
1560
1469
        Override return argument name of GetManagedObjects to be
 
1561
1470
        "objpath_interfaces_and_properties"
 
 
2253
2152
class ClientHandler(socketserver.BaseRequestHandler, object):
 
2254
2153
    """A class to handle client connections.
 
2256
2155
    Instantiated once for each connection to handle it.
 
2257
2156
    Note: This will run in its own forked process."""
 
2259
2158
    def handle(self):
 
2260
2159
        with contextlib.closing(self.server.child_pipe) as child_pipe:
 
2261
2160
            logger.info("TCP connection from: %s",
 
2262
2161
                        str(self.client_address))
 
2263
2162
            logger.debug("Pipe FD: %d",
 
2264
2163
                         self.server.child_pipe.fileno())
 
2266
2165
            session = gnutls.ClientSession(self.request)
 
2268
 
            # priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
2269
 
            #                       "+AES-256-CBC", "+SHA1",
 
2270
 
            #                       "+COMP-NULL", "+CTYPE-OPENPGP",
 
 
2167
            #priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
 
2168
            #                      "+AES-256-CBC", "+SHA1",
 
 
2169
            #                      "+COMP-NULL", "+CTYPE-OPENPGP",
 
2272
2171
            # Use a fallback default, since this MUST be set.
 
2273
2172
            priority = self.server.gnutls_priority
 
2274
2173
            if priority is None:
 
2275
2174
                priority = "NORMAL"
 
2276
 
            gnutls.priority_set_direct(session._c_object,
 
2277
 
                                       priority.encode("utf-8"),
 
 
2175
            gnutls.priority_set_direct(session._c_object, priority,
 
2280
2178
            # Start communication using the Mandos protocol
 
2281
2179
            # Get protocol number
 
2282
2180
            line = self.request.makefile().readline()
 
 
2504
 
class MultiprocessingMixIn:
 
 
2345
class MultiprocessingMixIn(object):
 
2505
2346
    """Like socketserver.ThreadingMixIn, but with multiprocessing"""
 
2507
2348
    def sub_process_main(self, request, address):
 
2509
2350
            self.finish_request(request, address)
 
2510
2351
        except Exception:
 
2511
2352
            self.handle_error(request, address)
 
2512
2353
        self.close_request(request)
 
2514
2355
    def process_request(self, request, address):
 
2515
2356
        """Start a new process to process the request."""
 
2516
 
        proc = multiprocessing.Process(target=self.sub_process_main,
 
2517
 
                                       args=(request, address))
 
 
2357
        proc = multiprocessing.Process(target = self.sub_process_main,
 
 
2358
                                       args = (request, address))
 
2522
 
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
 
 
2363
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
 
2523
2364
    """ adds a pipe to the MixIn """
 
2525
2366
    def process_request(self, request, client_address):
 
2526
2367
        """Overrides and wraps the original process_request().
 
2528
2369
        This function creates a new pipe in self.pipe
 
2530
2371
        parent_pipe, self.child_pipe = multiprocessing.Pipe()
 
2532
2373
        proc = MultiprocessingMixIn.process_request(self, request,
 
2533
2374
                                                    client_address)
 
2534
2375
        self.child_pipe.close()
 
2535
2376
        self.add_pipe(parent_pipe, proc)
 
2537
2378
    def add_pipe(self, parent_pipe, proc):
 
2538
2379
        """Dummy function; override as necessary"""
 
2539
2380
        raise NotImplementedError()
 
2542
2383
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
 
2543
 
                     socketserver.TCPServer):
 
 
2384
                     socketserver.TCPServer, object):
 
2544
2385
    """IPv6-capable TCP server.  Accepts 'None' as address and/or port
 
2547
2388
        enabled:        Boolean; whether this server is activated yet
 
2548
2389
        interface:      None or a network interface name (string)
 
2549
2390
        use_ipv6:       Boolean; to use IPv6 or not
 
2552
2393
    def __init__(self, server_address, RequestHandlerClass,
 
2553
2394
                 interface=None,
 
 
2588
2428
        # socket_wrapper(), if socketfd was set.
 
2589
2429
        socketserver.TCPServer.__init__(self, server_address,
 
2590
2430
                                        RequestHandlerClass)
 
2592
2432
    def server_bind(self):
 
2593
2433
        """This overrides the normal server_bind() function
 
2594
2434
        to bind to an interface if one was specified, and also NOT to
 
2595
2435
        bind to an address or port if they were not specified."""
 
2596
 
        global SO_BINDTODEVICE
 
2597
2436
        if self.interface is not None:
 
2598
2437
            if SO_BINDTODEVICE is None:
 
2599
 
                # Fall back to a hard-coded value which seems to be
 
2601
 
                logger.warning("SO_BINDTODEVICE not found, trying 25")
 
2602
 
                SO_BINDTODEVICE = 25
 
2604
 
                self.socket.setsockopt(
 
2605
 
                    socket.SOL_SOCKET, SO_BINDTODEVICE,
 
2606
 
                    (self.interface + "\0").encode("utf-8"))
 
2607
 
            except socket.error as error:
 
2608
 
                if error.errno == errno.EPERM:
 
2609
 
                    logger.error("No permission to bind to"
 
2610
 
                                 " interface %s", self.interface)
 
2611
 
                elif error.errno == errno.ENOPROTOOPT:
 
2612
 
                    logger.error("SO_BINDTODEVICE not available;"
 
2613
 
                                 " cannot bind to interface %s",
 
2615
 
                elif error.errno == errno.ENODEV:
 
2616
 
                    logger.error("Interface %s does not exist,"
 
2617
 
                                 " cannot bind", self.interface)
 
 
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)
 
2620
2459
        # Only bind(2) the socket if we really need to.
 
2621
2460
        if self.server_address[0] or self.server_address[1]:
 
2622
 
            if self.server_address[1]:
 
2623
 
                self.allow_reuse_address = True
 
2624
2461
            if not self.server_address[0]:
 
2625
2462
                if self.address_family == socket.AF_INET6:
 
2626
 
                    any_address = "::"  # in6addr_any
 
 
2463
                    any_address = "::" # in6addr_any
 
2628
 
                    any_address = "0.0.0.0"  # INADDR_ANY
 
 
2465
                    any_address = "0.0.0.0" # INADDR_ANY
 
2629
2466
                self.server_address = (any_address,
 
2630
2467
                                       self.server_address[1])
 
2631
2468
            elif not self.server_address[1]:
 
 
2665
2502
        self.gnutls_priority = gnutls_priority
 
2666
2503
        IPv6_TCPServer.__init__(self, server_address,
 
2667
2504
                                RequestHandlerClass,
 
2668
 
                                interface=interface,
 
 
2505
                                interface = interface,
 
 
2506
                                use_ipv6 = use_ipv6,
 
 
2507
                                socketfd = socketfd)
 
2672
2509
    def server_activate(self):
 
2673
2510
        if self.enabled:
 
2674
2511
            return socketserver.TCPServer.server_activate(self)
 
2676
2513
    def enable(self):
 
2677
2514
        self.enabled = True
 
2679
2516
    def add_pipe(self, parent_pipe, proc):
 
2680
2517
        # Call "handle_ipc" for both data and EOF events
 
2681
2518
        GLib.io_add_watch(
 
2682
 
            GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
2683
 
            GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
 
 
2519
            parent_pipe.fileno(),
 
 
2520
            GLib.IO_IN | GLib.IO_HUP,
 
2684
2521
            functools.partial(self.handle_ipc,
 
2685
 
                              parent_pipe=parent_pipe,
 
 
2522
                              parent_pipe = parent_pipe,
 
2688
2525
    def handle_ipc(self, source, condition,
 
2689
2526
                   parent_pipe=None,
 
2691
2528
                   client_object=None):
 
2692
2529
        # error, or the other end of multiprocessing.Pipe has closed
 
2693
2530
        if condition & (GLib.IO_ERR | GLib.IO_HUP):
 
2694
2531
            # Wait for other process to exit
 
2698
2535
        # Read a request from the child
 
2699
2536
        request = parent_pipe.recv()
 
2700
2537
        command = request[0]
 
2702
2539
        if command == 'init':
 
2703
 
            key_id = request[1].decode("ascii")
 
2704
 
            fpr = request[2].decode("ascii")
 
2705
 
            address = request[3]
 
 
2541
            address = request[2]
 
2707
2543
            for c in self.clients.values():
 
2708
 
                if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
 
2710
 
                if key_id and c.key_id == key_id:
 
2713
 
                if fpr and c.fingerprint == fpr:
 
 
2544
                if c.fingerprint == fpr:
 
2717
 
                logger.info("Client not found for key ID: %s, address"
 
2718
 
                            ": %s", key_id or fpr, address)
 
 
2548
                logger.info("Client not found for fingerprint: %s, ad"
 
 
2549
                            "dress: %s", fpr, address)
 
2719
2550
                if self.use_dbus:
 
2720
2551
                    # Emit D-Bus signal
 
2721
 
                    mandos_dbus_service.ClientNotFound(key_id or fpr,
 
 
2552
                    mandos_dbus_service.ClientNotFound(fpr,
 
2723
2554
                parent_pipe.send(False)
 
2726
2557
            GLib.io_add_watch(
 
2727
 
                GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
2728
 
                GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
 
 
2558
                parent_pipe.fileno(),
 
 
2559
                GLib.IO_IN | GLib.IO_HUP,
 
2729
2560
                functools.partial(self.handle_ipc,
 
2730
 
                                  parent_pipe=parent_pipe,
 
2732
 
                                  client_object=client))
 
 
2561
                                  parent_pipe = parent_pipe,
 
 
2563
                                  client_object = client))
 
2733
2564
            parent_pipe.send(True)
 
2734
2565
            # remove the old hook in favor of the new above hook on
 
 
2752
2583
                parent_pipe.send((
 
2753
2584
                    'data', client_object.__getattribute__(attrname)))
 
2755
2586
        if command == 'setattr':
 
2756
2587
            attrname = request[1]
 
2757
2588
            value = request[2]
 
2758
2589
            setattr(client_object, attrname, value)
 
2763
2594
def rfc3339_duration_to_delta(duration):
 
2764
2595
    """Parse an RFC 3339 "duration" and return a datetime.timedelta
 
2766
 
    >>> rfc3339_duration_to_delta("P7D") == datetime.timedelta(7)
 
2768
 
    >>> rfc3339_duration_to_delta("PT60S") == datetime.timedelta(0, 60)
 
2770
 
    >>> rfc3339_duration_to_delta("PT60M") == datetime.timedelta(0, 3600)
 
2772
 
    >>> rfc3339_duration_to_delta("PT24H") == datetime.timedelta(1)
 
2774
 
    >>> rfc3339_duration_to_delta("P1W") == datetime.timedelta(7)
 
2776
 
    >>> rfc3339_duration_to_delta("PT5M30S") == datetime.timedelta(0, 330)
 
2778
 
    >>> rfc3339_duration_to_delta("P1DT3M20S") == datetime.timedelta(1, 200)
 
 
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)
 
2782
2613
    # Parsing an RFC 3339 duration with regular expressions is not
 
2783
2614
    # possible - there would have to be multiple places for the same
 
2784
2615
    # values, like seconds.  The current code, while more esoteric, is
 
2785
2616
    # cleaner without depending on a parsing library.  If Python had a
 
2786
2617
    # built-in library for parsing we would use it, but we'd like to
 
2787
2618
    # avoid excessive use of external libraries.
 
2789
2620
    # New type for defining tokens, syntax, and semantics all-in-one
 
2790
2621
    Token = collections.namedtuple("Token", (
 
2791
2622
        "regexp",  # To match token; if "value" is not None, must have
 
 
2978
2806
    parser.add_argument("--no-zeroconf", action="store_false",
 
2979
2807
                        dest="zeroconf", help="Do not use Zeroconf",
 
2982
2810
    options = parser.parse_args()
 
 
2814
        fail_count, test_count = doctest.testmod()
 
 
2815
        sys.exit(os.EX_OK if fail_count == 0 else 1)
 
2984
2817
    # Default values for config file for server-global settings
 
2985
 
    if gnutls.has_rawpk:
 
2986
 
        priority = ("SECURE128:!CTYPE-X.509:+CTYPE-RAWPK:!RSA"
 
2987
 
                    ":!VERS-ALL:+VERS-TLS1.3:%PROFILE_ULTRA")
 
2989
 
        priority = ("SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
 
2990
 
                    ":+SIGN-DSA-SHA256")
 
2991
 
    server_defaults = {"interface": "",
 
2995
 
                       "priority": priority,
 
2996
 
                       "servicename": "Mandos",
 
3002
 
                       "statedir": "/var/lib/mandos",
 
3003
 
                       "foreground": "False",
 
 
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",
 
3008
2836
    # Parse config file for server-global settings
 
3009
 
    server_config = configparser.ConfigParser(server_defaults)
 
 
2837
    server_config = configparser.SafeConfigParser(server_defaults)
 
3010
2838
    del server_defaults
 
3011
2839
    server_config.read(os.path.join(options.configdir, "mandos.conf"))
 
3012
 
    # Convert the ConfigParser object to a dict
 
 
2840
    # Convert the SafeConfigParser object to a dict
 
3013
2841
    server_settings = server_config.defaults()
 
3014
2842
    # Use the appropriate methods on the non-string config options
 
3015
 
    for option in ("debug", "use_dbus", "use_ipv6", "restore",
 
3016
 
                   "foreground", "zeroconf"):
 
 
2843
    for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
 
3017
2844
        server_settings[option] = server_config.getboolean("DEFAULT",
 
3019
2846
    if server_settings["port"]:
 
 
3192
3019
        protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
 
3193
3020
        service = AvahiServiceToSyslog(
 
3194
 
            name=server_settings["servicename"],
 
3195
 
            servicetype="_mandos._tcp",
 
 
3021
            name = server_settings["servicename"],
 
 
3022
            servicetype = "_mandos._tcp",
 
 
3023
            protocol = protocol,
 
3198
3025
        if server_settings["interface"]:
 
3199
3026
            service.interface = if_nametoindex(
 
3200
3027
                server_settings["interface"].encode("utf-8"))
 
3202
3029
    global multiprocessing_manager
 
3203
3030
    multiprocessing_manager = multiprocessing.Manager()
 
3205
3032
    client_class = Client
 
3207
 
        client_class = functools.partial(ClientDBus, bus=bus)
 
 
3034
        client_class = functools.partial(ClientDBus, bus = bus)
 
3209
3036
    client_settings = Client.config_parser(client_config)
 
3210
3037
    old_client_settings = {}
 
3211
3038
    clients_data = {}
 
3213
3040
    # This is used to redirect stdout and stderr for checker processes
 
3215
 
    wnull = open(os.devnull, "w")  # A writable /dev/null
 
 
3042
    wnull = open(os.devnull, "w") # A writable /dev/null
 
3216
3043
    # Only used if server is running in foreground but not in debug
 
3218
3045
    if debug or not foreground:
 
3221
3048
    # Get client data and settings from last running state.
 
3222
3049
    if server_settings["restore"]:
 
3224
3051
            with open(stored_state_path, "rb") as stored_state:
 
3225
 
                if sys.version_info.major == 2:
 
 
3052
                if sys.version_info.major == 2:                
 
3226
3053
                    clients_data, old_client_settings = pickle.load(
 
3229
3056
                    bytes_clients_data, bytes_old_client_settings = (
 
3230
 
                        pickle.load(stored_state, encoding="bytes"))
 
3231
 
                    #   Fix bytes to strings
 
 
3057
                        pickle.load(stored_state, encoding = "bytes"))
 
 
3058
                    ### Fix bytes to strings
 
3234
 
                    clients_data = {(key.decode("utf-8")
 
3235
 
                                     if isinstance(key, bytes)
 
3238
 
                                    bytes_clients_data.items()}
 
 
3061
                    clients_data = { (key.decode("utf-8")
 
 
3062
                                      if isinstance(key, bytes)
 
 
3065
                                     bytes_clients_data.items() }
 
3239
3066
                    del bytes_clients_data
 
3240
3067
                    for key in clients_data:
 
3241
 
                        value = {(k.decode("utf-8")
 
3242
 
                                  if isinstance(k, bytes) else k): v
 
3244
 
                                 clients_data[key].items()}
 
 
3068
                        value = { (k.decode("utf-8")
 
 
3069
                                   if isinstance(k, bytes) else k): v
 
 
3071
                                  clients_data[key].items() }
 
3245
3072
                        clients_data[key] = value
 
3246
3073
                        # .client_structure
 
3247
3074
                        value["client_structure"] = [
 
3248
3075
                            (s.decode("utf-8")
 
3249
3076
                             if isinstance(s, bytes)
 
3250
3077
                             else s) for s in
 
3251
 
                            value["client_structure"]]
 
3252
 
                        # .name, .host, and .checker_command
 
3253
 
                        for k in ("name", "host", "checker_command"):
 
 
3078
                            value["client_structure"] ]
 
 
3080
                        for k in ("name", "host"):
 
3254
3081
                            if isinstance(value[k], bytes):
 
3255
3082
                                value[k] = value[k].decode("utf-8")
 
3256
 
                        if "key_id" not in value:
 
3257
 
                            value["key_id"] = ""
 
3258
 
                        elif "fingerprint" not in value:
 
3259
 
                            value["fingerprint"] = ""
 
3260
 
                    #  old_client_settings
 
 
3083
                    ## old_client_settings
 
3262
3085
                    old_client_settings = {
 
3263
3086
                        (key.decode("utf-8")
 
3264
3087
                         if isinstance(key, bytes)
 
3265
3088
                         else key): value
 
3266
3089
                        for key, value in
 
3267
 
                        bytes_old_client_settings.items()}
 
 
3090
                        bytes_old_client_settings.items() }
 
3268
3091
                    del bytes_old_client_settings
 
3269
 
                    # .host and .checker_command
 
3270
3093
                    for value in old_client_settings.values():
 
3271
 
                        for attribute in ("host", "checker_command"):
 
3272
 
                            if isinstance(value[attribute], bytes):
 
3273
 
                                value[attribute] = (value[attribute]
 
 
3094
                        if isinstance(value["host"], bytes):
 
 
3095
                            value["host"] = (value["host"]
 
3275
3097
            os.remove(stored_state_path)
 
3276
3098
        except IOError as e:
 
3277
3099
            if e.errno == errno.ENOENT: