File: //proc/self/root/lib/python3.9/site-packages/cockpit/__pycache__/beiboot.cpython-39.opt-1.pyc
a
���e�, � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
d dl m Z m
Z
mZmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
l m Z d d l m Z d d l m!Z!m"Z" d d
l#m$Z$ d dl%m&Z& d dl'm(Z(m)Z) d dl*m+Z+ e�,e-�Z.e
d�dd�Z/e
e0 d�dd�Z1G dd� de"�Z2ddiej3�Z3G dd� de)�Z4G d d � d ej5�Z6G d d � d e$�Z7G d d!� d!e(�Z8dd�d"d#�Z9dd�d$d%�Z:e-d&k�r�e:� dS )'� N)�Path)�Dict�Iterable�Optional�Sequence)� polyfills)�ferny)�
bootloader)�BridgeBeibootHelper)�
setup_logging)�ChannelRoutingRule)�PackagesChannel)�
JsonObject)�Packages�PackagesLoader)�Peer)�CockpitProblem)�Router� RoutingRule)�StdioTransport��returnc C s� t j�tj�d } | �� }tj�d�}|d u r:tj �
d�}tj |dd� t |d�}t
�d|� z:|�� |krzt
�d� t�|�� jd @ s�t
�d
� t�W n4 ttfy� t
�d � |�|� |�d � Y n0 |S )
Nzinteraction_client.pyZXDG_CACHE_HOMEz~/.cacheT)�exist_okzcockpit-client-askpasszChecking if %s exists...z. ... it exists but is not the same version...�@ z; ... it has the correct contents, but is not executable...z ... writing contents.i� )� importlibZ resources�filesr �__name__�
read_bytes�os�environ�get�path�
expanduser�makedirsr �logger�debug�
ValueError�stat�st_mode�FileNotFoundError� write_bytes�chmod)Zsrc_pathZsrc_dataZxdg_cache_homeZ dest_path� r, �3/usr/lib/python3.9/site-packages/cockpit/beiboot.py�ensure_ferny_askpass. s(
r. c c s< t �� D ].} | jD ]"}|jdv rt|jt�r|jV qqd S )N)� path-exists�path-not-exists)r Zload_manifestsZ
conditions�name�
isinstance�value�str)Zmanifest� conditionr, r, r- �get_interesting_filesJ s
r6 c @ sF e Zd ZU eeef ed< eeed�dd�Zeeef d�dd�Z dS ) �ProxyPackagesLoader� file_status)r5 r3 r c C s. |dkr| j | S |dkr&| j | S t�d S )Nr/ r0 )r8 �KeyError)�selfr5 r3 r, r, r- �check_conditionT s
z#ProxyPackagesLoader.check_condition�r8 c C s
|| _ d S �Nr< )r: r8 r, r, r- �__init___ s z ProxyPackagesLoader.__init__N)
r �
__module__� __qualname__r r4 �bool�__annotations__�objectr; r> r, r, r, r- r7 Q s
r7 �
report_existsz�
import os
def report_exists(files):
command('cockpit.report-exists', {name: os.path.exists(name) for name in files})
c sL e Zd ZU eed< eed�� fdd� Zeed�dd�Zdd �d
d �Z � Z
S ) �DefaultRoutingRule�peer)�routerrF c s t � �|� || _d S r= )�superr> rF )r: rG rF �� __class__r, r- r> p s zDefaultRoutingRule.__init__)�optionsr c C s | j S r= )rF )r: rK r, r, r- �
apply_rulet s z DefaultRoutingRule.apply_ruleNr c C s | j �� d S r= )rF �close�r: r, r, r- �shutdownw s zDefaultRoutingRule.shutdown) r r? r@ r rB r r> r rL rO �
__classcell__r, r, rI r- rE m s
rE c @ sZ e Zd ZU dZeed< ed�dd�Zeeeee d�dd�Z ee
e e ed d
�d d �Z
d S )
�AuthorizeResponder)z
ferny.askpass�cockpit.report-existsrG �rG c C s
|| _ d S r= rS )r: rG r, r, r- r> s zAuthorizeResponder.__init__)�messages�prompt�hintr c � sr |dkr d S dt �|�� ��� }| jj||||dd�I d H }|�d��� }t �|�� ��� }t �
dt |�� |S )NZnonezX-Conversation - F)rT rU rV ZechozX-Conversation -z Returning a %d chars password) �base64Z b64encode�encode�decoderG Zrequest_authorization� removeprefix�stripZ b64decoder$ r% �len)r: rT rU rV Z challengeZresponseZb64�passwdr, r, r- �
do_askpass� s � z AuthorizeResponder.do_askpassN)�command�args�fds�stderrr c � sN t �d|||� |dkrJ|\}tt|�d�| j_| jj�dt| jt g�� d S )NzGot ferny command %s %s %srR )�loaderr )
r$ r% r r7 rG �packagesZ
routing_rules�insertr r
)r: r_ r` ra rb r8 r, r, r- �do_custom_command� s
z$AuthorizeResponder.do_custom_command)r r? r@ Zcommandsr rB r> r4 r r^ �tuple�list�intrf r, r, r, r- rQ { s
rQ c sV e Zd ZU eed< eeejd�� fdd� Z dd�dd�Z
ee dd �� fd
d � Z � Z
S ) �SshPeer�always)rG � destinationr` c s || _ |j| _t� �|� d S r= )rl rk rH r> )r: rG rl r` rI r, r- r> � s zSshPeer.__init__Nr c � s t | �}t�t| j��}|�|� d}d}| jdkr\t� }d|��ddf}d| jt� |�f}t
j � d�r�d d
gd d � |D ��|�R }d}t
�d
||� | j|||dd�I d H }| js�ddgffg}ng }tjg |�dtt� �gf�|j�td�}|�|�� � |�� I d H d S )N)Zpython3z-icz# cockpit-bridger, Z localhostz SSH_ASKPASS=z DISPLAY=xzSSH_ASKPASS_REQUIRE=forceZsshz/.flatpak-infoz
flatpak-spawnz--hostc s s | ]}d |� �V qdS )z--env=Nr, )�.0Zkvr, r, r- � <genexpr>� � z/SshPeer.do_connect_transport.<locals>.<genexpr>z Launching command: cmd=%s env=%sT)rb Zstart_new_sessionZtry_execzcockpit-bridgerD )Zgadgets)r
r ZInteractionAgentrQ rG Z add_handlerrl r. �shlex�joinr r! �existsr$ r% Zspawnrk r Zmake_bootloaderrh r6 Zsteps�BEIBOOT_GADGETS�writerX � communicate) r: Zbeiboot_helperZagent�cmd�envZ ssh_askpassZ transportZexec_cockpit_bridge_stepsZstage1r, r, r- �do_connect_transport� sF
� ������z SshPeer.do_connect_transport)r_ �messager c s |dkr d S t � �||� d S )NZ authorize)rH �transport_control_received)r: r_ ry rI r, r- rz � s z"SshPeer.transport_control_received)r r? r@ rA rB r r4 �argparse� Namespacer> rx r rz rP r, r, rI r- rj � s
.rj c sN e Zd ZU dZee ed< eed< ej d�� fdd� Z
dd� Z d d
� Z � Z
S ) � SshBridgeNrd �ssh_peer)r` c s* t | |j|�| _t� �t| | j�g� d S r= )rj rl r~ rH r> rE )r: r` rI r, r- r> � s
�zSshBridge.__init__c C s d S r= r, rN r, r, r- � do_send_init� s zSshBridge.do_send_initc C s: t |�d�t�r | jdd� d|d< | jjf i |�� d S )NZ superuserzsuperuser-init-done)r_ F)r2 r �dict�
write_controlr~ )r: ry r, r, r- �do_init� s zSshBridge.do_init)r r? r@ rd r r rB rj r{ r| r> r r� rP r, r, rI r- r} � s
r} c
� sJ t �d� t| �}tt�� |� z~|j�� I d H }|�di �}t |t
�s^|j dddd� W d S d|d< |j r�d d
� |j j D �|d < |j f i |�� |j�
� W nt tjy� } zt�t|�� W Y d }~nJd }~0 t�y } z(|j f d|jd �|j�� W Y d }~d S d }~0 0 t �d
� z|�� I d H W n t�yD Y n0 d S )NzHi. How are you today?� capabilities�initzprotocol-errorzcapabilities must be a dict)r_ �problemry Tzexplicit-superuserc S s i | ]
}|d �qS r= r, )rm �pr, r, r- �
<dictcomp> ro zrun.<locals>.<dictcomp>rd )r_ r� z/Startup done. Looping until connection closes.)r$ r% r} r �asyncioZget_running_loopr~ �start�
setdefaultr2 r� r� rd Z
thaw_endpointr ZInteractionError�sys�exitr4 r r� �kwargsru �BrokenPipeError)r` Zbridgery r� �excr, r, r- �run s0
"
r� c C sl t �� tjdd�} | jdddd� | jddd� | jd d
d � | �� }t|jd � tj t |�|jd � d S )
Nz@cockpit-bridge is run automatically inside of a Cockpit session.)� descriptionz--always�
store_truez/Never try to run cockpit-bridge from the system)�action�helpz--debug)r� rl z5Name of the remote host to connect to, or 'localhost')r� )r% )
r �installr{ �ArgumentParser� add_argument�
parse_argsr r% r� r� )�parserr` r, r, r- �main) s r� �__main__);r{ r� rW Zimportlib.resourcesr Zloggingr rp r� �pathlibr �typingr r r r Zcockpitr Zcockpit._vendorr Zcockpit._vendor.beir Zcockpit.beipackr
Zcockpit.bridger Zcockpit.channelr Zcockpit.channelsr
Zcockpit.jsonutilr Zcockpit.packagesr r Z cockpit.peerr Zcockpit.protocolr Zcockpit.routerr r Zcockpit.transportsr Z getLoggerr r$ r. r4 r6 r7 rs rE ZInteractionResponderrQ rj r} r� r� r, r, r, r- �<module> sL
��
/?&