| OLD | NEW |
|---|
| (Empty) |
| 1 #!/usr/bin/python | |
| 2 | |
| 3 ############################################################################# | |
| 4 ############################################################################# | |
| 5 ## | |
| 6 ## Copyright (C) Sistina Software, Inc. 1997-2003 All rights reserved. | |
| 7 ## Copyright (C) 2004-2006 Red Hat, Inc. All rights reserved. | |
| 8 ## | |
| 9 ## This copyrighted material is made available to anyone wishing to use, | |
| 10 ## modify, copy, or redistribute it subject to the terms and conditions | |
| 11 ## of the GNU General Public License v.2. | |
| 12 ## | |
| 13 ############################################################################# | |
| 14 ## This APC Fence script uses snmp to control the APC power | |
| 15 ## switch. This script requires that net-snmp-utils be installed | |
| 16 ## on all nodes in the cluster, and that the powernet369.mib file be | |
| 17 ## located in /usr/share/snmp/mibs/ | |
| 18 ############################################################################# | |
| 19 ############################################################################# | |
| 20 | |
| 21 | |
| 22 | |
| 23 import getopt, sys | |
| 24 import os | |
| 25 import time | |
| 26 import select | |
| 27 import signal | |
| 28 from glob import glob | |
| 29 | |
| 30 #BEGIN_VERSION_GENERATION | |
| 31 FENCE_RELEASE_NAME="" | |
| 32 REDHAT_COPYRIGHT="" | |
| 33 BUILD_DATE="" | |
| 34 #END_VERSION_GENERATION | |
| 35 | |
| 36 POWER_ON="outletOn" | |
| 37 POWER_OFF="outletOff" | |
| 38 POWER_REBOOT="outletReboot" | |
| 39 | |
| 40 def usage(): | |
| 41 print "Usage:"; | |
| 42 print ""; | |
| 43 print "Options:"; | |
| 44 print " -a <ip> IP address or hostname of MasterSwitch"; | |
| 45 print " -h usage"; | |
| 46 print " -l <name> Login name"; | |
| 47 print " -n <num> Outlet number to change"; | |
| 48 print " -o <string> Action: Reboot (default), Off or On"; | |
| 49 print " -p <string> Login password"; | |
| 50 print " -q quiet mode"; | |
| 51 print " -V version"; | |
| 52 print " -v Log to file /tmp/apclog"; | |
| 53 | |
| 54 print sys.argv | |
| 55 sys.exit(0); | |
| 56 | |
| 57 | |
| 58 | |
| 59 def main(): | |
| 60 apc_base = "enterprises.apc.products.hardware." | |
| 61 apc_outletctl = "masterswitch.sPDUOutletControl.sPDUOutletControlTable.sPDUO
utletControlEntry.sPDUOutletCtl." | |
| 62 apc_outletstatus = "masterswitch.sPDUOutletStatus.sPDUOutletStatusMSPTable.s
PDUOutletStatusMSPEntry.sPDUOutletStatusMSP." | |
| 63 | |
| 64 address = "" | |
| 65 output = "" | |
| 66 port = "" | |
| 67 action = "outletReboot" | |
| 68 status_check = False | |
| 69 verbose = False | |
| 70 | |
| 71 if not glob('/usr/share/snmp/mibs/powernet*.mib'): | |
| 72 sys.stderr.write('This APC Fence script uses snmp to control the APC pow
er switch. This script requires that net-snmp-utils be installed on all nodes in
the cluster, and that the powernet369.mib file be located in /usr/share/snmp/mi
bs/\n') | |
| 73 sys.exit(1) | |
| 74 | |
| 75 if len(sys.argv) > 1: | |
| 76 try: | |
| 77 opts, args = getopt.getopt(sys.argv[1:], "a:hl:p:n:o:vV", ["help", "
output="]) | |
| 78 except getopt.GetoptError: | |
| 79 #print help info and quit | |
| 80 usage() | |
| 81 sys.exit(2) | |
| 82 | |
| 83 for o, a in opts: | |
| 84 if o == "-v": | |
| 85 verbose = True | |
| 86 if o == "-V": | |
| 87 print "%s\n" % FENCE_RELEASE_NAME | |
| 88 print "%s\n" % REDHAT_COPYRIGHT | |
| 89 print "%s\n" % BUILD_DATE | |
| 90 sys.exit(0) | |
| 91 if o in ("-h", "--help"): | |
| 92 usage() | |
| 93 sys.exit(0) | |
| 94 if o == "-n": | |
| 95 port = a | |
| 96 if o == "-o": | |
| 97 lcase = a.lower() #Lower case string | |
| 98 if lcase == "off": | |
| 99 action = "outletOff" | |
| 100 elif lcase == "on": | |
| 101 action = "outletOn" | |
| 102 elif lcase == "reboot": | |
| 103 action = "outletReboot" | |
| 104 elif lcase == "status": | |
| 105 #action = "sPDUOutletStatusMSPOutletState" | |
| 106 action = "" | |
| 107 status_check = True | |
| 108 else: | |
| 109 usage() | |
| 110 sys.exit() | |
| 111 if o == "-a": | |
| 112 address = a | |
| 113 | |
| 114 if address == "": | |
| 115 usage() | |
| 116 sys.exit(1) | |
| 117 | |
| 118 if port == "": | |
| 119 usage() | |
| 120 sys.exit(1) | |
| 121 | |
| 122 else: #Get opts from stdin | |
| 123 params = {} | |
| 124 #place params in dict | |
| 125 for line in sys.stdin: | |
| 126 val = line.split("=") | |
| 127 if len(val) == 2: | |
| 128 params[val[0].strip()] = val[1].strip() | |
| 129 | |
| 130 try: | |
| 131 address = params["ipaddr"] | |
| 132 except KeyError, e: | |
| 133 sys.stderr.write("FENCE: Missing ipaddr param for fence_apc...exitin
g") | |
| 134 sys.exit(1) | |
| 135 try: | |
| 136 login = params["login"] | |
| 137 except KeyError, e: | |
| 138 sys.stderr.write("FENCE: Missing login param for fence_apc...exiting
") | |
| 139 sys.exit(1) | |
| 140 | |
| 141 try: | |
| 142 passwd = params["passwd"] | |
| 143 except KeyError, e: | |
| 144 sys.stderr.write("FENCE: Missing passwd param for fence_apc...exitin
g") | |
| 145 sys.exit(1) | |
| 146 | |
| 147 try: | |
| 148 port = params["port"] | |
| 149 except KeyError, e: | |
| 150 sys.stderr.write("FENCE: Missing port param for fence_apc...exiting"
) | |
| 151 sys.exit(1) | |
| 152 | |
| 153 | |
| 154 try: | |
| 155 a = params["option"] | |
| 156 if a == "Off" or a == "OFF" or a == "off": | |
| 157 action = POWER_OFF | |
| 158 elif a == "On" or a == "ON" or a == "on": | |
| 159 action = POWER_ON | |
| 160 elif a == "Reboot" or a == "REBOOT" or a == "reboot": | |
| 161 action = POWER_REBOOT | |
| 162 except KeyError, e: | |
| 163 action = POWER_REBOOT | |
| 164 | |
| 165 ####End of stdin section | |
| 166 | |
| 167 apc_command = apc_base + apc_outletctl + port | |
| 168 | |
| 169 args_status = list() | |
| 170 args_off = list() | |
| 171 args_on = list() | |
| 172 | |
| 173 args_status.append("/usr/bin/snmpget") | |
| 174 args_status.append("-Oqu") #sets printing options | |
| 175 args_status.append("-v") | |
| 176 args_status.append("1") | |
| 177 args_status.append("-c") | |
| 178 args_status.append("private") | |
| 179 args_status.append("-m") | |
| 180 args_status.append("ALL") | |
| 181 args_status.append(address) | |
| 182 args_status.append(apc_command) | |
| 183 | |
| 184 args_off.append("/usr/bin/snmpset") | |
| 185 args_off.append("-Oqu") #sets printing options | |
| 186 args_off.append("-v") | |
| 187 args_off.append("1") | |
| 188 args_off.append("-c") | |
| 189 args_off.append("private") | |
| 190 args_off.append("-m") | |
| 191 args_off.append("ALL") | |
| 192 args_off.append(address) | |
| 193 args_off.append(apc_command) | |
| 194 args_off.append("i") | |
| 195 args_off.append("outletOff") | |
| 196 | |
| 197 args_on.append("/usr/bin/snmpset") | |
| 198 args_on.append("-Oqu") #sets printing options | |
| 199 args_on.append("-v") | |
| 200 args_on.append("1") | |
| 201 args_on.append("-c") | |
| 202 args_on.append("private") | |
| 203 args_on.append("-m") | |
| 204 args_on.append("ALL") | |
| 205 args_on.append(address) | |
| 206 args_on.append(apc_command) | |
| 207 args_on.append("i") | |
| 208 args_on.append("outletOn") | |
| 209 | |
| 210 cmdstr_status = ' '.join(args_status) | |
| 211 cmdstr_off = ' '.join(args_off) | |
| 212 cmdstr_on = ' '.join(args_on) | |
| 213 | |
| 214 ##This section issues the actual commands. Reboot is split into | |
| 215 ##Off, then On to make certain both actions work as planned. | |
| 216 ## | |
| 217 ##The status command just dumps the outlet status to stdout. | |
| 218 ##The status checks that are made when turning an outlet on or off, though, | |
| 219 ##use the execWithCaptureStatus so that the stdout from snmpget can be | |
| 220 ##examined and the desired operation confirmed. | |
| 221 | |
| 222 if status_check: | |
| 223 if verbose: | |
| 224 fd = open("/tmp/apclog", "w") | |
| 225 fd.write("Attempting the following command: %s\n" % cmdstr_status) | |
| 226 strr = os.system(cmdstr_status) | |
| 227 print strr | |
| 228 if verbose: | |
| 229 fd.write("Result: %s\n" % strr) | |
| 230 fd.close() | |
| 231 | |
| 232 else: | |
| 233 if action == POWER_OFF: | |
| 234 if verbose: | |
| 235 fd = open("/tmp/apclog", "w") | |
| 236 fd.write("Attempting the following command: %s\n" % cmdstr_off) | |
| 237 strr = os.system(cmdstr_off) | |
| 238 time.sleep(1) | |
| 239 strr,code = execWithCaptureStatus("/usr/bin/snmpget",args_status) | |
| 240 if verbose: | |
| 241 fd.write("Result: %s\n" % strr) | |
| 242 fd.close() | |
| 243 if strr.find(POWER_OFF) >= 0: | |
| 244 print "Success. Outlet off" | |
| 245 sys.exit(0) | |
| 246 else: | |
| 247 if verbose: | |
| 248 fd.write("Unable to power off apc outlet") | |
| 249 fd.close() | |
| 250 sys.exit(1) | |
| 251 | |
| 252 elif action == POWER_ON: | |
| 253 if verbose: | |
| 254 fd = open("/tmp/apclog", "w") | |
| 255 fd.write("Attempting the following command: %s\n" % cmdstr_on) | |
| 256 strr = os.system(cmdstr_on) | |
| 257 time.sleep(1) | |
| 258 strr,code = execWithCaptureStatus("/usr/bin/snmpget",args_status) | |
| 259 #strr = os.system(cmdstr_status) | |
| 260 if verbose: | |
| 261 fd.write("Result: %s\n" % strr) | |
| 262 if strr.find(POWER_ON) >= 0: | |
| 263 if verbose: | |
| 264 fd.close() | |
| 265 print "Success. Outlet On." | |
| 266 sys.exit(0) | |
| 267 else: | |
| 268 print "Unable to power on apc outlet" | |
| 269 if verbose: | |
| 270 fd.write("Unable to power on apc outlet") | |
| 271 fd.close() | |
| 272 sys.exit(1) | |
| 273 | |
| 274 elif action == POWER_REBOOT: | |
| 275 if verbose: | |
| 276 fd = open("/tmp/apclog", "w") | |
| 277 fd.write("Attempting the following command: %s\n" % cmdstr_off) | |
| 278 strr = os.system(cmdstr_off) | |
| 279 time.sleep(1) | |
| 280 strr,code = execWithCaptureStatus("/usr/bin/snmpget",args_status) | |
| 281 #strr = os.system(cmdstr_status) | |
| 282 if verbose: | |
| 283 fd.write("Result: %s\n" % strr) | |
| 284 if strr.find(POWER_OFF) < 0: | |
| 285 print "Unable to power off apc outlet" | |
| 286 if verbose: | |
| 287 fd.write("Unable to power off apc outlet") | |
| 288 fd.close() | |
| 289 sys.exit(1) | |
| 290 | |
| 291 if verbose: | |
| 292 fd.write("Attempting the following command: %s\n" % cmdstr_on) | |
| 293 strr = os.system(cmdstr_on) | |
| 294 time.sleep(1) | |
| 295 strr,code = execWithCaptureStatus("/usr/bin/snmpget",args_status) | |
| 296 #strr = os.system(cmdstr_status) | |
| 297 if verbose: | |
| 298 fd.write("Result: %s\n" % strr) | |
| 299 if strr.find(POWER_ON) >= 0: | |
| 300 if verbose: | |
| 301 fd.close() | |
| 302 print "Success: Outlet Rebooted." | |
| 303 sys.exit(0) | |
| 304 else: | |
| 305 print "Unable to power on apc outlet" | |
| 306 if verbose: | |
| 307 fd.write("Unable to power on apc outlet") | |
| 308 fd.close() | |
| 309 sys.exit(1) | |
| 310 | |
| 311 def execWithCaptureStatus(command, argv, searchPath = 0, root = '/', stdin = 0, | |
| 312 catchfd = 1, closefd = -1): | |
| 313 | |
| 314 if not os.access (root + command, os.X_OK): | |
| 315 raise RuntimeError, command + " cannot be run" | |
| 316 | |
| 317 (read, write) = os.pipe() | |
| 318 | |
| 319 childpid = os.fork() | |
| 320 if (not childpid): | |
| 321 if (root and root != '/'): os.chroot (root) | |
| 322 if isinstance(catchfd, tuple): | |
| 323 for fd in catchfd: | |
| 324 os.dup2(write, fd) | |
| 325 else: | |
| 326 os.dup2(write, catchfd) | |
| 327 os.close(write) | |
| 328 os.close(read) | |
| 329 | |
| 330 if closefd != -1: | |
| 331 os.close(closefd) | |
| 332 | |
| 333 if stdin: | |
| 334 os.dup2(stdin, 0) | |
| 335 os.close(stdin) | |
| 336 | |
| 337 if (searchPath): | |
| 338 os.execvp(command, argv) | |
| 339 else: | |
| 340 os.execv(command, argv) | |
| 341 | |
| 342 sys.exit(1) | |
| 343 | |
| 344 os.close(write) | |
| 345 | |
| 346 rc = "" | |
| 347 s = "1" | |
| 348 while (s): | |
| 349 select.select([read], [], []) | |
| 350 s = os.read(read, 1000) | |
| 351 rc = rc + s | |
| 352 | |
| 353 os.close(read) | |
| 354 | |
| 355 pid = -1 | |
| 356 status = -1 | |
| 357 try: | |
| 358 (pid, status) = os.waitpid(childpid, 0) | |
| 359 except OSError, (errno, msg): | |
| 360 print __name__, "waitpid:", msg | |
| 361 | |
| 362 if os.WIFEXITED(status) and (os.WEXITSTATUS(status) == 0): | |
| 363 status = os.WEXITSTATUS(status) | |
| 364 else: | |
| 365 status = -1 | |
| 366 | |
| 367 return (rc, status) | |
| 368 | |
| 369 if __name__ == "__main__": | |
| 370 main() | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 6124004:
Revert "Merge remote branch 'cros/upstream' into autotest-rebase" (Closed)
Base URL: ssh://git@gitrw.chromium.org:9222/autotest.git@master