def main(vm_name, guest_ip, host_ip, guest_primary_dns, path_logs): # Welcome print "\n--- AntiVMdetect by Mikael, @nsmfoo (modified by Jose Carlos R.) ---" print bcolors.OKGREEN + " [*]" + bcolors.ENDC + " Creating VirtualBox modifications .." dmi_info = {} for v in dmidecode.bios().values(): if type(v) == dict and v["dmi_type"] == 0: dmi_info["DmiBIOSVendor"] = v["data"]["Vendor"] dmi_info["DmiBIOSReleaseDate"] = v["data"]["Relase Date"] dmi_info["DmiBIOSVersion"] = v["data"]["Version"] biosversion = v["data"]["BIOS Revision"] try: dmi_info["DmiBIOSReleaseMajor"], dmi_info["DmiBIOSReleaseMinor"] = biosversion.split(".", 1) except: dmi_info["DmiBIOSReleaseMajor"] = "** No value to retrieve **" dmi_info["DmiBIOSReleaseMinor"] = "** No value to retrieve **" # python-dmidecode does not currently reveal all values .. this is plan B dmi_firmware = commands.getoutput("dmidecode -t0") try: dmi_info["DmiBIOSFirmwareMajor"], dmi_info["DmiBIOSFirmwareMinor"] = ( re.search("Firmware Revision: ([0-9A-Za-z. ]*)", dmi_firmware).group(1).split(".", 1) ) except: dmi_info["DmiBIOSFirmwareMajor"] = "** No value to retrieve **" dmi_info["DmiBIOSFirmwareMinor"] = "** No value to retrieve **" for v in dmidecode.baseboard().values(): if type(v) == dict and v["dmi_type"] == 2: serial_number = v["data"]["Serial Number"] dmi_info["DmiBoardVersion"] = v["data"]["Version"] if isinstance(v["data"]["Product Name"], int): dmi_info["DmiBoardProduct"] = str(v["data"]["Product Name"]) + " " else: dmi_info["DmiBoardProduct"] = v["data"]["Product Name"] dmi_info["DmiBoardVendor"] = v["data"]["Manufacturer"] # This is hopefully not the best solution .. try: s_number = [] if serial_number: # Get position if "/" in serial_number: for slash in re.finditer("/", serial_number): s_number.append(slash.start(0)) # Remove / from string new_serial = re.sub("/", "", serial_number) new_serial = serial_randomize(0, len(new_serial)) # Add / again for char in s_number: new_serial = new_serial[:char] + "/" + new_serial[char:] else: new_serial = serial_randomize(0, len(serial_number)) else: new_serial = "** No value to retrieve **" except: new_serial = "** No value to retrieve **" dmi_info["DmiBoardSerial"] = "string:" + new_serial # python-dmidecode does not reveal all values .. this is plan B dmi_board = commands.getoutput("dmidecode -t2") try: asset_tag = re.search("Asset Tag: ([0-9A-Za-z ]*)", dmi_board).group(1) except: asset_tag = "** No value to retrieve **" dmi_info["DmiBoardAssetTag"] = asset_tag try: loc_chassis = re.search("Location In Chassis: ([0-9A-Za-z ]*)", dmi_board).group(1) except: loc_chassis = "** No value to retrieve **" dmi_info["DmiBoardLocInChass"] = loc_chassis # Based on the list from http://www.dmtf.org/sites/default/files/standards/documents/DSP0134_3.0.0.pdf board_dict = { "Unknown": 1, "Other": 2, "Server Blade": 3, "Connectivity Switch": 4, "System Management Module": 5, "Processor Module": 6, "I/O Module": 7, "Memory Module": 8, "Daughter board": 9, "Motherboard": 10, "Processor/Memory Module": 11, "Processor/IO Module": 12, "Interconnect board": 13, } try: board_type = re.search("Type: ([0-9A-Za-z ]+)", dmi_board).group(1) board_type = str(board_dict.get(board_type)) except: board_type = "** No value to retrieve **" dmi_info["DmiBoardBoardType"] = board_type for v in dmidecode.system().values(): if type(v) == dict and v["dmi_type"] == 1: dmi_info["DmiSystemSKU"] = v["data"]["SKU Number"] system_family = v["data"]["Family"] system_serial = v["data"]["Serial Number"] dmi_info["DmiSystemVersion"] = "string:" + v["data"]["Version"] dmi_info["DmiSystemProduct"] = v["data"]["Product Name"] dmi_info["DmiSystemVendor"] = v["data"]["Manufacturer"] if not system_family: dmi_info["DmiSystemFamily"] = "Not Specified" else: dmi_info["DmiSystemFamily"] = system_family # Create a new UUID newuuid = str(uuid.uuid4()) dmi_info["DmiSystemUuid"] = newuuid.upper() # Create a new system serial number dmi_info["DmiSystemSerial"] = "string:" + (serial_randomize(0, len(system_serial))) for v in dmidecode.chassis().values(): dmi_info["DmiChassisVendor"] = v["data"]["Manufacturer"] chassi_serial = v["data"]["Serial Number"] dmi_info["DmiChassisVersion"] = v["data"]["Version"] dmi_info["DmiChassisType"] = v["data"]["Type"] # Based on the list from http://www.dmtf.org/sites/default/files/standards/documents/DSP0134_3.0.0.pdf chassi_dict = { "Other": 1, "Unknown": 2, "Desktop": 3, "Low Profile Desktop": 4, "Pizza Box": 5, "Mini Tower": 6, "Tower": 7, "Portable": 8, "Laptop": 9, "Notebook": 10, "Hand Held": 11, "Docking Station": 12, "All in One": 13, "Sub Notebook": 14, "Space-saving": 15, "Lunch Box": 16, "Main Server Chassis": 17, "Expansion Chassis": 18, "SubChassis": 19, "Bus Expansion Chassis": 20, "Peripheral Chassis": 21, } dmi_info["DmiChassisType"] = str(chassi_dict.get(dmi_info["DmiChassisType"])) # python-dmidecode does not reveal all values .. this is plan B chassi = commands.getoutput("dmidecode -t3") try: dmi_info["DmiChassisAssetTag"] = re.search("Asset Tag: ([0-9A-Za-z ]*)", chassi).group(1) except: dmi_info["DmiChassisAssetTag"] = "** No value to retrieve **" # Create a new chassi serial number, added string to make it be taken as a string even if it's a number dmi_info["DmiChassisSerial"] = "string:" + str(serial_randomize(0, len(chassi_serial))) for v in dmidecode.processor().values(): dmi_info["DmiProcVersion"] = v["data"]["Version"] dmi_info["DmiProcManufacturer"] = v["data"]["Manufacturer"]["Vendor"] # OEM strings try: for v in dmidecode.type(11).values(): oem_ver = v["data"]["Strings"]["3"] oem_rev = v["data"]["Strings"]["2"] except: pass try: dmi_info["DmiOEMVBoxVer"] = oem_ver dmi_info["DmiOEMVBoxRev"] = oem_rev except: dmi_info["DmiOEMVBoxVer"] = "** No value to retrieve **" dmi_info["DmiOEMVBoxRev"] = "** No value to retrieve **" # Write all data collected so far to file file_name = "vboxmods-" + vm_name.replace(" ", "_") + ".sh" logfile = file(path_logs + "/" + file_name, "w+") logfile.write("# Generated on: " + time.strftime("%H:%M:%S") + "\n") for k, v in sorted(dmi_info.iteritems()): if "** No value to retrieve **" in v: logfile.write( '# VBoxManage setextradata "' + vm_name + '" VBoxInternal/Devices/pcbios/0/Config/' + k + "\t" + v + "\n" ) else: logfile.write( 'VBoxManage setextradata "' + vm_name + '" VBoxInternal/Devices/pcbios/0/Config/' + k + "\t'" + v + "'\n" ) # Disk information disk_dmi = {} try: if os.path.exists("/dev/sda"): # Disk serial disk_serial = commands.getoutput( "hdparm -i /dev/sda | grep -o 'SerialNo=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'" ) # To avoid exceding 20 bytes serials if len(disk_serial) > 20: disk_dmi["SerialNumber"] = serial_randomize(0, 20) else: disk_dmi["SerialNumber"] = serial_randomize(0, len(disk_serial)) # Check for HP Legacy RAID elif os.path.exists("/dev/cciss/c0d0"): # Needs smartctl to be able to get the correct information if os.path.exists("/usr/sbin/smartctl"): hp_old_raid = commands.getoutput("smartctl -d cciss,1 -i /dev/cciss/c0d0") disk_serial = re.search("Serial number:([0-9A-Za-z ]*)", hp_old_raid).group(1).replace(" ", "") # To avoid exceding 20 bytes serials if len(disk_serial) > 20: disk_dmi["SerialNumber"] = serial_randomize(0, 20) else: disk_dmi["SerialNumber"] = serial_randomize(0, len(disk_serial)) else: print bcolors.WARNING + "Install smartmontools: apt-get install smartmontools" + bcolors.ENDC except OSError: print "Haz RAID?" print commands.getoutput("lspci | grep -i raid") # Disk firmeware rev try: if os.path.exists("/dev/sda"): disk_fwrev = commands.getoutput( "hdparm -i /dev/sda | grep -o 'FwRev=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'" ) disk_dmi["FirmwareRevision"] = disk_fwrev if len(disk_dmi["FirmwareRevision"]) < 8: disk_dmi["FirmwareRevision"] = disk_dmi["FirmwareRevision"][:8] elif os.path.exists("/dev/cciss/c0d0"): # Needs smartctl to be able to get the correct information if os.path.exists("/usr/sbin/smartctl"): hp_old_raid = commands.getoutput("smartctl -d cciss,1 -i /dev/cciss/c0d0") disk_dmi["FirmwareRevision"] = ( re.search("Revision:([0-9A-Za-z ]*)", hp_old_raid).group(1).replace(" ", "") ) if len(disk_dmi["FirmwareRevision"]) < 8: disk_dmi["FirmwareRevision"] = disk_dmi["FirmwareRevision"][:8] else: print "Install smartmontools: apt-get install smartmontools" except OSError: print "Haz RAID?" print commands.getoutput("lspci | grep -i raid") # Disk Model number try: if os.path.exists("/dev/sda"): disk_modelno = commands.getoutput( "hdparm -i /dev/sda | grep -o 'Model=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'" ) disk_dmi["ModelNumber"] = disk_modelno elif os.path.exists("/dev/cciss/c0d0"): # Needs smartctl to be able to get the correct information if os.path.exists("/usr/sbin/smartctl"): hp_old_raid = commands.getoutput("smartctl -d cciss,1 -i /dev/cciss/c0d0") disk_dmi["ModelNumber"] = re.search("Product:([0-9A-Za-z ]*)", hp_old_raid).group(1).replace(" ", "") else: print "Install smartmontools: apt-get install smartmontools" except OSError: print "Haz RAID?" print commands.getoutput("lspci | grep -i raid") # Write more things to file for k, v in disk_dmi.iteritems(): if "** No value to retrieve **" in v: logfile.write( '# VBoxManage setextradata "' + vm_name + '" VBoxInternal/Devices/piix3ide/0/Config/PrimaryMaster/' + k + "\t" + v + "\n" ) else: logfile.write( 'VBoxManage setextradata "' + vm_name + '" VBoxInternal/Devices/piix3ide/0/Config/PrimaryMaster/' + k + "\t'" + v + "'\n" ) # CD-ROM information cdrom_dmi = {} if os.path.islink("/dev/cdrom"): # CD-ROM serial cdrom_serial = commands.getoutput( "hdparm -i /dev/cdrom | grep -o 'SerialNo=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'" ) if cdrom_serial: cdrom_dmi["ATAPISerialNumber"] = serial_randomize(0, len(cdrom_serial)) else: cdrom_dmi["ATAPISerialNumber"] = "** No value to retrieve **" # CD-ROM firmeware rev cdrom_fwrev = commands.getoutput("cd-drive | grep Revision | grep ':' | awk {' print $3 \" \" $4'}") cdrom_dmi["ATAPIRevision"] = cdrom_fwrev.replace(" ", "") # CD-ROM Model numberA-Za-z0-9_\+\/ .\"- cdrom_modelno = commands.getoutput("cd-drive | grep Model | grep ':' | awk {' print $3 \" \" $4'}") cdrom_dmi["ATAPIProductId"] = cdrom_modelno # CD-ROM Vendor cdrom_vendor = commands.getoutput("cd-drive | grep Vendor | grep ':' | awk {' print $3 '}") cdrom_dmi["ATAPIVendorId"] = cdrom_vendor else: logfile.write("# No CD-ROM detected: ** No values to retrieve **\n") # And some more for k, v in cdrom_dmi.iteritems(): if "** No value to retrieve **" in v: logfile.write( '# VBoxManage setextradata "' + vm_name + '" VBoxInternal/Devices/piix3ide/0/Config/SecondaryMaster/' + k + "\t" + v + "\n" ) else: logfile.write( 'VBoxManage setextradata "' + vm_name + '" VBoxInternal/Devices/piix3ide/0/Config/SecondaryMaster/' + k + "\t'" + v + "'\n" ) # Get the DSDT image # os.system("dd if=/sys/firmware/acpi/tables/DSDT of=DSDT.bin >/dev/null 2>&1") # Write to file # Da un error de que la tabla ACPI es mayor de 64KB, aunque lo comente, si luego ejecuto el batch dentro del Guest, pasa las pruebas # logfile.write('VBoxManage setextradata '+vm_name+' VBoxInternal/Devices/acpi/0/Config/CustomTable\t\'' + os.getcwd() + '/DSDT.bin\'\n') acpi_misc = commands.getoutput('acpidump -s | grep DSDT | grep -o "\(([A-Za-z0-9].*)\)" | tr -d "()"') acpi_list = acpi_misc.split(" ") acpi_list = filter(None, acpi_list) logfile.write( 'VBoxManage setextradata "' + vm_name + "\" VBoxInternal/Devices/acpi/0/Config/AcpiOemId\t'" + acpi_list[1] + "'\n" ) logfile.write( 'VBoxManage setextradata "' + vm_name + "\" VBoxInternal/Devices/acpi/0/Config/AcpiCreatorId\t'" + acpi_list[4] + "'\n" ) logfile.write( 'VBoxManage setextradata "' + vm_name + "\" VBoxInternal/Devices/acpi/0/Config/AcpiCreatorRev\t'" + acpi_list[5] + "'\n" ) # Randomize MAC address, based on onboard interface MAC mac_seed = ":".join(re.findall("..", "%012x" % uuid.getnode()))[0:9] big_mac = mac_seed + "%02x:%02x:%02x" % (random.randint(0, 255), random.randint(0, 255), random.randint(0, 255)) le_big_mac = re.sub(":", "", big_mac) # The last thing! logfile.write('VBoxManage modifyvm "' + vm_name + '" --macaddress1\t' + le_big_mac + "\n") # Done! logfile.close() print " Finished: A template shell script has been created named:", file_name print bcolors.OKGREEN + "\n [*]" + bcolors.ENDC + " Creating guest based modification file (to be run inside the guest)" # Write all data to file file_name2 = "vboxmods-" + vm_name.replace(" ", "_") + ".bat" logfile = file("/srv/ftp/CopyThisOne!/" + file_name2, "w+") manu = acpi_list[1] # DSDT logfile.write("@reg copy HKLM\HARDWARE\ACPI\DSDT\VBOX__ HKLM\HARDWARE\ACPI\DSDT\\" + manu + " /s /f\n\n") logfile.write("@reg delete HKLM\HARDWARE\ACPI\DSDT\VBOX__ /f\n\n") logfile.write( "@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\" + manu + "\VBOXBIOS HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\" + manu + "\\" + acpi_list[2] + "___" + " /s /f\n\n" ) logfile.write("@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\" + manu + "\VBOXBIOS /f\n\n") logfile.write( "@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\" + manu + "\\" + acpi_list[2] + "___\\00000002 HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\" + manu + "\\" + acpi_list[2] + "___\\" + acpi_list[3] + " /s /f\n\n" ) logfile.write( "@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\" + manu + "\\" + acpi_list[2] + "___\\00000002 /f\n" ) # FADT logfile.write( "@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\" + manu + "\VBOXFACP HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\" + manu + "\\" + acpi_list[2] + "___ /s /f\n\n" ) logfile.write("@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\" + manu + "\VBOXFACP /f\n") logfile.write( "@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\" + manu + "\\" + acpi_list[2] + "___\\00000001 HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\" + manu + "\\" + acpi_list[2] + "___\\" + acpi_list[3] + " /s /f\n\n" ) logfile.write( "@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\" + manu + "\\" + acpi_list[2] + "___\\00000001 /f\n\n" ) # RSDT logfile.write( "@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\" + manu + "\VBOXRSDT HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\" + manu + "\\" + acpi_list[2] + "___ /s /f\r\n" ) logfile.write("@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\" + manu + "\VBOXRSDT /f\r\n") logfile.write( "@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\" + manu + "\\" + acpi_list[2] + "___\\00000001 HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\" + manu + "\\" + acpi_list[2] + "___\\" + acpi_list[3] + " /s /f\r\n" ) logfile.write( "@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\" + manu + "\\" + acpi_list[2] + "___\\00000001 /f\r\n" ) # SystemBiosVersion - TODO: get real values logfile.write( '@reg add HKEY_LOCAL_MACHINE\HARDWARE\DESCRIPTION\System /v SystemBiosVersion /t REG_MULTI_SZ /d "' + acpi_list[1] + " - " + acpi_list[0] + '" /f\n\n' ) # VideoBiosVersion - TODO: get real values logfile.write( '@reg add HKEY_LOCAL_MACHINE\HARDWARE\DESCRIPTION\System /v VideoBiosVersion /t REG_MULTI_SZ /d "' + acpi_list[0] + '" /f\n' ) # SystemBiosDate d_month, d_day, d_year = dmi_info["DmiBIOSReleaseDate"].split("/") if len(d_year) > 2: d_year = d_year[:2] logfile.write( '@reg add HKEY_LOCAL_MACHINE\HARDWARE\DESCRIPTION\System /v SystemBiosDate /t REG_MULTI_SZ /d "' + d_month + "/" + d_day + "/" + d_year + '" /f\n' ) # Prevent WMI identification logfile.write( '@reg add HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\services\PlugPlay /v Start /t REG_MULTI_SZ /d "4" /f\r\n' ) # The system have to be rebooted for this to work, don't know why # Disables FW logfile.write("netsh firewall set opmode disable\n") # Disables windows updates logfile.write( '@reg add "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate\Auto Update" /v AUOptions /t REG_DWORD /d 1 /f\n' ) # Change the IP and DNS shortScript = ( """ @echo off set _enabled=Habilitado set _dedicated=Dedicado for /f "tokens=1,2,3*" %%i in ('netsh int show interface') do ( if %%i equ %_enabled% ( if %%j equ %_dedicated% ( netsh int ip set address "%%k %%l" static """ + guest_ip + """ 255.255.255.0 """ + host_ip + """ 1 netsh int ip set dns "%%k %%l" static """ + guest_primary_dns + """ ) ) ) ipconfig /flushdns :EOF""" ) logfile.write(shortScript + "\n") logfile.close() print " Finished: A Windows batch file has been created named:", file_name2 return
def ismobile(self): ''' Returns a bool indicating whether or not the system in question is a laptop. The is mobile method is used by some rules that have alternate settings for laptops. @author: dkennel @regturn: bool - true if system is a laptop ''' ismobile = False dmitypes = ['LapTop', 'Portable', 'Notebook', 'Hand Held', 'Sub Notebook'] if DMI and self.euid == 0: try: chassis = dmidecode.chassis() for key in chassis: chassistype = chassis[key]['data']['Type'] if chassistype in dmitypes: ismobile = True except(IndexError, KeyError): # got unexpected data back from dmidecode pass elif os.path.exists('/usr/sbin/system_profiler'): profilerfetch = '/usr/sbin/system_profiler SPHardwareDataType' cmd3 = subprocess.Popen(profilerfetch, shell=True, stdout=subprocess.PIPE, close_fds=True) cmd3output = cmd3.stdout.readlines() for line in cmd3output: if re.search('Book', line): ismobile = True break return ismobile
def get_property_number(self): """ Find and return the Property number of the local machine @author: scmcleni @author: D. Kennel @return: int """ propnum = 0 try: if os.path.exists('/etc/property-number'): propertynumberfile = open('/etc/property-number', 'r') propnum = propertynumberfile.readline() propnum = propnum.strip() propertynumberfile.close() elif DMI and self.euid == 0: chassis = dmidecode.chassis() for key in chassis: propnum = chassis[key]['data']['Asset Tag'] if platform.system() == 'Darwin': pnfetch = '/usr/sbin/nvram asset_id 2>/dev/null' cmd = subprocess.Popen(pnfetch, shell=True, stdout=subprocess.PIPE, close_fds=True) cmdout = cmd.stdout.readline() cmdout = cmdout.split() try: propnum = cmdout[1] except(IndexError, KeyError): propnum = 0 except: pass # Failed to obtain property number return propnum
def get_property_number(self): """ Find and return the Property number of the local machine @author: scmcleni @author: D. Kennel @return: int """ propnum = 0 try: if os.path.exists('/etc/property-number'): propertynumberfile = open('/etc/property-number', 'r') propnum = propertynumberfile.readline() propnum = propnum.strip() propertynumberfile.close() elif DMI and self.euid == 0: chassis = dmidecode.chassis() for key in chassis: propnum = chassis[key]['data']['Asset Tag'] if platform.system() == 'Darwin': pnfetch = '/usr/sbin/nvram asset_id 2>/dev/null' cmd = subprocess.Popen(pnfetch, shell=True, stdout=subprocess.PIPE, close_fds=True) cmdout = cmd.stdout.readline() cmdout = cmdout.split() try: propnum = cmdout[1] except (IndexError, KeyError): propnum = 0 except: pass # Failed to obtain property number return propnum
def process_recommend_file(self, fname): matching_profile = None try: if not os.path.isfile(fname): return None config = ConfigObj(fname, list_values = False, interpolation = False) for section in list(config.keys()): match = True for option in list(config[section].keys()): value = config[section][option] if value == "": value = r"^$" if option == "virt": if not re.match(value, self.execute("virt-what")[1], re.S): match = False elif option == "system": if not re.match(value, self.read_file(consts.SYSTEM_RELEASE_FILE), re.S): match = False elif option[0] == "/": if not os.path.exists(option) or not re.match(value, self.read_file(option), re.S): match = False elif option[0:7] == "process": ps = procfs.pidstats() ps.reload_threads() if len(ps.find_by_regex(re.compile(value))) == 0: match = False elif option == "chassis_type": for chassis in dmidecode.chassis().values(): chassis_type = chassis["data"]["Type"] if re.match(value, chassis_type, re.IGNORECASE): break else: match = False elif option == "syspurpose_role": if have_syspurpose: s = syspurpose.files.SyspurposeStore( syspurpose.files.USER_SYSPURPOSE, raise_on_error = True) role = "" try: s.read_file() role = s.contents["role"] except (IOError, OSError, KeyError) as e: if hasattr(e, "errno") and e.errno != errno.ENOENT: log.error("Failed to load the syspurpose file: %s" % e) if re.match(value, role, re.IGNORECASE) is None: match = False else: log.error("Failed to process 'syspurpose_role' in '%s', the syspurpose module is not available" % fname) if match: # remove the ",.*" suffix r = re.compile(r",[^,]*$") matching_profile = r.sub("", section) break except (IOError, OSError, ConfigObjError) as e: log.error("error processing '%s', %s" % (fname, e)) return matching_profile
def get_chassis_serial_number(self): """ Find and return the Chassis serial number @author: dkennel @requires: string """ chassisserial = '0' if DMI and self.euid == 0: try: chassis = dmidecode.chassis() for key in chassis: chassisserial = chassis[key]['data']['Serial Number'] except(IndexError, KeyError): # got unexpected data back from dmidecode pass chassisserial = chassisserial.strip() return chassisserial
def get_chassis_manfacturer(self): """ Find and return the Chassis manufacterer @author: D. Kennel @return: string """ chassismfr = 'Unk' if DMI and self.euid == 0: try: chassis = dmidecode.chassis() for key in chassis: chassismfr = chassis[key]['data']['Manufacturer'] except(IndexError, KeyError): # got unexpected data back from dmidecode pass chassismfr = chassismfr.strip() return chassismfr
def get_chassis_manfacturer(self): """ Find and return the Chassis manufacterer @author: D. Kennel @return: string """ chassismfr = 'Unk' if DMI and self.euid == 0: try: chassis = dmidecode.chassis() for key in chassis: chassismfr = chassis[key]['data']['Manufacturer'] chassismfr = chassismfr.strip() except (IndexError, KeyError): # got unexpected data back from dmidecode pass return chassismfr
def testGetChassisSerNo(self): chassisserial = '0' if DMI: if self.euid == 0: try: chassis = dmidecode.chassis() for key in chassis: chassisserial = chassis[key]['data']['Serial Number'] except (IndexError, KeyError): # got unexpected data back from dmidecode pass unittestchassisserial = chassisserial.strip() envchassisserial = self.to.get_chassis_serial_number() self.assertEqual(unittestchassisserial, envchassisserial) print(('Ser: ' + self.to.get_chassis_serial_number())) else: msg = "Not running as root, Chassis serial number " + \ "information not availble\n" self.skipTest(msg) else: msg = "dmidecode module not available for import. " + \ "Unable to retrieve serial chassis number\n" self.skipTest(msg)
def testGetChassisSerNo(self): chassisserial = '0' if DMI: if self.euid == 0: try: chassis = dmidecode.chassis() for key in chassis: chassisserial = chassis[key]['data']['Serial Number'] except(IndexError, KeyError): # got unexpected data back from dmidecode pass unittestchassisserial = chassisserial.strip() envchassisserial = self.to.get_chassis_serial_number() self.assertEqual(unittestchassisserial, envchassisserial) print 'Ser: ' + self.to.get_chassis_serial_number() else: msg = "Not running as root, Chassis serial number " + \ "information not availble\n" self.skipTest(msg) else: msg = "dmidecode module not available for import. " + \ "Unable to retrieve serial chassis number\n" self.skipTest(msg)
dmi_info['DmiSystemVersion'] = "string:" + v['data']['Version'] dmi_info['DmiSystemProduct'] = v['data']['Product Name'] dmi_info['DmiSystemVendor'] = v['data']['Manufacturer'] if not system_family: dmi_info['DmiSystemFamily'] = "Not Specified" else: dmi_info['DmiSystemFamily'] = system_family # Create a new UUID newuuid = str(uuid.uuid4()) dmi_info['DmiSystemUuid'] = newuuid.upper() # Create a new system serial number dmi_info['DmiSystemSerial'] = (serial_randomize(0, len(system_serial))) for v in dmidecode.chassis().values(): dmi_info['DmiChassisVendor'] = v['data']['Manufacturer'] chassi_serial = v['data']['Serial Number'] dmi_info['DmiChassisVersion'] = v['data']['Version'] dmi_info['DmiChassisType'] = v['data']['Type'] # Based on the list from http://www.dmtf.org/sites/default/files/standards/documents/DSP0134_3.0.0.pdf chassi_dict = { 'Other': 1, 'Unknown': 2, 'Desktop': 3, 'Low Profile Desktop': 4, 'Pizza Box': 5, 'Mini Tower': 6, 'Tower': 7, 'Portable': 8,
def __init__(self): #Get all required system info for v in dmidecode.system().values(): if type(v) == dict and v['dmi_type'] == 1: self.Mfg = str((v['data']['Manufacturer'])) self.Model = str((v['data']['Product Name'])) self.Serial = str((v['data']['Serial Number'])) for x in dmidecode.chassis().values(): if type(x) == dict and x['dmi_type'] == 3: self.FFactor = str((x['data']['Type'])) for x in dmidecode.processor().values(): if type(x) == dict and x['dmi_type'] == 4: self.CPUMfg = str((x['data']['Manufacturer'])) self.CPUFam = str((x['data']['Family'])) self.CPUVer = str((x['data']['Version'])) self.CPUFrq = str((x['data']['Current Speed'])) for x in dmidecode.memory().values(): if type(x) == dict and x['dmi_type'] == 17: self.MEMSize = str((x['data']['Size'])) self.MEMType = str((x['data']['Type'])) self.MEMDeta = str((x['data']['Type Detail'])) self.MEMSpeed = str((x['data']['Speed'])) for x in dmidecode.baseboard().values(): if type(x) == dict and x['dmi_type'] == 10: if str((x['data']['dmi_on_board_devices'][0]['Type'])) == "Video": self.Video = str((x['data']['dmi_on_board_devices'][0]['Description'])) if str((x['data']['dmi_on_board_devices'][0]['Type'])) == "Sound": self.Audio = str((x['data']['dmi_on_board_devices'][0]['Description'])) if str((x['data']['dmi_on_board_devices'][0]['Type'])) == "Ethernet": self.Network = str((x['data']['dmi_on_board_devices'][0]['Description'])) ### Workarounds for things python-dmidecode doesn't do command = "grep \"model name\" /proc/cpuinfo | uniq | awk -F\" \" {' print $4 $5 \" \" $6 \" \" $8 '} | sed 's/(.\{1,2\})/ /g'" self.CPUName = subprocess.check_output(command, shell=True).strip() command = "MSiz=0; for Size in $(dmidecode -t 17 | grep \"Size:\" | awk {' print $2 '}); do MSiz=$(($MSiz+$Size)); done; echo $MSiz" self.MEMSize = subprocess.check_output(command, shell=True).strip() command = "dmidecode -t 17 | grep \"Type:\" | awk {' print $2 '} | uniq" self.MEMType = subprocess.check_output(command, shell=True).strip() if self.Video is None: command = "lspci | grep VGA | awk -F: {' print $3 '}" self.Video = subprocess.check_output(command, shell=True).strip() if self.Audio is None: command = "lspci | grep Audio | awk -F: {' print $3 '}" self.Audio = subprocess.check_output(command, shell=True).strip() if self.Network is None: command = "lspci | grep Ethernet | awk -F: {' print $3 '}" self.Network = subprocess.check_output(command, shell=True).strip() command = "lspci | grep 802.11 | awk -F: {' print $3 '}" self.WiFi = subprocess.check_output(command, shell=True).strip() command = "echo \"Not Yet Implemented\"" self.Battery = subprocess.check_output(command, shell=True).strip() ### Get hard drive info self.Drive = [] command = "lshw -C disk" CMDOutput = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE) Type = None Model = None Vendor = None Device = None Serial = None Size = 0 SectSize = 0 for line in CMDOutput.stdout: if "*-" in line: #print("Found: "+line) Line = line.split('-', 1) Type = Line[1].strip() if "product:" in line: #print("Found: "+line) Line = line.split(':', 1) Model = Line[1].strip() if "vendor:" in line: #print("Found: "+line) Line = line.split(':', 1) Vendor = Line[1].strip() if "logical name:" in line: #print("Found: "+line) Line = line.split(':', 1) Device = Line[1].strip() if "serial:" in line: #print("Found: "+line) Line = line.split(':', 1) Serial = Line[1].strip() if "size:" in line: command = "fdisk -l /dev/sda | grep \""+Device+":\" | awk {' print $5 '}" Size = subprocess.check_output(command, shell=True).strip() #print("Size: "+Size) command = "fdisk -l /dev/sda | grep \"Sector\" | awk {' print $4 '}" SectSize = subprocess.check_output(command, shell=True).strip() #print("SectSize: "+SectSize) if "configuration:" in line: Dev = Device.split('/', 2) if (Type == "cdrom"): self.Optical = Vendor+" "+Model else: if Vendor is None: Line = Model.split(' ', 1) Vendor = Line[0].strip() Model = Line[1].strip() self.Drive.append(self.Disk(Type, Model, Vendor, Device, Serial, Size, SectSize)) Type = None Model = None Vendor = None Device = None Serial = None Size = 0 SectSize = 0
dmi_info['DmiSystemVersion'] = "string:" + v['data']['Version'] dmi_info['DmiSystemProduct'] = v['data']['Product Name'] dmi_info['DmiSystemVendor'] = v['data']['Manufacturer'] if not system_family: dmi_info['DmiSystemFamily'] = "Not Specified" else: dmi_info['DmiSystemFamily'] = system_family # Create a new UUID newuuid = str(uuid.uuid4()) dmi_info['DmiSystemUuid'] = newuuid.upper() # Create a new system serial number dmi_info['DmiSystemSerial'] = (serial_randomize(0, len(system_serial))) for v in dmidecode.chassis().values(): dmi_info['DmiChassisVendor'] = v['data']['Manufacturer'] chassi_serial = v['data']['Serial Number'] dmi_info['DmiChassisVersion'] = v['data']['Version'] dmi_info['DmiChassisType'] = v['data']['Type'] # Based on the list from http://www.dmtf.org/sites/default/files/standards/documents/DSP0134_3.0.0.pdf chassi_dict = {'Other': 1, 'Unknown': 2, 'Desktop': 3, 'Low Profile Desktop': 4, 'Pizza Box': 5, 'Mini Tower': 6, 'Tower': 7, 'Portable': 8, 'Laptop': 9, 'Notebook': 10, 'Hand Held': 11, 'Docking Station': 12, 'All in One': 13, 'Sub Notebook': 14, 'Space-saving': 15, 'Lunch Box': 16, 'Main Server Chassis': 17, 'Expansion Chassis': 18, 'SubChassis': 19, 'Bus Expansion Chassis': 20, 'Peripheral Chassis': 21, 'RAID Chassis': 22, 'Rack Mount Chassis': 23, 'Sealed-case PC': 24, 'Multi-system chassis': 25, 'Compact PCI': 26, 'Advanced TCA': 27, 'Blade': 28, 'Blade Enclosure': 29, 'Tablet': 30, 'Convertible': 31, 'Detachable': 32} dmi_info['DmiChassisType'] = str(chassi_dict.get(dmi_info['DmiChassisType']))
def main(vm_name, guest_ip, host_ip, guest_primary_dns, path_logs): # Welcome print '\n--- AntiVMdetect by Mikael, @nsmfoo (modified by Jose Carlos R.) ---' print bcolors.OKGREEN+' [*]'+bcolors.ENDC+' Creating VirtualBox modifications ..' dmi_info = {} for v in dmidecode.bios().values(): if type(v) == dict and v['dmi_type'] == 0: dmi_info['DmiBIOSVendor'] = v['data']['Vendor'] dmi_info['DmiBIOSReleaseDate'] = v['data']['Relase Date'] dmi_info['DmiBIOSVersion'] = v['data']['Version'] biosversion = v['data']['BIOS Revision'] try: dmi_info['DmiBIOSReleaseMajor'], dmi_info['DmiBIOSReleaseMinor'] = biosversion.split('.', 1) except: dmi_info['DmiBIOSReleaseMajor'] = '** No value to retrieve **' dmi_info['DmiBIOSReleaseMinor'] = '** No value to retrieve **' # python-dmidecode does not currently reveal all values .. this is plan B dmi_firmware = commands.getoutput("dmidecode -t0") try: dmi_info['DmiBIOSFirmwareMajor'], dmi_info['DmiBIOSFirmwareMinor'] = re.search( "Firmware Revision: ([0-9A-Za-z. ]*)", dmi_firmware).group(1).split('.', 1) except: dmi_info['DmiBIOSFirmwareMajor'] = '** No value to retrieve **' dmi_info['DmiBIOSFirmwareMinor'] = '** No value to retrieve **' for v in dmidecode.baseboard().values(): if type(v) == dict and v['dmi_type'] == 2: serial_number = v['data']['Serial Number'] dmi_info['DmiBoardVersion'] = v['data']['Version'] if isinstance(v['data']['Product Name'], int ): dmi_info['DmiBoardProduct'] = str(v['data']['Product Name'])+ ' ' else: dmi_info['DmiBoardProduct'] = v['data']['Product Name'] dmi_info['DmiBoardVendor'] = v['data']['Manufacturer'] # This is hopefully not the best solution .. try: s_number = [] if serial_number: # Get position if '/' in serial_number: for slash in re.finditer('/', serial_number): s_number.append(slash.start(0)) # Remove / from string new_serial = re.sub('/', '', serial_number) new_serial = serial_randomize(0, len(new_serial)) # Add / again for char in s_number: new_serial = new_serial[:char] + '/' + new_serial[char:] else: new_serial = serial_randomize(0, len(serial_number)) else: new_serial = "** No value to retrieve **" except: new_serial = "** No value to retrieve **" dmi_info['DmiBoardSerial'] = "string:"+new_serial # python-dmidecode does not reveal all values .. this is plan B dmi_board = commands.getoutput("dmidecode -t2") try: asset_tag = re.search("Asset Tag: ([0-9A-Za-z ]*)", dmi_board).group(1) except: asset_tag = '** No value to retrieve **' dmi_info['DmiBoardAssetTag'] = asset_tag try: loc_chassis = re.search("Location In Chassis: ([0-9A-Za-z ]*)", dmi_board).group(1) except: loc_chassis = '** No value to retrieve **' dmi_info['DmiBoardLocInChass'] = loc_chassis # Based on the list from http://www.dmtf.org/sites/default/files/standards/documents/DSP0134_3.0.0.pdf board_dict = {'Unknown': 1, 'Other': 2, 'Server Blade': 3, 'Connectivity Switch': 4, 'System Management Module': 5, 'Processor Module': 6, 'I/O Module': 7, 'Memory Module': 8, 'Daughter board': 9, 'Motherboard': 10, 'Processor/Memory Module': 11, 'Processor/IO Module': 12, 'Interconnect board': 13} try: board_type = re.search("Type: ([0-9A-Za-z ]+)", dmi_board).group(1) board_type = str(board_dict.get(board_type)) except: board_type = '** No value to retrieve **' dmi_info['DmiBoardBoardType'] = board_type for v in dmidecode.system().values(): if type(v) == dict and v['dmi_type'] == 1: dmi_info['DmiSystemSKU'] = v['data']['SKU Number'] system_family = v['data']['Family'] system_serial = v['data']['Serial Number'] dmi_info['DmiSystemVersion'] = "string:" + v['data']['Version'] dmi_info['DmiSystemProduct'] = v['data']['Product Name'] dmi_info['DmiSystemVendor'] = v['data']['Manufacturer'] if not system_family: dmi_info['DmiSystemFamily'] = "Not Specified" else: dmi_info['DmiSystemFamily'] = system_family # Create a new UUID newuuid = str(uuid.uuid4()) dmi_info['DmiSystemUuid'] = newuuid.upper() # Create a new system serial number dmi_info['DmiSystemSerial'] = "string:"+(serial_randomize(0, len(system_serial))) for v in dmidecode.chassis().values(): dmi_info['DmiChassisVendor'] = v['data']['Manufacturer'] chassi_serial = v['data']['Serial Number'] dmi_info['DmiChassisVersion'] = v['data']['Version'] dmi_info['DmiChassisType'] = v['data']['Type'] # Based on the list from http://www.dmtf.org/sites/default/files/standards/documents/DSP0134_3.0.0.pdf chassi_dict = {'Other': 1, 'Unknown': 2, 'Desktop': 3, 'Low Profile Desktop': 4, 'Pizza Box': 5, 'Mini Tower': 6, 'Tower': 7, 'Portable': 8, 'Laptop': 9, 'Notebook': 10, 'Hand Held': 11, 'Docking Station': 12, 'All in One': 13, 'Sub Notebook': 14, 'Space-saving': 15, 'Lunch Box': 16, 'Main Server Chassis': 17, 'Expansion Chassis': 18, 'SubChassis': 19, 'Bus Expansion Chassis': 20, 'Peripheral Chassis': 21} dmi_info['DmiChassisType'] = str(chassi_dict.get(dmi_info['DmiChassisType'])) # python-dmidecode does not reveal all values .. this is plan B chassi = commands.getoutput("dmidecode -t3") try: dmi_info['DmiChassisAssetTag'] = re.search("Asset Tag: ([0-9A-Za-z ]*)", chassi).group(1) except: dmi_info['DmiChassisAssetTag'] = '** No value to retrieve **' # Create a new chassi serial number, added string to make it be taken as a string even if it's a number dmi_info['DmiChassisSerial'] = "string:"+str(serial_randomize(0, len(chassi_serial))) for v in dmidecode.processor().values(): dmi_info['DmiProcVersion'] = v['data']['Version'] dmi_info['DmiProcManufacturer'] = v['data']['Manufacturer']['Vendor'] # OEM strings try: for v in dmidecode.type(11).values(): oem_ver = v['data']['Strings']['3'] oem_rev = v['data']['Strings']['2'] except: pass try: dmi_info['DmiOEMVBoxVer'] = oem_ver dmi_info['DmiOEMVBoxRev'] = oem_rev except: dmi_info['DmiOEMVBoxVer'] = '** No value to retrieve **' dmi_info['DmiOEMVBoxRev'] = '** No value to retrieve **' # Write all data collected so far to file file_name="vboxmods-"+vm_name.replace(' ','_')+".sh" logfile = file(path_logs+'/'+file_name, 'w+') logfile.write('# Generated on: ' + time.strftime("%H:%M:%S") + '\n') for k, v in sorted(dmi_info.iteritems()): if '** No value to retrieve **' in v: logfile.write('# VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/pcbios/0/Config/' + k + '\t' + v + '\n') else: logfile.write('VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/pcbios/0/Config/' + k + '\t\'' + v + '\'\n') # Disk information disk_dmi = {} try: if os.path.exists("/dev/sda"): # Disk serial disk_serial = commands.getoutput( "hdparm -i /dev/sda | grep -o 'SerialNo=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'") # To avoid exceding 20 bytes serials if len(disk_serial)>20: disk_dmi['SerialNumber'] = (serial_randomize(0, 20)) else: disk_dmi['SerialNumber'] = (serial_randomize(0, len(disk_serial))) # Check for HP Legacy RAID elif os.path.exists("/dev/cciss/c0d0"): # Needs smartctl to be able to get the correct information if os.path.exists("/usr/sbin/smartctl"): hp_old_raid = commands.getoutput("smartctl -d cciss,1 -i /dev/cciss/c0d0") disk_serial = re.search("Serial number:([0-9A-Za-z ]*)", hp_old_raid).group(1).replace(" ", "") # To avoid exceding 20 bytes serials if len(disk_serial)>20: disk_dmi['SerialNumber'] = (serial_randomize(0, 20)) else: disk_dmi['SerialNumber'] = (serial_randomize(0, len(disk_serial))) else: print bcolors.WARNING+"Install smartmontools: apt-get install smartmontools"+bcolors.ENDC except OSError: print "Haz RAID?" print commands.getoutput("lspci | grep -i raid") # Disk firmeware rev try: if os.path.exists("/dev/sda"): disk_fwrev = commands.getoutput( "hdparm -i /dev/sda | grep -o 'FwRev=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'") disk_dmi['FirmwareRevision'] = disk_fwrev if len(disk_dmi['FirmwareRevision'])<8: disk_dmi['FirmwareRevision']=disk_dmi['FirmwareRevision'][:8] elif os.path.exists("/dev/cciss/c0d0"): # Needs smartctl to be able to get the correct information if os.path.exists("/usr/sbin/smartctl"): hp_old_raid = commands.getoutput("smartctl -d cciss,1 -i /dev/cciss/c0d0") disk_dmi['FirmwareRevision'] = re.search("Revision:([0-9A-Za-z ]*)", hp_old_raid).group(1).replace(" ", "") if len(disk_dmi['FirmwareRevision'])<8: disk_dmi['FirmwareRevision']=disk_dmi['FirmwareRevision'][:8] else: print "Install smartmontools: apt-get install smartmontools" except OSError: print "Haz RAID?" print commands.getoutput("lspci | grep -i raid") # Disk Model number try: if os.path.exists("/dev/sda"): disk_modelno = commands.getoutput( "hdparm -i /dev/sda | grep -o 'Model=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'") disk_dmi['ModelNumber'] = disk_modelno elif os.path.exists("/dev/cciss/c0d0"): # Needs smartctl to be able to get the correct information if os.path.exists("/usr/sbin/smartctl"): hp_old_raid = commands.getoutput("smartctl -d cciss,1 -i /dev/cciss/c0d0") disk_dmi['ModelNumber'] = re.search("Product:([0-9A-Za-z ]*)", hp_old_raid).group(1).replace(" ", "") else: print "Install smartmontools: apt-get install smartmontools" except OSError: print "Haz RAID?" print commands.getoutput("lspci | grep -i raid") # Write more things to file for k, v in disk_dmi.iteritems(): if '** No value to retrieve **' in v: logfile.write('# VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/piix3ide/0/Config/PrimaryMaster/' + k + '\t' + v + '\n') else: logfile.write('VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/piix3ide/0/Config/PrimaryMaster/' + k + '\t\'' + v + '\'\n') # CD-ROM information cdrom_dmi = {} if os.path.islink('/dev/cdrom'): # CD-ROM serial cdrom_serial = commands.getoutput( "hdparm -i /dev/cdrom | grep -o 'SerialNo=[A-Za-z0-9_\+\/ .\"-]*' | awk -F= '{print $2}'") if cdrom_serial: cdrom_dmi['ATAPISerialNumber'] = (serial_randomize(0, len(cdrom_serial))) else: cdrom_dmi['ATAPISerialNumber'] = "** No value to retrieve **" # CD-ROM firmeware rev cdrom_fwrev = commands.getoutput("cd-drive | grep Revision | grep ':' | awk {' print $3 \" \" $4'}") cdrom_dmi['ATAPIRevision'] = cdrom_fwrev.replace(" ", "") # CD-ROM Model numberA-Za-z0-9_\+\/ .\"- cdrom_modelno = commands.getoutput("cd-drive | grep Model | grep ':' | awk {' print $3 \" \" $4'}") cdrom_dmi['ATAPIProductId'] = cdrom_modelno # CD-ROM Vendor cdrom_vendor = commands.getoutput("cd-drive | grep Vendor | grep ':' | awk {' print $3 '}") cdrom_dmi['ATAPIVendorId'] = cdrom_vendor else: logfile.write('# No CD-ROM detected: ** No values to retrieve **\n') # And some more for k, v in cdrom_dmi.iteritems(): if '** No value to retrieve **' in v: logfile.write('# VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/piix3ide/0/Config/SecondaryMaster/' + k + '\t' + v + '\n') else: logfile.write('VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/piix3ide/0/Config/SecondaryMaster/' + k + '\t\'' + v + '\'\n') # Get the DSDT image #os.system("dd if=/sys/firmware/acpi/tables/DSDT of=DSDT.bin >/dev/null 2>&1") # Write to file # Da un error de que la tabla ACPI es mayor de 64KB, aunque lo comente, si luego ejecuto el batch dentro del Guest, pasa las pruebas #logfile.write('VBoxManage setextradata '+vm_name+' VBoxInternal/Devices/acpi/0/Config/CustomTable\t\'' + os.getcwd() + '/DSDT.bin\'\n') acpi_misc = commands.getoutput('acpidump -s | grep DSDT | grep -o "\(([A-Za-z0-9].*)\)" | tr -d "()"') acpi_list = acpi_misc.split(' ') acpi_list = filter(None, acpi_list) logfile.write('VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/acpi/0/Config/AcpiOemId\t\'' + acpi_list[1] + '\'\n') logfile.write('VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/acpi/0/Config/AcpiCreatorId\t\'' + acpi_list[4] + '\'\n') logfile.write('VBoxManage setextradata "'+vm_name+'" VBoxInternal/Devices/acpi/0/Config/AcpiCreatorRev\t\'' + acpi_list[5] + '\'\n') # Randomize MAC address, based on onboard interface MAC mac_seed = ':'.join(re.findall('..', '%012x' % uuid.getnode()))[0:9] big_mac = mac_seed + "%02x:%02x:%02x" % ( random.randint(0, 255), random.randint(0, 255), random.randint(0, 255), ) le_big_mac = re.sub(':', '', big_mac) # The last thing! logfile.write('VBoxManage modifyvm "'+vm_name+'" --macaddress1\t' + le_big_mac +'\n') # Done! logfile.close() print ' Finished: A template shell script has been created named:', file_name print bcolors.OKGREEN+'\n [*]'+bcolors.ENDC+' Creating guest based modification file (to be run inside the guest)' # Write all data to file file_name2="vboxmods-"+vm_name.replace(' ','_')+".bat" logfile = file('/srv/ftp/CopyThisOne!/' +file_name2, 'w+') manu = acpi_list[1] # DSDT logfile.write('@reg copy HKLM\HARDWARE\ACPI\DSDT\VBOX__ HKLM\HARDWARE\ACPI\DSDT\\' + manu + ' /s /f\n\n') logfile.write('@reg delete HKLM\HARDWARE\ACPI\DSDT\VBOX__ /f\n\n') logfile.write('@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\' + manu + '\VBOXBIOS HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\' + manu + '\\' + acpi_list[2] + '___' + ' /s /f\n\n') logfile.write('@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\' + manu + '\VBOXBIOS /f\n\n') logfile.write('@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\' + manu + '\\' + acpi_list[2] + '___\\00000002 HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\' + manu + '\\' + acpi_list[2] + '___\\' + acpi_list[3] + ' /s /f\n\n') logfile.write('@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\DSDT\\' + manu + '\\' + acpi_list[2] + '___\\00000002 /f\n') # FADT logfile.write('@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\' + manu + '\VBOXFACP HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\' + manu + '\\' + acpi_list[2] + '___ /s /f\n\n') logfile.write('@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\' + manu + '\VBOXFACP /f\n') logfile.write('@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\' + manu + '\\' + acpi_list[2] + '___\\00000001 HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\' + manu + '\\' + acpi_list[2] + '___\\' + acpi_list[3] + ' /s /f\n\n') logfile.write('@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\FADT\\' + manu + '\\' + acpi_list[2] + '___\\00000001 /f\n\n') # RSDT logfile.write('@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\' + manu + '\VBOXRSDT HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\' + manu + '\\' + acpi_list[2] + '___ /s /f\r\n') logfile.write('@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\' + manu + '\VBOXRSDT /f\r\n') logfile.write('@reg copy HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\' + manu + '\\' + acpi_list[2] + '___\\00000001 HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\' + manu + '\\' + acpi_list[2] + '___\\' + acpi_list[3] + ' /s /f\r\n') logfile.write('@reg delete HKEY_LOCAL_MACHINE\HARDWARE\ACPI\RSDT\\' + manu + '\\' + acpi_list[2] + '___\\00000001 /f\r\n') # SystemBiosVersion - TODO: get real values logfile.write('@reg add HKEY_LOCAL_MACHINE\HARDWARE\DESCRIPTION\System /v SystemBiosVersion /t REG_MULTI_SZ /d "' + acpi_list[1] + ' - ' + acpi_list[0] + '" /f\n\n') # VideoBiosVersion - TODO: get real values logfile.write('@reg add HKEY_LOCAL_MACHINE\HARDWARE\DESCRIPTION\System /v VideoBiosVersion /t REG_MULTI_SZ /d "' + acpi_list[0] + '" /f\n') # SystemBiosDate d_month, d_day, d_year = dmi_info['DmiBIOSReleaseDate'].split('/') if len(d_year) > 2: d_year = d_year[:2] logfile.write('@reg add HKEY_LOCAL_MACHINE\HARDWARE\DESCRIPTION\System /v SystemBiosDate /t REG_MULTI_SZ /d "' + d_month + '/' + d_day + '/' + d_year + '" /f\n') # Prevent WMI identification logfile.write('@reg add HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\services\PlugPlay /v Start /t REG_MULTI_SZ /d "4" /f\r\n') #The system have to be rebooted for this to work, don't know why # Disables FW logfile.write('netsh firewall set opmode disable\n') # Disables windows updates logfile.write('@reg add "HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows\CurrentVersion\WindowsUpdate\Auto Update" /v AUOptions /t REG_DWORD /d 1 /f\n') # Change the IP and DNS shortScript=''' @echo off set _enabled=Habilitado set _dedicated=Dedicado for /f "tokens=1,2,3*" %%i in ('netsh int show interface') do ( if %%i equ %_enabled% ( if %%j equ %_dedicated% ( netsh int ip set address "%%k %%l" static '''+guest_ip+''' 255.255.255.0 '''+host_ip+''' 1 netsh int ip set dns "%%k %%l" static '''+guest_primary_dns+''' ) ) ) ipconfig /flushdns :EOF''' logfile.write(shortScript+'\n') logfile.close() print ' Finished: A Windows batch file has been created named:', file_name2 return
def _getdmi(self): from pprint import pprint DMI = {} # get BIOS Data #tmp = dmidecode.bios() #pprint(tmp) for v in dmidecode.bios().values(): if type(v) == dict and v['dmi_type'] == 0: DMI['bios', 0, 'BIOS Revision'] = str( (v['data']['BIOS Revision'])) DMI['bios', 0, 'ROM Size'] = str((v['data']['ROM Size'])) try: DMI['bios', 0, 'Release Date'] = str( (v['data']['Relase Date'])) except (KeyError): DMI['bios', 0, 'Release Date'] = str( (v['data']['Release Date'])) DMI['bios', 0, 'Runtime Size'] = str( (v['data']['Runtime Size'])) DMI['bios', 0, 'Vendor'] = str((v['data']['Vendor'])) DMI['bios', 0, 'Version'] = str((v['data']['Version'])) # get System Data #tmp = dmidecode.system() #pprint(tmp) for v in dmidecode.system().values(): if type(v) == dict and v['dmi_type'] == 1: DMI['system', 0, 'Family'] = str((v['data']['Family'])) DMI['system', 0, 'Manufacturer'] = str( (v['data']['Manufacturer'])) DMI['system', 0, 'Product Name'] = str( (v['data']['Product Name'])) DMI['system', 0, 'SKU Number'] = str((v['data']['SKU Number'])) DMI['system', 0, 'Serial Number'] = str( (v['data']['Serial Number'])) DMI['system', 0, 'UUID'] = str((v['data']['UUID'])) DMI['system', 0, 'Version'] = str((v['data']['Version'])) DMI['system', 0, 'Wake-Up Type'] = str( (v['data']['Wake-Up Type'])) # get BaseBoard Data #tmp = dmidecode.baseboard() #pprint(tmp) for v in dmidecode.baseboard().values(): if type(v) == dict and v['dmi_type'] == 2: DMI['baseboard', 0, 'Manufacturer'] = str( (v['data']['Manufacturer'])) DMI['baseboard', 0, 'Product Name'] = str( (v['data']['Product Name'])) DMI['baseboard', 0, 'Serial Number'] = str( (v['data']['Serial Number'])) DMI['baseboard', 0, 'Version'] = str((v['data']['Version'])) # get chassis Data #tmp = dmidecode.chassis() #pprint(tmp) for v in dmidecode.chassis().values(): if type(v) == dict and v['dmi_type'] == 3: DMI['chassis', 0, 'Asset Tag'] = str((v['data']['Asset Tag'])) DMI['chassis', 0, 'Boot-Up State'] = str( (v['data']['Boot-Up State'])) DMI['chassis', 0, 'Lock'] = str((v['data']['Lock'])) DMI['chassis', 0, 'Manufacturer'] = str( (v['data']['Manufacturer'])) DMI['chassis', 0, 'Power Supply State'] = str( (v['data']['Power Supply State'])) DMI['chassis', 0, 'Security Status'] = str( (v['data']['Security Status'])) DMI['chassis', 0, 'Serial Number'] = str( (v['data']['Serial Number'])) DMI['chassis', 0, 'Thermal State'] = str( (v['data']['Thermal State'])) DMI['chassis', 0, 'Type'] = str((v['data']['Type'])) DMI['chassis', 0, 'Version'] = str((v['data']['Version'])) # get Processor Data #tmp = dmidecode.processor() #pprint(tmp) i = 0 for v in dmidecode.processor().values(): if type(v) == dict and v['dmi_type'] == 4: DMI['processor', i, 'Asset Tag'] = str( (v['data']['Asset Tag'])) DMI['processor', i, 'Characteristics'] = str( (v['data']['Characteristics'])) DMI['processor', i, 'Core Count'] = str( (v['data']['Core Count'])) DMI['processor', i, 'Core Enabled'] = str( (v['data']['Core Enabled'])) DMI['processor', i, 'Current Speed'] = str( (v['data']['Current Speed'])) DMI['processor', i, 'External Clock'] = str( (v['data']['External Clock'])) DMI['processor', i, 'Family'] = str((v['data']['Family'])) DMI['processor', i, 'L1 Cache Handle'] = str( (v['data']['L1 Cache Handle'])) DMI['processor', i, 'L2 Cache Handle'] = str( (v['data']['L2 Cache Handle'])) DMI['processor', i, 'L3 Cache Handle'] = str( (v['data']['L3 Cache Handle'])) DMI['processor', i, 'Manufacturer'] = str( (v['data']['Manufacturer']['Vendor'])) DMI['processor', i, 'Max Speed'] = str( (v['data']['Max Speed'])) DMI['processor', i, 'Part Number'] = str( (v['data']['Part Number'])) DMI['processor', i, 'Serial Number'] = str( (v['data']['Serial Number'])) DMI['processor', i, 'Socket Designation'] = str( (v['data']['Socket Designation'])) DMI['processor', i, 'Status'] = str((v['data']['Status'])) DMI['processor', i, 'Thread Count'] = str( (v['data']['Thread Count'])) DMI['processor', i, 'Type'] = str((v['data']['Type'])) DMI['processor', i, 'Upgrade'] = str((v['data']['Upgrade'])) DMI['processor', i, 'Version'] = str((v['data']['Version'])) DMI['processor', i, 'Voltage'] = str((v['data']['Voltage'])) i += 1 # get Memory Data #tmp = dmidecode.memory() #pprint(tmp) i = 0 for v in dmidecode.memory().values(): if type(v) == dict and v['dmi_type'] == 17: if str((v['data']['Size'])) != 'None': DMI['memory', i, 'Data Width'] = str( (v['data']['Data Width'])) DMI['memory', i, 'Error Information Handle'] = str( (v['data']['Error Information Handle'])) DMI['memory', i, 'Form Factor'] = str( (v['data']['Form Factor'])) DMI['memory', i, 'Bank Locator'] = str( (v['data']['Bank Locator'])) DMI['memory', i, 'Locator'] = str((v['data']['Locator'])) DMI['memory', i, 'Manufacturer'] = str( (v['data']['Manufacturer'])) DMI['memory', i, 'Part Number'] = str( (v['data']['Part Number'])) DMI['memory', i, 'Serial Number'] = str( (v['data']['Serial Number'])) DMI['memory', i, 'Size'] = str((v['data']['Size'])) DMI['memory', i, 'Speed'] = str((v['data']['Speed'])) DMI['memory', i, 'Type'] = str((v['data']['Type'])) i += 1 # get cache Data #tmp = dmidecode.cache() #pprint(tmp) # get connector Data #tmp = dmidecode.connector() #pprint(tmp) # get slot Data #tmp = dmidecode.slot() #pprint(tmp) return DMI
def _getdmi(self): from pprint import pprint DMI = { } # get BIOS Data #tmp = dmidecode.bios() #pprint(tmp) for v in dmidecode.bios().values(): if type(v) == dict and v['dmi_type'] == 0: DMI['bios',0,'BIOS Revision'] = str((v['data']['BIOS Revision'])) DMI['bios',0,'ROM Size'] = str((v['data']['ROM Size'])) try: DMI['bios',0,'Release Date'] = str((v['data']['Relase Date'])) except (KeyError): DMI['bios',0,'Release Date'] = str((v['data']['Release Date'])) DMI['bios',0,'Runtime Size'] = str((v['data']['Runtime Size'])) DMI['bios',0,'Vendor'] = str((v['data']['Vendor'])) DMI['bios',0,'Version'] = str((v['data']['Version'])) # get System Data #tmp = dmidecode.system() #pprint(tmp) for v in dmidecode.system().values(): if type(v) == dict and v['dmi_type'] == 1: DMI['system',0,'Family'] = str((v['data']['Family'])) DMI['system',0,'Manufacturer'] = str((v['data']['Manufacturer'])) DMI['system',0,'Product Name'] = str((v['data']['Product Name'])) DMI['system',0,'SKU Number'] = str((v['data']['SKU Number'])) DMI['system',0,'Serial Number'] = str((v['data']['Serial Number'])) DMI['system',0,'UUID'] = str((v['data']['UUID'])) DMI['system',0,'Version'] = str((v['data']['Version'])) DMI['system',0,'Wake-Up Type'] = str((v['data']['Wake-Up Type'])) # get BaseBoard Data #tmp = dmidecode.baseboard() #pprint(tmp) for v in dmidecode.baseboard().values(): if type(v) == dict and v['dmi_type'] == 2: DMI['baseboard',0,'Manufacturer'] = str((v['data']['Manufacturer'])) DMI['baseboard',0,'Product Name'] = str((v['data']['Product Name'])) DMI['baseboard',0,'Serial Number'] = str((v['data']['Serial Number'])) DMI['baseboard',0,'Version'] = str((v['data']['Version'])) # get chassis Data #tmp = dmidecode.chassis() #pprint(tmp) for v in dmidecode.chassis().values(): if type(v) == dict and v['dmi_type'] == 3: DMI['chassis',0,'Asset Tag'] = str((v['data']['Asset Tag'])) DMI['chassis',0,'Boot-Up State'] = str((v['data']['Boot-Up State'])) DMI['chassis',0,'Lock'] = str((v['data']['Lock'])) DMI['chassis',0,'Manufacturer'] = str((v['data']['Manufacturer'])) DMI['chassis',0,'Power Supply State'] = str((v['data']['Power Supply State'])) DMI['chassis',0,'Security Status'] = str((v['data']['Security Status'])) DMI['chassis',0,'Serial Number'] = str((v['data']['Serial Number'])) DMI['chassis',0,'Thermal State'] = str((v['data']['Thermal State'])) DMI['chassis',0,'Type'] = str((v['data']['Type'])) DMI['chassis',0,'Version'] = str((v['data']['Version'])) # get Processor Data #tmp = dmidecode.processor() #pprint(tmp) i = 0 for v in dmidecode.processor().values(): if type(v) == dict and v['dmi_type'] == 4: DMI['processor',i,'Asset Tag'] = str((v['data']['Asset Tag'])) DMI['processor',i,'Characteristics'] = str((v['data']['Characteristics'])) DMI['processor',i,'Core Count'] = str((v['data']['Core Count'])) DMI['processor',i,'Core Enabled'] = str((v['data']['Core Enabled'])) DMI['processor',i,'Current Speed'] =str((v['data']['Current Speed'])) DMI['processor',i,'External Clock'] = str((v['data']['External Clock'])) DMI['processor',i,'Family'] = str((v['data']['Family'])) DMI['processor',i,'L1 Cache Handle'] = str((v['data']['L1 Cache Handle'])) DMI['processor',i,'L2 Cache Handle'] = str((v['data']['L2 Cache Handle'])) DMI['processor',i,'L3 Cache Handle'] = str((v['data']['L3 Cache Handle'])) DMI['processor',i,'Manufacturer'] = str((v['data']['Manufacturer']['Vendor'])) DMI['processor',i,'Max Speed'] = str((v['data']['Max Speed'])) DMI['processor',i,'Part Number'] = str((v['data']['Part Number'])) DMI['processor',i,'Serial Number'] = str((v['data']['Serial Number'])) DMI['processor',i,'Socket Designation'] = str((v['data']['Socket Designation'])) DMI['processor',i,'Status'] = str((v['data']['Status'])) DMI['processor',i,'Thread Count'] = str((v['data']['Thread Count'])) DMI['processor',i,'Type'] = str((v['data']['Type'])) DMI['processor',i,'Upgrade'] = str((v['data']['Upgrade'])) DMI['processor',i,'Version'] = str((v['data']['Version'])) DMI['processor',i,'Voltage'] = str((v['data']['Voltage'])) i += 1 # get Memory Data #tmp = dmidecode.memory() #pprint(tmp) i = 0 for v in dmidecode.memory().values(): if type(v) == dict and v['dmi_type'] == 17 : if str((v['data']['Size'])) != 'None': DMI['memory',i,'Data Width'] = str((v['data']['Data Width'])) DMI['memory',i,'Error Information Handle'] = str((v['data']['Error Information Handle'])) DMI['memory',i,'Form Factor'] = str((v['data']['Form Factor'])) DMI['memory',i,'Bank Locator'] = str((v['data']['Bank Locator'])) DMI['memory',i,'Locator'] = str((v['data']['Locator'])) DMI['memory',i,'Manufacturer'] = str((v['data']['Manufacturer'])) DMI['memory',i,'Part Number'] = str((v['data']['Part Number'])) DMI['memory',i,'Serial Number'] = str((v['data']['Serial Number'])) DMI['memory',i,'Size'] = str((v['data']['Size'])) DMI['memory',i,'Speed'] = str((v['data']['Speed'])) DMI['memory',i,'Type'] = str((v['data']['Type'])) i += 1 # get cache Data #tmp = dmidecode.cache() #pprint(tmp) # get connector Data #tmp = dmidecode.connector() #pprint(tmp) # get slot Data #tmp = dmidecode.slot() #pprint(tmp) return DMI
def getData(self, options): """ Method getData Gets all information for system asset type @type options: dict @param options: passed options @rtype: void """ # getting data from dmidecode and parsing (chassis, system) for hwinfo in dmidecode.system().iteritems(): if hwinfo[1]['dmi_type'] == 1 and type(hwinfo[1]['data']) == dict: for iteminfo in hwinfo[1]['data'].iteritems(): tmpinfo = {} p = re.compile('\s+') key = p.sub('', iteminfo[0]) self.asset_info[0]['System' + key] = str(iteminfo[1]) for hwinfo in dmidecode.chassis().iteritems(): if hwinfo[1]['dmi_type'] == 3 and type(hwinfo[1]['data']) == dict: for iteminfo in hwinfo[1]['data'].iteritems(): tmpinfo = {} p = re.compile('\s+') key = p.sub('', iteminfo[0]) self.asset_info[0]['Chassis' + key] = str(iteminfo[1]) core_count = 0 core_enabled_count = 0 thread_count = 0 phys_cpu_count = 0 # counting totals for cpus, cores for hwinfo in dmidecode.processor().iteritems(): if hwinfo[1]['dmi_type'] == 4 and type(hwinfo[1]['data']) == dict: phys_cpu_count += 1 for iteminfo in hwinfo[1]['data'].iteritems(): if iteminfo[1] is not None and iteminfo[1] != '': p = re.compile('\s+') key = p.sub('', iteminfo[0]) if key == 'CoreCount': core_count += iteminfo[1] elif key == 'CoreEnabled': core_enabled_count += iteminfo[1] elif key == 'ThreadCount': thread_count += iteminfo[1] # we need this part in case dmidecode doesn't provide information # for cores and threads if thread_count == 0 or core_count == 0: cpuObj = systeminfo.proc.cpu.Cpu(self.confDir, self.cacheDir, self.viewDir) cpuObj.getData({}) logic_count = len(cpuObj.asset_info) phys_to_cores = {} for logic_cpu in cpuObj.asset_info: if logic_cpu.get('physicalid'): phys_cpu = logic_cpu.get('physicalid') cores = int(logic_cpu.get('cpucores'), 0) if not phys_to_cores.get('physcpu' + phys_cpu): phys_to_cores['physcpu' + phys_cpu] = cores core_count += cores if logic_count > core_count and phys_cpu_count != 0: thread_count = logic_count # getting memory info self.getMemInfo() self.asset_info[0]['OSCoreCount'] = str(core_count) self.asset_info[0]['OSCoreEnabled'] = str(core_enabled_count) self.asset_info[0]['OSThreadCount'] = str(thread_count) self.asset_info[0]['OSPhyscpuCount'] = str(phys_cpu_count) load = os.getloadavg() self.asset_info[0]['load'] = str(load[0]) + ' ' + str( load[1]) + ' ' + str(load[2]) self.asset_info[0]['machinetype'] = platform.machine() self.asset_info[0]['nodename'] = platform.node() self.asset_info[0]['osrelease'] = platform.release() self.asset_info[0]['osname'] = platform.system() self.asset_info[0]['osversion'] = platform.version() # getting info about distribution, first one is deprecated in newer versions of python try: distinfo = platform.dist() except AttributeError: distinfo = platform.linux_distribution() self.asset_info[0]['distname'] = distinfo[0] self.asset_info[0]['distver'] = distinfo[1] self.asset_info[0]['distid'] = distinfo[2] self.asset_info[0]['toolindex'] = self.asset_info[0][ 'SystemSerialNumber']
def getData(self, options): """ Method getData Gets all information for system asset type @type options: dict @param options: passed options @rtype: void """ # getting data from dmidecode and parsing (chassis, system) for hwinfo in dmidecode.system().iteritems(): if hwinfo[1]['dmi_type'] == 1 and type(hwinfo[1]['data']) == dict: for iteminfo in hwinfo[1]['data'].iteritems(): tmpinfo = {} p = re.compile('\s+') key = p.sub('', iteminfo[0]) self.asset_info[0]['System' + key] = str(iteminfo[1]) for hwinfo in dmidecode.chassis().iteritems(): if hwinfo[1]['dmi_type'] == 3 and type(hwinfo[1]['data']) == dict: for iteminfo in hwinfo[1]['data'].iteritems(): tmpinfo = {} p = re.compile('\s+') key = p.sub('', iteminfo[0]) self.asset_info[0]['Chassis' + key] = str(iteminfo[1]) core_count = 0 core_enabled_count = 0 thread_count = 0 phys_cpu_count = 0 # counting totals for cpus, cores for hwinfo in dmidecode.processor().iteritems(): if hwinfo[1]['dmi_type'] == 4 and type(hwinfo[1]['data']) == dict: phys_cpu_count += 1 for iteminfo in hwinfo[1]['data'].iteritems(): if iteminfo[1] is not None and iteminfo[1] != '': p = re.compile('\s+') key = p.sub('', iteminfo[0]) if key == 'CoreCount': core_count += iteminfo[1] elif key == 'CoreEnabled': core_enabled_count += iteminfo[1] elif key == 'ThreadCount': thread_count += iteminfo[1] # we need this part in case dmidecode doesn't provide information # for cores and threads if thread_count == 0 or core_count == 0: cpuObj = systeminfo.proc.cpu.Cpu(self.confDir, self.cacheDir, self.viewDir) cpuObj.getData({}) logic_count = len(cpuObj.asset_info) phys_to_cores = {} for logic_cpu in cpuObj.asset_info: if logic_cpu.get('physicalid'): phys_cpu = logic_cpu.get('physicalid') cores = int(logic_cpu.get('cpucores'), 0) if not phys_to_cores.get('physcpu' + phys_cpu): phys_to_cores['physcpu' + phys_cpu] = cores core_count += cores if logic_count > core_count and phys_cpu_count != 0: thread_count = logic_count # getting memory info self.getMemInfo() self.asset_info[0]['OSCoreCount'] = str(core_count) self.asset_info[0]['OSCoreEnabled'] = str(core_enabled_count) self.asset_info[0]['OSThreadCount'] = str(thread_count) self.asset_info[0]['OSPhyscpuCount'] = str(phys_cpu_count) load = os.getloadavg() self.asset_info[0]['load'] = str(load[0]) + ' ' + str(load[1]) + ' ' + str(load[2]) self.asset_info[0]['machinetype'] = platform.machine() self.asset_info[0]['nodename'] = platform.node() self.asset_info[0]['osrelease'] = platform.release() self.asset_info[0]['osname'] = platform.system() self.asset_info[0]['osversion'] = platform.version() # getting info about distribution, first one is deprecated in newer versions of python try: distinfo = platform.dist() except AttributeError: distinfo = platform.linux_distribution() self.asset_info[0]['distname'] = distinfo[0] self.asset_info[0]['distver'] = distinfo[1] self.asset_info[0]['distid'] = distinfo[2] self.asset_info[0]['toolindex'] = self.asset_info[0]['SystemSerialNumber']
print() #. Test for presence of important functions using /dev/mem... Using the legacy API #. This does not print any decoded info. If the call fails, either a warning will #. be issued or an exception will be raised. This test is now only used to check #. for presence of the legacy API, which "under the hood" uses #. dmidecode.QuerySection(name), where name can be 'bios', 'system', etc. if root_user: print("*** bios ***\n"); dmidecode.bios() print_warnings() print("*** system ***\n"); dmidecode.system() print_warnings() print("*** baseboard ***\n"); dmidecode.baseboard() print_warnings() print("*** chassis ***\n"); dmidecode.chassis() print_warnings() print("*** processor ***\n"); dmidecode.processor() print_warnings() print("*** memory ***\n"); dmidecode.memory() print_warnings() print("*** cache ***\n"); dmidecode.cache() print_warnings() print("*** connector ***\n"); dmidecode.connector() print_warnings() print("*** slot ***\n"); dmidecode.slot() print_warnings() #. Now test get/set of memory device file... print("*** get_dev()")