o g1v@sXdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZzddlZejejejejdZeedoGejZWn eyVdZdZYnwd d lmZmZmZmZm Z m!Z!m"Z"d 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)ConnectionTimeoutErrorInterfaceErrorNotSupportedErrorOperationalErrorProgrammingErrorReadTimeoutErrorWriteTimeoutError2ierrreturncCs"|jst|Sd|jd|jS)z`Reformat the IOError error message. This function reformats the IOError error message. zErrno z: )errnostrstrerror)rr Q/var/www/html/api-tag/env/lib/python3.10/site-packages/mysql/connector/network.py _strioerrorJs"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(r r r!sendjzNetworkBroker.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%r r r!recvr,zNetworkBroker.recvNN) __name__ __module__ __qualname____doc__rsocketrbytesrintr+ bytearrayr-r r r r!r#Rs( 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*r r r!__init__s zNetworkBrokerPlain.__init__cC|jdd|_dSzIncrement packet id.rNr;r=r r r!_set_next_pktnrz"NetworkBrokerPlain._set_next_pktnrr$r%pktc Csz||WdStjtfy}ztdd|d}~wty3}z td|t|fd|d}~wtyD}ztdd|d}~ww)z!Write packet to the comm channel. rNrvaluesi) sendallr3timeout TimeoutErrorrIOErrorrr"AttributeError)r*r$r%rDrr r r! _send_pkts    zNetworkBrokerPlain._send_pktrsizecCsVt|}t|}|r)|||}|dkr|dkrtdd||d}||8}|s |S)z(Read `size` bytes from the comm channel.rirFN)r6 memoryview recv_intor)r*r$rPrDpkt_viewreadr r r! _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. NrrBr3rr4rOr5r6rUrr+r-r r r r!r7s,    ,r7c seZdZdZdfdd Zedededeefdd Z dd d Z d e j d e deddffdd Z  dd e j d e dedeedeeddf fdd Zd e j dededdffdd Zd e j d e deffdd ZZS)NetworkBrokerCompressedr8rNcstd|_t|_dSr9)superr>_compressed_pktnrr _queue_readr= __class__r r!r>s  z NetworkBrokerCompressed.__init__r&pktnrcCsg}t|tkr8d}tt|tD]}|dtd||||t|dd}|t7}q||d}|tdt|ddtd|||S) z2Prepare a payload for sending to the MySQL server.rrVrWrrANrXrY)rZr[r\appendr]r^)r&ropktsr_r`r r r!_prepare_packetss$     &z(NetworkBrokerCompressed._prepare_packetscCr?r@)rkr=r r r!_set_next_compressed_pktnr rCz2NetworkBrokerCompressed._set_next_compressed_pktnrr$r%rDcs\t|}tdt|ddtd|jtdt|dd|}t|||S)z1Compress packet and write it to the comm channel.rXrrYrW)zlibcompressr]r^rZrkrjrO)r*r$r%rDcompressed_pktrmr r!rO$s  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. NrrXrYrW)rBr<rsrkr6joinrrrZr[rdr\rOMIN_COMPRESS_LENGTHrjr]r^) r*r$r%r&r'r( payload_prepr_r`rmr r!r+/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.rarrXrrbrYrrN) rjrUr6rt decompressrZr]rerdCOMPRESSED_PACKET_HEADER_LENGTHrkrlrp) r*r$r{r|rvrDr_pllrfrmr r!_recv_compressed_pktls>   z,NetworkBrokerCompressed._recv_compressed_pktc s|jscz4tj|td}td|dddd|dtd|dddd}|_}||||Wn+tj t fyL}zt dd |d }~wt yb}z t d |t|fd |d }~ww|jshd S|j}|d|_|S) z{Receive `one` or `several` packets from the MySQL server, enqueue them, and return the packet at the head. rarXrrYrbrrrErFNrGrH)rlrjrUr~r]rerkrr3rKrLrrMrr"popleftr<)r*r$r%rfr{r|rrDrmr r!r-s8    zNetworkBrokerCompressed.recvrhr.)r/r0r1r2r> staticmethodr4r5rrrrsr3rrOrr+rr6r- __classcell__r r rmr!ris@  =$0ric@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 d"e e ddf d#d$Zd/d%e e defd&d'Zed,d(d)Zeede fd*d+ZdS)0 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_hostr7 _netbrokerr=r r r!r>s zMySQLSocket.__init__cCs t|_dS)zIEnable network layer where transactions are made with compressed packets.N)rirr=r r r!switch_to_compressed_modes z%MySQLSocket.switch_to_compressed_modec Cs8z|jtj|jWdSttfyYdSw)z'Shut down the socket before closing it.N)r$shutdownr3 SHUT_RDWRcloserNOSErrorr=r r r!rs zMySQLSocket.shutdownc Cs*z|jWdSttfyYdSw)zClose the socket.N)r$rrNrr=r r r!close_connections zMySQLSocket.close_connectioncCs |dSN)rr=r r r!__del__s zMySQLSocket.__del__rKcCs ||_|jr|j|dSdS)zSet the connection timeout.N)rr$ settimeout)r*rKr r r!set_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_hostnamerGrH)r$familyrsslr wrap_socket NameErrorSSLErrorrMrr%r"CertificateErrorrNotImplementedError)r*rrrr r r! 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. NirFrT)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_locationsrMrload_cert_chain set_ciphersrxrrrr) r*rrrrrrr tls_version ssl_protocolcontextrr r r!build_ssl_contextsx        zMySQLSocket.build_ssl_contextr&r'r( write_timeoutc Cspz|js|jdur|j|rt|ndWnty(}zWYd}~nd}~ww|jj|j|j|||ddS)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`. N)r'r()rr$rfloatrrr+r%)r*r&r'r(rr`r r r!r+ss"   zMySQLSocket.send read_timeoutc Csdz|js|jdur|j|rt|ndWnty(}zWYd}~nd}~ww|j|j|jS)z.Get packet from the MySQL server comm channel.N)rr$rrrrr-r%)r*rr`r r r!r-s  zMySQLSocket.recvcCr))zOpen the socket.Nr r=r r r!open_connectionr,zMySQLSocket.open_connectioncCr))zGet the location of the socket.Nr r=r r r!r%r,zMySQLSocket.addressrh)NNNFFNN)NNNr)r/r0r1r2r>rrrrrr5rrrrboolrrr4r+r6r-rrpropertyr%r r r r!rsj    &   e   rcs^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)rjr>r_address)r*rrmr r!r>s  zMySQLUnixSocket.__init__cC|jSrrr=r r r!r%zMySQLUnixSocket.addressc Cszttjtj|_|j|j|j|jWdStjt fy5}z t d|j t |fd|d}~wt yL}z td|j t |fd|d}~wty^}ztt||d}~ww)NrH)r3AF_UNIX SOCK_STREAMr$rrconnectrrKrLrr%r"rMr Exceptionr)r*rr r r!rs8zMySQLUnixSocket.open_connectionargskwargscOstdtdS)zSwitch the socket to use SSL.z2SSL is disabled when using unix socket connectionsN)warningswarnWarning)r*rrr r r!rszMySQLUnixSocket.switch_to_ssl)rrh) r/r0r1r2rr>rr%rrrrr r rmr!rs 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)rjr>r server_portr_familyr)r*rrrrmr r!r>s zMySQLTCPSocket.__init__cCrrrr=r r r!r%rzMySQLTCPSocket.addressc Csd}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 |WdStjtfy}ztd|j|jt |fd|d}~wt 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 irHr)r3 getaddrinforrrSOL_TCPrAF_INET6AF_INETrrMr"rr$rrrrKrLrrrr) r*addrinfo addrinfosinforsocktypeprotor`sockaddrr r r!rs~   zMySQLTCPSocket.open_connection)rrFrh) r/r0r1r2rr5rr>rr%rrr r rmr!rs" r)0r2r3r]rrtabcrr collectionsrtypingrrrrr r rPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSrhasattrrTLS_V1_3_SUPPORTED ImportErrorerrorsrrrrrrrryr[rdr~rMrr"r#r7rirrrr r r r!sD   $ GeCj1