Google
Edit File: 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+