2
0
mirror of https://github.com/openvswitch/ovs synced 2025-10-15 14:17:18 +00:00
Files
openvswitch/xenserver/usr_lib_xsconsole_plugins-base_XSFeatureVSwitch.py
Raju Subramanian e0edde6fee Global replace of Nicira Networks.
Replaced all instances of Nicira Networks(, Inc) to Nicira, Inc.

Feature #10593
Signed-off-by: Raju Subramanian <rsubramanian@nicira.com>
Signed-off-by: Ben Pfaff <blp@nicira.com>
2012-05-02 17:08:02 -07:00

332 lines
12 KiB
Python

# Copyright (c) 2007-2011 Citrix Systems Inc.
# Copyright (c) 2009,2010,2011 Nicira, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 only.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
from XSConsoleLog import *
import os
import socket
import subprocess
vsctl="/usr/bin/ovs-vsctl"
if __name__ == "__main__":
raise Exception("This script is a plugin for xsconsole and cannot run independently")
from XSConsoleStandard import *
class VSwitchService:
service = {}
def __init__(self, name, processname=None):
self.name = name
self.processname = processname
if self.processname == None:
self.processname = name
def version(self):
try:
output = ShellPipe(["service", self.name, "version"]).Stdout()
except StandardError, e:
XSLogError("vswitch version retrieval error: " + str(e))
return "<unknown>"
for line in output:
if self.processname in line:
return line.split()[-1]
return "<unknown>"
def status(self):
try:
output = ShellPipe(["service", self.name, "status"]).Stdout()
except StandardError, e:
XSLogError("vswitch status retrieval error: " + str(e))
return "<unknown>"
if len(output) == 0:
return "<unknown>"
for line in output:
if self.processname not in line:
continue
elif "running" in line:
return "Running"
elif "stop" in line:
return "Stopped"
else:
return "<unknown>"
return "<unknown>"
def restart(self):
try:
ShellPipe(["service", self.name, "restart"]).Call()
except StandardError, e:
XSLogError("vswitch restart error: " + str(e))
@classmethod
def Inst(cls, name, processname=None):
key = name
if processname != None:
key = key + "-" + processname
if name not in cls.service:
cls.service[key] = VSwitchService(name, processname)
return cls.service[key]
class VSwitchConfig:
@staticmethod
def Get(action):
try:
arg = [vsctl, "--timeout=30", "-vANY:console:off"] + action.split()
output = ShellPipe(arg).Stdout()
except StandardError, e:
XSLogError("config retrieval error: " + str(e))
return "<unknown>"
if len(output) == 0:
output = ""
else:
output = output[0].strip()
return output
class VSwitchControllerDialogue(Dialogue):
def __init__(self):
Dialogue.__init__(self)
data=Data.Inst()
self.hostsInPool = 0
self.hostsUpdated = 0
self.xs_version = data.host.software_version.product_version('')
pool = data.GetPoolForThisHost()
if pool is not None:
self.controller = pool.get("vswitch_controller", "")
else:
self.controller = ""
choiceDefs = [
ChoiceDef(Lang("Set pool-wide controller"),
lambda: self.getController()),
ChoiceDef(Lang("Delete pool-wide controller"),
lambda: self.deleteController()),
ChoiceDef(Lang("Resync server controller config"),
lambda: self.syncController()),
# ChoiceDef(Lang("Restart ovs-vswitchd"),
# lambda: self.restartService("vswitch")),
]
self.menu = Menu(self, None, Lang("Configure Open vSwitch"), choiceDefs)
self.ChangeState("INITIAL")
def BuildPane(self):
pane = self.NewPane(DialoguePane(self.parent))
pane.TitleSet(Lang("Configure Open vSwitch"))
pane.AddBox()
def ChangeState(self, inState):
self.state = inState
self.BuildPane()
self.UpdateFields()
def UpdateFields(self):
self.Pane().ResetPosition()
getattr(self, "UpdateFields" + self.state)() # Dispatch method named 'UpdateFields'+self.state
def UpdateFieldsINITIAL(self):
pane = self.Pane()
pane.AddTitleField(Lang("Select an action"))
pane.AddMenuField(self.menu)
pane.AddKeyHelpField( { Lang("<Enter>") : Lang("OK"), Lang("<Esc>") : Lang("Cancel") } )
def UpdateFieldsGETCONTROLLER(self):
pane = self.Pane()
pane.ResetFields()
pane.AddTitleField(Lang("Enter IP address of controller"))
pane.AddInputField(Lang("Address", 16), self.controller, "address")
pane.AddKeyHelpField( { Lang("<Enter>") : Lang("OK"), Lang("<Esc>") : Lang("Exit") } )
if pane.CurrentInput() is None:
pane.InputIndexSet(0)
def HandleKey(self, inKey):
handled = False
if hasattr(self, "HandleKey" + self.state):
handled = getattr(self, "HandleKey" + self.state)(inKey)
if not handled and inKey == 'KEY_ESCAPE':
Layout.Inst().PopDialogue()
handled = True
return handled
def HandleKeyINITIAL(self, inKey):
return self.menu.HandleKey(inKey)
def HandleKeyGETCONTROLLER(self, inKey):
pane = self.Pane()
if pane.CurrentInput() is None:
pane.InputIndexSet(0)
if inKey == 'KEY_ENTER':
inputValues = pane.GetFieldValues()
self.controller = inputValues['address']
Layout.Inst().PopDialogue()
# Make sure the controller is specified as a valid dotted quad
try:
socket.inet_aton(self.controller)
except socket.error:
Layout.Inst().PushDialogue(InfoDialogue(Lang("Please enter in dotted quad format")))
return True
Layout.Inst().TransientBanner(Lang("Setting controller..."))
try:
self.SetController(self.controller)
Layout.Inst().PushDialogue(InfoDialogue(Lang("Setting controller successful")))
except Exception, e:
Layout.Inst().PushDialogue(InfoDialogue(Lang("Setting controller failed")))
self.ChangeState("INITIAL")
return True
else:
return pane.CurrentInput().HandleKey(inKey)
def restartService(self, name):
s = VSwitchService.Inst(name)
s.restart()
Layout.Inst().PopDialogue()
def getController(self):
self.ChangeState("GETCONTROLLER")
self.Pane().InputIndexSet(0)
def deleteController(self):
self.controller = ""
Layout.Inst().PopDialogue()
Layout.Inst().TransientBanner(Lang("Deleting controller..."))
try:
self.SetController(None)
Layout.Inst().PushDialogue(InfoDialogue(Lang("Controller deletion successful")))
except Exception, e:
Layout.Inst().PushDialogue(InfoDialogue(Lang("Controller deletion failed")))
def syncController(self):
Layout.Inst().PopDialogue()
Layout.Inst().TransientBanner(Lang("Resyncing controller setting..."))
try:
Task.Sync(lambda s: self._updateThisServer(s))
Layout.Inst().PushDialogue(InfoDialogue(Lang("Resyncing controller config successful")))
except Exception, e:
Layout.Inst().PushDialogue(InfoDialogue(Lang("Resyncing controller config failed")))
def SetController(self, ip):
self.hostsInPool = 0
self.hostsUpdated = 0
Task.Sync(lambda s: self._modifyPoolConfig(s, ip or ""))
# Should be done asynchronously, maybe with an external script?
Task.Sync(lambda s: self._updateActiveServers(s))
def _modifyPoolConfig(self, session, value):
"""Modify pool configuration.
If value == "" then delete configuration, otherwise set to value.
"""
pools = session.xenapi.pool.get_all()
# We assume there is only ever one pool...
if len(pools) == 0:
XSLogFatal(Lang("No pool found for host."))
return
if len(pools) > 1:
XSLogFatal(Lang("More than one pool for host."))
return
session.xenapi.pool.set_vswitch_controller(value)
Data.Inst().Update()
def _updateActiveServers(self, session):
hosts = session.xenapi.host.get_all()
self.hostsUpdated = 0
self.hostsInPool = len(hosts)
self.UpdateFields()
for host in hosts:
Layout.Inst().TransientBanner("Updating host %d out of %d"
% (self.hostsUpdated + 1, self.hostsInPool))
session.xenapi.host.call_plugin(host, "openvswitch-cfg-update", "update", {})
self.hostsUpdated = self.hostsUpdated + 1
def _updateThisServer(self, session):
data = Data.Inst()
host = data.host.opaqueref()
session.xenapi.host.call_plugin(host, "openvswitch-cfg-update", "update", {})
class XSFeatureVSwitch:
@classmethod
def StatusUpdateHandler(cls, inPane):
data = Data.Inst()
xs_version = data.host.software_version.product_version('')
inPane.AddTitleField(Lang("Open vSwitch"))
inPane.NewLine()
inPane.AddStatusField(Lang("Version", 20),
VSwitchService.Inst("openvswitch", "ovs-vswitchd").version())
inPane.NewLine()
pool = data.GetPoolForThisHost()
if pool is not None:
dbController = pool.get("vswitch_controller", "")
else:
dbController = ""
if dbController == "":
dbController = Lang("<None>")
inPane.AddStatusField(Lang("Controller (config)", 20), dbController)
controller = VSwitchConfig.Get("get-manager")
if controller == "":
controller = Lang("<None>")
elif controller[0:4] == "ssl:":
controller = controller.split(':')[1]
inPane.AddStatusField(Lang("Controller (in-use)", 20), controller)
inPane.NewLine()
inPane.AddStatusField(Lang("ovs-vswitchd status", 20),
VSwitchService.Inst("openvswitch", "ovs-vswitchd").status())
inPane.AddStatusField(Lang("ovsdb-server status", 20),
VSwitchService.Inst("openvswitch", "ovsdb-server").status())
inPane.AddKeyHelpField( {
Lang("<Enter>") : Lang("Reconfigure"),
Lang("<F5>") : Lang("Refresh")
})
@classmethod
def ActivateHandler(cls):
DialogueUtils.AuthenticatedOnly(lambda: Layout.Inst().PushDialogue(VSwitchControllerDialogue()))
def Register(self):
Importer.RegisterNamedPlugIn(
self,
'VSwitch', # Key of this plugin for replacement, etc.
{
'menuname' : 'MENU_NETWORK',
'menupriority' : 800,
'menutext' : Lang('Open vSwitch') ,
'statusupdatehandler' : self.StatusUpdateHandler,
'activatehandler' : self.ActivateHandler
}
)
# Register this plugin when module is imported, IFF vswitchd is running
if os.path.exists('/var/run/openvswitch/ovs-vswitchd.pid'):
XSFeatureVSwitch().Register()