157
130
                            u" after %i retries, exiting.",
 
158
131
                            self.rename_count)
 
159
132
            raise AvahiServiceError(u"Too many renames")
 
160
 
        self.name = unicode(self.server.GetAlternativeServiceName(self.name))
 
 
133
        self.name = server.GetAlternativeServiceName(self.name)
 
161
134
        logger.info(u"Changing Zeroconf service name to %r ...",
 
163
136
        syslogger.setFormatter(logging.Formatter
 
164
 
                               (u'Mandos (%s) [%%(process)d]:'
 
165
 
                                u' %%(levelname)s: %%(message)s'
 
 
137
                               ('Mandos (%s): %%(levelname)s:'
 
 
138
                                ' %%(message)s' % self.name))
 
170
 
        except dbus.exceptions.DBusException, error:
 
171
 
            logger.critical(u"DBusException: %s", error)
 
174
141
        self.rename_count += 1
 
175
142
    def remove(self):
 
176
143
        """Derived from the Avahi example code"""
 
177
 
        if self.group is not None:
 
 
144
        if group is not None:
 
180
147
        """Derived from the Avahi example code"""
 
181
 
        if self.group is None:
 
182
 
            self.group = dbus.Interface(
 
183
 
                self.bus.get_object(avahi.DBUS_NAME,
 
184
 
                                    self.server.EntryGroupNew()),
 
185
 
                avahi.DBUS_INTERFACE_ENTRY_GROUP)
 
186
 
            self.group.connect_to_signal('StateChanged',
 
188
 
                                         .entry_group_state_changed)
 
 
150
            group = dbus.Interface(bus.get_object
 
 
152
                                    server.EntryGroupNew()),
 
 
153
                                   avahi.DBUS_INTERFACE_ENTRY_GROUP)
 
 
154
            group.connect_to_signal('StateChanged',
 
 
155
                                    entry_group_state_changed)
 
189
156
        logger.debug(u"Adding Zeroconf service '%s' of type '%s' ...",
 
190
 
                     self.name, self.type)
 
191
 
        self.group.AddService(
 
194
 
            dbus.UInt32(0),     # flags
 
195
 
            self.name, self.type,
 
196
 
            self.domain, self.host,
 
197
 
            dbus.UInt16(self.port),
 
198
 
            avahi.string_array_to_txt_array(self.TXT))
 
200
 
    def entry_group_state_changed(self, state, error):
 
201
 
        """Derived from the Avahi example code"""
 
202
 
        logger.debug(u"Avahi entry group state change: %i", state)
 
204
 
        if state == avahi.ENTRY_GROUP_ESTABLISHED:
 
205
 
            logger.debug(u"Zeroconf service established.")
 
206
 
        elif state == avahi.ENTRY_GROUP_COLLISION:
 
207
 
            logger.info(u"Zeroconf service name collision.")
 
209
 
        elif state == avahi.ENTRY_GROUP_FAILURE:
 
210
 
            logger.critical(u"Avahi: Error in group state changed %s",
 
212
 
            raise AvahiGroupError(u"State changed: %s"
 
215
 
        """Derived from the Avahi example code"""
 
216
 
        if self.group is not None:
 
219
 
    def server_state_changed(self, state):
 
220
 
        """Derived from the Avahi example code"""
 
221
 
        logger.debug(u"Avahi server state change: %i", state)
 
222
 
        if state == avahi.SERVER_COLLISION:
 
223
 
            logger.error(u"Zeroconf server name collision")
 
225
 
        elif state == avahi.SERVER_RUNNING:
 
228
 
        """Derived from the Avahi example code"""
 
229
 
        if self.server is None:
 
230
 
            self.server = dbus.Interface(
 
231
 
                self.bus.get_object(avahi.DBUS_NAME,
 
232
 
                                    avahi.DBUS_PATH_SERVER),
 
233
 
                avahi.DBUS_INTERFACE_SERVER)
 
234
 
        self.server.connect_to_signal(u"StateChanged",
 
235
 
                                 self.server_state_changed)
 
236
 
        self.server_state_changed(self.server.GetState())
 
239
 
class Client(object):
 
 
157
                     service.name, service.type)
 
 
159
                self.interface,         # interface
 
 
160
                avahi.PROTO_INET6,      # protocol
 
 
161
                dbus.UInt32(0),         # flags
 
 
162
                self.name, self.type,
 
 
163
                self.domain, self.host,
 
 
164
                dbus.UInt16(self.port),
 
 
165
                avahi.string_array_to_txt_array(self.TXT))
 
 
168
# From the Avahi example code:
 
 
169
group = None                            # our entry group
 
 
170
# End of Avahi example code
 
 
173
def _datetime_to_dbus(dt, variant_level=0):
 
 
174
    """Convert a UTC datetime.datetime() to a D-Bus type."""
 
 
175
    return dbus.String(dt.isoformat(), variant_level=variant_level)
 
 
178
class Client(dbus.service.Object):
 
240
179
    """A representation of a client host served by this server.
 
243
 
    _approved:   bool(); 'None' if not yet approved/disapproved
 
244
 
    approval_delay: datetime.timedelta(); Time to wait for approval
 
245
 
    approval_duration: datetime.timedelta(); Duration of one approval
 
 
181
    name:       string; from the config file, used in log messages
 
 
182
    fingerprint: string (40 or 32 hexadecimal digits); used to
 
 
183
                 uniquely identify the client
 
 
184
    secret:     bytestring; sent verbatim (over TLS) to client
 
 
185
    host:       string; available for use by the checker command
 
 
186
    created:    datetime.datetime(); (UTC) object creation
 
 
187
    last_enabled: datetime.datetime(); (UTC)
 
 
189
    last_checked_ok: datetime.datetime(); (UTC) or None
 
 
190
    timeout:    datetime.timedelta(); How long from last_checked_ok
 
 
191
                                      until this client is invalid
 
 
192
    interval:   datetime.timedelta(); How often to start a new checker
 
 
193
    disable_hook:  If set, called by disable() as disable_hook(self)
 
246
194
    checker:    subprocess.Popen(); a running checker process used
 
247
195
                                    to see if the client lives.
 
248
196
                                    'None' if no process is running.
 
249
 
    checker_callback_tag: a gobject event source tag, or None
 
250
 
    checker_command: string; External command which is run to check
 
251
 
                     if client lives.  %() expansions are done at
 
 
197
    checker_initiator_tag: a gobject event source tag, or None
 
 
198
    disable_initiator_tag:    - '' -
 
 
199
    checker_callback_tag:  - '' -
 
 
200
    checker_command: string; External command which is run to check if
 
 
201
                     client lives.  %() expansions are done at
 
252
202
                     runtime with vars(self) as dict, so that for
 
253
203
                     instance %(name)s can be used in the command.
 
254
 
    checker_initiator_tag: a gobject event source tag, or None
 
255
 
    created:    datetime.datetime(); (UTC) object creation
 
256
 
    current_checker_command: string; current running checker_command
 
257
 
    disable_hook:  If set, called by disable() as disable_hook(self)
 
258
 
    disable_initiator_tag: a gobject event source tag, or None
 
260
 
    fingerprint: string (40 or 32 hexadecimal digits); used to
 
261
 
                 uniquely identify the client
 
262
 
    host:       string; available for use by the checker command
 
263
 
    interval:   datetime.timedelta(); How often to start a new checker
 
264
 
    last_approval_request: datetime.datetime(); (UTC) or None
 
265
 
    last_checked_ok: datetime.datetime(); (UTC) or None
 
266
 
    last_enabled: datetime.datetime(); (UTC)
 
267
 
    name:       string; from the config file, used in log messages and
 
269
 
    secret:     bytestring; sent verbatim (over TLS) to client
 
270
 
    timeout:    datetime.timedelta(); How long from last_checked_ok
 
271
 
                                      until this client is disabled
 
272
 
    runtime_expansions: Allowed attributes for runtime expansion.
 
 
204
    use_dbus: bool(); Whether to provide D-Bus interface and signals
 
 
205
    dbus_object_path: dbus.ObjectPath ; only set if self.use_dbus
 
275
 
    runtime_expansions = (u"approval_delay", u"approval_duration",
 
276
 
                          u"created", u"enabled", u"fingerprint",
 
277
 
                          u"host", u"interval", u"last_checked_ok",
 
278
 
                          u"last_enabled", u"name", u"timeout")
 
281
 
    def _timedelta_to_milliseconds(td):
 
282
 
        "Convert a datetime.timedelta() to milliseconds"
 
283
 
        return ((td.days * 24 * 60 * 60 * 1000)
 
284
 
                + (td.seconds * 1000)
 
285
 
                + (td.microseconds // 1000))
 
287
207
    def timeout_milliseconds(self):
 
288
208
        "Return the 'timeout' attribute in milliseconds"
 
289
 
        return self._timedelta_to_milliseconds(self.timeout)
 
 
209
        return ((self.timeout.days * 24 * 60 * 60 * 1000)
 
 
210
                + (self.timeout.seconds * 1000)
 
 
211
                + (self.timeout.microseconds // 1000))
 
291
213
    def interval_milliseconds(self):
 
292
214
        "Return the 'interval' attribute in milliseconds"
 
293
 
        return self._timedelta_to_milliseconds(self.interval)
 
295
 
    def approval_delay_milliseconds(self):
 
296
 
        return self._timedelta_to_milliseconds(self.approval_delay)
 
 
215
        return ((self.interval.days * 24 * 60 * 60 * 1000)
 
 
216
                + (self.interval.seconds * 1000)
 
 
217
                + (self.interval.microseconds // 1000))
 
298
 
    def __init__(self, name = None, disable_hook=None, config=None):
 
 
219
    def __init__(self, name = None, disable_hook=None, config=None,
 
299
221
        """Note: the 'checker' key in 'config' sets the
 
300
222
        'checker_command' attribute and *not* the 'checker'
 
 
507
421
        if self.checker_callback_tag:
 
508
422
            gobject.source_remove(self.checker_callback_tag)
 
509
423
            self.checker_callback_tag = None
 
510
 
        if getattr(self, u"checker", None) is None:
 
 
424
        if getattr(self, "checker", None) is None:
 
512
426
        logger.debug(u"Stopping checker for %(name)s", vars(self))
 
514
428
            os.kill(self.checker.pid, signal.SIGTERM)
 
516
430
            #if self.checker.poll() is None:
 
517
431
            #    os.kill(self.checker.pid, signal.SIGKILL)
 
518
432
        except OSError, error:
 
519
433
            if error.errno != errno.ESRCH: # No such process
 
521
435
        self.checker = None
 
523
 
def dbus_service_property(dbus_interface, signature=u"v",
 
524
 
                          access=u"readwrite", byte_arrays=False):
 
525
 
    """Decorators for marking methods of a DBusObjectWithProperties to
 
526
 
    become properties on the D-Bus.
 
528
 
    The decorated method will be called with no arguments by "Get"
 
529
 
    and with one argument by "Set".
 
531
 
    The parameters, where they are supported, are the same as
 
532
 
    dbus.service.method, except there is only "signature", since the
 
533
 
    type from Get() and the type sent to Set() is the same.
 
535
 
    # Encoding deeply encoded byte arrays is not supported yet by the
 
536
 
    # "Set" method, so we fail early here:
 
537
 
    if byte_arrays and signature != u"ay":
 
538
 
        raise ValueError(u"Byte arrays not supported for non-'ay'"
 
539
 
                         u" signature %r" % signature)
 
541
 
        func._dbus_is_property = True
 
542
 
        func._dbus_interface = dbus_interface
 
543
 
        func._dbus_signature = signature
 
544
 
        func._dbus_access = access
 
545
 
        func._dbus_name = func.__name__
 
546
 
        if func._dbus_name.endswith(u"_dbus_property"):
 
547
 
            func._dbus_name = func._dbus_name[:-14]
 
548
 
        func._dbus_get_args_options = {u'byte_arrays': byte_arrays }
 
553
 
class DBusPropertyException(dbus.exceptions.DBusException):
 
554
 
    """A base class for D-Bus property-related exceptions
 
556
 
    def __unicode__(self):
 
557
 
        return unicode(str(self))
 
560
 
class DBusPropertyAccessException(DBusPropertyException):
 
561
 
    """A property's access permissions disallows an operation.
 
566
 
class DBusPropertyNotFound(DBusPropertyException):
 
567
 
    """An attempt was made to access a non-existing property.
 
572
 
class DBusObjectWithProperties(dbus.service.Object):
 
573
 
    """A D-Bus object with properties.
 
575
 
    Classes inheriting from this can use the dbus_service_property
 
576
 
    decorator to expose methods as D-Bus properties.  It exposes the
 
577
 
    standard Get(), Set(), and GetAll() methods on the D-Bus.
 
581
 
    def _is_dbus_property(obj):
 
582
 
        return getattr(obj, u"_dbus_is_property", False)
 
584
 
    def _get_all_dbus_properties(self):
 
585
 
        """Returns a generator of (name, attribute) pairs
 
587
 
        return ((prop._dbus_name, prop)
 
589
 
                inspect.getmembers(self, self._is_dbus_property))
 
591
 
    def _get_dbus_property(self, interface_name, property_name):
 
592
 
        """Returns a bound method if one exists which is a D-Bus
 
593
 
        property with the specified name and interface.
 
595
 
        for name in (property_name,
 
596
 
                     property_name + u"_dbus_property"):
 
597
 
            prop = getattr(self, name, None)
 
599
 
                or not self._is_dbus_property(prop)
 
600
 
                or prop._dbus_name != property_name
 
601
 
                or (interface_name and prop._dbus_interface
 
602
 
                    and interface_name != prop._dbus_interface)):
 
606
 
        raise DBusPropertyNotFound(self.dbus_object_path + u":"
 
607
 
                                   + interface_name + u"."
 
610
 
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"ss",
 
612
 
    def Get(self, interface_name, property_name):
 
613
 
        """Standard D-Bus property Get() method, see D-Bus standard.
 
615
 
        prop = self._get_dbus_property(interface_name, property_name)
 
616
 
        if prop._dbus_access == u"write":
 
617
 
            raise DBusPropertyAccessException(property_name)
 
619
 
        if not hasattr(value, u"variant_level"):
 
621
 
        return type(value)(value, variant_level=value.variant_level+1)
 
623
 
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"ssv")
 
624
 
    def Set(self, interface_name, property_name, value):
 
625
 
        """Standard D-Bus property Set() method, see D-Bus standard.
 
627
 
        prop = self._get_dbus_property(interface_name, property_name)
 
628
 
        if prop._dbus_access == u"read":
 
629
 
            raise DBusPropertyAccessException(property_name)
 
630
 
        if prop._dbus_get_args_options[u"byte_arrays"]:
 
631
 
            # The byte_arrays option is not supported yet on
 
632
 
            # signatures other than "ay".
 
633
 
            if prop._dbus_signature != u"ay":
 
635
 
            value = dbus.ByteArray(''.join(unichr(byte)
 
639
 
    @dbus.service.method(dbus.PROPERTIES_IFACE, in_signature=u"s",
 
640
 
                         out_signature=u"a{sv}")
 
641
 
    def GetAll(self, interface_name):
 
642
 
        """Standard D-Bus property GetAll() method, see D-Bus
 
645
 
        Note: Will not include properties with access="write".
 
648
 
        for name, prop in self._get_all_dbus_properties():
 
650
 
                and interface_name != prop._dbus_interface):
 
651
 
                # Interface non-empty but did not match
 
653
 
            # Ignore write-only properties
 
654
 
            if prop._dbus_access == u"write":
 
657
 
            if not hasattr(value, u"variant_level"):
 
660
 
            all[name] = type(value)(value, variant_level=
 
661
 
                                    value.variant_level+1)
 
662
 
        return dbus.Dictionary(all, signature=u"sv")
 
664
 
    @dbus.service.method(dbus.INTROSPECTABLE_IFACE,
 
666
 
                         path_keyword='object_path',
 
667
 
                         connection_keyword='connection')
 
668
 
    def Introspect(self, object_path, connection):
 
669
 
        """Standard D-Bus method, overloaded to insert property tags.
 
671
 
        xmlstring = dbus.service.Object.Introspect(self, object_path,
 
674
 
            document = xml.dom.minidom.parseString(xmlstring)
 
675
 
            def make_tag(document, name, prop):
 
676
 
                e = document.createElement(u"property")
 
677
 
                e.setAttribute(u"name", name)
 
678
 
                e.setAttribute(u"type", prop._dbus_signature)
 
679
 
                e.setAttribute(u"access", prop._dbus_access)
 
681
 
            for if_tag in document.getElementsByTagName(u"interface"):
 
682
 
                for tag in (make_tag(document, name, prop)
 
684
 
                            in self._get_all_dbus_properties()
 
685
 
                            if prop._dbus_interface
 
686
 
                            == if_tag.getAttribute(u"name")):
 
687
 
                    if_tag.appendChild(tag)
 
688
 
                # Add the names to the return values for the
 
689
 
                # "org.freedesktop.DBus.Properties" methods
 
690
 
                if (if_tag.getAttribute(u"name")
 
691
 
                    == u"org.freedesktop.DBus.Properties"):
 
692
 
                    for cn in if_tag.getElementsByTagName(u"method"):
 
693
 
                        if cn.getAttribute(u"name") == u"Get":
 
694
 
                            for arg in cn.getElementsByTagName(u"arg"):
 
695
 
                                if (arg.getAttribute(u"direction")
 
697
 
                                    arg.setAttribute(u"name", u"value")
 
698
 
                        elif cn.getAttribute(u"name") == u"GetAll":
 
699
 
                            for arg in cn.getElementsByTagName(u"arg"):
 
700
 
                                if (arg.getAttribute(u"direction")
 
702
 
                                    arg.setAttribute(u"name", u"props")
 
703
 
            xmlstring = document.toxml(u"utf-8")
 
705
 
        except (AttributeError, xml.dom.DOMException,
 
706
 
                xml.parsers.expat.ExpatError), error:
 
707
 
            logger.error(u"Failed to override Introspection method",
 
712
 
class ClientDBus(Client, DBusObjectWithProperties):
 
713
 
    """A Client class using D-Bus
 
716
 
    dbus_object_path: dbus.ObjectPath
 
717
 
    bus: dbus.SystemBus()
 
720
 
    runtime_expansions = (Client.runtime_expansions
 
721
 
                          + (u"dbus_object_path",))
 
723
 
    # dbus.service.Object doesn't use super(), so we can't either.
 
725
 
    def __init__(self, bus = None, *args, **kwargs):
 
726
 
        self._approvals_pending = 0
 
728
 
        Client.__init__(self, *args, **kwargs)
 
729
 
        # Only now, when this client is initialized, can it show up on
 
731
 
        client_object_name = unicode(self.name).translate(
 
732
 
            {ord(u"."): ord(u"_"),
 
733
 
             ord(u"-"): ord(u"_")})
 
734
 
        self.dbus_object_path = (dbus.ObjectPath
 
735
 
                                 (u"/clients/" + client_object_name))
 
736
 
        DBusObjectWithProperties.__init__(self, self.bus,
 
737
 
                                          self.dbus_object_path)
 
739
 
    def _get_approvals_pending(self):
 
740
 
        return self._approvals_pending
 
741
 
    def _set_approvals_pending(self, value):
 
742
 
        old_value = self._approvals_pending
 
743
 
        self._approvals_pending = value
 
745
 
        if (hasattr(self, "dbus_object_path")
 
746
 
            and bval is not bool(old_value)):
 
747
 
            dbus_bool = dbus.Boolean(bval, variant_level=1)
 
748
 
            self.PropertyChanged(dbus.String(u"ApprovalPending"),
 
751
 
    approvals_pending = property(_get_approvals_pending,
 
752
 
                                 _set_approvals_pending)
 
753
 
    del _get_approvals_pending, _set_approvals_pending
 
756
 
    def _datetime_to_dbus(dt, variant_level=0):
 
757
 
        """Convert a UTC datetime.datetime() to a D-Bus type."""
 
758
 
        return dbus.String(dt.isoformat(),
 
759
 
                           variant_level=variant_level)
 
762
 
        oldstate = getattr(self, u"enabled", False)
 
763
 
        r = Client.enable(self)
 
764
 
        if oldstate != self.enabled:
 
766
 
            self.PropertyChanged(dbus.String(u"Enabled"),
 
767
 
                                 dbus.Boolean(True, variant_level=1))
 
768
 
            self.PropertyChanged(
 
769
 
                dbus.String(u"LastEnabled"),
 
770
 
                self._datetime_to_dbus(self.last_enabled,
 
774
 
    def disable(self, quiet = False):
 
775
 
        oldstate = getattr(self, u"enabled", False)
 
776
 
        r = Client.disable(self, quiet=quiet)
 
777
 
        if not quiet and oldstate != self.enabled:
 
779
 
            self.PropertyChanged(dbus.String(u"Enabled"),
 
780
 
                                 dbus.Boolean(False, variant_level=1))
 
783
 
    def __del__(self, *args, **kwargs):
 
785
 
            self.remove_from_connection()
 
788
 
        if hasattr(DBusObjectWithProperties, u"__del__"):
 
789
 
            DBusObjectWithProperties.__del__(self, *args, **kwargs)
 
790
 
        Client.__del__(self, *args, **kwargs)
 
792
 
    def checker_callback(self, pid, condition, command,
 
794
 
        self.checker_callback_tag = None
 
797
 
        self.PropertyChanged(dbus.String(u"CheckerRunning"),
 
798
 
                             dbus.Boolean(False, variant_level=1))
 
799
 
        if os.WIFEXITED(condition):
 
800
 
            exitstatus = os.WEXITSTATUS(condition)
 
802
 
            self.CheckerCompleted(dbus.Int16(exitstatus),
 
803
 
                                  dbus.Int64(condition),
 
804
 
                                  dbus.String(command))
 
807
 
            self.CheckerCompleted(dbus.Int16(-1),
 
808
 
                                  dbus.Int64(condition),
 
809
 
                                  dbus.String(command))
 
811
 
        return Client.checker_callback(self, pid, condition, command,
 
814
 
    def checked_ok(self, *args, **kwargs):
 
815
 
        r = Client.checked_ok(self, *args, **kwargs)
 
817
 
        self.PropertyChanged(
 
818
 
            dbus.String(u"LastCheckedOK"),
 
819
 
            (self._datetime_to_dbus(self.last_checked_ok,
 
823
 
    def need_approval(self, *args, **kwargs):
 
824
 
        r = Client.need_approval(self, *args, **kwargs)
 
826
 
        self.PropertyChanged(
 
827
 
            dbus.String(u"LastApprovalRequest"),
 
828
 
            (self._datetime_to_dbus(self.last_approval_request,
 
832
 
    def start_checker(self, *args, **kwargs):
 
833
 
        old_checker = self.checker
 
834
 
        if self.checker is not None:
 
835
 
            old_checker_pid = self.checker.pid
 
837
 
            old_checker_pid = None
 
838
 
        r = Client.start_checker(self, *args, **kwargs)
 
839
 
        # Only if new checker process was started
 
840
 
        if (self.checker is not None
 
841
 
            and old_checker_pid != self.checker.pid):
 
843
 
            self.CheckerStarted(self.current_checker_command)
 
844
 
            self.PropertyChanged(
 
845
 
                dbus.String(u"CheckerRunning"),
 
846
 
                dbus.Boolean(True, variant_level=1))
 
849
 
    def stop_checker(self, *args, **kwargs):
 
850
 
        old_checker = getattr(self, u"checker", None)
 
851
 
        r = Client.stop_checker(self, *args, **kwargs)
 
852
 
        if (old_checker is not None
 
853
 
            and getattr(self, u"checker", None) is None):
 
854
 
            self.PropertyChanged(dbus.String(u"CheckerRunning"),
 
855
 
                                 dbus.Boolean(False, variant_level=1))
 
858
 
    def _reset_approved(self):
 
859
 
        self._approved = None
 
862
 
    def approve(self, value=True):
 
863
 
        self.send_changedstate()
 
864
 
        self._approved = value
 
865
 
        gobject.timeout_add(self._timedelta_to_milliseconds
 
866
 
                            (self.approval_duration),
 
867
 
                            self._reset_approved)
 
870
 
    ## D-Bus methods, signals & properties
 
871
 
    _interface = u"se.bsnet.fukt.Mandos.Client"
 
 
437
            self.PropertyChanged(dbus.String(u"checker_running"),
 
 
438
                                 dbus.Boolean(False, variant_level=1))
 
 
440
    def still_valid(self):
 
 
441
        """Has the timeout not yet passed for this client?"""
 
 
442
        if not getattr(self, "enabled", False):
 
 
444
        now = datetime.datetime.utcnow()
 
 
445
        if self.last_checked_ok is None:
 
 
446
            return now < (self.created + self.timeout)
 
 
448
            return now < (self.last_checked_ok + self.timeout)
 
 
450
    ## D-Bus methods & signals
 
 
451
    _interface = u"org.mandos_system.Mandos.Client"
 
 
453
    # BumpTimeout - method
 
 
454
    BumpTimeout = dbus.service.method(_interface)(bump_timeout)
 
 
455
    BumpTimeout.__name__ = "BumpTimeout"
 
875
457
    # CheckerCompleted - signal
 
876
 
    @dbus.service.signal(_interface, signature=u"nxs")
 
877
 
    def CheckerCompleted(self, exitcode, waitstatus, command):
 
 
458
    @dbus.service.signal(_interface, signature="bqs")
 
 
459
    def CheckerCompleted(self, success, condition, command):
 
881
463
    # CheckerStarted - signal
 
882
 
    @dbus.service.signal(_interface, signature=u"s")
 
 
464
    @dbus.service.signal(_interface, signature="s")
 
883
465
    def CheckerStarted(self, command):
 
 
469
    # GetAllProperties - method
 
 
470
    @dbus.service.method(_interface, out_signature="a{sv}")
 
 
471
    def GetAllProperties(self):
 
 
473
        return dbus.Dictionary({
 
 
475
                    dbus.String(self.name, variant_level=1),
 
 
476
                dbus.String("fingerprint"):
 
 
477
                    dbus.String(self.fingerprint, variant_level=1),
 
 
479
                    dbus.String(self.host, variant_level=1),
 
 
480
                dbus.String("created"):
 
 
481
                    _datetime_to_dbus(self.created, variant_level=1),
 
 
482
                dbus.String("last_enabled"):
 
 
483
                    (_datetime_to_dbus(self.last_enabled,
 
 
485
                     if self.last_enabled is not None
 
 
486
                     else dbus.Boolean(False, variant_level=1)),
 
 
487
                dbus.String("enabled"):
 
 
488
                    dbus.Boolean(self.enabled, variant_level=1),
 
 
489
                dbus.String("last_checked_ok"):
 
 
490
                    (_datetime_to_dbus(self.last_checked_ok,
 
 
492
                     if self.last_checked_ok is not None
 
 
493
                     else dbus.Boolean (False, variant_level=1)),
 
 
494
                dbus.String("timeout"):
 
 
495
                    dbus.UInt64(self.timeout_milliseconds(),
 
 
497
                dbus.String("interval"):
 
 
498
                    dbus.UInt64(self.interval_milliseconds(),
 
 
500
                dbus.String("checker"):
 
 
501
                    dbus.String(self.checker_command,
 
 
503
                dbus.String("checker_running"):
 
 
504
                    dbus.Boolean(self.checker is not None,
 
 
508
    # IsStillValid - method
 
 
509
    IsStillValid = (dbus.service.method(_interface, out_signature="b")
 
 
511
    IsStillValid.__name__ = "IsStillValid"
 
887
513
    # PropertyChanged - signal
 
888
 
    @dbus.service.signal(_interface, signature=u"sv")
 
 
514
    @dbus.service.signal(_interface, signature="sv")
 
889
515
    def PropertyChanged(self, property, value):
 
894
 
    @dbus.service.signal(_interface)
 
897
 
        Is sent after a successful transfer of secret from the Mandos
 
898
 
        server to mandos-client
 
903
 
    @dbus.service.signal(_interface, signature=u"s")
 
904
 
    def Rejected(self, reason):
 
908
 
    # NeedApproval - signal
 
909
 
    @dbus.service.signal(_interface, signature=u"tb")
 
910
 
    def NeedApproval(self, timeout, default):
 
912
 
        return self.need_approval()
 
917
 
    @dbus.service.method(_interface, in_signature=u"b")
 
918
 
    def Approve(self, value):
 
922
 
    @dbus.service.method(_interface)
 
924
 
        return self.checked_ok()
 
 
519
    # SetChecker - method
 
 
520
    @dbus.service.method(_interface, in_signature="s")
 
 
521
    def SetChecker(self, checker):
 
 
522
        "D-Bus setter method"
 
 
523
        self.checker_command = checker
 
 
525
        self.PropertyChanged(dbus.String(u"checker"),
 
 
526
                             dbus.String(self.checker_command,
 
 
530
    @dbus.service.method(_interface, in_signature="s")
 
 
531
    def SetHost(self, host):
 
 
532
        "D-Bus setter method"
 
 
535
        self.PropertyChanged(dbus.String(u"host"),
 
 
536
                             dbus.String(self.host, variant_level=1))
 
 
538
    # SetInterval - method
 
 
539
    @dbus.service.method(_interface, in_signature="t")
 
 
540
    def SetInterval(self, milliseconds):
 
 
541
        self.interval = datetime.timedelta(0, 0, 0, milliseconds)
 
 
543
        self.PropertyChanged(dbus.String(u"interval"),
 
 
544
                             (dbus.UInt64(self.interval_milliseconds(),
 
 
548
    @dbus.service.method(_interface, in_signature="ay",
 
 
550
    def SetSecret(self, secret):
 
 
551
        "D-Bus setter method"
 
 
552
        self.secret = str(secret)
 
 
554
    # SetTimeout - method
 
 
555
    @dbus.service.method(_interface, in_signature="t")
 
 
556
    def SetTimeout(self, milliseconds):
 
 
557
        self.timeout = datetime.timedelta(0, 0, 0, milliseconds)
 
 
559
        self.PropertyChanged(dbus.String(u"timeout"),
 
 
560
                             (dbus.UInt64(self.timeout_milliseconds(),
 
926
563
    # Enable - method
 
927
 
    @dbus.service.method(_interface)
 
 
564
    Enable = dbus.service.method(_interface)(enable)
 
 
565
    Enable.__name__ = "Enable"
 
932
567
    # StartChecker - method
 
933
568
    @dbus.service.method(_interface)
 
 
944
579
    # StopChecker - method
 
945
 
    @dbus.service.method(_interface)
 
946
 
    def StopChecker(self):
 
951
 
    # ApprovalPending - property
 
952
 
    @dbus_service_property(_interface, signature=u"b", access=u"read")
 
953
 
    def ApprovalPending_dbus_property(self):
 
954
 
        return dbus.Boolean(bool(self.approvals_pending))
 
956
 
    # ApprovedByDefault - property
 
957
 
    @dbus_service_property(_interface, signature=u"b",
 
959
 
    def ApprovedByDefault_dbus_property(self, value=None):
 
960
 
        if value is None:       # get
 
961
 
            return dbus.Boolean(self.approved_by_default)
 
962
 
        self.approved_by_default = bool(value)
 
964
 
        self.PropertyChanged(dbus.String(u"ApprovedByDefault"),
 
965
 
                             dbus.Boolean(value, variant_level=1))
 
967
 
    # ApprovalDelay - property
 
968
 
    @dbus_service_property(_interface, signature=u"t",
 
970
 
    def ApprovalDelay_dbus_property(self, value=None):
 
971
 
        if value is None:       # get
 
972
 
            return dbus.UInt64(self.approval_delay_milliseconds())
 
973
 
        self.approval_delay = datetime.timedelta(0, 0, 0, value)
 
975
 
        self.PropertyChanged(dbus.String(u"ApprovalDelay"),
 
976
 
                             dbus.UInt64(value, variant_level=1))
 
978
 
    # ApprovalDuration - property
 
979
 
    @dbus_service_property(_interface, signature=u"t",
 
981
 
    def ApprovalDuration_dbus_property(self, value=None):
 
982
 
        if value is None:       # get
 
983
 
            return dbus.UInt64(self._timedelta_to_milliseconds(
 
984
 
                    self.approval_duration))
 
985
 
        self.approval_duration = datetime.timedelta(0, 0, 0, value)
 
987
 
        self.PropertyChanged(dbus.String(u"ApprovalDuration"),
 
988
 
                             dbus.UInt64(value, variant_level=1))
 
991
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
 
992
 
    def Name_dbus_property(self):
 
993
 
        return dbus.String(self.name)
 
995
 
    # Fingerprint - property
 
996
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
 
997
 
    def Fingerprint_dbus_property(self):
 
998
 
        return dbus.String(self.fingerprint)
 
1001
 
    @dbus_service_property(_interface, signature=u"s",
 
1002
 
                           access=u"readwrite")
 
1003
 
    def Host_dbus_property(self, value=None):
 
1004
 
        if value is None:       # get
 
1005
 
            return dbus.String(self.host)
 
1008
 
        self.PropertyChanged(dbus.String(u"Host"),
 
1009
 
                             dbus.String(value, variant_level=1))
 
1011
 
    # Created - property
 
1012
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
 
1013
 
    def Created_dbus_property(self):
 
1014
 
        return dbus.String(self._datetime_to_dbus(self.created))
 
1016
 
    # LastEnabled - property
 
1017
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
 
1018
 
    def LastEnabled_dbus_property(self):
 
1019
 
        if self.last_enabled is None:
 
1020
 
            return dbus.String(u"")
 
1021
 
        return dbus.String(self._datetime_to_dbus(self.last_enabled))
 
1023
 
    # Enabled - property
 
1024
 
    @dbus_service_property(_interface, signature=u"b",
 
1025
 
                           access=u"readwrite")
 
1026
 
    def Enabled_dbus_property(self, value=None):
 
1027
 
        if value is None:       # get
 
1028
 
            return dbus.Boolean(self.enabled)
 
1034
 
    # LastCheckedOK - property
 
1035
 
    @dbus_service_property(_interface, signature=u"s",
 
1036
 
                           access=u"readwrite")
 
1037
 
    def LastCheckedOK_dbus_property(self, value=None):
 
1038
 
        if value is not None:
 
1041
 
        if self.last_checked_ok is None:
 
1042
 
            return dbus.String(u"")
 
1043
 
        return dbus.String(self._datetime_to_dbus(self
 
1046
 
    # LastApprovalRequest - property
 
1047
 
    @dbus_service_property(_interface, signature=u"s", access=u"read")
 
1048
 
    def LastApprovalRequest_dbus_property(self):
 
1049
 
        if self.last_approval_request is None:
 
1050
 
            return dbus.String(u"")
 
1051
 
        return dbus.String(self.
 
1052
 
                           _datetime_to_dbus(self
 
1053
 
                                             .last_approval_request))
 
1055
 
    # Timeout - property
 
1056
 
    @dbus_service_property(_interface, signature=u"t",
 
1057
 
                           access=u"readwrite")
 
1058
 
    def Timeout_dbus_property(self, value=None):
 
1059
 
        if value is None:       # get
 
1060
 
            return dbus.UInt64(self.timeout_milliseconds())
 
1061
 
        self.timeout = datetime.timedelta(0, 0, 0, value)
 
1063
 
        self.PropertyChanged(dbus.String(u"Timeout"),
 
1064
 
                             dbus.UInt64(value, variant_level=1))
 
1065
 
        if getattr(self, u"disable_initiator_tag", None) is None:
 
1067
 
        # Reschedule timeout
 
1068
 
        gobject.source_remove(self.disable_initiator_tag)
 
1069
 
        self.disable_initiator_tag = None
 
1070
 
        time_to_die = (self.
 
1071
 
                       _timedelta_to_milliseconds((self
 
1076
 
        if time_to_die <= 0:
 
1077
 
            # The timeout has passed
 
1080
 
            self.disable_initiator_tag = (gobject.timeout_add
 
1081
 
                                          (time_to_die, self.disable))
 
1083
 
    # Interval - property
 
1084
 
    @dbus_service_property(_interface, signature=u"t",
 
1085
 
                           access=u"readwrite")
 
1086
 
    def Interval_dbus_property(self, value=None):
 
1087
 
        if value is None:       # get
 
1088
 
            return dbus.UInt64(self.interval_milliseconds())
 
1089
 
        self.interval = datetime.timedelta(0, 0, 0, value)
 
1091
 
        self.PropertyChanged(dbus.String(u"Interval"),
 
1092
 
                             dbus.UInt64(value, variant_level=1))
 
1093
 
        if getattr(self, u"checker_initiator_tag", None) is None:
 
1095
 
        # Reschedule checker run
 
1096
 
        gobject.source_remove(self.checker_initiator_tag)
 
1097
 
        self.checker_initiator_tag = (gobject.timeout_add
 
1098
 
                                      (value, self.start_checker))
 
1099
 
        self.start_checker()    # Start one now, too
 
1101
 
    # Checker - property
 
1102
 
    @dbus_service_property(_interface, signature=u"s",
 
1103
 
                           access=u"readwrite")
 
1104
 
    def Checker_dbus_property(self, value=None):
 
1105
 
        if value is None:       # get
 
1106
 
            return dbus.String(self.checker_command)
 
1107
 
        self.checker_command = value
 
1109
 
        self.PropertyChanged(dbus.String(u"Checker"),
 
1110
 
                             dbus.String(self.checker_command,
 
1113
 
    # CheckerRunning - property
 
1114
 
    @dbus_service_property(_interface, signature=u"b",
 
1115
 
                           access=u"readwrite")
 
1116
 
    def CheckerRunning_dbus_property(self, value=None):
 
1117
 
        if value is None:       # get
 
1118
 
            return dbus.Boolean(self.checker is not None)
 
1120
 
            self.start_checker()
 
1124
 
    # ObjectPath - property
 
1125
 
    @dbus_service_property(_interface, signature=u"o", access=u"read")
 
1126
 
    def ObjectPath_dbus_property(self):
 
1127
 
        return self.dbus_object_path # is already a dbus.ObjectPath
 
1130
 
    @dbus_service_property(_interface, signature=u"ay",
 
1131
 
                           access=u"write", byte_arrays=True)
 
1132
 
    def Secret_dbus_property(self, value):
 
1133
 
        self.secret = str(value)
 
 
580
    StopChecker = dbus.service.method(_interface)(stop_checker)
 
 
581
    StopChecker.__name__ = "StopChecker"
 
1138
 
class ProxyClient(object):
 
1139
 
    def __init__(self, child_pipe, fpr, address):
 
1140
 
        self._pipe = child_pipe
 
1141
 
        self._pipe.send(('init', fpr, address))
 
1142
 
        if not self._pipe.recv():
 
1145
 
    def __getattribute__(self, name):
 
1146
 
        if(name == '_pipe'):
 
1147
 
            return super(ProxyClient, self).__getattribute__(name)
 
1148
 
        self._pipe.send(('getattr', name))
 
1149
 
        data = self._pipe.recv()
 
1150
 
        if data[0] == 'data':
 
1152
 
        if data[0] == 'function':
 
1153
 
            def func(*args, **kwargs):
 
1154
 
                self._pipe.send(('funcall', name, args, kwargs))
 
1155
 
                return self._pipe.recv()[1]
 
1158
 
    def __setattr__(self, name, value):
 
1159
 
        if(name == '_pipe'):
 
1160
 
            return super(ProxyClient, self).__setattr__(name, value)
 
1161
 
        self._pipe.send(('setattr', name, value))
 
1164
 
class ClientHandler(socketserver.BaseRequestHandler, object):
 
1165
 
    """A class to handle client connections.
 
1167
 
    Instantiated once for each connection to handle it.
 
 
586
def peer_certificate(session):
 
 
587
    "Return the peer's OpenPGP certificate as a bytestring"
 
 
588
    # If not an OpenPGP certificate...
 
 
589
    if (gnutls.library.functions
 
 
590
        .gnutls_certificate_type_get(session._c_object)
 
 
591
        != gnutls.library.constants.GNUTLS_CRT_OPENPGP):
 
 
592
        # ...do the normal thing
 
 
593
        return session.peer_certificate
 
 
594
    list_size = ctypes.c_uint()
 
 
595
    cert_list = (gnutls.library.functions
 
 
596
                 .gnutls_certificate_get_peers
 
 
597
                 (session._c_object, ctypes.byref(list_size)))
 
 
598
    if list_size.value == 0:
 
 
601
    return ctypes.string_at(cert.data, cert.size)
 
 
604
def fingerprint(openpgp):
 
 
605
    "Convert an OpenPGP bytestring to a hexdigit fingerprint string"
 
 
606
    # New GnuTLS "datum" with the OpenPGP public key
 
 
607
    datum = (gnutls.library.types
 
 
608
             .gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
 
 
611
                             ctypes.c_uint(len(openpgp))))
 
 
612
    # New empty GnuTLS certificate
 
 
613
    crt = gnutls.library.types.gnutls_openpgp_crt_t()
 
 
614
    (gnutls.library.functions
 
 
615
     .gnutls_openpgp_crt_init(ctypes.byref(crt)))
 
 
616
    # Import the OpenPGP public key into the certificate
 
 
617
    (gnutls.library.functions
 
 
618
     .gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
 
 
619
                                gnutls.library.constants
 
 
620
                                .GNUTLS_OPENPGP_FMT_RAW))
 
 
621
    # Verify the self signature in the key
 
 
622
    crtverify = ctypes.c_uint()
 
 
623
    (gnutls.library.functions
 
 
624
     .gnutls_openpgp_crt_verify_self(crt, 0, ctypes.byref(crtverify)))
 
 
625
    if crtverify.value != 0:
 
 
626
        gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
 
 
627
        raise gnutls.errors.CertificateSecurityError("Verify failed")
 
 
628
    # New buffer for the fingerprint
 
 
629
    buf = ctypes.create_string_buffer(20)
 
 
630
    buf_len = ctypes.c_size_t()
 
 
631
    # Get the fingerprint from the certificate into the buffer
 
 
632
    (gnutls.library.functions
 
 
633
     .gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
 
 
634
                                         ctypes.byref(buf_len)))
 
 
635
    # Deinit the certificate
 
 
636
    gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
 
 
637
    # Convert the buffer to a Python bytestring
 
 
638
    fpr = ctypes.string_at(buf, buf_len.value)
 
 
639
    # Convert the bytestring to hexadecimal notation
 
 
640
    hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
 
 
644
class TCP_handler(SocketServer.BaseRequestHandler, object):
 
 
645
    """A TCP request handler class.
 
 
646
    Instantiated by IPv6_TCPServer for each request to handle it.
 
1168
647
    Note: This will run in its own forked process."""
 
1170
649
    def handle(self):
 
1171
 
        with contextlib.closing(self.server.child_pipe) as child_pipe:
 
1172
 
            logger.info(u"TCP connection from: %s",
 
1173
 
                        unicode(self.client_address))
 
1174
 
            logger.debug(u"Pipe FD: %d",
 
1175
 
                         self.server.child_pipe.fileno())
 
1177
 
            session = (gnutls.connection
 
1178
 
                       .ClientSession(self.request,
 
1180
 
                                      .X509Credentials()))
 
1182
 
            # Note: gnutls.connection.X509Credentials is really a
 
1183
 
            # generic GnuTLS certificate credentials object so long as
 
1184
 
            # no X.509 keys are added to it.  Therefore, we can use it
 
1185
 
            # here despite using OpenPGP certificates.
 
1187
 
            #priority = u':'.join((u"NONE", u"+VERS-TLS1.1",
 
1188
 
            #                      u"+AES-256-CBC", u"+SHA1",
 
1189
 
            #                      u"+COMP-NULL", u"+CTYPE-OPENPGP",
 
1191
 
            # Use a fallback default, since this MUST be set.
 
1192
 
            priority = self.server.gnutls_priority
 
1193
 
            if priority is None:
 
1194
 
                priority = u"NORMAL"
 
1195
 
            (gnutls.library.functions
 
1196
 
             .gnutls_priority_set_direct(session._c_object,
 
1199
 
            # Start communication using the Mandos protocol
 
1200
 
            # Get protocol number
 
1201
 
            line = self.request.makefile().readline()
 
1202
 
            logger.debug(u"Protocol version: %r", line)
 
1204
 
                if int(line.strip().split()[0]) > 1:
 
1206
 
            except (ValueError, IndexError, RuntimeError), error:
 
1207
 
                logger.error(u"Unknown protocol version: %s", error)
 
1210
 
            # Start GnuTLS connection
 
1213
 
            except gnutls.errors.GNUTLSError, error:
 
1214
 
                logger.warning(u"Handshake failed: %s", error)
 
1215
 
                # Do not run session.bye() here: the session is not
 
1216
 
                # established.  Just abandon the request.
 
1218
 
            logger.debug(u"Handshake succeeded")
 
1220
 
            approval_required = False
 
1223
 
                    fpr = self.fingerprint(self.peer_certificate
 
1225
 
                except (TypeError, gnutls.errors.GNUTLSError), error:
 
1226
 
                    logger.warning(u"Bad certificate: %s", error)
 
1228
 
                logger.debug(u"Fingerprint: %s", fpr)
 
1231
 
                    client = ProxyClient(child_pipe, fpr,
 
1232
 
                                         self.client_address)
 
1236
 
                if client.approval_delay:
 
1237
 
                    delay = client.approval_delay
 
1238
 
                    client.approvals_pending += 1
 
1239
 
                    approval_required = True
 
1242
 
                    if not client.enabled:
 
1243
 
                        logger.warning(u"Client %s is disabled",
 
1245
 
                        if self.server.use_dbus:
 
1247
 
                            client.Rejected("Disabled")                    
 
1250
 
                    if client._approved or not client.approval_delay:
 
1251
 
                        #We are approved or approval is disabled
 
1253
 
                    elif client._approved is None:
 
1254
 
                        logger.info(u"Client %s needs approval",
 
1256
 
                        if self.server.use_dbus:
 
1258
 
                            client.NeedApproval(
 
1259
 
                                client.approval_delay_milliseconds(),
 
1260
 
                                client.approved_by_default)
 
1262
 
                        logger.warning(u"Client %s was not approved",
 
1264
 
                        if self.server.use_dbus:
 
1266
 
                            client.Rejected("Denied")
 
1269
 
                    #wait until timeout or approved
 
1270
 
                    #x = float(client._timedelta_to_milliseconds(delay))
 
1271
 
                    time = datetime.datetime.now()
 
1272
 
                    client.changedstate.acquire()
 
1273
 
                    client.changedstate.wait(float(client._timedelta_to_milliseconds(delay) / 1000))
 
1274
 
                    client.changedstate.release()
 
1275
 
                    time2 = datetime.datetime.now()
 
1276
 
                    if (time2 - time) >= delay:
 
1277
 
                        if not client.approved_by_default:
 
1278
 
                            logger.warning("Client %s timed out while"
 
1279
 
                                           " waiting for approval",
 
1281
 
                            if self.server.use_dbus:
 
1283
 
                                client.Rejected("Approval timed out")
 
1288
 
                        delay -= time2 - time
 
1291
 
                while sent_size < len(client.secret):
 
1293
 
                        sent = session.send(client.secret[sent_size:])
 
1294
 
                    except (gnutls.errors.GNUTLSError), error:
 
1295
 
                        logger.warning("gnutls send failed")
 
1297
 
                    logger.debug(u"Sent: %d, remaining: %d",
 
1298
 
                                 sent, len(client.secret)
 
1299
 
                                 - (sent_size + sent))
 
1302
 
                logger.info(u"Sending secret to %s", client.name)
 
1303
 
                # bump the timeout as if seen
 
1305
 
                if self.server.use_dbus:
 
1310
 
                if approval_required:
 
1311
 
                    client.approvals_pending -= 1
 
1314
 
                except (gnutls.errors.GNUTLSError), error:
 
1315
 
                    logger.warning("GnuTLS bye failed")
 
1318
 
    def peer_certificate(session):
 
1319
 
        "Return the peer's OpenPGP certificate as a bytestring"
 
1320
 
        # If not an OpenPGP certificate...
 
1321
 
        if (gnutls.library.functions
 
1322
 
            .gnutls_certificate_type_get(session._c_object)
 
1323
 
            != gnutls.library.constants.GNUTLS_CRT_OPENPGP):
 
1324
 
            # ...do the normal thing
 
1325
 
            return session.peer_certificate
 
1326
 
        list_size = ctypes.c_uint(1)
 
1327
 
        cert_list = (gnutls.library.functions
 
1328
 
                     .gnutls_certificate_get_peers
 
1329
 
                     (session._c_object, ctypes.byref(list_size)))
 
1330
 
        if not bool(cert_list) and list_size.value != 0:
 
1331
 
            raise gnutls.errors.GNUTLSError(u"error getting peer"
 
1333
 
        if list_size.value == 0:
 
1336
 
        return ctypes.string_at(cert.data, cert.size)
 
1339
 
    def fingerprint(openpgp):
 
1340
 
        "Convert an OpenPGP bytestring to a hexdigit fingerprint"
 
1341
 
        # New GnuTLS "datum" with the OpenPGP public key
 
1342
 
        datum = (gnutls.library.types
 
1343
 
                 .gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
 
1346
 
                                 ctypes.c_uint(len(openpgp))))
 
1347
 
        # New empty GnuTLS certificate
 
1348
 
        crt = gnutls.library.types.gnutls_openpgp_crt_t()
 
1349
 
        (gnutls.library.functions
 
1350
 
         .gnutls_openpgp_crt_init(ctypes.byref(crt)))
 
1351
 
        # Import the OpenPGP public key into the certificate
 
1352
 
        (gnutls.library.functions
 
1353
 
         .gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
 
1354
 
                                    gnutls.library.constants
 
1355
 
                                    .GNUTLS_OPENPGP_FMT_RAW))
 
1356
 
        # Verify the self signature in the key
 
1357
 
        crtverify = ctypes.c_uint()
 
1358
 
        (gnutls.library.functions
 
1359
 
         .gnutls_openpgp_crt_verify_self(crt, 0,
 
1360
 
                                         ctypes.byref(crtverify)))
 
1361
 
        if crtverify.value != 0:
 
1362
 
            gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
 
1363
 
            raise (gnutls.errors.CertificateSecurityError
 
1365
 
        # New buffer for the fingerprint
 
1366
 
        buf = ctypes.create_string_buffer(20)
 
1367
 
        buf_len = ctypes.c_size_t()
 
1368
 
        # Get the fingerprint from the certificate into the buffer
 
1369
 
        (gnutls.library.functions
 
1370
 
         .gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
 
1371
 
                                             ctypes.byref(buf_len)))
 
1372
 
        # Deinit the certificate
 
1373
 
        gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
 
1374
 
        # Convert the buffer to a Python bytestring
 
1375
 
        fpr = ctypes.string_at(buf, buf_len.value)
 
1376
 
        # Convert the bytestring to hexadecimal notation
 
1377
 
        hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
 
1381
 
class MultiprocessingMixIn(object):
 
1382
 
    """Like socketserver.ThreadingMixIn, but with multiprocessing"""
 
1383
 
    def sub_process_main(self, request, address):
 
1385
 
            self.finish_request(request, address)
 
1387
 
            self.handle_error(request, address)
 
1388
 
        self.close_request(request)
 
1390
 
    def process_request(self, request, address):
 
1391
 
        """Start a new process to process the request."""
 
1392
 
        multiprocessing.Process(target = self.sub_process_main,
 
1393
 
                                args = (request, address)).start()
 
1395
 
class MultiprocessingMixInWithPipe(MultiprocessingMixIn, object):
 
1396
 
    """ adds a pipe to the MixIn """
 
1397
 
    def process_request(self, request, client_address):
 
1398
 
        """Overrides and wraps the original process_request().
 
1400
 
        This function creates a new pipe in self.pipe
 
1402
 
        parent_pipe, self.child_pipe = multiprocessing.Pipe()
 
1404
 
        super(MultiprocessingMixInWithPipe,
 
1405
 
              self).process_request(request, client_address)
 
1406
 
        self.child_pipe.close()
 
1407
 
        self.add_pipe(parent_pipe)
 
1409
 
    def add_pipe(self, parent_pipe):
 
1410
 
        """Dummy function; override as necessary"""
 
1413
 
class IPv6_TCPServer(MultiprocessingMixInWithPipe,
 
1414
 
                     socketserver.TCPServer, object):
 
1415
 
    """IPv6-capable TCP server.  Accepts 'None' as address and/or port
 
 
650
        logger.info(u"TCP connection from: %s",
 
 
651
                    unicode(self.client_address))
 
 
652
        session = (gnutls.connection
 
 
653
                   .ClientSession(self.request,
 
 
657
        line = self.request.makefile().readline()
 
 
658
        logger.debug(u"Protocol version: %r", line)
 
 
660
            if int(line.strip().split()[0]) > 1:
 
 
662
        except (ValueError, IndexError, RuntimeError), error:
 
 
663
            logger.error(u"Unknown protocol version: %s", error)
 
 
666
        # Note: gnutls.connection.X509Credentials is really a generic
 
 
667
        # GnuTLS certificate credentials object so long as no X.509
 
 
668
        # keys are added to it.  Therefore, we can use it here despite
 
 
669
        # using OpenPGP certificates.
 
 
671
        #priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
 
 
672
        #                "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
 
 
674
        # Use a fallback default, since this MUST be set.
 
 
675
        priority = self.server.settings.get("priority", "NORMAL")
 
 
676
        (gnutls.library.functions
 
 
677
         .gnutls_priority_set_direct(session._c_object,
 
 
682
        except gnutls.errors.GNUTLSError, error:
 
 
683
            logger.warning(u"Handshake failed: %s", error)
 
 
684
            # Do not run session.bye() here: the session is not
 
 
685
            # established.  Just abandon the request.
 
 
688
            fpr = fingerprint(peer_certificate(session))
 
 
689
        except (TypeError, gnutls.errors.GNUTLSError), error:
 
 
690
            logger.warning(u"Bad certificate: %s", error)
 
 
693
        logger.debug(u"Fingerprint: %s", fpr)
 
 
694
        for c in self.server.clients:
 
 
695
            if c.fingerprint == fpr:
 
 
699
            logger.warning(u"Client not found for fingerprint: %s",
 
 
703
        # Have to check if client.still_valid(), since it is possible
 
 
704
        # that the client timed out while establishing the GnuTLS
 
 
706
        if not client.still_valid():
 
 
707
            logger.warning(u"Client %(name)s is invalid",
 
 
711
        ## This won't work here, since we're in a fork.
 
 
712
        # client.bump_timeout()
 
 
714
        while sent_size < len(client.secret):
 
 
715
            sent = session.send(client.secret[sent_size:])
 
 
716
            logger.debug(u"Sent: %d, remaining: %d",
 
 
717
                         sent, len(client.secret)
 
 
718
                         - (sent_size + sent))
 
 
723
class IPv6_TCPServer(SocketServer.ForkingMixIn,
 
 
724
                     SocketServer.TCPServer, object):
 
 
725
    """IPv6 TCP server.  Accepts 'None' as address and/or port.
 
 
727
        settings:       Server settings
 
 
728
        clients:        Set() of Client objects
 
1418
729
        enabled:        Boolean; whether this server is activated yet
 
1419
 
        interface:      None or a network interface name (string)
 
1420
 
        use_ipv6:       Boolean; to use IPv6 or not
 
1422
 
    def __init__(self, server_address, RequestHandlerClass,
 
1423
 
                 interface=None, use_ipv6=True):
 
1424
 
        self.interface = interface
 
1426
 
            self.address_family = socket.AF_INET6
 
1427
 
        socketserver.TCPServer.__init__(self, server_address,
 
1428
 
                                        RequestHandlerClass)
 
 
731
    address_family = socket.AF_INET6
 
 
732
    def __init__(self, *args, **kwargs):
 
 
733
        if "settings" in kwargs:
 
 
734
            self.settings = kwargs["settings"]
 
 
735
            del kwargs["settings"]
 
 
736
        if "clients" in kwargs:
 
 
737
            self.clients = kwargs["clients"]
 
 
738
            del kwargs["clients"]
 
 
740
        super(IPv6_TCPServer, self).__init__(*args, **kwargs)
 
1429
741
    def server_bind(self):
 
1430
742
        """This overrides the normal server_bind() function
 
1431
743
        to bind to an interface if one was specified, and also NOT to
 
1432
744
        bind to an address or port if they were not specified."""
 
1433
 
        if self.interface is not None:
 
1434
 
            if SO_BINDTODEVICE is None:
 
1435
 
                logger.error(u"SO_BINDTODEVICE does not exist;"
 
1436
 
                             u" cannot bind to interface %s",
 
1440
 
                    self.socket.setsockopt(socket.SOL_SOCKET,
 
1444
 
                except socket.error, error:
 
1445
 
                    if error[0] == errno.EPERM:
 
1446
 
                        logger.error(u"No permission to"
 
1447
 
                                     u" bind to interface %s",
 
1449
 
                    elif error[0] == errno.ENOPROTOOPT:
 
1450
 
                        logger.error(u"SO_BINDTODEVICE not available;"
 
1451
 
                                     u" cannot bind to interface %s",
 
 
745
        if self.settings["interface"]:
 
 
746
            # 25 is from /usr/include/asm-i486/socket.h
 
 
747
            SO_BINDTODEVICE = getattr(socket, "SO_BINDTODEVICE", 25)
 
 
749
                self.socket.setsockopt(socket.SOL_SOCKET,
 
 
751
                                       self.settings["interface"])
 
 
752
            except socket.error, error:
 
 
753
                if error[0] == errno.EPERM:
 
 
754
                    logger.error(u"No permission to"
 
 
755
                                 u" bind to interface %s",
 
 
756
                                 self.settings["interface"])
 
1455
759
        # Only bind(2) the socket if we really need to.
 
1456
760
        if self.server_address[0] or self.server_address[1]:
 
1457
761
            if not self.server_address[0]:
 
1458
 
                if self.address_family == socket.AF_INET6:
 
1459
 
                    any_address = u"::" # in6addr_any
 
1461
 
                    any_address = socket.INADDR_ANY
 
1462
 
                self.server_address = (any_address,
 
 
763
                self.server_address = (in6addr_any,
 
1463
764
                                       self.server_address[1])
 
1464
765
            elif not self.server_address[1]:
 
1465
766
                self.server_address = (self.server_address[0],
 
1467
 
#                 if self.interface:
 
 
768
#                 if self.settings["interface"]:
 
1468
769
#                     self.server_address = (self.server_address[0],
 
1471
772
#                                            if_nametoindex
 
1473
 
            return socketserver.TCPServer.server_bind(self)
 
1476
 
class MandosServer(IPv6_TCPServer):
 
1480
 
        clients:        set of Client objects
 
1481
 
        gnutls_priority GnuTLS priority string
 
1482
 
        use_dbus:       Boolean; to emit D-Bus signals or not
 
1484
 
    Assumes a gobject.MainLoop event loop.
 
1486
 
    def __init__(self, server_address, RequestHandlerClass,
 
1487
 
                 interface=None, use_ipv6=True, clients=None,
 
1488
 
                 gnutls_priority=None, use_dbus=True):
 
1489
 
        self.enabled = False
 
1490
 
        self.clients = clients
 
1491
 
        if self.clients is None:
 
1492
 
            self.clients = set()
 
1493
 
        self.use_dbus = use_dbus
 
1494
 
        self.gnutls_priority = gnutls_priority
 
1495
 
        IPv6_TCPServer.__init__(self, server_address,
 
1496
 
                                RequestHandlerClass,
 
1497
 
                                interface = interface,
 
1498
 
                                use_ipv6 = use_ipv6)
 
 
775
            return super(IPv6_TCPServer, self).server_bind()
 
1499
776
    def server_activate(self):
 
1500
777
        if self.enabled:
 
1501
 
            return socketserver.TCPServer.server_activate(self)
 
 
778
            return super(IPv6_TCPServer, self).server_activate()
 
1502
779
    def enable(self):
 
1503
780
        self.enabled = True
 
1504
 
    def add_pipe(self, parent_pipe):
 
1505
 
        # Call "handle_ipc" for both data and EOF events
 
1506
 
        gobject.io_add_watch(parent_pipe.fileno(),
 
1507
 
                             gobject.IO_IN | gobject.IO_HUP,
 
1508
 
                             functools.partial(self.handle_ipc,
 
1509
 
                                               parent_pipe = parent_pipe))
 
1511
 
    def handle_ipc(self, source, condition, parent_pipe=None,
 
1512
 
                   client_object=None):
 
1514
 
            gobject.IO_IN: u"IN",   # There is data to read.
 
1515
 
            gobject.IO_OUT: u"OUT", # Data can be written (without
 
1517
 
            gobject.IO_PRI: u"PRI", # There is urgent data to read.
 
1518
 
            gobject.IO_ERR: u"ERR", # Error condition.
 
1519
 
            gobject.IO_HUP: u"HUP"  # Hung up (the connection has been
 
1520
 
                                    # broken, usually for pipes and
 
1523
 
        conditions_string = ' | '.join(name
 
1525
 
                                       condition_names.iteritems()
 
1526
 
                                       if cond & condition)
 
1527
 
        # error or the other end of multiprocessing.Pipe has closed
 
1528
 
        if condition & (gobject.IO_ERR | condition & gobject.IO_HUP):
 
1531
 
        # Read a request from the child
 
1532
 
        request = parent_pipe.recv()
 
1533
 
        command = request[0]
 
1535
 
        if command == 'init':
 
1537
 
            address = request[2]
 
1539
 
            for c in self.clients:
 
1540
 
                if c.fingerprint == fpr:
 
1544
 
                logger.warning(u"Client not found for fingerprint: %s, ad"
 
1545
 
                               u"dress: %s", fpr, address)
 
1548
 
                    mandos_dbus_service.ClientNotFound(fpr, address[0])
 
1549
 
                parent_pipe.send(False)
 
1552
 
            gobject.io_add_watch(parent_pipe.fileno(),
 
1553
 
                                 gobject.IO_IN | gobject.IO_HUP,
 
1554
 
                                 functools.partial(self.handle_ipc,
 
1555
 
                                                   parent_pipe = parent_pipe,
 
1556
 
                                                   client_object = client))
 
1557
 
            parent_pipe.send(True)
 
1558
 
            # remove the old hook in favor of the new above hook on same fileno
 
1560
 
        if command == 'funcall':
 
1561
 
            funcname = request[1]
 
1565
 
            parent_pipe.send(('data', getattr(client_object, funcname)(*args, **kwargs)))
 
1567
 
        if command == 'getattr':
 
1568
 
            attrname = request[1]
 
1569
 
            if callable(client_object.__getattribute__(attrname)):
 
1570
 
                parent_pipe.send(('function',))
 
1572
 
                parent_pipe.send(('data', client_object.__getattribute__(attrname)))
 
1574
 
        if command == 'setattr':
 
1575
 
            attrname = request[1]
 
1577
 
            setattr(client_object, attrname, value)
 
1582
783
def string_to_delta(interval):
 
1583
784
    """Parse a string and return a datetime.timedelta
 
1585
 
    >>> string_to_delta(u'7d')
 
 
786
    >>> string_to_delta('7d')
 
1586
787
    datetime.timedelta(7)
 
1587
 
    >>> string_to_delta(u'60s')
 
 
788
    >>> string_to_delta('60s')
 
1588
789
    datetime.timedelta(0, 60)
 
1589
 
    >>> string_to_delta(u'60m')
 
 
790
    >>> string_to_delta('60m')
 
1590
791
    datetime.timedelta(0, 3600)
 
1591
 
    >>> string_to_delta(u'24h')
 
 
792
    >>> string_to_delta('24h')
 
1592
793
    datetime.timedelta(1)
 
1593
794
    >>> string_to_delta(u'1w')
 
1594
795
    datetime.timedelta(7)
 
1595
 
    >>> string_to_delta(u'5m 30s')
 
 
796
    >>> string_to_delta('5m 30s')
 
1596
797
    datetime.timedelta(0, 330)
 
1598
799
    timevalue = datetime.timedelta(0)
 
 
1709
923
    # Default values for config file for server-global settings
 
1710
 
    server_defaults = { u"interface": u"",
 
1715
 
                        u"SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
 
1716
 
                        u"servicename": u"Mandos",
 
1717
 
                        u"use_dbus": u"True",
 
1718
 
                        u"use_ipv6": u"True",
 
 
924
    server_defaults = { "interface": "",
 
 
929
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
 
 
930
                        "servicename": "Mandos",
 
1722
934
    # Parse config file for server-global settings
 
1723
 
    server_config = configparser.SafeConfigParser(server_defaults)
 
 
935
    server_config = ConfigParser.SafeConfigParser(server_defaults)
 
1724
936
    del server_defaults
 
1725
 
    server_config.read(os.path.join(options.configdir,
 
 
937
    server_config.read(os.path.join(options.configdir, "mandos.conf"))
 
1727
938
    # Convert the SafeConfigParser object to a dict
 
1728
939
    server_settings = server_config.defaults()
 
1729
 
    # Use the appropriate methods on the non-string config options
 
1730
 
    for option in (u"debug", u"use_dbus", u"use_ipv6"):
 
1731
 
        server_settings[option] = server_config.getboolean(u"DEFAULT",
 
1733
 
    if server_settings["port"]:
 
1734
 
        server_settings["port"] = server_config.getint(u"DEFAULT",
 
 
940
    # Use getboolean on the boolean config options
 
 
941
    server_settings["debug"] = (server_config.getboolean
 
 
942
                                ("DEFAULT", "debug"))
 
 
943
    server_settings["use_dbus"] = (server_config.getboolean
 
 
944
                                   ("DEFAULT", "use_dbus"))
 
1736
945
    del server_config
 
1738
947
    # Override the settings from the config file with command line
 
1739
948
    # options, if set.
 
1740
 
    for option in (u"interface", u"address", u"port", u"debug",
 
1741
 
                   u"priority", u"servicename", u"configdir",
 
1742
 
                   u"use_dbus", u"use_ipv6", u"debuglevel"):
 
 
949
    for option in ("interface", "address", "port", "debug",
 
 
950
                   "priority", "servicename", "configdir",
 
1743
952
        value = getattr(options, option)
 
1744
953
        if value is not None:
 
1745
954
            server_settings[option] = value
 
1747
 
    # Force all strings to be unicode
 
1748
 
    for option in server_settings.keys():
 
1749
 
        if type(server_settings[option]) is str:
 
1750
 
            server_settings[option] = unicode(server_settings[option])
 
1751
956
    # Now we have our good server settings in "server_settings"
 
1753
 
    ##################################################################
 
1755
958
    # For convenience
 
1756
 
    debug = server_settings[u"debug"]
 
1757
 
    debuglevel = server_settings[u"debuglevel"]
 
1758
 
    use_dbus = server_settings[u"use_dbus"]
 
1759
 
    use_ipv6 = server_settings[u"use_ipv6"]
 
1761
 
    if server_settings[u"servicename"] != u"Mandos":
 
 
959
    debug = server_settings["debug"]
 
 
960
    use_dbus = server_settings["use_dbus"]
 
 
963
        syslogger.setLevel(logging.WARNING)
 
 
964
        console.setLevel(logging.WARNING)
 
 
966
    if server_settings["servicename"] != "Mandos":
 
1762
967
        syslogger.setFormatter(logging.Formatter
 
1763
 
                               (u'Mandos (%s) [%%(process)d]:'
 
1764
 
                                u' %%(levelname)s: %%(message)s'
 
1765
 
                                % server_settings[u"servicename"]))
 
 
968
                               ('Mandos (%s): %%(levelname)s:'
 
 
970
                                % server_settings["servicename"]))
 
1767
972
    # Parse config file with clients
 
1768
 
    client_defaults = { u"timeout": u"1h",
 
1770
 
                        u"checker": u"fping -q -- %%(host)s",
 
1772
 
                        u"approval_delay": u"0s",
 
1773
 
                        u"approval_duration": u"1s",
 
 
973
    client_defaults = { "timeout": "1h",
 
 
975
                        "checker": "fping -q -- %%(host)s",
 
1775
 
    client_config = configparser.SafeConfigParser(client_defaults)
 
1776
 
    client_config.read(os.path.join(server_settings[u"configdir"],
 
1779
 
    global mandos_dbus_service
 
1780
 
    mandos_dbus_service = None
 
1782
 
    tcp_server = MandosServer((server_settings[u"address"],
 
1783
 
                               server_settings[u"port"]),
 
1785
 
                              interface=(server_settings[u"interface"]
 
1789
 
                              server_settings[u"priority"],
 
1792
 
        pidfilename = u"/var/run/mandos.pid"
 
1794
 
            pidfile = open(pidfilename, u"w")
 
1796
 
            logger.error(u"Could not open file %r", pidfilename)
 
1799
 
        uid = pwd.getpwnam(u"_mandos").pw_uid
 
1800
 
        gid = pwd.getpwnam(u"_mandos").pw_gid
 
 
978
    client_config = ConfigParser.SafeConfigParser(client_defaults)
 
 
979
    client_config.read(os.path.join(server_settings["configdir"],
 
 
983
    tcp_server = IPv6_TCPServer((server_settings["address"],
 
 
984
                                 server_settings["port"]),
 
 
986
                                settings=server_settings,
 
 
988
    pidfilename = "/var/run/mandos.pid"
 
 
990
        pidfile = open(pidfilename, "w")
 
 
991
    except IOError, error:
 
 
992
        logger.error("Could not open file %r", pidfilename)
 
 
995
        uid = pwd.getpwnam("_mandos").pw_uid
 
 
996
        gid = pwd.getpwnam("_mandos").pw_gid
 
1801
997
    except KeyError:
 
1803
 
            uid = pwd.getpwnam(u"mandos").pw_uid
 
1804
 
            gid = pwd.getpwnam(u"mandos").pw_gid
 
 
999
            uid = pwd.getpwnam("mandos").pw_uid
 
 
1000
            gid = pwd.getpwnam("mandos").pw_gid
 
1805
1001
        except KeyError:
 
1807
 
                uid = pwd.getpwnam(u"nobody").pw_uid
 
1808
 
                gid = pwd.getpwnam(u"nobody").pw_gid
 
 
1003
                uid = pwd.getpwnam("nobody").pw_uid
 
 
1004
                gid = pwd.getpwnam("nogroup").pw_gid
 
1809
1005
            except KeyError:
 
1815
1011
    except OSError, error:
 
1816
1012
        if error[0] != errno.EPERM:
 
1819
 
    if not debug and not debuglevel:
 
1820
 
        syslogger.setLevel(logging.WARNING)
 
1821
 
        console.setLevel(logging.WARNING)
 
1823
 
        level = getattr(logging, debuglevel.upper())
 
1824
 
        syslogger.setLevel(level)
 
1825
 
        console.setLevel(level)
 
 
1016
    service = AvahiService(name = server_settings["servicename"],
 
 
1017
                           servicetype = "_mandos._tcp", )
 
 
1018
    if server_settings["interface"]:
 
 
1019
        service.interface = (if_nametoindex
 
 
1020
                             (server_settings["interface"]))
 
 
1025
    # From the Avahi example code
 
 
1026
    DBusGMainLoop(set_as_default=True )
 
 
1027
    main_loop = gobject.MainLoop()
 
 
1028
    bus = dbus.SystemBus()
 
 
1029
    server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
 
 
1030
                                           avahi.DBUS_PATH_SERVER),
 
 
1031
                            avahi.DBUS_INTERFACE_SERVER)
 
 
1032
    # End of Avahi example code
 
 
1034
        bus_name = dbus.service.BusName(u"org.mandos-system.Mandos",
 
 
1037
    clients.update(Set(Client(name = section,
 
 
1039
                              = dict(client_config.items(section)),
 
 
1040
                              use_dbus = use_dbus)
 
 
1041
                       for section in client_config.sections()))
 
 
1043
        logger.warning(u"No clients defined")
 
1828
 
        # Enable all possible GnuTLS debugging
 
1830
 
        # "Use a log level over 10 to enable all debugging options."
 
1832
 
        gnutls.library.functions.gnutls_global_set_log_level(11)
 
1834
 
        @gnutls.library.types.gnutls_log_func
 
1835
 
        def debug_gnutls(level, string):
 
1836
 
            logger.debug(u"GnuTLS: %s", string[:-1])
 
1838
 
        (gnutls.library.functions
 
1839
 
         .gnutls_global_set_log_function(debug_gnutls))
 
1841
1046
        # Redirect stdin so all checkers get /dev/null
 
1842
1047
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
 
1843
1048
        os.dup2(null, sys.stdin.fileno())
 
 
1847
1052
        # No console logging
 
1848
1053
        logger.removeHandler(console)
 
1850
 
    # Need to fork before connecting to D-Bus
 
1852
1054
        # Close all input and output, do double fork, etc.
 
1856
 
    # From the Avahi example code
 
1857
 
    DBusGMainLoop(set_as_default=True )
 
1858
 
    main_loop = gobject.MainLoop()
 
1859
 
    bus = dbus.SystemBus()
 
1860
 
    # End of Avahi example code
 
1863
 
            bus_name = dbus.service.BusName(u"se.bsnet.fukt.Mandos",
 
1864
 
                                            bus, do_not_queue=True)
 
1865
 
        except dbus.exceptions.NameExistsException, e:
 
1866
 
            logger.error(unicode(e) + u", disabling D-Bus")
 
1868
 
            server_settings[u"use_dbus"] = False
 
1869
 
            tcp_server.use_dbus = False
 
1870
 
    protocol = avahi.PROTO_INET6 if use_ipv6 else avahi.PROTO_INET
 
1871
 
    service = AvahiService(name = server_settings[u"servicename"],
 
1872
 
                           servicetype = u"_mandos._tcp",
 
1873
 
                           protocol = protocol, bus = bus)
 
1874
 
    if server_settings["interface"]:
 
1875
 
        service.interface = (if_nametoindex
 
1876
 
                             (str(server_settings[u"interface"])))
 
1878
 
    global multiprocessing_manager
 
1879
 
    multiprocessing_manager = multiprocessing.Manager()
 
1881
 
    client_class = Client
 
1883
 
        client_class = functools.partial(ClientDBus, bus = bus)
 
1884
 
    def client_config_items(config, section):
 
1885
 
        special_settings = {
 
1886
 
            "approved_by_default":
 
1887
 
                lambda: config.getboolean(section,
 
1888
 
                                          "approved_by_default"),
 
1890
 
        for name, value in config.items(section):
 
1892
 
                yield (name, special_settings[name]())
 
1896
 
    tcp_server.clients.update(set(
 
1897
 
            client_class(name = section,
 
1898
 
                         config= dict(client_config_items(
 
1899
 
                        client_config, section)))
 
1900
 
            for section in client_config.sections()))
 
1901
 
    if not tcp_server.clients:
 
1902
 
        logger.warning(u"No clients defined")
 
 
1059
        pidfile.write(str(pid) + "\n")
 
 
1063
        logger.error(u"Could not write to file %r with PID %d",
 
 
1066
        # "pidfile" was never created
 
 
1071
        "Cleanup function; run on exit"
 
 
1073
        # From the Avahi example code
 
 
1074
        if not group is None:
 
 
1077
        # End of Avahi example code
 
 
1080
            client = clients.pop()
 
 
1081
            client.disable_hook = None
 
 
1084
    atexit.register(cleanup)
 
1908
 
                pidfile.write(str(pid) + "\n")
 
1911
 
            logger.error(u"Could not write to file %r with PID %d",
 
1914
 
            # "pidfile" was never created
 
1918
1087
        signal.signal(signal.SIGINT, signal.SIG_IGN)
 
1920
1088
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
 
1921
1089
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
 
1924
 
        class MandosDBusService(dbus.service.Object):
 
 
1092
        class MandosServer(dbus.service.Object):
 
1925
1093
            """A D-Bus proxy object"""
 
1926
1094
            def __init__(self):
 
1927
 
                dbus.service.Object.__init__(self, bus, u"/")
 
1928
 
            _interface = u"se.bsnet.fukt.Mandos"
 
1930
 
            @dbus.service.signal(_interface, signature=u"o")
 
1931
 
            def ClientAdded(self, objpath):
 
1935
 
            @dbus.service.signal(_interface, signature=u"ss")
 
1936
 
            def ClientNotFound(self, fingerprint, address):
 
1940
 
            @dbus.service.signal(_interface, signature=u"os")
 
1941
 
            def ClientRemoved(self, objpath, name):
 
1945
 
            @dbus.service.method(_interface, out_signature=u"ao")
 
 
1095
                dbus.service.Object.__init__(self, bus,
 
 
1097
            _interface = u"org.mandos_system.Mandos"
 
 
1099
            @dbus.service.signal(_interface, signature="oa{sv}")
 
 
1100
            def ClientAdded(self, objpath, properties):
 
 
1104
            @dbus.service.signal(_interface, signature="o")
 
 
1105
            def ClientRemoved(self, objpath):
 
 
1109
            @dbus.service.method(_interface, out_signature="ao")
 
1946
1110
            def GetAllClients(self):
 
1948
 
                return dbus.Array(c.dbus_object_path
 
1949
 
                                  for c in tcp_server.clients)
 
1951
 
            @dbus.service.method(_interface,
 
1952
 
                                 out_signature=u"a{oa{sv}}")
 
 
1111
                return dbus.Array(c.dbus_object_path for c in clients)
 
 
1113
            @dbus.service.method(_interface, out_signature="a{oa{sv}}")
 
1953
1114
            def GetAllClientsWithProperties(self):
 
1955
1115
                return dbus.Dictionary(
 
1956
 
                    ((c.dbus_object_path, c.GetAll(u""))
 
1957
 
                     for c in tcp_server.clients),
 
1958
 
                    signature=u"oa{sv}")
 
1960
 
            @dbus.service.method(_interface, in_signature=u"o")
 
 
1116
                    ((c.dbus_object_path, c.GetAllProperties())
 
 
1120
            @dbus.service.method(_interface, in_signature="o")
 
1961
1121
            def RemoveClient(self, object_path):
 
1963
 
                for c in tcp_server.clients:
 
1964
1123
                    if c.dbus_object_path == object_path:
 
1965
 
                        tcp_server.clients.remove(c)
 
1966
 
                        c.remove_from_connection()
 
1967
1125
                        # Don't signal anything except ClientRemoved
 
1968
 
                        c.disable(quiet=True)
 
1969
1128
                        # Emit D-Bus signal
 
1970
 
                        self.ClientRemoved(object_path, c.name)
 
 
1129
                        self.ClientRemoved(object_path)
 
1972
 
                raise KeyError(object_path)
 
 
1132
            @dbus.service.method(_interface)
 
1976
 
        mandos_dbus_service = MandosDBusService()
 
1979
 
        "Cleanup function; run on exit"
 
1982
 
        while tcp_server.clients:
 
1983
 
            client = tcp_server.clients.pop()
 
1985
 
                client.remove_from_connection()
 
1986
 
            client.disable_hook = None
 
1987
 
            # Don't signal anything except ClientRemoved
 
1988
 
            client.disable(quiet=True)
 
1991
 
                mandos_dbus_service.ClientRemoved(client.dbus_object_path,
 
1994
 
    atexit.register(cleanup)
 
1996
 
    for client in tcp_server.clients:
 
 
1138
        mandos_server = MandosServer()
 
 
1140
    for client in clients:
 
1998
1142
            # Emit D-Bus signal
 
1999
 
            mandos_dbus_service.ClientAdded(client.dbus_object_path)
 
 
1143
            mandos_server.ClientAdded(client.dbus_object_path,
 
 
1144
                                      client.GetAllProperties())
 
2000
1145
        client.enable()
 
2002
1147
    tcp_server.enable()