11
11
# "AvahiService" class, and some lines in "main".
13
13
# Everything else is
14
# Copyright © 2008-2012 Teddy Hogeborn
15
# Copyright © 2008-2012 Björn Påhlsson
14
# Copyright © 2008-2014 Teddy Hogeborn
15
# Copyright © 2008-2014 Björn Påhlsson
17
17
# This program is free software: you can redistribute it and/or modify
18
18
# it under the terms of the GNU General Public License as published by
88
88
except ImportError:
89
89
SO_BINDTODEVICE = None
92
92
stored_state_file = "clients.pickle"
94
94
logger = logging.getLogger()
95
syslogger = (logging.handlers.SysLogHandler
96
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
97
address = str("/dev/log")))
100
98
if_nametoindex = (ctypes.cdll.LoadLibrary
116
114
def initlogger(debug, level=logging.WARNING):
117
115
"""init logger and add loglevel"""
117
syslogger = (logging.handlers.SysLogHandler
119
logging.handlers.SysLogHandler.LOG_DAEMON,
120
address = str("/dev/log")))
119
121
syslogger.setFormatter(logging.Formatter
120
122
('Mandos [%(process)d]: %(levelname)s:'
172
174
def password_encode(self, password):
173
175
# Passphrase can not be empty and can not contain newlines or
174
176
# NUL bytes. So we prefix it and hex encode it.
175
return b"mandos" + binascii.hexlify(password)
177
encoded = b"mandos" + binascii.hexlify(password)
178
if len(encoded) > 2048:
179
# GnuPG can't handle long passwords, so encode differently
180
encoded = (b"mandos" + password.replace(b"\\", b"\\\\")
181
.replace(b"\n", b"\\n")
182
.replace(b"\0", b"\\x00"))
177
185
def encrypt(self, data, password):
178
186
passphrase = self.password_encode(password)
440
448
runtime_expansions: Allowed attributes for runtime expansion.
441
449
expires: datetime.datetime(); time (UTC) when a client will be
442
450
disabled, or None
451
server_settings: The server_settings dict from main()
445
454
runtime_expansions = ("approval_delay", "approval_duration",
523
def __init__(self, settings, name = None):
532
def __init__(self, settings, name = None, server_settings=None):
534
if server_settings is None:
536
self.server_settings = server_settings
525
537
# adding all client settings
526
538
for setting, value in settings.iteritems():
527
539
setattr(self, setting, value)
680
692
# If a checker exists, make sure it is not a zombie
682
694
pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
683
except (AttributeError, OSError) as error:
684
if (isinstance(error, OSError)
685
and error.errno != errno.ECHILD):
695
except AttributeError:
697
except OSError as error:
698
if error.errno != errno.ECHILD:
689
702
logger.warning("Checker was a zombie")
711
724
# in normal mode, that is already done by daemon(),
712
725
# and in debug mode we don't want to. (Stdin is
713
726
# always replaced by /dev/null.)
727
# The exception is when not debugging but nevertheless
728
# running in the foreground; use the previously
731
if (not self.server_settings["debug"]
732
and self.server_settings["foreground"]):
733
popen_args.update({"stdout": wnull,
714
735
self.checker = subprocess.Popen(command,
717
739
except OSError as error:
718
740
logger.error("Failed to start subprocess",
914
936
# The byte_arrays option is not supported yet on
915
937
# signatures other than "ay".
916
938
if prop._dbus_signature != "ay":
939
raise ValueError("Byte arrays not supported for non-"
940
"'ay' signature {0!r}"
941
.format(prop._dbus_signature))
918
942
value = dbus.ByteArray(b''.join(chr(byte)
919
943
for byte in value))
1328
1352
*args, **kwargs)
1330
1354
def start_checker(self, *args, **kwargs):
1331
old_checker = self.checker
1332
if self.checker is not None:
1333
old_checker_pid = self.checker.pid
1335
old_checker_pid = None
1355
old_checker_pid = getattr(self.checker, "pid", None)
1336
1356
r = Client.start_checker(self, *args, **kwargs)
1337
1357
# Only if new checker process was started
1338
1358
if (self.checker is not None
1683
1703
logger.debug("Protocol version: %r", line)
1685
1705
if int(line.strip().split()[0]) > 1:
1706
raise RuntimeError(line)
1687
1707
except (ValueError, IndexError, RuntimeError) as error:
1688
1708
logger.error("Unknown protocol version: %s", error)
1897
1917
def add_pipe(self, parent_pipe, proc):
1898
1918
"""Dummy function; override as necessary"""
1899
raise NotImplementedError
1919
raise NotImplementedError()
1902
1922
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
1978
1998
if self.address_family == socket.AF_INET6:
1979
1999
any_address = "::" # in6addr_any
1981
any_address = socket.INADDR_ANY
2001
any_address = "0.0.0.0" # INADDR_ANY
1982
2002
self.server_address = (any_address,
1983
2003
self.server_address[1])
1984
2004
elif not self.server_address[1]:
2289
2309
help="Run self-test")
2290
2310
parser.add_argument("--debug", action="store_true",
2291
2311
help="Debug mode; run in foreground and log"
2312
" to terminal", default=None)
2293
2313
parser.add_argument("--debuglevel", metavar="LEVEL",
2294
2314
help="Debug level for stdout output")
2295
2315
parser.add_argument("--priority", help="GnuTLS"
2303
2323
parser.add_argument("--no-dbus", action="store_false",
2304
2324
dest="use_dbus", help="Do not provide D-Bus"
2305
" system bus interface")
2325
" system bus interface", default=None)
2306
2326
parser.add_argument("--no-ipv6", action="store_false",
2307
dest="use_ipv6", help="Do not use IPv6")
2327
dest="use_ipv6", help="Do not use IPv6",
2308
2329
parser.add_argument("--no-restore", action="store_false",
2309
2330
dest="restore", help="Do not restore stored"
2331
" state", default=None)
2311
2332
parser.add_argument("--socket", type=int,
2312
2333
help="Specify a file descriptor to a network"
2313
2334
" socket to use instead of creating one")
2314
2335
parser.add_argument("--statedir", metavar="DIR",
2315
2336
help="Directory to save/restore state in")
2316
2337
parser.add_argument("--foreground", action="store_true",
2317
help="Run in foreground")
2338
help="Run in foreground", default=None)
2319
2340
options = parser.parse_args()
2321
2342
if options.check:
2344
fail_count, test_count = doctest.testmod()
2345
sys.exit(os.EX_OK if fail_count == 0 else 1)
2326
2347
# Default values for config file for server-global settings
2327
2348
server_defaults = { "interface": "",
2379
2400
for option in server_settings.keys():
2380
2401
if type(server_settings[option]) is str:
2381
2402
server_settings[option] = unicode(server_settings[option])
2403
# Force all boolean options to be boolean
2404
for option in ("debug", "use_dbus", "use_ipv6", "restore",
2406
server_settings[option] = bool(server_settings[option])
2382
2407
# Debug implies foreground
2383
2408
if server_settings["debug"]:
2384
2409
server_settings["foreground"] = True
2432
2457
socketfd=(server_settings["socket"]
2434
2459
if not foreground:
2435
pidfilename = "/var/run/mandos.pid"
2460
pidfilename = "/run/mandos.pid"
2461
if not os.path.isdir("/run/."):
2462
pidfilename = "/var/run/mandos.pid"
2438
2465
pidfile = open(pidfilename, "w")
2524
2551
old_client_settings = {}
2525
2552
clients_data = {}
2554
# This is used to redirect stdout and stderr for checker processes
2556
wnull = open(os.devnull, "w") # A writable /dev/null
2557
# Only used if server is running in foreground but not in debug
2559
if debug or not foreground:
2527
2562
# Get client data and settings from last running state.
2528
2563
if server_settings["restore"]:
2546
2581
with PGPEngine() as pgp:
2547
2582
for client_name, client in clients_data.iteritems():
2583
# Skip removed clients
2584
if client_name not in client_settings:
2548
2587
# Decide which value to use after restoring saved state.
2549
2588
# We have three different values: Old config file,
2550
2589
# new config file, and saved state.
2612
2651
# Create all client objects
2613
2652
for client_name, client in clients_data.iteritems():
2614
2653
tcp_server.clients[client_name] = client_class(
2615
name = client_name, settings = client)
2654
name = client_name, settings = client,
2655
server_settings = server_settings)
2617
2657
if not tcp_server.clients:
2618
2658
logger.warning("No clients defined")
2718
2759
# A list of attributes that can not be pickled
2720
2761
exclude = set(("bus", "changedstate", "secret",
2762
"checker", "server_settings"))
2722
2763
for name, typ in (inspect.getmembers
2723
2764
(dbus.service.Object)):
2724
2765
exclude.add(name)