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-2015 Teddy Hogeborn
15
# Copyright © 2008-2015 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
236
190
.replace(b"\n", b"\\n")
237
191
.replace(b"\0", b"\\x00"))
240
194
def encrypt(self, data, password):
241
195
passphrase = self.password_encode(password)
242
196
with tempfile.NamedTemporaryFile(
243
197
dir=self.tempdir) as passfile:
244
198
passfile.write(passphrase)
246
proc = subprocess.Popen([self.gpg, '--symmetric',
200
proc = subprocess.Popen(['gpg', '--symmetric',
247
201
'--passphrase-file',
249
203
+ self.gnupgargs,
250
stdin=subprocess.PIPE,
251
stdout=subprocess.PIPE,
252
stderr=subprocess.PIPE)
253
ciphertext, err = proc.communicate(input=data)
204
stdin = subprocess.PIPE,
205
stdout = subprocess.PIPE,
206
stderr = subprocess.PIPE)
207
ciphertext, err = proc.communicate(input = data)
254
208
if proc.returncode != 0:
255
209
raise PGPError(err)
256
210
return ciphertext
258
212
def decrypt(self, data, password):
259
213
passphrase = self.password_encode(password)
260
214
with tempfile.NamedTemporaryFile(
261
dir=self.tempdir) as passfile:
215
dir = self.tempdir) as passfile:
262
216
passfile.write(passphrase)
264
proc = subprocess.Popen([self.gpg, '--decrypt',
218
proc = subprocess.Popen(['gpg', '--decrypt',
265
219
'--passphrase-file',
267
221
+ self.gnupgargs,
268
stdin=subprocess.PIPE,
269
stdout=subprocess.PIPE,
270
stderr=subprocess.PIPE)
271
decrypted_plaintext, err = proc.communicate(input=data)
222
stdin = subprocess.PIPE,
223
stdout = subprocess.PIPE,
224
stderr = subprocess.PIPE)
225
decrypted_plaintext, err = proc.communicate(input = data)
272
226
if proc.returncode != 0:
273
227
raise PGPError(err)
274
228
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
231
class AvahiError(Exception):
304
232
def __init__(self, value, *args, **kwargs):
305
233
self.value = value
495
423
class AvahiServiceToSyslog(AvahiService):
496
424
def rename(self, *args, **kwargs):
497
425
"""Add the new name to the syslog messages"""
498
ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
426
ret = AvahiService.rename(self, *args, **kwargs)
499
427
syslogger.setFormatter(logging.Formatter(
500
428
'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
501
429
.format(self.name)))
505
432
# 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)
433
class GnuTLS(object):
434
"""This isn't so much a class as it is a module-like namespace.
435
It is instantiated once, and simulates having a GnuTLS module."""
437
_library = ctypes.cdll.LoadLibrary(
438
ctypes.util.find_library("gnutls"))
439
_need_version = "3.3.0"
441
# Need to use class name "GnuTLS" here, since this method is
442
# called before the assignment to the "gnutls" global variable
444
if GnuTLS.check_version(self._need_version) is None:
445
raise GnuTLS.Error("Needs GnuTLS {} or later"
446
.format(self._need_version))
515
448
# Unless otherwise indicated, the constants and types below are
516
449
# all from the gnutls/gnutls.h C header file.
520
453
E_INTERRUPTED = -52
526
458
CRD_CERTIFICATE = 1
527
459
E_NO_CERTIFICATE_FOUND = -49
532
KEYID_USE_SHA256 = 1 # gnutls/x509.h
533
460
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
536
463
class session_int(ctypes.Structure):
538
465
session_t = ctypes.POINTER(session_int)
540
466
class certificate_credentials_st(ctypes.Structure):
542
468
certificate_credentials_t = ctypes.POINTER(
543
469
certificate_credentials_st)
544
470
certificate_type_t = ctypes.c_int
546
471
class datum_t(ctypes.Structure):
547
472
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
548
473
('size', ctypes.c_uint)]
550
474
class openpgp_crt_int(ctypes.Structure):
552
476
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
553
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
477
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
554
478
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
555
credentials_type_t = ctypes.c_int
479
credentials_type_t = ctypes.c_int #
556
480
transport_ptr_t = ctypes.c_void_p
557
481
close_request_t = ctypes.c_int
560
484
class Error(Exception):
561
def __init__(self, message=None, code=None, args=()):
485
# We need to use the class name "GnuTLS" here, since this
486
# exception might be raised from within GnuTLS.__init__,
487
# which is called before the assignment to the "gnutls"
488
# global variable has happened.
489
def __init__(self, message = None, code = None, args=()):
562
490
# Default usage is by a message string, but if a return
563
491
# code is passed, convert it to a string with
564
492
# gnutls.strerror()
566
494
if message is None and code is not None:
567
message = gnutls.strerror(code)
568
return super(gnutls.Error, self).__init__(
495
message = GnuTLS.strerror(code)
496
return super(GnuTLS.Error, self).__init__(
571
499
class CertificateSecurityError(Error):
575
503
class Credentials(object):
576
504
def __init__(self):
639
561
return _error_code(result)
640
562
result = func(*arguments)
643
565
# Unless otherwise indicated, the function declarations below are
644
566
# all from the gnutls/gnutls.h C header file.
647
569
priority_set_direct = _library.gnutls_priority_set_direct
648
570
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
649
571
ctypes.POINTER(ctypes.c_char_p)]
650
572
priority_set_direct.restype = _error_code
652
574
init = _library.gnutls_init
653
575
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
654
576
init.restype = _error_code
656
578
set_default_priority = _library.gnutls_set_default_priority
657
579
set_default_priority.argtypes = [session_t]
658
580
set_default_priority.restype = _error_code
660
582
record_send = _library.gnutls_record_send
661
583
record_send.argtypes = [session_t, ctypes.c_void_p,
663
585
record_send.restype = ctypes.c_ssize_t
664
586
record_send.errcheck = _retry_on_error
666
588
certificate_allocate_credentials = (
667
589
_library.gnutls_certificate_allocate_credentials)
668
590
certificate_allocate_credentials.argtypes = [
669
591
ctypes.POINTER(certificate_credentials_t)]
670
592
certificate_allocate_credentials.restype = _error_code
672
594
certificate_free_credentials = (
673
595
_library.gnutls_certificate_free_credentials)
674
certificate_free_credentials.argtypes = [
675
certificate_credentials_t]
596
certificate_free_credentials.argtypes = [certificate_credentials_t]
676
597
certificate_free_credentials.restype = None
678
599
handshake_set_private_extensions = (
679
600
_library.gnutls_handshake_set_private_extensions)
680
601
handshake_set_private_extensions.argtypes = [session_t,
682
603
handshake_set_private_extensions.restype = None
684
605
credentials_set = _library.gnutls_credentials_set
685
606
credentials_set.argtypes = [session_t, credentials_type_t,
687
608
credentials_set.restype = _error_code
689
610
strerror = _library.gnutls_strerror
690
611
strerror.argtypes = [ctypes.c_int]
691
612
strerror.restype = ctypes.c_char_p
693
614
certificate_type_get = _library.gnutls_certificate_type_get
694
615
certificate_type_get.argtypes = [session_t]
695
616
certificate_type_get.restype = _error_code
697
618
certificate_get_peers = _library.gnutls_certificate_get_peers
698
619
certificate_get_peers.argtypes = [session_t,
699
620
ctypes.POINTER(ctypes.c_uint)]
700
621
certificate_get_peers.restype = ctypes.POINTER(datum_t)
702
623
global_set_log_level = _library.gnutls_global_set_log_level
703
624
global_set_log_level.argtypes = [ctypes.c_int]
704
625
global_set_log_level.restype = None
706
627
global_set_log_function = _library.gnutls_global_set_log_function
707
628
global_set_log_function.argtypes = [log_func]
708
629
global_set_log_function.restype = None
710
631
deinit = _library.gnutls_deinit
711
632
deinit.argtypes = [session_t]
712
633
deinit.restype = None
714
635
handshake = _library.gnutls_handshake
715
636
handshake.argtypes = [session_t]
716
637
handshake.restype = _error_code
717
638
handshake.errcheck = _retry_on_error
719
640
transport_set_ptr = _library.gnutls_transport_set_ptr
720
641
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
721
642
transport_set_ptr.restype = None
723
644
bye = _library.gnutls_bye
724
645
bye.argtypes = [session_t, close_request_t]
725
646
bye.restype = _error_code
726
647
bye.errcheck = _retry_on_error
728
649
check_version = _library.gnutls_check_version
729
650
check_version.argtypes = [ctypes.c_char_p]
730
651
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
653
# All the function declarations below are from gnutls/openpgp.h
655
openpgp_crt_init = _library.gnutls_openpgp_crt_init
656
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
657
openpgp_crt_init.restype = _error_code
659
openpgp_crt_import = _library.gnutls_openpgp_crt_import
660
openpgp_crt_import.argtypes = [openpgp_crt_t,
661
ctypes.POINTER(datum_t),
663
openpgp_crt_import.restype = _error_code
665
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
666
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
667
ctypes.POINTER(ctypes.c_uint)]
668
openpgp_crt_verify_self.restype = _error_code
670
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
671
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
672
openpgp_crt_deinit.restype = None
674
openpgp_crt_get_fingerprint = (
675
_library.gnutls_openpgp_crt_get_fingerprint)
676
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
680
openpgp_crt_get_fingerprint.restype = _error_code
802
682
# Remove non-public functions
803
683
del _error_code, _retry_on_error
684
# Create the global "gnutls" object, simulating a module
806
687
def call_pipe(connection, # : multiprocessing.Connection
807
688
func, *args, **kwargs):
808
689
"""This function is meant to be called by multiprocessing.Process
810
691
This function runs func(*args, **kwargs), and writes the resulting
811
692
return value on the provided multiprocessing.Connection.
813
694
connection.send(func(*args, **kwargs))
814
695
connection.close()
817
697
class Client(object):
818
698
"""A representation of a client host served by this server.
821
701
approved: bool(); 'None' if not yet approved/disapproved
822
702
approval_delay: datetime.timedelta(); Time to wait for approval
824
704
checker: subprocess.Popen(); a running checker process used
825
705
to see if the client lives.
826
706
'None' if no process is running.
827
checker_callback_tag: a GLib event source tag, or None
707
checker_callback_tag: a gobject event source tag, or None
828
708
checker_command: string; External command which is run to check
829
709
if client lives. %() expansions are done at
830
710
runtime with vars(self) as dict, so that for
831
711
instance %(name)s can be used in the command.
832
checker_initiator_tag: a GLib event source tag, or None
712
checker_initiator_tag: a gobject event source tag, or None
833
713
created: datetime.datetime(); (UTC) object creation
834
714
client_structure: Object describing what attributes a client has
835
715
and is used for storing the client at exit
836
716
current_checker_command: string; current running checker_command
837
disable_initiator_tag: a GLib event source tag, or None
717
disable_initiator_tag: a gobject event source tag, or None
839
719
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
720
uniquely identify the client
843
721
host: string; available for use by the checker command
844
722
interval: datetime.timedelta(); How often to start a new checker
845
723
last_approval_request: datetime.datetime(); (UTC) or None
1002
875
logger.info("Disabling client %s", self.name)
1003
876
if getattr(self, "disable_initiator_tag", None) is not None:
1004
GLib.source_remove(self.disable_initiator_tag)
877
gobject.source_remove(self.disable_initiator_tag)
1005
878
self.disable_initiator_tag = None
1006
879
self.expires = None
1007
880
if getattr(self, "checker_initiator_tag", None) is not None:
1008
GLib.source_remove(self.checker_initiator_tag)
881
gobject.source_remove(self.checker_initiator_tag)
1009
882
self.checker_initiator_tag = None
1010
883
self.stop_checker()
1011
884
self.enabled = False
1013
886
self.send_changedstate()
1014
# Do not run this again if called by a GLib.timeout_add
887
# Do not run this again if called by a gobject.timeout_add
1017
890
def __del__(self):
1020
893
def init_checker(self):
1021
894
# Schedule a new checker to be started an 'interval' from now,
1022
895
# and every interval from then on.
1023
896
if self.checker_initiator_tag is not None:
1024
GLib.source_remove(self.checker_initiator_tag)
1025
self.checker_initiator_tag = GLib.timeout_add(
897
gobject.source_remove(self.checker_initiator_tag)
898
self.checker_initiator_tag = gobject.timeout_add(
1026
899
int(self.interval.total_seconds() * 1000),
1027
900
self.start_checker)
1028
901
# Schedule a disable() when 'timeout' has passed
1029
902
if self.disable_initiator_tag is not None:
1030
GLib.source_remove(self.disable_initiator_tag)
1031
self.disable_initiator_tag = GLib.timeout_add(
903
gobject.source_remove(self.disable_initiator_tag)
904
self.disable_initiator_tag = gobject.timeout_add(
1032
905
int(self.timeout.total_seconds() * 1000), self.disable)
1033
906
# Also start a new checker *right now*.
1034
907
self.start_checker()
1036
909
def checker_callback(self, source, condition, connection,
1038
911
"""The checker has completed, so take appropriate actions."""
1121
994
# The exception is when not debugging but nevertheless
1122
995
# running in the foreground; use the previously
1123
996
# created wnull.
1124
popen_args = {"close_fds": True,
997
popen_args = { "close_fds": True,
1127
1000
if (not self.server_settings["debug"]
1128
1001
and self.server_settings["foreground"]):
1129
1002
popen_args.update({"stdout": wnull,
1131
pipe = multiprocessing.Pipe(duplex=False)
1004
pipe = multiprocessing.Pipe(duplex = False)
1132
1005
self.checker = multiprocessing.Process(
1134
args=(pipe[1], subprocess.call, command),
1007
args = (pipe[1], subprocess.call, command),
1008
kwargs = popen_args)
1136
1009
self.checker.start()
1137
self.checker_callback_tag = GLib.io_add_watch(
1138
pipe[0].fileno(), GLib.IO_IN,
1010
self.checker_callback_tag = gobject.io_add_watch(
1011
pipe[0].fileno(), gobject.IO_IN,
1139
1012
self.checker_callback, pipe[0], command)
1140
# Re-run this periodically if run by GLib.timeout_add
1013
# Re-run this periodically if run by gobject.timeout_add
1143
1016
def stop_checker(self):
1144
1017
"""Force the checker process, if any, to stop."""
1145
1018
if self.checker_callback_tag:
1146
GLib.source_remove(self.checker_callback_tag)
1019
gobject.source_remove(self.checker_callback_tag)
1147
1020
self.checker_callback_tag = None
1148
1021
if getattr(self, "checker", None) is None:
1503
1376
exc_info=error)
1504
1377
return xmlstring
1508
1380
dbus.OBJECT_MANAGER_IFACE
1509
1381
except AttributeError:
1510
1382
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
1513
1384
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
1514
1385
"""A D-Bus object with an ObjectManager.
1516
1387
Classes inheriting from this exposes the standard
1517
1388
GetManagedObjects call and the InterfacesAdded and
1518
1389
InterfacesRemoved signals on the standard
1519
1390
"org.freedesktop.DBus.ObjectManager" interface.
1521
1392
Note: No signals are sent automatically; they must be sent
1524
1395
@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
1525
out_signature="a{oa{sa{sv}}}")
1396
out_signature = "a{oa{sa{sv}}}")
1526
1397
def GetManagedObjects(self):
1527
1398
"""This function must be overridden"""
1528
1399
raise NotImplementedError()
1530
1401
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
1531
signature="oa{sa{sv}}")
1402
signature = "oa{sa{sv}}")
1532
1403
def InterfacesAdded(self, object_path, interfaces_and_properties):
1535
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
1406
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
1536
1407
def InterfacesRemoved(self, object_path, interfaces):
1539
1410
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
1541
path_keyword='object_path',
1542
connection_keyword='connection')
1411
out_signature = "s",
1412
path_keyword = 'object_path',
1413
connection_keyword = 'connection')
1543
1414
def Introspect(self, object_path, connection):
1544
1415
"""Overloading of standard D-Bus method.
1546
1417
Override return argument name of GetManagedObjects to be
1547
1418
"objpath_interfaces_and_properties"
1626
1496
interface_names.add(alt_interface)
1627
1497
# Is this a D-Bus signal?
1628
1498
if getattr(attribute, "_dbus_is_signal", False):
1629
# Extract the original non-method undecorated
1630
# function by black magic
1631
1499
if sys.version_info.major == 2:
1500
# Extract the original non-method undecorated
1501
# function by black magic
1632
1502
nonmethod_func = (dict(
1633
1503
zip(attribute.func_code.co_freevars,
1634
1504
attribute.__closure__))
1635
1505
["func"].cell_contents)
1637
nonmethod_func = (dict(
1638
zip(attribute.__code__.co_freevars,
1639
attribute.__closure__))
1640
["func"].cell_contents)
1507
nonmethod_func = attribute
1641
1508
# Create a new, but exactly alike, function
1642
1509
# object, and decorate it to be a new D-Bus signal
1643
1510
# with the alternate D-Bus interface name
1644
new_function = copy_function(nonmethod_func)
1511
if sys.version_info.major == 2:
1512
new_function = types.FunctionType(
1513
nonmethod_func.func_code,
1514
nonmethod_func.func_globals,
1515
nonmethod_func.func_name,
1516
nonmethod_func.func_defaults,
1517
nonmethod_func.func_closure)
1519
new_function = types.FunctionType(
1520
nonmethod_func.__code__,
1521
nonmethod_func.__globals__,
1522
nonmethod_func.__name__,
1523
nonmethod_func.__defaults__,
1524
nonmethod_func.__closure__)
1645
1525
new_function = (dbus.service.signal(
1647
1527
attribute._dbus_signature)(new_function))
1842
1729
"ApprovedByDefault")
1843
1730
approval_delay = notifychangeproperty(
1844
1731
dbus.UInt64, "ApprovalDelay",
1845
type_func=lambda td: td.total_seconds() * 1000)
1732
type_func = lambda td: td.total_seconds() * 1000)
1846
1733
approval_duration = notifychangeproperty(
1847
1734
dbus.UInt64, "ApprovalDuration",
1848
type_func=lambda td: td.total_seconds() * 1000)
1735
type_func = lambda td: td.total_seconds() * 1000)
1849
1736
host = notifychangeproperty(dbus.String, "Host")
1850
1737
timeout = notifychangeproperty(
1851
1738
dbus.UInt64, "Timeout",
1852
type_func=lambda td: td.total_seconds() * 1000)
1739
type_func = lambda td: td.total_seconds() * 1000)
1853
1740
extended_timeout = notifychangeproperty(
1854
1741
dbus.UInt64, "ExtendedTimeout",
1855
type_func=lambda td: td.total_seconds() * 1000)
1742
type_func = lambda td: td.total_seconds() * 1000)
1856
1743
interval = notifychangeproperty(
1857
1744
dbus.UInt64, "Interval",
1858
type_func=lambda td: td.total_seconds() * 1000)
1745
type_func = lambda td: td.total_seconds() * 1000)
1859
1746
checker_command = notifychangeproperty(dbus.String, "Checker")
1860
1747
secret = notifychangeproperty(dbus.ByteArray, "Secret",
1861
1748
invalidate_only=True)
1863
1750
del notifychangeproperty
1865
1752
def __del__(self, *args, **kwargs):
1867
1754
self.remove_from_connection()
2239
2119
class ClientHandler(socketserver.BaseRequestHandler, object):
2240
2120
"""A class to handle client connections.
2242
2122
Instantiated once for each connection to handle it.
2243
2123
Note: This will run in its own forked process."""
2245
2125
def handle(self):
2246
2126
with contextlib.closing(self.server.child_pipe) as child_pipe:
2247
2127
logger.info("TCP connection from: %s",
2248
2128
str(self.client_address))
2249
2129
logger.debug("Pipe FD: %d",
2250
2130
self.server.child_pipe.fileno())
2252
2132
session = gnutls.ClientSession(self.request)
2254
# priority = ':'.join(("NONE", "+VERS-TLS1.1",
2255
# "+AES-256-CBC", "+SHA1",
2256
# "+COMP-NULL", "+CTYPE-OPENPGP",
2134
#priority = ':'.join(("NONE", "+VERS-TLS1.1",
2135
# "+AES-256-CBC", "+SHA1",
2136
# "+COMP-NULL", "+CTYPE-OPENPGP",
2258
2138
# Use a fallback default, since this MUST be set.
2259
2139
priority = self.server.gnutls_priority
2260
2140
if priority is None:
2261
2141
priority = "NORMAL"
2262
gnutls.priority_set_direct(session._c_object,
2263
priority.encode("utf-8"),
2142
gnutls.priority_set_direct(session._c_object, priority,
2266
2145
# Start communication using the Mandos protocol
2267
2146
# Get protocol number
2268
2147
line = self.request.makefile().readline()
2416
2273
cert = cert_list[0]
2417
2274
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
2277
def fingerprint(openpgp):
2455
2278
"Convert an OpenPGP bytestring to a hexdigit fingerprint"
2490
2312
class MultiprocessingMixIn(object):
2491
2313
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2493
2315
def sub_process_main(self, request, address):
2495
2317
self.finish_request(request, address)
2496
2318
except Exception:
2497
2319
self.handle_error(request, address)
2498
2320
self.close_request(request)
2500
2322
def process_request(self, request, address):
2501
2323
"""Start a new process to process the request."""
2502
proc = multiprocessing.Process(target=self.sub_process_main,
2503
args=(request, address))
2324
proc = multiprocessing.Process(target = self.sub_process_main,
2325
args = (request, address))
2508
2330
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2509
2331
""" adds a pipe to the MixIn """
2511
2333
def process_request(self, request, client_address):
2512
2334
"""Overrides and wraps the original process_request().
2514
2336
This function creates a new pipe in self.pipe
2516
2338
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2518
2340
proc = MultiprocessingMixIn.process_request(self, request,
2519
2341
client_address)
2520
2342
self.child_pipe.close()
2521
2343
self.add_pipe(parent_pipe, proc)
2523
2345
def add_pipe(self, parent_pipe, proc):
2524
2346
"""Dummy function; override as necessary"""
2525
2347
raise NotImplementedError()
2574
2395
# socket_wrapper(), if socketfd was set.
2575
2396
socketserver.TCPServer.__init__(self, server_address,
2576
2397
RequestHandlerClass)
2578
2399
def server_bind(self):
2579
2400
"""This overrides the normal server_bind() function
2580
2401
to bind to an interface if one was specified, and also NOT to
2581
2402
bind to an address or port if they were not specified."""
2582
global SO_BINDTODEVICE
2583
2403
if self.interface is not None:
2584
2404
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)
2405
logger.error("SO_BINDTODEVICE does not exist;"
2406
" cannot bind to interface %s",
2410
self.socket.setsockopt(
2411
socket.SOL_SOCKET, SO_BINDTODEVICE,
2412
(self.interface + "\0").encode("utf-8"))
2413
except socket.error as error:
2414
if error.errno == errno.EPERM:
2415
logger.error("No permission to bind to"
2416
" interface %s", self.interface)
2417
elif error.errno == errno.ENOPROTOOPT:
2418
logger.error("SO_BINDTODEVICE not available;"
2419
" cannot bind to interface %s",
2421
elif error.errno == errno.ENODEV:
2422
logger.error("Interface %s does not exist,"
2423
" cannot bind", self.interface)
2606
2426
# Only bind(2) the socket if we really need to.
2607
2427
if self.server_address[0] or self.server_address[1]:
2608
if self.server_address[1]:
2609
self.allow_reuse_address = True
2610
2428
if not self.server_address[0]:
2611
2429
if self.address_family == socket.AF_INET6:
2612
any_address = "::" # in6addr_any
2430
any_address = "::" # in6addr_any
2614
any_address = "0.0.0.0" # INADDR_ANY
2432
any_address = "0.0.0.0" # INADDR_ANY
2615
2433
self.server_address = (any_address,
2616
2434
self.server_address[1])
2617
2435
elif not self.server_address[1]:
2651
2469
self.gnutls_priority = gnutls_priority
2652
2470
IPv6_TCPServer.__init__(self, server_address,
2653
2471
RequestHandlerClass,
2654
interface=interface,
2472
interface = interface,
2473
use_ipv6 = use_ipv6,
2474
socketfd = socketfd)
2658
2476
def server_activate(self):
2659
2477
if self.enabled:
2660
2478
return socketserver.TCPServer.server_activate(self)
2662
2480
def enable(self):
2663
2481
self.enabled = True
2665
2483
def add_pipe(self, parent_pipe, proc):
2666
2484
# Call "handle_ipc" for both data and EOF events
2485
gobject.io_add_watch(
2668
2486
parent_pipe.fileno(),
2669
GLib.IO_IN | GLib.IO_HUP,
2487
gobject.IO_IN | gobject.IO_HUP,
2670
2488
functools.partial(self.handle_ipc,
2671
parent_pipe=parent_pipe,
2489
parent_pipe = parent_pipe,
2674
2492
def handle_ipc(self, source, condition,
2675
2493
parent_pipe=None,
2677
2495
client_object=None):
2678
2496
# error, or the other end of multiprocessing.Pipe has closed
2679
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2497
if condition & (gobject.IO_ERR | gobject.IO_HUP):
2680
2498
# Wait for other process to exit
2684
2502
# Read a request from the child
2685
2503
request = parent_pipe.recv()
2686
2504
command = request[0]
2688
2506
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:
2508
address = request[2]
2510
for c in self.clients.itervalues():
2511
if c.fingerprint == fpr:
2703
logger.info("Client not found for key ID: %s, address"
2704
": %s", key_id or fpr, address)
2515
logger.info("Client not found for fingerprint: %s, ad"
2516
"dress: %s", fpr, address)
2705
2517
if self.use_dbus:
2706
2518
# Emit D-Bus signal
2707
mandos_dbus_service.ClientNotFound(key_id or fpr,
2519
mandos_dbus_service.ClientNotFound(fpr,
2709
2521
parent_pipe.send(False)
2524
gobject.io_add_watch(
2713
2525
parent_pipe.fileno(),
2714
GLib.IO_IN | GLib.IO_HUP,
2526
gobject.IO_IN | gobject.IO_HUP,
2715
2527
functools.partial(self.handle_ipc,
2716
parent_pipe=parent_pipe,
2718
client_object=client))
2528
parent_pipe = parent_pipe,
2530
client_object = client))
2719
2531
parent_pipe.send(True)
2720
2532
# remove the old hook in favor of the new above hook on
3127
logger.debug("Did setuid/setgid to {}:{}".format(uid,
3129
2927
except OSError as error:
3130
logger.warning("Failed to setuid/setgid to {}:{}: {}"
3131
.format(uid, gid, os.strerror(error.errno)))
3132
2928
if error.errno != errno.EPERM:
3136
2932
# Enable all possible GnuTLS debugging
3138
2934
# "Use a log level over 10 to enable all debugging options."
3139
2935
# - GnuTLS manual
3140
2936
gnutls.global_set_log_level(11)
3142
2938
@gnutls.log_func
3143
2939
def debug_gnutls(level, string):
3144
2940
logger.debug("GnuTLS: %s", string[:-1])
3146
2942
gnutls.global_set_log_function(debug_gnutls)
3148
2944
# Redirect stdin so all checkers get /dev/null
3149
2945
null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
3150
2946
os.dup2(null, sys.stdin.fileno())
3154
2950
# Need to fork before connecting to D-Bus
3155
2951
if not foreground:
3156
2952
# 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.
2955
# multiprocessing will use threads, so before we use gobject we
2956
# need to inform gobject that threads will be used.
2957
gobject.threads_init()
3163
2959
global main_loop
3164
2960
# From the Avahi example code
3165
2961
DBusGMainLoop(set_as_default=True)
3166
main_loop = GLib.MainLoop()
2962
main_loop = gobject.MainLoop()
3167
2963
bus = dbus.SystemBus()
3168
2964
# End of Avahi example code
3183
2979
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
3184
2980
service = AvahiServiceToSyslog(
3185
name=server_settings["servicename"],
3186
servicetype="_mandos._tcp",
2981
name = server_settings["servicename"],
2982
servicetype = "_mandos._tcp",
2983
protocol = protocol,
3189
2985
if server_settings["interface"]:
3190
2986
service.interface = if_nametoindex(
3191
2987
server_settings["interface"].encode("utf-8"))
3193
2989
global multiprocessing_manager
3194
2990
multiprocessing_manager = multiprocessing.Manager()
3196
2992
client_class = Client
3198
client_class = functools.partial(ClientDBus, bus=bus)
2994
client_class = functools.partial(ClientDBus, bus = bus)
3200
2996
client_settings = Client.config_parser(client_config)
3201
2997
old_client_settings = {}
3202
2998
clients_data = {}
3204
3000
# This is used to redirect stdout and stderr for checker processes
3206
wnull = open(os.devnull, "w") # A writable /dev/null
3002
wnull = open(os.devnull, "w") # A writable /dev/null
3207
3003
# Only used if server is running in foreground but not in debug
3209
3005
if debug or not foreground:
3212
3008
# Get client data and settings from last running state.
3213
3009
if server_settings["restore"]:
3215
3011
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"]
3012
clients_data, old_client_settings = pickle.load(
3265
3014
os.remove(stored_state_path)
3266
3015
except IOError as e:
3267
3016
if e.errno == errno.ENOENT:
3367
3116
pidfilename, pid)
3369
3118
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)
3120
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
3121
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
3377
3125
@alternate_dbus_interfaces(
3378
{"se.recompile.Mandos": "se.bsnet.fukt.Mandos"})
3126
{ "se.recompile.Mandos": "se.bsnet.fukt.Mandos" })
3379
3127
class MandosDBusService(DBusObjectWithObjectManager):
3380
3128
"""A D-Bus proxy object"""
3382
3130
def __init__(self):
3383
3131
dbus.service.Object.__init__(self, bus, "/")
3385
3133
_interface = "se.recompile.Mandos"
3387
3135
@dbus.service.signal(_interface, signature="o")
3388
3136
def ClientAdded(self, objpath):
3392
3140
@dbus.service.signal(_interface, signature="ss")
3393
def ClientNotFound(self, key_id, address):
3141
def ClientNotFound(self, fingerprint, address):
3397
3145
@dbus_annotations({"org.freedesktop.DBus.Deprecated":
3399
3147
@dbus.service.signal(_interface, signature="os")
3400
3148
def ClientRemoved(self, objpath, name):
3404
3152
@dbus_annotations({"org.freedesktop.DBus.Deprecated":
3406
3154
@dbus.service.method(_interface, out_signature="ao")
3407
3155
def GetAllClients(self):
3409
3157
return dbus.Array(c.dbus_object_path for c in
3410
tcp_server.clients.values())
3158
tcp_server.clients.itervalues())
3412
3160
@dbus_annotations({"org.freedesktop.DBus.Deprecated":
3414
3162
@dbus.service.method(_interface,
3473
3221
self.ClientRemoved(client.dbus_object_path,
3476
3224
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
3227
"Cleanup function; run on exit"
3486
3229
service.cleanup()
3488
mp.active_children()
3231
multiprocessing.active_children()
3490
3233
if not (tcp_server.clients or client_settings):
3493
3236
# Store client before exiting. Secrets are encrypted with key
3494
3237
# based on what config file has. If config file is
3495
3238
# removed/edited, old secret will thus be unrecovable.
3497
3240
with PGPEngine() as pgp:
3498
for client in tcp_server.clients.values():
3241
for client in tcp_server.clients.itervalues():
3499
3242
key = client_settings[client.name]["secret"]
3500
3243
client.encrypted_secret = pgp.encrypt(client.secret,
3502
3245
client_dict = {}
3504
3247
# A list of attributes that can not be pickled
3506
exclude = {"bus", "changedstate", "secret",
3507
"checker", "server_settings"}
3249
exclude = { "bus", "changedstate", "secret",
3250
"checker", "server_settings" }
3508
3251
for name, typ in inspect.getmembers(dbus.service
3510
3253
exclude.add(name)
3512
3255
client_dict["encrypted_secret"] = (client
3513
3256
.encrypted_secret)
3514
3257
for attr in client.client_structure:
3515
3258
if attr not in exclude:
3516
3259
client_dict[attr] = getattr(client, attr)
3518
3261
clients[client.name] = client_dict
3519
3262
del client_settings[client.name]["secret"]
3522
3265
with tempfile.NamedTemporaryFile(