Adding upstream version 4.12.1.upstream/4.12.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 197 insertions, 0 deletions

diff --git a/agents/lpar/fence_lpar.py b/agents/lpar/fence_lpar.py
new file mode 100644
index 0000000..975971a
--- /dev/null
+++ b/agents/lpar/fence_lpar.py
@@ -0,0 +1,197 @@
+#!@PYTHON@ -tt
+
+#####
+##
+## The Following Agent Has Been Tested On:
+##
+##  Version
+## +---------------------------------------------+
+##  Tested on HMC
+##
+#####
+
+import sys, re
+import atexit
+import logging
+sys.path.append("@FENCEAGENTSLIBDIR@")
+from fencing import *
+from fencing import fail, fail_usage, EC_STATUS_HMC
+
+##
+## Transformation to standard ON/OFF status if possible
+def _normalize_status(status):
+	if status in ["Running", "Open Firmware", "Shutting Down", "Starting"]:
+		status = "on"
+	else:
+		status = "off"
+
+	return status
+
+def get_power_status(conn, options):
+	if options["--hmc-version"] == "3":
+		command = "lssyscfg -r lpar -m " + options["--managed"] + " -n " + options["--plug"] + " -F name,state\n"
+	elif options["--hmc-version"] in ["4", "IVM"]:
+		command = "lssyscfg -r lpar -m "+ options["--managed"] + \
+			" --filter 'lpar_names=" + options["--plug"] + "'\n"
+	else:
+		# Bad HMC Version cannot be reached
+		fail(EC_STATUS_HMC)
+
+	conn.send(command)
+	# First line (command) may cause parsing issues if long
+	conn.readline()
+	conn.log_expect(options["--command-prompt"], int(options["--power-timeout"]))
+
+	try:
+		if options["--hmc-version"] == "3":
+			status = re.compile("^" + options["--plug"] + ",(.*?),.*$",
+					    re.IGNORECASE | re.MULTILINE).search(conn.before).group(1)
+		elif options["--hmc-version"] in ["4", "IVM"]:
+			status = re.compile(",state=(.*?),", re.IGNORECASE).search(conn.before).group(1)
+	except AttributeError as e:
+		logging.debug("Command on HMC failed: {}\n{}".format(command, str(e)))
+		fail(EC_STATUS_HMC)
+
+	return _normalize_status(status)
+
+def is_comanaged(conn, options):
+	conn.send("lscomgmt -m " + options["--managed"] + "\n" )
+	conn.readline()
+	conn.log_expect(options["--command-prompt"], int(options["--power-timeout"]))
+
+	try:
+		cm = re.compile(",curr_master_mtms=(.*?),", re.IGNORECASE).search(conn.before).group(1)
+	except AttributeError as e:
+		cm = False
+
+	return cm
+
+def set_power_status(conn, options):
+	if options["--hmc-version"] == "3":
+		conn.send("chsysstate -o " + options["--action"] + " -r lpar -m " + options["--managed"]
+			+ " -n " + options["--plug"] + "\n")
+
+		# First line (command) may cause parsing issues if long
+		conn.readline()
+		conn.log_expect(options["--command-prompt"], int(options["--power-timeout"]))
+	elif options["--hmc-version"] in ["4", "IVM"]:
+		if options["--action"] == "on":
+			if is_comanaged(conn, options):
+				profile = ""
+			else:
+				profile = " -f `lssyscfg -r lpar -F curr_profile " + \
+				    " -m " + options["--managed"] + \
+				    " --filter \"lpar_names=" + options["--plug"] + "\"`"
+			conn.send("chsysstate -o on -r lpar" +
+				" -m " + options["--managed"] +
+				" -n " + options["--plug"] +
+				profile +
+				"\n")
+		else:
+			conn.send("chsysstate -o shutdown -r lpar --immed" +
+				" -m " + options["--managed"] + " -n " + options["--plug"] + "\n")
+
+		# First line (command) may cause parsing issues if long
+		conn.readline()
+		conn.log_expect(options["--command-prompt"], int(options["--power-timeout"]))
+
+def get_lpar_list(conn, options):
+	outlets = {}
+	if options["--hmc-version"] == "3":
+		conn.send("query_partition_names -m " + options["--managed"] + "\n")
+
+		## We have to remove first line (command)
+		conn.readline()
+		conn.log_expect(options["--command-prompt"], int(options["--power-timeout"]))
+
+		## We have to remove next 2 lines (header) and last line (part of new prompt)
+		####
+		res = re.search("^(.+?\n){2}(.*)\n.*$", conn.before, re.S)
+
+		if res == None:
+			fail_usage("Unable to parse output of list command")
+
+		lines = res.group(2).split("\n")
+		for outlet_line in lines:
+			outlets[outlet_line.rstrip()] = ("", "")
+	elif options["--hmc-version"] in ["4", "IVM"]:
+		sep = ":" if options["--hmc-version"] == "4" else ","
+
+		conn.send("lssyscfg -r lpar -m " + options["--managed"] +
+			" -F name" + sep + "state\n")
+
+		## We have to remove first line (command)
+		conn.readline()
+		conn.log_expect(options["--command-prompt"], int(options["--power-timeout"]))
+
+		## We have to remove last line (part of new prompt)
+		####
+		res = re.search("^(.*)\n.*$", conn.before, re.S)
+
+		if res == None:
+			fail_usage("Unable to parse output of list command")
+
+		lines = res.group(1).split("\n")
+		for outlet_line in lines:
+			try:
+				(port, status) = outlet_line.rstrip().split(sep)
+			except ValueError:
+				fail_usage('Output does not match expected HMC version, try different one');
+			outlets[port] = ("", _normalize_status(status))
+
+	return outlets
+
+def define_new_opts():
+	all_opt["managed"] = {
+		"getopt" : "s:",
+		"longopt" : "managed",
+		"help" : "-s, --managed=[id]             Name of the managed system",
+		"required" : "1",
+		"shortdesc" : "Managed system name",
+		"order" : 1}
+	all_opt["hmc_version"] = {
+		"getopt" : "H:",
+		"longopt" : "hmc-version",
+		"help" : "-H, --hmc-version=[version]    Force HMC version to use: (3|4|ivm) (default: 4)",
+		"required" : "0",
+		"shortdesc" : "Force HMC version to use",
+		"default" : "4",
+		"choices" : ["3", "4", "ivm"],
+		"order" : 1}
+
+def main():
+	device_opt = ["ipaddr", "login", "passwd", "secure", "cmd_prompt", \
+	                "port", "managed", "hmc_version"]
+
+	atexit.register(atexit_handler)
+
+	define_new_opts()
+
+	all_opt["login_timeout"]["default"] = "15"
+	all_opt["secure"]["default"] = "1"
+	all_opt["cmd_prompt"]["default"] = [r":~>", r"]\$", r"\$ "]
+
+	options = check_input(device_opt, process_input(device_opt), other_conditions = True)
+
+	docs = {}
+	docs["shortdesc"] = "Fence agent for IBM LPAR"
+	docs["longdesc"] = ""
+	docs["vendorurl"] = "http://www.ibm.com"
+	show_docs(options, docs)
+
+	if "--managed" not in options:
+		fail_usage("Failed: You have to enter name of managed system")
+
+	if options["--action"] == "validate-all":
+		sys.exit(0)
+
+	##
+	## Operate the fencing device
+	####
+	conn = fence_login(options)
+	result = fence_action(conn, options, set_power_status, get_power_status, get_lpar_list)
+	fence_logout(conn, "quit\r\n")
+	sys.exit(result)
+
+if __name__ == "__main__":
+	main()