X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~mdw/git/tripe/blobdiff_plain/29807d891d99bb0eeacce0677cf68d1d4d118a30..b10a8c3d8ffee143a0da007377fe641c24761782:/svc/conntrack.in

diff --git a/svc/conntrack.in b/svc/conntrack.in
index 18e2f730..b80f92ed 100644
--- a/svc/conntrack.in
+++ b/svc/conntrack.in
@@ -214,6 +214,7 @@ def localaddr(peer):
 
 _kick = T.Queue()
 def kickpeers():
+  lastip = {}
   while True:
     upness, reason = _kick.get()
 
@@ -233,35 +234,44 @@ def kickpeers():
       addr = localaddr(CF.testaddr)
       if addr is None:
         upness = False
+    else:
+      addr = None
 
     ## Now decide what to do.
     changes = []
     for g, pp in CF.groups:
 
       ## Find out which peer in the group ought to be active.
-      want = None                       # unequal to any string
-      if upness:
-        for t, p, a, m in pp:
-          if p is None:
-            aq = addr
-          else:
-            aq = localaddr(p)
-          if aq is not None and (aq & m) == a:
-            want = t
-            break
+      ip = None
+      map = {}
+      want = None
+      for t, p, a, m in pp:
+        if p is None or not upness:
+          ipq = addr
+        else:
+          ipq = localaddr(p)
+        if upness and ip is None and \
+              ipq is not None and (ipq & m) == a:
+          map[t] = 'up'
+          want = t
+          ip = ipq
+        else:
+          map[t] = 'down'
 
       ## Shut down the wrong ones.
       found = False
       for p in peers:
-        if p == want:
+        what = map.get(p, 'leave')
+        if what == 'up':
           found = True
-        elif p.startswith(g) and p != want:
+        elif what == 'down':
           changes.append(lambda p=p: SM.kill(p))
 
       ## Start the right one if necessary.
-      if want is not None and not found:
+      if want is not None and (not found or ip != lastip.get(g, None)):
         changes.append(lambda: T._simple(SM.svcsubmit('connect', 'active',
                                                       want)))
+      lastip[g] = ip
 
     ## Commit the changes.
     if changes:
@@ -388,6 +398,7 @@ class DBusMonitor (object):
     """
     me._mons = []
     me._loop = D.mainloop.glib.DBusGMainLoop()
+    me._state = 'startup'
     me._reconnect()
 
   def addmon(me, mon):
@@ -402,12 +413,20 @@ class DBusMonitor (object):
     if me._bus is not None:
       mon.attach(me._bus)
 
-  def _reconnect(me):
+  def _reconnect(me, hunoz = None):
     """
     Start connecting to the bus.
 
     If we fail the first time, retry periodically.
     """
+    if me._state == 'startup':
+      T.aside(SM.notify, 'conntrack', 'dbus-connection', 'startup')
+    elif me._state == 'connected':
+      T.aside(SM.notify, 'conntrack', 'dbus-connection', 'lost')
+    else:
+      T.aside(SM.notify, 'conntrack', 'dbus-connection',
+              'state=%s' % me._state)
+    me._state == 'reconnecting'
     me._bus = None
     if me._try_connect():
       G.timeout_add_seconds(5, me._try_connect)
@@ -419,13 +438,21 @@ class DBusMonitor (object):
     If we succeed, attach the monitors.
     """
     try:
-      bus = D.SystemBus(mainloop = me._loop, private = True)
-    except D.DBusException:
+      addr = OS.getenv('TRIPE_CONNTRACK_BUS')
+      if addr == 'SESSION':
+        bus = D.SessionBus(mainloop = me._loop, private = True)
+      elif addr is not None:
+        bus = D.bus.BusConnection(addr, mainloop = me._loop)
+      else:
+        bus = D.SystemBus(mainloop = me._loop, private = True)
+      for m in me._mons:
+        m.attach(bus)
+    except D.DBusException, e:
       return True
     me._bus = bus
+    me._state = 'connected'
     bus.call_on_disconnection(me._reconnect)
-    for m in me._mons:
-      m.attach(bus)
+    T.aside(SM.notify, 'conntrack', 'dbus-connection', 'connected')
     return False
 
 ###--------------------------------------------------------------------------