|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/jeepney/io/__pycache__/ |
Upload File : |
o
���` : � @ s� d dl Z d dlmZ d dlZd dlmZ d dlZd dlmZ zd dlm Z W n e
y5 d dl m Z Y nw d dl m
Z
mZ d dlZd dlmZ d dlmZmZ d d lmZ d d
lmZmZ d d lmZm Z m Z d d l m Z m Z d d
l!m"Z" ddl#m$Z$m%Z%m&Z&m'Z'm(Z( e�)e*�Z+g d�Z,edd� �Z-G dd� de�Z.d%dd�de.fdd�Z/G dd � d e%�Z0G d d � d �Z1G d d � d �Z2G d!d"� d"e �Z3e d%dd�d#d$��Z4dS )&� N)�contextmanager)�count)�Optional)�asynccontextmanager)�Value�Error)�Channel)�
Authenticator�BEGIN)�get_bus)�FileDescriptor� fds_buf_size)�Parser� MessageType�Message)� ProxyBase�
unwrap_msg)� message_bus� )�MessageFilters� FilterHandle� ReplyMatcher� RouterClosed�check_replyable)�open_dbus_connection�open_dbus_router�Proxyc
c sX � zd V W d S t y+ } z| jtjtjhv r t�d�d �t�d�| ��| �d } ~ ww )Nz this socket was already closedz socket connection broken: {})�OSError�errno�EBADF�ENOTSOCK�trio�ClosedResourceError�BrokenResourceError�format)�exc� r& �1/usr/lib/python3/dist-packages/jeepney/io/trio.py�)_translate_socket_errors_to_stream_errors0 s � ����r( c @ s~ e Zd ZdZddd�Zdd�defdd �Zd
efd d �Zd d
e fdd�Z
defdd�Z dd� Z dd� Z
dd� Zedd� �ZdS ) �DBusConnectiona� A plain D-Bus connection with no matching of replies.
This doesn't run any separate tasks: sending and receiving are done in
the task that calls those methods. It's suitable for implementing servers:
several worker tasks can receive requests and send replies.
For a typical client pattern, see :class:`DBusRouter`.
Implements trio's channel interface for Message objects.
Fc C sD || _ || _t� | _tdd�| _d | _t�� | _ t�� | _
d | _ d S )Nr )�start) �socket�
enable_fdsr �parserr �outgoing_serial� unique_namer! �Lock� send_lock� recv_lock�_leftover_to_send)�selfr+ r, r& r& r' �__init__I s
zDBusConnection.__init__N��serial�messagec � s� �| j 4 I dH �/ |du rt| j�}| jrt�d�nd}|j||d�}| �||�I dH W d �I dH dS 1 I dH s=w Y dS )z.Serialise and send a :class:`~.Message` objectN�i)�fds)r1 �nextr. r, �array� serialise�
_send_data)r4 r8 r7 r: �datar& r& r' �sendS s �
.�zDBusConnection.sendr? c � s� �| j jr
t�d��t� �Y | jr| �| j�I d H t|��0}|r5| j �|gtj j tj j
|fg�I d H }n | j � |�I d H }| �||�I d H W d � n1 sQw Y W d � d S W d � d S 1 siw Y d S )Nz!can't send data after sending EOF) r+ �did_shutdown_SHUT_WRr! r" r( r3 �_send_remainder�
memoryview�sendmsg�
SOL_SOCKET�
SCM_RIGHTSr@ )r4 r? r: �sentr& r& r' r> ^ s" �
� ��"�zDBusConnection._send_datar c � s� �z5|t |�k r1||d � �}| j�|�I d H }W d � n1 s"w Y ||7 }|t |�k sd | _W d S tjyF ||d � | _� w �N)�lenr+ r@ r3 r! � Cancelled)r4 r? � already_sent� remainingrG r& r& r' rB q s � � � �z DBusConnection._send_remainder�returnc � s� �| j 4 I dH �/ | j�� }|dur |W d �I dH S | �� I dH \}}|s/t�d��| j�||� q
1 I dH s=w Y dS )z5Return the next available message from the connectionNTz Socket closed at the other end)r2 r- �get_next_message�
_read_datar! � EndOfChannel�add_data)r4 �msg�br: r& r& r' �receive s �
�
��zDBusConnection.receivec � s� �| j rC| j�� }t� � | j�|t� �I d H \}}}}W d � n1 s&w Y |ttjdd�@ r<| � � t
d��|t � |�fS t� � | j�
d�I d H }W d � |g fS 1 s]w Y |g fS )N�
MSG_CTRUNCr z&Unable to receive all file descriptorsi )r, r- �
bytes_desiredr( r+ �recvmsgr
�getattrr! �_close� RuntimeErrorr � from_ancdata�recv)r4 �nbytesr? �ancdata�flags�_r& r& r' rO � s$ �
� �
��zDBusConnection._read_datac C s | j �� d | _d S rH )r+ �closer3 �r4 r& r& r' rY � s
zDBusConnection._closec � s �| � � dS )zClose the D-Bus connectionN)rY rb r& r& r' �aclose� s � zDBusConnection.aclosec C s� �t �� 4 I dH �-}t| �}|�|�I dH z |V W |�� I dH n|�� I dH w W d �I dH dS 1 I dH s<w Y dS )aY Temporarily wrap this connection as a :class:`DBusRouter`
To be used like::
async with conn.router() as req:
reply = await req.send_and_get_reply(msg)
While the router is running, you shouldn't use :meth:`receive`.
Once the router is closed, you can use the plain connection again.
N)r! � open_nursery�
DBusRouterr* rc )r4 �nursery�routerr& r& r' rg � s � ".�zDBusConnection.router)F)r )�__name__�
__module__� __qualname__�__doc__r5 r r@ �bytesr> rC rB rT rO rY rc r rg r& r& r& r' r) ? s
r) �SESSIONF�r, rM c � s� �t | �}t�|�I dH }t|d�}|D ]}|�|�I dH |�|�� I dH � q|�t�I dH t|j |d�}|�
� 4 I dH � }|� t �
� �I dH }|jd |_W d �I dH |S 1 I dH sbw Y |S )zHOpen a plain D-Bus connection
:return: :class:`DBusConnection`
Nrn r )r r! �open_unix_socketr �send_all�feed� receive_somer
r) r+ rg �send_and_get_replyr �Hello�bodyr/ ) �busr, �bus_addr�sock�authr�req_data�connrg �replyr& r& r' r � s �
��r c sF e Zd Zdef� fdd� Zedd� �Zdd� Zdd � Zd
d � Z � Z
S ) �TrioFilterHandle�filtersc s t � �|||� || _d S rH )�superr5 � send_channel)r4 r~ �rule�send_chn�recv_chn�� __class__r& r' r5 � s
zTrioFilterHandle.__init__c C s | j S rH ��queuerb r& r& r' �receive_channel� s z TrioFilterHandle.receive_channelc � s �| � � | j�� I d H d S rH )ra r� rc rb r& r& r' rc � s �zTrioFilterHandle.aclosec � s �| j S rH r� rb r& r& r' �
__aenter__� s �zTrioFilterHandle.__aenter__c � s �| � � I d H d S rH )rc )r4 �exc_type�exc_val�exc_tbr& r& r' � __aexit__� s �zTrioFilterHandle.__aexit__) rh ri rj r r5 �propertyr� rc r� r� �
__classcell__r& r& r� r' r} � s
r} c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �Futurez4A very simple Future for trio based on `trio.Event`.c C s d | _ t�� | _d S rH )�_outcomer! �Event�_eventrb r& r& r' r5 � s zFuture.__init__c C � t |�| _| j�� d S rH )r r� r� �set)r4 �resultr& r& r' �
set_result� �
zFuture.set_resultc C r� rH )r r� r� r� )r4 r% r& r& r' �
set_exception� r� zFuture.set_exceptionc � s �| j �� I d H | j�� S rH )r� �waitr� �unwraprb r& r& r' �get� s �
z
Future.getN)rh ri rj rk r5 r� r� r� r& r& r& r' r� � s r� c @ s� e Zd ZdZdZdZdefdd�Zedd� �Z dd�d d
�Z
d e fd d
�Z ddd�de
ej fdd�Zdejfdd�Zdd� Zde fdd�Zejfdd �ZdS ) re z�A client D-Bus connection which can wait for replies.
This runs a separate receiver task and dispatches received messages.
Nr{ c C s || _ t� | _t� | _d S rH )�_connr �_repliesr �_filters)r4 r{ r& r& r' r5 s zDBusRouter.__init__c C s | j jS rH )r� r/ rb r& r& r' r/ s zDBusRouter.unique_namer6 c � s �| j j||d�I dH dS )z/Send a message, don't wait for a reply
r6 N)r� r@ )r4 r8 r7 r& r& r' r@ s �zDBusRouter.sendrM c � s~ �t |� | jdu rtd��t| jj�}| j�|t� ��}| j ||d�I dH |�
� I dH W d � S 1 s8w Y dS )z�Send a method call message and wait for the reply
Returns the reply message (method return or error message type).
NzThis DBusRouter has stoppedr6 ) r �_rcv_cancel_scoper r; r� r. r� �catchr� r@ r� )r4 r8 r7 � reply_futr& r& r' rs s �
$�z DBusRouter.send_and_get_replyr )�channel�bufsizer� c C s, |du r t �|�\}}nd}t| j|||�S )a Create a filter for incoming messages
Usage::
async with router.filter(rule) as receive_channel:
matching_msg = await receive_channel.receive()
# OR:
send_chan, recv_chan = trio.open_memory_channel(1)
async with router.filter(rule, channel=send_chan):
matching_msg = await recv_chan.receive()
If the channel fills up,
The sending end of the channel is closed when leaving the ``async with``
block, whether or not it was passed in.
:param jeepney.MatchRule rule: Catch messages matching this rule
:param trio.MemorySendChannel channel: Send matching messages here
:param int bufsize: If no channel is passed in, create one with this size
N)r! �open_memory_channelr} r� )r4 r� r� r� � recv_channelr& r& r' �filter# s zDBusRouter.filterrf c � s, �| j d ur
td��|�| j�I d H | _ d S )Nz+DBusRouter receiver task is already running)r� rZ r* � _receiver)r4 rf r& r& r' r* @ s �
zDBusRouter.startc � s0 �| j dur| j �� d| _ t�d�I dH dS )z Stop the sender & receiver tasksNr )r� �cancelr! �sleeprb r& r& r' rc E s
�
zDBusRouter.acloserR c C sJ | j �|�rdS | j�|�D ]}z|j�|� W q tjy" Y qw dS )zHandle one received messageN)r� �dispatchr� �matchesr� � send_nowaitr! �
WouldBlock)r4 rR r� r& r& r' � _dispatchR s ��zDBusRouter._dispatchc � s� �t �� �K}d| _|�|� z | j�� I dH }| �|� qd| _| j�� t � d�� }| j
j � � D ]
}d|_
|j�� I dH q2W d � w 1 sJw Y w 1 sSw Y dS )z'Receiver loop - runs in a separate taskTNF� )r! � CancelScope�
is_running�startedr� rT r� r� �drop_all�
move_on_afterr� r~ �values�shieldr� rc )r4 � task_status�cscoperR �
cleanup_scoper� r& r& r' r� ] s$ �
�
� ��zDBusRouter._receiver)rh ri rj rk � _nursery_mgrr� r) r5 r� r/ r@ r rs r r! �MemorySendChannelr� �Nurseryr* rc r� �TASK_STATUS_IGNOREDr� r& r& r& r' re � s
re c s( e Zd ZdZ� fdd�Zdd� Z� ZS )r a� A trio proxy for calling D-Bus methods
You can call methods on the proxy object, such as ``await bus_proxy.Hello()``
to make a method call over D-Bus and wait for a reply. It will either
return a tuple of returned data, or raise :exc:`.DBusErrorResponse`.
The methods available are defined by the message generator you wrap.
:param msggen: A message generator object.
:param ~trio.DBusRouter router: Router to send and receive messages.
c s( t � �|� t|t�std��|| _d S )Nz)Proxy can only be used with DBusRequester)r r5 �
isinstancere � TypeError�_router)r4 �msggenrg r� r& r' r5 ~ s
zProxy.__init__c s � �fdd�}|S )Nc � s< �� | i |��}|j jtju sJ ��j�|�I d H }t|�S rH )�header� message_typer � method_callr� rs r )�args�kwargsrR r| ��make_msgr4 r& r' �inner� s
�z!Proxy._method_call.<locals>.innerr& )r4 r� r� r& r� r' � _method_call� s zProxy._method_call)rh ri rj rk r5 r� r� r& r& r� r' r s s
r c
C s� �t | |d�I dH }|4 I dH �- |�� 4 I dH �}|V W d �I dH n 1 I dH s-w Y W d �I dH dS 1 I dH sCw Y dS )a� Open a D-Bus 'router' to send and receive messages.
Use as an async context manager::
async with open_dbus_router() as req:
...
:param str bus: 'SESSION' or 'SYSTEM' or a supported address.
:return: :class:`DBusRouter`
This is a shortcut for::
conn = await open_dbus_connection()
async with conn:
async with conn.router() as req:
...
rn N)r rg )rv r, r{ �rtrr& r& r' r � s �*�.�r )rm )5r< �
contextlibr r � itertoolsr �logging�typingr r � ImportError�async_generator�outcomer r r! �trio.abcr � jeepney.authr r
� jeepney.busr � jeepney.fdsr r
�jeepney.low_levelr r r �jeepney.wrappersr r �jeepney.bus_messagesr �commonr r r r r � getLoggerrh �log�__all__r( r) r r} r� re r r r& r& r& r' �<module> sB �
~u