o j~ @sddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z m Z mZddlmZmZGdd d eZGd d d ZdS) N)logger)connect)QuicConnectionProtocol)QuicConfiguration)ConnectionTerminated QuicEventStreamDataReceived)OptionalcastcsXeZdZfddZdejddffdd Zdeddfd d Zd e ddfd d Z Z S)QuicTransportClientcs.tj|i|d|_d|_d|_d|_dSN)super__init__ _ack_waiterin_queuedisconnected_evt stream_id)selfargskwargs __class__/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/asysocks/unicomm/protocol/quic.pyrs  zQuicTransportClient.__init__ transportreturnNcstdt|S)Nz Connected!)printr connection_made)rrrrrrs z#QuicTransportClient.connection_madedatacs:|jdur |j|_d}|j|j|||dS)NF)r_quicget_next_available_stream_idsend_stream_datatransmit)rr end_streamrrrsends    zQuicTransportClient.sendeventcCs:t|tr|j|jdft|tr|jddSdS)NNN) isinstancerr put_nowaitrr)rr%rrrquic_event_received$s  z'QuicTransportClient.quic_event_received) __name__ __module__ __qualname__rasyncio BaseTransportrbytesr$rr) __classcell__rrrrr s   r c@s2eZdZdZd ddZddZddZd d ZdS) QUICSocketz Generic asynchronous TCP socket class, nothing SMB related. Creates the connection and channels incoming/outgoing bytes via asynchonous queues. NcCs.||_||_t|_t|_t|_dSr )settingssocketr-Queue out_queuerEventr)rr3targetrrrr1s   zQUICSocket.__init__cs|jrdS|jdS)zI Disconnects from the socket. Stops the reader and writer streams. N) disconnectedis_setset)rrrr disconnect:s zQUICSocket.disconnectc szSt|jj|jj|jtd4IdH4}tt|}|j|_|j|_|j s;|j IdH}| |IdH|j r&WdIdHWdS1IdHsMwYWdStjy_YdSty}}ztd|IdHWYd}~dSd}~ww)N) configurationcreate_protocolz[QUICSocket] handle_outgoing)rr2hostnameportr<r r rrr8r9r5getr$r-CancelledError Exceptionr exceptionr;)rclientrerrr run_innerCs$$   2 zQUICSocket.run_innerc sfztdgddtjd|_t||_WdSty2}zt dd|fWYd}~Sd}~ww) zk Main function to be called, connects to the target specified in settings, and starts reading/writing. smbTi)alpn_protocols is_clientmax_datagram_frame_size verify_mode)TNz[QUICSocket] mainFN) rssl CERT_NONEr<r- create_taskrFrDrBrrC)rrErrrrWs zQUICSocket.connectr&)r*r+r,__doc__rr;rFrrrrrr1,s    r1)enumr-rLasysocks.unicommraioquic.asyncio.clientraioquic.asyncio.protocolraioquic.quic.configurationraioquic.quic.eventsrrrtypingr r r r1rrrrs