diff options
Diffstat (limited to '')
| -rw-r--r-- | lib/fencing_snmp.py.py | 128 |
1 files changed, 128 insertions, 0 deletions
diff --git a/lib/fencing_snmp.py.py b/lib/fencing_snmp.py.py new file mode 100644 index 0000000..f9e5768 --- /dev/null +++ b/lib/fencing_snmp.py.py @@ -0,0 +1,128 @@ +#!@PYTHON@ -tt + +# For example of use please see fence_cisco_mds + +import re, pexpect +import logging +from fencing import * +from fencing import fail, fail_usage, EC_TIMED_OUT, run_delay, frun + +__all__ = ['FencingSnmp'] + +## do not add code here. +class FencingSnmp: + def __init__(self, options): + self.options = options + run_delay(options) + + def quote_for_run(self, string): + return string.replace(r"'", "'\\''") + + def complete_missed_params(self): + mapping = [[ + ['snmp-priv-passwd', 'password', '!snmp-sec-level'], + 'self.options["--snmp-sec-level"]="authPriv"' + ], [ + ['!snmp-version', 'community', '!username', '!snmp-priv-passwd', '!password'], + 'self.options["--snmp-version"]="2c"' + ]] + + for val in mapping: + e = val[0] + + res = True + + for item in e: + if item[0] == '!' and "--" + item[1:] in self.options: + res = False + break + + if item[0] != '!' and "--" + item[0:] not in self.options: + res = False + break + + if res: + exec(val[1]) + + def prepare_cmd(self, command): + cmd = "%s -m '' -Oeqn "% (command) + + self.complete_missed_params() + + #mapping from our option to snmpcmd option + mapping = (('snmp-version', 'v'), ('community', 'c')) + + for item in mapping: + if "--" + item[0] in self.options: + cmd += " -%s '%s'"% (item[1], self.quote_for_run(self.options["--" + item[0]])) + + # Some options make sense only for v3 (and for v1/2c can cause "problems") + if ("--snmp-version" in self.options) and (self.options["--snmp-version"] == "3"): + # Mapping from our options to snmpcmd options for v3 + mapping_v3 = (('snmp-auth-prot', 'a'), ('snmp-sec-level', 'l'), ('snmp-priv-prot', 'x'), \ + ('snmp-priv-passwd', 'X'), ('password', 'A'), ('username', 'u')) + for item in mapping_v3: + if "--"+item[0] in self.options: + cmd += " -%s '%s'"% (item[1], self.quote_for_run(self.options["--" + item[0]])) + + force_ipvx = "" + + if "--inet6-only" in self.options: + force_ipvx = "udp6:" + + if "--inet4-only" in self.options: + force_ipvx = "udp:" + + cmd += " '%s%s%s'"% (force_ipvx, self.quote_for_run(self.options["--ip"]), + "--ipport" in self.options and self.quote_for_run(":" + str(self.options["--ipport"])) or "") + return cmd + + def run_command(self, command, additional_timeout=0): + try: + logging.debug("%s\n", command) + + (res_output, res_code) = frun(command, + int(self.options["--shell-timeout"]) + + int(self.options["--login-timeout"]) + + additional_timeout, True) + + if res_code == None: + fail(EC_TIMED_OUT) + + logging.debug("%s\n", res_output) + + if (res_code != 0) or (re.search("^Error ", res_output, re.MULTILINE) != None): + fail_usage("Returned %d: %s"% (res_code, res_output)) + except pexpect.ExceptionPexpect: + fail_usage("Cannot run command %s"%(command)) + + return res_output + + def get(self, oid, additional_timeout=0): + cmd = "%s '%s'"% (self.prepare_cmd(self.options["--snmpget-path"]), self.quote_for_run(oid)) + + output = self.run_command(cmd, additional_timeout).splitlines() + + return output[len(output)-1].split(None, 1) + + def set(self, oid, value, additional_timeout=0): + mapping = ((int, 'i'), (str, 's')) + + type_of_value = '' + + for item in mapping: + if isinstance(value, item[0]): + type_of_value = item[1] + break + + cmd = "%s '%s' %s '%s'" % (self.prepare_cmd(self.options["--snmpset-path"]), + self.quote_for_run(oid), type_of_value, self.quote_for_run(str(value))) + + self.run_command(cmd, additional_timeout) + + def walk(self, oid, additional_timeout=0): + cmd = "%s '%s'"% (self.prepare_cmd(self.options["--snmpwalk-path"]), self.quote_for_run(oid)) + + output = self.run_command(cmd, additional_timeout).splitlines() + + return [x.split(None, 1) for x in output if x.startswith(".")] |