o `:@sddlZddlmZddlZddlmZddlZddlmZzddlm Z Wne y5ddl m Z Ynwddl m Z mZddlZddlmZddlmZmZdd lmZdd lmZmZdd lmZmZmZdd lmZm Z dd l!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'm(Z(e)e*Z+gdZ,eddZ-GdddeZ.d%ddde.fddZ/Gddde%Z0GdddZ1Gdd d Z2Gd!d"d"eZ3e d%ddd#d$Z4dS)&N)contextmanager)count)Optional)asynccontextmanager)ValueError)Channel) AuthenticatorBEGIN)get_bus)FileDescriptor fds_buf_size)Parser MessageTypeMessage) ProxyBase unwrap_msg) message_bus)MessageFilters FilterHandle ReplyMatcher RouterClosedcheck_replyable)open_dbus_connectionopen_dbus_routerProxyc csXzdVWdSty+}z|jtjtjhvrtddtd||d}~ww)Nzthis socket was already closedzsocket connection broken: {})OSErrorerrnoZEBADFENOTSOCKtrioClosedResourceErrorZBrokenResourceErrorformat)excr$1/usr/lib/python3/dist-packages/jeepney/io/trio.py)_translate_socket_errors_to_stream_errors0s  r&c@s~eZdZdZdddZdddefdd Zd efd d Zdd e fddZ defddZ ddZ ddZ ddZeddZdS)DBusConnectionaA 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. FcCsD||_||_t|_tdd|_d|_t|_ t|_ d|_ dS)Nr)start) socket enable_fdsrparserroutgoing_serial unique_namer ZLock send_lock recv_lock_leftover_to_send)selfr)r*r$r$r%__init__Is    zDBusConnection.__init__Nserialmessagec s|j4IdH/|durt|j}|jrtdnd}|j||d}|||IdHWdIdHdS1IdHs=wYdS)z.Serialise and send a :class:`~.Message` objectNi)fds)r.nextr,r*arrayZ serialise _send_data)r1r5r4r7datar$r$r%sendSs .zDBusConnection.sendr;c s|jjr tdtY|jr||jIdHt|0}|r5|j|gtjj tjj |fgIdH}n |j |IdH}|||IdHWdn1sQwYWddSWddS1siwYdS)Nz!can't send data after sending EOF) r)Zdid_shutdown_SHUT_WRr r!r&r0_send_remainder memoryviewZsendmsgZ SOL_SOCKETZ SCM_RIGHTSr<)r1r;r7sentr$r$r%r:^s"   "zDBusConnection._send_datarcsz5|t|kr1||d}|j|IdH}Wdn1s"wY||7}|t|ksd|_WdStjyF||d|_wN)lenr)r<r0r Z Cancelled)r1r;Z already_sentZ remainingr?r$r$r%r=qs  zDBusConnection._send_remainderreturnc s|j4IdH/ |j}|dur|WdIdHS|IdH\}}|s/td|j||q 1IdHs=wYdS)z5Return the next available message from the connectionNTzSocket closed at the other end)r/r+Zget_next_message _read_datar Z EndOfChannelZadd_data)r1msgbr7r$r$r%receives  zDBusConnection.receivecs|jrC|j}t|j|tIdH\}}}}Wdn1s&wY|ttjdd@r<| t d|t |fSt|j dIdH}Wd|gfS1s]wY|gfS)NZ MSG_CTRUNCrz&Unable to receive all file descriptorsi)r*r+Z bytes_desiredr&r)Zrecvmsgr getattrr _close RuntimeErrorr Z from_ancdataZrecv)r1nbytesr;Zancdataflags_r$r$r%rCs$  zDBusConnection._read_datacCs|jd|_dSr@)r)closer0r1r$r$r%rHs  zDBusConnection._closecs|dS)zClose the D-Bus connectionN)rHrNr$r$r%acloses zDBusConnection.aclosec Cst4IdH-}t|}||IdHz |VW|IdHn|IdHwWdIdHdS1IdHsr=rFrCrHrOrrRr$r$r$r%r'?s   r'SESSIONFr*rBc st|}t|IdH}t|d}|D]}||IdH||IdHq|tIdHt|j |d}| 4IdH}| t IdH}|jd|_WdIdH|S1IdHsbwY|S)zHOpen a plain D-Bus connection :return: :class:`DBusConnection` NrYr)r r Zopen_unix_socketr Zsend_allZfeedZ receive_somer r'r)rRsend_and_get_replyrZHellobodyr-) busr*Zbus_addrZsockZauthrZreq_dataconnrRreplyr$r$r%rs  rcsFeZdZdeffdd ZeddZddZdd Zd d Z Z S) TrioFilterHandlefilterscst|||||_dSr@)superr2 send_channel)r1r`ruleZsend_chnZrecv_chn __class__r$r%r2s zTrioFilterHandle.__init__cCs|jSr@queuerNr$r$r%receive_channelsz TrioFilterHandle.receive_channelcs||jIdHdSr@)rMrbrOrNr$r$r%rOszTrioFilterHandle.aclosecs|jSr@rfrNr$r$r% __aenter__szTrioFilterHandle.__aenter__cs|IdHdSr@)rO)r1exc_typeZexc_valexc_tbr$r$r% __aexit__szTrioFilterHandle.__aexit__) rSrTrUrr2propertyrhrOrirl __classcell__r$r$rdr%r_s r_c@s0eZdZdZddZddZddZdd Zd S) Futurez4A very simple Future for trio based on `trio.Event`.cCsd|_t|_dSr@)_outcomer ZEvent_eventrNr$r$r%r2szFuture.__init__cCt||_|jdSr@)rrprqset)r1resultr$r$r% set_result zFuture.set_resultcCrrr@)rrprqrs)r1r#r$r$r% set_exceptionrvzFuture.set_exceptioncs|jIdH|jSr@)rqwaitrpZunwraprNr$r$r%gets z Future.getN)rSrTrUrVr2rurwryr$r$r$r%ros  roc@seZdZdZdZdZdefddZeddZ ddd d Z d e fd d Z dddde ejfddZdejfddZddZde fddZejfddZdS)rPzA client D-Bus connection which can wait for replies. This runs a separate receiver task and dispatches received messages. Nr]cCs||_t|_t|_dSr@)_connr_repliesr_filters)r1r]r$r$r%r2s zDBusRouter.__init__cCs|jjSr@)rzr-rNr$r$r%r- szDBusRouter.unique_namer3cs|jj||dIdHdS)z/Send a message, don't wait for a reply r3N)rzr<)r1r5r4r$r$r%r<szDBusRouter.sendrBcs~t||jdurtdt|jj}|j|t}|j ||dIdH| IdHWdS1s8wYdS)zSend a method call message and wait for the reply Returns the reply message (method return or error message type). NzThis DBusRouter has stoppedr3) r_rcv_cancel_scoperr8rzr,r{Zcatchror<ry)r1r5r4Z reply_futr$r$r%rZs   $zDBusRouter.send_and_get_replyr)channelbufsizer~cCs,|dur t|\}}nd}t|j|||S)aCreate 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 Zopen_memory_channelr_r|)r1rcr~rZ recv_channelr$r$r%filter#szDBusRouter.filterrQcs,|jdur td||jIdH|_dS)Nz+DBusRouter receiver task is already running)r}rIr( _receiver)r1rQr$r$r%r(@s zDBusRouter.startcs0|jdur|jd|_tdIdHdS)z Stop the sender & receiver tasksNr)r}Zcancelr sleeprNr$r$r%rOEs   zDBusRouter.acloserDc CsJ|j|rdS|j|D]}z|j|Wqtjy"YqwdS)zHandle one received messageN)r{dispatchr|matchesrbZ send_nowaitr Z WouldBlock)r1rDrr$r$r% _dispatchRs zDBusRouter._dispatchcstK}d|_||z |jIdH}||qd|_|jt d}|j j D] }d|_ |jIdHq2Wdw1sJwYw1sSwYdS)z'Receiver loop - runs in a separate taskTNF)r Z CancelScopeZ is_runningZstartedrzrFrr{Zdrop_allZ move_on_afterr|r`valuesZshieldrbrO)r1Z task_statusZcscoperDZ cleanup_scoperr$r$r%r]s$     zDBusRouter._receiver)rSrTrUrVZ _nursery_mgrr}r'r2rmr-r<rrZrr ZMemorySendChannelrZNurseryr(rOrZTASK_STATUS_IGNOREDrr$r$r$r%rPs   rPcs(eZdZdZfddZddZZS)raA 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. cs(t|t|tstd||_dS)Nz)Proxy can only be used with DBusRequester)rar2 isinstancerP TypeError_router)r1ZmsggenrRrdr$r%r2~s   zProxy.__init__csfdd}|S)Ncs<|i|}|jjtjusJj|IdH}t|Sr@)headerZ message_typerZ method_callrrZr)argskwargsrDr^make_msgr1r$r%inners z!Proxy._method_call..innerr$)r1rrr$rr% _method_callszProxy._method_call)rSrTrUrVr2rrnr$r$rdr%rss rc Cst||dIdH}|4IdH-|4IdH}|VWdIdHn 1IdHs-wYWdIdHdS1IdHsCwYdS)aOpen 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: ... rYN)rrR)r\r*r]Zrtrr$r$r%rs*.r)rX)5r9 contextlibrr itertoolsrZloggingtypingrr ImportErrorasync_generatorZoutcomerrr Ztrio.abcrZ jeepney.authr r Z jeepney.busr Z jeepney.fdsr r Zjeepney.low_levelrrrZjeepney.wrappersrrZjeepney.bus_messagesrcommonrrrrrZ getLoggerrSlog__all__r&r'rr_rorPrrr$r$r$r%sB         ~u