Current File : //usr/lib/python3.9/site-packages/cockpit/__pycache__/bridge.cpython-39.pyc
a
��h0 � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl m
Z
mZmZmZmZmZmZ d dlmZ d dlmZmZ ddlmZ ddlm Z ddl m Z dd l m Z dd
l!m"Z"m#Z# dd l$m%Z% dd l&m'Z'm(Z(m)Z)m*Z* dd
l+m,Z,m-Z-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 ddl5m6Z6 ddl7m8Z8 e�9e:�Z;G dd� d�Z<G dd� de4e.�Z=dd�dd�Z>dd�dd�Z?e@d�d d �ZAe@dd �d d �ZBeCe
eCeCf d!�d"d#�ZDdd�d$d%�ZEd&d'�e@dd(�d)d*�ZFe:d+k�reF� dS ),� N)�Dict�Iterable�List�Optional�Sequence�Tuple�Type)�interaction_client)�bus� run_async� )� polyfills)� __version__)�ChannelRoutingRule)�
CHANNEL_TYPES)�Config� Environment)�EXPORTS)� JsonError�
JsonObject� JsonValue�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>; � z(InternalBus.__init__.<locals>.<listcomp>) �socket�
socketpairr
ZBus�new�detachZclientr$ r! )r+ r" Z
client_socketZ
server_socketr% r* r, �__init__7 s zInternalBus.__init__N)r( �obj�returnc C s | j �| j�||�� d S �N)r! �appendr$ r& )r+ r( r4 r% r% r, �export= s zInternalBus.export)�__name__�
__module__� __qualname__r r
ZSlot�__annotations__r r �strr Z
BaseObjectr3 r8 r% r% r% r, r 4 s
r c s� e Zd ZU eed< eed< ee ed< ee ed< e
j ed< e
j d�� fdd� Z e
d �d
d �Zee
d �d d
��Ze
dd�dd�Zdd �dd�Zdd �dd�Z� ZS )�Bridge�channels� internal_bus�packages�bridge_configs�args)rC 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�| _t� �t| �| j| j| j g� d S )
N)�
privilegedz
/superuserz/configz /environmentT)Zsudoz-kz-AZpython3z-icz# cockpit-bridge� --privilegedzSUDO_ASKPASS=ferny-askpass)rD Zspawn�environz /packages)r r r@ rB rC r rD �superuser_ruler8 r r r �
peers_rule�beipack� set_configsr rA r �packages_loadedr r r? �superr3 r )r+ rC �� __class__r% r, r3 H s:
��
�zBridge.__init__�r5 c C s� t �t�� �}t�|j�j}|g}t�� D ]&}|j |j
v r*|j|vr*|� |j� q*| j �
� | �� |j|j|||j|j |j|jd�d�S )N)�fullname�gid�group�groups�home�name�shell�uid)r? �
os_release�user)�pwd�getpwuid�os�getuid�grp�getgrgid�pw_gid�gr_name�getgrall�pw_name�gr_memr7 r? � capabilities�get_os_release�pw_gecos�pw_dir�pw_shell�pw_uid)r+ ZpwrR rS Zgrr% r% r, �infoo s$ ��z Bridge.infoc C sf zt ddd�} W nH tyX zt ddd�} W n$ tyR t�d� i Y Y S 0 Y n0 t| �� �S )Nz/etc/os-releasezutf-8)�encodingz/usr/lib/os-releasez6Neither /etc/os-release nor /usr/lib/os-release exists)�open�FileNotFoundError�logger�warning�parse_os_release�read)�filer% r% r, rf � s
zBridge.get_os_releaseN)�messager5 c C sD t �t��& t|d�}| j�|� W d � n1 s60 Y d S )N� superuser)�
contextlib�suppressr r rG �init)r+ rt ru r% r% r, �do_init� s
zBridge.do_initc C s@ ddid| � � dd�}| jd ur2t�| jj�|d< | �|� d S )Nzexplicit-superuserTrx r )re �commandz
os-release�versionrA )rf rA �dict�fromkeysZ
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 r6 )rA �get_bridge_configsrB rG rJ rH )r+ rB r% r% r, rK � s
zBridge.packages_loaded)r9 r: r; r r< r r r r r �argparse� Namespacer3 r rk � staticmethodrf ry r~ rK �
__classcell__r% r% rM r, r> A s
' r> rO 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.)ro �debugrZ r[ r\ r] rh rF ri rc r> r �asyncioZget_running_loopZ communicate�BrokenPipeError�ConnectionResetError)rC �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/ r0 r rz �filenoZrecv_fds�OSErrorr\ �dup2�close)ZoursZtheirsZ_msgr� Z_flagsZ_addrZ stderr_fdr# r% r% r, �try_to_receive_stderr� s 48 �r� c C sN t �t�d�j�sdS z dd l} W n ty6 Y dS 0 tj� | j
� � � dS )Nr� Fr T) �stat�S_ISSOCKr\ �fstat�st_modeZsystemd.journal� ImportError�logging�rootZ
addHandlerZjournalZJournalHandler)Zsystemdr% r% r, �setup_journald� s r� )r� r5 c C sv t �dd�}t� s tjdd� | s*|dkr>t�� jtjd� n4|rr|�d�D ]$}|� � }|s^qLt�|��tj� qLdS ) 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)
r\ �getenvr� r� Z basicConfig� getLoggerZsetLevel�DEBUG�split�strip)r� �modules�moduler% r% r, �
setup_logging� s r� )�textr5 c C sx i }| � � D ]f}|�� }|r |�d�r(q z |�d�\}}t�|�\}W n" tyh t�d|� Y q Y n0 |||< q |S )N�#�=z'Ignoring invalid line in os-release: %r)�
splitlinesr� �
startswithr� �shlex�
ValueErrorro rp )r� rX �line�k�vZv_parsedr% r% r, rq s
rq 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� r� �readliner� �replacer\ rF Z terminate�waitrn ro r� r� rp )�proc�token�excr% r% r, �start_ssh_agent s
�
r� F)rI )rI r5 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�ttjdd� t � �� D �dd�� d S dtjvr�|js�t� tt|�|j
d� d S )Nz@cockpit-bridge is run automatically inside of a Cockpit session.)� descriptionrE �
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 information)r� z Version: z
Protocol: 1c S s g | ]
}|j �qS r% )�__dict__)r' �configr% r% r, r- Q r. zmain.<locals>.<listcomp>r� )�indentr� )r
�installr� �ArgumentParser� add_argument�
parse_argsrI rD r� r� r� rA r Zshowr{ �printr Zbridges�json�dumpsr r\ rF r� r r� )rI �parserrC r% r% r, �main1 s0
"r� �__main__)Gr� r� rv r^ r� r� r\ rZ r� r/ r� r� �typingr r r r r r r Zcockpit._vendor.fernyr Z cockpit._vendor.systemd_ctypesr
r r� r
�_versionr Zchannelr r? r r� r r Zinternal_endpointsr Zjsonutilr r r r rA r r r Zpeerr Zremoter r� r ru r Z
transportsr r� r9 ro r r> r� r� �boolr� r� r= rq r� r� r% r% r% r, �<module> sN $
s+
Mini Shell