Current File : //usr/lib/python3.9/site-packages/cockpit/channels/__pycache__/dbus.cpython-39.opt-1.pyc
a
��h�V � @ s� d dl Z d dlZd dlZd dlZd dlZd dlm mZ d dl m
Z
d dl m Z m
Z
mZ ddlmZmZ e�e�ZG dd� d�Zdd � ZG d
d � d e�ZdS ) � N)�systemd_ctypes)�Bus�BusError�
introspection� )�Channel� ChannelErrorc @ sF e Zd Zdd� Zdd�dd�Zdd� Zdd d
�Zd d � Zdd
d�ZdS )�InterfaceCachec C s i | _ t� | _d S �N)�cache�set�old��self� r �9/usr/lib/python3.9/site-packages/cockpit/channels/dbus.py�__init__t s zInterfaceCache.__init__N��returnc C s | j �|� d S r
)r �update)r �
interfacesr r r �injectx s zInterfaceCache.injectc � sD |� ||dd�I d H \}t�|�}dd� |�d�D �}| �|� |S )Nz#org.freedesktop.DBus.IntrospectableZ
Introspectc S s i | ]}|j d t�|��qS )�name)Zattribr Zparse_interface)�.0�tagr r r �
<dictcomp>� � z2InterfaceCache.introspect_path.<locals>.<dictcomp>� interface)�call_method_async�ETZ
fromstring�findallr )r �bus� destination� object_pathZxml�etr r r r �introspect_path{ s �
z InterfaceCache.introspect_pathc � s` z | j | W S ty Y n0 |rT|rTz| �|||�I d H W n tyR Y n0 | j �|�S r
)r �KeyErrorr% r �get�r �interface_namer! r" r# r r r �
get_interface� s z InterfaceCache.get_interfacec � s0 || j v rd S | j �|� | �||||�I d H S r
)r
�addr* r( r r r �get_interface_if_new� s
z#InterfaceCache.get_interface_if_newc � sD | � ||||�I d H }|d u r.td|� d���d�|d | d �S )Nz
Interface z
is not found� �methods�in)r* r& �join)r r) �methodr! r" r# r r r r �
get_signature� s z InterfaceCache.get_signature)NNN)NNN) �__name__�
__module__� __qualname__r r r% r* r, r2 r r r r r s s
r c C s" dd� |� � D �| �|i �|< d S )Nc S s i | ]\}}||j �qS r )�value)r �k�vr r r r � r z!notify_update.<locals>.<dictcomp>)�items�
setdefault)�notify�pathr) �propsr r r �
notify_update� s r> c @ s� e Zd Zejdd�ZdZdZdZdZ dZ
dd� Z dd� Z d d
� Z
d d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � ZdS ) � DBusChannelr )�indentz
dbus-json3Nc
� s � fdd���fdd�}� j |dddd� jd� z$� j�dddd d
� j�I d H \}W n� ty� } z�|jd kr�z$� j�dddd d
� jd�I d H W q� ty� } z*t�d� j|j� � jd d� W Y d }~q�d }~0 0 nt�d� j|j� W Y d }~nd }~0 0 �|� d S )Nc s � j | kr | � _ � j| d� d S )N��owner)rB � send_jsonrA r r r �
send_owner� s
z9DBusChannel.setup_name_owner_tracking.<locals>.send_ownerc s( | � � \}}}� |dkr |nd d� d S )Nr- rA )�get_body)�message�_nameZ_old�new)rD r r �handler� s z6DBusChannel.setup_name_owner_tracking.<locals>.handlerzorg.freedesktop.DBusz/org/freedesktop/DBusZNameOwnerChanged)�senderr< r �memberZarg0Z GetNameOwner�sz)org.freedesktop.DBus.Error.NameHasNoOwnerZStartServiceByNameZsur z Failed to start service '%s': %srA z'Failed to get owner of service '%s': %s) �add_signal_handlerr r! r r �logger�debugrF rC )r rI � unique_name�errorZ start_errorr )r rD r �setup_name_owner_tracking� s8 ��
�$(z%DBusChannel.setup_name_owner_trackingc
s� t � � _|�d�� _g � _|�d�}|�d�}z�|d ur||d urT|dkrTtddd��t�d|� j� tj |� jd ud �� _
n�|d
kr�t�d � j� � j j j
� _
n`|d kr�t�d
� j� t�� � _
n>|dks�|d u r�t�d� j� t�� � _
ntdd|� d�d��W n@ t�yB } z&tdd|� d|� �d�|�W Y d }~n
d }~0 0 z� j
�d d� W n6 t�y� } z |jtjk�rx� W Y d }~n
d }~0 0 t�� � _� jd u�r�d d�� fdd� }� �|� � n� �� d S )Nr r! �addressZnonezprotocol-errorz0only one of "bus" and "address" can be specified�rF z get bus with address %s for %s)rS Z
bus_clientZinternalzget internal bus for %sZsessionzget session bus for %s�systemzget system bus for %sz
invalid bus "�"zfailed to connect to z bus: r r c � sp � j 4 I d H �H � �� I d H � jr4� j� jd� n� �ddi� W d �I d H ql1 I d H sb0 Y d S )N)rP Zproblemz not-found)�watch_processing_lockrR rB �ready�closer r r r � get_ready s
z&DBusChannel.do_open.<locals>.get_ready)r r r' r �matchesr rN rO r rH r! ZrouterZ internal_busZclientZ default_userZdefault_system�OSErrorZ attach_event�errnoZEBUSY�asyncio�LockrW � create_taskrX )r �optionsr! rS �exc�errrZ r r r �do_open� sB
0
zDBusChannel.do_openc s� t f i |��}d|d< d|vr2�jd ur2�j|d< |�d�dkrF|d= � �fdd�}�jd urzd|v rz|d �jkrz|}n� }d�d d
� |�� D ��}��� s�z�j�||�}W q�W q� ty� Y q�0 q��j �
|� d S ) N�signal�typerJ �path_namespace�/c s$ �j d ur �j | �� kr � | � d S r
)rB Z
get_senderrT �rI r r r � filter_owner s z4DBusChannel.add_signal_handler.<locals>.filter_owner�,c s s" | ]\}}|� d |� d�V qdS )z='�'Nr )r �keyr6 r r r � <genexpr> r z1DBusChannel.add_signal_handler.<locals>.<genexpr>) �dictr r' r0 r9 Z
is_closingr! � add_match�InterruptedErrorr[ �append)r rI �kwargs�rrj �funcZr_string�matchr ri r rM
s$
z DBusChannel.add_signal_handlerc s$ � �fdd�}�j |fi |�� d S )Nc s �� � | �� d S r
)r` rT ri r r � sync_handler, s z:DBusChannel.add_async_signal_handler.<locals>.sync_handler)rM )r rI rs rw r ri r �add_async_signal_handler+ s z$DBusChannel.add_async_signal_handlerc
� sx |d \}}}}|� d�}|� d�}|� d�}|d ur@|d9 }nd}|� d�} | d u rft|�dkrfd } | d u �r\z0t�d
||� | j�||| j| j|�I d H } W n� ty� }
z,| j |
jd |
j
� �gg|d � W Y d }
~
d S d }
~
0 t �y | j d
d|� d|� d�gg|d � Y d S t �yZ } z(| j dd | ��gg|d � W Y d } ~ d S d } ~ 0 0 z�| jj
| j|||| g|�R � } | jj| |d�I d H }
| j4 I d H �D | j |
�� g||d u�r�dnd |
�d�d� W d �I d H �q�1 I d H �s�0 Y W nt t�y< }
z$| j |
j|
j
gg|d � W Y d }
~
n@d }
~
0 t �yr t�d|� | j dt�� gg|d � Y n0 d S )N�call�id�flags�timeouti� l ���� rf r r- z%Doing introspection request for %s %szIntrospection: �rQ rz z(org.freedesktop.DBus.Error.UnknownMethodz Introspection data for method � z not availablez python.error)r| �<T)�replyrz r{ rf z do_call(%s): generic exception)r' �lenrN rO r r2 r! r r rC rF r& � ExceptionZmessage_new_method_callZ
call_asyncrW rE � exception� traceback�
format_exc)r rF r< �ifacer1 �args�cookier{ r| Z signaturerQ rb Z method_callr� r r r �do_call0 sT
" ��
�8, zDBusChannel.do_callc � s6 |d }t �d|� � fdd�}� j|fi |�� d S )N� add-matchzadding match %sc � sp t �d� � j4 I d H �> � j| �� | �� | �� t| �� �gd� W d �I d H ql1 I d H sb0 Y d S )Nz got match)re ) rN rO rW rC �get_pathr* �
get_member�listrE rT r r r � match_hitm s
�z+DBusChannel.do_add_match.<locals>.match_hit)rN rO rx )r rF rp r� r r r � do_add_matchi s
zDBusChannel.do_add_matchc � s� � �fdd�}�j ||dd� �j��j|dd�I d H \}|�� D ]f\}}|�� D ]T\} }
� d u sj| � krR�j�| �j�j|�I d H } | r�|�| | i� t||| |
� qRqBd S )Nc
� sd | � � }|dkr�| �� \}}t�d||� i }i }�j4 I d H �� |�� D ]T\}}� d u sb|� krJ�j�|�j�j |�I d H }|r�|�
||i� t ||||� qJ�j |d� �j |d� W d �I d H q�1 I d H s�0 Y n||dk�r`| �� \}} t�d|| � �j4 I d H �2 |t
�| �i}�j |d� W d �I d H �q`1 I d H �sV0 Y d S )NZInterfacesAddedzinterfaces added %s %s��meta�r; ZInterfacesRemovedzinterfaces removed %s %s)r� rE rN rO rW r9 r r, r! r r r> rC ro �fromkeys)
rF rK r< Zinterface_propsr� r; r r= �mmr �r) r r r rI ~ s* 8
z6DBusChannel.setup_objectmanager_watch.<locals>.handlerz"org.freedesktop.DBus.ObjectManager)r< r ZGetManagedObjects) rx r! r r r9 r r, r r> ) r r< r) r� r; rI Zobjects�pZifacesr� r= r� r r� r �setup_objectmanager_watchy s � � z%DBusChannel.setup_objectmanager_watchc � s� � fdd�}� j �� j� j|�I d H }|d ur@|�|�}||i}|�|� |r`� j|d|d� n� j|d|d� |D ]V} | �d�r�qtz2� j�� j|ddd| �I d H \}
t ||| |
� W qt t
y� Y qt0 qtd S ) Nc � s
� j 4 I d H �� | �� }| �� \}}}t�d||||� |D ]~}z&� j�� j|ddd||�I d H \}W nJ ty� } z2t�d||� j|t |�� W Y d }~q<W Y d }~n
d }~0 0 |||< q<i }t
||||� � j |d� W d �I d H �q1 I d H s�0 Y d S )NzNOTIFY: %s %s %s %s� org.freedesktop.DBus.PropertiesZGet�ssz+failed to fetch property %s.%s on %s %s: %sr� ) rW r� rE rN rO r! r r r �strr> rC ) rF r< r r= Zinvalids�invr� rb r; r r r rI � s& ��"
z-DBusChannel.setup_path_watch.<locals>.handlerr� )r rg )r r< zorg.freedesktop.DBus.ZGetAllrL ) r r% r! r r' r rx �
startswithr r> r ) r r< r) Zrecursive_propsr� r; rI Z this_metar r r= r r r �setup_path_watch� s4
��
� z DBusChannel.setup_path_watchc
� s| |d }|� d�}|� d�}|� d�}|� d�}|p6|}||k}|d u sP|d u r�t�d|� | jddgg|d � | jg |d
� d S z�| j4 I d H �| i }i } | �||||| �I d H |r�| �|||| �I d H | j|d � | j| d � | jg |d d
� W d �I d H �q&1 I d H �s 0 Y W nN t�yv }
z4t�d
||
j� | j|
j |
jgg|d � W Y d }
~
n
d }
~
0 0 d S )N�watchr< rg r rz z#ignored incomplete watch request %szx.y.zzNot Implementedr} )r� rz r� r� z#do_watch(%s) caught D-Bus error: %s)
r' rN rO rC rW r� r� r rF r ) r rF r� r< rg r) r� � recursiver� r; rQ r r r �do_watch� s2
DzDBusChannel.do_watchc � s | j �|d � d S )Nr� )r r )r rF r r r �do_meta� s zDBusChannel.do_metac C s� t �|�}t�d| j|� d|v r4| �| �|�� n^d|v rN| �| �|�� nDd|v rh| �| �|�� n*d|v r�| �| � |�� nt�d|� d S d S )Nzreceive dbus request %s %sry r� r� r� zignored dbus request %s)
�json�loadsrN rO r r` r� r� r� r� )r �datarF r r r �do_data� s
zDBusChannel.do_datac C s& | j D ] }|�� qg | _ | �� d S r
)r[ �cancelrY )r Zslotr r r �do_close s
zDBusChannel.do_close)r3 r4 r5 r Z JSONEncoderZ json_encoderZpayloadr[ r r! rB rR rd rM rx r� r� r� r� r� r� r� r� r r r r r? � s$ (3 9*0 r? )r^ r] r� Zloggingr� Zxml.etree.ElementTreeZetreeZ ElementTreer Zcockpit._vendorr Z cockpit._vendor.systemd_ctypesr r r Zchannelr r Z getLoggerr3 rN r r> r? r r r r �<module>% s
B2
Mini Shell