U
|
Z ±di1 ã @ sZ d dl Z d dlZd dlZd dlmZ d dlZd dlZd dlmZm Z m
|
Z
|
d dl Z d dl m Z d dlmZmZ eejejf ZeedddZeed d
|
d Zeed d d Zdd Zdd ZejG dd dZejG dd dZ ejG dd dZ ejG dd de j j Z ejG dd de j j!Z"G d d d ed Z#G d d d e j$j%ed Z&dS )!é N)Údeque)ÚUnionÚListÚOptional)Úcontextmanager)ÚFinalÚNoPublicConstructor)ÚipÚreturnc C s4 t | tjrtjjS t | tjr(tjjS ds0td S ©NF) Ú
|
isinstanceÚ ipaddressÚ IPv4AddressÚtrioÚsocketÚAF_INETÚ IPv6AddressÚAF_INET6ÚAssertionError)r © r úMd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/testing/_fake_net.pyÚ _family_for s
|
r )Úfamilyr
|
c C s8 | t jjkrt d¡S | t jjkr,t d¡S ds4td S )Nú0.0.0.0ú::F©r r r r Ú
|
ip_addressr r ©r r r r Ú_wildcard_ip_for s
|
|
|
r c C s8 | t jjkrt d¡S | t jjkr,t d¡S ds4td S )Nú 127.0.0.1ú::1Fr r r r r Ú_localhost_ip_for) s
|
|
|
r! c C s t | t | ¡d S ©N)ÚOSErrorÚosÚstrerror)Úcoder r r Ú _fake_err2 s r' c
|
C sf d}|D ]X}| ||t |
}t|}||d t |
< W 5 Q R X |t |7 }|t | kr qbq|S )Nr )ÚlenÚ
|
memoryview)ÚdataÚbuffersÚwrittenÚbufZ
|
next_pieceZmbufr r r Ú_scatter6 s
|
r. c @ s2 e Zd ZU eed< eed< dd Zedd ZdS )Ú UDPEndpointr Úportc C s( | j j| jf}t| j tjr$|d7 }|S )N)r r )r Ú
|
compressedr0 r r r )ÚselfÚsockaddrr r r Úas_python_sockaddrG s z UDPEndpoint.as_python_sockaddrc C s" |d d
\}}| t |¡|dS )Né )r r0 )r r )Úclsr3 r r0 r r r Úfrom_python_sockaddrM s z UDPEndpoint.from_python_sockaddrN) Ú__name__Ú
|
__module__Ú __qualname__Ú IPAddressÚ__annotations__Úintr4 Ú classmethodr7 r r r r r/ B s
|
|
r/ c @ s e Zd ZU eed< dS )Ú
|
UDPBindingÚlocalN)r8 r9 r: r/ r< r r r r r? S s
|
r? c @ s> e Zd ZU eed< eed< ejdd dZeed< dd Z d S )
|
Ú UDPPacketÚsourceÚ destinationc C s | ¡ S r" )Úhex)Úpr r r Ú<lambda>\ ó zUDPPacket.<lambda>)ÚreprÚpayloadc C s t | j| j|dS )N©rB rC rI )rA rC rB )r2 rI r r r Úreply^ s
|
ÿzUDPPacket.replyN)
|
r8 r9 r: r/ r< ÚattrÚibrI ÚbytesrK r r r r rA X s
|
rA c @ s* e Zd ZU ded< eeeddddZdS )ÚFakeSocketFactoryÚFakeNetÚfake_netÚ
|
FakeSocket)r ÚtypeÚprotor
|
c C s t | j|||¡S r" )rR Ú_createrQ )r2 r rS rT r r r r h s zFakeSocketFactory.socketN)r8 r9 r: r< r= r r r r r rO d s
|
rO c @ s> e Zd ZU ded< d eeeef dddZeddd Zd
|
S ) ÚFakeHostnameResolverrP rQ r )Úhostr0 c à s t dd S ©Nz FakeNet doesn't do fake DNS yet©ÚNotImplementedError)r2 rW r0 r rS rT Úflagsr r r Ú getaddrinfop s z FakeHostnameResolver.getaddrinfo)r[ c à s t dd S rX rY )r2 r3 r[ r r r Ú getnameinfou s z FakeHostnameResolver.getnameinfoN)r r r r ) r8 r9 r: r< Ústrr r= r\ r] r r r r rV l s
|
ÿ
|
ÿ rV c @ sP e Zd Zdd ZedddddZddd d
|
Zddd d Zddd dZdS )rP c C s@ t d¡ ¡ | _t d¡ ¡ | _ttdd| _i | _d | _ d S )Nz 1.0.0.0/8z1::/16iPÃ iÿÿ )
|
r Ú IPv4NetworkÚhostsZ_auto_ipv4_iterÚ IPv6NetworkÚiterÚrangeÚ_auto_port_iterÚ_boundÚ route_packet©r2 r r r Ú__init__z s
|
zFakeNet.__init__rR N)Úbindingr r
|
c C s" || j krttj || j |< d S r" )re r' ÚerrnoZ
|
EADDRINUSE)r2 ri r r r r Ú_bind s
|
|
z FakeNet._bind)r
|
c C s$ t j t| ¡ t j t| ¡ d S r" )r r Zset_custom_socket_factoryrO Z set_custom_hostname_resolverrV rg r r r Úenable s zFakeNet.enablec C s$ | j d kr| |¡ n
|
| |¡ d S r" )rf Údeliver_packet©r2 Úpacketr r r Ú send_packet s
|
zFakeNet.send_packetc C s, t |jd}|| jkr(| j| |¡ n d S )N©r@ )r? rC re Ú_deliver_packet)r2 ro ri r r r rm s
|
zFakeNet.deliver_packet) r8 r9 r: rh r? rk rl rp rm r r r r rP y s
|
|
rP )Ú metaclassc @ s e Zd ZeeeedddZdd Zdd Zdd Ze d
|
d d Z
|
d d Z dd Z dd Z d;ddZdd Zdd Zdd Zd d Zd d Zd d! Zd<d"d#Zd$d% Zd=d&d'Zd>d(d)Zd?d*d+Zd@d,d-ZdAd.d/Zd0d1 Z d2d3 Z d4d5 Z d6d7 Z d8d9 Z d:S )BrR )rQ r rS rT c C s || _ |stjj}|s tjj}|tjjtjjfkr@td| |tjjkrZtd| || _|| _ || _
|
d| _ t t d¡\| _| _d | _d S )Nz%FakeNet doesn't (yet) support family=z#FakeNet doesn't (yet) support type=FÚinf)Ú _fake_netr r r Ú SOCK_STREAMr rZ Ú
|
SOCK_DGRAMr rS rT Ú_closedZopen_memory_channelÚfloatÚ_packet_senderÚ_packet_receiverÚ_binding)r2 rQ r rS rT r r r rh s" ÿ zFakeSocket.__init__c C s | j rttj d S r" )rx r' rj ÚEBADFrg r r r Ú _check_closed· s zFakeSocket._check_closedc C s4 | j r
|
d S d| _ | jd k r&| jj| j= | j ¡ d S )NT)rx r| ru re r{ Úcloserg r r r r » s
|
zFakeSocket.closec à s$ t jj| j| j| j|d|dI d H S )NF)ÚaddressZ ipv6_v6onlyr@ )r Ú_socketÚ_resolve_address_nocprS r rT )r2 r r@ r r r r Ä s úz FakeSocket._resolve_address_nocp)ro c C s, z| j |¡ W n tjk
|
r& Y nX d S r" )rz Z send_nowaitr ZBrokenResourceErrorrn r r r rr Î s zFakeSocket._deliver_packetc à sÎ | ¡ | jd k r ttj tj ¡ I d H | j|ddI d H \}}t
|
|¡}t || j ks`t |t
|
d¡krzt
|
d¡}n|t
|
d¡krt
|
d¡}|dkr¦t| jj}tt||d}| j || ¡ || _d S )NTrq r r r r r )r~ r| Z _fake_errorrj ÚEINVALr ÚlowlevelÚ
|
checkpointr r r r r r Únextru rd r? r/ rk )r2 ÚaddrÚip_strr0 r ri r r r ÚbindÙ s
|
|
|
|
zFakeSocket.bindc à s t dd S )Nz0FakeNet does not (yet) support connected socketsrY )r2 Zpeerr r r Úconnectì s zFakeSocket.connectc Ç s6 | ¡ g }d}d }t|dkr(|\}nPt|dkr>|\}}n:t|dkrV|\}}}n"t|dkrp|\}}}}ntdtj ¡ I d H |d k r¤| j|ddI d H }|r°td |rÂtd
|
| |d krÔtt j
|
t |¡}| j d kr| t| jjdf¡I d H d |¡}t| j j||d }| j |¡ t|S ) Nr é r5 é é úwrong number of argumentsFrq ú&FakeNet doesn't support ancillary dataz"FakeNet send flags must be 0, not rG rJ )r~ r( Ú TypeErrorr r r
r rZ r' rj ÚENOTCONNr/ r7 r| r r r r1 ÚjoinrA r@ ru rp ) r2 ÚargsÚancdatar[ r r+ rC rI ro r r r Úsendmsgï sB
|
|
|
|
ý zFakeSocket.sendmsgr c à s| |dkrt d|dkr t d| ¡ g }d}| j ¡ I d H }|j ¡ }t|j|}|t|jk rp|t j
|
j O }||||fS )Nr r z&FakeNet doesn't support any recv flags) rZ r~ r{ ZreceiverB r4 r. rI r( r r Ú MSG_TRUNC) r2 r+ Ú
|
ancbufsizer[ r Ú msg_flagsro r r, r r r Ú recvmsg_into s
|
zFakeSocket.recvmsg_intoc C sJ | ¡ | jd k r | jj ¡ S | jtjjkr0dS | jtjjksBt dS d S )N)r r )r r )
|
r~ r| r@ r4 r r r r r r rg r r r Ú getsockname2 s
|
zFakeSocket.getsocknamec C s8 | ¡ | jd k r*| jjd k r*| jj ¡ S ttj d S r" )r~ r| Úremoter4 r' rj r rg r r r Ú getpeername< s
|
|
zFakeSocket.getpeernamec C s" | ¡ td| d| dd S )Nz%FakeNet doesn't implement getsockopt(ú, ú))r~ r# )r2 ÚlevelÚitemr r r Ú
|
getsockoptC s zFakeSocket.getsockoptc C sF | ¡ ||ftjjtjjfkr,|s,tdtd| d| dd S )Nz#FakeNet always has IPV6_V6ONLY=Truez%FakeNet doesn't implement setsockopt(r z, ...))r~ r r Ú IPPROTO_IPV6Ú IPV6_V6ONLYrZ r# )r2 r r Úvaluer r r Ú
|
setsockoptG s
|
zFakeSocket.setsockoptc C s | S r" r rg r r r Ú __enter__T s zFakeSocket.__enter__c G s | ¡ d S r" )r )r2 Úexc_infor r r Ú__exit__W s zFakeSocket.__exit__c à s | ||d ¡I d H S r" )Úsendto)r2 r* r[ r r r ÚsendZ s zFakeSocket.sendc Ç sR t |dkr|\}}d}n t |dkr2|\}}}ntd| |gg ||¡I d H S )Nr5 r r r )r( r r )r2 r r* r r[ r r r r© ] s zFakeSocket.sendtoc à s | ||¡I d H \}}|S r" )Úrecvfrom)r2 Úbufsizer[ r* r r r r Úrecvg s zFakeSocket.recvc à s | |||¡I d H \}}|S r" )Ú recvfrom_into)r2 r- Únbytesr[ Z got_bytesr r r r Ú recv_intok s zFakeSocket.recv_intoc à s" | ||¡I d H \}}}}||fS r" )Úrecvmsg)r2 r¬ r[ r* r r r r r r r« o s zFakeSocket.recvfromc à sB |dkr |t |kr td| |gd|¡I d H \}}}}||fS )Nr zpartial recvfrom_into)r( rZ r )r2 r- r¯ r[ Ú
|
got_nbytesr r r r r r r® s s ÿzFakeSocket.recvfrom_intoc à s> t |}| |g||¡I d H \}}}}t|d |
|||fS r" )Ú bytearrayr rN ) r2 r¬ r r[ r- r² r r r r r r r± { s ÿzFakeSocket.recvmsgc C s t dd S )Nz&can't get fileno() for FakeNet socketsrY rg r r r Úfileno s zFakeSocket.filenoc C s t dd S )Nz can't detach() a FakeNet socketrY rg r r r Údetach
s zFakeSocket.detachc C s dS r r rg r r r Úget_inheritable s zFakeSocket.get_inheritablec C s |r t dd S )Nz&FakeNet can't make inheritable socketsrY )r2 Ú inheritabler r r Úset_inheritable s zFakeSocket.set_inheritablec C s t dd S )NzFakeNet can't share socketsrY )r2 Z
|
process_idr r r Úshare s zFakeSocket.shareN)r r )r )r )r r )r )r r )r r )!r8 r9 r: rP r= rh r~ r r rA rr r r r r r r r¡ r¥ r¦ r¨ rª r© r r° r« r® r± r´ rµ r¶ r¸ r¹ r r r r rR s6
|
-
|
|
|
|
|
|
|
|
|
rR )'r rL r Ú collectionsr rj r$ Útypingr r r ÚenumÚ
|
contextlibr Z
|
trio._utilr r r r r; r= r r r! r' r. Úfrozenr/ r? rA ÚabcZ SocketFactoryrO ZHostnameResolverrV rP r Ú
|
SocketTyperR r r r r Ú<module> s6 #
|
