U ec R@sdZddlZddlZddlZddlZddlZddlmZzHddlZej ej ej dZ e edrdejneje d<e edo|ejZWnek rdZYnXd d lmZd d lmZmZmZd d lmZd dZddZGdddZGdddeZGdddeZdS)zGModule implementing low-level socket communication with MySQL servers. N)deque)TLSv1TLSv1.1TLSv1.2 PROTOCOL_TLSTLSv1.3 HAS_TLSv1_3F)MAX_PACKET_LENGTH)InterfaceErrorNotSupportedErrorOperationalError)init_bytearraycCs |jst|S|jd|jS)z_Reformat the IOError error message This function reformats the IOError error message.  )errnostrstrerror)errr;/tmp/pip-unpacked-wheel-ix8dg7li/mysql/connector/network.py _strioerror@srcCsg}t|}t}||krX|dtd||d|||d}t|}|d}q|td|ddtd|||S)z0Prepare a packet for sending to the MySQL serverz |jtj|j|`Wnttfk r8YnXdS)z&Shut down the socket before closing itN)r'shutdownsocket SHUT_RDWRcloser+AttributeErrorOSErrorr-rrrr7s  zBaseMySQLSocket.shutdownc Cs0z|j|`Wnttfk r*YnXdS)zClose the socketN)r'r:r+r;r<r-rrrclose_connections  z BaseMySQLSocket.close_connectioncCs |dSN)r7r-rrr__del__szBaseMySQLSocket.__del__Nc Cs|}|dkr|n||_t||j}|D]~}z|j|Wq,tk r|}ztd|t|fd|W5d}~XYq,t k r}ztdd|W5d}~XYq,Xq,dS)z Send packets to the MySQL serverNrvaluesr) r1r)r$r'sendallIOErrorr r6rr;)r.r packet_numbercompressed_packet_number_packetspacketrrrr send_plains"  zBaseMySQLSocket.send_plainc Cs|dkr|n||_|dkr*|n||_|j}t|}g}t}||krt||}d|} ~t | dd} t dt| ddt d|jd} | | | | dd} t| }|||kr>t | d|} t dt| ddt d|jd } | | | | |d} t| }|q| rt | } t dt| ddt d|jt d|dd} | | | ~ nt d|ddt d||} t| }|d kr t | } | t dt| ddt d|jt d|dd| nDt d|ddt d|jt dddd} | | | |D]} z|j | Wnltk r}ztd |t|fd |W5d}~XYn0tk r}ztd d|W5d}~XYnXqhdS)z+Send compressed packets to the MySQL serverN@rrrrs@r2r@rArCrD)r1r)r2r*rr r$joinzlibcompressrrrr'rErFr r6rr;)r.rrGrHr r"Zzpktsr#r!ZtmpbufZzbufheaderpktZ zip_packetrrrrsend_compresseds             $    zBaseMySQLSocket.send_compressedc Csztd}d}|dkrF|jd|}|s4tdd||7}t|}q|d|_td|dddd}|}|t|t |}|dd }|r|j ||}|dkr|dkrtdd||d }||8}q|WSt k r}zt d | t|fd |W5d }~XYnXd S) z%Receive packets from the MySQL serverrMrrDrrNr@rA) bytearrayr'recvr rr)runpackextend memoryview recv_intorFr r6r) r.rKZ packet_lenchunkZ payload_lenrestZ packet_viewreadrrrr recv_plains:       zBaseMySQLSocket.recv_plaincCsN|rJtd|dddd}|j|d|d||dd}qdS)zSplit compressed payloadrrrrXrVN)rr[r+r)r.Z packet_bunchpayload_lengthrrr_split_zipped_payloadsz%BaseMySQLSocket._split_zipped_payloadc CsFz|j}|d|_|WStk r.YnXtd}g}z`|jd}|rrt|dkrr||7}|jd}qL|rt|dkrtddt d|ddd d}|d|_ t d|d dd d}t |}t||kr|j|t|}|s tdd||}q|dkrB| ||j}|d|_|WS|||f|d kr^qt d}|jd}|rrt|dkrr||7}|jd}qrqrWn>tk r} ztd |t| fd | W5d} ~ XYnXt d} |D]\}} | t| 7} q| | ~ z|j}|d|_|WStk r@YnXdS)z0Receive compressed packets from the MySQL serverrrMr rWrDrrrXrVrNr@rAN)r+popleftr) IndexErrorrYr'rZrr rr[r*rrdrrFr r6rrQ decompress) r.rTrSrJZabyteZzip_payload_lengthrcZ zip_payloadr_rtmppayloadrrrrecv_compressed"sp                  zBaseMySQLSocket.recv_compressedcCs||_|jr|j|dS)zSet the connection timeoutN)r(r' settimeout)r.timeoutrrrset_connection_timeoutmsz&BaseMySQLSocket.set_connection_timeoutFc Cs|jstddz|r tj}n|r,tj}ntj}|dks>|sRt} |sd| _n|jdd|d} t s| dkrt |d kr|d } t | } t | } | dkrd |kr| j tjO_ d |kr| j tjO_ d |kr| j tjO_ d| _|| _| |rVz| |WnDttjfk rT} z|jtd | | W5d} ~ XYnX|rz| ||WnDttjfk r} z|jtd| | W5d} ~ XYnX|r| |t|dr| j|j|jd|_n| |j|_|rd| _|jg} tjdkrJ|jdkrJddg} t|j}| |dg|d d}g}| D]\}zt!|j"|Wn4tj#k r} z|$t%| W5d} ~ XYn Xd}qqV|s|jtdd&|Wnt'k r} zt(d| W5d} ~ XYntjtfk rJ} ztd|)t*| fd| W5d} ~ XYndtj#k r|} ztt%| | W5d} ~ XYn2t+k r} ztt%| | W5d} ~ XYnXdS)zSwitch the socket to use SSLirDNFT)reverserrr rrrzInvalid CA Certificate: zInvalid Certificate/Key: r,)server_hostnament localhost 127.0.0.1z"Unable to verify server identity: z, z&Python installation has no SSL supportr@rA),r'r ssl CERT_REQUIRED CERT_OPTIONAL CERT_NONEcreate_default_contextcheck_hostnamesortTLS_V1_3_SUPPORTEDr TLS_VERSIONS SSLContextoptions OP_NO_TLSv1_2 OP_NO_TLSv1_1 OP_NO_TLSv1 verify_modeload_default_certsload_verify_locationsrFSSLErrorr:load_cert_chain set_ciphershasattr wrap_socketr,osnamer8 gethostbyaddrr\match_hostname getpeercertCertificateErrorrrrP NameErrorr r6rr4)r.cacertkeyZ verify_certZverify_identityZ cipher_suitesZ tls_versions cert_reqscontextZ tls_versionZ ssl_protocolrZ hostnamesaliasesZ match_foundZerrshostnamerrr switch_to_sslss       " "       zBaseMySQLSocket.switch_to_ssl)NN)NN)FFNN)__name__ __module__ __qualname____doc__r/r1r2r5r6r7r=r?rLsendrUrbrZrdrkrnrrrrrr%Xs,    N"K r%cs:eZdZdZd fdd ZddZddZd d ZZS) MySQLUnixSocketzoMySQL socket class using UNIX sockets Opens a connection through the UNIX socket of the MySQL Server. /tmp/mysql.sockcst||_dSr>)superr/ unix_socket)r.r __class__rrr/s zMySQLUnixSocket.__init__cCs|jSr>)rr-rrrr6szMySQLUnixSocket.get_addressc Csz2ttjtj|_|j|j|j|jWnjtk rn}zt d| t |fd|W5d}~XYn0t k r}zt t ||W5d}~XYnXdS)NirA)r8AF_UNIX SOCK_STREAMr'rlr(connectrrFr r6r Exceptionr)r.rrrrr5szMySQLUnixSocket.open_connectioncOstdtdS)zSwitch the socket to use SSL.z2SSL is disabled when using unix socket connectionsN)warningswarnWarning)r.argskwargsrrrrszMySQLUnixSocket.switch_to_ssl)r) rrrrr/r6r5r __classcell__rrrrrs  rcs2eZdZdZd fdd ZddZd d ZZS) MySQLTCPSocketzXMySQL socket class using TCP/IP Opens a TCP/IP connection to the MySQL Server. rs Fcs&t||_||_||_d|_dS)Nr)rr/r, server_port force_ipv6_family)r.hostportrrrrr/ s  zMySQLTCPSocket.__init__cCs|jd|jS)N:)r,rr-rrrr6szMySQLTCPSocket.get_addressc Csdgd}zt|j|jdtjtj}|D]6}|jrJ|dtjkrJ|}qb|dtjkr*|}qbq*|jr|ddkrt d|j|ddkr|d}Wn<t k r}zt d| t |fd|W5d}~XYnX|\|_ }}}}z0t|j |||_|j|j|j|Wnpt k rX}z t d|j|jt |fd|W5d}~XYn2tk r}ztt||W5d}~XYnXdS)z.Open the TCP/IP connection to the MySQL serverNrzNo IPv6 address found for irA)r8 getaddrinfor,rrSOL_TCPrAF_INET6AF_INETr rFr6rrr'rlr(rrr r) r.ZaddrinfoZ addrinfosinforsocktypeprotorIZsockaddrrrrr5sX   zMySQLTCPSocket.open_connection)rsrF)rrrrr/r6r5rrrrrrsr) rrr8rrrQ collectionsrrtPROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2r|rrPROTOCOL_SSLv23rr{ ImportError constantsr errorsr r r utilsrrr$r%rrrrrrs<     !