255
124
self.rename_count = 0
256
125
self.max_renames = max_renames
257
self.protocol = protocol
258
self.group = None # our entry group
261
self.entry_group_state_changed_match = None
262
126
def rename(self):
263
127
"""Derived from the Avahi example code"""
264
128
if self.rename_count >= self.max_renames:
265
logger.critical("No suitable Zeroconf service name found"
266
" after %i retries, exiting.",
129
logger.critical(u"No suitable Zeroconf service name found"
130
u" after %i retries, exiting.",
267
131
self.rename_count)
268
raise AvahiServiceError("Too many renames")
269
self.name = unicode(self.server
270
.GetAlternativeServiceName(self.name))
271
logger.info("Changing Zeroconf service name to %r ...",
132
raise AvahiServiceError(u"Too many renames")
133
self.name = server.GetAlternativeServiceName(self.name)
134
logger.info(u"Changing Zeroconf service name to %r ...",
136
syslogger.setFormatter(logging.Formatter
137
('Mandos (%s): %%(levelname)s:'
138
' %%(message)s' % self.name))
276
except dbus.exceptions.DBusException as error:
277
logger.critical("DBusException: %s", error)
280
141
self.rename_count += 1
281
142
def remove(self):
282
143
"""Derived from the Avahi example code"""
283
if self.entry_group_state_changed_match is not None:
284
self.entry_group_state_changed_match.remove()
285
self.entry_group_state_changed_match = None
286
if self.group is not None:
144
if group is not None:
289
147
"""Derived from the Avahi example code"""
291
if self.group is None:
292
self.group = dbus.Interface(
293
self.bus.get_object(avahi.DBUS_NAME,
294
self.server.EntryGroupNew()),
295
avahi.DBUS_INTERFACE_ENTRY_GROUP)
296
self.entry_group_state_changed_match = (
297
self.group.connect_to_signal(
298
'StateChanged', self.entry_group_state_changed))
299
logger.debug("Adding Zeroconf service '%s' of type '%s' ...",
300
self.name, self.type)
301
self.group.AddService(
304
dbus.UInt32(0), # flags
305
self.name, self.type,
306
self.domain, self.host,
307
dbus.UInt16(self.port),
308
avahi.string_array_to_txt_array(self.TXT))
310
def entry_group_state_changed(self, state, error):
311
"""Derived from the Avahi example code"""
312
logger.debug("Avahi entry group state change: %i", state)
314
if state == avahi.ENTRY_GROUP_ESTABLISHED:
315
logger.debug("Zeroconf service established.")
316
elif state == avahi.ENTRY_GROUP_COLLISION:
317
logger.info("Zeroconf service name collision.")
319
elif state == avahi.ENTRY_GROUP_FAILURE:
320
logger.critical("Avahi: Error in group state changed %s",
322
raise AvahiGroupError("State changed: %s"
325
"""Derived from the Avahi example code"""
326
if self.group is not None:
329
except (dbus.exceptions.UnknownMethodException,
330
dbus.exceptions.DBusException):
334
def server_state_changed(self, state, error=None):
335
"""Derived from the Avahi example code"""
336
logger.debug("Avahi server state change: %i", state)
337
bad_states = { avahi.SERVER_INVALID:
338
"Zeroconf server invalid",
339
avahi.SERVER_REGISTERING: None,
340
avahi.SERVER_COLLISION:
341
"Zeroconf server name collision",
342
avahi.SERVER_FAILURE:
343
"Zeroconf server failure" }
344
if state in bad_states:
345
if bad_states[state] is not None:
347
logger.error(bad_states[state])
349
logger.error(bad_states[state] + ": %r", error)
351
elif state == avahi.SERVER_RUNNING:
355
logger.debug("Unknown state: %r", state)
357
logger.debug("Unknown state: %r: %r", state, error)
359
"""Derived from the Avahi example code"""
360
if self.server is None:
361
self.server = dbus.Interface(
362
self.bus.get_object(avahi.DBUS_NAME,
363
avahi.DBUS_PATH_SERVER,
364
follow_name_owner_changes=True),
365
avahi.DBUS_INTERFACE_SERVER)
366
self.server.connect_to_signal("StateChanged",
367
self.server_state_changed)
368
self.server_state_changed(self.server.GetState())
370
class AvahiServiceToSyslog(AvahiService):
372
"""Add the new name to the syslog messages"""
373
ret = AvahiService.rename(self)
374
syslogger.setFormatter(logging.Formatter
375
('Mandos (%s) [%%(process)d]:'
376
' %%(levelname)s: %%(message)s'
380
def timedelta_to_milliseconds(td):
381
"Convert a datetime.timedelta() to milliseconds"
382
return ((td.days * 24 * 60 * 60 * 1000)
383
+ (td.seconds * 1000)
384
+ (td.microseconds // 1000))
386
class Client(object):
150
group = dbus.Interface(bus.get_object
152
server.EntryGroupNew()),
153
avahi.DBUS_INTERFACE_ENTRY_GROUP)
154
group.connect_to_signal('StateChanged',
155
entry_group_state_changed)
156
logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
157
service.name, service.type)
159
self.interface, # interface
160
avahi.PROTO_INET6, # protocol
161
dbus.UInt32(0), # flags
162
self.name, self.type,
163
self.domain, self.host,
164
dbus.UInt16(self.port),
165
avahi.string_array_to_txt_array(self.TXT))
168
# From the Avahi example code:
169
group = None # our entry group
170
# End of Avahi example code
173
def _datetime_to_dbus(dt, variant_level=0):
174
"""Convert a UTC datetime.datetime() to a D-Bus type."""
175
return dbus.String(dt.isoformat(), variant_level=variant_level)
178
class Client(dbus.service.Object):
387
179
"""A representation of a client host served by this server.
390
approved: bool(); 'None' if not yet approved/disapproved
391
approval_delay: datetime.timedelta(); Time to wait for approval
392
approval_duration: datetime.timedelta(); Duration of one approval
181
name: string; from the config file, used in log messages
182
fingerprint: string (40 or 32 hexadecimal digits); used to
183
uniquely identify the client
184
secret: bytestring; sent verbatim (over TLS) to client
185
host: string; available for use by the checker command
186
created: datetime.datetime(); (UTC) object creation
187
last_enabled: datetime.datetime(); (UTC)
189
last_checked_ok: datetime.datetime(); (UTC) or None
190
timeout: datetime.timedelta(); How long from last_checked_ok
191
until this client is invalid
192
interval: datetime.timedelta(); How often to start a new checker
193
disable_hook: If set, called by disable() as disable_hook(self)
393
194
checker: subprocess.Popen(); a running checker process used
394
195
to see if the client lives.
395
196
'None' if no process is running.
396
checker_callback_tag: a gobject event source tag, or None
397
checker_command: string; External command which is run to check
398
if client lives. %() expansions are done at
197
checker_initiator_tag: a gobject event source tag, or None
198
disable_initiator_tag: - '' -
199
checker_callback_tag: - '' -
200
checker_command: string; External command which is run to check if
201
client lives. %() expansions are done at
399
202
runtime with vars(self) as dict, so that for
400
203
instance %(name)s can be used in the command.
401
checker_initiator_tag: a gobject event source tag, or None
402
created: datetime.datetime(); (UTC) object creation
403
client_structure: Object describing what attributes a client has
404
and is used for storing the client at exit
405
current_checker_command: string; current running checker_command
406
disable_initiator_tag: a gobject event source tag, or None
408
fingerprint: string (40 or 32 hexadecimal digits); used to
409
uniquely identify the client
410
host: string; available for use by the checker command
411
interval: datetime.timedelta(); How often to start a new checker
412
last_approval_request: datetime.datetime(); (UTC) or None
413
last_checked_ok: datetime.datetime(); (UTC) or None
415
last_checker_status: integer between 0 and 255 reflecting exit
416
status of last checker. -1 reflects crashed
418
last_enabled: datetime.datetime(); (UTC) or None
419
name: string; from the config file, used in log messages and
421
secret: bytestring; sent verbatim (over TLS) to client
422
timeout: datetime.timedelta(); How long from last_checked_ok
423
until this client is disabled
424
extended_timeout: extra long timeout when password has been sent
425
runtime_expansions: Allowed attributes for runtime expansion.
426
expires: datetime.datetime(); time (UTC) when a client will be
204
use_dbus: bool(); Whether to provide D-Bus interface and signals
205
dbus_object_path: dbus.ObjectPath ; only set if self.use_dbus
430
runtime_expansions = ("approval_delay", "approval_duration",
431
"created", "enabled", "fingerprint",
432
"host", "interval", "last_checked_ok",
433
"last_enabled", "name", "timeout")
434
client_defaults = { "timeout": "5m",
435
"extended_timeout": "15m",
437
"checker": "fping -q -- %%(host)s",
439
"approval_delay": "0s",
440
"approval_duration": "1s",
441
"approved_by_default": "True",
445
207
def timeout_milliseconds(self):
446
208
"Return the 'timeout' attribute in milliseconds"
447
return timedelta_to_milliseconds(self.timeout)
449
def extended_timeout_milliseconds(self):
450
"Return the 'extended_timeout' attribute in milliseconds"
451
return timedelta_to_milliseconds(self.extended_timeout)
209
return ((self.timeout.days * 24 * 60 * 60 * 1000)
210
+ (self.timeout.seconds * 1000)
211
+ (self.timeout.microseconds // 1000))
453
213
def interval_milliseconds(self):
454
214
"Return the 'interval' attribute in milliseconds"
455
return timedelta_to_milliseconds(self.interval)
215
return ((self.interval.days * 24 * 60 * 60 * 1000)
216
+ (self.interval.seconds * 1000)
217
+ (self.interval.microseconds // 1000))
457
def approval_delay_milliseconds(self):
458
return timedelta_to_milliseconds(self.approval_delay)
461
def config_parser(config):
462
""" Construct a new dict of client settings of this form:
463
{ client_name: {setting_name: value, ...}, ...}
464
with exceptions for any special settings as defined above"""
466
for client_name in config.sections():
467
section = dict(config.items(client_name))
468
client = settings[client_name] = {}
470
# Default copying each value from config to new dict
471
for setting, value in section.iteritems():
472
client[setting] = value
474
# Reformat values from string types to Python types
475
client["approved_by_default"] = config.getboolean(
476
client_name, "approved_by_default")
477
client["enabled"] = config.getboolean(client_name, "enabled")
479
client["fingerprint"] = (section["fingerprint"].upper()
481
if "secret" in section:
482
client["secret"] = section["secret"].decode("base64")
483
elif "secfile" in section:
484
with open(os.path.expanduser(os.path.expandvars
485
(section["secfile"])),
487
client["secret"] = secfile.read()
489
raise TypeError("No secret or secfile for section %s"
491
client["timeout"] = string_to_delta(section["timeout"])
492
client["extended_timeout"] = string_to_delta(
493
section["extended_timeout"])
494
client["interval"] = string_to_delta(section["interval"])
495
client["approval_delay"] = string_to_delta(
496
section["approval_delay"])
497
client["approval_duration"] = string_to_delta(
498
section["approval_duration"])
503
def __init__(self, config, name = None):
219
def __init__(self, name = None, disable_hook=None, config=None,
504
221
"""Note: the 'checker' key in 'config' sets the
505
222
'checker_command' attribute and *not* the 'checker'
508
logger.debug("Creating client %r", self.name)
227
logger.debug(u"Creating client %r", self.name)
228
self.use_dbus = use_dbus
230
self.dbus_object_path = (dbus.ObjectPath
232
+ self.name.replace(".", "_")))
233
dbus.service.Object.__init__(self, bus,
234
self.dbus_object_path)
509
235
# Uppercase and remove spaces from fingerprint for later
510
236
# comparison purposes with return value from the fingerprint()
512
self.fingerprint = config["fingerprint"]
513
logger.debug(" Fingerprint: %s", self.fingerprint)
514
self.secret = config["secret"]
515
self.host = config["host"]
238
self.fingerprint = (config["fingerprint"].upper()
240
logger.debug(u" Fingerprint: %s", self.fingerprint)
241
if "secret" in config:
242
self.secret = config["secret"].decode(u"base64")
243
elif "secfile" in config:
244
with closing(open(os.path.expanduser
246
(config["secfile"])))) as secfile:
247
self.secret = secfile.read()
249
raise TypeError(u"No secret or secfile for client %s"
251
self.host = config.get("host", "")
516
252
self.created = datetime.datetime.utcnow()
517
self.enabled = config["enabled"]
518
self.last_approval_request = None
520
self.last_enabled = datetime.datetime.utcnow()
522
self.last_enabled = None
254
self.last_enabled = None
523
255
self.last_checked_ok = None
524
self.last_checker_status = None
525
self.timeout = config["timeout"]
526
self.extended_timeout = config["extended_timeout"]
527
self.interval = config["interval"]
256
self.timeout = string_to_delta(config["timeout"])
257
self.interval = string_to_delta(config["interval"])
258
self.disable_hook = disable_hook
528
259
self.checker = None
529
260
self.checker_initiator_tag = None
530
261
self.disable_initiator_tag = None
532
self.expires = datetime.datetime.utcnow() + self.timeout
535
262
self.checker_callback_tag = None
536
263
self.checker_command = config["checker"]
537
self.current_checker_command = None
539
self.approved_by_default = config["approved_by_default"]
540
self.approvals_pending = 0
541
self.approval_delay = config["approval_delay"]
542
self.approval_duration = config["approval_duration"]
543
self.changedstate = (multiprocessing_manager
544
.Condition(multiprocessing_manager
546
self.client_structure = [attr for attr in
547
self.__dict__.iterkeys()
548
if not attr.startswith("_")]
549
self.client_structure.append("client_structure")
551
for name, t in inspect.getmembers(type(self),
555
if not name.startswith("_"):
556
self.client_structure.append(name)
558
# Send notice to process children that client state has changed
559
def send_changedstate(self):
560
with self.changedstate:
561
self.changedstate.notify_all()
563
265
def enable(self):
564
266
"""Start this client's checker and timeout hooks"""
565
if getattr(self, "enabled", False):
568
self.send_changedstate()
569
self.expires = datetime.datetime.utcnow() + self.timeout
571
267
self.last_enabled = datetime.datetime.utcnow()
574
def disable(self, quiet=True):
575
"""Disable this client."""
576
if not getattr(self, "enabled", False):
579
self.send_changedstate()
581
logger.info("Disabling client %s", self.name)
582
if getattr(self, "disable_initiator_tag", False):
583
gobject.source_remove(self.disable_initiator_tag)
584
self.disable_initiator_tag = None
586
if getattr(self, "checker_initiator_tag", False):
587
gobject.source_remove(self.checker_initiator_tag)
588
self.checker_initiator_tag = None
591
# Do not run this again if called by a gobject.timeout_add
597
def init_checker(self):
598
268
# Schedule a new checker to be started an 'interval' from now,
599
269
# and every interval from then on.
600
270
self.checker_initiator_tag = (gobject.timeout_add
601
271
(self.interval_milliseconds(),
602
272
self.start_checker))
273
# Also start a new checker *right now*.
603
275
# Schedule a disable() when 'timeout' has passed
604
276
self.disable_initiator_tag = (gobject.timeout_add
605
277
(self.timeout_milliseconds(),
607
# Also start a new checker *right now*.
282
self.PropertyChanged(dbus.String(u"enabled"),
283
dbus.Boolean(True, variant_level=1))
284
self.PropertyChanged(dbus.String(u"last_enabled"),
285
(_datetime_to_dbus(self.last_enabled,
289
"""Disable this client."""
290
if not getattr(self, "enabled", False):
292
logger.info(u"Disabling client %s", self.name)
293
if getattr(self, "disable_initiator_tag", False):
294
gobject.source_remove(self.disable_initiator_tag)
295
self.disable_initiator_tag = None
296
if getattr(self, "checker_initiator_tag", False):
297
gobject.source_remove(self.checker_initiator_tag)
298
self.checker_initiator_tag = None
300
if self.disable_hook:
301
self.disable_hook(self)
305
self.PropertyChanged(dbus.String(u"enabled"),
306
dbus.Boolean(False, variant_level=1))
307
# Do not run this again if called by a gobject.timeout_add
311
self.disable_hook = None
610
314
def checker_callback(self, pid, condition, command):
611
315
"""The checker has completed, so take appropriate actions."""
612
316
self.checker_callback_tag = None
613
317
self.checker = None
614
if os.WIFEXITED(condition):
615
self.last_checker_status = os.WEXITSTATUS(condition)
616
if self.last_checker_status == 0:
617
logger.info("Checker for %(name)s succeeded",
621
logger.info("Checker for %(name)s failed",
624
self.last_checker_status = -1
625
logger.warning("Checker for %(name)s crashed?",
320
self.PropertyChanged(dbus.String(u"checker_running"),
321
dbus.Boolean(False, variant_level=1))
322
if (os.WIFEXITED(condition)
323
and (os.WEXITSTATUS(condition) == 0)):
324
logger.info(u"Checker for %(name)s succeeded",
328
self.CheckerCompleted(dbus.Boolean(True),
329
dbus.UInt16(condition),
330
dbus.String(command))
332
elif not os.WIFEXITED(condition):
333
logger.warning(u"Checker for %(name)s crashed?",
337
self.CheckerCompleted(dbus.Boolean(False),
338
dbus.UInt16(condition),
339
dbus.String(command))
341
logger.info(u"Checker for %(name)s failed",
345
self.CheckerCompleted(dbus.Boolean(False),
346
dbus.UInt16(condition),
347
dbus.String(command))
628
def checked_ok(self, timeout=None):
349
def bump_timeout(self):
629
350
"""Bump up the timeout for this client.
631
351
This should only be called when the client has been seen,
635
timeout = self.timeout
636
354
self.last_checked_ok = datetime.datetime.utcnow()
637
if self.disable_initiator_tag is not None:
638
gobject.source_remove(self.disable_initiator_tag)
639
if getattr(self, "enabled", False):
640
self.disable_initiator_tag = (gobject.timeout_add
641
(timedelta_to_milliseconds
642
(timeout), self.disable))
643
self.expires = datetime.datetime.utcnow() + timeout
645
def need_approval(self):
646
self.last_approval_request = datetime.datetime.utcnow()
355
gobject.source_remove(self.disable_initiator_tag)
356
self.disable_initiator_tag = (gobject.timeout_add
357
(self.timeout_milliseconds(),
361
self.PropertyChanged(
362
dbus.String(u"last_checked_ok"),
363
(_datetime_to_dbus(self.last_checked_ok,
648
366
def start_checker(self):
649
367
"""Start a new checker subprocess if one is not running.
651
368
If a checker already exists, leave it running and do
653
370
# The reason for not killing a running checker is that if we
727
423
self.checker_callback_tag = None
728
424
if getattr(self, "checker", None) is None:
730
logger.debug("Stopping checker for %(name)s", vars(self))
426
logger.debug(u"Stopping checker for %(name)s", vars(self))
732
428
os.kill(self.checker.pid, signal.SIGTERM)
734
430
#if self.checker.poll() is None:
735
431
# os.kill(self.checker.pid, signal.SIGKILL)
736
except OSError as error:
432
except OSError, error:
737
433
if error.errno != errno.ESRCH: # No such process
739
435
self.checker = None
742
def dbus_service_property(dbus_interface, signature="v",
743
access="readwrite", byte_arrays=False):
744
"""Decorators for marking methods of a DBusObjectWithProperties to
745
become properties on the D-Bus.
747
The decorated method will be called with no arguments by "Get"
748
and with one argument by "Set".
750
The parameters, where they are supported, are the same as
751
dbus.service.method, except there is only "signature", since the
752
type from Get() and the type sent to Set() is the same.
754
# Encoding deeply encoded byte arrays is not supported yet by the
755
# "Set" method, so we fail early here:
756
if byte_arrays and signature != "ay":
757
raise ValueError("Byte arrays not supported for non-'ay'"
758
" signature %r" % signature)
760
func._dbus_is_property = True
761
func._dbus_interface = dbus_interface
762
func._dbus_signature = signature
763
func._dbus_access = access
764
func._dbus_name = func.__name__
765
if func._dbus_name.endswith("_dbus_property"):
766
func._dbus_name = func._dbus_name[:-14]
767
func._dbus_get_args_options = {'byte_arrays': byte_arrays }
772
class DBusPropertyException(dbus.exceptions.DBusException):
773
"""A base class for D-Bus property-related exceptions
775
def __unicode__(self):
776
return unicode(str(self))
779
class DBusPropertyAccessException(DBusPropertyException):
780
"""A property's access permissions disallows an operation.
785
class DBusPropertyNotFound(DBusPropertyException):
786
"""An attempt was made to access a non-existing property.
791
class DBusObjectWithProperties(dbus.service.Object):
792
"""A D-Bus object with properties.
794
Classes inheriting from this can use the dbus_service_property
795
decorator to expose methods as D-Bus properties. It exposes the
796
standard Get(), Set(), and GetAll() methods on the D-Bus.
800
def _is_dbus_property(obj):
801
return getattr(obj, "_dbus_is_property", False)
803
def _get_all_dbus_properties(self):
804
"""Returns a generator of (name, attribute) pairs
806
return ((prop.__get__(self)._dbus_name, prop.__get__(self))
807
for cls in self.__class__.__mro__
809
inspect.getmembers(cls, self._is_dbus_property))
811
def _get_dbus_property(self, interface_name, property_name):
812
"""Returns a bound method if one exists which is a D-Bus
813
property with the specified name and interface.
815
for cls in self.__class__.__mro__:
816
for name, value in (inspect.getmembers
817
(cls, self._is_dbus_property)):
818
if (value._dbus_name == property_name
819
and value._dbus_interface == interface_name):
820
return value.__get__(self)
823
raise DBusPropertyNotFound(self.dbus_object_path + ":"
824
+ interface_name + "."
827
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="ss",
829
def Get(self, interface_name, property_name):
830
"""Standard D-Bus property Get() method, see D-Bus standard.
832
prop = self._get_dbus_property(interface_name, property_name)
833
if prop._dbus_access == "write":
834
raise DBusPropertyAccessException(property_name)
836
if not hasattr(value, "variant_level"):
838
return type(value)(value, variant_level=value.variant_level+1)
840
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="ssv")
841
def Set(self, interface_name, property_name, value):
842
"""Standard D-Bus property Set() method, see D-Bus standard.
844
prop = self._get_dbus_property(interface_name, property_name)
845
if prop._dbus_access == "read":
846
raise DBusPropertyAccessException(property_name)
847
if prop._dbus_get_args_options["byte_arrays"]:
848
# The byte_arrays option is not supported yet on
849
# signatures other than "ay".
850
if prop._dbus_signature != "ay":
852
value = dbus.ByteArray(''.join(unichr(byte)
856
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="s",
857
out_signature="a{sv}")
858
def GetAll(self, interface_name):
859
"""Standard D-Bus property GetAll() method, see D-Bus
862
Note: Will not include properties with access="write".
865
for name, prop in self._get_all_dbus_properties():
867
and interface_name != prop._dbus_interface):
868
# Interface non-empty but did not match
870
# Ignore write-only properties
871
if prop._dbus_access == "write":
874
if not hasattr(value, "variant_level"):
875
properties[name] = value
877
properties[name] = type(value)(value, variant_level=
878
value.variant_level+1)
879
return dbus.Dictionary(properties, signature="sv")
881
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
883
path_keyword='object_path',
884
connection_keyword='connection')
885
def Introspect(self, object_path, connection):
886
"""Standard D-Bus method, overloaded to insert property tags.
888
xmlstring = dbus.service.Object.Introspect(self, object_path,
891
document = xml.dom.minidom.parseString(xmlstring)
892
def make_tag(document, name, prop):
893
e = document.createElement("property")
894
e.setAttribute("name", name)
895
e.setAttribute("type", prop._dbus_signature)
896
e.setAttribute("access", prop._dbus_access)
898
for if_tag in document.getElementsByTagName("interface"):
899
for tag in (make_tag(document, name, prop)
901
in self._get_all_dbus_properties()
902
if prop._dbus_interface
903
== if_tag.getAttribute("name")):
904
if_tag.appendChild(tag)
905
# Add the names to the return values for the
906
# "org.freedesktop.DBus.Properties" methods
907
if (if_tag.getAttribute("name")
908
== "org.freedesktop.DBus.Properties"):
909
for cn in if_tag.getElementsByTagName("method"):
910
if cn.getAttribute("name") == "Get":
911
for arg in cn.getElementsByTagName("arg"):
912
if (arg.getAttribute("direction")
914
arg.setAttribute("name", "value")
915
elif cn.getAttribute("name") == "GetAll":
916
for arg in cn.getElementsByTagName("arg"):
917
if (arg.getAttribute("direction")
919
arg.setAttribute("name", "props")
920
xmlstring = document.toxml("utf-8")
922
except (AttributeError, xml.dom.DOMException,
923
xml.parsers.expat.ExpatError) as error:
924
logger.error("Failed to override Introspection method",
929
def datetime_to_dbus (dt, variant_level=0):
930
"""Convert a UTC datetime.datetime() to a D-Bus type."""
932
return dbus.String("", variant_level = variant_level)
933
return dbus.String(dt.isoformat(),
934
variant_level=variant_level)
937
class AlternateDBusNamesMetaclass(DBusObjectWithProperties
939
"""Applied to an empty subclass of a D-Bus object, this metaclass
940
will add additional D-Bus attributes matching a certain pattern.
942
def __new__(mcs, name, bases, attr):
943
# Go through all the base classes which could have D-Bus
944
# methods, signals, or properties in them
945
for base in (b for b in bases
946
if issubclass(b, dbus.service.Object)):
947
# Go though all attributes of the base class
948
for attrname, attribute in inspect.getmembers(base):
949
# Ignore non-D-Bus attributes, and D-Bus attributes
950
# with the wrong interface name
951
if (not hasattr(attribute, "_dbus_interface")
952
or not attribute._dbus_interface
953
.startswith("se.recompile.Mandos")):
955
# Create an alternate D-Bus interface name based on
957
alt_interface = (attribute._dbus_interface
958
.replace("se.recompile.Mandos",
959
"se.bsnet.fukt.Mandos"))
960
# Is this a D-Bus signal?
961
if getattr(attribute, "_dbus_is_signal", False):
962
# Extract the original non-method function by
964
nonmethod_func = (dict(
965
zip(attribute.func_code.co_freevars,
966
attribute.__closure__))["func"]
968
# Create a new, but exactly alike, function
969
# object, and decorate it to be a new D-Bus signal
970
# with the alternate D-Bus interface name
971
new_function = (dbus.service.signal
973
attribute._dbus_signature)
975
nonmethod_func.func_code,
976
nonmethod_func.func_globals,
977
nonmethod_func.func_name,
978
nonmethod_func.func_defaults,
979
nonmethod_func.func_closure)))
980
# Define a creator of a function to call both the
981
# old and new functions, so both the old and new
982
# signals gets sent when the function is called
983
def fixscope(func1, func2):
984
"""This function is a scope container to pass
985
func1 and func2 to the "call_both" function
986
outside of its arguments"""
987
def call_both(*args, **kwargs):
988
"""This function will emit two D-Bus
989
signals by calling func1 and func2"""
990
func1(*args, **kwargs)
991
func2(*args, **kwargs)
993
# Create the "call_both" function and add it to
995
attr[attrname] = fixscope(attribute,
997
# Is this a D-Bus method?
998
elif getattr(attribute, "_dbus_is_method", False):
999
# Create a new, but exactly alike, function
1000
# object. Decorate it to be a new D-Bus method
1001
# with the alternate D-Bus interface name. Add it
1003
attr[attrname] = (dbus.service.method
1005
attribute._dbus_in_signature,
1006
attribute._dbus_out_signature)
1008
(attribute.func_code,
1009
attribute.func_globals,
1010
attribute.func_name,
1011
attribute.func_defaults,
1012
attribute.func_closure)))
1013
# Is this a D-Bus property?
1014
elif getattr(attribute, "_dbus_is_property", False):
1015
# Create a new, but exactly alike, function
1016
# object, and decorate it to be a new D-Bus
1017
# property with the alternate D-Bus interface
1018
# name. Add it to the class.
1019
attr[attrname] = (dbus_service_property
1021
attribute._dbus_signature,
1022
attribute._dbus_access,
1024
._dbus_get_args_options
1027
(attribute.func_code,
1028
attribute.func_globals,
1029
attribute.func_name,
1030
attribute.func_defaults,
1031
attribute.func_closure)))
1032
return type.__new__(mcs, name, bases, attr)
1035
class ClientDBus(Client, DBusObjectWithProperties):
1036
"""A Client class using D-Bus
1039
dbus_object_path: dbus.ObjectPath
1040
bus: dbus.SystemBus()
1043
runtime_expansions = (Client.runtime_expansions
1044
+ ("dbus_object_path",))
1046
# dbus.service.Object doesn't use super(), so we can't either.
1048
def __init__(self, bus = None, *args, **kwargs):
1050
Client.__init__(self, *args, **kwargs)
1052
self._approvals_pending = 0
1053
# Only now, when this client is initialized, can it show up on
1055
client_object_name = unicode(self.name).translate(
1056
{ord("."): ord("_"),
1057
ord("-"): ord("_")})
1058
self.dbus_object_path = (dbus.ObjectPath
1059
("/clients/" + client_object_name))
1060
DBusObjectWithProperties.__init__(self, self.bus,
1061
self.dbus_object_path)
1063
def notifychangeproperty(transform_func,
1064
dbus_name, type_func=lambda x: x,
1066
""" Modify a variable so that it's a property which announces
1067
its changes to DBus.
1069
transform_fun: Function that takes a value and a variant_level
1070
and transforms it to a D-Bus type.
1071
dbus_name: D-Bus name of the variable
1072
type_func: Function that transform the value before sending it
1073
to the D-Bus. Default: no transform
1074
variant_level: D-Bus variant level. Default: 1
1076
attrname = "_{0}".format(dbus_name)
1077
def setter(self, value):
1078
if hasattr(self, "dbus_object_path"):
1079
if (not hasattr(self, attrname) or
1080
type_func(getattr(self, attrname, None))
1081
!= type_func(value)):
1082
dbus_value = transform_func(type_func(value),
1085
self.PropertyChanged(dbus.String(dbus_name),
1087
setattr(self, attrname, value)
1089
return property(lambda self: getattr(self, attrname), setter)
1092
expires = notifychangeproperty(datetime_to_dbus, "Expires")
1093
approvals_pending = notifychangeproperty(dbus.Boolean,
1096
enabled = notifychangeproperty(dbus.Boolean, "Enabled")
1097
last_enabled = notifychangeproperty(datetime_to_dbus,
1099
checker = notifychangeproperty(dbus.Boolean, "CheckerRunning",
1100
type_func = lambda checker:
1101
checker is not None)
1102
last_checked_ok = notifychangeproperty(datetime_to_dbus,
1104
last_approval_request = notifychangeproperty(
1105
datetime_to_dbus, "LastApprovalRequest")
1106
approved_by_default = notifychangeproperty(dbus.Boolean,
1107
"ApprovedByDefault")
1108
approval_delay = notifychangeproperty(dbus.UInt64,
1111
timedelta_to_milliseconds)
1112
approval_duration = notifychangeproperty(
1113
dbus.UInt64, "ApprovalDuration",
1114
type_func = timedelta_to_milliseconds)
1115
host = notifychangeproperty(dbus.String, "Host")
1116
timeout = notifychangeproperty(dbus.UInt64, "Timeout",
1118
timedelta_to_milliseconds)
1119
extended_timeout = notifychangeproperty(
1120
dbus.UInt64, "ExtendedTimeout",
1121
type_func = timedelta_to_milliseconds)
1122
interval = notifychangeproperty(dbus.UInt64,
1125
timedelta_to_milliseconds)
1126
checker_command = notifychangeproperty(dbus.String, "Checker")
1128
del notifychangeproperty
1130
def __del__(self, *args, **kwargs):
1132
self.remove_from_connection()
1135
if hasattr(DBusObjectWithProperties, "__del__"):
1136
DBusObjectWithProperties.__del__(self, *args, **kwargs)
1137
Client.__del__(self, *args, **kwargs)
1139
def checker_callback(self, pid, condition, command,
1141
self.checker_callback_tag = None
1143
if os.WIFEXITED(condition):
1144
exitstatus = os.WEXITSTATUS(condition)
1146
self.CheckerCompleted(dbus.Int16(exitstatus),
1147
dbus.Int64(condition),
1148
dbus.String(command))
1151
self.CheckerCompleted(dbus.Int16(-1),
1152
dbus.Int64(condition),
1153
dbus.String(command))
1155
return Client.checker_callback(self, pid, condition, command,
1158
def start_checker(self, *args, **kwargs):
1159
old_checker = self.checker
1160
if self.checker is not None:
1161
old_checker_pid = self.checker.pid
1163
old_checker_pid = None
1164
r = Client.start_checker(self, *args, **kwargs)
1165
# Only if new checker process was started
1166
if (self.checker is not None
1167
and old_checker_pid != self.checker.pid):
1169
self.CheckerStarted(self.current_checker_command)
1172
def _reset_approved(self):
1173
self.approved = None
1176
def approve(self, value=True):
1177
self.send_changedstate()
1178
self.approved = value
1179
gobject.timeout_add(timedelta_to_milliseconds
1180
(self.approval_duration),
1181
self._reset_approved)
1184
## D-Bus methods, signals & properties
1185
_interface = "se.recompile.Mandos.Client"
437
self.PropertyChanged(dbus.String(u"checker_running"),
438
dbus.Boolean(False, variant_level=1))
440
def still_valid(self):
441
"""Has the timeout not yet passed for this client?"""
442
if not getattr(self, "enabled", False):
444
now = datetime.datetime.utcnow()
445
if self.last_checked_ok is None:
446
return now < (self.created + self.timeout)
448
return now < (self.last_checked_ok + self.timeout)
450
## D-Bus methods & signals
451
_interface = u"org.mandos_system.Mandos.Client"
453
# BumpTimeout - method
454
BumpTimeout = dbus.service.method(_interface)(bump_timeout)
455
BumpTimeout.__name__ = "BumpTimeout"
1189
457
# CheckerCompleted - signal
1190
@dbus.service.signal(_interface, signature="nxs")
1191
def CheckerCompleted(self, exitcode, waitstatus, command):
458
@dbus.service.signal(_interface, signature="bqs")
459
def CheckerCompleted(self, success, condition, command):
1266
579
# StopChecker - method
1267
@dbus.service.method(_interface)
1268
def StopChecker(self):
1273
# ApprovalPending - property
1274
@dbus_service_property(_interface, signature="b", access="read")
1275
def ApprovalPending_dbus_property(self):
1276
return dbus.Boolean(bool(self.approvals_pending))
1278
# ApprovedByDefault - property
1279
@dbus_service_property(_interface, signature="b",
1281
def ApprovedByDefault_dbus_property(self, value=None):
1282
if value is None: # get
1283
return dbus.Boolean(self.approved_by_default)
1284
self.approved_by_default = bool(value)
1286
# ApprovalDelay - property
1287
@dbus_service_property(_interface, signature="t",
1289
def ApprovalDelay_dbus_property(self, value=None):
1290
if value is None: # get
1291
return dbus.UInt64(self.approval_delay_milliseconds())
1292
self.approval_delay = datetime.timedelta(0, 0, 0, value)
1294
# ApprovalDuration - property
1295
@dbus_service_property(_interface, signature="t",
1297
def ApprovalDuration_dbus_property(self, value=None):
1298
if value is None: # get
1299
return dbus.UInt64(timedelta_to_milliseconds(
1300
self.approval_duration))
1301
self.approval_duration = datetime.timedelta(0, 0, 0, value)
1304
@dbus_service_property(_interface, signature="s", access="read")
1305
def Name_dbus_property(self):
1306
return dbus.String(self.name)
1308
# Fingerprint - property
1309
@dbus_service_property(_interface, signature="s", access="read")
1310
def Fingerprint_dbus_property(self):
1311
return dbus.String(self.fingerprint)
1314
@dbus_service_property(_interface, signature="s",
1316
def Host_dbus_property(self, value=None):
1317
if value is None: # get
1318
return dbus.String(self.host)
1319
self.host = unicode(value)
1321
# Created - property
1322
@dbus_service_property(_interface, signature="s", access="read")
1323
def Created_dbus_property(self):
1324
return datetime_to_dbus(self.created)
1326
# LastEnabled - property
1327
@dbus_service_property(_interface, signature="s", access="read")
1328
def LastEnabled_dbus_property(self):
1329
return datetime_to_dbus(self.last_enabled)
1331
# Enabled - property
1332
@dbus_service_property(_interface, signature="b",
1334
def Enabled_dbus_property(self, value=None):
1335
if value is None: # get
1336
return dbus.Boolean(self.enabled)
1342
# LastCheckedOK - property
1343
@dbus_service_property(_interface, signature="s",
1345
def LastCheckedOK_dbus_property(self, value=None):
1346
if value is not None:
1349
return datetime_to_dbus(self.last_checked_ok)
1351
# Expires - property
1352
@dbus_service_property(_interface, signature="s", access="read")
1353
def Expires_dbus_property(self):
1354
return datetime_to_dbus(self.expires)
1356
# LastApprovalRequest - property
1357
@dbus_service_property(_interface, signature="s", access="read")
1358
def LastApprovalRequest_dbus_property(self):
1359
return datetime_to_dbus(self.last_approval_request)
1361
# Timeout - property
1362
@dbus_service_property(_interface, signature="t",
1364
def Timeout_dbus_property(self, value=None):
1365
if value is None: # get
1366
return dbus.UInt64(self.timeout_milliseconds())
1367
self.timeout = datetime.timedelta(0, 0, 0, value)
1368
if getattr(self, "disable_initiator_tag", None) is None:
1370
# Reschedule timeout
1371
gobject.source_remove(self.disable_initiator_tag)
1372
self.disable_initiator_tag = None
1374
time_to_die = timedelta_to_milliseconds((self
1379
if time_to_die <= 0:
1380
# The timeout has passed
1383
self.expires = (datetime.datetime.utcnow()
1384
+ datetime.timedelta(milliseconds =
1386
self.disable_initiator_tag = (gobject.timeout_add
1387
(time_to_die, self.disable))
1389
# ExtendedTimeout - property
1390
@dbus_service_property(_interface, signature="t",
1392
def ExtendedTimeout_dbus_property(self, value=None):
1393
if value is None: # get
1394
return dbus.UInt64(self.extended_timeout_milliseconds())
1395
self.extended_timeout = datetime.timedelta(0, 0, 0, value)
1397
# Interval - property
1398
@dbus_service_property(_interface, signature="t",
1400
def Interval_dbus_property(self, value=None):
1401
if value is None: # get
1402
return dbus.UInt64(self.interval_milliseconds())
1403
self.interval = datetime.timedelta(0, 0, 0, value)
1404
if getattr(self, "checker_initiator_tag", None) is None:
1407
# Reschedule checker run
1408
gobject.source_remove(self.checker_initiator_tag)
1409
self.checker_initiator_tag = (gobject.timeout_add
1410
(value, self.start_checker))
1411
self.start_checker() # Start one now, too
1413
# Checker - property
1414
@dbus_service_property(_interface, signature="s",
1416
def Checker_dbus_property(self, value=None):
1417
if value is None: # get
1418
return dbus.String(self.checker_command)
1419
self.checker_command = unicode(value)
1421
# CheckerRunning - property
1422
@dbus_service_property(_interface, signature="b",
1424
def CheckerRunning_dbus_property(self, value=None):
1425
if value is None: # get
1426
return dbus.Boolean(self.checker is not None)
1428
self.start_checker()
1432
# ObjectPath - property
1433
@dbus_service_property(_interface, signature="o", access="read")
1434
def ObjectPath_dbus_property(self):
1435
return self.dbus_object_path # is already a dbus.ObjectPath
1438
@dbus_service_property(_interface, signature="ay",
1439
access="write", byte_arrays=True)
1440
def Secret_dbus_property(self, value):
1441
self.secret = str(value)
580
StopChecker = dbus.service.method(_interface)(stop_checker)
581
StopChecker.__name__ = "StopChecker"
1446
class ProxyClient(object):
1447
def __init__(self, child_pipe, fpr, address):
1448
self._pipe = child_pipe
1449
self._pipe.send(('init', fpr, address))
1450
if not self._pipe.recv():
1453
def __getattribute__(self, name):
1455
return super(ProxyClient, self).__getattribute__(name)
1456
self._pipe.send(('getattr', name))
1457
data = self._pipe.recv()
1458
if data[0] == 'data':
1460
if data[0] == 'function':
1461
def func(*args, **kwargs):
1462
self._pipe.send(('funcall', name, args, kwargs))
1463
return self._pipe.recv()[1]
1466
def __setattr__(self, name, value):
1468
return super(ProxyClient, self).__setattr__(name, value)
1469
self._pipe.send(('setattr', name, value))
1472
class ClientDBusTransitional(ClientDBus):
1473
__metaclass__ = AlternateDBusNamesMetaclass
1476
class ClientHandler(socketserver.BaseRequestHandler, object):
1477
"""A class to handle client connections.
1479
Instantiated once for each connection to handle it.
586
def peer_certificate(session):
587
"Return the peer's OpenPGP certificate as a bytestring"
588
# If not an OpenPGP certificate...
589
if (gnutls.library.functions
590
.gnutls_certificate_type_get(session._c_object)
591
!= gnutls.library.constants.GNUTLS_CRT_OPENPGP):
592
# ...do the normal thing
593
return session.peer_certificate
594
list_size = ctypes.c_uint()
595
cert_list = (gnutls.library.functions
596
.gnutls_certificate_get_peers
597
(session._c_object, ctypes.byref(list_size)))
598
if list_size.value == 0:
601
return ctypes.string_at(cert.data, cert.size)
604
def fingerprint(openpgp):
605
"Convert an OpenPGP bytestring to a hexdigit fingerprint string"
606
# New GnuTLS "datum" with the OpenPGP public key
607
datum = (gnutls.library.types
608
.gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
611
ctypes.c_uint(len(openpgp))))
612
# New empty GnuTLS certificate
613
crt = gnutls.library.types.gnutls_openpgp_crt_t()
614
(gnutls.library.functions
615
.gnutls_openpgp_crt_init(ctypes.byref(crt)))
616
# Import the OpenPGP public key into the certificate
617
(gnutls.library.functions
618
.gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
619
gnutls.library.constants
620
.GNUTLS_OPENPGP_FMT_RAW))
621
# Verify the self signature in the key
622
crtverify = ctypes.c_uint()
623
(gnutls.library.functions
624
.gnutls_openpgp_crt_verify_self(crt, 0, ctypes.byref(crtverify)))
625
if crtverify.value != 0:
626
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
627
raise gnutls.errors.CertificateSecurityError("Verify failed")
628
# New buffer for the fingerprint
629
buf = ctypes.create_string_buffer(20)
630
buf_len = ctypes.c_size_t()
631
# Get the fingerprint from the certificate into the buffer
632
(gnutls.library.functions
633
.gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
634
ctypes.byref(buf_len)))
635
# Deinit the certificate
636
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
637
# Convert the buffer to a Python bytestring
638
fpr = ctypes.string_at(buf, buf_len.value)
639
# Convert the bytestring to hexadecimal notation
640
hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
644
class TCP_handler(SocketServer.BaseRequestHandler, object):
645
"""A TCP request handler class.
646
Instantiated by IPv6_TCPServer for each request to handle it.
1480
647
Note: This will run in its own forked process."""
1482
649
def handle(self):
1483
with contextlib.closing(self.server.child_pipe) as child_pipe:
1484
logger.info("TCP connection from: %s",
1485
unicode(self.client_address))
1486
logger.debug("Pipe FD: %d",
1487
self.server.child_pipe.fileno())
1489
session = (gnutls.connection
1490
.ClientSession(self.request,
1492
.X509Credentials()))
1494
# Note: gnutls.connection.X509Credentials is really a
1495
# generic GnuTLS certificate credentials object so long as
1496
# no X.509 keys are added to it. Therefore, we can use it
1497
# here despite using OpenPGP certificates.
1499
#priority = ':'.join(("NONE", "+VERS-TLS1.1",
1500
# "+AES-256-CBC", "+SHA1",
1501
# "+COMP-NULL", "+CTYPE-OPENPGP",
1503
# Use a fallback default, since this MUST be set.
1504
priority = self.server.gnutls_priority
1505
if priority is None:
1507
(gnutls.library.functions
1508
.gnutls_priority_set_direct(session._c_object,
1511
# Start communication using the Mandos protocol
1512
# Get protocol number
1513
line = self.request.makefile().readline()
1514
logger.debug("Protocol version: %r", line)
1516
if int(line.strip().split()[0]) > 1:
1518
except (ValueError, IndexError, RuntimeError) as error:
1519
logger.error("Unknown protocol version: %s", error)
1522
# Start GnuTLS connection
1525
except gnutls.errors.GNUTLSError as error:
1526
logger.warning("Handshake failed: %s", error)
1527
# Do not run session.bye() here: the session is not
1528
# established. Just abandon the request.
1530
logger.debug("Handshake succeeded")
1532
approval_required = False
1535
fpr = self.fingerprint(self.peer_certificate
1538
gnutls.errors.GNUTLSError) as error:
1539
logger.warning("Bad certificate: %s", error)
1541
logger.debug("Fingerprint: %s", fpr)
1544
client = ProxyClient(child_pipe, fpr,
1545
self.client_address)
1549
if self.server.use_dbus:
1551
client.NewRequest(str(self.client_address))
1553
if client.approval_delay:
1554
delay = client.approval_delay
1555
client.approvals_pending += 1
1556
approval_required = True
1559
if not client.enabled:
1560
logger.info("Client %s is disabled",
1562
if self.server.use_dbus:
1564
client.Rejected("Disabled")
1567
if client.approved or not client.approval_delay:
1568
#We are approved or approval is disabled
1570
elif client.approved is None:
1571
logger.info("Client %s needs approval",
1573
if self.server.use_dbus:
1575
client.NeedApproval(
1576
client.approval_delay_milliseconds(),
1577
client.approved_by_default)
1579
logger.warning("Client %s was not approved",
1581
if self.server.use_dbus:
1583
client.Rejected("Denied")
1586
#wait until timeout or approved
1587
time = datetime.datetime.now()
1588
client.changedstate.acquire()
1589
(client.changedstate.wait
1590
(float(client.timedelta_to_milliseconds(delay)
1592
client.changedstate.release()
1593
time2 = datetime.datetime.now()
1594
if (time2 - time) >= delay:
1595
if not client.approved_by_default:
1596
logger.warning("Client %s timed out while"
1597
" waiting for approval",
1599
if self.server.use_dbus:
1601
client.Rejected("Approval timed out")
1606
delay -= time2 - time
1609
while sent_size < len(client.secret):
1611
sent = session.send(client.secret[sent_size:])
1612
except gnutls.errors.GNUTLSError as error:
1613
logger.warning("gnutls send failed")
1615
logger.debug("Sent: %d, remaining: %d",
1616
sent, len(client.secret)
1617
- (sent_size + sent))
1620
logger.info("Sending secret to %s", client.name)
1621
# bump the timeout using extended_timeout
1622
client.checked_ok(client.extended_timeout)
1623
if self.server.use_dbus:
1628
if approval_required:
1629
client.approvals_pending -= 1
1632
except gnutls.errors.GNUTLSError as error:
1633
logger.warning("GnuTLS bye failed")
1636
def peer_certificate(session):
1637
"Return the peer's OpenPGP certificate as a bytestring"
1638
# If not an OpenPGP certificate...
1639
if (gnutls.library.functions
1640
.gnutls_certificate_type_get(session._c_object)
1641
!= gnutls.library.constants.GNUTLS_CRT_OPENPGP):
1642
# ...do the normal thing
1643
return session.peer_certificate
1644
list_size = ctypes.c_uint(1)
1645
cert_list = (gnutls.library.functions
1646
.gnutls_certificate_get_peers
1647
(session._c_object, ctypes.byref(list_size)))
1648
if not bool(cert_list) and list_size.value != 0:
1649
raise gnutls.errors.GNUTLSError("error getting peer"
1651
if list_size.value == 0:
1654
return ctypes.string_at(cert.data, cert.size)
1657
def fingerprint(openpgp):
1658
"Convert an OpenPGP bytestring to a hexdigit fingerprint"
1659
# New GnuTLS "datum" with the OpenPGP public key
1660
datum = (gnutls.library.types
1661
.gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
1664
ctypes.c_uint(len(openpgp))))
1665
# New empty GnuTLS certificate
1666
crt = gnutls.library.types.gnutls_openpgp_crt_t()
1667
(gnutls.library.functions
1668
.gnutls_openpgp_crt_init(ctypes.byref(crt)))
1669
# Import the OpenPGP public key into the certificate
1670
(gnutls.library.functions
1671
.gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
1672
gnutls.library.constants
1673
.GNUTLS_OPENPGP_FMT_RAW))
1674
# Verify the self signature in the key
1675
crtverify = ctypes.c_uint()
1676
(gnutls.library.functions
1677
.gnutls_openpgp_crt_verify_self(crt, 0,
1678
ctypes.byref(crtverify)))
1679
if crtverify.value != 0:
1680
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
1681
raise (gnutls.errors.CertificateSecurityError
1683
# New buffer for the fingerprint
1684
buf = ctypes.create_string_buffer(20)
1685
buf_len = ctypes.c_size_t()
1686
# Get the fingerprint from the certificate into the buffer
1687
(gnutls.library.functions
1688
.gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
1689
ctypes.byref(buf_len)))
1690
# Deinit the certificate
1691
gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
1692
# Convert the buffer to a Python bytestring
1693
fpr = ctypes.string_at(buf, buf_len.value)
1694
# Convert the bytestring to hexadecimal notation
1695
hex_fpr = binascii.hexlify(fpr).upper()
1699
class MultiprocessingMixIn(object):
1700
"""Like socketserver.ThreadingMixIn, but with multiprocessing"""
1701
def sub_process_main(self, request, address):
1703
self.finish_request(request, address)
1705
self.handle_error(request, address)
1706
self.close_request(request)
1708
def process_request(self, request, address):
1709
"""Start a new process to process the request."""
1710
proc = multiprocessing.Process(target = self.sub_process_main,
1717
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
1718
""" adds a pipe to the MixIn """
1719
def process_request(self, request, client_address):
1720
"""Overrides and wraps the original process_request().
1722
This function creates a new pipe in self.pipe
1724
parent_pipe, self.child_pipe = multiprocessing.Pipe()
1726
proc = MultiprocessingMixIn.process_request(self, request,
1728
self.child_pipe.close()
1729
self.add_pipe(parent_pipe, proc)
1731
def add_pipe(self, parent_pipe, proc):
1732
"""Dummy function; override as necessary"""
1733
raise NotImplementedError
1736
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
1737
socketserver.TCPServer, object):
1738
"""IPv6-capable TCP server. Accepts 'None' as address and/or port
650
logger.info(u"TCP connection from: %s",
651
unicode(self.client_address))
652
session = (gnutls.connection
653
.ClientSession(self.request,
657
line = self.request.makefile().readline()
658
logger.debug(u"Protocol version: %r", line)
660
if int(line.strip().split()[0]) > 1:
662
except (ValueError, IndexError, RuntimeError), error:
663
logger.error(u"Unknown protocol version: %s", error)
666
# Note: gnutls.connection.X509Credentials is really a generic
667
# GnuTLS certificate credentials object so long as no X.509
668
# keys are added to it. Therefore, we can use it here despite
669
# using OpenPGP certificates.
671
#priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
672
# "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
674
# Use a fallback default, since this MUST be set.
675
priority = self.server.settings.get("priority", "NORMAL")
676
(gnutls.library.functions
677
.gnutls_priority_set_direct(session._c_object,
682
except gnutls.errors.GNUTLSError, error:
683
logger.warning(u"Handshake failed: %s", error)
684
# Do not run session.bye() here: the session is not
685
# established. Just abandon the request.
688
fpr = fingerprint(peer_certificate(session))
689
except (TypeError, gnutls.errors.GNUTLSError), error:
690
logger.warning(u"Bad certificate: %s", error)
693
logger.debug(u"Fingerprint: %s", fpr)
694
for c in self.server.clients:
695
if c.fingerprint == fpr:
699
logger.warning(u"Client not found for fingerprint: %s",
703
# Have to check if client.still_valid(), since it is possible
704
# that the client timed out while establishing the GnuTLS
706
if not client.still_valid():
707
logger.warning(u"Client %(name)s is invalid",
711
## This won't work here, since we're in a fork.
712
# client.bump_timeout()
714
while sent_size < len(client.secret):
715
sent = session.send(client.secret[sent_size:])
716
logger.debug(u"Sent: %d, remaining: %d",
717
sent, len(client.secret)
718
- (sent_size + sent))
723
class IPv6_TCPServer(SocketServer.ForkingMixIn,
724
SocketServer.TCPServer, object):
725
"""IPv6 TCP server. Accepts 'None' as address and/or port.
727
settings: Server settings
728
clients: Set() of Client objects
1741
729
enabled: Boolean; whether this server is activated yet
1742
interface: None or a network interface name (string)
1743
use_ipv6: Boolean; to use IPv6 or not
1745
def __init__(self, server_address, RequestHandlerClass,
1746
interface=None, use_ipv6=True):
1747
self.interface = interface
1749
self.address_family = socket.AF_INET6
1750
socketserver.TCPServer.__init__(self, server_address,
1751
RequestHandlerClass)
731
address_family = socket.AF_INET6
732
def __init__(self, *args, **kwargs):
733
if "settings" in kwargs:
734
self.settings = kwargs["settings"]
735
del kwargs["settings"]
736
if "clients" in kwargs:
737
self.clients = kwargs["clients"]
738
del kwargs["clients"]
740
super(IPv6_TCPServer, self).__init__(*args, **kwargs)
1752
741
def server_bind(self):
1753
742
"""This overrides the normal server_bind() function
1754
743
to bind to an interface if one was specified, and also NOT to
1755
744
bind to an address or port if they were not specified."""
1756
if self.interface is not None:
1757
if SO_BINDTODEVICE is None:
1758
logger.error("SO_BINDTODEVICE does not exist;"
1759
" cannot bind to interface %s",
1763
self.socket.setsockopt(socket.SOL_SOCKET,
1767
except socket.error as error:
1768
if error[0] == errno.EPERM:
1769
logger.error("No permission to"
1770
" bind to interface %s",
1772
elif error[0] == errno.ENOPROTOOPT:
1773
logger.error("SO_BINDTODEVICE not available;"
1774
" cannot bind to interface %s",
745
if self.settings["interface"]:
746
# 25 is from /usr/include/asm-i486/socket.h
747
SO_BINDTODEVICE = getattr(socket, "SO_BINDTODEVICE", 25)
749
self.socket.setsockopt(socket.SOL_SOCKET,
751
self.settings["interface"])
752
except socket.error, error:
753
if error[0] == errno.EPERM:
754
logger.error(u"No permission to"
755
u" bind to interface %s",
756
self.settings["interface"])
1778
759
# Only bind(2) the socket if we really need to.
1779
760
if self.server_address[0] or self.server_address[1]:
1780
761
if not self.server_address[0]:
1781
if self.address_family == socket.AF_INET6:
1782
any_address = "::" # in6addr_any
1784
any_address = socket.INADDR_ANY
1785
self.server_address = (any_address,
763
self.server_address = (in6addr_any,
1786
764
self.server_address[1])
1787
765
elif not self.server_address[1]:
1788
766
self.server_address = (self.server_address[0],
1790
# if self.interface:
768
# if self.settings["interface"]:
1791
769
# self.server_address = (self.server_address[0],
1794
772
# if_nametoindex
1796
return socketserver.TCPServer.server_bind(self)
1799
class MandosServer(IPv6_TCPServer):
1803
clients: set of Client objects
1804
gnutls_priority GnuTLS priority string
1805
use_dbus: Boolean; to emit D-Bus signals or not
1807
Assumes a gobject.MainLoop event loop.
1809
def __init__(self, server_address, RequestHandlerClass,
1810
interface=None, use_ipv6=True, clients=None,
1811
gnutls_priority=None, use_dbus=True):
1812
self.enabled = False
1813
self.clients = clients
1814
if self.clients is None:
1816
self.use_dbus = use_dbus
1817
self.gnutls_priority = gnutls_priority
1818
IPv6_TCPServer.__init__(self, server_address,
1819
RequestHandlerClass,
1820
interface = interface,
1821
use_ipv6 = use_ipv6)
775
return super(IPv6_TCPServer, self).server_bind()
1822
776
def server_activate(self):
1823
777
if self.enabled:
1824
return socketserver.TCPServer.server_activate(self)
778
return super(IPv6_TCPServer, self).server_activate()
1826
779
def enable(self):
1827
780
self.enabled = True
1829
def add_pipe(self, parent_pipe, proc):
1830
# Call "handle_ipc" for both data and EOF events
1831
gobject.io_add_watch(parent_pipe.fileno(),
1832
gobject.IO_IN | gobject.IO_HUP,
1833
functools.partial(self.handle_ipc,
1838
def handle_ipc(self, source, condition, parent_pipe=None,
1839
proc = None, client_object=None):
1841
gobject.IO_IN: "IN", # There is data to read.
1842
gobject.IO_OUT: "OUT", # Data can be written (without
1844
gobject.IO_PRI: "PRI", # There is urgent data to read.
1845
gobject.IO_ERR: "ERR", # Error condition.
1846
gobject.IO_HUP: "HUP" # Hung up (the connection has been
1847
# broken, usually for pipes and
1850
conditions_string = ' | '.join(name
1852
condition_names.iteritems()
1853
if cond & condition)
1854
# error, or the other end of multiprocessing.Pipe has closed
1855
if condition & (gobject.IO_ERR | condition & gobject.IO_HUP):
1856
# Wait for other process to exit
1860
# Read a request from the child
1861
request = parent_pipe.recv()
1862
command = request[0]
1864
if command == 'init':
1866
address = request[2]
1868
for c in self.clients.itervalues():
1869
if c.fingerprint == fpr:
1873
logger.info("Client not found for fingerprint: %s, ad"
1874
"dress: %s", fpr, address)
1877
mandos_dbus_service.ClientNotFound(fpr,
1879
parent_pipe.send(False)
1882
gobject.io_add_watch(parent_pipe.fileno(),
1883
gobject.IO_IN | gobject.IO_HUP,
1884
functools.partial(self.handle_ipc,
1890
parent_pipe.send(True)
1891
# remove the old hook in favor of the new above hook on
1894
if command == 'funcall':
1895
funcname = request[1]
1899
parent_pipe.send(('data', getattr(client_object,
1903
if command == 'getattr':
1904
attrname = request[1]
1905
if callable(client_object.__getattribute__(attrname)):
1906
parent_pipe.send(('function',))
1908
parent_pipe.send(('data', client_object
1909
.__getattribute__(attrname)))
1911
if command == 'setattr':
1912
attrname = request[1]
1914
setattr(client_object, attrname, value)
1919
783
def string_to_delta(interval):
1920
784
"""Parse a string and return a datetime.timedelta
1922
786
>>> string_to_delta('7d')
1923
787
datetime.timedelta(7)
1924
788
>>> string_to_delta('60s')
2134
1001
except KeyError:
2136
1003
uid = pwd.getpwnam("nobody").pw_uid
2137
gid = pwd.getpwnam("nobody").pw_gid
1004
gid = pwd.getpwnam("nogroup").pw_gid
2138
1005
except KeyError:
2144
except OSError as error:
1011
except OSError, error:
2145
1012
if error[0] != errno.EPERM:
2149
# Enable all possible GnuTLS debugging
2151
# "Use a log level over 10 to enable all debugging options."
2153
gnutls.library.functions.gnutls_global_set_log_level(11)
2155
@gnutls.library.types.gnutls_log_func
2156
def debug_gnutls(level, string):
2157
logger.debug("GnuTLS: %s", string[:-1])
2159
(gnutls.library.functions
2160
.gnutls_global_set_log_function(debug_gnutls))
2162
# Redirect stdin so all checkers get /dev/null
2163
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
2164
os.dup2(null, sys.stdin.fileno())
2168
# No console logging
2169
logger.removeHandler(console)
2171
# Need to fork before connecting to D-Bus
2173
# Close all input and output, do double fork, etc.
1016
service = AvahiService(name = server_settings["servicename"],
1017
servicetype = "_mandos._tcp", )
1018
if server_settings["interface"]:
1019
service.interface = (if_nametoindex
1020
(server_settings["interface"]))
2176
1022
global main_loop
2177
1025
# From the Avahi example code
2178
1026
DBusGMainLoop(set_as_default=True )
2179
1027
main_loop = gobject.MainLoop()
2180
1028
bus = dbus.SystemBus()
1029
server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
1030
avahi.DBUS_PATH_SERVER),
1031
avahi.DBUS_INTERFACE_SERVER)
2181
1032
# End of Avahi example code
2184
bus_name = dbus.service.BusName("se.recompile.Mandos",
2185
bus, do_not_queue=True)
2186
old_bus_name = (dbus.service.BusName
2187
("se.bsnet.fukt.Mandos", bus,
2189
except dbus.exceptions.NameExistsException as e:
2190
logger.error(unicode(e) + ", disabling D-Bus")
2192
server_settings["use_dbus"] = False
2193
tcp_server.use_dbus = False
2194
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
2195
service = AvahiServiceToSyslog(name =
2196
server_settings["servicename"],
2197
servicetype = "_mandos._tcp",
2198
protocol = protocol, bus = bus)
2199
if server_settings["interface"]:
2200
service.interface = (if_nametoindex
2201
(str(server_settings["interface"])))
2203
global multiprocessing_manager
2204
multiprocessing_manager = multiprocessing.Manager()
2206
client_class = Client
2208
client_class = functools.partial(ClientDBusTransitional,
2211
client_settings = Client.config_parser(client_config)
2212
old_client_settings = {}
2215
# Get client data and settings from last running state.
2216
if server_settings["restore"]:
2218
with open(stored_state_path, "rb") as stored_state:
2219
clients_data, old_client_settings = (pickle.load
2221
os.remove(stored_state_path)
2222
except IOError as e:
2223
logger.warning("Could not load persistent state: {0}"
2225
if e.errno != errno.ENOENT:
2228
with PGPEngine() as pgp:
2229
for client in clients_data:
2230
client_name = client["name"]
2232
# Decide which value to use after restoring saved state.
2233
# We have three different values: Old config file,
2234
# new config file, and saved state.
2235
# New config value takes precedence if it differs from old
2236
# config value, otherwise use saved state.
2237
for name, value in client_settings[client_name].items():
2239
# For each value in new config, check if it
2240
# differs from the old config value (Except for
2241
# the "secret" attribute)
2242
if (name != "secret" and
2243
value != old_client_settings[client_name]
2245
client[name] = value
2249
# Clients who has passed its expire date can still be
2250
# enabled if its last checker was successful. Clients
2251
# whose checker failed before we stored its state is
2252
# assumed to have failed all checkers during downtime.
2253
if client["enabled"]:
2254
if datetime.datetime.utcnow() >= client["expires"]:
2255
if not client["last_checked_ok"]:
2257
"disabling client {0} - Client never "
2258
"performed a successfull checker"
2259
.format(client["name"]))
2260
client["enabled"] = False
2261
elif client["last_checker_status"] != 0:
2263
"disabling client {0} - Client "
2264
"last checker failed with error code {1}"
2265
.format(client["name"],
2266
client["last_checker_status"]))
2267
client["enabled"] = False
2269
client["expires"] = (datetime.datetime
2271
+ client["timeout"])
2273
client["changedstate"] = (multiprocessing_manager
2275
(multiprocessing_manager
2277
client["checker"] = None
2279
new_client = (ClientDBusTransitional.__new__
2280
(ClientDBusTransitional))
2281
tcp_server.clients[client_name] = new_client
2282
new_client.bus = bus
2283
for name, value in client.iteritems():
2284
setattr(new_client, name, value)
2285
client_object_name = unicode(client_name).translate(
2286
{ord("."): ord("_"),
2287
ord("-"): ord("_")})
2288
new_client.dbus_object_path = (dbus.ObjectPath
2290
+ client_object_name))
2291
DBusObjectWithProperties.__init__(new_client,
2296
tcp_server.clients[client_name] = (Client.__new__
2298
for name, value in client.iteritems():
2299
setattr(tcp_server.clients[client_name],
2303
tcp_server.clients[client_name].secret = (
2304
pgp.decrypt(tcp_server.clients[client_name]
2306
client_settings[client_name]
2309
# If decryption fails, we use secret from new settings
2310
logger.debug("Failed to decrypt {0} old secret"
2311
.format(client_name))
2312
tcp_server.clients[client_name].secret = (
2313
client_settings[client_name]["secret"])
2315
# Create/remove clients based on new changes made to config
2316
for clientname in set(old_client_settings) - set(client_settings):
2317
del tcp_server.clients[clientname]
2318
for clientname in set(client_settings) - set(old_client_settings):
2319
tcp_server.clients[clientname] = (client_class(name = clientname,
2324
if not tcp_server.clients:
2325
logger.warning("No clients defined")
1034
bus_name = dbus.service.BusName(u"org.mandos-system.Mandos",
1037
clients.update(Set(Client(name = section,
1039
= dict(client_config.items(section)),
1040
use_dbus = use_dbus)
1041
for section in client_config.sections()))
1043
logger.warning(u"No clients defined")
1046
# Redirect stdin so all checkers get /dev/null
1047
null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
1048
os.dup2(null, sys.stdin.fileno())
1052
# No console logging
1053
logger.removeHandler(console)
1054
# Close all input and output, do double fork, etc.
1059
pidfile.write(str(pid) + "\n")
1063
logger.error(u"Could not write to file %r with PID %d",
1066
# "pidfile" was never created
1071
"Cleanup function; run on exit"
1073
# From the Avahi example code
1074
if not group is None:
1077
# End of Avahi example code
1080
client = clients.pop()
1081
client.disable_hook = None
1084
atexit.register(cleanup)
2331
pidfile.write(str(pid) + "\n".encode("utf-8"))
2334
logger.error("Could not write to file %r with PID %d",
2337
# "pidfile" was never created
2340
1087
signal.signal(signal.SIGINT, signal.SIG_IGN)
2342
1088
signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
2343
1089
signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
2346
class MandosDBusService(dbus.service.Object):
1092
class MandosServer(dbus.service.Object):
2347
1093
"""A D-Bus proxy object"""
2348
1094
def __init__(self):
2349
dbus.service.Object.__init__(self, bus, "/")
2350
_interface = "se.recompile.Mandos"
1095
dbus.service.Object.__init__(self, bus,
1097
_interface = u"org.mandos_system.Mandos"
1099
@dbus.service.signal(_interface, signature="oa{sv}")
1100
def ClientAdded(self, objpath, properties):
2352
1104
@dbus.service.signal(_interface, signature="o")
2353
def ClientAdded(self, objpath):
2357
@dbus.service.signal(_interface, signature="ss")
2358
def ClientNotFound(self, fingerprint, address):
2362
@dbus.service.signal(_interface, signature="os")
2363
def ClientRemoved(self, objpath, name):
1105
def ClientRemoved(self, objpath):
2367
1109
@dbus.service.method(_interface, out_signature="ao")
2368
1110
def GetAllClients(self):
2370
return dbus.Array(c.dbus_object_path
2372
tcp_server.clients.itervalues())
2374
@dbus.service.method(_interface,
2375
out_signature="a{oa{sv}}")
1111
return dbus.Array(c.dbus_object_path for c in clients)
1113
@dbus.service.method(_interface, out_signature="a{oa{sv}}")
2376
1114
def GetAllClientsWithProperties(self):
2378
1115
return dbus.Dictionary(
2379
((c.dbus_object_path, c.GetAll(""))
2380
for c in tcp_server.clients.itervalues()),
1116
((c.dbus_object_path, c.GetAllProperties())
2381
1118
signature="oa{sv}")
2383
1120
@dbus.service.method(_interface, in_signature="o")
2384
1121
def RemoveClient(self, object_path):
2386
for c in tcp_server.clients.itervalues():
2387
1123
if c.dbus_object_path == object_path:
2388
del tcp_server.clients[c.name]
2389
c.remove_from_connection()
2390
1125
# Don't signal anything except ClientRemoved
2391
c.disable(quiet=True)
2392
1128
# Emit D-Bus signal
2393
self.ClientRemoved(object_path, c.name)
1129
self.ClientRemoved(object_path)
2395
raise KeyError(object_path)
1132
@dbus.service.method(_interface)
2399
class MandosDBusServiceTransitional(MandosDBusService):
2400
__metaclass__ = AlternateDBusNamesMetaclass
2401
mandos_dbus_service = MandosDBusServiceTransitional()
2404
"Cleanup function; run on exit"
2407
multiprocessing.active_children()
2408
if not (tcp_server.clients or client_settings):
2411
# Store client before exiting. Secrets are encrypted with key
2412
# based on what config file has. If config file is
2413
# removed/edited, old secret will thus be unrecovable.
2415
with PGPEngine() as pgp:
2416
for client in tcp_server.clients.itervalues():
2417
key = client_settings[client.name]["secret"]
2418
client.encrypted_secret = pgp.encrypt(client.secret,
2422
# A list of attributes that will not be stored when
2424
exclude = set(("bus", "changedstate", "secret",
2426
for name, typ in (inspect.getmembers
2427
(dbus.service.Object)):
2430
client_dict["encrypted_secret"] = (client
2432
for attr in client.client_structure:
2433
if attr not in exclude:
2434
client_dict[attr] = getattr(client, attr)
2436
clients.append(client_dict)
2437
del client_settings[client.name]["secret"]
2440
with os.fdopen(os.open(stored_state_path,
2441
os.O_CREAT|os.O_WRONLY|os.O_TRUNC,
2442
0600), "wb") as stored_state:
2443
pickle.dump((clients, client_settings), stored_state)
2444
except (IOError, OSError) as e:
2445
logger.warning("Could not save persistent state: {0}"
2447
if e.errno not in (errno.ENOENT, errno.EACCES):
2450
# Delete all clients, and settings from config
2451
while tcp_server.clients:
2452
name, client = tcp_server.clients.popitem()
2454
client.remove_from_connection()
2455
# Don't signal anything except ClientRemoved
2456
client.disable(quiet=True)
2459
mandos_dbus_service.ClientRemoved(client
2462
client_settings.clear()
2464
atexit.register(cleanup)
2466
for client in tcp_server.clients.itervalues():
1138
mandos_server = MandosServer()
1140
for client in clients:
2468
1142
# Emit D-Bus signal
2469
mandos_dbus_service.ClientAdded(client.dbus_object_path)
2470
# Need to initiate checking of clients
2472
client.init_checker()
1143
mandos_server.ClientAdded(client.dbus_object_path,
1144
client.GetAllProperties())
2474
1147
tcp_server.enable()
2475
1148
tcp_server.server_activate()
2477
1150
# Find out what port we got
2478
1151
service.port = tcp_server.socket.getsockname()[1]
2480
logger.info("Now listening on address %r, port %d,"
2481
" flowinfo %d, scope_id %d"
2482
% tcp_server.socket.getsockname())
2484
logger.info("Now listening on address %r, port %d"
2485
% tcp_server.socket.getsockname())
1152
logger.info(u"Now listening on address %r, port %d, flowinfo %d,"
1153
u" scope_id %d" % tcp_server.socket.getsockname())
2487
1155
#service.interface = tcp_server.socket.getsockname()[3]
2490
1158
# From the Avahi example code
1159
server.connect_to_signal("StateChanged", server_state_changed)
2493
except dbus.exceptions.DBusException as error:
2494
logger.critical("DBusException: %s", error)
1161
server_state_changed(server.GetState())
1162
except dbus.exceptions.DBusException, error:
1163
logger.critical(u"DBusException: %s", error)
2497
1165
# End of Avahi example code