3
\�me � @ sd d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z e j e
�ZdZeddd g�Zd:ej_ed
d �ed d
�edd�edd�edd�edd�edd�edd�edd�ed d �ed d �ed d!�ed"d#�ed$d%�ed&d'�ed(d)�ed*d+�d,�Zeee d-�d.d/�Zeee d-�d0d1�Zeeee d2�d3d4�Zeee d-�d5d6�Zeed7�d8d9�ZdS );� N)�
namedtuple)�Optional)�subp)� is_container�
is_FreeBSDz/sys/class/dmi/id� KernelNames�linux�freebsdZboard_asset_tagzsmbios.planar.tagZ board_vendorzsmbios.planar.makerZ
board_namezsmbios.planar.productZ board_serialzsmbios.planar.serialZ
board_versionzsmbios.planar.versionZ bios_datezsmbios.bios.reldateZ bios_vendorzsmbios.bios.vendorZ bios_versionzsmbios.bios.versionZchassis_asset_tagzsmbios.chassis.tagZchassis_vendorzsmbios.chassis.makerZchassis_serialzsmbios.chassis.serialZchassis_versionzsmbios.chassis.versionZ
sys_vendorzsmbios.system.makerZ product_namezsmbios.system.productZproduct_serialzsmbios.system.serialZ product_uuidzsmbios.system.uuidZproduct_versionzsmbios.system.version)zbaseboard-asset-tagzbaseboard-manufacturerzbaseboard-product-namezbaseboard-serial-numberzbaseboard-versionzbios-release-datez bios-vendorz bios-versionzchassis-asset-tagzchassis-manufacturerzchassis-serial-numberzchassis-versionzsystem-manufacturerzsystem-product-namezsystem-serial-numberz system-uuidzsystem-version)�key�returnc C s� t j| �}|dks |jdkr dS djt|j�}tjd|� tjj |�sVtjd|� dS y"t
|d��}|j � }W dQ R X W n t k
r� tjd|� dS X |dt
|�d d kr�d
}y|jd �j� S tk
� r� } ztjd |||� W Y dd}~X nX dS )
z/
Reads dmi data from /sys/class/dmi/id
Nz{0}/{1}zquerying dmi data %szdid not find %s�rbzCould not read %s� �� �
� �utf8z-utf-8 decode of content (%s) in %s failed: %s)�DMIDECODE_TO_KERNEL�getr �format� DMI_SYS_PATH�LOG�debug�os�path�exists�open�read�PermissionError�len�decode�strip�UnicodeDecodeError�error)r
�kmapZ dmi_key_path�fpZkey_data�e� r& �/usr/lib/python3.6/dmi.py�_read_dmi_syspath? s2
r( c C s� t j| �}|dks |jdkr dS tjd|j� y6dd|jg}tj|�\}}|j� }tjd||j� |S tjk
r� } ztjd||� W Y dd}~X nX dS )z/
Reads dmi data from FreeBSD's kenv(1)
Nzquerying dmi data %sZkenvz-qzkenv returned '%s' for '%s'zfailed kenv cmd: %s
%s)r r r r r r r �ProcessExecutionError)r
r# �cmd�result�_errr% r& r& r' �
_read_kenve s
r- )r
�dmidecode_pathr c C s| yF|d| g}t j |�\}}|j� }tjd|| � |jdd�dkrDdS |S t jk
rv } ztjd||� dS d}~X nX dS )z
Calls out to dmidecode to get the data out. This is mostly for supporting
OS's without /sys/class/dmi/id support.
z--stringz dmidecode returned '%s' for '%s'�.� zfailed dmidecode cmd: %s
%sN)r r r r �replacer) )r
r. r* r+ r, r% r& r& r' �_call_dmidecode{ s
r2 c C s� t � r
dS t� rt| �S t| �}|dk r,|S dd� }tj� d }||�pN|d
ks`tjd|� dS tj d�}|rxt
| |�S tjd t � dS ) a�
Wrapper for reading DMI data.
If running in a container return None. This is because DMI data is
assumed to be not useful in a container as it does not represent the
container but rather the host.
This will do the following (returning the first that produces a
result):
1) Use a mapping to translate `key` from dmidecode naming to
sysfs naming and look in /sys/class/dmi/... for a value.
2) Use `key` as a sysfs key directly and look in /sys/class/dmi/...
3) Fall-back to passing `key` to `dmidecode --string`.
If all of the above fail to find a value, None will be returned.
Nc S s$ | dkp"| d dko"| dd � dkS )NZx86_64r �i� Z86r& )Zarchr& r& r' �is_x86� s z read_dmi_data.<locals>.is_x86� �aarch64�amd64z dmidata is not supported on %sZ dmidecodez0did not find either path %s or dmidecode command)r7 r8 ) r r r- r( r �unamer r r Zwhichr2 r )r
Z
syspath_valuer5 Z
uname_archr. r& r& r' �
read_dmi_data� s"
r: )�srcr c C s| d| kr | S t j� }xbtjd| �D ]R}||kr@tjd|| |� q"t|�}|sPd}tjd|| |� | jd|� d�|�} q"W | S )zBReplace __dmi.VARNAME__ with DMI values from either sysfs or kenv.�__z__dmi\.([^_]+)__z7Ignoring invalid __dmi.%s__ in %s. Expected one of: %s.r0 z'Replacing __dmi.%s__ in '%s' with '%s'.z__dmi.) r �keys�re�findallr Zwarningr: r r1 )r; Zvalid_dmi_keys�matchZ dmi_valuer& r& r' � sub_dmi_vars� s* rA )NN)Zloggingr r> � collectionsr Ztypingr Z cloudinitr Zcloudinit.utilr r Z getLogger�__name__r r r �__new__� __defaults__r �strr( r- r2 r: rA r& r& r& r' �<module> sR
&-