Adding upstream version 4.12.1.upstream/4.12.1upstream

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

1 files changed, 177 insertions, 0 deletions

diff --git a/agents/redfish/fence_redfish.py b/agents/redfish/fence_redfish.py
new file mode 100644
index 0000000..0f5af52
--- /dev/null
+++ b/agents/redfish/fence_redfish.py
@@ -0,0 +1,177 @@
+#!@PYTHON@ -tt
+
+# Copyright (c) 2018 Dell Inc. or its subsidiaries. All Rights Reserved.
+
+# Fence agent for devices that support the Redfish API Specification.
+
+import sys
+import re
+import logging
+import json
+import requests
+import atexit
+sys.path.append("@FENCEAGENTSLIBDIR@")
+
+from fencing import *
+from fencing import fail_usage, run_delay
+
+GET_HEADERS = {'accept': 'application/json', 'OData-Version': '4.0'}
+POST_HEADERS = {'content-type': 'application/json', 'accept': 'application/json',
+                'OData-Version': '4.0'}
+
+
+def get_power_status(conn, options):
+    response = send_get_request(options, options["--systems-uri"])
+    if response['ret'] is False:
+        fail_usage("Couldn't get power information")
+    data = response['data']
+
+    try:
+        logging.debug("PowerState is: " + data[u'PowerState'])
+    except Exception:
+        fail_usage("Unable to get PowerState: " + "https://" + options["--ip"] + ":" + str(options["--ipport"]) + options["--systems-uri"])
+
+    if data[u'PowerState'].strip() == "Off":
+        return "off"
+    else:
+        return "on"
+
+def set_power_status(conn, options):
+    action = {
+        'on' : "On",
+        'off': "ForceOff",
+        'reboot': "ForceRestart",
+        'diag': "Nmi"
+    }[options.get("original-action") or options["--action"]]
+
+    payload = {'ResetType': action}
+
+    # Search for 'Actions' key and extract URI from it
+    response = send_get_request(options, options["--systems-uri"])
+    if response['ret'] is False:
+        return {'ret': False}
+    data = response['data']
+    action_uri = data["Actions"]["#ComputerSystem.Reset"]["target"]
+
+    response = send_post_request(options, action_uri, payload)
+    if response['ret'] is False:
+        fail_usage("Error sending power command")
+    if options.get("original-action") == "diag":
+        return True
+    return
+
+def send_get_request(options, uri):
+    full_uri = "https://" + options["--ip"] + ":" + str(options["--ipport"]) + uri
+    try:
+        resp = requests.get(full_uri, verify=not "--ssl-insecure" in options,
+                            headers=GET_HEADERS,
+                            auth=(options["--username"], options["--password"]))
+        data = resp.json()
+    except Exception as e:
+        fail_usage("Failed: send_get_request: " + str(e))
+    return {'ret': True, 'data': data}
+
+def send_post_request(options, uri, payload):
+    full_uri = "https://" + options["--ip"] + ":" + str(options["--ipport"]) + uri
+    try:
+        requests.post(full_uri, data=json.dumps(payload),
+                      headers=POST_HEADERS, verify=not "--ssl-insecure" in options,
+                      auth=(options["--username"], options["--password"]))
+    except Exception as e:
+        fail_usage("Failed: send_post_request: " + str(e))
+    return {'ret': True}
+
+def find_systems_resource(options):
+    response = send_get_request(options, options["--redfish-uri"])
+    if response['ret'] is False:
+        return {'ret': False}
+    data = response['data']
+
+    if 'Systems' not in data:
+        # Systems resource not found"
+        return {'ret': False}
+    else:
+        response = send_get_request(options, data["Systems"]["@odata.id"])
+        if response['ret'] is False:
+            return {'ret': False}
+        data = response['data']
+
+        # need to be able to handle more than one entry
+        for member in data[u'Members']:
+            system_uri = member[u'@odata.id']
+        return {'ret': True, 'uri': system_uri}
+
+def define_new_opts():
+    all_opt["redfish-uri"] = {
+        "getopt" : ":",
+        "longopt" : "redfish-uri",
+        "help" : "--redfish-uri=[uri]            Base or starting Redfish URI",
+        "required" : "0",
+        "default" : "/redfish/v1",
+        "shortdesc" : "Base or starting Redfish URI",
+        "order": 1
+    }
+    all_opt["systems-uri"] = {
+        "getopt" : ":",
+        "longopt" : "systems-uri",
+        "help" : "--systems-uri=[uri]            Redfish Systems resource URI",
+        "required" : "0",
+        "shortdesc" : "Redfish Systems resource URI, i.e. /redfish/v1/Systems/System.Embedded.1",
+        "order": 1
+    }
+
+def main():
+    atexit.register(atexit_handler)
+    device_opt = ["ipaddr", "login", "passwd", "redfish-uri", "systems-uri",
+                  "ssl", "diag"]
+    define_new_opts()
+
+    opt = process_input(device_opt)
+
+    all_opt["ssl"]["default"] = "1"
+    options = check_input(device_opt, opt)
+
+    docs = {}
+    docs["shortdesc"] = "I/O Fencing agent for Redfish"
+    docs["longdesc"] = "fence_redfish is an I/O Fencing agent which can be used with \
+Out-of-Band controllers that support Redfish APIs. These controllers provide remote \
+access to control power on a server."
+    docs["vendorurl"] = "http://www.dmtf.org"
+    show_docs(options, docs)
+    run_delay(options)
+
+    ##
+    ## Operate the fencing device
+    ####
+
+    # Disable insecure-certificate-warning message
+    if "--ssl-insecure" in opt:
+        import urllib3
+        urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
+
+    # backwards compatibility for <ip>:<port>
+    if options["--ip"].count(":") == 1:
+        (options["--ip"], options["--ipport"]) = options["--ip"].split(":")
+
+    if "--systems-uri" not in opt:
+        # Systems URI not provided, find it
+        sysresult = find_systems_resource(options)
+        if sysresult['ret'] is False:
+            sys.exit(1)
+        else:
+            options["--systems-uri"] = sysresult["uri"]
+
+    reboot_fn = None
+    if options["--action"] == "diag":
+        # Diag is a special action that can't be verified so we will reuse reboot functionality
+        # to minimize impact on generic library
+        options["original-action"] = options["--action"]
+        options["--action"] = "reboot"
+        options["--method"] = "cycle"
+        reboot_fn = set_power_status
+
+    result = fence_action(None, options, set_power_status, get_power_status, None, reboot_fn)
+    sys.exit(result)
+
+if __name__ == "__main__":
+    main()