216
265
                       .replace(b"\n", b"\\n")
 
217
266
                       .replace(b"\0", b"\\x00"))
 
220
269
    def encrypt(self, data, password):
 
221
270
        passphrase = self.password_encode(password)
 
222
271
        with tempfile.NamedTemporaryFile(
 
223
272
                dir=self.tempdir) as passfile:
 
224
273
            passfile.write(passphrase)
 
226
 
            proc = subprocess.Popen([self.gpg, '--symmetric',
 
 
275
            proc = subprocess.Popen([self.gpg, "--symmetric",
 
229
278
                                    + self.gnupgargs,
 
230
 
                                    stdin = subprocess.PIPE,
 
231
 
                                    stdout = subprocess.PIPE,
 
232
 
                                    stderr = subprocess.PIPE)
 
233
 
            ciphertext, err = proc.communicate(input = data)
 
 
279
                                    stdin=subprocess.PIPE,
 
 
280
                                    stdout=subprocess.PIPE,
 
 
281
                                    stderr=subprocess.PIPE)
 
 
282
            ciphertext, err = proc.communicate(input=data)
 
234
283
        if proc.returncode != 0:
 
235
284
            raise PGPError(err)
 
236
285
        return ciphertext
 
238
287
    def decrypt(self, data, password):
 
239
288
        passphrase = self.password_encode(password)
 
240
289
        with tempfile.NamedTemporaryFile(
 
241
 
                dir = self.tempdir) as passfile:
 
 
290
                dir=self.tempdir) as passfile:
 
242
291
            passfile.write(passphrase)
 
244
 
            proc = subprocess.Popen([self.gpg, '--decrypt',
 
 
293
            proc = subprocess.Popen([self.gpg, "--decrypt",
 
247
296
                                    + self.gnupgargs,
 
248
 
                                    stdin = subprocess.PIPE,
 
249
 
                                    stdout = subprocess.PIPE,
 
250
 
                                    stderr = subprocess.PIPE)
 
251
 
            decrypted_plaintext, err = proc.communicate(input = data)
 
 
297
                                    stdin=subprocess.PIPE,
 
 
298
                                    stdout=subprocess.PIPE,
 
 
299
                                    stderr=subprocess.PIPE)
 
 
300
            decrypted_plaintext, err = proc.communicate(input=data)
 
252
301
        if proc.returncode != 0:
 
253
302
            raise PGPError(err)
 
254
303
        return decrypted_plaintext
 
256
306
# 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
 
 
308
    """This isn't so much a class as it is a module-like namespace."""
 
 
309
    IF_UNSPEC = -1               # avahi-common/address.h
 
 
310
    PROTO_UNSPEC = -1            # avahi-common/address.h
 
 
311
    PROTO_INET = 0               # avahi-common/address.h
 
 
312
    PROTO_INET6 = 1              # avahi-common/address.h
 
264
313
    DBUS_NAME = "org.freedesktop.Avahi"
 
265
314
    DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
 
266
315
    DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
 
267
316
    DBUS_PATH_SERVER = "/"
 
268
 
    def string_array_to_txt_array(self, t):
 
 
319
    def string_array_to_txt_array(t):
 
269
320
        return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
 
270
321
                           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
 
 
322
    ENTRY_GROUP_ESTABLISHED = 2  # avahi-common/defs.h
 
 
323
    ENTRY_GROUP_COLLISION = 3    # avahi-common/defs.h
 
 
324
    ENTRY_GROUP_FAILURE = 4      # avahi-common/defs.h
 
 
325
    SERVER_INVALID = 0           # avahi-common/defs.h
 
 
326
    SERVER_REGISTERING = 1       # avahi-common/defs.h
 
 
327
    SERVER_RUNNING = 2           # avahi-common/defs.h
 
 
328
    SERVER_COLLISION = 3         # avahi-common/defs.h
 
 
329
    SERVER_FAILURE = 4           # avahi-common/defs.h
 
281
332
class AvahiError(Exception):
 
282
333
    def __init__(self, value, *args, **kwargs):
 
 
473
524
class AvahiServiceToSyslog(AvahiService):
 
474
525
    def rename(self, *args, **kwargs):
 
475
526
        """Add the new name to the syslog messages"""
 
476
 
        ret = AvahiService.rename(self, *args, **kwargs)
 
 
527
        ret = super(AvahiServiceToSyslog, self).rename(*args,
 
477
529
        syslogger.setFormatter(logging.Formatter(
 
478
 
            'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
 
 
530
            "Mandos ({}) [%(process)d]: %(levelname)s: %(message)s"
 
479
531
            .format(self.name)))
 
482
535
# 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))
 
 
537
    """This isn't so much a class as it is a module-like namespace."""
 
 
539
    library = ctypes.util.find_library("gnutls")
 
 
541
        library = ctypes.util.find_library("gnutls-deb0")
 
 
542
    _library = ctypes.cdll.LoadLibrary(library)
 
498
545
    # Unless otherwise indicated, the constants and types below are
 
499
546
    # all from the gnutls/gnutls.h C header file.
 
503
550
    E_INTERRUPTED = -52
 
508
556
    CRD_CERTIFICATE = 1
 
509
557
    E_NO_CERTIFICATE_FOUND = -49
 
 
562
    KEYID_USE_SHA256 = 1        # gnutls/x509.h
 
510
563
    OPENPGP_FMT_RAW = 0         # gnutls/openpgp.h
 
513
 
    class session_int(ctypes.Structure):
 
 
566
    class _session_int(ctypes.Structure):
 
515
 
    session_t = ctypes.POINTER(session_int)
 
 
568
    session_t = ctypes.POINTER(_session_int)
 
516
570
    class certificate_credentials_st(ctypes.Structure):
 
518
572
    certificate_credentials_t = ctypes.POINTER(
 
519
573
        certificate_credentials_st)
 
520
574
    certificate_type_t = ctypes.c_int
 
521
576
    class datum_t(ctypes.Structure):
 
522
 
        _fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
 
523
 
                    ('size', ctypes.c_uint)]
 
524
 
    class openpgp_crt_int(ctypes.Structure):
 
 
577
        _fields_ = [("data", ctypes.POINTER(ctypes.c_ubyte)),
 
 
578
                    ("size", ctypes.c_uint)]
 
 
580
    class _openpgp_crt_int(ctypes.Structure):
 
526
 
    openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
 
527
 
    openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
 
 
582
    openpgp_crt_t = ctypes.POINTER(_openpgp_crt_int)
 
 
583
    openpgp_crt_fmt_t = ctypes.c_int  # gnutls/openpgp.h
 
528
584
    log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
 
529
585
    credentials_type_t = ctypes.c_int
 
530
586
    transport_ptr_t = ctypes.c_void_p
 
531
587
    close_request_t = ctypes.c_int
 
534
590
    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=()):
 
 
591
        def __init__(self, message=None, code=None, args=()):
 
540
592
            # Default usage is by a message string, but if a return
 
541
593
            # code is passed, convert it to a string with
 
542
594
            # gnutls.strerror()
 
544
596
            if message is None and code is not None:
 
545
 
                message = GnuTLS.strerror(code)
 
546
 
            return super(GnuTLS.Error, self).__init__(
 
 
597
                message = gnutls.strerror(code).decode(
 
 
598
                    "utf-8", errors="replace")
 
 
599
            return super(gnutls.Error, self).__init__(
 
549
602
    class CertificateSecurityError(Error):
 
 
606
        def __init__(self, cls):
 
 
609
        def from_param(self, obj):
 
 
610
            if not isinstance(obj, self.cls):
 
 
611
                raise TypeError("Not of type {}: {!r}"
 
 
612
                                .format(self.cls.__name__, obj))
 
 
613
            return ctypes.byref(obj.from_param(obj))
 
 
615
    class CastToVoidPointer:
 
 
616
        def __init__(self, cls):
 
 
619
        def from_param(self, obj):
 
 
620
            if not isinstance(obj, self.cls):
 
 
621
                raise TypeError("Not of type {}: {!r}"
 
 
622
                                .format(self.cls.__name__, obj))
 
 
623
            return ctypes.cast(obj.from_param(obj), ctypes.c_void_p)
 
 
625
    class With_from_param:
 
 
627
        def from_param(cls, obj):
 
 
628
            return obj._as_parameter_
 
553
 
    class Credentials(object):
 
 
631
    class Credentials(With_from_param):
 
554
632
        def __init__(self):
 
555
 
            self._c_object = gnutls.certificate_credentials_t()
 
556
 
            gnutls.certificate_allocate_credentials(
 
557
 
                ctypes.byref(self._c_object))
 
 
633
            self._as_parameter_ = gnutls.certificate_credentials_t()
 
 
634
            gnutls.certificate_allocate_credentials(self)
 
558
635
            self.type = gnutls.CRD_CERTIFICATE
 
560
637
        def __del__(self):
 
561
 
            gnutls.certificate_free_credentials(self._c_object)
 
563
 
    class ClientSession(object):
 
564
 
        def __init__(self, socket, credentials = None):
 
565
 
            self._c_object = gnutls.session_t()
 
566
 
            gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)
 
567
 
            gnutls.set_default_priority(self._c_object)
 
568
 
            gnutls.transport_set_ptr(self._c_object, socket.fileno())
 
569
 
            gnutls.handshake_set_private_extensions(self._c_object,
 
 
638
            gnutls.certificate_free_credentials(self)
 
 
640
    class ClientSession(With_from_param):
 
 
641
        def __init__(self, socket, credentials=None):
 
 
642
            self._as_parameter_ = gnutls.session_t()
 
 
643
            gnutls_flags = gnutls.CLIENT
 
 
644
            if gnutls.check_version(b"3.5.6"):
 
 
645
                gnutls_flags |= gnutls.NO_TICKETS
 
 
647
                gnutls_flags |= gnutls.ENABLE_RAWPK
 
 
648
            gnutls.init(self, gnutls_flags)
 
 
650
            gnutls.set_default_priority(self)
 
 
651
            gnutls.transport_set_ptr(self, socket.fileno())
 
 
652
            gnutls.handshake_set_private_extensions(self, True)
 
571
653
            self.socket = socket
 
572
654
            if credentials is None:
 
573
655
                credentials = gnutls.Credentials()
 
574
 
            gnutls.credentials_set(self._c_object, credentials.type,
 
575
 
                                   ctypes.cast(credentials._c_object,
 
 
656
            gnutls.credentials_set(self, credentials.type,
 
577
658
            self.credentials = credentials
 
579
660
        def __del__(self):
 
580
 
            gnutls.deinit(self._c_object)
 
582
663
        def handshake(self):
 
583
 
            return gnutls.handshake(self._c_object)
 
 
664
            return gnutls.handshake(self)
 
585
666
        def send(self, data):
 
586
667
            data = bytes(data)
 
587
668
            data_len = len(data)
 
588
669
            while data_len > 0:
 
589
 
                data_len -= gnutls.record_send(self._c_object,
 
 
670
                data_len -= gnutls.record_send(self, data[-data_len:],
 
594
 
            return gnutls.bye(self._c_object, gnutls.SHUT_RDWR)
 
 
674
            return gnutls.bye(self, gnutls.SHUT_RDWR)
 
596
676
    # Error handling functions
 
597
677
    def _error_code(result):
 
598
678
        """A function to raise exceptions on errors, suitable
 
599
 
        for the 'restype' attribute on ctypes functions"""
 
 
679
        for the "restype" attribute on ctypes functions"""
 
 
680
        if result >= gnutls.E_SUCCESS:
 
602
682
        if result == gnutls.E_NO_CERTIFICATE_FOUND:
 
603
 
            raise gnutls.CertificateSecurityError(code = result)
 
604
 
        raise gnutls.Error(code = result)
 
606
 
    def _retry_on_error(result, func, arguments):
 
 
683
            raise gnutls.CertificateSecurityError(code=result)
 
 
684
        raise gnutls.Error(code=result)
 
 
686
    def _retry_on_error(result, func, arguments,
 
 
687
                        _error_code=_error_code):
 
607
688
        """A function to retry on some errors, suitable
 
608
 
        for the 'errcheck' attribute on ctypes functions"""
 
 
689
        for the "errcheck" attribute on ctypes functions"""
 
 
690
        while result < gnutls.E_SUCCESS:
 
610
691
            if result not in (gnutls.E_INTERRUPTED, gnutls.E_AGAIN):
 
611
692
                return _error_code(result)
 
612
693
            result = func(*arguments)
 
615
696
    # Unless otherwise indicated, the function declarations below are
 
616
697
    # all from the gnutls/gnutls.h C header file.
 
619
700
    priority_set_direct = _library.gnutls_priority_set_direct
 
620
 
    priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
 
 
701
    priority_set_direct.argtypes = [ClientSession, ctypes.c_char_p,
 
621
702
                                    ctypes.POINTER(ctypes.c_char_p)]
 
622
703
    priority_set_direct.restype = _error_code
 
624
705
    init = _library.gnutls_init
 
625
 
    init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
 
 
706
    init.argtypes = [PointerTo(ClientSession), ctypes.c_int]
 
626
707
    init.restype = _error_code
 
628
709
    set_default_priority = _library.gnutls_set_default_priority
 
629
 
    set_default_priority.argtypes = [session_t]
 
 
710
    set_default_priority.argtypes = [ClientSession]
 
630
711
    set_default_priority.restype = _error_code
 
632
713
    record_send = _library.gnutls_record_send
 
633
 
    record_send.argtypes = [session_t, ctypes.c_void_p,
 
 
714
    record_send.argtypes = [ClientSession, ctypes.c_void_p,
 
635
716
    record_send.restype = ctypes.c_ssize_t
 
636
717
    record_send.errcheck = _retry_on_error
 
638
719
    certificate_allocate_credentials = (
 
639
720
        _library.gnutls_certificate_allocate_credentials)
 
640
721
    certificate_allocate_credentials.argtypes = [
 
641
 
        ctypes.POINTER(certificate_credentials_t)]
 
 
722
        PointerTo(Credentials)]
 
642
723
    certificate_allocate_credentials.restype = _error_code
 
644
725
    certificate_free_credentials = (
 
645
726
        _library.gnutls_certificate_free_credentials)
 
646
 
    certificate_free_credentials.argtypes = [certificate_credentials_t]
 
 
727
    certificate_free_credentials.argtypes = [Credentials]
 
647
728
    certificate_free_credentials.restype = None
 
649
730
    handshake_set_private_extensions = (
 
650
731
        _library.gnutls_handshake_set_private_extensions)
 
651
 
    handshake_set_private_extensions.argtypes = [session_t,
 
 
732
    handshake_set_private_extensions.argtypes = [ClientSession,
 
653
734
    handshake_set_private_extensions.restype = None
 
655
736
    credentials_set = _library.gnutls_credentials_set
 
656
 
    credentials_set.argtypes = [session_t, credentials_type_t,
 
 
737
    credentials_set.argtypes = [ClientSession, credentials_type_t,
 
 
738
                                CastToVoidPointer(Credentials)]
 
658
739
    credentials_set.restype = _error_code
 
660
741
    strerror = _library.gnutls_strerror
 
661
742
    strerror.argtypes = [ctypes.c_int]
 
662
743
    strerror.restype = ctypes.c_char_p
 
664
745
    certificate_type_get = _library.gnutls_certificate_type_get
 
665
 
    certificate_type_get.argtypes = [session_t]
 
 
746
    certificate_type_get.argtypes = [ClientSession]
 
666
747
    certificate_type_get.restype = _error_code
 
668
749
    certificate_get_peers = _library.gnutls_certificate_get_peers
 
669
 
    certificate_get_peers.argtypes = [session_t,
 
 
750
    certificate_get_peers.argtypes = [ClientSession,
 
670
751
                                      ctypes.POINTER(ctypes.c_uint)]
 
671
752
    certificate_get_peers.restype = ctypes.POINTER(datum_t)
 
673
754
    global_set_log_level = _library.gnutls_global_set_log_level
 
674
755
    global_set_log_level.argtypes = [ctypes.c_int]
 
675
756
    global_set_log_level.restype = None
 
677
758
    global_set_log_function = _library.gnutls_global_set_log_function
 
678
759
    global_set_log_function.argtypes = [log_func]
 
679
760
    global_set_log_function.restype = None
 
681
762
    deinit = _library.gnutls_deinit
 
682
 
    deinit.argtypes = [session_t]
 
 
763
    deinit.argtypes = [ClientSession]
 
683
764
    deinit.restype = None
 
685
766
    handshake = _library.gnutls_handshake
 
686
 
    handshake.argtypes = [session_t]
 
687
 
    handshake.restype = _error_code
 
 
767
    handshake.argtypes = [ClientSession]
 
 
768
    handshake.restype = ctypes.c_int
 
688
769
    handshake.errcheck = _retry_on_error
 
690
771
    transport_set_ptr = _library.gnutls_transport_set_ptr
 
691
 
    transport_set_ptr.argtypes = [session_t, transport_ptr_t]
 
 
772
    transport_set_ptr.argtypes = [ClientSession, transport_ptr_t]
 
692
773
    transport_set_ptr.restype = None
 
694
775
    bye = _library.gnutls_bye
 
695
 
    bye.argtypes = [session_t, close_request_t]
 
696
 
    bye.restype = _error_code
 
 
776
    bye.argtypes = [ClientSession, close_request_t]
 
 
777
    bye.restype = ctypes.c_int
 
697
778
    bye.errcheck = _retry_on_error
 
699
780
    check_version = _library.gnutls_check_version
 
700
781
    check_version.argtypes = [ctypes.c_char_p]
 
701
782
    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
 
 
784
    _need_version = b"3.3.0"
 
 
785
    if check_version(_need_version) is None:
 
 
786
        raise self.Error("Needs GnuTLS {} or later"
 
 
787
                         .format(_need_version))
 
 
789
    _tls_rawpk_version = b"3.6.6"
 
 
790
    has_rawpk = bool(check_version(_tls_rawpk_version))
 
 
794
        class pubkey_st(ctypes.Structure):
 
 
796
        pubkey_t = ctypes.POINTER(pubkey_st)
 
 
798
        x509_crt_fmt_t = ctypes.c_int
 
 
800
        # All the function declarations below are from
 
 
802
        pubkey_init = _library.gnutls_pubkey_init
 
 
803
        pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
 
 
804
        pubkey_init.restype = _error_code
 
 
806
        pubkey_import = _library.gnutls_pubkey_import
 
 
807
        pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
 
 
809
        pubkey_import.restype = _error_code
 
 
811
        pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
 
 
812
        pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
 
 
813
                                      ctypes.POINTER(ctypes.c_ubyte),
 
 
814
                                      ctypes.POINTER(ctypes.c_size_t)]
 
 
815
        pubkey_get_key_id.restype = _error_code
 
 
817
        pubkey_deinit = _library.gnutls_pubkey_deinit
 
 
818
        pubkey_deinit.argtypes = [pubkey_t]
 
 
819
        pubkey_deinit.restype = None
 
 
821
        # All the function declarations below are from
 
 
824
        openpgp_crt_init = _library.gnutls_openpgp_crt_init
 
 
825
        openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
 
 
826
        openpgp_crt_init.restype = _error_code
 
 
828
        openpgp_crt_import = _library.gnutls_openpgp_crt_import
 
 
829
        openpgp_crt_import.argtypes = [openpgp_crt_t,
 
 
830
                                       ctypes.POINTER(datum_t),
 
 
832
        openpgp_crt_import.restype = _error_code
 
 
834
        openpgp_crt_verify_self = \
 
 
835
            _library.gnutls_openpgp_crt_verify_self
 
 
836
        openpgp_crt_verify_self.argtypes = [
 
 
839
            ctypes.POINTER(ctypes.c_uint),
 
 
841
        openpgp_crt_verify_self.restype = _error_code
 
 
843
        openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
 
 
844
        openpgp_crt_deinit.argtypes = [openpgp_crt_t]
 
 
845
        openpgp_crt_deinit.restype = None
 
 
847
        openpgp_crt_get_fingerprint = (
 
 
848
            _library.gnutls_openpgp_crt_get_fingerprint)
 
 
849
        openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
 
 
853
        openpgp_crt_get_fingerprint.restype = _error_code
 
 
855
    if check_version(b"3.6.4"):
 
 
856
        certificate_type_get2 = _library.gnutls_certificate_type_get2
 
 
857
        certificate_type_get2.argtypes = [ClientSession, ctypes.c_int]
 
 
858
        certificate_type_get2.restype = _error_code
 
732
860
    # Remove non-public functions
 
733
861
    del _error_code, _retry_on_error
 
734
 
# Create the global "gnutls" object, simulating a module
 
737
864
def call_pipe(connection,       # : multiprocessing.Connection
 
738
865
              func, *args, **kwargs):
 
739
866
    """This function is meant to be called by multiprocessing.Process
 
741
868
    This function runs func(*args, **kwargs), and writes the resulting
 
742
869
    return value on the provided multiprocessing.Connection.
 
744
871
    connection.send(func(*args, **kwargs))
 
745
872
    connection.close()
 
747
 
class Client(object):
 
748
876
    """A representation of a client host served by this server.
 
751
 
    approved:   bool(); 'None' if not yet approved/disapproved
 
 
879
    approved:   bool(); None if not yet approved/disapproved
 
752
880
    approval_delay: datetime.timedelta(); Time to wait for approval
 
753
881
    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.
 
 
882
    checker: multiprocessing.Process(); a running checker process used
 
 
883
             to see if the client lives. None if no process is
 
757
885
    checker_callback_tag: a GLib event source tag, or None
 
758
886
    checker_command: string; External command which is run to check
 
759
887
                     if client lives.  %() expansions are done at
 
 
2117
2265
                           byte_arrays=True)
 
2118
2266
    def Secret_dbus_property(self, value):
 
2119
2267
        self.secret = bytes(value)
 
2124
 
class ProxyClient(object):
 
2125
 
    def __init__(self, child_pipe, fpr, address):
 
 
2273
    def __init__(self, child_pipe, key_id, fpr, address):
 
2126
2274
        self._pipe = child_pipe
 
2127
 
        self._pipe.send(('init', fpr, address))
 
 
2275
        self._pipe.send(("init", key_id, fpr, address))
 
2128
2276
        if not self._pipe.recv():
 
 
2277
            raise KeyError(key_id or fpr)
 
2131
2279
    def __getattribute__(self, name):
 
2133
2281
            return super(ProxyClient, self).__getattribute__(name)
 
2134
 
        self._pipe.send(('getattr', name))
 
 
2282
        self._pipe.send(("getattr", name))
 
2135
2283
        data = self._pipe.recv()
 
2136
 
        if data[0] == 'data':
 
 
2284
        if data[0] == "data":
 
2138
 
        if data[0] == 'function':
 
 
2286
        if data[0] == "function":
 
2140
2288
            def func(*args, **kwargs):
 
2141
 
                self._pipe.send(('funcall', name, args, kwargs))
 
 
2289
                self._pipe.send(("funcall", name, args, kwargs))
 
2142
2290
                return self._pipe.recv()[1]
 
2146
2294
    def __setattr__(self, name, value):
 
2148
2296
            return super(ProxyClient, self).__setattr__(name, value)
 
2149
 
        self._pipe.send(('setattr', name, value))
 
 
2297
        self._pipe.send(("setattr", name, value))
 
2152
2300
class ClientHandler(socketserver.BaseRequestHandler, object):
 
2153
2301
    """A class to handle client connections.
 
2155
2303
    Instantiated once for each connection to handle it.
 
2156
2304
    Note: This will run in its own forked process."""
 
2158
2306
    def handle(self):
 
2159
2307
        with contextlib.closing(self.server.child_pipe) as child_pipe:
 
2160
2308
            logger.info("TCP connection from: %s",
 
2161
2309
                        str(self.client_address))
 
2162
2310
            logger.debug("Pipe FD: %d",
 
2163
2311
                         self.server.child_pipe.fileno())
 
2165
2313
            session = gnutls.ClientSession(self.request)
 
2167
 
            #priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
2168
 
            #                      "+AES-256-CBC", "+SHA1",
 
2169
 
            #                      "+COMP-NULL", "+CTYPE-OPENPGP",
 
 
2315
            # priority = ":".join(("NONE", "+VERS-TLS1.1",
 
 
2316
            #                       "+AES-256-CBC", "+SHA1",
 
 
2317
            #                       "+COMP-NULL", "+CTYPE-OPENPGP",
 
2171
2319
            # Use a fallback default, since this MUST be set.
 
2172
2320
            priority = self.server.gnutls_priority
 
2173
2321
            if priority is None:
 
2174
2322
                priority = "NORMAL"
 
2175
 
            gnutls.priority_set_direct(session._c_object, priority,
 
 
2323
            gnutls.priority_set_direct(session,
 
 
2324
                                       priority.encode("utf-8"), None)
 
2178
2326
            # Start communication using the Mandos protocol
 
2179
2327
            # Get protocol number
 
2180
2328
            line = self.request.makefile().readline()
 
 
2502
2714
        self.gnutls_priority = gnutls_priority
 
2503
2715
        IPv6_TCPServer.__init__(self, server_address,
 
2504
2716
                                RequestHandlerClass,
 
2505
 
                                interface = interface,
 
2506
 
                                use_ipv6 = use_ipv6,
 
2507
 
                                socketfd = socketfd)
 
 
2717
                                interface=interface,
 
2509
2721
    def server_activate(self):
 
2510
2722
        if self.enabled:
 
2511
2723
            return socketserver.TCPServer.server_activate(self)
 
2513
2725
    def enable(self):
 
2514
2726
        self.enabled = True
 
2516
2728
    def add_pipe(self, parent_pipe, proc):
 
2517
2729
        # Call "handle_ipc" for both data and EOF events
 
2518
2730
        GLib.io_add_watch(
 
2519
 
            parent_pipe.fileno(),
 
2520
 
            GLib.IO_IN | GLib.IO_HUP,
 
 
2731
            GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
 
2732
            GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
 
2521
2733
            functools.partial(self.handle_ipc,
 
2522
 
                              parent_pipe = parent_pipe,
 
 
2734
                              parent_pipe=parent_pipe,
 
2525
2737
    def handle_ipc(self, source, condition,
 
2526
2738
                   parent_pipe=None,
 
2528
2740
                   client_object=None):
 
2529
2741
        # error, or the other end of multiprocessing.Pipe has closed
 
2530
2742
        if condition & (GLib.IO_ERR | GLib.IO_HUP):
 
2531
2743
            # Wait for other process to exit
 
2535
2747
        # Read a request from the child
 
2536
2748
        request = parent_pipe.recv()
 
2537
2749
        command = request[0]
 
2539
 
        if command == 'init':
 
2541
 
            address = request[2]
 
 
2751
        if command == "init":
 
 
2752
            key_id = request[1].decode("ascii")
 
 
2753
            fpr = request[2].decode("ascii")
 
 
2754
            address = request[3]
 
2543
2756
            for c in self.clients.values():
 
2544
 
                if c.fingerprint == fpr:
 
 
2757
                if key_id == ("E3B0C44298FC1C149AFBF4C8996FB924"
 
 
2758
                              "27AE41E4649B934CA495991B7852B855"):
 
 
2760
                if key_id and c.key_id == key_id:
 
 
2763
                if fpr and c.fingerprint == fpr:
 
2548
 
                logger.info("Client not found for fingerprint: %s, ad"
 
2549
 
                            "dress: %s", fpr, address)
 
 
2767
                logger.info("Client not found for key ID: %s, address"
 
 
2768
                            ": %s", key_id or fpr, address)
 
2550
2769
                if self.use_dbus:
 
2551
2770
                    # Emit D-Bus signal
 
2552
 
                    mandos_dbus_service.ClientNotFound(fpr,
 
 
2771
                    mandos_dbus_service.ClientNotFound(key_id or fpr,
 
2554
2773
                parent_pipe.send(False)
 
2557
2776
            GLib.io_add_watch(
 
2558
 
                parent_pipe.fileno(),
 
2559
 
                GLib.IO_IN | GLib.IO_HUP,
 
 
2777
                GLib.IOChannel.unix_new(parent_pipe.fileno()),
 
 
2778
                GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
 
2560
2779
                functools.partial(self.handle_ipc,
 
2561
 
                                  parent_pipe = parent_pipe,
 
2563
 
                                  client_object = client))
 
 
2780
                                  parent_pipe=parent_pipe,
 
 
2782
                                  client_object=client))
 
2564
2783
            parent_pipe.send(True)
 
2565
2784
            # remove the old hook in favor of the new above hook on
 
2568
 
        if command == 'funcall':
 
 
2787
        if command == "funcall":
 
2569
2788
            funcname = request[1]
 
2570
2789
            args = request[2]
 
2571
2790
            kwargs = request[3]
 
2573
 
            parent_pipe.send(('data', getattr(client_object,
 
 
2792
            parent_pipe.send(("data", getattr(client_object,
 
2574
2793
                                              funcname)(*args,
 
2577
 
        if command == 'getattr':
 
 
2796
        if command == "getattr":
 
2578
2797
            attrname = request[1]
 
2579
2798
            if isinstance(client_object.__getattribute__(attrname),
 
2580
 
                          collections.Callable):
 
2581
 
                parent_pipe.send(('function', ))
 
 
2799
                          collections.abc.Callable):
 
 
2800
                parent_pipe.send(("function", ))
 
2583
2802
                parent_pipe.send((
 
2584
 
                    'data', client_object.__getattribute__(attrname)))
 
2586
 
        if command == 'setattr':
 
 
2803
                    "data", client_object.__getattribute__(attrname)))
 
 
2805
        if command == "setattr":
 
2587
2806
            attrname = request[1]
 
2588
2807
            value = request[2]
 
2589
2808
            setattr(client_object, attrname, value)
 
2594
2813
def rfc3339_duration_to_delta(duration):
 
2595
2814
    """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)
 
 
2816
    >>> timedelta = datetime.timedelta
 
 
2817
    >>> rfc3339_duration_to_delta("P7D") == timedelta(7)
 
 
2819
    >>> rfc3339_duration_to_delta("PT60S") == timedelta(0, 60)
 
 
2821
    >>> rfc3339_duration_to_delta("PT60M") == timedelta(0, 3600)
 
 
2823
    >>> rfc3339_duration_to_delta("PT24H") == timedelta(1)
 
 
2825
    >>> rfc3339_duration_to_delta("P1W") == timedelta(7)
 
 
2827
    >>> rfc3339_duration_to_delta("PT5M30S") == timedelta(0, 330)
 
 
2829
    >>> rfc3339_duration_to_delta("P1DT3M20S") == timedelta(1, 200)
 
2613
2834
    # Parsing an RFC 3339 duration with regular expressions is not
 
2614
2835
    # possible - there would have to be multiple places for the same
 
2615
2836
    # values, like seconds.  The current code, while more esoteric, is
 
2616
2837
    # cleaner without depending on a parsing library.  If Python had a
 
2617
2838
    # built-in library for parsing we would use it, but we'd like to
 
2618
2839
    # avoid excessive use of external libraries.
 
2620
2841
    # New type for defining tokens, syntax, and semantics all-in-one
 
2621
2842
    Token = collections.namedtuple("Token", (
 
2622
2843
        "regexp",  # To match token; if "value" is not None, must have
 
 
3019
3246
        protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
 
3020
3247
        service = AvahiServiceToSyslog(
 
3021
 
            name = server_settings["servicename"],
 
3022
 
            servicetype = "_mandos._tcp",
 
3023
 
            protocol = protocol,
 
 
3248
            name=server_settings["servicename"],
 
 
3249
            servicetype="_mandos._tcp",
 
3025
3252
        if server_settings["interface"]:
 
3026
3253
            service.interface = if_nametoindex(
 
3027
3254
                server_settings["interface"].encode("utf-8"))
 
3029
3256
    global multiprocessing_manager
 
3030
3257
    multiprocessing_manager = multiprocessing.Manager()
 
3032
3259
    client_class = Client
 
3034
 
        client_class = functools.partial(ClientDBus, bus = bus)
 
 
3261
        client_class = functools.partial(ClientDBus, bus=bus)
 
3036
3263
    client_settings = Client.config_parser(client_config)
 
3037
3264
    old_client_settings = {}
 
3038
3265
    clients_data = {}
 
3040
3267
    # This is used to redirect stdout and stderr for checker processes
 
3042
 
    wnull = open(os.devnull, "w") # A writable /dev/null
 
 
3269
    wnull = open(os.devnull, "w")  # A writable /dev/null
 
3043
3270
    # Only used if server is running in foreground but not in debug
 
3045
3272
    if debug or not foreground:
 
3048
3275
    # Get client data and settings from last running state.
 
3049
3276
    if server_settings["restore"]:
 
3051
3278
            with open(stored_state_path, "rb") as stored_state:
 
3052
 
                if sys.version_info.major == 2:                
 
 
3279
                if sys.version_info.major == 2:
 
3053
3280
                    clients_data, old_client_settings = pickle.load(
 
3056
3283
                    bytes_clients_data, bytes_old_client_settings = (
 
3057
 
                        pickle.load(stored_state, encoding = "bytes"))
 
3058
 
                    ### Fix bytes to strings
 
 
3284
                        pickle.load(stored_state, encoding="bytes"))
 
 
3285
                    #   Fix bytes to strings
 
3061
 
                    clients_data = { (key.decode("utf-8")
 
3062
 
                                      if isinstance(key, bytes)
 
3065
 
                                     bytes_clients_data.items() }
 
 
3288
                    clients_data = {(key.decode("utf-8")
 
 
3289
                                     if isinstance(key, bytes)
 
 
3292
                                    bytes_clients_data.items()}
 
3066
3293
                    del bytes_clients_data
 
3067
3294
                    for key in clients_data:
 
3068
 
                        value = { (k.decode("utf-8")
 
3069
 
                                   if isinstance(k, bytes) else k): v
 
3071
 
                                  clients_data[key].items() }
 
 
3295
                        value = {(k.decode("utf-8")
 
 
3296
                                  if isinstance(k, bytes) else k): v
 
 
3298
                                 clients_data[key].items()}
 
3072
3299
                        clients_data[key] = value
 
3073
3300
                        # .client_structure
 
3074
3301
                        value["client_structure"] = [
 
3075
3302
                            (s.decode("utf-8")
 
3076
3303
                             if isinstance(s, bytes)
 
3077
3304
                             else s) for s in
 
3078
 
                            value["client_structure"] ]
 
3080
 
                        for k in ("name", "host"):
 
 
3305
                            value["client_structure"]]
 
 
3306
                        # .name, .host, and .checker_command
 
 
3307
                        for k in ("name", "host", "checker_command"):
 
3081
3308
                            if isinstance(value[k], bytes):
 
3082
3309
                                value[k] = value[k].decode("utf-8")
 
3083
 
                    ## old_client_settings
 
 
3310
                        if "key_id" not in value:
 
 
3311
                            value["key_id"] = ""
 
 
3312
                        elif "fingerprint" not in value:
 
 
3313
                            value["fingerprint"] = ""
 
 
3314
                    #  old_client_settings
 
3085
3316
                    old_client_settings = {
 
3086
3317
                        (key.decode("utf-8")
 
3087
3318
                         if isinstance(key, bytes)
 
3088
3319
                         else key): value
 
3089
3320
                        for key, value in
 
3090
 
                        bytes_old_client_settings.items() }
 
 
3321
                        bytes_old_client_settings.items()}
 
3091
3322
                    del bytes_old_client_settings
 
 
3323
                    # .host and .checker_command
 
3093
3324
                    for value in old_client_settings.values():
 
3094
 
                        if isinstance(value["host"], bytes):
 
3095
 
                            value["host"] = (value["host"]
 
 
3325
                        for attribute in ("host", "checker_command"):
 
 
3326
                            if isinstance(value[attribute], bytes):
 
 
3327
                                value[attribute] = (value[attribute]
 
3097
3329
            os.remove(stored_state_path)
 
3098
3330
        except IOError as e:
 
3099
3331
            if e.errno == errno.ENOENT: