File: //usr/share/netplan/netplan/cli/__pycache__/sriov.cpython-310.pyc
o
�
ed�F � @ 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�Zdeje fdd�Zdeje d eje fd
d �Zd d
� Zdd� Zdd� Zdd� Zdd� Zddd�Zddd�ZdS ) � N)� defaultdict)�ConfigurationErrorc @ s� e Zd ZdZdefdd�Zedefdd��Zedefdd ��Zd
edefd d �Z edefd
d��Z
ede fdd��Z ede fdd��Z
ede fdd��Zedefdd��Zedefdd��Zdededefd d �Zdefd 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
N�r )�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�subpathc 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��rt 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 � 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 � zPCIDevice.boundc C r )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 r zPCIDevice.is_pfc C r )zvDetermine if device is a SR-IOV Virtual Function
:return: whether device is a VF
:rtype: bool
�physfnr r r r r �is_vfb r zPCIDevice.is_vfc C sR g }d} z|� tj�t�| �d�|����� W n
ty# Y |S w |d7 }q)z�List Virtual Function addresses associated with a Physical Function
:return: List of PCI addresses of Virtual Functions
:rtype: list[str]
r Tzvirtfn{}� )�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( r0 r1 r r r r r $ s,
r r( c C s^ g }| D ](}|j s,td�|�d��}|�|j� |�|� W d � n1 s'w Y q|S )zBind unbound VFs to driver.z /sys/bus/pci/drivers/{}/bind�wtN�r �openr! �writer r )r( r � bound_vfs�vf�fr r r �bind_vfs� � ��rA r c C s^ g }| D ](}|j r,td�|�d��}|�|j� |�|� W d � n1 s'w Y q|S )z Unbind bound VFs from driver.z /sys/bus/pci/drivers/{}/unbindr: Nr; )r( r � unbound_vfsr? r@ r r r �
unbind_vfs� rB rD c C s� ||vrD|| }|j r<|j}|r|| v r|||< n+| D ] }|j|t�|�t�|�d�s,q||v r6td| ��|||< qn|| v rD|||< |�|d �S )N)�itf_name�
itf_driver�itf_macz3matched more than one interface for a PF device: %s)� has_match�set_name�match_interface�utils�get_interface_driver_name�get_interface_macaddressr �get)�
interfaces�np_state�pf_link�pfs�pf_devrI � interfacer r r �_get_target_interface� s(
�
� rU c
C s� t j�d| d�}z?t|��0}d}|�� D ] }|�� }|�d�r3|�dd�d }| W d � W S qW d � W dS 1 s@w Y W dS ty] } z
t d| t
|�f ��d}~ww ) 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 r< � readlines�strip�
startswith�split�IOError� RuntimeErrorr6 )�netdev� uevent_pathr@ �
pci_slot_name�line�er r r �_get_pci_slot_name� s"
��&���rd c
C s� |j �� D ]>\}}|jrt| ||jj|�rd||< z|j}W n tjy0 } zt|j � �d}~ww |dkr6qt| |||�} | rC||| < qdS )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.
Nr )
� ethernets�items�
sriov_linkrU �id�vf_count�
libnetplan�LibNetplanExceptionr �args)
rO rP � vf_countsr( rR �nid�netdef�countrc �pfr r r �get_vf_count_and_functions� s
����rr c
C s" |dkr t d| |f ��tj�d| d�}tj�|d�}tj�|d�}z t|��}t|�� �� �}W d � n1 s:w Y W n" tyV } z
t d| t
|�f ��d}~w t yb t d | ��w ||krpt d
| ||f ��z!t|d ��}|� t
|�� W d � W d S 1 s�w Y W d S t�y } zrd }|j
d
kr�t�d| � z8t|d ��
}|� d� W d � n1 s�w Y t|d ��}|� t
|�� W d � n1 s�w Y W n ty� } z| }W Y d} ~ nd} ~ ww d}|�rt d|| t
|�f ��W Y d}~d S d}~ww )zB
Allocate the required number of VFs for the selected PF.
� zDcannot allocate more VFs for PF %s than the SR-IOV maximum: %s > 256rV �devicer �sriov_totalvfsNz(failed parsing sriov_totalvfs for %s: %sz#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 r< �int�readrZ r] r^ r6 �
ValueErrorr= �errno�logging�warning)
rq ri �devdir� numvfs_path�
totalvfs_pathr@ �vf_maxrc �bail�e_innerr r r �set_numvfs_for_pf sh
�
��� � � ���
� ����
���r� c
C s� t j�d| d�}zHtt j�|d���}|�� �� dd� }W d � n1 s'w Y tt j�|d���}|�� �� dd� }W d � n1 sJw Y W n tyf } z
td| t|�f ��d}~ww d�||g�}d}||v rw dS dS ) z�
Perform any hardware-specific quirks for the given SR-IOV device to make
sure all the VF-count changes are applied.
rV rt �vendorrX Nz2could not determine vendor and device ID of %s: %s�:r ) r r r r< rz rZ r] r^ r6 )rq r r@ � device_id� vendor_idrc � combined_id�
quirk_devicesr r r � perform_hardware_specific_quirks1 s$ � ����
�r� �/c C s� d}t j�|d|d�}t j�t �|��}t j�|d| d�}t �|�D ]"} d| v rDt j�|| �}
t j�t �|
��} | |krD| dd� } nq"|sOtd||f ��ztjddd d
| d |d t |�g tj
tj
d
� W dS tj yu td| ��w )z@
Apply the hardware VLAN filtering for the selected VF.
Nz
sys/class/netrt �virtfn� zAcould not determine the VF index for %s while configuring vlan %s�ip�linkr- r, r? �vlan)�stdout�stderrzJfailed setting SR-IOV VLAN filter for vlan %s (ip link set command failed)) r r r r r �listdirr^ r. r/ r6 �DEVNULL�CalledProcessError) rq r? � vlan_name�vlan_id�prefix�vf_index� vf_devdir� vf_dev_id� pf_devdirr@ �dev_path�dev_idr r r �apply_vlan_filter_for_vfN s< �
�
� ���r� c
C sR t �� }|�|� t �� }|�|� | �� t�� }| j}t t
�}i }i }t |||||� g }|rE|� � D ]\} }
t
| |
�s?q5|�| � q5|rT|D ]} t| � qIt�� }|D ]/} || } | jr}|D ]}
| j|
d�r{| |v rw|| rwtd| ��|
|| < qaqV| |v r�| || < qV|� � D ]O\}}|| } | j}|dv r�t|�}t|�}|jr�t�d�||j�� |jr�| j }zt |j|j � |� dd|� W |s�t |j|j � q�|s�t |j|j � w w q�t!� }|j"� � D ]D\}} | j#�r&| j$}| j%}|�&|j'�} | �st�(d||j'f � q�|�&|j)j'�} | |v �rtd| |j'|f ��t*| | ||� |�+| � q�d S )
z
Go through all interfaces, identify which ones are SR-IOV VFs, create
them and perform all other necessary setup.
)rE z3matched more than one interface for a VF device: %s)� switchdev�legacyzFound VFs of {}: {}�eswitch�modezKSR-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),rj �Parser�load_yaml_hierarchy�State�import_parser_results�parse� netifacesrO rP r ry rr rf r� r r� rH rJ r �embedded_switch_moderd r r r} �debugr! r# r( � delay_virtual_functions_rebindrD r r0 rA r- �vlans�has_sriov_vlan_filter� vlan_linkr� rN rh r~ rg r� �add)�config_manager�rootdir�parserrP rO rm r( rR �vf_count_changedrq ri r? ro rT � netdef_id�iface� eswitch_moder �pcidev�rebind_delayed�filtered_vlans_setr� r� r� r r r �apply_sriov_configu s�
�
������� �
�
��r� )r� )r} r r. �typing� collectionsr �netplan.cli.utils�clirK �netplan.libnetplanrj �netplan.configmanagerr r� �objectr �IterablerA rD rU rd rr r� r� r� r� r r r r �<module> s& !/
'