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-2015 Teddy Hogeborn
15
# Copyright © 2008-2015 Björn Påhlsson
17
# This program is free software: you can redistribute it and/or modify
18
# it under the terms of the GNU General Public License as published by
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,
37
from future_builtins import *
39
from future_builtins import *
40
44
import SocketServer as socketserver
78
85
import dbus.service
82
from gi.repository import GObject as gobject
87
from gi.repository import GLib
84
88
from dbus.mainloop.glib import DBusGMainLoop
87
91
import xml.dom.minidom
94
if sys.version_info.major == 2:
98
# Add collections.abc.Callable if it does not exist
100
collections.abc.Callable
101
except AttributeError:
103
Callable = collections.Callable
104
collections.abc = abc
107
# Add shlex.quote if it does not exist
110
except AttributeError:
111
shlex.quote = re.escape
113
# Show warnings by default
114
if not sys.warnoptions:
116
warnings.simplefilter("default")
118
# Try to find the value of SO_BINDTODEVICE:
120
# This is where SO_BINDTODEVICE is in Python 3.3 (or 3.4?) and
121
# newer, and it is also the most natural place for it:
91
122
SO_BINDTODEVICE = socket.SO_BINDTODEVICE
92
123
except AttributeError:
125
# This is where SO_BINDTODEVICE was up to and including Python
94
127
from IN import SO_BINDTODEVICE
95
128
except ImportError:
96
SO_BINDTODEVICE = None
98
if sys.version_info.major == 2:
129
# In Python 2.7 it seems to have been removed entirely.
130
# Try running the C preprocessor:
132
cc = subprocess.Popen(["cc", "--language=c", "-E",
134
stdin=subprocess.PIPE,
135
stdout=subprocess.PIPE)
136
stdout = cc.communicate(
137
"#include <sys/socket.h>\nSO_BINDTODEVICE\n")[0]
138
SO_BINDTODEVICE = int(stdout.splitlines()[-1])
139
except (OSError, ValueError, IndexError):
141
SO_BINDTODEVICE = None
143
if sys.version_info < (3, 2):
144
configparser.Configparser = configparser.SafeConfigParser
102
147
stored_state_file = "clients.pickle"
104
149
logger = logging.getLogger()
150
logging.captureWarnings(True) # Show warnings via the logging system
108
154
if_nametoindex = ctypes.cdll.LoadLibrary(
109
155
ctypes.util.find_library("c")).if_nametoindex
110
156
except (OSError, AttributeError):
112
158
def if_nametoindex(interface):
113
159
"Get an interface index the hard way, i.e. using fcntl()"
114
160
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
190
265
.replace(b"\n", b"\\n")
191
266
.replace(b"\0", b"\\x00"))
194
269
def encrypt(self, data, password):
195
270
passphrase = self.password_encode(password)
196
271
with tempfile.NamedTemporaryFile(
197
272
dir=self.tempdir) as passfile:
198
273
passfile.write(passphrase)
200
proc = subprocess.Popen(['gpg', '--symmetric',
275
proc = subprocess.Popen([self.gpg, '--symmetric',
201
276
'--passphrase-file',
203
278
+ self.gnupgargs,
204
stdin = subprocess.PIPE,
205
stdout = subprocess.PIPE,
206
stderr = subprocess.PIPE)
207
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)
208
283
if proc.returncode != 0:
209
284
raise PGPError(err)
210
285
return ciphertext
212
287
def decrypt(self, data, password):
213
288
passphrase = self.password_encode(password)
214
289
with tempfile.NamedTemporaryFile(
215
dir = self.tempdir) as passfile:
290
dir=self.tempdir) as passfile:
216
291
passfile.write(passphrase)
218
proc = subprocess.Popen(['gpg', '--decrypt',
293
proc = subprocess.Popen([self.gpg, '--decrypt',
219
294
'--passphrase-file',
221
296
+ self.gnupgargs,
222
stdin = subprocess.PIPE,
223
stdout = subprocess.PIPE,
224
stderr = subprocess.PIPE)
225
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)
226
301
if proc.returncode != 0:
227
302
raise PGPError(err)
228
303
return decrypted_plaintext
306
# Pretend that we have an Avahi module
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
313
DBUS_NAME = "org.freedesktop.Avahi"
314
DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
315
DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
316
DBUS_PATH_SERVER = "/"
319
def string_array_to_txt_array(t):
320
return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
321
for s in t), signature="ay")
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
231
332
class AvahiError(Exception):
232
333
def __init__(self, value, *args, **kwargs):
233
334
self.value = value
423
524
class AvahiServiceToSyslog(AvahiService):
424
525
def rename(self, *args, **kwargs):
425
526
"""Add the new name to the syslog messages"""
426
ret = AvahiService.rename(self, *args, **kwargs)
527
ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
427
528
syslogger.setFormatter(logging.Formatter(
428
529
'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
429
530
.format(self.name)))
432
534
# Pretend that we have a GnuTLS module
433
class GnuTLS(object):
434
"""This isn't so much a class as it is a module-like namespace.
435
It is instantiated once, and simulates having a GnuTLS module."""
437
_library = ctypes.cdll.LoadLibrary(
438
ctypes.util.find_library("gnutls"))
439
_need_version = "3.3.0"
441
# Need to use class name "GnuTLS" here, since this method is
442
# called before the assignment to the "gnutls" global variable
444
if GnuTLS.check_version(self._need_version) is None:
445
raise GnuTLS.Error("Needs GnuTLS {} or later"
446
.format(self._need_version))
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)
448
544
# Unless otherwise indicated, the constants and types below are
449
545
# all from the gnutls/gnutls.h C header file.
453
549
E_INTERRUPTED = -52
458
555
CRD_CERTIFICATE = 1
459
556
E_NO_CERTIFICATE_FOUND = -49
561
KEYID_USE_SHA256 = 1 # gnutls/x509.h
460
562
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
463
565
class session_int(ctypes.Structure):
465
567
session_t = ctypes.POINTER(session_int)
466
569
class certificate_credentials_st(ctypes.Structure):
468
571
certificate_credentials_t = ctypes.POINTER(
469
572
certificate_credentials_st)
470
573
certificate_type_t = ctypes.c_int
471
575
class datum_t(ctypes.Structure):
472
576
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
473
577
('size', ctypes.c_uint)]
474
579
class openpgp_crt_int(ctypes.Structure):
476
581
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
477
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
582
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
478
583
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
479
credentials_type_t = ctypes.c_int #
584
credentials_type_t = ctypes.c_int
480
585
transport_ptr_t = ctypes.c_void_p
481
586
close_request_t = ctypes.c_int
484
589
class Error(Exception):
485
# We need to use the class name "GnuTLS" here, since this
486
# exception might be raised from within GnuTLS.__init__,
487
# which is called before the assignment to the "gnutls"
488
# global variable has happened.
489
def __init__(self, message = None, code = None, args=()):
590
def __init__(self, message=None, code=None, args=()):
490
591
# Default usage is by a message string, but if a return
491
592
# code is passed, convert it to a string with
492
593
# gnutls.strerror()
494
595
if message is None and code is not None:
495
message = GnuTLS.strerror(code)
496
return super(GnuTLS.Error, self).__init__(
596
message = gnutls.strerror(code)
597
return super(gnutls.Error, self).__init__(
499
600
class CertificateSecurityError(Error):
503
class Credentials(object):
504
605
def __init__(self):
505
606
self._c_object = gnutls.certificate_credentials_t()
506
607
gnutls.certificate_allocate_credentials(
507
608
ctypes.byref(self._c_object))
508
609
self.type = gnutls.CRD_CERTIFICATE
510
611
def __del__(self):
511
612
gnutls.certificate_free_credentials(self._c_object)
513
class ClientSession(object):
514
def __init__(self, socket, credentials = None):
615
def __init__(self, socket, credentials=None):
515
616
self._c_object = gnutls.session_t()
516
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)
517
624
gnutls.set_default_priority(self._c_object)
518
625
gnutls.transport_set_ptr(self._c_object, socket.fileno())
519
626
gnutls.handshake_set_private_extensions(self._c_object,
561
668
return _error_code(result)
562
669
result = func(*arguments)
565
672
# Unless otherwise indicated, the function declarations below are
566
673
# all from the gnutls/gnutls.h C header file.
569
676
priority_set_direct = _library.gnutls_priority_set_direct
570
677
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
571
678
ctypes.POINTER(ctypes.c_char_p)]
572
679
priority_set_direct.restype = _error_code
574
681
init = _library.gnutls_init
575
682
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
576
683
init.restype = _error_code
578
685
set_default_priority = _library.gnutls_set_default_priority
579
686
set_default_priority.argtypes = [session_t]
580
687
set_default_priority.restype = _error_code
582
689
record_send = _library.gnutls_record_send
583
690
record_send.argtypes = [session_t, ctypes.c_void_p,
585
692
record_send.restype = ctypes.c_ssize_t
586
693
record_send.errcheck = _retry_on_error
588
695
certificate_allocate_credentials = (
589
696
_library.gnutls_certificate_allocate_credentials)
590
697
certificate_allocate_credentials.argtypes = [
591
698
ctypes.POINTER(certificate_credentials_t)]
592
699
certificate_allocate_credentials.restype = _error_code
594
701
certificate_free_credentials = (
595
702
_library.gnutls_certificate_free_credentials)
596
certificate_free_credentials.argtypes = [certificate_credentials_t]
703
certificate_free_credentials.argtypes = [
704
certificate_credentials_t]
597
705
certificate_free_credentials.restype = None
599
707
handshake_set_private_extensions = (
600
708
_library.gnutls_handshake_set_private_extensions)
601
709
handshake_set_private_extensions.argtypes = [session_t,
603
711
handshake_set_private_extensions.restype = None
605
713
credentials_set = _library.gnutls_credentials_set
606
714
credentials_set.argtypes = [session_t, credentials_type_t,
608
716
credentials_set.restype = _error_code
610
718
strerror = _library.gnutls_strerror
611
719
strerror.argtypes = [ctypes.c_int]
612
720
strerror.restype = ctypes.c_char_p
614
722
certificate_type_get = _library.gnutls_certificate_type_get
615
723
certificate_type_get.argtypes = [session_t]
616
724
certificate_type_get.restype = _error_code
618
726
certificate_get_peers = _library.gnutls_certificate_get_peers
619
727
certificate_get_peers.argtypes = [session_t,
620
728
ctypes.POINTER(ctypes.c_uint)]
621
729
certificate_get_peers.restype = ctypes.POINTER(datum_t)
623
731
global_set_log_level = _library.gnutls_global_set_log_level
624
732
global_set_log_level.argtypes = [ctypes.c_int]
625
733
global_set_log_level.restype = None
627
735
global_set_log_function = _library.gnutls_global_set_log_function
628
736
global_set_log_function.argtypes = [log_func]
629
737
global_set_log_function.restype = None
631
739
deinit = _library.gnutls_deinit
632
740
deinit.argtypes = [session_t]
633
741
deinit.restype = None
635
743
handshake = _library.gnutls_handshake
636
744
handshake.argtypes = [session_t]
637
745
handshake.restype = _error_code
638
746
handshake.errcheck = _retry_on_error
640
748
transport_set_ptr = _library.gnutls_transport_set_ptr
641
749
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
642
750
transport_set_ptr.restype = None
644
752
bye = _library.gnutls_bye
645
753
bye.argtypes = [session_t, close_request_t]
646
754
bye.restype = _error_code
647
755
bye.errcheck = _retry_on_error
649
757
check_version = _library.gnutls_check_version
650
758
check_version.argtypes = [ctypes.c_char_p]
651
759
check_version.restype = ctypes.c_char_p
653
# All the function declarations below are from gnutls/openpgp.h
655
openpgp_crt_init = _library.gnutls_openpgp_crt_init
656
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
657
openpgp_crt_init.restype = _error_code
659
openpgp_crt_import = _library.gnutls_openpgp_crt_import
660
openpgp_crt_import.argtypes = [openpgp_crt_t,
661
ctypes.POINTER(datum_t),
663
openpgp_crt_import.restype = _error_code
665
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
666
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
667
ctypes.POINTER(ctypes.c_uint)]
668
openpgp_crt_verify_self.restype = _error_code
670
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
671
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
672
openpgp_crt_deinit.restype = None
674
openpgp_crt_get_fingerprint = (
675
_library.gnutls_openpgp_crt_get_fingerprint)
676
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
680
openpgp_crt_get_fingerprint.restype = _error_code
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
682
831
# Remove non-public functions
683
832
del _error_code, _retry_on_error
684
# Create the global "gnutls" object, simulating a module
687
835
def call_pipe(connection, # : multiprocessing.Connection
688
836
func, *args, **kwargs):
689
837
"""This function is meant to be called by multiprocessing.Process
691
839
This function runs func(*args, **kwargs), and writes the resulting
692
840
return value on the provided multiprocessing.Connection.
694
842
connection.send(func(*args, **kwargs))
695
843
connection.close()
697
class Client(object):
698
847
"""A representation of a client host served by this server.
701
850
approved: bool(); 'None' if not yet approved/disapproved
702
851
approval_delay: datetime.timedelta(); Time to wait for approval
703
852
approval_duration: datetime.timedelta(); Duration of one approval
704
checker: subprocess.Popen(); a running checker process used
705
to see if the client lives.
706
'None' if no process is running.
707
checker_callback_tag: a gobject event source tag, or None
853
checker: multiprocessing.Process(); a running checker process used
854
to see if the client lives. 'None' if no process is
856
checker_callback_tag: a GLib event source tag, or None
708
857
checker_command: string; External command which is run to check
709
858
if client lives. %() expansions are done at
710
859
runtime with vars(self) as dict, so that for
711
860
instance %(name)s can be used in the command.
712
checker_initiator_tag: a gobject event source tag, or None
861
checker_initiator_tag: a GLib event source tag, or None
713
862
created: datetime.datetime(); (UTC) object creation
714
863
client_structure: Object describing what attributes a client has
715
864
and is used for storing the client at exit
716
865
current_checker_command: string; current running checker_command
717
disable_initiator_tag: a gobject event source tag, or None
866
disable_initiator_tag: a GLib event source tag, or None
719
868
fingerprint: string (40 or 32 hexadecimal digits); used to
720
uniquely identify the client
869
uniquely identify an OpenPGP client
870
key_id: string (64 hexadecimal digits); used to uniquely identify
871
a client using raw public keys
721
872
host: string; available for use by the checker command
722
873
interval: datetime.timedelta(); How often to start a new checker
723
874
last_approval_request: datetime.datetime(); (UTC) or None
875
1031
logger.info("Disabling client %s", self.name)
876
1032
if getattr(self, "disable_initiator_tag", None) is not None:
877
gobject.source_remove(self.disable_initiator_tag)
1033
GLib.source_remove(self.disable_initiator_tag)
878
1034
self.disable_initiator_tag = None
879
1035
self.expires = None
880
1036
if getattr(self, "checker_initiator_tag", None) is not None:
881
gobject.source_remove(self.checker_initiator_tag)
1037
GLib.source_remove(self.checker_initiator_tag)
882
1038
self.checker_initiator_tag = None
883
1039
self.stop_checker()
884
1040
self.enabled = False
886
1042
self.send_changedstate()
887
# Do not run this again if called by a gobject.timeout_add
1043
# Do not run this again if called by a GLib.timeout_add
890
1046
def __del__(self):
893
1049
def init_checker(self):
894
1050
# Schedule a new checker to be started an 'interval' from now,
895
1051
# and every interval from then on.
896
1052
if self.checker_initiator_tag is not None:
897
gobject.source_remove(self.checker_initiator_tag)
898
self.checker_initiator_tag = gobject.timeout_add(
899
int(self.interval.total_seconds() * 1000),
1053
GLib.source_remove(self.checker_initiator_tag)
1054
self.checker_initiator_tag = GLib.timeout_add(
1055
random.randrange(int(self.interval.total_seconds() * 1000
900
1057
self.start_checker)
901
1058
# Schedule a disable() when 'timeout' has passed
902
1059
if self.disable_initiator_tag is not None:
903
gobject.source_remove(self.disable_initiator_tag)
904
self.disable_initiator_tag = gobject.timeout_add(
1060
GLib.source_remove(self.disable_initiator_tag)
1061
self.disable_initiator_tag = GLib.timeout_add(
905
1062
int(self.timeout.total_seconds() * 1000), self.disable)
906
1063
# Also start a new checker *right now*.
907
1064
self.start_checker()
909
1066
def checker_callback(self, source, condition, connection,
911
1068
"""The checker has completed, so take appropriate actions."""
912
self.checker_callback_tag = None
914
1069
# Read return code from connection (see call_pipe)
915
1070
returncode = connection.recv()
916
1071
connection.close()
1072
if self.checker is not None:
1074
self.checker_callback_tag = None
918
1077
if returncode >= 0:
919
1078
self.last_checker_status = returncode
920
1079
self.last_checker_signal = None
1376
1535
exc_info=error)
1377
1536
return xmlstring
1380
1540
dbus.OBJECT_MANAGER_IFACE
1381
1541
except AttributeError:
1382
1542
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
1384
1545
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
1385
1546
"""A D-Bus object with an ObjectManager.
1387
1548
Classes inheriting from this exposes the standard
1388
1549
GetManagedObjects call and the InterfacesAdded and
1389
1550
InterfacesRemoved signals on the standard
1390
1551
"org.freedesktop.DBus.ObjectManager" interface.
1392
1553
Note: No signals are sent automatically; they must be sent
1395
1556
@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
1396
out_signature = "a{oa{sa{sv}}}")
1557
out_signature="a{oa{sa{sv}}}")
1397
1558
def GetManagedObjects(self):
1398
1559
"""This function must be overridden"""
1399
1560
raise NotImplementedError()
1401
1562
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
1402
signature = "oa{sa{sv}}")
1563
signature="oa{sa{sv}}")
1403
1564
def InterfacesAdded(self, object_path, interfaces_and_properties):
1406
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
1567
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
1407
1568
def InterfacesRemoved(self, object_path, interfaces):
1410
1571
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
1411
out_signature = "s",
1412
path_keyword = 'object_path',
1413
connection_keyword = 'connection')
1573
path_keyword='object_path',
1574
connection_keyword='connection')
1414
1575
def Introspect(self, object_path, connection):
1415
1576
"""Overloading of standard D-Bus method.
1417
1578
Override return argument name of GetManagedObjects to be
1418
1579
"objpath_interfaces_and_properties"
2312
class MultiprocessingMixIn(object):
2522
class MultiprocessingMixIn:
2313
2523
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2315
2525
def sub_process_main(self, request, address):
2317
2527
self.finish_request(request, address)
2318
2528
except Exception:
2319
2529
self.handle_error(request, address)
2320
2530
self.close_request(request)
2322
2532
def process_request(self, request, address):
2323
2533
"""Start a new process to process the request."""
2324
proc = multiprocessing.Process(target = self.sub_process_main,
2325
args = (request, address))
2534
proc = multiprocessing.Process(target=self.sub_process_main,
2535
args=(request, address))
2330
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2540
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
2331
2541
""" adds a pipe to the MixIn """
2333
2543
def process_request(self, request, client_address):
2334
2544
"""Overrides and wraps the original process_request().
2336
2546
This function creates a new pipe in self.pipe
2338
2548
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2340
2550
proc = MultiprocessingMixIn.process_request(self, request,
2341
2551
client_address)
2342
2552
self.child_pipe.close()
2343
2553
self.add_pipe(parent_pipe, proc)
2345
2555
def add_pipe(self, parent_pipe, proc):
2346
2556
"""Dummy function; override as necessary"""
2347
2557
raise NotImplementedError()
2350
2560
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
2351
socketserver.TCPServer, object):
2561
socketserver.TCPServer):
2352
2562
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
2355
2565
enabled: Boolean; whether this server is activated yet
2356
2566
interface: None or a network interface name (string)
2357
2567
use_ipv6: Boolean; to use IPv6 or not
2360
2570
def __init__(self, server_address, RequestHandlerClass,
2361
2571
interface=None,
2395
2606
# socket_wrapper(), if socketfd was set.
2396
2607
socketserver.TCPServer.__init__(self, server_address,
2397
2608
RequestHandlerClass)
2399
2610
def server_bind(self):
2400
2611
"""This overrides the normal server_bind() function
2401
2612
to bind to an interface if one was specified, and also NOT to
2402
2613
bind to an address or port if they were not specified."""
2614
global SO_BINDTODEVICE
2403
2615
if self.interface is not None:
2404
2616
if SO_BINDTODEVICE is None:
2405
logger.error("SO_BINDTODEVICE does not exist;"
2406
" cannot bind to interface %s",
2410
self.socket.setsockopt(
2411
socket.SOL_SOCKET, SO_BINDTODEVICE,
2412
(self.interface + "\0").encode("utf-8"))
2413
except socket.error as error:
2414
if error.errno == errno.EPERM:
2415
logger.error("No permission to bind to"
2416
" interface %s", self.interface)
2417
elif error.errno == errno.ENOPROTOOPT:
2418
logger.error("SO_BINDTODEVICE not available;"
2419
" cannot bind to interface %s",
2421
elif error.errno == errno.ENODEV:
2422
logger.error("Interface %s does not exist,"
2423
" cannot bind", self.interface)
2617
# Fall back to a hard-coded value which seems to be
2619
logger.warning("SO_BINDTODEVICE not found, trying 25")
2620
SO_BINDTODEVICE = 25
2622
self.socket.setsockopt(
2623
socket.SOL_SOCKET, SO_BINDTODEVICE,
2624
(self.interface + "\0").encode("utf-8"))
2625
except socket.error as error:
2626
if error.errno == errno.EPERM:
2627
logger.error("No permission to bind to"
2628
" interface %s", self.interface)
2629
elif error.errno == errno.ENOPROTOOPT:
2630
logger.error("SO_BINDTODEVICE not available;"
2631
" cannot bind to interface %s",
2633
elif error.errno == errno.ENODEV:
2634
logger.error("Interface %s does not exist,"
2635
" cannot bind", self.interface)
2426
2638
# Only bind(2) the socket if we really need to.
2427
2639
if self.server_address[0] or self.server_address[1]:
2640
if self.server_address[1]:
2641
self.allow_reuse_address = True
2428
2642
if not self.server_address[0]:
2429
2643
if self.address_family == socket.AF_INET6:
2430
any_address = "::" # in6addr_any
2644
any_address = "::" # in6addr_any
2432
any_address = "0.0.0.0" # INADDR_ANY
2646
any_address = "0.0.0.0" # INADDR_ANY
2433
2647
self.server_address = (any_address,
2434
2648
self.server_address[1])
2435
2649
elif not self.server_address[1]:
2469
2683
self.gnutls_priority = gnutls_priority
2470
2684
IPv6_TCPServer.__init__(self, server_address,
2471
2685
RequestHandlerClass,
2472
interface = interface,
2473
use_ipv6 = use_ipv6,
2474
socketfd = socketfd)
2686
interface=interface,
2476
2690
def server_activate(self):
2477
2691
if self.enabled:
2478
2692
return socketserver.TCPServer.server_activate(self)
2480
2694
def enable(self):
2481
2695
self.enabled = True
2483
2697
def add_pipe(self, parent_pipe, proc):
2484
2698
# Call "handle_ipc" for both data and EOF events
2485
gobject.io_add_watch(
2486
parent_pipe.fileno(),
2487
gobject.IO_IN | gobject.IO_HUP,
2700
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2701
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2488
2702
functools.partial(self.handle_ipc,
2489
parent_pipe = parent_pipe,
2703
parent_pipe=parent_pipe,
2492
2706
def handle_ipc(self, source, condition,
2493
2707
parent_pipe=None,
2495
2709
client_object=None):
2496
2710
# error, or the other end of multiprocessing.Pipe has closed
2497
if condition & (gobject.IO_ERR | gobject.IO_HUP):
2711
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2498
2712
# Wait for other process to exit
2502
2716
# Read a request from the child
2503
2717
request = parent_pipe.recv()
2504
2718
command = request[0]
2506
2720
if command == 'init':
2508
address = request[2]
2510
for c in self.clients.itervalues():
2511
if c.fingerprint == fpr:
2721
key_id = request[1].decode("ascii")
2722
fpr = request[2].decode("ascii")
2723
address = request[3]
2725
for c in self.clients.values():
2726
if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
2728
if key_id and c.key_id == key_id:
2731
if fpr and c.fingerprint == fpr:
2515
logger.info("Client not found for fingerprint: %s, ad"
2516
"dress: %s", fpr, address)
2735
logger.info("Client not found for key ID: %s, address"
2736
": %s", key_id or fpr, address)
2517
2737
if self.use_dbus:
2518
2738
# Emit D-Bus signal
2519
mandos_dbus_service.ClientNotFound(fpr,
2739
mandos_dbus_service.ClientNotFound(key_id or fpr,
2521
2741
parent_pipe.send(False)
2524
gobject.io_add_watch(
2525
parent_pipe.fileno(),
2526
gobject.IO_IN | gobject.IO_HUP,
2745
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2746
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2527
2747
functools.partial(self.handle_ipc,
2528
parent_pipe = parent_pipe,
2530
client_object = client))
2748
parent_pipe=parent_pipe,
2750
client_object=client))
2531
2751
parent_pipe.send(True)
2532
2752
# remove the old hook in favor of the new above hook on
2536
2756
funcname = request[1]
2537
2757
args = request[2]
2538
2758
kwargs = request[3]
2540
2760
parent_pipe.send(('data', getattr(client_object,
2541
2761
funcname)(*args,
2544
2764
if command == 'getattr':
2545
2765
attrname = request[1]
2546
2766
if isinstance(client_object.__getattribute__(attrname),
2547
collections.Callable):
2767
collections.abc.Callable):
2548
2768
parent_pipe.send(('function', ))
2550
2770
parent_pipe.send((
2551
2771
'data', client_object.__getattribute__(attrname)))
2553
2773
if command == 'setattr':
2554
2774
attrname = request[1]
2555
2775
value = request[2]
2556
2776
setattr(client_object, attrname, value)
2561
2781
def rfc3339_duration_to_delta(duration):
2562
2782
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
2564
>>> rfc3339_duration_to_delta("P7D")
2565
datetime.timedelta(7)
2566
>>> rfc3339_duration_to_delta("PT60S")
2567
datetime.timedelta(0, 60)
2568
>>> rfc3339_duration_to_delta("PT60M")
2569
datetime.timedelta(0, 3600)
2570
>>> rfc3339_duration_to_delta("PT24H")
2571
datetime.timedelta(1)
2572
>>> rfc3339_duration_to_delta("P1W")
2573
datetime.timedelta(7)
2574
>>> rfc3339_duration_to_delta("PT5M30S")
2575
datetime.timedelta(0, 330)
2576
>>> rfc3339_duration_to_delta("P1DT3M20S")
2577
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)
2580
2800
# Parsing an RFC 3339 duration with regular expressions is not
2581
2801
# possible - there would have to be multiple places for the same
2582
2802
# values, like seconds. The current code, while more esoteric, is
2583
2803
# cleaner without depending on a parsing library. If Python had a
2584
2804
# built-in library for parsing we would use it, but we'd like to
2585
2805
# avoid excessive use of external libraries.
2587
2807
# New type for defining tokens, syntax, and semantics all-in-one
2588
2808
Token = collections.namedtuple("Token", (
2589
2809
"regexp", # To match token; if "value" is not None, must have
2773
2996
parser.add_argument("--no-zeroconf", action="store_false",
2774
2997
dest="zeroconf", help="Do not use Zeroconf",
2777
3000
options = parser.parse_args()
2781
fail_count, test_count = doctest.testmod()
2782
sys.exit(os.EX_OK if fail_count == 0 else 1)
2784
3002
# Default values for config file for server-global settings
2785
server_defaults = { "interface": "",
2790
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
2791
":+SIGN-DSA-SHA256",
2792
"servicename": "Mandos",
2798
"statedir": "/var/lib/mandos",
2799
"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",
2803
3026
# Parse config file for server-global settings
2804
server_config = configparser.SafeConfigParser(server_defaults)
3027
server_config = configparser.ConfigParser(server_defaults)
2805
3028
del server_defaults
2806
3029
server_config.read(os.path.join(options.configdir, "mandos.conf"))
2807
# Convert the SafeConfigParser object to a dict
3030
# Convert the ConfigParser object to a dict
2808
3031
server_settings = server_config.defaults()
2809
3032
# Use the appropriate methods on the non-string config options
2810
for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
3033
for option in ("debug", "use_dbus", "use_ipv6", "restore",
3034
"foreground", "zeroconf"):
2811
3035
server_settings[option] = server_config.getboolean("DEFAULT",
2813
3037
if server_settings["port"]:
2979
3210
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
2980
3211
service = AvahiServiceToSyslog(
2981
name = server_settings["servicename"],
2982
servicetype = "_mandos._tcp",
2983
protocol = protocol,
3212
name=server_settings["servicename"],
3213
servicetype="_mandos._tcp",
2985
3216
if server_settings["interface"]:
2986
3217
service.interface = if_nametoindex(
2987
3218
server_settings["interface"].encode("utf-8"))
2989
3220
global multiprocessing_manager
2990
3221
multiprocessing_manager = multiprocessing.Manager()
2992
3223
client_class = Client
2994
client_class = functools.partial(ClientDBus, bus = bus)
3225
client_class = functools.partial(ClientDBus, bus=bus)
2996
3227
client_settings = Client.config_parser(client_config)
2997
3228
old_client_settings = {}
2998
3229
clients_data = {}
3000
3231
# This is used to redirect stdout and stderr for checker processes
3002
wnull = open(os.devnull, "w") # A writable /dev/null
3233
wnull = open(os.devnull, "w") # A writable /dev/null
3003
3234
# Only used if server is running in foreground but not in debug
3005
3236
if debug or not foreground:
3008
3239
# Get client data and settings from last running state.
3009
3240
if server_settings["restore"]:
3011
3242
with open(stored_state_path, "rb") as stored_state:
3012
clients_data, old_client_settings = pickle.load(
3243
if sys.version_info.major == 2:
3244
clients_data, old_client_settings = pickle.load(
3247
bytes_clients_data, bytes_old_client_settings = (
3248
pickle.load(stored_state, encoding="bytes"))
3249
# Fix bytes to strings
3252
clients_data = {(key.decode("utf-8")
3253
if isinstance(key, bytes)
3256
bytes_clients_data.items()}
3257
del bytes_clients_data
3258
for key in clients_data:
3259
value = {(k.decode("utf-8")
3260
if isinstance(k, bytes) else k): v
3262
clients_data[key].items()}
3263
clients_data[key] = value
3265
value["client_structure"] = [
3267
if isinstance(s, bytes)
3269
value["client_structure"]]
3270
# .name, .host, and .checker_command
3271
for k in ("name", "host", "checker_command"):
3272
if isinstance(value[k], bytes):
3273
value[k] = value[k].decode("utf-8")
3274
if "key_id" not in value:
3275
value["key_id"] = ""
3276
elif "fingerprint" not in value:
3277
value["fingerprint"] = ""
3278
# old_client_settings
3280
old_client_settings = {
3281
(key.decode("utf-8")
3282
if isinstance(key, bytes)
3285
bytes_old_client_settings.items()}
3286
del bytes_old_client_settings
3287
# .host and .checker_command
3288
for value in old_client_settings.values():
3289
for attribute in ("host", "checker_command"):
3290
if isinstance(value[attribute], bytes):
3291
value[attribute] = (value[attribute]
3014
3293
os.remove(stored_state_path)
3015
3294
except IOError as e:
3016
3295
if e.errno == errno.ENOENT:
3116
3395
pidfilename, pid)
3118
3397
del pidfilename
3120
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
3121
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
3399
for termsig in (signal.SIGHUP, signal.SIGTERM):
3400
GLib.unix_signal_add(GLib.PRIORITY_HIGH, termsig,
3401
lambda: main_loop.quit() and False)
3125
3405
@alternate_dbus_interfaces(
3126
{ "se.recompile.Mandos": "se.bsnet.fukt.Mandos" })
3406
{"se.recompile.Mandos": "se.bsnet.fukt.Mandos"})
3127
3407
class MandosDBusService(DBusObjectWithObjectManager):
3128
3408
"""A D-Bus proxy object"""
3130
3410
def __init__(self):
3131
3411
dbus.service.Object.__init__(self, bus, "/")
3133
3413
_interface = "se.recompile.Mandos"
3135
3415
@dbus.service.signal(_interface, signature="o")
3136
3416
def ClientAdded(self, objpath):
3140
3420
@dbus.service.signal(_interface, signature="ss")
3141
def ClientNotFound(self, fingerprint, address):
3421
def ClientNotFound(self, key_id, address):
3145
3425
@dbus_annotations({"org.freedesktop.DBus.Deprecated":
3147
3427
@dbus.service.signal(_interface, signature="os")
3148
3428
def ClientRemoved(self, objpath, name):
3152
3432
@dbus_annotations({"org.freedesktop.DBus.Deprecated":
3154
3434
@dbus.service.method(_interface, out_signature="ao")
3155
3435
def GetAllClients(self):
3157
3437
return dbus.Array(c.dbus_object_path for c in
3158
tcp_server.clients.itervalues())
3438
tcp_server.clients.values())
3160
3440
@dbus_annotations({"org.freedesktop.DBus.Deprecated":
3162
3442
@dbus.service.method(_interface,