File: //usr/share/netplan/netplan/cli/__pycache__/sriov.cpython-38.pyc
U
ci�_�5 � @ s� d dl Z d dlZd dlZd dlmZ d dlm mZ d dlm Z d dl
Z
dd� Z dd� Z dd � Z
d
d � Zdd
d�Zdd� ZdS )� N)� defaultdict)�ConfigurationErrorc 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�
macaddress�driverz3matched 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� interface� r �'/usr/share/netplan/netplan/cli/sriov.py�_get_target_interface s6
�
��
� 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�link� z9could not match physical interface for the defined PF: %sNz;more VFs allocated than the explicit size declared: %s > %s)r �itemsr r �logging�warningr ) r r � vf_counts�vfsr Zexplicit_countsZethernet�settingsZ explicit_num�pfr �countr r r �get_vf_count_and_functionsE s4
�r# 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 > 256�/sys/class/net�deviceZ sriov_numvfsZsriov_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 �os�path�join�open�int�read�strip�IOError� RuntimeError�str�
ValueError�write�errnor r )
r! �vf_count�devdirZ numvfs_pathZ
totalvfs_path�fZvf_max�eZbailZe_innerr r r �set_numvfs_for_pfv sH
�
$ � (r; 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.
r% r& Zvendor� Nz2could not determine vendor and device ID of %s: %s�:r ) r* r+ r, r- r/ r0 r1 r2 r3 )r! r8 r9 Z device_idZ vendor_idr: Z combined_idZ
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} 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/netr& Zvirtfn� zAcould not determine the VF index for %s while configuring vlan %s�ipr �set�dev�vf�vlan)�stdout�stderrzJfailed setting SR-IOV VLAN filter for vlan %s (ip link set command failed)) r* r+ r, �basename�readlink�listdirr2 �
subprocess�
check_callr3 �DEVNULL�CalledProcessError) r! rD Z vlan_name�vlan_id�prefixZvf_indexZ vf_devdirZ vf_dev_idZ pf_devdirr9 Zdev_pathZdev_idr r r �apply_vlan_filter_for_vf� s>
�
��
�rQ c C s� | � � t�� }tt�}i }i }t|| |||� g }|r`|�� D ] \}}t||�sTq@|�|� q@|r~|D ] }t |� qht�� }|D ]z}| j
� |�} | � d�}
|
r�|
� d�} |D ]:} | r�t �
| | �s�q�||kr�|| r�td| ��| ||< q�q�||kr�|||< q�t� }
| j�� D ]�\}} | � d�dk�r| � d�}| � d�}|�sLtd| ��|� |�}|�srt�d ||f � �q| j
� |�� d�}|� |�}||
k�r�td
|||f ��t||||� |
�|� �qd S ) 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: %sr Zsriovr �idz)no id property defined for SR-IOV vlan %szKSR-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)�parse� netifacesr r r. r# r r; �appendr> r r r
r r rB �vlansr r rQ �add)r r r r r Zvf_count_changedr! r7 rD r r r r Zfiltered_vlans_setrE r rO Zvf_parent_entryr r r �apply_sriov_config� st �
�
�
�rX )r? )r r* rK � collectionsr �netplan.cli.utils�clir
�netplan.configmanagerr rT r r# r; r> rQ rX r r r r �<module> s '1/
'