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(