a î½jÄAã@sNdZddlTdZdZdZdZdZdZdZdZ dZ dZ dZ dZ Gdd „d ƒZd S) z$Idé)Ú*éééééc@sÌeZdZdZd.dd„Zdd„Zdd„Zd d „Zd d „Ze fd d„Z dd„Z d/dd„Z ddifdd„Z ddifdd„Zdd„Zdd„Zd0dd„Zd1d d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,d-„ZdS)2ÚSessiona£ The Session class instance is used for storing all session-related info like credentials, socket/xml stream/session state flags, roster items (in case of client type connection) etc. Session object have no means of discovering is any info is ready to be read. Instead you should use poll() (recomended) or select() methods for this purpose. Session can be one of two types: 'server' and 'client'. 'server' session handles inbound connection and 'client' one used to create an outbound one. Session instance have multitude of internal attributes. The most imporant is the 'peer' one. It is set once the peer is authenticated (client). NcCs||_|rd|_||_t|_nd|_d|_t|_||_|j|_|j |_ |j |_ d|_ |j |_ d|_|j j|_i|_||_|jdkrštt ¡ƒdd…|_nd|_d|_d|_d|_g|_i|_g|_t|_g|_ttt t!fD]}||j"vrÜ|j #|¡qÜg|_"d|_$d|_%| &¡dS)aI When the session is created it's type (client/server) is determined from the beginning. socket argument is the pre-created socket-like object. It must have the following methods: send, recv, fileno, close. owner is the 'master' instance that have Dispatcher plugged into it and generally will take care about all session events. xmlns is the stream namespace that will be used. Client must set this argument If server sets this argument than stream will be dropped if opened with some another namespace. peer is the name of peer instance. This is the flag that differentiates client session from server session. Client must set it to the name of the server that will be connected, server must leave this argument alone. ÚclientÚserverNrÚsessionrÚ)'ÚxmlnsÚTYPÚpeerÚSOCKET_UNCONNECTEDÚ _socket_stateÚ SOCKET_ALIVEÚ_sockÚsendÚ_sendÚrecvÚ_recvÚfilenoÚ _registeredÚ DispatcherZDBG_LINEÚDEBUGZ _expectedÚ_ownerÚreprÚrandomÚIDÚ sendbufferÚ_stream_pos_queuedÚ_stream_pos_sentÚdeliver_key_queueÚdeliver_queue_mapÚ stanza_queueÚSESSION_NOT_AUTHEDÚ_session_stateÚwaiting_featuresÚNS_TLSÚNS_SASLÚNS_BINDÚ NS_SESSIONÚfeaturesÚappendÚfeature_in_processÚ slave_sessionÚ StartStream)ÚselfÚsocketÚownerr rÚfeature©r6úR/Users/admin/.hermes/hermes-agent/venv/lib/python3.9/site-packages/xmpp/session.pyÚ__init__8sD  "zSession.__init__cCsbt|_t ¡|_d|j_|j|j_|jj|_|j |j_ |j dkrT|j |j_ | ¡n |j|j_ dS)zâ This method is used to initialise the internal xml expat parser and to send initial stream header (in case of client connection). Should be used after initial connection and after every stream restart.rr N)ÚSTREAM__NOT_OPENEDÚ _stream_stateZ simplexmlZ NodeBuilderÚStreamZ_dispatch_depthÚ _dispatchÚdispatchÚParseÚ _stream_closeÚstream_footer_receivedrÚ_catch_stream_idÚstream_header_receivedÚ _stream_open©r2r6r6r7r1ks       zSession.StartStreamcCshz| d¡}Wnd}Yn0t|ƒrF| t| ¡ƒd|d¡n| dd¡| t¡tdƒ‚|S)z‚ Reads all pending incoming data. Raises IOError on disconnection. Blocks until at least one byte is read.i(r ú Úgotz!Socket error while receiving dataÚerrorzPeer disconnected)rÚlenrrrÚset_socket_stateÚ SOCKET_DEADÚIOError)r2Zreceivedr6r6r7Úreceive{s  zSession.receivecCsBt|tƒr| ¡ d¡}nt|ƒtdƒkr4| d¡}| |¡dS)zÈ Put chunk into "immidiatedly send" queue. Should only be used for auth/TLS stuff and like. If you just want to shedule regular stanza for delivery use enqueue method. úutf-8r N)Ú isinstanceÚNodeÚ__str__ÚencodeÚtypeÚenqueue)r2Úchunkr6r6r7ÚsendnowŠszSession.sendnowcCs<t|tƒr|j |¡n|j|7_|jtkr8| ¡dS)zä Takes Protocol instance as argument. Puts stanza into "send" fifo queue. Items into the send queue are hold until stream authenticated. After that this method is effectively the same as "sendnow" method.N)rNÚProtocolr%r.r rrÚ push_queue)r2Ústanzar6r6r7rS“s zSession.enqueuecCsè|jtks|jtkr€|j |¡|jD]}|jt|j ||ƒddq&|j D]}|jt||ƒddqLigg|_ |_|_ dS|j t krä|j D]L}|  ¡ d¡}|j|7_|jt|ƒ7_||j |j<|j |j¡qg|_ dS)zä If stream is authenticated than move items from "send" queue to "immidiatedly send" queue. Else if the stream is failed then return all queued stanzas with error passed as argument. Otherwise do nothing.r©ÚtrustedNrM)r:ÚSTREAM__CLOSEDrrJrZdeactivatesessionr#r<ÚErrorr$r%r'ÚSESSION_AUTHEDrPrQr r!rHr.)r2Z failreasonÚkeyrXÚtxtr6r6r7rWœs       zSession.push_queuecCsÊ|jrÆz| |j¡}Wn*| t¡| dd¡| ¡YS0| t| ¡ƒd|jd|…d¡|j|7_|j|d…|_|j|_ |j rÆ|j |j dkrÆ|j |j d=|j   |j d¡qŽdS)z\ Put the "immidiatedly send" queue content on the wire. Blocks until at least one byte sent.zSocket error while sending datarGrENÚsentr) r rrIrJrÚterminate_streamrrr"Z_stream_pos_deliveredr#r$Úremove)r2r`r6r6r7Ú flush_queueµs  &zSession.flush_queuercCsF|jjd7_|jtks|rB| | ¡d¡||_|j ||¡SdS)a" This is callback that is used to pass the received stanza forth to owner's dispatcher _if_ the stream is authorised. Otherwise the stanza is just dropped. The 'trusted' argument is used to emulate stanza receive. This method is used internally. rr=N) rZpacketsr:ÚSTREAM__OPENEDrrPrZrr=)r2rXrZr6r6r7r<És zSession._dispatchÚstreamcCs<d|vs|ds| t¡S|d|_d|vr8|j |¡dS)z[ This callback is used to detect the stream namespace of incoming stream. Used internally. ÚidÚversionN)raÚSTREAM_INVALID_XMLrrZDialback)r2ÚnsÚtagÚattrsr6r6r7rAÕs  zSession._catch_stream_idcCsd}|jdkr|d|j7}n<|d|j7}d|vrJ|d|jjd7}n|d|d7}d|vrr|d |d7}|jr€|j}nt}|d tt|f7}d |vs¨|jdkr°|d 7}|  |d ¡|  t ¡|jdkrÖdS|dkrè|  t ¡S|tkrú|  t¡S|jj|jkr|  t¡Sd|vr(|  t¡S|d|jjvrD|  t¡S|d ¡|_|jdkrd |vrtdƒ}t|jvr–|jj t¡|jjjjt|jvrÐ|jj t¡|jjjD]}||jjj_ qºn4t!|jvrê|jj" t!¡t#|jvr|jj$ t#¡|  |¡dS)a This callback is used to handle opening stream tag of the incoming stream. In the case of client session it just make some validation. Server session also sends server headers and if the stream valid the features node. Used internally. z5 Nrer zstream:features)%rrrrZ servernamesr Z NS_SERVERZ NS_DIALBACKZ NS_STREAMSrUÚset_stream_staterdrarhZSTREAM_INVALID_NAMESPACEr;ZSTREAM_BAD_NAMESPACE_PREFIXZSTREAM_IMPROPER_ADDRESSINGZSTREAM_HOST_UNKNOWNÚlowerZournamerOr)r(ZNTÚstarttlsÚ setNamespaceÚTÚrequiredr*Z mechanismsZSASLZ mechanismr+Úbindr,r )r2rirjrkÚtextr r-Zmecr6r6r7rCÜs@     zSession._stream_opencCs$||jvr|j |¡| |¡dS)z/ Declare some stream feature as activated one. N)r-r.Ú unfeature©r2r5r6r6r7r5szSession.featurecCs||jvr|j |¡dS)z‹ Declare some feature as illegal. Illegal features can not be used. Example: BIND feature becomes illegal after Non-SASL auth. N)r(rbrwr6r6r7rv szSession.unfeaturercCsZ|jtkrdS| t¡| d¡| t¡| ¡|j ¡|rN|j |¡|  ¡dS)zS Write the closing stream tag and destroy the underlaying socket. Used internally. Nz) r:r[rnÚSTREAM__CLOSINGrUrWrZ flush_queuesZunregistersessionÚ_destroy_socket)r2Ú unregisterr6r6r7r?s    zSession._stream_closecCs¢|jtkrdS|jtkr,| t¡| ¡n,| t¡tdd}| t¡|j|dd|r€t |t ƒrr|  |¡n|  t |ƒ¡|j |d|jrž|j t¡dS)aƒ Notify the peer about stream closure. Ensure that xmlstream is not brokes - i.e. if the stream isn't opened yet - open it before closure. If the error condition is specified than create a stream error and send it along with closing stream tag. Emulate receiving 'unavailable' type presence just before stream closure. NÚ unavailable)ÚtyprrY)rz)r:rxrdrnrCZPresencerqZ NS_CLIENTr<rNrOrUZ ErrorNoder?r0raZSTREAM_REMOTE_CONNECTION_FAILED)r2rGrzÚpr6r6r7ras       zSession.terminate_streamcCs:d|j_d|j_d|j_|j ¡|j ¡| t¡dS)zD Break cyclic dependancies to let python's GC free memory right now.N) r;r=r@rBÚdestroyrÚcloserIrJrDr6r6r7ry2s   zSession._destroy_socketcCs"|jrtd||jfƒ‚||_dS)z] Declare some feature as "negotiating now" to prevent other features from start negotiating. zStarting feature %s over %s !N©r/Ú Exception©r2Úfr6r6r7Ú start_feature;szSession.start_featurecCs&|j|krtd||jfƒ‚d|_dS)zQ Declare some feature as "negotiated" to allow other features start negotiating. z#Stopping feature %s instead of %s !Nr€r‚r6r6r7Ú stop_feature@szSession.stop_featurecCs|j|kr||_dS)z» Change the underlaying socket state. Socket starts with SOCKET_UNCONNECTED state and then proceeds (possibly) to SOCKET_ALIVE and then to SOCKET_DEAD N)r©r2Znewstater6r6r7rIEszSession.set_socket_statecCs.|j|kr*|jtkr$|tkr$|j|_||_dS)zÔ Change the session state. Session starts with SESSION_NOT_AUTHED state and then comes through SESSION_AUTHED, SESSION_BOUND, SESSION_OPENED and SESSION_CLOSED states. N)r'r]r"r!r†r6r6r7Úset_session_stateLs   ÿzSession.set_session_statecCs|j|kr||_dS)aA Change the underlaying XML stream state Stream starts with STREAM__NOT_OPENED and then proceeds with STREAM__OPENED, STREAM__CLOSING and STREAM__CLOSED states. Note that some features (like TLS and SASL) requires stream re-start so this state can have non-linear changes. N)r:r†r6r6r7rnWszSession.set_stream_state)NN)r)r)Nr)Ú__name__Ú __module__Ú __qualname__Ú__doc__r8r1rLrUrSZERR_RECIPIENT_UNAVAILABLErWrcr<rArCr5rvr?raryr„r…rIr‡rnr6r6r6r7r,s* 3   )   rN)Ú __version__ÚprotocolrrrJr9rdrxr[r&r]Z SESSION_BOUNDZSESSION_OPENEDZSESSION_CLOSEDrr6r6r6r7Ús