File: //usr/share/netplan/netplan_cli/cli/__pycache__/sriov.cpython-310.pyc
o
ϐxe�F � @ s� d dl Z d dlZd dlZd dlZd dlmZ ddlmZ ddlm Z d dl
Z
d dl Z G dd� de �Z
d eje
fd
d �Zd eje
d eje
fd
d�Zdd� Zdd� Zdd� Zdd� Zdd� Zd dd�Zd d d �ZdS ) � N)� defaultdict� )�utils� )�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/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{}r )�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* r2 r3 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� � ��rC 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_vfsrA rB r r r �
unbind_vfs� rD rF 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)�
iface_name� iface_driver� iface_macz3matched more than one interface for a PF device: %s)�
_has_match�set_name�_match_interfacer �get_interface_driver_name�get_interface_macaddressr �get)�
interfaces�np_state�pf_link�pfs�pf_devrK � interfacer r r �_get_target_interface� s(
�
� rV 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 r z'failed parsing PCI slot name for %s: %s) r r r r> � readlines�strip�
startswith�split�IOError� RuntimeErrorr8 )�netdev� uevent_pathrB �
pci_slot_name�line�er r r �_get_pci_slot_name� s"
��&���rd c
C s� |j �� D ]E\}}|j�d�r t| ||j�d�j|�r d||< z|j}W n tjy7 } zt t
|���d}~ww |dkr=qt| |||�} | rJ||| < 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.
�sriovNr ) � ethernets�items�linksrO rV �id� _vf_count�netplan�NetplanExceptionr r8 )
rP rQ � vf_countsr* rS �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 > 256rW �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^ r8 �
ValueErrorr? �errno�logging�warning)
rq �vf_count�devdir� numvfs_path�
totalvfs_pathrB �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.
rW rt �vendorr Nz2could not determine vendor and device ID of %s: %s�:r ) r r r r> rz rZ r] r^ r8 )rq r� rB � 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. rA �vlan)�stdout�stderrzJfailed setting SR-IOV VLAN filter for vlan %s (ip link set command failed)) r r r r r �listdirr^ r0 r1 r8 �DEVNULL�CalledProcessError) rq rA � vlan_name�vlan_id�prefix�vf_index� vf_devdir� vf_dev_id� pf_devdirrB �dev_path�dev_idr r r �apply_vlan_filter_for_vfN s< �
�
� ���r� c
C s^ 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 ]J\}} | j#�r,| j$�%d�}| j&}|�%|j'�} | �st�(d||j'f � q�|�%|j$�%d �j'�} | |v �r td
| |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.
)rG z3matched more than one interface for a VF device: %s)� switchdev�legacyzFound VFs of {}: {}�eswitch�moder� zKSR-IOV vlan defined for %s but link %s is either not a VF or has no matchesre zJinterface %s for netplan device %s (%s) already has an SR-IOV vlan definedN)+rk �Parser�load_yaml_hierarchy�State�import_parser_results�parse� netifacesrP rQ r ry rr rg r� r" r� rJ rL r �_embedded_switch_moderd r r r} �debugr# r% r* � _delay_virtual_functions_rebindrF r r2 rC r/ �vlans�_has_sriov_vlan_filterrh rO �_vlan_idri r~ r� �add)�config_manager�rootdir�parserrQ rP rm r* rS �vf_count_changedrq r rA ro rU � netdef_id�iface� eswitch_moder �pcidev�rebind_delayed�filtered_vlans_setr� r� r� r r r �apply_sriov_configu s�
�
������� �
�
��r� )r� )r} r r0 �typing� collectionsr r r �
configmanagerr rk r� �objectr �IterablerC rF rV rd rr r� r� r� r� r r r r �<module> s& !/
'