/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-06-17 22:26:40 UTC
  • mfrom: (301.1.1 release)
  • Revision ID: teddy@recompile.se-20120617222640-49scfn0husnrr5bi
Tags: version-1.6.0-1
* Makefile (version): Changed to "1.6.0".
* NEWS (Version 1.6.0): New entry.
* debian/changelog (1.6.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
 
37
39
import SocketServer as socketserver
38
40
import socket
39
41
import argparse
65
67
import types
66
68
import binascii
67
69
import tempfile
 
70
import itertools
68
71
 
69
72
import dbus
70
73
import dbus.service
85
88
    except ImportError:
86
89
        SO_BINDTODEVICE = None
87
90
 
88
 
version = "1.5.3"
 
91
version = "1.6.0"
89
92
stored_state_file = "clients.pickle"
90
93
 
91
94
logger = logging.getLogger()
148
151
    def __enter__(self):
149
152
        return self
150
153
    
151
 
    def __exit__ (self, exc_type, exc_value, traceback):
 
154
    def __exit__(self, exc_type, exc_value, traceback):
152
155
        self._cleanup()
153
156
        return False
154
157
    
175
178
    
176
179
    def encrypt(self, data, password):
177
180
        self.gnupg.passphrase = self.password_encode(password)
178
 
        with open(os.devnull) as devnull:
 
181
        with open(os.devnull, "w") as devnull:
179
182
            try:
180
183
                proc = self.gnupg.run(['--symmetric'],
181
184
                                      create_fhs=['stdin', 'stdout'],
192
195
    
193
196
    def decrypt(self, data, password):
194
197
        self.gnupg.passphrase = self.password_encode(password)
195
 
        with open(os.devnull) as devnull:
 
198
        with open(os.devnull, "w") as devnull:
196
199
            try:
197
200
                proc = self.gnupg.run(['--decrypt'],
198
201
                                      create_fhs=['stdin', 'stdout'],
199
202
                                      attach_fhs={'stderr': devnull})
200
 
                with contextlib.closing(proc.handles['stdin'] ) as f:
 
203
                with contextlib.closing(proc.handles['stdin']) as f:
201
204
                    f.write(data)
202
205
                with contextlib.closing(proc.handles['stdout']) as f:
203
206
                    decrypted_plaintext = f.read()
208
211
        return decrypted_plaintext
209
212
 
210
213
 
211
 
 
212
214
class AvahiError(Exception):
213
215
    def __init__(self, value, *args, **kwargs):
214
216
        self.value = value
243
245
    server: D-Bus Server
244
246
    bus: dbus.SystemBus()
245
247
    """
 
248
    
246
249
    def __init__(self, interface = avahi.IF_UNSPEC, name = None,
247
250
                 servicetype = None, port = None, TXT = None,
248
251
                 domain = "", host = "", max_renames = 32768,
261
264
        self.server = None
262
265
        self.bus = bus
263
266
        self.entry_group_state_changed_match = None
 
267
    
264
268
    def rename(self):
265
269
        """Derived from the Avahi example code"""
266
270
        if self.rename_count >= self.max_renames:
276
280
        try:
277
281
            self.add()
278
282
        except dbus.exceptions.DBusException as error:
279
 
            logger.critical("DBusException: %s", error)
 
283
            logger.critical("D-Bus Exception", exc_info=error)
280
284
            self.cleanup()
281
285
            os._exit(1)
282
286
        self.rename_count += 1
 
287
    
283
288
    def remove(self):
284
289
        """Derived from the Avahi example code"""
285
290
        if self.entry_group_state_changed_match is not None:
287
292
            self.entry_group_state_changed_match = None
288
293
        if self.group is not None:
289
294
            self.group.Reset()
 
295
    
290
296
    def add(self):
291
297
        """Derived from the Avahi example code"""
292
298
        self.remove()
309
315
            dbus.UInt16(self.port),
310
316
            avahi.string_array_to_txt_array(self.TXT))
311
317
        self.group.Commit()
 
318
    
312
319
    def entry_group_state_changed(self, state, error):
313
320
        """Derived from the Avahi example code"""
314
321
        logger.debug("Avahi entry group state change: %i", state)
321
328
        elif state == avahi.ENTRY_GROUP_FAILURE:
322
329
            logger.critical("Avahi: Error in group state changed %s",
323
330
                            unicode(error))
324
 
            raise AvahiGroupError("State changed: %s"
325
 
                                  % unicode(error))
 
331
            raise AvahiGroupError("State changed: {0!s}"
 
332
                                  .format(error))
 
333
    
326
334
    def cleanup(self):
327
335
        """Derived from the Avahi example code"""
328
336
        if self.group is not None:
333
341
                pass
334
342
            self.group = None
335
343
        self.remove()
 
344
    
336
345
    def server_state_changed(self, state, error=None):
337
346
        """Derived from the Avahi example code"""
338
347
        logger.debug("Avahi server state change: %i", state)
357
366
                logger.debug("Unknown state: %r", state)
358
367
            else:
359
368
                logger.debug("Unknown state: %r: %r", state, error)
 
369
    
360
370
    def activate(self):
361
371
        """Derived from the Avahi example code"""
362
372
        if self.server is None:
369
379
                                 self.server_state_changed)
370
380
        self.server_state_changed(self.server.GetState())
371
381
 
 
382
 
372
383
class AvahiServiceToSyslog(AvahiService):
373
384
    def rename(self):
374
385
        """Add the new name to the syslog messages"""
375
386
        ret = AvahiService.rename(self)
376
387
        syslogger.setFormatter(logging.Formatter
377
 
                               ('Mandos (%s) [%%(process)d]:'
378
 
                                ' %%(levelname)s: %%(message)s'
379
 
                                % self.name))
 
388
                               ('Mandos ({0}) [%(process)d]:'
 
389
                                ' %(levelname)s: %(message)s'
 
390
                                .format(self.name)))
380
391
        return ret
381
392
 
 
393
 
382
394
def timedelta_to_milliseconds(td):
383
395
    "Convert a datetime.timedelta() to milliseconds"
384
396
    return ((td.days * 24 * 60 * 60 * 1000)
385
397
            + (td.seconds * 1000)
386
398
            + (td.microseconds // 1000))
387
 
        
 
399
 
 
400
 
388
401
class Client(object):
389
402
    """A representation of a client host served by this server.
390
403
    
429
442
    """
430
443
    
431
444
    runtime_expansions = ("approval_delay", "approval_duration",
432
 
                          "created", "enabled", "fingerprint",
433
 
                          "host", "interval", "last_checked_ok",
 
445
                          "created", "enabled", "expires",
 
446
                          "fingerprint", "host", "interval",
 
447
                          "last_approval_request", "last_checked_ok",
434
448
                          "last_enabled", "name", "timeout")
435
449
    client_defaults = { "timeout": "5m",
436
450
                        "extended_timeout": "15m",
457
471
    
458
472
    def approval_delay_milliseconds(self):
459
473
        return timedelta_to_milliseconds(self.approval_delay)
460
 
 
 
474
    
461
475
    @staticmethod
462
476
    def config_parser(config):
463
477
        """Construct a new dict of client settings of this form:
488
502
                          "rb") as secfile:
489
503
                    client["secret"] = secfile.read()
490
504
            else:
491
 
                raise TypeError("No secret or secfile for section %s"
492
 
                                % section)
 
505
                raise TypeError("No secret or secfile for section {0}"
 
506
                                .format(section))
493
507
            client["timeout"] = string_to_delta(section["timeout"])
494
508
            client["extended_timeout"] = string_to_delta(
495
509
                section["extended_timeout"])
504
518
            client["last_checker_status"] = -2
505
519
        
506
520
        return settings
507
 
        
508
 
        
 
521
    
509
522
    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."""
513
523
        self.name = name
514
524
        # adding all client settings
515
525
        for setting, value in settings.iteritems():
524
534
        else:
525
535
            self.last_enabled = None
526
536
            self.expires = None
527
 
       
 
537
        
528
538
        logger.debug("Creating client %r", self.name)
529
539
        # Uppercase and remove spaces from fingerprint for later
530
540
        # comparison purposes with return value from the fingerprint()
532
542
        logger.debug("  Fingerprint: %s", self.fingerprint)
533
543
        self.created = settings.get("created",
534
544
                                    datetime.datetime.utcnow())
535
 
 
 
545
        
536
546
        # attributes specific for this server instance
537
547
        self.checker = None
538
548
        self.checker_initiator_tag = None
566
576
        if getattr(self, "enabled", False):
567
577
            # Already enabled
568
578
            return
569
 
        self.send_changedstate()
570
579
        self.expires = datetime.datetime.utcnow() + self.timeout
571
580
        self.enabled = True
572
581
        self.last_enabled = datetime.datetime.utcnow()
573
582
        self.init_checker()
 
583
        self.send_changedstate()
574
584
    
575
585
    def disable(self, quiet=True):
576
586
        """Disable this client."""
577
587
        if not getattr(self, "enabled", False):
578
588
            return False
579
589
        if not quiet:
580
 
            self.send_changedstate()
581
 
        if not quiet:
582
590
            logger.info("Disabling client %s", self.name)
583
 
        if getattr(self, "disable_initiator_tag", False):
 
591
        if getattr(self, "disable_initiator_tag", None) is not None:
584
592
            gobject.source_remove(self.disable_initiator_tag)
585
593
            self.disable_initiator_tag = None
586
594
        self.expires = None
587
 
        if getattr(self, "checker_initiator_tag", False):
 
595
        if getattr(self, "checker_initiator_tag", None) is not None:
588
596
            gobject.source_remove(self.checker_initiator_tag)
589
597
            self.checker_initiator_tag = None
590
598
        self.stop_checker()
591
599
        self.enabled = False
 
600
        if not quiet:
 
601
            self.send_changedstate()
592
602
        # Do not run this again if called by a gobject.timeout_add
593
603
        return False
594
604
    
598
608
    def init_checker(self):
599
609
        # Schedule a new checker to be started an 'interval' from now,
600
610
        # and every interval from then on.
 
611
        if self.checker_initiator_tag is not None:
 
612
            gobject.source_remove(self.checker_initiator_tag)
601
613
        self.checker_initiator_tag = (gobject.timeout_add
602
614
                                      (self.interval_milliseconds(),
603
615
                                       self.start_checker))
604
616
        # Schedule a disable() when 'timeout' has passed
 
617
        if self.disable_initiator_tag is not None:
 
618
            gobject.source_remove(self.disable_initiator_tag)
605
619
        self.disable_initiator_tag = (gobject.timeout_add
606
620
                                   (self.timeout_milliseconds(),
607
621
                                    self.disable))
638
652
            timeout = self.timeout
639
653
        if self.disable_initiator_tag is not None:
640
654
            gobject.source_remove(self.disable_initiator_tag)
 
655
            self.disable_initiator_tag = None
641
656
        if getattr(self, "enabled", False):
642
657
            self.disable_initiator_tag = (gobject.timeout_add
643
658
                                          (timedelta_to_milliseconds
653
668
        If a checker already exists, leave it running and do
654
669
        nothing."""
655
670
        # The reason for not killing a running checker is that if we
656
 
        # did that, then if a checker (for some reason) started
657
 
        # running slowly and taking more than 'interval' time, the
658
 
        # client would inevitably timeout, since no checker would get
659
 
        # a chance to run to completion.  If we instead leave running
 
671
        # did that, and if a checker (for some reason) started running
 
672
        # slowly and taking more than 'interval' time, then the client
 
673
        # would inevitably timeout, since no checker would get a
 
674
        # chance to run to completion.  If we instead leave running
660
675
        # checkers alone, the checker would have to take more time
661
676
        # than 'timeout' for the client to be disabled, which is as it
662
677
        # should be.
676
691
                                      self.current_checker_command)
677
692
        # Start a new checker if needed
678
693
        if self.checker is None:
 
694
            # Escape attributes for the shell
 
695
            escaped_attrs = dict(
 
696
                (attr, re.escape(unicode(getattr(self, attr))))
 
697
                for attr in
 
698
                self.runtime_expansions)
679
699
            try:
680
 
                # In case checker_command has exactly one % operator
681
 
                command = self.checker_command % self.host
682
 
            except TypeError:
683
 
                # Escape attributes for the shell
684
 
                escaped_attrs = dict(
685
 
                    (attr,
686
 
                     re.escape(unicode(str(getattr(self, attr, "")),
687
 
                                       errors=
688
 
                                       'replace')))
689
 
                    for attr in
690
 
                    self.runtime_expansions)
691
 
                
692
 
                try:
693
 
                    command = self.checker_command % escaped_attrs
694
 
                except TypeError as error:
695
 
                    logger.error('Could not format string "%s":'
696
 
                                 ' %s', self.checker_command, error)
697
 
                    return True # Try again later
 
700
                command = self.checker_command % escaped_attrs
 
701
            except TypeError as error:
 
702
                logger.error('Could not format string "%s"',
 
703
                             self.checker_command, exc_info=error)
 
704
                return True # Try again later
698
705
            self.current_checker_command = command
699
706
            try:
700
707
                logger.info("Starting checker %r for %s",
706
713
                self.checker = subprocess.Popen(command,
707
714
                                                close_fds=True,
708
715
                                                shell=True, cwd="/")
709
 
                self.checker_callback_tag = (gobject.child_watch_add
710
 
                                             (self.checker.pid,
711
 
                                              self.checker_callback,
712
 
                                              data=command))
713
 
                # The checker may have completed before the gobject
714
 
                # watch was added.  Check for this.
715
 
                pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
716
 
                if pid:
717
 
                    gobject.source_remove(self.checker_callback_tag)
718
 
                    self.checker_callback(pid, status, command)
719
716
            except OSError as error:
720
 
                logger.error("Failed to start subprocess: %s",
721
 
                             error)
 
717
                logger.error("Failed to start subprocess",
 
718
                             exc_info=error)
 
719
            self.checker_callback_tag = (gobject.child_watch_add
 
720
                                         (self.checker.pid,
 
721
                                          self.checker_callback,
 
722
                                          data=command))
 
723
            # The checker may have completed before the gobject
 
724
            # watch was added.  Check for this.
 
725
            pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
 
726
            if pid:
 
727
                gobject.source_remove(self.checker_callback_tag)
 
728
                self.checker_callback(pid, status, command)
722
729
        # Re-run this periodically if run by gobject.timeout_add
723
730
        return True
724
731
    
731
738
            return
732
739
        logger.debug("Stopping checker for %(name)s", vars(self))
733
740
        try:
734
 
            os.kill(self.checker.pid, signal.SIGTERM)
 
741
            self.checker.terminate()
735
742
            #time.sleep(0.5)
736
743
            #if self.checker.poll() is None:
737
 
            #    os.kill(self.checker.pid, signal.SIGKILL)
 
744
            #    self.checker.kill()
738
745
        except OSError as error:
739
746
            if error.errno != errno.ESRCH: # No such process
740
747
                raise
757
764
    # "Set" method, so we fail early here:
758
765
    if byte_arrays and signature != "ay":
759
766
        raise ValueError("Byte arrays not supported for non-'ay'"
760
 
                         " signature %r" % signature)
 
767
                         " signature {0!r}".format(signature))
761
768
    def decorator(func):
762
769
        func._dbus_is_property = True
763
770
        func._dbus_interface = dbus_interface
771
778
    return decorator
772
779
 
773
780
 
 
781
def dbus_interface_annotations(dbus_interface):
 
782
    """Decorator for marking functions returning interface annotations
 
783
    
 
784
    Usage:
 
785
    
 
786
    @dbus_interface_annotations("org.example.Interface")
 
787
    def _foo(self):  # Function name does not matter
 
788
        return {"org.freedesktop.DBus.Deprecated": "true",
 
789
                "org.freedesktop.DBus.Property.EmitsChangedSignal":
 
790
                    "false"}
 
791
    """
 
792
    def decorator(func):
 
793
        func._dbus_is_interface = True
 
794
        func._dbus_interface = dbus_interface
 
795
        func._dbus_name = dbus_interface
 
796
        return func
 
797
    return decorator
 
798
 
 
799
 
 
800
def dbus_annotations(annotations):
 
801
    """Decorator to annotate D-Bus methods, signals or properties
 
802
    Usage:
 
803
    
 
804
    @dbus_service_property("org.example.Interface", signature="b",
 
805
                           access="r")
 
806
    @dbus_annotations({{"org.freedesktop.DBus.Deprecated": "true",
 
807
                        "org.freedesktop.DBus.Property."
 
808
                        "EmitsChangedSignal": "false"})
 
809
    def Property_dbus_property(self):
 
810
        return dbus.Boolean(False)
 
811
    """
 
812
    def decorator(func):
 
813
        func._dbus_annotations = annotations
 
814
        return func
 
815
    return decorator
 
816
 
 
817
 
774
818
class DBusPropertyException(dbus.exceptions.DBusException):
775
819
    """A base class for D-Bus property-related exceptions
776
820
    """
799
843
    """
800
844
    
801
845
    @staticmethod
802
 
    def _is_dbus_property(obj):
803
 
        return getattr(obj, "_dbus_is_property", False)
 
846
    def _is_dbus_thing(thing):
 
847
        """Returns a function testing if an attribute is a D-Bus thing
 
848
        
 
849
        If called like _is_dbus_thing("method") it returns a function
 
850
        suitable for use as predicate to inspect.getmembers().
 
851
        """
 
852
        return lambda obj: getattr(obj, "_dbus_is_{0}".format(thing),
 
853
                                   False)
804
854
    
805
 
    def _get_all_dbus_properties(self):
 
855
    def _get_all_dbus_things(self, thing):
806
856
        """Returns a generator of (name, attribute) pairs
807
857
        """
808
 
        return ((prop.__get__(self)._dbus_name, prop.__get__(self))
 
858
        return ((getattr(athing.__get__(self), "_dbus_name",
 
859
                         name),
 
860
                 athing.__get__(self))
809
861
                for cls in self.__class__.__mro__
810
 
                for name, prop in
811
 
                inspect.getmembers(cls, self._is_dbus_property))
 
862
                for name, athing in
 
863
                inspect.getmembers(cls,
 
864
                                   self._is_dbus_thing(thing)))
812
865
    
813
866
    def _get_dbus_property(self, interface_name, property_name):
814
867
        """Returns a bound method if one exists which is a D-Bus
816
869
        """
817
870
        for cls in  self.__class__.__mro__:
818
871
            for name, value in (inspect.getmembers
819
 
                                (cls, self._is_dbus_property)):
 
872
                                (cls,
 
873
                                 self._is_dbus_thing("property"))):
820
874
                if (value._dbus_name == property_name
821
875
                    and value._dbus_interface == interface_name):
822
876
                    return value.__get__(self)
864
918
        Note: Will not include properties with access="write".
865
919
        """
866
920
        properties = {}
867
 
        for name, prop in self._get_all_dbus_properties():
 
921
        for name, prop in self._get_all_dbus_things("property"):
868
922
            if (interface_name
869
923
                and interface_name != prop._dbus_interface):
870
924
                # Interface non-empty but did not match
885
939
                         path_keyword='object_path',
886
940
                         connection_keyword='connection')
887
941
    def Introspect(self, object_path, connection):
888
 
        """Standard D-Bus method, overloaded to insert property tags.
 
942
        """Overloading of standard D-Bus method.
 
943
        
 
944
        Inserts property tags and interface annotation tags.
889
945
        """
890
946
        xmlstring = dbus.service.Object.Introspect(self, object_path,
891
947
                                                   connection)
898
954
                e.setAttribute("access", prop._dbus_access)
899
955
                return e
900
956
            for if_tag in document.getElementsByTagName("interface"):
 
957
                # Add property tags
901
958
                for tag in (make_tag(document, name, prop)
902
959
                            for name, prop
903
 
                            in self._get_all_dbus_properties()
 
960
                            in self._get_all_dbus_things("property")
904
961
                            if prop._dbus_interface
905
962
                            == if_tag.getAttribute("name")):
906
963
                    if_tag.appendChild(tag)
 
964
                # Add annotation tags
 
965
                for typ in ("method", "signal", "property"):
 
966
                    for tag in if_tag.getElementsByTagName(typ):
 
967
                        annots = dict()
 
968
                        for name, prop in (self.
 
969
                                           _get_all_dbus_things(typ)):
 
970
                            if (name == tag.getAttribute("name")
 
971
                                and prop._dbus_interface
 
972
                                == if_tag.getAttribute("name")):
 
973
                                annots.update(getattr
 
974
                                              (prop,
 
975
                                               "_dbus_annotations",
 
976
                                               {}))
 
977
                        for name, value in annots.iteritems():
 
978
                            ann_tag = document.createElement(
 
979
                                "annotation")
 
980
                            ann_tag.setAttribute("name", name)
 
981
                            ann_tag.setAttribute("value", value)
 
982
                            tag.appendChild(ann_tag)
 
983
                # Add interface annotation tags
 
984
                for annotation, value in dict(
 
985
                    itertools.chain.from_iterable(
 
986
                        annotations().iteritems()
 
987
                        for name, annotations in
 
988
                        self._get_all_dbus_things("interface")
 
989
                        if name == if_tag.getAttribute("name")
 
990
                        )).iteritems():
 
991
                    ann_tag = document.createElement("annotation")
 
992
                    ann_tag.setAttribute("name", annotation)
 
993
                    ann_tag.setAttribute("value", value)
 
994
                    if_tag.appendChild(ann_tag)
907
995
                # Add the names to the return values for the
908
996
                # "org.freedesktop.DBus.Properties" methods
909
997
                if (if_tag.getAttribute("name")
924
1012
        except (AttributeError, xml.dom.DOMException,
925
1013
                xml.parsers.expat.ExpatError) as error:
926
1014
            logger.error("Failed to override Introspection method",
927
 
                         error)
 
1015
                         exc_info=error)
928
1016
        return xmlstring
929
1017
 
930
1018
 
931
 
def datetime_to_dbus (dt, variant_level=0):
 
1019
def datetime_to_dbus(dt, variant_level=0):
932
1020
    """Convert a UTC datetime.datetime() to a D-Bus type."""
933
1021
    if dt is None:
934
1022
        return dbus.String("", variant_level = variant_level)
936
1024
                       variant_level=variant_level)
937
1025
 
938
1026
 
939
 
class AlternateDBusNamesMetaclass(DBusObjectWithProperties
940
 
                                  .__metaclass__):
941
 
    """Applied to an empty subclass of a D-Bus object, this metaclass
942
 
    will add additional D-Bus attributes matching a certain pattern.
 
1027
def alternate_dbus_interfaces(alt_interface_names, deprecate=True):
 
1028
    """A class decorator; applied to a subclass of
 
1029
    dbus.service.Object, it will add alternate D-Bus attributes with
 
1030
    interface names according to the "alt_interface_names" mapping.
 
1031
    Usage:
 
1032
    
 
1033
    @alternate_dbus_interfaces({"org.example.Interface":
 
1034
                                    "net.example.AlternateInterface"})
 
1035
    class SampleDBusObject(dbus.service.Object):
 
1036
        @dbus.service.method("org.example.Interface")
 
1037
        def SampleDBusMethod():
 
1038
            pass
 
1039
    
 
1040
    The above "SampleDBusMethod" on "SampleDBusObject" will be
 
1041
    reachable via two interfaces: "org.example.Interface" and
 
1042
    "net.example.AlternateInterface", the latter of which will have
 
1043
    its D-Bus annotation "org.freedesktop.DBus.Deprecated" set to
 
1044
    "true", unless "deprecate" is passed with a False value.
 
1045
    
 
1046
    This works for methods and signals, and also for D-Bus properties
 
1047
    (from DBusObjectWithProperties) and interfaces (from the
 
1048
    dbus_interface_annotations decorator).
943
1049
    """
944
 
    def __new__(mcs, name, bases, attr):
945
 
        # Go through all the base classes which could have D-Bus
946
 
        # methods, signals, or properties in them
947
 
        for base in (b for b in bases
948
 
                     if issubclass(b, dbus.service.Object)):
949
 
            # Go though all attributes of the base class
950
 
            for attrname, attribute in inspect.getmembers(base):
 
1050
    def wrapper(cls):
 
1051
        for orig_interface_name, alt_interface_name in (
 
1052
            alt_interface_names.iteritems()):
 
1053
            attr = {}
 
1054
            interface_names = set()
 
1055
            # Go though all attributes of the class
 
1056
            for attrname, attribute in inspect.getmembers(cls):
951
1057
                # Ignore non-D-Bus attributes, and D-Bus attributes
952
1058
                # with the wrong interface name
953
1059
                if (not hasattr(attribute, "_dbus_interface")
954
1060
                    or not attribute._dbus_interface
955
 
                    .startswith("se.recompile.Mandos")):
 
1061
                    .startswith(orig_interface_name)):
956
1062
                    continue
957
1063
                # Create an alternate D-Bus interface name based on
958
1064
                # the current name
959
1065
                alt_interface = (attribute._dbus_interface
960
 
                                 .replace("se.recompile.Mandos",
961
 
                                          "se.bsnet.fukt.Mandos"))
 
1066
                                 .replace(orig_interface_name,
 
1067
                                          alt_interface_name))
 
1068
                interface_names.add(alt_interface)
962
1069
                # Is this a D-Bus signal?
963
1070
                if getattr(attribute, "_dbus_is_signal", False):
964
1071
                    # Extract the original non-method function by
979
1086
                                nonmethod_func.func_name,
980
1087
                                nonmethod_func.func_defaults,
981
1088
                                nonmethod_func.func_closure)))
 
1089
                    # Copy annotations, if any
 
1090
                    try:
 
1091
                        new_function._dbus_annotations = (
 
1092
                            dict(attribute._dbus_annotations))
 
1093
                    except AttributeError:
 
1094
                        pass
982
1095
                    # Define a creator of a function to call both the
983
 
                    # old and new functions, so both the old and new
984
 
                    # signals gets sent when the function is called
 
1096
                    # original and alternate functions, so both the
 
1097
                    # original and alternate signals gets sent when
 
1098
                    # the function is called
985
1099
                    def fixscope(func1, func2):
986
1100
                        """This function is a scope container to pass
987
1101
                        func1 and func2 to the "call_both" function
994
1108
                        return call_both
995
1109
                    # Create the "call_both" function and add it to
996
1110
                    # the class
997
 
                    attr[attrname] = fixscope(attribute,
998
 
                                              new_function)
 
1111
                    attr[attrname] = fixscope(attribute, new_function)
999
1112
                # Is this a D-Bus method?
1000
1113
                elif getattr(attribute, "_dbus_is_method", False):
1001
1114
                    # Create a new, but exactly alike, function
1012
1125
                                        attribute.func_name,
1013
1126
                                        attribute.func_defaults,
1014
1127
                                        attribute.func_closure)))
 
1128
                    # Copy annotations, if any
 
1129
                    try:
 
1130
                        attr[attrname]._dbus_annotations = (
 
1131
                            dict(attribute._dbus_annotations))
 
1132
                    except AttributeError:
 
1133
                        pass
1015
1134
                # Is this a D-Bus property?
1016
1135
                elif getattr(attribute, "_dbus_is_property", False):
1017
1136
                    # Create a new, but exactly alike, function
1031
1150
                                        attribute.func_name,
1032
1151
                                        attribute.func_defaults,
1033
1152
                                        attribute.func_closure)))
1034
 
        return type.__new__(mcs, name, bases, attr)
1035
 
 
1036
 
 
 
1153
                    # Copy annotations, if any
 
1154
                    try:
 
1155
                        attr[attrname]._dbus_annotations = (
 
1156
                            dict(attribute._dbus_annotations))
 
1157
                    except AttributeError:
 
1158
                        pass
 
1159
                # Is this a D-Bus interface?
 
1160
                elif getattr(attribute, "_dbus_is_interface", False):
 
1161
                    # Create a new, but exactly alike, function
 
1162
                    # object.  Decorate it to be a new D-Bus interface
 
1163
                    # with the alternate D-Bus interface name.  Add it
 
1164
                    # to the class.
 
1165
                    attr[attrname] = (dbus_interface_annotations
 
1166
                                      (alt_interface)
 
1167
                                      (types.FunctionType
 
1168
                                       (attribute.func_code,
 
1169
                                        attribute.func_globals,
 
1170
                                        attribute.func_name,
 
1171
                                        attribute.func_defaults,
 
1172
                                        attribute.func_closure)))
 
1173
            if deprecate:
 
1174
                # Deprecate all alternate interfaces
 
1175
                iname="_AlternateDBusNames_interface_annotation{0}"
 
1176
                for interface_name in interface_names:
 
1177
                    @dbus_interface_annotations(interface_name)
 
1178
                    def func(self):
 
1179
                        return { "org.freedesktop.DBus.Deprecated":
 
1180
                                     "true" }
 
1181
                    # Find an unused name
 
1182
                    for aname in (iname.format(i)
 
1183
                                  for i in itertools.count()):
 
1184
                        if aname not in attr:
 
1185
                            attr[aname] = func
 
1186
                            break
 
1187
            if interface_names:
 
1188
                # Replace the class with a new subclass of it with
 
1189
                # methods, signals, etc. as created above.
 
1190
                cls = type(b"{0}Alternate".format(cls.__name__),
 
1191
                           (cls,), attr)
 
1192
        return cls
 
1193
    return wrapper
 
1194
 
 
1195
 
 
1196
@alternate_dbus_interfaces({"se.recompile.Mandos":
 
1197
                                "se.bsnet.fukt.Mandos"})
1037
1198
class ClientDBus(Client, DBusObjectWithProperties):
1038
1199
    """A Client class using D-Bus
1039
1200
    
1059
1220
                                 ("/clients/" + client_object_name))
1060
1221
        DBusObjectWithProperties.__init__(self, self.bus,
1061
1222
                                          self.dbus_object_path)
1062
 
        
 
1223
    
1063
1224
    def notifychangeproperty(transform_func,
1064
1225
                             dbus_name, type_func=lambda x: x,
1065
1226
                             variant_level=1):
1088
1249
        
1089
1250
        return property(lambda self: getattr(self, attrname), setter)
1090
1251
    
1091
 
    
1092
1252
    expires = notifychangeproperty(datetime_to_dbus, "Expires")
1093
1253
    approvals_pending = notifychangeproperty(dbus.Boolean,
1094
1254
                                             "ApprovalPending",
1176
1336
        return False
1177
1337
    
1178
1338
    def approve(self, value=True):
1179
 
        self.send_changedstate()
1180
1339
        self.approved = value
1181
1340
        gobject.timeout_add(timedelta_to_milliseconds
1182
1341
                            (self.approval_duration),
1183
1342
                            self._reset_approved)
1184
 
    
 
1343
        self.send_changedstate()
1185
1344
    
1186
1345
    ## D-Bus methods, signals & properties
1187
1346
    _interface = "se.recompile.Mandos.Client"
1188
1347
    
 
1348
    ## Interfaces
 
1349
    
 
1350
    @dbus_interface_annotations(_interface)
 
1351
    def _foo(self):
 
1352
        return { "org.freedesktop.DBus.Property.EmitsChangedSignal":
 
1353
                     "false"}
 
1354
    
1189
1355
    ## Signals
1190
1356
    
1191
1357
    # CheckerCompleted - signal
1364
1530
    def Timeout_dbus_property(self, value=None):
1365
1531
        if value is None:       # get
1366
1532
            return dbus.UInt64(self.timeout_milliseconds())
 
1533
        old_timeout = self.timeout
1367
1534
        self.timeout = datetime.timedelta(0, 0, 0, value)
1368
 
        # Reschedule timeout
 
1535
        # Reschedule disabling
1369
1536
        if self.enabled:
1370
1537
            now = datetime.datetime.utcnow()
1371
 
            time_to_die = timedelta_to_milliseconds(
1372
 
                (self.last_checked_ok + self.timeout) - now)
1373
 
            if time_to_die <= 0:
 
1538
            self.expires += self.timeout - old_timeout
 
1539
            if self.expires <= now:
1374
1540
                # The timeout has passed
1375
1541
                self.disable()
1376
1542
            else:
1377
 
                self.expires = (now +
1378
 
                                datetime.timedelta(milliseconds =
1379
 
                                                   time_to_die))
1380
1543
                if (getattr(self, "disable_initiator_tag", None)
1381
1544
                    is None):
1382
1545
                    return
1383
1546
                gobject.source_remove(self.disable_initiator_tag)
1384
 
                self.disable_initiator_tag = (gobject.timeout_add
1385
 
                                              (time_to_die,
1386
 
                                               self.disable))
 
1547
                self.disable_initiator_tag = (
 
1548
                    gobject.timeout_add(
 
1549
                        timedelta_to_milliseconds(self.expires - now),
 
1550
                        self.disable))
1387
1551
    
1388
1552
    # ExtendedTimeout - property
1389
1553
    @dbus_service_property(_interface, signature="t",
1468
1632
        self._pipe.send(('setattr', name, value))
1469
1633
 
1470
1634
 
1471
 
class ClientDBusTransitional(ClientDBus):
1472
 
    __metaclass__ = AlternateDBusNamesMetaclass
1473
 
 
1474
 
 
1475
1635
class ClientHandler(socketserver.BaseRequestHandler, object):
1476
1636
    """A class to handle client connections.
1477
1637
    
1581
1741
                    #wait until timeout or approved
1582
1742
                    time = datetime.datetime.now()
1583
1743
                    client.changedstate.acquire()
1584
 
                    (client.changedstate.wait
1585
 
                     (float(client.timedelta_to_milliseconds(delay)
1586
 
                            / 1000)))
 
1744
                    client.changedstate.wait(
 
1745
                        float(timedelta_to_milliseconds(delay)
 
1746
                              / 1000))
1587
1747
                    client.changedstate.release()
1588
1748
                    time2 = datetime.datetime.now()
1589
1749
                    if (time2 - time) >= delay:
1605
1765
                    try:
1606
1766
                        sent = session.send(client.secret[sent_size:])
1607
1767
                    except gnutls.errors.GNUTLSError as error:
1608
 
                        logger.warning("gnutls send failed")
 
1768
                        logger.warning("gnutls send failed",
 
1769
                                       exc_info=error)
1609
1770
                        return
1610
1771
                    logger.debug("Sent: %d, remaining: %d",
1611
1772
                                 sent, len(client.secret)
1625
1786
                try:
1626
1787
                    session.bye()
1627
1788
                except gnutls.errors.GNUTLSError as error:
1628
 
                    logger.warning("GnuTLS bye failed")
 
1789
                    logger.warning("GnuTLS bye failed",
 
1790
                                   exc_info=error)
1629
1791
    
1630
1792
    @staticmethod
1631
1793
    def peer_certificate(session):
1703
1865
    def process_request(self, request, address):
1704
1866
        """Start a new process to process the request."""
1705
1867
        proc = multiprocessing.Process(target = self.sub_process_main,
1706
 
                                       args = (request,
1707
 
                                               address))
 
1868
                                       args = (request, address))
1708
1869
        proc.start()
1709
1870
        return proc
1710
1871
 
1738
1899
        use_ipv6:       Boolean; to use IPv6 or not
1739
1900
    """
1740
1901
    def __init__(self, server_address, RequestHandlerClass,
1741
 
                 interface=None, use_ipv6=True):
 
1902
                 interface=None, use_ipv6=True, socketfd=None):
 
1903
        """If socketfd is set, use that file descriptor instead of
 
1904
        creating a new one with socket.socket().
 
1905
        """
1742
1906
        self.interface = interface
1743
1907
        if use_ipv6:
1744
1908
            self.address_family = socket.AF_INET6
 
1909
        if socketfd is not None:
 
1910
            # Save the file descriptor
 
1911
            self.socketfd = socketfd
 
1912
            # Save the original socket.socket() function
 
1913
            self.socket_socket = socket.socket
 
1914
            # To implement --socket, we monkey patch socket.socket.
 
1915
            # 
 
1916
            # (When socketserver.TCPServer is a new-style class, we
 
1917
            # could make self.socket into a property instead of monkey
 
1918
            # patching socket.socket.)
 
1919
            # 
 
1920
            # Create a one-time-only replacement for socket.socket()
 
1921
            @functools.wraps(socket.socket)
 
1922
            def socket_wrapper(*args, **kwargs):
 
1923
                # Restore original function so subsequent calls are
 
1924
                # not affected.
 
1925
                socket.socket = self.socket_socket
 
1926
                del self.socket_socket
 
1927
                # This time only, return a new socket object from the
 
1928
                # saved file descriptor.
 
1929
                return socket.fromfd(self.socketfd, *args, **kwargs)
 
1930
            # Replace socket.socket() function with wrapper
 
1931
            socket.socket = socket_wrapper
 
1932
        # The socketserver.TCPServer.__init__ will call
 
1933
        # socket.socket(), which might be our replacement,
 
1934
        # socket_wrapper(), if socketfd was set.
1745
1935
        socketserver.TCPServer.__init__(self, server_address,
1746
1936
                                        RequestHandlerClass)
 
1937
    
1747
1938
    def server_bind(self):
1748
1939
        """This overrides the normal server_bind() function
1749
1940
        to bind to an interface if one was specified, and also NOT to
1757
1948
                try:
1758
1949
                    self.socket.setsockopt(socket.SOL_SOCKET,
1759
1950
                                           SO_BINDTODEVICE,
1760
 
                                           str(self.interface
1761
 
                                               + '\0'))
 
1951
                                           str(self.interface + '\0'))
1762
1952
                except socket.error as error:
1763
 
                    if error[0] == errno.EPERM:
1764
 
                        logger.error("No permission to"
1765
 
                                     " bind to interface %s",
1766
 
                                     self.interface)
1767
 
                    elif error[0] == errno.ENOPROTOOPT:
 
1953
                    if error.errno == errno.EPERM:
 
1954
                        logger.error("No permission to bind to"
 
1955
                                     " interface %s", self.interface)
 
1956
                    elif error.errno == errno.ENOPROTOOPT:
1768
1957
                        logger.error("SO_BINDTODEVICE not available;"
1769
1958
                                     " cannot bind to interface %s",
1770
1959
                                     self.interface)
 
1960
                    elif error.errno == errno.ENODEV:
 
1961
                        logger.error("Interface %s does not exist,"
 
1962
                                     " cannot bind", self.interface)
1771
1963
                    else:
1772
1964
                        raise
1773
1965
        # Only bind(2) the socket if we really need to.
1803
1995
    """
1804
1996
    def __init__(self, server_address, RequestHandlerClass,
1805
1997
                 interface=None, use_ipv6=True, clients=None,
1806
 
                 gnutls_priority=None, use_dbus=True):
 
1998
                 gnutls_priority=None, use_dbus=True, socketfd=None):
1807
1999
        self.enabled = False
1808
2000
        self.clients = clients
1809
2001
        if self.clients is None:
1813
2005
        IPv6_TCPServer.__init__(self, server_address,
1814
2006
                                RequestHandlerClass,
1815
2007
                                interface = interface,
1816
 
                                use_ipv6 = use_ipv6)
 
2008
                                use_ipv6 = use_ipv6,
 
2009
                                socketfd = socketfd)
1817
2010
    def server_activate(self):
1818
2011
        if self.enabled:
1819
2012
            return socketserver.TCPServer.server_activate(self)
1832
2025
    
1833
2026
    def handle_ipc(self, source, condition, parent_pipe=None,
1834
2027
                   proc = None, client_object=None):
1835
 
        condition_names = {
1836
 
            gobject.IO_IN: "IN",   # There is data to read.
1837
 
            gobject.IO_OUT: "OUT", # Data can be written (without
1838
 
                                    # blocking).
1839
 
            gobject.IO_PRI: "PRI", # There is urgent data to read.
1840
 
            gobject.IO_ERR: "ERR", # Error condition.
1841
 
            gobject.IO_HUP: "HUP"  # Hung up (the connection has been
1842
 
                                    # broken, usually for pipes and
1843
 
                                    # sockets).
1844
 
            }
1845
 
        conditions_string = ' | '.join(name
1846
 
                                       for cond, name in
1847
 
                                       condition_names.iteritems()
1848
 
                                       if cond & condition)
1849
2028
        # error, or the other end of multiprocessing.Pipe has closed
1850
 
        if condition & (gobject.IO_ERR | condition & gobject.IO_HUP):
 
2029
        if condition & (gobject.IO_ERR | gobject.IO_HUP):
1851
2030
            # Wait for other process to exit
1852
2031
            proc.join()
1853
2032
            return False
1943
2122
            elif suffix == "w":
1944
2123
                delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
1945
2124
            else:
1946
 
                raise ValueError("Unknown suffix %r" % suffix)
 
2125
                raise ValueError("Unknown suffix {0!r}"
 
2126
                                 .format(suffix))
1947
2127
        except (ValueError, IndexError) as e:
1948
2128
            raise ValueError(*(e.args))
1949
2129
        timevalue += delta
1963
2143
        sys.exit()
1964
2144
    if not noclose:
1965
2145
        # Close all standard open file descriptors
1966
 
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
 
2146
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
1967
2147
        if not stat.S_ISCHR(os.fstat(null).st_mode):
1968
2148
            raise OSError(errno.ENODEV,
1969
 
                          "%s not a character device"
1970
 
                          % os.path.devnull)
 
2149
                          "{0} not a character device"
 
2150
                          .format(os.devnull))
1971
2151
        os.dup2(null, sys.stdin.fileno())
1972
2152
        os.dup2(null, sys.stdout.fileno())
1973
2153
        os.dup2(null, sys.stderr.fileno())
1982
2162
    
1983
2163
    parser = argparse.ArgumentParser()
1984
2164
    parser.add_argument("-v", "--version", action="version",
1985
 
                        version = "%%(prog)s %s" % version,
 
2165
                        version = "%(prog)s {0}".format(version),
1986
2166
                        help="show version number and exit")
1987
2167
    parser.add_argument("-i", "--interface", metavar="IF",
1988
2168
                        help="Bind to interface IF")
2013
2193
    parser.add_argument("--no-restore", action="store_false",
2014
2194
                        dest="restore", help="Do not restore stored"
2015
2195
                        " state")
 
2196
    parser.add_argument("--socket", type=int,
 
2197
                        help="Specify a file descriptor to a network"
 
2198
                        " socket to use instead of creating one")
2016
2199
    parser.add_argument("--statedir", metavar="DIR",
2017
2200
                        help="Directory to save/restore state in")
 
2201
    parser.add_argument("--foreground", action="store_true",
 
2202
                        help="Run in foreground")
2018
2203
    
2019
2204
    options = parser.parse_args()
2020
2205
    
2035
2220
                        "use_ipv6": "True",
2036
2221
                        "debuglevel": "",
2037
2222
                        "restore": "True",
2038
 
                        "statedir": "/var/lib/mandos"
 
2223
                        "socket": "",
 
2224
                        "statedir": "/var/lib/mandos",
 
2225
                        "foreground": "False",
2039
2226
                        }
2040
2227
    
2041
2228
    # Parse config file for server-global settings
2046
2233
    # Convert the SafeConfigParser object to a dict
2047
2234
    server_settings = server_config.defaults()
2048
2235
    # Use the appropriate methods on the non-string config options
2049
 
    for option in ("debug", "use_dbus", "use_ipv6"):
 
2236
    for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
2050
2237
        server_settings[option] = server_config.getboolean("DEFAULT",
2051
2238
                                                           option)
2052
2239
    if server_settings["port"]:
2053
2240
        server_settings["port"] = server_config.getint("DEFAULT",
2054
2241
                                                       "port")
 
2242
    if server_settings["socket"]:
 
2243
        server_settings["socket"] = server_config.getint("DEFAULT",
 
2244
                                                         "socket")
 
2245
        # Later, stdin will, and stdout and stderr might, be dup'ed
 
2246
        # over with an opened os.devnull.  But we don't want this to
 
2247
        # happen with a supplied network socket.
 
2248
        if 0 <= server_settings["socket"] <= 2:
 
2249
            server_settings["socket"] = os.dup(server_settings
 
2250
                                               ["socket"])
2055
2251
    del server_config
2056
2252
    
2057
2253
    # Override the settings from the config file with command line
2059
2255
    for option in ("interface", "address", "port", "debug",
2060
2256
                   "priority", "servicename", "configdir",
2061
2257
                   "use_dbus", "use_ipv6", "debuglevel", "restore",
2062
 
                   "statedir"):
 
2258
                   "statedir", "socket", "foreground"):
2063
2259
        value = getattr(options, option)
2064
2260
        if value is not None:
2065
2261
            server_settings[option] = value
2068
2264
    for option in server_settings.keys():
2069
2265
        if type(server_settings[option]) is str:
2070
2266
            server_settings[option] = unicode(server_settings[option])
 
2267
    # Debug implies foreground
 
2268
    if server_settings["debug"]:
 
2269
        server_settings["foreground"] = True
2071
2270
    # Now we have our good server settings in "server_settings"
2072
2271
    
2073
2272
    ##################################################################
2079
2278
    use_ipv6 = server_settings["use_ipv6"]
2080
2279
    stored_state_path = os.path.join(server_settings["statedir"],
2081
2280
                                     stored_state_file)
 
2281
    foreground = server_settings["foreground"]
2082
2282
    
2083
2283
    if debug:
2084
2284
        initlogger(debug, logging.DEBUG)
2091
2291
    
2092
2292
    if server_settings["servicename"] != "Mandos":
2093
2293
        syslogger.setFormatter(logging.Formatter
2094
 
                               ('Mandos (%s) [%%(process)d]:'
2095
 
                                ' %%(levelname)s: %%(message)s'
2096
 
                                % server_settings["servicename"]))
 
2294
                               ('Mandos ({0}) [%(process)d]:'
 
2295
                                ' %(levelname)s: %(message)s'
 
2296
                                .format(server_settings
 
2297
                                        ["servicename"])))
2097
2298
    
2098
2299
    # Parse config file with clients
2099
2300
    client_config = configparser.SafeConfigParser(Client
2112
2313
                              use_ipv6=use_ipv6,
2113
2314
                              gnutls_priority=
2114
2315
                              server_settings["priority"],
2115
 
                              use_dbus=use_dbus)
2116
 
    if not debug:
 
2316
                              use_dbus=use_dbus,
 
2317
                              socketfd=(server_settings["socket"]
 
2318
                                        or None))
 
2319
    if not foreground:
2117
2320
        pidfilename = "/var/run/mandos.pid"
 
2321
        pidfile = None
2118
2322
        try:
2119
2323
            pidfile = open(pidfilename, "w")
2120
 
        except IOError:
2121
 
            logger.error("Could not open file %r", pidfilename)
 
2324
        except IOError as e:
 
2325
            logger.error("Could not open file %r", pidfilename,
 
2326
                         exc_info=e)
2122
2327
    
2123
 
    try:
2124
 
        uid = pwd.getpwnam("_mandos").pw_uid
2125
 
        gid = pwd.getpwnam("_mandos").pw_gid
2126
 
    except KeyError:
 
2328
    for name in ("_mandos", "mandos", "nobody"):
2127
2329
        try:
2128
 
            uid = pwd.getpwnam("mandos").pw_uid
2129
 
            gid = pwd.getpwnam("mandos").pw_gid
 
2330
            uid = pwd.getpwnam(name).pw_uid
 
2331
            gid = pwd.getpwnam(name).pw_gid
 
2332
            break
2130
2333
        except KeyError:
2131
 
            try:
2132
 
                uid = pwd.getpwnam("nobody").pw_uid
2133
 
                gid = pwd.getpwnam("nobody").pw_gid
2134
 
            except KeyError:
2135
 
                uid = 65534
2136
 
                gid = 65534
 
2334
            continue
 
2335
    else:
 
2336
        uid = 65534
 
2337
        gid = 65534
2137
2338
    try:
2138
2339
        os.setgid(gid)
2139
2340
        os.setuid(uid)
2140
2341
    except OSError as error:
2141
 
        if error[0] != errno.EPERM:
 
2342
        if error.errno != errno.EPERM:
2142
2343
            raise error
2143
2344
    
2144
2345
    if debug:
2156
2357
         .gnutls_global_set_log_function(debug_gnutls))
2157
2358
        
2158
2359
        # Redirect stdin so all checkers get /dev/null
2159
 
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
 
2360
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
2160
2361
        os.dup2(null, sys.stdin.fileno())
2161
2362
        if null > 2:
2162
2363
            os.close(null)
2163
2364
    
2164
2365
    # Need to fork before connecting to D-Bus
2165
 
    if not debug:
 
2366
    if not foreground:
2166
2367
        # Close all input and output, do double fork, etc.
2167
2368
        daemon()
2168
2369
    
 
2370
    # multiprocessing will use threads, so before we use gobject we
 
2371
    # need to inform gobject that threads will be used.
2169
2372
    gobject.threads_init()
2170
2373
    
2171
2374
    global main_loop
2172
2375
    # From the Avahi example code
2173
 
    DBusGMainLoop(set_as_default=True )
 
2376
    DBusGMainLoop(set_as_default=True)
2174
2377
    main_loop = gobject.MainLoop()
2175
2378
    bus = dbus.SystemBus()
2176
2379
    # End of Avahi example code
2182
2385
                            ("se.bsnet.fukt.Mandos", bus,
2183
2386
                             do_not_queue=True))
2184
2387
        except dbus.exceptions.NameExistsException as e:
2185
 
            logger.error(unicode(e) + ", disabling D-Bus")
 
2388
            logger.error("Disabling D-Bus:", exc_info=e)
2186
2389
            use_dbus = False
2187
2390
            server_settings["use_dbus"] = False
2188
2391
            tcp_server.use_dbus = False
2200
2403
    
2201
2404
    client_class = Client
2202
2405
    if use_dbus:
2203
 
        client_class = functools.partial(ClientDBusTransitional,
2204
 
                                         bus = bus)
 
2406
        client_class = functools.partial(ClientDBus, bus = bus)
2205
2407
    
2206
2408
    client_settings = Client.config_parser(client_config)
2207
2409
    old_client_settings = {}
2215
2417
                                                     (stored_state))
2216
2418
            os.remove(stored_state_path)
2217
2419
        except IOError as e:
2218
 
            logger.warning("Could not load persistent state: {0}"
2219
 
                           .format(e))
2220
 
            if e.errno != errno.ENOENT:
 
2420
            if e.errno == errno.ENOENT:
 
2421
                logger.warning("Could not load persistent state: {0}"
 
2422
                                .format(os.strerror(e.errno)))
 
2423
            else:
 
2424
                logger.critical("Could not load persistent state:",
 
2425
                                exc_info=e)
2221
2426
                raise
2222
2427
        except EOFError as e:
2223
2428
            logger.warning("Could not load persistent state: "
2224
 
                           "EOFError: {0}".format(e))
 
2429
                           "EOFError:", exc_info=e)
2225
2430
    
2226
2431
    with PGPEngine() as pgp:
2227
2432
        for client_name, client in clients_data.iteritems():
2280
2485
                             .format(client_name))
2281
2486
                client["secret"] = (
2282
2487
                    client_settings[client_name]["secret"])
2283
 
 
2284
2488
    
2285
2489
    # Add/remove clients based on new changes made to config
2286
2490
    for client_name in (set(old_client_settings)
2289
2493
    for client_name in (set(client_settings)
2290
2494
                        - set(old_client_settings)):
2291
2495
        clients_data[client_name] = client_settings[client_name]
2292
 
 
 
2496
    
2293
2497
    # Create all client objects
2294
2498
    for client_name, client in clients_data.iteritems():
2295
2499
        tcp_server.clients[client_name] = client_class(
2297
2501
    
2298
2502
    if not tcp_server.clients:
2299
2503
        logger.warning("No clients defined")
2300
 
        
2301
 
    if not debug:
2302
 
        try:
2303
 
            with pidfile:
2304
 
                pid = os.getpid()
2305
 
                pidfile.write(str(pid) + "\n".encode("utf-8"))
2306
 
            del pidfile
2307
 
        except IOError:
2308
 
            logger.error("Could not write to file %r with PID %d",
2309
 
                         pidfilename, pid)
2310
 
        except NameError:
2311
 
            # "pidfile" was never created
2312
 
            pass
 
2504
    
 
2505
    if not foreground:
 
2506
        if pidfile is not None:
 
2507
            try:
 
2508
                with pidfile:
 
2509
                    pid = os.getpid()
 
2510
                    pidfile.write(str(pid) + "\n".encode("utf-8"))
 
2511
            except IOError:
 
2512
                logger.error("Could not write to file %r with PID %d",
 
2513
                             pidfilename, pid)
 
2514
        del pidfile
2313
2515
        del pidfilename
2314
 
        signal.signal(signal.SIGINT, signal.SIG_IGN)
2315
2516
    
2316
2517
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
2317
2518
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
2318
2519
    
2319
2520
    if use_dbus:
2320
 
        class MandosDBusService(dbus.service.Object):
 
2521
        @alternate_dbus_interfaces({"se.recompile.Mandos":
 
2522
                                        "se.bsnet.fukt.Mandos"})
 
2523
        class MandosDBusService(DBusObjectWithProperties):
2321
2524
            """A D-Bus proxy object"""
2322
2525
            def __init__(self):
2323
2526
                dbus.service.Object.__init__(self, bus, "/")
2324
2527
            _interface = "se.recompile.Mandos"
2325
2528
            
 
2529
            @dbus_interface_annotations(_interface)
 
2530
            def _foo(self):
 
2531
                return { "org.freedesktop.DBus.Property"
 
2532
                         ".EmitsChangedSignal":
 
2533
                             "false"}
 
2534
            
2326
2535
            @dbus.service.signal(_interface, signature="o")
2327
2536
            def ClientAdded(self, objpath):
2328
2537
                "D-Bus signal"
2370
2579
            
2371
2580
            del _interface
2372
2581
        
2373
 
        class MandosDBusServiceTransitional(MandosDBusService):
2374
 
            __metaclass__ = AlternateDBusNamesMetaclass
2375
 
        mandos_dbus_service = MandosDBusServiceTransitional()
 
2582
        mandos_dbus_service = MandosDBusService()
2376
2583
    
2377
2584
    def cleanup():
2378
2585
        "Cleanup function; run on exit"
2411
2618
                del client_settings[client.name]["secret"]
2412
2619
        
2413
2620
        try:
2414
 
            tempfd, tempname = tempfile.mkstemp(suffix=".pickle",
2415
 
                                                prefix="clients-",
2416
 
                                                dir=os.path.dirname
2417
 
                                                (stored_state_path))
2418
 
            with os.fdopen(tempfd, "wb") as stored_state:
 
2621
            with (tempfile.NamedTemporaryFile
 
2622
                  (mode='wb', suffix=".pickle", prefix='clients-',
 
2623
                   dir=os.path.dirname(stored_state_path),
 
2624
                   delete=False)) as stored_state:
2419
2625
                pickle.dump((clients, client_settings), stored_state)
 
2626
                tempname=stored_state.name
2420
2627
            os.rename(tempname, stored_state_path)
2421
2628
        except (IOError, OSError) as e:
2422
 
            logger.warning("Could not save persistent state: {0}"
2423
 
                           .format(e))
2424
2629
            if not debug:
2425
2630
                try:
2426
2631
                    os.remove(tempname)
2427
2632
                except NameError:
2428
2633
                    pass
2429
 
            if e.errno not in set((errno.ENOENT, errno.EACCES,
2430
 
                                   errno.EEXIST)):
 
2634
            if e.errno in (errno.ENOENT, errno.EACCES, errno.EEXIST):
 
2635
                logger.warning("Could not save persistent state: {0}"
 
2636
                               .format(os.strerror(e.errno)))
 
2637
            else:
 
2638
                logger.warning("Could not save persistent state:",
 
2639
                               exc_info=e)
2431
2640
                raise e
2432
2641
        
2433
2642
        # Delete all clients, and settings from config
2461
2670
    service.port = tcp_server.socket.getsockname()[1]
2462
2671
    if use_ipv6:
2463
2672
        logger.info("Now listening on address %r, port %d,"
2464
 
                    " flowinfo %d, scope_id %d"
2465
 
                    % tcp_server.socket.getsockname())
 
2673
                    " flowinfo %d, scope_id %d",
 
2674
                    *tcp_server.socket.getsockname())
2466
2675
    else:                       # IPv4
2467
 
        logger.info("Now listening on address %r, port %d"
2468
 
                    % tcp_server.socket.getsockname())
 
2676
        logger.info("Now listening on address %r, port %d",
 
2677
                    *tcp_server.socket.getsockname())
2469
2678
    
2470
2679
    #service.interface = tcp_server.socket.getsockname()[3]
2471
2680
    
2474
2683
        try:
2475
2684
            service.activate()
2476
2685
        except dbus.exceptions.DBusException as error:
2477
 
            logger.critical("DBusException: %s", error)
 
2686
            logger.critical("D-Bus Exception", exc_info=error)
2478
2687
            cleanup()
2479
2688
            sys.exit(1)
2480
2689
        # End of Avahi example code
2487
2696
        logger.debug("Starting main loop")
2488
2697
        main_loop.run()
2489
2698
    except AvahiError as error:
2490
 
        logger.critical("AvahiError: %s", error)
 
2699
        logger.critical("Avahi Error", exc_info=error)
2491
2700
        cleanup()
2492
2701
        sys.exit(1)
2493
2702
    except KeyboardInterrupt: