Current File : //usr/share/crypto-policies/python/__pycache__/build-crypto-policies.cpython-39.pyc
a
C�h� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZdZdd� Zdd� Z dd� Z
dd d
�Z d d � Z d
d� Z
edkr|e
� dS )� Nzreload-cmds.shc O s t | dtji|�� d S )N�file)�print�sys�stderr)�args�kwargs� r �:/usr/share/crypto-policies/python/build-crypto-policies.py�eprint s r
c C s� t jdd�} | jdddd� | jdddd� | jd td
d d � | jd
ddd� | jdddd� | jddd� | jddd� | �� S )zParse the command lineF)� allow_abbrevz--flat�
store_truez1put all the generated files in a single directory)�action�helpz--testz7compare the generated config file with the existing onez--policyZPOLICYz"generate the specified policy only)�type�metavarr z --reloadcmdszDalso save reload cmds into reload-cmds.sh script in output directoryz--strictzfail on warnings� policydirz5a directory with base policy definition files (*.pol))r � outputdirz.a target directory with generated config files)�argparse�ArgumentParser� add_argument�str�
parse_args)�parserr r r r s. ��
�����r c
C s� | j r$tj�| j|� d|� d��}n`tj�| j|�}tj�|�srzt�|� W n" typ td|� �� Y dS 0 tj�||d �}| j �rVz�t
|dd��}|� � }W d � n1 s�0 Y ||k�rtd|� d|� d �� d
d � }t j
||�||�||d d
�} tj�| � W dS W dS t�y2 Y n$ t�yT td|� �� Y dS 0 td|� d|� �� t
|ddd��}|�|� W d � n1 �s�0 Y t� dS )N�-z.txtz#Cannot create directory for policy F�utf-8)�encodingz Config for z for policy z differs from the existing one:c S s dd� | � d�D �S )Nc S s g | ] }|d �qS )�
r )�.0�lr r r �
<listcomp>G � z.save_config.<locals>.lines.<locals>.<listcomp>r )�split)�sr r r �linesF s zsave_config.<locals>.linesz.new)ZfromfileZtofileTz Error reading generated file zSaving config for �w��moder )Zflat�os�path�joinr �isdir�mkdir�OSErrorr
�test�open�read�difflibZ unified_diffr r �
writelines�FileNotFoundErrorr �write)
�cmdline� policy_nameZ config_name�configr( �dirpath�fZ
old_configr# �diffr r r � save_config0 sH
� &
� *r: c
C s( d}|d u rg }z t j|g|�R d| ji�}W n6 tyd } z tdt|� � W Y d }~dS d }~0 0 dd� tt�D �}|D ]�}tj| }|� } | � |�
| j ��}
|dv s�| � |
��rz(d�
|g|��} t| | | j|
�s�d }W n. t�y td
| j � td � d }Y n0 q|td
| j � d}q||S )Nr r zError: � c S s g | ]}d |v r|�qS �� Generatorr �r �gr r r r h r z build_policy.<locals>.<listcomp>> ZEMPTYz GOST-ONLY�:� zError saving config for z Keeping original configuration� zError testing config for � )�cryptopoliciesZUnscopedCryptoPolicyr �
ValueErrorr
r �dir�policygenerators�__dict__Zgenerate_configZscopedZSCOPESZ test_configr) r: Z CONFIG_NAMEr, ) r4 r5 �subpolicy_names�errZcp�e�
generatorsr? �cls�genr6 �namer r r � build_policy[ s: ��
rP c C s� d}dd� t t�D �}tj�| jt�}zRt|ddd��0}|D ]}tj| }|� |j
� q<W d � n1 sl0 Y W n t y� t d� d}Y n0 |S ) Nr c S s g | ]}d |v r|�qS r< r r> r r r r � r z$save_reload_cmds.<locals>.<listcomp>r$ r r% zError saving reload cmds� )
rF rG r' r( r) r �RELOAD_CMD_NAMEr. rH r3 Z
RELOAD_CMDr, r
)r4 rJ rL r( r8 r? rM r r r �save_reload_cmds~ s
0
rS c C s� t � } d}| jrt�d� | jrPdd� | j�� �d�D �}|^}}t| ||�}nzt� | j
��^}|D ]H}|j � d�sb|j
dd�rbtj�|j �\}}|d krbt| |�}|rb q�qbW d
� n1 s�0 Y |s�| jr�t| �}t�|� d
S ) z!The actual command implementationr �errorc S s g | ] }|r|�qS r r )r �nr r r r � r zmain.<locals>.<listcomp>r@ �.F)�follow_symlinksz.polN)r �strict�warnings�filterwarningsZpolicy�upperr! rP r' �scandirr rO �
startswith�is_filer( �splitextZ
reloadcmdsrS r �exit)r4 rJ �namesr5 rI �sd�i�extr r r �main� s(
$
re �__main__)N)r r0 r' r rY rD rG rR r
r r: rP rS re �__name__r r r r �<module> s +
#
Mini Shell