o bz|jtj|j|`WdStjtfyYdSw)z&Shut down the socket before closing itN)r'shutdownsocket SHUT_RDWRcloser+errorAttributeErrorr,rrrr:s  zBaseMySQLSocket.shutdownc Cs0z |j|`WdStjtfyYdSw)zClose the socketN)r'r=r+r;r>r?r,rrrclose_connections   z BaseMySQLSocket.close_connectioncCs |dSN)r:r,rrr__del__s zBaseMySQLSocket.__del__Nc Cs|dur|jn||_t||j}|D]/}z|j|Wqty6}z tjd|t |fdd}~wt yBtjddwdS)z Send packets to the MySQL serverNrvaluesr) r1r)r$r'sendallIOErrorr OperationalErrorr9rr?)r-r packet_numbercompressed_packet_numberpacketspacketrrrr send_plains    zBaseMySQLSocket.send_plainc Cs|dur|jn||_|dur|jn||_|j}t|}g}tj}||krt||}d|} ~t | dd} t dt| ddt d|jd} | | | | dd} t| }|j||krt | d|} t dt| ddt d|jd } | | | | |d} t| }|j||ksg| rt | } t dt| ddt d|jt d|dd} | | | ~ net d|ddt d||} t| }|d kr t | } | t dt| ddt d|jt d|dd| n"t d|ddt d|jt dddd} | | | |D]2} z |j| Wq1tyV}z tjd |t|fd d}~wtyctjd dwdS)z+Send compressed packets to the MySQL serverN@rrrrs@r2rCrDrFrG)r1r)r3r*rr rr$joinzlibcompressrrrr'rHrIr rJr9rr?)r-rrKrLr r"Zzpktsr#r!ZtmpbufZzbufheaderpktZ zip_packetrrrrsend_compresseds              zBaseMySQLSocket.send_compressedc Cszrtd}d}|dkr'|jd|}|stjdd||7}t|}|dks |d|_td|dddd}|}| t|t |}|dd }|rp|j ||}|dkrd|dkrdtjdd||d }||8}|sO|WSt y}z tj d |t|fd d }~ww) %Receive packets from the MySQL serverrPrrGrrNrCrD) bytearrayr'recvr r rr)runpackextend memoryview recv_intorIrJr9r) r-rNZ packet_lenchunk payload_lenrestZ packet_viewreadrrrr recv_plains<     zBaseMySQLSocket.recv_plainc Cszdtd}d}|dkr'|jd|}|stjdd||7}t|}|dks |d|_td|dddd}|}t d}|dkr`|j|}|sRtjdd||7}|t|}|dksD||WSt y}}z tj d | t |fd d }~ww) rYrPrrZr[rGrrr\rCrDN)r]r'r^r r rr)rr_r rIrJr9r)r-rVZ header_lenrcrdrepayloadrrrrrecv_py26_plains8      zBaseMySQLSocket.recv_py26_plainr)rjcCsT|r(td|dddd}|j|d|d||dd}|sdSdS)zSplit compressed payloadrrrr\rZN)rr_r+r)r-Z packet_bunchpayload_lengthrrr_split_zipped_payload5s  z%BaseMySQLSocket._split_zipped_payloadc CsNz |j}|d|_|WStyYnwtd}g}z|jd}|r>t|dkr>||7}|jd}|r>t|dks,|rt|dkrLtj ddt d|ddd d}|d|_ t d|d dd d}t |}t||kr|j|t|}|stj dd||}t||ksw|dkr|||j}|d|_|WS|||f|d krn&t d}|jd}|rt|dkr||7}|jd}|rt|dks|s@Wnty} z tjd |t| fd d} ~ wwt d} |D] \}} | t| 7} q|| ~ z |j}|d|_|WSty&YdSw)z0Receive compressed packets from the MySQL serverrrPr r[rGrrr\rZrQrCrDN)r+popleftr) IndexErrorr]r'r^rr r rr_r*r rmrrIrJr9rrT decompress) r-rWrVrMZabyteZzip_payload_lengthrlZ zip_payloadrcrtmprhrrrrecv_compressed<s                (    zBaseMySQLSocket.recv_compressedcCs ||_|jr|j|dSdS)zSet the connection timeoutN)r(r' settimeout)r-timeoutrrrset_connection_timeoutsz&BaseMySQLSocket.set_connection_timeoutFc Cs^|js tjddzT|rtj}n |rtj}ntj}|dus |s*t} |s)d| _nK|j dd|d} t sD| dkrDt |d krD|d } t | } t | } | dkrud |vr]| jtjO_d |vri| jtjO_d |vru| jtjO_d| _|| _| |rz| |Wnttjfy} z |jtd | d} ~ ww|rz| ||Wnttjfy} z |jtd| d} ~ ww|r| |t|dr| j|j|jd|_n| |j|_|rYd| _|jg} tjdkr|jdkrddg} t !|j}| "|dg|d d}g}| D],}z t#|j$|Wntj%yC} z|&t'| WYd} ~ qd} ~ wwd}|s\|jtdd(|WdSWdSt)ykt*dtjtfy} z tjd|+t,| fdd} ~ wtj%y} ztt'| d} ~ wt-y} ztt'| d} ~ ww)zSwitch the socket to use SSLirGNFT)reverserrr rrrzInvalid CA Certificate: {}zInvalid Certificate/Key: {} server_host)server_hostnament localhost 127.0.0.1z$Unable to verify server identity: {}z, z&Python installation has no SSL supportrCrD).r'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_locationsrISSLErrorr=rload_cert_chain set_ciphershasattr wrap_socketrxosnamer; gethostbyaddrr`match_hostname getpeercertCertificateErrorrrrS NameErrorZNotSupportedErrorr9rr6)r-cacertkeyZ verify_certZverify_identityZ cipher_suitesZ tls_versions cert_reqscontextZ tls_version ssl_protocolrZ hostnamesaliasesZ match_foundZerrshostnamerrr switch_to_ssls             zBaseMySQLSocket.switch_to_ssl)NN)FFNN)__name__ __module__ __qualname____doc__r.propertyr1r3r7r9r:r@rBrOsendrXrgrisys version_infor^rmrsrvrrrrrr%Zs@     E!Lr%cs2eZdZdZd fdd ZddZddZZS) MySQLUnixSocketzoMySQL socket class using UNIX sockets Opens a connection through the UNIX socket of the MySQL Server. /tmp/mysql.sockcstt|||_dSrA)superrr. unix_socket)r-r __class__rrr.s zMySQLUnixSocket.__init__cCs|jSrA)rr,rrrr9szMySQLUnixSocket.get_addressc Cszttjtj|_|j|j|j|jWdSty3}z t j d| t |fdd}~wt yE}zt t|d}~ww)NirD)r;AF_UNIX SOCK_STREAMr'rtr(connectrrIr r r9r Exceptionr)r-rrrrr7szMySQLUnixSocket.open_connection)rrrrrr.r9r7 __classcell__rrrrrs rcs2eZdZdZd fdd ZddZd d ZZS) MySQLTCPSocketzXMySQL socket class using TCP/IP Opens a TCP/IP connection to the MySQL Server. r| Fcs*tt|||_||_||_d|_dS)Nr)rrr.rx server_port force_ipv6_family)r-hostportrrrrr.s  zMySQLTCPSocket.__init__cCsd|j|jS)Nz{0}:{1})rrxrr,rrrr9szMySQLTCPSocket.get_addressc Csldgd}zIt|j|jdtjtj}|D]}|jr%|dtjkr%|}n |dtjkr0|}nq|jrC|ddurCt d |j|ddurM|d}Wnt yg}z t j d| t|fdd}~ww|\|_}}}}zt|j|||_|j|j|j|WdSt y}zt j d|j|jt|fdd}~wty}zt t|d}~ww)z7Open the TCP/IP connection to the MySQL server NrzNo IPv6 address found for {0}irD)r; getaddrinforxrrSOL_TCPrAF_INET6AF_INETr r rrIr9rrr'rtr(rrrJr) r-addrinfo addrinfosinforsocktypeproto_sockaddrrrrr7s\   zMySQLTCPSocket.open_connection)r|rFrrrrrr s r)r collectionsrrr;rrrTr}PROTOCOL_TLSv1PROTOCOL_TLSv1_1PROTOCOL_TLSv1_2rrrPROTOCOL_SSLv23rrr r r utilsr rr$objectr%rrrrrrsD