File: //usr/share/netplan/netplan/cli/__pycache__/sriov.cpython-38.pyc
U
��~f N � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlm mZ d dl m
Z
d dl m Z d dl
Z
G dd� de�Zeje d�dd�Zeje eje d �d
d �Zd d
� Zdd� Zdd� Zdd� Zdd� Zddd�Zd dd�ZdS ) � N)� defaultdict)�ConfigurationErrorc @ s� e Zd ZdZed�dd�Zeed�dd��Zeed�dd ��Zeed
�d d �Z eed�d
d��Z
ee d�dd��Z ee d�dd��Z
ee d�dd��Zeed�dd��Zeed�dd��Zeeed�dd�Zed�d d �Zd S ) � PCIDevicez.Helper class for interaction with a PCI device��pci_addrc C s
|| _ dS )znInitialise a new PCI device handler
:param pci_addr: PCI address of device
:type: str
Nr )�selfr � r �'/usr/share/netplan/netplan/cli/sriov.py�__init__' s zPCIDevice.__init__)�returnc C s dS )zusysfs path (can be overridden for testing)
:return: full path to /sys filesystem
:rtype: str
z/sysr �r r r r �sys. s z
PCIDevice.sysc C s t j�| jd| j�S )zq/sys path for PCI device
:return: full path to PCI device in /sys filesystem
:rtype: str
zbus/pci/devices)�os�path�joinr
r r r r r r 6 s zPCIDevice.path)�subpathr c C s t j�| j|�S )z�/sys subpath helper for PCI device
:param subpath: subpath to construct path for
:type: str
:return: self.path + subpath
:rtype: str
)r r r )r r r r r r > s zPCIDevice.subpathc C s2 d}t j�| �d��r.t j�t �| �d���}|S )zjKernel driver for PCI device
:return: kernel driver in use for device
:rtype: str
� �driver)r r �existsr �basename�readlink)r r r r r r G s zPCIDevice.driverc C s t j�| �d��S )z�Determine if device is bound to a kernel driver
:return: whether device is bound to a kernel driver
:rtype: bool
r �r r r r r r r r �boundR s zPCIDevice.boundc C s t j�| �d��S )zwDetermine if device is a SR-IOV Physical Function
:return: whether device is a PF
:rtype: bool
� sriov_numvfsr r r r r �is_pfZ s zPCIDevice.is_pfc C s t j�| �d��S )zvDetermine if device is a SR-IOV Virtual Function
:return: whether device is a VF
:rtype: bool
Zphysfnr r r r r �is_vfb s zPCIDevice.is_vfc C sX g }d}z(|� tj�t�| �d�|����� W n tk
rH Y qTY nX |d7 }q|S )z�List Virtual Function addresses associated with a Physical Function
:return: List of PCI addresses of Virtual Functions
:rtype: list[str]
r zvirtfn{}� )�appendr r r r r �format�FileNotFoundError)r �vf_addrs�ir r r r j s ��
zPCIDevice.vf_addrsc C s dd� | j D �S )z�List Virtual Function associated with a Physical Function
:return: List of PCI devices of Virtual Functions
:rtype: list[PCIDevice]
c S s g | ] }t |��qS r )r )�.0�addrr r r �
<listcomp>� s z!PCIDevice.vfs.<locals>.<listcomp>)r r r r r �vfs~ s z
PCIDevice.vfs)�obj_name�prop�valuec C s$ t �dd|dd�| j�||g� dS )z�Set devlink options for the PCI device
:param obj_name: devlink object to set options on
:type: str
:param prop: property to set
:type: str
:param value: value to set for property
:type: str
z
/sbin/devlink�dev�setzpci/{}N)�
subprocess�
check_callr r )r r& r' r( r r r � devlink_set� s
��zPCIDevice.devlink_setc C s | j S )zbString represenation of object
:return: PCI address of string
:rtype: str
r r r r r �__str__� s zPCIDevice.__str__N)�__name__�
__module__� __qualname__�__doc__�strr
�propertyr
r r r �boolr r r �listr r% r- r. r r r r r $ s*
r )r% c
C sJ g }| D ]<}|j std�|�d�� }|�|j� |�|� W 5 Q R X q|S )zBind unbound VFs to driver.z /sys/bus/pci/drivers/{}/bind�wt�r �openr �writer r )r% r Z bound_vfs�vf�fr r r �bind_vfs� s r= )r% r c
C sJ g }| D ]<}|j rtd�|�d�� }|�|j� |�|� W 5 Q R X q|S )z Unbind bound VFs from driver.z /sys/bus/pci/drivers/{}/unbindr7 r8 )r% r Z unbound_vfsr; r< r r r �
unbind_vfs� s r> c C s� ||kr�|j | }|�d�}|r�|�d�}|r@|| kr@|||< q�|�d�}|�d�}|�d�} | D ]R}
|rvt�|
|�rb|r�t�|
|�rb| r�t�|
| �s�qb||kr�td| ��|
||< qbn|| kr�|||< |�|d �S )N�matchzset-name�name�
macaddressr z3matched more than one interface for a PF device: %s)� ethernets�get�utils�is_interface_matching_name� is_interface_matching_macaddress�!is_interface_matching_driver_namer ) �
interfaces�config_manager�pf_link�pfsZpf_devZpf_match�set_name�by_nameZby_macZ by_driver� interfacer r r �_get_target_interface� s6
�
��
� rO c
C s� t j�d| d�}zbt|��P}d}|�� D ]<}|�� }|�d�r(|�dd�d }| W 5 Q R � W S q(W 5 Q R X W n6 tk
r� } zt d| t
|�f ��W 5 d}~X Y nX dS ) z5
Read PCI slot name for given interface name
�/sys/class/netz
device/ueventNzPCI_SLOT_NAME=�=� r z'failed parsing PCI slot name for %s: %s) r r r r9 � readlines�strip�
startswith�split�IOError� RuntimeErrorr3 )ZnetdevZ uevent_pathr< Z
pci_slot_name�line�er r r �_get_pci_slot_name� s
&r[ c C s� i }|j �� D ]�\}}|s q|dkr&q|�d�}|rPt| |||�} | r||| < q|�d�}
|
r|
|j krt| ||
|� |
|kr�|||
d7 < nt�d|
� qd||< q|�� D ]8\} } | |kr�|| | kr�td|| | f ��| || < q�dS )z�
Go through the list of netplan ethernet devices and identify which are
PFs and VFs, matching the former with actual networking interfaces.
Count how many VFs each PF will need.
�rendererzvirtual-function-count�linkr z9could not match physical interface for the defined PF: %sNz;more VFs allocated than the explicit size declared: %s > %s)rB �itemsrC rO �logging�warningr ) rH rI � vf_countsr% rK Zexplicit_countsZethernet�settingsZ explicit_num�pfrJ �countr r r �get_vf_count_and_functions� s4
�re c
C s� |dkrt d| |f ��tj�d| d�}tj�|d�}tj�|d�}z(t|��}t|�� �� �}W 5 Q R X W nT tk
r� } zt d| t
|�f ��W 5 d}~X Y n t k
r� t d | ��Y nX ||kr�t d
| ||f ��z(t|d ��}|� t
|�� W 5 Q R X W n� tk
�r� } z�d }|j
d
k�r�t�d| � zHt|d ��}|� d� W 5 Q R X t|d ��}|� t
|�� W 5 Q R X W n( tk
�r� } z| }W 5 d} ~ X Y nX d}|�r�t d|| t
|�f ��W 5 d}~X Y nX d S )zB
Allocate the required number of VFs for the selected PF.
� zDcannot allocate more VFs for PF %s than the SR-IOV maximum: %s > 256rP �devicer Zsriov_totalvfsz(failed parsing sriov_totalvfs for %s: %sNz#invalid sriov_totalvfs value for %szKcannot allocate more VFs for PF %s than supported: %s > %s (sriov_totalvfs)�wT� zLdevice or resource busy while setting sriov_numvfs for %s, trying workaround�0Fz,failed setting sriov_numvfs to %s for %s: %s)r r r r r9 �int�readrT rW rX r3 �
ValueErrorr: �errnor_ r` )
rc �vf_count�devdirZ numvfs_pathZ
totalvfs_pathr< Zvf_maxrZ ZbailZe_innerr r r �set_numvfs_for_pf" sH
�
$ � (rq c
C s� t j�d| d�}zhtt j�|d���}|�� �� dd� }W 5 Q R X tt j�|d���}|�� �� dd� }W 5 Q R X W n6 tk
r� } ztd| t|�f ��W 5 d}~X Y nX d�||g�}d}||kr�dS ) z�
Perform any hardware-specific quirks for the given SR-IOV device to make
sure all the VF-count changes are applied.
rP rg ZvendorrR Nz2could not determine vendor and device ID of %s: %s�:r ) r r r r9 rl rT rW rX r3 )rc rp r< Z device_idZ vendor_idrZ Z combined_idZ
quirk_devicesr r r � perform_hardware_specific_quirksQ s "&
rs �/c C s� d}t j�|d|d�}t j�t �|��}t j�|d| d�}t �|�D ]D} d| krDt j�|| �}
t j�t �|
��} | |krD| dd� } q�qD|s�td||f ��z.tjddd d
| d |d t |�g tj
tj
d
� W n" tj k
r� td| ��Y nX dS )z@
Apply the hardware VLAN filtering for the selected VF.
Nz
sys/class/netrg Zvirtfn� zAcould not determine the VF index for %s while configuring vlan %s�ipr] r* r) r; �vlan)�stdout�stderrzJfailed setting SR-IOV VLAN filter for vlan %s (ip link set command failed)) r r r r r �listdirrX r+ r, r3 �DEVNULL�CalledProcessError) rc r; Z vlan_name�vlan_id�prefixZvf_indexZ vf_devdirZ vf_dev_idZ pf_devdirr< Zdev_pathZdev_idr r r �apply_vlan_filter_for_vfn s>
�
��
�r c
C s� t �� }|�|� t �� }|�|� | �� t�� }tt �}i }i }t
|| |||� g }|r�|� � D ] \} }
t | |
�sxqd|�
| � qd|r�|D ] } t| � q�t�� }|D ]�} | j�| �} | �d�}
|
�r|
�d�}|D ]>}|r�t�||�s�q�| |k�r
|| �r
td| ��||| < q�q�| |kr�| || < q�|� � D ]�\}}|| }|j}|dk�r0t|�}t|�}|j�r0t�d�||j�� |j �r0|j }z t |j |j � |�!dd|� W 5 |�s�t |j |j � X �q0t"� }| j#� � D ]�\}} | �d�d k�r�| �d
�}| �d �}|�|�} | �s(t�$d ||f � �q�| j�|��d
�}|�|�} | |k�r`td
| ||f ��t%| | ||� |�&| � �q�dS )z
Go through all interfaces, identify which ones are SR-IOV VFs, create
them and perform all other necessary setup.
r? r@ z3matched more than one interface for a VF device: %s)Z switchdevZlegacyzFound VFs of {}: {}Zeswitch�moder\ Zsriovr] �idzKSR-IOV vlan defined for %s but link %s is either not a VF or has no matcheszJinterface %s for netplan device %s (%s) already has an SR-IOV vlan definedN)'�
libnetplan�ParserZload_yaml_hierarchy�StateZimport_parser_results�parse� netifacesrH r rk re r^ rq r rs rB rC rD rE r Zembedded_switch_moder[ r r r_ �debugr r r% Z delay_virtual_functions_rebindr= r r> r- r* �vlansr` r �add) rI Zrootdir�parserZnp_staterH ra r% rK Zvf_count_changedrc ro r; rb r? rM rN Z netdef_id�ifaceZnetdefZ eswitch_moder ZpcidevZrebind_delayedZfiltered_vlans_setrw r] r} Zvf_parent_entryr r r �apply_sriov_config� s�
�
�
�r� )rt )rt )r_ r r+ �typing� collectionsr �netplan.cli.utils�clirD Znetplan.libnetplanr� �netplan.configmanagerr r� �objectr �Iterabler= r> rO r[ re rq rs r r� r r r r �<module> s$ '1/
'