U gF@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZmZddlZddlZddlmZmZe rddlZddlmZmZddlmZddlmZmZmZeejejfZ de!d <d d d d d Z"d d dddZ#d d dddZ$d ddddZ%ddd dddZ&e dddZ'ej(GdddZ)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)-) annotationsN) TYPE_CHECKINGAnyIterableNoReturnTypeVarUnionoverload)NoPublicConstructorfinal) AddressFamily SocketKind) TracebackType)BufferSelf TypeAliasr IPAddressint)ipreturncCs4t|tjrtjjSt|tjr(tjjStddS)Nz!Unhandled IPAddress instance type) isinstance ipaddress IPv4AddresstriosocketAF_INET IPv6AddressAF_INET6NotImplementedError)rr:/tmp/pip-unpacked-wheel-ks04xdmi/trio/testing/_fake_net.py _family_for*s   r!)familyrcCs8|tjjkrtdS|tjjkr,tdStddS)N0.0.0.0::Unhandled ip address familyrrrr ip_addressrrr"rrr _wildcard_ip_for2s     r)cCs8|tjjkrtdS|tjjkr,tdStddS)N 127.0.0.1::1r%r&r(rrr _localhost_ip_for;s     r,r)codercCst|t|dSN)OSErrorosstrerror)r-rrr _fake_errCsr2bytesIterable[Buffer])databuffersrc Cshd}|D]Z}|||t|j}t|}||dt|<W5QRX|t|7}|t|krqdq|S)Nr) memoryviewnbyteslen)r5r6writtenbufZ next_pieceZmbufrrr _scatterGs   r< T_UDPEndpoint UDPEndpoint)boundc@sBeZdZUded<ded<ddddZed dd d d d ZdS)r>rrrport+tuple[str, int] | tuple[str, int, int, int]rcCs(|jj|jf}t|jtjr$|d7}|S)N)rr)r compressedr@rrr)selfsockaddrrrr as_python_sockaddr[s zUDPEndpoint.as_python_sockaddrztype[T_UDPEndpoint]r=)clsrErcCs"|dd\}}|t||dS)N)rr@)rr')rGrErr@rrr from_python_sockaddrdsz UDPEndpoint.from_python_sockaddrN)__name__ __module__ __qualname____annotations__rF classmethodrIrrrr r>Vs  c@seZdZUded<dS) UDPBindingr>localN)rJrKrLrMrrrr rOms rOc@sFeZdZUded<ded<ejdddZded<ddd d d Zd S) UDPPacketr>source destinationcCs|Sr.)hex)prrr wzUDPPacket.)reprr3payload)rYrcCst|j|j|dS)NrRrSrY)rQrSrR)rDrYrrr replyzs zUDPPacket.replyN)rJrKrLrMattrsfieldrYr[rrrr rQss rQc@s*eZdZUded<dddddddZdS) FakeSocketFactoryFakeNetfake_netr FakeSocket)r"type_protorcCst|j|||Sr.)ra_creater`)rDr"rbrcrrr rszFakeSocketFactory.socketN)rJrKrLrMrrrrr r^s r^c @sDeZdZUded<dddddddddd d Zd dd d ddZdS)FakeHostnameResolverr_r`rz bytes | Nonezbytes | str | int | Nonerz]list[tuple[AddressFamily, SocketKind, int, str, tuple[str, int] | tuple[str, int, int, int]]])hostr@r"typercflagsrcs tddSNzFakeNet doesn't do fake DNS yetr)rDrfr@r"rgrcrhrrr getaddrinfosz FakeHostnameResolver.getaddrinforAztuple[str, str])rErhrcs tddSrirj)rDrErhrrr getnameinfosz FakeHostnameResolver.getnameinfoN)rrrr)rJrKrLrMrkrlrrrr res rec@sZeZdZddddZdddddd Zddd d Zd dd ddZd dd ddZdS)r_NonerBcCs@td|_td|_ttdd|_i|_ d|_ dS)Nz 1.0.0.0/8z1::/16iPi) r IPv4NetworkhostsZ_auto_ipv4_iter IPv6NetworkZ_auto_ipv6_iteriterrange_auto_port_iter_bound route_packetrDrrr __init__s zFakeNet.__init__rOra)bindingrrcCs"||jkrttj||j|<dSr.)rtr2errnoZ EADDRINUSE)rDrxrrrr _binds  z FakeNet._bindcCs$tjt|tjt|dSr.)rrZset_custom_socket_factoryr^Zset_custom_hostname_resolverrervrrr enableszFakeNet.enablerQpacketrcCs$|jdkr||n ||dSr.)rudeliver_packetrDr}rrr send_packets  zFakeNet.send_packetcCs,t|jd}||jkr(|j||ndS)NrP)rOrSrt_deliver_packet)rDr}rxrrr r~s  zFakeNet.deliver_packetN)rJrKrLrwrzr{rr~rrrr r_s  r_c@s:eZdZdddddddZeddd d Zeddd d Zeddd dZddddZddddZ ddddddZ dddddZ ddddd Z dd!d"d#d$Z d~d(d)dd*dd+d,d-Zejd.kseseejd/reZdd(ddd0d1d2d3Zejd.kses eejd/r eZd4dd5d6Zd4dd7d8Zedddd9d:d;Zeddddd;Zdddd?d@d=dAd;ZedddBddCdDdEZeddddddFdGdEZddddHd?ddFdIdEZdJddKdLZdMdNdOddPdQdRZddSdddTdUdVZedSdWddXdYdZZedSdd[dd\d]dZZd^dd_d`dZZddddrz)rDrip_strr@_rrxrrr binds"      zFakeSocket.bindr)peerrcs tddS)Nz0FakeNet does not (yet) support connected socketsrj)rDrrrr connect4szFakeSocket.connectrrNr4z!Iterable[tuple[int, int, Buffer]]z Any | None)r6ancdatarhrrcs|tjIdH|dk r4|j|ddIdH}|r@td|rRtd||dkrdttjt |}|j dkr| t |jjdfIdHd|}|j dk stt|j j||d}|j|t|S)NFr&FakeNet doesn't support ancillary dataz"FakeNet send flags must be 0, not rrWrZ)rrrrrrr2ryENOTCONNr>rIrrr)r"rCjoinrrQrPrrr9)rDr6rrhrrSrYr}rrr _sendmsg7s,     zFakeSocket._sendmsgwin32sendmsgz2tuple[int, list[tuple[int, int, bytes]], int, Any])r6 ancbufsizerhrc s|dkrtd|dkr td|jdkr2td|g}d}|jIdH}|j}t|j|}|t |jkr|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.) rrrrZreceiverRrFr<rYr9rr MSG_TRUNC) rDr6rrhr msg_flagsr}rr:rrr _recvmsg_intocs"    zFakeSocket._recvmsg_intorAcCsJ||jdk r|jjS|jtjjkr0dS|jtjjksBt dSdS)N)r#r)r$r) rrrPrFr"rrrrrrvrrr getsocknames  zFakeSocket.getsocknamecCsb||jdk rTt|jds&td|jjdk rTt|jjtsHtd|jjStt j dS)NremotezGThis method seems to assume that self._binding has a remote UDPEndpointz,Self._binding.remote should be a UDPEndpoint) rrhasattrrrrr>rFr2ryrrvrrr getpeernames"   zFakeSocket.getpeername)leveloptnamercCsdSr.r)rDrrrrr getsockoptszFakeSocket.getsockoptr3)rrbuflenrcCsdSr.rrDrrrrrr rsz int | Nonez int | bytescCs"|td|d|ddS)Nz%FakeNet doesn't implement getsockopt(, ))rr/rrrr rsz int | Buffer)rrvaluercCsdSr.r)rDrrrrrr setsockoptszFakeSocket.setsockopt)rrroptlenrcCsdSr.rrDrrrrrrr rszint | Buffer | NonecCsF|||ftjjtjjfkr,|s,tdtd|d|ddS)Nz#FakeNet always has IPV6_V6ONLY=Truez%FakeNet doesn't implement setsockopt(rz, ...))rrr IPPROTO_IPV6 IPV6_V6ONLYrr/rrrr rsrcCs|Sr.rrvrrr __enter__szFakeSocket.__enter__z#builtins.type[BaseException] | NonezBaseException | NonezTracebackType | None)exc_type exc_value tracebackrcCs |dSr.)r)rDrrrrrr __exit__szFakeSocket.__exit__r)r5rhrcs|||dIdHSr.)sendto)rDr5rhrrr sendszFakeSocket.sendz!tuple[object, ...] | str | Buffer)_FakeSocket__data_FakeSocket__addressrcsdSr.r)rDrrrrr rszFakeSocket.sendtoz(tuple[object, ...] | str | None | Buffer)r_FakeSocket__flagsrrcsdSr.r)rDrrrrrr rsr)argsrcsRt|dkr|\}}d}n t|dkr2|\}}}ntd||gg||IdHS)NrHrzwrong number of arguments)r9 TypeErrorr)rDrr5rrhrrr rs   )bufsizerhrcs|||IdH\}}|Sr.)recvfrom)rDrrhr5rrrr recvszFakeSocket.recv)r;r8rhrcs||||IdH\}}|Sr.) recvfrom_into)rDr;r8rhZ got_bytesrrrr recv_intoszFakeSocket.recv_intoztuple[bytes, Any]cs"|||IdH\}}}}||fSr.)_recvmsg)rDrrhr5rrrrrr r szFakeSocket.recvfromztuple[int, Any]csD|dkr|t|jkrtd||gd|IdH\}}}}||fS)Nrzpartial recvfrom_into)r7r8rr)rDr;r8rh got_nbytesrrrrrr rszFakeSocket.recvfrom_intoz4tuple[bytes, list[tuple[int, int, bytes]], int, Any])rrrhrc s>t|}||g||IdH\}}}}t|d||||fSr.) bytearrayrr3) rDrrrhr;rrrrrrr rszFakeSocket._recvmsgcCs tddS)Nz&can't get fileno() for FakeNet socketsrjrvrrr fileno0szFakeSocket.filenocCs tddS)Nzcan't detach() a FakeNet socketrjrvrrr detach3szFakeSocket.detachcCsdS)NFrrvrrr get_inheritable6szFakeSocket.get_inheritable) inheritablercCs|r tddS)Nz&FakeNet can't make inheritable socketsrj)rDrrrr set_inheritable9szFakeSocket.set_inheritableshare) process_idrcCs tddS)NzFakeNet can't share socketsrj)rDrrrr rAszFakeSocket.share)rrN)rr)N)N)r)r)rr)r)rr)rr)+rJrKrLrwpropertyrgr"rcrrrrrrrsysplatformrrrrr recvmsg_intorrr rrrrrrrrrrrrecvmsgrrrrrrrrr ras  '           ra) metaclass)4 __future__rrryrr0rrtypingrrrrrrr r\rZ trio._utilr r builtinsr r typesrZtyping_extensionsrrrrrrrMr!r)r,r2r<r=frozenr>rOrQabcZ SocketFactoryr^ZHostnameResolverrer_ SocketTyperarrrr  sJ $    #