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
261
'--passphrase-file',
248
263
+ self.gnupgargs,
249
stdin = subprocess.PIPE,
250
stdout = subprocess.PIPE,
251
stderr = subprocess.PIPE)
252
ciphertext, err = proc.communicate(input = data)
264
stdin=subprocess.PIPE,
265
stdout=subprocess.PIPE,
266
stderr=subprocess.PIPE)
267
ciphertext, err = proc.communicate(input=data)
253
268
if proc.returncode != 0:
254
269
raise PGPError(err)
255
270
return ciphertext
257
272
def decrypt(self, data, password):
258
273
passphrase = self.password_encode(password)
259
274
with tempfile.NamedTemporaryFile(
260
dir = self.tempdir) as passfile:
275
dir=self.tempdir) as passfile:
261
276
passfile.write(passphrase)
263
278
proc = subprocess.Popen([self.gpg, '--decrypt',
264
279
'--passphrase-file',
266
281
+ self.gnupgargs,
267
stdin = subprocess.PIPE,
268
stdout = subprocess.PIPE,
269
stderr = subprocess.PIPE)
270
decrypted_plaintext, err = proc.communicate(input = data)
282
stdin=subprocess.PIPE,
283
stdout=subprocess.PIPE,
284
stderr=subprocess.PIPE)
285
decrypted_plaintext, err = proc.communicate(input=data)
271
286
if proc.returncode != 0:
272
287
raise PGPError(err)
273
288
return decrypted_plaintext
275
291
# 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
293
"""This isn't so much a class as it is a module-like namespace."""
294
IF_UNSPEC = -1 # avahi-common/address.h
295
PROTO_UNSPEC = -1 # avahi-common/address.h
296
PROTO_INET = 0 # avahi-common/address.h
297
PROTO_INET6 = 1 # avahi-common/address.h
283
298
DBUS_NAME = "org.freedesktop.Avahi"
284
299
DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
285
300
DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
286
301
DBUS_PATH_SERVER = "/"
287
def string_array_to_txt_array(self, t):
304
def string_array_to_txt_array(t):
288
305
return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
289
306
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
307
ENTRY_GROUP_ESTABLISHED = 2 # avahi-common/defs.h
308
ENTRY_GROUP_COLLISION = 3 # avahi-common/defs.h
309
ENTRY_GROUP_FAILURE = 4 # avahi-common/defs.h
310
SERVER_INVALID = 0 # avahi-common/defs.h
311
SERVER_REGISTERING = 1 # avahi-common/defs.h
312
SERVER_RUNNING = 2 # avahi-common/defs.h
313
SERVER_COLLISION = 3 # avahi-common/defs.h
314
SERVER_FAILURE = 4 # avahi-common/defs.h
300
317
class AvahiError(Exception):
301
318
def __init__(self, value, *args, **kwargs):
492
509
class AvahiServiceToSyslog(AvahiService):
493
510
def rename(self, *args, **kwargs):
494
511
"""Add the new name to the syslog messages"""
495
ret = AvahiService.rename(self, *args, **kwargs)
512
ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
496
513
syslogger.setFormatter(logging.Formatter(
497
514
'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
498
515
.format(self.name)))
501
519
# 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))
521
"""This isn't so much a class as it is a module-like namespace."""
523
library = ctypes.util.find_library("gnutls")
525
library = ctypes.util.find_library("gnutls-deb0")
526
_library = ctypes.cdll.LoadLibrary(library)
517
529
# Unless otherwise indicated, the constants and types below are
518
530
# all from the gnutls/gnutls.h C header file.
522
534
E_INTERRUPTED = -52
527
540
CRD_CERTIFICATE = 1
528
541
E_NO_CERTIFICATE_FOUND = -49
546
KEYID_USE_SHA256 = 1 # gnutls/x509.h
529
547
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
532
550
class session_int(ctypes.Structure):
534
552
session_t = ctypes.POINTER(session_int)
535
554
class certificate_credentials_st(ctypes.Structure):
537
556
certificate_credentials_t = ctypes.POINTER(
538
557
certificate_credentials_st)
539
558
certificate_type_t = ctypes.c_int
540
560
class datum_t(ctypes.Structure):
541
561
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
542
562
('size', ctypes.c_uint)]
543
564
class openpgp_crt_int(ctypes.Structure):
545
566
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
546
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
567
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
547
568
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
548
569
credentials_type_t = ctypes.c_int
549
570
transport_ptr_t = ctypes.c_void_p
550
571
close_request_t = ctypes.c_int
553
574
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=()):
575
def __init__(self, message=None, code=None, args=()):
559
576
# Default usage is by a message string, but if a return
560
577
# code is passed, convert it to a string with
561
578
# gnutls.strerror()
563
580
if message is None and code is not None:
564
message = GnuTLS.strerror(code)
565
return super(GnuTLS.Error, self).__init__(
581
message = gnutls.strerror(code)
582
return super(gnutls.Error, self).__init__(
568
585
class CertificateSecurityError(Error):
572
class Credentials(object):
573
590
def __init__(self):
574
591
self._c_object = gnutls.certificate_credentials_t()
575
592
gnutls.certificate_allocate_credentials(
576
593
ctypes.byref(self._c_object))
577
594
self.type = gnutls.CRD_CERTIFICATE
579
596
def __del__(self):
580
597
gnutls.certificate_free_credentials(self._c_object)
582
class ClientSession(object):
583
def __init__(self, socket, credentials = None):
600
def __init__(self, socket, credentials=None):
584
601
self._c_object = gnutls.session_t()
585
gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)
602
gnutls_flags = gnutls.CLIENT
603
if gnutls.check_version(b"3.5.6"):
604
gnutls_flags |= gnutls.NO_TICKETS
606
gnutls_flags |= gnutls.ENABLE_RAWPK
607
gnutls.init(ctypes.byref(self._c_object), gnutls_flags)
586
609
gnutls.set_default_priority(self._c_object)
587
610
gnutls.transport_set_ptr(self._c_object, socket.fileno())
588
611
gnutls.handshake_set_private_extensions(self._c_object,
630
653
return _error_code(result)
631
654
result = func(*arguments)
634
657
# Unless otherwise indicated, the function declarations below are
635
658
# all from the gnutls/gnutls.h C header file.
638
661
priority_set_direct = _library.gnutls_priority_set_direct
639
662
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
640
663
ctypes.POINTER(ctypes.c_char_p)]
641
664
priority_set_direct.restype = _error_code
643
666
init = _library.gnutls_init
644
667
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
645
668
init.restype = _error_code
647
670
set_default_priority = _library.gnutls_set_default_priority
648
671
set_default_priority.argtypes = [session_t]
649
672
set_default_priority.restype = _error_code
651
674
record_send = _library.gnutls_record_send
652
675
record_send.argtypes = [session_t, ctypes.c_void_p,
654
677
record_send.restype = ctypes.c_ssize_t
655
678
record_send.errcheck = _retry_on_error
657
680
certificate_allocate_credentials = (
658
681
_library.gnutls_certificate_allocate_credentials)
659
682
certificate_allocate_credentials.argtypes = [
660
683
ctypes.POINTER(certificate_credentials_t)]
661
684
certificate_allocate_credentials.restype = _error_code
663
686
certificate_free_credentials = (
664
687
_library.gnutls_certificate_free_credentials)
665
certificate_free_credentials.argtypes = [certificate_credentials_t]
688
certificate_free_credentials.argtypes = [
689
certificate_credentials_t]
666
690
certificate_free_credentials.restype = None
668
692
handshake_set_private_extensions = (
669
693
_library.gnutls_handshake_set_private_extensions)
670
694
handshake_set_private_extensions.argtypes = [session_t,
672
696
handshake_set_private_extensions.restype = None
674
698
credentials_set = _library.gnutls_credentials_set
675
699
credentials_set.argtypes = [session_t, credentials_type_t,
677
701
credentials_set.restype = _error_code
679
703
strerror = _library.gnutls_strerror
680
704
strerror.argtypes = [ctypes.c_int]
681
705
strerror.restype = ctypes.c_char_p
683
707
certificate_type_get = _library.gnutls_certificate_type_get
684
708
certificate_type_get.argtypes = [session_t]
685
709
certificate_type_get.restype = _error_code
687
711
certificate_get_peers = _library.gnutls_certificate_get_peers
688
712
certificate_get_peers.argtypes = [session_t,
689
713
ctypes.POINTER(ctypes.c_uint)]
690
714
certificate_get_peers.restype = ctypes.POINTER(datum_t)
692
716
global_set_log_level = _library.gnutls_global_set_log_level
693
717
global_set_log_level.argtypes = [ctypes.c_int]
694
718
global_set_log_level.restype = None
696
720
global_set_log_function = _library.gnutls_global_set_log_function
697
721
global_set_log_function.argtypes = [log_func]
698
722
global_set_log_function.restype = None
700
724
deinit = _library.gnutls_deinit
701
725
deinit.argtypes = [session_t]
702
726
deinit.restype = None
704
728
handshake = _library.gnutls_handshake
705
729
handshake.argtypes = [session_t]
706
730
handshake.restype = _error_code
707
731
handshake.errcheck = _retry_on_error
709
733
transport_set_ptr = _library.gnutls_transport_set_ptr
710
734
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
711
735
transport_set_ptr.restype = None
713
737
bye = _library.gnutls_bye
714
738
bye.argtypes = [session_t, close_request_t]
715
739
bye.restype = _error_code
716
740
bye.errcheck = _retry_on_error
718
742
check_version = _library.gnutls_check_version
719
743
check_version.argtypes = [ctypes.c_char_p]
720
744
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
746
_need_version = b"3.3.0"
747
if check_version(_need_version) is None:
748
raise self.Error("Needs GnuTLS {} or later"
749
.format(_need_version))
751
_tls_rawpk_version = b"3.6.6"
752
has_rawpk = bool(check_version(_tls_rawpk_version))
756
class pubkey_st(ctypes.Structure):
758
pubkey_t = ctypes.POINTER(pubkey_st)
760
x509_crt_fmt_t = ctypes.c_int
762
# All the function declarations below are from gnutls/abstract.h
763
pubkey_init = _library.gnutls_pubkey_init
764
pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
765
pubkey_init.restype = _error_code
767
pubkey_import = _library.gnutls_pubkey_import
768
pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
770
pubkey_import.restype = _error_code
772
pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
773
pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
774
ctypes.POINTER(ctypes.c_ubyte),
775
ctypes.POINTER(ctypes.c_size_t)]
776
pubkey_get_key_id.restype = _error_code
778
pubkey_deinit = _library.gnutls_pubkey_deinit
779
pubkey_deinit.argtypes = [pubkey_t]
780
pubkey_deinit.restype = None
782
# All the function declarations below are from gnutls/openpgp.h
784
openpgp_crt_init = _library.gnutls_openpgp_crt_init
785
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
786
openpgp_crt_init.restype = _error_code
788
openpgp_crt_import = _library.gnutls_openpgp_crt_import
789
openpgp_crt_import.argtypes = [openpgp_crt_t,
790
ctypes.POINTER(datum_t),
792
openpgp_crt_import.restype = _error_code
794
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
795
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
796
ctypes.POINTER(ctypes.c_uint)]
797
openpgp_crt_verify_self.restype = _error_code
799
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
800
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
801
openpgp_crt_deinit.restype = None
803
openpgp_crt_get_fingerprint = (
804
_library.gnutls_openpgp_crt_get_fingerprint)
805
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
809
openpgp_crt_get_fingerprint.restype = _error_code
811
if check_version(b"3.6.4"):
812
certificate_type_get2 = _library.gnutls_certificate_type_get2
813
certificate_type_get2.argtypes = [session_t, ctypes.c_int]
814
certificate_type_get2.restype = _error_code
751
816
# Remove non-public functions
752
817
del _error_code, _retry_on_error
753
# Create the global "gnutls" object, simulating a module
756
820
def call_pipe(connection, # : multiprocessing.Connection
757
821
func, *args, **kwargs):
758
822
"""This function is meant to be called by multiprocessing.Process
760
824
This function runs func(*args, **kwargs), and writes the resulting
761
825
return value on the provided multiprocessing.Connection.
763
827
connection.send(func(*args, **kwargs))
764
828
connection.close()
766
class Client(object):
767
832
"""A representation of a client host served by this server.
770
835
approved: bool(); 'None' if not yet approved/disapproved
771
836
approval_delay: datetime.timedelta(); Time to wait for approval
772
837
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.
838
checker: multiprocessing.Process(); a running checker process used
839
to see if the client lives. 'None' if no process is
776
841
checker_callback_tag: a GLib event source tag, or None
777
842
checker_command: string; External command which is run to check
778
843
if client lives. %() expansions are done at
1447
1520
exc_info=error)
1448
1521
return xmlstring
1451
1525
dbus.OBJECT_MANAGER_IFACE
1452
1526
except AttributeError:
1453
1527
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
1455
1530
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
1456
1531
"""A D-Bus object with an ObjectManager.
1458
1533
Classes inheriting from this exposes the standard
1459
1534
GetManagedObjects call and the InterfacesAdded and
1460
1535
InterfacesRemoved signals on the standard
1461
1536
"org.freedesktop.DBus.ObjectManager" interface.
1463
1538
Note: No signals are sent automatically; they must be sent
1466
1541
@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
1467
out_signature = "a{oa{sa{sv}}}")
1542
out_signature="a{oa{sa{sv}}}")
1468
1543
def GetManagedObjects(self):
1469
1544
"""This function must be overridden"""
1470
1545
raise NotImplementedError()
1472
1547
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
1473
signature = "oa{sa{sv}}")
1548
signature="oa{sa{sv}}")
1474
1549
def InterfacesAdded(self, object_path, interfaces_and_properties):
1477
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
1552
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
1478
1553
def InterfacesRemoved(self, object_path, interfaces):
1481
1556
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
1482
out_signature = "s",
1483
path_keyword = 'object_path',
1484
connection_keyword = 'connection')
1558
path_keyword='object_path',
1559
connection_keyword='connection')
1485
1560
def Introspect(self, object_path, connection):
1486
1561
"""Overloading of standard D-Bus method.
1488
1563
Override return argument name of GetManagedObjects to be
1489
1564
"objpath_interfaces_and_properties"
2365
class MultiprocessingMixIn(object):
2507
class MultiprocessingMixIn:
2366
2508
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2368
2510
def sub_process_main(self, request, address):
2370
2512
self.finish_request(request, address)
2371
2513
except Exception:
2372
2514
self.handle_error(request, address)
2373
2515
self.close_request(request)
2375
2517
def process_request(self, request, address):
2376
2518
"""Start a new process to process the request."""
2377
proc = multiprocessing.Process(target = self.sub_process_main,
2378
args = (request, address))
2519
proc = multiprocessing.Process(target=self.sub_process_main,
2520
args=(request, address))
2383
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2525
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
2384
2526
""" adds a pipe to the MixIn """
2386
2528
def process_request(self, request, client_address):
2387
2529
"""Overrides and wraps the original process_request().
2389
2531
This function creates a new pipe in self.pipe
2391
2533
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2393
2535
proc = MultiprocessingMixIn.process_request(self, request,
2394
2536
client_address)
2395
2537
self.child_pipe.close()
2396
2538
self.add_pipe(parent_pipe, proc)
2398
2540
def add_pipe(self, parent_pipe, proc):
2399
2541
"""Dummy function; override as necessary"""
2400
2542
raise NotImplementedError()
2403
2545
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
2404
socketserver.TCPServer, object):
2546
socketserver.TCPServer):
2405
2547
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
2408
2550
enabled: Boolean; whether this server is activated yet
2409
2551
interface: None or a network interface name (string)
2410
2552
use_ipv6: Boolean; to use IPv6 or not
2413
2555
def __init__(self, server_address, RequestHandlerClass,
2414
2556
interface=None,
2523
2668
self.gnutls_priority = gnutls_priority
2524
2669
IPv6_TCPServer.__init__(self, server_address,
2525
2670
RequestHandlerClass,
2526
interface = interface,
2527
use_ipv6 = use_ipv6,
2528
socketfd = socketfd)
2671
interface=interface,
2530
2675
def server_activate(self):
2531
2676
if self.enabled:
2532
2677
return socketserver.TCPServer.server_activate(self)
2534
2679
def enable(self):
2535
2680
self.enabled = True
2537
2682
def add_pipe(self, parent_pipe, proc):
2538
2683
# Call "handle_ipc" for both data and EOF events
2539
2684
GLib.io_add_watch(
2540
parent_pipe.fileno(),
2541
GLib.IO_IN | GLib.IO_HUP,
2685
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2686
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2542
2687
functools.partial(self.handle_ipc,
2543
parent_pipe = parent_pipe,
2688
parent_pipe=parent_pipe,
2546
2691
def handle_ipc(self, source, condition,
2547
2692
parent_pipe=None,
2549
2694
client_object=None):
2550
2695
# error, or the other end of multiprocessing.Pipe has closed
2551
2696
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2552
2697
# Wait for other process to exit
2556
2701
# Read a request from the child
2557
2702
request = parent_pipe.recv()
2558
2703
command = request[0]
2560
2705
if command == 'init':
2562
address = request[2]
2706
key_id = request[1].decode("ascii")
2707
fpr = request[2].decode("ascii")
2708
address = request[3]
2564
2710
for c in self.clients.values():
2565
if c.fingerprint == fpr:
2711
if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
2713
if key_id and c.key_id == key_id:
2716
if fpr and c.fingerprint == fpr:
2569
logger.info("Client not found for fingerprint: %s, ad"
2570
"dress: %s", fpr, address)
2720
logger.info("Client not found for key ID: %s, address"
2721
": %s", key_id or fpr, address)
2571
2722
if self.use_dbus:
2572
2723
# Emit D-Bus signal
2573
mandos_dbus_service.ClientNotFound(fpr,
2724
mandos_dbus_service.ClientNotFound(key_id or fpr,
2575
2726
parent_pipe.send(False)
2578
2729
GLib.io_add_watch(
2579
parent_pipe.fileno(),
2580
GLib.IO_IN | GLib.IO_HUP,
2730
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2731
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2581
2732
functools.partial(self.handle_ipc,
2582
parent_pipe = parent_pipe,
2584
client_object = client))
2733
parent_pipe=parent_pipe,
2735
client_object=client))
2585
2736
parent_pipe.send(True)
2586
2737
# remove the old hook in favor of the new above hook on
2604
2755
parent_pipe.send((
2605
2756
'data', client_object.__getattribute__(attrname)))
2607
2758
if command == 'setattr':
2608
2759
attrname = request[1]
2609
2760
value = request[2]
2610
2761
setattr(client_object, attrname, value)
2615
2766
def rfc3339_duration_to_delta(duration):
2616
2767
"""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)
2769
>>> rfc3339_duration_to_delta("P7D") == datetime.timedelta(7)
2771
>>> rfc3339_duration_to_delta("PT60S") == datetime.timedelta(0, 60)
2773
>>> rfc3339_duration_to_delta("PT60M") == datetime.timedelta(0, 3600)
2775
>>> rfc3339_duration_to_delta("PT24H") == datetime.timedelta(1)
2777
>>> rfc3339_duration_to_delta("P1W") == datetime.timedelta(7)
2779
>>> rfc3339_duration_to_delta("PT5M30S") == datetime.timedelta(0, 330)
2781
>>> rfc3339_duration_to_delta("P1DT3M20S") == datetime.timedelta(1, 200)
2634
2785
# Parsing an RFC 3339 duration with regular expressions is not
2635
2786
# possible - there would have to be multiple places for the same
2636
2787
# values, like seconds. The current code, while more esoteric, is
2637
2788
# cleaner without depending on a parsing library. If Python had a
2638
2789
# built-in library for parsing we would use it, but we'd like to
2639
2790
# avoid excessive use of external libraries.
2641
2792
# New type for defining tokens, syntax, and semantics all-in-one
2642
2793
Token = collections.namedtuple("Token", (
2643
2794
"regexp", # To match token; if "value" is not None, must have
2827
2981
parser.add_argument("--no-zeroconf", action="store_false",
2828
2982
dest="zeroconf", help="Do not use Zeroconf",
2831
2985
options = parser.parse_args()
2835
fail_count, test_count = doctest.testmod()
2836
sys.exit(os.EX_OK if fail_count == 0 else 1)
2838
2987
# 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",
2988
if gnutls.has_rawpk:
2989
priority = ("SECURE128:!CTYPE-X.509:+CTYPE-RAWPK:!RSA"
2990
":!VERS-ALL:+VERS-TLS1.3:%PROFILE_ULTRA")
2992
priority = ("SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
2993
":+SIGN-DSA-SHA256")
2994
server_defaults = {"interface": "",
2998
"priority": priority,
2999
"servicename": "Mandos",
3005
"statedir": "/var/lib/mandos",
3006
"foreground": "False",
2857
3011
# Parse config file for server-global settings
2858
server_config = configparser.SafeConfigParser(server_defaults)
3012
server_config = configparser.ConfigParser(server_defaults)
2859
3013
del server_defaults
2860
3014
server_config.read(os.path.join(options.configdir, "mandos.conf"))
2861
# Convert the SafeConfigParser object to a dict
3015
# Convert the ConfigParser object to a dict
2862
3016
server_settings = server_config.defaults()
2863
3017
# Use the appropriate methods on the non-string config options
2864
for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
3018
for option in ("debug", "use_dbus", "use_ipv6", "restore",
3019
"foreground", "zeroconf"):
2865
3020
server_settings[option] = server_config.getboolean("DEFAULT",
2867
3022
if server_settings["port"]:
3040
3195
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
3041
3196
service = AvahiServiceToSyslog(
3042
name = server_settings["servicename"],
3043
servicetype = "_mandos._tcp",
3044
protocol = protocol,
3197
name=server_settings["servicename"],
3198
servicetype="_mandos._tcp",
3046
3201
if server_settings["interface"]:
3047
3202
service.interface = if_nametoindex(
3048
3203
server_settings["interface"].encode("utf-8"))
3050
3205
global multiprocessing_manager
3051
3206
multiprocessing_manager = multiprocessing.Manager()
3053
3208
client_class = Client
3055
client_class = functools.partial(ClientDBus, bus = bus)
3210
client_class = functools.partial(ClientDBus, bus=bus)
3057
3212
client_settings = Client.config_parser(client_config)
3058
3213
old_client_settings = {}
3059
3214
clients_data = {}
3061
3216
# This is used to redirect stdout and stderr for checker processes
3063
wnull = open(os.devnull, "w") # A writable /dev/null
3218
wnull = open(os.devnull, "w") # A writable /dev/null
3064
3219
# Only used if server is running in foreground but not in debug
3066
3221
if debug or not foreground:
3069
3224
# Get client data and settings from last running state.
3070
3225
if server_settings["restore"]:
3072
3227
with open(stored_state_path, "rb") as stored_state:
3073
if sys.version_info.major == 2:
3228
if sys.version_info.major == 2:
3074
3229
clients_data, old_client_settings = pickle.load(
3077
3232
bytes_clients_data, bytes_old_client_settings = (
3078
pickle.load(stored_state, encoding = "bytes"))
3079
### Fix bytes to strings
3233
pickle.load(stored_state, encoding="bytes"))
3234
# Fix bytes to strings
3082
clients_data = { (key.decode("utf-8")
3083
if isinstance(key, bytes)
3086
bytes_clients_data.items() }
3237
clients_data = {(key.decode("utf-8")
3238
if isinstance(key, bytes)
3241
bytes_clients_data.items()}
3087
3242
del bytes_clients_data
3088
3243
for key in clients_data:
3089
value = { (k.decode("utf-8")
3090
if isinstance(k, bytes) else k): v
3092
clients_data[key].items() }
3244
value = {(k.decode("utf-8")
3245
if isinstance(k, bytes) else k): v
3247
clients_data[key].items()}
3093
3248
clients_data[key] = value
3094
3249
# .client_structure
3095
3250
value["client_structure"] = [
3096
3251
(s.decode("utf-8")
3097
3252
if isinstance(s, bytes)
3098
3253
else s) for s in
3099
value["client_structure"] ]
3101
for k in ("name", "host"):
3254
value["client_structure"]]
3255
# .name, .host, and .checker_command
3256
for k in ("name", "host", "checker_command"):
3102
3257
if isinstance(value[k], bytes):
3103
3258
value[k] = value[k].decode("utf-8")
3104
## old_client_settings
3259
if "key_id" not in value:
3260
value["key_id"] = ""
3261
elif "fingerprint" not in value:
3262
value["fingerprint"] = ""
3263
# old_client_settings
3106
3265
old_client_settings = {
3107
3266
(key.decode("utf-8")
3108
3267
if isinstance(key, bytes)
3109
3268
else key): value
3110
3269
for key, value in
3111
bytes_old_client_settings.items() }
3270
bytes_old_client_settings.items()}
3112
3271
del bytes_old_client_settings
3272
# .host and .checker_command
3114
3273
for value in old_client_settings.values():
3115
if isinstance(value["host"], bytes):
3116
value["host"] = (value["host"]
3274
for attribute in ("host", "checker_command"):
3275
if isinstance(value[attribute], bytes):
3276
value[attribute] = (value[attribute]
3118
3278
os.remove(stored_state_path)
3119
3279
except IOError as e:
3120
3280
if e.errno == errno.ENOENT: