Current File : //lib/python3.9/site-packages/cockpit/__pycache__/bridge.cpython-39.pyc
a
��j�. � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
mZmZmZmZmZmZmZ d dlmZ d dlmZmZ ddlmZ ddl m 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/ ddl0m1Z1 ddl2m3Z3 ddl4m5Z5 ddl6m7Z7 ddl8m9Z9 e�:e;�Z<G dd� d�Z=G dd� de5e/�Z>dd�dd�Z?dd�dd�Z@eAd�d d �ZBeAdd �d d �ZCeDeeDeDf d!�d"d#�ZEdd�d$d%�ZFd&d'�eAdd(�d)d*�ZGe;d+k�r eG� 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>0 � z(InternalBus.__init__.<locals>.<listcomp>) �socket�
socketpairr
ZBus�new�detachZclientr$ r! )r+ r" Z
client_socketZ
server_socketr% r* r, �__init__, s zInternalBus.__init__N)r( �obj�returnc C s | j �| j�||�� d S �N)r! �appendr$ r& )r+ r( r4 r% r% r, �export2 s zInternalBus.export)�__name__�
__module__� __qualname__r r
ZSlot�__annotations__r r �strr Z
BaseObjectr3 r8 r% r% r% r, r ) 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��tddd tjd
d d gd
��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-AZpython3�-ic�# cockpit-bridge� --privilegedzSUDO_ASKPASS=ferny-askpass)rD �spawn�environZStartTransientUnitrE rF rG )rD Zpolkit�methodrH z /packages)r r r@ rB rC r rD �superuser_ruler8 r r r �
peers_rule�beipack� set_configsr �sys�
executablerA r �packages_loadedr r r? �superr3 r )r+ rC �� __class__r% r, r3 = sF
� ��
�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+ ZpwrX rY Zgrr% r% r, �infoj 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, rl � 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 rK �init)r+ rz r{ r% r% r, �do_init� s
zBridge.do_initc C s@ ddid| � � dd�}| jd ur2t�| jj�|d< | �|� d S )Nzexplicit-superuserTr~ r )rk �commandz
os-release�versionrA )rl 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 rK rN rL )r+ rB r% r% r, rQ � s
zBridge.packages_loaded)r9 r: r; r r< r r r r r �argparse� Namespacer3 r rq � staticmethodrl r r� rQ �
__classcell__r% r% rS r, r> 6 s
- r> rU 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.)ru �debugr` ra rb rc rn rI ro ri 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 r� �filenoZrecv_fds�OSErrorrb �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_ISSOCKrb �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)
rb �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�
ValueErrorru rv )r� r^ �line�k�vZv_parsedr% r% r, rw s
rw 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� �replacerb rI Z terminate�waitrt ru r� r� rv )�proc�token�excr% r% r, �start_ssh_agent s
�
r� F)rM )rM 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.)� descriptionrG �
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- L r. zmain.<locals>.<listcomp>r� )�indentr� )r
�installr� �ArgumentParser� add_argument�
parse_argsrM rD r� r� r� rA r Zshowr� �printr Zbridges�json�dumpsr� rb rI r� r r� )rM �parserrC r% r% r, �main, s0
"r� �__main__)Hr� r� r| rd r� r� rb r` r� r/ r� r� rO �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 r{ r Z
transportsr r� r9 ru r r> r� r� �boolr� r� r= rw r� r� r% r% r% r, �<module> sP $
y+
Mini Shell