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-2020 Teddy Hogeborn
15
# Copyright © 2008-2020 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
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.
456
555
CRD_CERTIFICATE = 1
457
556
E_NO_CERTIFICATE_FOUND = -49
561
KEYID_USE_SHA256 = 1 # gnutls/x509.h
458
562
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
461
565
class session_int(ctypes.Structure):
463
567
session_t = ctypes.POINTER(session_int)
464
569
class certificate_credentials_st(ctypes.Structure):
466
571
certificate_credentials_t = ctypes.POINTER(
467
572
certificate_credentials_st)
468
573
certificate_type_t = ctypes.c_int
469
575
class datum_t(ctypes.Structure):
470
576
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
471
577
('size', ctypes.c_uint)]
472
579
class openpgp_crt_int(ctypes.Structure):
474
581
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
475
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
582
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
476
583
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
477
credentials_type_t = ctypes.c_int #
584
credentials_type_t = ctypes.c_int
478
585
transport_ptr_t = ctypes.c_void_p
479
586
close_request_t = ctypes.c_int
482
589
class Error(Exception):
483
# We need to use the class name "GnuTLS" here, since this
484
# exception might be raised from within GnuTLS.__init__,
485
# which is called before the assignment to the "gnutls"
486
# global variable happens.
487
def __init__(self, message = None, code = None, args=()):
590
def __init__(self, message=None, code=None, args=()):
488
591
# Default usage is by a message string, but if a return
489
592
# code is passed, convert it to a string with
490
593
# gnutls.strerror()
491
595
if message is None and code is not None:
492
message = GnuTLS.strerror(code)
493
return super(GnuTLS.Error, self).__init__(
596
message = gnutls.strerror(code)
597
return super(gnutls.Error, self).__init__(
496
600
class CertificateSecurityError(Error):
500
class Credentials(object):
501
605
def __init__(self):
502
606
self._c_object = gnutls.certificate_credentials_t()
503
607
gnutls.certificate_allocate_credentials(
504
608
ctypes.byref(self._c_object))
505
609
self.type = gnutls.CRD_CERTIFICATE
507
611
def __del__(self):
508
612
gnutls.certificate_free_credentials(self._c_object)
510
class ClientSession(object):
511
def __init__(self, socket, credentials = None):
615
def __init__(self, socket, credentials=None):
512
616
self._c_object = gnutls.session_t()
513
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)
514
624
gnutls.set_default_priority(self._c_object)
515
625
gnutls.transport_set_ptr(self._c_object, socket.fileno())
516
626
gnutls.handshake_set_private_extensions(self._c_object,
522
632
ctypes.cast(credentials._c_object,
523
633
ctypes.c_void_p))
524
634
self.credentials = credentials
526
636
def __del__(self):
527
637
gnutls.deinit(self._c_object)
529
639
def handshake(self):
530
640
return gnutls.handshake(self._c_object)
532
642
def send(self, data):
533
643
data = bytes(data)
536
return gnutls.record_send(self._c_object, data, len(data))
646
data_len -= gnutls.record_send(self._c_object,
539
651
return gnutls.bye(self._c_object, gnutls.SHUT_RDWR)
541
# Error handling function
653
# Error handling functions
542
654
def _error_code(result):
543
655
"""A function to raise exceptions on errors, suitable
544
656
for the 'restype' attribute on ctypes functions"""
547
659
if result == gnutls.E_NO_CERTIFICATE_FOUND:
548
raise gnutls.CertificateSecurityError(code = result)
549
raise gnutls.Error(code = result)
660
raise gnutls.CertificateSecurityError(code=result)
661
raise gnutls.Error(code=result)
663
def _retry_on_error(result, func, arguments):
664
"""A function to retry on some errors, suitable
665
for the 'errcheck' attribute on ctypes functions"""
667
if result not in (gnutls.E_INTERRUPTED, gnutls.E_AGAIN):
668
return _error_code(result)
669
result = func(*arguments)
551
672
# Unless otherwise indicated, the function declarations below are
552
673
# all from the gnutls/gnutls.h C header file.
555
676
priority_set_direct = _library.gnutls_priority_set_direct
556
677
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
557
678
ctypes.POINTER(ctypes.c_char_p)]
558
679
priority_set_direct.restype = _error_code
560
681
init = _library.gnutls_init
561
682
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
562
683
init.restype = _error_code
564
685
set_default_priority = _library.gnutls_set_default_priority
565
686
set_default_priority.argtypes = [session_t]
566
687
set_default_priority.restype = _error_code
568
689
record_send = _library.gnutls_record_send
569
690
record_send.argtypes = [session_t, ctypes.c_void_p,
571
692
record_send.restype = ctypes.c_ssize_t
693
record_send.errcheck = _retry_on_error
573
695
certificate_allocate_credentials = (
574
696
_library.gnutls_certificate_allocate_credentials)
575
697
certificate_allocate_credentials.argtypes = [
576
698
ctypes.POINTER(certificate_credentials_t)]
577
699
certificate_allocate_credentials.restype = _error_code
579
701
certificate_free_credentials = (
580
702
_library.gnutls_certificate_free_credentials)
581
certificate_free_credentials.argtypes = [certificate_credentials_t]
703
certificate_free_credentials.argtypes = [
704
certificate_credentials_t]
582
705
certificate_free_credentials.restype = None
584
707
handshake_set_private_extensions = (
585
708
_library.gnutls_handshake_set_private_extensions)
586
709
handshake_set_private_extensions.argtypes = [session_t,
588
711
handshake_set_private_extensions.restype = None
590
713
credentials_set = _library.gnutls_credentials_set
591
714
credentials_set.argtypes = [session_t, credentials_type_t,
593
716
credentials_set.restype = _error_code
595
718
strerror = _library.gnutls_strerror
596
719
strerror.argtypes = [ctypes.c_int]
597
720
strerror.restype = ctypes.c_char_p
599
722
certificate_type_get = _library.gnutls_certificate_type_get
600
723
certificate_type_get.argtypes = [session_t]
601
724
certificate_type_get.restype = _error_code
603
726
certificate_get_peers = _library.gnutls_certificate_get_peers
604
727
certificate_get_peers.argtypes = [session_t,
605
728
ctypes.POINTER(ctypes.c_uint)]
606
729
certificate_get_peers.restype = ctypes.POINTER(datum_t)
608
731
global_set_log_level = _library.gnutls_global_set_log_level
609
732
global_set_log_level.argtypes = [ctypes.c_int]
610
733
global_set_log_level.restype = None
612
735
global_set_log_function = _library.gnutls_global_set_log_function
613
736
global_set_log_function.argtypes = [log_func]
614
737
global_set_log_function.restype = None
616
739
deinit = _library.gnutls_deinit
617
740
deinit.argtypes = [session_t]
618
741
deinit.restype = None
620
743
handshake = _library.gnutls_handshake
621
744
handshake.argtypes = [session_t]
622
745
handshake.restype = _error_code
746
handshake.errcheck = _retry_on_error
624
748
transport_set_ptr = _library.gnutls_transport_set_ptr
625
749
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
626
750
transport_set_ptr.restype = None
628
752
bye = _library.gnutls_bye
629
753
bye.argtypes = [session_t, close_request_t]
630
754
bye.restype = _error_code
755
bye.errcheck = _retry_on_error
632
757
check_version = _library.gnutls_check_version
633
758
check_version.argtypes = [ctypes.c_char_p]
634
759
check_version.restype = ctypes.c_char_p
636
# All the function declarations below are from gnutls/openpgp.h
638
openpgp_crt_init = _library.gnutls_openpgp_crt_init
639
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
640
openpgp_crt_init.restype = _error_code
642
openpgp_crt_import = _library.gnutls_openpgp_crt_import
643
openpgp_crt_import.argtypes = [openpgp_crt_t,
644
ctypes.POINTER(datum_t),
646
openpgp_crt_import.restype = _error_code
648
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
649
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
650
ctypes.POINTER(ctypes.c_uint)]
651
openpgp_crt_verify_self.restype = _error_code
653
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
654
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
655
openpgp_crt_deinit.restype = None
657
openpgp_crt_get_fingerprint = (
658
_library.gnutls_openpgp_crt_get_fingerprint)
659
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
663
openpgp_crt_get_fingerprint.restype = _error_code
665
# Remove non-public function
667
# Create the global "gnutls" object, simulating a module
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
831
# Remove non-public functions
832
del _error_code, _retry_on_error
670
835
def call_pipe(connection, # : multiprocessing.Connection
671
836
func, *args, **kwargs):
672
837
"""This function is meant to be called by multiprocessing.Process
674
839
This function runs func(*args, **kwargs), and writes the resulting
675
840
return value on the provided multiprocessing.Connection.
677
842
connection.send(func(*args, **kwargs))
678
843
connection.close()
680
class Client(object):
681
847
"""A representation of a client host served by this server.
684
850
approved: bool(); 'None' if not yet approved/disapproved
685
851
approval_delay: datetime.timedelta(); Time to wait for approval
686
852
approval_duration: datetime.timedelta(); Duration of one approval
687
checker: subprocess.Popen(); a running checker process used
688
to see if the client lives.
689
'None' if no process is running.
690
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
691
857
checker_command: string; External command which is run to check
692
858
if client lives. %() expansions are done at
693
859
runtime with vars(self) as dict, so that for
694
860
instance %(name)s can be used in the command.
695
checker_initiator_tag: a gobject event source tag, or None
861
checker_initiator_tag: a GLib event source tag, or None
696
862
created: datetime.datetime(); (UTC) object creation
697
863
client_structure: Object describing what attributes a client has
698
864
and is used for storing the client at exit
699
865
current_checker_command: string; current running checker_command
700
disable_initiator_tag: a gobject event source tag, or None
866
disable_initiator_tag: a GLib event source tag, or None
702
868
fingerprint: string (40 or 32 hexadecimal digits); used to
703
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
704
872
host: string; available for use by the checker command
705
873
interval: datetime.timedelta(); How often to start a new checker
706
874
last_approval_request: datetime.datetime(); (UTC) or None
858
1031
logger.info("Disabling client %s", self.name)
859
1032
if getattr(self, "disable_initiator_tag", None) is not None:
860
gobject.source_remove(self.disable_initiator_tag)
1033
GLib.source_remove(self.disable_initiator_tag)
861
1034
self.disable_initiator_tag = None
862
1035
self.expires = None
863
1036
if getattr(self, "checker_initiator_tag", None) is not None:
864
gobject.source_remove(self.checker_initiator_tag)
1037
GLib.source_remove(self.checker_initiator_tag)
865
1038
self.checker_initiator_tag = None
866
1039
self.stop_checker()
867
1040
self.enabled = False
869
1042
self.send_changedstate()
870
# 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
873
1046
def __del__(self):
876
1049
def init_checker(self):
877
1050
# Schedule a new checker to be started an 'interval' from now,
878
1051
# and every interval from then on.
879
1052
if self.checker_initiator_tag is not None:
880
gobject.source_remove(self.checker_initiator_tag)
881
self.checker_initiator_tag = gobject.timeout_add(
882
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
883
1057
self.start_checker)
884
1058
# Schedule a disable() when 'timeout' has passed
885
1059
if self.disable_initiator_tag is not None:
886
gobject.source_remove(self.disable_initiator_tag)
887
self.disable_initiator_tag = gobject.timeout_add(
1060
GLib.source_remove(self.disable_initiator_tag)
1061
self.disable_initiator_tag = GLib.timeout_add(
888
1062
int(self.timeout.total_seconds() * 1000), self.disable)
889
1063
# Also start a new checker *right now*.
890
1064
self.start_checker()
892
1066
def checker_callback(self, source, condition, connection,
894
1068
"""The checker has completed, so take appropriate actions."""
895
self.checker_callback_tag = None
897
1069
# Read return code from connection (see call_pipe)
898
1070
returncode = connection.recv()
899
1071
connection.close()
1072
if self.checker is not None:
1074
self.checker_callback_tag = None
901
1077
if returncode >= 0:
902
1078
self.last_checker_status = returncode
903
1079
self.last_checker_signal = None
2301
class MultiprocessingMixIn(object):
2522
class MultiprocessingMixIn:
2302
2523
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2304
2525
def sub_process_main(self, request, address):
2306
2527
self.finish_request(request, address)
2307
2528
except Exception:
2308
2529
self.handle_error(request, address)
2309
2530
self.close_request(request)
2311
2532
def process_request(self, request, address):
2312
2533
"""Start a new process to process the request."""
2313
proc = multiprocessing.Process(target = self.sub_process_main,
2314
args = (request, address))
2534
proc = multiprocessing.Process(target=self.sub_process_main,
2535
args=(request, address))
2319
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2540
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
2320
2541
""" adds a pipe to the MixIn """
2322
2543
def process_request(self, request, client_address):
2323
2544
"""Overrides and wraps the original process_request().
2325
2546
This function creates a new pipe in self.pipe
2327
2548
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2329
2550
proc = MultiprocessingMixIn.process_request(self, request,
2330
2551
client_address)
2331
2552
self.child_pipe.close()
2332
2553
self.add_pipe(parent_pipe, proc)
2334
2555
def add_pipe(self, parent_pipe, proc):
2335
2556
"""Dummy function; override as necessary"""
2336
2557
raise NotImplementedError()
2339
2560
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
2340
socketserver.TCPServer, object):
2561
socketserver.TCPServer):
2341
2562
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
2344
2565
enabled: Boolean; whether this server is activated yet
2345
2566
interface: None or a network interface name (string)
2346
2567
use_ipv6: Boolean; to use IPv6 or not
2349
2570
def __init__(self, server_address, RequestHandlerClass,
2350
2571
interface=None,
2384
2606
# socket_wrapper(), if socketfd was set.
2385
2607
socketserver.TCPServer.__init__(self, server_address,
2386
2608
RequestHandlerClass)
2388
2610
def server_bind(self):
2389
2611
"""This overrides the normal server_bind() function
2390
2612
to bind to an interface if one was specified, and also NOT to
2391
2613
bind to an address or port if they were not specified."""
2614
global SO_BINDTODEVICE
2392
2615
if self.interface is not None:
2393
2616
if SO_BINDTODEVICE is None:
2394
logger.error("SO_BINDTODEVICE does not exist;"
2395
" cannot bind to interface %s",
2399
self.socket.setsockopt(
2400
socket.SOL_SOCKET, SO_BINDTODEVICE,
2401
(self.interface + "\0").encode("utf-8"))
2402
except socket.error as error:
2403
if error.errno == errno.EPERM:
2404
logger.error("No permission to bind to"
2405
" interface %s", self.interface)
2406
elif error.errno == errno.ENOPROTOOPT:
2407
logger.error("SO_BINDTODEVICE not available;"
2408
" cannot bind to interface %s",
2410
elif error.errno == errno.ENODEV:
2411
logger.error("Interface %s does not exist,"
2412
" 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)
2415
2638
# Only bind(2) the socket if we really need to.
2416
2639
if self.server_address[0] or self.server_address[1]:
2640
if self.server_address[1]:
2641
self.allow_reuse_address = True
2417
2642
if not self.server_address[0]:
2418
2643
if self.address_family == socket.AF_INET6:
2419
any_address = "::" # in6addr_any
2644
any_address = "::" # in6addr_any
2421
any_address = "0.0.0.0" # INADDR_ANY
2646
any_address = "0.0.0.0" # INADDR_ANY
2422
2647
self.server_address = (any_address,
2423
2648
self.server_address[1])
2424
2649
elif not self.server_address[1]:
2458
2683
self.gnutls_priority = gnutls_priority
2459
2684
IPv6_TCPServer.__init__(self, server_address,
2460
2685
RequestHandlerClass,
2461
interface = interface,
2462
use_ipv6 = use_ipv6,
2463
socketfd = socketfd)
2686
interface=interface,
2465
2690
def server_activate(self):
2466
2691
if self.enabled:
2467
2692
return socketserver.TCPServer.server_activate(self)
2469
2694
def enable(self):
2470
2695
self.enabled = True
2472
2697
def add_pipe(self, parent_pipe, proc):
2473
2698
# Call "handle_ipc" for both data and EOF events
2474
gobject.io_add_watch(
2475
parent_pipe.fileno(),
2476
gobject.IO_IN | gobject.IO_HUP,
2700
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2701
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2477
2702
functools.partial(self.handle_ipc,
2478
parent_pipe = parent_pipe,
2703
parent_pipe=parent_pipe,
2481
2706
def handle_ipc(self, source, condition,
2482
2707
parent_pipe=None,
2484
2709
client_object=None):
2485
2710
# error, or the other end of multiprocessing.Pipe has closed
2486
if condition & (gobject.IO_ERR | gobject.IO_HUP):
2711
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2487
2712
# Wait for other process to exit
2491
2716
# Read a request from the child
2492
2717
request = parent_pipe.recv()
2493
2718
command = request[0]
2495
2720
if command == 'init':
2497
address = request[2]
2499
for c in self.clients.itervalues():
2500
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:
2504
logger.info("Client not found for fingerprint: %s, ad"
2505
"dress: %s", fpr, address)
2735
logger.info("Client not found for key ID: %s, address"
2736
": %s", key_id or fpr, address)
2506
2737
if self.use_dbus:
2507
2738
# Emit D-Bus signal
2508
mandos_dbus_service.ClientNotFound(fpr,
2739
mandos_dbus_service.ClientNotFound(key_id or fpr,
2510
2741
parent_pipe.send(False)
2513
gobject.io_add_watch(
2514
parent_pipe.fileno(),
2515
gobject.IO_IN | gobject.IO_HUP,
2745
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2746
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2516
2747
functools.partial(self.handle_ipc,
2517
parent_pipe = parent_pipe,
2519
client_object = client))
2748
parent_pipe=parent_pipe,
2750
client_object=client))
2520
2751
parent_pipe.send(True)
2521
2752
# remove the old hook in favor of the new above hook on
2525
2756
funcname = request[1]
2526
2757
args = request[2]
2527
2758
kwargs = request[3]
2529
2760
parent_pipe.send(('data', getattr(client_object,
2530
2761
funcname)(*args,
2533
2764
if command == 'getattr':
2534
2765
attrname = request[1]
2535
2766
if isinstance(client_object.__getattribute__(attrname),
2536
collections.Callable):
2767
collections.abc.Callable):
2537
2768
parent_pipe.send(('function', ))
2539
2770
parent_pipe.send((
2540
2771
'data', client_object.__getattribute__(attrname)))
2542
2773
if command == 'setattr':
2543
2774
attrname = request[1]
2544
2775
value = request[2]
2545
2776
setattr(client_object, attrname, value)
2550
2781
def rfc3339_duration_to_delta(duration):
2551
2782
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
2553
>>> rfc3339_duration_to_delta("P7D")
2554
datetime.timedelta(7)
2555
>>> rfc3339_duration_to_delta("PT60S")
2556
datetime.timedelta(0, 60)
2557
>>> rfc3339_duration_to_delta("PT60M")
2558
datetime.timedelta(0, 3600)
2559
>>> rfc3339_duration_to_delta("PT24H")
2560
datetime.timedelta(1)
2561
>>> rfc3339_duration_to_delta("P1W")
2562
datetime.timedelta(7)
2563
>>> rfc3339_duration_to_delta("PT5M30S")
2564
datetime.timedelta(0, 330)
2565
>>> rfc3339_duration_to_delta("P1DT3M20S")
2566
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)
2569
2800
# Parsing an RFC 3339 duration with regular expressions is not
2570
2801
# possible - there would have to be multiple places for the same
2571
2802
# values, like seconds. The current code, while more esoteric, is
2572
2803
# cleaner without depending on a parsing library. If Python had a
2573
2804
# built-in library for parsing we would use it, but we'd like to
2574
2805
# avoid excessive use of external libraries.
2576
2807
# New type for defining tokens, syntax, and semantics all-in-one
2577
2808
Token = collections.namedtuple("Token", (
2578
2809
"regexp", # To match token; if "value" is not None, must have
2968
3210
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
2969
3211
service = AvahiServiceToSyslog(
2970
name = server_settings["servicename"],
2971
servicetype = "_mandos._tcp",
2972
protocol = protocol,
3212
name=server_settings["servicename"],
3213
servicetype="_mandos._tcp",
2974
3216
if server_settings["interface"]:
2975
3217
service.interface = if_nametoindex(
2976
3218
server_settings["interface"].encode("utf-8"))
2978
3220
global multiprocessing_manager
2979
3221
multiprocessing_manager = multiprocessing.Manager()
2981
3223
client_class = Client
2983
client_class = functools.partial(ClientDBus, bus = bus)
3225
client_class = functools.partial(ClientDBus, bus=bus)
2985
3227
client_settings = Client.config_parser(client_config)
2986
3228
old_client_settings = {}
2987
3229
clients_data = {}
2989
3231
# This is used to redirect stdout and stderr for checker processes
2991
wnull = open(os.devnull, "w") # A writable /dev/null
3233
wnull = open(os.devnull, "w") # A writable /dev/null
2992
3234
# Only used if server is running in foreground but not in debug
2994
3236
if debug or not foreground:
2997
3239
# Get client data and settings from last running state.
2998
3240
if server_settings["restore"]:
3000
3242
with open(stored_state_path, "rb") as stored_state:
3001
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]
3003
3293
os.remove(stored_state_path)
3004
3294
except IOError as e:
3005
3295
if e.errno == errno.ENOENT: