File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/virt.cpython-310.pyc
o
�N�g� � @ sp d Z ddlZddlZddlZddlZddlZddlZddlZddlZ e�
e �Z dd� Z
dd� ZG dd� d�ZdS ) z:
This module contains routines shared by the virt system.
� Nc
C s� zet �t�d���� }tj�| �j� d�d }tj�
||� d|� ��}t j j
�|d��0}tj�| ��}|�|�� � W d � n1 sDw Y W d � |W S W d � |W S 1 s^w Y |W S tys } z|�d}~ww )z�
Attempts to download a file specified by 'url'
:param url: The full remote path of the file which should be downloaded.
:param dir: The path the file should be downloaded to.
� �/����-�wbN)�hashlib�md5�os�urandom� hexdigest�urllib�parse�urlparse�path�split�join�salt�utils�files�fopen�request�urlopen�writeZrease� Exception)�url�dirZrandZremote_filenameZfull_directory�file�response�err� r �C/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/virt.py�download_remote s* � �(� ����r! c C s& t �d�}|�tj�| �j�rdS dS )z�
Checks to see if the path provided contains ftp, http, or https. Returns
the full path if it is found.
:param cmdline_path: The path to the initrd image or the kernel
z
^(ht|f)tps?\bTF)�re�compile�matchr r
r �scheme)Z cmdline_pathZregexr r r � check_remote+ s
r& c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �VirtKeyz.
Used to manage key signing requests.
c C sR || _ || _|| _tj�| j d d|�}tj�|�s t�|� tj�||�| _d S )N�pki_dirZvirtkeys)�opts�hyper�idr r r �isdir�makedirs)�selfr* Zid_r) r r r r �__init__? s
zVirtKey.__init__c C s z!t jj�| jd��}t|�� �}W d � n1 sw Y W n# ty4 t� d| j
| j � Y dS t yD t� d| j� Y dS w t
�
� | dkrXt�d| j
| j � dS tj�| jd d | j
�}t jj�|d
��
}|�|� W d � n1 s|w Y | �� d S ) z)
Accept the provided key
�rNzJRequest to sign key for minion '%s' on hyper '%s' denied: no authorizationFzInvalid expiry data in %siX zORequest to sign key for minion "%s" on hyper "%s" denied: authorization expiredr( Zminions�w+T)r r r r r �int�read�OSError�log�errorr+ r* �
ValueError�time�warningr r r) r �void)r. Zpub�fp_ZexpiryZpubfnr r r �acceptH s: �� � �� �zVirtKey.acceptc C sN t jj�| jd��}|�ttt�� ��� W d � dS 1 s w Y dS )z@
Prepare the master to expect a signing request
r1 NT) r r r r r r �strr2 r8 )r. r; r r r � authorizel s
��zVirtKey.authorizec C s( z t �| j� W dS ty Y dS w )z7
Invalidate any existing authorization
TF)r �unlinkr r4 )r. r r r r: t s �z VirtKey.voidN)�__name__�
__module__� __qualname__�__doc__r/ r< r> r: r r r r r' : s $ r' )rC r �loggingr r" r8 r � urllib.parseZsalt.utils.filesr � getLoggerr@ r5 r! r& r' r r r r �<module> s