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 and
 
 
183
    fingerprint: string (40 or 32 hexadecimal digits); used to
 
 
184
                 uniquely identify the client
 
 
185
    secret:     bytestring; sent verbatim (over TLS) to client
 
 
186
    host:       string; available for use by the checker command
 
 
187
    created:    datetime.datetime(); (UTC) object creation
 
 
188
    last_enabled: datetime.datetime(); (UTC)
 
 
190
    last_checked_ok: datetime.datetime(); (UTC) or None
 
 
191
    timeout:    datetime.timedelta(); How long from last_checked_ok
 
 
192
                                      until this client is invalid
 
 
193
    interval:   datetime.timedelta(); How often to start a new checker
 
 
194
    disable_hook:  If set, called by disable() as disable_hook(self)
 
246
195
    checker:    subprocess.Popen(); a running checker process used
 
247
196
                                    to see if the client lives.
 
248
197
                                    '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
 
 
198
    checker_initiator_tag: a gobject event source tag, or None
 
 
199
    disable_initiator_tag:    - '' -
 
 
200
    checker_callback_tag:  - '' -
 
 
201
    checker_command: string; External command which is run to check if
 
 
202
                     client lives.  %() expansions are done at
 
252
203
                     runtime with vars(self) as dict, so that for
 
253
204
                     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.
 
 
205
    use_dbus: bool(); Whether to provide D-Bus interface and signals
 
 
206
    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
208
    def timeout_milliseconds(self):
 
288
209
        "Return the 'timeout' attribute in milliseconds"
 
289
 
        return self._timedelta_to_milliseconds(self.timeout)
 
 
210
        return ((self.timeout.days * 24 * 60 * 60 * 1000)
 
 
211
                + (self.timeout.seconds * 1000)
 
 
212
                + (self.timeout.microseconds // 1000))
 
291
214
    def interval_milliseconds(self):
 
292
215
        "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)
 
 
216
        return ((self.interval.days * 24 * 60 * 60 * 1000)
 
 
217
                + (self.interval.seconds * 1000)
 
 
218
                + (self.interval.microseconds // 1000))
 
298
 
    def __init__(self, name = None, disable_hook=None, config=None):
 
 
220
    def __init__(self, name = None, disable_hook=None, config=None,
 
299
222
        """Note: the 'checker' key in 'config' sets the
 
300
223
        'checker_command' attribute and *not* the 'checker'
 
 
507
422
        if self.checker_callback_tag:
 
508
423
            gobject.source_remove(self.checker_callback_tag)
 
509
424
            self.checker_callback_tag = None
 
510
 
        if getattr(self, u"checker", None) is None:
 
 
425
        if getattr(self, "checker", None) is None:
 
512
427
        logger.debug(u"Stopping checker for %(name)s", vars(self))
 
514
429
            os.kill(self.checker.pid, signal.SIGTERM)
 
516
431
            #if self.checker.poll() is None:
 
517
432
            #    os.kill(self.checker.pid, signal.SIGKILL)
 
518
433
        except OSError, error:
 
519
434
            if error.errno != errno.ESRCH: # No such process
 
521
436
        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
 
 
438
            self.PropertyChanged(dbus.String(u"checker_running"),
 
 
439
                                 dbus.Boolean(False, variant_level=1))
 
 
441
    def still_valid(self):
 
 
442
        """Has the timeout not yet passed for this client?"""
 
 
443
        if not getattr(self, "enabled", False):
 
 
445
        now = datetime.datetime.utcnow()
 
 
446
        if self.last_checked_ok is None:
 
 
447
            return now < (self.created + self.timeout)
 
 
449
            return now < (self.last_checked_ok + self.timeout)
 
 
451
    ## D-Bus methods & signals
 
871
452
    _interface = u"se.bsnet.fukt.Mandos.Client"
 
 
455
    CheckedOK = dbus.service.method(_interface)(checked_ok)
 
 
456
    CheckedOK.__name__ = "CheckedOK"
 
875
458
    # CheckerCompleted - signal
 
876
 
    @dbus.service.signal(_interface, signature=u"nxs")
 
 
459
    @dbus.service.signal(_interface, signature="nxs")
 
877
460
    def CheckerCompleted(self, exitcode, waitstatus, command):
 
881
464
    # CheckerStarted - signal
 
882
 
    @dbus.service.signal(_interface, signature=u"s")
 
 
465
    @dbus.service.signal(_interface, signature="s")
 
883
466
    def CheckerStarted(self, command):
 
 
470
    # GetAllProperties - method
 
 
471
    @dbus.service.method(_interface, out_signature="a{sv}")
 
 
472
    def GetAllProperties(self):
 
 
474
        return dbus.Dictionary({
 
 
476
                    dbus.String(self.name, variant_level=1),
 
 
477
                dbus.String("fingerprint"):
 
 
478
                    dbus.String(self.fingerprint, variant_level=1),
 
 
480
                    dbus.String(self.host, variant_level=1),
 
 
481
                dbus.String("created"):
 
 
482
                    _datetime_to_dbus(self.created, variant_level=1),
 
 
483
                dbus.String("last_enabled"):
 
 
484
                    (_datetime_to_dbus(self.last_enabled,
 
 
486
                     if self.last_enabled is not None
 
 
487
                     else dbus.Boolean(False, variant_level=1)),
 
 
488
                dbus.String("enabled"):
 
 
489
                    dbus.Boolean(self.enabled, variant_level=1),
 
 
490
                dbus.String("last_checked_ok"):
 
 
491
                    (_datetime_to_dbus(self.last_checked_ok,
 
 
493
                     if self.last_checked_ok is not None
 
 
494
                     else dbus.Boolean (False, variant_level=1)),
 
 
495
                dbus.String("timeout"):
 
 
496
                    dbus.UInt64(self.timeout_milliseconds(),
 
 
498
                dbus.String("interval"):
 
 
499
                    dbus.UInt64(self.interval_milliseconds(),
 
 
501
                dbus.String("checker"):
 
 
502
                    dbus.String(self.checker_command,
 
 
504
                dbus.String("checker_running"):
 
 
505
                    dbus.Boolean(self.checker is not None,
 
 
507
                dbus.String("object_path"):
 
 
508
                    dbus.ObjectPath(self.dbus_object_path,
 
 
512
    # IsStillValid - method
 
 
513
    IsStillValid = (dbus.service.method(_interface, out_signature="b")
 
 
515
    IsStillValid.__name__ = "IsStillValid"
 
887
517
    # PropertyChanged - signal
 
888
 
    @dbus.service.signal(_interface, signature=u"sv")
 
 
518
    @dbus.service.signal(_interface, signature="sv")
 
889
519
    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()
 
 
523
    # SetChecker - method
 
 
524
    @dbus.service.method(_interface, in_signature="s")
 
 
525
    def SetChecker(self, checker):
 
 
526
        "D-Bus setter method"
 
 
527
        self.checker_command = checker
 
 
529
        self.PropertyChanged(dbus.String(u"checker"),
 
 
530
                             dbus.String(self.checker_command,
 
 
534
    @dbus.service.method(_interface, in_signature="s")
 
 
535
    def SetHost(self, host):
 
 
536
        "D-Bus setter method"
 
 
539
        self.PropertyChanged(dbus.String(u"host"),
 
 
540
                             dbus.String(self.host, variant_level=1))
 
 
542
    # SetInterval - method
 
 
543
    @dbus.service.method(_interface, in_signature="t")
 
 
544
    def SetInterval(self, milliseconds):
 
 
545
        self.interval = datetime.timedelta(0, 0, 0, milliseconds)
 
 
547
        self.PropertyChanged(dbus.String(u"interval"),
 
 
548
                             (dbus.UInt64(self.interval_milliseconds(),
 
 
552
    @dbus.service.method(_interface, in_signature="ay",
 
 
554
    def SetSecret(self, secret):
 
 
555
        "D-Bus setter method"
 
 
556
        self.secret = str(secret)
 
 
558
    # SetTimeout - method
 
 
559
    @dbus.service.method(_interface, in_signature="t")
 
 
560
    def SetTimeout(self, milliseconds):
 
 
561
        self.timeout = datetime.timedelta(0, 0, 0, milliseconds)
 
 
563
        self.PropertyChanged(dbus.String(u"timeout"),
 
 
564
                             (dbus.UInt64(self.timeout_milliseconds(),
 
926
567
    # Enable - method
 
927
 
    @dbus.service.method(_interface)
 
 
568
    Enable = dbus.service.method(_interface)(enable)
 
 
569
    Enable.__name__ = "Enable"
 
932
571
    # StartChecker - method
 
933
572
    @dbus.service.method(_interface)
 
 
944
583
    # 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)
 
 
584
    StopChecker = dbus.service.method(_interface)(stop_checker)
 
 
585
    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.
 
 
590
def peer_certificate(session):
 
 
591
    "Return the peer's OpenPGP certificate as a bytestring"
 
 
592
    # If not an OpenPGP certificate...
 
 
593
    if (gnutls.library.functions
 
 
594
        .gnutls_certificate_type_get(session._c_object)
 
 
595
        != gnutls.library.constants.GNUTLS_CRT_OPENPGP):
 
 
596
        # ...do the normal thing
 
 
597
        return session.peer_certificate
 
 
598
    list_size = ctypes.c_uint(1)
 
 
599
    cert_list = (gnutls.library.functions
 
 
600
                 .gnutls_certificate_get_peers
 
 
601
                 (session._c_object, ctypes.byref(list_size)))
 
 
602
    if not bool(cert_list) and list_size.value != 0:
 
 
603
        raise gnutls.errors.GNUTLSError("error getting peer"
 
 
605
    if list_size.value == 0:
 
 
608
    return ctypes.string_at(cert.data, cert.size)
 
 
611
def fingerprint(openpgp):
 
 
612
    "Convert an OpenPGP bytestring to a hexdigit fingerprint string"
 
 
613
    # New GnuTLS "datum" with the OpenPGP public key
 
 
614
    datum = (gnutls.library.types
 
 
615
             .gnutls_datum_t(ctypes.cast(ctypes.c_char_p(openpgp),
 
 
618
                             ctypes.c_uint(len(openpgp))))
 
 
619
    # New empty GnuTLS certificate
 
 
620
    crt = gnutls.library.types.gnutls_openpgp_crt_t()
 
 
621
    (gnutls.library.functions
 
 
622
     .gnutls_openpgp_crt_init(ctypes.byref(crt)))
 
 
623
    # Import the OpenPGP public key into the certificate
 
 
624
    (gnutls.library.functions
 
 
625
     .gnutls_openpgp_crt_import(crt, ctypes.byref(datum),
 
 
626
                                gnutls.library.constants
 
 
627
                                .GNUTLS_OPENPGP_FMT_RAW))
 
 
628
    # Verify the self signature in the key
 
 
629
    crtverify = ctypes.c_uint()
 
 
630
    (gnutls.library.functions
 
 
631
     .gnutls_openpgp_crt_verify_self(crt, 0, ctypes.byref(crtverify)))
 
 
632
    if crtverify.value != 0:
 
 
633
        gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
 
 
634
        raise gnutls.errors.CertificateSecurityError("Verify failed")
 
 
635
    # New buffer for the fingerprint
 
 
636
    buf = ctypes.create_string_buffer(20)
 
 
637
    buf_len = ctypes.c_size_t()
 
 
638
    # Get the fingerprint from the certificate into the buffer
 
 
639
    (gnutls.library.functions
 
 
640
     .gnutls_openpgp_crt_get_fingerprint(crt, ctypes.byref(buf),
 
 
641
                                         ctypes.byref(buf_len)))
 
 
642
    # Deinit the certificate
 
 
643
    gnutls.library.functions.gnutls_openpgp_crt_deinit(crt)
 
 
644
    # Convert the buffer to a Python bytestring
 
 
645
    fpr = ctypes.string_at(buf, buf_len.value)
 
 
646
    # Convert the bytestring to hexadecimal notation
 
 
647
    hex_fpr = u''.join(u"%02X" % ord(char) for char in fpr)
 
 
651
class TCP_handler(SocketServer.BaseRequestHandler, object):
 
 
652
    """A TCP request handler class.
 
 
653
    Instantiated by IPv6_TCPServer for each request to handle it.
 
1168
654
    Note: This will run in its own forked process."""
 
1170
656
    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
 
 
657
        logger.info(u"TCP connection from: %s",
 
 
658
                    unicode(self.client_address))
 
 
659
        session = (gnutls.connection
 
 
660
                   .ClientSession(self.request,
 
 
664
        line = self.request.makefile().readline()
 
 
665
        logger.debug(u"Protocol version: %r", line)
 
 
667
            if int(line.strip().split()[0]) > 1:
 
 
669
        except (ValueError, IndexError, RuntimeError), error:
 
 
670
            logger.error(u"Unknown protocol version: %s", error)
 
 
673
        # Note: gnutls.connection.X509Credentials is really a generic
 
 
674
        # GnuTLS certificate credentials object so long as no X.509
 
 
675
        # keys are added to it.  Therefore, we can use it here despite
 
 
676
        # using OpenPGP certificates.
 
 
678
        #priority = ':'.join(("NONE", "+VERS-TLS1.1", "+AES-256-CBC",
 
 
679
        #                "+SHA1", "+COMP-NULL", "+CTYPE-OPENPGP",
 
 
681
        # Use a fallback default, since this MUST be set.
 
 
682
        priority = self.server.settings.get("priority", "NORMAL")
 
 
683
        (gnutls.library.functions
 
 
684
         .gnutls_priority_set_direct(session._c_object,
 
 
689
        except gnutls.errors.GNUTLSError, error:
 
 
690
            logger.warning(u"Handshake failed: %s", error)
 
 
691
            # Do not run session.bye() here: the session is not
 
 
692
            # established.  Just abandon the request.
 
 
694
        logger.debug(u"Handshake succeeded")
 
 
696
            fpr = fingerprint(peer_certificate(session))
 
 
697
        except (TypeError, gnutls.errors.GNUTLSError), error:
 
 
698
            logger.warning(u"Bad certificate: %s", error)
 
 
701
        logger.debug(u"Fingerprint: %s", fpr)
 
 
703
        for c in self.server.clients:
 
 
704
            if c.fingerprint == fpr:
 
 
708
            logger.warning(u"Client not found for fingerprint: %s",
 
 
712
        # Have to check if client.still_valid(), since it is possible
 
 
713
        # that the client timed out while establishing the GnuTLS
 
 
715
        if not client.still_valid():
 
 
716
            logger.warning(u"Client %(name)s is invalid",
 
 
720
        ## This won't work here, since we're in a fork.
 
 
721
        # client.checked_ok()
 
 
723
        while sent_size < len(client.secret):
 
 
724
            sent = session.send(client.secret[sent_size:])
 
 
725
            logger.debug(u"Sent: %d, remaining: %d",
 
 
726
                         sent, len(client.secret)
 
 
727
                         - (sent_size + sent))
 
 
732
class IPv6_TCPServer(SocketServer.ForkingMixIn,
 
 
733
                     SocketServer.TCPServer, object):
 
 
734
    """IPv6 TCP server.  Accepts 'None' as address and/or port.
 
 
736
        settings:       Server settings
 
 
737
        clients:        Set() of Client objects
 
1418
738
        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)
 
 
740
    address_family = socket.AF_INET6
 
 
741
    def __init__(self, *args, **kwargs):
 
 
742
        if "settings" in kwargs:
 
 
743
            self.settings = kwargs["settings"]
 
 
744
            del kwargs["settings"]
 
 
745
        if "clients" in kwargs:
 
 
746
            self.clients = kwargs["clients"]
 
 
747
            del kwargs["clients"]
 
 
749
        super(IPv6_TCPServer, self).__init__(*args, **kwargs)
 
1429
750
    def server_bind(self):
 
1430
751
        """This overrides the normal server_bind() function
 
1431
752
        to bind to an interface if one was specified, and also NOT to
 
1432
753
        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",
 
 
754
        if self.settings["interface"]:
 
 
755
            # 25 is from /usr/include/asm-i486/socket.h
 
 
756
            SO_BINDTODEVICE = getattr(socket, "SO_BINDTODEVICE", 25)
 
 
758
                self.socket.setsockopt(socket.SOL_SOCKET,
 
 
760
                                       self.settings["interface"])
 
 
761
            except socket.error, error:
 
 
762
                if error[0] == errno.EPERM:
 
 
763
                    logger.error(u"No permission to"
 
 
764
                                 u" bind to interface %s",
 
 
765
                                 self.settings["interface"])
 
1455
768
        # Only bind(2) the socket if we really need to.
 
1456
769
        if self.server_address[0] or self.server_address[1]:
 
1457
770
            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,
 
 
772
                self.server_address = (in6addr_any,
 
1463
773
                                       self.server_address[1])
 
1464
774
            elif not self.server_address[1]:
 
1465
775
                self.server_address = (self.server_address[0],
 
1467
 
#                 if self.interface:
 
 
777
#                 if self.settings["interface"]:
 
1468
778
#                     self.server_address = (self.server_address[0],
 
1471
781
#                                            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)
 
 
784
            return super(IPv6_TCPServer, self).server_bind()
 
1499
785
    def server_activate(self):
 
1500
786
        if self.enabled:
 
1501
 
            return socketserver.TCPServer.server_activate(self)
 
 
787
            return super(IPv6_TCPServer, self).server_activate()
 
1502
788
    def enable(self):
 
1503
789
        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
792
def string_to_delta(interval):
 
1583
793
    """Parse a string and return a datetime.timedelta
 
1585
 
    >>> string_to_delta(u'7d')
 
 
795
    >>> string_to_delta('7d')
 
1586
796
    datetime.timedelta(7)
 
1587
 
    >>> string_to_delta(u'60s')
 
 
797
    >>> string_to_delta('60s')
 
1588
798
    datetime.timedelta(0, 60)
 
1589
 
    >>> string_to_delta(u'60m')
 
 
799
    >>> string_to_delta('60m')
 
1590
800
    datetime.timedelta(0, 3600)
 
1591
 
    >>> string_to_delta(u'24h')
 
 
801
    >>> string_to_delta('24h')
 
1592
802
    datetime.timedelta(1)
 
1593
803
    >>> string_to_delta(u'1w')
 
1594
804
    datetime.timedelta(7)
 
1595
 
    >>> string_to_delta(u'5m 30s')
 
 
805
    >>> string_to_delta('5m 30s')
 
1596
806
    datetime.timedelta(0, 330)
 
1598
808
    timevalue = datetime.timedelta(0)
 
 
1709
932
    # 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",
 
 
933
    server_defaults = { "interface": "",
 
 
938
                        "SECURE256:!CTYPE-X.509:+CTYPE-OPENPGP",
 
 
939
                        "servicename": "Mandos",
 
1722
943
    # Parse config file for server-global settings
 
1723
 
    server_config = configparser.SafeConfigParser(server_defaults)
 
 
944
    server_config = ConfigParser.SafeConfigParser(server_defaults)
 
1724
945
    del server_defaults
 
1725
 
    server_config.read(os.path.join(options.configdir,
 
 
946
    server_config.read(os.path.join(options.configdir, "mandos.conf"))
 
1727
947
    # Convert the SafeConfigParser object to a dict
 
1728
948
    server_settings = server_config.defaults()
 
1729
949
    # 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",
 
 
950
    server_settings["debug"] = server_config.getboolean("DEFAULT",
 
 
952
    server_settings["use_dbus"] = server_config.getboolean("DEFAULT",
 
1733
954
    if server_settings["port"]:
 
1734
 
        server_settings["port"] = server_config.getint(u"DEFAULT",
 
 
955
        server_settings["port"] = server_config.getint("DEFAULT",
 
1736
957
    del server_config
 
1738
959
    # Override the settings from the config file with command line
 
1739
960
    # 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"):
 
 
961
    for option in ("interface", "address", "port", "debug",
 
 
962
                   "priority", "servicename", "configdir",
 
1743
964
        value = getattr(options, option)
 
1744
965
        if value is not None:
 
1745
966
            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
968
    # Now we have our good server settings in "server_settings"
 
1753
 
    ##################################################################
 
1755
970
    # 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"]
 
 
971
    debug = server_settings["debug"]
 
 
972
    use_dbus = server_settings["use_dbus"]
 
1761
 
    if server_settings[u"servicename"] != u"Mandos":
 
 
974
    def sigsegvhandler(signum, frame):
 
 
975
        raise RuntimeError('Segmentation fault')
 
 
978
        syslogger.setLevel(logging.WARNING)
 
 
979
        console.setLevel(logging.WARNING)
 
 
981
        signal.signal(signal.SIGSEGV, sigsegvhandler)
 
 
983
    if server_settings["servicename"] != "Mandos":
 
1762
984
        syslogger.setFormatter(logging.Formatter
 
1763
 
                               (u'Mandos (%s) [%%(process)d]:'
 
1764
 
                                u' %%(levelname)s: %%(message)s'
 
1765
 
                                % server_settings[u"servicename"]))
 
 
985
                               ('Mandos (%s): %%(levelname)s:'
 
 
987
                                % server_settings["servicename"]))
 
1767
989
    # 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",
 
 
990
    client_defaults = { "timeout": "1h",
 
 
992
                        "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
 
 
995
    client_config = ConfigParser.SafeConfigParser(client_defaults)
 
 
996
    client_config.read(os.path.join(server_settings["configdir"],
 
 
1000
    tcp_server = IPv6_TCPServer((server_settings["address"],
 
 
1001
                                 server_settings["port"]),
 
 
1003
                                settings=server_settings,
 
 
1005
    pidfilename = "/var/run/mandos.pid"
 
 
1007
        pidfile = open(pidfilename, "w")
 
 
1008
    except IOError, error:
 
 
1009
        logger.error("Could not open file %r", pidfilename)
 
 
1012
        uid = pwd.getpwnam("_mandos").pw_uid
 
 
1013
        gid = pwd.getpwnam("_mandos").pw_gid
 
1801
1014
    except KeyError:
 
1803
 
            uid = pwd.getpwnam(u"mandos").pw_uid
 
1804
 
            gid = pwd.getpwnam(u"mandos").pw_gid
 
 
1016
            uid = pwd.getpwnam("mandos").pw_uid
 
 
1017
            gid = pwd.getpwnam("mandos").pw_gid
 
1805
1018
        except KeyError:
 
1807
 
                uid = pwd.getpwnam(u"nobody").pw_uid
 
1808
 
                gid = pwd.getpwnam(u"nobody").pw_gid
 
 
1020
                uid = pwd.getpwnam("nobody").pw_uid
 
 
1021
                gid = pwd.getpwnam("nogroup").pw_gid
 
1809
1022
            except KeyError:
 
 
1816
1029
        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)
 
 
1032
    # Enable all possible GnuTLS debugging
 
1828
 
        # Enable all possible GnuTLS debugging
 
1830
1034
        # "Use a log level over 10 to enable all debugging options."
 
1831
1035
        # - GnuTLS manual
 
1832
1036
        gnutls.library.functions.gnutls_global_set_log_level(11)
 
1834
1038
        @gnutls.library.types.gnutls_log_func
 
1835
1039
        def debug_gnutls(level, string):
 
1836
 
            logger.debug(u"GnuTLS: %s", string[:-1])
 
 
1040
            logger.debug("GnuTLS: %s", string[:-1])
 
1838
1042
        (gnutls.library.functions
 
1839
1043
         .gnutls_global_set_log_function(debug_gnutls))
 
1841
 
        # Redirect stdin so all checkers get /dev/null
 
1842
 
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
 
1843
 
        os.dup2(null, sys.stdin.fileno())
 
1847
 
        # No console logging
 
1848
 
        logger.removeHandler(console)
 
1850
 
    # Need to fork before connecting to D-Bus
 
1852
 
        # Close all input and output, do double fork, etc.
 
 
1046
    service = AvahiService(name = server_settings["servicename"],
 
 
1047
                           servicetype = "_mandos._tcp", )
 
 
1048
    if server_settings["interface"]:
 
 
1049
        service.interface = (if_nametoindex
 
 
1050
                             (server_settings["interface"]))
 
1855
1052
    global main_loop
 
1856
1055
    # From the Avahi example code
 
1857
1056
    DBusGMainLoop(set_as_default=True )
 
1858
1057
    main_loop = gobject.MainLoop()
 
1859
1058
    bus = dbus.SystemBus()
 
 
1059
    server = dbus.Interface(bus.get_object(avahi.DBUS_NAME,
 
 
1060
                                           avahi.DBUS_PATH_SERVER),
 
 
1061
                            avahi.DBUS_INTERFACE_SERVER)
 
1860
1062
    # 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:
 
 
1064
        bus_name = dbus.service.BusName(u"se.bsnet.fukt.Mandos", bus)
 
 
1066
    clients.update(Set(Client(name = section,
 
 
1068
                              = dict(client_config.items(section)),
 
 
1069
                              use_dbus = use_dbus)
 
 
1070
                       for section in client_config.sections()))
 
1902
1072
        logger.warning(u"No clients defined")
 
 
1075
        # Redirect stdin so all checkers get /dev/null
 
 
1076
        null = os.open(os.path.devnull, os.O_NOCTTY | os.O_RDWR)
 
 
1077
        os.dup2(null, sys.stdin.fileno())
 
 
1081
        # No console logging
 
 
1082
        logger.removeHandler(console)
 
 
1083
        # Close all input and output, do double fork, etc.
 
 
1088
        pidfile.write(str(pid) + "\n")
 
 
1092
        logger.error(u"Could not write to file %r with PID %d",
 
 
1095
        # "pidfile" was never created
 
 
1100
        "Cleanup function; run on exit"
 
 
1102
        # From the Avahi example code
 
 
1103
        if not group is None:
 
 
1106
        # End of Avahi example code
 
 
1109
            client = clients.pop()
 
 
1110
            client.disable_hook = None
 
 
1113
    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
1116
        signal.signal(signal.SIGINT, signal.SIG_IGN)
 
1920
1117
    signal.signal(signal.SIGHUP, lambda signum, frame: sys.exit())
 
1921
1118
    signal.signal(signal.SIGTERM, lambda signum, frame: sys.exit())
 
1924
 
        class MandosDBusService(dbus.service.Object):
 
 
1121
        class MandosServer(dbus.service.Object):
 
1925
1122
            """A D-Bus proxy object"""
 
1926
1123
            def __init__(self):
 
1927
 
                dbus.service.Object.__init__(self, bus, u"/")
 
 
1124
                dbus.service.Object.__init__(self, bus, "/")
 
1928
1125
            _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")
 
 
1127
            @dbus.service.signal(_interface, signature="oa{sv}")
 
 
1128
            def ClientAdded(self, objpath, properties):
 
 
1132
            @dbus.service.signal(_interface, signature="os")
 
1941
1133
            def ClientRemoved(self, objpath, name):
 
1945
 
            @dbus.service.method(_interface, out_signature=u"ao")
 
 
1137
            @dbus.service.method(_interface, out_signature="ao")
 
1946
1138
            def GetAllClients(self):
 
1948
 
                return dbus.Array(c.dbus_object_path
 
1949
 
                                  for c in tcp_server.clients)
 
 
1140
                return dbus.Array(c.dbus_object_path for c in clients)
 
1951
 
            @dbus.service.method(_interface,
 
1952
 
                                 out_signature=u"a{oa{sv}}")
 
 
1142
            @dbus.service.method(_interface, out_signature="a{oa{sv}}")
 
1953
1143
            def GetAllClientsWithProperties(self):
 
1955
1145
                return dbus.Dictionary(
 
1956
 
                    ((c.dbus_object_path, c.GetAll(u""))
 
1957
 
                     for c in tcp_server.clients),
 
1958
 
                    signature=u"oa{sv}")
 
 
1146
                    ((c.dbus_object_path, c.GetAllProperties())
 
1960
 
            @dbus.service.method(_interface, in_signature=u"o")
 
 
1150
            @dbus.service.method(_interface, in_signature="o")
 
1961
1151
            def RemoveClient(self, object_path):
 
1963
 
                for c in tcp_server.clients:
 
1964
1154
                    if c.dbus_object_path == object_path:
 
1965
 
                        tcp_server.clients.remove(c)
 
1966
 
                        c.remove_from_connection()
 
1967
1156
                        # Don't signal anything except ClientRemoved
 
1968
 
                        c.disable(quiet=True)
 
1969
1159
                        # Emit D-Bus signal
 
1970
1160
                        self.ClientRemoved(object_path, c.name)
 
1972
 
                raise KeyError(object_path)
 
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:
 
 
1166
        mandos_server = MandosServer()
 
 
1168
    for client in clients:
 
1998
1170
            # Emit D-Bus signal
 
1999
 
            mandos_dbus_service.ClientAdded(client.dbus_object_path)
 
 
1171
            mandos_server.ClientAdded(client.dbus_object_path,
 
 
1172
                                      client.GetAllProperties())
 
2000
1173
        client.enable()
 
2002
1175
    tcp_server.enable()