_iOdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZmZmZmZddlmZmZdd lmZmZmZdd lmZdd l m!Z!m"Z"m#Z#e!e"fZ$e r dd l%m&Z&dd l'm(Z( ddl)m*Z*m+Z+m,Z,m)Z)d?dZ-n#e.$rd?dZ-YnwxYwdZ/dZ0dZ1ej2dkru ddl3Z4n#e.$rddl4Z4YnwxYwdZ5 e4j6e4j7d5Z8e5e8ddZ9e5e8ddZ:dddn #1swxYwYn#e;$rdZ9dZ:YnwxYwdZGd1d2Z?dDd6Z@dEd8ZAdFd:ZBdDd;ZCdEd<ZDdGd>ZEdS)Hz)Pool utilities and shared helper methods.) annotationsN) TYPE_CHECKINGAnyNoReturnOptionalUnion)_csot) _getaddrinfo) AutoReconnectConnectionFailureNetworkTimeout_CertificateError)_get_timeout_detailsformat_timeout_details)AsyncNetworkingInterfaceNetworkingInterfacePyMongoProtocol) PoolOptions) PYSSLErrorSSLError_has_sni)_sslConn)_Address)F_GETFDF_SETFD FD_CLOEXECfcntlfdintreturnNoneclt|t}t|t|tzdS)z8Set the close-on-exec flag on the given file descriptor.N)rrrr)rflagss 6D:\Nbitz\venv\Lib\site-packages\pymongo/pool_shared.py_set_non_inheritable_non_atomicr%4s.b'"" b'5:-.....cdS)z6Dummy function for platforms that don't provide fcntl.N)rs r$r%r%=sr&x win32c tj||\}}t|S#ttf$r|cYSwxYwN)winreg QueryValueExrOSError ValueError)keynamedefaultvalue_s r$_queryr8KsT *355HE1u:: $   NNN s &)??z2SYSTEM\CurrentControlSet\Services\Tcpip\Parameters KeepAliveTimeimKeepAliveIntervalctttdz}tttdz}|tks |tkr%|t jd||fdSdS)Nr;)min_WINDOWS_TCP_IDLE_MS_MAX_TCP_KEEPIDLE_WINDOWS_TCP_INTERVAL_MS_MAX_TCP_KEEPINTVLioctlsocketSIO_KEEPALIVE_VALS)sockidle_ms interval_mss r$_set_keepalive_timesrIasr*,=,DEE24F4MNN ) ) )[;S-S-S JJv01g{2K L L L L L.T-Sr&rF socket.socket tcp_optionstr max_valuectt|rrtt|} |tj|}||kr#|tj||dSdS#t $rYdSwxYwdSr.)hasattrrDgetattr getsockopt IPPROTO_TCP setsockoptr1)rFrKrMsockoptr5s r$_set_tcp_optionrUis 6: & & fj11G //&*>>>>r&addressrerror Exception msg_prefix Optional[str]timeout_detailsOptional[dict[str, float]]rcJ|\}}| d|||fz}n|d|}|r||z}d|vr|t|z }t|tjrt ||t|t r!dt |vrt ||t||)z9Convert a socket.error to ConnectionFailure and raise it.Nz %s:%d: %sz: zconfigured timeouts timed out)r isinstancerDtimeoutr SSLErrorsrLr )r\r]r_rahostportmsgs r$_raise_connection_failurerk|sJD$ T4//    3C'' %o666%(( ,S!!u, E9 % %,+U*C*C S!!u,C  e+r&c6eZdZddZddZed dZdS) _CancellationContextr r!cd|_dS)NF _cancelledselfs r$__init__z_CancellationContext.__init__s r&cd|_dS)zCancel this context.TNrorqs r$cancelz_CancellationContext.cancels r&boolc|jS)zWas cancel called?rorqs r$ cancelledz_CancellationContext.cancelleds r&N)r r!)r rv)__name__ __module__ __qualname__rsrupropertyrxr(r&r$rmrms\    Xr&rmroptionsrc K|\}}|drttdstdtjtj}t | |dtj ||d{V|S#t$r| wxYwtj }tjr|dkr tj}d}t!|||tjd{VD]}|\}} } } } tj|| t%tdd z| }n&#t$rtj|| | }YnwxYwt | |tjtjd t-j} | |j} n| d krtjd |tjtjd t9||dtjtj || | d{V|| |cS#tj$r:}| tjd }||_ Yd}~d}~wt$r!}| |}Yd}~d}~wwxYw||td)Given (host, port) and PoolOptions, connect and return a raw socket object. Can raise socket.error. This is a modified version of create_connection from CPython >= 2.7. .sockAF_UNIX-UNIX-sockets are not supported on this systemFN localhostfamilytype SOCK_CLOEXECrr=rdTrfgetaddrinfo failed)!endswithrOrDr rr%fileno setblockingasyncioget_running_loop sock_connectr1closeAF_INEThas_ipv6 AF_UNSPECr SOCK_STREAMrPrSrR TCP_NODELAYr remainingconnect_timeoutrf SOL_SOCKET SO_KEEPALIVErIwait_for settimeout TimeoutError __cause__r\r}rhrirFrerrresafsocktypeprotodummysarfes r$_async_create_connectionrswJD$ }}W vy)) U#$STT T}V^,,' 666    U # # #*,,99$EE E E E E E E EK    JJLLL  ^F "4;..! C!$V&BTUUUUUUUUU$$),&HeUB 6=XPQ0R0R%RTYZZDD 6 6 6=Xu55DDD 6 ( 666  OOF.0BA F F Fo''G!1An[111 OOF-v/BD I I I  & & &   U # # #"(**77bAA7        OOG $ $ $KKK#    JJLLL.--CCMMMMMM    JJLLLCCCCCC   *+++sDAC C%-E11 FF9DKL6/L L6L11L6Union[socket.socket, _sslConn]cKt||d{V}|j}|||j|S|d} t drJt j}|dtj |j ||d{V}n5t j}|d|j |d{V}nt#t$r| tgtR$r@}| t|}t!||d|Yd}~nd}~wwxYw|jr[|jsT|jsM t)j||n##t$r| wxYw||j|S)Given (host, port) and PoolOptions, return a raw configured socket. Can raise socket.error, ConnectionFailure, or _CertificateError. Sets socket's SSL and timeout options. NrFserver_hostnameSSL handshake failed: rahostname)r _ssl_contextrsocket_timeoutrrrrun_in_executor functoolspartial wrap_socketrrr1rgrrk verify_modecheck_hostnametls_allow_invalid_hostnamessslmatch_hostname getpeercert) r\r}rF ssl_contextrhloopssl_sockexcdetailss r$_async_configured_socketrs?*'7;; ; ; ; ; ; ;D&K ./// 1:Dc E?? W+--D!11!+"94QUVVVHH +--D!11$ 8OQUVVVVVVVVH     i ccc 'w//!'30HZabbbbbbbbb c  * 3     x3355 E E E E E     NN      ./// Os%BC1E6EE (F F)rc Kt||d{V}|j}|j |>tt j fd|d{VS|d} t j fd|||d{V\}}nL#t$rtgtR$r,}t|}t||d|Yd}~nd}~wwxYw|j r\|j sU|jsN tj|d | n##t$r|wxYwt||fS) zGiven (host, port) and PoolOptions, return a configured AsyncNetworkingInterface. Can raise socket.error, ConnectionFailure, or _CertificateError. Sets protocol's SSL and timeout options. Nc$tSNrrrsr$z0_configured_protocol_interface..?s888r&r[rc$tSrrrsr$rz0_configured_protocol_interface..HsOG444r&)rFrrrrpeercertr)rrrrrrcreate_connectionrr1rgrrkrrrrrget_extra_infoabort) r\r}rFrrh transportprotocolrrrfs @r$_configured_protocol_interfacer/s"*'7;; ; ; ; ; ; ;D&K$G'*,,>>8888t?         1:Dc%,$<$>$>$P$P 4 4 4 4  %Q% %        88    i ccc'w//!'30HZabbbbbbbbb c  * 3     y77 CCd S S S S S     OO      $Y$9 : ::s$16B((C1"C,,C1 )D44 Ec (|\}}|drttdstdtjtj}t | |||S#t$r| wxYwtj }tj r|dkr tj }d}tj |||tjD]x}|\}} } } } tj|| ttddz| }n&#t$rtj|| | }YnwxYwt | |tjtjd t'j} | |j} n| dkrtjd || |tjtjd t5||| |cS#t$r!}|}| Yd}~rd}~wwxYw||td ) rrrrrNrrrr=rdTr)rrOrDr rr%rconnectr1rrrr getaddrinforrPrSrRrr rrrfrrrrIrs r$_create_connectionresJD$ }}W vy)) U#$STT T}V^,,' 666  LL   K    JJLLL  ^F "4;..! C!$V&BTUUU),&HeUB 6=XPQ0R0R%RTYZZDD 6 6 6=Xu55DDD 6 ( 666  OOF.0BA F F Fo''G!1An[111 OOG $ $ $ OOF-v/BD I I I  & & & LL   KKK   C JJLLLLLLLL   *+++s7?B B6-E E$#E$ CI I>I99I>ct||}|j}|||j|S|d} t dr|||}n||}nt#t $r|tgtR$r@}|t|}t||d|Yd}~nd}~wwxYw|j r[|j sT|jsM tj||n##t $r|wxYw||j|S)rNrTrrrr)rrrrrrrrr1rgrrkrrrrrrr\r}rFrrhrrrs r$_configured_socketrs gw / /D&K ./// 1:Dc D>> 5"..tT.JJHH"..t44H     i ccc 'w//!'30HZabbbbbbbbb c  * 3     x3355 E E E E E     NN      ./// Os#> 5"..tT.JJHH"..t44H     i ccc 'w//!'30HZabbbbbbbbb c  * 3     x3355 E E E E E     NN      ./// x ( ((s$ rs0/"""""" 555555 POOOOOOO``````````,,,,,,>>>>>>>>>>  " )222222(((((( E999999999999///// EEEEEEEEE E<7       ( V^  %'\   N #)6##H#H '-vc3F'M'M $  N N N N N N N N N N N N N N N ((( '#'   ( MMMM    ????!%26 ,,,,,:        M,M,M,M,`5555p3;3;3;3;lD,D,D,D,N....b.).).).).).)sZ8B BB*B// B;:B;D C?3 D ?DD DD DD