U
K��^� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZe�d�Zdd� Zdej ej dddd�dd �Z
ej ej ddddd
d �d d
�Z ej ej ej dd�dd�Z
ej ej ej dddd�dd�Zddd�Zdd� Zdd� ZdS ) � Nzsubiquitycore.utilsc C s( | d krt j�� } n| �� } d| d< | S )N�C�LC_ALL)�os�environ�copy)�env� r �+/usr/share/subiquity/subiquitycore/utils.py�
_clean_env s
r
zutf-8�replace)�input�stdout�stderr�encoding�errorsr c
K s� |dkrt j|d< n
|�|�}t�d| � z\t j| f|||t|�d�|��}|r�t|jt �rj|j�
|�|_t|j t �r�|j �
|�|_ W n6 t j k
r� } zt�dt
| �� � W 5 d} ~ X Y nX t�d|j|j� |S dS )z�A wrapper around subprocess.run with logging and different defaults.
We never ever want a subprocess to inherit our file descriptors!
N�stdinzrun_command called: %s)r r
r r zrun_command %sz"run_command %s exited with code %s)�
subprocess�DEVNULL�encode�log�debug�runr
�
isinstancer
�bytes�decoder �CalledProcessError�str�args�
returncode)
�cmdr r
r r r r �kwZcp�er r r � run_command$ s(
�� r" F)r
r r r r r �checkc
� s� |d kr d|kr0t j|d< nt j|d< |�|�}t�d| � tj| ||t|�d�|��I d H } | j |d�I d H \}}|r�|d k r�|�
|�}|d k r�|�
|�}t�d| | j � |r�| j dkr�t � | j | ��nt �
| | j ||�S d S )Nr zarun_command called: %s�r
r r )r z#arun_command %s exited with code %sr )r r �PIPEr r r �asyncio�create_subprocess_execr
Z communicater r r ZCompletedProcess)
r r
r r r r r r# r �procr r r � arun_command@ s:
��
�r) )r
r r r c � s. t �d| � tj| ||t|�d�|��I d H S )Nzastart_command called: %sr$ )r r r& r' r
)r r
r r r r r r r �astart_commandZ s ��r* )r r
r r r r c K s, t �d| � tj| f|||t|�d�|��S )z�A wrapper around subprocess.Popen with logging and different defaults.
We never ever want a subprocess to inherit our file descriptors!
zstart_command called: %s)r r
r r )r r r �Popenr
)r r r
r r r r r r r r �
start_commandc s ��r, �SHA-512c sd ddddd�}||kr0t d�|d�|�� ����d� d }d�� fd
d �|D ��}t�| || | �S ) Nz$6$z$5$z$1$� )r- zSHA-256ZMD5ZDESz&Invalid algo({}), must be one of: {}. �,z@abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789./z c s g | ]}t �� ��qS r )�randomZchoice)�.0�c�Zsalt_setr r �
<listcomp>{ s z"crypt_password.<locals>.<listcomp>)� Exception�format�join�keys�crypt)ZpasswdZalgoZalgosZsaltr r3 r �crypt_passwordp s �r: c C s t �d� tdddddg� dS )z> Stop console-conf service; which also restores getty service z disabling console-conf service� systemctl�stop�
--no-blockzconsole-conf@*.servicez serial-console-conf@*.serviceN�r �infor" r r r r �disable_console_conf s
�r@ c C sJ t �d� tdddg� tddg� tddd d
g� tdd d d d
g� dS )z; Stop subiquity service; which also restores getty service zdisabling subiquity service�mkdirz-pz/run/subiquityZtouchz/run/subiquity/completer; �startr= zgetty@tty1.servicer< z(snap.subiquity.subiquity-service.servicezserial-subiquity@*.serviceNr> r r r r �disable_subiquity� s
�rC )r- )r& r9 Zloggingr r0 r Z getLoggerr r
r% r" r) r r* r, r: r@ rC r r r r �<module> sD
� � � �
|