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