2
# -*- mode: python; coding: utf-8 -*-
2
# -*- mode: python; after-save-hook: (lambda () (let ((command (if (fboundp 'file-local-name) (file-local-name (buffer-file-name)) (or (file-remote-p (buffer-file-name) 'localname) (buffer-file-name))))) (if (= (progn (if (get-buffer "*Test*") (kill-buffer "*Test*")) (process-file-shell-command (format "%s --check" (shell-quote-argument command)) nil "*Test*")) 0) (let ((w (get-buffer-window "*Test*"))) (if w (delete-window w))) (progn (with-current-buffer "*Test*" (compilation-mode)) (display-buffer "*Test*" '(display-buffer-in-side-window)))))); coding: utf-8 -*-
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-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
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
19
21
# the Free Software Foundation, either version 3 of the License, or
20
22
# (at your option) any later version.
22
# This program is distributed in the hope that it will be useful,
23
# but WITHOUT ANY WARRANTY; without even the implied warranty of
24
# Mandos is distributed in the hope that it will be useful, but
25
# WITHOUT ANY WARRANTY; without even the implied warranty of
24
26
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
25
27
# GNU General Public License for more details.
27
29
# You should have received a copy of the GNU General Public License
28
# along with this program. If not, see
29
# <http://www.gnu.org/licenses/>.
30
# along with Mandos. If not, see <http://www.gnu.org/licenses/>.
31
32
# Contact the authors at <mandos@recompile.se>.
34
35
from __future__ import (division, absolute_import, print_function,
246
269
'--passphrase-file',
248
271
+ self.gnupgargs,
249
stdin = subprocess.PIPE,
250
stdout = subprocess.PIPE,
251
stderr = subprocess.PIPE)
252
ciphertext, err = proc.communicate(input = data)
272
stdin=subprocess.PIPE,
273
stdout=subprocess.PIPE,
274
stderr=subprocess.PIPE)
275
ciphertext, err = proc.communicate(input=data)
253
276
if proc.returncode != 0:
254
277
raise PGPError(err)
255
278
return ciphertext
257
280
def decrypt(self, data, password):
258
281
passphrase = self.password_encode(password)
259
282
with tempfile.NamedTemporaryFile(
260
dir = self.tempdir) as passfile:
283
dir=self.tempdir) as passfile:
261
284
passfile.write(passphrase)
263
286
proc = subprocess.Popen([self.gpg, '--decrypt',
264
287
'--passphrase-file',
266
289
+ self.gnupgargs,
267
stdin = subprocess.PIPE,
268
stdout = subprocess.PIPE,
269
stderr = subprocess.PIPE)
270
decrypted_plaintext, err = proc.communicate(input = data)
290
stdin=subprocess.PIPE,
291
stdout=subprocess.PIPE,
292
stderr=subprocess.PIPE)
293
decrypted_plaintext, err = proc.communicate(input=data)
271
294
if proc.returncode != 0:
272
295
raise PGPError(err)
273
296
return decrypted_plaintext
275
299
# Pretend that we have an Avahi module
277
"""This isn't so much a class as it is a module-like namespace.
278
It is instantiated once, and simulates having an Avahi module."""
279
IF_UNSPEC = -1 # avahi-common/address.h
280
PROTO_UNSPEC = -1 # avahi-common/address.h
281
PROTO_INET = 0 # avahi-common/address.h
282
PROTO_INET6 = 1 # avahi-common/address.h
301
"""This isn't so much a class as it is a module-like namespace."""
302
IF_UNSPEC = -1 # avahi-common/address.h
303
PROTO_UNSPEC = -1 # avahi-common/address.h
304
PROTO_INET = 0 # avahi-common/address.h
305
PROTO_INET6 = 1 # avahi-common/address.h
283
306
DBUS_NAME = "org.freedesktop.Avahi"
284
307
DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
285
308
DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
286
309
DBUS_PATH_SERVER = "/"
287
def string_array_to_txt_array(self, t):
312
def string_array_to_txt_array(t):
288
313
return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
289
314
for s in t), signature="ay")
290
ENTRY_GROUP_ESTABLISHED = 2 # avahi-common/defs.h
291
ENTRY_GROUP_COLLISION = 3 # avahi-common/defs.h
292
ENTRY_GROUP_FAILURE = 4 # avahi-common/defs.h
293
SERVER_INVALID = 0 # avahi-common/defs.h
294
SERVER_REGISTERING = 1 # avahi-common/defs.h
295
SERVER_RUNNING = 2 # avahi-common/defs.h
296
SERVER_COLLISION = 3 # avahi-common/defs.h
297
SERVER_FAILURE = 4 # avahi-common/defs.h
315
ENTRY_GROUP_ESTABLISHED = 2 # avahi-common/defs.h
316
ENTRY_GROUP_COLLISION = 3 # avahi-common/defs.h
317
ENTRY_GROUP_FAILURE = 4 # avahi-common/defs.h
318
SERVER_INVALID = 0 # avahi-common/defs.h
319
SERVER_REGISTERING = 1 # avahi-common/defs.h
320
SERVER_RUNNING = 2 # avahi-common/defs.h
321
SERVER_COLLISION = 3 # avahi-common/defs.h
322
SERVER_FAILURE = 4 # avahi-common/defs.h
300
325
class AvahiError(Exception):
301
326
def __init__(self, value, *args, **kwargs):
492
517
class AvahiServiceToSyslog(AvahiService):
493
518
def rename(self, *args, **kwargs):
494
519
"""Add the new name to the syslog messages"""
495
ret = AvahiService.rename(self, *args, **kwargs)
520
ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
496
521
syslogger.setFormatter(logging.Formatter(
497
522
'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
498
523
.format(self.name)))
501
527
# Pretend that we have a GnuTLS module
502
class GnuTLS(object):
503
"""This isn't so much a class as it is a module-like namespace.
504
It is instantiated once, and simulates having a GnuTLS module."""
506
_library = ctypes.cdll.LoadLibrary(
507
ctypes.util.find_library("gnutls"))
508
_need_version = b"3.3.0"
510
# Need to use class name "GnuTLS" here, since this method is
511
# called before the assignment to the "gnutls" global variable
513
if GnuTLS.check_version(self._need_version) is None:
514
raise GnuTLS.Error("Needs GnuTLS {} or later"
515
.format(self._need_version))
529
"""This isn't so much a class as it is a module-like namespace."""
531
library = ctypes.util.find_library("gnutls")
533
library = ctypes.util.find_library("gnutls-deb0")
534
_library = ctypes.cdll.LoadLibrary(library)
517
537
# Unless otherwise indicated, the constants and types below are
518
538
# all from the gnutls/gnutls.h C header file.
522
542
E_INTERRUPTED = -52
527
548
CRD_CERTIFICATE = 1
528
549
E_NO_CERTIFICATE_FOUND = -49
554
KEYID_USE_SHA256 = 1 # gnutls/x509.h
529
555
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
532
558
class session_int(ctypes.Structure):
534
560
session_t = ctypes.POINTER(session_int)
535
562
class certificate_credentials_st(ctypes.Structure):
537
564
certificate_credentials_t = ctypes.POINTER(
538
565
certificate_credentials_st)
539
566
certificate_type_t = ctypes.c_int
540
568
class datum_t(ctypes.Structure):
541
569
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
542
570
('size', ctypes.c_uint)]
543
572
class openpgp_crt_int(ctypes.Structure):
545
574
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
546
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
575
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
547
576
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
548
577
credentials_type_t = ctypes.c_int
549
578
transport_ptr_t = ctypes.c_void_p
550
579
close_request_t = ctypes.c_int
553
582
class Error(Exception):
554
# We need to use the class name "GnuTLS" here, since this
555
# exception might be raised from within GnuTLS.__init__,
556
# which is called before the assignment to the "gnutls"
557
# global variable has happened.
558
def __init__(self, message = None, code = None, args=()):
583
def __init__(self, message=None, code=None, args=()):
559
584
# Default usage is by a message string, but if a return
560
585
# code is passed, convert it to a string with
561
586
# gnutls.strerror()
563
588
if message is None and code is not None:
564
message = GnuTLS.strerror(code)
565
return super(GnuTLS.Error, self).__init__(
589
message = gnutls.strerror(code)
590
return super(gnutls.Error, self).__init__(
568
593
class CertificateSecurityError(Error):
572
class Credentials(object):
573
598
def __init__(self):
574
599
self._c_object = gnutls.certificate_credentials_t()
575
600
gnutls.certificate_allocate_credentials(
576
601
ctypes.byref(self._c_object))
577
602
self.type = gnutls.CRD_CERTIFICATE
579
604
def __del__(self):
580
605
gnutls.certificate_free_credentials(self._c_object)
582
class ClientSession(object):
583
def __init__(self, socket, credentials = None):
608
def __init__(self, socket, credentials=None):
584
609
self._c_object = gnutls.session_t()
585
gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)
610
gnutls_flags = gnutls.CLIENT
611
if gnutls.check_version(b"3.5.6"):
612
gnutls_flags |= gnutls.NO_TICKETS
614
gnutls_flags |= gnutls.ENABLE_RAWPK
615
gnutls.init(ctypes.byref(self._c_object), gnutls_flags)
586
617
gnutls.set_default_priority(self._c_object)
587
618
gnutls.transport_set_ptr(self._c_object, socket.fileno())
588
619
gnutls.handshake_set_private_extensions(self._c_object,
630
661
return _error_code(result)
631
662
result = func(*arguments)
634
665
# Unless otherwise indicated, the function declarations below are
635
666
# all from the gnutls/gnutls.h C header file.
638
669
priority_set_direct = _library.gnutls_priority_set_direct
639
670
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
640
671
ctypes.POINTER(ctypes.c_char_p)]
641
672
priority_set_direct.restype = _error_code
643
674
init = _library.gnutls_init
644
675
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
645
676
init.restype = _error_code
647
678
set_default_priority = _library.gnutls_set_default_priority
648
679
set_default_priority.argtypes = [session_t]
649
680
set_default_priority.restype = _error_code
651
682
record_send = _library.gnutls_record_send
652
683
record_send.argtypes = [session_t, ctypes.c_void_p,
654
685
record_send.restype = ctypes.c_ssize_t
655
686
record_send.errcheck = _retry_on_error
657
688
certificate_allocate_credentials = (
658
689
_library.gnutls_certificate_allocate_credentials)
659
690
certificate_allocate_credentials.argtypes = [
660
691
ctypes.POINTER(certificate_credentials_t)]
661
692
certificate_allocate_credentials.restype = _error_code
663
694
certificate_free_credentials = (
664
695
_library.gnutls_certificate_free_credentials)
665
certificate_free_credentials.argtypes = [certificate_credentials_t]
696
certificate_free_credentials.argtypes = [
697
certificate_credentials_t]
666
698
certificate_free_credentials.restype = None
668
700
handshake_set_private_extensions = (
669
701
_library.gnutls_handshake_set_private_extensions)
670
702
handshake_set_private_extensions.argtypes = [session_t,
672
704
handshake_set_private_extensions.restype = None
674
706
credentials_set = _library.gnutls_credentials_set
675
707
credentials_set.argtypes = [session_t, credentials_type_t,
677
709
credentials_set.restype = _error_code
679
711
strerror = _library.gnutls_strerror
680
712
strerror.argtypes = [ctypes.c_int]
681
713
strerror.restype = ctypes.c_char_p
683
715
certificate_type_get = _library.gnutls_certificate_type_get
684
716
certificate_type_get.argtypes = [session_t]
685
717
certificate_type_get.restype = _error_code
687
719
certificate_get_peers = _library.gnutls_certificate_get_peers
688
720
certificate_get_peers.argtypes = [session_t,
689
721
ctypes.POINTER(ctypes.c_uint)]
690
722
certificate_get_peers.restype = ctypes.POINTER(datum_t)
692
724
global_set_log_level = _library.gnutls_global_set_log_level
693
725
global_set_log_level.argtypes = [ctypes.c_int]
694
726
global_set_log_level.restype = None
696
728
global_set_log_function = _library.gnutls_global_set_log_function
697
729
global_set_log_function.argtypes = [log_func]
698
730
global_set_log_function.restype = None
700
732
deinit = _library.gnutls_deinit
701
733
deinit.argtypes = [session_t]
702
734
deinit.restype = None
704
736
handshake = _library.gnutls_handshake
705
737
handshake.argtypes = [session_t]
706
738
handshake.restype = _error_code
707
739
handshake.errcheck = _retry_on_error
709
741
transport_set_ptr = _library.gnutls_transport_set_ptr
710
742
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
711
743
transport_set_ptr.restype = None
713
745
bye = _library.gnutls_bye
714
746
bye.argtypes = [session_t, close_request_t]
715
747
bye.restype = _error_code
716
748
bye.errcheck = _retry_on_error
718
750
check_version = _library.gnutls_check_version
719
751
check_version.argtypes = [ctypes.c_char_p]
720
752
check_version.restype = ctypes.c_char_p
722
# All the function declarations below are from gnutls/openpgp.h
724
openpgp_crt_init = _library.gnutls_openpgp_crt_init
725
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
726
openpgp_crt_init.restype = _error_code
728
openpgp_crt_import = _library.gnutls_openpgp_crt_import
729
openpgp_crt_import.argtypes = [openpgp_crt_t,
730
ctypes.POINTER(datum_t),
732
openpgp_crt_import.restype = _error_code
734
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
735
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
736
ctypes.POINTER(ctypes.c_uint)]
737
openpgp_crt_verify_self.restype = _error_code
739
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
740
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
741
openpgp_crt_deinit.restype = None
743
openpgp_crt_get_fingerprint = (
744
_library.gnutls_openpgp_crt_get_fingerprint)
745
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
749
openpgp_crt_get_fingerprint.restype = _error_code
754
_need_version = b"3.3.0"
755
if check_version(_need_version) is None:
756
raise self.Error("Needs GnuTLS {} or later"
757
.format(_need_version))
759
_tls_rawpk_version = b"3.6.6"
760
has_rawpk = bool(check_version(_tls_rawpk_version))
764
class pubkey_st(ctypes.Structure):
766
pubkey_t = ctypes.POINTER(pubkey_st)
768
x509_crt_fmt_t = ctypes.c_int
770
# All the function declarations below are from gnutls/abstract.h
771
pubkey_init = _library.gnutls_pubkey_init
772
pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
773
pubkey_init.restype = _error_code
775
pubkey_import = _library.gnutls_pubkey_import
776
pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
778
pubkey_import.restype = _error_code
780
pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
781
pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
782
ctypes.POINTER(ctypes.c_ubyte),
783
ctypes.POINTER(ctypes.c_size_t)]
784
pubkey_get_key_id.restype = _error_code
786
pubkey_deinit = _library.gnutls_pubkey_deinit
787
pubkey_deinit.argtypes = [pubkey_t]
788
pubkey_deinit.restype = None
790
# All the function declarations below are from gnutls/openpgp.h
792
openpgp_crt_init = _library.gnutls_openpgp_crt_init
793
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
794
openpgp_crt_init.restype = _error_code
796
openpgp_crt_import = _library.gnutls_openpgp_crt_import
797
openpgp_crt_import.argtypes = [openpgp_crt_t,
798
ctypes.POINTER(datum_t),
800
openpgp_crt_import.restype = _error_code
802
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
803
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
804
ctypes.POINTER(ctypes.c_uint)]
805
openpgp_crt_verify_self.restype = _error_code
807
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
808
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
809
openpgp_crt_deinit.restype = None
811
openpgp_crt_get_fingerprint = (
812
_library.gnutls_openpgp_crt_get_fingerprint)
813
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
817
openpgp_crt_get_fingerprint.restype = _error_code
819
if check_version(b"3.6.4"):
820
certificate_type_get2 = _library.gnutls_certificate_type_get2
821
certificate_type_get2.argtypes = [session_t, ctypes.c_int]
822
certificate_type_get2.restype = _error_code
751
824
# Remove non-public functions
752
825
del _error_code, _retry_on_error
753
# Create the global "gnutls" object, simulating a module
756
828
def call_pipe(connection, # : multiprocessing.Connection
757
829
func, *args, **kwargs):
758
830
"""This function is meant to be called by multiprocessing.Process
760
832
This function runs func(*args, **kwargs), and writes the resulting
761
833
return value on the provided multiprocessing.Connection.
763
835
connection.send(func(*args, **kwargs))
764
836
connection.close()
766
class Client(object):
767
840
"""A representation of a client host served by this server.
770
843
approved: bool(); 'None' if not yet approved/disapproved
771
844
approval_delay: datetime.timedelta(); Time to wait for approval
772
845
approval_duration: datetime.timedelta(); Duration of one approval
773
checker: subprocess.Popen(); a running checker process used
774
to see if the client lives.
775
'None' if no process is running.
846
checker: multiprocessing.Process(); a running checker process used
847
to see if the client lives. 'None' if no process is
776
849
checker_callback_tag: a GLib event source tag, or None
777
850
checker_command: string; External command which is run to check
778
851
if client lives. %() expansions are done at
1447
1528
exc_info=error)
1448
1529
return xmlstring
1451
1533
dbus.OBJECT_MANAGER_IFACE
1452
1534
except AttributeError:
1453
1535
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
1455
1538
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
1456
1539
"""A D-Bus object with an ObjectManager.
1458
1541
Classes inheriting from this exposes the standard
1459
1542
GetManagedObjects call and the InterfacesAdded and
1460
1543
InterfacesRemoved signals on the standard
1461
1544
"org.freedesktop.DBus.ObjectManager" interface.
1463
1546
Note: No signals are sent automatically; they must be sent
1466
1549
@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
1467
out_signature = "a{oa{sa{sv}}}")
1550
out_signature="a{oa{sa{sv}}}")
1468
1551
def GetManagedObjects(self):
1469
1552
"""This function must be overridden"""
1470
1553
raise NotImplementedError()
1472
1555
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
1473
signature = "oa{sa{sv}}")
1556
signature="oa{sa{sv}}")
1474
1557
def InterfacesAdded(self, object_path, interfaces_and_properties):
1477
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
1560
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
1478
1561
def InterfacesRemoved(self, object_path, interfaces):
1481
1564
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
1482
out_signature = "s",
1483
path_keyword = 'object_path',
1484
connection_keyword = 'connection')
1566
path_keyword='object_path',
1567
connection_keyword='connection')
1485
1568
def Introspect(self, object_path, connection):
1486
1569
"""Overloading of standard D-Bus method.
1488
1571
Override return argument name of GetManagedObjects to be
1489
1572
"objpath_interfaces_and_properties"
2365
class MultiprocessingMixIn(object):
2515
class MultiprocessingMixIn:
2366
2516
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2368
2518
def sub_process_main(self, request, address):
2370
2520
self.finish_request(request, address)
2371
2521
except Exception:
2372
2522
self.handle_error(request, address)
2373
2523
self.close_request(request)
2375
2525
def process_request(self, request, address):
2376
2526
"""Start a new process to process the request."""
2377
proc = multiprocessing.Process(target = self.sub_process_main,
2378
args = (request, address))
2527
proc = multiprocessing.Process(target=self.sub_process_main,
2528
args=(request, address))
2383
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2533
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
2384
2534
""" adds a pipe to the MixIn """
2386
2536
def process_request(self, request, client_address):
2387
2537
"""Overrides and wraps the original process_request().
2389
2539
This function creates a new pipe in self.pipe
2391
2541
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2393
2543
proc = MultiprocessingMixIn.process_request(self, request,
2394
2544
client_address)
2395
2545
self.child_pipe.close()
2396
2546
self.add_pipe(parent_pipe, proc)
2398
2548
def add_pipe(self, parent_pipe, proc):
2399
2549
"""Dummy function; override as necessary"""
2400
2550
raise NotImplementedError()
2403
2553
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
2404
socketserver.TCPServer, object):
2554
socketserver.TCPServer):
2405
2555
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
2408
2558
enabled: Boolean; whether this server is activated yet
2409
2559
interface: None or a network interface name (string)
2410
2560
use_ipv6: Boolean; to use IPv6 or not
2413
2563
def __init__(self, server_address, RequestHandlerClass,
2414
2564
interface=None,
2523
2676
self.gnutls_priority = gnutls_priority
2524
2677
IPv6_TCPServer.__init__(self, server_address,
2525
2678
RequestHandlerClass,
2526
interface = interface,
2527
use_ipv6 = use_ipv6,
2528
socketfd = socketfd)
2679
interface=interface,
2530
2683
def server_activate(self):
2531
2684
if self.enabled:
2532
2685
return socketserver.TCPServer.server_activate(self)
2534
2687
def enable(self):
2535
2688
self.enabled = True
2537
2690
def add_pipe(self, parent_pipe, proc):
2538
2691
# Call "handle_ipc" for both data and EOF events
2539
2692
GLib.io_add_watch(
2540
parent_pipe.fileno(),
2541
GLib.IO_IN | GLib.IO_HUP,
2693
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2694
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2542
2695
functools.partial(self.handle_ipc,
2543
parent_pipe = parent_pipe,
2696
parent_pipe=parent_pipe,
2546
2699
def handle_ipc(self, source, condition,
2547
2700
parent_pipe=None,
2549
2702
client_object=None):
2550
2703
# error, or the other end of multiprocessing.Pipe has closed
2551
2704
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2552
2705
# Wait for other process to exit
2556
2709
# Read a request from the child
2557
2710
request = parent_pipe.recv()
2558
2711
command = request[0]
2560
2713
if command == 'init':
2562
address = request[2]
2714
key_id = request[1].decode("ascii")
2715
fpr = request[2].decode("ascii")
2716
address = request[3]
2564
2718
for c in self.clients.values():
2565
if c.fingerprint == fpr:
2719
if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
2721
if key_id and c.key_id == key_id:
2724
if fpr and c.fingerprint == fpr:
2569
logger.info("Client not found for fingerprint: %s, ad"
2570
"dress: %s", fpr, address)
2728
logger.info("Client not found for key ID: %s, address"
2729
": %s", key_id or fpr, address)
2571
2730
if self.use_dbus:
2572
2731
# Emit D-Bus signal
2573
mandos_dbus_service.ClientNotFound(fpr,
2732
mandos_dbus_service.ClientNotFound(key_id or fpr,
2575
2734
parent_pipe.send(False)
2578
2737
GLib.io_add_watch(
2579
parent_pipe.fileno(),
2580
GLib.IO_IN | GLib.IO_HUP,
2738
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2739
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2581
2740
functools.partial(self.handle_ipc,
2582
parent_pipe = parent_pipe,
2584
client_object = client))
2741
parent_pipe=parent_pipe,
2743
client_object=client))
2585
2744
parent_pipe.send(True)
2586
2745
# remove the old hook in favor of the new above hook on
2590
2749
funcname = request[1]
2591
2750
args = request[2]
2592
2751
kwargs = request[3]
2594
2753
parent_pipe.send(('data', getattr(client_object,
2595
2754
funcname)(*args,
2598
2757
if command == 'getattr':
2599
2758
attrname = request[1]
2600
2759
if isinstance(client_object.__getattribute__(attrname),
2601
collections.Callable):
2760
collections.abc.Callable):
2602
2761
parent_pipe.send(('function', ))
2604
2763
parent_pipe.send((
2605
2764
'data', client_object.__getattribute__(attrname)))
2607
2766
if command == 'setattr':
2608
2767
attrname = request[1]
2609
2768
value = request[2]
2610
2769
setattr(client_object, attrname, value)
2615
2774
def rfc3339_duration_to_delta(duration):
2616
2775
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
2618
>>> rfc3339_duration_to_delta("P7D")
2619
datetime.timedelta(7)
2620
>>> rfc3339_duration_to_delta("PT60S")
2621
datetime.timedelta(0, 60)
2622
>>> rfc3339_duration_to_delta("PT60M")
2623
datetime.timedelta(0, 3600)
2624
>>> rfc3339_duration_to_delta("PT24H")
2625
datetime.timedelta(1)
2626
>>> rfc3339_duration_to_delta("P1W")
2627
datetime.timedelta(7)
2628
>>> rfc3339_duration_to_delta("PT5M30S")
2629
datetime.timedelta(0, 330)
2630
>>> rfc3339_duration_to_delta("P1DT3M20S")
2631
datetime.timedelta(1, 200)
2777
>>> rfc3339_duration_to_delta("P7D") == datetime.timedelta(7)
2779
>>> rfc3339_duration_to_delta("PT60S") == datetime.timedelta(0, 60)
2781
>>> rfc3339_duration_to_delta("PT60M") == datetime.timedelta(0, 3600)
2783
>>> rfc3339_duration_to_delta("PT24H") == datetime.timedelta(1)
2785
>>> rfc3339_duration_to_delta("P1W") == datetime.timedelta(7)
2787
>>> rfc3339_duration_to_delta("PT5M30S") == datetime.timedelta(0, 330)
2789
>>> rfc3339_duration_to_delta("P1DT3M20S") == datetime.timedelta(1, 200)
2634
2793
# Parsing an RFC 3339 duration with regular expressions is not
2635
2794
# possible - there would have to be multiple places for the same
2636
2795
# values, like seconds. The current code, while more esoteric, is
2637
2796
# cleaner without depending on a parsing library. If Python had a
2638
2797
# built-in library for parsing we would use it, but we'd like to
2639
2798
# avoid excessive use of external libraries.
2641
2800
# New type for defining tokens, syntax, and semantics all-in-one
2642
2801
Token = collections.namedtuple("Token", (
2643
2802
"regexp", # To match token; if "value" is not None, must have
2827
2989
parser.add_argument("--no-zeroconf", action="store_false",
2828
2990
dest="zeroconf", help="Do not use Zeroconf",
2831
2993
options = parser.parse_args()
2835
fail_count, test_count = doctest.testmod()
2836
sys.exit(os.EX_OK if fail_count == 0 else 1)
2838
2995
# Default values for config file for server-global settings
2839
server_defaults = { "interface": "",
2844
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
2845
":+SIGN-DSA-SHA256",
2846
"servicename": "Mandos",
2852
"statedir": "/var/lib/mandos",
2853
"foreground": "False",
2996
if gnutls.has_rawpk:
2997
priority = ("SECURE128:!CTYPE-X.509:+CTYPE-RAWPK:!RSA"
2998
":!VERS-ALL:+VERS-TLS1.3:%PROFILE_ULTRA")
3000
priority = ("SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
3001
":+SIGN-DSA-SHA256")
3002
server_defaults = {"interface": "",
3006
"priority": priority,
3007
"servicename": "Mandos",
3013
"statedir": "/var/lib/mandos",
3014
"foreground": "False",
2857
3019
# Parse config file for server-global settings
2858
server_config = configparser.SafeConfigParser(server_defaults)
3020
server_config = configparser.ConfigParser(server_defaults)
2859
3021
del server_defaults
2860
3022
server_config.read(os.path.join(options.configdir, "mandos.conf"))
2861
# Convert the SafeConfigParser object to a dict
3023
# Convert the ConfigParser object to a dict
2862
3024
server_settings = server_config.defaults()
2863
3025
# Use the appropriate methods on the non-string config options
2864
for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
3026
for option in ("debug", "use_dbus", "use_ipv6", "restore",
3027
"foreground", "zeroconf"):
2865
3028
server_settings[option] = server_config.getboolean("DEFAULT",
2867
3030
if server_settings["port"]:
3040
3203
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
3041
3204
service = AvahiServiceToSyslog(
3042
name = server_settings["servicename"],
3043
servicetype = "_mandos._tcp",
3044
protocol = protocol,
3205
name=server_settings["servicename"],
3206
servicetype="_mandos._tcp",
3046
3209
if server_settings["interface"]:
3047
3210
service.interface = if_nametoindex(
3048
3211
server_settings["interface"].encode("utf-8"))
3050
3213
global multiprocessing_manager
3051
3214
multiprocessing_manager = multiprocessing.Manager()
3053
3216
client_class = Client
3055
client_class = functools.partial(ClientDBus, bus = bus)
3218
client_class = functools.partial(ClientDBus, bus=bus)
3057
3220
client_settings = Client.config_parser(client_config)
3058
3221
old_client_settings = {}
3059
3222
clients_data = {}
3061
3224
# This is used to redirect stdout and stderr for checker processes
3063
wnull = open(os.devnull, "w") # A writable /dev/null
3226
wnull = open(os.devnull, "w") # A writable /dev/null
3064
3227
# Only used if server is running in foreground but not in debug
3066
3229
if debug or not foreground:
3069
3232
# Get client data and settings from last running state.
3070
3233
if server_settings["restore"]:
3072
3235
with open(stored_state_path, "rb") as stored_state:
3073
if sys.version_info.major == 2:
3236
if sys.version_info.major == 2:
3074
3237
clients_data, old_client_settings = pickle.load(
3077
3240
bytes_clients_data, bytes_old_client_settings = (
3078
pickle.load(stored_state, encoding = "bytes"))
3079
### Fix bytes to strings
3241
pickle.load(stored_state, encoding="bytes"))
3242
# Fix bytes to strings
3082
clients_data = { (key.decode("utf-8")
3083
if isinstance(key, bytes)
3086
bytes_clients_data.items() }
3245
clients_data = {(key.decode("utf-8")
3246
if isinstance(key, bytes)
3249
bytes_clients_data.items()}
3087
3250
del bytes_clients_data
3088
3251
for key in clients_data:
3089
value = { (k.decode("utf-8")
3090
if isinstance(k, bytes) else k): v
3092
clients_data[key].items() }
3252
value = {(k.decode("utf-8")
3253
if isinstance(k, bytes) else k): v
3255
clients_data[key].items()}
3093
3256
clients_data[key] = value
3094
3257
# .client_structure
3095
3258
value["client_structure"] = [
3096
3259
(s.decode("utf-8")
3097
3260
if isinstance(s, bytes)
3098
3261
else s) for s in
3099
value["client_structure"] ]
3101
for k in ("name", "host"):
3262
value["client_structure"]]
3263
# .name, .host, and .checker_command
3264
for k in ("name", "host", "checker_command"):
3102
3265
if isinstance(value[k], bytes):
3103
3266
value[k] = value[k].decode("utf-8")
3104
## old_client_settings
3267
if "key_id" not in value:
3268
value["key_id"] = ""
3269
elif "fingerprint" not in value:
3270
value["fingerprint"] = ""
3271
# old_client_settings
3106
3273
old_client_settings = {
3107
3274
(key.decode("utf-8")
3108
3275
if isinstance(key, bytes)
3109
3276
else key): value
3110
3277
for key, value in
3111
bytes_old_client_settings.items() }
3278
bytes_old_client_settings.items()}
3112
3279
del bytes_old_client_settings
3280
# .host and .checker_command
3114
3281
for value in old_client_settings.values():
3115
if isinstance(value["host"], bytes):
3116
value["host"] = (value["host"]
3282
for attribute in ("host", "checker_command"):
3283
if isinstance(value[attribute], bytes):
3284
value[attribute] = (value[attribute]
3118
3286
os.remove(stored_state_path)
3119
3287
except IOError as e:
3120
3288
if e.errno == errno.ENOENT: