From: David Howells <dhowells@redhat.com>
Date: Wed, 9 Apr 2014 14:40:36 +0000 (+0100)
Subject: Move VL connection security determination into cell.py
X-Git-Url: https://www.infradead.org/git/?a=commitdiff_plain;h=2de121b9bbf9fd5ad9a7e2281ca8052632e3bf7f;p=users%2Fdhowells%2Fkafs-utils.git

Move VL connection security determination into cell.py

Move the Volume Location DB connection security handling into
cell.open_vl_server() as it's common to all vos commands.  It can be
configured directly by passing in a dictionary holding the appropriate
parameters.

Signed-off-by: David Howells <dhowells@redhat.com>
---

diff --git a/suite/commands/vos/listvldb.py b/suite/commands/vos/listvldb.py
index a9fd153..b3adfca 100644
--- a/suite/commands/vos/listvldb.py
+++ b/suite/commands/vos/listvldb.py
@@ -108,17 +108,7 @@ def print_record(params, vldb):
 def main(params):
     # Get a list of VLDB servers to query
     cell = params["cell"]
-
-    if "localauth" in params:
-        raise RuntimeError("Don't support -localauth yet")
-    elif "noauth" in params:
-        security = None
-    elif "encrypt" in params:
-        security = kafs.RXRPC_SECURITY_ENCRYPT
-    else:
-        security = None
-
-    z_conn = cell.open_vl_server(security)
+    z_conn = cell.open_vl_server(params)
     quiet = "quiet" in params
 
     if "name" in params:
diff --git a/suite/lib/cell.py b/suite/lib/cell.py
index a4938ee..d7bf79f 100644
--- a/suite/lib/cell.py
+++ b/suite/lib/cell.py
@@ -119,19 +119,27 @@ class cell:
         return addrs
 
     # Open a VL Server connection
-    def open_vl_server(self, security=None):
+    def open_vl_server(self, params=None):
         if self.__vlconn:
             return
 
+        key = None
+        security = None
+        if params:
+            key = "afs@" + self.__name.upper()
+            if "localauth" in params:
+                raise RuntimeError("Don't support -localauth yet")
+            elif "noauth" in params:
+                security = 0
+                key = None
+            elif "encrypt" in params:
+                security = kafs.RXRPC_SECURITY_ENCRYPT
+            else:
+                security = kafs.RXRPC_SECURITY_PLAIN
+
         for vlserver in self.query_vl_addrs():
             debug("Trying", vlserver)
 
-            if security != None:
-                key = "afs@" + self.__name.upper()
-            else:
-                key = None
-                security = 0
-
             z_conn = kafs.rx_new_connection(vlserver, kafs.VL_PORT, kafs.VL_SERVICE,
                                             key, security)
             try: