o .jV@sddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ej ZzddlmZWn eyDeZYnwzejjZWn eyVeZYnwejrlddlZddlmZdd lmZmZdd d ZeduZGdddZGdddeZ dddZ!dZ"dS)) annotationsN)timeout)open_dgram_connectionDatagramReaderDatagramWriter) TimeoutError)Literal)_TYPE_PEER_CERT_RET_TYPE_PEER_CERT_RET_DICTreturnboolcCs@ddl}|dkr|dddkrt|ddkSdS)NrWindows)37T)platformsystempython_version_tupleint)rrX/home/thesage/.local/lib/python3.10/site-packages/urllib3_future/contrib/ssa/__init__.py)_can_shutdown_and_close_selector_loop_bug!s rFc@s,eZdZdZejejddfdTddZdUddZdVddZ dVddZ dVddZ e j dkr0dWddZdXd"d#ZejdYd$d%ZejdZd(d%Z d[d\d)d%Zd]d+d,Zd^d/d0Zddd1d_d9d:Zd`dad=d>Zd`dad?d@ZdbdAdBZdcdEdFZdcdGdHZdcdIdJZd[dddKdLZdedMdNZdfdPdQZdgdRdSZdS)h AsyncSocketa This class is brought to add a level of abstraction to an asyncio transport (reader, or writer) We don't want to have two distinct code (async/sync) but rather a unified and easily verifiable code base. 'ssa' stands for Simplified - Socket - Asynchronous. rNfamilysocket.AddressFamilytypesocket.SocketKindprotorfileno int | Noner NonecCs||_||_||_||_d|_t|_tj|j|j|j|d|_ |j dd|_ d|_ t |_t |_d|_d|_d|_d|_dS)NF)rrr!r"r)rrr!_fileno_connect_calledasyncioEvent _establishedsocket_sock settimeout_writer_reader Semaphore_writer_semaphore_reader_semaphore_addr_external_timeout _tls_in_tls _ech_config)selfrrr!r"rrr__init__>s$     zAsyncSocket.__init__cCs|jdur|jS|jSN)r%r+r"r6rrrr"aszAsyncSocket.filenoc s|jrdS|jdur dSz |jddu}Wn ty"d}Ynw|r3tdIdH|jjz+td4IdH|j IdHWdIdHWdS1IdHsWwYWdSt yhYdSw)N ssl_objectFrr) r&r-get_extra_infoAttributeErrorr'sleep transportabortr wait_closedOSError)r6is_sslrrrwait_for_closeds(   2 zAsyncSocket.wait_for_closecCs|jdur%z|jWnty$d|_|jd|_d|_YdSwt}|s>t dkr>t t dr>t t t jr>d}zd}d}d}t |jdrz |jtjd}Wn[tyd}z |jd}WnFtyz tj|jd}Wn ttfyYn'wz z |tjd}WntytdtYnwW|n|wYnwYnwt |jtjrt |jd r|s|sz |jd}Wn ttfyYnw|r|st |jd r|sz|jjWn tttfyYnwWnGtyAt |jtjrz|jWn-tttfyYn!wt |jd r?|s?z|jjWn tttfy>YnwYnwd|_|jdS) ziThe closing procedure is a real challenge. There's a ton of edge cases. Act carefully around this method.NFrSelectorEventLoopTshutdown)r"zurllib3-future is unable to properly close your async socket. This mean that you are probably using an asyncio implementation like uvloop that does not support shutdown() or/and close() on the socket transport. This will lead to unclosed socket (fd).closer+)r-rFr<r&r)clearr.!_CPYTHON_SELECTOR_CLOSE_BUG_EXISTrrhasattrr' isinstanceget_running_looprDr+rEr*SHUT_RD TypeErrorr"rA ValueErrorwarningswarnResourceWarningdetach)r6edge_case_close_bug_existuvloop_edge_case_bugshutdown_called close_called direct_sockrrrrFs                  zAsyncSocket.closecs|jIdHdSr8)r)waitr9rrrwait_for_readinesszAsyncSocket.wait_for_readinesswin32controloption!int | tuple[int, int, int] | boolcCs|j||Sr8)r+ioctl)r6r\r]rrrr_szAsyncSocket.ioctl_AsyncSocket__level_AsyncSocket__optname_AsyncSocket__value int | bytescCs|j|||dSr8)r+ setsockopt)r6r`rarbrrrrdszAsyncSocket.setsockoptcCdSr8r)r6r`rarrr getsockoptzAsyncSocket.getsockoptbuflenbytescCrer8rr6r`rarhrrrrf rgcCs&|dur |j||S|j|||Sr8)r+rfrjrrrrf srcCs |jduS)NF)r&r9rrrshould_connect zAsyncSocket.should_connectaddr+tuple[str, int] | tuple[str, int, int, int]c s|jrtdd|_tdkrIddlm}m}|dd\}}||sI||sItj |||j |j dIdH}|sCt d||dd }|jdurz-t|j4IdHt|j|IdHWdIdHn 1IdHsuwYWn9tttfy}zd |_t|d}~wtytd wzt|j|IdHWn tytd w|j t jks|j d krtj|jd IdH\|_|_n|j t jkrt|jd IdH\|_|_|jdusJ|jd |j|_||_|j !dS)Nz>attempted to connect twice on a already established connectionTrr)is_ipv4is_ipv6)rrzunable to resolve hostname rrFz@Likely FD Kernel/Loop Racing Allocation Error. You should retry.)sockr*)"r&rArrresolver.utilsrorpr'rK getaddrinforrr*gaierrorr3r sock_connectr+FutureTimeoutErrorAsyncioTimeoutErrorrStandardTimeoutError RuntimeErrorConnectionError SOCK_STREAMopen_connectionr.r- SOCK_DGRAMrr;r2r)set)r6rmrorphostportreserrrconnectsb    (   zAsyncSocket.connectserver_hostnamessl_handshake_timeoutctxssl.SSLContextr str | Noner float | NoneSSLAsyncSocketc sT|jIdH|jz ttjjddWn ty!Ynwztjj}t|ds3dd}||_ Wn ty=Ynw|j t j kr|j dusKJt|j tjsTJ|j j}|j IdH|j jj|j j||d||dIdH}|dur|jr}tdtd ||j _|j j}|j |_||_|d du|_||_ntd |jt|_|S) N_start_tls_compatibleT _force_closecSsFd|_|jdur!z |j|WdSty |jYdSwdS)NT)_closed _ssl_protocol_abortrM)r6excrrrrms  z-AsyncSocket.wrap_socket.._force_closeF) server_siderrzTAsyncio TLS-in-TLS failed. The transport failed silently during inner TLS handshake.zMAsyncio start TLS failed. The transport failed silently during TLS handshake. sslcontextzUnsupported socket type) r)rXrGsetattrr'sslproto_SSLProtocolTransportr<rIrrr*r{r-rJ StreamWriter _protocoldrain_loop start_tls _transportr4rAr>_stream_writerr; _over_ssl_tls_ctxryr~r __class__) r6rrr_ssl_tprprotocol new_transportr>rrr wrap_socketRsh         zAsyncSocket.wrap_socketsizebytes | list[bytes]c s&|dkrd}|jdusJ|js|jIdH|jIdHzi|jdur}z2t|j4IdH|jj|dIdHWdIdHWW|j S1IdHsUwYWn!t t t fym}zt |d}~wty|}zt|d}~ww|jj|dIdHW|j S|j w)aeReceive data from the socket. Returns ``bytes`` for a single datagram (or stream chunk), or ``list[bytes]`` when GRO / batch-receive delivered multiple coalesced datagrams in one syscall. The caller can then feed all segments to the QUIC state-machine in a tight loop before probing, avoiding per-datagram overhead.riN)n)r.r)is_setrXr1acquirer3rreadreleaservrwrrxrAConnectionResetError)r6rrrrrrecvs0   zAsyncSocket.recvcs|j|dIdHS)MJust an alias for recv(), it is needed due to our custom AsyncSocks override.)rNr)r6rrrr read_exactszAsyncSocket.read_exactcs|IdHS)rNrr9rrrrszAsyncSocket.readdata,bytes | bytearray | memoryview | list[bytes]cs|jdusJ|jIdH|jIdHz!z|j||jIdHWnty1wW|jdS|jwr8) r-r)rXr0rwriter Exceptionrr6rrrrsendalls  zAsyncSocket.sendallcs||IdHdS)zPJust an alias for sendall(), it is needed due to our custom AsyncSocks override.Nrrrrr write_allszAsyncSocket.write_allcs||IdHdSr8rrrrrsendrZzAsyncSocket.sendcCs ||_dSr8r3)r6rbrrrr,rlzAsyncSocket.settimeoutcC|jSr8rr9rrr gettimeoutszAsyncSocket.gettimeouttuple[str, int]cC |jSr8)r+ getpeernamer9rrrrrlzAsyncSocket.getpeernamecCs|j|dSr8)r+bind)r6rmrrrrszAsyncSocket.bind) rrrr r!rr"r#r r$)r r)r r$)r\rr]r^r r$)r`rrarrbrcr r$)r`rrarr r)r`rrarrhrr rir8)r`rrarrhr#r rcr r)rmrnr r$rrrrrrr r)r)rrr r)r r)rrr r$)rbrr r$)r r)r r)rmrr r$) __name__ __module__ __qualname____doc__r*AF_INETr{r7r"rCrFrYsysrr_rdtypingoverloadrfrkrrrrrrrrr,rrrrrrrr5sF  #  " s       @ T       rcseZdZUded<ded<ej d)d*d d Zejd+dd Zd,d-dd Zd.ddZed/ddZ d.ddZ ed0ddZ ed/ddZ d1dd Z d!d!d"d2fd'd(ZZS)3rrrrr4. binary_formLiteral[False]r _TYPE_PEER_CERT_RET_DICT | NonecCrer8rr6rrrr getpeercertszSSLAsyncSocket.getpeercert Literal[True] bytes | NonecCrer8rrrrrrrgFr cCs|jj|dS)N)r)sslobjrrrrrrsrcCrr8)rselected_alpn_protocolr9rrrrrlz%SSLAsyncSocket.selected_alpn_protocolssl.SSLSocket | ssl.SSLObjectcCs*|jdur|jd}|dur|Std)Nr:zC"ssl_object" could not be extracted from this SslAsyncSock instance)r-r;ry)r6rrrrrs zSSLAsyncSocket.sslobjcCrr8)rversionr9rrrr rlzSSLAsyncSocket.versioncCs|jjSr8)rcontextr9rrrrszSSLAsyncSocket.contextcCrr8)rr9rrr_sslobjszSSLAsyncSocket._sslobjtuple[str, str, int] | NonecCrr8)rcipherr9rrrrrlzSSLAsyncSocket.cipherNrrrrrcs d|_tj|||dIdHS)NTr)r4superr)r6rrrrrrrs zSSLAsyncSocket.wrap_socket).)rrr r)rrr r)F)rrr r )r r)r r)r r)r rr)rrr__annotations__rrrrpropertyrrrrrr __classcell__rrrrrs*         rcCsXddl}|dkr*|dkr*zddl}Wn ty YdSwt|dr(dSdSdS)uA bug exist in PyPy asyncio implementation that prevent us to use a DGRAM socket. This piece of code inform us, potentially, if PyPy has fixed the winapi implementation. See https://github.com/pypy/pypy/issues/4008 and https://github.com/jawah/niquests/pull/87 The stacktrace look as follows: File "C:\hostedtoolcache\windows\PyPy.10.13†\Libsyncio\windows_events.py", line 594, in connect _overlapped.WSAConnect(conn.fileno(), address) AttributeError: module '_overlapped' has no attribute 'WSAConnect' rNrPyPyF WSAConnectT)rrpython_implementation _overlapped ImportErrorrI)rrrrr_has_complete_support_dgram+s    r)rrrr)# __future__rr'rr*rrrO_timeoutr_grorrrrxconcurrent.futuresrrvr exceptionsrwr< TYPE_CHECKINGssltyping_extensionsr _typingr r rrHrrr__all__rrrrsB        ; >