3
c8h�3 � @ sD d dl Z d dlZd dlZd dlmZmZ dd� Zdd� ZdZdZ d Z
d
Z d Z d Z
d
ZdZdZd Zd
Zd Zd
ZdZdZd Zd
Zd
ZdZdZd Z d
Z d
Z dZ dZ d Z!d
Z"d Z#d Z$dZ%dZ&dZ'dZ(d
Z)dZ*dZ+d Z,d
Z-d Z.d
Z/dZ0dZ1d Z2d
Z3d Z4d Z5d
Z6dZ7dZ8d Z9d
Z:d Z;d Z<dZ=dd� Z>dd � Z?d d � Z@d d � ZAd!d"� ZBd#d$� ZCdS )%� N)�PCI�USBc
C sF t j� } | j� }g }�x*|jd�jd�jd�jd�jd�D �]}|j}||jt|�dt|�t|�d�}|d d krzd
|d< |d d kr�d |d < |dkr�t |d
�r�t |d
�|d< |j
|d< |j dkr�q:t |d�r�q:t |d�dkr�q:t |d�dkr�q:n�|dk�rJt |d�}|�r"|j d�\|d< |d< t |d�}|�r�|j d�\|d< |d< n>|dk�r�t |d ��rnt |d �|d< t |d ��r�t |d �|d< t |d
��r4t |d
�dk�r4t |d ��s�t |d ��r4t |d ��r�t |d �}t
jd |�} nt |d �}t
jd!|�} | �r4| jd"�|d< | jd#�|d< | jd$�|d< | jd%�|d< |j|� q:W |S )&aE Returns list of dictionaries with keys for every device in system
(values are provide as example):
'bus' : 'pci'
'driver' : 'pcieport-driver'
'pciType' : '1'
'detached' : '0'
'class' : 'OTHER'
'desc' : 'Intel Corporation|5000 Series Chipset PCI Express x4 Port 2'
�pci�usb�blockZccw�scsi�0)�bus�driverZpciTypeZdetached�classZdescr N�OTHERr
�unknownZID_BUSr �device� partitionZDM_NAMEZMAJOR�1�7�PCI_ID�:Zprop1Zprop2Z
PCI_SUBSYS_IDZprop3Zprop4� ID_VENDOR_ID� ID_MODEL_IDZID_PATHZDEVPATHz .*scsi-(\d+):(\d+):(\d+):(\d+)z!.*/(\d+):(\d+):(\d+):(\d+)/block/� � � � )�pyudevZContextZ list_devicesZmatch_subsystem� subsystemr
�_clasify_pci_type�_clasify_class�_get_device_desc�_get_device_propertyZsys_name� device_type�split�re�search�group�append)
�contextZdevices�resultr r Z result_item� pci_classZ
pci_subsys�path�m� r+ �#/usr/lib/python3.6/hardware_udev.py� get_devices sj ,
r- c C s$ t j� } | d | d | d d�}|S )z� Return dictionaries with keys (values are provided as example):
'system.formfactor': 'unknown'
'system.kernel.version': '2.6.18-128.1.6.el5xen'
'system.kernel.machine': 'i686'
'system.kernel.name': 'Linux'
r r r )zsystem.kernel.namezsystem.kernel.versionzsystem.kernel.machine)�os�uname)r/ r' r+ r+ r, �get_computer_infob s
r0 r Z00Z01Z02Z03Z04Z80�2�3�4�6Z05Z06Z07Z08r �9�Cc C s | dkr dS dS dS )z) return 1 if device is PCI, otherwise -1 r r z-1Nr+ )r r+ r+ r, r � s r c C s, y | j j|�S tk
r& | j|�S X dS )zS return the property of the given device independent of the implementation version N)Z
properties�get�AttributeError)r Zpropr+ r+ r, r � s r c C s: t t| d��\}}| j}|tkr$dS |tkrD|tkr8dS |tkrDdS t| d�}|rr|j� }d|krfdS d|krrdS |�rX|tkr�dS |t kr�|t
kr�d S |t kr�d
S n�|t kr�|t
kr�d S |tkr�d S |tkr�d
S |tkr�dS nx|tko�|tk� r�dS |tk�r|tk�rdS |tk�r6|tk�r&dS |tk�rXdS n"|tk�rX|tk�sT|tk�rXdS |dk�r�t| d��s�t| d��r�t| d�dk�r�dS dS n|dk�r�dS |dk�r�| jd k�r�t| �}|d k�s�|d k�r�dS |d k�r�d S |d!k�r�dS dS t j d"| j �}|�rd#S |dk�r d S |d$k�s2|d%k�r6dS d&S )'a Clasify type of device. Returned value is one of following string:
NETWORK, KEYBOARD, MOUSE, VIDEO, USB, IDE, SCSI, RAID, MODEM, SCANNER
CAPTURE, AUDIO, FIREWIRE, SOCKET, CDROM, HD, FLOPPY, TAPE, PRINTER, OTHER
or None if it is neither PCI nor USB device.
Z PCI_CLASSr ZKEYBOARDZMOUSEZ ID_SERIALZkeyboardZmouseZVIDEOr ZFIREWIREZIDEZSCSIZRAIDZFLOPPYZMODEMZSCANNERZCAPTUREZAUDIOZSOCKETr ZID_CDROMZID_TYPEZcdZCDROMZHDZsoundr Z scsi_devicer � r ZTAPE� z .*/lp\d+$ZPRINTERr r N) �_parse_pci_classr r �PCI_BASE_CLASS_NETWORK�PCI_BASE_CLASS_INPUT�PCI_CLASS_INPUT_KEYBOARD�PCI_CLASS_INPUT_MOUSE�lower�PCI_BASE_CLASS_DISPLAY�PCI_BASE_CLASS_SERIAL�PCI_CLASS_SERIAL_USB�PCI_CLASS_SERIAL_FIREWIRE�PCI_BASE_CLASS_STORAGE�PCI_CLASS_STORAGE_IDE�PCI_CLASS_STORAGE_SCSI�PCI_CLASS_STORAGE_RAID�PCI_CLASS_STORAGE_FLOPPY� PCI_BASE_CLASS_COMMUNICATION� PCI_CLASS_COMMUNICATION_MODEM�PCI_CLASS_INPUT_SCANNER�PCI_BASE_CLASS_MULTIMEDIA�PCI_CLASS_MULTIMEDIA_VIDEO�PCI_CLASS_MULTIMEDIA_AUDIO�PCI_BASE_CLASS_BRIDGE�PCI_CLASS_BRIDGE_PCMCIA�PCI_CLASS_BRIDGE_CARDBUSr �_get_scsi_dev_typer" r# �sys_path)r Z
base_classZ sub_classr Z id_serialZdev_typer* r+ r+ r, r � s�
r c C sL | j }d}d}|dkrLt| d�jd�\}}t� }d|j|�|j||�f }n�|dk�r&t| d�}t� }|r�d|j|�|j|t| d��f }n�| jd kr�| jd
kr�d }n| jd kr�d
}nd}nh| jdko�t| d��r:t| d�jd�dd� \}}dt |d� }dt |d� }d|j|�|j||�f }n|dk�r:t| d�}|�rD|S dS dS )z. Return human readable description of device. Nr r r z%s|%sr r r Z
usb_interfaceZusbhidzUSB HID InterfaceZhubzUSB Hub Interfacez
USB InterfaceZ
usb_deviceZPRODUCT�/r z%.4x� r ZID_MODEL� )
r r r! r Z
get_vendorZ
get_devicer r r
�int) r r Zcommandr' Z vendor_idZ device_idr r Zmodel_idr+ r+ r, r s8
"
r c C s( | dkr dS | dd� | dd � fS dS )
a= Parse Class Code. Return touple of
[base class code, sub-class code]
You are usually interested to only first two.
The third - "specific register-level programming interface" is ignored.
For details, see the PCI Local Bus Specification 2.1/2.2 Section 6.2.1 Device Identification
N� r r )NNi�������rZ ���r+ )r( r+ r+ r, r; B s r; c C s> yt d| j d�}W n tk
r( dS X |j� }|j� |S )zQ Return SCSI type of device in raw format as presented in /sys/...devpath../type z%s/type�rr ���)�openrT �IOError�readline�close)r �fr' r+ r+ r, rS N s rS )Dr r. r" Zhwdatar r r- r0 rE rG rF rI ZPCI_CLASS_STORAGE_IPIrH ZPCI_CLASS_STORAGE_OTHERr< ZPCI_CLASS_NETWORK_ETHERNETZ PCI_CLASS_NETWORK_TOKEN_RINGZPCI_CLASS_NETWORK_FDDIZPCI_CLASS_NETWORK_ATMZPCI_CLASS_NETWORK_OTHERrA ZPCI_CLASS_DISPLAY_VGAZPCI_CLASS_DISPLAY_XGAZPCI_CLASS_DISPLAY_3DZPCI_CLASS_DISPLAY_OTHERrM rN rO ZPCI_CLASS_MULTIMEDIA_PHONEZPCI_CLASS_MULTIMEDIA_OTHERrP ZPCI_CLASS_BRIDGE_HOSTZPCI_CLASS_BRIDGE_ISAZPCI_CLASS_BRIDGE_EISAZPCI_CLASS_BRIDGE_MCZPCI_CLASS_BRIDGE_PCIrQ ZPCI_CLASS_BRIDGE_NUBUSrR ZPCI_CLASS_BRIDGE_RACEWAYZPCI_CLASS_BRIDGE_OTHERrJ Z PCI_CLASS_COMMUNICATION_SERIALZ PCI_CLASS_COMMUNICATION_PARALLELZ#PCI_CLASS_COMMUNICATION_MULTISERIALrK Z PCI_CLASS_COMMUNICATION_OTHERr= r> ZPCI_CLASS_INPUT_PENr? rL ZPCI_CLASS_INPUT_GAMEPORTZPCI_CLASS_INPUT_OTHERrB rD ZPCI_CLASS_SERIAL_ACCESSZPCI_CLASS_SERIAL_SSArC ZPCI_CLASS_SERIAL_FIBERZPCI_CLASS_SERIAL_SMBUSr r r r r; rS r+ r+ r+ r, �<module> s� M`"