2
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-2016 Teddy Hogeborn
15
# Copyright © 2008-2016 Björn Påhlsson
17
# This program is free software: you can redistribute it and/or modify
18
# it under the terms of the GNU General Public License as published by
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,
247
227
'--passphrase-file',
249
229
+ self.gnupgargs,
250
stdin=subprocess.PIPE,
251
stdout=subprocess.PIPE,
252
stderr=subprocess.PIPE)
253
ciphertext, err = proc.communicate(input=data)
230
stdin = subprocess.PIPE,
231
stdout = subprocess.PIPE,
232
stderr = subprocess.PIPE)
233
ciphertext, err = proc.communicate(input = data)
254
234
if proc.returncode != 0:
255
235
raise PGPError(err)
256
236
return ciphertext
258
238
def decrypt(self, data, password):
259
239
passphrase = self.password_encode(password)
260
240
with tempfile.NamedTemporaryFile(
261
dir=self.tempdir) as passfile:
241
dir = self.tempdir) as passfile:
262
242
passfile.write(passphrase)
264
244
proc = subprocess.Popen([self.gpg, '--decrypt',
265
245
'--passphrase-file',
267
247
+ self.gnupgargs,
268
stdin=subprocess.PIPE,
269
stdout=subprocess.PIPE,
270
stderr=subprocess.PIPE)
271
decrypted_plaintext, err = proc.communicate(input=data)
248
stdin = subprocess.PIPE,
249
stdout = subprocess.PIPE,
250
stderr = subprocess.PIPE)
251
decrypted_plaintext, err = proc.communicate(input = data)
272
252
if proc.returncode != 0:
273
253
raise PGPError(err)
274
254
return decrypted_plaintext
277
256
# Pretend that we have an Avahi module
278
257
class Avahi(object):
279
258
"""This isn't so much a class as it is a module-like namespace.
280
259
It is instantiated once, and simulates having an Avahi module."""
281
IF_UNSPEC = -1 # avahi-common/address.h
282
PROTO_UNSPEC = -1 # avahi-common/address.h
283
PROTO_INET = 0 # avahi-common/address.h
284
PROTO_INET6 = 1 # avahi-common/address.h
260
IF_UNSPEC = -1 # avahi-common/address.h
261
PROTO_UNSPEC = -1 # avahi-common/address.h
262
PROTO_INET = 0 # avahi-common/address.h
263
PROTO_INET6 = 1 # avahi-common/address.h
285
264
DBUS_NAME = "org.freedesktop.Avahi"
286
265
DBUS_INTERFACE_ENTRY_GROUP = DBUS_NAME + ".EntryGroup"
287
266
DBUS_INTERFACE_SERVER = DBUS_NAME + ".Server"
288
267
DBUS_PATH_SERVER = "/"
290
268
def string_array_to_txt_array(self, t):
291
269
return dbus.Array((dbus.ByteArray(s.encode("utf-8"))
292
270
for s in t), signature="ay")
293
ENTRY_GROUP_ESTABLISHED = 2 # avahi-common/defs.h
294
ENTRY_GROUP_COLLISION = 3 # avahi-common/defs.h
295
ENTRY_GROUP_FAILURE = 4 # avahi-common/defs.h
296
SERVER_INVALID = 0 # avahi-common/defs.h
297
SERVER_REGISTERING = 1 # avahi-common/defs.h
298
SERVER_RUNNING = 2 # avahi-common/defs.h
299
SERVER_COLLISION = 3 # avahi-common/defs.h
300
SERVER_FAILURE = 4 # avahi-common/defs.h
271
ENTRY_GROUP_ESTABLISHED = 2 # avahi-common/defs.h
272
ENTRY_GROUP_COLLISION = 3 # avahi-common/defs.h
273
ENTRY_GROUP_FAILURE = 4 # avahi-common/defs.h
274
SERVER_INVALID = 0 # avahi-common/defs.h
275
SERVER_REGISTERING = 1 # avahi-common/defs.h
276
SERVER_RUNNING = 2 # avahi-common/defs.h
277
SERVER_COLLISION = 3 # avahi-common/defs.h
278
SERVER_FAILURE = 4 # avahi-common/defs.h
304
281
class AvahiError(Exception):
305
282
def __init__(self, value, *args, **kwargs):
306
283
self.value = value
496
473
class AvahiServiceToSyslog(AvahiService):
497
474
def rename(self, *args, **kwargs):
498
475
"""Add the new name to the syslog messages"""
499
ret = super(AvahiServiceToSyslog, self).rename(*args, **kwargs)
476
ret = AvahiService.rename(self, *args, **kwargs)
500
477
syslogger.setFormatter(logging.Formatter(
501
478
'Mandos ({}) [%(process)d]: %(levelname)s: %(message)s'
502
479
.format(self.name)))
506
482
# Pretend that we have a GnuTLS module
507
483
class GnuTLS(object):
508
484
"""This isn't so much a class as it is a module-like namespace.
509
485
It is instantiated once, and simulates having a GnuTLS module."""
511
library = ctypes.util.find_library("gnutls")
513
library = ctypes.util.find_library("gnutls-deb0")
514
_library = ctypes.cdll.LoadLibrary(library)
487
_library = ctypes.cdll.LoadLibrary(
488
ctypes.util.find_library("gnutls"))
516
489
_need_version = b"3.3.0"
517
_tls_rawpk_version = b"3.6.6"
519
490
def __init__(self):
520
# Need to use "self" here, since this method is called before
521
# the assignment to the "gnutls" global variable happens.
522
if self.check_version(self._need_version) is None:
523
raise self.Error("Needs GnuTLS {} or later"
524
.format(self._need_version))
491
# Need to use class name "GnuTLS" here, since this method is
492
# called before the assignment to the "gnutls" global variable
494
if GnuTLS.check_version(self._need_version) is None:
495
raise GnuTLS.Error("Needs GnuTLS {} or later"
496
.format(self._need_version))
526
498
# Unless otherwise indicated, the constants and types below are
527
499
# all from the gnutls/gnutls.h C header file.
531
503
E_INTERRUPTED = -52
537
508
CRD_CERTIFICATE = 1
538
509
E_NO_CERTIFICATE_FOUND = -49
543
KEYID_USE_SHA256 = 1 # gnutls/x509.h
544
510
OPENPGP_FMT_RAW = 0 # gnutls/openpgp.h
547
513
class session_int(ctypes.Structure):
549
515
session_t = ctypes.POINTER(session_int)
551
516
class certificate_credentials_st(ctypes.Structure):
553
518
certificate_credentials_t = ctypes.POINTER(
554
519
certificate_credentials_st)
555
520
certificate_type_t = ctypes.c_int
557
521
class datum_t(ctypes.Structure):
558
522
_fields_ = [('data', ctypes.POINTER(ctypes.c_ubyte)),
559
523
('size', ctypes.c_uint)]
561
524
class openpgp_crt_int(ctypes.Structure):
563
526
openpgp_crt_t = ctypes.POINTER(openpgp_crt_int)
564
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
527
openpgp_crt_fmt_t = ctypes.c_int # gnutls/openpgp.h
565
528
log_func = ctypes.CFUNCTYPE(None, ctypes.c_int, ctypes.c_char_p)
566
529
credentials_type_t = ctypes.c_int
567
530
transport_ptr_t = ctypes.c_void_p
568
531
close_request_t = ctypes.c_int
571
534
class Error(Exception):
572
535
# We need to use the class name "GnuTLS" here, since this
573
536
# exception might be raised from within GnuTLS.__init__,
574
537
# which is called before the assignment to the "gnutls"
575
538
# global variable has happened.
576
def __init__(self, message=None, code=None, args=()):
539
def __init__(self, message = None, code = None, args=()):
577
540
# Default usage is by a message string, but if a return
578
541
# code is passed, convert it to a string with
579
542
# gnutls.strerror()
654
611
return _error_code(result)
655
612
result = func(*arguments)
658
615
# Unless otherwise indicated, the function declarations below are
659
616
# all from the gnutls/gnutls.h C header file.
662
619
priority_set_direct = _library.gnutls_priority_set_direct
663
620
priority_set_direct.argtypes = [session_t, ctypes.c_char_p,
664
621
ctypes.POINTER(ctypes.c_char_p)]
665
622
priority_set_direct.restype = _error_code
667
624
init = _library.gnutls_init
668
625
init.argtypes = [ctypes.POINTER(session_t), ctypes.c_int]
669
626
init.restype = _error_code
671
628
set_default_priority = _library.gnutls_set_default_priority
672
629
set_default_priority.argtypes = [session_t]
673
630
set_default_priority.restype = _error_code
675
632
record_send = _library.gnutls_record_send
676
633
record_send.argtypes = [session_t, ctypes.c_void_p,
678
635
record_send.restype = ctypes.c_ssize_t
679
636
record_send.errcheck = _retry_on_error
681
638
certificate_allocate_credentials = (
682
639
_library.gnutls_certificate_allocate_credentials)
683
640
certificate_allocate_credentials.argtypes = [
684
641
ctypes.POINTER(certificate_credentials_t)]
685
642
certificate_allocate_credentials.restype = _error_code
687
644
certificate_free_credentials = (
688
645
_library.gnutls_certificate_free_credentials)
689
certificate_free_credentials.argtypes = [
690
certificate_credentials_t]
646
certificate_free_credentials.argtypes = [certificate_credentials_t]
691
647
certificate_free_credentials.restype = None
693
649
handshake_set_private_extensions = (
694
650
_library.gnutls_handshake_set_private_extensions)
695
651
handshake_set_private_extensions.argtypes = [session_t,
697
653
handshake_set_private_extensions.restype = None
699
655
credentials_set = _library.gnutls_credentials_set
700
656
credentials_set.argtypes = [session_t, credentials_type_t,
702
658
credentials_set.restype = _error_code
704
660
strerror = _library.gnutls_strerror
705
661
strerror.argtypes = [ctypes.c_int]
706
662
strerror.restype = ctypes.c_char_p
708
664
certificate_type_get = _library.gnutls_certificate_type_get
709
665
certificate_type_get.argtypes = [session_t]
710
666
certificate_type_get.restype = _error_code
712
668
certificate_get_peers = _library.gnutls_certificate_get_peers
713
669
certificate_get_peers.argtypes = [session_t,
714
670
ctypes.POINTER(ctypes.c_uint)]
715
671
certificate_get_peers.restype = ctypes.POINTER(datum_t)
717
673
global_set_log_level = _library.gnutls_global_set_log_level
718
674
global_set_log_level.argtypes = [ctypes.c_int]
719
675
global_set_log_level.restype = None
721
677
global_set_log_function = _library.gnutls_global_set_log_function
722
678
global_set_log_function.argtypes = [log_func]
723
679
global_set_log_function.restype = None
725
681
deinit = _library.gnutls_deinit
726
682
deinit.argtypes = [session_t]
727
683
deinit.restype = None
729
685
handshake = _library.gnutls_handshake
730
686
handshake.argtypes = [session_t]
731
687
handshake.restype = _error_code
732
688
handshake.errcheck = _retry_on_error
734
690
transport_set_ptr = _library.gnutls_transport_set_ptr
735
691
transport_set_ptr.argtypes = [session_t, transport_ptr_t]
736
692
transport_set_ptr.restype = None
738
694
bye = _library.gnutls_bye
739
695
bye.argtypes = [session_t, close_request_t]
740
696
bye.restype = _error_code
741
697
bye.errcheck = _retry_on_error
743
699
check_version = _library.gnutls_check_version
744
700
check_version.argtypes = [ctypes.c_char_p]
745
701
check_version.restype = ctypes.c_char_p
747
has_rawpk = bool(check_version(_tls_rawpk_version))
751
class pubkey_st(ctypes.Structure):
753
pubkey_t = ctypes.POINTER(pubkey_st)
755
x509_crt_fmt_t = ctypes.c_int
757
# All the function declarations below are from gnutls/abstract.h
758
pubkey_init = _library.gnutls_pubkey_init
759
pubkey_init.argtypes = [ctypes.POINTER(pubkey_t)]
760
pubkey_init.restype = _error_code
762
pubkey_import = _library.gnutls_pubkey_import
763
pubkey_import.argtypes = [pubkey_t, ctypes.POINTER(datum_t),
765
pubkey_import.restype = _error_code
767
pubkey_get_key_id = _library.gnutls_pubkey_get_key_id
768
pubkey_get_key_id.argtypes = [pubkey_t, ctypes.c_int,
769
ctypes.POINTER(ctypes.c_ubyte),
770
ctypes.POINTER(ctypes.c_size_t)]
771
pubkey_get_key_id.restype = _error_code
773
pubkey_deinit = _library.gnutls_pubkey_deinit
774
pubkey_deinit.argtypes = [pubkey_t]
775
pubkey_deinit.restype = None
777
# All the function declarations below are from gnutls/openpgp.h
779
openpgp_crt_init = _library.gnutls_openpgp_crt_init
780
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
781
openpgp_crt_init.restype = _error_code
783
openpgp_crt_import = _library.gnutls_openpgp_crt_import
784
openpgp_crt_import.argtypes = [openpgp_crt_t,
785
ctypes.POINTER(datum_t),
787
openpgp_crt_import.restype = _error_code
789
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
790
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
791
ctypes.POINTER(ctypes.c_uint)]
792
openpgp_crt_verify_self.restype = _error_code
794
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
795
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
796
openpgp_crt_deinit.restype = None
798
openpgp_crt_get_fingerprint = (
799
_library.gnutls_openpgp_crt_get_fingerprint)
800
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
804
openpgp_crt_get_fingerprint.restype = _error_code
806
if check_version("3.6.4"):
807
certificate_type_get2 = _library.gnutls_certificate_type_get2
808
certificate_type_get2.argtypes = [session_t, ctypes.c_int]
809
certificate_type_get2.restype = _error_code
703
# All the function declarations below are from gnutls/openpgp.h
705
openpgp_crt_init = _library.gnutls_openpgp_crt_init
706
openpgp_crt_init.argtypes = [ctypes.POINTER(openpgp_crt_t)]
707
openpgp_crt_init.restype = _error_code
709
openpgp_crt_import = _library.gnutls_openpgp_crt_import
710
openpgp_crt_import.argtypes = [openpgp_crt_t,
711
ctypes.POINTER(datum_t),
713
openpgp_crt_import.restype = _error_code
715
openpgp_crt_verify_self = _library.gnutls_openpgp_crt_verify_self
716
openpgp_crt_verify_self.argtypes = [openpgp_crt_t, ctypes.c_uint,
717
ctypes.POINTER(ctypes.c_uint)]
718
openpgp_crt_verify_self.restype = _error_code
720
openpgp_crt_deinit = _library.gnutls_openpgp_crt_deinit
721
openpgp_crt_deinit.argtypes = [openpgp_crt_t]
722
openpgp_crt_deinit.restype = None
724
openpgp_crt_get_fingerprint = (
725
_library.gnutls_openpgp_crt_get_fingerprint)
726
openpgp_crt_get_fingerprint.argtypes = [openpgp_crt_t,
730
openpgp_crt_get_fingerprint.restype = _error_code
811
732
# Remove non-public functions
812
733
del _error_code, _retry_on_error
813
734
# Create the global "gnutls" object, simulating a module
814
735
gnutls = GnuTLS()
817
737
def call_pipe(connection, # : multiprocessing.Connection
818
738
func, *args, **kwargs):
819
739
"""This function is meant to be called by multiprocessing.Process
821
741
This function runs func(*args, **kwargs), and writes the resulting
822
742
return value on the provided multiprocessing.Connection.
824
744
connection.send(func(*args, **kwargs))
825
745
connection.close()
828
747
class Client(object):
829
748
"""A representation of a client host served by this server.
832
751
approved: bool(); 'None' if not yet approved/disapproved
833
752
approval_delay: datetime.timedelta(); Time to wait for approval
1514
1428
exc_info=error)
1515
1429
return xmlstring
1519
1432
dbus.OBJECT_MANAGER_IFACE
1520
1433
except AttributeError:
1521
1434
dbus.OBJECT_MANAGER_IFACE = "org.freedesktop.DBus.ObjectManager"
1524
1436
class DBusObjectWithObjectManager(DBusObjectWithAnnotations):
1525
1437
"""A D-Bus object with an ObjectManager.
1527
1439
Classes inheriting from this exposes the standard
1528
1440
GetManagedObjects call and the InterfacesAdded and
1529
1441
InterfacesRemoved signals on the standard
1530
1442
"org.freedesktop.DBus.ObjectManager" interface.
1532
1444
Note: No signals are sent automatically; they must be sent
1535
1447
@dbus.service.method(dbus.OBJECT_MANAGER_IFACE,
1536
out_signature="a{oa{sa{sv}}}")
1448
out_signature = "a{oa{sa{sv}}}")
1537
1449
def GetManagedObjects(self):
1538
1450
"""This function must be overridden"""
1539
1451
raise NotImplementedError()
1541
1453
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE,
1542
signature="oa{sa{sv}}")
1454
signature = "oa{sa{sv}}")
1543
1455
def InterfacesAdded(self, object_path, interfaces_and_properties):
1546
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature="oas")
1458
@dbus.service.signal(dbus.OBJECT_MANAGER_IFACE, signature = "oas")
1547
1459
def InterfacesRemoved(self, object_path, interfaces):
1550
1462
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
1552
path_keyword='object_path',
1553
connection_keyword='connection')
1463
out_signature = "s",
1464
path_keyword = 'object_path',
1465
connection_keyword = 'connection')
1554
1466
def Introspect(self, object_path, connection):
1555
1467
"""Overloading of standard D-Bus method.
1557
1469
Override return argument name of GetManagedObjects to be
1558
1470
"objpath_interfaces_and_properties"
1853
1762
"ApprovedByDefault")
1854
1763
approval_delay = notifychangeproperty(
1855
1764
dbus.UInt64, "ApprovalDelay",
1856
type_func=lambda td: td.total_seconds() * 1000)
1765
type_func = lambda td: td.total_seconds() * 1000)
1857
1766
approval_duration = notifychangeproperty(
1858
1767
dbus.UInt64, "ApprovalDuration",
1859
type_func=lambda td: td.total_seconds() * 1000)
1768
type_func = lambda td: td.total_seconds() * 1000)
1860
1769
host = notifychangeproperty(dbus.String, "Host")
1861
1770
timeout = notifychangeproperty(
1862
1771
dbus.UInt64, "Timeout",
1863
type_func=lambda td: td.total_seconds() * 1000)
1772
type_func = lambda td: td.total_seconds() * 1000)
1864
1773
extended_timeout = notifychangeproperty(
1865
1774
dbus.UInt64, "ExtendedTimeout",
1866
type_func=lambda td: td.total_seconds() * 1000)
1775
type_func = lambda td: td.total_seconds() * 1000)
1867
1776
interval = notifychangeproperty(
1868
1777
dbus.UInt64, "Interval",
1869
type_func=lambda td: td.total_seconds() * 1000)
1778
type_func = lambda td: td.total_seconds() * 1000)
1870
1779
checker_command = notifychangeproperty(dbus.String, "Checker")
1871
1780
secret = notifychangeproperty(dbus.ByteArray, "Secret",
1872
1781
invalidate_only=True)
1874
1783
del notifychangeproperty
1876
1785
def __del__(self, *args, **kwargs):
1878
1787
self.remove_from_connection()
2250
2152
class ClientHandler(socketserver.BaseRequestHandler, object):
2251
2153
"""A class to handle client connections.
2253
2155
Instantiated once for each connection to handle it.
2254
2156
Note: This will run in its own forked process."""
2256
2158
def handle(self):
2257
2159
with contextlib.closing(self.server.child_pipe) as child_pipe:
2258
2160
logger.info("TCP connection from: %s",
2259
2161
str(self.client_address))
2260
2162
logger.debug("Pipe FD: %d",
2261
2163
self.server.child_pipe.fileno())
2263
2165
session = gnutls.ClientSession(self.request)
2265
# priority = ':'.join(("NONE", "+VERS-TLS1.1",
2266
# "+AES-256-CBC", "+SHA1",
2267
# "+COMP-NULL", "+CTYPE-OPENPGP",
2167
#priority = ':'.join(("NONE", "+VERS-TLS1.1",
2168
# "+AES-256-CBC", "+SHA1",
2169
# "+COMP-NULL", "+CTYPE-OPENPGP",
2269
2171
# Use a fallback default, since this MUST be set.
2270
2172
priority = self.server.gnutls_priority
2271
2173
if priority is None:
2272
2174
priority = "NORMAL"
2273
gnutls.priority_set_direct(session._c_object,
2274
priority.encode("utf-8"),
2175
gnutls.priority_set_direct(session._c_object, priority,
2277
2178
# Start communication using the Mandos protocol
2278
2179
# Get protocol number
2279
2180
line = self.request.makefile().readline()
2427
2306
cert = cert_list[0]
2428
2307
return ctypes.string_at(cert.data, cert.size)
2431
def key_id(certificate):
2432
"Convert a certificate bytestring to a hexdigit key ID"
2433
# New GnuTLS "datum" with the public key
2434
datum = gnutls.datum_t(
2435
ctypes.cast(ctypes.c_char_p(certificate),
2436
ctypes.POINTER(ctypes.c_ubyte)),
2437
ctypes.c_uint(len(certificate)))
2438
# XXX all these need to be created in the gnutls "module"
2439
# New empty GnuTLS certificate
2440
pubkey = gnutls.pubkey_t()
2441
gnutls.pubkey_init(ctypes.byref(pubkey))
2442
# Import the raw public key into the certificate
2443
gnutls.pubkey_import(pubkey,
2444
ctypes.byref(datum),
2445
gnutls.X509_FMT_DER)
2446
# New buffer for the key ID
2447
buf = ctypes.create_string_buffer(32)
2448
buf_len = ctypes.c_size_t(len(buf))
2449
# Get the key ID from the raw public key into the buffer
2450
gnutls.pubkey_get_key_id(pubkey,
2451
gnutls.KEYID_USE_SHA256,
2452
ctypes.cast(ctypes.byref(buf),
2453
ctypes.POINTER(ctypes.c_ubyte)),
2454
ctypes.byref(buf_len))
2455
# Deinit the certificate
2456
gnutls.pubkey_deinit(pubkey)
2458
# Convert the buffer to a Python bytestring
2459
key_id = ctypes.string_at(buf, buf_len.value)
2460
# Convert the bytestring to hexadecimal notation
2461
hex_key_id = binascii.hexlify(key_id).upper()
2465
2310
def fingerprint(openpgp):
2466
2311
"Convert an OpenPGP bytestring to a hexdigit fingerprint"
2501
2345
class MultiprocessingMixIn(object):
2502
2346
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
2504
2348
def sub_process_main(self, request, address):
2506
2350
self.finish_request(request, address)
2507
2351
except Exception:
2508
2352
self.handle_error(request, address)
2509
2353
self.close_request(request)
2511
2355
def process_request(self, request, address):
2512
2356
"""Start a new process to process the request."""
2513
proc = multiprocessing.Process(target=self.sub_process_main,
2514
args=(request, address))
2357
proc = multiprocessing.Process(target = self.sub_process_main,
2358
args = (request, address))
2519
2363
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
2520
2364
""" adds a pipe to the MixIn """
2522
2366
def process_request(self, request, client_address):
2523
2367
"""Overrides and wraps the original process_request().
2525
2369
This function creates a new pipe in self.pipe
2527
2371
parent_pipe, self.child_pipe = multiprocessing.Pipe()
2529
2373
proc = MultiprocessingMixIn.process_request(self, request,
2530
2374
client_address)
2531
2375
self.child_pipe.close()
2532
2376
self.add_pipe(parent_pipe, proc)
2534
2378
def add_pipe(self, parent_pipe, proc):
2535
2379
"""Dummy function; override as necessary"""
2536
2380
raise NotImplementedError()
2585
2428
# socket_wrapper(), if socketfd was set.
2586
2429
socketserver.TCPServer.__init__(self, server_address,
2587
2430
RequestHandlerClass)
2589
2432
def server_bind(self):
2590
2433
"""This overrides the normal server_bind() function
2591
2434
to bind to an interface if one was specified, and also NOT to
2592
2435
bind to an address or port if they were not specified."""
2593
global SO_BINDTODEVICE
2594
2436
if self.interface is not None:
2595
2437
if SO_BINDTODEVICE is None:
2596
# Fall back to a hard-coded value which seems to be
2598
logger.warning("SO_BINDTODEVICE not found, trying 25")
2599
SO_BINDTODEVICE = 25
2601
self.socket.setsockopt(
2602
socket.SOL_SOCKET, SO_BINDTODEVICE,
2603
(self.interface + "\0").encode("utf-8"))
2604
except socket.error as error:
2605
if error.errno == errno.EPERM:
2606
logger.error("No permission to bind to"
2607
" interface %s", self.interface)
2608
elif error.errno == errno.ENOPROTOOPT:
2609
logger.error("SO_BINDTODEVICE not available;"
2610
" cannot bind to interface %s",
2612
elif error.errno == errno.ENODEV:
2613
logger.error("Interface %s does not exist,"
2614
" cannot bind", self.interface)
2438
logger.error("SO_BINDTODEVICE does not exist;"
2439
" cannot bind to interface %s",
2443
self.socket.setsockopt(
2444
socket.SOL_SOCKET, SO_BINDTODEVICE,
2445
(self.interface + "\0").encode("utf-8"))
2446
except socket.error as error:
2447
if error.errno == errno.EPERM:
2448
logger.error("No permission to bind to"
2449
" interface %s", self.interface)
2450
elif error.errno == errno.ENOPROTOOPT:
2451
logger.error("SO_BINDTODEVICE not available;"
2452
" cannot bind to interface %s",
2454
elif error.errno == errno.ENODEV:
2455
logger.error("Interface %s does not exist,"
2456
" cannot bind", self.interface)
2617
2459
# Only bind(2) the socket if we really need to.
2618
2460
if self.server_address[0] or self.server_address[1]:
2619
2461
if not self.server_address[0]:
2620
2462
if self.address_family == socket.AF_INET6:
2621
any_address = "::" # in6addr_any
2463
any_address = "::" # in6addr_any
2623
any_address = "0.0.0.0" # INADDR_ANY
2465
any_address = "0.0.0.0" # INADDR_ANY
2624
2466
self.server_address = (any_address,
2625
2467
self.server_address[1])
2626
2468
elif not self.server_address[1]:
2660
2502
self.gnutls_priority = gnutls_priority
2661
2503
IPv6_TCPServer.__init__(self, server_address,
2662
2504
RequestHandlerClass,
2663
interface=interface,
2505
interface = interface,
2506
use_ipv6 = use_ipv6,
2507
socketfd = socketfd)
2667
2509
def server_activate(self):
2668
2510
if self.enabled:
2669
2511
return socketserver.TCPServer.server_activate(self)
2671
2513
def enable(self):
2672
2514
self.enabled = True
2674
2516
def add_pipe(self, parent_pipe, proc):
2675
2517
# Call "handle_ipc" for both data and EOF events
2676
2518
GLib.io_add_watch(
2677
2519
parent_pipe.fileno(),
2678
2520
GLib.IO_IN | GLib.IO_HUP,
2679
2521
functools.partial(self.handle_ipc,
2680
parent_pipe=parent_pipe,
2522
parent_pipe = parent_pipe,
2683
2525
def handle_ipc(self, source, condition,
2684
2526
parent_pipe=None,
2686
2528
client_object=None):
2687
2529
# error, or the other end of multiprocessing.Pipe has closed
2688
2530
if condition & (GLib.IO_ERR | GLib.IO_HUP):
2689
2531
# Wait for other process to exit
2693
2535
# Read a request from the child
2694
2536
request = parent_pipe.recv()
2695
2537
command = request[0]
2697
2539
if command == 'init':
2698
key_id = request[1].decode("ascii")
2699
fpr = request[2].decode("ascii")
2700
address = request[3]
2541
address = request[2]
2702
2543
for c in self.clients.values():
2703
if key_id == "E3B0C44298FC1C149AFBF4C8996FB92427AE41E4649B934CA495991B7852B855":
2705
if key_id and c.key_id == key_id:
2708
if fpr and c.fingerprint == fpr:
2544
if c.fingerprint == fpr:
2712
logger.info("Client not found for key ID: %s, address"
2713
": %s", key_id or fpr, address)
2548
logger.info("Client not found for fingerprint: %s, ad"
2549
"dress: %s", fpr, address)
2714
2550
if self.use_dbus:
2715
2551
# Emit D-Bus signal
2716
mandos_dbus_service.ClientNotFound(key_id or fpr,
2552
mandos_dbus_service.ClientNotFound(fpr,
2718
2554
parent_pipe.send(False)
2721
2557
GLib.io_add_watch(
2722
2558
parent_pipe.fileno(),
2723
2559
GLib.IO_IN | GLib.IO_HUP,
2724
2560
functools.partial(self.handle_ipc,
2725
parent_pipe=parent_pipe,
2727
client_object=client))
2561
parent_pipe = parent_pipe,
2563
client_object = client))
2728
2564
parent_pipe.send(True)
2729
2565
# remove the old hook in favor of the new above hook on
3192
3019
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
3193
3020
service = AvahiServiceToSyslog(
3194
name=server_settings["servicename"],
3195
servicetype="_mandos._tcp",
3021
name = server_settings["servicename"],
3022
servicetype = "_mandos._tcp",
3023
protocol = protocol,
3198
3025
if server_settings["interface"]:
3199
3026
service.interface = if_nametoindex(
3200
3027
server_settings["interface"].encode("utf-8"))
3202
3029
global multiprocessing_manager
3203
3030
multiprocessing_manager = multiprocessing.Manager()
3205
3032
client_class = Client
3207
client_class = functools.partial(ClientDBus, bus=bus)
3034
client_class = functools.partial(ClientDBus, bus = bus)
3209
3036
client_settings = Client.config_parser(client_config)
3210
3037
old_client_settings = {}
3211
3038
clients_data = {}
3213
3040
# This is used to redirect stdout and stderr for checker processes
3215
wnull = open(os.devnull, "w") # A writable /dev/null
3042
wnull = open(os.devnull, "w") # A writable /dev/null
3216
3043
# Only used if server is running in foreground but not in debug
3218
3045
if debug or not foreground:
3221
3048
# Get client data and settings from last running state.
3222
3049
if server_settings["restore"]:
3224
3051
with open(stored_state_path, "rb") as stored_state:
3225
if sys.version_info.major == 2:
3052
if sys.version_info.major == 2:
3226
3053
clients_data, old_client_settings = pickle.load(
3229
3056
bytes_clients_data, bytes_old_client_settings = (
3230
pickle.load(stored_state, encoding="bytes"))
3231
# Fix bytes to strings
3057
pickle.load(stored_state, encoding = "bytes"))
3058
### Fix bytes to strings
3234
clients_data = {(key.decode("utf-8")
3235
if isinstance(key, bytes)
3238
bytes_clients_data.items()}
3061
clients_data = { (key.decode("utf-8")
3062
if isinstance(key, bytes)
3065
bytes_clients_data.items() }
3239
3066
del bytes_clients_data
3240
3067
for key in clients_data:
3241
value = {(k.decode("utf-8")
3242
if isinstance(k, bytes) else k): v
3244
clients_data[key].items()}
3068
value = { (k.decode("utf-8")
3069
if isinstance(k, bytes) else k): v
3071
clients_data[key].items() }
3245
3072
clients_data[key] = value
3246
3073
# .client_structure
3247
3074
value["client_structure"] = [
3248
3075
(s.decode("utf-8")
3249
3076
if isinstance(s, bytes)
3250
3077
else s) for s in
3251
value["client_structure"]]
3078
value["client_structure"] ]
3252
3079
# .name & .host
3253
3080
for k in ("name", "host"):
3254
3081
if isinstance(value[k], bytes):
3255
3082
value[k] = value[k].decode("utf-8")
3256
if not value.has_key("key_id"):
3257
value["key_id"] = ""
3258
elif not value.has_key("fingerprint"):
3259
value["fingerprint"] = ""
3260
# old_client_settings
3083
## old_client_settings
3262
3085
old_client_settings = {
3263
3086
(key.decode("utf-8")
3264
3087
if isinstance(key, bytes)
3265
3088
else key): value
3266
3089
for key, value in
3267
bytes_old_client_settings.items()}
3090
bytes_old_client_settings.items() }
3268
3091
del bytes_old_client_settings
3270
3093
for value in old_client_settings.values():