hjlUdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZ ddlZejejejejdZeedoejZn#e$rdZdZYnwxYwd d lmZmZmZmZd Z e!e"d <d Z#e!e"d<dZ$e!e"d<dZ%e!e"d<de&de'fdZ(GddeZ)Gdde)Z*Gdde*Z+GddeZ,Gdde,Z-Gd d!e,Z.dS)"zGModule implementing low-level socket communication with MySQL servers. N)ABCabstractmethod)deque)AnyDequeListOptionalTupleUnion)TLSv1TLSv1.1TLSv1.2TLSv1.3 HAS_TLSv1_3F)InterfaceErrorNotSupportedErrorOperationalErrorProgrammingError2MIN_COMPRESS_LENGTHiMAX_PAYLOAD_LENGTHPACKET_HEADER_LENGTHCOMPRESSED_PACKET_HEADER_LENGTHerrreturncP|jst|n|jd|jS)z`Reformat the IOError error message. This function reformats the IOError error message.  )errnostrstrerror)rs _/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/mysql/connector/network.py _strioerrorr%Gs- 9 G3s888SY*G*G*G*GGceZdZdZe d dejdededee dee ddf d Z edejdede fd Z dS) NetworkBrokeraPBroker class interface. The network object is a broker used as a delegate by a socket object. Whenever the socket wants to deliver or get packets to or from the MySQL server it needs to rely on its network broker (netbroker). The netbroker sends `payloads` and receives `packets`. A packet is a bytes sequence, it has a header and body (referred to as payload). The first `PACKET_HEADER_LENGTH` or `COMPRESSED_PACKET_HEADER_LENGTH` (as appropriate) bytes correspond to the `header`, the remaining ones represent the `payload`. The maximum payload length allowed to be sent per packet to the server is `MAX_PAYLOAD_LENGTH`. When `send` is called with a payload whose length is greater than `MAX_PAYLOAD_LENGTH` the netbroker breaks it down into packets, so the caller of `send` can provide payloads of arbitrary length. Finally, data received by the netbroker comes directly from the server, expect to get a packet for each call to `recv`. The received packet contains a header and payload, the latter respecting `MAX_PAYLOAD_LENGTH`. Nsockaddresspayload packet_numbercompressed_packet_numberrcdS)aSend `payload` to the MySQL server. If provided a payload whose length is greater than `MAX_PAYLOAD_LENGTH`, it is broken down into packets. Args: sock: Object holding the socket connection. address: Socket's location. payload: Packet's body to send. packet_number: Sequence id (packet ID) to attach to the header when sending plain packets. compressed_packet_number: Same as `packet_number` but used when sending compressed packets. Raises: :class:`OperationalError`: If something goes wrong while sending packets to the MySQL server. N)selfr)r*r+r,r-s r$sendzNetworkBroker.sendgr&cdS)a)Get the next available packet from the MySQL server. Args: sock: Object holding the socket connection. address: Socket's location. Returns: packet: A packet from the MySQL server. Raises: :class:`OperationalError`: If something goes wrong while receiving packets from the MySQL server. :class:`InterfaceError`: If something goes wrong while receiving packets from the MySQL server. Nr/)r0r)r*s r$recvzNetworkBroker.recvr2r&NN) __name__ __module__ __qualname____doc__rsocketr"bytesr intr1 bytearrayr4r/r&r$r(r(Os. (,26   m     }  #+3-      ^ 6      ^   r&r(c eZdZdZddZddZdejdededdfd Z ddejd e de fd Z ddejded ede e de e ddf dZdejdede fdZdS)NetworkBrokerPlain,Broker class for MySQL socket communication.rNcd|_dSN_pktnrr0s r$__init__zNetworkBrokerPlain.__init__s  r&c*|jdzdz|_dSzIncrement packet id.rNrDrFs r$_set_next_pktnrz"NetworkBrokerPlain._set_next_pktnrs{Q#- r&r)r*pktc ||dS#t$r&}td|t|f|d}~wt$r}td|d}~wwxYw)z!Write packet to the comm channel.r!valuesNir!)sendallIOErrorrr%AttributeError)r0r)r*rLrs r$ _send_pktzNetworkBrokerPlain._send_pkts 8 LL        "G[-=-=#>  8 8 8"...C 7 8s  A'!A A'A""A'rsizect|}t|}|rC|||}|dkr|dkrtd||d}||z}|C|S)z(Read `size` bytes from the comm channel.rirQN)r= memoryview recv_intor)r0r)rVrLpkt_viewreads r$ _recv_chunkzNetworkBrokerPlain._recv_chunks|ooc?? >>(D11DqyyTAXX$40000H DLD    r&r+r,r-c X||n||_t|tkrd}t t|tzD]e}|||dt jd|jz|||tzz||tz }f||d}|||t jdt|ddt jd|jz|zdS)zSend payload to the MySQL server. If provided a payload whose length is greater than `MAX_PAYLOAD_LENGTH`, it is broken down into packets. Nr> - -#k$ 445fv0B'BBCD$$&&&,,fgg&G    Kc'll + +AaC 0k$ ,, -      r&c> ||t}tjd|dddzd|dc}|_||||zS#t $r&}t d|t|f|d}~wwxYw) z+Receive `one` packet from the MySQL server.rVr`rrarNrON)r\rrdunpackrErSrr%)r0r)r*header payload_lenrs r$r4zNetworkBrokerPlain.recvs %%d1E%FFF dF1Q3K'$9::1=q  %K D,,T ,DDD D   "G[-=-=#>  sA)A,, B6!BBrN)rr5)r6r7r8r9rGrKr:r"r;rUr<r=r\r r1r4r/r&r$r?r?s566.... 8fm 8c 8 8$ 8 8 8 8   S     "(,26 * * m* *  *  } * #+3- *  * * * * Xr&r?c eZdZdZdfd ZedededeefdZ ddZ d e j d e d eddffd Z dd e j d e ded eedeeddf fd Zd e j dededdffd Zd e j d e deffd ZxZS)NetworkBrokerCompressedr@rNc|td|_t|_dSrB)superrG_compressed_pktnrr _queue_read)r0 __class__s r$rGz NetworkBrokerCompressed.__init__s3 !#-2WWr&r+pktnrcg}t|tkrd}tt|tzD]R}|dt jd|z|||tzz|dzdz}|tz }S||d}|t jdt|ddt jd|z|z|S) z2Prepare a payload for sending to the MySQL server.rr^r_rrJNr`ra)rbrrcappendrdre)r+rvpktsrfrgs r$_prepare_packetsz(NetworkBrokerCompressed._prepare_packetss w<<- - -F3w<<+==>> - - #k$../fv0B'BBCD c),,fgg&G Kc'll + +AaC 06;tU3K3K Kg U    r&c*|jdzdz|_dSrI)rsrFs r$_set_next_compressed_pktnrz2NetworkBrokerCompressed._set_next_compressed_pktnrs"&"81"<!Cr&r)r*rLcXtj|}tjdt |ddtjd|jztjdt |ddz|z}t |||S)z1Compress packet and write it to the comm channel.r`rrar_)zlibcompressrdrerbrsrrrU)r0r)r*rLcompressed_pktrus r$rUz!NetworkBrokerCompressed._send_pktss++ Kc.11 2 21Q3 7k$ 677 8k$C))!A#. /  ww  w444r&r,r-c ||n||_||n||_t d|||j}t|ttz krd}tt|tzD]G}| |||||tz||tz }H| ||||ddSt|tkr| |||dSt ||tjdt|ddtjd|jztjddddz|zdS)zSend `payload` as compressed packets to the MySQL server. If provided a payload whose length is greater than `MAX_PAYLOAD_LENGTH`, it is broken down into packets. Nr&rr`rar_)rKrEr|rsr=joinrzrbrrrcrUrrrrdre) r0r)r*r+r,r- payload_preprfrgrus r$r1zNetworkBrokerCompressed.send(s   " " " "'DK # +  + + - - - -%=D " ~~**4+@+@$++V+VWW w<<-0DD D DF3|,,0BBCC - -'<BT9T0T#U//111,, NN4,vww*? @ @ @ @ @ 7||111tWl;;;;;!!Kc,&7&7881=k$(>??@k$**1Q3/0##r&compressed_plluncompressed_pllct||}|dkr|n ttj|}d}|t |krdt jd|||tzdz dzd}t|zt ||z krt|t}t jd|dddzd|dt jd|dddzd c}|_ }t||}||dkr|ntj|z }|j |||tz|z|t|zz }|t |kbd Sd S) z&Handle reading of a compressed packet.rirr`rrjrarrN) rrr\r=r~ decompressrbrdrkrrrsrtrx) r0r)rrrrLrfpllrlrus r$_recv_compressed_pktz,NetworkBrokerCompressed._recv_compressed_pktes,,T,GG 1$$ N4?>::;; s3xx-c&6,@#@1#DDEOC$c)CHHv,===,,T8W,XXM$qs g(=>>qA1IM$qs g(=>>qA "*$ "'!4!4T!4!O!O'1,,#N88   # #C:N1NQT1T(T$U V V V *S0 0F?s3xxr&c|js t|t}t jd|dddzd|dt jd|dddzd c}|_}||||n3#t$r&}td|t|f |d }~wwxYw|jsd S|j }|d|_ |S) z{Receive `one` or `several` packets from the MySQL server, enqueue them, and return the packet at the head. rir`rrarjrrrNrON) rtrrr\rrdrkrsrrSrr%popleftrE) r0r)r*rlrrrrLrus r$r4zNetworkBrokerCompressed.recvs2  ,,T8W,XXM$qs g(=>>qA1IM$qs g(=>>qA "*$ ))$@PQQQQ   &S1A1A'B   4&&((!f  sBB%% C/!CCrnr5)r6r7r8r9rG staticmethodr;r<rrzr|r:r"rUr r1rr=r4 __classcell__rus@r$rprps66555555 %U \2DDDD 5fm 5c 5 5$ 5 5 5 5 5 5 (,26 ;;m;; ;  } ; #+3- ; ;;;;;;z.1M.136.1JM.1 .1.1.1.1.1.1`r&rpceZdZdZddZddZddZddZddZd e e ddfd Z d e d e ddfd Z dde e de e de e de ede ede ee de ee de fdZ d dede e de e ddfdZdefdZeddZeede fdZdS)! MySQLSocketzMySQL socket communication interface. Examples: Subclasses: network.MySQLTCPSocket and network.MySQLUnixSocket. rNcVd|_d|_d|_t|_dS)zsNetwork layer where transactions are made with plain (uncompressed) packets is enabled by default. N)r)_connection_timeout server_hostr? _netbrokerrFs r$rGzMySQLSocket.__init__s, .2 26 *.);)=)=r&c,t|_dS)zIEnable network layer where transactions are made with compressed packets.N)rprrFs r$switch_to_compressed_modez%MySQLSocket.switch_to_compressed_modes133r&c |jtj|jdS#t t f$rYdSwxYw)z'Shut down the socket before closing it.N)r)shutdownr: SHUT_RDWRcloserTOSErrorrFs r$rzMySQLSocket.shutdowns_  I  v/ 0 0 0 IOO     (    DD s=AAAcj |jdS#ttf$rYdSwxYw)zClose the socket.N)r)rrTrrFs r$close_connectionzMySQLSocket.close_connectionsF  IOO     (    DD s 22c.|dSN)rrFs r$__del__zMySQLSocket.__del__s r&timeoutcZ||_|jr|j|dSdS)zSet the connection timeout.N)rr) settimeout)r0rs r$set_connection_timeoutz"MySQLSocket.set_connection_timeouts:#* 9 * I  ) ) ) ) ) * *r& ssl_contexthostc:|jJ|jjdkrtdtt d ||j||_dS#t $r}t d|d}~wtjtf$r+}td|j t|f|d}~wtj $r"}tt||d}~wt$r"}tt||d}~wwxYw)aUpgrade an existing connection to TLS. Args: ssl_context (ssl.SSLContext): The SSL Context to be used. host (str): Server host name. Returns: None. Raises: ProgrammingError: If the transport does not expose the socket instance. NotSupportedError: If Python installation has no SSL support. Nrz,SSL is not supported when using Unix sockets&Python installation has no SSL support)server_hostnamerNrO)r)familyrsslr wrap_socket NameErrorSSLErrorrSrr*r%CertificateErrorr"NotImplementedError)r0rrrs r$ switch_to_sslzMySQLSocket.switch_to_sslsFy$$$ 9 q "#QRR R ;#$LMM M 4#// 4/PPDIII W W W#$LMMSV V g&    DL+c2B2B#C # 4 4 4 S** 3" 4 4 4 S** 3 4s<!A## D-A==D&B<<DC++ D8DDFssl_cassl_certssl_keyssl_verify_certssl_verify_identity tls_versionstls_cipher_suitesc|jstdttd|g}|g} |r|d|d}t |} tj| } |dkrZd |vr| xjtjzc_d |vr| xjtj zc_d |vr| xjtj zc_ntj } || _ |rtj | _n%|rtj| _ntj| _| |rH | |n1#t&tjf$r} td | | d} ~ wwxYw|rI | ||n1#t&tjf$r} td | | d} ~ wwxYw|r(| d|| S#t0$r} td| d} ~ wt&t2tjtjf$r"} tt7| | d} ~ wwxYw)aBuild a SSLContext. Args: ssl_ca: Certificate authority, opptional. ssl_cert: SSL certificate, optional. ssl_key: Private key, optional. ssl_verify_cert: Verify the SSL certificate if `True`. ssl_verify_identity: Verify host identity if `True`. tls_versions: TLS protocol versions, optional. tls_cipher_suites: Set of steps that helps to establish a secure connection. Returns: ssl_context (ssl.SSLContext): An SSL Context ready be used. Raises: NotSupportedError: Python installation has no SSL support. InterfaceError: Socket undefined or invalid ssl data. irQNrT)reverserrrr r zInvalid CA Certificate: zInvalid Certificate/Key: :)r)rrrsort TLS_VERSIONS SSLContextoptions OP_NO_TLSv1_2 OP_NO_TLSv1_1 OP_NO_TLSv1create_default_contextcheck_hostname CERT_REQUIRED verify_mode CERT_OPTIONAL CERT_NONEload_default_certsload_verify_locationsrSrload_cert_chain set_ciphersrrrrr") r0rrrrrrr tls_version ssl_protocolcontextrs r$build_ssl_contextzMySQLSocket.build_ssl_contexts8y - t,,, , ;#$LMM M  L  $ " 4 4 7!!$!///*1o +K8 .66)++ 443+<< 443+<<l223?:466%8G " 4&)&7##$ 4&)&7##&)m#  & & ( ( ( TT11&9999.TTT()IC)I)IJJPSST UU++Hg>>>>.UUU()JS)J)JKKQTTU! A##CHH->$?$?@@@N W W W#$LMMSV V    L   4 4 4 !S** 3  4slD HEHF/FFH F$#H$G:G  G.H I' H*I'I""I'r+r,r-cT|j|j|j|||S)z#Send `payload` to the MySQL server.)r,r-)rr1r)r*)r0r+r,r-s r$r1zMySQLSocket.sendfs6## I L '%= $   r&cL|j|j|jS)z.Get packet from the MySQL server comm channel.)rr4r)r*rFs r$r4zMySQLSocket.recvus##DIt|<<>>>4444*hsm***** "4"4C"4D"4"4"4"4L!%"&!%*/.3,015\4\4 \43-\4# \4 "$ \4 &d^ \4tCy)\4$DI.\4 \4\4\4\4B(,26      }  #+3-       =i====^....^X...r&rcfeZdZdZd deddffd ZedefdZd dZd e d e ddfd Z xZ S)MySQLUnixSocketzpMySQL socket class using UNIX sockets. Opens a connection through the UNIX socket of the MySQL Server. /tmp/mysql.sock unix_socketrNcdt||_||_dSr)rrrGr_address)r0rrus r$rGzMySQLUnixSocket.__init__s,  +( r&c|jSrrrFs r$r*zMySQLUnixSocket.address }r&c tjtjtj|_|j|j|j|jdS#t$r+}td|j t|f|d}~wt$r"}tt||d}~wwxYw)NirO)r:AF_UNIX SOCK_STREAMr)rrconnectrrSrr*r% Exceptionr")r0rs r$rzMySQLUnixSocket.open_connections 4  2DI I !9 : : : I  d. / / / / /    DL+c2B2B#C  4 4 4 S** 3 4s$A,A00 C:&B  C-C  Cargskwargsc:tjdtdS)zSwitch the socket to use SSL.z2SSL is disabled when using unix socket connectionsN)warningswarnWarning)r0rrs r$rzMySQLUnixSocket.switch_to_ssls(  @      r&)rrn) r6r7r8r9r"rGrr*rrrrrs@r$rrs ))C))))))) X 4 4 4 4  $'         r&rc `eZdZdZ d dedededd ffd Zedefd Z dd Z xZ S)MySQLTCPSocketzYMySQL socket class using TCP/IP. Opens a TCP/IP connection to the MySQL Server. 127.0.0.1 Frport force_ipv6rNct||_||_||_d|_|d||_dS)Nrr)rrrGr server_portr_familyr)r0rrrrus r$rGzMySQLTCPSocket.__init__sR  $ $ * $--t-- r&c|jSrrrFs r$r*zMySQLTCPSocket.addressrr&c td} tj|j|jdtjtj}|D]=}|jr|dtjkr|}n|dtjkr|}n>|jr|dtd|j|d|d}n8#t$r+}td|j t|f|d}~wwxYw|\|_ }}}} tj|j |||_|j|j|j|dS#t$r1}td|j|jt|f|d}~wt$$r"}t't)||d}~wwxYw)z/Open the TCP/IP connection to the MySQL server.)NNNNNrNzNo IPv6 address found for irO)r: getaddrinforrrSOL_TCPrAF_INET6AF_INETrrSr*r%rr)rrrrrr") r0addrinfo addrinfosinforsocktypeprotorgsockaddrs r$rzMySQLTCPSocket.open_connections +  *  " I"  ?tAw&/'A'A#HE7fn,,#HE- V8A;#6$%T$BR%T%TUUU{"$Q<    DL+c2B2B#C   8@4x8 6 dlHeDDDI I !9 : : : I  h ' ' ' ' '    $$$$  6 6 6"3s88,,# 5 6s=B,B11 C&;&C!!C&7AE F7,F F7F22F7)rrFrn) r6r7r8r9r"r<rrGrr*rrrs@r$rrs  . . . . .  . . . . . .X8686868686868686r&r)/r9r:rdrr~abcrr collectionsrtypingrrrr r r rPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSrhasattrrTLS_V1_3_SUPPORTED ImportErrorerrorsrrrrrr<__annotations__rrrrSr"r%r(r?rprrrr/r&r$r s>  ########;;;;;;;;;;;;;;;; JJJ#''# L !m44H CCC S#C###c'((((HWHHHHHD D D D D CD D D N^^^^^^^^B}}}}}0}}}@I.I.I.I.I.#I.I.I.X$ $ $ $ $ k$ $ $ NO6O6O6O6O6[O6O6O6O6O6s2A&& A21A2