/mandos/release

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/release

« back to all changes in this revision

Viewing changes to mandos

  • Committer: Teddy Hogeborn
  • Date: 2012-01-01 05:06:20 UTC
  • Revision ID: teddy@recompile.se-20120101050620-mpqzg5l19gppibhf
Tags: version-1.5.0-1
* Makefile (version): Changed to "1.5.0".
* NEWS (Version 1.5.0): New entry.
* debian/changelog (1.5.0-1): - '' -

Show diffs side-by-side

added added

removed removed

Lines of Context:
34
34
from __future__ import (division, absolute_import, print_function,
35
35
                        unicode_literals)
36
36
 
37
 
from future_builtins import *
38
 
 
39
37
import SocketServer as socketserver
40
38
import socket
41
39
import argparse
67
65
import types
68
66
import binascii
69
67
import tempfile
70
 
import itertools
71
68
 
72
69
import dbus
73
70
import dbus.service
88
85
    except ImportError:
89
86
        SO_BINDTODEVICE = None
90
87
 
91
 
version = "1.5.3"
 
88
version = "1.5.0"
92
89
stored_state_file = "clients.pickle"
93
90
 
94
91
logger = logging.getLogger()
178
175
    
179
176
    def encrypt(self, data, password):
180
177
        self.gnupg.passphrase = self.password_encode(password)
181
 
        with open(os.devnull, "w") as devnull:
 
178
        with open(os.devnull) as devnull:
182
179
            try:
183
180
                proc = self.gnupg.run(['--symmetric'],
184
181
                                      create_fhs=['stdin', 'stdout'],
195
192
    
196
193
    def decrypt(self, data, password):
197
194
        self.gnupg.passphrase = self.password_encode(password)
198
 
        with open(os.devnull, "w") as devnull:
 
195
        with open(os.devnull) as devnull:
199
196
            try:
200
197
                proc = self.gnupg.run(['--decrypt'],
201
198
                                      create_fhs=['stdin', 'stdout'],
202
199
                                      attach_fhs={'stderr': devnull})
203
 
                with contextlib.closing(proc.handles['stdin']) as f:
 
200
                with contextlib.closing(proc.handles['stdin'] ) as f:
204
201
                    f.write(data)
205
202
                with contextlib.closing(proc.handles['stdout']) as f:
206
203
                    decrypted_plaintext = f.read()
211
208
        return decrypted_plaintext
212
209
 
213
210
 
 
211
 
214
212
class AvahiError(Exception):
215
213
    def __init__(self, value, *args, **kwargs):
216
214
        self.value = value
245
243
    server: D-Bus Server
246
244
    bus: dbus.SystemBus()
247
245
    """
248
 
    
249
246
    def __init__(self, interface = avahi.IF_UNSPEC, name = None,
250
247
                 servicetype = None, port = None, TXT = None,
251
248
                 domain = "", host = "", max_renames = 32768,
264
261
        self.server = None
265
262
        self.bus = bus
266
263
        self.entry_group_state_changed_match = None
267
 
    
268
264
    def rename(self):
269
265
        """Derived from the Avahi example code"""
270
266
        if self.rename_count >= self.max_renames:
280
276
        try:
281
277
            self.add()
282
278
        except dbus.exceptions.DBusException as error:
283
 
            logger.critical("D-Bus Exception", exc_info=error)
 
279
            logger.critical("DBusException: %s", error)
284
280
            self.cleanup()
285
281
            os._exit(1)
286
282
        self.rename_count += 1
287
 
    
288
283
    def remove(self):
289
284
        """Derived from the Avahi example code"""
290
285
        if self.entry_group_state_changed_match is not None:
292
287
            self.entry_group_state_changed_match = None
293
288
        if self.group is not None:
294
289
            self.group.Reset()
295
 
    
296
290
    def add(self):
297
291
        """Derived from the Avahi example code"""
298
292
        self.remove()
315
309
            dbus.UInt16(self.port),
316
310
            avahi.string_array_to_txt_array(self.TXT))
317
311
        self.group.Commit()
318
 
    
319
312
    def entry_group_state_changed(self, state, error):
320
313
        """Derived from the Avahi example code"""
321
314
        logger.debug("Avahi entry group state change: %i", state)
328
321
        elif state == avahi.ENTRY_GROUP_FAILURE:
329
322
            logger.critical("Avahi: Error in group state changed %s",
330
323
                            unicode(error))
331
 
            raise AvahiGroupError("State changed: {0!s}"
332
 
                                  .format(error))
333
 
    
 
324
            raise AvahiGroupError("State changed: %s"
 
325
                                  % unicode(error))
334
326
    def cleanup(self):
335
327
        """Derived from the Avahi example code"""
336
328
        if self.group is not None:
341
333
                pass
342
334
            self.group = None
343
335
        self.remove()
344
 
    
345
336
    def server_state_changed(self, state, error=None):
346
337
        """Derived from the Avahi example code"""
347
338
        logger.debug("Avahi server state change: %i", state)
366
357
                logger.debug("Unknown state: %r", state)
367
358
            else:
368
359
                logger.debug("Unknown state: %r: %r", state, error)
369
 
    
370
360
    def activate(self):
371
361
        """Derived from the Avahi example code"""
372
362
        if self.server is None:
384
374
        """Add the new name to the syslog messages"""
385
375
        ret = AvahiService.rename(self)
386
376
        syslogger.setFormatter(logging.Formatter
387
 
                               ('Mandos ({0}) [%(process)d]:'
388
 
                                ' %(levelname)s: %(message)s'
389
 
                                .format(self.name)))
 
377
                               ('Mandos (%s) [%%(process)d]:'
 
378
                                ' %%(levelname)s: %%(message)s'
 
379
                                % self.name))
390
380
        return ret
391
381
 
392
382
def timedelta_to_milliseconds(td):
394
384
    return ((td.days * 24 * 60 * 60 * 1000)
395
385
            + (td.seconds * 1000)
396
386
            + (td.microseconds // 1000))
397
 
 
 
387
        
398
388
class Client(object):
399
389
    """A representation of a client host served by this server.
400
390
    
425
415
    last_checked_ok: datetime.datetime(); (UTC) or None
426
416
    last_checker_status: integer between 0 and 255 reflecting exit
427
417
                         status of last checker. -1 reflects crashed
428
 
                         checker, -2 means no checker completed yet.
 
418
                         checker, or None.
429
419
    last_enabled: datetime.datetime(); (UTC) or None
430
420
    name:       string; from the config file, used in log messages and
431
421
                        D-Bus identifiers
432
422
    secret:     bytestring; sent verbatim (over TLS) to client
433
423
    timeout:    datetime.timedelta(); How long from last_checked_ok
434
424
                                      until this client is disabled
435
 
    extended_timeout:   extra long timeout when secret has been sent
 
425
    extended_timeout:   extra long timeout when password has been sent
436
426
    runtime_expansions: Allowed attributes for runtime expansion.
437
427
    expires:    datetime.datetime(); time (UTC) when a client will be
438
428
                disabled, or None
467
457
    
468
458
    def approval_delay_milliseconds(self):
469
459
        return timedelta_to_milliseconds(self.approval_delay)
470
 
    
 
460
 
471
461
    @staticmethod
472
462
    def config_parser(config):
473
463
        """Construct a new dict of client settings of this form:
498
488
                          "rb") as secfile:
499
489
                    client["secret"] = secfile.read()
500
490
            else:
501
 
                raise TypeError("No secret or secfile for section {0}"
502
 
                                .format(section))
 
491
                raise TypeError("No secret or secfile for section %s"
 
492
                                % section)
503
493
            client["timeout"] = string_to_delta(section["timeout"])
504
494
            client["extended_timeout"] = string_to_delta(
505
495
                section["extended_timeout"])
511
501
            client["checker_command"] = section["checker"]
512
502
            client["last_approval_request"] = None
513
503
            client["last_checked_ok"] = None
514
 
            client["last_checker_status"] = -2
 
504
            client["last_checker_status"] = None
515
505
        
516
506
        return settings
517
 
    
 
507
        
 
508
        
518
509
    def __init__(self, settings, name = None):
 
510
        """Note: the 'checker' key in 'config' sets the
 
511
        'checker_command' attribute and *not* the 'checker'
 
512
        attribute."""
519
513
        self.name = name
520
514
        # adding all client settings
521
515
        for setting, value in settings.iteritems():
530
524
        else:
531
525
            self.last_enabled = None
532
526
            self.expires = None
533
 
        
 
527
       
534
528
        logger.debug("Creating client %r", self.name)
535
529
        # Uppercase and remove spaces from fingerprint for later
536
530
        # comparison purposes with return value from the fingerprint()
538
532
        logger.debug("  Fingerprint: %s", self.fingerprint)
539
533
        self.created = settings.get("created",
540
534
                                    datetime.datetime.utcnow())
541
 
        
 
535
 
542
536
        # attributes specific for this server instance
543
537
        self.checker = None
544
538
        self.checker_initiator_tag = None
632
626
            logger.warning("Checker for %(name)s crashed?",
633
627
                           vars(self))
634
628
    
635
 
    def checked_ok(self):
636
 
        """Assert that the client has been seen, alive and well."""
637
 
        self.last_checked_ok = datetime.datetime.utcnow()
638
 
        self.last_checker_status = 0
639
 
        self.bump_timeout()
640
 
    
641
 
    def bump_timeout(self, timeout=None):
642
 
        """Bump up the timeout for this client."""
 
629
    def checked_ok(self, timeout=None):
 
630
        """Bump up the timeout for this client.
 
631
        
 
632
        This should only be called when the client has been seen,
 
633
        alive and well.
 
634
        """
643
635
        if timeout is None:
644
636
            timeout = self.timeout
 
637
        self.last_checked_ok = datetime.datetime.utcnow()
645
638
        if self.disable_initiator_tag is not None:
646
639
            gobject.source_remove(self.disable_initiator_tag)
647
640
        if getattr(self, "enabled", False):
659
652
        If a checker already exists, leave it running and do
660
653
        nothing."""
661
654
        # The reason for not killing a running checker is that if we
662
 
        # did that, and if a checker (for some reason) started running
663
 
        # slowly and taking more than 'interval' time, then the client
664
 
        # would inevitably timeout, since no checker would get a
665
 
        # chance to run to completion.  If we instead leave running
 
655
        # did that, then if a checker (for some reason) started
 
656
        # running slowly and taking more than 'interval' time, the
 
657
        # client would inevitably timeout, since no checker would get
 
658
        # a chance to run to completion.  If we instead leave running
666
659
        # checkers alone, the checker would have to take more time
667
660
        # than 'timeout' for the client to be disabled, which is as it
668
661
        # should be.
698
691
                try:
699
692
                    command = self.checker_command % escaped_attrs
700
693
                except TypeError as error:
701
 
                    logger.error('Could not format string "%s"',
702
 
                                 self.checker_command, exc_info=error)
 
694
                    logger.error('Could not format string "%s":'
 
695
                                 ' %s', self.checker_command, error)
703
696
                    return True # Try again later
704
697
            self.current_checker_command = command
705
698
            try:
723
716
                    gobject.source_remove(self.checker_callback_tag)
724
717
                    self.checker_callback(pid, status, command)
725
718
            except OSError as error:
726
 
                logger.error("Failed to start subprocess",
727
 
                             exc_info=error)
 
719
                logger.error("Failed to start subprocess: %s",
 
720
                             error)
728
721
        # Re-run this periodically if run by gobject.timeout_add
729
722
        return True
730
723
    
737
730
            return
738
731
        logger.debug("Stopping checker for %(name)s", vars(self))
739
732
        try:
740
 
            self.checker.terminate()
 
733
            os.kill(self.checker.pid, signal.SIGTERM)
741
734
            #time.sleep(0.5)
742
735
            #if self.checker.poll() is None:
743
 
            #    self.checker.kill()
 
736
            #    os.kill(self.checker.pid, signal.SIGKILL)
744
737
        except OSError as error:
745
738
            if error.errno != errno.ESRCH: # No such process
746
739
                raise
763
756
    # "Set" method, so we fail early here:
764
757
    if byte_arrays and signature != "ay":
765
758
        raise ValueError("Byte arrays not supported for non-'ay'"
766
 
                         " signature {0!r}".format(signature))
 
759
                         " signature %r" % signature)
767
760
    def decorator(func):
768
761
        func._dbus_is_property = True
769
762
        func._dbus_interface = dbus_interface
777
770
    return decorator
778
771
 
779
772
 
780
 
def dbus_interface_annotations(dbus_interface):
781
 
    """Decorator for marking functions returning interface annotations
782
 
    
783
 
    Usage:
784
 
    
785
 
    @dbus_interface_annotations("org.example.Interface")
786
 
    def _foo(self):  # Function name does not matter
787
 
        return {"org.freedesktop.DBus.Deprecated": "true",
788
 
                "org.freedesktop.DBus.Property.EmitsChangedSignal":
789
 
                    "false"}
790
 
    """
791
 
    def decorator(func):
792
 
        func._dbus_is_interface = True
793
 
        func._dbus_interface = dbus_interface
794
 
        func._dbus_name = dbus_interface
795
 
        return func
796
 
    return decorator
797
 
 
798
 
 
799
 
def dbus_annotations(annotations):
800
 
    """Decorator to annotate D-Bus methods, signals or properties
801
 
    Usage:
802
 
    
803
 
    @dbus_service_property("org.example.Interface", signature="b",
804
 
                           access="r")
805
 
    @dbus_annotations({{"org.freedesktop.DBus.Deprecated": "true",
806
 
                        "org.freedesktop.DBus.Property."
807
 
                        "EmitsChangedSignal": "false"})
808
 
    def Property_dbus_property(self):
809
 
        return dbus.Boolean(False)
810
 
    """
811
 
    def decorator(func):
812
 
        func._dbus_annotations = annotations
813
 
        return func
814
 
    return decorator
815
 
 
816
 
 
817
773
class DBusPropertyException(dbus.exceptions.DBusException):
818
774
    """A base class for D-Bus property-related exceptions
819
775
    """
842
798
    """
843
799
    
844
800
    @staticmethod
845
 
    def _is_dbus_thing(thing):
846
 
        """Returns a function testing if an attribute is a D-Bus thing
847
 
        
848
 
        If called like _is_dbus_thing("method") it returns a function
849
 
        suitable for use as predicate to inspect.getmembers().
850
 
        """
851
 
        return lambda obj: getattr(obj, "_dbus_is_{0}".format(thing),
852
 
                                   False)
 
801
    def _is_dbus_property(obj):
 
802
        return getattr(obj, "_dbus_is_property", False)
853
803
    
854
 
    def _get_all_dbus_things(self, thing):
 
804
    def _get_all_dbus_properties(self):
855
805
        """Returns a generator of (name, attribute) pairs
856
806
        """
857
 
        return ((getattr(athing.__get__(self), "_dbus_name",
858
 
                         name),
859
 
                 athing.__get__(self))
 
807
        return ((prop.__get__(self)._dbus_name, prop.__get__(self))
860
808
                for cls in self.__class__.__mro__
861
 
                for name, athing in
862
 
                inspect.getmembers(cls,
863
 
                                   self._is_dbus_thing(thing)))
 
809
                for name, prop in
 
810
                inspect.getmembers(cls, self._is_dbus_property))
864
811
    
865
812
    def _get_dbus_property(self, interface_name, property_name):
866
813
        """Returns a bound method if one exists which is a D-Bus
868
815
        """
869
816
        for cls in  self.__class__.__mro__:
870
817
            for name, value in (inspect.getmembers
871
 
                                (cls,
872
 
                                 self._is_dbus_thing("property"))):
 
818
                                (cls, self._is_dbus_property)):
873
819
                if (value._dbus_name == property_name
874
820
                    and value._dbus_interface == interface_name):
875
821
                    return value.__get__(self)
917
863
        Note: Will not include properties with access="write".
918
864
        """
919
865
        properties = {}
920
 
        for name, prop in self._get_all_dbus_things("property"):
 
866
        for name, prop in self._get_all_dbus_properties():
921
867
            if (interface_name
922
868
                and interface_name != prop._dbus_interface):
923
869
                # Interface non-empty but did not match
938
884
                         path_keyword='object_path',
939
885
                         connection_keyword='connection')
940
886
    def Introspect(self, object_path, connection):
941
 
        """Overloading of standard D-Bus method.
942
 
        
943
 
        Inserts property tags and interface annotation tags.
 
887
        """Standard D-Bus method, overloaded to insert property tags.
944
888
        """
945
889
        xmlstring = dbus.service.Object.Introspect(self, object_path,
946
890
                                                   connection)
953
897
                e.setAttribute("access", prop._dbus_access)
954
898
                return e
955
899
            for if_tag in document.getElementsByTagName("interface"):
956
 
                # Add property tags
957
900
                for tag in (make_tag(document, name, prop)
958
901
                            for name, prop
959
 
                            in self._get_all_dbus_things("property")
 
902
                            in self._get_all_dbus_properties()
960
903
                            if prop._dbus_interface
961
904
                            == if_tag.getAttribute("name")):
962
905
                    if_tag.appendChild(tag)
963
 
                # Add annotation tags
964
 
                for typ in ("method", "signal", "property"):
965
 
                    for tag in if_tag.getElementsByTagName(typ):
966
 
                        annots = dict()
967
 
                        for name, prop in (self.
968
 
                                           _get_all_dbus_things(typ)):
969
 
                            if (name == tag.getAttribute("name")
970
 
                                and prop._dbus_interface
971
 
                                == if_tag.getAttribute("name")):
972
 
                                annots.update(getattr
973
 
                                              (prop,
974
 
                                               "_dbus_annotations",
975
 
                                               {}))
976
 
                        for name, value in annots.iteritems():
977
 
                            ann_tag = document.createElement(
978
 
                                "annotation")
979
 
                            ann_tag.setAttribute("name", name)
980
 
                            ann_tag.setAttribute("value", value)
981
 
                            tag.appendChild(ann_tag)
982
 
                # Add interface annotation tags
983
 
                for annotation, value in dict(
984
 
                    itertools.chain.from_iterable(
985
 
                        annotations().iteritems()
986
 
                        for name, annotations in
987
 
                        self._get_all_dbus_things("interface")
988
 
                        if name == if_tag.getAttribute("name")
989
 
                        )).iteritems():
990
 
                    ann_tag = document.createElement("annotation")
991
 
                    ann_tag.setAttribute("name", annotation)
992
 
                    ann_tag.setAttribute("value", value)
993
 
                    if_tag.appendChild(ann_tag)
994
906
                # Add the names to the return values for the
995
907
                # "org.freedesktop.DBus.Properties" methods
996
908
                if (if_tag.getAttribute("name")
1011
923
        except (AttributeError, xml.dom.DOMException,
1012
924
                xml.parsers.expat.ExpatError) as error:
1013
925
            logger.error("Failed to override Introspection method",
1014
 
                         exc_info=error)
 
926
                         error)
1015
927
        return xmlstring
1016
928
 
1017
929
 
1023
935
                       variant_level=variant_level)
1024
936
 
1025
937
 
1026
 
def alternate_dbus_interfaces(alt_interface_names, deprecate=True):
1027
 
    """A class decorator; applied to a subclass of
1028
 
    dbus.service.Object, it will add alternate D-Bus attributes with
1029
 
    interface names according to the "alt_interface_names" mapping.
1030
 
    Usage:
1031
 
    
1032
 
    @alternate_dbus_names({"org.example.Interface":
1033
 
                               "net.example.AlternateInterface"})
1034
 
    class SampleDBusObject(dbus.service.Object):
1035
 
        @dbus.service.method("org.example.Interface")
1036
 
        def SampleDBusMethod():
1037
 
            pass
1038
 
    
1039
 
    The above "SampleDBusMethod" on "SampleDBusObject" will be
1040
 
    reachable via two interfaces: "org.example.Interface" and
1041
 
    "net.example.AlternateInterface", the latter of which will have
1042
 
    its D-Bus annotation "org.freedesktop.DBus.Deprecated" set to
1043
 
    "true", unless "deprecate" is passed with a False value.
1044
 
    
1045
 
    This works for methods and signals, and also for D-Bus properties
1046
 
    (from DBusObjectWithProperties) and interfaces (from the
1047
 
    dbus_interface_annotations decorator).
 
938
class AlternateDBusNamesMetaclass(DBusObjectWithProperties
 
939
                                  .__metaclass__):
 
940
    """Applied to an empty subclass of a D-Bus object, this metaclass
 
941
    will add additional D-Bus attributes matching a certain pattern.
1048
942
    """
1049
 
    def wrapper(cls):
1050
 
        for orig_interface_name, alt_interface_name in (
1051
 
            alt_interface_names.iteritems()):
1052
 
            attr = {}
1053
 
            interface_names = set()
1054
 
            # Go though all attributes of the class
1055
 
            for attrname, attribute in inspect.getmembers(cls):
 
943
    def __new__(mcs, name, bases, attr):
 
944
        # Go through all the base classes which could have D-Bus
 
945
        # methods, signals, or properties in them
 
946
        for base in (b for b in bases
 
947
                     if issubclass(b, dbus.service.Object)):
 
948
            # Go though all attributes of the base class
 
949
            for attrname, attribute in inspect.getmembers(base):
1056
950
                # Ignore non-D-Bus attributes, and D-Bus attributes
1057
951
                # with the wrong interface name
1058
952
                if (not hasattr(attribute, "_dbus_interface")
1059
953
                    or not attribute._dbus_interface
1060
 
                    .startswith(orig_interface_name)):
 
954
                    .startswith("se.recompile.Mandos")):
1061
955
                    continue
1062
956
                # Create an alternate D-Bus interface name based on
1063
957
                # the current name
1064
958
                alt_interface = (attribute._dbus_interface
1065
 
                                 .replace(orig_interface_name,
1066
 
                                          alt_interface_name))
1067
 
                interface_names.add(alt_interface)
 
959
                                 .replace("se.recompile.Mandos",
 
960
                                          "se.bsnet.fukt.Mandos"))
1068
961
                # Is this a D-Bus signal?
1069
962
                if getattr(attribute, "_dbus_is_signal", False):
1070
963
                    # Extract the original non-method function by
1085
978
                                nonmethod_func.func_name,
1086
979
                                nonmethod_func.func_defaults,
1087
980
                                nonmethod_func.func_closure)))
1088
 
                    # Copy annotations, if any
1089
 
                    try:
1090
 
                        new_function._dbus_annotations = (
1091
 
                            dict(attribute._dbus_annotations))
1092
 
                    except AttributeError:
1093
 
                        pass
1094
981
                    # Define a creator of a function to call both the
1095
 
                    # original and alternate functions, so both the
1096
 
                    # original and alternate signals gets sent when
1097
 
                    # the function is called
 
982
                    # old and new functions, so both the old and new
 
983
                    # signals gets sent when the function is called
1098
984
                    def fixscope(func1, func2):
1099
985
                        """This function is a scope container to pass
1100
986
                        func1 and func2 to the "call_both" function
1107
993
                        return call_both
1108
994
                    # Create the "call_both" function and add it to
1109
995
                    # the class
1110
 
                    attr[attrname] = fixscope(attribute, new_function)
 
996
                    attr[attrname] = fixscope(attribute,
 
997
                                              new_function)
1111
998
                # Is this a D-Bus method?
1112
999
                elif getattr(attribute, "_dbus_is_method", False):
1113
1000
                    # Create a new, but exactly alike, function
1124
1011
                                        attribute.func_name,
1125
1012
                                        attribute.func_defaults,
1126
1013
                                        attribute.func_closure)))
1127
 
                    # Copy annotations, if any
1128
 
                    try:
1129
 
                        attr[attrname]._dbus_annotations = (
1130
 
                            dict(attribute._dbus_annotations))
1131
 
                    except AttributeError:
1132
 
                        pass
1133
1014
                # Is this a D-Bus property?
1134
1015
                elif getattr(attribute, "_dbus_is_property", False):
1135
1016
                    # Create a new, but exactly alike, function
1149
1030
                                        attribute.func_name,
1150
1031
                                        attribute.func_defaults,
1151
1032
                                        attribute.func_closure)))
1152
 
                    # Copy annotations, if any
1153
 
                    try:
1154
 
                        attr[attrname]._dbus_annotations = (
1155
 
                            dict(attribute._dbus_annotations))
1156
 
                    except AttributeError:
1157
 
                        pass
1158
 
                # Is this a D-Bus interface?
1159
 
                elif getattr(attribute, "_dbus_is_interface", False):
1160
 
                    # Create a new, but exactly alike, function
1161
 
                    # object.  Decorate it to be a new D-Bus interface
1162
 
                    # with the alternate D-Bus interface name.  Add it
1163
 
                    # to the class.
1164
 
                    attr[attrname] = (dbus_interface_annotations
1165
 
                                      (alt_interface)
1166
 
                                      (types.FunctionType
1167
 
                                       (attribute.func_code,
1168
 
                                        attribute.func_globals,
1169
 
                                        attribute.func_name,
1170
 
                                        attribute.func_defaults,
1171
 
                                        attribute.func_closure)))
1172
 
            if deprecate:
1173
 
                # Deprecate all alternate interfaces
1174
 
                iname="_AlternateDBusNames_interface_annotation{0}"
1175
 
                for interface_name in interface_names:
1176
 
                    @dbus_interface_annotations(interface_name)
1177
 
                    def func(self):
1178
 
                        return { "org.freedesktop.DBus.Deprecated":
1179
 
                                     "true" }
1180
 
                    # Find an unused name
1181
 
                    for aname in (iname.format(i)
1182
 
                                  for i in itertools.count()):
1183
 
                        if aname not in attr:
1184
 
                            attr[aname] = func
1185
 
                            break
1186
 
            if interface_names:
1187
 
                # Replace the class with a new subclass of it with
1188
 
                # methods, signals, etc. as created above.
1189
 
                cls = type(b"{0}Alternate".format(cls.__name__),
1190
 
                           (cls,), attr)
1191
 
        return cls
1192
 
    return wrapper
1193
 
 
1194
 
 
1195
 
@alternate_dbus_interfaces({"se.recompile.Mandos":
1196
 
                                "se.bsnet.fukt.Mandos"})
 
1033
        return type.__new__(mcs, name, bases, attr)
 
1034
 
 
1035
 
1197
1036
class ClientDBus(Client, DBusObjectWithProperties):
1198
1037
    """A Client class using D-Bus
1199
1038
    
1210
1049
    def __init__(self, bus = None, *args, **kwargs):
1211
1050
        self.bus = bus
1212
1051
        Client.__init__(self, *args, **kwargs)
 
1052
        self._approvals_pending = 0
 
1053
        
 
1054
        self._approvals_pending = 0
1213
1055
        # Only now, when this client is initialized, can it show up on
1214
1056
        # the D-Bus
1215
1057
        client_object_name = unicode(self.name).translate(
1219
1061
                                 ("/clients/" + client_object_name))
1220
1062
        DBusObjectWithProperties.__init__(self, self.bus,
1221
1063
                                          self.dbus_object_path)
1222
 
    
 
1064
        
1223
1065
    def notifychangeproperty(transform_func,
1224
1066
                             dbus_name, type_func=lambda x: x,
1225
1067
                             variant_level=1):
1248
1090
        
1249
1091
        return property(lambda self: getattr(self, attrname), setter)
1250
1092
    
 
1093
    
1251
1094
    expires = notifychangeproperty(datetime_to_dbus, "Expires")
1252
1095
    approvals_pending = notifychangeproperty(dbus.Boolean,
1253
1096
                                             "ApprovalPending",
1260
1103
                                       checker is not None)
1261
1104
    last_checked_ok = notifychangeproperty(datetime_to_dbus,
1262
1105
                                           "LastCheckedOK")
1263
 
    last_checker_status = notifychangeproperty(dbus.Int16,
1264
 
                                               "LastCheckerStatus")
1265
1106
    last_approval_request = notifychangeproperty(
1266
1107
        datetime_to_dbus, "LastApprovalRequest")
1267
1108
    approved_by_default = notifychangeproperty(dbus.Boolean,
1341
1182
                            (self.approval_duration),
1342
1183
                            self._reset_approved)
1343
1184
    
 
1185
    
1344
1186
    ## D-Bus methods, signals & properties
1345
1187
    _interface = "se.recompile.Mandos.Client"
1346
1188
    
1347
 
    ## Interfaces
1348
 
    
1349
 
    @dbus_interface_annotations(_interface)
1350
 
    def _foo(self):
1351
 
        return { "org.freedesktop.DBus.Property.EmitsChangedSignal":
1352
 
                     "false"}
1353
 
    
1354
1189
    ## Signals
1355
1190
    
1356
1191
    # CheckerCompleted - signal
1392
1227
        "D-Bus signal"
1393
1228
        return self.need_approval()
1394
1229
    
 
1230
    # NeRwequest - signal
 
1231
    @dbus.service.signal(_interface, signature="s")
 
1232
    def NewRequest(self, ip):
 
1233
        """D-Bus signal
 
1234
        Is sent after a client request a password.
 
1235
        """
 
1236
        pass
 
1237
    
1395
1238
    ## Methods
1396
1239
    
1397
1240
    # Approve - method
1507
1350
            return
1508
1351
        return datetime_to_dbus(self.last_checked_ok)
1509
1352
    
1510
 
    # LastCheckerStatus - property
1511
 
    @dbus_service_property(_interface, signature="n",
1512
 
                           access="read")
1513
 
    def LastCheckerStatus_dbus_property(self):
1514
 
        return dbus.Int16(self.last_checker_status)
1515
 
    
1516
1353
    # Expires - property
1517
1354
    @dbus_service_property(_interface, signature="s", access="read")
1518
1355
    def Expires_dbus_property(self):
1633
1470
        self._pipe.send(('setattr', name, value))
1634
1471
 
1635
1472
 
 
1473
class ClientDBusTransitional(ClientDBus):
 
1474
    __metaclass__ = AlternateDBusNamesMetaclass
 
1475
 
 
1476
 
1636
1477
class ClientHandler(socketserver.BaseRequestHandler, object):
1637
1478
    """A class to handle client connections.
1638
1479
    
1706
1547
                except KeyError:
1707
1548
                    return
1708
1549
                
 
1550
                if self.server.use_dbus:
 
1551
                    # Emit D-Bus signal
 
1552
                    client.NewRequest(str(self.client_address))
 
1553
                
1709
1554
                if client.approval_delay:
1710
1555
                    delay = client.approval_delay
1711
1556
                    client.approvals_pending += 1
1766
1611
                    try:
1767
1612
                        sent = session.send(client.secret[sent_size:])
1768
1613
                    except gnutls.errors.GNUTLSError as error:
1769
 
                        logger.warning("gnutls send failed",
1770
 
                                       exc_info=error)
 
1614
                        logger.warning("gnutls send failed")
1771
1615
                        return
1772
1616
                    logger.debug("Sent: %d, remaining: %d",
1773
1617
                                 sent, len(client.secret)
1776
1620
                
1777
1621
                logger.info("Sending secret to %s", client.name)
1778
1622
                # bump the timeout using extended_timeout
1779
 
                client.bump_timeout(client.extended_timeout)
 
1623
                client.checked_ok(client.extended_timeout)
1780
1624
                if self.server.use_dbus:
1781
1625
                    # Emit D-Bus signal
1782
1626
                    client.GotSecret()
1787
1631
                try:
1788
1632
                    session.bye()
1789
1633
                except gnutls.errors.GNUTLSError as error:
1790
 
                    logger.warning("GnuTLS bye failed",
1791
 
                                   exc_info=error)
 
1634
                    logger.warning("GnuTLS bye failed")
1792
1635
    
1793
1636
    @staticmethod
1794
1637
    def peer_certificate(session):
1866
1709
    def process_request(self, request, address):
1867
1710
        """Start a new process to process the request."""
1868
1711
        proc = multiprocessing.Process(target = self.sub_process_main,
1869
 
                                       args = (request, address))
 
1712
                                       args = (request,
 
1713
                                               address))
1870
1714
        proc.start()
1871
1715
        return proc
1872
1716
 
1994
1838
    
1995
1839
    def handle_ipc(self, source, condition, parent_pipe=None,
1996
1840
                   proc = None, client_object=None):
 
1841
        condition_names = {
 
1842
            gobject.IO_IN: "IN",   # There is data to read.
 
1843
            gobject.IO_OUT: "OUT", # Data can be written (without
 
1844
                                    # blocking).
 
1845
            gobject.IO_PRI: "PRI", # There is urgent data to read.
 
1846
            gobject.IO_ERR: "ERR", # Error condition.
 
1847
            gobject.IO_HUP: "HUP"  # Hung up (the connection has been
 
1848
                                    # broken, usually for pipes and
 
1849
                                    # sockets).
 
1850
            }
 
1851
        conditions_string = ' | '.join(name
 
1852
                                       for cond, name in
 
1853
                                       condition_names.iteritems()
 
1854
                                       if cond & condition)
1997
1855
        # error, or the other end of multiprocessing.Pipe has closed
1998
 
        if condition & (gobject.IO_ERR | gobject.IO_HUP):
 
1856
        if condition & (gobject.IO_ERR | condition & gobject.IO_HUP):
1999
1857
            # Wait for other process to exit
2000
1858
            proc.join()
2001
1859
            return False
2091
1949
            elif suffix == "w":
2092
1950
                delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
2093
1951
            else:
2094
 
                raise ValueError("Unknown suffix {0!r}"
2095
 
                                 .format(suffix))
 
1952
                raise ValueError("Unknown suffix %r" % suffix)
2096
1953
        except (ValueError, IndexError) as e:
2097
1954
            raise ValueError(*(e.args))
2098
1955
        timevalue += delta
2112
1969
        sys.exit()
2113
1970
    if not noclose:
2114
1971
        # Close all standard open file descriptors
2115
 
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
 
1972
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
2116
1973
        if not stat.S_ISCHR(os.fstat(null).st_mode):
2117
1974
            raise OSError(errno.ENODEV,
2118
 
                          "{0} not a character device"
2119
 
                          .format(os.devnull))
 
1975
                          "%s not a character device"
 
1976
                          % os.path.devnull)
2120
1977
        os.dup2(null, sys.stdin.fileno())
2121
1978
        os.dup2(null, sys.stdout.fileno())
2122
1979
        os.dup2(null, sys.stderr.fileno())
2131
1988
    
2132
1989
    parser = argparse.ArgumentParser()
2133
1990
    parser.add_argument("-v", "--version", action="version",
2134
 
                        version = "%(prog)s {0}".format(version),
 
1991
                        version = "%%(prog)s %s" % version,
2135
1992
                        help="show version number and exit")
2136
1993
    parser.add_argument("-i", "--interface", metavar="IF",
2137
1994
                        help="Bind to interface IF")
2240
2097
    
2241
2098
    if server_settings["servicename"] != "Mandos":
2242
2099
        syslogger.setFormatter(logging.Formatter
2243
 
                               ('Mandos ({0}) [%(process)d]:'
2244
 
                                ' %(levelname)s: %(message)s'
2245
 
                                .format(server_settings
2246
 
                                        ["servicename"])))
 
2100
                               ('Mandos (%s) [%%(process)d]:'
 
2101
                                ' %%(levelname)s: %%(message)s'
 
2102
                                % server_settings["servicename"]))
2247
2103
    
2248
2104
    # Parse config file with clients
2249
2105
    client_config = configparser.SafeConfigParser(Client
2267
2123
        pidfilename = "/var/run/mandos.pid"
2268
2124
        try:
2269
2125
            pidfile = open(pidfilename, "w")
2270
 
        except IOError as e:
2271
 
            logger.error("Could not open file %r", pidfilename,
2272
 
                         exc_info=e)
 
2126
        except IOError:
 
2127
            logger.error("Could not open file %r", pidfilename)
2273
2128
    
2274
 
    for name in ("_mandos", "mandos", "nobody"):
 
2129
    try:
 
2130
        uid = pwd.getpwnam("_mandos").pw_uid
 
2131
        gid = pwd.getpwnam("_mandos").pw_gid
 
2132
    except KeyError:
2275
2133
        try:
2276
 
            uid = pwd.getpwnam(name).pw_uid
2277
 
            gid = pwd.getpwnam(name).pw_gid
2278
 
            break
 
2134
            uid = pwd.getpwnam("mandos").pw_uid
 
2135
            gid = pwd.getpwnam("mandos").pw_gid
2279
2136
        except KeyError:
2280
 
            continue
2281
 
    else:
2282
 
        uid = 65534
2283
 
        gid = 65534
 
2137
            try:
 
2138
                uid = pwd.getpwnam("nobody").pw_uid
 
2139
                gid = pwd.getpwnam("nobody").pw_gid
 
2140
            except KeyError:
 
2141
                uid = 65534
 
2142
                gid = 65534
2284
2143
    try:
2285
2144
        os.setgid(gid)
2286
2145
        os.setuid(uid)
2303
2162
         .gnutls_global_set_log_function(debug_gnutls))
2304
2163
        
2305
2164
        # Redirect stdin so all checkers get /dev/null
2306
 
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
 
2165
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
2307
2166
        os.dup2(null, sys.stdin.fileno())
2308
2167
        if null > 2:
2309
2168
            os.close(null)
2317
2176
    
2318
2177
    global main_loop
2319
2178
    # From the Avahi example code
2320
 
    DBusGMainLoop(set_as_default=True)
 
2179
    DBusGMainLoop(set_as_default=True )
2321
2180
    main_loop = gobject.MainLoop()
2322
2181
    bus = dbus.SystemBus()
2323
2182
    # End of Avahi example code
2329
2188
                            ("se.bsnet.fukt.Mandos", bus,
2330
2189
                             do_not_queue=True))
2331
2190
        except dbus.exceptions.NameExistsException as e:
2332
 
            logger.error("Disabling D-Bus:", exc_info=e)
 
2191
            logger.error(unicode(e) + ", disabling D-Bus")
2333
2192
            use_dbus = False
2334
2193
            server_settings["use_dbus"] = False
2335
2194
            tcp_server.use_dbus = False
2347
2206
    
2348
2207
    client_class = Client
2349
2208
    if use_dbus:
2350
 
        client_class = functools.partial(ClientDBus, bus = bus)
 
2209
        client_class = functools.partial(ClientDBusTransitional,
 
2210
                                         bus = bus)
2351
2211
    
2352
2212
    client_settings = Client.config_parser(client_config)
2353
2213
    old_client_settings = {}
2361
2221
                                                     (stored_state))
2362
2222
            os.remove(stored_state_path)
2363
2223
        except IOError as e:
2364
 
            if e.errno == errno.ENOENT:
2365
 
                logger.warning("Could not load persistent state: {0}"
2366
 
                                .format(os.strerror(e.errno)))
2367
 
            else:
2368
 
                logger.critical("Could not load persistent state:",
2369
 
                                exc_info=e)
 
2224
            logger.warning("Could not load persistent state: {0}"
 
2225
                           .format(e))
 
2226
            if e.errno != errno.ENOENT:
2370
2227
                raise
2371
2228
        except EOFError as e:
2372
2229
            logger.warning("Could not load persistent state: "
2373
 
                           "EOFError:", exc_info=e)
 
2230
                           "EOFError: {0}".format(e))
2374
2231
    
2375
2232
    with PGPEngine() as pgp:
2376
2233
        for client_name, client in clients_data.iteritems():
2393
2250
            
2394
2251
            # Clients who has passed its expire date can still be
2395
2252
            # enabled if its last checker was successful.  Clients
2396
 
            # whose checker succeeded before we stored its state is
2397
 
            # assumed to have successfully run all checkers during
2398
 
            # downtime.
 
2253
            # whose checker failed before we stored its state is
 
2254
            # assumed to have failed all checkers during downtime.
2399
2255
            if client["enabled"]:
2400
2256
                if datetime.datetime.utcnow() >= client["expires"]:
2401
2257
                    if not client["last_checked_ok"]:
2402
2258
                        logger.warning(
2403
2259
                            "disabling client {0} - Client never "
2404
 
                            "performed a successful checker"
2405
 
                            .format(client_name))
 
2260
                            "performed a successfull checker"
 
2261
                            .format(client["name"]))
2406
2262
                        client["enabled"] = False
2407
2263
                    elif client["last_checker_status"] != 0:
2408
2264
                        logger.warning(
2409
2265
                            "disabling client {0} - Client "
2410
2266
                            "last checker failed with error code {1}"
2411
 
                            .format(client_name,
 
2267
                            .format(client["name"],
2412
2268
                                    client["last_checker_status"]))
2413
2269
                        client["enabled"] = False
2414
2270
                    else:
2417
2273
                                             + client["timeout"])
2418
2274
                        logger.debug("Last checker succeeded,"
2419
2275
                                     " keeping {0} enabled"
2420
 
                                     .format(client_name))
 
2276
                                     .format(client["name"]))
2421
2277
            try:
2422
2278
                client["secret"] = (
2423
2279
                    pgp.decrypt(client["encrypted_secret"],
2429
2285
                             .format(client_name))
2430
2286
                client["secret"] = (
2431
2287
                    client_settings[client_name]["secret"])
 
2288
 
2432
2289
    
2433
2290
    # Add/remove clients based on new changes made to config
2434
2291
    for client_name in (set(old_client_settings)
2437
2294
    for client_name in (set(client_settings)
2438
2295
                        - set(old_client_settings)):
2439
2296
        clients_data[client_name] = client_settings[client_name]
2440
 
    
2441
 
    # Create all client objects
 
2297
 
 
2298
    # Create clients all clients
2442
2299
    for client_name, client in clients_data.iteritems():
2443
2300
        tcp_server.clients[client_name] = client_class(
2444
2301
            name = client_name, settings = client)
2445
2302
    
2446
2303
    if not tcp_server.clients:
2447
2304
        logger.warning("No clients defined")
2448
 
    
 
2305
        
2449
2306
    if not debug:
2450
2307
        try:
2451
2308
            with pidfile:
2465
2322
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
2466
2323
    
2467
2324
    if use_dbus:
2468
 
        @alternate_dbus_interfaces({"se.recompile.Mandos":
2469
 
                                        "se.bsnet.fukt.Mandos"})
2470
 
        class MandosDBusService(DBusObjectWithProperties):
 
2325
        class MandosDBusService(dbus.service.Object):
2471
2326
            """A D-Bus proxy object"""
2472
2327
            def __init__(self):
2473
2328
                dbus.service.Object.__init__(self, bus, "/")
2474
2329
            _interface = "se.recompile.Mandos"
2475
2330
            
2476
 
            @dbus_interface_annotations(_interface)
2477
 
            def _foo(self):
2478
 
                return { "org.freedesktop.DBus.Property"
2479
 
                         ".EmitsChangedSignal":
2480
 
                             "false"}
2481
 
            
2482
2331
            @dbus.service.signal(_interface, signature="o")
2483
2332
            def ClientAdded(self, objpath):
2484
2333
                "D-Bus signal"
2526
2375
            
2527
2376
            del _interface
2528
2377
        
2529
 
        mandos_dbus_service = MandosDBusService()
 
2378
        class MandosDBusServiceTransitional(MandosDBusService):
 
2379
            __metaclass__ = AlternateDBusNamesMetaclass
 
2380
        mandos_dbus_service = MandosDBusServiceTransitional()
2530
2381
    
2531
2382
    def cleanup():
2532
2383
        "Cleanup function; run on exit"
2565
2416
                del client_settings[client.name]["secret"]
2566
2417
        
2567
2418
        try:
2568
 
            with (tempfile.NamedTemporaryFile
2569
 
                  (mode='wb', suffix=".pickle", prefix='clients-',
2570
 
                   dir=os.path.dirname(stored_state_path),
2571
 
                   delete=False)) as stored_state:
 
2419
            tempfd, tempname = tempfile.mkstemp(suffix=".pickle",
 
2420
                                                prefix="clients-",
 
2421
                                                dir=os.path.dirname
 
2422
                                                (stored_state_path))
 
2423
            with os.fdopen(tempfd, "wb") as stored_state:
2572
2424
                pickle.dump((clients, client_settings), stored_state)
2573
 
                tempname=stored_state.name
2574
2425
            os.rename(tempname, stored_state_path)
2575
2426
        except (IOError, OSError) as e:
 
2427
            logger.warning("Could not save persistent state: {0}"
 
2428
                           .format(e))
2576
2429
            if not debug:
2577
2430
                try:
2578
2431
                    os.remove(tempname)
2579
2432
                except NameError:
2580
2433
                    pass
2581
 
            if e.errno in (errno.ENOENT, errno.EACCES, errno.EEXIST):
2582
 
                logger.warning("Could not save persistent state: {0}"
2583
 
                               .format(os.strerror(e.errno)))
2584
 
            else:
2585
 
                logger.warning("Could not save persistent state:",
2586
 
                               exc_info=e)
 
2434
            if e.errno not in set((errno.ENOENT, errno.EACCES,
 
2435
                                   errno.EEXIST)):
2587
2436
                raise e
2588
2437
        
2589
2438
        # Delete all clients, and settings from config
2617
2466
    service.port = tcp_server.socket.getsockname()[1]
2618
2467
    if use_ipv6:
2619
2468
        logger.info("Now listening on address %r, port %d,"
2620
 
                    " flowinfo %d, scope_id %d",
2621
 
                    *tcp_server.socket.getsockname())
 
2469
                    " flowinfo %d, scope_id %d"
 
2470
                    % tcp_server.socket.getsockname())
2622
2471
    else:                       # IPv4
2623
 
        logger.info("Now listening on address %r, port %d",
2624
 
                    *tcp_server.socket.getsockname())
 
2472
        logger.info("Now listening on address %r, port %d"
 
2473
                    % tcp_server.socket.getsockname())
2625
2474
    
2626
2475
    #service.interface = tcp_server.socket.getsockname()[3]
2627
2476
    
2630
2479
        try:
2631
2480
            service.activate()
2632
2481
        except dbus.exceptions.DBusException as error:
2633
 
            logger.critical("D-Bus Exception", exc_info=error)
 
2482
            logger.critical("DBusException: %s", error)
2634
2483
            cleanup()
2635
2484
            sys.exit(1)
2636
2485
        # End of Avahi example code
2643
2492
        logger.debug("Starting main loop")
2644
2493
        main_loop.run()
2645
2494
    except AvahiError as error:
2646
 
        logger.critical("Avahi Error", exc_info=error)
 
2495
        logger.critical("AvahiError: %s", error)
2647
2496
        cleanup()
2648
2497
        sys.exit(1)
2649
2498
    except KeyboardInterrupt: