+++ /dev/null
-#!/usr/bin/python
-#
-# Copyright (c) 2016 Cisco and/or its affiliates.
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at:
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-import requests
-import xml.etree.ElementTree as et
-import re
-
-from requests.packages.urllib3.exceptions import InsecureRequestWarning
-requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
-
-BASEDN = "sys/rack-unit-1"
-
-###
-### Helper function - iterate through a list in pairs
-###
-def chunks(lst, chunksize):
- """Yield successive n-sized chunks from l."""
- for i in range(0, len(lst), chunksize):
- yield lst[i:i+chunksize]
-
-###
-### Helper function: Perform an XML request to CIMC
-###
-def xml_req(ip, xml, debug=False):
- if debug:
- print "DEBUG: XML-REQUEST:"
- et.dump(xml)
- headers = {'Content-Type': 'text/xml'}
- req = requests.post('https://' + ip + '/nuova', headers=headers,
- verify=False, data=et.tostring(xml))
- resp = et.fromstring(req.content)
- if debug:
- print "DEBUG: XML-RESPONSE:"
- et.dump(resp)
-
- if resp.tag == 'error':
- if debug:
- print "XML response contains error:"
- et.dump(error)
- raise RuntimeError('XML response contains error')
- return resp
-
-###
-### Authenticate (Log-In) to CIMC and obtain a cookie
-###
-def login(ip, username, password):
- reqxml = et.Element('aaaLogin',
- attrib={'inName':username, 'inPassword':password})
- respxml = xml_req(ip, reqxml)
- try:
- cookie = respxml.attrib['outCookie']
- except:
- print "Cannot find cookie in CIMC server response."
- print "CIMC server output:"
- et.dump(respxml)
- raise
-
- return cookie
-
-###
-### Log out from CIMC.
-###
-### Note: There is a maximum session limit in CIMC and sessions to take a long
-### time (10 minutes) to time out. Therefore, calling this function during
-### testing is essential, otherwise one will quickly exhaust all available
-### sessions.
-###
-def logout(ip, cookie):
- reqxml = et.Element('aaaLogout', attrib={'cookie': cookie,
- 'inCookie': cookie})
- xml_req(ip, reqxml)
-
-###
-### Power off the host
-###
-def powerOff(ip, cookie, debug=False):
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'false',
- 'dn': BASEDN})
- inconfig = et.SubElement(reqxml, 'inConfig')
- et.SubElement(inconfig, 'computeRackUnit',
- attrib={'adminPower': 'down', 'dn': BASEDN})
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-
-###
-### Power on the host
-###
-def powerOn(ip, cookie, debug=False):
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'false',
- 'dn': BASEDN})
- inconfig = et.SubElement(reqxml, 'inConfig')
- et.SubElement(inconfig, 'computeRackUnit',
- attrib={'adminPower': 'up', 'dn': BASEDN})
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-
-###
-### Restore BIOS to default settings
-###
-def restoreBiosDefaultSettings(ip, cookie, debug=False):
- reqxml = et.Element('configResolveClass',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'classId': 'biosPlatformDefaults'})
- respxml = xml_req(ip, reqxml, debug)
-
- configs = respxml.find('outConfigs')
- defaults = configs.find('biosPlatformDefaults')
-
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'dn': "{}/bios/bios-settings".format(BASEDN)})
- inconfig = et.SubElement(reqxml, 'inConfig')
- biosset = et.SubElement(inconfig, 'biosSettings')
- biosset.extend(defaults)
-
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-
-###
-### Apply specified BIOS settings.
-###
-### These must be a list of strings in XML format. Not currently very
-### user friendly. Format can either be obtained from CIMC
-### documention, or by setting them manually and then fetching
-### BIOS settings via CIMC XML API.
-###
-def setBiosSettings(ip, cookie, settings, debug=False):
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'dn': "{}/bios/bios-settings".format(BASEDN)})
- inconfig = et.SubElement(reqxml, 'inConfig')
- biosset = et.SubElement(inconfig, 'biosSettings')
- print "Applying settings:"
- print settings
- for s in settings:
- x = et.fromstring(s)
- et.dump(x)
- biosset.append(et.fromstring(s))
-
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-###
-### Delete any existing virtual drives
-###
-### WARNING: THIS WILL ERASE ALL DATA ON ALL DISKS, WITHOUT ANY CONFIRMATION
-### QUESTION.
-###
-### The server must be POWERED ON for this to succeed.
-###
-def deleteAllVirtualDrives(ip, cookie, debug=False):
- reqxml = et.Element('configResolveClass',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'classId': 'storageController'})
- respxml = xml_req(ip, reqxml, debug)
-
- configs = respxml.find('outConfigs')
- for sc in configs.iter('storageController'):
- if debug:
- print "DEBUG: SC DN {} ID {}".format(sc.attrib['dn'],
- sc.attrib['id'])
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'dn': sc.attrib['dn']})
- inconfig = et.SubElement(reqxml, 'inConfig')
- et.SubElement(inconfig, 'storageController',
- attrib={'adminAction': 'delete-all-vds-reset-pds',
- 'dn': sc.attrib['dn']})
- xml_req(ip, reqxml, debug)
-
-###
-### Create a single RAID-10 across all drives.
-###
-### The server must be POWERED ON for this to succeed.
-###
-def createRaid10_all(ip, cookie, debug=False):
- reqxml = et.Element('configResolveClass',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'classId': 'storageController'})
- respxml = xml_req(ip, reqxml, debug)
-
- configs = respxml.find('outConfigs')
- for sc in configs.iter('storageController'):
- if debug:
- print "DEBUG: SC DN {} ID {}".format(sc.attrib['dn'],
- sc.attrib['id'])
- #
- # Find disk size and number of disks
- #
- disks = []
- total_size = 0
- for pd in sc.iter('storageLocalDisk'):
- if debug:
- print "DEBUG: PD {} size {}".format(pd.attrib['id'],
- pd.attrib['coercedSize'])
- disks.append(pd.attrib['id'])
- total_size += int(pd.attrib['coercedSize'].split(' ')[0])
-
- #
- # Create a RAID10 array of all available disks, as in:
- # [1,2][3,4][5,6][7,8][9,10][11,12][13,14][15,16][17,18]
- #
- raid_size = total_size/2
- raid_span = ''
- for p in list(chunks(disks, 2)):
- raid_span += "[{},{}]".format(p[0], p[1])
-
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'dn': sc.attrib['dn']})
- inconfig = et.SubElement(reqxml, 'inConfig')
- et.SubElement(inconfig,
- 'storageVirtualDriveCreatorUsingUnusedPhysicalDrive',
- attrib={'virtualDriveName': 'raid10-all',
- 'size': str(raid_size)+' MB',
- 'raidLevel': '10', 'driveGroup': raid_span,
- 'adminState': 'trigger'})
-
- xml_req(ip, reqxml, debug)
-
-###
-### Create a single RAID across from empty drives as provided.
-###
-### The server must be POWERED ON for this to succeed.
-###
-def createRaid(ip, cookie, name, raidlevel, size, drives, debug=False):
- reqxml = et.Element('configResolveClass',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'classId': 'storageController'})
- respxml = xml_req(ip, reqxml, debug)
-
- configs = respxml.find('outConfigs')
- for sc in configs.iter('storageController'):
- if debug:
- print "DEBUG: SC DN {} ID {}".format(sc.attrib['dn'],
- sc.attrib['id'])
-
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'dn': sc.attrib['dn']})
- inconfig = et.SubElement(reqxml, 'inConfig')
- et.SubElement(inconfig,
- 'storageVirtualDriveCreatorUsingUnusedPhysicalDrive',
- attrib={'virtualDriveName': name,
- 'size': str(size)+' MB',
- 'raidLevel': raidlevel,
- 'driveGroup': drives,
- 'adminState': 'trigger'})
-
- xml_req(ip, reqxml, debug)
-
-###
-### Enable Serial-Over-LAN (SOL) console and redirect BIOS output to
-### serial console
-###
-def enableConsoleRedir(ip, cookie, debug=False):
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'false',
- 'dn': "{}/bios/bios-settings".format(BASEDN)})
- inconfig = et.SubElement(reqxml, 'inConfig')
- bs = et.SubElement(inconfig, 'biosSettings',
- attrib={'dn': "{}/bios/bios-settings".format(BASEDN)})
- et.SubElement(bs,
- 'biosVfConsoleRedirection',
- attrib={'vpConsoleRedirection': 'com-0',
- 'vpBaudRate': '115200'})
- respxml = xml_req(ip, reqxml, debug)
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'false',
- 'dn': BASEDN+'/sol-if'})
- inconfig = et.SubElement(reqxml, 'inConfig')
- et.SubElement(inconfig, 'solIf',
- attrib={'dn': BASEDN+'/sol-if', 'adminState': 'enable',
- 'speed': '115200', 'comport': 'com0'})
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-
-###
-### Boot into UEFI bootloader (we may use this to "park" the host in
-### powered-on state)
-###
-def bootIntoUefi(ip, cookie, debug=False):
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'false',
- 'dn': BASEDN+'/boot-policy'})
- inconfig = et.SubElement(reqxml, 'inConfig')
- bootDef = et.SubElement(inconfig, 'lsbootDef',
- attrib={'dn': BASEDN+'/boot-policy',
- 'rebootOnUpdate': 'yes'})
- et.SubElement(bootDef, 'lsbootEfi',
- attrib={'rn': 'efi-read-only', 'order': '1',
- 'type': 'efi'})
-
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-
-###
-### Boot via PXE. Reboot immediately.
-###
-def bootPXE(ip, cookie, debug=False):
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'false',
- 'dn': BASEDN+'/boot-policy'})
- inconfig = et.SubElement(reqxml, 'inConfig')
- bootDef = et.SubElement(inconfig, 'lsbootDef',
- attrib={'dn': BASEDN+'/boot-policy',
- 'rebootOnUpdate': 'yes'})
- et.SubElement(bootDef, 'lsbootLan',
- attrib={'rn': 'lan-read-only', 'order': '1',
- 'type': 'lan', 'prot': 'pxe'})
-
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-
-
-###
-### Boot via Local HDD first, then via PXE. Do not reboot immediately.
-###
-def bootHDDPXE(ip, cookie, debug=False):
- reqxml = et.Element('configConfMo',
- attrib={'cookie': cookie, 'inHierarchical': 'false',
- 'dn': BASEDN+'/boot-policy'})
- inconfig = et.SubElement(reqxml, 'inConfig')
- bootDef = et.SubElement(inconfig, 'lsbootDef',
- attrib={'dn': BASEDN+'/boot-policy',
- 'rebootOnUpdate': 'no'})
- storage = et.SubElement(bootDef, 'lsbootStorage',
- attrib={'rn': 'storage-read-write',
- 'access': 'read-write',
- 'order': '1', 'type': 'storage'})
- et.SubElement(storage, 'lsbootLocalStorage',
- attrib={'rn': 'local-storage'})
- et.SubElement(bootDef, 'lsbootLan',
- attrib={'rn': 'lan-read-only', 'order': '2',
- 'type': 'lan', 'prot': 'pxe'})
-
- respxml = xml_req(ip, reqxml, debug)
- return respxml
-
-###
-### Return LOM port 1 MAC address
-###
-def getLOMMacAddress(ip, cookie, debug=False):
- reqxml = et.Element('configResolveClass',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'classId': 'networkAdapterUnit'})
- respxml = xml_req(ip, reqxml, debug)
- reqxml = et.Element('configResolveDn',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'dn': BASEDN+'/network-adapter-L/eth-1'})
- respxml = xml_req(ip, reqxml, debug)
-
- oc = respxml.find('outConfig')
- netw = oc.find('networkAdapterEthIf')
- if debug:
- print "DEBUG: MAC address is {}".format(netw.get('mac'))
- return netw.get('mac')
-
-###
-### Return all port MAC addresses
-###
-def getMacAddresses(ip, cookie, debug=False):
- maclist = {}
- reqxml = et.Element('configResolveClass',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'classId': 'networkAdapterUnit'})
- respxml = xml_req(ip, reqxml, debug)
- oc = respxml.find('outConfigs')
- for adapter in oc.iter('networkAdapterUnit'):
- if debug:
- print "DEBUG: ADAPTER SLOT {} MODEL {}".format(adapter.attrib['slot'],
- adapter.attrib['model'])
- slot = adapter.attrib['slot']
- maclist[slot] = {}
- for port in adapter.iter('networkAdapterEthIf'):
- if debug:
- print "DEBUG: SLOT {} PORT {} MAC {}".format(slot,
- port.attrib['id'],
- port.attrib['mac'])
- maclist[slot][port.attrib['id']] = port.attrib['mac'].lower()
-
- reqxml = et.Element('configResolveClass',
- attrib={'cookie': cookie, 'inHierarchical': 'true',
- 'classId': 'adaptorUnit'})
- respxml = xml_req(ip, reqxml, debug)
- oc = respxml.find('outConfigs')
- for adapter in oc.iter('adaptorUnit'):
- if debug:
- print "DEBUG: VIC ADAPTER SLOT {} MODEL {}".format(adapter.attrib['pciSlot'],
- adapter.attrib['model'])
- slot = adapter.attrib['pciSlot']
- maclist[slot] = {}
- for port in adapter.iter('adaptorHostEthIf'):
- portnum = int(re.sub('eth([0-9]+)', '\\1', port.attrib['name']))+1
- if debug:
- print "DEBUG: VIC SLOT {} PORT {} MAC {}".format(slot,
- portnum,
- port.attrib['mac'])
- maclist[slot][portnum] = port.attrib['mac'].lower()
-
- return maclist
Code Review / csit.git / blobdiff