o DfY@sdZddlZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZmZdd lmZejejejejhZd ZeZd Zd Zed ZddddddZefddZ GdddZ!Gddde!Z"Gddde!Z#dddZ$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_KEEPCNTcCsd|}t|}|r|d}|drt|d}||fSd|vr.|dd\}}t|}||fS)z1Convert hostname:port string to host, port tuple.r:) IPV6_LITERALmatchgroupintrsplit)hostdefaultportmrG/home/ubuntu/webapp/venv/lib/python3.10/site-packages/amqp/transport.py to_host_port(s   r c@seZdZdZ   d$ddZdZddZd d Zed d Z d dZ ddZ ddZ ddZ d%ddZddZddZddZddZefd d!Zd"d#ZdS)&_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. NTcKsDd|_d|_||_t|_t|\|_|_||_||_ ||_ ||_ dSNF) connectedsockraise_on_initial_eintr EMPTY_BUFFER _read_bufferr rrconnect_timeout read_timeout write_timeoutsocket_settings)selfrr(r)r*r+r%kwargsrrr__init__Ws z_AbstractTransport.__init__) connectionr$r%r'rrr(r)r*r+__dict__ __weakref__c Cs|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)r,srcdstrrr__repr__ts ""*z_AbstractTransport.__repr__c Csrz |jrWdS||j|j|j||j|j|jd|_WdSt t fy8|j r7|js7|j d|_ w)NT) r#_connectrrr( _init_socketr+r)r*OSErrorrr$closer,rrrconnect|s    z_AbstractTransport.connectc cs|dur |jVdS|j}|}||kr||zLz|jVWn7tyC}zdt|vr4tdt|vr>td}~wtyY}z |jtj krTtd}~wwW||krf||dSdS||krr||ww)N timed outzThe operation did not complete) r$ gettimeout settimeoutrstrsockettimeoutr>errno EWOULDBLOCK)r,rGr$prevexcrrrhaving_timeouts8       z!_AbstractTransport.having_timeoutc Cst||tjtjt}t|D]S\}}|\}}} } } z*t||| |_zt|jdWn ty4Ynw|j ||j | WdStj yb|jrT|j d|_|dt |kr`YqwdS)NTr)rF getaddrinfo AF_UNSPEC SOCK_STREAMr enumerater$r NotImplementedErrorrDrAerrorr?len) r,rrrGentriesiresafsocktypeproto canonnamesarrrr<s0   z_AbstractTransport._connectc Cs|jd|jtjtjd||tj|ftj|ffD]!\}}|dur@t |}t ||d}|jtj|t d||q| | t dS)Nri@Bll)r$rD setsockoptrF SOL_SOCKET SO_KEEPALIVE_set_socket_options SO_SNDTIMEO SO_RCVTIMEOrr_setup_transport_writeAMQP_PROTOCOL_HEADER)r,r+r)r*rGintervalsecusecrrrr=s    z_AbstractTransport._init_socketc Csi}tD]C}d}|dkr zddlm}Wntyd}Yn wtt|r*tt|}|rG|tvr7t|||<qtt|rG|ttt|||<q|S)Nrr)r) rrFr ImportErrorhasattrgetattrDEFAULT_SOCKET_SETTINGS getsockoptr )r,r$tcp_optsoptenumrrr_get_tcp_socket_defaultss(     z+_AbstractTransport._get_tcp_socket_defaultscCs@||j}|r |||D] \}}|jt||qdSN)rrr$updateitemsr]r )r,r+rorpvalrrrr`s  z&_AbstractTransport._set_socket_optionsFcCtd)z#Read exactly n bytes from the peer.Must be overridden in subclassrQ)r,ninitialrrr_readz_AbstractTransport._readcCdS)z.Do any additional initialization of the class.Nrr@rrrrcz#_AbstractTransport._setup_transportcCr~)z8Do any preliminary work in shutting down the connection.Nrr@rrr_shutdown_transportrz&_AbstractTransport._shutdown_transportcCrw)z&Completely write a string to the peer.rxry)r,srrrrdr}z_AbstractTransport._writecCs|jdurshutdownrF SHUT_RDWRr?r#r@rrrr?s$      z_AbstractTransport.closec Csn|j}t}zJ|dd}||7}|d|\}}}|tkr@|t}z||t} Wntjttfy7||7}wd|| g} n||} || 7}t|d} WnUtjy^||j |_ ttfy} z9t | tj rt j dkr| jtjkr||j |_ tt | trdt| vr||j |_ t| jtvrd|_d } ~ ww| d kr||| fStd | 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>BHIrntrBFNzReceived frame_end z#04xz while expecting 0xce)r|r&SIGNED_INT_MAXrFrGr>rjoinordr' isinstancerRosnamerHrIrE_UNAVAILr#r) r,rreadread_frame_buffer frame_header frame_typechannelsizepart1part2payload frame_endrKrrr read_framesR           z_AbstractTransport.read_framec CsLz||WdStjyty%}z |jtvr d|_d}~wwr")rdrFrGr>rHrr#)r,rrKrrrwriteYs z_AbstractTransport.write)NNNNT)F)r7 __module__ __qualname____doc__r. __slots__r;rArrLr<r=rrr`r|rcrrdr?rrrrrrrr!7s,      Br!cseZdZdZdfdd ZdZddZddd Zdd d Z    dddZ ddZ d e j e j e jffddZddZZS) 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 s6t|tr|ni|_t|_tj|fd|i|dS)Nr()rdictssloptsr&r'superr.)r,rr(sslr- __class__rrr.s zSSLTransport.__init__)rcCs>|j|jfi|j|_|j|j|j|jj|_dS)z!Wrap the socket in an SSL object.N) _wrap_socketr$rrDr( do_handshaker _quick_recvr@rrrrcs zSSLTransport._setup_transportcKs*|r |j||fi|S|j|fi|Srs) _wrap_context_wrap_socket_sni)r,r$contextrrrrrszSSLTransport._wrap_socketcKs(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`. Nr)rcreate_default_contextcheck_hostname wrap_socket)r,r$rr ctx_optionsctxrrrrszSSLTransport._wrap_contextFTc Cs||||| d} | dur|rtjntj} t| } |dur#| |||dur,| || dur5| | z tjo<| du| _Wn t yHYnw|durP|| _ |duri| j tj kri|r`tj 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)r,r$keyfilecertfiler cert_reqsca_certsrrrciphers ssl_versionoptsrpurposerrrrsDD        zSSLTransport._wrap_socket_snicCs|jdur |j|_dSdS)z/Unwrap a SSL socket, so we can call shutdown().N)r$unwrapr@rrrr/s z SSLTransport._shutdown_transportc Cs|j}|j}zDt||krIz ||t|}Wn!ty8}z|j|vr3|r-|jr-tWYd}~qd}~ww|s?td||7}t||ks Wn||_|d|||d}|_|S)N%Server unexpectedly closed connectionrr'rSr>rHr%rFrG r,rzr{_errnosrecvrbufrrKresultrrrr|4s0     zSSLTransport._readcCsT|jj}|r(z||}Wn tyd}Ynw|std||d}|sdSdS)z+Write a string out to the SSL socket fully.rz Socket closedN)r$r ValueErrorr>)r,rrrzrrrrdPs    zSSLTransport._write)NNrs) NNFNNFTNNN)r7rrrr.rrcrrrrrHENOENTEAGAINEINTRr|rd __classcell__rrrrrds$"   x rc@s.eZdZdZddZdejejffddZdS) TCPTransportz~Transport that deals directly with TCP socket. All parameters are :class:`~amqp.transport._AbstractTransport` class. cCs|jj|_t|_|jj|_dSrs)r$sendallrdr&r'rrr@rrrrcgs zTCPTransport._setup_transportFc Cs|j}|j}zDt||krIz ||t|}Wn!ty8}z|j|vr3|r-|jr-tWYd}~qd}~ww|s?td||7}t||ks Wn||_|d|||d}|_|S)z%Read exactly n bytes from the socket.Nrrrrrrr|ns0      zTCPTransport._readN) r7rrrrcrHrrr|rrrrrasrFcKs"|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(rr- transportrrr Transports rr")%rrHrrerFr contextlibrrstructrr exceptionsrplatformrr utilsr rrrrIr AMQP_PORTbytesr&rrecompilerrmr r!rrrrrrrs@      /~&