o 4@HhG@sUddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z ddlZddlZddlmZmZe rgddlZddlmZddlmZmZddlmZddlmZmZmZdd lmZe ej ej!fZ"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-j0Z1eGd*d+d+Z2eGd,d-d-ejj3ed.Z4dS)3) annotationsN) TYPE_CHECKINGAnyNoReturnUnionoverload)NoPublicConstructorfinal)Iterable) AddressFamily SocketKind) TracebackType)BufferSelf TypeAlias) AddressFormatr IPAddressipreturnintcCs0t|tjr tjjSt|tjrtjjStd)Nz!Unhandled IPAddress instance type) isinstance ipaddress IPv4AddresstriosocketAF_INET IPv6AddressAF_INET6NotImplementedError)rrV/home/air/sanwanet/gpt-api/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_for3    r*cCr#)N 127.0.0.1::1r&r'r)rrr _localhost_ip_for<r+r.codercCst|t|N)OSErrorosstrerror)r/rrr _fake_errDsr4databytesbuffersIterable[Buffer]c Cs~d}|D]8}|||t|j}t|}||dt|<Wdn1s(wY|t|7}|t|kr<|Sq|S)Nr) memoryviewnbyteslen)r5r7writtenbuf next_piecembufrrr _scatterHs   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 compressedrBrrr)selfsockaddrrrr as_python_sockaddrYs zUDPEndpoint.as_python_sockaddrrGcCs"|dd\}}|t||dS)N)rrB)rr()clsrGrrBrrr from_python_sockaddrbsz UDPEndpoint.from_python_sockaddrNrrC)rGrCrrA)__name__ __module__ __qualname____annotations__rH classmethodrKrrrr rATs   rAc@seZdZUded<dS) UDPBindingrAlocalN)rMrNrOrPrrrr rRks rRc@s@eZdZUded<ded<ejdddZded<d d d Zd S) UDPPacketrAsource destinationcCs|Sr0)hex)prrr uszUDPPacket.)reprr6payloadrcCst|j|j|dS)NrUrVr[)rTrVrU)rFr[rrr replyxs zUDPPacket.replyN)r[r6rrT)rMrNrOrPattrsfieldr[r]rrrr rTqs rTc@s eZdZUded<d d d Zd S) FakeSocketFactoryFakeNetfake_netr"rtype_protor FakeSocketcCst|j|||Sr0)re_createrb)rFr"rcrdrrr rszFakeSocketFactory.socketN)r"rrcrrdrrre)rMrNrOrPrrrrr r`s r`c@s4eZdZUded<    ddddZdddZdS)FakeHostnameResolverrarbrhost bytes | NonerBbytes | str | int | Noner"rtyperdflagsrqlist[tuple[AddressFamily, SocketKind, int, str, tuple[str, int] | tuple[str, int, int, int] | tuple[int, bytes]]]c tdNzFakeNet doesn't do fake DNS yetr)rFrhrBr"rkrdrlrrr getaddrinfosz FakeHostnameResolver.getaddrinforGrCtuple[str, str]crnrorp)rFrGrlrrr getnameinfosz FakeHostnameResolver.getnameinfoN)rrrr)rhrirBrjr"rrkrrdrrlrrrm)rGrCrlrrrr)rMrNrOrPrqrsrrrr rgs  rgc@s>eZdZdddZdd d Zdd d ZdddZdddZdS)rarNonecCs@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_packetrFrrr __init__s  zFakeNet.__init__bindingrRrrecCs"||jvr ttj||j|<dSr0)r}r4errno EADDRINUSE)rFrrrrr _binds  z FakeNet._bindcCs$tjt|tjt|dSr0)rrset_custom_socket_factoryr`set_custom_hostname_resolverrgrrrr enableszFakeNet.enablepacketrTcCs&|jdur ||dS||dSr0)r~deliver_packetrFrrrr send_packets zFakeNet.send_packetcCs0t|jd}||jvr|j||dS dS)NrS)rRrVr}_deliver_packet)rFrrrrr rs  zFakeNet.deliver_packetNrrt)rrRrrerrtrrTrrt)rMrNrOrrrrrrrrr ras    rac@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)rerbrar"r rkr rdrrrtcCs||_|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_channelrTfloat_packet_sender_packet_receiver_binding)rFrbr"rkrdrrr rs&  zFakeSocket.__init__cC|jSr0)rrrrr rkzFakeSocket.typecCrr0)rrrrr r"rzFakeSocket.familycCrr0)rrrrr rdrzFakeSocket.protocCs|jr ttjdSdSr0)rr4rEBADFrrrr _check_closedszFakeSocket._check_closedcCs4|jrdSd|_|jdur|jj|j=|jdS)NT)rrrr}rcloserrrr rs   zFakeSocket.closeaddressobjectrSbooltuple[str, int]cs&tjj|j|j|j|d|dIdHS)NF)r ipv6_v6onlyrS)r_socket_resolve_address_nocprkr"rd)rFrrSrrr rs z FakeSocket._resolve_address_nocprrTcCs>ttj|j|WddS1swYdSr0) contextlibsuppressrBrokenResourceErrorr send_nowaitrrrr rs"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)rrr4rEINVALrlowlevel checkpointrrr(r!r"nextrr|rRrAr)rFrip_strrB_rrrrr binds$       zFakeSocket.bindpeerrcrn)Nz0FakeNet does not (yet) support connected socketsrp)rFrrrr connect2szFakeSocket.connectrrNr7r8ancdata!Iterable[tuple[int, int, Buffer]]rlAddressFormat | 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\)rrrrrrr4rENOTCONNrArKrrr*r"rEjoinrTrSrrr;)rFr7rrlrrVr[rrrr _sendmsg5s.     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.) rrrrreceiverUrHr@r[r;rr MSG_TRUNC) rFr7rrlr msg_flagsrrr<rrr _recvmsg_intoas$      zFakeSocket._recvmsg_intorCcCsF||jdur|jjS|jtjjkrdS|jtjjks!JdS)N)r$r)r%r) rrrSrHr"rrrrrrrr 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) rrhasattrrrrArHr4rrrrrr getpeernames"   zFakeSocket.getpeernameleveloptnamecCdSr0r)rFrrrrr getsockoptzFakeSocket.getsockoptbuflenr6cCrr0rrFrrrrrr rr int | None int | bytescCs|td|d|d)Nz%FakeNet doesn't implement getsockopt(, ))rr1rrrr rsvalue int | BuffercCrr0r)rFrrrrrr setsockoptrzFakeSocket.setsockoptoptlencCrr0rrFrrrrrrr rsint | 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|Sr0rrrrr __enter__rzFakeSocket.__enter__exc_type#builtins.type[BaseException] | None exc_valueBaseException | None tracebackTracebackType | NonecCs |dSr0)r)rFrrrrrr __exit__s zFakeSocket.__exit__r5rcs|||dIdHSr0)sendto)rFr5rlrrr sendszFakeSocket.send_FakeSocket__data_FakeSocket__address!tuple[object, ...] | str | BuffercdSr0r)rFrrrrr rszFakeSocket.sendto_FakeSocket__flags(tuple[object, ...] | str | Buffer | Nonecrr0r)rFrrrrrr rsargsrcsTt|dkr|\}}d}nt|dkr|\}}}ntd||gg||IdHS)NrIrzwrong number of arguments)r; TypeErrorr)rFrr5rrlrrr rs   bufsizecs|||IdH\}}|Sr0)recvfrom)rFrrlr5_addressrrr recv szFakeSocket.recvr=r:cs||||IdH\}}|Sr0) recvfrom_into)rFr=r:rl got_bytesrrrr recv_intoszFakeSocket.recv_intotuple[bytes, AddressFormat]cs$|||IdH\}}}}||fSr0)_recvmsg)rFrrlr5_ancdata _msg_flagsrrrr rszFakeSocket.recvfromtuple[int, AddressFormat]csF|dkr|t|jkrtd||gd|IdH\}}}}||fS)Nrzpartial recvfrom_into)r9r:rr)rFr=r:rl got_nbytesrrrrrr rszFakeSocket.recvfrom_into>tuple[bytes, list[tuple[int, int, bytes]], int, AddressFormat]c s@t|}||g||IdH\}}}}t|d||||fSr0) bytearrayrr6) rFrrrlr=rrrrrrr r)szFakeSocket._recvmsgcCtd)Nz&can't get fileno() for FakeNet socketsrprrrr fileno<zFakeSocket.filenocCr)Nzcan't detach() a FakeNet socketrprrrr detach?rzFakeSocket.detachcCsdS)NFrrrrr get_inheritableBrzFakeSocket.get_inheritable inheritablecCs|rtddS)Nz&FakeNet can't make inheritable socketsrp)rFrrrr set_inheritableEszFakeSocket.set_inheritableshare process_idcCr)NzFakeNet can't share socketsrp)rFrrrr rMrzFakeSocket.share) rbrar"r rkr rdrrrt)rr )rr )rrr)rrrSrrrr)rrrrt)rrrr)rrN) r7r8rrrlrrrrrrD)r7r8rrrlrrrrL)rrrrrr)rrrrrrrr6r0)rrrrrrrr)rrrrrrrrt) rrrrrrtrrrrt) rrrrrrrrrrt)rr)rrrrrrrrt)r)r5rrlrrr)rrrrrr)rrrrrrrr)rrrr)rrrlrrr6)r=rr:rrlrrr)rrrlrrr)r=rr:rrlrrr)rrrrrlrrr)rr)rrrrt)rrrr6)+rMrNrOrpropertyrkr"rdrrrrrrrsysplatformrrrrr recvmsg_intorrrrrrrrrrrrrrrecvmsgrrrrrrrrr res          '  $                           re) metaclass)rrrr)r"rrr)r/rrr)r5r6r7r8rr)5 __future__rrrrr2rr typingrrrrrr^r trio._utilrr builtinscollections.abcr r r typesr typing_extensionsrrr trio._socketrrrrrPr!r*r.r4r@frozenrArRrTabc SocketFactoryr`HostnameResolverrgra SocketTypererrrr sL        #