관리-도구
편집 파일: beiboot.cpython-39.pyc
a ��g' � @ 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 �� '