101
81
except ImportError:
102
82
SO_BINDTODEVICE = None
104
if sys.version_info.major == 2:
108
stored_state_file = "clients.pickle"
110
logger = logging.getLogger()
114
if_nametoindex = ctypes.cdll.LoadLibrary(
115
ctypes.util.find_library("c")).if_nametoindex
116
except (OSError, AttributeError):
118
def if_nametoindex(interface):
119
"Get an interface index the hard way, i.e. using fcntl()"
120
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
121
with contextlib.closing(socket.socket()) as s:
122
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
123
struct.pack(b"16s16x", interface))
124
interface_index = struct.unpack("I", ifreq[16:20])[0]
125
return interface_index
128
def initlogger(debug, level=logging.WARNING):
129
"""init logger and add loglevel"""
132
syslogger = (logging.handlers.SysLogHandler(
133
facility = logging.handlers.SysLogHandler.LOG_DAEMON,
134
address = "/dev/log"))
135
syslogger.setFormatter(logging.Formatter
136
('Mandos [%(process)d]: %(levelname)s:'
138
logger.addHandler(syslogger)
141
console = logging.StreamHandler()
142
console.setFormatter(logging.Formatter('%(asctime)s %(name)s'
146
logger.addHandler(console)
147
logger.setLevel(level)
150
class PGPError(Exception):
151
"""Exception if encryption/decryption fails"""
155
class PGPEngine(object):
156
"""A simple class for OpenPGP symmetric encryption & decryption"""
159
self.tempdir = tempfile.mkdtemp(prefix="mandos-")
160
self.gnupgargs = ['--batch',
161
'--home', self.tempdir,
169
def __exit__(self, exc_type, exc_value, traceback):
177
if self.tempdir is not None:
178
# Delete contents of tempdir
179
for root, dirs, files in os.walk(self.tempdir,
181
for filename in files:
182
os.remove(os.path.join(root, filename))
184
os.rmdir(os.path.join(root, dirname))
186
os.rmdir(self.tempdir)
189
def password_encode(self, password):
190
# Passphrase can not be empty and can not contain newlines or
191
# NUL bytes. So we prefix it and hex encode it.
192
encoded = b"mandos" + binascii.hexlify(password)
193
if len(encoded) > 2048:
194
# GnuPG can't handle long passwords, so encode differently
195
encoded = (b"mandos" + password.replace(b"\\", b"\\\\")
196
.replace(b"\n", b"\\n")
197
.replace(b"\0", b"\\x00"))
200
def encrypt(self, data, password):
201
passphrase = self.password_encode(password)
202
with tempfile.NamedTemporaryFile(
203
dir=self.tempdir) as passfile:
204
passfile.write(passphrase)
206
proc = subprocess.Popen(['gpg', '--symmetric',
210
stdin = subprocess.PIPE,
211
stdout = subprocess.PIPE,
212
stderr = subprocess.PIPE)
213
ciphertext, err = proc.communicate(input = data)
214
if proc.returncode != 0:
218
def decrypt(self, data, password):
219
passphrase = self.password_encode(password)
220
with tempfile.NamedTemporaryFile(
221
dir = self.tempdir) as passfile:
222
passfile.write(passphrase)
224
proc = subprocess.Popen(['gpg', '--decrypt',
228
stdin = subprocess.PIPE,
229
stdout = subprocess.PIPE,
230
stderr = subprocess.PIPE)
231
decrypted_plaintext, err = proc.communicate(input = data)
232
if proc.returncode != 0:
234
return decrypted_plaintext
87
#logger = logging.getLogger('mandos')
88
logger = logging.Logger('mandos')
89
syslogger = (logging.handlers.SysLogHandler
90
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
91
address = str("/dev/log")))
92
syslogger.setFormatter(logging.Formatter
93
('Mandos [%(process)d]: %(levelname)s:'
95
logger.addHandler(syslogger)
97
console = logging.StreamHandler()
98
console.setFormatter(logging.Formatter('%(name)s [%(process)d]:'
101
logger.addHandler(console)
237
103
class AvahiError(Exception):
238
104
def __init__(self, value, *args, **kwargs):
239
105
self.value = value
240
return super(AvahiError, self).__init__(value, *args,
106
super(AvahiError, self).__init__(value, *args, **kwargs)
107
def __unicode__(self):
108
return unicode(repr(self.value))
244
110
class AvahiServiceError(AvahiError):
248
113
class AvahiGroupError(AvahiError):
453
291
interval: datetime.timedelta(); How often to start a new checker
454
292
last_approval_request: datetime.datetime(); (UTC) or None
455
293
last_checked_ok: datetime.datetime(); (UTC) or None
456
last_checker_status: integer between 0 and 255 reflecting exit
457
status of last checker. -1 reflects crashed
458
checker, -2 means no checker completed yet.
459
last_enabled: datetime.datetime(); (UTC) or None
294
last_enabled: datetime.datetime(); (UTC)
460
295
name: string; from the config file, used in log messages and
461
296
D-Bus identifiers
462
297
secret: bytestring; sent verbatim (over TLS) to client
463
298
timeout: datetime.timedelta(); How long from last_checked_ok
464
299
until this client is disabled
465
extended_timeout: extra long timeout when secret has been sent
466
300
runtime_expansions: Allowed attributes for runtime expansion.
467
expires: datetime.datetime(); time (UTC) when a client will be
469
server_settings: The server_settings dict from main()
472
303
runtime_expansions = ("approval_delay", "approval_duration",
473
"created", "enabled", "expires",
474
"fingerprint", "host", "interval",
475
"last_approval_request", "last_checked_ok",
304
"created", "enabled", "fingerprint",
305
"host", "interval", "last_checked_ok",
476
306
"last_enabled", "name", "timeout")
479
"extended_timeout": "PT15M",
481
"checker": "fping -q -- %%(host)s",
483
"approval_delay": "PT0S",
484
"approval_duration": "PT1S",
485
"approved_by_default": "True",
490
def config_parser(config):
491
"""Construct a new dict of client settings of this form:
492
{ client_name: {setting_name: value, ...}, ...}
493
with exceptions for any special settings as defined above.
494
NOTE: Must be a pure function. Must return the same result
495
value given the same arguments.
498
for client_name in config.sections():
499
section = dict(config.items(client_name))
500
client = settings[client_name] = {}
502
client["host"] = section["host"]
503
# Reformat values from string types to Python types
504
client["approved_by_default"] = config.getboolean(
505
client_name, "approved_by_default")
506
client["enabled"] = config.getboolean(client_name,
509
# Uppercase and remove spaces from fingerprint for later
510
# comparison purposes with return value from the
511
# fingerprint() function
512
client["fingerprint"] = (section["fingerprint"].upper()
514
if "secret" in section:
515
client["secret"] = section["secret"].decode("base64")
516
elif "secfile" in section:
517
with open(os.path.expanduser(os.path.expandvars
518
(section["secfile"])),
520
client["secret"] = secfile.read()
522
raise TypeError("No secret or secfile for section {}"
524
client["timeout"] = string_to_delta(section["timeout"])
525
client["extended_timeout"] = string_to_delta(
526
section["extended_timeout"])
527
client["interval"] = string_to_delta(section["interval"])
528
client["approval_delay"] = string_to_delta(
529
section["approval_delay"])
530
client["approval_duration"] = string_to_delta(
531
section["approval_duration"])
532
client["checker_command"] = section["checker"]
533
client["last_approval_request"] = None
534
client["last_checked_ok"] = None
535
client["last_checker_status"] = -2
539
def __init__(self, settings, name = None, server_settings=None):
309
def _timedelta_to_milliseconds(td):
310
"Convert a datetime.timedelta() to milliseconds"
311
return ((td.days * 24 * 60 * 60 * 1000)
312
+ (td.seconds * 1000)
313
+ (td.microseconds // 1000))
315
def timeout_milliseconds(self):
316
"Return the 'timeout' attribute in milliseconds"
317
return self._timedelta_to_milliseconds(self.timeout)
319
def interval_milliseconds(self):
320
"Return the 'interval' attribute in milliseconds"
321
return self._timedelta_to_milliseconds(self.interval)
323
def approval_delay_milliseconds(self):
324
return self._timedelta_to_milliseconds(self.approval_delay)
326
def __init__(self, name = None, disable_hook=None, config=None):
327
"""Note: the 'checker' key in 'config' sets the
328
'checker_command' attribute and *not* the 'checker'
541
if server_settings is None:
543
self.server_settings = server_settings
544
# adding all client settings
545
for setting, value in settings.items():
546
setattr(self, setting, value)
549
if not hasattr(self, "last_enabled"):
550
self.last_enabled = datetime.datetime.utcnow()
551
if not hasattr(self, "expires"):
552
self.expires = (datetime.datetime.utcnow()
555
self.last_enabled = None
558
333
logger.debug("Creating client %r", self.name)
334
# Uppercase and remove spaces from fingerprint for later
335
# comparison purposes with return value from the fingerprint()
337
self.fingerprint = (config["fingerprint"].upper()
559
339
logger.debug(" Fingerprint: %s", self.fingerprint)
560
self.created = settings.get("created",
561
datetime.datetime.utcnow())
563
# attributes specific for this server instance
340
if "secret" in config:
341
self.secret = config["secret"].decode("base64")
342
elif "secfile" in config:
343
with open(os.path.expanduser(os.path.expandvars
344
(config["secfile"])),
346
self.secret = secfile.read()
348
raise TypeError("No secret or secfile for client %s"
350
self.host = config.get("host", "")
351
self.created = datetime.datetime.utcnow()
353
self.last_approval_request = None
354
self.last_enabled = None
355
self.last_checked_ok = None
356
self.timeout = string_to_delta(config["timeout"])
357
self.interval = string_to_delta(config["interval"])
358
self.disable_hook = disable_hook
564
359
self.checker = None
565
360
self.checker_initiator_tag = None
566
361
self.disable_initiator_tag = None
567
362
self.checker_callback_tag = None
363
self.checker_command = config["checker"]
568
364
self.current_checker_command = None
365
self.last_connect = None
366
self._approved = None
367
self.approved_by_default = config.get("approved_by_default",
570
369
self.approvals_pending = 0
571
self.changedstate = multiprocessing_manager.Condition(
572
multiprocessing_manager.Lock())
573
self.client_structure = [attr
574
for attr in self.__dict__.iterkeys()
575
if not attr.startswith("_")]
576
self.client_structure.append("client_structure")
578
for name, t in inspect.getmembers(
579
type(self), lambda obj: isinstance(obj, property)):
580
if not name.startswith("_"):
581
self.client_structure.append(name)
370
self.approval_delay = string_to_delta(
371
config["approval_delay"])
372
self.approval_duration = string_to_delta(
373
config["approval_duration"])
374
self.changedstate = multiprocessing_manager.Condition(multiprocessing_manager.Lock())
583
# Send notice to process children that client state has changed
584
376
def send_changedstate(self):
585
with self.changedstate:
586
self.changedstate.notify_all()
377
self.changedstate.acquire()
378
self.changedstate.notify_all()
379
self.changedstate.release()
588
381
def enable(self):
589
382
"""Start this client's checker and timeout hooks"""
590
383
if getattr(self, "enabled", False):
591
384
# Already enabled
593
self.expires = datetime.datetime.utcnow() + self.timeout
386
self.send_changedstate()
595
387
self.last_enabled = datetime.datetime.utcnow()
597
self.send_changedstate()
388
# Schedule a new checker to be started an 'interval' from now,
389
# and every interval from then on.
390
self.checker_initiator_tag = (gobject.timeout_add
391
(self.interval_milliseconds(),
393
# Schedule a disable() when 'timeout' has passed
394
self.disable_initiator_tag = (gobject.timeout_add
395
(self.timeout_milliseconds(),
398
# Also start a new checker *right now*.
599
401
def disable(self, quiet=True):
600
402
"""Disable this client."""
601
403
if not getattr(self, "enabled", False):
406
self.send_changedstate()
604
408
logger.info("Disabling client %s", self.name)
605
if getattr(self, "disable_initiator_tag", None) is not None:
409
if getattr(self, "disable_initiator_tag", False):
606
410
gobject.source_remove(self.disable_initiator_tag)
607
411
self.disable_initiator_tag = None
609
if getattr(self, "checker_initiator_tag", None) is not None:
412
if getattr(self, "checker_initiator_tag", False):
610
413
gobject.source_remove(self.checker_initiator_tag)
611
414
self.checker_initiator_tag = None
612
415
self.stop_checker()
416
if self.disable_hook:
417
self.disable_hook(self)
613
418
self.enabled = False
615
self.send_changedstate()
616
419
# Do not run this again if called by a gobject.timeout_add
619
422
def __del__(self):
423
self.disable_hook = None
622
def init_checker(self):
623
# Schedule a new checker to be started an 'interval' from now,
624
# and every interval from then on.
625
if self.checker_initiator_tag is not None:
626
gobject.source_remove(self.checker_initiator_tag)
627
self.checker_initiator_tag = gobject.timeout_add(
628
int(self.interval.total_seconds() * 1000),
630
# Schedule a disable() when 'timeout' has passed
631
if self.disable_initiator_tag is not None:
632
gobject.source_remove(self.disable_initiator_tag)
633
self.disable_initiator_tag = gobject.timeout_add(
634
int(self.timeout.total_seconds() * 1000), self.disable)
635
# Also start a new checker *right now*.
638
426
def checker_callback(self, pid, condition, command):
639
427
"""The checker has completed, so take appropriate actions."""
640
428
self.checker_callback_tag = None
641
429
self.checker = None
642
430
if os.WIFEXITED(condition):
643
self.last_checker_status = os.WEXITSTATUS(condition)
644
if self.last_checker_status == 0:
431
exitstatus = os.WEXITSTATUS(condition)
645
433
logger.info("Checker for %(name)s succeeded",
647
435
self.checked_ok()
649
logger.info("Checker for %(name)s failed", vars(self))
437
logger.info("Checker for %(name)s failed",
651
self.last_checker_status = -1
652
440
logger.warning("Checker for %(name)s crashed?",
655
443
def checked_ok(self):
656
"""Assert that the client has been seen, alive and well."""
444
"""Bump up the timeout for this client.
446
This should only be called when the client has been seen,
657
449
self.last_checked_ok = datetime.datetime.utcnow()
658
self.last_checker_status = 0
661
def bump_timeout(self, timeout=None):
662
"""Bump up the timeout for this client."""
664
timeout = self.timeout
665
if self.disable_initiator_tag is not None:
666
gobject.source_remove(self.disable_initiator_tag)
667
self.disable_initiator_tag = None
668
if getattr(self, "enabled", False):
669
self.disable_initiator_tag = gobject.timeout_add(
670
int(timeout.total_seconds() * 1000), self.disable)
671
self.expires = datetime.datetime.utcnow() + timeout
450
gobject.source_remove(self.disable_initiator_tag)
451
self.disable_initiator_tag = (gobject.timeout_add
452
(self.timeout_milliseconds(),
673
455
def need_approval(self):
674
456
self.last_approval_request = datetime.datetime.utcnow()
703
484
self.current_checker_command)
704
485
# Start a new checker if needed
705
486
if self.checker is None:
706
# Escape attributes for the shell
708
attr: re.escape(str(getattr(self, attr)))
709
for attr in self.runtime_expansions }
711
command = self.checker_command % escaped_attrs
712
except TypeError as error:
713
logger.error('Could not format string "%s"',
714
self.checker_command,
716
return True # Try again later
488
# In case checker_command has exactly one % operator
489
command = self.checker_command % self.host
491
# Escape attributes for the shell
492
escaped_attrs = dict(
494
re.escape(unicode(str(getattr(self, attr, "")),
498
self.runtime_expansions)
501
command = self.checker_command % escaped_attrs
502
except TypeError as error:
503
logger.error('Could not format string "%s":'
504
' %s', self.checker_command, error)
505
return True # Try again later
717
506
self.current_checker_command = command
719
logger.info("Starting checker %r for %s", command,
508
logger.info("Starting checker %r for %s",
721
510
# We don't need to redirect stdout and stderr, since
722
511
# in normal mode, that is already done by daemon(),
723
512
# and in debug mode we don't want to. (Stdin is
724
513
# always replaced by /dev/null.)
725
# The exception is when not debugging but nevertheless
726
# running in the foreground; use the previously
729
if (not self.server_settings["debug"]
730
and self.server_settings["foreground"]):
731
popen_args.update({"stdout": wnull,
733
514
self.checker = subprocess.Popen(command,
738
except OSError as error:
739
logger.error("Failed to start subprocess",
742
self.checker_callback_tag = gobject.child_watch_add(
743
self.checker.pid, self.checker_callback, data=command)
744
# The checker may have completed before the gobject
745
# watch was added. Check for this.
517
self.checker_callback_tag = (gobject.child_watch_add
519
self.checker_callback,
521
# The checker may have completed before the gobject
522
# watch was added. Check for this.
747
523
pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
525
gobject.source_remove(self.checker_callback_tag)
526
self.checker_callback(pid, status, command)
748
527
except OSError as error:
749
if error.errno == errno.ECHILD:
750
# This should never happen
751
logger.error("Child process vanished",
756
gobject.source_remove(self.checker_callback_tag)
757
self.checker_callback(pid, status, command)
528
logger.error("Failed to start subprocess: %s",
758
530
# Re-run this periodically if run by gobject.timeout_add
873
600
class DBusObjectWithProperties(dbus.service.Object):
874
601
"""A D-Bus object with properties.
876
603
Classes inheriting from this can use the dbus_service_property
877
604
decorator to expose methods as D-Bus properties. It exposes the
878
605
standard Get(), Set(), and GetAll() methods on the D-Bus.
882
def _is_dbus_thing(thing):
883
"""Returns a function testing if an attribute is a D-Bus thing
885
If called like _is_dbus_thing("method") it returns a function
886
suitable for use as predicate to inspect.getmembers().
888
return lambda obj: getattr(obj, "_dbus_is_{}".format(thing),
609
def _is_dbus_property(obj):
610
return getattr(obj, "_dbus_is_property", False)
891
def _get_all_dbus_things(self, thing):
612
def _get_all_dbus_properties(self):
892
613
"""Returns a generator of (name, attribute) pairs
894
return ((getattr(athing.__get__(self), "_dbus_name", name),
895
athing.__get__(self))
896
for cls in self.__class__.__mro__
898
inspect.getmembers(cls, self._is_dbus_thing(thing)))
615
return ((prop._dbus_name, prop)
617
inspect.getmembers(self, self._is_dbus_property))
900
619
def _get_dbus_property(self, interface_name, property_name):
901
620
"""Returns a bound method if one exists which is a D-Bus
902
621
property with the specified name and interface.
904
for cls in self.__class__.__mro__:
905
for name, value in inspect.getmembers(
906
cls, self._is_dbus_thing("property")):
907
if (value._dbus_name == property_name
908
and value._dbus_interface == interface_name):
909
return value.__get__(self)
623
for name in (property_name,
624
property_name + "_dbus_property"):
625
prop = getattr(self, name, None)
627
or not self._is_dbus_property(prop)
628
or prop._dbus_name != property_name
629
or (interface_name and prop._dbus_interface
630
and interface_name != prop._dbus_interface)):
911
633
# No such property
912
raise DBusPropertyNotFound("{}:{}.{}".format(
913
self.dbus_object_path, interface_name, property_name))
634
raise DBusPropertyNotFound(self.dbus_object_path + ":"
635
+ interface_name + "."
915
@dbus.service.method(dbus.PROPERTIES_IFACE,
638
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="ss",
917
639
out_signature="v")
918
640
def Get(self, interface_name, property_name):
919
641
"""Standard D-Bus property Get() method, see D-Bus standard.
966
685
if not hasattr(value, "variant_level"):
967
properties[name] = value
969
properties[name] = type(value)(
970
value, variant_level = value.variant_level + 1)
971
return dbus.Dictionary(properties, signature="sv")
973
@dbus.service.signal(dbus.PROPERTIES_IFACE, signature="sa{sv}as")
974
def PropertiesChanged(self, interface_name, changed_properties,
975
invalidated_properties):
976
"""Standard D-Bus PropertiesChanged() signal, see D-Bus
688
all[name] = type(value)(value, variant_level=
689
value.variant_level+1)
690
return dbus.Dictionary(all, signature="sv")
981
692
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
982
693
out_signature="s",
983
694
path_keyword='object_path',
984
695
connection_keyword='connection')
985
696
def Introspect(self, object_path, connection):
986
"""Overloading of standard D-Bus method.
988
Inserts property tags and interface annotation tags.
697
"""Standard D-Bus method, overloaded to insert property tags.
990
699
xmlstring = dbus.service.Object.Introspect(self, object_path,
993
702
document = xml.dom.minidom.parseString(xmlstring)
995
703
def make_tag(document, name, prop):
996
704
e = document.createElement("property")
997
705
e.setAttribute("name", name)
998
706
e.setAttribute("type", prop._dbus_signature)
999
707
e.setAttribute("access", prop._dbus_access)
1002
709
for if_tag in document.getElementsByTagName("interface"):
1004
710
for tag in (make_tag(document, name, prop)
1006
in self._get_all_dbus_things("property")
712
in self._get_all_dbus_properties()
1007
713
if prop._dbus_interface
1008
714
== if_tag.getAttribute("name")):
1009
715
if_tag.appendChild(tag)
1010
# Add annotation tags
1011
for typ in ("method", "signal", "property"):
1012
for tag in if_tag.getElementsByTagName(typ):
1014
for name, prop in (self.
1015
_get_all_dbus_things(typ)):
1016
if (name == tag.getAttribute("name")
1017
and prop._dbus_interface
1018
== if_tag.getAttribute("name")):
1019
annots.update(getattr(
1020
prop, "_dbus_annotations", {}))
1021
for name, value in annots.items():
1022
ann_tag = document.createElement(
1024
ann_tag.setAttribute("name", name)
1025
ann_tag.setAttribute("value", value)
1026
tag.appendChild(ann_tag)
1027
# Add interface annotation tags
1028
for annotation, value in dict(
1029
itertools.chain.from_iterable(
1030
annotations().items()
1031
for name, annotations
1032
in self._get_all_dbus_things("interface")
1033
if name == if_tag.getAttribute("name")
1035
ann_tag = document.createElement("annotation")
1036
ann_tag.setAttribute("name", annotation)
1037
ann_tag.setAttribute("value", value)
1038
if_tag.appendChild(ann_tag)
1039
716
# Add the names to the return values for the
1040
717
# "org.freedesktop.DBus.Properties" methods
1041
718
if (if_tag.getAttribute("name")
1056
733
except (AttributeError, xml.dom.DOMException,
1057
734
xml.parsers.expat.ExpatError) as error:
1058
735
logger.error("Failed to override Introspection method",
1060
737
return xmlstring
1063
def datetime_to_dbus(dt, variant_level=0):
1064
"""Convert a UTC datetime.datetime() to a D-Bus type."""
1066
return dbus.String("", variant_level = variant_level)
1067
return dbus.String(dt.isoformat(), variant_level=variant_level)
1070
def alternate_dbus_interfaces(alt_interface_names, deprecate=True):
1071
"""A class decorator; applied to a subclass of
1072
dbus.service.Object, it will add alternate D-Bus attributes with
1073
interface names according to the "alt_interface_names" mapping.
1076
@alternate_dbus_interfaces({"org.example.Interface":
1077
"net.example.AlternateInterface"})
1078
class SampleDBusObject(dbus.service.Object):
1079
@dbus.service.method("org.example.Interface")
1080
def SampleDBusMethod():
1083
The above "SampleDBusMethod" on "SampleDBusObject" will be
1084
reachable via two interfaces: "org.example.Interface" and
1085
"net.example.AlternateInterface", the latter of which will have
1086
its D-Bus annotation "org.freedesktop.DBus.Deprecated" set to
1087
"true", unless "deprecate" is passed with a False value.
1089
This works for methods and signals, and also for D-Bus properties
1090
(from DBusObjectWithProperties) and interfaces (from the
1091
dbus_interface_annotations decorator).
1095
for orig_interface_name, alt_interface_name in (
1096
alt_interface_names.items()):
1098
interface_names = set()
1099
# Go though all attributes of the class
1100
for attrname, attribute in inspect.getmembers(cls):
1101
# Ignore non-D-Bus attributes, and D-Bus attributes
1102
# with the wrong interface name
1103
if (not hasattr(attribute, "_dbus_interface")
1104
or not attribute._dbus_interface.startswith(
1105
orig_interface_name)):
1107
# Create an alternate D-Bus interface name based on
1109
alt_interface = attribute._dbus_interface.replace(
1110
orig_interface_name, alt_interface_name)
1111
interface_names.add(alt_interface)
1112
# Is this a D-Bus signal?
1113
if getattr(attribute, "_dbus_is_signal", False):
1114
# Extract the original non-method undecorated
1115
# function by black magic
1116
nonmethod_func = (dict(
1117
zip(attribute.func_code.co_freevars,
1118
attribute.__closure__))
1119
["func"].cell_contents)
1120
# Create a new, but exactly alike, function
1121
# object, and decorate it to be a new D-Bus signal
1122
# with the alternate D-Bus interface name
1123
new_function = (dbus.service.signal(
1124
alt_interface, attribute._dbus_signature)
1125
(types.FunctionType(
1126
nonmethod_func.func_code,
1127
nonmethod_func.func_globals,
1128
nonmethod_func.func_name,
1129
nonmethod_func.func_defaults,
1130
nonmethod_func.func_closure)))
1131
# Copy annotations, if any
1133
new_function._dbus_annotations = dict(
1134
attribute._dbus_annotations)
1135
except AttributeError:
1137
# Define a creator of a function to call both the
1138
# original and alternate functions, so both the
1139
# original and alternate signals gets sent when
1140
# the function is called
1141
def fixscope(func1, func2):
1142
"""This function is a scope container to pass
1143
func1 and func2 to the "call_both" function
1144
outside of its arguments"""
1146
def call_both(*args, **kwargs):
1147
"""This function will emit two D-Bus
1148
signals by calling func1 and func2"""
1149
func1(*args, **kwargs)
1150
func2(*args, **kwargs)
1153
# Create the "call_both" function and add it to
1155
attr[attrname] = fixscope(attribute, new_function)
1156
# Is this a D-Bus method?
1157
elif getattr(attribute, "_dbus_is_method", False):
1158
# Create a new, but exactly alike, function
1159
# object. Decorate it to be a new D-Bus method
1160
# with the alternate D-Bus interface name. Add it
1163
dbus.service.method(
1165
attribute._dbus_in_signature,
1166
attribute._dbus_out_signature)
1167
(types.FunctionType(attribute.func_code,
1168
attribute.func_globals,
1169
attribute.func_name,
1170
attribute.func_defaults,
1171
attribute.func_closure)))
1172
# Copy annotations, if any
1174
attr[attrname]._dbus_annotations = dict(
1175
attribute._dbus_annotations)
1176
except AttributeError:
1178
# Is this a D-Bus property?
1179
elif getattr(attribute, "_dbus_is_property", False):
1180
# Create a new, but exactly alike, function
1181
# object, and decorate it to be a new D-Bus
1182
# property with the alternate D-Bus interface
1183
# name. Add it to the class.
1184
attr[attrname] = (dbus_service_property(
1185
alt_interface, attribute._dbus_signature,
1186
attribute._dbus_access,
1187
attribute._dbus_get_args_options
1189
(types.FunctionType(
1190
attribute.func_code,
1191
attribute.func_globals,
1192
attribute.func_name,
1193
attribute.func_defaults,
1194
attribute.func_closure)))
1195
# Copy annotations, if any
1197
attr[attrname]._dbus_annotations = dict(
1198
attribute._dbus_annotations)
1199
except AttributeError:
1201
# Is this a D-Bus interface?
1202
elif getattr(attribute, "_dbus_is_interface", False):
1203
# Create a new, but exactly alike, function
1204
# object. Decorate it to be a new D-Bus interface
1205
# with the alternate D-Bus interface name. Add it
1208
dbus_interface_annotations(alt_interface)
1209
(types.FunctionType(attribute.func_code,
1210
attribute.func_globals,
1211
attribute.func_name,
1212
attribute.func_defaults,
1213
attribute.func_closure)))
1215
# Deprecate all alternate interfaces
1216
iname="_AlternateDBusNames_interface_annotation{}"
1217
for interface_name in interface_names:
1219
@dbus_interface_annotations(interface_name)
1221
return { "org.freedesktop.DBus.Deprecated":
1223
# Find an unused name
1224
for aname in (iname.format(i)
1225
for i in itertools.count()):
1226
if aname not in attr:
1230
# Replace the class with a new subclass of it with
1231
# methods, signals, etc. as created above.
1232
cls = type(b"{}Alternate".format(cls.__name__),
1239
@alternate_dbus_interfaces({"se.recompile.Mandos":
1240
"se.bsnet.fukt.Mandos"})
1241
740
class ClientDBus(Client, DBusObjectWithProperties):
1242
741
"""A Client class using D-Bus
1249
748
runtime_expansions = (Client.runtime_expansions
1250
+ ("dbus_object_path", ))
1252
_interface = "se.recompile.Mandos.Client"
749
+ ("dbus_object_path",))
1254
751
# dbus.service.Object doesn't use super(), so we can't either.
1256
753
def __init__(self, bus = None, *args, **kwargs):
754
self._approvals_pending = 0
1258
756
Client.__init__(self, *args, **kwargs)
1259
757
# Only now, when this client is initialized, can it show up on
1261
client_object_name = str(self.name).translate(
759
client_object_name = unicode(self.name).translate(
1262
760
{ord("."): ord("_"),
1263
761
ord("-"): ord("_")})
1264
self.dbus_object_path = dbus.ObjectPath(
1265
"/clients/" + client_object_name)
762
self.dbus_object_path = (dbus.ObjectPath
763
("/clients/" + client_object_name))
1266
764
DBusObjectWithProperties.__init__(self, self.bus,
1267
765
self.dbus_object_path)
1269
def notifychangeproperty(transform_func, dbus_name,
1270
type_func=lambda x: x,
1272
invalidate_only=False,
1273
_interface=_interface):
1274
""" Modify a variable so that it's a property which announces
1275
its changes to DBus.
1277
transform_fun: Function that takes a value and a variant_level
1278
and transforms it to a D-Bus type.
1279
dbus_name: D-Bus name of the variable
1280
type_func: Function that transform the value before sending it
1281
to the D-Bus. Default: no transform
1282
variant_level: D-Bus variant level. Default: 1
1284
attrname = "_{}".format(dbus_name)
1286
def setter(self, value):
1287
if hasattr(self, "dbus_object_path"):
1288
if (not hasattr(self, attrname) or
1289
type_func(getattr(self, attrname, None))
1290
!= type_func(value)):
1292
self.PropertiesChanged(
1293
_interface, dbus.Dictionary(),
1294
dbus.Array((dbus_name, )))
1296
dbus_value = transform_func(
1298
variant_level = variant_level)
1299
self.PropertyChanged(dbus.String(dbus_name),
1301
self.PropertiesChanged(
1303
dbus.Dictionary({ dbus.String(dbus_name):
1306
setattr(self, attrname, value)
1308
return property(lambda self: getattr(self, attrname), setter)
1310
expires = notifychangeproperty(datetime_to_dbus, "Expires")
1311
approvals_pending = notifychangeproperty(dbus.Boolean,
1314
enabled = notifychangeproperty(dbus.Boolean, "Enabled")
1315
last_enabled = notifychangeproperty(datetime_to_dbus,
1317
checker = notifychangeproperty(
1318
dbus.Boolean, "CheckerRunning",
1319
type_func = lambda checker: checker is not None)
1320
last_checked_ok = notifychangeproperty(datetime_to_dbus,
1322
last_checker_status = notifychangeproperty(dbus.Int16,
1323
"LastCheckerStatus")
1324
last_approval_request = notifychangeproperty(
1325
datetime_to_dbus, "LastApprovalRequest")
1326
approved_by_default = notifychangeproperty(dbus.Boolean,
1327
"ApprovedByDefault")
1328
approval_delay = notifychangeproperty(
1329
dbus.UInt64, "ApprovalDelay",
1330
type_func = lambda td: td.total_seconds() * 1000)
1331
approval_duration = notifychangeproperty(
1332
dbus.UInt64, "ApprovalDuration",
1333
type_func = lambda td: td.total_seconds() * 1000)
1334
host = notifychangeproperty(dbus.String, "Host")
1335
timeout = notifychangeproperty(
1336
dbus.UInt64, "Timeout",
1337
type_func = lambda td: td.total_seconds() * 1000)
1338
extended_timeout = notifychangeproperty(
1339
dbus.UInt64, "ExtendedTimeout",
1340
type_func = lambda td: td.total_seconds() * 1000)
1341
interval = notifychangeproperty(
1342
dbus.UInt64, "Interval",
1343
type_func = lambda td: td.total_seconds() * 1000)
1344
checker_command = notifychangeproperty(dbus.String, "Checker")
1345
secret = notifychangeproperty(dbus.ByteArray, "Secret",
1346
invalidate_only=True)
1348
del notifychangeproperty
767
def _get_approvals_pending(self):
768
return self._approvals_pending
769
def _set_approvals_pending(self, value):
770
old_value = self._approvals_pending
771
self._approvals_pending = value
773
if (hasattr(self, "dbus_object_path")
774
and bval is not bool(old_value)):
775
dbus_bool = dbus.Boolean(bval, variant_level=1)
776
self.PropertyChanged(dbus.String("ApprovalPending"),
779
approvals_pending = property(_get_approvals_pending,
780
_set_approvals_pending)
781
del _get_approvals_pending, _set_approvals_pending
784
def _datetime_to_dbus(dt, variant_level=0):
785
"""Convert a UTC datetime.datetime() to a D-Bus type."""
786
return dbus.String(dt.isoformat(),
787
variant_level=variant_level)
790
oldstate = getattr(self, "enabled", False)
791
r = Client.enable(self)
792
if oldstate != self.enabled:
794
self.PropertyChanged(dbus.String("Enabled"),
795
dbus.Boolean(True, variant_level=1))
796
self.PropertyChanged(
797
dbus.String("LastEnabled"),
798
self._datetime_to_dbus(self.last_enabled,
802
def disable(self, quiet = False):
803
oldstate = getattr(self, "enabled", False)
804
r = Client.disable(self, quiet=quiet)
805
if not quiet and oldstate != self.enabled:
807
self.PropertyChanged(dbus.String("Enabled"),
808
dbus.Boolean(False, variant_level=1))
1350
811
def __del__(self, *args, **kwargs):
1375
839
return Client.checker_callback(self, pid, condition, command,
1376
840
*args, **kwargs)
842
def checked_ok(self, *args, **kwargs):
843
Client.checked_ok(self, *args, **kwargs)
845
self.PropertyChanged(
846
dbus.String("LastCheckedOK"),
847
(self._datetime_to_dbus(self.last_checked_ok,
850
def need_approval(self, *args, **kwargs):
851
r = Client.need_approval(self, *args, **kwargs)
853
self.PropertyChanged(
854
dbus.String("LastApprovalRequest"),
855
(self._datetime_to_dbus(self.last_approval_request,
1378
859
def start_checker(self, *args, **kwargs):
1379
old_checker_pid = getattr(self.checker, "pid", None)
860
old_checker = self.checker
861
if self.checker is not None:
862
old_checker_pid = self.checker.pid
864
old_checker_pid = None
1380
865
r = Client.start_checker(self, *args, **kwargs)
1381
866
# Only if new checker process was started
1382
867
if (self.checker is not None
1383
868
and old_checker_pid != self.checker.pid):
1384
869
# Emit D-Bus signal
1385
870
self.CheckerStarted(self.current_checker_command)
871
self.PropertyChanged(
872
dbus.String("CheckerRunning"),
873
dbus.Boolean(True, variant_level=1))
876
def stop_checker(self, *args, **kwargs):
877
old_checker = getattr(self, "checker", None)
878
r = Client.stop_checker(self, *args, **kwargs)
879
if (old_checker is not None
880
and getattr(self, "checker", None) is None):
881
self.PropertyChanged(dbus.String("CheckerRunning"),
882
dbus.Boolean(False, variant_level=1))
1388
885
def _reset_approved(self):
1389
self.approved = None
886
self._approved = None
1392
889
def approve(self, value=True):
1393
self.approved = value
1394
gobject.timeout_add(int(self.approval_duration.total_seconds()
1395
* 1000), self._reset_approved)
1396
890
self.send_changedstate()
891
self._approved = value
892
gobject.timeout_add(self._timedelta_to_milliseconds
893
(self.approval_duration),
894
self._reset_approved)
1398
897
## D-Bus methods, signals & properties
898
_interface = "se.bsnet.fukt.Mandos.Client"
1556
1061
# LastCheckedOK - property
1557
@dbus_service_property(_interface,
1062
@dbus_service_property(_interface, signature="s",
1559
1063
access="readwrite")
1560
1064
def LastCheckedOK_dbus_property(self, value=None):
1561
1065
if value is not None:
1562
1066
self.checked_ok()
1564
return datetime_to_dbus(self.last_checked_ok)
1566
# LastCheckerStatus - property
1567
@dbus_service_property(_interface, signature="n", access="read")
1568
def LastCheckerStatus_dbus_property(self):
1569
return dbus.Int16(self.last_checker_status)
1571
# Expires - property
1572
@dbus_service_property(_interface, signature="s", access="read")
1573
def Expires_dbus_property(self):
1574
return datetime_to_dbus(self.expires)
1068
if self.last_checked_ok is None:
1069
return dbus.String("")
1070
return dbus.String(self._datetime_to_dbus(self
1576
1073
# LastApprovalRequest - property
1577
1074
@dbus_service_property(_interface, signature="s", access="read")
1578
1075
def LastApprovalRequest_dbus_property(self):
1579
return datetime_to_dbus(self.last_approval_request)
1076
if self.last_approval_request is None:
1077
return dbus.String("")
1078
return dbus.String(self.
1079
_datetime_to_dbus(self
1080
.last_approval_request))
1581
1082
# Timeout - property
1582
@dbus_service_property(_interface,
1083
@dbus_service_property(_interface, signature="t",
1584
1084
access="readwrite")
1585
1085
def Timeout_dbus_property(self, value=None):
1586
1086
if value is None: # get
1587
return dbus.UInt64(self.timeout.total_seconds() * 1000)
1588
old_timeout = self.timeout
1087
return dbus.UInt64(self.timeout_milliseconds())
1589
1088
self.timeout = datetime.timedelta(0, 0, 0, value)
1590
# Reschedule disabling
1592
now = datetime.datetime.utcnow()
1593
self.expires += self.timeout - old_timeout
1594
if self.expires <= now:
1595
# The timeout has passed
1598
if (getattr(self, "disable_initiator_tag", None)
1601
gobject.source_remove(self.disable_initiator_tag)
1602
self.disable_initiator_tag = gobject.timeout_add(
1603
int((self.expires - now).total_seconds() * 1000),
1606
# ExtendedTimeout - property
1607
@dbus_service_property(_interface,
1610
def ExtendedTimeout_dbus_property(self, value=None):
1611
if value is None: # get
1612
return dbus.UInt64(self.extended_timeout.total_seconds()
1614
self.extended_timeout = datetime.timedelta(0, 0, 0, value)
1090
self.PropertyChanged(dbus.String("Timeout"),
1091
dbus.UInt64(value, variant_level=1))
1092
if getattr(self, "disable_initiator_tag", None) is None:
1094
# Reschedule timeout
1095
gobject.source_remove(self.disable_initiator_tag)
1096
self.disable_initiator_tag = None
1097
time_to_die = (self.
1098
_timedelta_to_milliseconds((self
1103
if time_to_die <= 0:
1104
# The timeout has passed
1107
self.disable_initiator_tag = (gobject.timeout_add
1108
(time_to_die, self.disable))
1616
1110
# Interval - property
1617
@dbus_service_property(_interface,
1111
@dbus_service_property(_interface, signature="t",
1619
1112
access="readwrite")
1620
1113
def Interval_dbus_property(self, value=None):
1621
1114
if value is None: # get
1622
return dbus.UInt64(self.interval.total_seconds() * 1000)
1115
return dbus.UInt64(self.interval_milliseconds())
1623
1116
self.interval = datetime.timedelta(0, 0, 0, value)
1118
self.PropertyChanged(dbus.String("Interval"),
1119
dbus.UInt64(value, variant_level=1))
1624
1120
if getattr(self, "checker_initiator_tag", None) is None:
1627
# Reschedule checker run
1628
gobject.source_remove(self.checker_initiator_tag)
1629
self.checker_initiator_tag = gobject.timeout_add(
1630
value, self.start_checker)
1631
self.start_checker() # Start one now, too
1122
# Reschedule checker run
1123
gobject.source_remove(self.checker_initiator_tag)
1124
self.checker_initiator_tag = (gobject.timeout_add
1125
(value, self.start_checker))
1126
self.start_checker() # Start one now, too
1633
1128
# Checker - property
1634
@dbus_service_property(_interface,
1129
@dbus_service_property(_interface, signature="s",
1636
1130
access="readwrite")
1637
1131
def Checker_dbus_property(self, value=None):
1638
1132
if value is None: # get
1639
1133
return dbus.String(self.checker_command)
1640
self.checker_command = str(value)
1134
self.checker_command = value
1136
self.PropertyChanged(dbus.String("Checker"),
1137
dbus.String(self.checker_command,
1642
1140
# CheckerRunning - property
1643
@dbus_service_property(_interface,
1141
@dbus_service_property(_interface, signature="b",
1645
1142
access="readwrite")
1646
1143
def CheckerRunning_dbus_property(self, value=None):
1647
1144
if value is None: # get
1872
1367
def fingerprint(openpgp):
1873
1368
"Convert an OpenPGP bytestring to a hexdigit fingerprint"
1874
1369
# New GnuTLS "datum" with the OpenPGP public key
1875
datum = gnutls.library.types.gnutls_datum_t(
1876
ctypes.cast(ctypes.c_char_p(openpgp),
1877
ctypes.POINTER(ctypes.c_ubyte)),
1878
ctypes.c_uint(len(openpgp)))
1370
datum = (gnutls.library.types
1371
.gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
1374
ctypes.c_uint(len(openpgp))))
1879
1375
# New empty GnuTLS certificate
1880
1376
crt = gnutls.library.types.gnutls_openpgp_crt_t()
1881
gnutls.library.functions.gnutls_openpgp_crt_init(
1377
(gnutls.library.functions
1378
.gnutls_openpgp_crt_init(ctypes.byref(crt)))
1883
1379
# Import the OpenPGP public key into the certificate
1884
gnutls.library.functions.gnutls_openpgp_crt_import(
1885
crt, ctypes.byref(datum),
1886
gnutls.library.constants.GNUTLS_OPENPGP_FMT_RAW)
1380
(gnutls.library.functions
1381
.gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
1382
gnutls.library.constants
1383
.GNUTLS_OPENPGP_FMT_RAW))
1887
1384
# Verify the self signature in the key
1888
1385
crtverify = ctypes.c_uint()
1889
gnutls.library.functions.gnutls_openpgp_crt_verify_self(
1890
crt, 0, ctypes.byref(crtverify))
1386
(gnutls.library.functions
1387
.gnutls_openpgp_crt_verify_self(crt, 0,
1388
ctypes.byref(crtverify)))
1891
1389
if crtverify.value != 0:
1892
1390
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
1893
raise gnutls.errors.CertificateSecurityError(
1391
raise (gnutls.errors.CertificateSecurityError
1895
1393
# New buffer for the fingerprint
1896
1394
buf = ctypes.create_string_buffer(20)
1897
1395
buf_len = ctypes.c_size_t()
1898
1396
# Get the fingerprint from the certificate into the buffer
1899
gnutls.library.functions.gnutls_openpgp_crt_get_fingerprint(
1900
crt, ctypes.byref(buf), ctypes.byref(buf_len))
1397
(gnutls.library.functions
1398
.gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
1399
ctypes.byref(buf_len)))
1901
1400
# Deinit the certificate
1902
1401
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
1903
1402
# Convert the buffer to a Python bytestring
1904
1403
fpr = ctypes.string_at(buf, buf_len.value)
1905
1404
# Convert the bytestring to hexadecimal notation
1906
hex_fpr = binascii.hexlify(fpr).upper()
1405
hex_fpr = ''.join("%02X" % ord(char) for char in fpr)
1910
1409
class MultiprocessingMixIn(object):
1911
1410
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
1913
1411
def sub_process_main(self, request, address):
1915
1413
self.finish_request(request, address)
1917
1415
self.handle_error(request, address)
1918
1416
self.close_request(request)
1920
1418
def process_request(self, request, address):
1921
1419
"""Start a new process to process the request."""
1922
proc = multiprocessing.Process(target = self.sub_process_main,
1923
args = (request, address))
1420
multiprocessing.Process(target = self.sub_process_main,
1421
args = (request, address)).start()
1928
1423
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
1929
1424
""" adds a pipe to the MixIn """
1931
1425
def process_request(self, request, client_address):
1932
1426
"""Overrides and wraps the original process_request().
1934
1428
This function creates a new pipe in self.pipe
1936
1430
parent_pipe, self.child_pipe = multiprocessing.Pipe()
1938
proc = MultiprocessingMixIn.process_request(self, request,
1432
super(MultiprocessingMixInWithPipe,
1433
self).process_request(request, client_address)
1940
1434
self.child_pipe.close()
1941
self.add_pipe(parent_pipe, proc)
1943
def add_pipe(self, parent_pipe, proc):
1435
self.add_pipe(parent_pipe)
1437
def add_pipe(self, parent_pipe):
1944
1438
"""Dummy function; override as necessary"""
1945
raise NotImplementedError()
1439
raise NotImplementedError
1948
1441
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
1949
1442
socketserver.TCPServer, object):
2052
1512
Assumes a gobject.MainLoop event loop.
2055
1514
def __init__(self, server_address, RequestHandlerClass,
2059
gnutls_priority=None,
1515
interface=None, use_ipv6=True, clients=None,
1516
gnutls_priority=None, use_dbus=True):
2062
1517
self.enabled = False
2063
1518
self.clients = clients
2064
1519
if self.clients is None:
1520
self.clients = set()
2066
1521
self.use_dbus = use_dbus
2067
1522
self.gnutls_priority = gnutls_priority
2068
1523
IPv6_TCPServer.__init__(self, server_address,
2069
1524
RequestHandlerClass,
2070
1525
interface = interface,
2071
use_ipv6 = use_ipv6,
2072
socketfd = socketfd)
1526
use_ipv6 = use_ipv6)
2074
1527
def server_activate(self):
2075
1528
if self.enabled:
2076
1529
return socketserver.TCPServer.server_activate(self)
2078
1530
def enable(self):
2079
1531
self.enabled = True
2081
def add_pipe(self, parent_pipe, proc):
1532
def add_pipe(self, parent_pipe):
2082
1533
# Call "handle_ipc" for both data and EOF events
2083
gobject.io_add_watch(
2084
parent_pipe.fileno(),
2085
gobject.IO_IN | gobject.IO_HUP,
2086
functools.partial(self.handle_ipc,
2087
parent_pipe = parent_pipe,
2090
def handle_ipc(self, source, condition,
1534
gobject.io_add_watch(parent_pipe.fileno(),
1535
gobject.IO_IN | gobject.IO_HUP,
1536
functools.partial(self.handle_ipc,
1537
parent_pipe = parent_pipe))
1539
def handle_ipc(self, source, condition, parent_pipe=None,
2093
1540
client_object=None):
2094
# error, or the other end of multiprocessing.Pipe has closed
2095
if condition & (gobject.IO_ERR | gobject.IO_HUP):
2096
# Wait for other process to exit
1542
gobject.IO_IN: "IN", # There is data to read.
1543
gobject.IO_OUT: "OUT", # Data can be written (without
1545
gobject.IO_PRI: "PRI", # There is urgent data to read.
1546
gobject.IO_ERR: "ERR", # Error condition.
1547
gobject.IO_HUP: "HUP" # Hung up (the connection has been
1548
# broken, usually for pipes and
1551
conditions_string = ' | '.join(name
1553
condition_names.iteritems()
1554
if cond & condition)
1555
# error or the other end of multiprocessing.Pipe has closed
1556
if condition & (gobject.IO_ERR | condition & gobject.IO_HUP):
2100
1559
# Read a request from the child
2114
1573
"dress: %s", fpr, address)
2115
1574
if self.use_dbus:
2116
1575
# Emit D-Bus signal
2117
mandos_dbus_service.ClientNotFound(fpr,
1576
mandos_dbus_service.ClientNotFound(fpr, address[0])
2119
1577
parent_pipe.send(False)
2122
gobject.io_add_watch(
2123
parent_pipe.fileno(),
2124
gobject.IO_IN | gobject.IO_HUP,
2125
functools.partial(self.handle_ipc,
2126
parent_pipe = parent_pipe,
2128
client_object = client))
1580
gobject.io_add_watch(parent_pipe.fileno(),
1581
gobject.IO_IN | gobject.IO_HUP,
1582
functools.partial(self.handle_ipc,
1583
parent_pipe = parent_pipe,
1584
client_object = client))
2129
1585
parent_pipe.send(True)
2130
# remove the old hook in favor of the new above hook on
1586
# remove the old hook in favor of the new above hook on same fileno
2133
1588
if command == 'funcall':
2134
1589
funcname = request[1]
2135
1590
args = request[2]
2136
1591
kwargs = request[3]
2138
parent_pipe.send(('data', getattr(client_object,
1593
parent_pipe.send(('data', getattr(client_object, funcname)(*args, **kwargs)))
2142
1595
if command == 'getattr':
2143
1596
attrname = request[1]
2144
1597
if callable(client_object.__getattribute__(attrname)):
2145
parent_pipe.send(('function', ))
1598
parent_pipe.send(('function',))
2148
'data', client_object.__getattribute__(attrname)))
1600
parent_pipe.send(('data', client_object.__getattribute__(attrname)))
2150
1602
if command == 'setattr':
2151
1603
attrname = request[1]
2152
1604
value = request[2]
2153
1605
setattr(client_object, attrname, value)
2158
def rfc3339_duration_to_delta(duration):
2159
"""Parse an RFC 3339 "duration" and return a datetime.timedelta
2161
>>> rfc3339_duration_to_delta("P7D")
2162
datetime.timedelta(7)
2163
>>> rfc3339_duration_to_delta("PT60S")
2164
datetime.timedelta(0, 60)
2165
>>> rfc3339_duration_to_delta("PT60M")
2166
datetime.timedelta(0, 3600)
2167
>>> rfc3339_duration_to_delta("PT24H")
2168
datetime.timedelta(1)
2169
>>> rfc3339_duration_to_delta("P1W")
2170
datetime.timedelta(7)
2171
>>> rfc3339_duration_to_delta("PT5M30S")
2172
datetime.timedelta(0, 330)
2173
>>> rfc3339_duration_to_delta("P1DT3M20S")
2174
datetime.timedelta(1, 200)
2177
# Parsing an RFC 3339 duration with regular expressions is not
2178
# possible - there would have to be multiple places for the same
2179
# values, like seconds. The current code, while more esoteric, is
2180
# cleaner without depending on a parsing library. If Python had a
2181
# built-in library for parsing we would use it, but we'd like to
2182
# avoid excessive use of external libraries.
2184
# New type for defining tokens, syntax, and semantics all-in-one
2185
Token = collections.namedtuple("Token", (
2186
"regexp", # To match token; if "value" is not None, must have
2187
# a "group" containing digits
2188
"value", # datetime.timedelta or None
2189
"followers")) # Tokens valid after this token
2190
# RFC 3339 "duration" tokens, syntax, and semantics; taken from
2191
# the "duration" ABNF definition in RFC 3339, Appendix A.
2192
token_end = Token(re.compile(r"$"), None, frozenset())
2193
token_second = Token(re.compile(r"(\d+)S"),
2194
datetime.timedelta(seconds=1),
2195
frozenset((token_end, )))
2196
token_minute = Token(re.compile(r"(\d+)M"),
2197
datetime.timedelta(minutes=1),
2198
frozenset((token_second, token_end)))
2199
token_hour = Token(re.compile(r"(\d+)H"),
2200
datetime.timedelta(hours=1),
2201
frozenset((token_minute, token_end)))
2202
token_time = Token(re.compile(r"T"),
2204
frozenset((token_hour, token_minute,
2206
token_day = Token(re.compile(r"(\d+)D"),
2207
datetime.timedelta(days=1),
2208
frozenset((token_time, token_end)))
2209
token_month = Token(re.compile(r"(\d+)M"),
2210
datetime.timedelta(weeks=4),
2211
frozenset((token_day, token_end)))
2212
token_year = Token(re.compile(r"(\d+)Y"),
2213
datetime.timedelta(weeks=52),
2214
frozenset((token_month, token_end)))
2215
token_week = Token(re.compile(r"(\d+)W"),
2216
datetime.timedelta(weeks=1),
2217
frozenset((token_end, )))
2218
token_duration = Token(re.compile(r"P"), None,
2219
frozenset((token_year, token_month,
2220
token_day, token_time,
2222
# Define starting values
2223
value = datetime.timedelta() # Value so far
2225
followers = frozenset((token_duration, )) # Following valid tokens
2226
s = duration # String left to parse
2227
# Loop until end token is found
2228
while found_token is not token_end:
2229
# Search for any currently valid tokens
2230
for token in followers:
2231
match = token.regexp.match(s)
2232
if match is not None:
2234
if token.value is not None:
2235
# Value found, parse digits
2236
factor = int(match.group(1), 10)
2237
# Add to value so far
2238
value += factor * token.value
2239
# Strip token from string
2240
s = token.regexp.sub("", s, 1)
2243
# Set valid next tokens
2244
followers = found_token.followers
2247
# No currently valid tokens were found
2248
raise ValueError("Invalid RFC 3339 duration: {!r}"
2254
1610
def string_to_delta(interval):
2255
1611
"""Parse a string and return a datetime.timedelta
2385
1744
"debug": "False",
2387
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:!RSA"
2388
":+SIGN-RSA-SHA224:+SIGN-RSA-RMD160",
1746
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
2389
1747
"servicename": "Mandos",
2390
1748
"use_dbus": "True",
2391
1749
"use_ipv6": "True",
2392
1750
"debuglevel": "",
2395
"statedir": "/var/lib/mandos",
2396
"foreground": "False",
2400
1753
# Parse config file for server-global settings
2401
1754
server_config = configparser.SafeConfigParser(server_defaults)
2402
1755
del server_defaults
2403
server_config.read(os.path.join(options.configdir, "mandos.conf"))
1756
server_config.read(os.path.join(options.configdir,
2404
1758
# Convert the SafeConfigParser object to a dict
2405
1759
server_settings = server_config.defaults()
2406
1760
# Use the appropriate methods on the non-string config options
2407
for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
1761
for option in ("debug", "use_dbus", "use_ipv6"):
2408
1762
server_settings[option] = server_config.getboolean("DEFAULT",
2410
1764
if server_settings["port"]:
2411
1765
server_settings["port"] = server_config.getint("DEFAULT",
2413
if server_settings["socket"]:
2414
server_settings["socket"] = server_config.getint("DEFAULT",
2416
# Later, stdin will, and stdout and stderr might, be dup'ed
2417
# over with an opened os.devnull. But we don't want this to
2418
# happen with a supplied network socket.
2419
if 0 <= server_settings["socket"] <= 2:
2420
server_settings["socket"] = os.dup(server_settings
2422
1767
del server_config
2424
1769
# Override the settings from the config file with command line
2425
1770
# options, if set.
2426
1771
for option in ("interface", "address", "port", "debug",
2427
"priority", "servicename", "configdir", "use_dbus",
2428
"use_ipv6", "debuglevel", "restore", "statedir",
2429
"socket", "foreground", "zeroconf"):
1772
"priority", "servicename", "configdir",
1773
"use_dbus", "use_ipv6", "debuglevel"):
2430
1774
value = getattr(options, option)
2431
1775
if value is not None:
2432
1776
server_settings[option] = value
2434
1778
# Force all strings to be unicode
2435
1779
for option in server_settings.keys():
2436
if isinstance(server_settings[option], bytes):
2437
server_settings[option] = (server_settings[option]
2439
# Force all boolean options to be boolean
2440
for option in ("debug", "use_dbus", "use_ipv6", "restore",
2441
"foreground", "zeroconf"):
2442
server_settings[option] = bool(server_settings[option])
2443
# Debug implies foreground
2444
if server_settings["debug"]:
2445
server_settings["foreground"] = True
1780
if type(server_settings[option]) is str:
1781
server_settings[option] = unicode(server_settings[option])
2446
1782
# Now we have our good server settings in "server_settings"
2448
1784
##################################################################
2450
if (not server_settings["zeroconf"]
2451
and not (server_settings["port"]
2452
or server_settings["socket"] != "")):
2453
parser.error("Needs port or socket to work without Zeroconf")
2455
1786
# For convenience
2456
1787
debug = server_settings["debug"]
2457
1788
debuglevel = server_settings["debuglevel"]
2458
1789
use_dbus = server_settings["use_dbus"]
2459
1790
use_ipv6 = server_settings["use_ipv6"]
2460
stored_state_path = os.path.join(server_settings["statedir"],
2462
foreground = server_settings["foreground"]
2463
zeroconf = server_settings["zeroconf"]
2466
initlogger(debug, logging.DEBUG)
2471
level = getattr(logging, debuglevel.upper())
2472
initlogger(debug, level)
2474
1792
if server_settings["servicename"] != "Mandos":
2475
syslogger.setFormatter(
2476
logging.Formatter('Mandos ({}) [%(process)d]:'
2477
' %(levelname)s: %(message)s'.format(
2478
server_settings["servicename"])))
1793
syslogger.setFormatter(logging.Formatter
1794
('Mandos (%s) [%%(process)d]:'
1795
' %%(levelname)s: %%(message)s'
1796
% server_settings["servicename"]))
2480
1798
# Parse config file with clients
2481
client_config = configparser.SafeConfigParser(Client
1799
client_defaults = { "timeout": "1h",
1801
"checker": "fping -q -- %%(host)s",
1803
"approval_delay": "0s",
1804
"approval_duration": "1s",
1806
client_config = configparser.SafeConfigParser(client_defaults)
2483
1807
client_config.read(os.path.join(server_settings["configdir"],
2484
1808
"clients.conf"))
2486
1810
global mandos_dbus_service
2487
1811
mandos_dbus_service = None
2490
if server_settings["socket"] != "":
2491
socketfd = server_settings["socket"]
2492
tcp_server = MandosServer(
2493
(server_settings["address"], server_settings["port"]),
2495
interface=(server_settings["interface"] or None),
2497
gnutls_priority=server_settings["priority"],
2501
pidfilename = "/run/mandos.pid"
2502
if not os.path.isdir("/run/."):
2503
pidfilename = "/var/run/mandos.pid"
1813
tcp_server = MandosServer((server_settings["address"],
1814
server_settings["port"]),
1816
interface=(server_settings["interface"]
1820
server_settings["priority"],
1823
pidfilename = "/var/run/mandos.pid"
2506
pidfile = codecs.open(pidfilename, "w", encoding="utf-8")
2507
except IOError as e:
2508
logger.error("Could not open file %r", pidfilename,
1825
pidfile = open(pidfilename, "w")
1827
logger.error("Could not open file %r", pidfilename)
2511
for name in ("_mandos", "mandos", "nobody"):
1830
uid = pwd.getpwnam("_mandos").pw_uid
1831
gid = pwd.getpwnam("_mandos").pw_gid
2513
uid = pwd.getpwnam(name).pw_uid
2514
gid = pwd.getpwnam(name).pw_gid
1834
uid = pwd.getpwnam("mandos").pw_uid
1835
gid = pwd.getpwnam("mandos").pw_gid
2516
1836
except KeyError:
1838
uid = pwd.getpwnam("nobody").pw_uid
1839
gid = pwd.getpwnam("nobody").pw_gid
2524
1846
except OSError as error:
2525
if error.errno != errno.EPERM:
1847
if error[0] != errno.EPERM:
1850
if not debug and not debuglevel:
1851
syslogger.setLevel(logging.WARNING)
1852
console.setLevel(logging.WARNING)
1854
level = getattr(logging, debuglevel.upper())
1855
syslogger.setLevel(level)
1856
console.setLevel(level)
2529
1859
# Enable all possible GnuTLS debugging
2536
1866
def debug_gnutls(level, string):
2537
1867
logger.debug("GnuTLS: %s", string[:-1])
2539
gnutls.library.functions.gnutls_global_set_log_function(
1869
(gnutls.library.functions
1870
.gnutls_global_set_log_function(debug_gnutls))
2542
1872
# Redirect stdin so all checkers get /dev/null
2543
null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
1873
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
2544
1874
os.dup2(null, sys.stdin.fileno())
1878
# No console logging
1879
logger.removeHandler(console)
2548
1881
# Need to fork before connecting to D-Bus
2550
1883
# Close all input and output, do double fork, etc.
2553
# multiprocessing will use threads, so before we use gobject we
2554
# need to inform gobject that threads will be used.
2555
gobject.threads_init()
2557
1886
global main_loop
2558
1887
# From the Avahi example code
2559
DBusGMainLoop(set_as_default=True)
1888
DBusGMainLoop(set_as_default=True )
2560
1889
main_loop = gobject.MainLoop()
2561
1890
bus = dbus.SystemBus()
2562
1891
# End of Avahi example code
2565
bus_name = dbus.service.BusName("se.recompile.Mandos",
2568
old_bus_name = dbus.service.BusName(
2569
"se.bsnet.fukt.Mandos", bus,
2571
except dbus.exceptions.DBusException as e:
2572
logger.error("Disabling D-Bus:", exc_info=e)
1894
bus_name = dbus.service.BusName("se.bsnet.fukt.Mandos",
1895
bus, do_not_queue=True)
1896
except dbus.exceptions.NameExistsException as e:
1897
logger.error(unicode(e) + ", disabling D-Bus")
2573
1898
use_dbus = False
2574
1899
server_settings["use_dbus"] = False
2575
1900
tcp_server.use_dbus = False
2577
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
2578
service = AvahiServiceToSyslog(
2579
name = server_settings["servicename"],
2580
servicetype = "_mandos._tcp",
2581
protocol = protocol,
2583
if server_settings["interface"]:
2584
service.interface = if_nametoindex(
2585
server_settings["interface"].encode("utf-8"))
1901
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
1902
service = AvahiService(name = server_settings["servicename"],
1903
servicetype = "_mandos._tcp",
1904
protocol = protocol, bus = bus)
1905
if server_settings["interface"]:
1906
service.interface = (if_nametoindex
1907
(str(server_settings["interface"])))
2587
1909
global multiprocessing_manager
2588
1910
multiprocessing_manager = multiprocessing.Manager()
2590
1912
client_class = Client
2592
1914
client_class = functools.partial(ClientDBus, bus = bus)
2594
client_settings = Client.config_parser(client_config)
2595
old_client_settings = {}
2598
# This is used to redirect stdout and stderr for checker processes
2600
wnull = open(os.devnull, "w") # A writable /dev/null
2601
# Only used if server is running in foreground but not in debug
2603
if debug or not foreground:
2606
# Get client data and settings from last running state.
2607
if server_settings["restore"]:
2609
with open(stored_state_path, "rb") as stored_state:
2610
clients_data, old_client_settings = pickle.load(
2612
os.remove(stored_state_path)
2613
except IOError as e:
2614
if e.errno == errno.ENOENT:
2615
logger.warning("Could not load persistent state:"
2616
" {}".format(os.strerror(e.errno)))
2618
logger.critical("Could not load persistent state:",
2621
except EOFError as e:
2622
logger.warning("Could not load persistent state: "
2626
with PGPEngine() as pgp:
2627
for client_name, client in clients_data.items():
2628
# Skip removed clients
2629
if client_name not in client_settings:
2632
# Decide which value to use after restoring saved state.
2633
# We have three different values: Old config file,
2634
# new config file, and saved state.
2635
# New config value takes precedence if it differs from old
2636
# config value, otherwise use saved state.
2637
for name, value in client_settings[client_name].items():
2639
# For each value in new config, check if it
2640
# differs from the old config value (Except for
2641
# the "secret" attribute)
2642
if (name != "secret"
2644
old_client_settings[client_name][name])):
2645
client[name] = value
2649
# Clients who has passed its expire date can still be
2650
# enabled if its last checker was successful. Clients
2651
# whose checker succeeded before we stored its state is
2652
# assumed to have successfully run all checkers during
2654
if client["enabled"]:
2655
if datetime.datetime.utcnow() >= client["expires"]:
2656
if not client["last_checked_ok"]:
2658
"disabling client {} - Client never "
2659
"performed a successful checker".format(
2661
client["enabled"] = False
2662
elif client["last_checker_status"] != 0:
2664
"disabling client {} - Client last"
2665
" checker failed with error code"
2668
client["last_checker_status"]))
2669
client["enabled"] = False
2671
client["expires"] = (
2672
datetime.datetime.utcnow()
2673
+ client["timeout"])
2674
logger.debug("Last checker succeeded,"
2675
" keeping {} enabled".format(
1915
def client_config_items(config, section):
1916
special_settings = {
1917
"approved_by_default":
1918
lambda: config.getboolean(section,
1919
"approved_by_default"),
1921
for name, value in config.items(section):
2678
client["secret"] = pgp.decrypt(
2679
client["encrypted_secret"],
2680
client_settings[client_name]["secret"])
2682
# If decryption fails, we use secret from new settings
2683
logger.debug("Failed to decrypt {} old secret".format(
2685
client["secret"] = (client_settings[client_name]
2688
# Add/remove clients based on new changes made to config
2689
for client_name in (set(old_client_settings)
2690
- set(client_settings)):
2691
del clients_data[client_name]
2692
for client_name in (set(client_settings)
2693
- set(old_client_settings)):
2694
clients_data[client_name] = client_settings[client_name]
2696
# Create all client objects
2697
for client_name, client in clients_data.items():
2698
tcp_server.clients[client_name] = client_class(
2701
server_settings = server_settings)
1923
yield (name, special_settings[name]())
1927
tcp_server.clients.update(set(
1928
client_class(name = section,
1929
config= dict(client_config_items(
1930
client_config, section)))
1931
for section in client_config.sections()))
2703
1932
if not tcp_server.clients:
2704
1933
logger.warning("No clients defined")
2707
if pidfile is not None:
2711
print(pid, file=pidfile)
2713
logger.error("Could not write to file %r with PID %d",
1939
pidfile.write(str(pid) + "\n".encode("utf-8"))
1942
logger.error("Could not write to file %r with PID %d",
1945
# "pidfile" was never created
2716
1947
del pidfilename
1949
signal.signal(signal.SIGINT, signal.SIG_IGN)
2718
1951
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
2719
1952
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
2723
@alternate_dbus_interfaces(
2724
{ "se.recompile.Mandos": "se.bsnet.fukt.Mandos" })
2725
class MandosDBusService(DBusObjectWithProperties):
1955
class MandosDBusService(dbus.service.Object):
2726
1956
"""A D-Bus proxy object"""
2728
1957
def __init__(self):
2729
1958
dbus.service.Object.__init__(self, bus, "/")
2731
_interface = "se.recompile.Mandos"
2733
@dbus_interface_annotations(_interface)
2736
"org.freedesktop.DBus.Property.EmitsChangedSignal":
1959
_interface = "se.bsnet.fukt.Mandos"
2739
1961
@dbus.service.signal(_interface, signature="o")
2740
1962
def ClientAdded(self, objpath):
2788
2010
"Cleanup function; run on exit"
2792
multiprocessing.active_children()
2794
if not (tcp_server.clients or client_settings):
2797
# Store client before exiting. Secrets are encrypted with key
2798
# based on what config file has. If config file is
2799
# removed/edited, old secret will thus be unrecovable.
2801
with PGPEngine() as pgp:
2802
for client in tcp_server.clients.itervalues():
2803
key = client_settings[client.name]["secret"]
2804
client.encrypted_secret = pgp.encrypt(client.secret,
2808
# A list of attributes that can not be pickled
2810
exclude = { "bus", "changedstate", "secret",
2811
"checker", "server_settings" }
2812
for name, typ in inspect.getmembers(dbus.service
2816
client_dict["encrypted_secret"] = (client
2818
for attr in client.client_structure:
2819
if attr not in exclude:
2820
client_dict[attr] = getattr(client, attr)
2822
clients[client.name] = client_dict
2823
del client_settings[client.name]["secret"]
2826
with tempfile.NamedTemporaryFile(
2830
dir=os.path.dirname(stored_state_path),
2831
delete=False) as stored_state:
2832
pickle.dump((clients, client_settings), stored_state)
2833
tempname = stored_state.name
2834
os.rename(tempname, stored_state_path)
2835
except (IOError, OSError) as e:
2841
if e.errno in (errno.ENOENT, errno.EACCES, errno.EEXIST):
2842
logger.warning("Could not save persistent state: {}"
2843
.format(os.strerror(e.errno)))
2845
logger.warning("Could not save persistent state:",
2849
# Delete all clients, and settings from config
2850
2013
while tcp_server.clients:
2851
name, client = tcp_server.clients.popitem()
2014
client = tcp_server.clients.pop()
2853
2016
client.remove_from_connection()
2017
client.disable_hook = None
2854
2018
# Don't signal anything except ClientRemoved
2855
2019
client.disable(quiet=True)
2857
2021
# Emit D-Bus signal
2858
mandos_dbus_service.ClientRemoved(
2859
client.dbus_object_path, client.name)
2860
client_settings.clear()
2022
mandos_dbus_service.ClientRemoved(client.dbus_object_path,
2862
2025
atexit.register(cleanup)
2864
for client in tcp_server.clients.itervalues():
2027
for client in tcp_server.clients:
2866
2029
# Emit D-Bus signal
2867
2030
mandos_dbus_service.ClientAdded(client.dbus_object_path)
2868
# Need to initiate checking of clients
2870
client.init_checker()
2872
2033
tcp_server.enable()
2873
2034
tcp_server.server_activate()
2875
2036
# Find out what port we got
2877
service.port = tcp_server.socket.getsockname()[1]
2037
service.port = tcp_server.socket.getsockname()[1]
2879
2039
logger.info("Now listening on address %r, port %d,"
2880
" flowinfo %d, scope_id %d",
2881
*tcp_server.socket.getsockname())
2040
" flowinfo %d, scope_id %d"
2041
% tcp_server.socket.getsockname())
2883
logger.info("Now listening on address %r, port %d",
2884
*tcp_server.socket.getsockname())
2043
logger.info("Now listening on address %r, port %d"
2044
% tcp_server.socket.getsockname())
2886
2046
#service.interface = tcp_server.socket.getsockname()[3]
2890
# From the Avahi example code
2893
except dbus.exceptions.DBusException as error:
2894
logger.critical("D-Bus Exception", exc_info=error)
2897
# End of Avahi example code
2049
# From the Avahi example code
2052
except dbus.exceptions.DBusException as error:
2053
logger.critical("DBusException: %s", error)
2056
# End of Avahi example code
2899
2058
gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
2900
2059
lambda *args, **kwargs: