a lJhn@sRdZddgZddlZddlZddlZz"ddlZejejejej dZ Wne y\dZYn0ddl m Z mZddlmZddlmZmZmZmZmZd d lmZmZmZmZmZmZd d lmZm Z m!Z!m"Z"d d l#m$Z$m%Z%e&e'dddZ(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_connection)errreturncCs |jst|S|jd|jS)z`Reformat the IOError error message. This function reformats the IOError error message.  )errnostrstrerror)rr$M/var/www/shaz/venv/lib/python3.9/site-packages/mysql/connector/aio/network.py _strioerrorIsr&c @s\eZdZdZed eeeee ee ee ddddZ ed e j eee e dddZdS) 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_timeoutrcsdS)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%writeis zNetworkBroker.writereaderr* read_timeoutrcsdS)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%readszNetworkBroker.read)NNN)N)__name__ __module__ __qualname____doc__r rr"bytesrintr0asyncio StreamReader bytearrayr4r$r$r$r%r'Qs,r'c @seZdZdZddddZeeeeefdddZ de edd d d Z e e edd d dZdejee eedddZde e ee ee ee eddddZdeje e eedddZdS)NetworkBrokerPlain,Broker class for MySQL socket communication.NrcCs d|_dSN_pktnrr/r$r$r%__init__szNetworkBrokerPlain.__init__pktrcCsBt|tkrtdtd|dddd|d}}||fS)-Recover the header information from a packet.3Can't recover header info from an incomplete packet||t|IdH}||\}|_|||||IdHWStyx}z"td|t|fd|WYd}~n d}~00dS)z+Receive `one` packet from the MySQL server.Nr\r])rjrrTrDr`rr&)r/r2r*r3headerZ payload_lenrr$r$r%r4s zNetworkBrokerPlain.read)N)rN)NNN)N)r5r6r7r8rF staticmethodr9rr:rTrrZrr"rbr;r<r=rjr0r4r$r$r$r%r>sJ     3r>c seZdZdZddfdd ZeeeeedddZ eee eeefd d d Z de edd d dZ eeeddfdd Zdeeee ee ee eddfdd Zdejee eddfdd Zdejee eedfdd ZZS)NetworkBrokerCompressedr?Nr@cstd|_t|_dSrA)superrF_compressed_pktnrr _queue_readrE __class__r$r%rF-s z NetworkBrokerCompressed.__init__)r+pktnrrcCsd}g}tt|tD]>}|dtd||||t|dd}|t7}q|tdt||ddtd|||d|S) z2Prepare a payload for sending to the MySQL server.rrkrlrrXrKrLN)rmrNrappendrPrn)r+ryroZpktsrpr$r$r%_prepare_packets2s(     z(NetworkBrokerCompressed._prepare_packetsrGcCsbt|tkrtdtd|dddd|dtd|dddd}}}|||fS)rIrJrKrrLrM)rNrrOrPrQ)rHcompressed_pllrSuncompressed_pllr$r$r%rTKs  z"NetworkBrokerCompressed.get_headerrUcCs0|dur|jd7_n||_|jd;_dSrW)rurYr$r$r%_set_next_compressed_pktnrYsz2NetworkBrokerCompressed._set_next_compressed_pktnrr[csbt|}tdt|ddtd|jtdt|dd|}t|||IdHS)z1Compress packet and write it to the comm channel.rKrrLrlN)zlibcompressrPrnrNrurtrb)r/r)r*rHcompressed_pktrwr$r%rbas  z"NetworkBrokerCompressed._write_pktr(c s|||||z&td|||j}t|ttkrd}t t|tD]<} t | |||||t|IdH||t7}qTt | ||||d|IdHnt|t krt | ||||IdHnXt t ||tdt|ddtd|jtdddd||IdHWn:t jt jfyv} ztdd| WYd} ~ n d} ~ 00dS) 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. rdrNrKrLrlrer_)rZrr=joinr{rDrNrrrmr;rfrbrrtrPrnrurgrhr) r/r)r*r+r,r-r.Z payload_preprorprrwr$r%r0qsV         zNetworkBrokerCompressed.write)r2r~r3rc sttt|||IdH}d}|t|krtd|||tddd}t|t||krt|t |IdH}| |\}|_ }t|||IdH} ||dkr| nt| 7}|j |||t||t|7}q$dS)z&Handle reading of a compressed packet.NrrKrrM)r=r decompressrtrjrNrPrQrrrTrurvrz) r/r2r~r3rHrorRrqrrrwr$r%_read_compressed_pkts>   z,NetworkBrokerCompressed._read_compressed_pktr1c s|jszht|t|IdH}||\}|_}|dkrX|jt|||IdHn||||IdHWn:ty}z"t d|t |fd|WYd}~n d}~00|jsdS|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) rvrtrjrrTrurzrr`rr&popleftrD) r/r2r*r3rqr~rrrHrwr$r%r4s8     zNetworkBrokerCompressed.read)N)NNN)N)N)r5r6r7r8rFrrr9r:r r{rrTrrrr"rbr0r;r<rr=r4 __classcell__r$r$rwr%rs*sL  M3rsc @seZdZdZddddZeedddZee ddd d Z ddd d Z e dd dZ eddddZddddZejddddZd!eeeeeeeddddZd"eeedddZdddddggfeeeeeeee ee eeeeeeejddd ZdS)# MySQLSocketzMySQL socket communication interface. Examples: Subclasses: network.MySQLTCPSocket and network.MySQLUnixSocket. Nr@cCs*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_connectedrEr$r$r%rF s zMySQLSocket.__init__cCs|jS)zSocket location.)rrEr$r$r%r*+szMySQLSocket.addresskwargsrcsdS)zOpen the socket.Nr$r/rr$r$r%r0szMySQLSocket.open_connectionc sl|jrbz6|j|jjdur*|jj|jIdHWn$ty`}z WYd}~n d}~00d|_dS)zClose the connection.NF)rclose transportabortZ wait_closed Exceptionr)r/rpr$r$r%close_connection4s   zMySQLSocket.close_connectioncCs|jS)zCheck if the socket is connected. Return: bool: Returns `True` if the socket is connected to MySQL server. )rrEr$r$r% is_connectedCszMySQLSocket.is_connected)timeoutrcCs ||_dS)zSet the connection timeout.N)r)r/rr$r$r%set_connection_timeoutKsz"MySQLSocket.set_connection_timeoutcCs t|_dS)zIEnable network layer where transactions are made with compressed packets.N)rsrrEr$r$r%switch_to_compressed_modeOsz%MySQLSocket.switch_to_compressed_mode) ssl_contextrcsD|jdusJ|jjd}|jdkr.td|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)rrZget_extra_infofamilyrZ start_tls)r/rrr$r$r% switch_to_sslSs  zMySQLSocket.switch_to_ssl)r+r,r-r.rcs&|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.write)r3rcs|j|j|j|IdHS)z#Read packets from the MySQL server.N)rr4rr*)r/r3r$r$r%r4}szMySQLSocket.readF)ssl_cassl_certssl_keyssl_verify_certssl_verify_identity tls_versionstls_cipher_suitesrc Csd}|jstddtdur$tdz|r|jdd|d}t|} t| } |dkrd |vrr| jtjO_d |vr| jtj O_d |vr| jtj O_nt } || _ |rtj | _n|rtj| _ntj| _| |r0z| |Wn<ttjfy.} ztd | | WYd} ~ n d} ~ 00|rz| ||Wn<ttjfy} ztd | | WYd} ~ n d} ~ 00|r|d kr| d|| WSty} ztd| WYd} ~ nHd} ~ 0tttjtjfy} ztt| | WYd} ~ n d} ~ 00dS)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_locationsr`SSLErrorload_cert_chain set_ciphersr NameErrorrNotImplementedErrorCertificateErrorr") r/rrrrrrrZ tls_versionZ ssl_protocolcontextrr$r$r%build_ssl_contexts`      &&zMySQLSocket.build_ssl_context)NNN)N)r5r6r7r8rFpropertyr"r*r r rrboolrr:rrrrrr9rr0r=r4r rr$r$r$r%rsN    rcs<eZdZdZd eeedfdd Zedd d d Z 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 F)hostport force_ipv6cs0t||_||_||_|d||_dS)Nr)rtrF_host_portZ _force_ipv6r)r/rrrrwr$r%rFs  zMySQLTcpSocket.__init__Nrcs0tf|j|jd|IdH\|_|_d|_dS)zOpen TCP/IP connection.)rrNT)rrrrrrrr$r$r%rs zMySQLTcpSocket.open_connection)rrF) r5r6r7r8r"r:rrFr rrr$r$rwr%rs   cs8eZdZdZd edfdd Zedddd ZZS) rzbMySQL socket class using UNIX sockets. Args: unix_socket: UNIX socket file path. /tmp/mysql.sock) unix_socketcst||_dS)N)rtrFr)r/rrwr$r%rFs zMySQLUnixSocket.__init__Nrcs.tjfd|ji|IdH\|_|_d|_dS)zOpen UNIX socket connection.pathNT)r;Zopen_unix_connectionrrrrrr$r$r%rs zMySQLUnixSocket.open_connection)r) r5r6r7r8r"rFr rrr$r$rwr%rs)/r8__all__r;rPrrPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSr ImportErrorabcrr collectionsr typingr r r rrerrorsrrrrrrnetworkrrrrutilsrrr`r"r&r'r>rsrrrr$r$r$r%s:     Tp4