wip

re6st/db.py

@@ -5,7 +5,8 @@ from . import utils
 class PeerDB(object):
 
     # internal ip = temp arg/attribute
-    def __init__(self, db_path, registry, key_path, prefix, db_size=200):
+    def __init__(self, db_path, registry, key_path, network, prefix,
+                       db_size=200):
         self._prefix = prefix
         self._db_size = db_size
         self._key_path = key_path
@@ -31,21 +32,29 @@ class PeerDB(object):
         try:
             a = q("SELECT value FROM config WHERE name='registry'").next()[0]
         except StopIteration:
-            logging.info("Private IP of registry not in cache."
-                         " Asking registry via its public IP ...")
-            retry = 1
-            while True:
-                try:
-                    a = self._registry.getPrivateAddress(self._prefix)
-                    break
-                except socket.error, e:
-                    logging.warning(e)
-                    time.sleep(retry)
-                    retry = min(60, retry * 2)
-            q("INSERT INTO config VALUES ('registry',?)", (a,))
-        self.registry_ip = utils.binFromIp(a)
+            a = self._updateRegistryIP()
+        else:
+            self.registry_ip = utils.binFromIp(a)
+            if not self.registry_ip.starswith(network):
+                a = self._updateRegistryIP()
         logging.info("Cache initialized. Registry IP is %s", a)
 
+    def _updateRegistryIP(self):
+        logging.info("Asking registry its private IP...")
+        retry = 1
+        while True:
+            try:
+                a = self._registry.getPrivateAddress(self._prefix)
+                break
+            except socket.error, e:
+                logging.warning(e)
+                time.sleep(retry)
+                retry = min(60, retry * 2)
+        self._db.execute("INSERT OR REPLACE INTO config VALUES ('registry',?)",
+                         (a,))
+        self.registry_ip = utils.binFromIp(a)
+        return a
+
     def log(self):
         if logging.getLogger().isEnabledFor(5):
             logging.trace("Cache:")

re6stnet

@@ -268,7 +268,7 @@ def main():
             # Create and open read_only pipe to get server events
             r_pipe, write_pipe = os.pipe()
             read_pipe = os.fdopen(r_pipe)
-            peer_db = db.PeerDB(db_path, registry, config.key, prefix)
+            peer_db = db.PeerDB(db_path, registry, config.key, network, prefix)
             tunnel_manager = tunnel.TunnelManager(write_pipe, peer_db,
                 config.openvpn_args, timeout, config.tunnel_refresh,
                 config.client_count, config.iface_list, network, prefix,