o gn@sTdZddgZddlZddlZddlZzddlZejejejej dZ Wn e y.dZYnwddl m Z mZddlmZddlmZmZmZmZmZd d lmZmZmZmZmZmZd d lmZm Z m!Z!m"Z"d d l#m$Z$m%Z%de&de'fddZ(Gddde Z)Gddde)Z*Gddde*Z+Gddde Z,Gddde,Z-Gddde,Z.dS)zFModule implementing low-level socket communication with MySQL servers.MySQLTcpSocketMySQLUnixSocketN)TLSv1TLSv1.1TLSv1.2TLSv1.3)ABCabstractmethod)deque)AnyDequeListOptionalTuple)InterfaceErrorNotSupportedErrorOperationalErrorProgrammingErrorReadTimeoutErrorWriteTimeoutError)COMPRESSED_PACKET_HEADER_LENGTHMAX_PAYLOAD_LENGTHMIN_COMPRESS_LENGTHPACKET_HEADER_LENGTH) StreamWriteropen_connectionerrreturncCs |jst|S|jd|jS)z`Reformat the IOError error message. This function reformats the IOError error message.  )errnostrstrerror)rr$U/var/www/html/api-tag/env/lib/python3.10/site-packages/mysql/connector/aio/network.py _strioerrorIs r&c@sveZdZdZe   ddedededee dee dee d dfd d Z e dd e j ded ee d e fddZdS) 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`. Nwriteraddresspayload packet_numbercompressed_packet_number write_timeoutrcdS)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. write_timeout: Timeout in seconds before which sending a packet to the server should finish else WriteTimeoutError is raised. Raises: :class:`OperationalError`: If something goes wrong while sending packets to the MySQL server. Nr$)selfr(r)r*r+r,r-r$r$r%writeizNetworkBroker.writereader read_timeoutcr.)aMGet the next available packet from the MySQL server. Args: sock: Object holding the socket connection. address: Socket's location. read_timeout: Timeout in seconds before which reading a packet from the server should finish. Returns: packet: A packet from the MySQL server. Raises: :class:`OperationalError`: If something goes wrong while receiving packets from the MySQL server. :class:`ReadTimeoutError`: If the time to receive a packet from the server takes longer than `read_timeout`. :class:`InterfaceError`: If something goes wrong while receiving packets from the MySQL server. Nr$)r/r2r)r3r$r$r%readr1zNetworkBroker.readNNNN)__name__ __module__ __qualname____doc__r rr"bytesrintr0asyncio StreamReader bytearrayr4r$r$r$r%r'Qs@ r'c@seZdZdZdddZededeeeffddZ dd e eddfd d Z d e d e deddfddZ  d dejdede edefddZ   d!d e d e dede ede ede eddfddZ ddejd e de edefddZdS)"NetworkBrokerPlain,Broker class for MySQL socket communication.rNcCs d|_dSN_pktnrr/r$r$r%__init__s zNetworkBrokerPlain.__init__pktcCsBt|tkr tdtd|dddd|d}}||fS)-Recover the header information from a packet.3Can't recover header info from an incomplete packetr?rkr0r4r$r$r$r%r@sl     3r@cs$eZdZdZd fdd Zedededeefdd Z ed ede eeeffd d Z d!d e eddfddZ deded eddffdd Z   d"dededede ede ede eddffdd Z d!dejdede eddffdd Z d!dejdede edeffdd ZZS)#NetworkBrokerCompressedrArNcstd|_t|_dSrB)superrG_compressed_pktnrr _queue_readrF __class__r$r%rG-s  z NetworkBrokerCompressed.__init__r*pktnrcCsd}g}tt|tD]}|dtd||||t|dd}|t7}q |tdt||ddtd|||d|S) z2Prepare a payload for sending to the MySQL server.rrlrmrrXrKrLN)rnrNrappendrPro)r*r|rppktsrqr$r$r%_prepare_packets2s(     z(NetworkBrokerCompressed._prepare_packetsrHcCsbt|tkr tdtd|dddd|dtd|dddd}}}|||fS)rIrJrKrrLrM)rNrrOrPrQ)rHcompressed_pllrSuncompressed_pllr$r$r%rTKs  z"NetworkBrokerCompressed.get_headerrUcCrVrW)rxrYr$r$r%_set_next_compressed_pktnrYr[z2NetworkBrokerCompressed._set_next_compressed_pktnrr(r)csdt|}tdt|ddtd|jtdt|dd|}t|||IdHS)z1Compress packet and write it to the comm channel.rKrrLrmN)zlibcompressrProrNrxrwrc)r/r(r)rHcompressed_pktrzr$r%rcas  z"NetworkBrokerCompressed._write_pktr+r,r-c st||||ztd|||j}t|ttkr^d}t t|tD]} t | |||||t|IdH||t7}q*t | ||||d|IdHWdSt|t krut | ||||IdHWdSt t ||tdt|ddtd|jtdddd||IdHWdSt jt jfy} ztdd| d} ~ ww) 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. rerNrKrLrmrfr_)rZrr?joinrrErNrrrnr=rgrcrrwrProrxrhrir) r/r(r)r*r+r,r- payload_preprprqrrzr$r%r0qs\        zNetworkBrokerCompressed.writer2rr3c sttt|||IdH}d}|t|kr}td|||tddd}t|t||krat|t |IdH}| |\}|_ }t|||IdH} ||dkr[| nt| 7}|j |||t||t|7}|t|ksdSdS)z&Handle reading of a compressed packet.NrrKrrM)r?r decompressrwrkrNrPrQrrrTrxryr}) r/r2rr3rHrprRrrrrrzr$r%_read_compressed_pktsB     z,NetworkBrokerCompressed._read_compressed_pktc s|jsPz4t|t|IdH}||\}|_}|dkr-|jt|||IdHn ||||IdHWntyO}z t d|t |fd|d}~ww|jsUdS|j }|d|_ |S)z{Receive `one` or `several` packets from the MySQL server, enqueue them, and return the packet at the head. Nrr\r]rL) ryrwrkrrTrxr}rrarr&popleftrE) r/r2r)r3rrrrrrHrzr$r%r4s@       zNetworkBrokerCompressed.readrtr6r5)r7r8r9r:rGrur;r<r rrrTrrrr"rcr0r=r>rr?r4 __classcell__r$r$rzr%rv*sn M3rvc@s4eZdZdZd*ddZedefddZede ddfd d Z d*d d Z de fd dZ deddfddZd*ddZdejddfddZ   d+dedeedeedeeddf ddZd,deedefddZdddd d ggfd!eed"eed#eed$ee d%ee d&eeed'eeedejfd(d)ZdS)- MySQLSocketzMySQL socket communication interface. Examples: Subclasses: network.MySQLTCPSocket and network.MySQLUnixSocket. rNcCs*d|_d|_d|_d|_t|_d|_dS)zsNetwork layer where transactions are made with plain (uncompressed) packets is enabled by default. NF)_reader_writer_connection_timeout_addressr@ _netbroker _is_connectedrFr$r$r%rG s  zMySQLSocket.__init__cC|jS)zSocket location.)rrFr$r$r%r)+szMySQLSocket.addresskwargscr.)zOpen the socket.Nr$r/rr$r$r%r0r1zMySQLSocket.open_connectionc sn|jr2z|j|jjdur|jj|jIdHWnty1}zWYd}~nd}~wwd|_dS)zClose the connection.NF)rclose transportabort wait_closed Exceptionr)r/rqr$r$r%close_connection4s     zMySQLSocket.close_connectioncCr)zCheck if the socket is connected. Return: bool: Returns `True` if the socket is connected to MySQL server. )rrFr$r$r% is_connectedCszMySQLSocket.is_connectedtimeoutcCs ||_dS)zSet the connection timeout.N)r)r/rr$r$r%set_connection_timeoutKs z"MySQLSocket.set_connection_timeoutcCs t|_dS)zIEnable network layer where transactions are made with compressed packets.N)rvrrFr$r$r%switch_to_compressed_modeOs z%MySQLSocket.switch_to_compressed_mode ssl_contextcsF|jdusJ|jjd}|jdkrtd|j|IdHdS)a?Upgrade an existing stream-based connection to TLS. The `start_tls()` method from `asyncio.streams.StreamWriter` is only available in Python 3.11. This method is used as a workaround. The MySQL TLS negotiation happens in the middle of the TCP connection. Therefore, passing a socket to open connection will cause it to negotiate TLS on an existing connection. Args: ssl_context: The SSL Context to be used. Raises: RuntimeError: If the transport does not expose the socket instance. Nsocketrz,SSL is not supported when using Unix sockets)rrget_extra_infofamilyr start_tls)r/rrr$r$r% switch_to_sslSs  zMySQLSocket.switch_to_sslr*r+r,r-cs(|jj|j|j||||dIdHdS)z!Send packets to the MySQL server.)r+r,r-N)rr0rr))r/r*r+r,r-r$r$r%r0lszMySQLSocket.writer3cs|j|j|j|IdHS)z#Read packets from the MySQL server.N)rr4rr))r/r3r$r$r%r4}szMySQLSocket.readFssl_cassl_certssl_keyssl_verify_certssl_verify_identity tls_versionstls_cipher_suitesc Csd}|js tddtdurtdz|rQ|jdd|d}t|} t| } |dkrPd |vr8| jtjO_d |vrD| jtj O_d |vrP| jtj O_nt } || _ |r_tj | _n |rftj| _ntj| _| |rz| |Wnttjfy} ztd | | d} ~ ww|rz| ||Wnttjfy} ztd | | d} ~ ww|r|d kr| d|| WSty} ztd| d} ~ wtttjtjfy} ztt| | d} ~ ww)zBuild a SSLContext.Nir_z&Python installation has no SSL supportT)reverserrrrrzInvalid CA Certificate: zInvalid Certificate/Key: :)rrssl RuntimeErrorsort 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_locationsraSSLErrorload_cert_chain set_ciphersr NameErrorrNotImplementedErrorCertificateErrorr") r/rrrrrrr tls_version ssl_protocolcontextrr$r$r%build_ssl_contextsp         zMySQLSocket.build_ssl_contextrtr5r6)r7r8r9r:rGpropertyr"r)r r rrboolrr<rrrrrr;rr0r?r4r rr$r$r$r%rsf        rcsDeZdZdZ ddededeffdd Zd ed d fd dZ Z S)rzMySQL socket class using TCP/IP. Args: host: MySQL host name. port: MySQL port. force_ipv6: Force IPv6 usage. 127.0.0.1 Fhostport force_ipv6cs0t||_||_||_|d||_dS)Nr)rwrG_host_port _force_ipv6r)r/rrrrzr$r%rGs zMySQLTcpSocket.__init__rrNcs2td|j|jd|IdH\|_|_d|_dS)zOpen TCP/IP connection.)rrNTr$)rrrrrrrr$r$r%rs zMySQLTcpSocket.open_connection)rrF) r7r8r9r:r"r<rrGr rrr$r$rzr%rs  cs:eZdZdZd deffdd Zdeddfd d ZZS) rzbMySQL socket class using UNIX sockets. Args: unix_socket: UNIX socket file path. /tmp/mysql.sock unix_socketcst||_dSr6)rwrGr)r/rrzr$r%rGs  zMySQLUnixSocket.__init__rrNcs0tjdd|ji|IdH\|_|_d|_dS)zOpen UNIX socket connection.pathNTr$)r=open_unix_connectionrrrrrr$r$r%rs  zMySQLUnixSocket.open_connection)r) r7r8r9r:r"rGr rrr$r$rzr%rs)/r:__all__r=rPrrPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSr ImportErrorabcrr collectionsr typingr r r rrerrorsrrrrrrnetworkrrrrutilsrrrar"r&r'r@rvrrrr$r$r$r%s>   Tp4