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 -*-
2
# -*- mode: python; 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-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
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
21
19
# the Free Software Foundation, either version 3 of the License, or
22
20
# (at your option) any later version.
24
# Mandos is distributed in the hope that it will be useful, but
25
# WITHOUT ANY WARRANTY; without even the implied warranty of
22
# This program is distributed in the hope that it will be useful,
23
# but WITHOUT ANY WARRANTY; without even the implied warranty of
26
24
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
27
25
# GNU General Public License for more details.
29
27
# You should have received a copy of the GNU General Public License
30
# along with Mandos. If not, see <http://www.gnu.org/licenses/>.
28
# along with this program. If not, see
29
# <http://www.gnu.org/licenses/>.
32
31
# Contact the authors at <mandos@recompile.se>.
35
34
from __future__ import (division, absolute_import, print_function,
39
from future_builtins import *
37
from future_builtins import *
44
40
import SocketServer as socketserver
265
190
.replace(b"\n", b"\\n")
266
191
.replace(b"\0", b"\\x00"))
269
194
def encrypt(self, data, password):
270
195
passphrase = self.password_encode(password)
271
196
with tempfile.NamedTemporaryFile(
272
197
dir=self.tempdir) as passfile:
273
198
passfile.write(passphrase)
275
proc = subprocess.Popen([self.gpg, '--symmetric',
200
proc = subprocess.Popen(['gpg', '--symmetric',
276
201
'--passphrase-file',
278
203
+ self.gnupgargs,
279
stdin=subprocess.PIPE,
280
stdout=subprocess.PIPE,
281
stderr=subprocess.PIPE)
282
ciphertext, err = proc.communicate(input=data)
204
stdin = subprocess.PIPE,
205
stdout = subprocess.PIPE,
206
stderr = subprocess.PIPE)
207
ciphertext, err = proc.communicate(input = data)
283
208
if proc.returncode != 0:
284
209
raise PGPError(err)
285
210
return ciphertext
287
212
def decrypt(self, data, password):
288
213
passphrase = self.password_encode(password)
289
214
with tempfile.NamedTemporaryFile(
290
dir=self.tempdir) as passfile:
215
dir = self.tempdir) as passfile:
291
216
passfile.write(passphrase)
293
proc = subprocess.Popen([self.gpg, '--decrypt',
218
proc = subprocess.Popen(['gpg', '--decrypt',
294
219
'--passphrase-file',
296
221
+ self.gnupgargs,
297
stdin=subprocess.PIPE,
298
stdout=subprocess.PIPE,
299
stderr=subprocess.PIPE)
300
decrypted_plaintext, err = proc.communicate(input=data)
222
stdin = subprocess.PIPE,
223
stdout = subprocess.PIPE,
224
stderr = subprocess.PIPE)
225
decrypted_plaintext, err = proc.communicate(input = data)
301
226
if proc.returncode != 0:
302
227
raise PGPError(err)
303
228
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
332
231
class AvahiError(Exception):
333
232
def __init__(self, value, *args, **kwargs):
334
233
self.value = value
524
423
class AvahiServiceToSyslog(AvahiService):
525
424
def rename(self, *args, **kwargs):
526
425
"""Add the new name to the syslog messages"""
527
ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
426
ret = AvahiService.rename(self, *args, **kwargs)
528
427
syslogger.setFormatter(logging.Formatter(
529
428
'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
530
429
.format(self.name)))
534
432
# Pretend that we have a GnuTLS module
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)
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))
544
448
# Unless otherwise indicated, the constants and types below are
545
449
# all from the gnutls/gnutls.h C header file.
555
456
CRD_CERTIFICATE = 1
556
457
E_NO_CERTIFICATE_FOUND = -49
561
KEYID_USE_SHA256 = 1 # gnutls/x509.h
562
458
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
565
461
class session_int(ctypes.Structure):
567
463
session_t = ctypes.POINTER(session_int)
569
464
class certificate_credentials_st(ctypes.Structure):
571
466
certificate_credentials_t = ctypes.POINTER(
572
467
certificate_credentials_st)
573
468
certificate_type_t = ctypes.c_int
575
469
class datum_t(ctypes.Structure):
576
470
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
577
471
('size', ctypes.c_uint)]
579
472
class openpgp_crt_int(ctypes.Structure):
581
474
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
582
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
475
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
583
476
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
584
credentials_type_t = ctypes.c_int
477
credentials_type_t = ctypes.c_int #
585
478
transport_ptr_t = ctypes.c_void_p
586
479
close_request_t = ctypes.c_int
589
482
class Error(Exception):
590
def __init__(self, message=None, code=None, args=()):
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=()):
591
488
# Default usage is by a message string, but if a return
592
489
# code is passed, convert it to a string with
593
490
# gnutls.strerror()
595
491
if message is None and code is not None:
596
message = gnutls.strerror(code)
597
return super(gnutls.Error, self).__init__(
492
message = GnuTLS.strerror(code)
493
return super(GnuTLS.Error, self).__init__(
600
496
class CertificateSecurityError(Error):
500
class Credentials(object):
605
501
def __init__(self):
606
502
self._c_object = gnutls.certificate_credentials_t()
607
503
gnutls.certificate_allocate_credentials(
608
504
ctypes.byref(self._c_object))
609
505
self.type = gnutls.CRD_CERTIFICATE
611
507
def __del__(self):
612
508
gnutls.certificate_free_credentials(self._c_object)
615
def __init__(self, socket, credentials=None):
510
class ClientSession(object):
511
def __init__(self, socket, credentials = None):
616
512
self._c_object = gnutls.session_t()
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)
513
gnutls.init(ctypes.byref(self._c_object), gnutls.CLIENT)
624
514
gnutls.set_default_priority(self._c_object)
625
515
gnutls.transport_set_ptr(self._c_object, socket.fileno())
626
516
gnutls.handshake_set_private_extensions(self._c_object,
632
522
ctypes.cast(credentials._c_object,
633
523
ctypes.c_void_p))
634
524
self.credentials = credentials
636
526
def __del__(self):
637
527
gnutls.deinit(self._c_object)
639
529
def handshake(self):
640
530
return gnutls.handshake(self._c_object)
642
532
def send(self, data):
643
533
data = bytes(data)
646
data_len -= gnutls.record_send(self._c_object,
536
return gnutls.record_send(self._c_object, data, len(data))
651
539
return gnutls.bye(self._c_object, gnutls.SHUT_RDWR)
653
# Error handling functions
541
# Error handling function
654
542
def _error_code(result):
655
543
"""A function to raise exceptions on errors, suitable
656
544
for the 'restype' attribute on ctypes functions"""
659
547
if result == gnutls.E_NO_CERTIFICATE_FOUND:
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)
548
raise gnutls.CertificateSecurityError(code = result)
549
raise gnutls.Error(code = result)
672
551
# Unless otherwise indicated, the function declarations below are
673
552
# all from the gnutls/gnutls.h C header file.
676
555
priority_set_direct = _library.gnutls_priority_set_direct
677
556
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
678
557
ctypes.POINTER(ctypes.c_char_p)]
679
558
priority_set_direct.restype = _error_code
681
560
init = _library.gnutls_init
682
561
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
683
562
init.restype = _error_code
685
564
set_default_priority = _library.gnutls_set_default_priority
686
565
set_default_priority.argtypes = [session_t]
687
566
set_default_priority.restype = _error_code
689
568
record_send = _library.gnutls_record_send
690
569
record_send.argtypes = [session_t, ctypes.c_void_p,
692
571
record_send.restype = ctypes.c_ssize_t
693
record_send.errcheck = _retry_on_error
695
573
certificate_allocate_credentials = (
696
574
_library.gnutls_certificate_allocate_credentials)
697
575
certificate_allocate_credentials.argtypes = [
698
576
ctypes.POINTER(certificate_credentials_t)]
699
577
certificate_allocate_credentials.restype = _error_code
701
579
certificate_free_credentials = (
702
580
_library.gnutls_certificate_free_credentials)
703
certificate_free_credentials.argtypes = [
704
certificate_credentials_t]
581
certificate_free_credentials.argtypes = [certificate_credentials_t]
705
582
certificate_free_credentials.restype = None
707
584
handshake_set_private_extensions = (
708
585
_library.gnutls_handshake_set_private_extensions)
709
586
handshake_set_private_extensions.argtypes = [session_t,
711
588
handshake_set_private_extensions.restype = None
713
590
credentials_set = _library.gnutls_credentials_set
714
591
credentials_set.argtypes = [session_t, credentials_type_t,
716
593
credentials_set.restype = _error_code
718
595
strerror = _library.gnutls_strerror
719
596
strerror.argtypes = [ctypes.c_int]
720
597
strerror.restype = ctypes.c_char_p
722
599
certificate_type_get = _library.gnutls_certificate_type_get
723
600
certificate_type_get.argtypes = [session_t]
724
601
certificate_type_get.restype = _error_code
726
603
certificate_get_peers = _library.gnutls_certificate_get_peers
727
604
certificate_get_peers.argtypes = [session_t,
728
605
ctypes.POINTER(ctypes.c_uint)]
729
606
certificate_get_peers.restype = ctypes.POINTER(datum_t)
731
608
global_set_log_level = _library.gnutls_global_set_log_level
732
609
global_set_log_level.argtypes = [ctypes.c_int]
733
610
global_set_log_level.restype = None
735
612
global_set_log_function = _library.gnutls_global_set_log_function
736
613
global_set_log_function.argtypes = [log_func]
737
614
global_set_log_function.restype = None
739
616
deinit = _library.gnutls_deinit
740
617
deinit.argtypes = [session_t]
741
618
deinit.restype = None
743
620
handshake = _library.gnutls_handshake
744
621
handshake.argtypes = [session_t]
745
622
handshake.restype = _error_code
746
handshake.errcheck = _retry_on_error
748
624
transport_set_ptr = _library.gnutls_transport_set_ptr
749
625
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
750
626
transport_set_ptr.restype = None
752
628
bye = _library.gnutls_bye
753
629
bye.argtypes = [session_t, close_request_t]
754
630
bye.restype = _error_code
755
bye.errcheck = _retry_on_error
757
632
check_version = _library.gnutls_check_version
758
633
check_version.argtypes = [ctypes.c_char_p]
759
634
check_version.restype = ctypes.c_char_p
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
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
835
670
def call_pipe(connection, # : multiprocessing.Connection
836
671
func, *args, **kwargs):
837
672
"""This function is meant to be called by multiprocessing.Process
839
674
This function runs func(*args, **kwargs), and writes the resulting
840
675
return value on the provided multiprocessing.Connection.
842
677
connection.send(func(*args, **kwargs))
843
678
connection.close()
680
class Client(object):
847
681
"""A representation of a client host served by this server.
850
684
approved: bool(); 'None' if not yet approved/disapproved
851
685
approval_delay: datetime.timedelta(); Time to wait for approval
852
686
approval_duration: datetime.timedelta(); Duration of one approval
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
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
857
691
checker_command: string; External command which is run to check
858
692
if client lives. %() expansions are done at
859
693
runtime with vars(self) as dict, so that for
860
694
instance %(name)s can be used in the command.
861
checker_initiator_tag: a GLib event source tag, or None
695
checker_initiator_tag: a gobject event source tag, or None
862
696
created: datetime.datetime(); (UTC) object creation
863
697
client_structure: Object describing what attributes a client has
864
698
and is used for storing the client at exit
865
699
current_checker_command: string; current running checker_command
866
disable_initiator_tag: a GLib event source tag, or None
700
disable_initiator_tag: a gobject event source tag, or None
868
702
fingerprint: string (40 or 32 hexadecimal digits); used to
869
uniquely identify an OpenPGP client
870
key_id: string (64 hexadecimal digits); used to uniquely identify
871
a client using raw public keys
703
uniquely identify the client
872
704
host: string; available for use by the checker command
873
705
interval: datetime.timedelta(); How often to start a new checker
874
706
last_approval_request: datetime.datetime(); (UTC) or None
1031
858
logger.info("Disabling client %s", self.name)
1032
859
if getattr(self, "disable_initiator_tag", None) is not None:
1033
GLib.source_remove(self.disable_initiator_tag)
860
gobject.source_remove(self.disable_initiator_tag)
1034
861
self.disable_initiator_tag = None
1035
862
self.expires = None
1036
863
if getattr(self, "checker_initiator_tag", None) is not None:
1037
GLib.source_remove(self.checker_initiator_tag)
864
gobject.source_remove(self.checker_initiator_tag)
1038
865
self.checker_initiator_tag = None
1039
866
self.stop_checker()
1040
867
self.enabled = False
1042
869
self.send_changedstate()
1043
# Do not run this again if called by a GLib.timeout_add
870
# Do not run this again if called by a gobject.timeout_add
1046
873
def __del__(self):
1049
876
def init_checker(self):
1050
877
# Schedule a new checker to be started an 'interval' from now,
1051
878
# and every interval from then on.
1052
879
if self.checker_initiator_tag is not None:
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
880
gobject.source_remove(self.checker_initiator_tag)
881
self.checker_initiator_tag = gobject.timeout_add(
882
int(self.interval.total_seconds() * 1000),
1057
883
self.start_checker)
1058
884
# Schedule a disable() when 'timeout' has passed
1059
885
if self.disable_initiator_tag is not None:
1060
GLib.source_remove(self.disable_initiator_tag)
1061
self.disable_initiator_tag = GLib.timeout_add(
886
gobject.source_remove(self.disable_initiator_tag)
887
self.disable_initiator_tag = gobject.timeout_add(
1062
888
int(self.timeout.total_seconds() * 1000), self.disable)
1063
889
# Also start a new checker *right now*.
1064
890
self.start_checker()
1066
892
def checker_callback(self, source, condition, connection,
1068
894
"""The checker has completed, so take appropriate actions."""
895
self.checker_callback_tag = None
1069
897
# Read return code from connection (see call_pipe)
1070
898
returncode = connection.recv()
1071
899
connection.close()
1072
if self.checker is not None:
1074
self.checker_callback_tag = None
1077
901
if returncode >= 0:
1078
902
self.last_checker_status = returncode
1079
903
self.last_checker_signal = None
2522
class MultiprocessingMixIn:
2301
class MultiprocessingMixIn(object):
2523
2302
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2525
2304
def sub_process_main(self, request, address):
2527
2306
self.finish_request(request, address)
2528
2307
except Exception:
2529
2308
self.handle_error(request, address)
2530
2309
self.close_request(request)
2532
2311
def process_request(self, request, address):
2533
2312
"""Start a new process to process the request."""
2534
proc = multiprocessing.Process(target=self.sub_process_main,
2535
args=(request, address))
2313
proc = multiprocessing.Process(target = self.sub_process_main,
2314
args = (request, address))
2540
class MultiprocessingMixInWithPipe(MultiprocessingMixIn):
2319
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2541
2320
""" adds a pipe to the MixIn """
2543
2322
def process_request(self, request, client_address):
2544
2323
"""Overrides and wraps the original process_request().
2546
2325
This function creates a new pipe in self.pipe
2548
2327
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2550
2329
proc = MultiprocessingMixIn.process_request(self, request,
2551
2330
client_address)
2552
2331
self.child_pipe.close()
2553
2332
self.add_pipe(parent_pipe, proc)
2555
2334
def add_pipe(self, parent_pipe, proc):
2556
2335
"""Dummy function; override as necessary"""
2557
2336
raise NotImplementedError()
2560
2339
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
2561
socketserver.TCPServer):
2340
socketserver.TCPServer, object):
2562
2341
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
2565
2344
enabled: Boolean; whether this server is activated yet
2566
2345
interface: None or a network interface name (string)
2567
2346
use_ipv6: Boolean; to use IPv6 or not
2570
2349
def __init__(self, server_address, RequestHandlerClass,
2571
2350
interface=None,
2606
2384
# socket_wrapper(), if socketfd was set.
2607
2385
socketserver.TCPServer.__init__(self, server_address,
2608
2386
RequestHandlerClass)
2610
2388
def server_bind(self):
2611
2389
"""This overrides the normal server_bind() function
2612
2390
to bind to an interface if one was specified, and also NOT to
2613
2391
bind to an address or port if they were not specified."""
2614
global SO_BINDTODEVICE
2615
2392
if self.interface is not None:
2616
2393
if SO_BINDTODEVICE is None:
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)
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)
2638
2415
# Only bind(2) the socket if we really need to.
2639
2416
if self.server_address[0] or self.server_address[1]:
2640
if self.server_address[1]:
2641
self.allow_reuse_address = True
2642
2417
if not self.server_address[0]:
2643
2418
if self.address_family == socket.AF_INET6:
2644
any_address = "::" # in6addr_any
2419
any_address = "::" # in6addr_any
2646
any_address = "0.0.0.0" # INADDR_ANY
2421
any_address = "0.0.0.0" # INADDR_ANY
2647
2422
self.server_address = (any_address,
2648
2423
self.server_address[1])
2649
2424
elif not self.server_address[1]:
2683
2458
self.gnutls_priority = gnutls_priority
2684
2459
IPv6_TCPServer.__init__(self, server_address,
2685
2460
RequestHandlerClass,
2686
interface=interface,
2461
interface = interface,
2462
use_ipv6 = use_ipv6,
2463
socketfd = socketfd)
2690
2465
def server_activate(self):
2691
2466
if self.enabled:
2692
2467
return socketserver.TCPServer.server_activate(self)
2694
2469
def enable(self):
2695
2470
self.enabled = True
2697
2472
def add_pipe(self, parent_pipe, proc):
2698
2473
# Call "handle_ipc" for both data and EOF events
2700
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2701
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2474
gobject.io_add_watch(
2475
parent_pipe.fileno(),
2476
gobject.IO_IN | gobject.IO_HUP,
2702
2477
functools.partial(self.handle_ipc,
2703
parent_pipe=parent_pipe,
2478
parent_pipe = parent_pipe,
2706
2481
def handle_ipc(self, source, condition,
2707
2482
parent_pipe=None,
2709
2484
client_object=None):
2710
2485
# error, or the other end of multiprocessing.Pipe has closed
2711
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2486
if condition & (gobject.IO_ERR | gobject.IO_HUP):
2712
2487
# Wait for other process to exit
2716
2491
# Read a request from the child
2717
2492
request = parent_pipe.recv()
2718
2493
command = request[0]
2720
2495
if command == 'init':
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:
2497
address = request[2]
2499
for c in self.clients.itervalues():
2500
if c.fingerprint == fpr:
2735
logger.info("Client not found for key ID: %s, address"
2736
": %s", key_id or fpr, address)
2504
logger.info("Client not found for fingerprint: %s, ad"
2505
"dress: %s", fpr, address)
2737
2506
if self.use_dbus:
2738
2507
# Emit D-Bus signal
2739
mandos_dbus_service.ClientNotFound(key_id or fpr,
2508
mandos_dbus_service.ClientNotFound(fpr,
2741
2510
parent_pipe.send(False)
2745
GLib.IOChannel.unix_new(parent_pipe.fileno()),
2746
GLib.PRIORITY_DEFAULT, GLib.IO_IN | GLib.IO_HUP,
2513
gobject.io_add_watch(
2514
parent_pipe.fileno(),
2515
gobject.IO_IN | gobject.IO_HUP,
2747
2516
functools.partial(self.handle_ipc,
2748
parent_pipe=parent_pipe,
2750
client_object=client))
2517
parent_pipe = parent_pipe,
2519
client_object = client))
2751
2520
parent_pipe.send(True)
2752
2521
# remove the old hook in favor of the new above hook on
2756
2525
funcname = request[1]
2757
2526
args = request[2]
2758
2527
kwargs = request[3]
2760
2529
parent_pipe.send(('data', getattr(client_object,
2761
2530
funcname)(*args,
2764
2533
if command == 'getattr':
2765
2534
attrname = request[1]
2766
2535
if isinstance(client_object.__getattribute__(attrname),
2767
collections.abc.Callable):
2536
collections.Callable):
2768
2537
parent_pipe.send(('function', ))
2770
2539
parent_pipe.send((
2771
2540
'data', client_object.__getattribute__(attrname)))
2773
2542
if command == 'setattr':
2774
2543
attrname = request[1]
2775
2544
value = request[2]
2776
2545
setattr(client_object, attrname, value)
2781
2550
def rfc3339_duration_to_delta(duration):
2782
2551
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
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)
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)
2800
2569
# Parsing an RFC 3339 duration with regular expressions is not
2801
2570
# possible - there would have to be multiple places for the same
2802
2571
# values, like seconds. The current code, while more esoteric, is
2803
2572
# cleaner without depending on a parsing library. If Python had a
2804
2573
# built-in library for parsing we would use it, but we'd like to
2805
2574
# avoid excessive use of external libraries.
2807
2576
# New type for defining tokens, syntax, and semantics all-in-one
2808
2577
Token = collections.namedtuple("Token", (
2809
2578
"regexp", # To match token; if "value" is not None, must have
3210
2968
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
3211
2969
service = AvahiServiceToSyslog(
3212
name=server_settings["servicename"],
3213
servicetype="_mandos._tcp",
2970
name = server_settings["servicename"],
2971
servicetype = "_mandos._tcp",
2972
protocol = protocol,
3216
2974
if server_settings["interface"]:
3217
2975
service.interface = if_nametoindex(
3218
2976
server_settings["interface"].encode("utf-8"))
3220
2978
global multiprocessing_manager
3221
2979
multiprocessing_manager = multiprocessing.Manager()
3223
2981
client_class = Client
3225
client_class = functools.partial(ClientDBus, bus=bus)
2983
client_class = functools.partial(ClientDBus, bus = bus)
3227
2985
client_settings = Client.config_parser(client_config)
3228
2986
old_client_settings = {}
3229
2987
clients_data = {}
3231
2989
# This is used to redirect stdout and stderr for checker processes
3233
wnull = open(os.devnull, "w") # A writable /dev/null
2991
wnull = open(os.devnull, "w") # A writable /dev/null
3234
2992
# Only used if server is running in foreground but not in debug
3236
2994
if debug or not foreground:
3239
2997
# Get client data and settings from last running state.
3240
2998
if server_settings["restore"]:
3242
3000
with open(stored_state_path, "rb") as stored_state:
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]
3001
clients_data, old_client_settings = pickle.load(
3293
3003
os.remove(stored_state_path)
3294
3004
except IOError as e:
3295
3005
if e.errno == errno.ENOENT: