Mini Shell
3
{)ce} � @ s" d dl Z d dlZd dlZd dlZd dlZddlmZ e jd�Ze jd�Z e j
e j � ej d�Z
ej d�Zej d�Zej d �Zej d
�Zej d �Zej d �Zej d
�Zej d�Zej d�Zej d�Zej d�Zej d�Zej d�ZG dd� de�Z G dd� de�Z d dd�Z d dd�Z d d d �Z dS )!� N� )�utilZblivetZtestdataz .*/int13_dev([0-9a-fA-F]+)/*$z2^(PCIX|PCI|XPRS|HTPT)\s*(\S*)\s*channel: (\S*)\s*$z'^ATAPI\s*device: (\S*)\s*lun: (\S*)\s*$z^ATA\s*device: (\S*)\s*$z"^SCSI\s*id: (\S*)\s*lun: (\S*)\s*$z ^USB\s*serial_number: (\S*)\s*$z^1394\s*eui: (\S*)\s*$z%^FIBRE\s*wwid: (\S*)\s*lun: (\S*)\s*$z ^I2O\s*identity_tag: (\S*)\s*$z ^RAID\s*identity_tag: (\S*)\s*$z^SATA\s*device: (\S*)\s*$z&^SATA\s*device: (\S*)\s*pmp: (\S*)\s*$z,^SAS\s*sas_address: (\S*)\s*lun: \(\S*\)\s*$z!^(\S*)\s*unknown: (\S*) (\S*)\s*$c @ sb e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�EddEntryz[ This object merely collects what the /sys/firmware/edd/* entries can
provide.
Nc C s� t j|p
ddd�| _t j|| jd�| _tj|�}t|jd�dd�| _d | _ t j
| jd�| _ d | _ d | _
d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _d | _ d | _ d | _ d | _ | j � d S )N� )�root� � )�base�version)!r �Pathr � sysfspath�re_bios_device_number�match�int�group�bios_device_number� sysfslink�get_sysfs_attrr
�mbr_sig�sectors�host_bus�pci_dev�channel� interface�type� atapi_device� atapi_lun�
ata_device�ata_pmp�scsi_id�scsi_lun�
usb_serial�ieee1394_eui64�
fibre_wwid� fibre_lun� i2o_identity� sas_address�sas_lun�load)�selfr r r � r* �/usr/lib/python3.6/edd.py�__init__F s6
zEddEntry.__init__c C sL d}| j d k r|d7 }| jd k r(|d7 }t| j| j| jd k g�rH|d7 }| jd k rZ|d7 }t| jd k | jd k g�rz|d7 }| j d k r�|d7 }| j
d k r�|d7 }t| j d k | j d k g�r�|d 7 }| j
d k r�|d
7 }| jd k r�|d 7 }t| j| jg�r�|d 7 }| jd k �r|d
7 }t| j| jg��r(|d7 }tj| j�}||d< ||d< || S )Nzh%(t)spath: %(sysfspath)s version: %(version)s %(nl)s%(t)smbr_signature: %(mbr_sig)s sectors: %(sectors)sz %(type)sz(%(nl)s%(t)ssysfs pci path: %(sysfslink)szK%(nl)s%(t)shost_bus: %(host_bus)s pci_dev: %(pci_dev)s channel: %(channel)sz%%(nl)s%(t)sinterface: "%(interface)s"zB%(nl)s%(t)satapi_device: %(atapi_device)s atapi_lun: %(atapi_lun)sz%%(nl)s%(t)sata_device: %(ata_device)sz, ata_pmp: %(ata_pmp)sz7%(nl)s%(t)sscsi_id: %(scsi_id)s, scsi_lun: %(scsi_lun)sz%%(nl)s%(t)susb_serial: %(usb_serial)sz'%(nl)s%(t)s1394_eui: %(ieee1394_eui64)sz8%(nl)s%(t)sfibre wwid: %(fibre_wwid)s lun: %(fibre_lun)sz)%(nl)s%(t)si2o_identity: %(i2o_identity)sz<%(nl)s%(t)ssas_address: %(sas_address)s sas_lun: %(sas_lun)s�t�nl)r r �anyr r r r r r r r r r r! r" r# r$ r% r& r' �copy�__dict__)r) Zline_padZ separator�s�dr* r* r+ �_fmt� s>
z
EddEntry._fmtc C sT t | |j� r(t || j� r(| j|jkS tj| j�}tj|j�}|d= |d= ||kS )Nr )�
isinstance� __class__r0 r1 )r) �other�ldict�rdictr* r* r+ �__gt__� s zEddEntry.__gt__c C sT t | |j� r(t || j� r(| j|jkS tj| j�}tj|j�}|d= |d= ||kS )Nr )r5 r6 r0 r1 )r) r7 r8 r9 r* r* r+ �__eq__� s zEddEntry.__eq__c C sT t | |j� r(t || j� r(| j|jk S tj| j�}tj|j�}|d= |d= ||k S )Nr )r5 r6 r0 r1 )r) r7 r8 r9 r* r* r+ �__lt__� s zEddEntry.__lt__c C s | j dd�S )N� �
)r4 )r) r* r* r+ �__str__ s zEddEntry.__str__c C s d| j dd�f S )Nz <EddEntry%s>� r )r4 )r) r* r* r+ �__repr__ s zEddEntry.__repr__c C s t | �| S )N)�str)r) �idxr* r* r+ � __getitem__ s zEddEntry.__getitem__c C s t t| ��S )N)�lenrB )r) r* r* r+ �__len__ s zEddEntry.__len__c C s� t j| jd�}|| _|�r �y�|j� d | _| jdkrdtj|�}t|j d��| _
t|j d��| _ �nR| jdkr�t j|�}t|j d��| _
�n*| jdkr�tj|�}t|j d��| _t|j d��| _�n�| jdkr�tj|�}t|j d�d d
�| _�n�| jd k�r tj|�}t|j d�d d
�| _�n�| jd k�r`tj|�}t|j d�d d
�| _t|j d�d d
�| _�nV| jd
k�r�tj|�}t|j d�d d
�| _�n(| jdk�r�tj|�}t|j d�d d
�| _n�| jdk�rt j|�}|�r�t|j d��| _
t|j d��| _ nt j|�}t|j d��| _
n�| jdk�r�t j|�}t j|�}|�rdt|j d�d d
�| _!t|j d�d d
�| _"n@|�r�t|j d�d d
�| _!t|j d�d d
�| _"nt#j$d| j|� nt#j$d| j|� W nF t%k
�r� } z(|j&dk�r�t#j$d| j|� n|�W Y d d }~X nX t j| jd�| _'t j| jd�}|�r.t|�| _(t j| jd�}|�r�t)j|�}|�r||j d�| _*|j d�| _+t|j d��| _,nt#j$d| j|� d S )Nr r �ATAPIr r �ATA�SCSI�USBr )r �1394�FIBRE�I2O�RAID�SATA�SAS� z'edd: can not match interface for %s: %sz*'NoneType' object has no attribute 'group'Z
mbr_signaturer r z&edd: can not match host_bus for %s: %s)-r r r r �splitr �re_interface_atapir r r r r �re_interface_atar �re_interface_scsir r �re_interface_usbr! �re_interface_1394r" �re_interface_fibrer# r$ �re_interface_i2or% �re_interface_raidZ
raid_array�re_interface_edd4_satar �re_interface_edd3_sata�re_interface_sas�re_interface_unknownr& r' �log�warning�AttributeError�argsr r �re_host_bus_pcir r r )r) r r Z sas_matchZ
unknown_match�er Zhbusr* r* r+ r( s�
z
EddEntry.load)N)�__name__�
__module__� __qualname__�__doc__r, r4 r: r; r< r? rA rD rF r( r* r* r* r+ r @ s
~&
r c @ sJ e Zd ZdZddd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�
EddMatcherz� This object tries to match given entry to a disk device name.
Assuming, heuristic analysis and guessing hapens here.
Nc C s || _ |p d| _d S )Nr )�eddr )r) � edd_entryr r* r* r+ r, k s zEddMatcher.__init__c
sT t jd� jd�}g }� fdd�}g }�x*|j� D �] }t jd� jd�}t j|� jd�}t j||d�}tjd||� dg|jd� }t|�d kr�q0|d
d � j j
f kr�q0|d j d
�s�q0|d } t |d dd � �}
d� j j
i} d| } tj| �}
|
jdj|dd � ��}|�s(|j|| |
||d�� q0x$||| |
||�D ]}|j|� �q:W q0W x�|D ]~}xv|f |�D ]h}|�r�tjd� j j
� j j� j j
� j j|d � n2tjd� j j
� j j� j j
� j j|d � |j|� �qhW �qXW t|�dk�rtjd� j jdd� |D �� t|�dk�r.|d d � j _|d d S tjd � j j
� j j� j j
� j j� d S ) Nz /sys/block/*)r c 3 s� t jt j| dd� d|g �� jd�}tt j|d��}� jjdkrT|� jjkrhd S n|� jjd krhd S | dd� d|f g }d g| d
|f g }t j|�}t j
|�}t j|d |f g �} t j| � jd�} t j|d |f g �}
t j|
� jd�}
x�| |
gD ]�} x�| j � D ]�} |j | j
�}
|
d k�r,�q|
jd
�d k�r�t|
jd��}� jjdk�rb|dk�sp� jj|k�r�t j|� jd�|jd�d d�V n>t|
jd��} � jj| k�rt j|� jd�|jd�d d�V �qW �q W d S )Nr � �ata_port)r �port_norH r zlink%dz.*zdev%d\.(\d+)(\.(\d+)){0,1}$z dev%d.*.*zdev%d.*rQ � �/)�link�path���rs )r r Z
join_pathsr r r rj r r �re�compile�globr �ondiskr r rR r )�
componentsrm � ata_port_idxrr rq �fnrn �expZ
expmatcherZpmp�devZataglobZatapathr r )r) r* r+ �
match_ports s>
z7EddMatcher.devname_from_ata_pci_dev.<locals>.match_portrp )rr rq zsysfs link: "%s" -> "%s"z
/sys/block� � z0000:%s� ZatarQ Zdevicez2target\d+:0:%(device)s/\d+:0:%(device)s:0/block/.*� )rx rm ry rr rq zQedd: ignoring possible extra match for ATA device %s channel %s ata %d pmp %s: %srr zNedd: using possible extra match for ATA device %s channel %s ata %d pmp %s: %sr z7edd: Found too many ATA devices for EDD device 0x%x: %sc S s g | ] }|d �qS )rq r* )�.0�ar* r* r+ �
<listcomp>� s z7EddMatcher.devname_from_ata_pci_dev.<locals>.<listcomp>r rq zFedd: Could not find ATA device for pci dev %s channel %s ata %d pmp %s)r r r rv �sysfs_readlink� testdata_log�debugrR rE rj r �
startswithr r rt ru r �join�appendr_ r` r r �errorr r )r) �patternZretriesr} �answersrr Z
emptyslashrq rx rm ry rb r{ �matcherr ZportZretryr* )r) r+ �devname_from_ata_pci_devo sr 2
z#EddMatcher.devname_from_ata_pci_devc C s | j jd ks| j jd kr d S d}| j j| j j| j jd�}tj|| | jd d�}g }xB|j� D ]6}tj |j
�}x$|D ] }|| }|j ||d�� qtW q^W t |�dkr�t
jd| j jdd � |D �� t |�d
kr�|d
d | j _|d
d S t
jd
| j j| j j| j j| j j� d S )Nza../devices/pci0000:00/0000:%(pci_dev)s/virtio*/host*/target*:0:%(dev)d/*:0:%(dev)d:%(lun)d/block/)r r| �lunz /sys/block/)r )rq rr r z:Found too many VirtIO SCSI devices for EDD device 0x%x: %sc S s g | ] }|d �qS )rq r* )r� r� r* r* r+ r� s z?EddMatcher.devname_from_virtio_scsi_pci_dev.<locals>.<listcomp>r rq rr zRedd: Could not find VirtIO SCSI device for pci dev %s channel %s scsi id %s lun %s)rj r r r r r r rv �os�listdirrw r� rE r_ r� r r �infor ) r) �tmplrb r� r� �mp�
block_entries�berq r* r* r+ � devname_from_virtio_scsi_pci_dev s.
z+EddMatcher.devname_from_virtio_scsi_pci_devc C s� d}| j j| j j| j j| j jd�}tj|| | jd d�}g }xB|j� D ]6}t j
|j �}x$|D ] }|| }|j ||d�� q^W qHW t
|�dkr�tjd| j jdd � |D �� t
|�d
kr�|d
d | j _|d
d S tjd
| j j| j j| j j| j j� d S )Nzn../devices/pci0000:00/0000:%(pci_dev)s/host%(chan)d/target%(chan)d:0:%(dev)d/%(chan)d:0:%(dev)d:%(lun)d/block/)r Zchanr| r� z /sys/block/)r )rq rr r z3Found too many SCSI devices for EDD device 0x%x: %sc S s g | ] }|d �qS )rq r* )r� r� r* r* r+ r� : s z8EddMatcher.devname_from_scsi_pci_dev.<locals>.<listcomp>r rq rr zKedd: Could not find SCSI device for pci dev %s channel %s scsi id %s lun %s)rj r r r r r r r rv r� r� rw r� rE r_ r� r r r` ) r) r� rb r� r� r� r� r� rq r* r* r+ �devname_from_scsi_pci_dev$ s.
z$EddMatcher.devname_from_scsi_pci_devc C s� t jd| jjf | jd d�}g }xB|j� D ]6}tj|j�}x$|D ] }|| }|j ||d�� qBW q,W t
|�dkr�t j d| jj
dd� |D �� t
|�d kr�|d d
| j_|d d S t jd | jj| jj� d S )
Nz,../devices/pci0000:00/0000:%s/virtio*/block/z /sys/block/)r )rq rr r z5Found too many VirtIO devices for EDD device 0x%x: %sc S s g | ] }|d �qS )rq r* )r� r� r* r* r+ r� R s z8EddMatcher.devname_from_virt_pci_dev.<locals>.<listcomp>r rq rr z;edd: Could not find Virtio device for pci dev %s channel %s)r r rj r r rv r� r� rw r� rE r_ r� r r r� r )r) r� r� r� r� r� rq r* r* r+ �devname_from_virt_pci_devD s&
z$EddMatcher.devname_from_virt_pci_devc C s� | j jd krd S | j� }|d k r$|S | j� }|d k r8|S d}| j j|krntjd| j j| j j� tjd | j j� | j jdkr�| j j d k r�| j
� }n| j jd kr�| j � }| j j r�t
jd
| jd�| j j d }tj|j�}tjd||� |S )NrG rJ rK rM rN rL rP z.edd: interface type %s is not implemented (%s)zedd: interface details: %srH rO rI z /sys/block/)r z/devicezsysfs link: "%s" -> "%s")rG rJ rK rM rN rL rP )rH rO )rj r r� r� r r_ r` r r r r� r� r r r r r� �readlinkrw r� r� )r) �nameZ unsupportedrr rq r* r* r+ �devname_from_pci_dev] s.
z EddMatcher.devname_from_pci_devc C sL t jd| jd�}x6|j� D ]*\}}|| jjkrt j||d�| j_|S qW dS )z� Try to match the edd entry based on its mbr signature.
This will obviously fail for a fresh drive/image, but in extreme
cases can also show false positives for randomly matching data.
z /sys/block/)r )rq N)r r r �itemsrj r r� r )r) �mbr_dictZsysblockr� r r* r* r+ �match_via_mbrsigsz s z EddMatcher.match_via_mbrsigs)N) re rf rg rh r, r� r� r� r� r� r� r* r* r* r+ ri d s
! ri c C sh i }t jd| d�}xP|j� D ]D}tj|�}td|jd�f dd�}tjd||� t || d�||< q W |S )Nz /sys/firmware/edd/int13_dev*/)r z0x%sr r )r z edd: found device 0x%x at %s)
r r rv r
r r r r_ r� r )r Z
edd_data_dictZglobstrrr r Zbiosdevr* r* r+ �collect_edd_data� s
r� c
C sp i }�xX| D �]N}d}y�t jd|d�|j }tj|jtj�}tj|dd� tj|d�}t j
d|�}t j
d|�}d j d
d � |D ��}tj |� t
jd ||� W nX tk
r� } z<t
jd
|| � tjd|jt| �� |dkr�tj |� w W Y dd} ~ X nX d| }
|
dk�rtjd|j� q n8x6|j� D ]*\} } | |
k�r$tjd|
| |j� i S �q$W |
||j< q W tjd|� |S )z� Read MBR signatures from devices.
Returns a dict mapping device names to their MBR signatures. It is not
guaranteed this will succeed, with a new disk for instance.
r z/dev)r i� r r �IZBBBBr c S s g | ]}d |f �qS )z%02xr* )r� �xr* r* r+ r� � s z collect_mbrs.<locals>.<listcomp>z device %s data[440:443] = %sz!device %s data[440:443] raised %sz+edd: could not read mbrsig from disk %s: %sNz0x%08xZ
0x00000000z1edd: MBR signature on %s is zero. new disk image?z,edd: dupicite MBR signature %s for %s and %sz!edd: collected mbr signatures: %srs )r r r� r� �openrw �O_RDONLY�lseek�read�struct�unpackr� �closer� r� �OSErrorr_ r� rB r� r� )
�devicesr r� r| �fdrr �dataZmbrsigZsdatard Z
mbrsig_strZdev_nameZmbrsig_str_oldr* r* r+ � collect_mbrs� sB
r� c
C s� t | |d�}t|d�}i }x�|j� D ]�\}}t||d�}|j� }tjd||� |rbtjd||� n |j|�}|r~tjd||� |r�|j |�} | r�tjd| ||� i S |||< q$tj
d|� q$W |S )a Generates the 'device name' -> 'edd number' mapping.
The EDD kernel module that exposes /sys/firmware/edd is thoroughly
broken, the information there is incomplete and sometimes downright
wrong. So after we mine out all useful information that the files under
/sys/firmware/edd/int13_*/ can provide, we resort to heuristics and
guessing. Our first attempt is, by looking at the device type int
'interface', attempting to map pci device number, channel number etc. to
a sysfs path, check that the path really exists, then read the device
name (e.g 'sda') from there. Should this fail we try to match contents
of 'mbr_signature' to a real MBR signature found on the existing block
devices.
)r z edd: data extracted from 0x%x:%rz%edd: matched 0x%x to %s using PCI devz%edd: matched 0x%x to %s using MBR sigz5edd: both edd entries 0x%x and 0x%x seem to map to %sz#edd: unable to match edd entry 0x%x) r� r� r� ri r� r_ r� r� r� �getr� )
r� r r� Zedd_entries_dictZedd_dictZ
edd_numberrk r� r� Zold_edd_numberr* r* r+ � get_edd_dict� s*
r� )N)N)N)!Zloggingr� rt r� r0 r r Z getLoggerr_ r� ZsetLevel�DEBUGru r
rc rS rT rU rV rW rX rY rZ r\ r[ r] r^ �objectr ri r� r� r� r* r* r* r+ �<module> s>
& &
/