U jm@sLdZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZz2ddlZejejejejdZeedoejZWnek rdZdZYnXd d lmZmZmZmZd Z d Z!d Z"dZ#e$e%dddZ&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)InterfaceErrorNotSupportedErrorOperationalErrorProgrammingError2i)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/network.py _strioerrorEsr c @sPeZdZdZed ejeeee ee ddddZ eejee dddZ 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_numberrcCsdS)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'rrrsendes zNetworkBroker.sendr#r$rcCsdS)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$rrrrecvszNetworkBroker.recv)NN) __name__ __module__ __qualname____doc__rsocketrbytesrintr) bytearrayr+rrrrr!Msr!c@seZdZdZddddZddddZejeeddd d Z deje e d d dZ dejeee e e e ddddZejee dddZdS)NetworkBrokerPlain,Broker class for MySQL socket communication.NrcCs d|_dSN_pktnrr(rrr__init__szNetworkBrokerPlain.__init__cCs|jdd|_dSzIncrement packet id.rNr9r;rrr_set_next_pktnrsz"NetworkBrokerPlain._set_next_pktnrr#r$pktrc Csxz||Wndtk rF}ztd|t|fd|W5d}~XYn.tk rr}ztdd|W5d}~XYnXdS)z!Write packet to the comm channel.rvaluesNir)sendallIOErrorrr AttributeError)r(r#r$rArrrr _send_pkts zNetworkBrokerPlain._send_pktr)r#sizercCsTt|}t|}|rP|||}|dkr:|dkr:tdd||d}||8}q|S)z(Read `size` bytes from the comm channel.rirEN)r3 memoryview recv_intor)r(r#rJrAZpkt_viewreadrrr _recv_chunks    zNetworkBrokerPlain._recv_chunkr"c Cs|dkr|n||_t|tkrd}tt|tD]@}|||dtd|j|||t||t7}q8||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}|dtd||||t|dd}|t7}q$||d}|tdt|ddtd|||S) z2Prepare a payload for sending to the MySQL server.rrOrPrr>NrQrR)rSrTrUappendrVrW)r%reZpktsrXrYrrr_prepare_packetss$     &z(NetworkBrokerCompressed._prepare_packetscCs|jdd|_dSr=)rar;rrr_set_next_compressed_pktnrsz2NetworkBrokerCompressed._set_next_compressed_pktnrr@cs\t|}tdt|ddtd|jtdt|dd|}t|||S)z1Compress packet and write it to the comm channel.rQrrRrP)zlibcompressrVrWrSrar`rI)r(r#r$rAcompressed_pktrcrrrIs  z!NetworkBrokerCompressed._send_pktr"c s|dkr|n||_|dkr*|n||_td|||j}t|tt krd}t t|tD].}| |||||t||t7}ql| ||||dnft|t kr| |||nJt ||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. NrrQrRrP)r?r:rhrar3joinrgrSrTr\rUrIMIN_COMPRESS_LENGTHr`rVrW) r(r#r$r%r&r'Z payload_preprXrYrcrrr)&s>     zNetworkBrokerCompressed.send)r#compressed_plluncompressed_pllrc s"tj||d}|dkr|n tt|}d}|t|krtd|||tddd}t|t||krtj|t d}td|dddd|dtd|dddd}|_ }tj||d}||dkr|nt|7}|j |||t||t|7}q.d S) z&Handle reading of a compressed packet.rZrrQrr[rRrrN) r`rNr3ri decompressrSrVr]r\COMPRESSED_PACKET_HEADER_LENGTHrarbrf) r(r#rorprkrArXZpllr^rcrr_recv_compressed_pktcs> z,NetworkBrokerCompressed._recv_compressed_pktr*c s|jszhtj|td}td|dddd|dtd|dddd}|_}||||Wn8tk r}zt d|t |fd |W5d }~XYnX|jsd S|j }|d|_ |S) z{Receive `one` or `several` packets from the MySQL server, enqueue them, and return the packet at the head. rZrQrrRr[rrrBrCN) rbr`rNrrrVr]rarsrGrr popleftr:)r(r#r$r^rorprrArcrrr+s0   zNetworkBrokerCompressed.recv)NN)r,r-r.r/r< staticmethodr1r2rrgrhr0rrIrr)rsr3r+ __classcell__rrrcrr_s,>0r_c @seZdZdZddddZddddZdddd Zddd d Zddd d Ze e ddddZ e e ddddZd!e e e e e e e ee ee ee e ee e dddZd"ee e e e ddddZedddZeddddZeee ddd ZdS)# MySQLSocketzMySQL socket communication interface. Examples: Subclasses: network.MySQLTCPSocket and network.MySQLUnixSocket. Nr6cCsd|_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<szMySQLSocket.__init__cCs t|_dS)zIEnable network layer where transactions are made with compressed packets.N)r_rzr;rrrswitch_to_compressed_modesz%MySQLSocket.switch_to_compressed_modec Cs:z|jtj|jWnttfk r4YnXdS)z'Shut down the socket before closing it.N)r#shutdownr0 SHUT_RDWRcloserHOSErrorr;rrrr|s zMySQLSocket.shutdownc Cs,z|jWnttfk r&YnXdS)zClose the socket.N)r#r~rHrr;rrrclose_connectionszMySQLSocket.close_connectioncCs |dSN)r|r;rrr__del__szMySQLSocket.__del__)timeoutrcCs||_|jr|j|dS)zSet the connection timeout.N)rxr# settimeout)r(rrrrset_connection_timeoutsz"MySQLSocket.set_connection_timeout) ssl_contexthostrc Cs|jdk st|jjdkr"tdtdkr2tdz|j|j|d|_Wntk rt}ztd|W5d}~XYntjt fk r}zt d|j t |fd|W5d}~XYnbtj k r}zt t||W5d}~XYn2tk r}zt t||W5d}~XYnXdS)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#AssertionErrorfamilyrsslr wrap_socket NameErrorSSLErrorrGrr$r CertificateErrorrNotImplementedError)r(rrrrrr switch_to_ssls(  zMySQLSocket.switch_to_sslF)ssl_cassl_certssl_keyssl_verify_certssl_verify_identity tls_versionstls_cipher_suitesrc Cs|jstddtdkr td|dkr,g}|dkr8g}zt|r|jdd|d}t|} t| } |dkrd |kr| jtjO_d |kr| jtj O_d |kr| jtj O_nt } || _ |rtj | _n|rtj| _ntj| _| |rBz| |Wn:ttjfk r@} ztd | | W5d} ~ XYnX|rz| ||Wn:ttjfk r} ztd | | W5d} ~ XYnX|r| d|| WStk r} ztd| W5d} ~ XYn>tttjtjfk r} ztt| | W5d} ~ XYnXdS)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. irENrT)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_ciphersrmrrrr) r(rrrrrrrZ tls_versionZ ssl_protocolcontextrrrrbuild_ssl_contextsf     ""zMySQLSocket.build_ssl_context)r%r&r'rcCs|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')rzr)r#r$)r(r%r&r'rrrr)dszMySQLSocket.sendcCs|j|j|jS)z.Get packet from the MySQL server comm channel.)rzr+r#r$r;rrrr+{szMySQLSocket.recvcCsdS)zOpen the socket.Nrr;rrropen_connectionszMySQLSocket.open_connectioncCsdS)zGet the location of the socket.Nrr;rrrr$szMySQLSocket.address)NNNFFNN)NN)r,r-r.r/r<r{r|rrrr2rrrrboolrrr1r)r3r+rrpropertyr$rrrrrwsN &   a rwcs\eZdZdZdeddfdd Zeeddd Zddd d Ze e dd d dZ Z S)MySQLUnixSocketzpMySQL socket class using UNIX sockets. Opens a connection through the UNIX socket of the MySQL Server. /tmp/mysql.sockN) unix_socketrcst||_||_dSr)r`r<r_address)r(rrcrrr<s zMySQLUnixSocket.__init__r6cCs|jSrrr;rrrr$szMySQLUnixSocket.addressc Csz2ttjtj|_|j|j|j|jWnhtk rl}zt d|j t |fd|W5d}~XYn0t k r}zt t ||W5d}~XYnXdS)NirC)r0AF_UNIX SOCK_STREAMr#rrxconnectrrGrr$r Exceptionr)r(rrrrrs  zMySQLUnixSocket.open_connection)argskwargsrcOstdtdS)zSwitch the socket to use SSL.z2SSL is disabled when using unix socket connectionsN)warningswarnWarning)r(rrrrrrszMySQLUnixSocket.switch_to_ssl)r) r,r-r.r/rr<rr$rrrrvrrrcrrsrcsNeZdZdZdeeeddfdd Zeed d d Z dd d d Z Z S)MySQLTCPSocketzYMySQL socket class using TCP/IP. Opens a TCP/IP connection to the MySQL Server. 127.0.0.1 FN)rport force_ipv6rcs6t||_||_||_d|_|d||_dS)Nrr)r`r<ry server_portr_familyr)r(rrrrcrrr<s  zMySQLTCPSocket.__init__r6cCs|jSrrr;rrrr$szMySQLTCPSocket.addressc Csd}zt|j|jdtjtj}|D]6}|jrD|dtjkrD|}q\|dtjkr$|}q\q$|jr~|ddkr~t d|j|ddkr|d}Wn:t k r}zt d|j t |fd|W5d}~XYnX|\|_ }}}}z0t|j |||_|j|j|j|Wnpt k rP}z t d|j|jt |fd|W5d}~XYn2tk r}ztt||W5d}~XYnXdS)z/Open the TCP/IP connection to the MySQL server.)NNNNNrNzNo IPv6 address found for irC)r0 getaddrinforyrrSOL_TCPrAF_INET6AF_INETrrGr$r rr#rrxrrrr) r(ZaddrinfoZ addrinfosinforsocktypeprotorYZsockaddrrrrrsZ     zMySQLTCPSocket.open_connection)rrF) r,r-r.r/rr2rr<rr$rrvrrrcrrs r)-r/r0rVrriabcrr collectionsrtypingrrrrr r rPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2 PROTOCOL_TLSrhasattrrZTLS_V1_3_SUPPORTED ImportErrorerrorsrrrrrnrTr\rrrGrr r!r4r_rwrrrrrrs@   GaAU)