o j+&@sddlZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z GdddZGdd d Z ddlZddlZGd d d ejZGd d d ZGdddeZGdddeZdS)N) UniTarget) PacketizerStreamPacketizer) PacketizerSSL) UNITransportc @seZdZd*dejdejdededef ddZ d d Z d d Z d+d dZ ddZ ddZddZd*ddZddZddZddZddZdd Zd!d"Zd#d$Zd%ejfd&d'Zd%ejfd(d)ZdS), UniConnectionNreaderwriter packetizerpeer_ip peer_portcCsV||_||_||_||_||_d|_d|_t|_ t |_ t|_ |j dSNF)rr r r r packetizer_taskclosingasyncioEvent closed_evtLock read_lock read_resumeset)selfrr r r r r/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/asysocks/unicomm/common/connection.py__init__ s   zUniConnection.__init__cs|SNrrrrr __aenter__zUniConnection.__aenter__cs|IdHdSr)close)rexc_typeexctbrrr __aexit__szUniConnection.__aexit__cCs<t|jdr |j||S|dkr|jdur|j|jfS|S)Nget_extra_infopeername)hasattrr r$r r )rnamedefaultrrrr$!s  zUniConnection.get_extra_infocCs |jSr)r get_peer_certificaterrrrr)* z"UniConnection.get_peer_certificatecCs,|j}t|jtr||j_dS||_dSr)r flush_buffer isinstancer)rr rem_datarrrchange_packetizer-s    zUniConnection.change_packetizercOs|jj|i|Sr)r packetizer_control)rargskwrrrr/4z UniConnection.packetizer_controlcsV|dur|j}|durt}d|_tj|_t|||_|j|j|j IdHdSr ) r sslcreate_default_contextcheck_hostname CERT_NONE verify_moder do_handshakerr )rssl_ctxr rrrwrap_ssl7s zUniConnection.wrap_sslcs*d|_|jdur|j|jdSNT)rr rrrrrrrrAs   zUniConnection.closecsdSrrrrrrdrainGrzUniConnection.draincs>|j|2z3dHW}|j||jIdHq6dSr)r data_outr writer<)rdatapacketrrrr>Js  zUniConnection.writecs$|2z 3dHW}|S6dSr)read)rr@rrrread_oneOszUniConnection.read_onec CszQd}|jdur5|j|2z3dHW}|durn|Vq6|j|jjIdH}|dkr0n|jdus d}|j|2z3dHW}|durKWdS|Vq=6WdStyh}z dVWYd}~dSd}~ww)NF)rr data_inrrA buffer_size Exception)rr?resulterrrrASs.   zUniConnection.readcsNt|jts td |j|jjIdH}|j|IdH|dkr&dSq )NzrBrArJrMProtocolrXr[rrrrr s$$     rc@seZdZddZdS)UniUDPConnectioncCs||_||_||_dSr)socketr?addr)rrer?rfrrrrs zUniUDPConnection.__init__N)r\r]r^rrrrrrds rdc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS)DatagramEndpointProtocolz8Datagram protocol for the endpoint high-level interface.cCs ||_dSr) _endpoint)rendpointrrrrr*z!DatagramEndpointProtocol.__init__cCs ||j_dSr)rh _transport)rrYrrrrVs z(DatagramEndpointProtocol.connection_madecCs4|dusJ|jjdur|jjd|jdSr)rh_write_ready_future set_resultr)rr!rrrrSs  z(DatagramEndpointProtocol.connection_lostcCs|j||dSr)rh feed_datagramrr?rfrrrdatagram_receivedr2z*DatagramEndpointProtocol.datagram_receivedcCsd}t||dS)Nz Endpoint received an error: {!r})warningswarnformat)rr!msgrrrerror_receivedsz'DatagramEndpointProtocol.error_receivedcCs*|jjdusJ|jjj}||j_dSr)rhrkrj_loop create_future)rlooprrr pause_writings z&DatagramEndpointProtocol.pause_writingcCs*|jjdusJ|jjdd|j_dSr)rhrkrlrrrrresume_writings z'DatagramEndpointProtocol.resume_writingN) r\r]r^__doc__rrVrSrortrxryrrrrrgs rgc@seZdZdZddefddZddZdd Zdd d Zdd dZ dddZ ddZ ddZ ddZ ddZeddZeddZdS)EndpointzHigh-level interface for UDP enpoints. Can either be local or remote. It is initialized with an optional queue size for the incoming datagrams. NtargetcCs4||_|dur d}t||_d|_d|_d|_dS)NrF)r|rQueue_queue_closedrjrk)rr| queue_sizerrrrs  zEndpoint.__init__cCs8z |j||fWdStjytdYdSw)NzEndpoint queue is full)r~ put_nowaitr QueueFullrprqrnrrrrms zEndpoint.feed_datagramcCs>|jrdSd|_|jr|dd|jr|jdSdSr;)rr~emptyrmrjrrrrrrs  zEndpoint.closecs(|dur|j|jjf}|||S)%Send a datagram to the given address.N)r|get_ip_or_hostnameportsendrnrrrr> s zEndpoint.writeFcCsZ|js+|IdH\}}|dur|dS||f}|dur#|Vn|V|jrdSdSnWait for an incoming datagram and return it with the corresponding address. This method is a coroutine. NT)rreceiver)r with_addrr?rfrGrrrrAsz Endpoint.readcs<|IdH\}}|dur|dS|dur||fS|Sr)rr)rrr?rfrrrrB szEndpoint.read_onecCsJ|jrtdt|jjdurt|j||}dS|j||dS)rEnpoint is closedTN)rIOErrorr iscoroutinerjsendtorW)rr?rfrZrrrr-s z Endpoint.sendcsF|jr |jr td|jIdH\}}|durtd||fS)rrN)r~rrrgetrnrrrr7szEndpoint.receivecCs$|jrtd|j|dS)z Close the transport immediately.rN)rrrjabortrrrrrrCs  zEndpoint.abortcs |jdur|jIdHdSdS)z4Drain the transport buffer below the low-water mark.N)rkrrrrr<Js zEndpoint.draincCs|jdS)z-The endpoint address as a (host, port) tuple.re)rjr$ getsocknamerrrraddressQszEndpoint.addresscCs|jS)z0Indicates whether the endpoint is closed or not.)rrrrrclosedVszEndpoint.closedr)F)r\r]r^rzrrrmrr>rArBrrrr<propertyrrrrrrr{s         r{c@seZdZdZdS) LocalEndpointzyHigh-level interface for UDP local enpoints. It is initialized with an optional queue size for the incoming datagrams. N)r\r]r^rzrrrrr\srcs,eZdZdZfddZfddZZS)RemoteEndpointzzHigh-level interface for UDP remote enpoints. It is initialized with an optional queue size for the incoming datagrams. cst|ddS)z#Send a datagram to the remote host.N)superrrI __class__rrrhszRemoteEndpoint.sendcstIdH\}}|S)zU Wait for an incoming datagram from the remote host. This method is a coroutine. N)rrrnrrrrlszRemoteEndpoint.receive)r\r]r^rzrr __classcell__rrrrrcs r)r3copyrrQ ipaddressasysocks.unicomm.common.targetr#asysocks.unicomm.common.packetizersrr'asysocks.unicomm.common.packetizers.sslr!asysocks.unicomm.common.transportrrrdrpDatagramProtocolrgr{rrrrrrs&     )x