diff --git a/fence/agents/azure_arm/Makefile.am b/fence/agents/azure_arm/Makefile.am new file mode 100644 index 00000000..e61ed898 --- /dev/null +++ b/fence/agents/azure_arm/Makefile.am @@ -0,0 +1,17 @@ +MAINTAINERCLEANFILES= Makefile.in + +TARGET= fence_azure_arm + +SRC= fence_azure_arm + +EXTRA_DIST= $(SRC) + +sbin_SCRIPTS= $(TARGET) + +man_MANS= fence_azure_arm.8 + +FENCE_TEST_ARGS = -l test -p test -a test -n 1 + +include $(top_srcdir)/make/fencebuild.mk +include $(top_srcdir)/make/fenceman.mk + diff --git a/fence/agents/azure_arm/fence_azure_arm.py b/fence/agents/azure_arm/fence_azure_arm.py new file mode 100644 index 00000000..a3211a2f --- /dev/null +++ b/fence/agents/azure_arm/fence_azure_arm.py @@ -0,0 +1,134 @@ +#!@PYTHON@ -tt + +import sys, re, pexpect +import logging +import atexit +sys.path.append("/usr/share/fence") +from fencing import * +from fencing import fail, fail_usage, EC_TIMED_OUT, run_delay + +#BEGIN_VERSION_GENERATION +RELEASE_VERSION="4.0.24.6-7e576" +BUILD_DATE="(built Thu Nov 3 15:43:03 STD 2016)" +REDHAT_COPYRIGHT="Copyright (C) Red Hat, Inc. 2004-2010 All rights reserved." +#END_VERSION_GENERATION + +compute_client = None + +def get_nodes_list(conn, options): + result = {} + if compute_client: + rgName = options["--resourceGroup"] + vms = compute_client.virtual_machines.list(rgName) + for vm in vms: + result[vm.name] = ("", None) + + return result + +def get_power_status(conn, options): + logging.info("getting power status for VM " + options["--plug"]) + + if compute_client: + rgName = options["--resourceGroup"] + vmName = options["--plug"] + + powerState = "unknown" + vmStatus = compute_client.virtual_machines.get(rgName, vmName, "instanceView") + for status in vmStatus.instance_view.statuses: + if status.code.startswith("PowerState"): + powerState = status.code + break + + logging.info("Found power state of VM: " + powerState) + if powerState == "PowerState/running": + return "on" + + return "off" + +def set_power_status(conn, options): + logging.info("setting power status for VM " + options["--plug"] + " to " + options["--action"]) + + if compute_client: + rgName = options["--resourceGroup"] + vmName = options["--plug"] + + if (options["--action"]=="off"): + logging.info("Deallocating " + vmName + "in resource group " + rgName) + compute_client.virtual_machines.deallocate(rgName, vmName) + elif (options["--action"]=="reboot"): + logging.info("Restarting " + vmName + "in resource group " + rgName) + compute_client.virtual_machines.restart(rgName, vmName) + + sys.exit(0) + + sys.exit(1) + +# Main agent method +def main(): + global compute_client + + device_opt = ["resourceGroup", "login", "passwd", "tenantId", "subscriptionId","port"] + + atexit.register(atexit_handler) + + all_opt["resourceGroup"] = { + "getopt" : "rg:", + "longopt" : "resourceGroup", + "help" : "-rg, --resourceGroup=[name] Name of the resource group", + "shortdesc" : "Name of resource group.", + "required" : "1", + "order" : 2 + } + all_opt["tenantId"] = { + "getopt" : "tid:", + "longopt" : "tenantId", + "help" : "-tid, --tenantId=[name] Id of the Azure Active Directory tenant", + "shortdesc" : "Id of Azure Active Directory tenant.", + "required" : "1", + "order" : 3 + } + all_opt["subscriptionId"] = { + "getopt" : "sid:", + "longopt" : "subscriptionId", + "help" : "-sid, --subscriptionId=[name] Id of the Azure subscription", + "shortdesc" : "Id of the Azure subscription.", + "required" : "1", + "order" : 4 + } + + options = check_input(device_opt, process_input(device_opt)) + + docs = {} + docs["shortdesc"] = "Fence agent for Azure Resource Manager" + docs["longdesc"] = "Used to deallocate virtual machines and to report power state of virtual machines running in Azure" + docs["vendorurl"] = "http://www.microsoft.com" + show_docs(options, docs) + + run_delay(options) + + try: + from azure.common.credentials import ServicePrincipalCredentials + from azure.mgmt.compute import ComputeManagementClient + + tenantid = options["--tenantId"] + servicePrincipal = options["--username"] + spPassword = options["--password"] + subscriptionId = options["--subscriptionId"] + credentials = ServicePrincipalCredentials( + client_id = servicePrincipal, + secret = spPassword, + tenant = tenantid + ) + compute_client = ComputeManagementClient( + credentials, + subscriptionId + ) + except ImportError: + fail_usage("Azure Resource Manager Pyhton SDK not found or not accessible") + + # Operate the fencing device + result = fence_action(None, options, set_power_status, get_power_status, get_nodes_list) + sys.exit(result) + +if __name__ == "__main__": + main() diff --git a/tests/data/metadata/fence_azure_arm.xml b/tests/data/metadata/fence_azure_arm.xml new file mode 100644 index 00000000..1bafc3e6 --- /dev/null +++ b/tests/data/metadata/fence_azure_arm.xml @@ -0,0 +1,113 @@ + + +Used to deallocate virtual machines and to report power state of virtual machines running in Azure +http://www.microsoft.com + + + + + Fencing action + + + + + Login name + + + + + Login password or passphrase + + + + + Script to run to retrieve password + + + + + Physical plug number on device, UUID or identification of machine + + + + + Name of resource group. + + + + + Id of Azure Active Directory tenant. + + + + + Id of the Azure subscription. + + + + + Verbose mode + + + + + Write debug information to given file + + + + + Display version information and exit + + + + + Display help and exit + + + + + Separator for CSV created by 'list' operation + + + + + Wait X seconds before fencing is started + + + + + Wait X seconds for cmd prompt after login + + + + + Test X seconds for status change after ON/OFF + + + + + Wait X seconds after issuing ON/OFF + + + + + Wait X seconds for cmd prompt after issuing command + + + + + Count of attempts to retry power on + + + + + + + + + + + + + +