netdev-dpdk: fix mbuf leaks
[cascardo/ovs.git] / xenserver / etc_xapi.d_plugins_openvswitch-cfg-update
index bceccbf..e7404e3 100755 (executable)
@@ -1,10 +1,10 @@
 #!/usr/bin/env python
 #
 # xapi plugin script to update the cache of configuration items in the
-# ovs-vswitchd configuration that are managed in the xapi database when 
+# ovs-vswitchd configuration that are managed in the xapi database when
 # integrated with Citrix management tools.
 
-# Copyright (C) 2009, 2010, 2011 Nicira Networks, Inc.
+# Copyright (C) 2009, 2010, 2011, 2012, 2013 Nicira, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # TBD:   the system in a bad state if anything goes wrong.
 
 import XenAPIPlugin
-import XenAPI
 import os
 import subprocess
 import syslog
 import re
 
-vsctl="/usr/bin/ovs-vsctl"
-ofctl="/usr/bin/ovs-ofctl"
-cacert_filename="/etc/openvswitch/vswitchd.cacert"
+vsctl = '/usr/bin/ovs-vsctl'
+ofctl = '/usr/bin/ovs-ofctl'
+cacert_filename = '/etc/openvswitch/vswitchd.cacert'
+ovsdb_port = '6640'
+
 
 # Delete the CA certificate, so that we go back to boot-strapping mode
 def delete_cacert():
@@ -40,36 +41,39 @@ def delete_cacert():
         # Ignore error if file doesn't exist
         pass
 
+
 def update(session, args):
     # Refresh bridge network UUIDs in case this host joined or left a pool.
-    script = "/opt/xensource/libexec/interface-reconfigure"
+    script = '/opt/xensource/libexec/interface-reconfigure'
     try:
-        retval = subprocess.call([script, "rewrite"])
+        retval = subprocess.call([script, 'rewrite'])
         if retval != 0:
-            syslog.syslog("%s exited with status %d" % (script, retval))
+            syslog.syslog('%s exited with status %d' % (script, retval))
     except OSError, e:
-        syslog.syslog("%s: failed to execute (%s)" % (script, e.strerror))
+        syslog.syslog('%s: failed to execute (%s)' % (script, e.strerror))
 
     pools = session.xenapi.pool.get_all()
     # We assume there is only ever one pool...
     if len(pools) == 0:
-        raise XenAPIPlugin.Failure("NO_POOL_FOR_HOST", [])
+        raise XenAPIPlugin.Failure('NO_POOL_FOR_HOST', [])
     if len(pools) > 1:
-        raise XenAPIPlugin.Failure("MORE_THAN_ONE_POOL_FOR_HOST", [])
+        raise XenAPIPlugin.Failure('MORE_THAN_ONE_POOL_FOR_HOST', [])
     new_controller = False
     pool = session.xenapi.pool.get_record(pools[0])
-    controller = pool.get("vswitch_controller", "")
-    ret_str = ""
-    currentController = vswitchCurrentController()
-    if controller == "" and currentController != "":
+    controller = pool.get('vswitch_controller')
+    ret_str = ''
+    currentControllers = vswitchCurrentControllers()
+
+    if not controller and currentControllers:
         delete_cacert()
         try:
             emergency_reset(session, None)
         except:
             pass
         removeControllerCfg()
-        ret_str += "Successfully removed controller config.  "
-    elif controller != currentController:
+        ret_str += 'Successfully removed controller config.  '
+    # controller cannot be empty, otherwise, this will always be True.
+    elif controller and controller not in currentControllers:
         delete_cacert()
         try:
             emergency_reset(session, None)
@@ -77,33 +81,32 @@ def update(session, args):
             pass
         setControllerCfg(controller)
         new_controller = True
-        ret_str += "Successfully set controller to %s.  " % controller
+        ret_str += 'Successfully set controller to %s.  ' % controller
 
     try:
-        pool_fail_mode = pool["other_config"]["vswitch-controller-fail-mode"]
+        pool_fail_mode = pool['other_config']['vswitch-controller-fail-mode']
     except KeyError, e:
         pool_fail_mode = None
 
     bton = {}
 
-    for n in session.xenapi.network.get_all():
-        rec = session.xenapi.network.get_record(n)
+    for rec in session.xenapi.network.get_all_records().values():
         try:
             bton[rec['bridge']] = rec
         except KeyError:
             pass
 
-    # If new controller, get managagment MAC addresses from XAPI now
+    # If new controller, get management MAC addresses from XAPI now
     # in case fail_mode set to secure which may affect XAPI access
     mgmt_bridge = None
     host_mgmt_mac = None
     host_mgmt_device = None
     pool_mgmt_macs = {}
     if new_controller:
-        for n in session.xenapi.PIF.get_all():
-            rec = session.xenapi.PIF.get_record(n)
-            if rec.get('management', False):
-                pool_mgmt_macs[rec.get('MAC')] = rec.get('device')
+        query = 'field "management"="true"'
+        recs = session.xenapi.PIF.get_all_records_where(query)
+        for rec in recs.itervalues():
+            pool_mgmt_macs[rec.get('MAC')] = rec.get('device')
 
     dib_changed = False
     fail_mode_changed = False
@@ -138,11 +141,12 @@ def update(session, args):
             dib_changed = True
 
         # Change bridge fail_mode if XAPI state differs from OVS state.
-        bridge_fail_mode = vswitchCfgQuery(["get", "Bridge",
-            bridge, "fail_mode"]).strip('[]"')
+        bridge_fail_mode = vswitchCfgQuery(['get', 'Bridge',
+                                            bridge, 'fail_mode']).strip('[]"')
 
         try:
-            fail_mode = bton[bridge]["other_config"]["vswitch-controller-fail-mode"]
+            other_config = bton[bridge]['other_config']
+            fail_mode = other_config['vswitch-controller-fail-mode']
         except KeyError, e:
             fail_mode = None
 
@@ -154,98 +158,112 @@ def update(session, args):
 
         if bridge_fail_mode != fail_mode:
             vswitchCfgMod(['--', 'set', 'Bridge', bridge,
-                "fail_mode=%s" % fail_mode])
+                           'fail_mode=%s' % fail_mode])
             fail_mode_changed = True
 
         # Determine local mgmt MAC address if host being added to secure
         # pool so we can add default flows to allow management traffic
-        if new_controller and fail_mode_changed and pool_fail_mode == "secure":
-            oc = vswitchCfgQuery(["get", "Bridge", bridge, "other-config"])
+        if new_controller and fail_mode_changed and pool_fail_mode == 'secure':
+            oc = vswitchCfgQuery(['get', 'Bridge', bridge, 'other-config'])
             m = re.match('.*hwaddr="([0-9a-fA-F:].*)".*', oc)
             if m and m.group(1) in pool_mgmt_macs.keys():
                 mgmt_bridge = bridge
                 host_mgmt_mac = m.group(1)
                 host_mgmt_device = pool_mgmt_macs[host_mgmt_mac]
 
-    if host_mgmt_mac is not None and mgmt_bridge is not None and \
-        host_mgmt_device is not None:
-        tp = "idle_timeout=0,priority=0"
-        port = vswitchCfgQuery(["get", "interface", host_mgmt_device, "ofport"])
-        addFlow(mgmt_bridge, "%s,in_port=%s,arp,nw_proto=1,actions=local" % \
-            (tp, port))
-        addFlow(mgmt_bridge, "%s,in_port=local,arp,dl_src=%s,actions=%s" % \
-            (tp, host_mgmt_mac, port))
-        addFlow(mgmt_bridge, "%s,in_port=%s,dl_dst=%s,actions=local" % \
-            (tp, port, host_mgmt_mac))
-        addFlow(mgmt_bridge, "%s,in_port=local,dl_src=%s,actions=%s" % \
-            (tp, host_mgmt_mac, port))
+    if (host_mgmt_mac is not None and mgmt_bridge is not None and
+            host_mgmt_device is not None):
+        tp = 'idle_timeout=0,priority=0'
+        port = vswitchCfgQuery(['get', 'interface', host_mgmt_device,
+                                'ofport'])
+
+        addFlow(mgmt_bridge, '%s,in_port=%s,arp,nw_proto=1,actions=local' %
+                (tp, port))
+        addFlow(mgmt_bridge, '%s,in_port=local,arp,dl_src=%s,actions=%s' %
+                (tp, host_mgmt_mac, port))
+        addFlow(mgmt_bridge, '%s,in_port=%s,dl_dst=%s,actions=local' %
+                (tp, port, host_mgmt_mac))
+        addFlow(mgmt_bridge, '%s,in_port=local,dl_src=%s,actions=%s' %
+                (tp, host_mgmt_mac, port))
 
     if dib_changed:
-        ret_str += "Updated in-band management.  "
+        ret_str += 'Updated in-band management.  '
     if fail_mode_changed:
-        ret_str += "Updated fail_mode.  "
+        ret_str += 'Updated fail_mode.  '
 
     if ret_str != '':
         return ret_str
     else:
-        return "No change to configuration"
-
-def vswitchCurrentController():
-    controller = vswitchCfgQuery(["get-manager"])
-    if controller == "":
-        return controller
-    if len(controller) < 4 or controller[0:4] != "ssl:":
-        return controller
-    else:
-        return controller.split(':')[1]
+        return 'No change to configuration'
+
+
+def vswitchCurrentControllers():
+    controllers = vswitchCfgQuery(['get-manager'])
+
+    def parse_controller(controller):
+        if controller.startswith('ssl:'):
+            return controller.split(':')[1]
+
+        return controller.split(':')[0]
+
+    return [parse_controller(controller)
+            for controller in controllers.split('\n')
+            if controller]
+
 
 def removeControllerCfg():
-    vswitchCfgMod(["--", "del-manager",
-                   "--", "del-ssl"])
+    vswitchCfgMod(['--', 'del-manager',
+                   '--', 'del-ssl'])
+
 
 def setControllerCfg(controller):
     # /etc/xensource/xapi-ssl.pem is mentioned twice below because it
     # contains both the private key and the certificate.
-    vswitchCfgMod(["--", "del-manager",
-                   "--", "del-ssl",
-                   "--", "--bootstrap", "set-ssl",
-                   "/etc/xensource/xapi-ssl.pem",
-                   "/etc/xensource/xapi-ssl.pem",
+    vswitchCfgMod(['--', 'del-manager',
+                   '--', 'del-ssl',
+                   '--', '--bootstrap', 'set-ssl',
+                   '/etc/xensource/xapi-ssl.pem',
+                   '/etc/xensource/xapi-ssl.pem',
                    cacert_filename,
-                   "--", "set-manager", 'ssl:' + controller + ':6632'])
+                   '--', 'set-manager', 'ssl:' + controller + ':' + ovsdb_port])
+
 
 def vswitchCfgQuery(action_args):
-    cmd = [vsctl, "--timeout=5", "-vANY:console:emer"] + action_args
+    cmd = [vsctl, '-vconsole:off'] + action_args
     output = subprocess.Popen(cmd, stdout=subprocess.PIPE).communicate()
-    if len(output) == 0 or output[0] == None:
-        output = ""
+    if len(output) == 0 or output[0] is None:
+        output = ''
     else:
         output = output[0].strip()
     return output
 
+
 def vswitchCfgMod(action_args):
-    cmd = [vsctl, "--timeout=5", "-vANY:console:emer"] + action_args
+    cmd = [vsctl, '--timeout=5', '-vconsole:off'] + action_args
     exitcode = subprocess.call(cmd)
     if exitcode != 0:
-        raise XenAPIPlugin.Failure("VSWITCH_CONFIG_MOD_FAILURE",
-                                   [ str(exitcode) , str(action_args) ])
+        raise XenAPIPlugin.Failure('VSWITCH_CONFIG_MOD_FAILURE',
+                                   [str(exitcode), str(action_args)])
+
 
 def emergency_reset(session, args):
-    cmd = [vsctl, "--timeout=5", "emer-reset"]
+    cmd = [vsctl, '--timeout=5', 'emer-reset']
     exitcode = subprocess.call(cmd)
     if exitcode != 0:
-        raise XenAPIPlugin.Failure("VSWITCH_EMER_RESET_FAILURE",
-                                   [ str(exitcode) ])
+        raise XenAPIPlugin.Failure('VSWITCH_EMER_RESET_FAILURE',
+                                   [str(exitcode)])
+
+    return 'Successfully reset configuration'
 
-    return "Successfully reset configuration"
 
 def addFlow(switch, flow):
-    cmd = [ofctl, "add-flow", switch, flow]
+    cmd = [ofctl, 'add-flow', switch, flow]
     exitcode = subprocess.call(cmd)
     if exitcode != 0:
-        raise XenAPIPlugin.Failure("VSWITCH_ADD_FLOW_FAILURE",
-                                   [ str(exitcode) , str(switch), str(flow) ])
-    
-if __name__ == "__main__":
-    XenAPIPlugin.dispatch({"update": update,
-                           "emergency_reset": emergency_reset})
+        raise XenAPIPlugin.Failure('VSWITCH_ADD_FLOW_FAILURE',
+                                   [str(exitcode), str(switch), str(flow)])
+
+
+if __name__ == '__main__':
+    XenAPIPlugin.dispatch({'update': update,
+                           'emergency_reset': emergency_reset})