o ACe?@sUddlmZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZddlZddlZddlmZmZerWddlmZmZddlmZddlmZmZmZe ejejfZde d <d/d dZ!d0ddZ"d0ddZ#d1ddZ$d2ddZ%e ddd Z&ej'Gd!ddZ(ej'Gd"d#d#Z)ej'Gd$d%d%Z*ej'Gd&d'd'ej+j,Z-ej'Gd(d)d)ej+j.Z/eGd*d+d+Z0eGd,d-d-ejj1ed.Z2dS)3) annotationsN) TYPE_CHECKINGAnyIterableNoReturnTypeVarUnionoverload)NoPublicConstructorfinal) AddressFamily SocketKind) TracebackType)BufferSelf TypeAliasr IPAddressipreturnintcCs0t|tjr tjjSt|tjrtjjStd)Nz!Unhandled IPAddress instance type) isinstance ipaddress IPv4AddresstriosocketAF_INET IPv6AddressAF_INET6NotImplementedError)rr:usr/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_for0    r*cCr#)N 127.0.0.1::1r&r'r)rrr _localhost_ip_for8r+r.codercCst|t|N)OSErrorosstrerror)r/rrr _fake_err@sr4databytesbuffersIterable[Buffer]c Cs~d}|D]8}|||t|j}t|}||dt|<Wdn1s(wY|t|7}|t|kr<|Sq|S)Nr) memoryviewnbyteslen)r5r7writtenbufZ next_pieceZmbufrrr _scatterDs   r> T_UDPEndpoint UDPEndpoint)boundc@s6eZdZUded<ded<dddZedd dZdS)r@rrrportr+tuple[str, int] | tuple[str, int, int, int]cCs(|jj|jf}t|jtjr|d7}|S)Nrr)r compressedrBrrr)selfsockaddrrrr as_python_sockaddrXs zUDPEndpoint.as_python_sockaddrclstype[T_UDPEndpoint]rGr?cCs"|dd\}}|t||dS)N)rrB)rr()rIrGrrBrrr from_python_sockaddrasz UDPEndpoint.from_python_sockaddrNrrC)rIrJrGrCrr?)__name__ __module__ __qualname____annotations__rH classmethodrLrrrr r@Ss   c@seZdZUded<dS) UDPBindingr@localN)rNrOrPrQrrrr rSis rSc@s@eZdZUded<ded<ejdddZded<d d d Zd S) UDPPacketr@source destinationcCs|Sr0)hex)prrr sszUDPPacket.)reprr6payloadrcCst|j|j|dS)NrVrWr\)rUrWrV)rFr\rrr replyus zUDPPacket.replyN)r\r6rrU)rNrOrPrQattrZibr\r^rrrr rUos rUc@s eZdZUded<d d d Zd S) FakeSocketFactoryFakeNetfake_netr"rtypeprotor FakeSocketcCst|j|||Sr0)re_createrb)rFr"rcrdrrr rszFakeSocketFactory.socketN)r"rrcrrdrrre)rNrOrPrQrrrrr r`{s r`c@s4eZdZUded<    ddddZdddZdS)FakeHostnameResolverrarbrhostbytes | str | NonerBbytes | str | int | Noner"rrcrdflagsr]list[tuple[AddressFamily, SocketKind, int, str, tuple[str, int] | tuple[str, int, int, int]]]c tdNzFakeNet doesn't do fake DNS yetr)rFrhrBr"rcrdrkrrr getaddrinfosz FakeHostnameResolver.getaddrinforGrCtuple[str, str]crmrnro)rFrGrkrrr getnameinfosz FakeHostnameResolver.getnameinfoN)rrrr)rhrirBrjr"rrcrrdrrkrrrl)rGrCrkrrrq)rNrOrPrQrprrrrrr rgs  rgc@s>eZdZdddZdd d Zdd d ZdddZdddZdS)rarNonecCs@td|_td|_ttdd|_i|_d|_ dS)Nz 1.0.0.0/8z1::/16iPi) rZ IPv4NetworkZhostsZ_auto_ipv4_iterZ IPv6Networkiterrange_auto_port_iter_bound route_packetrFrrr __init__s  zFakeNet.__init__bindingrSrrecCs"||jvr ttj||j|<dSr0)rwr4errnoZ EADDRINUSE)rFr{rrrr _binds  z FakeNet._bindcCs$tjt|tjt|dSr0)rrZset_custom_socket_factoryr`Zset_custom_hostname_resolverrgryrrr enableszFakeNet.enablepacketrUcCs&|jdur ||dS||dSr0)rxdeliver_packetrFrrrr send_packets zFakeNet.send_packetcCs0t|jd}||jvr|j||dS dS)NrT)rSrWrw_deliver_packet)rFrr{rrr rs  zFakeNet.deliver_packetNrrs)r{rSrrerrsrrUrrs)rNrOrPrzr}r~rrrrrr ras    rac@seZdZdzd d Zed{d d Zed|ddZed}ddZd~ddZd~ddZ dddZ dd d!Z dd#d$Z dd'd(Z dd+d,Z - -ddd3d4Zdd6d7Zdd8d9Zeddttj|j|WddS1swYdSr0) contextlibsuppressrZBrokenResourceErrorrZ send_nowaitrrrr rs"zFakeSocket._deliver_packetaddrcs||jdurttjtjIdH|j|ddIdH\}}t |}t ||j ks1J|t dkr>t d}n |t dkrJt d}|dkrTt |jj}tt||d}|j||||_dS)NTrr$r,r%r-r)rrr4r|EINVALrlowlevel checkpointrrr(r!r"nextrrvrSr@r})rFrZip_strrBrr{rrr binds"       zFakeSocket.bindpeerrcrm)Nz0FakeNet does not (yet) support connected socketsro)rFrrrr connect#szFakeSocket.connectargsrc sD|g}d}d}t|dkr|\}n(t|dkr |\}}nt|dkr,|\}}}nt|dkr9|\}}}}ntdtjIdH|durS|j|ddIdH}|rYtd |rbtd ||durktt j t |}|j dur|t|jjdfIdHd |}|j dusJt|j j||d }|j|t|S) NrrKwrong number of argumentsFr&FakeNet doesn't support ancillary dataz"FakeNet send flags must be 0, not r])rr; TypeErrorrrrrrr4r|ENOTCONNr@rLrrr*r"rEjoinrUrTrr) rFrancdatarkrr7rWr\rrrr sendmsg&sF           zFakeSocket.sendmsgrr7r8 ancbufsizerk2tuple[int, list[tuple[int, int, bytes]], int, Any]c s~|dkr td|dkrtd|g}d}|jIdH}|j}t|j|}|t|jkr9|t j j O}||||fS)Nrrz&FakeNet doesn't support any recv flags) rrrZreceiverVrHr>r\r;rr MSG_TRUNC) rFr7rrkr msg_flagsrrr<rrr recvmsg_intoXs    zFakeSocket.recvmsg_intorCcCsF||jdur|jjS|jtjjkrdS|jtjjks!JdS)N)r$r)r%r) rrrTrHr"rrrrryrrr getsocknamess  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@rHr4r|rryrrr getpeername~s   zFakeSocket.getpeernameleveloptnamecCdSr0r)rFrrrrr getsockoptzFakeSocket.getsockoptbuflenr6cCrr0rrFrrrrrr rrN int | None int | bytescCs|td|d|d)Nz%FakeNet doesn't implement getsockopt(, ))rr1rrrr rsvalue int | BuffercCrr0r)rFrrrrrr setsockoptrzFakeSocket.setsockoptoptlencCrr0rrFrrrrrrr rrint | Buffer | NonecCsB|||ftjjtjjfkr|stdtd|d|d)Nz#FakeNet always has IPV6_V6ONLY=Truez%FakeNet doesn't implement setsockopt(rz, ...))rrr IPPROTO_IPV6 IPV6_V6ONLYrr1rrrr rsrcCs|Sr0rryrrr __enter__zFakeSocket.__enter__exc_type#builtins.type[BaseException] | None exc_valueBaseException | None tracebackTracebackType | NonecCs |dSr0)r)rFrrrrrr __exit__s zFakeSocket.__exit__r5rcs|||dIdHSr0)sendto)rFr5rkrrr sendszFakeSocket.sendcsTt|dkr|\}}d}nt|dkr|\}}}ntd||gg||IdHS)NrKrrr)r;rr)rFrr5rrkrrr rs   zFakeSocket.sendtobufsizecs|||IdH\}}|Sr0)recvfrom)rFrrkr5rrrr recvszFakeSocket.recvr=r:cs||||IdH\}}|Sr0) recvfrom_into)rFr=r:rkZ got_bytesrrrr recv_intoszFakeSocket.recv_intotuple[bytes, Any]cs$|||IdH\}}}}||fSr0)recvmsg)rFrrkr5rrrrrr rszFakeSocket.recvfromtuple[int, Any]csF|dkr|t|jkrtd||gd|IdH\}}}}||fS)Nrzpartial recvfrom_into)r9r:rr)rFr=r:rk got_nbytesrrrrrr rszFakeSocket.recvfrom_into4tuple[bytes, list[tuple[int, int, bytes]], int, Any]c s@t|}||g||IdH\}}}}t|d||||fSr0) bytearrayrr6) rFrrrkr=rrrrrrr rs zFakeSocket.recvmsgcCtd)Nz&can't get fileno() for FakeNet socketsroryrrr filenozFakeSocket.filenocCr)Nzcan't detach() a FakeNet socketroryrrr detachrzFakeSocket.detachcCsdS)NFrryrrr get_inheritablerzFakeSocket.get_inheritable inheritablecCs|rtddS)Nz&FakeNet can't make inheritable socketsro)rFrrrr set_inheritableszFakeSocket.set_inheritable process_idcCr)NzFakeNet can't share socketsro)rFrrrr sharerzFakeSocket.share)rbrar"r rcr rdr)rr )rr )rrr)rrrTrrrr)rrrrs)rrrr)rrrrrD)r7r8rrrkrrrrM)rrrrrr)rrrrrrrr6r0)rrrrrrrr)rrrrrrrrs) rrrrrrsrrrrs) rrrrrrrrrrs)rr)rrrrrrrrs)r)r5rrkrrr)rrrkrrr6)r=rr:rrkrrr)rrrkrrr)r=rr:rrkrrr)rrrrrkrrr)rr)rrrrs)rrrr6)#rNrOrPrzpropertyrcr"rdrrrrrrrrrrr rrrrrrrrrrrrrrrrrrrr res`        5                 re) metaclass)rrrr)r"rrr)r/rrr)r5r6r7r8rr)3 __future__rbuiltinsrr|rr2typingrrrrrrr r_rZ trio._utilr r rr r typesrtyping_extensionsrrrrrrrQr!r*r.r4r>r?frozenr@rSrUabcZ SocketFactoryr`ZHostnameResolverrgra SocketTypererrrr sF$         #