Current File : //lib/python3.9/site-packages/cockpit/_vendor/bei/__pycache__/beiboot.cpython-39.pyc
a
��h' � @ s2 d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
m Z ddl m
Z
d eeee
e d�dd�Zdd �eee
e d
�d d �Zdd �eee
e d
�d
d�Zddd�eeee
e d�dd�Zeee dd�dd�Ze
e edd�dd�Ze
e edd�dd�Zdd�d d �Zed k�r.e� dS ) � N)�IO�List�Sequence�Tuple� )�make_bootloaderF)�local�tty�sh�returnc C sD | r
t jnd}|r |df}n
|ddf}|r@d�dd� |D ��f}|S )NZpython3z-iqz-icz8" - beiboot - "; import sys; sys.ps1 = ''; sys.ps2 = '';� c s s | ]}t �|�V qd S )N)�shlex�quote)�.0�arg� r �?/usr/lib/python3.9/site-packages/cockpit/_vendor/bei/beiboot.py� <genexpr>. � z%get_python_command.<locals>.<genexpr>)�sys�
executable�join)r r r
Z interpreter�commandr r r �get_python_command s
�r �r )�argsr r c G s& dg| rdgnd�|�t | dd��R S )N�sshz-tr T�r r
�r �r r r r r �get_ssh_command3 s ��
�r c G s( dddg| rdgnd�|�t | d��R S )NZpodman�execz
--interactivez--ttyr r r r r r r �get_container_command: s ���r" r )r r r
r c G s g |�t d| |d��R S )NT)r r r
r )r r
r r r r � get_commandA s r# )�src�dstr c s* d d�� �fdd� }t j|dd��� d S )N�r c sR � �: d} t ��| �}|s q0� �|� � �� q
W d � n1 sD0 Y d S )Ni )�os�read�write�flush)Z
block_size�data�r% r$ r r �_threadF s
z!splice_in_thread.<locals>._threadT)�target�daemon)� threading�Thread�start)r$ r% r- r r, r �splice_in_threadE s r3 )r �scriptr c C sh t j| t jd��D}|jd us J �|j�|� td|j� t�|�� � W d � n1 sZ0 Y d S )N)�stdinr ) �
subprocess�Popen�PIPEr5 r) r3 r �exit�wait)r r4 �procr r r �send_and_spliceT s
r< c s� dd l }G � �fdd�d|j�}|�|� �}tj| tj|d��|� � jd usNJ �� j�tddt ��g dffg|j
d�� � � � j� � t
�|�� � td� j� t�� �� � W d � n1 s�0 Y d S ) Nr c s, e Zd Zeeee edd�� �fdd� ZdS )z%send_xz_and_splice.<locals>.ResponderN)r r �fds�stderrr c � s0 � j d usJ �|dkr,� j ��� � j �� d S )Nzbeiboot.provide)r5 r) r* )�selfr r r= r> �r; r4 r r �do_custom_commanda s z7send_xz_and_splice.<locals>.Responder.do_custom_command)�__name__�
__module__� __qualname__�strr r �intrA r r@ r r � Responder` s
�rG )r5 r> Zboot_xzz script.py.xzT)Zgadgets)�fernyZInteractionResponderZInteractionAgentr6 r7 r8 r5 r) r �lenZBEIBOOT_GADGETS�encoder* �asyncio�runZ communicater3 r r9 r: )r r4 rH rG Zagentr r@ r �send_xz_and_splice] s ��
rM r& c C s� t �� } | jdddd� | jddd� | jdd d� | jd
d d � | �� }|j oZt�d
�}|jg krrt|d�}nd|jd
dkr�t |jdd � d|i�}n<|jd
dkr�t
|jdd � d|i�}nt |j||j d��}|j�r t
|jd��}|�� }W d � n1 �s
0 Y t||� n`|j�rjt
|jd��}|�� }W d � n1 �sT0 Y t||� ntj|d
g|�R � d S )Nz--sh�
store_truez/Pass Python interpreter command as shell-script)�action�helpz--xzzthe xz to run remotely)rP z--scriptz2the script to run remotely (must be repl-friendly)r �*)�nargsr r r r r � containerr �rb)�argparse�ArgumentParser� add_argument�
parse_argsr4 r' �isattyr r r r" r# r
�openr( r< �xzrM �execlp)�parserr r r �filer4 r r r �mainx s6 ��
( ( r_ �__main__)FFF)rU rK r' r
r6 r r0 �typingr r r r Z
bootloaderr �boolrE r r r" r# rF �bytesr3 r< rM r_ rB r r r r �<module> s0 �� '
Mini Shell