File: //proc/self/root/usr/lib/python3.9/site-packages/cockpit/__pycache__/superuser.cpython-39.pyc
a
���e�$ � @ s: d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
m Z m Z m
Z
d dlmZ d dlmZ d dlmZmZ ddlmZ dd lmZmZ dd
lmZ dd l m Z m Z m Z dd l m!Z! dd
l"m#Z#m$Z$m%Z% e�&e'�Z(G dd� de �Z)G dd� dej*�Z+G dd� de+�Z,G dd� de%e+ej-dd�Z.dS )� N)�TemporaryDirectory)�List�Optional�Sequence�Tuple)�ferny)�make_bootloader)�Variant�bus� )�BridgeBeibootHelper)�
JsonObject�get_str)� BridgeConfig)�ConfiguredPeer�Peer� PeerError)� PolkitAgent)�Router� RoutingError� RoutingRulec sB e Zd ZU ejed< eeejd�� fdd� Zdd�dd�Z � Z
S ) �
SuperuserPeer� responder)�router�configr c s t � �||� || _d S �N)�super�__init__r )�selfr r r �� __class__� �5/usr/lib/python3.9/site-packages/cockpit/superuser.pyr - s zSuperuserPeer.__init__N��returnc � st t �� 4 I d H ��F}d| jv rDt�d| j� |�t| j��I d H nt�d| j� t� | j�}d| j
v r�|� t � �}t�
|�}d|� �g}n| j
}| j| j||dd�I d H }d| jv r�t�d | j� t| d
g�}|�|� |�t|jtjd ��� � z|�� I d H W n: tj�y@ } z td t|�d
�|�W Y d }~n
d }~0 0 W d �I d H �qp1 I d H �sf0 Y d S )NZpkexecz-connecting polkit superuser peer transport %rz1connecting non-polkit superuser peer transport %rzSUDO_ASKPASS=ferny-askpassz
SUDO_ASKPASS=T)�stderrZstart_new_sessionz# cockpit-bridgez$going to beiboot superuser bridge %rz --privileged)Zgadgetszauthentication-failed)�message)�
contextlib�AsyncExitStack�args�logger�debug�enter_async_contextr r r ZInteractionAgent�env�
enter_contextr Zwrite_askpass_to_tmpdirZspawnr Z add_handler�writer ZstepsZBEIBOOT_GADGETS�encodeZ communicateZInteractionErrorr �str) r �contextZagentZtmpdirZ
ferny_askpassr- Z transport�helper�excr! r! r" �do_connect_transport1 s*
z"SuperuserPeer.do_connect_transport) �__name__�
__module__� __qualname__r �InteractionResponder�__annotations__r r r r5 �
__classcell__r! r! r r" r * s
r c @ s* e Zd ZdZeeee edd�dd�ZdS )�CockpitResponder)z
ferny.askpass�cockpit.send-stderrN)�commandr) �fdsr% r$ c
� sj |dkrft j |d d��>}|�d� |�dgt jt jt�ddg�fg� W d � n1 s\0 Y d S )Nr= r )�fileno� �i� )�socket�popZsendmsgZ
SOL_SOCKETZ
SCM_RIGHTS�array)r r> r) r? r% Zsockr! r! r" �do_custom_commandS s
z"CockpitResponder.do_custom_command) r6 r7 r8 Zcommandsr1 r r �intrG r! r! r! r" r< P s r< c @ s. e Zd Zed�dd�Zeeeed�dd�ZdS )�AuthorizeResponder�r c C s
|| _ d S r rJ )r r r! r! r" r \ s zAuthorizeResponder.__init__��messages�prompt�hintr$ c � s6 d� dd� t�� �d�D ��}| j�d|� ��I d H S )N� c s s | ]}|d �V qdS )Z02xNr! ��.0�cr! r! r" � <genexpr>` � z0AuthorizeResponder.do_askpass.<locals>.<genexpr>�asciizplain1:)�join�getpassZgetuserr0 r Zrequest_authorization)r rL rM rN Zhexuserr! r! r" �
do_askpass_ s z AuthorizeResponder.do_askpassN)r6 r7 r8 r r r1 rX r! r! r! r" rI [ s rI c st e Zd ZU dZee ed< eej ed< ee
ed< e j �
ddddd�Ze j jdg d�Ze j jdd d�Ze j jd
i d�Zeee d �d d
�Zeeeee d�dd�Zdd�eed�� fdd�Zdd� Zee j dd�dd�Z ee d �d d �Z d d � Z d!d"� Z!edd#�d$d%�Z"d&d'� Z#e j j$dgd(�edd)�d*d+��Z%e j �$� dd,�d-d.��Z&e j j$dgd(�edd/�d0d1��Z'� Z(S )2�SuperuserRoutingRuler! �superuser_configs�pending_prompt�peer�s�b�as)�value�none�a{sv})�optionsr$ c C s< |� d�}|r| jdkr d S | js*|dkr0| jS td��d S )N� superuser�root�tryz
access-denied)�get�currentr\ r )r rc rd r! r! r" �
apply_ruler s
z SuperuserRoutingRule.apply_rulerK c � sd | j d u sJ �|dk}t�� �� | _ z2t�d|� | �d|d||� | j I d H W d | _ S d | _ 0 d S )NZconfirmzprompting for %srO )r[ �asyncioZget_running_loopZ
create_futurer* r+ rM )r rL rM rN Zechor! r! r" rX � s �z SuperuserRoutingRule.do_askpassF��
privileged)r rl c s8 t � �|� d | _d | _d | _|s.t�� dkr4d| _d S )Nr re )r r r[ r\ �startup�os�getuidrh )r r rl r r! r" r � s z SuperuserRoutingRule.__init__c C s d| _ d | _d S )Nra )rh r\ �r r! r! r" � peer_done� s z SuperuserRoutingRule.peer_doneN)�namer r$ c
� s | j dkrt�dd��| jd u s$J �| jd u s2J �| jD ]}||jdfv r8 qdq8t�dd|� d���d| _ t| j||�| _| j� | j
� z | jj | jj d�I d H W nZ t
jy� t�d d
�d �Y n: ttfy� } z t�dt|��|�W Y d }~n
d }~0 0 | jjj| _ d S ) Nra zcockpit.Superuser.Errorz Superuser bridge already running�anyz Unknown superuser bridge type "�"�init)� init_hostz!cockpit.Superuser.Error.CancelledzOperation aborted)rh r
ZBusErrorr\ rm rZ rr r r �add_done_callbackrq �startrv rj ZCancelledError�OSErrorr r1 r )r rr r r r4 r! r! r" �go� s$
(zSuperuserRoutingRule.go)�configsc C s� t �dt|�� dd� |D �}t|�| _dd� | jD �| _dd� |D �| _t �d| j� | jd ur�| jj| jvr�t �d| jjj � | �
� d S ) Nzset_configs() with %d itemsc S s g | ]}|j r|�qS r! rk �rQ r r! r! r" �
<listcomp>� rT z4SuperuserRoutingRule.set_configs.<locals>.<listcomp>c S s g | ]
}|j �qS r! )rr r| r! r! r" r} � rT c S s* i | ]"}|j r|j td t|j �id��qS )�labelrb )r~ r rP r! r! r" �
<dictcomp>� rT z4SuperuserRoutingRule.set_configs.<locals>.<dictcomp>z bridges are now %sz= stopping superuser bridge '%s': it disappeared from configs) r* r+ �len�tuplerZ �bridges�methodsr\ r rr �stop)r r{ r! r! r" � set_configs� s
z SuperuserRoutingRule.set_configsc C s | j d ur| j �� d | _ d S r )r[ �cancelrp r! r! r" �
cancel_prompt� s
z"SuperuserRoutingRule.cancel_promptc C s. | � � | jd ur | j�� | jd u s*J �d S r )r� r\ �closerp r! r! r" �shutdown� s
z SuperuserRoutingRule.shutdown)�paramsr$ c C s< t |dd�}t| j�}t�| �||��| _| j�| j� d S )N�idrs ) r rI r rj Z create_taskrz �
_init_taskrw �
_init_done)r r� rr r r! r! r" ru � s
zSuperuserRoutingRule.initc C s& t �d|�� � | jjdd� | `d S )Nzsuperuser init done! %szsuperuser-init-done)r> )r* r+ � exceptionr Z
write_controlr� )r Ztaskr! r! r" r� � s z SuperuserRoutingRule._init_done)Zin_types)rr r$ c � s | � || �I d H d S r )rz )r rr r! r! r" rx � s zSuperuserRoutingRule.startr# c C s | � � d S r )r� rp r! r! r" r� � s zSuperuserRoutingRule.stop)�replyr$ c C s0 | j d ur"t�d� | j �|� n
t�d� d S )Nz responding to pending promptz!got Answer, but no prompt pending)r[ r* r+ Z
set_result)r r� r! r! r" �answer� s
zSuperuserRoutingRule.answer))r6 r7 r8 rZ r r r: r rj ZFuturer r
Z InterfaceZSignalrM ZPropertyr� rh r� r
r ri r1 rX r �boolr rq r r9 rz r� r� r� ru r� ZMethodrx r� r� r; r! r! r r" rY d s.
rY zcockpit.Superuser)Z interface)/rF rj r' rW Zloggingrn rD Ztempfiler �typingr r r r Zcockpit._vendorr Z cockpit._vendor.bei.bootloaderr Z cockpit._vendor.systemd_ctypesr r
Zbeipackr Zjsonutilr
r Zpackagesr r\ r r r Zpolkitr r r r r Z getLoggerr6 r* r r9 r< rI ZObjectrY r! r! r! r" �<module> s,
&