def test_imc_context_manager_no_timeout(): try: import ConfigParser except: import configparser as ConfigParser from imcsdk.imchandle import ImcHandle from ..connection import info host = 'imc' config = ConfigParser.RawConfigParser() config.read(info.CONNECTION_CFG_FILEPATH) hostname = config.get(host, "hostname") username = config.get(host, "username") password = config.get(host, "password") handle = ImcHandle(hostname, username, password) with handle: server_dn = get_server_dn(handle) mo = handle.query_dn(server_dn, timeout=600) usr_lbl = "test-lbl2" mo.usr_lbl = usr_lbl handle.set_mo(mo, timeout=600) mo = handle.query_dn(server_dn) assert_equal(mo.usr_lbl, usr_lbl)
def test_imc_context_manager_no_timeout(): try: import ConfigParser except: import configparser as ConfigParser from imcsdk.imchandle import ImcHandle from ..connection import info host = 'imc' config = ConfigParser.RawConfigParser() config.read(info.CONNECTION_CFG_FILEPATH) hostname = config.get(host, "hostname") username = config.get(host, "username") password = config.get(host, "password") handle = ImcHandle(hostname, username, password) with handle: server_dn = get_server_dn(handle) mo = handle.query_dn(server_dn, timeout=600) usr_lbl = "test-lbl2" mo.usr_lbl = usr_lbl handle.set_mo(mo, timeout=600) mo = handle.query_dn(server_dn) assert_equal(mo.usr_lbl, usr_lbl)
def reboot(cimc_ip, cimc_user, cimc_password): handle = ImcHandle(cimc_ip, cimc_user, cimc_password) handle.login(auto_refresh=False) mo = handle.config_resolve_dn("sys/rack-unit-1") mo.admin_power = ComputeRackUnit.ADMIN_POWER_BMC_RESET_IMMEDIATE try: handle.set_mo(mo) # handle.set_imc_managedobject( # mo, class_id="ComputeRackUnit", # params={ # ComputeRackUnit.ADMIN_POWER: # ComputeRackUnit.CONST_ADMIN_POWER_BMC_RESET_IMMEDIATE, # ComputeRackUnit.DN: "sys/rack-unit-1"}) except Exception as e: print(e)
def update_snmp(): """ Update IMC SNMP and Trap Destination """ handle = ImcHandle(IMC_HOST, IMC_USER, IMC_PASS) handle.login() imc_snmp = handle.query_dn('sys/svc-ext/snmp-svc') imc_snmp.admin_state = 'enabled' imc_snmp.community = 'public' handle.set_mo(imc_snmp) imc_snmp_trap_dest = handle.query_dn('sys/svc-ext/snmp-svc/snmp-trap-1') imc_snmp_trap_dest.admin_state = 'enabled' imc_snmp_trap_dest.hostname = '10.10.10.10' imc_snmp_trap_dest.version = 'v2c' handle.set_mo(imc_snmp_trap_dest) handle.logout()
handle.login() from imcsdk.mometa.storage.StorageVirtualDriveCreatorUsingUnusedPhysicalDrive import StorageVirtualDriveCreatorUsingUnusedPhysicalDrive for raid_params in settings_file['raid_config']: print "Create virtual drive %s with RAID level %s" % ( raid_params['drive_name'], raid_params['raid_level']) mo = StorageVirtualDriveCreatorUsingUnusedPhysicalDrive( parent_mo_or_dn=raid_params['dn'], virtual_drive_name=raid_params['drive_name'], raid_level=raid_params['raid_level'], size=raid_params['size'], drive_group=raid_params['drive_group'], write_policy="Write Through", admin_state="trigger") handle.set_mo(mo) if raid_params['boot_drive'] == 'yes': print "Set virtual drive %s as boot drive" % raid_params[ 'drive_name'] from imcsdk.mometa.storage.StorageVirtualDrive import StorageVirtualDrive mo = StorageVirtualDrive(parent_mo_or_dn=raid_params['dn'], id="0", admin_action="set-boot-drive") handle.set_mo(mo) handle.logout() pause = raw_input('') # Redfish API operations # ---------------------- # check CIMC fw version and update if not at desired version
from imcsdk.imchandle import ImcHandle # Create a connection handle handle = ImcHandle("192.168.1.23", "admin", "password") # Create a connection handle handle.login() # Query for the NTP Server imc_ntp_server_mo = handle.query_dn("sys/svc-ext/ntp-svc") print "NTP config inicial: " print imc_ntp_server_mo # Set NTP Server parameters imc_ntp_server_mo.ntp_enable = "yes" imc_ntp_server_mo.ntp_server1 = "5.5.5.5" # Commit the changes to the IMC handle.set_mo(imc_ntp_server_mo) print "NTP config applied" # Verify changes applied imc_ntp_server_mo = handle.query_dn("sys/svc-ext/ntp-svc") print "NTP new config: " print imc_ntp_server_mo # Logout from the server handle.logout()
class CimcServer(LabServer): RAID_0, RAID_1, RAID_10 = '0', '1', '10' def __init__(self, pod, dic): super(CimcServer, self).__init__(pod=pod, dic=dic) self._handle = None self._dump_xml = False self._logout_on_each_command = False def logger(self, message): self.log('CIMC ' + message) @property def handle(self): from imcsdk.imchandle import ImcHandle from urllib2 import URLError if self._handle is None: self._handle = ImcHandle(self.oob_ip, self.oob_username, self.oob_password) try: self._handle.login() except URLError as ex: raise RuntimeError('{}: {} {}'.format(self, ex, self.oob_ip)) return self._handle def cimc_get_raid_battery_status(self): return self._handle.query_classid(class_id='storageRaidBattery') def _cimc_bios_lom(self, status): self.logger('{} all LOM'.format(status)) actual = self._handle.query_classid(class_id='BiosVfLOMPortOptionROM') if actual.Status != status: params = {'Dn': 'sys/rack-unit-1/bios/bios-settings/LOMPort-OptionROM', 'VpLOMPortsAllState': status, 'vpLOMPort0State': status, 'vpLOMPort1State': status} self._handle.set_mo('set_imc_managedobject', in_mo=None, class_id='BiosVfLOMPortOptionROM', params=params) self.cimc_power_cycle() def cimc_enable_loms_in_bios(self): self._cimc_bios_lom(status='Enable') def cimc_disable_loms_in_bios(self): self._cimc_bios_lom(status='Disable') def cimc_list_all_nics(self): self.log('{} getting CIMC NICs'.format(self.oob_ip)) r1 = self.handle.query_classid(class_id='networkAdapterEthIf') # physical Intel NICs (or others in PCI-E slots) r2 = self.handle.query_classid(class_id='adaptorExtEthIf') # physical Cisco VIC in MLOM return {x.dn: x.mac for x in r1 + r2} def cimc_list_all_nics_and_vnics(self): r1 = self.handle.query_classid(class_id='networkAdapterEthIf') # physical Intel NICs (or others in PCI-E slots) r2 = self.handle.query_classid(class_id='adaptorExtEthIf') # physical Cisco VIC in MLOM r3 = self.handle.query_classid(class_id='adaptorHostEthIf') # virtual Cisco vNIC in MLOM return {x.dn: x.mac for x in r1 + r2 + r3} def cimc_list_pci_nic(self): r = self.handle.query_classid('networkAdapterUnit') return [{'Dn': x.Dn, 'Model': x.Model} for x in r] def cimc_list_vnics(self): ans1 = self.handle.query_classid('adaptorHostEthIf') ans2 = self.handle.query_classid('adaptorEthGenProfile') vnics = {x.Name: {'mac': x.Mac, 'uplink': x.UplinkPort, 'pci-slot': x.UsnicCount, 'dn': x.Dn, 'mtu': x.Mtu, 'name': x.Name, 'pxe-boot': x.PxeBoot} for x in ans1} vlans_dict = {x.Dn.replace('/general', ''): {'vlan': x.Vlan, 'vlan_mode': x.VlanMode} for x in ans2} for vnic in vnics.values(): vlans = vlans_dict[vnic.get('dn')] vnic.update(vlans) return vnics def cimc_enable_sol(self): self.logger('enabling SOL') params = {'dn': 'sys/rack-unit-1/sol-if', 'adminState': 'enable', 'speed': '115200'} self.cimc_set_mo_by_class_id(class_id='solIf', params=params) def create_storage(self, raid=RAID_1, disks_needed=2, clean_vds=False): """ :param clean_vds: Clean all virtual drives before creating any :param raid: Please select from '0','1','10','5','6' :param disks_needed: Number of disks needed :return: """ if raid not in [self.RAID_0, self.RAID_1, self.RAID_10]: raise ValueError('RAID request is not correct. Use one of the {0}. Got: {1}'.format(','.join([self.RAID_0, self.RAID_1, self.RAID_10]), raid)) virtual_drives_list = self.handle.query_classid(class_id='storageVirtualDrive') if virtual_drives_list: if clean_vds: self.logger('Cleaning Virtual Drives to create new one.') for vd in virtual_drives_list: self.handle.remove_mo(vd) else: self.logger('Virtual Drive already exists.') return disks = self.handle.query_classid('storageLocalDisk') # get 2 or more disks to form RAID disks_by_size = {} map(lambda x: disks_by_size.setdefault(x.get_attr('CoercedSize'), []).append(x), disks) available_disks = filter(lambda x: len(disks_by_size[x]) > disks_needed, disks_by_size.keys()) if len(available_disks) == 0: raise Exception('Not enough disks to build RAID {0}. Minimum required are {1}.'.format(raid, disks_needed)) size = available_disks[0] drive_group = ','.join(map(lambda x: x.Id, disks_by_size[size])[:disks_needed]) params = {'raidLevel': raid, 'size': size, 'virtualDriveName': "RAID", 'dn': "sys/rack-unit-1/board/storage-SAS-SLOT-HBA/virtual-drive-create", 'driveGroup': '[{0}]'.format(drive_group), 'adminState': 'trigger', 'writePolicy': 'Write Through'} self.logger('Creating Virtual Drive RAID {0}. Using storage {0}'.format(raid, drive_group)) self.cimc_set_mo_by_class_id(class_id="storageVirtualDriveCreatorUsingUnusedPhysicalDrive", params=params) def cimc_delete_all_vnics(self): self.logger('deleting all vNICs') vnic_names = self.cimc_list_vnics().keys() for vnic_name in vnic_names: self.cimc_delete_vnic(vnic_name=vnic_name) def cimc_delete_vnic(self, vnic_name): dn = 'sys/rack-unit-1/adaptor-MLOM/host-eth-{}'.format(vnic_name) if 'eth0' in vnic_name or 'eth1' in vnic_name: # no way to delete eth0 or eth1, so reset them to default self.logger(message='Resetting vNIC ' + vnic_name) ethX = self.handle.query_dn(dn) ethX.mac = 'AUTO' ethX.mtu = 1500 ethX.pxe_boot = False ethX.uplink_port = vnic_name[-1] self.handle.set_mo(ethX) profile = self.handle.query_dn('sys/rack-unit-1/adaptor-{}/host-eth-{}/general'.format(pci_slot_id, name)) profile.vlan = None profile.order = 'ANY' self.handle.set_mo(profile) else: self.logger(message='Deleting vNIC ' + dn) vnic = self.handle.query_dn(dn=dn) self.handle.remove_mo(vnic) def cimc_set_ssh_timeout(self, timeout=3600): self.cmd('set_imc_managedobject', in_mo=None, class_id='commHttps', params={'Dn': 'sys/svc-ext/https-svc', 'sessionTimeout': str(timeout)}) def cimc_change_boot_order(self, pxe_order=1, hdd_order=2): self.logger('updating boot order: PXE as #{0}, HDD as #{1}'.format(pxe_order, hdd_order)) boot_configs = [{'params': {'Dn': 'sys/rack-unit-1/boot-policy/lan-read-only', 'Order': pxe_order, 'Access': 'read-only'}, 'class_id': 'LsbootLan'}, {'params': {'Dn': 'sys/rack-unit-1/boot-policy/storage-read-write', 'Order': hdd_order, 'Access': 'read-write'}, 'class_id': 'LsbootStorage'}] for boot_config in boot_configs: boot_device = self.cmd('get_imc_managedobject', in_mo=None, class_id=None, params={'Dn': boot_config['params']['Dn']}) if boot_device: self.cmd('set_imc_managedobject', in_mo=boot_device, class_id=None, params=boot_config['params']) else: self.cmd('add_imc_managedobject', in_mo=None, class_id=boot_config['class_id'], params=boot_config['params']) def cimc_create_vnic(self, pci_slot_id, uplink_port, order, name, mac, vlan, is_pxe_enabled): from imcsdk.mometa.adaptor.AdaptorHostEthIf import AdaptorHostEthIf self.logger(message='creating vNIC {} on PCI id {} uplink {} mac={} vlan={} pxe: {}'.format(name, pci_slot_id, uplink_port, mac, vlan, is_pxe_enabled)) if name in ['eth0', 'eth1']: # eth0 and eth1 are default, only possible to modify there parameters, no way to rename ot delete ethX = self.handle.query_dn('sys/rack-unit-1/adaptor-{}/host-eth-{}'.format(pci_slot_id, name)) ethX.mac = mac ethX.mtu = 1500 ethX.pxe_boot = is_pxe_enabled self.handle.set_mo(ethX) else: vnic = AdaptorHostEthIf(parent_mo_or_dn='sys/rack-unit-1/adaptor-' + pci_slot_id, name=name, mac=mac, mtu=1500, pxe_boot=is_pxe_enabled, uplink_port=uplink_port) self.handle.add_mo(vnic) profile = self.handle.query_dn('sys/rack-unit-1/adaptor-{}/host-eth-{}/general'.format(pci_slot_id, name)) profile.vlan = vlan profile.order = order self.handle.set_mo(profile) def cimc_get_power_status(self): return self.handle.query_classid('computeRackUnit')[0].get_attr('OperPower') def _cimc_power(self, requested_state): import time self.logger('power {0}'.format(requested_state)) rack = self.handle.query_dn('sys/rack-unit-1') rack.admin_power = requested_state self.handle.set_mo(rack) time.sleep(120) # wait for the server to come up def cimc_power_down(self): current_power_state = self.cimc_get_power_status() if current_power_state == 'on': self._cimc_power('down') else: self.log(' is already OFF') def cimc_power_up(self): current_power_state = self.cimc_get_power_status() if current_power_state == 'off': self._cimc_power('up') else: self.log(' is already ON') def cimc_reset(self): pass # from ImcSdk.ImcMos import ComputeRackUnit # mo = handle.config_resolve_dn("sys/rack-unit-1") # mo.admin_power = ComputeRackUnit.CONST_ADMIN_POWER_BMC_RESET_IMMEDIATE # set_imc_managedobject(mo, class_id="ComputeRackUnit", params={ComputeRackUnit.ADMIN_POWER: # ComputeRackUnit.CONST_ADMIN_POWER_BMC_RESET_IMMEDIATE, # ComputeRackUnit.DN: "sys/rack-unit-1"}) def cimc_power_cycle(self): current_power_state = self.cimc_get_power_status() self._cimc_power('up' if current_power_state == 'off' else 'cycle-immediate') def cimc_recreate_vnics(self): actual_vnics = self.cimc_list_all_nics_and_vnics() actual_loms = [x for x in self.cimc_list_pci_nic_ports() if 'L' in x['Dn']] for nic_order, nic in enumerate(self.lom_nics_dic.values()): # NIC order starts from 0 names = nic.get_names() macs = nic.get_macs() port_ids = nic.get_port_ids() for name, mac, port_id in zip(names, macs, port_ids): if port_id in ['LOM-1', 'LOM-2']: actual_mac = actual_loms[port_id]['mac'] if mac.upper() != actual_mac.upper(): raise ValueError('{}: "{}" actual mac is "{}" while requested "{}". Edit lab config!'.format(self.id, port_id, actual_mac, mac)) else: if 'eth' not in self.get_nics() and nic.is_ssh(): # if no NIC called eth and it's nic on ssh network, use default eth0, eth1 if name in actual_vnics: self.cimc_delete_vnic(vnic_name=name) name = 'eth' + name[-1] if name in actual_vnics: if mac == actual_vnics[name]['mac'] and str(nic.get_vlan_id()) == str(actual_vnics[name]['vlan']): # this nic is already in CIMC self.logger(message='vNIC {} is already configured'.format(name)) if name in actual_vnics: actual_vnics.pop(name) continue else: self.logger('deleting {} since mac or vlan is not correct: {}'.format(name, actual_vnics[name])) self.cimc_delete_vnic(vnic_name=name) pci_slot_id, uplink_port = port_id.split('/') self.cimc_create_vnic(pci_slot_id=pci_slot_id, uplink_port=uplink_port, order='ANY', name=name, mac=mac, vlan=nic.get_vlan_id(), is_pxe_enabled=nic.is_pxe()) if name in actual_vnics: actual_vnics.pop(name) for vnic_name in actual_vnics.keys(): # delete all actual which are not requested self.cimc_delete_vnic(vnic_name) def cimc_configure(self, is_debug=False): self._dump_xml = is_debug lab_type = self.pod.get_type() self.logger('configuring for {}'.format(lab_type)) self.cimc_set_hostname() self.cimc_power_up() if self.pod.get_type() == self.pod.MERCURY: self.cimc_disable_loms_in_bios() else: is_any_nic_on_lom = any(map(lambda x: x.is_on_lom(), self.get_nics().values())) if is_any_nic_on_lom: self.cimc_enable_loms_in_bios() self.cimc_recreate_vnics() self.cimc_change_boot_order(pxe_order=1, hdd_order=2) # if lab_type == self.lab().LAB_MERCURY: # self.create_storage('1', 2, True) def cimc_get_adaptors(self): r = self.handle.query_classid('AdaptorUnit') return r def cimc_set_mlom_adaptor(self, pci_slot, n_vnics): self.logger(message='allowing MLOM-{} to use up to {} vNICs'.format(pci_slot, n_vnics)) # self.cimc_set_mo_by_class_id(class_id='adaptorUnit', params={'dn': 'sys/rack-unit-1/adaptor-{0}'.format(pci_slot), 'description': str(self)}) self.cimc_set_mo_by_class_id(class_id='adaptorGenProfile', params={'dn': 'sys/rack-unit-1/adaptor-{0}/general'.format(pci_slot), 'fipMode': 'enabled', 'vntagMode': 'enabled', 'numOfVMFexIfs': n_vnics}) def cimc_get_mgmt_nic(self): return self.handle.query_classid('mgmtIf')[0] def cimc_set_hostname(self): current = self.cimc_get_mgmt_nic() new_cimc_hostname = '{}-ru{}-{}'.format(self.pod, self.hardware, self.id) if new_cimc_hostname != current.hostname: self.logger(message='setting hostname to {}'.format(new_cimc_hostname)) current.hostname = new_cimc_hostname self.handle.set_mo(current) else: self.logger(message='hostname is already {}'.format(new_cimc_hostname)) @staticmethod def cimc_deduce_wiring_by_lldp(pod): import yaml from lab.wire import Wire cimc_info_file_path = pod.get_artifact_file_path('cimc-{}.yaml'.format(pod)) try: with open(cimc_info_file_path) as f: cimc_info_dic = yaml.load(f.read()) except IOError: cimc_info_dic = {} for cimc in sorted(pod.cimc_servers_dic.values()): cimc_info_dic[cimc.oob_ip] = cimc.cimc_list_all_nics() # returns dic of {port_id: mac} with open(cimc_info_file_path, mode='w') as f: yaml.dump(cimc_info_dic, f) wires_cfg = [] for node in sorted(pod.cimc_servers_dic.values()): for port1, mac in cimc_info_dic[node.oob_ip].items(): neis = [x.find_neighbour_with_mac(mac=mac, cimc_port_id=port1) for x in pod.switches] neis = filter(lambda n: n is not None, neis) if neis: assert len(neis) == 1, 'More then 1 switch is found connected to {} {}'.format(node, port1) nei = neis[0] wires_cfg.append({'node1': node.id, 'port1': port1, 'mac': mac, 'node2': nei.n9.id, 'port2': nei.port.port_id, 'pc-id': nei.port.pc_id}) else: wires_cfg.append({'node1': node.id, 'port1': port1, 'mac': mac, 'node2': None, 'port2': None, 'pc-id': None}) pod.wires.extend(Wire.add_wires(pod=pod, wires_cfg=wires_cfg))
# Query for SNMP settings imc_snmp_mo = handle.query_dn("sys/svc-ext/snmp-svc") print "SNMP config inicial: " print imc_snmp_mo # Set SNMP parameters imc_snmp_mo.admin_state = "enabled" imc_snmp_mo.com2_sec = "full" imc_snmp_mo.community = "private" imc_snmp_mo.trap_community = "private" imc_snmp_mo.sys_contact = "Mr XPTO" imc_snmp_mo.sys_location = "Datacenter Y" # Commit the changes to the IMC handle.set_mo(imc_snmp_mo) # Verify changes applied imc_snmp_mo = handle.query_dn("sys/svc-ext/snmp-svc") print "SNMP new config: " print imc_snmp_mo # Query for SNMP Trap1 settings imc_snmp_trap1_mo = handle.query_dn("sys/svc-ext/snmp-svc/snmp-trap-1") print "SNMP Trap-1config inicial: " print imc_snmp_trap1_mo # Set SNMP Trap-1 parameters imc_snmp_trap1_mo.admin_state = "enabled"
class CimcServer(LabServer): RAID_0, RAID_1, RAID_10 = '0', '1', '10' def __init__(self, pod, dic): super(CimcServer, self).__init__(pod=pod, dic=dic) self._handle = None self._dump_xml = False self._logout_on_each_command = False def logger(self, message): self.log('CIMC ' + message) @property def handle(self): from imcsdk.imchandle import ImcHandle from urllib2 import URLError if self._handle is None: self._handle = ImcHandle(self.oob_ip, self.oob_username, self.oob_password) try: self._handle.login() except URLError as ex: raise RuntimeError('{}: {} {}'.format(self, ex, self.oob_ip)) return self._handle def cimc_get_raid_battery_status(self): return self._handle.query_classid(class_id='storageRaidBattery') def _cimc_bios_lom(self, status): self.logger('{} all LOM'.format(status)) actual = self._handle.query_classid(class_id='BiosVfLOMPortOptionROM') if actual.Status != status: params = { 'Dn': 'sys/rack-unit-1/bios/bios-settings/LOMPort-OptionROM', 'VpLOMPortsAllState': status, 'vpLOMPort0State': status, 'vpLOMPort1State': status } self._handle.set_mo('set_imc_managedobject', in_mo=None, class_id='BiosVfLOMPortOptionROM', params=params) self.cimc_power_cycle() def cimc_enable_loms_in_bios(self): self._cimc_bios_lom(status='Enable') def cimc_disable_loms_in_bios(self): self._cimc_bios_lom(status='Disable') def cimc_list_all_nics(self): self.log('{} getting CIMC NICs'.format(self.oob_ip)) r1 = self.handle.query_classid( class_id='networkAdapterEthIf' ) # physical Intel NICs (or others in PCI-E slots) r2 = self.handle.query_classid( class_id='adaptorExtEthIf') # physical Cisco VIC in MLOM return {x.dn: x.mac for x in r1 + r2} def cimc_list_all_nics_and_vnics(self): r1 = self.handle.query_classid( class_id='networkAdapterEthIf' ) # physical Intel NICs (or others in PCI-E slots) r2 = self.handle.query_classid( class_id='adaptorExtEthIf') # physical Cisco VIC in MLOM r3 = self.handle.query_classid( class_id='adaptorHostEthIf') # virtual Cisco vNIC in MLOM return {x.dn: x.mac for x in r1 + r2 + r3} def cimc_list_pci_nic(self): r = self.handle.query_classid('networkAdapterUnit') return [{'Dn': x.Dn, 'Model': x.Model} for x in r] def cimc_list_vnics(self): ans1 = self.handle.query_classid('adaptorHostEthIf') ans2 = self.handle.query_classid('adaptorEthGenProfile') vnics = { x.Name: { 'mac': x.Mac, 'uplink': x.UplinkPort, 'pci-slot': x.UsnicCount, 'dn': x.Dn, 'mtu': x.Mtu, 'name': x.Name, 'pxe-boot': x.PxeBoot } for x in ans1 } vlans_dict = { x.Dn.replace('/general', ''): { 'vlan': x.Vlan, 'vlan_mode': x.VlanMode } for x in ans2 } for vnic in vnics.values(): vlans = vlans_dict[vnic.get('dn')] vnic.update(vlans) return vnics def cimc_enable_sol(self): self.logger('enabling SOL') params = { 'dn': 'sys/rack-unit-1/sol-if', 'adminState': 'enable', 'speed': '115200' } self.cimc_set_mo_by_class_id(class_id='solIf', params=params) def create_storage(self, raid=RAID_1, disks_needed=2, clean_vds=False): """ :param clean_vds: Clean all virtual drives before creating any :param raid: Please select from '0','1','10','5','6' :param disks_needed: Number of disks needed :return: """ if raid not in [self.RAID_0, self.RAID_1, self.RAID_10]: raise ValueError( 'RAID request is not correct. Use one of the {0}. Got: {1}'. format(','.join([self.RAID_0, self.RAID_1, self.RAID_10]), raid)) virtual_drives_list = self.handle.query_classid( class_id='storageVirtualDrive') if virtual_drives_list: if clean_vds: self.logger('Cleaning Virtual Drives to create new one.') for vd in virtual_drives_list: self.handle.remove_mo(vd) else: self.logger('Virtual Drive already exists.') return disks = self.handle.query_classid('storageLocalDisk') # get 2 or more disks to form RAID disks_by_size = {} map( lambda x: disks_by_size.setdefault(x.get_attr('CoercedSize'), []). append(x), disks) available_disks = filter( lambda x: len(disks_by_size[x]) > disks_needed, disks_by_size.keys()) if len(available_disks) == 0: raise Exception( 'Not enough disks to build RAID {0}. Minimum required are {1}.' .format(raid, disks_needed)) size = available_disks[0] drive_group = ','.join( map(lambda x: x.Id, disks_by_size[size])[:disks_needed]) params = { 'raidLevel': raid, 'size': size, 'virtualDriveName': "RAID", 'dn': "sys/rack-unit-1/board/storage-SAS-SLOT-HBA/virtual-drive-create", 'driveGroup': '[{0}]'.format(drive_group), 'adminState': 'trigger', 'writePolicy': 'Write Through' } self.logger( 'Creating Virtual Drive RAID {0}. Using storage {0}'.format( raid, drive_group)) self.cimc_set_mo_by_class_id( class_id="storageVirtualDriveCreatorUsingUnusedPhysicalDrive", params=params) def cimc_delete_all_vnics(self): self.logger('deleting all vNICs') vnic_names = self.cimc_list_vnics().keys() for vnic_name in vnic_names: self.cimc_delete_vnic(vnic_name=vnic_name) def cimc_delete_vnic(self, vnic_name): dn = 'sys/rack-unit-1/adaptor-MLOM/host-eth-{}'.format(vnic_name) if 'eth0' in vnic_name or 'eth1' in vnic_name: # no way to delete eth0 or eth1, so reset them to default self.logger(message='Resetting vNIC ' + vnic_name) ethX = self.handle.query_dn(dn) ethX.mac = 'AUTO' ethX.mtu = 1500 ethX.pxe_boot = False ethX.uplink_port = vnic_name[-1] self.handle.set_mo(ethX) profile = self.handle.query_dn( 'sys/rack-unit-1/adaptor-{}/host-eth-{}/general'.format( pci_slot_id, name)) profile.vlan = None profile.order = 'ANY' self.handle.set_mo(profile) else: self.logger(message='Deleting vNIC ' + dn) vnic = self.handle.query_dn(dn=dn) self.handle.remove_mo(vnic) def cimc_set_ssh_timeout(self, timeout=3600): self.cmd('set_imc_managedobject', in_mo=None, class_id='commHttps', params={ 'Dn': 'sys/svc-ext/https-svc', 'sessionTimeout': str(timeout) }) def cimc_change_boot_order(self, pxe_order=1, hdd_order=2): self.logger('updating boot order: PXE as #{0}, HDD as #{1}'.format( pxe_order, hdd_order)) boot_configs = [{ 'params': { 'Dn': 'sys/rack-unit-1/boot-policy/lan-read-only', 'Order': pxe_order, 'Access': 'read-only' }, 'class_id': 'LsbootLan' }, { 'params': { 'Dn': 'sys/rack-unit-1/boot-policy/storage-read-write', 'Order': hdd_order, 'Access': 'read-write' }, 'class_id': 'LsbootStorage' }] for boot_config in boot_configs: boot_device = self.cmd('get_imc_managedobject', in_mo=None, class_id=None, params={'Dn': boot_config['params']['Dn']}) if boot_device: self.cmd('set_imc_managedobject', in_mo=boot_device, class_id=None, params=boot_config['params']) else: self.cmd('add_imc_managedobject', in_mo=None, class_id=boot_config['class_id'], params=boot_config['params']) def cimc_create_vnic(self, pci_slot_id, uplink_port, order, name, mac, vlan, is_pxe_enabled): from imcsdk.mometa.adaptor.AdaptorHostEthIf import AdaptorHostEthIf self.logger( message= 'creating vNIC {} on PCI id {} uplink {} mac={} vlan={} pxe: {}'. format(name, pci_slot_id, uplink_port, mac, vlan, is_pxe_enabled)) if name in [ 'eth0', 'eth1' ]: # eth0 and eth1 are default, only possible to modify there parameters, no way to rename ot delete ethX = self.handle.query_dn( 'sys/rack-unit-1/adaptor-{}/host-eth-{}'.format( pci_slot_id, name)) ethX.mac = mac ethX.mtu = 1500 ethX.pxe_boot = is_pxe_enabled self.handle.set_mo(ethX) else: vnic = AdaptorHostEthIf( parent_mo_or_dn='sys/rack-unit-1/adaptor-' + pci_slot_id, name=name, mac=mac, mtu=1500, pxe_boot=is_pxe_enabled, uplink_port=uplink_port) self.handle.add_mo(vnic) profile = self.handle.query_dn( 'sys/rack-unit-1/adaptor-{}/host-eth-{}/general'.format( pci_slot_id, name)) profile.vlan = vlan profile.order = order self.handle.set_mo(profile) def cimc_get_power_status(self): return self.handle.query_classid('computeRackUnit')[0].get_attr( 'OperPower') def _cimc_power(self, requested_state): import time self.logger('power {0}'.format(requested_state)) rack = self.handle.query_dn('sys/rack-unit-1') rack.admin_power = requested_state self.handle.set_mo(rack) time.sleep(120) # wait for the server to come up def cimc_power_down(self): current_power_state = self.cimc_get_power_status() if current_power_state == 'on': self._cimc_power('down') else: self.log(' is already OFF') def cimc_power_up(self): current_power_state = self.cimc_get_power_status() if current_power_state == 'off': self._cimc_power('up') else: self.log(' is already ON') def cimc_reset(self): pass # from ImcSdk.ImcMos import ComputeRackUnit # mo = handle.config_resolve_dn("sys/rack-unit-1") # mo.admin_power = ComputeRackUnit.CONST_ADMIN_POWER_BMC_RESET_IMMEDIATE # set_imc_managedobject(mo, class_id="ComputeRackUnit", params={ComputeRackUnit.ADMIN_POWER: # ComputeRackUnit.CONST_ADMIN_POWER_BMC_RESET_IMMEDIATE, # ComputeRackUnit.DN: "sys/rack-unit-1"}) def cimc_power_cycle(self): current_power_state = self.cimc_get_power_status() self._cimc_power('up' if current_power_state == 'off' else 'cycle-immediate') def cimc_recreate_vnics(self): actual_vnics = self.cimc_list_all_nics_and_vnics() actual_loms = [ x for x in self.cimc_list_pci_nic_ports() if 'L' in x['Dn'] ] for nic_order, nic in enumerate( self.lom_nics_dic.values()): # NIC order starts from 0 names = nic.get_names() macs = nic.get_macs() port_ids = nic.get_port_ids() for name, mac, port_id in zip(names, macs, port_ids): if port_id in ['LOM-1', 'LOM-2']: actual_mac = actual_loms[port_id]['mac'] if mac.upper() != actual_mac.upper(): raise ValueError( '{}: "{}" actual mac is "{}" while requested "{}". Edit lab config!' .format(self.id, port_id, actual_mac, mac)) else: if 'eth' not in self.get_nics() and nic.is_ssh( ): # if no NIC called eth and it's nic on ssh network, use default eth0, eth1 if name in actual_vnics: self.cimc_delete_vnic(vnic_name=name) name = 'eth' + name[-1] if name in actual_vnics: if mac == actual_vnics[name]['mac'] and str( nic.get_vlan_id()) == str( actual_vnics[name] ['vlan']): # this nic is already in CIMC self.logger( message='vNIC {} is already configured'.format( name)) if name in actual_vnics: actual_vnics.pop(name) continue else: self.logger( 'deleting {} since mac or vlan is not correct: {}' .format(name, actual_vnics[name])) self.cimc_delete_vnic(vnic_name=name) pci_slot_id, uplink_port = port_id.split('/') self.cimc_create_vnic(pci_slot_id=pci_slot_id, uplink_port=uplink_port, order='ANY', name=name, mac=mac, vlan=nic.get_vlan_id(), is_pxe_enabled=nic.is_pxe()) if name in actual_vnics: actual_vnics.pop(name) for vnic_name in actual_vnics.keys( ): # delete all actual which are not requested self.cimc_delete_vnic(vnic_name) def cimc_configure(self, is_debug=False): self._dump_xml = is_debug lab_type = self.pod.get_type() self.logger('configuring for {}'.format(lab_type)) self.cimc_set_hostname() self.cimc_power_up() if self.pod.get_type() == self.pod.MERCURY: self.cimc_disable_loms_in_bios() else: is_any_nic_on_lom = any( map(lambda x: x.is_on_lom(), self.get_nics().values())) if is_any_nic_on_lom: self.cimc_enable_loms_in_bios() self.cimc_recreate_vnics() self.cimc_change_boot_order(pxe_order=1, hdd_order=2) # if lab_type == self.lab().LAB_MERCURY: # self.create_storage('1', 2, True) def cimc_get_adaptors(self): r = self.handle.query_classid('AdaptorUnit') return r def cimc_set_mlom_adaptor(self, pci_slot, n_vnics): self.logger(message='allowing MLOM-{} to use up to {} vNICs'.format( pci_slot, n_vnics)) # self.cimc_set_mo_by_class_id(class_id='adaptorUnit', params={'dn': 'sys/rack-unit-1/adaptor-{0}'.format(pci_slot), 'description': str(self)}) self.cimc_set_mo_by_class_id( class_id='adaptorGenProfile', params={ 'dn': 'sys/rack-unit-1/adaptor-{0}/general'.format(pci_slot), 'fipMode': 'enabled', 'vntagMode': 'enabled', 'numOfVMFexIfs': n_vnics }) def cimc_get_mgmt_nic(self): return self.handle.query_classid('mgmtIf')[0] def cimc_set_hostname(self): current = self.cimc_get_mgmt_nic() new_cimc_hostname = '{}-ru{}-{}'.format(self.pod, self.hardware, self.id) if new_cimc_hostname != current.hostname: self.logger( message='setting hostname to {}'.format(new_cimc_hostname)) current.hostname = new_cimc_hostname self.handle.set_mo(current) else: self.logger( message='hostname is already {}'.format(new_cimc_hostname)) @staticmethod def cimc_deduce_wiring_by_lldp(pod): import yaml from lab.wire import Wire cimc_info_file_path = pod.get_artifact_file_path( 'cimc-{}.yaml'.format(pod)) try: with open(cimc_info_file_path) as f: cimc_info_dic = yaml.load(f.read()) except IOError: cimc_info_dic = {} for cimc in sorted(pod.cimc_servers_dic.values()): cimc_info_dic[cimc.oob_ip] = cimc.cimc_list_all_nics( ) # returns dic of {port_id: mac} with open(cimc_info_file_path, mode='w') as f: yaml.dump(cimc_info_dic, f) wires_cfg = [] for node in sorted(pod.cimc_servers_dic.values()): for port1, mac in cimc_info_dic[node.oob_ip].items(): neis = [ x.find_neighbour_with_mac(mac=mac, cimc_port_id=port1) for x in pod.switches ] neis = filter(lambda n: n is not None, neis) if neis: assert len( neis ) == 1, 'More then 1 switch is found connected to {} {}'.format( node, port1) nei = neis[0] wires_cfg.append({ 'node1': node.id, 'port1': port1, 'mac': mac, 'node2': nei.n9.id, 'port2': nei.port.port_id, 'pc-id': nei.port.pc_id }) else: wires_cfg.append({ 'node1': node.id, 'port1': port1, 'mac': mac, 'node2': None, 'port2': None, 'pc-id': None }) pod.wires.extend(Wire.add_wires(pod=pod, wires_cfg=wires_cfg))