QjZ|dZddlZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZdd lmZej&ej(ej*ej,hZd ZeZd Zd Zej:d ZddddddZefdZ GddZ!Gdde!Z"Gdde!Z#ddZ$y)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_KEEPCNTc|}tj|}|r@|jd}|jdrt|jd}||fSd|vr |j dd\}}t|}||fS)z1Convert hostname:port string to host, port tuple.r:) IPV6_LITERALmatchgroupintrsplit)hostdefaultportms J/var/www/html/truck-me/venv/lib/python3.12/site-packages/amqp/transport.py to_host_portr!(s D4 Awwqz 771:qwwqz?D : $;S!,JD$t9D :ceZdZdZ ddZdZdZdZedZ dZ d Z d Z d Z dd Zd ZdZdZdZefdZdZy)_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. Nc d|_d|_||_t|_t |\|_|_||_||_ ||_ ||_ yNF) 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__WsU &<#(+D1 49.(*.r") connectionr(r)r+rrr,r-r.r/__dict__ __weakref__c |jr|jjdd|jjd} |jjdd|jjd}dt |j d|d|dt|d d Sdt |j d t|d d S#tj$r}d|}Yd}~od}~wwxYw) NrrrzERROR:  z at z#x>z: (disconnected) at )r( getsockname getpeernamesocketerrortype__name__id)r0srcdstes r __repr__z_AbstractTransport.__repr__ts 99YY**,Q/0$))2G2G2I!2L1MNC $..034Adii6K6K6Ma6P5QRtDz**+2cU$se44}AN NtDz**++?4}AN N LL $sm $s =CC<-C77C<c |jry|j|j|j|j|j |j |j|jd|_y#ttf$r;|jr-|js!|jjd|_ wxYw)NT) r'_connectrrr, _init_socketr/r-r.OSErrorrr(closer0s r connectz_AbstractTransport.connects ~~ MM$))TYY0D0D E   $$d&7&79K9K  "DN" yy !    s A9A)A99A Cc#<K||jy|j}|j}||k7r|j| |j ||k7r|j|yy#t$rH}dt |vrt j dt |vrt j d}~wt$r7}|jtjk(rt j d}~wwxYw#||k7r|j|wwxYww)N timed outzThe operation did not complete) r( gettimeout settimeoutrstrr;timeoutrGerrno EWOULDBLOCK)r0rPr(prevexcs r having_timeoutz!_AbstractTransport.having_timeouts ?))O99D??$Dw( *iid?OOD)# #c(* ..**5SA ..** 99 1 11 ..**  d?OOD)#sCADA.D. C=7AB:: C=2C88C==DDDcPtj||tjtjt}t |D]x\}}|\}}} } } tj||| |_ t|j d|j j||j j| yy#t$rYEwxYw#tj$rB|j r|j jd|_|dzt|k\rYwxYw)NTr)r; getaddrinfo AF_UNSPEC SOCK_STREAMr enumerater(r NotImplementedErrorrNrJr<rHlen) r0rrrPentriesiresafsocktypeproto canonnamesas r rEz_AbstractTransport._connects$$ $((&*<*  40 $$W- !!"%#  +<< 99IIOO%  q5CL()  s7C1C6C C  C C  CAD%$D%c |jjd|jjtjtj d|j |tj|ftj|ffD]Z\}}| t|}t||z dz}|jjtj|td||\|j|jty)Nri@Bll)r(rN setsockoptr; SOL_SOCKET SO_KEEPALIVE_set_socket_options SO_SNDTIMEO SO_RCVTIMEOrr_setup_transport_writeAMQP_PROTOCOL_HEADER)r0r/r-r.rPintervalsecusecs r rFz_AbstractTransport._init_sockets T" V..0C0CQG   1$*#5#5}"E#)#5#5|"D"F  GX#(mHsNg56 $$%%wsD)    ()r"c@i}tD]}d}|dk(r ddlm}n t t|rt t|}|s5|t vr t |||<Jt t|s[|jtt t|||<|S#t$rd}YawxYw)Nrr)r) r r;r ImportErrorhasattrgetattrDEFAULT_SOCKET_SETTINGS getsockoptr )r0r(tcp_optsoptenums r _get_tcp_socket_defaultsz+_AbstractTransport._get_tcp_socket_defaultss! 7CD((?%vs+11%|j] |j |jjtj  |jj d|_d|_y#t$rY_wxYw#t$rYDwxYw#t$rY8wxYwr&)r(rrGshutdownr; SHUT_RDWRrHr'rIs r rHz_AbstractTransport.closes 99  ((*  ""6#3#34  !DI!       s4A2)B B2 A>=A> B  B  BBc|j}t} |dd}||z }|d|\}}}|tkDr0|t} ||tz } dj|| g} n||} || z }t|d} | d k(r||| fSt'd | d d #tjt t f$r||z }wxYw#tj$r||jz|_ t t f$r} t| tjrXtjdk(rE| jtjk(r(||jz|_ tjt| t r5dt!| vr(||jz|_ tj| jt"vrd|_d } ~ wwxYw)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"rntrLFNzReceived frame_end z#04xz while expecting 0xce)rr*SIGNED_INT_MAXr;rPrGrjoinordr+ isinstancer<osnamerQrRrO_UNAVAILr'r) r0rreadread_frame_buffer frame_header frame_typechannelsizepart1part2payload frame_endrTs r read_framez_AbstractTransport.read_frameszz(, 4=L  - (.v|(D %Jn$^, !67E((E5>2t*  ( DG I2  w/ /!%i%55JKM MM:&.% ~~  1D4E4E ED  " 3 -"''T/II!2!22 %68I8I$I!nn&&#x([CH-D%68I8I$I!nn&&yy(!& ' s00CB 2C %CC6G>CGGc |j|y#tj$rt$r}|jt vrd|_d}~wwxYwr&)rnr;rPrGrQrr')r0rrTs r writez_AbstractTransport.write\sG  KKN~~   yy(!&  sAA  A)NNNNT)F)r> __module__ __qualname____doc__r2 __slots__rCrJrrUrErFr}rjrrmrrnrHrrrrr"r r$r$7s~>.226>B / I O(**4.*&*4D  D,!'@MDr"r$ceZdZdZd fd ZdZdZd dZd dZ ddZ d Z de je je 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 c rt|tr|ni|_t|_t ||fd|i|y)Nr,)rdictssloptsr*r+superr2)r0rr,sslr1 __class__s r r2zSSLTransport.__init__s=(d3s (   ="1 =5; =r")rc|j|jfi|j|_|jj|j|jj |jj |_y)z!Wrap the socket in an SSL object.N) _wrap_socketr(rrNr, do_handshaker _quick_recvrIs r rmzSSLTransport._setup_transports[%D%%dii@4<<@  T112  99>>r"c T|r|j||fi|S|j|fi|Sr) _wrap_context_wrap_socket_sni)r0r(contextrs r rzSSLTransport._wrap_sockets7 %4%%dG?w? ?$t$$T5W55r"c `tjdi|}||_|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_contexts5*((7;7+st/w//r"Fc T||||| d} | "|rtjntj} tj| } || j |||| j || | j |  tjxr| du| _||| _ |d| jtjk7rG|rtjjntjj}| j|| j di| }|S#t$rYwxYw)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_snis8H&'>$8.    '',,  ..-    # #Hg 6    ) )( 3      (  ;t ;  "  "+G    3 3s}} D ''[[,,   & &w /"w""*T* #   s=D D'&D'c\|j |jj|_yy)z/Unwrap a SSL socket, so we can call shutdown().N)r(unwraprIs r rz SSLTransport._shutdown_transport2s% 99  ((*DI !r"c|j}|j} t||kr6 ||t|z }|s td||z }t||kr6|d|||dc}|_|S#t$r;}|j|vr'|r |j rt jYd}~d}~wwxYw#||_xYw)N%Server unexpectedly closed connectionrr+r\rGrQr)r;rP r0rr_errnosrecvrbufrrTresults r rzSSLTransport._read7s    d)a- QT]+A!"IJJ d)a-"%)!Hd12h!! yyG+"t'B'B"(.."22   $D  9B7A0 B70 B490B/)B7.B//B44B77 Cc|jj}|r ||}|s td||d}|ryy#t$rd}Y$wxYw)z+Write a string out to the SSL socket fully.rz Socket closedN)r(r ValueErrorrG)r0rrrs r rnzSSLTransport._writeSsX  !Ho..!"A    s9 AA)NNr) NNFNNFTNNN)r>rrrr2rrmrrrrrQENOENTEAGAINEINTRrrn __classcell__)rs@r rrgsg D=  I*6 02=A6:@EDH37 vp+ %||U\\5;;?8r"rcNeZdZdZdZdej ejffdZy) TCPTransportz~Transport that deals directly with TCP socket. All parameters are :class:`~amqp.transport._AbstractTransport` class. c|jj|_t|_|jj |_yr)r(sendallrnr*r+rrrIs r rmzTCPTransport._setup_transportjs-ii'' (99>>r"Fc|j}|j} t||kr6 ||t|z }|s td||z }t||kr6|d|||dc}|_|S#t$r;}|j|vr'|r |j rt jYd}~d}~wwxYw#||_xYw)z%Read exactly n bytes from the socket.Nrrrs r rzTCPTransport._readqs   d)a-QT]+A!"IJJ d)a- %)!Hd12h!! yyG+"t'B'B"(.."22    $D  rN) r>rrrrmrQrrrrr"r rrds% * %u||U[[.Ir"rc 6|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 Transportrs$8!$ I T N? Nv NNr"r&)%rrQrrer;r contextlibrrstructrr exceptionsrplatformr r utilsr rrrrRr AMQP_PORTbytesr*rrocompilerrxr!r$rrrrr"r rs %'- LL%++u||U5F5F G  w /rzz:;   ) mm` z%zz#%#LOr"