19
19
# You should have received a copy of the GNU General Public License
 
20
20
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
22
 
# Contact the authors at <mandos@fukt.bsnet.se>.
 
 
22
# Contact the authors at <mandos@recompile.se>.
 
25
 
from __future__ import division, absolute_import, print_function, unicode_literals
 
 
25
from __future__ import (division, absolute_import, print_function,
 
 
48
49
logging.getLogger('dbus.proxies').setLevel(logging.CRITICAL)
 
50
51
# Some useful constants
 
51
 
domain = 'se.bsnet.fukt'
 
 
52
domain = 'se.recompile'
 
52
53
server_interface = domain + '.Mandos'
 
53
54
client_interface = domain + '.Mandos.Client'
 
56
57
# Always run in monochrome mode
 
57
58
urwid.curses_display.curses.has_colors = lambda : False
 
 
86
87
        self.proxy = proxy_object # Mandos Client proxy object
 
88
89
        self.properties = dict()
 
89
 
        self.proxy.connect_to_signal("PropertyChanged",
 
90
 
                                     self.property_changed,
 
 
90
        self.property_changed_match = (
 
 
91
            self.proxy.connect_to_signal("PropertyChanged",
 
 
92
                                         self.property_changed,
 
94
96
        self.properties.update(
 
95
97
            self.proxy.GetAll(client_interface,
 
96
98
                              dbus_interface = dbus.PROPERTIES_IFACE))
 
98
 
        #XXX This break good super behaviour!
 
 
100
        #XXX This breaks good super behaviour
 
99
101
#        super(MandosClientPropertyCache, self).__init__(
 
100
102
#            *args, **kwargs)
 
 
106
108
        # Update properties dict with new value
 
107
109
        self.properties[property] = value
 
 
111
    def delete(self, *args, **kwargs):
 
 
112
        self.property_changed_match.remove()
 
 
113
        super(MandosClientPropertyCache, self).__init__(
 
110
117
class MandosClientWidget(urwid.FlowWidget, MandosClientPropertyCache):
 
 
154
161
        if self.need_approval:
 
155
162
            self.using_timer(True)
 
157
 
        self.proxy.connect_to_signal("CheckerCompleted",
 
158
 
                                     self.checker_completed,
 
161
 
        self.proxy.connect_to_signal("CheckerStarted",
 
162
 
                                     self.checker_started,
 
165
 
        self.proxy.connect_to_signal("GotSecret",
 
169
 
        self.proxy.connect_to_signal("NeedApproval",
 
173
 
        self.proxy.connect_to_signal("Rejected",
 
 
164
        self.match_objects = (
 
 
165
            self.proxy.connect_to_signal("CheckerCompleted",
 
 
166
                                         self.checker_completed,
 
 
169
            self.proxy.connect_to_signal("CheckerStarted",
 
 
170
                                         self.checker_started,
 
 
173
            self.proxy.connect_to_signal("GotSecret",
 
 
177
            self.proxy.connect_to_signal("NeedApproval",
 
 
181
            self.proxy.connect_to_signal("Rejected",
 
 
185
        #self.logger('Created client %s' % (self.properties["Name"]))
 
178
187
    def property_changed(self, property=None, value=None):
 
179
188
        super(self, MandosClientWidget).property_changed(property,
 
 
305
315
                message = "Denial in %s. (a)pprove?"
 
306
316
            message = message % unicode(timer).rsplit(".", 1)[0]
 
307
317
        elif self.last_checker_failed:
 
308
 
            timeout = datetime.timedelta(milliseconds
 
311
 
            last_ok = isoformat_to_datetime(
 
312
 
                max((self.properties["LastCheckedOK"]
 
313
 
                     or self.properties["Created"]),
 
314
 
                    self.properties["LastEnabled"]))
 
315
 
            timer = timeout - (datetime.datetime.utcnow() - last_ok)
 
 
318
            # When checker has failed, print a timer until client expires
 
 
319
            expires = self.properties["Expires"]
 
 
321
                timer = datetime.timedelta(0)
 
 
323
                expires = datetime.datetime.strptime(expires,
 
 
324
                                                     '%Y-%m-%dT%H:%M:%S.%f')
 
 
325
                timer = expires - datetime.datetime.utcnow()
 
316
326
            message = ('A checker has failed! Time until client'
 
317
327
                       ' gets disabled: %s'
 
318
328
                           % unicode(timer).rsplit(".", 1)[0])
 
 
337
347
            self.update_hook()
 
339
349
    def update_timer(self):
 
 
350
        """called by gobject. Will indefinitely loop until
 
 
351
        gobject.source_remove() on tag is called"""
 
342
353
        return True             # Keep calling this
 
 
355
    def delete(self, *args, **kwargs):
 
345
356
        if self._update_timer_callback_tag is not None:
 
346
357
            gobject.source_remove(self._update_timer_callback_tag)
 
347
358
            self._update_timer_callback_tag = None
 
 
359
        for match in self.match_objects:
 
 
361
        self.match_objects = ()
 
348
362
        if self.delete_hook is not None:
 
349
363
            self.delete_hook(self)
 
 
364
        return super(MandosClientWidget, self).delete(*args, **kwargs)
 
351
366
    def render(self, maxcolrow, focus=False):
 
352
367
        """Render differently if we have focus.
 
 
359
374
        This overrides the method from urwid.FlowWidget"""
 
361
 
            self.proxy.Enable(dbus_interface = client_interface)
 
 
376
            self.proxy.Enable(dbus_interface = client_interface,
 
363
 
            self.proxy.Disable(dbus_interface = client_interface)
 
 
379
            self.proxy.Disable(dbus_interface = client_interface,
 
365
382
            self.proxy.Approve(dbus.Boolean(True, variant_level=1),
 
366
 
                               dbus_interface = client_interface)
 
 
383
                               dbus_interface = client_interface,
 
368
386
            self.proxy.Approve(dbus.Boolean(False, variant_level=1),
 
369
 
                                  dbus_interface = client_interface)
 
 
387
                                  dbus_interface = client_interface,
 
370
389
        elif key == "R" or key == "_" or key == "ctrl k":
 
371
390
            self.server_proxy_object.RemoveClient(self.proxy
 
374
 
            self.proxy.StartChecker(dbus_interface = client_interface)
 
 
394
            self.proxy.StartChecker(dbus_interface = client_interface,
 
376
 
            self.proxy.StopChecker(dbus_interface = client_interface)
 
 
397
            self.proxy.StopChecker(dbus_interface = client_interface,
 
378
 
            self.proxy.CheckedOK(dbus_interface = client_interface)
 
 
400
            self.proxy.CheckedOK(dbus_interface = client_interface,
 
380
403
#         elif key == "p" or key == "=":
 
381
404
#             self.proxy.pause()
 
 
572
595
        #self.log_message("Wrap mode: " + self.log_wrap)
 
574
597
    def find_and_remove_client(self, path, name):
 
575
 
        """Find an client from its object path and remove it.
 
 
598
        """Find a client by its object path and remove it.
 
577
600
        This is connected to the ClientRemoved signal from the
 
578
601
        Mandos server object."""
 
 
580
603
            client = self.clients_dict[path]
 
 
606
            self.log_message("Unknown client %r (%r) removed", name,
 
584
 
        self.remove_client(client, path)
 
586
611
    def add_new_client(self, path):
 
587
612
        client_proxy_object = self.bus.get_object(self.busname, path)