summaryrefslogtreecommitdiffstats
path: root/agents/lpar/fence_lpar.py
blob: 975971a576b5b3f997b291f9df9574abab5b4ac8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
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()