o VAf/@s4ddlZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dZ d Z d Z d Zd ZGd ddeZddZGdddeedZedkreZeejeejeejeejeejeejeejeejedej deej!e"ddddSdS)N) _fireinfo)bios)cpu)device) hypervisor)networkz/sys/class/dmi/idz/etc/fireinfo-id)ZOEMzO.E.M.zo.e.m.zN/Azn/aZ12345Z54321Z202020ZChassisZchassiszDefault stringZEVALzNot ApplicableNoneemptyZSerialzSystem Serial NumberZSSNZXXXXXz$01010101-0101-0101-0101-010101010101z$00020003-0004-0005-0006-000700080009z$03000200-0400-0500-0006-000700080009z$11111111-1111-1111-1111-111111111111Z0000000Z00000000z$3b903780-4f79-1018-816e-aeb2724778a7)ZNAcs(eZdZfddZfddZZS) Singletoncstt||||d|_dSN)superr __init__instance)clsnamebasesdict __class__4/usr/lib/python3.10/site-packages/fireinfo/system.pyr@s zSingleton.__init__cs(|jdurtt|j|i||_|jSr )rr r __call__)rargskwrrrrDs zSingleton.__call__)__name__ __module__ __qualname__rr __classcell__rrrrr ?s r cCsbtj|sdSzt|}|WdWS1swYWdSty0YdSw)z! Read all data from filename. N)ospathexistsopenreadstripIOError)filenamefrrrread_from_fileKs   ( r(c@seZdZddZddZeddZeddZed d Zed d Z ed dZ eddZ eddZ eddZ ddZeddZeddZeddZeddZedd Zed!d"Zed#d$Zed%d&Zd'd(Zed)d*Zed+d,Zd-S).SystemcCs>t||_g|_|t|_t|_|j|_ dSr ) rZBIOSdevicesscanrZCPUrZ HypervisorZ read_cpuinfo_System__cpuinfoselfrrrrYs   zSystem.__init__c Cs:i}|j|j|j|j|j|j|j|jd|d<g|d<|jD],}|j |j |j |j |j d}|ddkrC|j |d<|j|d<|d|q|j|jj|jj|jj|jj|jj|jj|jj|jjd |d <|jjrt|jj|d <|j|j|j|jd |d <|jrd|jji|d<t|j |j!|dS)N)modelvendorvirtuallanguagereleasekernel_releasememory root_sizesystemr*) subsystemr0r/ deviceclassdriverr8Zpci sub_model sub_vendor) archr0r/ model_stringsteppingflagsspeedfamilycountrbogomips)ZgreenZblueZorangeZredrr0r)Zprofile_version public_id private_idprofile)"r/r0r1r2r3r4r5r6r*r8lowerr9r:r;r<appendr=rr>r?r@rArBrCrDrZ has_greenZhas_blueZ has_orangeZhas_redrPROFILE_VERSIONrErF)r.prdrrrrGes^       zSystem.profilecC tdS)Nrunamer-rrrr=s z System.archcCs$|j}|sdSt|}|S)zi This returns a globally (hopefully) ID to identify the host later (by request) in the database. (0000000000000000000000000000000000000000) secret_idhashlibsha1encode hexdigest)r.rEhrrrrEs zSystem.public_idcCs:d}t|jD]}||7}q|sdSt|}|S)z The private ID is built out of the _unique_id and used to permit a host to do changes on the database. No one could ever guess this without access to the host. rQ)reversedrRrSrTrUrV)r.rFirWrrrrFs zSystem.private_idcCs,tjtr ttSt|j}| S)zX Read a "secret" ID from a file if available or calculate it from the hardware. ) rr r!SECRET_ID_FILEr(rSrT _unique_idrUrV)r.rWrrrrRs zSystem.secret_idcCsg}dD]}ttjt|}||qg}|D]7}|dur qtD] }||kr,d}nq"|rsz$System._unique_id..ZQM000z/opt/pakfire/db/uuid#) r(rr join SYS_CLASS_DMIrIINVALID_ID_STRINGS_EXACT_MATCHINVALID_ID_STRINGSallroot_disk_serial startswith)r.ZidsfileidZ_idsrZrgrrrr\sD     zSystem._unique_idcCsd}tj|s dSt|d)}|D]}|dd\}}|dkr/|WdSqWddS1s;wYdS)zK Return the language code of IPFire or "unknown" if we cannot get it. z/var/ipfire/main/settingsunknownr=rZLANGUAGEN)rr r!r" readlinessplitr$)r.r&r'linekeyvalrrrr2s    "zSystem.languagecCs tdpdS)z( Return the system release string. z/etc/system-releaserkr(r-rrrr3. zSystem.releasecCstdS)z# Return the bios vendor name. z/sys/class/dmi/id/bios_vendorrsr-rrr bios_vendor5zSystem.bios_vendorcCsz|jd}Wn tyYdSw|drd|fSz |dd\}}W||fStyG|dr9d}|}n d}|}Y||fSY||fSw) NHardware)NNz ARM-VersatileZARM rZBCMZBroadcom)r,KeyErrorrhro ValueError)r.svmrrrvendor_model_tuple<s(   zSystem.vendor_model_tuplecCs |durdSddd|DS)zE Will remove all non-printable characters from the given string NrXcSsg|] }|tjvr|qSr)stringZ printable)r^xrrr Ysz(System.escape_string..)rb)r{rrr escape_stringQszSystem.escape_stringcCsdd}dD]}ttjt|}|r||Sqtjdr'|jdd}n| \}}||S)z8 Return the vendor string of this system (if any). N)Z sys_vendorZ board_vendorZchassis_vendorz/proc/device-treerw) r(rr rbrcrr!r,getr~)r.retrir}rrrr0[s   z System.vendorcCsdd}dD]}ttjt|}|r||Sqtd}|r%|dd}|s-|\}}||S)z7 Return the model string of this system (if any). N)Z product_nameZ board_modelZ chassis_modelz/proc/device-tree/modelrX)r(rr rbrcrreplacer~)r.rrir|rrrr/ms   z System.modelcCsJtdd}|}t|dWdS1swYdS)z0 Return the amount of memory in kilobytes. z /proc/meminforlrN)r"readliner$intro)r.r' firstlinerrrr5s  $z System.memorycCrM)z( Return the kernel release string. rOr-rrrr4rtzSystem.kernel_releasecCstdd_}|D]Q}|sq |dd\}}}}|dkr[|dks[|dd}|d d kr<|dd WdS|d tjvrP|dd }|d tjvsC|WdSq WddS1sgwYdS) z, Return the dev node of the root disk. z /proc/mountsrlrx/ZrootfsNrK)r"rnrorZdigits)r.r'rpZdevZ mountpointZfsrestrrr root_disks$     "zSystem.root_diskcCs\d|j}tj|s dSt|d}t|ddWdS1s'wYdS)z5 Return the size of the root disk in kilobytes. z/sys/block/%s/sizeNrlii)rrr r!r"rr)r.r r'rrrr6s   $zSystem.root_sizecCs:z td|j}Wn tyYdSw|r|SdS)z: Return the serial number of the root disk (if any). z/dev/%sN)rZget_harddisk_serialrOSErrorr$)r.serialrrrrgs zSystem.root_disk_serialc Cshg|_dtjfdtjff}|D]"\}}tj|sqt|}|D]}|j|tj ||q!qdS)zS Scan for all devices (PCI/USB) in the system and append them to our list. z/sys/bus/pci/devicesz/sys/bus/usb/devicesN) r*rZ PCIDeviceZ USBDevicerr r!listdirrIrb)r.Ztoscanr rZdirlistdirrrrr+s   z System.scancCs|jjS)z; Say if the host is running in a virtual environment. )rr1r-rrrr1rvzSystem.virtualcCstS)z& Reference to the network class. )rZNetworkr-rrrrrvzSystem.networkN)rrrrrGpropertyr=rErFrRr\r2r3rur~ staticmethodrr0r/r5r4rr6rgr+r1rrrrrr)XsR P     9             r)) metaclass__main__z ------------ z ------------ TrN)Z sort_keysindent)#rSZjsonrrrXrrrrrrrJrcr[rerdtyper r(objectr)rr{printr=r2r3rur5Zkernelrr6r*dumpsrGrrrrsD