| Server IP : 202.155.9.38 / Your IP : 216.73.216.124 Web Server : LiteSpeed System : Linux srv733443859 5.15.0-179-generic #189-Ubuntu SMP Tue May 5 18:20:56 UTC 2026 x86_64 User : rabbi4843 ( 1044) PHP Version : 8.1.34 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : ON Directory : /proc/752016/root/usr/share/netplan/netplan_cli/cli/__pycache__/ |
Upload File : |
o
��itS � @ sZ d dl Z d dlZd dlZd dlZd dlZd dlmZmZmZmZ ddl m
Z
ddl m Z d dl
Z
d dlZG dd� de�Zd eje fd
d �Zd eje d eje fd
d�Zde
jded efdd�Zde
jd ee fdd�Zdd� Zde
jd eeef fdd�Zde
jd eee f fdd�Z de
jd ee fd d �Z d d � Z d d!� Z d'd#d$�Z!d'd%d&�Z"dS )(� N)�Dict�List�Optional�Set� )�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 �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
�
/sbin/devlink�dev�setzpci/{}N)�
subprocess�
check_callr& r )r
r. r/ r0 r r r � devlink_set� s
��zPCIDevice.devlink_setc C sj d| j � �}ztjddddd|gtjd�}W n tjy! Y dS w t�|�}|�di ��|i ��d d�S )
z�Query eswitch mode via devlink for the PCI device
:return: the eswitch mode or '__undetermined' if it can't be retrieved
:rtype: str
zpci/r1 z-jr2 �eswitch�show)�stderr�__undetermined�mode)r r4 � check_output�DEVNULL�CalledProcessError�json�loads�get)r
�pci�output� json_outputr r r �devlink_eswitch_mode� s" �
� �
z PCIDevice.devlink_eswitch_modec 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- r6 rE rF 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� � ��rV 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/{}/unbindrO NrP )r- r � unbound_vfsrT rU r r r �
unbind_vfs� rW rY �netdef� interfacec C s | j |t�|�t�|�d�S )N)�
iface_name� iface_driver� iface_mac)�_match_interfacer �get_interface_driver_name�get_interface_macaddress)rZ r[ r r r �_interface_matches� s
�rb c C s� t �� }| jr<| j}|r||v r|S t� }|D ]}t| |�s qt|�dkr,td| j ��|� |� q|r:t
|�d S dS | j|v rD| jS dS )z�
Try to match a netdef with the real system network interface.
Throws ConfigurationError if there is more than one match.
r z3matched more than one interface for a PF device: %sr N) � netifaces�
interfaces�
_has_match�set_namer3 rb �lenr �id�addrN )rZ rd rf �matchesr[ r r r � _get_interface_name_for_netdef� s$
�rk 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 rQ � readlines�strip�
startswith�split�IOError� RuntimeErrorrK )�netdev� uevent_pathrU �
pci_slot_name�line�er r r �_get_pci_slot_name s"
��&���ry �np_statec C s� i }| j �� D ]O}|j�d� }r t| |j � }r |||j< q|jr.t|� }r.|||j< z|j}W n tj yF } zt
t |���d}~ww |dkrVt|� }rV|||j< q|S )z�
Go through the list of netplan ethernet devices and identify which are
PFs matching them with actual network interfaces.
�sriovNr ) � ethernets�values�linksrA rk rh �_embedded_switch_mode� _vf_count�netplan�NetplanExceptionr rK )rz �pfsrZ �link�iface�countrx r r r �_get_physical_functions s(
�
��
�r� c C sf i }| j �� D ])}z|j}W n tjy! } ztt|���d}~ww |dkr0t|� }r0|||< q|S )z�
Go through the list of netplan ethernet devices and identify which ones
have VFs. netdef._vf_count ultimately calls _netplan_state_get_vf_count_for_def
from libnetplan which return MAX(sriov_explicit_vf_count, number of VF netdefs).
Nr )r| r} r� r� r� r rK rk )rz � vf_countsrZ r� rx r� r r r �_get_vf_number_per_pf0 s
�� �r� c C sD t � }| j�� D ]}|j�d� }r t| |j �r |�|j� q|S )zk
Go through the list of netplan ethernet devices and identify which ones
are virtual functions
r{ )r3 r| r} r~ rA rk rh ri )rz r- rZ r� r r r �_get_virtual_functionsC s �r� 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 > 256rl �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 rQ �int�readro rr rs rK �
ValueErrorrR �errno�logging�warning)
�pf�vf_count�devdir� numvfs_path�
totalvfs_pathrU �vf_maxrx �bail�e_innerr r r �set_numvfs_for_pfQ 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.
rl r� �vendorr Nz2could not determine vendor and device ID of %s: %s�:r ) r r r rQ r� ro rr rs rK )r� r� rU � device_id� vendor_idrx � combined_id�
quirk_devicesr r r � perform_hardware_specific_quirks� 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/netr� �virtfn� zAcould not determine the VF index for %s while configuring vlan %s�ipr� r3 r2 rT �vlan)�stdoutr9 zJfailed setting SR-IOV VLAN filter for vlan %s (ip link set command failed)) r r r r r �listdirrs r4 r5 rK r= r> ) r� rT � vlan_name�vlan_id�prefix�vf_index� vf_devdir� vf_dev_id� pf_devdirrU �dev_path�dev_idr r r �apply_vlan_filter_for_vf� s< �
�
� ���r� c C s� t �� }|�|� t �� }|�|� | �� t�� }| j}t |�}t
|�}t |�}g }|rA|� � D ]\} }
t
| |
�s;q1|�| � q1|rP|D ]} t| � qEt�� }i } |D ]/} || }
|
jr{|D ]}|
j|d�ry| | v ru| | rutd| ��|| | < q_qT| |v r�| | | < qT|� � D ]\}}|| }
|
j}|dv �rt|�}t|�}|�� }||k�r|j�rt�d�||j�� |j r�z t |j |j � W n t y� } zt� d|� dt!|�� �� W Y d}~nd}~ww t�d|� d |� d
|� �� |�"d d |� |j �r|
j#�st$|j |j � q�t%� }|j&� � D ]L\}}
|
j'�r[|
j(�)d
�}|
j*}| �)|j+�} | �s6t� d||j+f � �q|�)|j(�)d�j+�} | |v �rOtd| |j+|f ��t,| | ||� |�-| � �qdS )z
Go through all interfaces, identify which ones are SR-IOV VFs, create
them and perform all other necessary setup.
)r\ z3matched more than one interface for a VF device: %s)� switchdev�legacyzFound VFs of {}: {}zUnbinding of VFs for z failed: NzChanging eswitch mode from z to z for: r7 r; r� zKSR-IOV vlan defined for %s but link %s is either not a VF or has no matchesr{ zJinterface %s for netplan device %s (%s) already has an SR-IOV vlan defined).r� �Parser�load_yaml_hierarchy�State�import_parser_results�parserc rd rz r� r� r� �itemsr� r% r� re r_ r r ry r
rE r" r� �debugr& r( r- rY r � Exceptionr� rK r6 � _delay_virtual_functions_rebindrV r3 �vlans�_has_sriov_vlan_filterr~ rA �_vlan_idrh r� ri )�config_manager�rootdir�parserrz rd r� �vfs_setr� �vf_count_changedr� r� r- rT rZ r[ � netdef_idr� � eswitch_moder �pcidev�current_eswitch_mode_systemrx �filtered_vlans_setr� r� r� r r r �apply_sriov_config� s�
���
&�� � �
�
��r� )r� )#r? r� r r4 �typingr r r r r r �
configmanagerr r� rc �objectr
�IterablerV rY �
NetDefinitionrK rM rb rk ry r� r� r� r� r� r� r� r� r� r r r r �<module> s0 $ /
'