chiark / gitweb /
mon/tripemon.in (xlate_bytes): Trim a character from the docstring.
[tripe] / mon / tripemon.in
CommitLineData
060ca767 1#! @PYTHON@
984b6d33
MW
2### -*- mode: python; coding: utf-8 -*-
3###
4### Graphical monitor for tripe server
5###
6### (c) 2007 Straylight/Edgeware
7###
8
9###----- Licensing notice ---------------------------------------------------
10###
11### This file is part of Trivial IP Encryption (TrIPE).
12###
11ad66c2
MW
13### TrIPE is free software: you can redistribute it and/or modify it under
14### the terms of the GNU General Public License as published by the Free
15### Software Foundation; either version 3 of the License, or (at your
16### option) any later version.
984b6d33 17###
11ad66c2
MW
18### TrIPE is distributed in the hope that it will be useful, but WITHOUT
19### ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
20### FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
21### for more details.
984b6d33
MW
22###
23### You should have received a copy of the GNU General Public License
11ad66c2 24### along with TrIPE. If not, see <https://www.gnu.org/licenses/>.
984b6d33
MW
25
26###--------------------------------------------------------------------------
27### Dependencies.
060ca767 28
29import socket as S
984b6d33
MW
30import tripe as T
31import mLib as M
32from sys import argv, exit, stdin, stdout, stderr, exc_info, excepthook
060ca767 33import os as OS
34from os import environ
984b6d33 35import math as MATH
060ca767 36import sets as SET
984b6d33
MW
37from optparse import OptionParser
38import time as TIME
c55f0b7c 39import re as RX
060ca767 40from cStringIO import StringIO
41
a69f4417
MW
42try:
43 if OS.getenv('TRIPEMON_FORCE_GI'): raise ImportError
44 import pygtk
45 pygtk.require('2.0')
46 import gtk as G
47 import gobject as GO
48 import gtk.gdk as GDK
49 GL = GO
dcd075c8 50 GDK.KEY_Escape = G.keysyms.Escape
a69f4417
MW
51 def raise_window(w): w.window.raise_()
52 combo_box_text = G.combo_box_new_text
53 def set_entry_bg(e, c): e.modify_base(G.STATE_NORMAL, c)
54except ImportError:
55 from gi.repository import GObject as GO, GLib as GL, Gtk as G, Gdk as GDK
56 G.WINDOW_TOPLEVEL = G.WindowType.TOPLEVEL
57 G.EXPAND = G.AttachOptions.EXPAND
58 G.SHRINK = G.AttachOptions.SHRINK
59 G.FILL = G.AttachOptions.FILL
60 G.SORT_ASCENDING = G.SortType.ASCENDING
61 G.POLICY_AUTOMATIC = G.PolicyType.AUTOMATIC
62 G.SHADOW_IN = G.ShadowType.IN
63 G.SELECTION_NONE = G.SelectionMode.NONE
64 G.DIALOG_MODAL = G.DialogFlags.MODAL
65 G.RESPONSE_CANCEL = G.ResponseType.CANCEL
66 G.RESPONSE_NONE = G.ResponseType.NONE
67 def raise_window(w): getattr(w.get_window(), 'raise')()
68 combo_box_text = G.ComboBoxText
69 def set_entry_bg(e, c): e.modify_bg(G.StateType.NORMAL, c)
060ca767 70
984b6d33
MW
71if OS.getenv('TRIPE_DEBUG_MONITOR') is not None:
72 T._debug = 1
060ca767 73
984b6d33
MW
74###--------------------------------------------------------------------------
75### Doing things later.
060ca767 76
984b6d33
MW
77def uncaught():
78 """Report an uncaught exception."""
79 excepthook(*exc_info())
060ca767 80
984b6d33
MW
81def xwrap(func):
82 """
83 Return a function which behaves like FUNC, but reports exceptions via
84 uncaught.
85 """
86 def _(*args, **kw):
87 try:
88 return func(*args, **kw)
89 except SystemExit:
90 raise
91 except:
92 uncaught()
93 raise
94 return _
060ca767 95
984b6d33
MW
96def invoker(func, *args, **kw):
97 """
98 Return a function which throws away its arguments and calls
99 FUNC(*ARGS, **KW).
060ca767 100
984b6d33
MW
101 If for loops worked by binding rather than assignment then we wouldn't need
102 this kludge.
103 """
104 return lambda *hunoz, **hukairz: xwrap(func)(*args, **kw)
060ca767 105
984b6d33
MW
106def cr(func, *args, **kw):
107 """Return a function which invokes FUNC(*ARGS, **KW) in a coroutine."""
22b47552
MW
108 name = T.funargstr(func, args, kw)
109 return lambda *hunoz, **hukairz: \
110 T.Coroutine(xwrap(func), name = name).switch(*args, **kw)
060ca767 111
984b6d33
MW
112def incr(func):
113 """Decorator: runs its function in a coroutine of its own."""
22b47552
MW
114 return lambda *args, **kw: \
115 (T.Coroutine(func, name = T.funargstr(func, args, kw))
116 .switch(*args, **kw))
060ca767 117
984b6d33
MW
118###--------------------------------------------------------------------------
119### Random bits of infrastructure.
060ca767 120
984b6d33
MW
121## Program name, shorn of extraneous stuff.
122M.ego(argv[0])
123moan = M.moan
124die = M.die
060ca767 125
984b6d33
MW
126class HookList (object):
127 """
128 Notification hook list.
060ca767 129
984b6d33
MW
130 Other objects can add functions onto the hook list. When the hook list is
131 run, the functions are called in the order in which they were registered.
132 """
060ca767 133
060ca767 134 def __init__(me):
984b6d33 135 """Basic initialization: create the hook list."""
060ca767 136 me.list = []
984b6d33 137
060ca767 138 def add(me, func, obj):
984b6d33 139 """Add FUNC to the list of hook functions."""
060ca767 140 me.list.append((obj, func))
984b6d33 141
060ca767 142 def prune(me, obj):
984b6d33 143 """Remove hook functions registered with the given OBJ."""
060ca767 144 new = []
145 for o, f in me.list:
146 if o is not obj:
147 new.append((o, f))
148 me.list = new
984b6d33 149
060ca767 150 def run(me, *args, **kw):
984b6d33 151 """Invoke the hook functions with arguments *ARGS and **KW."""
060ca767 152 for o, hook in me.list:
153 rc = hook(*args, **kw)
154 if rc is not None: return rc
155 return None
156
157class HookClient (object):
984b6d33
MW
158 """
159 Mixin for classes which are clients of hooks.
160
161 It keeps track of the hooks it's a client of, and has the ability to
162 extricate itself from all of them. This is useful because weak objects
163 don't seem to work well.
164 """
060ca767 165 def __init__(me):
984b6d33 166 """Basic initialization."""
060ca767 167 me.hooks = SET.Set()
984b6d33 168
060ca767 169 def hook(me, hk, func):
984b6d33 170 """Add FUNC to the hook list HK."""
060ca767 171 hk.add(func, me)
172 me.hooks.add(hk)
984b6d33 173
060ca767 174 def unhook(me, hk):
984b6d33 175 """Remove myself from the hook list HK."""
060ca767 176 hk.prune(me)
177 me.hooks.discard(hk)
984b6d33 178
060ca767 179 def unhookall(me):
984b6d33 180 """Remove myself from all hook lists."""
060ca767 181 for hk in me.hooks:
182 hk.prune(me)
183 me.hooks.clear()
060ca767 184
984b6d33
MW
185class struct (object):
186 """A very simple dumb data container object."""
187 def __init__(me, **kw):
188 me.__dict__.update(kw)
060ca767 189
984b6d33
MW
190## Matches ISO date format yyyy-mm-ddThh:mm:ss.
191rx_time = RX.compile(r'^(\d+)-(\d+)-(\d+)T(\d+):(\d+):(\d+)$')
060ca767 192
984b6d33
MW
193###--------------------------------------------------------------------------
194### Connections.
060ca767 195
690a6ec1
MW
196class GIOWatcher (object):
197 """
198 Monitor I/O events using glib.
199 """
a69f4417 200 def __init__(me, conn, mc = GL.main_context_default()):
690a6ec1
MW
201 me._conn = conn
202 me._watch = None
203 me._mc = mc
204 def connected(me, sock):
a69f4417 205 me._watch = GL.io_add_watch(sock, GL.IO_IN,
690a6ec1
MW
206 lambda *hunoz: me._conn.receive())
207 def disconnected(me):
a69f4417 208 GL.source_remove(me._watch)
690a6ec1
MW
209 me._watch = None
210 def iterate(me):
211 me._mc.iteration(True)
212
984b6d33
MW
213class Connection (T.TripeCommandDispatcher):
214 """
215 The main connection to the server.
060ca767 216
984b6d33
MW
217 The improvement over the TripeCommandDispatcher is that the Connection
218 provides hooklists for NOTE, WARN and TRACE messages, and for connect and
219 disconnect events.
060ca767 220
984b6d33 221 This class knows about the Glib I/O dispatcher system, and plugs into it.
060ca767 222
984b6d33 223 Hooks:
060ca767 224
984b6d33
MW
225 * connecthook(): a connection to the server has been established
226 * disconnecthook(): the connection has been dropped
227 * notehook(TOKEN, ...): server issued a notification
228 * warnhook(TOKEN, ...): server issued a warning
229 * tracehook(TOKEN, ...): server issued a trace message
230 """
060ca767 231
984b6d33
MW
232 def __init__(me, socket):
233 """Create a new Connection."""
234 T.TripeCommandDispatcher.__init__(me, socket)
060ca767 235 me.connecthook = HookList()
236 me.disconnecthook = HookList()
060ca767 237 me.notehook = HookList()
984b6d33
MW
238 me.warnhook = HookList()
239 me.tracehook = HookList()
240 me.handler['NOTE'] = lambda _, *rest: me.notehook.run(*rest)
241 me.handler['WARN'] = lambda _, *rest: me.warnhook.run(*rest)
242 me.handler['TRACE'] = lambda _, *rest: me.tracehook.run(*rest)
690a6ec1 243 me.iowatch = GIOWatcher(me)
984b6d33 244
060ca767 245 def connected(me):
984b6d33
MW
246 """Handles reconnection to the server, and signals the hook."""
247 T.TripeCommandDispatcher.connected(me)
984b6d33
MW
248 me.connecthook.run()
249
250 def disconnected(me, reason):
251 """Handles disconnection from the server, and signals the hook."""
984b6d33
MW
252 me.disconnecthook.run(reason)
253 T.TripeCommandDispatcher.disconnected(me, reason)
254
255###--------------------------------------------------------------------------
256### Watching the peers go by.
257
258class MonitorObject (object):
259 """
260 An object with hooks it uses to notify others of changes in its state.
261 These are the objects tracked by the MonitorList class.
262
263 The object has a name, an `aliveness' state indicated by the `alivep' flag,
264 and hooks.
265
266 Hooks:
267
268 * changehook(): the object has changed its state
269 * deadhook(): the object has been destroyed
270
271 Subclass responsibilities:
272
273 * update(INFO): update internal state based on the provided INFO, and run
274 the changehook.
275 """
276
277 def __init__(me, name):
278 """Initialize the object with the given NAME."""
060ca767 279 me.name = name
984b6d33
MW
280 me.deadhook = HookList()
281 me.changehook = HookList()
282 me.alivep = True
283
284 def dead(me):
285 """Mark the object as dead; invoke the deadhook."""
286 me.alivep = False
287 me.deadhook.run()
288
289class Peer (MonitorObject):
290 """
291 An object representing a connected peer.
292
293 As well as the standard hooks, a peer has a pinghook, which isn't used
294 directly by this class.
295
296 Hooks:
297
298 * pinghook(): invoked by the Pinger (q.v.) when ping statistics change
299
300 Attributes provided are:
301
302 * addr = a vaguely human-readable representation of the peer's address
303 * ifname = the peer's interface name
304 * tunnel = the kind of tunnel the peer is using
305 * keepalive = the peer's keepalive interval in seconds
306 * ping['EPING'] and ping['PING'] = pingstate statistics (maintained by
307 the Pinger)
308 """
309
310 def __init__(me, name):
311 """Initialize the object with the given name."""
312 MonitorObject.__init__(me, name)
313 me.pinghook = HookList()
1ff5474e 314 me.__dict__.update(conn.algs(name))
984b6d33
MW
315 me.update()
316
317 def update(me, hunoz = None):
318 """Update the peer, fetching information about it from the server."""
d1034a00 319 me._setaddr(conn.addr(me.name))
6411163d
MW
320 me.ifname = conn.ifname(me.name)
321 me.__dict__.update(conn.peerinfo(me.name))
322 me.changehook.run()
323
324 def _setaddr(me, addr):
325 """Set the peer's address."""
060ca767 326 if addr[0] == 'INET':
327 ipaddr, port = addr[1:]
328 try:
329 name = S.gethostbyaddr(ipaddr)[0]
330 me.addr = 'INET %s:%s [%s]' % (name, port, ipaddr)
331 except S.herror:
332 me.addr = 'INET %s:%s' % (ipaddr, port)
333 else:
334 me.addr = ' '.join(addr)
6411163d
MW
335
336 def setaddr(me, addr):
337 """Informs the object of a change to its address to ADDR."""
338 me._setaddr(addr)
984b6d33
MW
339 me.changehook.run()
340
341 def setifname(me, newname):
342 """Informs the object of a change to its interface name to NEWNAME."""
343 me.ifname = newname
344 me.changehook.run()
345
346class Service (MonitorObject):
347 """
348 Represents a service.
349
350 Additional attributes are:
351
352 * version = the service version
353 """
354 def __init__(me, name, version):
355 MonitorObject.__init__(me, name)
356 me.version = version
357
358 def update(me, version):
359 """Tell the Service that its version has changed to VERSION."""
360 me.version = version
361 me.changehook.run()
362
363class MonitorList (object):
364 """
365 Maintains a collection of MonitorObjects.
366
367 The MonitorList can be indexed by name to retrieve the individual objects;
368 iteration generates the individual objects. More complicated operations
369 can be done on the `table' dictionary directly.
370
371 Hooks addhook(OBJ) and delhook(OBJ) are invoked when objects are added or
372 deleted.
060ca767 373
984b6d33
MW
374 Subclass responsibilities:
375
376 * list(): return a list of (NAME, INFO) pairs.
377
378 * make(NAME, INFO): returns a new MonitorObject for the given NAME; INFO
379 is from the output of list().
380 """
381
382 def __init__(me):
383 """Initialize a new MonitorList."""
384 me.table = {}
385 me.addhook = HookList()
386 me.delhook = HookList()
387
388 def update(me):
389 """
390 Refresh the list of objects:
391
392 We add new object which have appeared, delete ones which have vanished,
393 and update any which persist.
394 """
395 new = {}
396 for name, stuff in me.list():
397 new[name] = True
398 me.add(name, stuff)
399 for name in me.table.copy():
400 if name not in new:
401 me.remove(name)
402
403 def add(me, name, stuff):
404 """
405 Add a new object created by make(NAME, STUFF) if it doesn't already
406 exist. If it does, update it.
407 """
408 if name not in me.table:
409 obj = me.make(name, stuff)
410 me.table[name] = obj
411 me.addhook.run(obj)
412 else:
413 me.table[name].update(stuff)
414
415 def remove(me, name):
416 """
417 Remove the object called NAME from the list.
418
419 The object becomes dead.
420 """
421 if name in me.table:
422 obj = me.table[name]
423 del me.table[name]
424 me.delhook.run(obj)
425 obj.dead()
426
427 def __getitem__(me, name):
428 """Retrieve the object called NAME."""
429 return me.table[name]
430
431 def __iter__(me):
432 """Iterate over the objects."""
433 return me.table.itervalues()
434
435class PeerList (MonitorList):
436 """The list of the known peers."""
437 def list(me):
438 return [(name, None) for name in conn.list()]
439 def make(me, name, stuff):
440 return Peer(name)
441
442class ServiceList (MonitorList):
443 """The list of the registered services."""
444 def list(me):
445 return conn.svclist()
446 def make(me, name, stuff):
447 return Service(name, stuff)
448
449class Monitor (HookClient):
450 """
451 The main monitor: keeps track of the changes happening to the server.
452
453 Exports the peers, services MonitorLists, and a (plain Python) list
454 autopeers of peers which the connect service knows how to start by name.
455
456 Hooks provided:
457
458 * autopeershook(): invoked when the auto-peers list changes.
459 """
460 def __init__(me):
461 """Initialize the Monitor."""
462 HookClient.__init__(me)
463 me.peers = PeerList()
464 me.services = ServiceList()
465 me.hook(conn.connecthook, me._connected)
466 me.hook(conn.notehook, me._notify)
467 me.autopeershook = HookList()
468 me.autopeers = None
469
470 def _connected(me):
471 """Handle a successful connection by starting the setup coroutine."""
472 me._setup()
473
474 @incr
475 def _setup(me):
476 """Coroutine function: initialize for a new connection."""
477 conn.watch('-A+wnt')
478 me.peers.update()
479 me.services.update()
480 me._updateautopeers()
481
482 def _updateautopeers(me):
483 """Update the auto-peers list from the connect service."""
484 if 'connect' in me.services.table:
485 me.autopeers = [' '.join(line)
c1f93f03 486 for line in conn.svcsubmit('connect', 'list-active')]
984b6d33
MW
487 me.autopeers.sort()
488 else:
489 me.autopeers = None
490 me.autopeershook.run()
491
492 def _notify(me, code, *rest):
493 """
494 Handle notifications from the server.
495
496 ADD, KILL and NEWIFNAME notifications get passed up to the PeerList;
497 SVCCLAIM and SVCRELEASE get passed up to the ServiceList. Finally,
498 peerdb-update notifications from the watch service cause us to refresh
499 the auto-peers list.
500 """
501 if code == 'ADD':
690a6ec1 502 T.aside(me.peers.add, rest[0], None)
984b6d33 503 elif code == 'KILL':
690a6ec1 504 T.aside(me.peers.remove, rest[0])
984b6d33
MW
505 elif code == 'NEWIFNAME':
506 try:
507 me.peers[rest[0]].setifname(rest[2])
508 except KeyError:
509 pass
6411163d
MW
510 elif code == 'NEWADDR':
511 try:
512 me.peers[rest[0]].setaddr(rest[1:])
513 except KeyError:
514 pass
984b6d33 515 elif code == 'SVCCLAIM':
690a6ec1 516 T.aside(me.services.add, rest[0], rest[1])
984b6d33 517 if rest[0] == 'connect':
690a6ec1 518 T.aside(me._updateautopeers)
984b6d33 519 elif code == 'SVCRELEASE':
690a6ec1 520 T.aside(me.services.remove, rest[0])
984b6d33 521 if rest[0] == 'connect':
690a6ec1 522 T.aside(me._updateautopeers)
984b6d33
MW
523 elif code == 'USER':
524 if not rest: return
525 if rest[0] == 'watch' and \
526 rest[1] == 'peerdb-update':
690a6ec1 527 T.aside(me._updateautopeers)
984b6d33
MW
528
529###--------------------------------------------------------------------------
530### Window management cruft.
060ca767 531
532class MyWindowMixin (G.Window, HookClient):
984b6d33
MW
533 """
534 Mixin for windows which call a closehook when they're destroyed. It's also
535 a hookclient, and will release its hooks when it's destroyed.
536
537 Hooks:
538
539 * closehook(): called when the window is closed.
540 """
541
060ca767 542 def mywininit(me):
984b6d33 543 """Initialization function. Note that it's not called __init__!"""
060ca767 544 me.closehook = HookList()
545 HookClient.__init__(me)
546 me.connect('destroy', invoker(me.close))
984b6d33 547
060ca767 548 def close(me):
984b6d33 549 """Close the window, invoking the closehook and releasing all hooks."""
060ca767 550 me.closehook.run()
551 me.destroy()
552 me.unhookall()
984b6d33 553
060ca767 554class MyWindow (MyWindowMixin):
984b6d33 555 """A version of MyWindowMixin suitable as a single parent class."""
060ca767 556 def __init__(me, kind = G.WINDOW_TOPLEVEL):
557 G.Window.__init__(me, kind)
558 me.mywininit()
984b6d33 559
dcd075c8
MW
560class TrivialWindowMixin (MyWindowMixin):
561 """A simple window which you can close with Escape."""
562 def mywininit(me):
563 super(TrivialWindowMixin, me).mywininit()
564 me.connect('key-press-event', me._keypress)
565 def _keypress(me, _, ev):
566 if ev.keyval == GDK.KEY_Escape: me.destroy()
567
568class TrivialWindow (MyWindow, TrivialWindowMixin):
569 pass
570
060ca767 571class MyDialog (G.Dialog, MyWindowMixin, HookClient):
572 """A dialogue box with a closehook and sensible button binding."""
984b6d33 573
060ca767 574 def __init__(me, title = None, flags = 0, buttons = []):
984b6d33
MW
575 """
576 The BUTTONS are a list of (STOCKID, THUNK) pairs: call the appropriate
577 THUNK when the button is pressed. The other arguments are just like
578 GTK's Dialog class.
579 """
060ca767 580 i = 0
581 br = []
582 me.rmap = []
583 for b, f in buttons:
584 br.append(b)
585 br.append(i)
586 me.rmap.append(f)
587 i += 1
588 G.Dialog.__init__(me, title, None, flags, tuple(br))
060ca767 589 me.mywininit()
590 me.set_default_response(i - 1)
591 me.connect('response', me.respond)
984b6d33 592
060ca767 593 def respond(me, hunoz, rid, *hukairz):
984b6d33 594 """Dispatch responses to the appropriate thunks."""
060ca767 595 if rid >= 0: me.rmap[rid]()
596
ca6eb20c 597def makeactiongroup(name, acts):
984b6d33
MW
598 """
599 Creates an ActionGroup called NAME.
600
601 ACTS is a list of tuples containing:
602
603 * ACT: an action name
604 * LABEL: the label string for the action
605 * ACCEL: accelerator string, or None
606 * FUNC: thunk to call when the action is invoked
607 """
ca6eb20c 608 actgroup = G.ActionGroup(name)
609 for act, label, accel, func in acts:
610 a = G.Action(act, label, None, None)
611 if func: a.connect('activate', invoker(func))
612 actgroup.add_action_with_accel(a, accel)
613 return actgroup
614
615class GridPacker (G.Table):
984b6d33
MW
616 """
617 Like a Table, but with more state: makes filling in the widgets easier.
618 """
619
ca6eb20c 620 def __init__(me):
984b6d33 621 """Initialize a new GridPacker."""
ca6eb20c 622 G.Table.__init__(me)
623 me.row = 0
624 me.col = 0
625 me.rows = 1
626 me.cols = 1
627 me.set_border_width(4)
628 me.set_col_spacings(4)
629 me.set_row_spacings(4)
984b6d33 630
ca6eb20c 631 def pack(me, w, width = 1, newlinep = False,
632 xopt = G.EXPAND | G.FILL | G.SHRINK, yopt = 0,
633 xpad = 0, ypad = 0):
984b6d33
MW
634 """
635 Packs a new widget.
636
637 W is the widget to add. XOPY, YOPT, XPAD and YPAD are as for Table.
638 WIDTH is how many cells to take up horizontally. NEWLINEP is whether to
639 start a new line for this widget. Returns W.
640 """
ca6eb20c 641 if newlinep:
642 me.row += 1
643 me.col = 0
644 bot = me.row + 1
645 right = me.col + width
646 if bot > me.rows or right > me.cols:
647 if bot > me.rows: me.rows = bot
648 if right > me.cols: me.cols = right
649 me.resize(me.rows, me.cols)
650 me.attach(w, me.col, me.col + width, me.row, me.row + 1,
651 xopt, yopt, xpad, ypad)
652 me.col += width
653 return w
984b6d33 654
ca6eb20c 655 def labelled(me, lab, w, newlinep = False, **kw):
984b6d33
MW
656 """
657 Packs a labelled widget.
658
659 Other arguments are as for pack. Returns W.
660 """
661 label = G.Label(lab + ' ')
ca6eb20c 662 label.set_alignment(1.0, 0)
663 me.pack(label, newlinep = newlinep, xopt = G.FILL)
664 me.pack(w, **kw)
665 return w
984b6d33 666
ca6eb20c 667 def info(me, label, text = None, len = 18, **kw):
984b6d33
MW
668 """
669 Packs an information widget with a label.
670
671 LABEL is the label; TEXT is the initial text; LEN is the estimated length
672 in characters. Returns the entry widget.
673 """
674 e = G.Label()
ca6eb20c 675 if text is not None: e.set_text(text)
676 e.set_width_chars(len)
984b6d33
MW
677 e.set_selectable(True)
678 e.set_alignment(0.0, 0.5)
ca6eb20c 679 me.labelled(label, e, **kw)
680 return e
681
060ca767 682class WindowSlot (HookClient):
984b6d33
MW
683 """
684 A place to store a window -- specificially a MyWindowMixin.
685
686 If the window is destroyed, remember this; when we come to open the window,
687 raise it if it already exists; otherwise make a new one.
688 """
060ca767 689 def __init__(me, createfunc):
984b6d33
MW
690 """
691 Constructor: CREATEFUNC must return a new Window which supports the
692 closehook protocol.
693 """
060ca767 694 HookClient.__init__(me)
695 me.createfunc = createfunc
696 me.window = None
984b6d33 697
060ca767 698 def open(me):
699 """Opens the window, creating it if necessary."""
700 if me.window:
a69f4417 701 raise_window(me.window)
060ca767 702 else:
703 me.window = me.createfunc()
704 me.hook(me.window.closehook, me.closed)
984b6d33 705
060ca767 706 def closed(me):
984b6d33 707 """Handles the window being closed."""
060ca767 708 me.unhook(me.window.closehook)
709 me.window = None
710
984b6d33
MW
711class MyTreeView (G.TreeView):
712 def __init__(me, model):
713 G.TreeView.__init__(me, model)
714 me.set_rules_hint(True)
715
716class MyScrolledWindow (G.ScrolledWindow):
717 def __init__(me):
718 G.ScrolledWindow.__init__(me)
719 me.set_policy(G.POLICY_AUTOMATIC, G.POLICY_AUTOMATIC)
720 me.set_shadow_type(G.SHADOW_IN)
721
722## Matches a signed integer.
723rx_num = RX.compile(r'^[-+]?\d+$')
724
725## The colour red.
c739b22c 726c_red = GDK.color_parse('#ff6666')
984b6d33 727
060ca767 728class ValidationError (Exception):
729 """Raised by ValidatingEntry.get_text() if the text isn't valid."""
730 pass
984b6d33 731
060ca767 732class ValidatingEntry (G.Entry):
984b6d33
MW
733 """
734 Like an Entry, but makes the text go red if the contents are invalid.
735
736 If get_text is called, and the text is invalid, ValidationError is raised.
737 The attribute validp reflects whether the contents are currently valid.
738 """
739
060ca767 740 def __init__(me, valid, text = '', size = -1, *arg, **kw):
984b6d33
MW
741 """
742 Make a validating Entry.
743
744 VALID is a regular expression or a predicate on strings. TEXT is the
745 default text to insert. SIZE is the size of the box to set, in
746 characters (ish). Other arguments are passed to Entry.
747 """
060ca767 748 G.Entry.__init__(me, *arg, **kw)
c739b22c
MW
749 me.connect("changed", me._check)
750 me.connect("state-changed", me._check)
060ca767 751 if callable(valid):
752 me.validate = valid
753 else:
754 me.validate = RX.compile(valid).match
755 me.ensure_style()
060ca767 756 if size != -1: me.set_width_chars(size)
757 me.set_activates_default(True)
758 me.set_text(text)
c739b22c 759 me._check()
984b6d33 760
c739b22c 761 def _check(me, *hunoz):
984b6d33 762 """Check the current text and update validp and the text colour."""
060ca767 763 if me.validate(G.Entry.get_text(me)):
764 me.validp = True
a69f4417 765 set_entry_bg(me, None)
060ca767 766 else:
767 me.validp = False
a69f4417 768 set_entry_bg(me, me.is_sensitive() and c_red or None)
984b6d33 769
060ca767 770 def get_text(me):
984b6d33
MW
771 """
772 Return the text in the Entry if it's valid. If it isn't, raise
773 ValidationError.
774 """
060ca767 775 if not me.validp:
776 raise ValidationError
777 return G.Entry.get_text(me)
778
779def numericvalidate(min = None, max = None):
984b6d33
MW
780 """
781 Return a validation function for numbers.
782
783 Entry must consist of an optional sign followed by digits, and the
784 resulting integer must be within the given bounds.
785 """
060ca767 786 return lambda x: (rx_num.match(x) and
787 (min is None or long(x) >= min) and
788 (max is None or long(x) <= max))
789
984b6d33
MW
790###--------------------------------------------------------------------------
791### Various minor dialog boxen.
060ca767 792
11ad66c2
MW
793GPL = """\
794TrIPE is free software: you can redistribute it and/or modify it under
795the terms of the GNU General Public License as published by the Free
796Software Foundation; either version 3 of the License, or (at your
797option) any later version.
060ca767 798
11ad66c2
MW
799TrIPE is distributed in the hope that it will be useful, but WITHOUT
800ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
801FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
802for more details.
060ca767 803
804You should have received a copy of the GNU General Public License
11ad66c2 805along with TrIPE. If not, see <https://www.gnu.org/licenses/>."""
060ca767 806
dcd075c8 807class AboutBox (G.AboutDialog, TrivialWindowMixin):
060ca767 808 """The program `About' box."""
809 def __init__(me):
810 G.AboutDialog.__init__(me)
811 me.mywininit()
812 me.set_name('TrIPEmon')
984b6d33 813 me.set_version(T.VERSION)
060ca767 814 me.set_license(GPL)
984b6d33
MW
815 me.set_authors(['Mark Wooding <mdw@distorted.org.uk>'])
816 me.set_comments('A graphical monitor for the TrIPE VPN server')
817 me.set_copyright('Copyright © 2006-2008 Straylight/Edgeware')
818 me.connect('response', me.respond)
060ca767 819 me.show()
984b6d33
MW
820 def respond(me, hunoz, rid, *hukairz):
821 if rid == G.RESPONSE_CANCEL:
822 me.close()
060ca767 823aboutbox = WindowSlot(AboutBox)
824
825def moanbox(msg):
826 """Report an error message in a window."""
984b6d33 827 d = G.Dialog('Error from %s' % M.quis,
060ca767 828 flags = G.DIALOG_MODAL,
829 buttons = ((G.STOCK_OK, G.RESPONSE_NONE)))
830 label = G.Label(msg)
831 label.set_padding(20, 20)
a69f4417 832 d.vbox.pack_start(label, True, True, 0)
060ca767 833 label.show()
834 d.run()
835 d.destroy()
836
ca6eb20c 837def unimplemented(*hunoz):
838 """Indicator of laziness."""
839 moanbox("I've not written that bit yet.")
840
984b6d33
MW
841###--------------------------------------------------------------------------
842### Logging windows.
ca6eb20c 843
984b6d33
MW
844class LogModel (G.ListStore):
845 """
846 A simple list of log messages, usable as the model for a TreeView.
ca6eb20c 847
984b6d33
MW
848 The column headings are stored in the `cols' attribute.
849 """
060ca767 850
060ca767 851 def __init__(me, columns):
984b6d33
MW
852 """
853 COLUMNS must be a list of column name strings. We add a time column to
854 the left.
855 """
060ca767 856 me.cols = ('Time',) + columns
857 G.ListStore.__init__(me, *((GO.TYPE_STRING,) * len(me.cols)))
984b6d33 858
060ca767 859 def add(me, *entries):
984b6d33
MW
860 """
861 Adds a new log message, with a timestamp.
862
863 The ENTRIES are the contents for the list columns.
864 """
865 now = TIME.strftime('%Y-%m-%d %H:%M:%S')
866 me.append((now, ) + entries)
060ca767 867
868class TraceLogModel (LogModel):
869 """Log model for trace messages."""
870 def __init__(me):
871 LogModel.__init__(me, ('Message',))
872 def notify(me, line):
873 """Call with a new trace message."""
874 me.add(line)
875
876class WarningLogModel (LogModel):
984b6d33
MW
877 """
878 Log model for warnings.
879
880 We split the category out into a separate column.
881 """
060ca767 882 def __init__(me):
883 LogModel.__init__(me, ('Category', 'Message'))
984b6d33 884 def notify(me, tag, *rest):
060ca767 885 """Call with a new warning message."""
984b6d33 886 me.add(tag, ' '.join([T.quotify(w) for w in rest]))
060ca767 887
dcd075c8 888class LogViewer (TrivialWindow):
984b6d33
MW
889 """
890 A log viewer window.
891
892 Its contents are a TreeView showing the log.
893
894 Attributes:
895
896 * model: an appropriate LogModel
897 * list: a TreeView widget to display the log
898 """
899
060ca767 900 def __init__(me, model):
984b6d33
MW
901 """
902 Create a log viewer showing the LogModel MODEL.
903 """
dcd075c8 904 TrivialWindow.__init__(me)
060ca767 905 me.model = model
984b6d33
MW
906 scr = MyScrolledWindow()
907 me.list = MyTreeView(me.model)
060ca767 908 i = 0
909 for c in me.model.cols:
984b6d33
MW
910 crt = G.CellRendererText()
911 me.list.append_column(G.TreeViewColumn(c, crt, text = i))
060ca767 912 i += 1
984b6d33 913 crt.set_property('family', 'monospace')
060ca767 914 me.set_default_size(440, 256)
915 scr.add(me.list)
916 me.add(scr)
917 me.show_all()
918
984b6d33
MW
919###--------------------------------------------------------------------------
920### Pinging peers.
921
922class pingstate (struct):
923 """
924 Information kept for each peer by the Pinger.
925
926 Important attributes:
927
928 * peer = the peer name
929 * command = PING or EPING
930 * n = how many pings we've sent so far
931 * ngood = how many returned
932 * nmiss = how many didn't return
933 * nmissrun = how many pings since the last good one
934 * tlast = round-trip time for the last (good) ping
935 * ttot = total roung trip time
936 """
937 pass
938
939class Pinger (T.Coroutine, HookClient):
940 """
941 Coroutine which pings known peers and collects statistics.
942
943 Interesting attributes:
944
945 * _map: dict mapping peer names to Peer objects
946 * _q: event queue for notifying pinger coroutine
947 * _timer: gobject timer for waking the coroutine
948 """
949
950 def __init__(me):
951 """
952 Initialize the pinger.
953
954 We watch the monitor's PeerList to track which peers we should ping. We
955 maintain an event queue and put all the events on that.
956
957 The statistics for a PEER are held in the Peer object, in PEER.ping[CMD],
958 where CMD is 'PING' or 'EPING'.
959 """
960 T.Coroutine.__init__(me)
961 HookClient.__init__(me)
962 me._map = {}
963 me._q = T.Queue()
964 me._timer = None
965 me.hook(conn.connecthook, me._connected)
966 me.hook(conn.disconnecthook, me._disconnected)
967 me.hook(monitor.peers.addhook,
690a6ec1 968 lambda p: T.defer(me._q.put, (p, 'ADD', None)))
984b6d33 969 me.hook(monitor.peers.delhook,
690a6ec1 970 lambda p: T.defer(me._q.put, (p, 'KILL', None)))
984b6d33
MW
971 if conn.connectedp(): me.connected()
972
973 def _connected(me):
974 """Respond to connection: start pinging thngs."""
a69f4417 975 me._timer = GL.timeout_add(1000, me._timerfunc)
984b6d33
MW
976
977 def _timerfunc(me):
978 """Timer function: put a timer event on the queue."""
979 me._q.put((None, 'TIMER', None))
980 return True
981
982 def _disconnected(me, reason):
983 """Respond to disconnection: stop pinging."""
a69f4417 984 GL.source_remove(me._timer)
984b6d33
MW
985
986 def run(me):
987 """
988 Coroutine function: read events from the queue and process them.
989
990 Interesting events:
991
992 * (PEER, 'KILL', None): remove PEER from the interesting peers list
993 * (PEER, 'ADD', None): add PEER to the list
994 * (PEER, 'INFO', TOKENS): result from a PING command
995 * (None, 'TIMER', None): interval timer went off: send more pings
996 """
997 while True:
998 tag, code, stuff = me._q.get()
999 if code == 'KILL':
1000 name = tag.name
1001 if name in me._map:
1002 del me._map[name]
1003 elif not conn.connectedp():
1004 pass
1005 elif code == 'ADD':
1006 p = tag
1007 p.ping = {}
1008 for cmd in 'PING', 'EPING':
1009 ps = pingstate(command = cmd, peer = p,
1010 n = 0, ngood = 0, nmiss = 0, nmissrun = 0,
1011 tlast = 0, ttot = 0)
1012 p.ping[cmd] = ps
1013 me._map[p.name] = p
1014 elif code == 'INFO':
1015 ps = tag
1016 if stuff[0] == 'ping-ok':
1017 t = float(stuff[1])
1018 ps.ngood += 1
1019 ps.nmissrun = 0
1020 ps.tlast = t
1021 ps.ttot += t
1022 else:
1023 ps.nmiss += 1
1024 ps.nmissrun += 1
1025 ps.n += 1
1026 ps.peer.pinghook.run(ps.peer, ps.command, ps)
1027 elif code == 'TIMER':
1028 for name, p in me._map.iteritems():
1029 for cmd, ps in p.ping.iteritems():
1030 conn.rawcommand(T.TripeAsynchronousCommand(me._q, ps, [
1031 cmd, '-background', conn.bgtag(), '--', name]))
1032
1033###--------------------------------------------------------------------------
1034### Random dialogue boxes.
1035
1036class AddPeerDialog (MyDialog):
1037 """
1038 Let the user create a new peer the low-level way.
1039
1040 Interesting attributes:
1041
1042 * e_name, e_addr, e_port, c_keepalive, l_tunnel: widgets in the dialog
1043 """
1044
1045 def __init__(me):
1046 """Initialize the dialogue."""
1047 MyDialog.__init__(me, 'Add peer',
1048 buttons = [(G.STOCK_CANCEL, me.destroy),
1049 (G.STOCK_OK, me.ok)])
1050 me._setup()
1051
1052 @incr
1053 def _setup(me):
1054 """Coroutine function: background setup for AddPeerDialog."""
1055 table = GridPacker()
a69f4417 1056 me.vbox.pack_start(table, True, True, 0)
984b6d33
MW
1057 me.e_name = table.labelled('Name',
1058 ValidatingEntry(r'^[^\s.:]+$', '', 16),
1059 width = 3)
1060 me.e_addr = table.labelled('Address',
1061 ValidatingEntry(r'^[a-zA-Z0-9.-]+$', '', 24),
1062 newlinep = True)
1063 me.e_port = table.labelled('Port',
1064 ValidatingEntry(numericvalidate(0, 65535),
1065 '4070',
1066 5))
a69f4417 1067 me.l_tunnel = table.labelled('Tunnel', combo_box_text(),
984b6d33 1068 newlinep = True, width = 3)
0adda4f8 1069 me.tuns = ['(Default)'] + conn.tunnels()
984b6d33
MW
1070 for t in me.tuns:
1071 me.l_tunnel.append_text(t)
1072 me.l_tunnel.set_active(0)
15e8239f
MW
1073
1074 def tickybox_sensitivity(tickybox, target):
1075 tickybox.connect('toggled',
1076 lambda t: target.set_sensitive (t.get_active()))
1077
1078 me.c_keepalive = G.CheckButton('Keepalives')
984b6d33 1079 table.pack(me.c_keepalive, newlinep = True, xopt = G.FILL)
984b6d33
MW
1080 me.e_keepalive = ValidatingEntry(r'^\d+[hms]?$', '', 5)
1081 me.e_keepalive.set_sensitive(False)
15e8239f 1082 tickybox_sensitivity(me.c_keepalive, me.e_keepalive)
984b6d33 1083 table.pack(me.e_keepalive, width = 3)
15e8239f 1084
f6208364
MW
1085 me.c_cork = G.CheckButton('Cork')
1086 table.pack(me.c_cork, newlinep = True, width = 4, xopt = G.FILL)
1087
15e8239f
MW
1088 me.c_mobile = G.CheckButton('Mobile')
1089 table.pack(me.c_mobile, newlinep = True, width = 4, xopt = G.FILL)
1090
1091 me.c_peerkey = G.CheckButton('Peer key tag')
1092 table.pack(me.c_peerkey, newlinep = True, xopt = G.FILL)
1093 me.e_peerkey = ValidatingEntry(r'^[^.:\s]+$', '', 16)
1094 me.e_peerkey.set_sensitive(False)
1095 tickybox_sensitivity(me.c_peerkey, me.e_peerkey)
1096 table.pack(me.e_peerkey, width = 3)
1097
1098 me.c_privkey = G.CheckButton('Private key tag')
1099 table.pack(me.c_privkey, newlinep = True, xopt = G.FILL)
1100 me.e_privkey = ValidatingEntry(r'^[^.:\s]+$', '', 16)
1101 me.e_privkey.set_sensitive(False)
1102 tickybox_sensitivity(me.c_privkey, me.e_privkey)
1103 table.pack(me.e_privkey, width = 3)
1104
984b6d33
MW
1105 me.show_all()
1106
1107 def ok(me):
1108 """Handle an OK press: create the peer."""
1109 try:
984b6d33 1110 t = me.l_tunnel.get_active()
15e8239f
MW
1111 me._addpeer(me.e_name.get_text(),
1112 me.e_addr.get_text(),
1113 me.e_port.get_text(),
1114 keepalive = (me.c_keepalive.get_active() and
1115 me.e_keepalive.get_text() or None),
1116 tunnel = t and me.tuns[t] or None,
f6208364 1117 cork = me.c_cork.get_active() or None,
f40cd7f1 1118 mobile = me.c_mobile.get_active() or None,
15e8239f
MW
1119 key = (me.c_peerkey.get_active() and
1120 me.e_peerkey.get_text() or None),
1121 priv = (me.c_privkey.get_active() and
1122 me.e_privkey.get_text() or None))
984b6d33
MW
1123 except ValidationError:
1124 GDK.beep()
1125 return
1126
1127 @incr
15e8239f 1128 def _addpeer(me, *args, **kw):
984b6d33
MW
1129 """Coroutine function: actually do the ADD command."""
1130 try:
15e8239f 1131 conn.add(*args, **kw)
984b6d33
MW
1132 me.destroy()
1133 except T.TripeError, exc:
690a6ec1 1134 T.defer(moanbox, ' '.join(exc))
984b6d33 1135
dcd075c8 1136class ServInfo (TrivialWindow):
984b6d33
MW
1137 """
1138 Show information about the server and available services.
1139
1140 Interesting attributes:
1141
1142 * e: maps SERVINFO keys to entry widgets
1143 * svcs: Gtk ListStore describing services (columns are name and version)
1144 """
1145
1146 def __init__(me):
dcd075c8 1147 TrivialWindow.__init__(me)
984b6d33
MW
1148 me.set_title('TrIPE server info')
1149 table = GridPacker()
1150 me.add(table)
1151 me.e = {}
1152 def add(label, tag, text = None, **kw):
1153 me.e[tag] = table.info(label, text, **kw)
1154 add('Implementation', 'implementation')
1155 add('Version', 'version', newlinep = True)
1156 me.svcs = G.ListStore(*(GO.TYPE_STRING,) * 2)
1157 me.svcs.set_sort_column_id(0, G.SORT_ASCENDING)
1158 scr = MyScrolledWindow()
1159 lb = MyTreeView(me.svcs)
1160 i = 0
1161 for title in 'Service', 'Version':
1162 lb.append_column(G.TreeViewColumn(
1163 title, G.CellRendererText(), text = i))
1164 i += 1
1165 for svc in monitor.services:
1166 me.svcs.append([svc.name, svc.version])
1167 scr.add(lb)
1168 table.pack(scr, width = 2, newlinep = True,
1169 yopt = G.EXPAND | G.FILL | G.SHRINK)
1170 me.update()
1171 me.hook(conn.connecthook, me.update)
1172 me.hook(monitor.services.addhook, me.addsvc)
1173 me.hook(monitor.services.delhook, me.delsvc)
1174 me.show_all()
1175
1176 def addsvc(me, svc):
1177 me.svcs.append([svc.name, svc.version])
1178
1179 def delsvc(me, svc):
1180 for i in xrange(len(me.svcs)):
1181 if me.svcs[i][0] == svc.name:
1182 me.svcs.remove(me.svcs.get_iter(i))
1183 break
1184 @incr
1185 def update(me):
1186 info = conn.servinfo()
1187 for i in me.e:
1188 me.e[i].set_text(info[i])
1189
1190class TraceOptions (MyDialog):
1191 """Tracing options window."""
1192 def __init__(me):
1193 MyDialog.__init__(me, title = 'Tracing options',
1194 buttons = [(G.STOCK_CLOSE, me.destroy),
1195 (G.STOCK_OK, cr(me.ok))])
1196 me._setup()
1197
1198 @incr
1199 def _setup(me):
1200 me.opts = []
1201 for ch, st, desc in conn.trace():
1202 if ch.isupper(): continue
1203 text = desc[0].upper() + desc[1:]
1204 ticky = G.CheckButton(text)
1205 ticky.set_active(st == '+')
a69f4417 1206 me.vbox.pack_start(ticky, True, True, 0)
984b6d33
MW
1207 me.opts.append((ch, ticky))
1208 me.show_all()
1209 def ok(me):
1210 on = []
1211 off = []
1212 for ch, ticky in me.opts:
1213 if ticky.get_active():
1214 on.append(ch)
1215 else:
1216 off.append(ch)
1217 setting = ''.join(on) + '-' + ''.join(off)
1218 conn.trace(setting)
1219 me.destroy()
1220
1221###--------------------------------------------------------------------------
1222### Peer window.
060ca767 1223
1224def xlate_time(t):
984b6d33 1225 """Translate a TrIPE-format time to something human-readable."""
060ca767 1226 if t == 'NEVER': return '(never)'
37941236 1227 YY, MM, DD, hh, mm, ss = map(int, rx_time.match(t).group(1, 2, 3, 4, 5, 6))
984b6d33
MW
1228 ago = TIME.time() - TIME.mktime((YY, MM, DD, hh, mm, ss, 0, 0, -1))
1229 ago = MATH.floor(ago); unit = 's'
37941236 1230 for n, u in [(60, 'min'), (60, 'hrs'), (24, 'days')]:
1231 if ago < 2*n: break
1232 ago /= n
1233 unit = u
1234 return '%04d:%02d:%02d %02d:%02d:%02d (%.1f %s ago)' % \
1235 (YY, MM, DD, hh, mm, ss, ago, unit)
060ca767 1236def xlate_bytes(b):
9a06b923 1237 """Translate a raw byte count into something a human might want to read."""
060ca767 1238 suff = 'B'
1239 b = int(b)
1240 for s in 'KMG':
1241 if b < 4096: break
1242 b /= 1024
1243 suff = s
1244 return '%d %s' % (b, suff)
1245
1246## How to translate peer stats. Maps the stat name to a translation
1247## function.
1248statsxlate = \
1249 [('start-time', xlate_time),
1250 ('last-packet-time', xlate_time),
1251 ('last-keyexch-time', xlate_time),
1252 ('bytes-in', xlate_bytes),
1253 ('bytes-out', xlate_bytes),
1254 ('keyexch-bytes-in', xlate_bytes),
1255 ('keyexch-bytes-out', xlate_bytes),
1256 ('ip-bytes-in', xlate_bytes),
1257 ('ip-bytes-out', xlate_bytes)]
1258
1259## How to lay out the stats dialog. Format is (LABEL, FORMAT): LABEL is
1260## the label to give the entry box; FORMAT is the format string to write into
1261## the entry.
1262statslayout = \
1263 [('Start time', '%(start-time)s'),
1ff5474e
MW
1264 ('Private key', '%(current-key)s'),
1265 ('Diffie-Hellman group',
1266 '%(kx-group)s '
1267 '(%(kx-group-order-bits)s-bit order, '
1268 '%(kx-group-elt-bits)s-bit elements)'),
1269 ('Cipher',
1270 '%(cipher)s (%(cipher-keysz)s-bit key, %(cipher-blksz)s-bit block)'),
1271 ('Mac', '%(mac)s (%(mac-keysz)s-bit key, %(mac-tagsz)s-bit tag)'),
1272 ('Hash', '%(hash)s (%(hash-sz)s-bit output)'),
060ca767 1273 ('Last key-exchange', '%(last-keyexch-time)s'),
1274 ('Last packet', '%(last-packet-time)s'),
1275 ('Packets in/out',
1276 '%(packets-in)s (%(bytes-in)s) / %(packets-out)s (%(bytes-out)s)'),
1277 ('Key-exchange in/out',
1278 '%(keyexch-packets-in)s (%(keyexch-bytes-in)s) / %(keyexch-packets-out)s (%(keyexch-bytes-out)s)'),
1279 ('IP in/out',
984b6d33 1280 '%(ip-packets-in)s (%(ip-bytes-in)s) / %(ip-packets-out)s (%(ip-bytes-out)s)'),
060ca767 1281 ('Rejected packets', '%(rejected-packets)s')]
1282
dcd075c8 1283class PeerWindow (TrivialWindow):
984b6d33
MW
1284 """
1285 Show information about a peer.
1286
1287 This gives a graphical view of the server's peer statistics.
1288
1289 Interesting attributes:
1290
1291 * e: dict mapping keys (mostly matching label widget texts, though pings
1292 use command names) to entry widgets so that we can update them easily
1293 * peer: the peer this window shows information about
1294 * cr: the info-fetching coroutine, or None if crrrently disconnected
1295 * doupate: whether the info-fetching corouting should continue running
1296 """
1297
1298 def __init__(me, peer):
1299 """Construct a PeerWindow, showing information about PEER."""
1300
dcd075c8 1301 TrivialWindow.__init__(me)
060ca767 1302 me.set_title('TrIPE statistics: %s' % peer.name)
060ca767 1303 me.peer = peer
984b6d33 1304
060ca767 1305 table = GridPacker()
1306 me.add(table)
984b6d33
MW
1307
1308 ## Utility for adding fields.
060ca767 1309 me.e = {}
984b6d33
MW
1310 def add(label, text = None, key = None):
1311 if key is None: key = label
1312 me.e[key] = table.info(label, text, len = 42, newlinep = True)
1313
1314 ## Build the dialogue box.
060ca767 1315 add('Peer name', peer.name)
1316 add('Tunnel', peer.tunnel)
1317 add('Interface', peer.ifname)
1318 add('Keepalives',
1319 (peer.keepalive == '0' and 'never') or '%s s' % peer.keepalive)
1320 add('Address', peer.addr)
984b6d33
MW
1321 add('Transport pings', key = 'PING')
1322 add('Encrypted pings', key = 'EPING')
1323
1324 for label, format in statslayout:
1325 add(label)
1326
1327 ## Hook onto various interesting events.
1328 me.hook(conn.connecthook, me.tryupdate)
1329 me.hook(conn.disconnecthook, me.stopupdate)
060ca767 1330 me.hook(me.closehook, me.stopupdate)
1331 me.hook(me.peer.deadhook, me.dead)
984b6d33 1332 me.hook(me.peer.changehook, me.change)
060ca767 1333 me.hook(me.peer.pinghook, me.ping)
984b6d33
MW
1334 me.cr = None
1335 me.doupdate = True
060ca767 1336 me.tryupdate()
984b6d33
MW
1337
1338 ## Format the ping statistics.
1339 for cmd, ps in me.peer.ping.iteritems():
1340 me.ping(me.peer, cmd, ps)
1341
1342 ## And show the window.
060ca767 1343 me.show_all()
984b6d33
MW
1344
1345 def change(me):
1346 """Update the display in response to a notification."""
1347 me.e['Interface'].set_text(me.peer.ifname)
1348
1349 def _update(me):
1350 """
1351 Main display-updating coroutine.
1352
1353 This does an update, sleeps for a while, and starts again. If the
1354 me.doupdate flag goes low, we stop the loop.
1355 """
1356 while me.peer.alivep and conn.connectedp() and me.doupdate:
1357 stat = conn.stats(me.peer.name)
1358 for s, trans in statsxlate:
1359 stat[s] = trans(stat[s])
1ff5474e 1360 stat.update(me.peer.__dict__)
984b6d33
MW
1361 for label, format in statslayout:
1362 me.e[label].set_text(format % stat)
a69f4417 1363 GL.timeout_add(1000, lambda: me.cr.switch() and False)
984b6d33
MW
1364 me.cr.parent.switch()
1365 me.cr = None
1366
060ca767 1367 def tryupdate(me):
984b6d33
MW
1368 """Start the updater coroutine, if it's not going already."""
1369 if me.cr is None:
22b47552
MW
1370 me.cr = T.Coroutine(me._update,
1371 name = 'update-peer-window %s' % me.peer.name)
984b6d33
MW
1372 me.cr.switch()
1373
1374 def stopupdate(me, *hunoz, **hukairz):
1375 """Stop the update coroutine, by setting me.doupdate."""
1376 me.doupdate = False
1377
060ca767 1378 def dead(me):
984b6d33 1379 """Called when the peer is killed."""
060ca767 1380 me.set_title('TrIPE statistics: %s [defunct]' % me.peer.name)
1381 me.e['Peer name'].set_text('%s [defunct]' % me.peer.name)
1382 me.stopupdate()
060ca767 1383
984b6d33
MW
1384 def ping(me, peer, cmd, ps):
1385 """Called when a ping result for the peer is reported."""
1386 s = '%d/%d' % (ps.ngood, ps.n)
1387 if ps.n:
1388 s += ' (%.1f%%)' % (ps.ngood * 100.0/ps.n)
1389 if ps.ngood:
1390 s += '; %.2f ms (last %.1f ms)' % (ps.ttot/ps.ngood, ps.tlast);
1391 me.e[ps.command].set_text(s)
1392
1393###--------------------------------------------------------------------------
1394### Cryptographic status.
1395
dcd075c8 1396class CryptoInfo (TrivialWindow):
984b6d33
MW
1397 """Simple display of cryptographic algorithms in use."""
1398 def __init__(me):
dcd075c8 1399 TrivialWindow.__init__(me)
984b6d33 1400 me.set_title('Cryptographic algorithms')
690a6ec1 1401 T.aside(me.populate)
984b6d33 1402 def populate(me):
060ca767 1403 table = GridPacker()
984b6d33 1404 me.add(table)
060ca767 1405
984b6d33
MW
1406 crypto = conn.algs()
1407 table.info('Diffie-Hellman group',
1408 '%s (%d-bit order, %d-bit elements)' %
1409 (crypto['kx-group'],
1410 int(crypto['kx-group-order-bits']),
1411 int(crypto['kx-group-elt-bits'])),
1412 len = 32)
1413 table.info('Data encryption',
1414 '%s (%d-bit key; %s)' %
1415 (crypto['cipher'],
1416 int(crypto['cipher-keysz']) * 8,
1417 crypto['cipher-blksz'] == '0'
1418 and 'stream cipher'
1419 or '%d-bit block' % (int(crypto['cipher-blksz']) * 8)),
1420 newlinep = True)
1421 table.info('Message authentication',
1422 '%s (%d-bit key; %d-bit tag)' %
1423 (crypto['mac'],
1424 int(crypto['mac-keysz']) * 8,
1425 int(crypto['mac-tagsz']) * 8),
1426 newlinep = True)
1427 table.info('Hash function',
1428 '%s (%d-bit output)' %
1429 (crypto['hash'],
1430 int(crypto['hash-sz']) * 8),
1431 newlinep = True)
060ca767 1432
984b6d33
MW
1433 me.show_all()
1434
1435###--------------------------------------------------------------------------
1436### Main monitor window.
060ca767 1437
1438class MonitorWindow (MyWindow):
1439
984b6d33
MW
1440 """
1441 The main monitor window.
1442
1443 This class creates, populates and maintains the main monitor window.
1444
1445 Lots of attributes:
1446
1447 * warnings, trace: log models for server output
1448 * warnview, traceview, traceopts, addpeerwin, cryptoinfo, servinfo:
1449 WindowSlot objects for ancillary windows
1450 * ui: Gtk UIManager object for the menu system
1451 * apmenu: pair of identical autoconnecting peer menus
1452 * listmodel: Gtk ListStore for connected peers; contains peer name,
1453 address, and ping times (transport and encrypted, value and colour)
1454 * status: Gtk Statusbar at the bottom of the window
1455 * _kidding: an unpleasant backchannel between the apchange method (which
1456 builds the apmenus) and the menu handler, forced on us by a Gtk
1457 misfeature
1458
1459 Also installs attributes on Peer objects:
1460
1461 * i: index of peer's entry in listmodel
1462 * win: WindowSlot object for the peer's PeerWindow
1463 """
1464
1465 def __init__(me):
1466 """Construct the window."""
1467
1468 ## Basic stuff.
060ca767 1469 MyWindow.__init__(me)
1470 me.set_title('TrIPE monitor')
984b6d33
MW
1471
1472 ## Hook onto diagnostic outputs.
060ca767 1473 me.warnings = WarningLogModel()
984b6d33 1474 me.hook(conn.warnhook, me.warnings.notify)
060ca767 1475 me.trace = TraceLogModel()
984b6d33 1476 me.hook(conn.tracehook, me.trace.notify)
060ca767 1477
984b6d33 1478 ## Make slots to store the various ancillary singleton windows.
060ca767 1479 me.warnview = WindowSlot(lambda: LogViewer(me.warnings))
1480 me.traceview = WindowSlot(lambda: LogViewer(me.trace))
984b6d33
MW
1481 me.traceopts = WindowSlot(lambda: TraceOptions())
1482 me.addpeerwin = WindowSlot(lambda: AddPeerDialog())
1483 me.cryptoinfo = WindowSlot(lambda: CryptoInfo())
1484 me.servinfo = WindowSlot(lambda: ServInfo())
060ca767 1485
984b6d33 1486 ## Main window structure.
060ca767 1487 vbox = G.VBox()
1488 me.add(vbox)
1489
984b6d33 1490 ## UI manager makes our menus. (We're too cheap to have a toolbar.)
060ca767 1491 me.ui = G.UIManager()
1492 actgroup = makeactiongroup('monitor',
1493 [('file-menu', '_File', None, None),
984b6d33
MW
1494 ('connect', '_Connect', '<Control>C', conn.connect),
1495 ('disconnect', '_Disconnect', '<Control>D',
1496 lambda: conn.disconnect(None)),
1497 ('quit', '_Quit', '<Control>Q', me.close),
060ca767 1498 ('server-menu', '_Server', None, None),
984b6d33
MW
1499 ('daemon', 'Run in _background', None, cr(conn.daemon)),
1500 ('server-version', 'Server version', '<Control>V', me.servinfo.open),
1501 ('crypto-algs', 'Cryptographic algorithms',
1502 '<Control>Y', me.cryptoinfo.open),
1503 ('reload-keys', 'Reload keys', '<Control>R', cr(conn.reload)),
1504 ('server-quit', 'Terminate server', None, cr(conn.quit)),
1505 ('conn-peer', 'Connect peer', None, None),
060ca767 1506 ('logs-menu', '_Logs', None, None),
984b6d33
MW
1507 ('show-warnings', 'Show _warnings', '<Control>W', me.warnview.open),
1508 ('show-trace', 'Show _trace', '<Control>T', me.traceview.open),
060ca767 1509 ('trace-options', 'Trace _options...', None, me.traceopts.open),
1510 ('help-menu', '_Help', None, None),
1511 ('about', '_About tripemon...', None, aboutbox.open),
984b6d33 1512 ('add-peer', '_Add peer...', '<Control>A', me.addpeerwin.open),
060ca767 1513 ('kill-peer', '_Kill peer', None, me.killpeer),
1514 ('force-kx', 'Force key e_xchange', None, me.forcekx)])
984b6d33
MW
1515
1516 ## Menu structures.
060ca767 1517 uidef = '''
1518 <ui>
1519 <menubar>
1520 <menu action="file-menu">
1521 <menuitem action="quit"/>
1522 </menu>
1523 <menu action="server-menu">
984b6d33 1524 <menuitem action="connect"/>
060ca767 1525 <menuitem action="disconnect"/>
1526 <separator/>
ca6eb20c 1527 <menuitem action="server-version"/>
984b6d33 1528 <menuitem action="crypto-algs"/>
060ca767 1529 <menuitem action="add-peer"/>
984b6d33 1530 <menuitem action="conn-peer"/>
060ca767 1531 <menuitem action="daemon"/>
ca6eb20c 1532 <menuitem action="reload-keys"/>
060ca767 1533 <separator/>
1534 <menuitem action="server-quit"/>
1535 </menu>
1536 <menu action="logs-menu">
1537 <menuitem action="show-warnings"/>
1538 <menuitem action="show-trace"/>
1539 <menuitem action="trace-options"/>
1540 </menu>
1541 <menu action="help-menu">
1542 <menuitem action="about"/>
1543 </menu>
1544 </menubar>
1545 <popup name="peer-popup">
1546 <menuitem action="add-peer"/>
984b6d33 1547 <menuitem action="conn-peer"/>
060ca767 1548 <menuitem action="kill-peer"/>
1549 <menuitem action="force-kx"/>
1550 </popup>
1551 </ui>
1552 '''
984b6d33
MW
1553
1554 ## Populate the UI manager.
060ca767 1555 me.ui.insert_action_group(actgroup, 0)
1556 me.ui.add_ui_from_string(uidef)
984b6d33
MW
1557
1558 ## Construct the menu bar.
a69f4417 1559 vbox.pack_start(me.ui.get_widget('/menubar'), False, True, 0)
060ca767 1560 me.add_accel_group(me.ui.get_accel_group())
060ca767 1561
984b6d33
MW
1562 ## Construct and attach the auto-peers menu. (This is a horrible bodge
1563 ## because we can't attach the same submenu in two different places.)
1564 me.apmenu = G.Menu(), G.Menu()
1565 me.ui.get_widget('/menubar/server-menu/conn-peer') \
1566 .set_submenu(me.apmenu[0])
1567 me.ui.get_widget('/peer-popup/conn-peer').set_submenu(me.apmenu[1])
1568
1569 ## Construct the main list model, and listen on hooks which report
1570 ## changes to the available peers.
060ca767 1571 me.listmodel = G.ListStore(*(GO.TYPE_STRING,) * 6)
1572 me.listmodel.set_sort_column_id(0, G.SORT_ASCENDING)
984b6d33
MW
1573 me.hook(monitor.peers.addhook, me.addpeer)
1574 me.hook(monitor.peers.delhook, me.delpeer)
1575 me.hook(monitor.autopeershook, me.apchange)
060ca767 1576
984b6d33
MW
1577 ## Construct the list viewer and put it in a scrolling window.
1578 scr = MyScrolledWindow()
1579 me.list = MyTreeView(me.listmodel)
060ca767 1580 me.list.append_column(G.TreeViewColumn('Peer name',
1581 G.CellRendererText(),
1582 text = 0))
1583 me.list.append_column(G.TreeViewColumn('Address',
1584 G.CellRendererText(),
1585 text = 1))
1586 me.list.append_column(G.TreeViewColumn('T-ping',
1587 G.CellRendererText(),
1588 text = 2,
1589 foreground = 3))
1590 me.list.append_column(G.TreeViewColumn('E-ping',
1591 G.CellRendererText(),
1592 text = 4,
1593 foreground = 5))
1594 me.list.get_column(1).set_expand(True)
1595 me.list.connect('row-activated', me.activate)
1596 me.list.connect('button-press-event', me.buttonpress)
1597 me.list.set_reorderable(True)
1598 me.list.get_selection().set_mode(G.SELECTION_NONE)
1599 scr.add(me.list)
a69f4417 1600 vbox.pack_start(scr, True, True, 0)
060ca767 1601
984b6d33
MW
1602 ## Construct the status bar, and listen on hooks which report changes to
1603 ## connection status.
1604 me.status = G.Statusbar()
a69f4417 1605 vbox.pack_start(me.status, False, True, 0)
984b6d33
MW
1606 me.hook(conn.connecthook, cr(me.connected))
1607 me.hook(conn.disconnecthook, me.disconnected)
1608 me.hook(conn.notehook, me.notify)
1609
1610 ## Set a plausible default window size.
1611 me.set_default_size(512, 180)
060ca767 1612
1613 def addpeer(me, peer):
984b6d33 1614 """Hook: announces that PEER has been added."""
060ca767 1615 peer.i = me.listmodel.append([peer.name, peer.addr,
1616 '???', 'green', '???', 'green'])
984b6d33
MW
1617 peer.win = WindowSlot(lambda: PeerWindow(peer))
1618 me.hook(peer.pinghook, me._ping)
1619 me.apchange()
1620
060ca767 1621 def delpeer(me, peer):
984b6d33 1622 """Hook: announces that PEER has been removed."""
060ca767 1623 me.listmodel.remove(peer.i)
984b6d33
MW
1624 me.unhook(peer.pinghook)
1625 me.apchange()
1626
060ca767 1627 def path_peer(me, path):
984b6d33
MW
1628 """Return the peer corresponding to a given list-model PATH."""
1629 return monitor.peers[me.listmodel[path][0]]
1630
1631 def apchange(me):
1632 """
1633 Hook: announces that a change has been made to the peers available for
1634 automated connection.
1635
1636 This populates both auto-peer menus and keeps them in sync. (As
1637 mentioned above, we can't attach the same submenu to two separate parent
1638 menu items. So we end up with two identical menus instead. Yes, this
1639 does suck.)
1640 """
1641
1642 ## The set_active method of a CheckMenuItem works by maybe activating the
1643 ## menu item. This signals our handler. But we don't actually want to
1644 ## signal the handler unless the user actually frobbed the item. So the
1645 ## _kidding flag is used as an underhanded way of telling the handler
1646 ## that we don't actually want it to do anything. Of course, this sucks
1647 ## mightily.
1648 me._kidding = True
1649
1650 ## Iterate over the two menus.
1651 for m in 0, 1:
1652 menu = me.apmenu[m]
1653 existing = menu.get_children()
1654 if monitor.autopeers is None:
1655
1656 ## No peers, so empty out the menu.
1657 for item in existing:
1658 menu.remove(item)
1659
1660 else:
1661
1662 ## Insert the new items into the menu. (XXX this seems buggy XXX)
37bc5549 1663 ## Tick the peers which are actually connected.
984b6d33
MW
1664 i = j = 0
1665 for peer in monitor.autopeers:
1666 if j < len(existing) and \
1667 existing[j].get_child().get_text() == peer:
1668 item = existing[j]
1669 j += 1
1670 else:
1671 item = G.CheckMenuItem(peer, use_underline = False)
1672 item.connect('activate', invoker(me._addautopeer, peer))
1673 menu.insert(item, i)
1674 item.set_active(peer in monitor.peers.table)
1675 i += 1
1676
1677 ## Make all the menu items visible.
1678 menu.show_all()
1679
1680 ## Set the parent menu items sensitive if and only if there are any peers
1681 ## to connect.
1682 for name in ['/menubar/server-menu/conn-peer', '/peer-popup/conn-peer']:
1683 me.ui.get_widget(name).set_sensitive(bool(monitor.autopeers))
1684
1685 ## And now allow the handler to do its business normally.
1686 me._kidding = False
1687
1688 def _addautopeer(me, peer):
1689 """
1690 Automatically connect an auto-peer.
1691
1692 This method is invoked from the main coroutine. Since the actual
1693 connection needs to issue administration commands, we must spawn a new
1694 child coroutine for it.
1695 """
1696 if me._kidding:
1697 return
22b47552
MW
1698 T.Coroutine(me._addautopeer_hack,
1699 name = '_addautopeerhack %s' % peer).switch(peer)
984b6d33
MW
1700
1701 def _addautopeer_hack(me, peer):
1702 """Make an automated connection to PEER in response to a user click."""
1703 if me._kidding:
1704 return
1705 try:
1706 T._simple(conn.svcsubmit('connect', 'active', peer))
1707 except T.TripeError, exc:
690a6ec1 1708 T.defer(moanbox, ' '.join(exc.args))
984b6d33 1709 me.apchange()
060ca767 1710
1711 def activate(me, l, path, col):
984b6d33
MW
1712 """
1713 Handle a double-click on a peer in the main list: open a PeerInfo window.
1714 """
060ca767 1715 peer = me.path_peer(path)
1716 peer.win.open()
984b6d33 1717
060ca767 1718 def buttonpress(me, l, ev):
984b6d33
MW
1719 """
1720 Handle a mouse click on the main list.
1721
1722 Currently we're only interested in button-3, which pops up the peer menu.
1723 For future reference, we stash the peer that was clicked in me.menupeer.
1724 """
060ca767 1725 if ev.button == 3:
984b6d33
MW
1726 x, y = int(ev.x), int(ev.y)
1727 r = me.list.get_path_at_pos(x, y)
060ca767 1728 for i in '/peer-popup/kill-peer', '/peer-popup/force-kx':
984b6d33 1729 me.ui.get_widget(i).set_sensitive(conn.connectedp() and
060ca767 1730 r is not None)
984b6d33
MW
1731 me.ui.get_widget('/peer-popup/conn-peer'). \
1732 set_sensitive(bool(monitor.autopeers))
060ca767 1733 if r:
1734 me.menupeer = me.path_peer(r[0])
1735 else:
1736 me.menupeer = None
984b6d33
MW
1737 me.ui.get_widget('/peer-popup').popup(
1738 None, None, None, ev.button, ev.time)
060ca767 1739
1740 def killpeer(me):
984b6d33
MW
1741 """Kill a peer from the popup menu."""
1742 cr(conn.kill, me.menupeer.name)()
1743
060ca767 1744 def forcekx(me):
984b6d33
MW
1745 """Kickstart a key-exchange from the popup menu."""
1746 cr(conn.forcekx, me.menupeer.name)()
1747
1748 _columnmap = {'PING': (2, 3), 'EPING': (4, 5)}
1749 def _ping(me, p, cmd, ps):
1750 """Hook: responds to ping reports."""
1751 textcol, colourcol = me._columnmap[cmd]
1752 if ps.nmissrun:
1753 me.listmodel[p.i][textcol] = '(miss %d)' % ps.nmissrun
1754 me.listmodel[p.i][colourcol] = 'red'
060ca767 1755 else:
984b6d33
MW
1756 me.listmodel[p.i][textcol] = '%.1f ms' % ps.tlast
1757 me.listmodel[p.i][colourcol] = 'black'
1758
060ca767 1759 def setstatus(me, status):
984b6d33 1760 """Update the message in the status bar."""
060ca767 1761 me.status.pop(0)
1762 me.status.push(0, status)
984b6d33
MW
1763
1764 def notify(me, note, *rest):
1765 """Hook: invoked when interesting notifications occur."""
060ca767 1766 if note == 'DAEMON':
1767 me.ui.get_widget('/menubar/server-menu/daemon').set_sensitive(False)
984b6d33 1768
060ca767 1769 def connected(me):
984b6d33
MW
1770 """
1771 Hook: invoked when a connection is made to the server.
1772
1773 Make options which require a server connection sensitive.
1774 """
1775 me.setstatus('Connected (port %s)' % conn.port())
060ca767 1776 me.ui.get_widget('/menubar/server-menu/connect').set_sensitive(False)
1777 for i in ('/menubar/server-menu/disconnect',
1778 '/menubar/server-menu/server-version',
1779 '/menubar/server-menu/add-peer',
1780 '/menubar/server-menu/server-quit',
1781 '/menubar/logs-menu/trace-options'):
1782 me.ui.get_widget(i).set_sensitive(True)
984b6d33
MW
1783 me.ui.get_widget('/menubar/server-menu/conn-peer'). \
1784 set_sensitive(bool(monitor.autopeers))
060ca767 1785 me.ui.get_widget('/menubar/server-menu/daemon'). \
984b6d33
MW
1786 set_sensitive(conn.servinfo()['daemon'] == 'nil')
1787
1788 def disconnected(me, reason):
1789 """
1790 Hook: invoked when the connection to the server is lost.
1791
1792 Make most options insensitive.
1793 """
060ca767 1794 me.setstatus('Disconnected')
1795 me.ui.get_widget('/menubar/server-menu/connect').set_sensitive(True)
1796 for i in ('/menubar/server-menu/disconnect',
1797 '/menubar/server-menu/server-version',
1798 '/menubar/server-menu/add-peer',
984b6d33 1799 '/menubar/server-menu/conn-peer',
060ca767 1800 '/menubar/server-menu/daemon',
1801 '/menubar/server-menu/server-quit',
1802 '/menubar/logs-menu/trace-options'):
1803 me.ui.get_widget(i).set_sensitive(False)
984b6d33
MW
1804 if reason: moanbox(reason)
1805
1806###--------------------------------------------------------------------------
1807### Main program.
1808
1809def parse_options():
1810 """
1811 Parse command-line options.
1812
1813 Process the boring ones. Return all of them, for later.
1814 """
1815 op = OptionParser(usage = '%prog [-a FILE] [-d DIR]',
1816 version = '%prog (tripe version 1.0.0)')
1817 op.add_option('-a', '--admin-socket',
1818 metavar = 'FILE', dest = 'tripesock', default = T.tripesock,
1819 help = 'Select socket to connect to [default %default]')
1820 op.add_option('-d', '--directory',
1821 metavar = 'DIR', dest = 'dir', default = T.configdir,
1822 help = 'Select current diretory [default %default]')
1823 opts, args = op.parse_args()
1824 if args: op.error('no arguments permitted')
1825 OS.chdir(opts.dir)
1826 return opts
1827
1828def init(opts):
1829 """Initialization."""
1830
1831 global conn, monitor, pinger
1832
984b6d33
MW
1833 ## Try to establish a connection.
1834 conn = Connection(opts.tripesock)
1835
1836 ## Make the main interesting coroutines and objects.
1837 monitor = Monitor()
1838 pinger = Pinger()
1839 pinger.switch()
060ca767 1840
984b6d33 1841def main():
060ca767 1842
984b6d33
MW
1843 ## Main window.
1844 root = MonitorWindow()
1845 conn.connect()
1846 root.show_all()
060ca767 1847
984b6d33 1848 ## Main loop.
060ca767 1849 HookClient().hook(root.closehook, exit)
690a6ec1 1850 conn.mainloop()
060ca767 1851
1852if __name__ == '__main__':
984b6d33
MW
1853 opts = parse_options()
1854 init(opts)
060ca767 1855 main()
1856
984b6d33 1857###----- That's all, folks --------------------------------------------------