YfYjdZddlZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZdd lmZejejejejhZd ZeZd Zd Zejd ZddddddZefdZ GddZ!Gdde!Z"Gdde!Z#ddZ$dS)zTransport implementation.N)contextmanager)SSLError)packunpack)UnexpectedFrame)KNOWN_TCP_OPTSSOL_TCP) set_cloexeci(isAMQP z\[([\.0-9a-f:]+)\](?::(\d+))?i< ) TCP_NODELAYTCP_USER_TIMEOUT TCP_KEEPIDLE TCP_KEEPINTVL TCP_KEEPCNTc8|}t|}|rM|d}|dr"t|d}n,d|vr(|dd\}}t|}||fS)z1Convert hostname:port string to host, port tuple.r:) IPV6_LITERALmatchgroupintrsplit)hostdefaultportms @/var/www/html/env/lib/python3.11/site-packages/amqp/transport.py to_host_portr!(s D4  Awwqzz 771:: #qwwqzz??D $;;S!,,JD$t99D :ceZdZdZ ddZdZdZdZedZ d Z d Z d Z d Z ddZdZdZdZdZefdZdZdS)_AbstractTransportaCommon superclass for TCP and SSL transports. PARAMETERS: host: str Broker address in format ``HOSTNAME:PORT``. connect_timeout: int Timeout of creating new connection. read_timeout: int sets ``SO_RCVTIMEO`` parameter of socket. write_timeout: int sets ``SO_SNDTIMEO`` parameter of socket. socket_settings: dict dictionary containing `optname` and ``optval`` passed to ``setsockopt(2)``. raise_on_initial_eintr: bool when True, ``socket.timeout`` is raised when exception is received during first read. See ``_read()`` for details. NTc d|_d|_||_t|_t |\|_|_||_||_ ||_ ||_ dSNF) connectedsockraise_on_initial_eintr EMPTY_BUFFER _read_bufferr!rrconnect_timeout read_timeout write_timeoutsocket_settings)selfrr,r-r.r/r)kwargss r __init__z_AbstractTransport.__init__Ws^ &<#(+D11 49.(*.r") connectionr(r)r+rrr,r-r.r/__dict__ __weakref__c |jr|jdd|jd}|jdd|jd}dt|jd|d|dt |dd Sdt|jd t |dd S) Nrrr z at z#x>z: (disconnected) at )r( getsockname getpeernametype__name__id)r0srcdsts r __repr__z_AbstractTransport.__repr__ts 9 OY**,,Q/NN$)2G2G2I2I!2LNNCY**,,Q/NN$)2G2G2I2I!2LNNCNtDzz*NNcNNsNN4NNNN NNtDzz*NN4NNNN Nr"cH |jrdS||j|j|j||j|j|jd|_dS#ttf$r0|j r'|js |j d|_ wxYw)NT) r'_connectrrr, _init_socketr/r-r.OSErrorrr(closer0s r connectz_AbstractTransport.connect|s ~  MM$)TY0D E E E   $d&79K    "DNNN"   y ! ! !!!    sA AA AB!c#FK| |jVdS|j}|}||kr|| |jVn#t$rN}dt |vrt jdt |vrt jd}~wt$r.}|jtj krt jd}~wwxYw ||kr||dSdS#||kr||wwxYw)N timed outzThe operation did not complete) r( gettimeout settimeoutrstrsockettimeoutrDerrno EWOULDBLOCK)r0rNr(prevexcs r having_timeoutz!_AbstractTransport.having_timeoutsI ?)OOOOO9D??$$Dw((( *i   #c((** .***5SAA .***   9 111 .***  d??OOD)))))#?7d??OOD))))#s7 AD CA B$$ C1)CCDD c:tj||tjtjt}t |D]\}}|\}}} } } tj||| |_ t|jdn#t$rYnwxYw|j ||j | dS#tj $rA|jr|j d|_|dzt|krYwxYwdS)NTr)rM getaddrinfo AF_UNSPEC SOCK_STREAMr enumerater(r NotImplementedErrorrKrGerrorrElen) r0rrrNentriesiresafsocktypeproto canonnamesas r rBz_AbstractTransport._connectsG$ $(&*>  40000*D $$W--- !!"%%%<   9&IOO%%%  q5CLL(()(    s7C+BC B C B7CA DDc |jd|jtjtjd||tj|ftj|ffD]`\}}|Yt|}t||z dz}|jtj|td||a| | tdS)Nri@Bll)r(rK setsockoptrM SOL_SOCKET SO_KEEPALIVE_set_socket_options SO_SNDTIMEO SO_RCVTIMEOrr_setup_transport_writeAMQP_PROTOCOL_HEADER)r0r/r-r.rNintervalsecusecs r rCz_AbstractTransport._init_sockets T""" V.0CQGGG   111$*#5}"E#)#5|"D"F   GX#(mmHsNg566 $$%wsD))  ()))))r"czi}tD]}d}|dkr ddlm}n<#t$rd}Yn.wxYwt t|rt t|}|r`|t vrt |||<jt t|r1|tt t|||<|S)Nrr)r) r rMr ImportErrorhasattrgetattrDEFAULT_SOCKET_SETTINGS getsockoptr )r0r(tcp_optsoptenums r _get_tcp_socket_defaultsz+_AbstractTransport._get_tcp_socket_defaultss! 7 7CD(((???????"DDD%% ,vs++ 7111%|j |n#t$rYnwxYw |jtjn#t$rYnwxYw |jn#t$rYnwxYwd|_d|_dSr&)r(rrDshutdownrM SHUT_RDWRrEr'rFs r rEz_AbstractTransport.closes 9  ((****       ""6#34444      !!!!    DIs/ ++$A A! A!%A?? B  B c|j}t} |dd}||z }|d|\}}}|tkrc|t} ||tz } n&#tjt t f$r||z }wxYwd|| g} n ||} || z }t|d} n#tj$r||j z|_ t t f$r} t| tj rGtj dkr7| jtjkr"||j z|_ tjt| t r3dt!| vr"||j z|_ tj| jt"vrd|_d } ~ wwxYw| d kr||| fSt'd | d d )aParse AMQP frame. Frame has following format:: 0 1 3 7 size+7 size+8 +------+---------+---------+ +-------------+ +-----------+ | type | channel | size | | payload | | frame-end | +------+---------+---------+ +-------------+ +-----------+ octet short long 'size' octets octet Tz>BHIr"rntrIFNzReceived frame_end z#04xz while expecting 0xce)rr*SIGNED_INT_MAXrMrNrDrjoinordr+ isinstancerZosnamerOrPrL_UNAVAILr'r) r0rreadread_frame_buffer frame_header frame_typechannelsizepart1part2payload frame_endrRs r read_framez_AbstractTransport.read_frames,z(, 44==L  - (.v|(D(D %Jn$$^,, D!677EE:&.%  ((E5>22$t**  ( DDGG II~    1D4E ED  "   3 -- '24'T//I!222 %68I$I!n&&&#x(( '[CHH-D-D%68I$I!n&&&y((!& ' *   w/ /!KiKKKKMM Ms1.226>B / / / / I OOO(**^*4.***&*444DDDD      DDD,!'@M@M@M@MDr"r$ceZdZdZdfd ZdZdZddZddZ dd Z d Z de j e j e jffd Zd ZxZS) SSLTransportaTransport that works over SSL. PARAMETERS: host: str Broker address in format ``HOSTNAME:PORT``. connect_timeout: int Timeout of creating new connection. ssl: bool|dict parameters of TLS subsystem. - when ``ssl`` is not dictionary, defaults of TLS are used - otherwise: - if ``ssl`` dictionary contains ``context`` key, :attr:`~SSLTransport._wrap_context` is used for wrapping socket. ``context`` is a dictionary passed to :attr:`~SSLTransport._wrap_context` as context parameter. All others items from ``ssl`` argument are passed as ``sslopts``. - if ``ssl`` dictionary does not contain ``context`` key, :attr:`~SSLTransport._wrap_socket_sni` is used for wrapping socket. All items in ``ssl`` argument are passed to :attr:`~SSLTransport._wrap_socket_sni` as parameters. kwargs: additional arguments of :class:`~amqp.transport._AbstractTransport` class Nc t|tr|ni|_t|_t j|fd|i|dS)Nr,)rdictssloptsr*r+superr2)r0rr,sslr1 __class__s r r2zSSLTransport.__init__sb(d33;ss (  = ="1 =5; = = = = =r")rc|j|jfi|j|_|j|j|j|jj|_dS)z!Wrap the socket in an SSL object.N) _wrap_socketr(rrKr, do_handshaker _quick_recvrFs r rlzSSLTransport._setup_transportsa%D%di@@4<@@  T1222    9>r"c @|r|j||fi|S|j|fi|Sr~) _wrap_context_wrap_socket_sni)r0r(contextrs r rzSSLTransport._wrap_socketsC  @%4%dG??w?? ?$t$T55W555r"c Ntjdi|}||_|j|fi|S)uWrap socket without SNI headers. PARAMETERS: sock: socket.socket Socket to be wrapped. sslopts: dict Parameters of :attr:`ssl.SSLContext.wrap_socket`. check_hostname Whether to match the peer cert’s hostname. See :attr:`ssl.SSLContext.check_hostname` for details. ctx_options Parameters of :attr:`ssl.create_default_context`. r)rcreate_default_contextcheck_hostname wrap_socket)r0r(rr ctx_optionsctxs r rzSSLTransport._wrap_contexts<*(77;77+st//w///r"FTc ||||| d} | |r tjn tj} tj| } || |||| || | |  tjo| du| _n#t$rYnwxYw||| _ |N| j tj kr9|rtj j ntj j}| || jdi| }|S)uSocket wrap with SNI headers. stdlib :attr:`ssl.SSLContext.wrap_socket` method augmented with support for setting the server_hostname field required for SNI hostname header. PARAMETERS: sock: socket.socket Socket to be wrapped. keyfile: str Path to the private key certfile: str Path to the certificate server_side: bool Identifies whether server-side or client-side behavior is desired from this socket. See :attr:`~ssl.SSLContext.wrap_socket` for details. cert_reqs: ssl.VerifyMode When set to other than :attr:`ssl.CERT_NONE`, peers certificate is checked. Possible values are :attr:`ssl.CERT_NONE`, :attr:`ssl.CERT_OPTIONAL` and :attr:`ssl.CERT_REQUIRED`. ca_certs: str Path to “certification authority” (CA) certificates used to validate other peers’ certificates when ``cert_reqs`` is other than :attr:`ssl.CERT_NONE`. do_handshake_on_connect: bool Specifies whether to do the SSL handshake automatically. See :attr:`~ssl.SSLContext.wrap_socket` for details. suppress_ragged_eofs (bool): See :attr:`~ssl.SSLContext.wrap_socket` for details. server_hostname: str Specifies the hostname of the service which we are connecting to. See :attr:`~ssl.SSLContext.wrap_socket` for details. ciphers: str Available ciphers for sockets created with this context. See :attr:`ssl.SSLContext.set_ciphers` ssl_version: Protocol of the SSL Context. The value is one of ``ssl.PROTOCOL_*`` constants. )r( server_sidedo_handshake_on_connectsuppress_ragged_eofsserver_hostnameNr)rPROTOCOL_TLS_SERVERPROTOCOL_TLS_CLIENT SSLContextload_cert_chainload_verify_locations set_ciphersHAS_SNIrAttributeError verify_mode CERT_NONEPurpose CLIENT_AUTH SERVER_AUTHload_default_certsr)r0r(keyfilecertfiler cert_reqsca_certsrrrciphers ssl_versionoptsrpurposes r rzSSLTransport._wrap_socket_snis`H&'>$8.     -'',  .--    # #Hg 6 6 6    ) )( 3 3 3      ( ( (  ;t ;  " "    D   "+G    3s} D D- ''[,   & &w / / /"w"**T** sB B#"B#cT|j |j|_dSdS)z/Unwrap a SSL socket, so we can call shutdown().N)r(unwraprFs r rz SSLTransport._shutdown_transport/s+ 9  ((**DIII ! r"c|j}|j} t||kr ||t|z }n=#t$r0}|j|vr!|r|jrt jYd}~dd}~wwxYw|std||z }t||kn #||_xYw|d|||dc}|_|S)N%Server unexpectedly closed connectionrr+r[rDrOr)rMrN r0rr_errnosrecvrbufrrRresults r rzSSLTransport._read4s   d))a-- QT]++AAyG++"3t'B3"(."2"22 K!"IJJJ d))a--  $D  $(!Hd122h!! 9B'AB' A: %A5/B'4A55A::,B'' B1c|jj}|r> ||}n#t$rd}YnwxYw|std||d}|r"Fc|j}|j} t||kr ||t|z }n=#t$r0}|j|vr!|r|jrt jYd}~dd}~wwxYw|std||z }t||kn #||_xYw|d|||dc}|_|S)z%Read exactly n bytes from the socket.Nrrrs r rzTCPTransport._readns   d))a--QT]++AAyG++"3t'B3"(."2"22   K!"IJJJ d))a--  $D  $(!Hd122h!! rN) r<rrrrlrOrrrrr"r rrasQ *** %u|U[.Ir"rFc :|rtnt}||f||d|S)aCreate transport. Given a few parameters from the Connection constructor, select and create a subclass of :class:`~amqp.transport._AbstractTransport`. PARAMETERS: host: str Broker address in format ``HOSTNAME:PORT``. connect_timeout: int Timeout of creating new connection. ssl: bool|dict If set, :class:`~amqp.transport.SSLTransport` is used and ``ssl`` parameter is passed to it. Otherwise :class:`~amqp.transport.TCPTransport` is used. kwargs: additional arguments of :class:`~amqp.transport._AbstractTransport` class )r,r)rr)rr,rr1 transports r Transportrs18!$5 I 9T N? N Nv N NNr"r&)%rrOrrerMr contextlibrrstructrr exceptionsrplatformr r utilsr rrrrPr AMQP_PORTbytesr*rrncompilerrwr!r$rrrrr"r rs %%%%%%''''''-------- L%+u|U5F G  uww /rz:;;   )    jjjjjjjjZ zzzzz%zzzz#####%###LOOOOOOr"