o F=2gn@sLdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZzddlZejejejejdZeedoGejZWn eyVdZdZYnwd d lmZmZmZmZd Z d Z!d Z"dZ#de$de%fddZ&GdddeZ'Gddde'Z(Gddde(Z)GdddeZ*Gddde*Z+Gddde*Z,dS)zGModule implementing low-level socket communication with MySQL servers. N)ABCabstractmethod)deque)AnyDequeListOptionalTupleUnion)TLSv1TLSv1.1TLSv1.2TLSv1.3 HAS_TLSv1_3F)InterfaceErrorNotSupportedErrorOperationalErrorProgrammingError2ierrreturncCs"|jst|Sd|jd|jS)z`Reformat the IOError error message. This function reformats the IOError error message. zErrno z: )errnostrstrerror)rrf/var/www/mastermindingenieria.com/MONITOR/venv/lib/python3.10/site-packages/mysql/connector/network.py _strioerrorEs"rc@sbeZdZdZe  d dejdededee dee ddf d d Z edejdede fd d 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_numberrcCdS)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%rrrsendezNetworkBroker.sendcCr&)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"rrrrecvr)zNetworkBroker.recvNN) __name__ __module__ __qualname____doc__rsocketrbytesrintr( bytearrayr*rrrrr Ms( r c @seZdZdZdddZdddZdejd ed eddfd d Z ddejde de fddZ  ddejd edede e de e ddf ddZdejd ede fddZdS)NetworkBrokerPlain,Broker class for MySQL socket communication.rNcCs d|_dSN_pktnrr'rrr__init__s zNetworkBrokerPlain.__init__cC|jdd|_dSzIncrement packet id.rNr8r:rrr_set_next_pktnrz"NetworkBrokerPlain._set_next_pktnrr!r"pktc Csbz||WdSty}z td|t|fd|d}~wty0}ztdd|d}~ww)z!Write packet to the comm channel.rvaluesNir)sendallIOErrorrrAttributeError)r'r!r"rArrrr _send_pkts  zNetworkBrokerPlain._send_pktrsizecCsVt|}t|}|r)|||}|dkr|dkrtdd||d}||8}|s |S)z(Read `size` bytes from the comm channel.rirEN)r3 memoryview recv_intor)r'r!rJrApkt_viewreadrrr _recv_chunks   zNetworkBrokerPlain._recv_chunkr#r$r%c Cs|dur |n||_t|tkrCd}tt|tD] }|||dtd|j|||t||t7}q||d}|||tdt|ddtd|j|dS)zSend payload to the MySQL server. If provided a payload whose length is greater than `MAX_PAYLOAD_LENGTH`, it is broken down into packets. NrNrRrS)rTrUrVappendrWrX)r#rhpktsrYrZrrr_prepare_packetss$     &z(NetworkBrokerCompressed._prepare_packetscCr<r=)rdr:rrr_set_next_compressed_pktnrr@z2NetworkBrokerCompressed._set_next_compressed_pktnrr!r"rAcs\t|}tdt|ddtd|jtdt|dd|}t|||S)z1Compress packet and write it to the comm channel.rRrrSrQ)zlibcompressrWrXrTrdrcrI)r'r!r"rAcompressed_pktrfrrrIs  z!NetworkBrokerCompressed._send_pktr$r%c s"|dur |n||_|dur|n||_td|||j}t|tt kr[d}t t|tD]}| |||||t||t7}q6| ||||ddSt|t krj| |||dSt ||tdt|ddtd|jtdddd|dS)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. NrrRrSrQ)r?r9rlrdr3joinrkrTrUr]rVrIMIN_COMPRESS_LENGTHrcrWrX) r'r!r"r#r$r% payload_preprYrZrfrrr(&s:     zNetworkBrokerCompressed.sendcompressed_plluncompressed_pllc s.tj||d}|dkr|ntt|}d}|t|krtd|||tddd}t|t||krytj|t d}td|dddd|dtd|dddd}|_ }tj||d}||dkrs|nt|7}|j |||t||t|7}|t|ksd Sd S) z&Handle reading of a compressed packet.r[rrRrr\rSrrN) rcrOr3rm decompressrTrWr^r]COMPRESSED_PACKET_HEADER_LENGTHrdreri) r'r!rtrurorArYpllr_rfrr_recv_compressed_pktcs>   z,NetworkBrokerCompressed._recv_compressed_pktc s|jsOz4tj|td}td|dddd|dtd|dddd}|_}||||WntyN}z t d|t |fd |d }~ww|jsTd S|j }|d|_ |S) z{Receive `one` or `several` packets from the MySQL server, enqueue them, and return the packet at the head. r[rRrrSr\rrrBrCN) rercrOrwrWr^rdryrGrrpopleftr9)r'r!r"r_rtrurrArfrrr*s2   zNetworkBrokerCompressed.recvrar+)r,r-r.r/r; staticmethodr1r2rrkrlr0rrIrr(ryr3r* __classcell__rrrfrrbs@  =$0rbc@s&eZdZdZd*ddZd*ddZd*dd Zd*d d Zd*d d Zde e ddfddZ de de ddfddZ       d+de e de e de e de ede ede ee de ee de fddZ  d,ded e e d!e e ddfd"d#Zdefd$d%Zed*d&d'Zeede fd(d)ZdS)- MySQLSocketzMySQL socket communication interface. Examples: Subclasses: network.MySQLTCPSocket and network.MySQLUnixSocket. rNcCsd|_d|_d|_t|_dS)zsNetwork layer where transactions are made with plain (uncompressed) packets is enabled by default. N)r!_connection_timeout server_hostr4 _netbrokerr:rrrr;s zMySQLSocket.__init__cCs t|_dS)zIEnable network layer where transactions are made with compressed packets.N)rbrr:rrrswitch_to_compressed_modes z%MySQLSocket.switch_to_compressed_modec Cs8z|jtj|jWdSttfyYdSw)z'Shut down the socket before closing it.N)r!shutdownr0 SHUT_RDWRcloserHOSErrorr:rrrrs zMySQLSocket.shutdownc Cs*z|jWdSttfyYdSw)zClose the socket.N)r!rrHrr:rrrclose_connections zMySQLSocket.close_connectioncCs |dSN)rr:rrr__del__s zMySQLSocket.__del__timeoutcCs ||_|jr|j|dSdS)zSet the connection timeout.N)r~r! settimeout)r'rrrrset_connection_timeoutsz"MySQLSocket.set_connection_timeout ssl_contexthostc Cs|jdusJ|jjdkrtdtdurtdz |j|j|d|_WdSty6}ztd|d}~wtjtfyP}z t d|j t |fd|d}~wtj yc}zt t ||d}~wtyu}zt t ||d}~ww)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_hostnamerBrC)r!familyrsslr wrap_socket NameErrorSSLErrorrGrr"rCertificateErrorrNotImplementedError)r'rrrrrr switch_to_ssls0  zMySQLSocket.switch_to_sslFssl_cassl_certssl_keyssl_verify_certssl_verify_identity tls_versionstls_cipher_suitesc Csd}|js tddtdurtd|durg}|durg}z|r]|jdd|d}t|} t| } |dkr\d |vrD| jtjO_d |vrP| jtj O_d |vr\| jtj O_nt } || _ |rktj | _n |rrtj| _ntj| _| |rz| |Wnttjfy} ztd | | d} ~ ww|rz| ||Wnttjfy} ztd | | d} ~ ww|r|d kr| d|| WSty} ztd| d} ~ wtttjtjfy} ztt| | d} ~ ww)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. NirErT)reverserrr r 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_locationsrGrload_cert_chain set_ciphersrqrrrr) r'rrrrrrr tls_version ssl_protocolcontextrrrrbuild_ssl_contextsx        zMySQLSocket.build_ssl_contextr#r$r%cCs|jj|j|j|||dS)atSend `payload` to the MySQL server. NOTE: if `payload` is an instance of `bytearray`, then `payload` might be changed by this method - `bytearray` is similar to passing a variable by reference. If you're sure you won't read `payload` after invoking `send()`, then you can use `bytearray.` Otherwise, you must use `bytes`. )r$r%)rr(r!r")r'r#r$r%rrrr(hszMySQLSocket.sendcCs|j|j|jS)z.Get packet from the MySQL server comm channel.)rr*r!r"r:rrrr*szMySQLSocket.recvcCr&)zOpen the socket.Nrr:rrropen_connectionr)zMySQLSocket.open_connectioncCr&)zGet the location of the socket.Nrr:rrrr"r)zMySQLSocket.addressra)NNNFFNNr+)r,r-r.r/r;rrrrrr2rrrrboolrrr1r(r3r*rrpropertyr"rrrrr}sd    &   e  r}cs^eZdZdZddeddffdd Zedefdd Zdd d Zd e d e ddfddZ Z S)MySQLUnixSocketzpMySQL socket class using UNIX sockets. Opens a connection through the UNIX socket of the MySQL Server. /tmp/mysql.sock unix_socketrNcst||_||_dSr)rcr;r_address)r'rrfrrr;s  zMySQLUnixSocket.__init__cC|jSrrr:rrrr"zMySQLUnixSocket.addressc Cszttjtj|_|j|j|j|jWdSty2}z t d|j t |fd|d}~wt yD}zt t ||d}~ww)NirC)r0AF_UNIX SOCK_STREAMr!rr~connectrrGrr"r Exceptionr)r'rrrrrs$zMySQLUnixSocket.open_connectionargskwargscOstdtdS)zSwitch the socket to use SSL.z2SSL is disabled when using unix socket connectionsN)warningswarnWarning)r'rrrrrrszMySQLUnixSocket.switch_to_ssl)rra) r,r-r.r/rr;rr"rrrr|rrrfrrs rc sVeZdZdZ   ddedededd ffd d Zedefd d Z dddZ Z S)MySQLTCPSocketzYMySQL socket class using TCP/IP. Opens a TCP/IP connection to the MySQL Server. 127.0.0.1 Frport force_ipv6rNcs6t||_||_||_d|_|d||_dS)Nrr)rcr;r server_portr_familyr)r'rrrrfrrr;s zMySQLTCPSocket.__init__cCrrrr:rrrr"rzMySQLTCPSocket.addressc Csfd}zHt|j|jdtjtj}|D]}|jr"|dtjkr"|}n |dtjkr-|}nq|jr?|ddur?t d|j|ddurI|d}Wnt yd}zt d|j|jt |fd|d}~ww|\|_ }}}}zt|j |||_ |j |j|j |WdSt y}zt d|j|jt |fd|d}~wty}ztt||d}~ww)z/Open the TCP/IP connection to the MySQL server.)NNNNNrNzNo IPv6 address found for irC)r0 getaddrinforrrSOL_TCPrAF_INET6AF_INETrrGrrr!rr~rrrr) r'addrinfo addrinfosinforsocktypeprotorZsockaddrrrrrsh   zMySQLTCPSocket.open_connection)rrFra) r,r-r.r/rr2rr;rr"rr|rrrfrrs" r)-r/r0rWrrmabcrr collectionsrtypingrrrrr r rPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSrhasattrrTLS_V1_3_SUPPORTED ImportErrorerrorsrrrrrrrUr]rwrGrrr r4rbr}rrrrrrsD   GaAY)