File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/tokens/__pycache__/localfs.cpython-310.pyc
o
�N�gy � @ sz d Z ddlZddlZddlZddlZddlZddlZddlZddl m
Z
e� e �Z
dZdd� Zdd� Zd d
� Zd d � ZdS )
zu
Stores eauth tokens in the filesystem of the master. Location is configured by the master config option 'token_dir'
� N)�DEFAULT_HASH_TYPEZlocalfsc C s2 t t| �dt��}t|t�d���� �}tj� | d |�}|� d�}tj�
|�rBt|t�d���� �}tj� | d |�}tj�
|�s(||d< zAt j j
�d��) t j j
�|d��}|�t j�|�� W d � n1 skw Y W d � n1 szw Y t�||� W |S ty� t�d |� i Y S w )
a�
Mint a new token using the config option hash_type and store tdata with 'token' attribute set
to the token.
This module uses the hash of random 512 bytes as a token.
:param opts: Salt master config options
:param tdata: Token data to be stored with 'token' attribute of this dict set to the token.
:returns: tdata with token if successful. Empty dict if failed.
� hash_typei � token_dirz.tmp�token� zw+bNz6Authentication failure: can not write token file "%s".)�getattr�hashlib�getr �str�os�urandom� hexdigest�path�join�isfile�salt�utils�filesZ set_umask�fopen�write�payload�dumps�rename�OSError�log�warning)�opts�tdatar �tok�t_pathZ temp_t_path�fp_� r! �G/opt/saltstack/salt/lib/python3.10/site-packages/salt/tokens/localfs.py�mk_token s,
� �� � � �r# c C s� t j�| d |�}tjj�| d |�si S t j�|�s i S z&tjj� |d��}tj
� |� � �}|W d � W S 1 s<w Y W dS t
yT t�d|� i Y S w )z�
Fetch the token data from the store.
:param opts: Salt master config options
:param tok: Token value to get
:returns: Token data if successful. Empty dict if failed.
r �rbNz5Authentication failure: can not read token file "%s".)r r r r r ZverifyZ
clean_pathr r r r �loads�readr r r )r r r r r r! r! r" � get_token1 s (� �r' c C sD t j�| d |�}zt �|� i W S ty! t�d|� Y dS w )z�
Remove token from the store.
:param opts: Salt master config options
:param tok: Token to remove
:returns: Empty dict if successful. None if failed.
r zCould not remove token %sN)r r r �remover r r )r r r r! r! r" �rm_tokenG s
�r) c C s: g }t jj�| d �D ]\}}}|D ]}|�|� qq |S )zy
List all tokens in the store.
:param opts: Salt master config options
:returns: List of dicts (tokens)
r )r r r Zos_walk�append)r �ret�dirpath�dirnames� filenamesr r! r! r" � list_tokensW s �r/ )�__doc__r �loggingr Z salt.payloadr Zsalt.utils.filesZsalt.utils.pathZsalt.utils.verifyZ salt.configr � getLogger�__name__r Z__virtualname__r# r' r) r/ r! r! r! r" �<module> s