o dTilG@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z ddlZddlZddlmZmZe reddlZddlmZddlmZmZddlmZddlmZmZdd lmZejej BZ!d e"d <d/ddZ#d0ddZ$d0ddZ%d1ddZ&d2ddZ'ej(Gd d!d!Z)ej(Gd"d#d#Z*ej(Gd$d%d%Z+ej(Gd&d'd'ej,j-Z.ej(Gd(d)d)ej,j/Z0eGd*d+d+Z1eGd,d-d-ejj2ed.Z3dS)3) annotationsN) TYPE_CHECKINGAnyNoReturn TypeAliasoverload)NoPublicConstructorfinal)Iterable) AddressFamily SocketKind) TracebackType)BufferSelf) AddressFormatr IPAddressipreturnintcCs0t|tjr tjjSt|tjrtjjStd)Nz!Unhandled IPAddress instance type) isinstance ipaddress IPv4AddresstriosocketAF_INET IPv6AddressAF_INET6NotImplementedError)rrR/home/air/sos-pdf/back/venv/lib/python3.10/site-packages/trio/testing/_fake_net.py _family_for%s  r familycC4|tjjkr tdS|tjjkrtdStd)N0.0.0.0::Unhandled ip address familyrrrr ip_addressrrr!rrr_wildcard_ip_for-    r)cCr")N 127.0.0.1::1r%r&r(rrr_localhost_ip_for6r*r-codercCst|t|N)OSErrorosstrerror)r.rrr _fake_err>sr3databytesbuffersIterable[Buffer]c Cs~d}|D]8}|||t|j}t|}||dt|<Wdn1s(wY|t|7}|t|kr<|Sq|S)Nr) memoryviewnbyteslen)r4r6writtenbuf next_piecembufrrr_scatterBs   r?c@s6eZdZUded<ded<d ddZedd d Zd S) UDPEndpointrrrportr+tuple[str, int] | tuple[str, int, int, int]cCs(|jj|jf}t|jtjr|d7}|S)Nrr)r compressedrArrr)selfsockaddrrrras_python_sockaddrSs zUDPEndpoint.as_python_sockaddrrFcCs"|dd\}}|t||dS)N)rrA)rr')clsrFrrArrrfrom_python_sockaddr\sz UDPEndpoint.from_python_sockaddrNrrB)rFrBrr@)__name__ __module__ __qualname____annotations__rG classmethodrJrrrrr@Ns   r@c@seZdZUded<dS) UDPBindingr@localN)rLrMrNrOrrrrrQes rQc@s@eZdZUded<ded<ejdddZded<d d d Zd S) UDPPacketr@source destinationcCs|Sr/)hex)prrroszUDPPacket.)reprr5payloadrcCst|j|j|dS)NrTrUrZ)rSrUrT)rErZrrrreplyrs zUDPPacket.replyN)rZr5rrS)rLrMrNrOattrsfieldrZr\rrrrrSks rSc@s eZdZUded<d d d Zd S) FakeSocketFactoryFakeNetfake_netr!rtype_protor FakeSocketcCst|j|||Sr/)rd_createra)rEr!rbrcrrrr~szFakeSocketFactory.socketN)r!rrbrrcrrrd)rLrMrNrOrrrrrr_zs r_c@s4eZdZUded<    ddddZdddZdS)FakeHostnameResolverr`rarhost bytes | NonerAbytes | str | int | Noner!rtypercflagsrqlist[tuple[AddressFamily, SocketKind, int, str, tuple[str, int] | tuple[str, int, int, int] | tuple[int, bytes]]]c tdNzFakeNet doesn't do fake DNS yetr)rErgrAr!rjrcrkrrr getaddrinfosz FakeHostnameResolver.getaddrinforFrBtuple[str, str]crmrnro)rErFrkrrr getnameinfosz FakeHostnameResolver.getnameinfoN)rrrr)rgrhrArir!rrjrrcrrkrrrl)rFrBrkrrrq)rLrMrNrOrprrrrrrrfs  rfc@s>eZdZdddZdd d Zdd d ZdddZdddZdS)r`rNonecCs@td|_td|_ttdd|_i|_ d|_ dS)Nz 1.0.0.0/8z1::/16iPi) r IPv4Networkhosts_auto_ipv4_iter IPv6Network_auto_ipv6_iteriterrange_auto_port_iter_bound route_packetrErrr__init__s  zFakeNet.__init__bindingrQrrdcCs"||jvr ttj||j|<dSr/)r|r3errno EADDRINUSE)rErrrrr_binds  z FakeNet._bindcCs$tjt|tjt|dSr/)rrset_custom_socket_factoryr_set_custom_hostname_resolverrfr~rrrenableszFakeNet.enablepacketrScCs&|jdur ||dS||dSr/)r}deliver_packetrErrrr send_packets zFakeNet.send_packetcCs0t|jd}||jvr|j||dS dS)NrR)rQrUr|_deliver_packet)rErrrrrrs  zFakeNet.deliver_packetNrrs)rrQrrdrrsrrSrrs)rLrMrNrrrrrrrrrr`s    r`c@sFeZdZdd d Zedd dZedddZedddZdddZdddZ dddZ dd d!Z dd#d$Z dd'd(Z ) * +ddd2d3Zejd4ksResTeejd5rTeZ * *ddd8d9Zejd4ksieskeejd5rkeZdd;d<Zdd=d>ZeddAdBZeddEdBZ +dddHdBZeddKdLZeddNdLZ +dddPdLZddRdSZddZd[Zddd^d_ZeddcddZeddgddZddjddZdddldmZ dddpdqZ! *dddsdtZ" * *dddvdwZ# * *dddydzZ$ejd4kseseejd5re$Z%dd{d|Z&dd}d~Z'dddZ(dddZ)ejd4kseseejdr!dddZ*d+Sd+Sd+S)rdrar`r!r rjr rcrrrscCs||_|s tjj}|stjj}|tjjtjjfvr td||tjjkr-td|||_||_ ||_ d|_ tj t td\|_|_d|_dS)Nz%FakeNet doesn't (yet) support family=z#FakeNet doesn't (yet) support type=Finf) _fake_netrrr SOCK_STREAMrr SOCK_DGRAM_family_type_proto_closedopen_memory_channelrSfloat_packet_sender_packet_receiver_binding)rErar!rjrcrrrrs&  zFakeSocket.__init__cC|jSr/)rr~rrrrjzFakeSocket.typecCrr/)rr~rrrr!rzFakeSocket.familycCrr/)rr~rrrrcrzFakeSocket.protocCs|jr ttjdSdSr/)rr3rEBADFr~rrr _check_closedszFakeSocket._check_closedcCs4|jrdSd|_|jdur|jj|j=|jdS)NT)rrrr|rcloser~rrrrs   zFakeSocket.closeaddressobjectrRbooltuple[str, int]cs&tjj|j|j|j|d|dIdHS)NF)r ipv6_v6onlyrR)r_socket_resolve_address_nocprjr!rc)rErrRrrrrs z FakeSocket._resolve_address_nocprrScCs>ttj|j|WddS1swYdSr/) contextlibsuppressrBrokenResourceErrorr send_nowaitrrrrrs"zFakeSocket._deliver_packetaddrcs||jdurttjtjIdH|j|ddIdH^}}}|gks,Jdt |}t ||j ks:J|t dkrGt d}n |t dkrSt d}|dkr]t |jj}tt||d}|j||||_dS) NTrzTODO: handle other values?r#r+r$r,r)rrr3rEINVALrlowlevel checkpointrrr'r r!nextrr{rQr@r)rErip_strrA_rrrrrbinds$       zFakeSocket.bindpeerrcrm)Nz0FakeNet does not (yet) support connected socketsro)rErrrrconnect,szFakeSocket.connectrrNr6r7ancdata!Iterable[tuple[int, int, Buffer]]rkAddressFormat | Nonecs|tjIdH|dur|j|ddIdH}|r!td|r*td||dur3ttjt |}|j durK| t |jjdfIdHd|}|j dusWJt|j j||d}|j|t|S)NFr&FakeNet doesn't support ancillary dataz"FakeNet send flags must be 0, not rr[)rrrrrrr3rENOTCONNr@rJrrr)r!rDjoinrSrRrrr:)rEr6rrkrrUrZrrrr_sendmsg/s.     zFakeSocket._sendmsgwin32sendmsg ancbufsizeZtuple[int, list[tuple[int, int, bytes]], int, tuple[str, int] | tuple[str, int, int, int]]c s|dkr td|dkrtd|jdurtd|g}d}|jIdH}|j}t|j|}|t |jkrB|t j j O}||||fS)Nrrz&FakeNet doesn't support any recv flagszThe code will most likely hang if you try to receive on a fakesocket without a binding. If that is not the case, or you explicitly want to test that, remove this warning.) rrrrreceiverTrGr?rZr:rr MSG_TRUNC) rEr6rrkr msg_flagsrrr;rrr _recvmsg_into[s$      zFakeSocket._recvmsg_intorBcCsF||jdur|jjS|jtjjkrdS|jtjjks!JdS)N)r#r)r$r) rrrRrGr!rrrrr~rrr getsocknames  zFakeSocket.getsocknamecCsb||jdur*t|jdsJd|jjdur*t|jjts$Jd|jjSttj dS)NremotezGThis method seems to assume that self._binding has a remote UDPEndpointz,Self._binding.remote should be a UDPEndpoint) rrhasattrrrr@rGr3rrr~rrr getpeernames"   zFakeSocket.getpeernameleveloptnamecCdSr/r)rErrrrr getsockoptzFakeSocket.getsockoptbuflenr5cCrr/rrErrrrrrrr int | None int | bytescCs|td|d|d)Nz%FakeNet doesn't implement getsockopt(, ))rr0rrrrrsvalue int | BuffercCrr/r)rErrrrrr setsockoptrzFakeSocket.setsockoptoptlencCrr/rrErrrrrrrrsint | Buffer | NonecCsB|||ftjjtjjfkr|stdtd|d|d)Nz#FakeNet always has IPV6_V6ONLY=Truez%FakeNet doesn't implement setsockopt(rz, ...))rrr IPPROTO_IPV6 IPV6_V6ONLYrr0rrrrrsrcCs|Sr/rr~rrr __enter__rzFakeSocket.__enter__exc_type#builtins.type[BaseException] | None exc_valueBaseException | None tracebackTracebackType | NonecCs |dSr/)r)rErrrrrr__exit__s zFakeSocket.__exit__r4rcs|||dIdHSr/)sendto)rEr4rkrrrsendszFakeSocket.send_FakeSocket__data_FakeSocket__address!tuple[object, ...] | str | BuffercdSr/r)rErrrrrrszFakeSocket.sendto_FakeSocket__flags(tuple[object, ...] | str | Buffer | Nonecrr/r)rErrrrrrrsargsrcsTt|dkr|\}}d}nt|dkr|\}}}ntd||gg||IdHS)NrHrzwrong number of arguments)r: TypeErrorr)rErr4rrkrrrrs   bufsizecs|||IdH\}}|Sr/)recvfrom)rErrkr4_addressrrrrecvszFakeSocket.recvr<r9cs||||IdH\}}|Sr/) recvfrom_into)rEr<r9rk got_bytesrrrr recv_intoszFakeSocket.recv_intotuple[bytes, AddressFormat]cs$|||IdH\}}}}||fSr/)_recvmsg)rErrkr4_ancdata _msg_flagsrrrrr szFakeSocket.recvfromtuple[int, AddressFormat]csF|dkr|t|jkrtd||gd|IdH\}}}}||fS)Nrzpartial recvfrom_into)r8r9rr)rEr<r9rk got_nbytesrrrrrrrszFakeSocket.recvfrom_into>tuple[bytes, list[tuple[int, int, bytes]], int, AddressFormat]c s@t|}||g||IdH\}}}}t|d||||fSr/) bytearrayrr5) rErrrkr<rrrrrrrr#szFakeSocket._recvmsgcCtd)Nz&can't get fileno() for FakeNet socketsror~rrrfileno6zFakeSocket.filenocCr)Nzcan't detach() a FakeNet socketror~rrrdetach9rzFakeSocket.detachcCsdS)NFrr~rrrget_inheritable<rzFakeSocket.get_inheritable inheritablecCs|rtddS)Nz&FakeNet can't make inheritable socketsro)rErrrrset_inheritable?szFakeSocket.set_inheritableshare process_idcCr)NzFakeNet can't share socketsro)rErrrrrGrzFakeSocket.share) rar`r!r rjr rcrrrs)rr )rr )rrr)rrrRrrrr)rrrrs)rrrr)rrN) r6r7rrrkrrrrrrC)r6r7rrrkrrrrK)rrrrrr)rrrrrrrr5r/)rrrrrrrr)rrrrrrrrs) rrrrrrsrrrrs) rrrrrrrrrrs)rr)rrrrrrrrs)r)r4rrkrrr)rrrrrr)rrrrrrrr)rrrr)rrrkrrr5)r<rr9rrkrrr)rrrkrrr)r<rr9rrkrrr)rrrrrkrrr)rr)rrrrs)rrrr5)+rLrMrNrpropertyrjr!rcrrrrrrrsysplatformrrrrr recvmsg_intorrrrrrrrrrrrrrrecvmsgrrrrrrrrrrds          '  $                           rd) metaclass)rrrr)r!rrr)r.rrr)r4r5r6r7rr)4 __future__rrrrr1rr typingrrrrrr]r trio._utilrr builtinscollections.abcr r r typesr typing_extensionsrr trio._socketrrrrrOr r)r-r3r?frozenr@rQrSabc SocketFactoryr_HostnameResolverrfr` SocketTyperdrrrrsL        #