81
86
SO_BINDTODEVICE = None
90
stored_state_path = "/var/lib/mandos/clients.pickle"
86
#logger = logging.getLogger(u'mandos')
87
logger = logging.Logger(u'mandos')
92
logger = logging.getLogger()
88
93
syslogger = (logging.handlers.SysLogHandler
89
94
(facility = logging.handlers.SysLogHandler.LOG_DAEMON,
90
address = "/dev/log"))
91
syslogger.setFormatter(logging.Formatter
92
(u'Mandos [%(process)d]: %(levelname)s:'
94
logger.addHandler(syslogger)
96
console = logging.StreamHandler()
97
console.setFormatter(logging.Formatter(u'%(name)s [%(process)d]:'
100
logger.addHandler(console)
95
address = str("/dev/log")))
98
if_nametoindex = (ctypes.cdll.LoadLibrary
99
(ctypes.util.find_library("c"))
101
except (OSError, AttributeError):
102
def if_nametoindex(interface):
103
"Get an interface index the hard way, i.e. using fcntl()"
104
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
105
with contextlib.closing(socket.socket()) as s:
106
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
107
struct.pack(str("16s16x"),
109
interface_index = struct.unpack(str("I"),
111
return interface_index
114
def initlogger(level=logging.WARNING):
115
"""init logger and add loglevel"""
117
syslogger.setFormatter(logging.Formatter
118
('Mandos [%(process)d]: %(levelname)s:'
120
logger.addHandler(syslogger)
122
console = logging.StreamHandler()
123
console.setFormatter(logging.Formatter('%(asctime)s %(name)s'
127
logger.addHandler(console)
128
logger.setLevel(level)
131
class CryptoError(Exception):
135
class Crypto(object):
136
"""A simple class for OpenPGP symmetric encryption & decryption"""
138
self.gnupg = GnuPGInterface.GnuPG()
139
self.tempdir = tempfile.mkdtemp(prefix="mandos-")
140
self.gnupg = GnuPGInterface.GnuPG()
141
self.gnupg.options.meta_interactive = False
142
self.gnupg.options.homedir = self.tempdir
143
self.gnupg.options.extra_args.extend(['--force-mdc',
149
def __exit__ (self, exc_type, exc_value, traceback):
157
if self.tempdir is not None:
158
# Delete contents of tempdir
159
for root, dirs, files in os.walk(self.tempdir,
161
for filename in files:
162
os.remove(os.path.join(root, filename))
164
os.rmdir(os.path.join(root, dirname))
166
os.rmdir(self.tempdir)
169
def password_encode(self, password):
170
# Passphrase can not be empty and can not contain newlines or
171
# NUL bytes. So we prefix it and hex encode it.
172
return b"mandos" + binascii.hexlify(password)
174
def encrypt(self, data, password):
175
self.gnupg.passphrase = self.password_encode(password)
176
with open(os.devnull) as devnull:
178
proc = self.gnupg.run(['--symmetric'],
179
create_fhs=['stdin', 'stdout'],
180
attach_fhs={'stderr': devnull})
181
with contextlib.closing(proc.handles['stdin']) as f:
183
with contextlib.closing(proc.handles['stdout']) as f:
184
ciphertext = f.read()
188
self.gnupg.passphrase = None
191
def decrypt(self, data, password):
192
self.gnupg.passphrase = self.password_encode(password)
193
with open(os.devnull) as devnull:
195
proc = self.gnupg.run(['--decrypt'],
196
create_fhs=['stdin', 'stdout'],
197
attach_fhs={'stderr': devnull})
198
with contextlib.closing(proc.handles['stdin'] ) as f:
200
with contextlib.closing(proc.handles['stdout']) as f:
201
decrypted_plaintext = f.read()
205
self.gnupg.passphrase = None
206
return decrypted_plaintext
102
210
class AvahiError(Exception):
103
211
def __init__(self, value, *args, **kwargs):
150
258
self.group = None # our entry group
151
259
self.server = None
261
self.entry_group_state_changed_match = None
153
262
def rename(self):
154
263
"""Derived from the Avahi example code"""
155
264
if self.rename_count >= self.max_renames:
156
logger.critical(u"No suitable Zeroconf service name found"
157
u" after %i retries, exiting.",
265
logger.critical("No suitable Zeroconf service name found"
266
" after %i retries, exiting.",
158
267
self.rename_count)
159
raise AvahiServiceError(u"Too many renames")
160
self.name = unicode(self.server.GetAlternativeServiceName(self.name))
161
logger.info(u"Changing Zeroconf service name to %r ...",
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 ...",
163
syslogger.setFormatter(logging.Formatter
164
(u'Mandos (%s) [%%(process)d]:'
165
u' %%(levelname)s: %%(message)s'
170
except dbus.exceptions.DBusException, error:
171
logger.critical(u"DBusException: %s", error)
276
except dbus.exceptions.DBusException as error:
277
logger.critical("DBusException: %s", error)
174
280
self.rename_count += 1
175
281
def remove(self):
176
282
"""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
177
286
if self.group is not None:
178
287
self.group.Reset()
180
289
"""Derived from the Avahi example code"""
181
291
if self.group is None:
182
292
self.group = dbus.Interface(
183
293
self.bus.get_object(avahi.DBUS_NAME,
184
294
self.server.EntryGroupNew()),
185
295
avahi.DBUS_INTERFACE_ENTRY_GROUP)
186
self.group.connect_to_signal('StateChanged',
188
.entry_group_state_changed)
189
logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
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' ...",
190
300
self.name, self.type)
191
301
self.group.AddService(
199
309
self.group.Commit()
200
310
def entry_group_state_changed(self, state, error):
201
311
"""Derived from the Avahi example code"""
202
logger.debug(u"Avahi entry group state change: %i", state)
312
logger.debug("Avahi entry group state change: %i", state)
204
314
if state == avahi.ENTRY_GROUP_ESTABLISHED:
205
logger.debug(u"Zeroconf service established.")
315
logger.debug("Zeroconf service established.")
206
316
elif state == avahi.ENTRY_GROUP_COLLISION:
207
logger.warning(u"Zeroconf service name collision.")
317
logger.info("Zeroconf service name collision.")
209
319
elif state == avahi.ENTRY_GROUP_FAILURE:
210
logger.critical(u"Avahi: Error in group state changed %s",
320
logger.critical("Avahi: Error in group state changed %s",
212
raise AvahiGroupError(u"State changed: %s"
322
raise AvahiGroupError("State changed: %s"
213
323
% unicode(error))
214
324
def cleanup(self):
215
325
"""Derived from the Avahi example code"""
216
326
if self.group is not None:
329
except (dbus.exceptions.UnknownMethodException,
330
dbus.exceptions.DBusException):
218
332
self.group = None
219
def server_state_changed(self, state):
334
def server_state_changed(self, state, error=None):
220
335
"""Derived from the Avahi example code"""
221
logger.debug(u"Avahi server state change: %i", state)
222
if state == avahi.SERVER_COLLISION:
223
logger.error(u"Zeroconf server name collision")
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)
225
351
elif state == avahi.SERVER_RUNNING:
355
logger.debug("Unknown state: %r", state)
357
logger.debug("Unknown state: %r: %r", state, error)
227
358
def activate(self):
228
359
"""Derived from the Avahi example code"""
229
360
if self.server is None:
230
361
self.server = dbus.Interface(
231
362
self.bus.get_object(avahi.DBUS_NAME,
232
avahi.DBUS_PATH_SERVER),
363
avahi.DBUS_PATH_SERVER,
364
follow_name_owner_changes=True),
233
365
avahi.DBUS_INTERFACE_SERVER)
234
self.server.connect_to_signal(u"StateChanged",
366
self.server.connect_to_signal("StateChanged",
235
367
self.server_state_changed)
236
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))
239
386
class Client(object):
240
387
"""A representation of a client host served by this server.
263
411
interval: datetime.timedelta(); How often to start a new checker
264
412
last_approval_request: datetime.datetime(); (UTC) or None
265
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
266
418
last_enabled: datetime.datetime(); (UTC)
267
419
name: string; from the config file, used in log messages and
268
420
D-Bus identifiers
269
421
secret: bytestring; sent verbatim (over TLS) to client
270
422
timeout: datetime.timedelta(); How long from last_checked_ok
271
423
until this client is disabled
424
extended_timeout: extra long timeout when password has been sent
272
425
runtime_expansions: Allowed attributes for runtime expansion.
426
expires: datetime.datetime(); time (UTC) when a client will be
275
runtime_expansions = (u"approval_delay", u"approval_duration",
276
u"created", u"enabled", u"fingerprint",
277
u"host", u"interval", u"last_checked_ok",
278
u"last_enabled", u"name", u"timeout")
281
def _timedelta_to_milliseconds(td):
282
"Convert a datetime.timedelta() to milliseconds"
283
return ((td.days * 24 * 60 * 60 * 1000)
284
+ (td.seconds * 1000)
285
+ (td.microseconds // 1000))
430
runtime_expansions = ("approval_delay", "approval_duration",
431
"created", "enabled", "fingerprint",
432
"host", "interval", "last_checked_ok",
433
"last_enabled", "name", "timeout")
287
435
def timeout_milliseconds(self):
288
436
"Return the 'timeout' attribute in milliseconds"
289
return self._timedelta_to_milliseconds(self.timeout)
437
return _timedelta_to_milliseconds(self.timeout)
439
def extended_timeout_milliseconds(self):
440
"Return the 'extended_timeout' attribute in milliseconds"
441
return _timedelta_to_milliseconds(self.extended_timeout)
291
443
def interval_milliseconds(self):
292
444
"Return the 'interval' attribute in milliseconds"
293
return self._timedelta_to_milliseconds(self.interval)
445
return _timedelta_to_milliseconds(self.interval)
295
447
def approval_delay_milliseconds(self):
296
return self._timedelta_to_milliseconds(self.approval_delay)
448
return _timedelta_to_milliseconds(self.approval_delay)
298
def __init__(self, name = None, disable_hook=None, config=None):
450
def __init__(self, name = None, config=None):
299
451
"""Note: the 'checker' key in 'config' sets the
300
452
'checker_command' attribute and *not* the 'checker'
303
455
if config is None:
305
logger.debug(u"Creating client %r", self.name)
457
logger.debug("Creating client %r", self.name)
306
458
# Uppercase and remove spaces from fingerprint for later
307
459
# comparison purposes with return value from the fingerprint()
309
self.fingerprint = (config[u"fingerprint"].upper()
311
logger.debug(u" Fingerprint: %s", self.fingerprint)
312
if u"secret" in config:
313
self.secret = config[u"secret"].decode(u"base64")
314
elif u"secfile" in config:
461
self.fingerprint = (config["fingerprint"].upper()
463
logger.debug(" Fingerprint: %s", self.fingerprint)
464
if "secret" in config:
465
self.secret = config["secret"].decode("base64")
466
elif "secfile" in config:
315
467
with open(os.path.expanduser(os.path.expandvars
316
(config[u"secfile"])),
468
(config["secfile"])),
317
469
"rb") as secfile:
318
470
self.secret = secfile.read()
320
raise TypeError(u"No secret or secfile for client %s"
472
raise TypeError("No secret or secfile for client %s"
322
self.host = config.get(u"host", u"")
474
self.host = config.get("host", "")
323
475
self.created = datetime.datetime.utcnow()
325
477
self.last_approval_request = None
326
self.last_enabled = None
478
self.last_enabled = datetime.datetime.utcnow()
327
479
self.last_checked_ok = None
328
self.timeout = string_to_delta(config[u"timeout"])
329
self.interval = string_to_delta(config[u"interval"])
330
self.disable_hook = disable_hook
480
self.last_checker_status = None
481
self.timeout = string_to_delta(config["timeout"])
482
self.extended_timeout = string_to_delta(config
483
["extended_timeout"])
484
self.interval = string_to_delta(config["interval"])
331
485
self.checker = None
332
486
self.checker_initiator_tag = None
333
487
self.disable_initiator_tag = None
488
self.expires = datetime.datetime.utcnow() + self.timeout
334
489
self.checker_callback_tag = None
335
self.checker_command = config[u"checker"]
490
self.checker_command = config["checker"]
336
491
self.current_checker_command = None
337
self.last_connect = None
338
492
self._approved = None
339
self.approved_by_default = config.get(u"approved_by_default",
493
self.approved_by_default = config.get("approved_by_default",
341
495
self.approvals_pending = 0
342
496
self.approval_delay = string_to_delta(
343
config[u"approval_delay"])
497
config["approval_delay"])
344
498
self.approval_duration = string_to_delta(
345
config[u"approval_duration"])
346
self.changedstate = multiprocessing_manager.Condition(multiprocessing_manager.Lock())
499
config["approval_duration"])
500
self.changedstate = (multiprocessing_manager
501
.Condition(multiprocessing_manager
503
self.client_structure = [attr for attr in
504
self.__dict__.iterkeys()
505
if not attr.startswith("_")]
506
self.client_structure.append("client_structure")
508
for name, t in inspect.getmembers(type(self),
512
if not name.startswith("_"):
513
self.client_structure.append(name)
515
# Send notice to process children that client state has changed
348
516
def send_changedstate(self):
349
self.changedstate.acquire()
350
self.changedstate.notify_all()
351
self.changedstate.release()
517
with self.changedstate:
518
self.changedstate.notify_all()
353
520
def enable(self):
354
521
"""Start this client's checker and timeout hooks"""
355
if getattr(self, u"enabled", False):
522
if getattr(self, "enabled", False):
356
523
# Already enabled
358
525
self.send_changedstate()
526
self.expires = datetime.datetime.utcnow() + self.timeout
359
528
self.last_enabled = datetime.datetime.utcnow()
531
def disable(self, quiet=True):
532
"""Disable this client."""
533
if not getattr(self, "enabled", False):
536
self.send_changedstate()
538
logger.info("Disabling client %s", self.name)
539
if getattr(self, "disable_initiator_tag", False):
540
gobject.source_remove(self.disable_initiator_tag)
541
self.disable_initiator_tag = None
543
if getattr(self, "checker_initiator_tag", False):
544
gobject.source_remove(self.checker_initiator_tag)
545
self.checker_initiator_tag = None
548
# Do not run this again if called by a gobject.timeout_add
554
def init_checker(self):
360
555
# Schedule a new checker to be started an 'interval' from now,
361
556
# and every interval from then on.
362
557
self.checker_initiator_tag = (gobject.timeout_add
366
561
self.disable_initiator_tag = (gobject.timeout_add
367
562
(self.timeout_milliseconds(),
370
564
# Also start a new checker *right now*.
371
565
self.start_checker()
373
def disable(self, quiet=True):
374
"""Disable this client."""
375
if not getattr(self, "enabled", False):
378
self.send_changedstate()
380
logger.info(u"Disabling client %s", self.name)
381
if getattr(self, u"disable_initiator_tag", False):
382
gobject.source_remove(self.disable_initiator_tag)
383
self.disable_initiator_tag = None
384
if getattr(self, u"checker_initiator_tag", False):
385
gobject.source_remove(self.checker_initiator_tag)
386
self.checker_initiator_tag = None
388
if self.disable_hook:
389
self.disable_hook(self)
391
# Do not run this again if called by a gobject.timeout_add
395
self.disable_hook = None
398
567
def checker_callback(self, pid, condition, command):
399
568
"""The checker has completed, so take appropriate actions."""
400
569
self.checker_callback_tag = None
401
570
self.checker = None
402
571
if os.WIFEXITED(condition):
403
exitstatus = os.WEXITSTATUS(condition)
405
logger.info(u"Checker for %(name)s succeeded",
572
self.last_checker_status = os.WEXITSTATUS(condition)
573
if self.last_checker_status == 0:
574
logger.info("Checker for %(name)s succeeded",
407
576
self.checked_ok()
409
logger.info(u"Checker for %(name)s failed",
578
logger.info("Checker for %(name)s failed",
412
logger.warning(u"Checker for %(name)s crashed?",
581
self.last_checker_status = -1
582
logger.warning("Checker for %(name)s crashed?",
415
def checked_ok(self):
585
def checked_ok(self, timeout=None):
416
586
"""Bump up the timeout for this client.
418
588
This should only be called when the client has been seen,
592
timeout = self.timeout
421
593
self.last_checked_ok = datetime.datetime.utcnow()
422
gobject.source_remove(self.disable_initiator_tag)
423
self.disable_initiator_tag = (gobject.timeout_add
424
(self.timeout_milliseconds(),
594
if self.disable_initiator_tag is not None:
595
gobject.source_remove(self.disable_initiator_tag)
596
if getattr(self, "enabled", False):
597
self.disable_initiator_tag = (gobject.timeout_add
598
(_timedelta_to_milliseconds
599
(timeout), self.disable))
600
self.expires = datetime.datetime.utcnow() + timeout
427
602
def need_approval(self):
428
603
self.last_approval_request = datetime.datetime.utcnow()
581
757
def _is_dbus_property(obj):
582
return getattr(obj, u"_dbus_is_property", False)
758
return getattr(obj, "_dbus_is_property", False)
584
760
def _get_all_dbus_properties(self):
585
761
"""Returns a generator of (name, attribute) pairs
587
return ((prop._dbus_name, prop)
763
return ((prop.__get__(self)._dbus_name, prop.__get__(self))
764
for cls in self.__class__.__mro__
588
765
for name, prop in
589
inspect.getmembers(self, self._is_dbus_property))
766
inspect.getmembers(cls, self._is_dbus_property))
591
768
def _get_dbus_property(self, interface_name, property_name):
592
769
"""Returns a bound method if one exists which is a D-Bus
593
770
property with the specified name and interface.
595
for name in (property_name,
596
property_name + u"_dbus_property"):
597
prop = getattr(self, name, None)
599
or not self._is_dbus_property(prop)
600
or prop._dbus_name != property_name
601
or (interface_name and prop._dbus_interface
602
and interface_name != prop._dbus_interface)):
772
for cls in self.__class__.__mro__:
773
for name, value in (inspect.getmembers
774
(cls, self._is_dbus_property)):
775
if (value._dbus_name == property_name
776
and value._dbus_interface == interface_name):
777
return value.__get__(self)
605
779
# No such property
606
raise DBusPropertyNotFound(self.dbus_object_path + u":"
607
+ interface_name + u"."
780
raise DBusPropertyNotFound(self.dbus_object_path + ":"
781
+ interface_name + "."
610
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"ss",
784
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="ss",
612
786
def Get(self, interface_name, property_name):
613
787
"""Standard D-Bus property Get() method, see D-Bus standard.
615
789
prop = self._get_dbus_property(interface_name, property_name)
616
if prop._dbus_access == u"write":
790
if prop._dbus_access == "write":
617
791
raise DBusPropertyAccessException(property_name)
619
if not hasattr(value, u"variant_level"):
793
if not hasattr(value, "variant_level"):
621
795
return type(value)(value, variant_level=value.variant_level+1)
623
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"ssv")
797
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="ssv")
624
798
def Set(self, interface_name, property_name, value):
625
799
"""Standard D-Bus property Set() method, see D-Bus standard.
627
801
prop = self._get_dbus_property(interface_name, property_name)
628
if prop._dbus_access == u"read":
802
if prop._dbus_access == "read":
629
803
raise DBusPropertyAccessException(property_name)
630
if prop._dbus_get_args_options[u"byte_arrays"]:
804
if prop._dbus_get_args_options["byte_arrays"]:
631
805
# The byte_arrays option is not supported yet on
632
806
# signatures other than "ay".
633
if prop._dbus_signature != u"ay":
807
if prop._dbus_signature != "ay":
635
809
value = dbus.ByteArray(''.join(unichr(byte)
636
810
for byte in value))
639
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"s",
640
out_signature=u"a{sv}")
813
@dbus.service.method(dbus.PROPERTIES_IFACE, in_signature="s",
814
out_signature="a{sv}")
641
815
def GetAll(self, interface_name):
642
816
"""Standard D-Bus property GetAll() method, see D-Bus
645
819
Note: Will not include properties with access="write".
648
822
for name, prop in self._get_all_dbus_properties():
649
823
if (interface_name
650
824
and interface_name != prop._dbus_interface):
651
825
# Interface non-empty but did not match
653
827
# Ignore write-only properties
654
if prop._dbus_access == u"write":
828
if prop._dbus_access == "write":
657
if not hasattr(value, u"variant_level"):
831
if not hasattr(value, "variant_level"):
832
properties[name] = value
660
all[name] = type(value)(value, variant_level=
661
value.variant_level+1)
662
return dbus.Dictionary(all, signature=u"sv")
834
properties[name] = type(value)(value, variant_level=
835
value.variant_level+1)
836
return dbus.Dictionary(properties, signature="sv")
664
838
@dbus.service.method(dbus.INTROSPECTABLE_IFACE,
666
840
path_keyword='object_path',
667
841
connection_keyword='connection')
668
842
def Introspect(self, object_path, connection):
674
848
document = xml.dom.minidom.parseString(xmlstring)
675
849
def make_tag(document, name, prop):
676
e = document.createElement(u"property")
677
e.setAttribute(u"name", name)
678
e.setAttribute(u"type", prop._dbus_signature)
679
e.setAttribute(u"access", prop._dbus_access)
850
e = document.createElement("property")
851
e.setAttribute("name", name)
852
e.setAttribute("type", prop._dbus_signature)
853
e.setAttribute("access", prop._dbus_access)
681
for if_tag in document.getElementsByTagName(u"interface"):
855
for if_tag in document.getElementsByTagName("interface"):
682
856
for tag in (make_tag(document, name, prop)
684
858
in self._get_all_dbus_properties()
685
859
if prop._dbus_interface
686
== if_tag.getAttribute(u"name")):
860
== if_tag.getAttribute("name")):
687
861
if_tag.appendChild(tag)
688
862
# Add the names to the return values for the
689
863
# "org.freedesktop.DBus.Properties" methods
690
if (if_tag.getAttribute(u"name")
691
== u"org.freedesktop.DBus.Properties"):
692
for cn in if_tag.getElementsByTagName(u"method"):
693
if cn.getAttribute(u"name") == u"Get":
694
for arg in cn.getElementsByTagName(u"arg"):
695
if (arg.getAttribute(u"direction")
697
arg.setAttribute(u"name", u"value")
698
elif cn.getAttribute(u"name") == u"GetAll":
699
for arg in cn.getElementsByTagName(u"arg"):
700
if (arg.getAttribute(u"direction")
702
arg.setAttribute(u"name", u"props")
703
xmlstring = document.toxml(u"utf-8")
864
if (if_tag.getAttribute("name")
865
== "org.freedesktop.DBus.Properties"):
866
for cn in if_tag.getElementsByTagName("method"):
867
if cn.getAttribute("name") == "Get":
868
for arg in cn.getElementsByTagName("arg"):
869
if (arg.getAttribute("direction")
871
arg.setAttribute("name", "value")
872
elif cn.getAttribute("name") == "GetAll":
873
for arg in cn.getElementsByTagName("arg"):
874
if (arg.getAttribute("direction")
876
arg.setAttribute("name", "props")
877
xmlstring = document.toxml("utf-8")
704
878
document.unlink()
705
879
except (AttributeError, xml.dom.DOMException,
706
xml.parsers.expat.ExpatError), error:
707
logger.error(u"Failed to override Introspection method",
880
xml.parsers.expat.ExpatError) as error:
881
logger.error("Failed to override Introspection method",
886
def datetime_to_dbus (dt, variant_level=0):
887
"""Convert a UTC datetime.datetime() to a D-Bus type."""
889
return dbus.String("", variant_level = variant_level)
890
return dbus.String(dt.isoformat(),
891
variant_level=variant_level)
894
class AlternateDBusNamesMetaclass(DBusObjectWithProperties
896
"""Applied to an empty subclass of a D-Bus object, this metaclass
897
will add additional D-Bus attributes matching a certain pattern.
899
def __new__(mcs, name, bases, attr):
900
# Go through all the base classes which could have D-Bus
901
# methods, signals, or properties in them
902
for base in (b for b in bases
903
if issubclass(b, dbus.service.Object)):
904
# Go though all attributes of the base class
905
for attrname, attribute in inspect.getmembers(base):
906
# Ignore non-D-Bus attributes, and D-Bus attributes
907
# with the wrong interface name
908
if (not hasattr(attribute, "_dbus_interface")
909
or not attribute._dbus_interface
910
.startswith("se.recompile.Mandos")):
912
# Create an alternate D-Bus interface name based on
914
alt_interface = (attribute._dbus_interface
915
.replace("se.recompile.Mandos",
916
"se.bsnet.fukt.Mandos"))
917
# Is this a D-Bus signal?
918
if getattr(attribute, "_dbus_is_signal", False):
919
# Extract the original non-method function by
921
nonmethod_func = (dict(
922
zip(attribute.func_code.co_freevars,
923
attribute.__closure__))["func"]
925
# Create a new, but exactly alike, function
926
# object, and decorate it to be a new D-Bus signal
927
# with the alternate D-Bus interface name
928
new_function = (dbus.service.signal
930
attribute._dbus_signature)
932
nonmethod_func.func_code,
933
nonmethod_func.func_globals,
934
nonmethod_func.func_name,
935
nonmethod_func.func_defaults,
936
nonmethod_func.func_closure)))
937
# Define a creator of a function to call both the
938
# old and new functions, so both the old and new
939
# signals gets sent when the function is called
940
def fixscope(func1, func2):
941
"""This function is a scope container to pass
942
func1 and func2 to the "call_both" function
943
outside of its arguments"""
944
def call_both(*args, **kwargs):
945
"""This function will emit two D-Bus
946
signals by calling func1 and func2"""
947
func1(*args, **kwargs)
948
func2(*args, **kwargs)
950
# Create the "call_both" function and add it to
952
attr[attrname] = fixscope(attribute,
954
# Is this a D-Bus method?
955
elif getattr(attribute, "_dbus_is_method", False):
956
# Create a new, but exactly alike, function
957
# object. Decorate it to be a new D-Bus method
958
# with the alternate D-Bus interface name. Add it
960
attr[attrname] = (dbus.service.method
962
attribute._dbus_in_signature,
963
attribute._dbus_out_signature)
965
(attribute.func_code,
966
attribute.func_globals,
968
attribute.func_defaults,
969
attribute.func_closure)))
970
# Is this a D-Bus property?
971
elif getattr(attribute, "_dbus_is_property", False):
972
# Create a new, but exactly alike, function
973
# object, and decorate it to be a new D-Bus
974
# property with the alternate D-Bus interface
975
# name. Add it to the class.
976
attr[attrname] = (dbus_service_property
978
attribute._dbus_signature,
979
attribute._dbus_access,
981
._dbus_get_args_options
984
(attribute.func_code,
985
attribute.func_globals,
987
attribute.func_defaults,
988
attribute.func_closure)))
989
return type.__new__(mcs, name, bases, attr)
712
992
class ClientDBus(Client, DBusObjectWithProperties):
713
993
"""A Client class using D-Bus
720
1000
runtime_expansions = (Client.runtime_expansions
721
+ (u"dbus_object_path",))
1001
+ ("dbus_object_path",))
723
1003
# dbus.service.Object doesn't use super(), so we can't either.
725
1005
def __init__(self, bus = None, *args, **kwargs):
1007
Client.__init__(self, *args, **kwargs)
726
1009
self._approvals_pending = 0
728
Client.__init__(self, *args, **kwargs)
729
1010
# Only now, when this client is initialized, can it show up on
731
1012
client_object_name = unicode(self.name).translate(
732
{ord(u"."): ord(u"_"),
733
ord(u"-"): ord(u"_")})
1013
{ord("."): ord("_"),
1014
ord("-"): ord("_")})
734
1015
self.dbus_object_path = (dbus.ObjectPath
735
(u"/clients/" + client_object_name))
1016
("/clients/" + client_object_name))
736
1017
DBusObjectWithProperties.__init__(self, self.bus,
737
1018
self.dbus_object_path)
739
def _get_approvals_pending(self):
740
return self._approvals_pending
741
def _set_approvals_pending(self, value):
742
old_value = self._approvals_pending
743
self._approvals_pending = value
745
if (hasattr(self, "dbus_object_path")
746
and bval is not bool(old_value)):
747
dbus_bool = dbus.Boolean(bval, variant_level=1)
748
self.PropertyChanged(dbus.String(u"ApprovalPending"),
751
approvals_pending = property(_get_approvals_pending,
752
_set_approvals_pending)
753
del _get_approvals_pending, _set_approvals_pending
756
def _datetime_to_dbus(dt, variant_level=0):
757
"""Convert a UTC datetime.datetime() to a D-Bus type."""
758
return dbus.String(dt.isoformat(),
759
variant_level=variant_level)
762
oldstate = getattr(self, u"enabled", False)
763
r = Client.enable(self)
764
if oldstate != self.enabled:
766
self.PropertyChanged(dbus.String(u"Enabled"),
767
dbus.Boolean(True, variant_level=1))
768
self.PropertyChanged(
769
dbus.String(u"LastEnabled"),
770
self._datetime_to_dbus(self.last_enabled,
774
def disable(self, quiet = False):
775
oldstate = getattr(self, u"enabled", False)
776
r = Client.disable(self, quiet=quiet)
777
if not quiet and oldstate != self.enabled:
779
self.PropertyChanged(dbus.String(u"Enabled"),
780
dbus.Boolean(False, variant_level=1))
1020
def notifychangeproperty(transform_func,
1021
dbus_name, type_func=lambda x: x,
1023
""" Modify a variable so that it's a property which announces
1024
its changes to DBus.
1026
transform_fun: Function that takes a value and a variant_level
1027
and transforms it to a D-Bus type.
1028
dbus_name: D-Bus name of the variable
1029
type_func: Function that transform the value before sending it
1030
to the D-Bus. Default: no transform
1031
variant_level: D-Bus variant level. Default: 1
1033
attrname = "_{0}".format(dbus_name)
1034
def setter(self, value):
1035
if hasattr(self, "dbus_object_path"):
1036
if (not hasattr(self, attrname) or
1037
type_func(getattr(self, attrname, None))
1038
!= type_func(value)):
1039
dbus_value = transform_func(type_func(value),
1042
self.PropertyChanged(dbus.String(dbus_name),
1044
setattr(self, attrname, value)
1046
return property(lambda self: getattr(self, attrname), setter)
1049
expires = notifychangeproperty(datetime_to_dbus, "Expires")
1050
approvals_pending = notifychangeproperty(dbus.Boolean,
1053
enabled = notifychangeproperty(dbus.Boolean, "Enabled")
1054
last_enabled = notifychangeproperty(datetime_to_dbus,
1056
checker = notifychangeproperty(dbus.Boolean, "CheckerRunning",
1057
type_func = lambda checker:
1058
checker is not None)
1059
last_checked_ok = notifychangeproperty(datetime_to_dbus,
1061
last_approval_request = notifychangeproperty(
1062
datetime_to_dbus, "LastApprovalRequest")
1063
approved_by_default = notifychangeproperty(dbus.Boolean,
1064
"ApprovedByDefault")
1065
approval_delay = notifychangeproperty(dbus.UInt16,
1068
_timedelta_to_milliseconds)
1069
approval_duration = notifychangeproperty(
1070
dbus.UInt16, "ApprovalDuration",
1071
type_func = _timedelta_to_milliseconds)
1072
host = notifychangeproperty(dbus.String, "Host")
1073
timeout = notifychangeproperty(dbus.UInt16, "Timeout",
1075
_timedelta_to_milliseconds)
1076
extended_timeout = notifychangeproperty(
1077
dbus.UInt16, "ExtendedTimeout",
1078
type_func = _timedelta_to_milliseconds)
1079
interval = notifychangeproperty(dbus.UInt16,
1082
_timedelta_to_milliseconds)
1083
checker_command = notifychangeproperty(dbus.String, "Checker")
1085
del notifychangeproperty
783
1087
def __del__(self, *args, **kwargs):
785
1089
self.remove_from_connection()
786
1090
except LookupError:
788
if hasattr(DBusObjectWithProperties, u"__del__"):
1092
if hasattr(DBusObjectWithProperties, "__del__"):
789
1093
DBusObjectWithProperties.__del__(self, *args, **kwargs)
790
1094
Client.__del__(self, *args, **kwargs)
951
1230
# ApprovalPending - property
952
@dbus_service_property(_interface, signature=u"b", access=u"read")
1231
@dbus_service_property(_interface, signature="b", access="read")
953
1232
def ApprovalPending_dbus_property(self):
954
1233
return dbus.Boolean(bool(self.approvals_pending))
956
1235
# ApprovedByDefault - property
957
@dbus_service_property(_interface, signature=u"b",
1236
@dbus_service_property(_interface, signature="b",
959
1238
def ApprovedByDefault_dbus_property(self, value=None):
960
1239
if value is None: # get
961
1240
return dbus.Boolean(self.approved_by_default)
962
1241
self.approved_by_default = bool(value)
964
self.PropertyChanged(dbus.String(u"ApprovedByDefault"),
965
dbus.Boolean(value, variant_level=1))
967
1243
# ApprovalDelay - property
968
@dbus_service_property(_interface, signature=u"t",
1244
@dbus_service_property(_interface, signature="t",
970
1246
def ApprovalDelay_dbus_property(self, value=None):
971
1247
if value is None: # get
972
1248
return dbus.UInt64(self.approval_delay_milliseconds())
973
1249
self.approval_delay = datetime.timedelta(0, 0, 0, value)
975
self.PropertyChanged(dbus.String(u"ApprovalDelay"),
976
dbus.UInt64(value, variant_level=1))
978
1251
# ApprovalDuration - property
979
@dbus_service_property(_interface, signature=u"t",
1252
@dbus_service_property(_interface, signature="t",
981
1254
def ApprovalDuration_dbus_property(self, value=None):
982
1255
if value is None: # get
983
return dbus.UInt64(self._timedelta_to_milliseconds(
1256
return dbus.UInt64(_timedelta_to_milliseconds(
984
1257
self.approval_duration))
985
1258
self.approval_duration = datetime.timedelta(0, 0, 0, value)
987
self.PropertyChanged(dbus.String(u"ApprovalDuration"),
988
dbus.UInt64(value, variant_level=1))
990
1260
# Name - property
991
@dbus_service_property(_interface, signature=u"s", access=u"read")
1261
@dbus_service_property(_interface, signature="s", access="read")
992
1262
def Name_dbus_property(self):
993
1263
return dbus.String(self.name)
995
1265
# Fingerprint - property
996
@dbus_service_property(_interface, signature=u"s", access=u"read")
1266
@dbus_service_property(_interface, signature="s", access="read")
997
1267
def Fingerprint_dbus_property(self):
998
1268
return dbus.String(self.fingerprint)
1000
1270
# Host - property
1001
@dbus_service_property(_interface, signature=u"s",
1002
access=u"readwrite")
1271
@dbus_service_property(_interface, signature="s",
1003
1273
def Host_dbus_property(self, value=None):
1004
1274
if value is None: # get
1005
1275
return dbus.String(self.host)
1006
1276
self.host = value
1008
self.PropertyChanged(dbus.String(u"Host"),
1009
dbus.String(value, variant_level=1))
1011
1278
# Created - property
1012
@dbus_service_property(_interface, signature=u"s", access=u"read")
1279
@dbus_service_property(_interface, signature="s", access="read")
1013
1280
def Created_dbus_property(self):
1014
return dbus.String(self._datetime_to_dbus(self.created))
1281
return dbus.String(datetime_to_dbus(self.created))
1016
1283
# LastEnabled - property
1017
@dbus_service_property(_interface, signature=u"s", access=u"read")
1284
@dbus_service_property(_interface, signature="s", access="read")
1018
1285
def LastEnabled_dbus_property(self):
1019
if self.last_enabled is None:
1020
return dbus.String(u"")
1021
return dbus.String(self._datetime_to_dbus(self.last_enabled))
1286
return datetime_to_dbus(self.last_enabled)
1023
1288
# Enabled - property
1024
@dbus_service_property(_interface, signature=u"b",
1025
access=u"readwrite")
1289
@dbus_service_property(_interface, signature="b",
1026
1291
def Enabled_dbus_property(self, value=None):
1027
1292
if value is None: # get
1028
1293
return dbus.Boolean(self.enabled)
1034
1299
# LastCheckedOK - property
1035
@dbus_service_property(_interface, signature=u"s",
1036
access=u"readwrite")
1300
@dbus_service_property(_interface, signature="s",
1037
1302
def LastCheckedOK_dbus_property(self, value=None):
1038
1303
if value is not None:
1039
1304
self.checked_ok()
1041
if self.last_checked_ok is None:
1042
return dbus.String(u"")
1043
return dbus.String(self._datetime_to_dbus(self
1306
return datetime_to_dbus(self.last_checked_ok)
1308
# Expires - property
1309
@dbus_service_property(_interface, signature="s", access="read")
1310
def Expires_dbus_property(self):
1311
return datetime_to_dbus(self.expires)
1046
1313
# LastApprovalRequest - property
1047
@dbus_service_property(_interface, signature=u"s", access=u"read")
1314
@dbus_service_property(_interface, signature="s", access="read")
1048
1315
def LastApprovalRequest_dbus_property(self):
1049
if self.last_approval_request is None:
1050
return dbus.String(u"")
1051
return dbus.String(self.
1052
_datetime_to_dbus(self
1053
.last_approval_request))
1316
return datetime_to_dbus(self.last_approval_request)
1055
1318
# Timeout - property
1056
@dbus_service_property(_interface, signature=u"t",
1057
access=u"readwrite")
1319
@dbus_service_property(_interface, signature="t",
1058
1321
def Timeout_dbus_property(self, value=None):
1059
1322
if value is None: # get
1060
1323
return dbus.UInt64(self.timeout_milliseconds())
1061
1324
self.timeout = datetime.timedelta(0, 0, 0, value)
1063
self.PropertyChanged(dbus.String(u"Timeout"),
1064
dbus.UInt64(value, variant_level=1))
1065
if getattr(self, u"disable_initiator_tag", None) is None:
1325
if getattr(self, "disable_initiator_tag", None) is None:
1067
1327
# Reschedule timeout
1068
1328
gobject.source_remove(self.disable_initiator_tag)
1069
1329
self.disable_initiator_tag = None
1070
time_to_die = (self.
1071
_timedelta_to_milliseconds((self
1331
time_to_die = _timedelta_to_milliseconds((self
1076
1336
if time_to_die <= 0:
1077
1337
# The timeout has passed
1340
self.expires = (datetime.datetime.utcnow()
1341
+ datetime.timedelta(milliseconds =
1080
1343
self.disable_initiator_tag = (gobject.timeout_add
1081
1344
(time_to_die, self.disable))
1346
# ExtendedTimeout - property
1347
@dbus_service_property(_interface, signature="t",
1349
def ExtendedTimeout_dbus_property(self, value=None):
1350
if value is None: # get
1351
return dbus.UInt64(self.extended_timeout_milliseconds())
1352
self.extended_timeout = datetime.timedelta(0, 0, 0, value)
1083
1354
# Interval - property
1084
@dbus_service_property(_interface, signature=u"t",
1085
access=u"readwrite")
1355
@dbus_service_property(_interface, signature="t",
1086
1357
def Interval_dbus_property(self, value=None):
1087
1358
if value is None: # get
1088
1359
return dbus.UInt64(self.interval_milliseconds())
1089
1360
self.interval = datetime.timedelta(0, 0, 0, value)
1091
self.PropertyChanged(dbus.String(u"Interval"),
1092
dbus.UInt64(value, variant_level=1))
1093
if getattr(self, u"checker_initiator_tag", None) is None:
1361
if getattr(self, "checker_initiator_tag", None) is None:
1095
1363
# Reschedule checker run
1096
1364
gobject.source_remove(self.checker_initiator_tag)
1097
1365
self.checker_initiator_tag = (gobject.timeout_add
1098
1366
(value, self.start_checker))
1099
1367
self.start_checker() # Start one now, too
1101
1369
# Checker - property
1102
@dbus_service_property(_interface, signature=u"s",
1103
access=u"readwrite")
1370
@dbus_service_property(_interface, signature="s",
1104
1372
def Checker_dbus_property(self, value=None):
1105
1373
if value is None: # get
1106
1374
return dbus.String(self.checker_command)
1107
1375
self.checker_command = value
1109
self.PropertyChanged(dbus.String(u"Checker"),
1110
dbus.String(self.checker_command,
1113
1377
# CheckerRunning - property
1114
@dbus_service_property(_interface, signature=u"b",
1115
access=u"readwrite")
1378
@dbus_service_property(_interface, signature="b",
1116
1380
def CheckerRunning_dbus_property(self, value=None):
1117
1381
if value is None: # get
1118
1382
return dbus.Boolean(self.checker is not None)
1170
1438
def handle(self):
1171
1439
with contextlib.closing(self.server.child_pipe) as child_pipe:
1172
logger.info(u"TCP connection from: %s",
1440
logger.info("TCP connection from: %s",
1173
1441
unicode(self.client_address))
1174
logger.debug(u"Pipe FD: %d",
1442
logger.debug("Pipe FD: %d",
1175
1443
self.server.child_pipe.fileno())
1177
1445
session = (gnutls.connection
1178
1446
.ClientSession(self.request,
1179
1447
gnutls.connection
1180
1448
.X509Credentials()))
1182
1450
# Note: gnutls.connection.X509Credentials is really a
1183
1451
# generic GnuTLS certificate credentials object so long as
1184
1452
# no X.509 keys are added to it. Therefore, we can use it
1185
1453
# here despite using OpenPGP certificates.
1187
#priority = u':'.join((u"NONE", u"+VERS-TLS1.1",
1188
# u"+AES-256-CBC", u"+SHA1",
1189
# u"+COMP-NULL", u"+CTYPE-OPENPGP",
1455
#priority = ':'.join(("NONE", "+VERS-TLS1.1",
1456
# "+AES-256-CBC", "+SHA1",
1457
# "+COMP-NULL", "+CTYPE-OPENPGP",
1191
1459
# Use a fallback default, since this MUST be set.
1192
1460
priority = self.server.gnutls_priority
1193
1461
if priority is None:
1194
priority = u"NORMAL"
1195
1463
(gnutls.library.functions
1196
1464
.gnutls_priority_set_direct(session._c_object,
1197
1465
priority, None))
1199
1467
# Start communication using the Mandos protocol
1200
1468
# Get protocol number
1201
1469
line = self.request.makefile().readline()
1202
logger.debug(u"Protocol version: %r", line)
1470
logger.debug("Protocol version: %r", line)
1204
1472
if int(line.strip().split()[0]) > 1:
1205
1473
raise RuntimeError
1206
except (ValueError, IndexError, RuntimeError), error:
1207
logger.error(u"Unknown protocol version: %s", error)
1474
except (ValueError, IndexError, RuntimeError) as error:
1475
logger.error("Unknown protocol version: %s", error)
1210
1478
# Start GnuTLS connection
1212
1480
session.handshake()
1213
except gnutls.errors.GNUTLSError, error:
1214
logger.warning(u"Handshake failed: %s", error)
1481
except gnutls.errors.GNUTLSError as error:
1482
logger.warning("Handshake failed: %s", error)
1215
1483
# Do not run session.bye() here: the session is not
1216
1484
# established. Just abandon the request.
1218
logger.debug(u"Handshake succeeded")
1486
logger.debug("Handshake succeeded")
1220
1488
approval_required = False
1223
1491
fpr = self.fingerprint(self.peer_certificate
1225
except (TypeError, gnutls.errors.GNUTLSError), error:
1226
logger.warning(u"Bad certificate: %s", error)
1494
gnutls.errors.GNUTLSError) as error:
1495
logger.warning("Bad certificate: %s", error)
1228
logger.debug(u"Fingerprint: %s", fpr)
1497
logger.debug("Fingerprint: %s", fpr)
1498
if self.server.use_dbus:
1500
client.NewRequest(str(self.client_address))
1231
1503
client = ProxyClient(child_pipe, fpr,
1232
1504
self.client_address)
1499
1777
def server_activate(self):
1500
1778
if self.enabled:
1501
1779
return socketserver.TCPServer.server_activate(self)
1502
1781
def enable(self):
1503
1782
self.enabled = True
1504
def add_pipe(self, parent_pipe):
1784
def add_pipe(self, parent_pipe, proc):
1505
1785
# Call "handle_ipc" for both data and EOF events
1506
1786
gobject.io_add_watch(parent_pipe.fileno(),
1507
1787
gobject.IO_IN | gobject.IO_HUP,
1508
1788
functools.partial(self.handle_ipc,
1509
parent_pipe = parent_pipe))
1511
1793
def handle_ipc(self, source, condition, parent_pipe=None,
1512
client_object=None):
1794
proc = None, client_object=None):
1513
1795
condition_names = {
1514
gobject.IO_IN: u"IN", # There is data to read.
1515
gobject.IO_OUT: u"OUT", # Data can be written (without
1796
gobject.IO_IN: "IN", # There is data to read.
1797
gobject.IO_OUT: "OUT", # Data can be written (without
1517
gobject.IO_PRI: u"PRI", # There is urgent data to read.
1518
gobject.IO_ERR: u"ERR", # Error condition.
1519
gobject.IO_HUP: u"HUP" # Hung up (the connection has been
1799
gobject.IO_PRI: "PRI", # There is urgent data to read.
1800
gobject.IO_ERR: "ERR", # Error condition.
1801
gobject.IO_HUP: "HUP" # Hung up (the connection has been
1520
1802
# broken, usually for pipes and
1536
1820
fpr = request[1]
1537
1821
address = request[2]
1539
for c in self.clients:
1823
for c in self.clients.itervalues():
1540
1824
if c.fingerprint == fpr:
1544
logger.warning(u"Client not found for fingerprint: %s, ad"
1545
u"dress: %s", fpr, address)
1828
logger.info("Client not found for fingerprint: %s, ad"
1829
"dress: %s", fpr, address)
1546
1830
if self.use_dbus:
1547
1831
# Emit D-Bus signal
1548
mandos_dbus_service.ClientNotFound(fpr, address[0])
1832
mandos_dbus_service.ClientNotFound(fpr,
1549
1834
parent_pipe.send(False)
1552
1837
gobject.io_add_watch(parent_pipe.fileno(),
1553
1838
gobject.IO_IN | gobject.IO_HUP,
1554
1839
functools.partial(self.handle_ipc,
1555
parent_pipe = parent_pipe,
1556
client_object = client))
1557
1845
parent_pipe.send(True)
1558
# remove the old hook in favor of the new above hook on same fileno
1846
# remove the old hook in favor of the new above hook on
1560
1849
if command == 'funcall':
1561
1850
funcname = request[1]
1562
1851
args = request[2]
1563
1852
kwargs = request[3]
1565
parent_pipe.send(('data', getattr(client_object, funcname)(*args, **kwargs)))
1854
parent_pipe.send(('data', getattr(client_object,
1567
1858
if command == 'getattr':
1568
1859
attrname = request[1]
1569
1860
if callable(client_object.__getattribute__(attrname)):
1570
1861
parent_pipe.send(('function',))
1572
parent_pipe.send(('data', client_object.__getattribute__(attrname)))
1863
parent_pipe.send(('data', client_object
1864
.__getattribute__(attrname)))
1574
1866
if command == 'setattr':
1575
1867
attrname = request[1]
1576
1868
value = request[2]
1577
1869
setattr(client_object, attrname, value)
1582
1874
def string_to_delta(interval):
1583
1875
"""Parse a string and return a datetime.timedelta
1585
>>> string_to_delta(u'7d')
1877
>>> string_to_delta('7d')
1586
1878
datetime.timedelta(7)
1587
>>> string_to_delta(u'60s')
1879
>>> string_to_delta('60s')
1588
1880
datetime.timedelta(0, 60)
1589
>>> string_to_delta(u'60m')
1881
>>> string_to_delta('60m')
1590
1882
datetime.timedelta(0, 3600)
1591
>>> string_to_delta(u'24h')
1883
>>> string_to_delta('24h')
1592
1884
datetime.timedelta(1)
1593
>>> string_to_delta(u'1w')
1885
>>> string_to_delta('1w')
1594
1886
datetime.timedelta(7)
1595
>>> string_to_delta(u'5m 30s')
1887
>>> string_to_delta('5m 30s')
1596
1888
datetime.timedelta(0, 330)
1598
1890
timevalue = datetime.timedelta(0)
1601
1893
suffix = unicode(s[-1])
1602
1894
value = int(s[:-1])
1604
1896
delta = datetime.timedelta(value)
1605
elif suffix == u"s":
1606
1898
delta = datetime.timedelta(0, value)
1607
elif suffix == u"m":
1608
1900
delta = datetime.timedelta(0, 0, 0, 0, value)
1609
elif suffix == u"h":
1610
1902
delta = datetime.timedelta(0, 0, 0, 0, 0, value)
1611
elif suffix == u"w":
1612
1904
delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
1614
raise ValueError(u"Unknown suffix %r" % suffix)
1615
except (ValueError, IndexError), e:
1616
raise ValueError(e.message)
1906
raise ValueError("Unknown suffix %r" % suffix)
1907
except (ValueError, IndexError) as e:
1908
raise ValueError(*(e.args))
1617
1909
timevalue += delta
1618
1910
return timevalue
1621
def if_nametoindex(interface):
1622
"""Call the C function if_nametoindex(), or equivalent
1624
Note: This function cannot accept a unicode string."""
1625
global if_nametoindex
1627
if_nametoindex = (ctypes.cdll.LoadLibrary
1628
(ctypes.util.find_library(u"c"))
1630
except (OSError, AttributeError):
1631
logger.warning(u"Doing if_nametoindex the hard way")
1632
def if_nametoindex(interface):
1633
"Get an interface index the hard way, i.e. using fcntl()"
1634
SIOCGIFINDEX = 0x8933 # From /usr/include/linux/sockios.h
1635
with contextlib.closing(socket.socket()) as s:
1636
ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
1637
struct.pack(str(u"16s16x"),
1639
interface_index = struct.unpack(str(u"I"),
1641
return interface_index
1642
return if_nametoindex(interface)
1645
1913
def daemon(nochdir = False, noclose = False):
1646
1914
"""See daemon(3). Standard BSD Unix function.
1672
1940
##################################################################
1673
1941
# Parsing of options, both command line and config file
1675
parser = optparse.OptionParser(version = "%%prog %s" % version)
1676
parser.add_option("-i", u"--interface", type=u"string",
1677
metavar="IF", help=u"Bind to interface IF")
1678
parser.add_option("-a", u"--address", type=u"string",
1679
help=u"Address to listen for requests on")
1680
parser.add_option("-p", u"--port", type=u"int",
1681
help=u"Port number to receive requests on")
1682
parser.add_option("--check", action=u"store_true",
1683
help=u"Run self-test")
1684
parser.add_option("--debug", action=u"store_true",
1685
help=u"Debug mode; run in foreground and log to"
1687
parser.add_option("--debuglevel", type=u"string", metavar="LEVEL",
1688
help=u"Debug level for stdout output")
1689
parser.add_option("--priority", type=u"string", help=u"GnuTLS"
1690
u" priority string (see GnuTLS documentation)")
1691
parser.add_option("--servicename", type=u"string",
1692
metavar=u"NAME", help=u"Zeroconf service name")
1693
parser.add_option("--configdir", type=u"string",
1694
default=u"/etc/mandos", metavar=u"DIR",
1695
help=u"Directory to search for configuration"
1697
parser.add_option("--no-dbus", action=u"store_false",
1698
dest=u"use_dbus", help=u"Do not provide D-Bus"
1699
u" system bus interface")
1700
parser.add_option("--no-ipv6", action=u"store_false",
1701
dest=u"use_ipv6", help=u"Do not use IPv6")
1702
options = parser.parse_args()[0]
1943
parser = argparse.ArgumentParser()
1944
parser.add_argument("-v", "--version", action="version",
1945
version = "%%(prog)s %s" % version,
1946
help="show version number and exit")
1947
parser.add_argument("-i", "--interface", metavar="IF",
1948
help="Bind to interface IF")
1949
parser.add_argument("-a", "--address",
1950
help="Address to listen for requests on")
1951
parser.add_argument("-p", "--port", type=int,
1952
help="Port number to receive requests on")
1953
parser.add_argument("--check", action="store_true",
1954
help="Run self-test")
1955
parser.add_argument("--debug", action="store_true",
1956
help="Debug mode; run in foreground and log"
1958
parser.add_argument("--debuglevel", metavar="LEVEL",
1959
help="Debug level for stdout output")
1960
parser.add_argument("--priority", help="GnuTLS"
1961
" priority string (see GnuTLS documentation)")
1962
parser.add_argument("--servicename",
1963
metavar="NAME", help="Zeroconf service name")
1964
parser.add_argument("--configdir",
1965
default="/etc/mandos", metavar="DIR",
1966
help="Directory to search for configuration"
1968
parser.add_argument("--no-dbus", action="store_false",
1969
dest="use_dbus", help="Do not provide D-Bus"
1970
" system bus interface")
1971
parser.add_argument("--no-ipv6", action="store_false",
1972
dest="use_ipv6", help="Do not use IPv6")
1973
parser.add_argument("--no-restore", action="store_false",
1974
dest="restore", help="Do not restore stored"
1975
" state", default=True)
1977
options = parser.parse_args()
1704
1979
if options.check:
1709
1984
# Default values for config file for server-global settings
1710
server_defaults = { u"interface": u"",
1715
u"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
1716
u"servicename": u"Mandos",
1717
u"use_dbus": u"True",
1718
u"use_ipv6": u"True",
1985
server_defaults = { "interface": "",
1990
"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
1991
"servicename": "Mandos",
1722
1997
# Parse config file for server-global settings
1723
1998
server_config = configparser.SafeConfigParser(server_defaults)
1724
1999
del server_defaults
1725
2000
server_config.read(os.path.join(options.configdir,
1727
2002
# Convert the SafeConfigParser object to a dict
1728
2003
server_settings = server_config.defaults()
1729
2004
# Use the appropriate methods on the non-string config options
1730
for option in (u"debug", u"use_dbus", u"use_ipv6"):
1731
server_settings[option] = server_config.getboolean(u"DEFAULT",
2005
for option in ("debug", "use_dbus", "use_ipv6"):
2006
server_settings[option] = server_config.getboolean("DEFAULT",
1733
2008
if server_settings["port"]:
1734
server_settings["port"] = server_config.getint(u"DEFAULT",
2009
server_settings["port"] = server_config.getint("DEFAULT",
1736
2011
del server_config
1738
2013
# Override the settings from the config file with command line
1739
2014
# options, if set.
1740
for option in (u"interface", u"address", u"port", u"debug",
1741
u"priority", u"servicename", u"configdir",
1742
u"use_dbus", u"use_ipv6", u"debuglevel"):
2015
for option in ("interface", "address", "port", "debug",
2016
"priority", "servicename", "configdir",
2017
"use_dbus", "use_ipv6", "debuglevel", "restore"):
1743
2018
value = getattr(options, option)
1744
2019
if value is not None:
1745
2020
server_settings[option] = value
1753
2028
##################################################################
1755
2030
# For convenience
1756
debug = server_settings[u"debug"]
1757
debuglevel = server_settings[u"debuglevel"]
1758
use_dbus = server_settings[u"use_dbus"]
1759
use_ipv6 = server_settings[u"use_ipv6"]
1761
if server_settings[u"servicename"] != u"Mandos":
2031
debug = server_settings["debug"]
2032
debuglevel = server_settings["debuglevel"]
2033
use_dbus = server_settings["use_dbus"]
2034
use_ipv6 = server_settings["use_ipv6"]
2037
initlogger(logging.DEBUG)
2042
level = getattr(logging, debuglevel.upper())
2045
if server_settings["servicename"] != "Mandos":
1762
2046
syslogger.setFormatter(logging.Formatter
1763
(u'Mandos (%s) [%%(process)d]:'
1764
u' %%(levelname)s: %%(message)s'
1765
% server_settings[u"servicename"]))
2047
('Mandos (%s) [%%(process)d]:'
2048
' %%(levelname)s: %%(message)s'
2049
% server_settings["servicename"]))
1767
2051
# Parse config file with clients
1768
client_defaults = { u"timeout": u"1h",
1770
u"checker": u"fping -q -- %%(host)s",
1772
u"approval_delay": u"0s",
1773
u"approval_duration": u"1s",
2052
client_defaults = { "timeout": "5m",
2053
"extended_timeout": "15m",
2055
"checker": "fping -q -- %%(host)s",
2057
"approval_delay": "0s",
2058
"approval_duration": "1s",
1775
2060
client_config = configparser.SafeConfigParser(client_defaults)
1776
client_config.read(os.path.join(server_settings[u"configdir"],
2061
client_config.read(os.path.join(server_settings["configdir"],
1779
2064
global mandos_dbus_service
1780
2065
mandos_dbus_service = None
1782
tcp_server = MandosServer((server_settings[u"address"],
1783
server_settings[u"port"]),
2067
tcp_server = MandosServer((server_settings["address"],
2068
server_settings["port"]),
1785
interface=(server_settings[u"interface"]
2070
interface=(server_settings["interface"]
1787
2072
use_ipv6=use_ipv6,
1788
2073
gnutls_priority=
1789
server_settings[u"priority"],
2074
server_settings["priority"],
1790
2075
use_dbus=use_dbus)
1792
pidfilename = u"/var/run/mandos.pid"
2077
pidfilename = "/var/run/mandos.pid"
1794
pidfile = open(pidfilename, u"w")
2079
pidfile = open(pidfilename, "w")
1795
2080
except IOError:
1796
logger.error(u"Could not open file %r", pidfilename)
2081
logger.error("Could not open file %r", pidfilename)
1799
uid = pwd.getpwnam(u"_mandos").pw_uid
1800
gid = pwd.getpwnam(u"_mandos").pw_gid
2084
uid = pwd.getpwnam("_mandos").pw_uid
2085
gid = pwd.getpwnam("_mandos").pw_gid
1801
2086
except KeyError:
1803
uid = pwd.getpwnam(u"mandos").pw_uid
1804
gid = pwd.getpwnam(u"mandos").pw_gid
2088
uid = pwd.getpwnam("mandos").pw_uid
2089
gid = pwd.getpwnam("mandos").pw_gid
1805
2090
except KeyError:
1807
uid = pwd.getpwnam(u"nobody").pw_uid
1808
gid = pwd.getpwnam(u"nobody").pw_gid
2092
uid = pwd.getpwnam("nobody").pw_uid
2093
gid = pwd.getpwnam("nobody").pw_gid
1809
2094
except KeyError:
1815
except OSError, error:
2100
except OSError as error:
1816
2101
if error[0] != errno.EPERM:
1819
if not debug and not debuglevel:
1820
syslogger.setLevel(logging.WARNING)
1821
console.setLevel(logging.WARNING)
1823
level = getattr(logging, debuglevel.upper())
1824
syslogger.setLevel(level)
1825
console.setLevel(level)
1828
2105
# Enable all possible GnuTLS debugging
1856
2137
# End of Avahi example code
1859
bus_name = dbus.service.BusName(u"se.bsnet.fukt.Mandos",
2140
bus_name = dbus.service.BusName("se.recompile.Mandos",
1860
2141
bus, do_not_queue=True)
1861
except dbus.exceptions.NameExistsException, e:
1862
logger.error(unicode(e) + u", disabling D-Bus")
2142
old_bus_name = (dbus.service.BusName
2143
("se.bsnet.fukt.Mandos", bus,
2145
except dbus.exceptions.NameExistsException as e:
2146
logger.error(unicode(e) + ", disabling D-Bus")
1863
2147
use_dbus = False
1864
server_settings[u"use_dbus"] = False
2148
server_settings["use_dbus"] = False
1865
2149
tcp_server.use_dbus = False
1866
2150
protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
1867
service = AvahiService(name = server_settings[u"servicename"],
1868
servicetype = u"_mandos._tcp",
1869
protocol = protocol, bus = bus)
2151
service = AvahiServiceToSyslog(name =
2152
server_settings["servicename"],
2153
servicetype = "_mandos._tcp",
2154
protocol = protocol, bus = bus)
1870
2155
if server_settings["interface"]:
1871
2156
service.interface = (if_nametoindex
1872
(str(server_settings[u"interface"])))
1875
# Close all input and output, do double fork, etc.
2157
(str(server_settings["interface"])))
1878
2159
global multiprocessing_manager
1879
2160
multiprocessing_manager = multiprocessing.Manager()
1881
2162
client_class = Client
1883
client_class = functools.partial(ClientDBus, bus = bus)
1884
def client_config_items(config, section):
1885
special_settings = {
1886
"approved_by_default":
1887
lambda: config.getboolean(section,
1888
"approved_by_default"),
1890
for name, value in config.items(section):
2164
client_class = functools.partial(ClientDBusTransitional,
2167
special_settings = {
2168
# Some settings need to be accessd by special methods;
2169
# booleans need .getboolean(), etc. Here is a list of them:
2170
"approved_by_default":
2172
client_config.getboolean(section, "approved_by_default"),
2174
# Construct a new dict of client settings of this form:
2175
# { client_name: {setting_name: value, ...}, ...}
2176
# with exceptions for any special settings as defined above
2177
client_settings = dict((clientname,
2180
if setting not in special_settings
2181
else special_settings[setting]
2183
for setting, value in
2184
client_config.items(clientname)))
2185
for clientname in client_config.sections())
2187
old_client_settings = {}
2190
# Get client data and settings from last running state.
2191
if server_settings["restore"]:
2193
with open(stored_state_path, "rb") as stored_state:
2194
clients_data, old_client_settings = (pickle.load
2196
os.remove(stored_state_path)
2197
except IOError as e:
2198
logger.warning("Could not load persistent state: {0}"
2200
if e.errno != errno.ENOENT:
2203
with Crypto() as crypt:
2204
for client in clients_data:
2205
client_name = client["name"]
2207
# Decide which value to use after restoring saved state.
2208
# We have three different values: Old config file,
2209
# new config file, and saved state.
2210
# New config value takes precedence if it differs from old
2211
# config value, otherwise use saved state.
2212
for name, value in client_settings[client_name].items():
2214
# For each value in new config, check if it
2215
# differs from the old config value (Except for
2216
# the "secret" attribute)
2217
if (name != "secret" and
2218
value != old_client_settings[client_name]
2220
setattr(client, name, value)
2224
# Clients who has passed its expire date can still be
2225
# enabled if its last checker was sucessful. Clients
2226
# whose checker failed before we stored its state is
2227
# assumed to have failed all checkers during downtime.
2228
if client["enabled"] and client["last_checked_ok"]:
2229
if ((datetime.datetime.utcnow()
2230
- client["last_checked_ok"])
2231
> client["interval"]):
2232
if client["last_checker_status"] != 0:
2233
client["enabled"] = False
2235
client["expires"] = (datetime.datetime
2237
+ client["timeout"])
2239
client["changedstate"] = (multiprocessing_manager
2241
(multiprocessing_manager
2244
new_client = (ClientDBusTransitional.__new__
2245
(ClientDBusTransitional))
2246
tcp_server.clients[client_name] = new_client
2247
new_client.bus = bus
2248
for name, value in client.iteritems():
2249
setattr(new_client, name, value)
2250
client_object_name = unicode(client_name).translate(
2251
{ord("."): ord("_"),
2252
ord("-"): ord("_")})
2253
new_client.dbus_object_path = (dbus.ObjectPath
2255
+ client_object_name))
2256
DBusObjectWithProperties.__init__(new_client,
2261
tcp_server.clients[client_name] = (Client.__new__
2263
for name, value in client.iteritems():
2264
setattr(tcp_server.clients[client_name],
1892
yield (name, special_settings[name]())
1896
tcp_server.clients.update(set(
1897
client_class(name = section,
1898
config= dict(client_config_items(
1899
client_config, section)))
1900
for section in client_config.sections()))
2268
tcp_server.clients[client_name].secret = (
2269
crypt.decrypt(tcp_server.clients[client_name]
2271
client_settings[client_name]
2274
# If decryption fails, we use secret from new settings
2275
tcp_server.clients[client_name].secret = (
2276
client_settings[client_name]["secret"])
2278
# Create/remove clients based on new changes made to config
2279
for clientname in set(old_client_settings) - set(client_settings):
2280
del tcp_server.clients[clientname]
2281
for clientname in set(client_settings) - set(old_client_settings):
2282
tcp_server.clients[clientname] = (client_class(name
1901
2288
if not tcp_server.clients:
1902
logger.warning(u"No clients defined")
2289
logger.warning("No clients defined")
1907
2294
pid = os.getpid()
1908
pidfile.write(str(pid) + "\n")
2295
pidfile.write(str(pid) + "\n".encode("utf-8"))
1910
2297
except IOError:
1911
logger.error(u"Could not write to file %r with PID %d",
2298
logger.error("Could not write to file %r with PID %d",
1912
2299
pidfilename, pid)
1913
2300
except NameError:
1914
2301
# "pidfile" was never created
1924
2311
class MandosDBusService(dbus.service.Object):
1925
2312
"""A D-Bus proxy object"""
1926
2313
def __init__(self):
1927
dbus.service.Object.__init__(self, bus, u"/")
1928
_interface = u"se.bsnet.fukt.Mandos"
2314
dbus.service.Object.__init__(self, bus, "/")
2315
_interface = "se.recompile.Mandos"
1930
@dbus.service.signal(_interface, signature=u"o")
2317
@dbus.service.signal(_interface, signature="o")
1931
2318
def ClientAdded(self, objpath):
1935
@dbus.service.signal(_interface, signature=u"ss")
2322
@dbus.service.signal(_interface, signature="ss")
1936
2323
def ClientNotFound(self, fingerprint, address):
1940
@dbus.service.signal(_interface, signature=u"os")
2327
@dbus.service.signal(_interface, signature="os")
1941
2328
def ClientRemoved(self, objpath, name):
1945
@dbus.service.method(_interface, out_signature=u"ao")
2332
@dbus.service.method(_interface, out_signature="ao")
1946
2333
def GetAllClients(self):
1948
2335
return dbus.Array(c.dbus_object_path
1949
for c in tcp_server.clients)
2337
tcp_server.clients.itervalues())
1951
2339
@dbus.service.method(_interface,
1952
out_signature=u"a{oa{sv}}")
2340
out_signature="a{oa{sv}}")
1953
2341
def GetAllClientsWithProperties(self):
1955
2343
return dbus.Dictionary(
1956
((c.dbus_object_path, c.GetAll(u""))
1957
for c in tcp_server.clients),
1958
signature=u"oa{sv}")
2344
((c.dbus_object_path, c.GetAll(""))
2345
for c in tcp_server.clients.itervalues()),
1960
@dbus.service.method(_interface, in_signature=u"o")
2348
@dbus.service.method(_interface, in_signature="o")
1961
2349
def RemoveClient(self, object_path):
1963
for c in tcp_server.clients:
2351
for c in tcp_server.clients.itervalues():
1964
2352
if c.dbus_object_path == object_path:
1965
tcp_server.clients.remove(c)
2353
del tcp_server.clients[c.name]
1966
2354
c.remove_from_connection()
1967
2355
# Don't signal anything except ClientRemoved
1968
2356
c.disable(quiet=True)
1976
mandos_dbus_service = MandosDBusService()
2364
class MandosDBusServiceTransitional(MandosDBusService):
2365
__metaclass__ = AlternateDBusNamesMetaclass
2366
mandos_dbus_service = MandosDBusServiceTransitional()
1979
2369
"Cleanup function; run on exit"
1980
2370
service.cleanup()
2372
multiprocessing.active_children()
2373
if not (tcp_server.clients or client_settings):
2376
# Store client before exiting. Secrets are encrypted with key
2377
# based on what config file has. If config file is
2378
# removed/edited, old secret will thus be unrecovable.
2380
with Crypto() as crypt:
2381
for client in tcp_server.clients.itervalues():
2382
key = client_settings[client.name]["secret"]
2383
client.encrypted_secret = crypt.encrypt(client.secret,
2387
# A list of attributes that will not be stored when
2389
exclude = set(("bus", "changedstate", "secret"))
2390
for name, typ in (inspect.getmembers
2391
(dbus.service.Object)):
2394
client_dict["encrypted_secret"] = (client
2396
for attr in client.client_structure:
2397
if attr not in exclude:
2398
client_dict[attr] = getattr(client, attr)
2400
clients.append(client_dict)
2401
del client_settings[client.name]["secret"]
2404
with os.fdopen(os.open(stored_state_path,
2405
os.O_CREAT|os.O_WRONLY|os.O_TRUNC,
2406
0600), "wb") as stored_state:
2407
pickle.dump((clients, client_settings), stored_state)
2408
except (IOError, OSError) as e:
2409
logger.warning("Could not save persistent state: {0}"
2411
if e.errno not in (errno.ENOENT, errno.EACCES):
2414
# Delete all clients, and settings from config
1982
2415
while tcp_server.clients:
1983
client = tcp_server.clients.pop()
2416
name, client = tcp_server.clients.popitem()
1985
2418
client.remove_from_connection()
1986
client.disable_hook = None
1987
2419
# Don't signal anything except ClientRemoved
1988
2420
client.disable(quiet=True)
1990
2422
# Emit D-Bus signal
1991
mandos_dbus_service.ClientRemoved(client.dbus_object_path,
2423
mandos_dbus_service.ClientRemoved(client
2426
client_settings.clear()
1994
2428
atexit.register(cleanup)
1996
for client in tcp_server.clients:
2430
for client in tcp_server.clients.itervalues():
1998
2432
# Emit D-Bus signal
1999
2433
mandos_dbus_service.ClientAdded(client.dbus_object_path)
2434
# Need to initiate checking of clients
2436
client.init_checker()
2002
2438
tcp_server.enable()
2003
2439
tcp_server.server_activate()