U j`a@sLdZddgZddlZddlZddlZz"ddlZejejejej dZ Wne k r^dZYnXddl m Z mZddlmZddlmZmZmZmZmZd d lmZmZmZmZd d lmZmZmZm 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)InterfaceErrorNotSupportedErrorOperationalErrorProgrammingError)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"?/tmp/pip-unpacked-wheel-d6bt0v6z/mysql/connector/aio/network.py _strioerrorGsr$c @sNeZdZdZed eeeee ee ddddZ ee j ee 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_numberrcsdS)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. Nr")selfr'r(r)r*r+r"r"r#writegs zNetworkBroker.writereaderr(rcsdS)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")r,r/r(r"r"r#readszNetworkBroker.read)NN)__name__ __module__ __qualname____doc__r rr bytesrintr-asyncio StreamReader bytearrayr0r"r"r"r#r%Osr%c@seZdZdZddddZeeeeefdddZ de edd d d Z e e edd d dZdejeedddZde e ee ee eddddZeje edddZdS)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|j|tdIdH}||\}|_||j||dIdHWStk rv}ztd|t|fd|W5d}~XYnXdS)z+Receive `one` packet from the MySQL server.r_NrXrY)rbrrPr@r\rr$)r,r/r(headerZ payload_lenrr"r"r#r0s zNetworkBrokerPlain.read)N)r)NN)r1r2r3r4rB staticmethodr5rr6rPrrVrr r^r7r8r9rbr-r0r"r"r"r#r:s.    &r:cseZdZdZddfdd ZeeeeedddZ eee eeefd d d Z de edd d dZ eeeddfdd Zdeeee ee eddfdd Zejeddfdd Zejeedfdd ZZS)NetworkBrokerCompressedr;Nr<cstd|_t|_dSr=)superrB_compressed_pktnrr _queue_readrA __class__r"r#rBs 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.rrcrdrrTrGrHN)rerJrappendrLrf)r)rrrgZpktsrhr"r"r#_prepare_packetss(     z(NetworkBrokerCompressed._prepare_packetsrCcCsbt|tkrtdtd|dddd|dtd|dddd}}}|||fS)rErFrGrrHrI)rJrrKrLrM)rDcompressed_pllrOuncompressed_pllr"r"r#rP!s  z"NetworkBrokerCompressed.get_headerrQcCs0|dkr|jd7_n||_|jd;_dSrS)rnrUr"r"r#_set_next_compressed_pktnr/sz2NetworkBrokerCompressed._set_next_compressed_pktnrrWcsbt|}tdt|ddtd|jtdt|dd|}t|||IdHS)z1Compress packet and write it to the comm channel.rGrrHrdN)zlibcompressrLrfrJrnrmr^)r,r'r(rDcompressed_pktrpr"r#r^7s  z"NetworkBrokerCompressed._write_pktr&c s||||td|||j}t|ttkrd}t t|tD]4}| |||||tIdH||t7}qP| ||||dIdHnrt|t kr| |||IdHnPt ||t dt|ddt d|jt dddd|IdHdS)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. r`rNrGrHrd)rVryr9joinrtr@rJrrrer^rrmrLrfrn) r,r'r(r)r*r+Z payload_preprgrhrpr"r#r-Bs6      zNetworkBrokerCompressed.write)r/rwrc stttj||dIdH}d}|t|krtd|||tddd}t|t||krtj|t dIdH}| |\}|_ }tj||dIdH}||dkr|nt|7}|j |||t||t|7}q$dS)z&Handle reading of a compressed packet.riNrrGrrI)r9rz decompressrmrbrJrLrMrrrPrnrors) r,r/rwrDrgrNrjrxr|rpr"r#_read_compressed_pktys:  z,NetworkBrokerCompressed._read_compressed_pktr.c s|jszftj|tdIdH}||\}|_}|dkrX|jtj||dIdHn|||IdHWn8tk r}zt d|t |fd|W5d}~XYnX|jsdS|j }|d|_ |S)z{Receive `one` or `several` packets from the MySQL server, enqueue them, and return the packet at the head. riNrrXrYrH) rormrbrrPrnrsrr\rr$popleftr@)r,r/r(rjrwrxrrDrpr"r#r0s8    zNetworkBrokerCompressed.read)N)NN)r1r2r3r4rBrkr5r6r rtrrPrryrr r^r-r7r8rr9r0 __classcell__r"r"rpr#rls. 8(rlc @seZdZdZddddZeedddZee ddd d Z ddd d Z e dd dZ eddddZddddZejddddZd eeeeeddddZedddZdddddggfeeeeeeee ee eeeeeeejdddZdS)! 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_connectedrAr"r"r#rBs zMySQLSocket.__init__cCs|jS)zSocket location.)rrAr"r"r#r(szMySQLSocket.addresskwargsrcsdS)zOpen the socket.Nr"r,rr"r"r#rszMySQLSocket.open_connectioncsB|jr8|j|jjdk r(|jj|jIdHd|_dS)zClose the connection.NF)rclose transportabortZ wait_closedrrAr"r"r#close_connections    zMySQLSocket.close_connectioncCs|jS)zCheck if the socket is connected. Return: bool: Returns `True` if the socket is connected to MySQL server. )rrAr"r"r# is_connectedszMySQLSocket.is_connected)timeoutrcCs ||_dS)zSet the connection timeout.N)r)r,rr"r"r#set_connection_timeoutsz"MySQLSocket.set_connection_timeoutcCs t|_dS)zIEnable network layer where transactions are made with compressed packets.N)rlrrAr"r"r#switch_to_compressed_modesz%MySQLSocket.switch_to_compressed_mode) ssl_contextrcsD|jdk st|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)rAssertionErrorrZget_extra_infofamilyrZ start_tls)r,rrr"r"r# switch_to_ssls  zMySQLSocket.switch_to_ssl)r)r*r+rcs$|jj|j|j|||dIdHdS)z!Send packets to the MySQL server.)r*r+N)rr-rr()r,r)r*r+r"r"r#r-szMySQLSocket.writecs|j|j|jIdHS)z#Read packets from the MySQL server.N)rr0rr(rAr"r"r#r0&szMySQLSocket.readF)ssl_cassl_certssl_keyssl_verify_certssl_verify_identity tls_versionstls_cipher_suitesrc Cs|jstddtdkr tdzt|r|jdd|d}t|} t| } |dkrd |krn| jtjO_d |kr| jtj O_d |kr| jtj O_nt } || _ |rtj | _n|rtj| _ntj| _| |r*z| |Wn:ttjfk r(} ztd | | W5d} ~ XYnX|r|z| ||Wn:ttjfk rz} ztd | | W5d} ~ XYnX|r| d|| WStk r} ztd| W5d} ~ XYn>tttjtjfk r} ztt| | W5d} ~ XYnXdS)zBuild a SSLContext.ir[Nz&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_context*s^      ""zMySQLSocket.build_ssl_context)NN)r1r2r3r4rBpropertyr r(r r rrboolrr6rrrrrr5rr-r9r0r rr"r"r"r#rsJ     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)rmrB_host_portZ _force_ipv6r)r,rrrrpr"r#rBzs  zMySQLTcpSocket.__init__Nrcs0tf|j|jd|IdH\|_|_d|_dS)zOpen TCP/IP connection.)rrNT)rrrrrrrr"r"r#rszMySQLTcpSocket.open_connection)rrF) r1r2r3r4r r6rrBr rrr"r"rpr#rqs  cs8eZdZdZd edfdd Zedddd ZZS) rzbMySQL socket class using UNIX sockets. Args: unix_socket: UNIX socket file path. /tmp/mysql.sock) unix_socketcst||_dS)N)rmrBr)r,rrpr"r#rBs zMySQLUnixSocket.__init__Nrcs.tjfd|ji|IdH\|_|_d|_dS)zOpen UNIX socket connection.pathNT)r7Zopen_unix_connectionrrrrrr"r"r#rs zMySQLUnixSocket.open_connection)r) r1r2r3r4r rBr rrr"r"rpr#rs)-r4__all__r7rLrzrPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSr ImportErrorabcrr collectionsr typingr r r rrerrorsrrrrnetworkrrrrutilsrrr\r r$r%r:rlrrrr"r"r"r#s8   GjJ)