/mandos/trunk

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

« back to all changes in this revision

Viewing changes to mandos

  • Committer: Teddy Hogeborn
  • Date: 2014-08-09 13:12:55 UTC
  • Revision ID: teddy@recompile.se-20140809131255-lp31j98u2pl0xpe6
mandos: Stop using str() and remove unnecessary unicode() calls.

* mandos (if_nametoindex): Use "bytes" literal instead of str().
  (initlogger): Use a unicode string for log device.
  (AvahiError.__unicode__): Removed.
  (DBusPropertyException.__unicode__): - '' -
  (ClientDBus.Secret_dbus_property): Use bytes() instead of str().
  (IPv6_TCPServer.server_bind): Use .encode() instead of str().
  (string_to_delta): Removed unnecessary unicode() call.
  (main): Use "isinstance(x, bytes)" instead of "type(x) is str", use
          .decode() instead of unicode(), and use .encode() instead of
          str().

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
#!/usr/bin/python
 
1
#!/usr/bin/python2.7
2
2
# -*- mode: python; coding: utf-8 -*-
3
3
4
4
# Mandos server - give out binary blobs to connecting clients.
11
11
# "AvahiService" class, and some lines in "main".
12
12
13
13
# Everything else is
14
 
# Copyright © 2008-2012 Teddy Hogeborn
15
 
# Copyright © 2008-2012 Björn Påhlsson
 
14
# Copyright © 2008-2014 Teddy Hogeborn
 
15
# Copyright © 2008-2014 Björn Påhlsson
16
16
17
17
# This program is free software: you can redistribute it and/or modify
18
18
# it under the terms of the GNU General Public License as published by
68
68
import binascii
69
69
import tempfile
70
70
import itertools
 
71
import collections
71
72
 
72
73
import dbus
73
74
import dbus.service
78
79
import ctypes.util
79
80
import xml.dom.minidom
80
81
import inspect
81
 
import GnuPGInterface
82
82
 
83
83
try:
84
84
    SO_BINDTODEVICE = socket.SO_BINDTODEVICE
88
88
    except ImportError:
89
89
        SO_BINDTODEVICE = None
90
90
 
91
 
version = "1.5.5"
 
91
version = "1.6.8"
92
92
stored_state_file = "clients.pickle"
93
93
 
94
94
logger = logging.getLogger()
95
 
syslogger = (logging.handlers.SysLogHandler
96
 
             (facility = logging.handlers.SysLogHandler.LOG_DAEMON,
97
 
              address = str("/dev/log")))
 
95
syslogger = None
98
96
 
99
97
try:
100
98
    if_nametoindex = (ctypes.cdll.LoadLibrary
106
104
        SIOCGIFINDEX = 0x8933  # From /usr/include/linux/sockios.h
107
105
        with contextlib.closing(socket.socket()) as s:
108
106
            ifreq = fcntl.ioctl(s, SIOCGIFINDEX,
109
 
                                struct.pack(str("16s16x"),
110
 
                                            interface))
111
 
        interface_index = struct.unpack(str("I"),
112
 
                                        ifreq[16:20])[0]
 
107
                                struct.pack(b"16s16x", interface))
 
108
        interface_index = struct.unpack("I", ifreq[16:20])[0]
113
109
        return interface_index
114
110
 
115
111
 
116
112
def initlogger(debug, level=logging.WARNING):
117
113
    """init logger and add loglevel"""
118
114
    
 
115
    global syslogger
 
116
    syslogger = (logging.handlers.SysLogHandler
 
117
                 (facility =
 
118
                  logging.handlers.SysLogHandler.LOG_DAEMON,
 
119
                  address = "/dev/log"))
119
120
    syslogger.setFormatter(logging.Formatter
120
121
                           ('Mandos [%(process)d]: %(levelname)s:'
121
122
                            ' %(message)s'))
139
140
class PGPEngine(object):
140
141
    """A simple class for OpenPGP symmetric encryption & decryption"""
141
142
    def __init__(self):
142
 
        self.gnupg = GnuPGInterface.GnuPG()
143
143
        self.tempdir = tempfile.mkdtemp(prefix="mandos-")
144
 
        self.gnupg = GnuPGInterface.GnuPG()
145
 
        self.gnupg.options.meta_interactive = False
146
 
        self.gnupg.options.homedir = self.tempdir
147
 
        self.gnupg.options.extra_args.extend(['--force-mdc',
148
 
                                              '--quiet',
149
 
                                              '--no-use-agent'])
 
144
        self.gnupgargs = ['--batch',
 
145
                          '--home', self.tempdir,
 
146
                          '--force-mdc',
 
147
                          '--quiet',
 
148
                          '--no-use-agent']
150
149
    
151
150
    def __enter__(self):
152
151
        return self
174
173
    def password_encode(self, password):
175
174
        # Passphrase can not be empty and can not contain newlines or
176
175
        # NUL bytes.  So we prefix it and hex encode it.
177
 
        return b"mandos" + binascii.hexlify(password)
 
176
        encoded = b"mandos" + binascii.hexlify(password)
 
177
        if len(encoded) > 2048:
 
178
            # GnuPG can't handle long passwords, so encode differently
 
179
            encoded = (b"mandos" + password.replace(b"\\", b"\\\\")
 
180
                       .replace(b"\n", b"\\n")
 
181
                       .replace(b"\0", b"\\x00"))
 
182
        return encoded
178
183
    
179
184
    def encrypt(self, data, password):
180
 
        self.gnupg.passphrase = self.password_encode(password)
181
 
        with open(os.devnull, "w") as devnull:
182
 
            try:
183
 
                proc = self.gnupg.run(['--symmetric'],
184
 
                                      create_fhs=['stdin', 'stdout'],
185
 
                                      attach_fhs={'stderr': devnull})
186
 
                with contextlib.closing(proc.handles['stdin']) as f:
187
 
                    f.write(data)
188
 
                with contextlib.closing(proc.handles['stdout']) as f:
189
 
                    ciphertext = f.read()
190
 
                proc.wait()
191
 
            except IOError as e:
192
 
                raise PGPError(e)
193
 
        self.gnupg.passphrase = None
 
185
        passphrase = self.password_encode(password)
 
186
        with tempfile.NamedTemporaryFile(dir=self.tempdir
 
187
                                         ) as passfile:
 
188
            passfile.write(passphrase)
 
189
            passfile.flush()
 
190
            proc = subprocess.Popen(['gpg', '--symmetric',
 
191
                                     '--passphrase-file',
 
192
                                     passfile.name]
 
193
                                    + self.gnupgargs,
 
194
                                    stdin = subprocess.PIPE,
 
195
                                    stdout = subprocess.PIPE,
 
196
                                    stderr = subprocess.PIPE)
 
197
            ciphertext, err = proc.communicate(input = data)
 
198
        if proc.returncode != 0:
 
199
            raise PGPError(err)
194
200
        return ciphertext
195
201
    
196
202
    def decrypt(self, data, password):
197
 
        self.gnupg.passphrase = self.password_encode(password)
198
 
        with open(os.devnull, "w") as devnull:
199
 
            try:
200
 
                proc = self.gnupg.run(['--decrypt'],
201
 
                                      create_fhs=['stdin', 'stdout'],
202
 
                                      attach_fhs={'stderr': devnull})
203
 
                with contextlib.closing(proc.handles['stdin']) as f:
204
 
                    f.write(data)
205
 
                with contextlib.closing(proc.handles['stdout']) as f:
206
 
                    decrypted_plaintext = f.read()
207
 
                proc.wait()
208
 
            except IOError as e:
209
 
                raise PGPError(e)
210
 
        self.gnupg.passphrase = None
 
203
        passphrase = self.password_encode(password)
 
204
        with tempfile.NamedTemporaryFile(dir = self.tempdir
 
205
                                         ) as passfile:
 
206
            passfile.write(passphrase)
 
207
            passfile.flush()
 
208
            proc = subprocess.Popen(['gpg', '--decrypt',
 
209
                                     '--passphrase-file',
 
210
                                     passfile.name]
 
211
                                    + self.gnupgargs,
 
212
                                    stdin = subprocess.PIPE,
 
213
                                    stdout = subprocess.PIPE,
 
214
                                    stderr = subprocess.PIPE)
 
215
            decrypted_plaintext, err = proc.communicate(input
 
216
                                                        = data)
 
217
        if proc.returncode != 0:
 
218
            raise PGPError(err)
211
219
        return decrypted_plaintext
212
220
 
213
221
 
214
222
class AvahiError(Exception):
215
223
    def __init__(self, value, *args, **kwargs):
216
224
        self.value = value
217
 
        super(AvahiError, self).__init__(value, *args, **kwargs)
218
 
    def __unicode__(self):
219
 
        return unicode(repr(self.value))
 
225
        return super(AvahiError, self).__init__(value, *args,
 
226
                                                **kwargs)
220
227
 
221
228
class AvahiServiceError(AvahiError):
222
229
    pass
233
240
               Used to optionally bind to the specified interface.
234
241
    name: string; Example: 'Mandos'
235
242
    type: string; Example: '_mandos._tcp'.
236
 
                  See <http://www.dns-sd.org/ServiceTypes.html>
 
243
     See <https://www.iana.org/assignments/service-names-port-numbers>
237
244
    port: integer; what port to announce
238
245
    TXT: list of strings; TXT record for the service
239
246
    domain: string; Domain to publish on, default to .local if empty.
328
335
        elif state == avahi.ENTRY_GROUP_FAILURE:
329
336
            logger.critical("Avahi: Error in group state changed %s",
330
337
                            unicode(error))
331
 
            raise AvahiGroupError("State changed: {0!s}"
 
338
            raise AvahiGroupError("State changed: {!s}"
332
339
                                  .format(error))
333
340
    
334
341
    def cleanup(self):
385
392
        """Add the new name to the syslog messages"""
386
393
        ret = AvahiService.rename(self)
387
394
        syslogger.setFormatter(logging.Formatter
388
 
                               ('Mandos ({0}) [%(process)d]:'
 
395
                               ('Mandos ({}) [%(process)d]:'
389
396
                                ' %(levelname)s: %(message)s'
390
397
                                .format(self.name)))
391
398
        return ret
392
399
 
393
400
 
394
 
def timedelta_to_milliseconds(td):
395
 
    "Convert a datetime.timedelta() to milliseconds"
396
 
    return ((td.days * 24 * 60 * 60 * 1000)
397
 
            + (td.seconds * 1000)
398
 
            + (td.microseconds // 1000))
399
 
 
400
 
 
401
401
class Client(object):
402
402
    """A representation of a client host served by this server.
403
403
    
439
439
    runtime_expansions: Allowed attributes for runtime expansion.
440
440
    expires:    datetime.datetime(); time (UTC) when a client will be
441
441
                disabled, or None
 
442
    server_settings: The server_settings dict from main()
442
443
    """
443
444
    
444
445
    runtime_expansions = ("approval_delay", "approval_duration",
446
447
                          "fingerprint", "host", "interval",
447
448
                          "last_approval_request", "last_checked_ok",
448
449
                          "last_enabled", "name", "timeout")
449
 
    client_defaults = { "timeout": "5m",
450
 
                        "extended_timeout": "15m",
451
 
                        "interval": "2m",
 
450
    client_defaults = { "timeout": "PT5M",
 
451
                        "extended_timeout": "PT15M",
 
452
                        "interval": "PT2M",
452
453
                        "checker": "fping -q -- %%(host)s",
453
454
                        "host": "",
454
 
                        "approval_delay": "0s",
455
 
                        "approval_duration": "1s",
 
455
                        "approval_delay": "PT0S",
 
456
                        "approval_duration": "PT1S",
456
457
                        "approved_by_default": "True",
457
458
                        "enabled": "True",
458
459
                        }
459
460
    
460
 
    def timeout_milliseconds(self):
461
 
        "Return the 'timeout' attribute in milliseconds"
462
 
        return timedelta_to_milliseconds(self.timeout)
463
 
    
464
 
    def extended_timeout_milliseconds(self):
465
 
        "Return the 'extended_timeout' attribute in milliseconds"
466
 
        return timedelta_to_milliseconds(self.extended_timeout)
467
 
    
468
 
    def interval_milliseconds(self):
469
 
        "Return the 'interval' attribute in milliseconds"
470
 
        return timedelta_to_milliseconds(self.interval)
471
 
    
472
 
    def approval_delay_milliseconds(self):
473
 
        return timedelta_to_milliseconds(self.approval_delay)
474
 
    
475
461
    @staticmethod
476
462
    def config_parser(config):
477
463
        """Construct a new dict of client settings of this form:
502
488
                          "rb") as secfile:
503
489
                    client["secret"] = secfile.read()
504
490
            else:
505
 
                raise TypeError("No secret or secfile for section {0}"
 
491
                raise TypeError("No secret or secfile for section {}"
506
492
                                .format(section))
507
493
            client["timeout"] = string_to_delta(section["timeout"])
508
494
            client["extended_timeout"] = string_to_delta(
519
505
        
520
506
        return settings
521
507
    
522
 
    def __init__(self, settings, name = None):
 
508
    def __init__(self, settings, name = None, server_settings=None):
523
509
        self.name = name
 
510
        if server_settings is None:
 
511
            server_settings = {}
 
512
        self.server_settings = server_settings
524
513
        # adding all client settings
525
 
        for setting, value in settings.iteritems():
 
514
        for setting, value in settings.items():
526
515
            setattr(self, setting, value)
527
516
        
528
517
        if self.enabled:
611
600
        if self.checker_initiator_tag is not None:
612
601
            gobject.source_remove(self.checker_initiator_tag)
613
602
        self.checker_initiator_tag = (gobject.timeout_add
614
 
                                      (self.interval_milliseconds(),
 
603
                                      (int(self.interval
 
604
                                           .total_seconds() * 1000),
615
605
                                       self.start_checker))
616
606
        # Schedule a disable() when 'timeout' has passed
617
607
        if self.disable_initiator_tag is not None:
618
608
            gobject.source_remove(self.disable_initiator_tag)
619
609
        self.disable_initiator_tag = (gobject.timeout_add
620
 
                                   (self.timeout_milliseconds(),
621
 
                                    self.disable))
 
610
                                      (int(self.timeout
 
611
                                           .total_seconds() * 1000),
 
612
                                       self.disable))
622
613
        # Also start a new checker *right now*.
623
614
        self.start_checker()
624
615
    
655
646
            self.disable_initiator_tag = None
656
647
        if getattr(self, "enabled", False):
657
648
            self.disable_initiator_tag = (gobject.timeout_add
658
 
                                          (timedelta_to_milliseconds
659
 
                                           (timeout), self.disable))
 
649
                                          (int(timeout.total_seconds()
 
650
                                               * 1000), self.disable))
660
651
            self.expires = datetime.datetime.utcnow() + timeout
661
652
    
662
653
    def need_approval(self):
679
670
        # If a checker exists, make sure it is not a zombie
680
671
        try:
681
672
            pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
682
 
        except (AttributeError, OSError) as error:
683
 
            if (isinstance(error, OSError)
684
 
                and error.errno != errno.ECHILD):
685
 
                raise error
 
673
        except AttributeError:
 
674
            pass
 
675
        except OSError as error:
 
676
            if error.errno != errno.ECHILD:
 
677
                raise
686
678
        else:
687
679
            if pid:
688
680
                logger.warning("Checker was a zombie")
692
684
        # Start a new checker if needed
693
685
        if self.checker is None:
694
686
            # 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)
 
687
            escaped_attrs = { attr:
 
688
                                  re.escape(unicode(getattr(self,
 
689
                                                            attr)))
 
690
                              for attr in self.runtime_expansions }
699
691
            try:
700
692
                command = self.checker_command % escaped_attrs
701
693
            except TypeError as error:
710
702
                # in normal mode, that is already done by daemon(),
711
703
                # and in debug mode we don't want to.  (Stdin is
712
704
                # always replaced by /dev/null.)
 
705
                # The exception is when not debugging but nevertheless
 
706
                # running in the foreground; use the previously
 
707
                # created wnull.
 
708
                popen_args = {}
 
709
                if (not self.server_settings["debug"]
 
710
                    and self.server_settings["foreground"]):
 
711
                    popen_args.update({"stdout": wnull,
 
712
                                       "stderr": wnull })
713
713
                self.checker = subprocess.Popen(command,
714
714
                                                close_fds=True,
715
 
                                                shell=True, cwd="/")
 
715
                                                shell=True, cwd="/",
 
716
                                                **popen_args)
716
717
            except OSError as error:
717
718
                logger.error("Failed to start subprocess",
718
719
                             exc_info=error)
 
720
                return True
719
721
            self.checker_callback_tag = (gobject.child_watch_add
720
722
                                         (self.checker.pid,
721
723
                                          self.checker_callback,
722
724
                                          data=command))
723
725
            # The checker may have completed before the gobject
724
726
            # watch was added.  Check for this.
725
 
            pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
 
727
            try:
 
728
                pid, status = os.waitpid(self.checker.pid, os.WNOHANG)
 
729
            except OSError as error:
 
730
                if error.errno == errno.ECHILD:
 
731
                    # This should never happen
 
732
                    logger.error("Child process vanished",
 
733
                                 exc_info=error)
 
734
                    return True
 
735
                raise
726
736
            if pid:
727
737
                gobject.source_remove(self.checker_callback_tag)
728
738
                self.checker_callback(pid, status, command)
764
774
    # "Set" method, so we fail early here:
765
775
    if byte_arrays and signature != "ay":
766
776
        raise ValueError("Byte arrays not supported for non-'ay'"
767
 
                         " signature {0!r}".format(signature))
 
777
                         " signature {!r}".format(signature))
768
778
    def decorator(func):
769
779
        func._dbus_is_property = True
770
780
        func._dbus_interface = dbus_interface
818
828
class DBusPropertyException(dbus.exceptions.DBusException):
819
829
    """A base class for D-Bus property-related exceptions
820
830
    """
821
 
    def __unicode__(self):
822
 
        return unicode(str(self))
823
 
 
 
831
    pass
824
832
 
825
833
class DBusPropertyAccessException(DBusPropertyException):
826
834
    """A property's access permissions disallows an operation.
849
857
        If called like _is_dbus_thing("method") it returns a function
850
858
        suitable for use as predicate to inspect.getmembers().
851
859
        """
852
 
        return lambda obj: getattr(obj, "_dbus_is_{0}".format(thing),
 
860
        return lambda obj: getattr(obj, "_dbus_is_{}".format(thing),
853
861
                                   False)
854
862
    
855
863
    def _get_all_dbus_things(self, thing):
904
912
            # The byte_arrays option is not supported yet on
905
913
            # signatures other than "ay".
906
914
            if prop._dbus_signature != "ay":
907
 
                raise ValueError
 
915
                raise ValueError("Byte arrays not supported for non-"
 
916
                                 "'ay' signature {!r}"
 
917
                                 .format(prop._dbus_signature))
908
918
            value = dbus.ByteArray(b''.join(chr(byte)
909
919
                                            for byte in value))
910
920
        prop(value)
974
984
                                              (prop,
975
985
                                               "_dbus_annotations",
976
986
                                               {}))
977
 
                        for name, value in annots.iteritems():
 
987
                        for name, value in annots.items():
978
988
                            ann_tag = document.createElement(
979
989
                                "annotation")
980
990
                            ann_tag.setAttribute("name", name)
983
993
                # Add interface annotation tags
984
994
                for annotation, value in dict(
985
995
                    itertools.chain.from_iterable(
986
 
                        annotations().iteritems()
 
996
                        annotations().items()
987
997
                        for name, annotations in
988
998
                        self._get_all_dbus_things("interface")
989
999
                        if name == if_tag.getAttribute("name")
990
 
                        )).iteritems():
 
1000
                        )).items():
991
1001
                    ann_tag = document.createElement("annotation")
992
1002
                    ann_tag.setAttribute("name", annotation)
993
1003
                    ann_tag.setAttribute("value", value)
1049
1059
    """
1050
1060
    def wrapper(cls):
1051
1061
        for orig_interface_name, alt_interface_name in (
1052
 
            alt_interface_names.iteritems()):
 
1062
            alt_interface_names.items()):
1053
1063
            attr = {}
1054
1064
            interface_names = set()
1055
1065
            # Go though all attributes of the class
1068
1078
                interface_names.add(alt_interface)
1069
1079
                # Is this a D-Bus signal?
1070
1080
                if getattr(attribute, "_dbus_is_signal", False):
1071
 
                    # Extract the original non-method function by
1072
 
                    # black magic
 
1081
                    # Extract the original non-method undecorated
 
1082
                    # function by black magic
1073
1083
                    nonmethod_func = (dict(
1074
1084
                            zip(attribute.func_code.co_freevars,
1075
1085
                                attribute.__closure__))["func"]
1172
1182
                                        attribute.func_closure)))
1173
1183
            if deprecate:
1174
1184
                # Deprecate all alternate interfaces
1175
 
                iname="_AlternateDBusNames_interface_annotation{0}"
 
1185
                iname="_AlternateDBusNames_interface_annotation{}"
1176
1186
                for interface_name in interface_names:
1177
1187
                    @dbus_interface_annotations(interface_name)
1178
1188
                    def func(self):
1187
1197
            if interface_names:
1188
1198
                # Replace the class with a new subclass of it with
1189
1199
                # methods, signals, etc. as created above.
1190
 
                cls = type(b"{0}Alternate".format(cls.__name__),
 
1200
                cls = type(b"{}Alternate".format(cls.__name__),
1191
1201
                           (cls,), attr)
1192
1202
        return cls
1193
1203
    return wrapper
1234
1244
                   to the D-Bus.  Default: no transform
1235
1245
        variant_level: D-Bus variant level.  Default: 1
1236
1246
        """
1237
 
        attrname = "_{0}".format(dbus_name)
 
1247
        attrname = "_{}".format(dbus_name)
1238
1248
        def setter(self, value):
1239
1249
            if hasattr(self, "dbus_object_path"):
1240
1250
                if (not hasattr(self, attrname) or
1270
1280
    approval_delay = notifychangeproperty(dbus.UInt64,
1271
1281
                                          "ApprovalDelay",
1272
1282
                                          type_func =
1273
 
                                          timedelta_to_milliseconds)
 
1283
                                          lambda td: td.total_seconds()
 
1284
                                          * 1000)
1274
1285
    approval_duration = notifychangeproperty(
1275
1286
        dbus.UInt64, "ApprovalDuration",
1276
 
        type_func = timedelta_to_milliseconds)
 
1287
        type_func = lambda td: td.total_seconds() * 1000)
1277
1288
    host = notifychangeproperty(dbus.String, "Host")
1278
1289
    timeout = notifychangeproperty(dbus.UInt64, "Timeout",
1279
 
                                   type_func =
1280
 
                                   timedelta_to_milliseconds)
 
1290
                                   type_func = lambda td:
 
1291
                                       td.total_seconds() * 1000)
1281
1292
    extended_timeout = notifychangeproperty(
1282
1293
        dbus.UInt64, "ExtendedTimeout",
1283
 
        type_func = timedelta_to_milliseconds)
 
1294
        type_func = lambda td: td.total_seconds() * 1000)
1284
1295
    interval = notifychangeproperty(dbus.UInt64,
1285
1296
                                    "Interval",
1286
1297
                                    type_func =
1287
 
                                    timedelta_to_milliseconds)
 
1298
                                    lambda td: td.total_seconds()
 
1299
                                    * 1000)
1288
1300
    checker_command = notifychangeproperty(dbus.String, "Checker")
1289
1301
    
1290
1302
    del notifychangeproperty
1318
1330
                                       *args, **kwargs)
1319
1331
    
1320
1332
    def start_checker(self, *args, **kwargs):
1321
 
        old_checker = self.checker
1322
 
        if self.checker is not None:
1323
 
            old_checker_pid = self.checker.pid
1324
 
        else:
1325
 
            old_checker_pid = None
 
1333
        old_checker_pid = getattr(self.checker, "pid", None)
1326
1334
        r = Client.start_checker(self, *args, **kwargs)
1327
1335
        # Only if new checker process was started
1328
1336
        if (self.checker is not None
1337
1345
    
1338
1346
    def approve(self, value=True):
1339
1347
        self.approved = value
1340
 
        gobject.timeout_add(timedelta_to_milliseconds
1341
 
                            (self.approval_duration),
1342
 
                            self._reset_approved)
 
1348
        gobject.timeout_add(int(self.approval_duration.total_seconds()
 
1349
                                * 1000), self._reset_approved)
1343
1350
        self.send_changedstate()
1344
1351
    
1345
1352
    ## D-Bus methods, signals & properties
1448
1455
                           access="readwrite")
1449
1456
    def ApprovalDelay_dbus_property(self, value=None):
1450
1457
        if value is None:       # get
1451
 
            return dbus.UInt64(self.approval_delay_milliseconds())
 
1458
            return dbus.UInt64(self.approval_delay.total_seconds()
 
1459
                               * 1000)
1452
1460
        self.approval_delay = datetime.timedelta(0, 0, 0, value)
1453
1461
    
1454
1462
    # ApprovalDuration - property
1456
1464
                           access="readwrite")
1457
1465
    def ApprovalDuration_dbus_property(self, value=None):
1458
1466
        if value is None:       # get
1459
 
            return dbus.UInt64(timedelta_to_milliseconds(
1460
 
                    self.approval_duration))
 
1467
            return dbus.UInt64(self.approval_duration.total_seconds()
 
1468
                               * 1000)
1461
1469
        self.approval_duration = datetime.timedelta(0, 0, 0, value)
1462
1470
    
1463
1471
    # Name - property
1529
1537
                           access="readwrite")
1530
1538
    def Timeout_dbus_property(self, value=None):
1531
1539
        if value is None:       # get
1532
 
            return dbus.UInt64(self.timeout_milliseconds())
 
1540
            return dbus.UInt64(self.timeout.total_seconds() * 1000)
1533
1541
        old_timeout = self.timeout
1534
1542
        self.timeout = datetime.timedelta(0, 0, 0, value)
1535
1543
        # Reschedule disabling
1546
1554
                gobject.source_remove(self.disable_initiator_tag)
1547
1555
                self.disable_initiator_tag = (
1548
1556
                    gobject.timeout_add(
1549
 
                        timedelta_to_milliseconds(self.expires - now),
1550
 
                        self.disable))
 
1557
                        int((self.expires - now).total_seconds()
 
1558
                            * 1000), self.disable))
1551
1559
    
1552
1560
    # ExtendedTimeout - property
1553
1561
    @dbus_service_property(_interface, signature="t",
1554
1562
                           access="readwrite")
1555
1563
    def ExtendedTimeout_dbus_property(self, value=None):
1556
1564
        if value is None:       # get
1557
 
            return dbus.UInt64(self.extended_timeout_milliseconds())
 
1565
            return dbus.UInt64(self.extended_timeout.total_seconds()
 
1566
                               * 1000)
1558
1567
        self.extended_timeout = datetime.timedelta(0, 0, 0, value)
1559
1568
    
1560
1569
    # Interval - property
1562
1571
                           access="readwrite")
1563
1572
    def Interval_dbus_property(self, value=None):
1564
1573
        if value is None:       # get
1565
 
            return dbus.UInt64(self.interval_milliseconds())
 
1574
            return dbus.UInt64(self.interval.total_seconds() * 1000)
1566
1575
        self.interval = datetime.timedelta(0, 0, 0, value)
1567
1576
        if getattr(self, "checker_initiator_tag", None) is None:
1568
1577
            return
1601
1610
    @dbus_service_property(_interface, signature="ay",
1602
1611
                           access="write", byte_arrays=True)
1603
1612
    def Secret_dbus_property(self, value):
1604
 
        self.secret = str(value)
 
1613
        self.secret = bytes(value)
1605
1614
    
1606
1615
    del _interface
1607
1616
 
1673
1682
            logger.debug("Protocol version: %r", line)
1674
1683
            try:
1675
1684
                if int(line.strip().split()[0]) > 1:
1676
 
                    raise RuntimeError
 
1685
                    raise RuntimeError(line)
1677
1686
            except (ValueError, IndexError, RuntimeError) as error:
1678
1687
                logger.error("Unknown protocol version: %s", error)
1679
1688
                return
1728
1737
                        if self.server.use_dbus:
1729
1738
                            # Emit D-Bus signal
1730
1739
                            client.NeedApproval(
1731
 
                                client.approval_delay_milliseconds(),
1732
 
                                client.approved_by_default)
 
1740
                                client.approval_delay.total_seconds()
 
1741
                                * 1000, client.approved_by_default)
1733
1742
                    else:
1734
1743
                        logger.warning("Client %s was not approved",
1735
1744
                                       client.name)
1741
1750
                    #wait until timeout or approved
1742
1751
                    time = datetime.datetime.now()
1743
1752
                    client.changedstate.acquire()
1744
 
                    client.changedstate.wait(
1745
 
                        float(timedelta_to_milliseconds(delay)
1746
 
                              / 1000))
 
1753
                    client.changedstate.wait(delay.total_seconds())
1747
1754
                    client.changedstate.release()
1748
1755
                    time2 = datetime.datetime.now()
1749
1756
                    if (time2 - time) >= delay:
1886
1893
    
1887
1894
    def add_pipe(self, parent_pipe, proc):
1888
1895
        """Dummy function; override as necessary"""
1889
 
        raise NotImplementedError
 
1896
        raise NotImplementedError()
1890
1897
 
1891
1898
 
1892
1899
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
1948
1955
                try:
1949
1956
                    self.socket.setsockopt(socket.SOL_SOCKET,
1950
1957
                                           SO_BINDTODEVICE,
1951
 
                                           str(self.interface
1952
 
                                               + '\0'))
 
1958
                                           (self.interface + "\0")
 
1959
                                           .encode("utf-8"))
1953
1960
                except socket.error as error:
1954
1961
                    if error.errno == errno.EPERM:
1955
 
                        logger.error("No permission to"
1956
 
                                     " bind to interface %s",
1957
 
                                     self.interface)
 
1962
                        logger.error("No permission to bind to"
 
1963
                                     " interface %s", self.interface)
1958
1964
                    elif error.errno == errno.ENOPROTOOPT:
1959
1965
                        logger.error("SO_BINDTODEVICE not available;"
1960
1966
                                     " cannot bind to interface %s",
1961
1967
                                     self.interface)
1962
1968
                    elif error.errno == errno.ENODEV:
1963
 
                        logger.error("Interface %s does not"
1964
 
                                     " exist, cannot bind",
1965
 
                                     self.interface)
 
1969
                        logger.error("Interface %s does not exist,"
 
1970
                                     " cannot bind", self.interface)
1966
1971
                    else:
1967
1972
                        raise
1968
1973
        # Only bind(2) the socket if we really need to.
1971
1976
                if self.address_family == socket.AF_INET6:
1972
1977
                    any_address = "::" # in6addr_any
1973
1978
                else:
1974
 
                    any_address = socket.INADDR_ANY
 
1979
                    any_address = "0.0.0.0" # INADDR_ANY
1975
1980
                self.server_address = (any_address,
1976
1981
                                       self.server_address[1])
1977
1982
            elif not self.server_address[1]:
2093
2098
        return True
2094
2099
 
2095
2100
 
 
2101
def rfc3339_duration_to_delta(duration):
 
2102
    """Parse an RFC 3339 "duration" and return a datetime.timedelta
 
2103
    
 
2104
    >>> rfc3339_duration_to_delta("P7D")
 
2105
    datetime.timedelta(7)
 
2106
    >>> rfc3339_duration_to_delta("PT60S")
 
2107
    datetime.timedelta(0, 60)
 
2108
    >>> rfc3339_duration_to_delta("PT60M")
 
2109
    datetime.timedelta(0, 3600)
 
2110
    >>> rfc3339_duration_to_delta("PT24H")
 
2111
    datetime.timedelta(1)
 
2112
    >>> rfc3339_duration_to_delta("P1W")
 
2113
    datetime.timedelta(7)
 
2114
    >>> rfc3339_duration_to_delta("PT5M30S")
 
2115
    datetime.timedelta(0, 330)
 
2116
    >>> rfc3339_duration_to_delta("P1DT3M20S")
 
2117
    datetime.timedelta(1, 200)
 
2118
    """
 
2119
    
 
2120
    # Parsing an RFC 3339 duration with regular expressions is not
 
2121
    # possible - there would have to be multiple places for the same
 
2122
    # values, like seconds.  The current code, while more esoteric, is
 
2123
    # cleaner without depending on a parsing library.  If Python had a
 
2124
    # built-in library for parsing we would use it, but we'd like to
 
2125
    # avoid excessive use of external libraries.
 
2126
    
 
2127
    # New type for defining tokens, syntax, and semantics all-in-one
 
2128
    Token = collections.namedtuple("Token",
 
2129
                                   ("regexp", # To match token; if
 
2130
                                              # "value" is not None,
 
2131
                                              # must have a "group"
 
2132
                                              # containing digits
 
2133
                                    "value",  # datetime.timedelta or
 
2134
                                              # None
 
2135
                                    "followers")) # Tokens valid after
 
2136
                                                  # this token
 
2137
    # RFC 3339 "duration" tokens, syntax, and semantics; taken from
 
2138
    # the "duration" ABNF definition in RFC 3339, Appendix A.
 
2139
    token_end = Token(re.compile(r"$"), None, frozenset())
 
2140
    token_second = Token(re.compile(r"(\d+)S"),
 
2141
                         datetime.timedelta(seconds=1),
 
2142
                         frozenset((token_end,)))
 
2143
    token_minute = Token(re.compile(r"(\d+)M"),
 
2144
                         datetime.timedelta(minutes=1),
 
2145
                         frozenset((token_second, token_end)))
 
2146
    token_hour = Token(re.compile(r"(\d+)H"),
 
2147
                       datetime.timedelta(hours=1),
 
2148
                       frozenset((token_minute, token_end)))
 
2149
    token_time = Token(re.compile(r"T"),
 
2150
                       None,
 
2151
                       frozenset((token_hour, token_minute,
 
2152
                                  token_second)))
 
2153
    token_day = Token(re.compile(r"(\d+)D"),
 
2154
                      datetime.timedelta(days=1),
 
2155
                      frozenset((token_time, token_end)))
 
2156
    token_month = Token(re.compile(r"(\d+)M"),
 
2157
                        datetime.timedelta(weeks=4),
 
2158
                        frozenset((token_day, token_end)))
 
2159
    token_year = Token(re.compile(r"(\d+)Y"),
 
2160
                       datetime.timedelta(weeks=52),
 
2161
                       frozenset((token_month, token_end)))
 
2162
    token_week = Token(re.compile(r"(\d+)W"),
 
2163
                       datetime.timedelta(weeks=1),
 
2164
                       frozenset((token_end,)))
 
2165
    token_duration = Token(re.compile(r"P"), None,
 
2166
                           frozenset((token_year, token_month,
 
2167
                                      token_day, token_time,
 
2168
                                      token_week)))
 
2169
    # Define starting values
 
2170
    value = datetime.timedelta() # Value so far
 
2171
    found_token = None
 
2172
    followers = frozenset((token_duration,)) # Following valid tokens
 
2173
    s = duration                # String left to parse
 
2174
    # Loop until end token is found
 
2175
    while found_token is not token_end:
 
2176
        # Search for any currently valid tokens
 
2177
        for token in followers:
 
2178
            match = token.regexp.match(s)
 
2179
            if match is not None:
 
2180
                # Token found
 
2181
                if token.value is not None:
 
2182
                    # Value found, parse digits
 
2183
                    factor = int(match.group(1), 10)
 
2184
                    # Add to value so far
 
2185
                    value += factor * token.value
 
2186
                # Strip token from string
 
2187
                s = token.regexp.sub("", s, 1)
 
2188
                # Go to found token
 
2189
                found_token = token
 
2190
                # Set valid next tokens
 
2191
                followers = found_token.followers
 
2192
                break
 
2193
        else:
 
2194
            # No currently valid tokens were found
 
2195
            raise ValueError("Invalid RFC 3339 duration")
 
2196
    # End token found
 
2197
    return value
 
2198
 
 
2199
 
2096
2200
def string_to_delta(interval):
2097
2201
    """Parse a string and return a datetime.timedelta
2098
2202
    
2109
2213
    >>> string_to_delta('5m 30s')
2110
2214
    datetime.timedelta(0, 330)
2111
2215
    """
 
2216
    
 
2217
    try:
 
2218
        return rfc3339_duration_to_delta(interval)
 
2219
    except ValueError:
 
2220
        pass
 
2221
    
2112
2222
    timevalue = datetime.timedelta(0)
2113
2223
    for s in interval.split():
2114
2224
        try:
2115
 
            suffix = unicode(s[-1])
 
2225
            suffix = s[-1]
2116
2226
            value = int(s[:-1])
2117
2227
            if suffix == "d":
2118
2228
                delta = datetime.timedelta(value)
2125
2235
            elif suffix == "w":
2126
2236
                delta = datetime.timedelta(0, 0, 0, 0, 0, 0, value)
2127
2237
            else:
2128
 
                raise ValueError("Unknown suffix {0!r}"
 
2238
                raise ValueError("Unknown suffix {!r}"
2129
2239
                                 .format(suffix))
2130
 
        except (ValueError, IndexError) as e:
 
2240
        except IndexError as e:
2131
2241
            raise ValueError(*(e.args))
2132
2242
        timevalue += delta
2133
2243
    return timevalue
2148
2258
        # Close all standard open file descriptors
2149
2259
        null = os.open(os.devnull, os.O_NOCTTY | os.O_RDWR)
2150
2260
        if not stat.S_ISCHR(os.fstat(null).st_mode):
2151
 
            raise OSError(errno.ENODEV,
2152
 
                          "{0} not a character device"
 
2261
            raise OSError(errno.ENODEV, "{} not a character device"
2153
2262
                          .format(os.devnull))
2154
2263
        os.dup2(null, sys.stdin.fileno())
2155
2264
        os.dup2(null, sys.stdout.fileno())
2165
2274
    
2166
2275
    parser = argparse.ArgumentParser()
2167
2276
    parser.add_argument("-v", "--version", action="version",
2168
 
                        version = "%(prog)s {0}".format(version),
 
2277
                        version = "%(prog)s {}".format(version),
2169
2278
                        help="show version number and exit")
2170
2279
    parser.add_argument("-i", "--interface", metavar="IF",
2171
2280
                        help="Bind to interface IF")
2177
2286
                        help="Run self-test")
2178
2287
    parser.add_argument("--debug", action="store_true",
2179
2288
                        help="Debug mode; run in foreground and log"
2180
 
                        " to terminal")
 
2289
                        " to terminal", default=None)
2181
2290
    parser.add_argument("--debuglevel", metavar="LEVEL",
2182
2291
                        help="Debug level for stdout output")
2183
2292
    parser.add_argument("--priority", help="GnuTLS"
2190
2299
                        " files")
2191
2300
    parser.add_argument("--no-dbus", action="store_false",
2192
2301
                        dest="use_dbus", help="Do not provide D-Bus"
2193
 
                        " system bus interface")
 
2302
                        " system bus interface", default=None)
2194
2303
    parser.add_argument("--no-ipv6", action="store_false",
2195
 
                        dest="use_ipv6", help="Do not use IPv6")
 
2304
                        dest="use_ipv6", help="Do not use IPv6",
 
2305
                        default=None)
2196
2306
    parser.add_argument("--no-restore", action="store_false",
2197
2307
                        dest="restore", help="Do not restore stored"
2198
 
                        " state")
 
2308
                        " state", default=None)
2199
2309
    parser.add_argument("--socket", type=int,
2200
2310
                        help="Specify a file descriptor to a network"
2201
2311
                        " socket to use instead of creating one")
2202
2312
    parser.add_argument("--statedir", metavar="DIR",
2203
2313
                        help="Directory to save/restore state in")
 
2314
    parser.add_argument("--foreground", action="store_true",
 
2315
                        help="Run in foreground", default=None)
 
2316
    parser.add_argument("--no-zeroconf", action="store_false",
 
2317
                        dest="zeroconf", help="Do not use Zeroconf",
 
2318
                        default=None)
2204
2319
    
2205
2320
    options = parser.parse_args()
2206
2321
    
2207
2322
    if options.check:
2208
2323
        import doctest
2209
 
        doctest.testmod()
2210
 
        sys.exit()
 
2324
        fail_count, test_count = doctest.testmod()
 
2325
        sys.exit(os.EX_OK if fail_count == 0 else 1)
2211
2326
    
2212
2327
    # Default values for config file for server-global settings
2213
2328
    server_defaults = { "interface": "",
2215
2330
                        "port": "",
2216
2331
                        "debug": "False",
2217
2332
                        "priority":
2218
 
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
 
2333
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP:+SIGN-RSA-SHA224:+SIGN-RSA-RMD160",
2219
2334
                        "servicename": "Mandos",
2220
2335
                        "use_dbus": "True",
2221
2336
                        "use_ipv6": "True",
2222
2337
                        "debuglevel": "",
2223
2338
                        "restore": "True",
2224
2339
                        "socket": "",
2225
 
                        "statedir": "/var/lib/mandos"
 
2340
                        "statedir": "/var/lib/mandos",
 
2341
                        "foreground": "False",
 
2342
                        "zeroconf": "True",
2226
2343
                        }
2227
2344
    
2228
2345
    # Parse config file for server-global settings
2233
2350
    # Convert the SafeConfigParser object to a dict
2234
2351
    server_settings = server_config.defaults()
2235
2352
    # Use the appropriate methods on the non-string config options
2236
 
    for option in ("debug", "use_dbus", "use_ipv6"):
 
2353
    for option in ("debug", "use_dbus", "use_ipv6", "foreground"):
2237
2354
        server_settings[option] = server_config.getboolean("DEFAULT",
2238
2355
                                                           option)
2239
2356
    if server_settings["port"]:
2255
2372
    for option in ("interface", "address", "port", "debug",
2256
2373
                   "priority", "servicename", "configdir",
2257
2374
                   "use_dbus", "use_ipv6", "debuglevel", "restore",
2258
 
                   "statedir", "socket"):
 
2375
                   "statedir", "socket", "foreground", "zeroconf"):
2259
2376
        value = getattr(options, option)
2260
2377
        if value is not None:
2261
2378
            server_settings[option] = value
2262
2379
    del options
2263
2380
    # Force all strings to be unicode
2264
2381
    for option in server_settings.keys():
2265
 
        if type(server_settings[option]) is str:
2266
 
            server_settings[option] = unicode(server_settings[option])
 
2382
        if isinstance(server_settings[option], bytes):
 
2383
            server_settings[option] = (server_settings[option]
 
2384
                                       .decode("utf-8"))
 
2385
    # Force all boolean options to be boolean
 
2386
    for option in ("debug", "use_dbus", "use_ipv6", "restore",
 
2387
                   "foreground", "zeroconf"):
 
2388
        server_settings[option] = bool(server_settings[option])
 
2389
    # Debug implies foreground
 
2390
    if server_settings["debug"]:
 
2391
        server_settings["foreground"] = True
2267
2392
    # Now we have our good server settings in "server_settings"
2268
2393
    
2269
2394
    ##################################################################
2270
2395
    
 
2396
    if (not server_settings["zeroconf"] and
 
2397
        not (server_settings["port"]
 
2398
             or server_settings["socket"] != "")):
 
2399
            parser.error("Needs port or socket to work without"
 
2400
                         " Zeroconf")
 
2401
    
2271
2402
    # For convenience
2272
2403
    debug = server_settings["debug"]
2273
2404
    debuglevel = server_settings["debuglevel"]
2275
2406
    use_ipv6 = server_settings["use_ipv6"]
2276
2407
    stored_state_path = os.path.join(server_settings["statedir"],
2277
2408
                                     stored_state_file)
 
2409
    foreground = server_settings["foreground"]
 
2410
    zeroconf = server_settings["zeroconf"]
2278
2411
    
2279
2412
    if debug:
2280
2413
        initlogger(debug, logging.DEBUG)
2287
2420
    
2288
2421
    if server_settings["servicename"] != "Mandos":
2289
2422
        syslogger.setFormatter(logging.Formatter
2290
 
                               ('Mandos ({0}) [%(process)d]:'
 
2423
                               ('Mandos ({}) [%(process)d]:'
2291
2424
                                ' %(levelname)s: %(message)s'
2292
2425
                                .format(server_settings
2293
2426
                                        ["servicename"])))
2301
2434
    global mandos_dbus_service
2302
2435
    mandos_dbus_service = None
2303
2436
    
 
2437
    socketfd = None
 
2438
    if server_settings["socket"] != "":
 
2439
        socketfd = server_settings["socket"]
2304
2440
    tcp_server = MandosServer((server_settings["address"],
2305
2441
                               server_settings["port"]),
2306
2442
                              ClientHandler,
2310
2446
                              gnutls_priority=
2311
2447
                              server_settings["priority"],
2312
2448
                              use_dbus=use_dbus,
2313
 
                              socketfd=(server_settings["socket"]
2314
 
                                        or None))
2315
 
    if not debug:
2316
 
        pidfilename = "/var/run/mandos.pid"
 
2449
                              socketfd=socketfd)
 
2450
    if not foreground:
 
2451
        pidfilename = "/run/mandos.pid"
 
2452
        if not os.path.isdir("/run/."):
 
2453
            pidfilename = "/var/run/mandos.pid"
 
2454
        pidfile = None
2317
2455
        try:
2318
2456
            pidfile = open(pidfilename, "w")
2319
2457
        except IOError as e:
2335
2473
        os.setuid(uid)
2336
2474
    except OSError as error:
2337
2475
        if error.errno != errno.EPERM:
2338
 
            raise error
 
2476
            raise
2339
2477
    
2340
2478
    if debug:
2341
2479
        # Enable all possible GnuTLS debugging
2358
2496
            os.close(null)
2359
2497
    
2360
2498
    # Need to fork before connecting to D-Bus
2361
 
    if not debug:
 
2499
    if not foreground:
2362
2500
        # Close all input and output, do double fork, etc.
2363
2501
        daemon()
2364
2502
    
2384
2522
            use_dbus = False
2385
2523
            server_settings["use_dbus"] = False
2386
2524
            tcp_server.use_dbus = False
2387
 
    protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
2388
 
    service = AvahiServiceToSyslog(name =
2389
 
                                   server_settings["servicename"],
2390
 
                                   servicetype = "_mandos._tcp",
2391
 
                                   protocol = protocol, bus = bus)
2392
 
    if server_settings["interface"]:
2393
 
        service.interface = (if_nametoindex
2394
 
                             (str(server_settings["interface"])))
 
2525
    if zeroconf:
 
2526
        protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
 
2527
        service = AvahiServiceToSyslog(name =
 
2528
                                       server_settings["servicename"],
 
2529
                                       servicetype = "_mandos._tcp",
 
2530
                                       protocol = protocol, bus = bus)
 
2531
        if server_settings["interface"]:
 
2532
            service.interface = (if_nametoindex
 
2533
                                 (server_settings["interface"]
 
2534
                                  .encode("utf-8")))
2395
2535
    
2396
2536
    global multiprocessing_manager
2397
2537
    multiprocessing_manager = multiprocessing.Manager()
2404
2544
    old_client_settings = {}
2405
2545
    clients_data = {}
2406
2546
    
 
2547
    # This is used to redirect stdout and stderr for checker processes
 
2548
    global wnull
 
2549
    wnull = open(os.devnull, "w") # A writable /dev/null
 
2550
    # Only used if server is running in foreground but not in debug
 
2551
    # mode
 
2552
    if debug or not foreground:
 
2553
        wnull.close()
 
2554
    
2407
2555
    # Get client data and settings from last running state.
2408
2556
    if server_settings["restore"]:
2409
2557
        try:
2413
2561
            os.remove(stored_state_path)
2414
2562
        except IOError as e:
2415
2563
            if e.errno == errno.ENOENT:
2416
 
                logger.warning("Could not load persistent state: {0}"
 
2564
                logger.warning("Could not load persistent state: {}"
2417
2565
                                .format(os.strerror(e.errno)))
2418
2566
            else:
2419
2567
                logger.critical("Could not load persistent state:",
2424
2572
                           "EOFError:", exc_info=e)
2425
2573
    
2426
2574
    with PGPEngine() as pgp:
2427
 
        for client_name, client in clients_data.iteritems():
 
2575
        for client_name, client in clients_data.items():
 
2576
            # Skip removed clients
 
2577
            if client_name not in client_settings:
 
2578
                continue
 
2579
            
2428
2580
            # Decide which value to use after restoring saved state.
2429
2581
            # We have three different values: Old config file,
2430
2582
            # new config file, and saved state.
2451
2603
                if datetime.datetime.utcnow() >= client["expires"]:
2452
2604
                    if not client["last_checked_ok"]:
2453
2605
                        logger.warning(
2454
 
                            "disabling client {0} - Client never "
 
2606
                            "disabling client {} - Client never "
2455
2607
                            "performed a successful checker"
2456
2608
                            .format(client_name))
2457
2609
                        client["enabled"] = False
2458
2610
                    elif client["last_checker_status"] != 0:
2459
2611
                        logger.warning(
2460
 
                            "disabling client {0} - Client "
2461
 
                            "last checker failed with error code {1}"
 
2612
                            "disabling client {} - Client last"
 
2613
                            " checker failed with error code {}"
2462
2614
                            .format(client_name,
2463
2615
                                    client["last_checker_status"]))
2464
2616
                        client["enabled"] = False
2467
2619
                                             .utcnow()
2468
2620
                                             + client["timeout"])
2469
2621
                        logger.debug("Last checker succeeded,"
2470
 
                                     " keeping {0} enabled"
 
2622
                                     " keeping {} enabled"
2471
2623
                                     .format(client_name))
2472
2624
            try:
2473
2625
                client["secret"] = (
2476
2628
                                ["secret"]))
2477
2629
            except PGPError:
2478
2630
                # If decryption fails, we use secret from new settings
2479
 
                logger.debug("Failed to decrypt {0} old secret"
 
2631
                logger.debug("Failed to decrypt {} old secret"
2480
2632
                             .format(client_name))
2481
2633
                client["secret"] = (
2482
2634
                    client_settings[client_name]["secret"])
2490
2642
        clients_data[client_name] = client_settings[client_name]
2491
2643
    
2492
2644
    # Create all client objects
2493
 
    for client_name, client in clients_data.iteritems():
 
2645
    for client_name, client in clients_data.items():
2494
2646
        tcp_server.clients[client_name] = client_class(
2495
 
            name = client_name, settings = client)
 
2647
            name = client_name, settings = client,
 
2648
            server_settings = server_settings)
2496
2649
    
2497
2650
    if not tcp_server.clients:
2498
2651
        logger.warning("No clients defined")
2499
2652
    
2500
 
    if not debug:
2501
 
        try:
2502
 
            with pidfile:
2503
 
                pid = os.getpid()
2504
 
                pidfile.write(str(pid) + "\n".encode("utf-8"))
2505
 
            del pidfile
2506
 
        except IOError:
2507
 
            logger.error("Could not write to file %r with PID %d",
2508
 
                         pidfilename, pid)
2509
 
        except NameError:
2510
 
            # "pidfile" was never created
2511
 
            pass
 
2653
    if not foreground:
 
2654
        if pidfile is not None:
 
2655
            try:
 
2656
                with pidfile:
 
2657
                    pid = os.getpid()
 
2658
                    pidfile.write("{}\n".format(pid).encode("utf-8"))
 
2659
            except IOError:
 
2660
                logger.error("Could not write to file %r with PID %d",
 
2661
                             pidfilename, pid)
 
2662
        del pidfile
2512
2663
        del pidfilename
2513
2664
    
2514
2665
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
2580
2731
    
2581
2732
    def cleanup():
2582
2733
        "Cleanup function; run on exit"
2583
 
        service.cleanup()
 
2734
        if zeroconf:
 
2735
            service.cleanup()
2584
2736
        
2585
2737
        multiprocessing.active_children()
 
2738
        wnull.close()
2586
2739
        if not (tcp_server.clients or client_settings):
2587
2740
            return
2588
2741
        
2599
2752
                
2600
2753
                # A list of attributes that can not be pickled
2601
2754
                # + secret.
2602
 
                exclude = set(("bus", "changedstate", "secret",
2603
 
                               "checker"))
 
2755
                exclude = { "bus", "changedstate", "secret",
 
2756
                            "checker", "server_settings" }
2604
2757
                for name, typ in (inspect.getmembers
2605
2758
                                  (dbus.service.Object)):
2606
2759
                    exclude.add(name)
2629
2782
                except NameError:
2630
2783
                    pass
2631
2784
            if e.errno in (errno.ENOENT, errno.EACCES, errno.EEXIST):
2632
 
                logger.warning("Could not save persistent state: {0}"
 
2785
                logger.warning("Could not save persistent state: {}"
2633
2786
                               .format(os.strerror(e.errno)))
2634
2787
            else:
2635
2788
                logger.warning("Could not save persistent state:",
2636
2789
                               exc_info=e)
2637
 
                raise e
 
2790
                raise
2638
2791
        
2639
2792
        # Delete all clients, and settings from config
2640
2793
        while tcp_server.clients:
2664
2817
    tcp_server.server_activate()
2665
2818
    
2666
2819
    # Find out what port we got
2667
 
    service.port = tcp_server.socket.getsockname()[1]
 
2820
    if zeroconf:
 
2821
        service.port = tcp_server.socket.getsockname()[1]
2668
2822
    if use_ipv6:
2669
2823
        logger.info("Now listening on address %r, port %d,"
2670
2824
                    " flowinfo %d, scope_id %d",
2676
2830
    #service.interface = tcp_server.socket.getsockname()[3]
2677
2831
    
2678
2832
    try:
2679
 
        # From the Avahi example code
2680
 
        try:
2681
 
            service.activate()
2682
 
        except dbus.exceptions.DBusException as error:
2683
 
            logger.critical("D-Bus Exception", exc_info=error)
2684
 
            cleanup()
2685
 
            sys.exit(1)
2686
 
        # End of Avahi example code
 
2833
        if zeroconf:
 
2834
            # From the Avahi example code
 
2835
            try:
 
2836
                service.activate()
 
2837
            except dbus.exceptions.DBusException as error:
 
2838
                logger.critical("D-Bus Exception", exc_info=error)
 
2839
                cleanup()
 
2840
                sys.exit(1)
 
2841
            # End of Avahi example code
2687
2842
        
2688
2843
        gobject.io_add_watch(tcp_server.fileno(), gobject.IO_IN,
2689
2844
                             lambda *args, **kwargs: