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
276
'--passphrase-file',
248
278
+ self.gnupgargs,
249
stdin = subprocess.PIPE,
250
stdout = subprocess.PIPE,
251
stderr = subprocess.PIPE)
252
ciphertext, err = proc.communicate(input = data)
279
stdin=subprocess.PIPE,
280
stdout=subprocess.PIPE,
281
stderr=subprocess.PIPE)
282
ciphertext, err = proc.communicate(input=data)
253
283
if proc.returncode != 0:
254
284
raise PGPError(err)
255
285
return ciphertext
257
287
def decrypt(self, data, password):
258
288
passphrase = self.password_encode(password)
259
289
with tempfile.NamedTemporaryFile(
260
dir = self.tempdir) as passfile:
290
dir=self.tempdir) as passfile:
261
291
passfile.write(passphrase)
263
293
proc = subprocess.Popen([self.gpg, '--decrypt',
264
294
'--passphrase-file',
266
296
+ self.gnupgargs,
267
stdin = subprocess.PIPE,
268
stdout = subprocess.PIPE,
269
stderr = subprocess.PIPE)
270
decrypted_plaintext, err = proc.communicate(input = data)
297
stdin=subprocess.PIPE,
298
stdout=subprocess.PIPE,
299
stderr=subprocess.PIPE)
300
decrypted_plaintext, err = proc.communicate(input=data)
271
301
if proc.returncode != 0:
272
302
raise PGPError(err)
273
303
return decrypted_plaintext
275
306
# 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
308
"""This isn't so much a class as it is a module-like namespace."""
309
IF_UNSPEC = -1 # avahi-common/address.h
310
PROTO_UNSPEC = -1 # avahi-common/address.h
311
PROTO_INET = 0 # avahi-common/address.h
312
PROTO_INET6 = 1 # avahi-common/address.h
283
313
DBUS_NAME = "org.freedesktop.Avahi"
284
314
DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
285
315
DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
286
316
DBUS_PATH_SERVER = "/"
287
def string_array_to_txt_array(self, t):
319
def string_array_to_txt_array(t):
288
320
return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
289
321
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
322
ENTRY_GROUP_ESTABLISHED = 2 # avahi-common/defs.h
323
ENTRY_GROUP_COLLISION = 3 # avahi-common/defs.h
324
ENTRY_GROUP_FAILURE = 4 # avahi-common/defs.h
325
SERVER_INVALID = 0 # avahi-common/defs.h
326
SERVER_REGISTERING = 1 # avahi-common/defs.h
327
SERVER_RUNNING = 2 # avahi-common/defs.h
328
SERVER_COLLISION = 3 # avahi-common/defs.h
329
SERVER_FAILURE = 4 # avahi-common/defs.h
300
332
class AvahiError(Exception):
301
333
def __init__(self, value, *args, **kwargs):
492
524
class AvahiServiceToSyslog(AvahiService):
493
525
def rename(self, *args, **kwargs):
494
526
"""Add the new name to the syslog messages"""
495
ret = AvahiService.rename(self, *args, **kwargs)
527
ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
496
528
syslogger.setFormatter(logging.Formatter(
497
529
'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
498
530
.format(self.name)))
501
534
# 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))
536
"""This isn't so much a class as it is a module-like namespace."""
538
library = ctypes.util.find_library("gnutls")
540
library = ctypes.util.find_library("gnutls-deb0")
541
_library = ctypes.cdll.LoadLibrary(library)
517
544
# Unless otherwise indicated, the constants and types below are
518
545
# all from the gnutls/gnutls.h C header file.
522
549
E_INTERRUPTED = -52
527
555
CRD_CERTIFICATE = 1
528
556
E_NO_CERTIFICATE_FOUND = -49
561
KEYID_USE_SHA256 = 1 # gnutls/x509.h
529
562
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
532
565
class session_int(ctypes.Structure):
534
567
session_t = ctypes.POINTER(session_int)
535
569
class certificate_credentials_st(ctypes.Structure):
537
571
certificate_credentials_t = ctypes.POINTER(
538
572
certificate_credentials_st)
539
573
certificate_type_t = ctypes.c_int
540
575
class datum_t(ctypes.Structure):
541
576
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
542
577
('size', ctypes.c_uint)]
543
579
class openpgp_crt_int(ctypes.Structure):
545
581
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
546
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
582
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
547
583
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
548
584
credentials_type_t = ctypes.c_int
549
585
transport_ptr_t = ctypes.c_void_p
550
586
close_request_t = ctypes.c_int
553
589
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=()):
590
def __init__(self, message=None, code=None, args=()):
559
591
# Default usage is by a message string, but if a return
560
592
# code is passed, convert it to a string with
561
593
# gnutls.strerror()
563
595
if message is None and code is not None:
564
message = GnuTLS.strerror(code)
565
return super(GnuTLS.Error, self).__init__(
596
message = gnutls.strerror(code)
597
return super(gnutls.Error, self).__init__(
568
600
class CertificateSecurityError(Error):
572
class Credentials(object):
573
605
def __init__(self):
574
606
self._c_object = gnutls.certificate_credentials_t()
575
607
gnutls.certificate_allocate_credentials(
576
608
ctypes.byref(self._c_object))
577
609
self.type = gnutls.CRD_CERTIFICATE
579
611
def __del__(self):
580
612
gnutls.certificate_free_credentials(self._c_object)
582
class ClientSession(object):
583
def __init__(self, socket, credentials = None):
615
def __init__(self, socket, credentials=None):
584
616
self._c_object = gnutls.session_t()
585
gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)
617
gnutls_flags = gnutls.CLIENT
618
if gnutls.check_version(b"3.5.6"):
619
gnutls_flags |= gnutls.NO_TICKETS
621
gnutls_flags |= gnutls.ENABLE_RAWPK
622
gnutls.init(ctypes.byref(self._c_object), gnutls_flags)
586
624
gnutls.set_default_priority(self._c_object)
587
625
gnutls.transport_set_ptr(self._c_object, socket.fileno())
588
626
gnutls.handshake_set_private_extensions(self._c_object,
630
668
return _error_code(result)
631
669
result = func(*arguments)
634
672
# Unless otherwise indicated, the function declarations below are
635
673
# all from the gnutls/gnutls.h C header file.
638
676
priority_set_direct = _library.gnutls_priority_set_direct
639
677
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
640
678
ctypes.POINTER(ctypes.c_char_p)]
641
679
priority_set_direct.restype = _error_code
643
681
init = _library.gnutls_init
644
682
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
645
683
init.restype = _error_code
647
685
set_default_priority = _library.gnutls_set_default_priority
648
686
set_default_priority.argtypes = [session_t]
649
687
set_default_priority.restype = _error_code
651
689
record_send = _library.gnutls_record_send
652
690
record_send.argtypes = [session_t, ctypes.c_void_p,
654
692
record_send.restype = ctypes.c_ssize_t
655
693
record_send.errcheck = _retry_on_error
657
695
certificate_allocate_credentials = (
658
696
_library.gnutls_certificate_allocate_credentials)
659
697
certificate_allocate_credentials.argtypes = [
660
698
ctypes.POINTER(certificate_credentials_t)]
661
699
certificate_allocate_credentials.restype = _error_code
663
701
certificate_free_credentials = (
664
702
_library.gnutls_certificate_free_credentials)
665
certificate_free_credentials.argtypes = [certificate_credentials_t]
703
certificate_free_credentials.argtypes = [
704
certificate_credentials_t]
666
705
certificate_free_credentials.restype = None
668
707
handshake_set_private_extensions = (
669
708
_library.gnutls_handshake_set_private_extensions)
670
709
handshake_set_private_extensions.argtypes = [session_t,
672
711
handshake_set_private_extensions.restype = None
674
713
credentials_set = _library.gnutls_credentials_set
675
714
credentials_set.argtypes = [session_t, credentials_type_t,
677
716
credentials_set.restype = _error_code
679
718
strerror = _library.gnutls_strerror
680
719
strerror.argtypes = [ctypes.c_int]
681
720
strerror.restype = ctypes.c_char_p
683
722
certificate_type_get = _library.gnutls_certificate_type_get
684
723
certificate_type_get.argtypes = [session_t]
685
724
certificate_type_get.restype = _error_code
687
726
certificate_get_peers = _library.gnutls_certificate_get_peers
688
727
certificate_get_peers.argtypes = [session_t,
689
728
ctypes.POINTER(ctypes.c_uint)]
690
729
certificate_get_peers.restype = ctypes.POINTER(datum_t)
692
731
global_set_log_level = _library.gnutls_global_set_log_level
693
732
global_set_log_level.argtypes = [ctypes.c_int]
694
733
global_set_log_level.restype = None
696
735
global_set_log_function = _library.gnutls_global_set_log_function
697
736
global_set_log_function.argtypes = [log_func]
698
737
global_set_log_function.restype = None
700
739
deinit = _library.gnutls_deinit
701
740
deinit.argtypes = [session_t]
702
741
deinit.restype = None
704
743
handshake = _library.gnutls_handshake
705
744
handshake.argtypes = [session_t]
706
745
handshake.restype = _error_code
707
746
handshake.errcheck = _retry_on_error
709
748
transport_set_ptr = _library.gnutls_transport_set_ptr
710
749
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
711
750
transport_set_ptr.restype = None
713
752
bye = _library.gnutls_bye
714
753
bye.argtypes = [session_t, close_request_t]
715
754
bye.restype = _error_code
716
755
bye.errcheck = _retry_on_error
718
757
check_version = _library.gnutls_check_version
719
758
check_version.argtypes = [ctypes.c_char_p]
720
759
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
761
_need_version = b"3.3.0"
762
if check_version(_need_version) is None:
763
raise self.Error("Needs GnuTLS {} or later"
764
.format(_need_version))
766
_tls_rawpk_version = b"3.6.6"
767
has_rawpk = bool(check_version(_tls_rawpk_version))
771
class pubkey_st(ctypes.Structure):
773
pubkey_t = ctypes.POINTER(pubkey_st)
775
x509_crt_fmt_t = ctypes.c_int
777
# All the function declarations below are from gnutls/abstract.h
778
pubkey_init = _library.gnutls_pubkey_init
779
pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
780
pubkey_init.restype = _error_code
782
pubkey_import = _library.gnutls_pubkey_import
783
pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
785
pubkey_import.restype = _error_code
787
pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
788
pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
789
ctypes.POINTER(ctypes.c_ubyte),
790
ctypes.POINTER(ctypes.c_size_t)]
791
pubkey_get_key_id.restype = _error_code
793
pubkey_deinit = _library.gnutls_pubkey_deinit
794
pubkey_deinit.argtypes = [pubkey_t]
795
pubkey_deinit.restype = None
797
# All the function declarations below are from gnutls/openpgp.h
799
openpgp_crt_init = _library.gnutls_openpgp_crt_init
800
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
801
openpgp_crt_init.restype = _error_code
803
openpgp_crt_import = _library.gnutls_openpgp_crt_import
804
openpgp_crt_import.argtypes = [openpgp_crt_t,
805
ctypes.POINTER(datum_t),
807
openpgp_crt_import.restype = _error_code
809
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
810
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
811
ctypes.POINTER(ctypes.c_uint)]
812
openpgp_crt_verify_self.restype = _error_code
814
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
815
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
816
openpgp_crt_deinit.restype = None
818
openpgp_crt_get_fingerprint = (
819
_library.gnutls_openpgp_crt_get_fingerprint)
820
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
824
openpgp_crt_get_fingerprint.restype = _error_code
826
if check_version(b"3.6.4"):
827
certificate_type_get2 = _library.gnutls_certificate_type_get2
828
certificate_type_get2.argtypes = [session_t, ctypes.c_int]
829
certificate_type_get2.restype = _error_code
751
831
# Remove non-public functions
752
832
del _error_code, _retry_on_error
753
# Create the global "gnutls" object, simulating a module
756
835
def call_pipe(connection, # : multiprocessing.Connection
757
836
func, *args, **kwargs):
758
837
"""This function is meant to be called by multiprocessing.Process
760
839
This function runs func(*args, **kwargs), and writes the resulting
761
840
return value on the provided multiprocessing.Connection.
763
842
connection.send(func(*args, **kwargs))
764
843
connection.close()
766
class Client(object):
767
847
"""A representation of a client host served by this server.
770
850
approved: bool(); 'None' if not yet approved/disapproved
771
851
approval_delay: datetime.timedelta(); Time to wait for approval
772
852
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.
853
checker: multiprocessing.Process(); a running checker process used
854
to see if the client lives. 'None' if no process is
776
856
checker_callback_tag: a GLib event source tag, or None
777
857
checker_command: string; External command which is run to check
778
858
if client lives. %() expansions are done at
1447
1535
exc_info=error)
1448
1536
return xmlstring
1451
1540
dbus.OBJECT_MANAGER_IFACE
1452
1541
except AttributeError:
1453
1542
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
1455
1545
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
1456
1546
"""A D-Bus object with an ObjectManager.
1458
1548
Classes inheriting from this exposes the standard
1459
1549
GetManagedObjects call and the InterfacesAdded and
1460
1550
InterfacesRemoved signals on the standard
1461
1551
"org.freedesktop.DBus.ObjectManager" interface.
1463
1553
Note: No signals are sent automatically; they must be sent
1466
1556
@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
1467
out_signature = "a{oa{sa{sv}}}")
1557
out_signature="a{oa{sa{sv}}}")
1468
1558
def GetManagedObjects(self):
1469
1559
"""This function must be overridden"""
1470
1560
raise NotImplementedError()
1472
1562
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
1473
signature = "oa{sa{sv}}")
1563
signature="oa{sa{sv}}")
1474
1564
def InterfacesAdded(self, object_path, interfaces_and_properties):
1477
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
1567
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
1478
1568
def InterfacesRemoved(self, object_path, interfaces):
1481
1571
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
1482
out_signature = "s",
1483
path_keyword = 'object_path',
1484
connection_keyword = 'connection')
1573
path_keyword='object_path',
1574
connection_keyword='connection')
1485
1575
def Introspect(self, object_path, connection):
1486
1576
"""Overloading of standard D-Bus method.
1488
1578
Override return argument name of GetManagedObjects to be
1489
1579
"objpath_interfaces_and_properties"
2365
class MultiprocessingMixIn(object):
2522
class MultiprocessingMixIn:
2366
2523
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2368
2525
def sub_process_main(self, request, address):
2370
2527
self.finish_request(request, address)
2371
2528
except Exception:
2372
2529
self.handle_error(request, address)
2373
2530
self.close_request(request)
2375
2532
def process_request(self, request, address):
2376
2533
"""Start a new process to process the request."""
2377
proc = multiprocessing.Process(target = self.sub_process_main,
2378
args = (request, address))
2534
proc = multiprocessing.Process(target=self.sub_process_main,
2535
args=(request, address))
2383
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2540
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
2384
2541
""" adds a pipe to the MixIn """
2386
2543
def process_request(self, request, client_address):
2387
2544
"""Overrides and wraps the original process_request().
2389
2546
This function creates a new pipe in self.pipe
2391
2548
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2393
2550
proc = MultiprocessingMixIn.process_request(self, request,
2394
2551
client_address)
2395
2552
self.child_pipe.close()
2396
2553
self.add_pipe(parent_pipe, proc)
2398
2555
def add_pipe(self, parent_pipe, proc):
2399
2556
"""Dummy function; override as necessary"""
2400
2557
raise NotImplementedError()
2403
2560
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
2404
socketserver.TCPServer, object):
2561
socketserver.TCPServer):
2405
2562
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
2408
2565
enabled: Boolean; whether this server is activated yet
2409
2566
interface: None or a network interface name (string)
2410
2567
use_ipv6: Boolean; to use IPv6 or not
2413
2570
def __init__(self, server_address, RequestHandlerClass,
2414
2571
interface=None,
2522
2683
self.gnutls_priority = gnutls_priority
2523
2684
IPv6_TCPServer.__init__(self, server_address,
2524
2685
RequestHandlerClass,
2525
interface = interface,
2526
use_ipv6 = use_ipv6,
2527
socketfd = socketfd)
2686
interface=interface,
2529
2690
def server_activate(self):
2530
2691
if self.enabled:
2531
2692
return socketserver.TCPServer.server_activate(self)
2533
2694
def enable(self):
2534
2695
self.enabled = True
2536
2697
def add_pipe(self, parent_pipe, proc):
2537
2698
# Call "handle_ipc" for both data and EOF events
2538
2699
GLib.io_add_watch(
2539
parent_pipe.fileno(),
2540
GLib.IO_IN | GLib.IO_HUP,
2700
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2701
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2541
2702
functools.partial(self.handle_ipc,
2542
parent_pipe = parent_pipe,
2703
parent_pipe=parent_pipe,
2545
2706
def handle_ipc(self, source, condition,
2546
2707
parent_pipe=None,
2548
2709
client_object=None):
2549
2710
# error, or the other end of multiprocessing.Pipe has closed
2550
2711
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2551
2712
# Wait for other process to exit
2555
2716
# Read a request from the child
2556
2717
request = parent_pipe.recv()
2557
2718
command = request[0]
2559
2720
if command == 'init':
2561
address = request[2]
2721
key_id = request[1].decode("ascii")
2722
fpr = request[2].decode("ascii")
2723
address = request[3]
2563
2725
for c in self.clients.values():
2564
if c.fingerprint == fpr:
2726
if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
2728
if key_id and c.key_id == key_id:
2731
if fpr and c.fingerprint == fpr:
2568
logger.info("Client not found for fingerprint: %s, ad"
2569
"dress: %s", fpr, address)
2735
logger.info("Client not found for key ID: %s, address"
2736
": %s", key_id or fpr, address)
2570
2737
if self.use_dbus:
2571
2738
# Emit D-Bus signal
2572
mandos_dbus_service.ClientNotFound(fpr,
2739
mandos_dbus_service.ClientNotFound(key_id or fpr,
2574
2741
parent_pipe.send(False)
2577
2744
GLib.io_add_watch(
2578
parent_pipe.fileno(),
2579
GLib.IO_IN | GLib.IO_HUP,
2745
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2746
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2580
2747
functools.partial(self.handle_ipc,
2581
parent_pipe = parent_pipe,
2583
client_object = client))
2748
parent_pipe=parent_pipe,
2750
client_object=client))
2584
2751
parent_pipe.send(True)
2585
2752
# remove the old hook in favor of the new above hook on
2589
2756
funcname = request[1]
2590
2757
args = request[2]
2591
2758
kwargs = request[3]
2593
2760
parent_pipe.send(('data', getattr(client_object,
2594
2761
funcname)(*args,
2597
2764
if command == 'getattr':
2598
2765
attrname = request[1]
2599
2766
if isinstance(client_object.__getattribute__(attrname),
2600
collections.Callable):
2767
collections.abc.Callable):
2601
2768
parent_pipe.send(('function', ))
2603
2770
parent_pipe.send((
2604
2771
'data', client_object.__getattribute__(attrname)))
2606
2773
if command == 'setattr':
2607
2774
attrname = request[1]
2608
2775
value = request[2]
2609
2776
setattr(client_object, attrname, value)
2614
2781
def rfc3339_duration_to_delta(duration):
2615
2782
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
2617
>>> rfc3339_duration_to_delta("P7D")
2618
datetime.timedelta(7)
2619
>>> rfc3339_duration_to_delta("PT60S")
2620
datetime.timedelta(0, 60)
2621
>>> rfc3339_duration_to_delta("PT60M")
2622
datetime.timedelta(0, 3600)
2623
>>> rfc3339_duration_to_delta("PT24H")
2624
datetime.timedelta(1)
2625
>>> rfc3339_duration_to_delta("P1W")
2626
datetime.timedelta(7)
2627
>>> rfc3339_duration_to_delta("PT5M30S")
2628
datetime.timedelta(0, 330)
2629
>>> rfc3339_duration_to_delta("P1DT3M20S")
2630
datetime.timedelta(1, 200)
2784
>>> rfc3339_duration_to_delta("P7D") == datetime.timedelta(7)
2786
>>> rfc3339_duration_to_delta("PT60S") == datetime.timedelta(0, 60)
2788
>>> rfc3339_duration_to_delta("PT60M") == datetime.timedelta(0, 3600)
2790
>>> rfc3339_duration_to_delta("PT24H") == datetime.timedelta(1)
2792
>>> rfc3339_duration_to_delta("P1W") == datetime.timedelta(7)
2794
>>> rfc3339_duration_to_delta("PT5M30S") == datetime.timedelta(0, 330)
2796
>>> rfc3339_duration_to_delta("P1DT3M20S") == datetime.timedelta(1, 200)
2633
2800
# Parsing an RFC 3339 duration with regular expressions is not
2634
2801
# possible - there would have to be multiple places for the same
2635
2802
# values, like seconds. The current code, while more esoteric, is
2636
2803
# cleaner without depending on a parsing library. If Python had a
2637
2804
# built-in library for parsing we would use it, but we'd like to
2638
2805
# avoid excessive use of external libraries.
2640
2807
# New type for defining tokens, syntax, and semantics all-in-one
2641
2808
Token = collections.namedtuple("Token", (
2642
2809
"regexp", # To match token; if "value" is not None, must have
2826
2996
parser.add_argument("--no-zeroconf", action="store_false",
2827
2997
dest="zeroconf", help="Do not use Zeroconf",
2830
3000
options = parser.parse_args()
2834
fail_count, test_count = doctest.testmod()
2835
sys.exit(os.EX_OK if fail_count == 0 else 1)
2837
3002
# Default values for config file for server-global settings
2838
server_defaults = { "interface": "",
2843
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
2844
":+SIGN-DSA-SHA256",
2845
"servicename": "Mandos",
2851
"statedir": "/var/lib/mandos",
2852
"foreground": "False",
3003
if gnutls.has_rawpk:
3004
priority = ("SECURE128:!CTYPE-X.509:+CTYPE-RAWPK:!RSA"
3005
":!VERS-ALL:+VERS-TLS1.3:%PROFILE_ULTRA")
3007
priority = ("SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
3008
":+SIGN-DSA-SHA256")
3009
server_defaults = {"interface": "",
3013
"priority": priority,
3014
"servicename": "Mandos",
3020
"statedir": "/var/lib/mandos",
3021
"foreground": "False",
2856
3026
# Parse config file for server-global settings
2857
server_config = configparser.SafeConfigParser(server_defaults)
3027
server_config = configparser.ConfigParser(server_defaults)
2858
3028
del server_defaults
2859
3029
server_config.read(os.path.join(options.configdir, "mandos.conf"))
2860
# Convert the SafeConfigParser object to a dict
3030
# Convert the ConfigParser object to a dict
2861
3031
server_settings = server_config.defaults()
2862
3032
# Use the appropriate methods on the non-string config options
2863
for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
3033
for option in ("debug", "use_dbus", "use_ipv6", "restore",
3034
"foreground", "zeroconf"):
2864
3035
server_settings[option] = server_config.getboolean("DEFAULT",
2866
3037
if server_settings["port"]:
3039
3210
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
3040
3211
service = AvahiServiceToSyslog(
3041
name = server_settings["servicename"],
3042
servicetype = "_mandos._tcp",
3043
protocol = protocol,
3212
name=server_settings["servicename"],
3213
servicetype="_mandos._tcp",
3045
3216
if server_settings["interface"]:
3046
3217
service.interface = if_nametoindex(
3047
3218
server_settings["interface"].encode("utf-8"))
3049
3220
global multiprocessing_manager
3050
3221
multiprocessing_manager = multiprocessing.Manager()
3052
3223
client_class = Client
3054
client_class = functools.partial(ClientDBus, bus = bus)
3225
client_class = functools.partial(ClientDBus, bus=bus)
3056
3227
client_settings = Client.config_parser(client_config)
3057
3228
old_client_settings = {}
3058
3229
clients_data = {}
3060
3231
# This is used to redirect stdout and stderr for checker processes
3062
wnull = open(os.devnull, "w") # A writable /dev/null
3233
wnull = open(os.devnull, "w") # A writable /dev/null
3063
3234
# Only used if server is running in foreground but not in debug
3065
3236
if debug or not foreground:
3068
3239
# Get client data and settings from last running state.
3069
3240
if server_settings["restore"]:
3071
3242
with open(stored_state_path, "rb") as stored_state:
3072
if sys.version_info.major == 2:
3243
if sys.version_info.major == 2:
3073
3244
clients_data, old_client_settings = pickle.load(
3076
3247
bytes_clients_data, bytes_old_client_settings = (
3077
pickle.load(stored_state, encoding = "bytes"))
3078
### Fix bytes to strings
3248
pickle.load(stored_state, encoding="bytes"))
3249
# Fix bytes to strings
3081
clients_data = { (key.decode("utf-8")
3082
if isinstance(key, bytes)
3085
bytes_clients_data.items() }
3252
clients_data = {(key.decode("utf-8")
3253
if isinstance(key, bytes)
3256
bytes_clients_data.items()}
3086
3257
del bytes_clients_data
3087
3258
for key in clients_data:
3088
value = { (k.decode("utf-8")
3089
if isinstance(k, bytes) else k): v
3091
clients_data[key].items() }
3259
value = {(k.decode("utf-8")
3260
if isinstance(k, bytes) else k): v
3262
clients_data[key].items()}
3092
3263
clients_data[key] = value
3093
3264
# .client_structure
3094
3265
value["client_structure"] = [
3095
3266
(s.decode("utf-8")
3096
3267
if isinstance(s, bytes)
3097
3268
else s) for s in
3098
value["client_structure"] ]
3100
for k in ("name", "host"):
3269
value["client_structure"]]
3270
# .name, .host, and .checker_command
3271
for k in ("name", "host", "checker_command"):
3101
3272
if isinstance(value[k], bytes):
3102
3273
value[k] = value[k].decode("utf-8")
3103
## old_client_settings
3274
if "key_id" not in value:
3275
value["key_id"] = ""
3276
elif "fingerprint" not in value:
3277
value["fingerprint"] = ""
3278
# old_client_settings
3105
3280
old_client_settings = {
3106
3281
(key.decode("utf-8")
3107
3282
if isinstance(key, bytes)
3108
3283
else key): value
3109
3284
for key, value in
3110
bytes_old_client_settings.items() }
3285
bytes_old_client_settings.items()}
3111
3286
del bytes_old_client_settings
3287
# .host and .checker_command
3113
3288
for value in old_client_settings.values():
3114
if isinstance(value["host"], bytes):
3115
value["host"] = (value["host"]
3289
for attribute in ("host", "checker_command"):
3290
if isinstance(value[attribute], bytes):
3291
value[attribute] = (value[attribute]
3117
3293
os.remove(stored_state_path)
3118
3294
except IOError as e:
3119
3295
if e.errno == errno.ENOENT: