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,8 +32,15 @@ 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 ...")
+            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:
@@ -42,9 +50,10 @@ class PeerDB(object):
                 logging.warning(e)
                 time.sleep(retry)
                 retry = min(60, retry * 2)
-            q("INSERT INTO config VALUES ('registry',?)", (a,))
+        self._db.execute("INSERT OR REPLACE INTO config VALUES ('registry',?)",
+                         (a,))
         self.registry_ip = utils.binFromIp(a)
-        logging.info("Cache initialized. Registry IP is %s", a)
+        return a
 
     def log(self):
         if logging.getLogger().isEnabledFor(5):

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,