o G)fm@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|S|jd|jS)z`Reformat the IOError error message. This function reformats the IOError error message.  )errnostrstrerror)rrT/var/www/html/flask-app/venv/lib/python3.12/site-packages/mysql/connector/network.py _strioerrorEs r c@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.rNr9r;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)sendallIOErrorrr AttributeError)r(r"r#rBrrrr _send_pkts  zNetworkBrokerPlain._send_pktrsizecCsVt|}t|}|r)|||}|dkr|dkrtdd||d}||8}|s |S)z(Read `size` bytes from the comm channel.rirFN)r4 memoryview recv_intor)r(r"rKrBpkt_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. Nr)rer;rrr_set_next_compressed_pktnrrAz2NetworkBrokerCompressed._set_next_compressed_pktnrr"r#rBcs\t|}tdt|ddtd|jtdt|dd|}t|||S)z1Compress packet and write it to the comm channel.rSrrTrR)zlibcompressrXrYrUrerdrJ)r(r"r#rBcompressed_pktrgrrrJs  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. NrrSrTrR)r@r:rmrer4joinrlrUrVr^rWrJMIN_COMPRESS_LENGTHrdrXrY) r(r"r#r$r%r& payload_preprZr[rgrrr)&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\rrSrr]rTrrN) rdrPr4rn decompressrUrXr_r^COMPRESSED_PACKET_HEADER_LENGTHrerfrj) r(r"rurvrprBrZpllr`rgrr_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\rSrrTr]rrrCrDN) rfrdrPrxrXr_rerzrHrr popleftr:)r(r"r#r`rurvrrBrgrrr+s2   zNetworkBrokerCompressed.recvrbr,)r-r.r/r0r< staticmethodr2r3rrlrmr1rrJrr)rzr4r+ __classcell__rrrgrrcs@  =$0rcc@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_hostr5 _netbrokerr;rrrr<s zMySQLSocket.__init__cCs t|_dS)zIEnable network layer where transactions are made with compressed packets.N)rcrr;rrrswitch_to_compressed_modes z%MySQLSocket.switch_to_compressed_modec Cs8z|jtj|jWdSttfyYdSw)z'Shut down the socket before closing it.N)r"shutdownr1 SHUT_RDWRcloserIOSErrorr;rrrrs zMySQLSocket.shutdownc Cs*z|jWdSttfyYdSw)zClose the socket.N)r"rrIrr;rrrclose_connections zMySQLSocket.close_connectioncCs |dSN)rr;rrr__del__s zMySQLSocket.__del__timeoutcCs ||_|jr|j|dSdS)zSet the connection timeout.N)rr" 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_hostnamerCrD)r"familyrsslr wrap_socket NameErrorSSLErrorrHrr#r CertificateErrorrNotImplementedError)r(rrrrrr switch_to_ssls0  zMySQLSocket.switch_to_sslFssl_cassl_certssl_keyssl_verify_certssl_verify_identity tls_versionstls_cipher_suitesc Cs|jstddtdurtd|durg}|durg}z|r[|jdd|d}t|} t| } |dkrZd |vrB| jtjO_d |vrN| jtj O_d |vrZ| jtj O_nt } || _ |ritj | _n |rptj| _ntj| _| |rz| |Wnttjfy} ztd | | d} ~ ww|rz| ||Wnttjfy} ztd | | d} ~ ww|r| 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. irFNrT)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_locationsrHrload_cert_chain set_ciphersrrrrrr) r(rrrrrrr tls_version ssl_protocolcontextrrrrbuild_ssl_contextsv       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)dszMySQLSocket.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.addressrb)NNNFFNNr,)r-r.r/r0r<rrrrrr3rrrrboolrrr2r)r4r+rrpropertyr#rrrrr~sd    &   a  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)rdr<r_address)r(rrgrrr<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)NirD)r1AF_UNIX SOCK_STREAMr"rrconnectrrHrr#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)rrb) r-r.r/r0rr<rr#rrrr}rrrgrrs 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)rdr<r server_portr_familyr)r(rrrrgrrr<s zMySQLTCPSocket.__init__cCrrrr;rrrr#rzMySQLTCPSocket.addressc Csbd}zHt|j|jdtjtj}|D]}|jr"|dtjkr"|}n |dtjkr-|}nq|jr?|ddur?t d|j|ddurI|d}Wnt yb}z t d|j t |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 irD)r1 getaddrinforrrSOL_TCPrAF_INET6AF_INETrrHr#r rr"rrrrrr) r(addrinfo addrinfosinforsocktypeprotor[sockaddrrrrrsf   zMySQLTCPSocket.open_connection)rrFrb) r-r.r/r0rr3rr<rr#rr}rrrgrrs" r)-r0r1rXrrnabcrr collectionsrtypingrrrrr r rPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSrhasattrrTLS_V1_3_SUPPORTED ImportErrorerrorsrrrrrsrVr^rxrHrr r!r5rcr~rrrrrrsD   GaAU)