File: //usr/lib/python3.9/site-packages/cockpit/__pycache__/bridge.cpython-39.pyc
a
���el) � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z m Z m
Z
mZmZmZ d dlmZ d dlmZmZ ddlmZ ddlmZ ddlmZ dd l m Z dd
l m Z m Z dd l!m"Z" dd l#m$Z$m%Z%m&Z& dd
l'm(Z(m)Z)m*Z* ddl+m,Z, ddl-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 e�5e6�Z7G dd� d�Z8G dd� de0e*�Z9dd�dd�Z:dd� Z;e<d �d d �Z=dd�d d �Z>d!d"�e<dd#�d$d%�Z?e6d&k�r�e?� dS )'� N)�Iterable�List�Optional�Sequence�Tuple�Type)�interaction_client)�bus� run_async� )� polyfills)� __version__)�ChannelRoutingRule)�
CHANNEL_TYPES)�Config� Environment)�EXPORTS)� JsonError�
JsonObject�get_dict)� BridgeConfig�Packages�PackagesListener)�PeersRoutingRule)�HostRoutingRule)�Router)�SuperuserRoutingRule)�StdioTransportc @ sP e Zd ZU eej ed< eee e
ej f d�dd�Z e ej dd�dd�Z
dS ) � InternalBus� exportees)�exportsc sN t �� \}}tjj|�� d�� _tjj|�� dd�� _� fdd�|D �� _d S )N)�fdT)r! �serverc s g | ]\}}� j �||� ��qS � )r" �
add_object)�.0�path�cls��selfr# �2/usr/lib/python3.9/site-packages/cockpit/bridge.py�
<listcomp>9 � z(InternalBus.__init__.<locals>.<listcomp>) �socket�
socketpairr ZBus�new�detachZclientr" r )r) r Z
client_socketZ
server_socketr# r( r* �__init__5 s zInternalBus.__init__N)r& �obj�returnc C s | j �| j�||�� d S �N)r �appendr" r$ )r) r& r2 r# r# r* �export; s zInternalBus.export)�__name__�
__module__� __qualname__r r ZSlot�__annotations__r r �strr Z
BaseObjectr1 r6 r# r# r# r* r 2 s
r c s� e Zd ZU eed< ee ed< ee ed< e j
ed< e j
d�� fdd� Z e dd � �Z
ed
d �d d
�Zd
d�dd�Zd
d�dd�Zed�� fdd� Z� ZS )�Bridge� internal_bus�packages�bridge_configs�args)r@ c s� t t�| _g | _|| _t| |jd�| _| j�d| j� | j�dt � � | j�dt
� � t | �| _ |j
r�| j�tdg d�dgd��f� d | _n0|jr�d | _n"t| �| _| j�d | j� | �� t� �t| �| jt| t�| j g� d S )
N)�
privilegedz
/superuserz/configz /environmentT)Zsudoz-kz-AZpython3z-icz# cockpit-bridge� --privilegedzSUDO_ASKPASS=ferny-askpass)rA Zspawn�environz /packages)r r r= r? r@ r rA �superuser_ruler6 r r r �
peers_rule�beipack� set_configsr r> r �packages_loaded�superr1 r r r )r) r@ �� __class__r# r* r1 E s8
��
�zBridge.__init__c C s� zt ddd�} W nH tyX zt ddd�} W n$ tyR t�d� i Y Y S 0 Y n0 i }| �� D ]f}|�� }|rf|�d�r�qfz |�d�\}}t�|�\}W n" t y� t�d|� Y qfY n0 |||< qf|S ) Nz/etc/os-releasezutf-8)�encodingz/usr/lib/os-releasez6Neither /etc/os-release nor /usr/lib/os-release exists�#�=z'Ignoring invalid line in os-release: %r)
�open�FileNotFoundError�logger�warning� readlines�strip�
startswith�split�shlex�
ValueError)�fileZ
os_release�line�k�vZv_parsedr# r# r* �get_os_releasej s*
zBridge.get_os_releaseN)�messager3 c C sD t �t��& t|d�}| j�|� W d � n1 s60 Y d S )N� superuser)�
contextlib�suppressr r rD �init)r) r^ r_ r# r# r* �do_init� s
zBridge.do_init�r3 c C sL ddi| � � d�}| jd ur2dd� | jjD �|d< | jf ddd �|�� d S )
Nzexplicit-superuserT)Z capabilitiesz
os-releasec S s i | ]
}|d �qS r4 r# )r% �pr# r# r* �
<dictcomp>� r, z'Bridge.do_send_init.<locals>.<dictcomp>r> rb r )�command�version)r] r> Z
write_control)r) Z init_argsr# r# r* � do_send_init� s �
zBridge.do_send_initc C s@ | j s
J �| j �� }| j|kr<| j�|� | j�|� || _d S r4 )r> �get_bridge_configsr? rD rG rE )r) r? r# r# r* rH � s
zBridge.packages_loadedc s | j �� t� �� S r4 )rD Z
cancel_promptrI � eof_receivedr( rJ r# r* rk � s
zBridge.eof_received)r7 r8 r9 r r: r r r r �argparse� Namespacer1 � staticmethodr] r rc ri rH �boolrk �
__classcell__r# r# rJ r* r< ? s
%
r< rd c � s� t �d� t�t�� �}|jtjd< |jtjd< |j tjd< t �d� t
| �}t t �
� |� t �d� z|�� I d H W n ttfy� Y n0 d S )NzHi. How are you today?�HOMEZSHELLZUSERzStarting the router.z/Startup done. Looping until connection closes.)rQ �debug�pwd�getpwuid�os�getuid�pw_dirrC �pw_shell�pw_namer< r �asyncioZget_running_loopZ communicate�BrokenPipeError�ConnectionResetError)r@ �me�routerr# r# r* �run� s
r c
C s� z�t �� \} }| �` |�& tjdd|�� gd� W d � n1 sD0 Y t �| dd�\}}}}W d � n1 sx0 Y W n ty� Y d S 0 z6|\}t�dd� t�|d� W |D ]}t� |� q�n|D ]}t� |� q�0 d S )N� zcockpit.send-stderr)�fdsr z ferny.end)
r- r. r rg �filenoZrecv_fds�OSErrorru �dup2�close)ZoursZtheirsZ_msgr� Z_flagsZ_addrZ stderr_fdr! r# r# r* �try_to_receive_stderr� s 48 �r� �rr c C sp t �dd�}tjdd� | s$|dkr8t�� jtjd� n4|rl|�d�D ]$}|�� }|sXqFt�|��tj� qFdS ) zOSetup our logger with optional filtering of modules if COCKPIT_DEBUG env is setZ
COCKPIT_DEBUG� z#%(name)s-%(levelname)s: %(message)s)�format�all)�level�,N) ru �getenv�loggingZ basicConfig� getLoggerZsetLevel�DEBUGrV rT )rr �modules�moduler# r# r* �
setup_logging� s r� c
C s� zxt jg d�t jt jdd�} | jd us*J �tj| j�� dd�D ]&}|�d�r>|�ddd�tj d< qvq>| �
� | � � W nJ t y� t
�d � Y n0 ty� } zt
�d
|� W Y d }~n
d }~0 0 d S ) N)z ssh-agent�shz-ecz)echo SSH_AUTH_SOCK=$SSH_AUTH_SOCK; read aT)�stdin�stdoutZuniversal_newlines)Zpunctuation_charszSSH_AUTH_SOCK=r� r �
SSH_AUTH_SOCKz,Couldn't start ssh-agent (FileNotFoundError)z Could not start ssh-agent: %s)�
subprocess�Popen�PIPEr� rW �readlinerU �replaceru rC Z terminate�waitrP rQ rr r� rR )�proc�token�excr# r# r* �start_ssh_agent� s
�
r� F)rF )rF r3 c C s t �� tjdd�}|jdddd� |jdddd� |jd dd
d� |jd dd d� |jd
ddd� |�� }| |_|jr~t� t |j
d� |j r�t � �
� d S |jr�tdt� d�� d S |jr�ttjt � �� dd�� d S dtjvr�|js�t� tt|�|j
d� d S )Nz@cockpit-bridge is run automatically inside of a Cockpit session.)� descriptionrB �
store_truez Privileged copy of the bridge)�action�helpz
--packagesz Show Cockpit package informationz --bridgesz Show Cockpit bridges informationz--debugz"Enable debug output (very verbose)z --versionz Show Cockpit version informationr� z Version: z
Protocol: 1r� )�indentr� )r �installrl �ArgumentParser� add_argument�
parse_argsrF rA r� r� rr r> r Zshowrh �printr
Zbridges�json�dumpsrj ru rC r� r
r )rF �parserr@ r# r# r* �main� s0
r� �__main__)@rl rz r` r� r� ru rs rW r- r� �typingr r r r r r Zcockpit._vendor.fernyr Z cockpit._vendor.systemd_ctypesr r
r� r Z_versionr
Zchannelr Zchannelsr Zconfigr r Zinternal_endpointsr Zjsonutilr r r r> r r r Zpeerr Zremoter r~ r r_ r Z
transportsr r� r7 rQ r r< r r� ro r� r� r� r# r# r# r* �<module> sF
e+