File: //opt/saltstack/salt/lib/python3.10/site-packages/zmq/auth/__pycache__/certs.cpython-310.pyc
o
�N�g� � @ s2 d Z ddlZddlZddlZddlmZmZmZmZ ddl Z dZ
dZ d dee ej
f de d ee ef d
eee ef d eee e f d e d
dfdd�Z d dee ej
f de d eee e f d
ee e f fdd�Zdee ej
f d
eeee f fdd�Zd dee ej
f d
eeef fdd�Zg d�ZdS ) z10MQ authentication related functions and classes.� N)�Dict�Optional�Tuple�Unionz�# **** Generated on {0} by pyzmq ****
# ZeroMQ CURVE **Secret** Certificate
# DO NOT PROVIDE THIS FILE TO OTHER USERS nor change its permissions.
a
# **** Generated on {0} by pyzmq ****
# ZeroMQ CURVE Public Certificate
# Exchange securely, or use a secure mechanism to verify the contents
# of this file after exchange. Store public certificates in your home
# directory, in the .curve subdirectory.
�utf-8� key_filename�banner�
public_key�
secret_key�metadata�encoding�returnc C s t |t�r
|�|�}t |t�r|�|�}t| ddd��e}|�|�tj�� �� |�d� |rW|�� D ]$\}}t |t�r@|�|�}t |t�rJ|�|�}|�d|� d|� d�� q2|�d� |�d |� d
�� |rx|�d |� d
�� W d � d S W d � d S 1 s�w Y d S )
zCreate a certificate file�w�utf8)r z metadata
z z = �
zcurve
z public-key = "z"
z secret-key = "N) �
isinstance�bytes�decode�open�write�format�datetime�now�items) r r r r
r r �f�k�v� r �B/opt/saltstack/salt/lib/python3.10/site-packages/zmq/auth/certs.py�_write_key_file s*
�"�r �key_dir�namec C sj t �� \}}tj�| |�}|� d�}|� d�}tj�� }t|t� |�|� t|t
� |�|||d� ||fS )zeCreate zmq certificates.
Returns the file paths to the public and secret certificate files.
z .key_secretz.key)r
r ) �zmqZ
curve_keypair�os�path�joinr r r �_cert_public_bannerr �_cert_secret_banner) r r! r r r
�
base_filenameZsecret_key_fileZpublic_key_filer r r r �create_certificates> s
�r) �filenamec C s� d}d}t j�| �std| � ���t| d��=}|D ]2}|�� }|�d�r%q|�d�r5|�dd�d �d�}|�d �rE|�dd�d �d�}|rK|rK nqW d � n1 sVw Y |du retd
| ��||fS ) a Load public and secret key from a zmq certificate.
Returns (public_key, secret_key)
If the certificate file only contains the public key,
secret_key will be None.
If there is no public key found in the file, ValueError will be raised.
NzInvalid certificate file: �rb� #s
public-key� =� s '"s
secret-keyzNo public key found in %s) r# r$ �exists�OSErrorr �strip�
startswith�split�
ValueError)r* r r
r �liner r r �load_certificateZ s(
� � r6 �.� directoryc C s\ i }t j�| �std| � ���t j�| d�}t�|�}|D ]}t|�\}}|r+d||< q |S )z5Load public keys from all certificates in a directoryz Invalid certificate directory: z*.keyT)r# r$ �isdirr0 r% �globr6 )r8 �certsZ glob_stringZ
cert_files� cert_filer �_r r r �load_certificates} s
�r> )r) r6 r> )NNr )N)r7 )�__doc__r r: r# �typingr r r r r" r' r& �str�PathLiker r r) r6 �boolr> �__all__r r r r �<module> sV
� ��
����
�#� ���
� �
�&#