2
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,
 
39
 
    from future_builtins import *
 
 
37
from future_builtins import *
 
44
40
    import SocketServer as socketserver
 
 
247
212
                                     '--passphrase-file',
 
249
214
                                    + self.gnupgargs,
 
250
 
                                    stdin=subprocess.PIPE,
 
251
 
                                    stdout=subprocess.PIPE,
 
252
 
                                    stderr=subprocess.PIPE)
 
253
 
            ciphertext, err = proc.communicate(input=data)
 
 
215
                                    stdin = subprocess.PIPE,
 
 
216
                                    stdout = subprocess.PIPE,
 
 
217
                                    stderr = subprocess.PIPE)
 
 
218
            ciphertext, err = proc.communicate(input = data)
 
254
219
        if proc.returncode != 0:
 
255
220
            raise PGPError(err)
 
256
221
        return ciphertext
 
258
223
    def decrypt(self, data, password):
 
259
224
        passphrase = self.password_encode(password)
 
260
225
        with tempfile.NamedTemporaryFile(
 
261
 
                dir=self.tempdir) as passfile:
 
 
226
                dir = self.tempdir) as passfile:
 
262
227
            passfile.write(passphrase)
 
264
229
            proc = subprocess.Popen([self.gpg, '--decrypt',
 
265
230
                                     '--passphrase-file',
 
267
232
                                    + self.gnupgargs,
 
268
 
                                    stdin=subprocess.PIPE,
 
269
 
                                    stdout=subprocess.PIPE,
 
270
 
                                    stderr=subprocess.PIPE)
 
271
 
            decrypted_plaintext, err = proc.communicate(input=data)
 
 
233
                                    stdin = subprocess.PIPE,
 
 
234
                                    stdout = subprocess.PIPE,
 
 
235
                                    stderr = subprocess.PIPE)
 
 
236
            decrypted_plaintext, err = proc.communicate(input = data)
 
272
237
        if proc.returncode != 0:
 
273
238
            raise PGPError(err)
 
274
239
        return decrypted_plaintext
 
277
 
# Pretend that we have an Avahi module
 
279
 
    """This isn't so much a class as it is a module-like namespace."""
 
280
 
    IF_UNSPEC = -1               # avahi-common/address.h
 
281
 
    PROTO_UNSPEC = -1            # avahi-common/address.h
 
282
 
    PROTO_INET = 0               # avahi-common/address.h
 
283
 
    PROTO_INET6 = 1              # avahi-common/address.h
 
284
 
    DBUS_NAME = "org.freedesktop.Avahi"
 
285
 
    DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
 
286
 
    DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
 
287
 
    DBUS_PATH_SERVER = "/"
 
290
 
    def string_array_to_txt_array(t):
 
291
 
        return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
 
292
 
                           for s in t), signature="ay")
 
293
 
    ENTRY_GROUP_ESTABLISHED = 2  # avahi-common/defs.h
 
294
 
    ENTRY_GROUP_COLLISION = 3    # avahi-common/defs.h
 
295
 
    ENTRY_GROUP_FAILURE = 4      # avahi-common/defs.h
 
296
 
    SERVER_INVALID = 0           # avahi-common/defs.h
 
297
 
    SERVER_REGISTERING = 1       # avahi-common/defs.h
 
298
 
    SERVER_RUNNING = 2           # avahi-common/defs.h
 
299
 
    SERVER_COLLISION = 3         # avahi-common/defs.h
 
300
 
    SERVER_FAILURE = 4           # avahi-common/defs.h
 
303
242
class AvahiError(Exception):
 
304
243
    def __init__(self, value, *args, **kwargs):
 
305
244
        self.value = value
 
 
495
434
class AvahiServiceToSyslog(AvahiService):
 
496
435
    def rename(self, *args, **kwargs):
 
497
436
        """Add the new name to the syslog messages"""
 
498
 
        ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
 
 
437
        ret = AvahiService.rename(self, *args, **kwargs)
 
499
438
        syslogger.setFormatter(logging.Formatter(
 
500
439
            'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
 
501
440
            .format(self.name)))
 
505
443
# Pretend that we have a GnuTLS module
 
506
 
class gnutls(object):
 
507
 
    """This isn't so much a class as it is a module-like namespace."""
 
509
 
    library = ctypes.util.find_library("gnutls")
 
511
 
        library = ctypes.util.find_library("gnutls-deb0")
 
512
 
    _library = ctypes.cdll.LoadLibrary(library)
 
 
444
class GnuTLS(object):
 
 
445
    """This isn't so much a class as it is a module-like namespace.
 
 
446
    It is instantiated once, and simulates having a GnuTLS module."""
 
 
448
    _library = ctypes.cdll.LoadLibrary(
 
 
449
        ctypes.util.find_library("gnutls"))
 
 
450
    _need_version = "3.3.0"
 
 
452
        # Need to use class name "GnuTLS" here, since this method is
 
 
453
        # called before the assignment to the "gnutls" global variable
 
 
455
        if GnuTLS.check_version(self._need_version) is None:
 
 
456
            raise GnuTLS.Error("Needs GnuTLS {} or later"
 
 
457
                               .format(self._need_version))
 
515
459
    # Unless otherwise indicated, the constants and types below are
 
516
460
    # all from the gnutls/gnutls.h C header file.
 
520
464
    E_INTERRUPTED = -52
 
526
469
    CRD_CERTIFICATE = 1
 
527
470
    E_NO_CERTIFICATE_FOUND = -49
 
532
 
    KEYID_USE_SHA256 = 1        # gnutls/x509.h
 
533
471
    OPENPGP_FMT_RAW = 0         # gnutls/openpgp.h
 
536
474
    class session_int(ctypes.Structure):
 
538
476
    session_t = ctypes.POINTER(session_int)
 
540
477
    class certificate_credentials_st(ctypes.Structure):
 
542
479
    certificate_credentials_t = ctypes.POINTER(
 
543
480
        certificate_credentials_st)
 
544
481
    certificate_type_t = ctypes.c_int
 
546
482
    class datum_t(ctypes.Structure):
 
547
483
        _fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
 
548
484
                    ('size', ctypes.c_uint)]
 
550
485
    class openpgp_crt_int(ctypes.Structure):
 
552
487
    openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
 
553
 
    openpgp_crt_fmt_t = ctypes.c_int  # gnutls/openpgp.h
 
 
488
    openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
 
554
489
    log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
 
555
 
    credentials_type_t = ctypes.c_int
 
 
490
    credentials_type_t = ctypes.c_int # 
 
556
491
    transport_ptr_t = ctypes.c_void_p
 
557
492
    close_request_t = ctypes.c_int
 
560
495
    class Error(Exception):
 
561
 
        def __init__(self, message=None, code=None, args=()):
 
 
496
        # We need to use the class name "GnuTLS" here, since this
 
 
497
        # exception might be raised from within GnuTLS.__init__,
 
 
498
        # which is called before the assignment to the "gnutls"
 
 
499
        # global variable has happened.
 
 
500
        def __init__(self, message = None, code = None, args=()):
 
562
501
            # Default usage is by a message string, but if a return
 
563
502
            # code is passed, convert it to a string with
 
564
503
            # gnutls.strerror()
 
566
505
            if message is None and code is not None:
 
567
 
                message = gnutls.strerror(code)
 
568
 
            return super(gnutls.Error, self).__init__(
 
 
506
                message = GnuTLS.strerror(code)
 
 
507
            return super(GnuTLS.Error, self).__init__(
 
571
510
    class CertificateSecurityError(Error):
 
575
514
    class Credentials(object):
 
576
515
        def __init__(self):
 
 
639
572
                return _error_code(result)
 
640
573
            result = func(*arguments)
 
643
576
    # Unless otherwise indicated, the function declarations below are
 
644
577
    # all from the gnutls/gnutls.h C header file.
 
647
580
    priority_set_direct = _library.gnutls_priority_set_direct
 
648
581
    priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
 
649
582
                                    ctypes.POINTER(ctypes.c_char_p)]
 
650
583
    priority_set_direct.restype = _error_code
 
652
585
    init = _library.gnutls_init
 
653
586
    init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
 
654
587
    init.restype = _error_code
 
656
589
    set_default_priority = _library.gnutls_set_default_priority
 
657
590
    set_default_priority.argtypes = [session_t]
 
658
591
    set_default_priority.restype = _error_code
 
660
593
    record_send = _library.gnutls_record_send
 
661
594
    record_send.argtypes = [session_t, ctypes.c_void_p,
 
663
596
    record_send.restype = ctypes.c_ssize_t
 
664
597
    record_send.errcheck = _retry_on_error
 
666
599
    certificate_allocate_credentials = (
 
667
600
        _library.gnutls_certificate_allocate_credentials)
 
668
601
    certificate_allocate_credentials.argtypes = [
 
669
602
        ctypes.POINTER(certificate_credentials_t)]
 
670
603
    certificate_allocate_credentials.restype = _error_code
 
672
605
    certificate_free_credentials = (
 
673
606
        _library.gnutls_certificate_free_credentials)
 
674
 
    certificate_free_credentials.argtypes = [
 
675
 
        certificate_credentials_t]
 
 
607
    certificate_free_credentials.argtypes = [certificate_credentials_t]
 
676
608
    certificate_free_credentials.restype = None
 
678
610
    handshake_set_private_extensions = (
 
679
611
        _library.gnutls_handshake_set_private_extensions)
 
680
612
    handshake_set_private_extensions.argtypes = [session_t,
 
682
614
    handshake_set_private_extensions.restype = None
 
684
616
    credentials_set = _library.gnutls_credentials_set
 
685
617
    credentials_set.argtypes = [session_t, credentials_type_t,
 
687
619
    credentials_set.restype = _error_code
 
689
621
    strerror = _library.gnutls_strerror
 
690
622
    strerror.argtypes = [ctypes.c_int]
 
691
623
    strerror.restype = ctypes.c_char_p
 
693
625
    certificate_type_get = _library.gnutls_certificate_type_get
 
694
626
    certificate_type_get.argtypes = [session_t]
 
695
627
    certificate_type_get.restype = _error_code
 
697
629
    certificate_get_peers = _library.gnutls_certificate_get_peers
 
698
630
    certificate_get_peers.argtypes = [session_t,
 
699
631
                                      ctypes.POINTER(ctypes.c_uint)]
 
700
632
    certificate_get_peers.restype = ctypes.POINTER(datum_t)
 
702
634
    global_set_log_level = _library.gnutls_global_set_log_level
 
703
635
    global_set_log_level.argtypes = [ctypes.c_int]
 
704
636
    global_set_log_level.restype = None
 
706
638
    global_set_log_function = _library.gnutls_global_set_log_function
 
707
639
    global_set_log_function.argtypes = [log_func]
 
708
640
    global_set_log_function.restype = None
 
710
642
    deinit = _library.gnutls_deinit
 
711
643
    deinit.argtypes = [session_t]
 
712
644
    deinit.restype = None
 
714
646
    handshake = _library.gnutls_handshake
 
715
647
    handshake.argtypes = [session_t]
 
716
648
    handshake.restype = _error_code
 
717
649
    handshake.errcheck = _retry_on_error
 
719
651
    transport_set_ptr = _library.gnutls_transport_set_ptr
 
720
652
    transport_set_ptr.argtypes = [session_t, transport_ptr_t]
 
721
653
    transport_set_ptr.restype = None
 
723
655
    bye = _library.gnutls_bye
 
724
656
    bye.argtypes = [session_t, close_request_t]
 
725
657
    bye.restype = _error_code
 
726
658
    bye.errcheck = _retry_on_error
 
728
660
    check_version = _library.gnutls_check_version
 
729
661
    check_version.argtypes = [ctypes.c_char_p]
 
730
662
    check_version.restype = ctypes.c_char_p
 
732
 
    _need_version = b"3.3.0"
 
733
 
    if check_version(_need_version) is None:
 
734
 
        raise self.Error("Needs GnuTLS {} or later"
 
735
 
                         .format(_need_version))
 
737
 
    _tls_rawpk_version = b"3.6.6"
 
738
 
    has_rawpk = bool(check_version(_tls_rawpk_version))
 
742
 
        class pubkey_st(ctypes.Structure):
 
744
 
        pubkey_t = ctypes.POINTER(pubkey_st)
 
746
 
        x509_crt_fmt_t = ctypes.c_int
 
748
 
        # All the function declarations below are from gnutls/abstract.h
 
749
 
        pubkey_init = _library.gnutls_pubkey_init
 
750
 
        pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
 
751
 
        pubkey_init.restype = _error_code
 
753
 
        pubkey_import = _library.gnutls_pubkey_import
 
754
 
        pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
 
756
 
        pubkey_import.restype = _error_code
 
758
 
        pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
 
759
 
        pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
 
760
 
                                      ctypes.POINTER(ctypes.c_ubyte),
 
761
 
                                      ctypes.POINTER(ctypes.c_size_t)]
 
762
 
        pubkey_get_key_id.restype = _error_code
 
764
 
        pubkey_deinit = _library.gnutls_pubkey_deinit
 
765
 
        pubkey_deinit.argtypes = [pubkey_t]
 
766
 
        pubkey_deinit.restype = None
 
768
 
        # All the function declarations below are from gnutls/openpgp.h
 
770
 
        openpgp_crt_init = _library.gnutls_openpgp_crt_init
 
771
 
        openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
 
772
 
        openpgp_crt_init.restype = _error_code
 
774
 
        openpgp_crt_import = _library.gnutls_openpgp_crt_import
 
775
 
        openpgp_crt_import.argtypes = [openpgp_crt_t,
 
776
 
                                       ctypes.POINTER(datum_t),
 
778
 
        openpgp_crt_import.restype = _error_code
 
780
 
        openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
 
781
 
        openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
 
782
 
                                            ctypes.POINTER(ctypes.c_uint)]
 
783
 
        openpgp_crt_verify_self.restype = _error_code
 
785
 
        openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
 
786
 
        openpgp_crt_deinit.argtypes = [openpgp_crt_t]
 
787
 
        openpgp_crt_deinit.restype = None
 
789
 
        openpgp_crt_get_fingerprint = (
 
790
 
            _library.gnutls_openpgp_crt_get_fingerprint)
 
791
 
        openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
 
795
 
        openpgp_crt_get_fingerprint.restype = _error_code
 
797
 
    if check_version("3.6.4"):
 
798
 
        certificate_type_get2 = _library.gnutls_certificate_type_get2
 
799
 
        certificate_type_get2.argtypes = [session_t, ctypes.c_int]
 
800
 
        certificate_type_get2.restype = _error_code
 
 
664
    # All the function declarations below are from gnutls/openpgp.h
 
 
666
    openpgp_crt_init = _library.gnutls_openpgp_crt_init
 
 
667
    openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
 
 
668
    openpgp_crt_init.restype = _error_code
 
 
670
    openpgp_crt_import = _library.gnutls_openpgp_crt_import
 
 
671
    openpgp_crt_import.argtypes = [openpgp_crt_t,
 
 
672
                                   ctypes.POINTER(datum_t),
 
 
674
    openpgp_crt_import.restype = _error_code
 
 
676
    openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
 
 
677
    openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
 
 
678
                                        ctypes.POINTER(ctypes.c_uint)]
 
 
679
    openpgp_crt_verify_self.restype = _error_code
 
 
681
    openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
 
 
682
    openpgp_crt_deinit.argtypes = [openpgp_crt_t]
 
 
683
    openpgp_crt_deinit.restype = None
 
 
685
    openpgp_crt_get_fingerprint = (
 
 
686
        _library.gnutls_openpgp_crt_get_fingerprint)
 
 
687
    openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
 
 
691
    openpgp_crt_get_fingerprint.restype = _error_code
 
802
693
    # Remove non-public functions
 
803
694
    del _error_code, _retry_on_error
 
 
695
# Create the global "gnutls" object, simulating a module
 
806
698
def call_pipe(connection,       # : multiprocessing.Connection
 
807
699
              func, *args, **kwargs):
 
808
700
    """This function is meant to be called by multiprocessing.Process
 
810
702
    This function runs func(*args, **kwargs), and writes the resulting
 
811
703
    return value on the provided multiprocessing.Connection.
 
813
705
    connection.send(func(*args, **kwargs))
 
814
706
    connection.close()
 
817
708
class Client(object):
 
818
709
    """A representation of a client host served by this server.
 
821
712
    approved:   bool(); 'None' if not yet approved/disapproved
 
822
713
    approval_delay: datetime.timedelta(); Time to wait for approval
 
 
824
715
    checker:    subprocess.Popen(); a running checker process used
 
825
716
                                    to see if the client lives.
 
826
717
                                    'None' if no process is running.
 
827
 
    checker_callback_tag: a GLib event source tag, or None
 
 
718
    checker_callback_tag: a gobject event source tag, or None
 
828
719
    checker_command: string; External command which is run to check
 
829
720
                     if client lives.  %() expansions are done at
 
830
721
                     runtime with vars(self) as dict, so that for
 
831
722
                     instance %(name)s can be used in the command.
 
832
 
    checker_initiator_tag: a GLib event source tag, or None
 
 
723
    checker_initiator_tag: a gobject event source tag, or None
 
833
724
    created:    datetime.datetime(); (UTC) object creation
 
834
725
    client_structure: Object describing what attributes a client has
 
835
726
                      and is used for storing the client at exit
 
836
727
    current_checker_command: string; current running checker_command
 
837
 
    disable_initiator_tag: a GLib event source tag, or None
 
 
728
    disable_initiator_tag: a gobject event source tag, or None
 
839
730
    fingerprint: string (40 or 32 hexadecimal digits); used to
 
840
 
                 uniquely identify an OpenPGP client
 
841
 
    key_id: string (64 hexadecimal digits); used to uniquely identify
 
842
 
            a client using raw public keys
 
 
731
                 uniquely identify the client
 
843
732
    host:       string; available for use by the checker command
 
844
733
    interval:   datetime.timedelta(); How often to start a new checker
 
845
734
    last_approval_request: datetime.datetime(); (UTC) or None
 
 
1002
886
            logger.info("Disabling client %s", self.name)
 
1003
887
        if getattr(self, "disable_initiator_tag", None) is not None:
 
1004
 
            GLib.source_remove(self.disable_initiator_tag)
 
 
888
            gobject.source_remove(self.disable_initiator_tag)
 
1005
889
            self.disable_initiator_tag = None
 
1006
890
        self.expires = None
 
1007
891
        if getattr(self, "checker_initiator_tag", None) is not None:
 
1008
 
            GLib.source_remove(self.checker_initiator_tag)
 
 
892
            gobject.source_remove(self.checker_initiator_tag)
 
1009
893
            self.checker_initiator_tag = None
 
1010
894
        self.stop_checker()
 
1011
895
        self.enabled = False
 
1013
897
            self.send_changedstate()
 
1014
 
        # Do not run this again if called by a GLib.timeout_add
 
 
898
        # Do not run this again if called by a gobject.timeout_add
 
1017
901
    def __del__(self):
 
1020
904
    def init_checker(self):
 
1021
905
        # Schedule a new checker to be started an 'interval' from now,
 
1022
906
        # and every interval from then on.
 
1023
907
        if self.checker_initiator_tag is not None:
 
1024
 
            GLib.source_remove(self.checker_initiator_tag)
 
1025
 
        self.checker_initiator_tag = GLib.timeout_add(
 
 
908
            gobject.source_remove(self.checker_initiator_tag)
 
 
909
        self.checker_initiator_tag = gobject.timeout_add(
 
1026
910
            int(self.interval.total_seconds() * 1000),
 
1027
911
            self.start_checker)
 
1028
912
        # Schedule a disable() when 'timeout' has passed
 
1029
913
        if self.disable_initiator_tag is not None:
 
1030
 
            GLib.source_remove(self.disable_initiator_tag)
 
1031
 
        self.disable_initiator_tag = GLib.timeout_add(
 
 
914
            gobject.source_remove(self.disable_initiator_tag)
 
 
915
        self.disable_initiator_tag = gobject.timeout_add(
 
1032
916
            int(self.timeout.total_seconds() * 1000), self.disable)
 
1033
917
        # Also start a new checker *right now*.
 
1034
918
        self.start_checker()
 
1036
920
    def checker_callback(self, source, condition, connection,
 
1038
922
        """The checker has completed, so take appropriate actions."""
 
 
1121
1005
            # The exception is when not debugging but nevertheless
 
1122
1006
            # running in the foreground; use the previously
 
1123
1007
            # created wnull.
 
1124
 
            popen_args = {"close_fds": True,
 
 
1008
            popen_args = { "close_fds": True,
 
1127
1011
            if (not self.server_settings["debug"]
 
1128
1012
                and self.server_settings["foreground"]):
 
1129
1013
                popen_args.update({"stdout": wnull,
 
1131
 
            pipe = multiprocessing.Pipe(duplex=False)
 
 
1015
            pipe = multiprocessing.Pipe(duplex = False)
 
1132
1016
            self.checker = multiprocessing.Process(
 
1134
 
                args=(pipe[1], subprocess.call, command),
 
 
1018
                args = (pipe[1], subprocess.call, command),
 
 
1019
                kwargs = popen_args)
 
1136
1020
            self.checker.start()
 
1137
 
            self.checker_callback_tag = GLib.io_add_watch(
 
1138
 
                pipe[0].fileno(), GLib.IO_IN,
 
 
1021
            self.checker_callback_tag = gobject.io_add_watch(
 
 
1022
                pipe[0].fileno(), gobject.IO_IN,
 
1139
1023
                self.checker_callback, pipe[0], command)
 
1140
 
        # Re-run this periodically if run by GLib.timeout_add
 
 
1024
        # Re-run this periodically if run by gobject.timeout_add
 
1143
1027
    def stop_checker(self):
 
1144
1028
        """Force the checker process, if any, to stop."""
 
1145
1029
        if self.checker_callback_tag:
 
1146
 
            GLib.source_remove(self.checker_callback_tag)
 
 
1030
            gobject.source_remove(self.checker_callback_tag)
 
1147
1031
            self.checker_callback_tag = None
 
1148
1032
        if getattr(self, "checker", None) is None:
 
 
1503
1387
                         exc_info=error)
 
1504
1388
        return xmlstring
 
1508
1391
    dbus.OBJECT_MANAGER_IFACE
 
1509
1392
except AttributeError:
 
1510
1393
    dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
 
1513
1395
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
 
1514
1396
    """A D-Bus object with an ObjectManager.
 
1516
1398
    Classes inheriting from this exposes the standard
 
1517
1399
    GetManagedObjects call and the InterfacesAdded and
 
1518
1400
    InterfacesRemoved signals on the standard
 
1519
1401
    "org.freedesktop.DBus.ObjectManager" interface.
 
1521
1403
    Note: No signals are sent automatically; they must be sent
 
1524
1406
    @dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
 
1525
 
                         out_signature="a{oa{sa{sv}}}")
 
 
1407
                         out_signature = "a{oa{sa{sv}}}")
 
1526
1408
    def GetManagedObjects(self):
 
1527
1409
        """This function must be overridden"""
 
1528
1410
        raise NotImplementedError()
 
1530
1412
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
 
1531
 
                         signature="oa{sa{sv}}")
 
 
1413
                         signature = "oa{sa{sv}}")
 
1532
1414
    def InterfacesAdded(self, object_path, interfaces_and_properties):
 
1535
 
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
 
 
1417
    @dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
 
1536
1418
    def InterfacesRemoved(self, object_path, interfaces):
 
1539
1421
    @dbus.service.method(dbus.INTROSPECTABLE_IFACE,
 
1541
 
                         path_keyword='object_path',
 
1542
 
                         connection_keyword='connection')
 
 
1422
                         out_signature = "s",
 
 
1423
                         path_keyword = 'object_path',
 
 
1424
                         connection_keyword = 'connection')
 
1543
1425
    def Introspect(self, object_path, connection):
 
1544
1426
        """Overloading of standard D-Bus method.
 
1546
1428
        Override return argument name of GetManagedObjects to be
 
1547
1429
        "objpath_interfaces_and_properties"
 
 
1626
1507
                interface_names.add(alt_interface)
 
1627
1508
                # Is this a D-Bus signal?
 
1628
1509
                if getattr(attribute, "_dbus_is_signal", False):
 
1629
 
                    # Extract the original non-method undecorated
 
1630
 
                    # function by black magic
 
1631
1510
                    if sys.version_info.major == 2:
 
 
1511
                        # Extract the original non-method undecorated
 
 
1512
                        # function by black magic
 
1632
1513
                        nonmethod_func = (dict(
 
1633
1514
                            zip(attribute.func_code.co_freevars,
 
1634
1515
                                attribute.__closure__))
 
1635
1516
                                          ["func"].cell_contents)
 
1637
 
                        nonmethod_func = (dict(
 
1638
 
                            zip(attribute.__code__.co_freevars,
 
1639
 
                                attribute.__closure__))
 
1640
 
                                          ["func"].cell_contents)
 
 
1518
                        nonmethod_func = attribute
 
1641
1519
                    # Create a new, but exactly alike, function
 
1642
1520
                    # object, and decorate it to be a new D-Bus signal
 
1643
1521
                    # with the alternate D-Bus interface name
 
1644
 
                    new_function = copy_function(nonmethod_func)
 
 
1522
                    if sys.version_info.major == 2:
 
 
1523
                        new_function = types.FunctionType(
 
 
1524
                            nonmethod_func.func_code,
 
 
1525
                            nonmethod_func.func_globals,
 
 
1526
                            nonmethod_func.func_name,
 
 
1527
                            nonmethod_func.func_defaults,
 
 
1528
                            nonmethod_func.func_closure)
 
 
1530
                        new_function = types.FunctionType(
 
 
1531
                            nonmethod_func.__code__,
 
 
1532
                            nonmethod_func.__globals__,
 
 
1533
                            nonmethod_func.__name__,
 
 
1534
                            nonmethod_func.__defaults__,
 
 
1535
                            nonmethod_func.__closure__)
 
1645
1536
                    new_function = (dbus.service.signal(
 
1647
1538
                        attribute._dbus_signature)(new_function))
 
 
1842
1740
                                               "ApprovedByDefault")
 
1843
1741
    approval_delay = notifychangeproperty(
 
1844
1742
        dbus.UInt64, "ApprovalDelay",
 
1845
 
        type_func=lambda td: td.total_seconds() * 1000)
 
 
1743
        type_func = lambda td: td.total_seconds() * 1000)
 
1846
1744
    approval_duration = notifychangeproperty(
 
1847
1745
        dbus.UInt64, "ApprovalDuration",
 
1848
 
        type_func=lambda td: td.total_seconds() * 1000)
 
 
1746
        type_func = lambda td: td.total_seconds() * 1000)
 
1849
1747
    host = notifychangeproperty(dbus.String, "Host")
 
1850
1748
    timeout = notifychangeproperty(
 
1851
1749
        dbus.UInt64, "Timeout",
 
1852
 
        type_func=lambda td: td.total_seconds() * 1000)
 
 
1750
        type_func = lambda td: td.total_seconds() * 1000)
 
1853
1751
    extended_timeout = notifychangeproperty(
 
1854
1752
        dbus.UInt64, "ExtendedTimeout",
 
1855
 
        type_func=lambda td: td.total_seconds() * 1000)
 
 
1753
        type_func = lambda td: td.total_seconds() * 1000)
 
1856
1754
    interval = notifychangeproperty(
 
1857
1755
        dbus.UInt64, "Interval",
 
1858
 
        type_func=lambda td: td.total_seconds() * 1000)
 
 
1756
        type_func = lambda td: td.total_seconds() * 1000)
 
1859
1757
    checker_command = notifychangeproperty(dbus.String, "Checker")
 
1860
1758
    secret = notifychangeproperty(dbus.ByteArray, "Secret",
 
1861
1759
                                  invalidate_only=True)
 
1863
1761
    del notifychangeproperty
 
1865
1763
    def __del__(self, *args, **kwargs):
 
1867
1765
            self.remove_from_connection()
 
 
2239
2130
class ClientHandler(socketserver.BaseRequestHandler, object):
 
2240
2131
    """A class to handle client connections.
 
2242
2133
    Instantiated once for each connection to handle it.
 
2243
2134
    Note: This will run in its own forked process."""
 
2245
2136
    def handle(self):
 
2246
2137
        with contextlib.closing(self.server.child_pipe) as child_pipe:
 
2247
2138
            logger.info("TCP connection from: %s",
 
2248
2139
                        str(self.client_address))
 
2249
2140
            logger.debug("Pipe FD: %d",
 
2250
2141
                         self.server.child_pipe.fileno())
 
2252
2143
            session = gnutls.ClientSession(self.request)
 
2254
 
            # priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
2255
 
            #                       "+AES-256-CBC", "+SHA1",
 
2256
 
            #                       "+COMP-NULL", "+CTYPE-OPENPGP",
 
 
2145
            #priority = ':'.join(("NONE", "+VERS-TLS1.1",
 
 
2146
            #                      "+AES-256-CBC", "+SHA1",
 
 
2147
            #                      "+COMP-NULL", "+CTYPE-OPENPGP",
 
2258
2149
            # Use a fallback default, since this MUST be set.
 
2259
2150
            priority = self.server.gnutls_priority
 
2260
2151
            if priority is None:
 
2261
2152
                priority = "NORMAL"
 
2262
 
            gnutls.priority_set_direct(session._c_object,
 
2263
 
                                       priority.encode("utf-8"),
 
 
2153
            gnutls.priority_set_direct(session._c_object, priority,
 
2266
2156
            # Start communication using the Mandos protocol
 
2267
2157
            # Get protocol number
 
2268
2158
            line = self.request.makefile().readline()
 
 
2416
2284
        cert = cert_list[0]
 
2417
2285
        return ctypes.string_at(cert.data, cert.size)
 
2420
 
    def key_id(certificate):
 
2421
 
        "Convert a certificate bytestring to a hexdigit key ID"
 
2422
 
        # New GnuTLS "datum" with the public key
 
2423
 
        datum = gnutls.datum_t(
 
2424
 
            ctypes.cast(ctypes.c_char_p(certificate),
 
2425
 
                        ctypes.POINTER(ctypes.c_ubyte)),
 
2426
 
            ctypes.c_uint(len(certificate)))
 
2427
 
        # XXX all these need to be created in the gnutls "module"
 
2428
 
        # New empty GnuTLS certificate
 
2429
 
        pubkey = gnutls.pubkey_t()
 
2430
 
        gnutls.pubkey_init(ctypes.byref(pubkey))
 
2431
 
        # Import the raw public key into the certificate
 
2432
 
        gnutls.pubkey_import(pubkey,
 
2433
 
                             ctypes.byref(datum),
 
2434
 
                             gnutls.X509_FMT_DER)
 
2435
 
        # New buffer for the key ID
 
2436
 
        buf = ctypes.create_string_buffer(32)
 
2437
 
        buf_len = ctypes.c_size_t(len(buf))
 
2438
 
        # Get the key ID from the raw public key into the buffer
 
2439
 
        gnutls.pubkey_get_key_id(pubkey,
 
2440
 
                                 gnutls.KEYID_USE_SHA256,
 
2441
 
                                 ctypes.cast(ctypes.byref(buf),
 
2442
 
                                             ctypes.POINTER(ctypes.c_ubyte)),
 
2443
 
                                 ctypes.byref(buf_len))
 
2444
 
        # Deinit the certificate
 
2445
 
        gnutls.pubkey_deinit(pubkey)
 
2447
 
        # Convert the buffer to a Python bytestring
 
2448
 
        key_id = ctypes.string_at(buf, buf_len.value)
 
2449
 
        # Convert the bytestring to hexadecimal notation
 
2450
 
        hex_key_id = binascii.hexlify(key_id).upper()
 
2454
2288
    def fingerprint(openpgp):
 
2455
2289
        "Convert an OpenPGP bytestring to a hexdigit fingerprint"
 
 
2490
2323
class MultiprocessingMixIn(object):
 
2491
2324
    """Like socketserver.ThreadingMixIn, but with multiprocessing"""
 
2493
2326
    def sub_process_main(self, request, address):
 
2495
2328
            self.finish_request(request, address)
 
2496
2329
        except Exception:
 
2497
2330
            self.handle_error(request, address)
 
2498
2331
        self.close_request(request)
 
2500
2333
    def process_request(self, request, address):
 
2501
2334
        """Start a new process to process the request."""
 
2502
 
        proc = multiprocessing.Process(target=self.sub_process_main,
 
2503
 
                                       args=(request, address))
 
 
2335
        proc = multiprocessing.Process(target = self.sub_process_main,
 
 
2336
                                       args = (request, address))
 
2508
2341
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
 
2509
2342
    """ adds a pipe to the MixIn """
 
2511
2344
    def process_request(self, request, client_address):
 
2512
2345
        """Overrides and wraps the original process_request().
 
2514
2347
        This function creates a new pipe in self.pipe
 
2516
2349
        parent_pipe, self.child_pipe = multiprocessing.Pipe()
 
2518
2351
        proc = MultiprocessingMixIn.process_request(self, request,
 
2519
2352
                                                    client_address)
 
2520
2353
        self.child_pipe.close()
 
2521
2354
        self.add_pipe(parent_pipe, proc)
 
2523
2356
    def add_pipe(self, parent_pipe, proc):
 
2524
2357
        """Dummy function; override as necessary"""
 
2525
2358
        raise NotImplementedError()
 
 
2574
2406
        # socket_wrapper(), if socketfd was set.
 
2575
2407
        socketserver.TCPServer.__init__(self, server_address,
 
2576
2408
                                        RequestHandlerClass)
 
2578
2410
    def server_bind(self):
 
2579
2411
        """This overrides the normal server_bind() function
 
2580
2412
        to bind to an interface if one was specified, and also NOT to
 
2581
2413
        bind to an address or port if they were not specified."""
 
2582
 
        global SO_BINDTODEVICE
 
2583
2414
        if self.interface is not None:
 
2584
2415
            if SO_BINDTODEVICE is None:
 
2585
 
                # Fall back to a hard-coded value which seems to be
 
2587
 
                logger.warning("SO_BINDTODEVICE not found, trying 25")
 
2588
 
                SO_BINDTODEVICE = 25
 
2590
 
                self.socket.setsockopt(
 
2591
 
                    socket.SOL_SOCKET, SO_BINDTODEVICE,
 
2592
 
                    (self.interface + "\0").encode("utf-8"))
 
2593
 
            except socket.error as error:
 
2594
 
                if error.errno == errno.EPERM:
 
2595
 
                    logger.error("No permission to bind to"
 
2596
 
                                 " interface %s", self.interface)
 
2597
 
                elif error.errno == errno.ENOPROTOOPT:
 
2598
 
                    logger.error("SO_BINDTODEVICE not available;"
 
2599
 
                                 " cannot bind to interface %s",
 
2601
 
                elif error.errno == errno.ENODEV:
 
2602
 
                    logger.error("Interface %s does not exist,"
 
2603
 
                                 " cannot bind", self.interface)
 
 
2416
                logger.error("SO_BINDTODEVICE does not exist;"
 
 
2417
                             " cannot bind to interface %s",
 
 
2421
                    self.socket.setsockopt(
 
 
2422
                        socket.SOL_SOCKET, SO_BINDTODEVICE,
 
 
2423
                        (self.interface + "\0").encode("utf-8"))
 
 
2424
                except socket.error as error:
 
 
2425
                    if error.errno == errno.EPERM:
 
 
2426
                        logger.error("No permission to bind to"
 
 
2427
                                     " interface %s", self.interface)
 
 
2428
                    elif error.errno == errno.ENOPROTOOPT:
 
 
2429
                        logger.error("SO_BINDTODEVICE not available;"
 
 
2430
                                     " cannot bind to interface %s",
 
 
2432
                    elif error.errno == errno.ENODEV:
 
 
2433
                        logger.error("Interface %s does not exist,"
 
 
2434
                                     " cannot bind", self.interface)
 
2606
2437
        # Only bind(2) the socket if we really need to.
 
2607
2438
        if self.server_address[0] or self.server_address[1]:
 
2608
 
            if self.server_address[1]:
 
2609
 
                self.allow_reuse_address = True
 
2610
2439
            if not self.server_address[0]:
 
2611
2440
                if self.address_family == socket.AF_INET6:
 
2612
 
                    any_address = "::"  # in6addr_any
 
 
2441
                    any_address = "::" # in6addr_any
 
2614
 
                    any_address = "0.0.0.0"  # INADDR_ANY
 
 
2443
                    any_address = "0.0.0.0" # INADDR_ANY
 
2615
2444
                self.server_address = (any_address,
 
2616
2445
                                       self.server_address[1])
 
2617
2446
            elif not self.server_address[1]:
 
 
2651
2480
        self.gnutls_priority = gnutls_priority
 
2652
2481
        IPv6_TCPServer.__init__(self, server_address,
 
2653
2482
                                RequestHandlerClass,
 
2654
 
                                interface=interface,
 
 
2483
                                interface = interface,
 
 
2484
                                use_ipv6 = use_ipv6,
 
 
2485
                                socketfd = socketfd)
 
2658
2487
    def server_activate(self):
 
2659
2488
        if self.enabled:
 
2660
2489
            return socketserver.TCPServer.server_activate(self)
 
2662
2491
    def enable(self):
 
2663
2492
        self.enabled = True
 
2665
2494
    def add_pipe(self, parent_pipe, proc):
 
2666
2495
        # Call "handle_ipc" for both data and EOF events
 
 
2496
        gobject.io_add_watch(
 
2668
2497
            parent_pipe.fileno(),
 
2669
 
            GLib.IO_IN | GLib.IO_HUP,
 
 
2498
            gobject.IO_IN | gobject.IO_HUP,
 
2670
2499
            functools.partial(self.handle_ipc,
 
2671
 
                              parent_pipe=parent_pipe,
 
 
2500
                              parent_pipe = parent_pipe,
 
2674
2503
    def handle_ipc(self, source, condition,
 
2675
2504
                   parent_pipe=None,
 
2677
2506
                   client_object=None):
 
2678
2507
        # error, or the other end of multiprocessing.Pipe has closed
 
2679
 
        if condition & (GLib.IO_ERR | GLib.IO_HUP):
 
 
2508
        if condition & (gobject.IO_ERR | gobject.IO_HUP):
 
2680
2509
            # Wait for other process to exit
 
2684
2513
        # Read a request from the child
 
2685
2514
        request = parent_pipe.recv()
 
2686
2515
        command = request[0]
 
2688
2517
        if command == 'init':
 
2689
 
            key_id = request[1].decode("ascii")
 
2690
 
            fpr = request[2].decode("ascii")
 
2691
 
            address = request[3]
 
2693
 
            for c in self.clients.values():
 
2694
 
                if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
 
2696
 
                if key_id and c.key_id == key_id:
 
2699
 
                if fpr and c.fingerprint == fpr:
 
 
2519
            address = request[2]
 
 
2521
            for c in self.clients.itervalues():
 
 
2522
                if c.fingerprint == fpr:
 
2703
 
                logger.info("Client not found for key ID: %s, address"
 
2704
 
                            ": %s", key_id or fpr, address)
 
 
2526
                logger.info("Client not found for fingerprint: %s, ad"
 
 
2527
                            "dress: %s", fpr, address)
 
2705
2528
                if self.use_dbus:
 
2706
2529
                    # Emit D-Bus signal
 
2707
 
                    mandos_dbus_service.ClientNotFound(key_id or fpr,
 
 
2530
                    mandos_dbus_service.ClientNotFound(fpr,
 
2709
2532
                parent_pipe.send(False)
 
 
2535
            gobject.io_add_watch(
 
2713
2536
                parent_pipe.fileno(),
 
2714
 
                GLib.IO_IN | GLib.IO_HUP,
 
 
2537
                gobject.IO_IN | gobject.IO_HUP,
 
2715
2538
                functools.partial(self.handle_ipc,
 
2716
 
                                  parent_pipe=parent_pipe,
 
2718
 
                                  client_object=client))
 
 
2539
                                  parent_pipe = parent_pipe,
 
 
2541
                                  client_object = client))
 
2719
2542
            parent_pipe.send(True)
 
2720
2543
            # remove the old hook in favor of the new above hook on
 
 
3127
 
            logger.debug("Did setuid/setgid to {}:{}".format(uid,
 
3129
2940
    except OSError as error:
 
3130
 
        logger.warning("Failed to setuid/setgid to {}:{}: {}"
 
3131
 
                       .format(uid, gid, os.strerror(error.errno)))
 
3132
2941
        if error.errno != errno.EPERM:
 
3136
2945
        # Enable all possible GnuTLS debugging
 
3138
2947
        # "Use a log level over 10 to enable all debugging options."
 
3139
2948
        # - GnuTLS manual
 
3140
2949
        gnutls.global_set_log_level(11)
 
3142
2951
        @gnutls.log_func
 
3143
2952
        def debug_gnutls(level, string):
 
3144
2953
            logger.debug("GnuTLS: %s", string[:-1])
 
3146
2955
        gnutls.global_set_log_function(debug_gnutls)
 
3148
2957
        # Redirect stdin so all checkers get /dev/null
 
3149
2958
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
 
3150
2959
        os.dup2(null, sys.stdin.fileno())
 
3154
2963
    # Need to fork before connecting to D-Bus
 
3155
2964
    if not foreground:
 
3156
2965
        # Close all input and output, do double fork, etc.
 
3159
 
    # multiprocessing will use threads, so before we use GLib we need
 
3160
 
    # to inform GLib that threads will be used.
 
 
2968
    # multiprocessing will use threads, so before we use gobject we
 
 
2969
    # need to inform gobject that threads will be used.
 
 
2970
    gobject.threads_init()
 
3163
2972
    global main_loop
 
3164
2973
    # From the Avahi example code
 
3165
2974
    DBusGMainLoop(set_as_default=True)
 
3166
 
    main_loop = GLib.MainLoop()
 
 
2975
    main_loop = gobject.MainLoop()
 
3167
2976
    bus = dbus.SystemBus()
 
3168
2977
    # End of Avahi example code
 
 
3183
2992
        protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
 
3184
2993
        service = AvahiServiceToSyslog(
 
3185
 
            name=server_settings["servicename"],
 
3186
 
            servicetype="_mandos._tcp",
 
 
2994
            name = server_settings["servicename"],
 
 
2995
            servicetype = "_mandos._tcp",
 
 
2996
            protocol = protocol,
 
3189
2998
        if server_settings["interface"]:
 
3190
2999
            service.interface = if_nametoindex(
 
3191
3000
                server_settings["interface"].encode("utf-8"))
 
3193
3002
    global multiprocessing_manager
 
3194
3003
    multiprocessing_manager = multiprocessing.Manager()
 
3196
3005
    client_class = Client
 
3198
 
        client_class = functools.partial(ClientDBus, bus=bus)
 
 
3007
        client_class = functools.partial(ClientDBus, bus = bus)
 
3200
3009
    client_settings = Client.config_parser(client_config)
 
3201
3010
    old_client_settings = {}
 
3202
3011
    clients_data = {}
 
3204
3013
    # This is used to redirect stdout and stderr for checker processes
 
3206
 
    wnull = open(os.devnull, "w")  # A writable /dev/null
 
 
3015
    wnull = open(os.devnull, "w") # A writable /dev/null
 
3207
3016
    # Only used if server is running in foreground but not in debug
 
3209
3018
    if debug or not foreground:
 
3212
3021
    # Get client data and settings from last running state.
 
3213
3022
    if server_settings["restore"]:
 
3215
3024
            with open(stored_state_path, "rb") as stored_state:
 
3216
 
                if sys.version_info.major == 2:
 
3217
 
                    clients_data, old_client_settings = pickle.load(
 
3220
 
                    bytes_clients_data, bytes_old_client_settings = (
 
3221
 
                        pickle.load(stored_state, encoding="bytes"))
 
3222
 
                    #   Fix bytes to strings
 
3225
 
                    clients_data = {(key.decode("utf-8")
 
3226
 
                                     if isinstance(key, bytes)
 
3229
 
                                    bytes_clients_data.items()}
 
3230
 
                    del bytes_clients_data
 
3231
 
                    for key in clients_data:
 
3232
 
                        value = {(k.decode("utf-8")
 
3233
 
                                  if isinstance(k, bytes) else k): v
 
3235
 
                                 clients_data[key].items()}
 
3236
 
                        clients_data[key] = value
 
3238
 
                        value["client_structure"] = [
 
3240
 
                             if isinstance(s, bytes)
 
3242
 
                            value["client_structure"]]
 
3244
 
                        for k in ("name", "host"):
 
3245
 
                            if isinstance(value[k], bytes):
 
3246
 
                                value[k] = value[k].decode("utf-8")
 
3247
 
                        if not value.has_key("key_id"):
 
3248
 
                            value["key_id"] = ""
 
3249
 
                        elif not value.has_key("fingerprint"):
 
3250
 
                            value["fingerprint"] = ""
 
3251
 
                    #  old_client_settings
 
3253
 
                    old_client_settings = {
 
3254
 
                        (key.decode("utf-8")
 
3255
 
                         if isinstance(key, bytes)
 
3258
 
                        bytes_old_client_settings.items()}
 
3259
 
                    del bytes_old_client_settings
 
3261
 
                    for value in old_client_settings.values():
 
3262
 
                        if isinstance(value["host"], bytes):
 
3263
 
                            value["host"] = (value["host"]
 
 
3025
                clients_data, old_client_settings = pickle.load(
 
3265
3027
            os.remove(stored_state_path)
 
3266
3028
        except IOError as e:
 
3267
3029
            if e.errno == errno.ENOENT:
 
 
3367
3129
                             pidfilename, pid)
 
3369
3131
        del pidfilename
 
3371
 
    for termsig in (signal.SIGHUP, signal.SIGTERM):
 
3372
 
        GLib.unix_signal_add(GLib.PRIORITY_HIGH, termsig,
 
3373
 
                             lambda: main_loop.quit() and False)
 
 
3133
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
 
 
3134
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
 
3377
3138
        @alternate_dbus_interfaces(
 
3378
 
            {"se.recompile.Mandos": "se.bsnet.fukt.Mandos"})
 
 
3139
            { "se.recompile.Mandos": "se.bsnet.fukt.Mandos" })
 
3379
3140
        class MandosDBusService(DBusObjectWithObjectManager):
 
3380
3141
            """A D-Bus proxy object"""
 
3382
3143
            def __init__(self):
 
3383
3144
                dbus.service.Object.__init__(self, bus, "/")
 
3385
3146
            _interface = "se.recompile.Mandos"
 
3387
3148
            @dbus.service.signal(_interface, signature="o")
 
3388
3149
            def ClientAdded(self, objpath):
 
3392
3153
            @dbus.service.signal(_interface, signature="ss")
 
3393
 
            def ClientNotFound(self, key_id, address):
 
 
3154
            def ClientNotFound(self, fingerprint, address):
 
3397
3158
            @dbus_annotations({"org.freedesktop.DBus.Deprecated":
 
3399
3160
            @dbus.service.signal(_interface, signature="os")
 
3400
3161
            def ClientRemoved(self, objpath, name):
 
3404
3165
            @dbus_annotations({"org.freedesktop.DBus.Deprecated":
 
3406
3167
            @dbus.service.method(_interface, out_signature="ao")
 
3407
3168
            def GetAllClients(self):
 
3409
3170
                return dbus.Array(c.dbus_object_path for c in
 
3410
 
                                  tcp_server.clients.values())
 
 
3171
                                  tcp_server.clients.itervalues())
 
3412
3173
            @dbus_annotations({"org.freedesktop.DBus.Deprecated":
 
3414
3175
            @dbus.service.method(_interface,
 
 
3473
3234
                    self.ClientRemoved(client.dbus_object_path,
 
3476
3237
        mandos_dbus_service = MandosDBusService()
 
3478
 
    # Save modules to variables to exempt the modules from being
 
3479
 
    # unloaded before the function registered with atexit() is run.
 
3480
 
    mp = multiprocessing
 
3484
3240
        "Cleanup function; run on exit"
 
3486
3242
            service.cleanup()
 
3488
 
        mp.active_children()
 
 
3244
        multiprocessing.active_children()
 
3490
3246
        if not (tcp_server.clients or client_settings):
 
3493
3249
        # Store client before exiting. Secrets are encrypted with key
 
3494
3250
        # based on what config file has. If config file is
 
3495
3251
        # removed/edited, old secret will thus be unrecovable.
 
3497
3253
        with PGPEngine() as pgp:
 
3498
 
            for client in tcp_server.clients.values():
 
 
3254
            for client in tcp_server.clients.itervalues():
 
3499
3255
                key = client_settings[client.name]["secret"]
 
3500
3256
                client.encrypted_secret = pgp.encrypt(client.secret,
 
3502
3258
                client_dict = {}
 
3504
3260
                # A list of attributes that can not be pickled
 
3506
 
                exclude = {"bus", "changedstate", "secret",
 
3507
 
                           "checker", "server_settings"}
 
 
3262
                exclude = { "bus", "changedstate", "secret",
 
 
3263
                            "checker", "server_settings" }
 
3508
3264
                for name, typ in inspect.getmembers(dbus.service
 
3510
3266
                    exclude.add(name)
 
3512
3268
                client_dict["encrypted_secret"] = (client
 
3513
3269
                                                   .encrypted_secret)
 
3514
3270
                for attr in client.client_structure:
 
3515
3271
                    if attr not in exclude:
 
3516
3272
                        client_dict[attr] = getattr(client, attr)
 
3518
3274
                clients[client.name] = client_dict
 
3519
3275
                del client_settings[client.name]["secret"]
 
3522
3278
            with tempfile.NamedTemporaryFile(