U
|
£ý°d)W ã @ sJ d Z ddlZddlZddlZddlZddlZddlmZ zHddlZej ej
|
ej dZ e edrdejneje d< e edo|ejZW n ek
|
r dZY nX dd lmZmZmZmZmZmZ d
|
d lmZ d
|
d l m Z m Z m Z d
|
d l m!Z! d
|
dl"m#Z# e$e%dddZ&e'e(ee' dddZ)G dd dZ*G dd de*Z+G dd de*Z,dS )zGModule implementing low-level socket communication with MySQL servers.
|
é N)Údeque)ÚTLSv1úTLSv1.1úTLSv1.2Ú PROTOCOL_TLSúTLSv1.3Ú HAS_TLSv1_3F)ÚAnyÚDequeÚListÚOptionalÚTupleÚUnioné )ÚMAX_PACKET_LENGTH)ÚInterfaceErrorÚNotSupportedErrorÚOperationalError)ÚStrOrBytesPath)Úinit_bytearray)ÚerrÚreturnc C s | j st| S | j d| j S )z_Reformat the IOError error message
|
|
This function reformats the IOError error message.
|
ú )ÚerrnoÚstrÚstrerror)r © r úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\mysql/connector/network.pyÚ _strioerrorE s r )ÚbufÚpktnrr c C s g }t | }t}||krX| dt d|¡ | d|
¡ | |d
} t | }|d }q| t d|¡dd
t d|¡ | ¡ |S )z0Prepare a packet for sending to the MySQL serveró ÿÿÿú<BNr ú<Ir é )Úlenr ÚappendÚstructÚpack)r r ÚpktsÚpllenÚ maxpktlenr r r Ú_prepare_packetsO s "
|
*r, c
|
@ s2 e Zd ZdZddddZedddZeddd Zedd
|
d Z edd d Z
|
ddddZ ddddZ ddddZ d'eee ee ddddZeZd(eee ee ddddZedddZeZeddd d Zee dd d Zee dd d!d"Zd)eeeeeee ee e dd$d%d&Z dS )*ÚBaseMySQLSocketa Base class for MySQL socket communication
|
|
This class should not be used directly but overloaded, changing the
|
at least the open_connection()-method. Examples of subclasses are
|
mysql.connector.network.MySQLTCPSocket
|
mysql.connector.network.MySQLUnixSocket
|
N©r c C s0 d | _ d | _d| _d| _t | _d | _d| _d S )Néÿÿÿÿi )ÚsockÚ_connection_timeoutÚ_packet_numberÚ_compressed_packet_numberr Ú _packet_queueÚ server_hostZrecvsize©Úselfr r r Ú__init__f s zBaseMySQLSocket.__init__c C s" | j d | _ | j dkr d| _ | j S )z Increments the packet numberr éÿ r )r2 r6 r r r Únext_packet_numberp s
|
z"BaseMySQLSocket.next_packet_numberc C s" | j d | _ | j dkr d| _ | j S )z'Increments the compressed packet numberr r9 r )r3 r6 r r r Ú next_compressed_packet_numberw s
|
z-BaseMySQLSocket.next_compressed_packet_numberc C s t dS )zOpen the socketN©ÚNotImplementedErrorr6 r r r Úopen_connection~ s z BaseMySQLSocket.open_connectionc C s t dS )z Get the location of the socketNr< r6 r r r Ú get_address s zBaseMySQLSocket.get_addressc C s> z | j tj¡ | j ¡ | `W n ttfk
|
r8 Y nX dS )z&Shut down the socket before closing itN)r0 ÚshutdownÚsocketÚ SHUT_RDWRÚcloser4 ÚAttributeErrorÚOSErrorr6 r r r r@ s
|
zBaseMySQLSocket.shutdownc C s0 z| j ¡ | `W n ttfk
|
r* Y nX dS )zClose the socketN)r0 rC r4 rD rE r6 r r r Úclose_connection s
|
|
z BaseMySQLSocket.close_connectionc C s | ¡ d S ©N)r@ r6 r r r Ú__del__ s zBaseMySQLSocket.__del__)r Ú packet_numberÚcompressed_packet_numberr c C s° |}|dkr| ¡ n|| _t|| j}|D ]~}z| j |¡ W q, tk
|
r| } z td| ¡ t|fd|W 5 d}~X Y q, t k
|
r¨ } ztdd|W 5 d}~X Y q,X q,dS )z Send packets to the MySQL serverNé ©r ÚvalueséÖ ©r )
|
r: r2 r, r0 ÚsendallÚIOErrorr r? r rD )r7 r rI rJ Ú_ÚpacketsÚpacketr r r r Ú
|
send_plain s"
|
ÿþzBaseMySQLSocket.send_plainc C sò |dkr| ¡ n|| _|dkr*| ¡ n|| _| j}t|}g }t}||krt||}d |¡} ~t | dd
¡}
|
t
|
dt|
|
¡dd
t
|
d| j¡ d } | | |
|
¡ | dd
} t| }| ¡ ||kr>t | d|
¡}
|
t
|
dt|
|
¡dd
t
|
d| j¡ d } | | |
|
¡ | |d
} t| }| ¡ qÌ| rt | ¡}
|
t
|
dt|
|
¡dd
t
|
d| j¡ t
|
d|¡dd
} | | |
|
¡ ~ nÊt
|
d|¡dd
t
|
d|¡ | } t| }|d
|
kr t | ¡}
|
| t
|
dt|
|
¡dd
t
|
d| j¡ t
|
d|¡dd
|
|
¡ nDt
|
d|¡dd
t
|
d| j¡ t
|
dd¡dd
} | | | ¡ |D ]} z| j | ¡ W nl tk
|
rº } z td | ¡ t|fd |W 5 d}~X Y n0 tk
|
rè } ztd d|W 5 d}~X Y nX qhdS )z+Send compressed packets to the MySQL serverNó é @ r# r r$ r" s @ r! é2 rK rL rN rO )r: r2 r; r3 r% r r, ÚjoinÚzlibÚcompressr' r( r& r0 rP rQ r r? r rD )r7 r rI rJ r r* Zzpktsr+ r) ZtmpbufZzbufÚheaderÚpktZ
|
zip_packetr r r r Úsend_compressed´ s
|
|
|
|
|
ÿþÿ
|
ÿþÿ
|
|
ÿþÿ$
|
|
ÿþýÿ ÿþÿ ÿþz BaseMySQLSocket.send_compressedc
|
C s zØt d}d}|dk rF| j d| ¡}|s4tdd||7 }t|}q|d | _t d|dd
d ¡d }|}| t |¡ t |}|dd
}|rÔ| j
|
||¡}|dkr¾|dkr¾tdd||d
}||8 }q|W S t k
|
r } z t d
|
| ¡ t|fd |W 5 d }~X Y nX d S ) z%Receive packets from the MySQL serverrV r é éÝ rO r$ r# ó NrK rL )Ú bytearrayr0 Úrecvr r% r2 r' ÚunpackÚextendÚ
|
memoryviewÚ recv_intorQ r r? r ) r7 rT Z
|
packet_lenÚchunkZ payload_lenÚrestZ packet_viewÚreadr r r r Ú
|
recv_plain s:
|
|
|
|
|
ÿþzBaseMySQLSocket.recv_plain)Ú packet_bunchr c C sN |rJt d|dd
d ¡d }| j |d|d
¡ ||d d
}q dS )zSplit compressed payloadr# r r$ ra r_ N)r' rd r4 r& )r7 rl Úpayload_lengthr r r Ú_split_zipped_payload+ s z%BaseMySQLSocket._split_zipped_payloadc
|
C sF z| j ¡ }|d | _|W S tk
|
r. Y nX td}g }z`| j d¡}|rrt|dk rr||7 }| j d¡}qL|rt|dk rtddt
|
d|dd
d ¡d }|d | _ t
|
d|d
|
d
d ¡d }t |}t||k r| j |t| ¡}|s
|
tdd|| }qØ|dkrB| |¡ | j ¡ }|d | _|W S | ||f¡ |d kr^qt d}| j d¡}|rrt|dk rr||7 }| j d¡}qrqrW n> tk
|
rÜ } z td | ¡ t| fd | W 5 d} ~ X Y nX t d}
|
|D ]\}} |
|
t | ¡7 }
|
qê| |
|
¡ ~
|
z| j ¡ }|d | _|W S tk
|
r@ Y nX dS )z0Receive compressed packets from the MySQL serverr$ rV r é r` rO r# r ra r_ rW rK rL N)r4 Úpopleftr2 Ú
|
IndexErrorrb r0 rc r% r r' rd r3 r rn r& rQ r r? r rZ Ú
|
decompress) r7 r] r\ rS ZabyteZzip_payload_lengthrm Z zip_payloadrh r ÚtmpÚpayloadr r r Úrecv_compressed2 sr
|
|
|
|
|
|
|
|
|
|
|
ÿþ
|
|
|
z BaseMySQLSocket.recv_compressed)Útimeoutr c C s || _ | jr| j |¡ dS )zSet the connection timeoutN)r1 r0 Ú
|
settimeout)r7 rv r r r Úset_connection_timeout~ s z&BaseMySQLSocket.set_connection_timeoutF)ÚcaÚcertÚkeyÚ verify_certÚverify_identityÚ cipher_suitesÚ tls_versionsr c C s¾ | j stddzÔ|r tj}n|r,tj}ntj}|dks>|sRt ¡ } |sèd| _n|jdd |d }
|
t s|
|
dkrt
|
|d kr|d }
|
t |
|
} t | ¡} |
|
dkrèd
|
|kr¸| j tjO _ d |krÐ| j tjO _ d |krè| j tjO _ d| _|| _| ¡ |rVz| |¡ W nD ttjfk
|
rT } z | j ¡ td | | W 5 d} ~ X Y nX |r²z| ||¡ W nD ttjfk
|
r° } z | j ¡ td| | W 5 d} ~ X Y nX |rÂ| |¡ t| drä| j| j | jd| _ n| | j ¡| _ |räd| _| jr| jgng } t j dkrV| jdkrVddg} t | j¡}| |d g|d ¡ d}g }| D ]\}zt !| j "¡ |¡ W n4 tj#k
|
r° } z| $t%| ¡ W 5 d} ~ X Y n X d} qÀqb|sä| j ¡ tdd &|¡ W nÒ t'k
|
r } zt(d| W 5 d} ~ X Y n¦ tjtfk
|
rV } z td| )¡ t*| fd| W 5 d} ~ X Y nd tj#k
|
r } ztt%| | W 5 d} ~ X Y n2 t+k
|
r¸ } ztt%| | W 5 d} ~ X Y nX dS )z Switch the socket to use SSLi rO NFT)Úreverser r r r r r zInvalid CA Certificate: zInvalid Certificate/Key: r5 )Úserver_hostnameÚntÚ localhostú 127.0.0.1z"Unable to verify server identity: z, z&Python installation has no SSL supportrK rL ),r0 r ÚsslÚ CERT_REQUIREDÚ CERT_OPTIONALÚ CERT_NONEÚcreate_default_contextÚcheck_hostnameÚsortÚTLS_V1_3_SUPPORTEDr% Ú TLS_VERSIONSÚ
|
SSLContextÚoptionsÚ OP_NO_TLSv1_2Ú OP_NO_TLSv1_1Ú OP_NO_TLSv1Ú verify_modeÚload_default_certsÚload_verify_locationsrQ ÚSSLErrorrC Úload_cert_chainÚ set_ciphersÚhasattrÚ wrap_socketr5 ÚosÚnamerA Ú gethostbyaddrre Úmatch_hostnameÚ getpeercertÚCertificateErrorr& r rY Ú NameErrorr r? r r= )r7 ry rz r{ r| r} r~ r Ú cert_reqsÚcontextZ tls_versionZ ssl_protocolr Z hostnamesÚaliasesZ match_foundZerrsÚhostnamer r r Ú switch_to_ssl s®
|
ÿþ
|
ý
|
|
"
|
"
|
ÿ
|
|
|
ÿ ÿþ z BaseMySQLSocket.switch_to_ssl)NN)NN)FFNN) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r8 Úintr: r; r r> r? r@ rF rH Úbytesr rU Úsendr^ rb rk rc rn ru rx r Úboolr r r¦ r r r r r- ] sZ
|
üû üû S"L ø
|
÷r- c sX e Zd ZdZdedd fdd Zeddd Zddd
|
d Zeedd d dZ Z
|
S )ÚMySQLUnixSocketzoMySQL socket class using UNIX sockets
|
|
Opens a connection through the UNIX socket of the MySQL Server.
|
ú/tmp/mysql.sockN)Ú unix_socketr c s t ¡ || _d S rG )Úsuperr8 r± )r7 r± ©Ú __class__r r r8 ú s
|
zMySQLUnixSocket.__init__r. c C s | j S rG )r± r6 r r r r? þ s zMySQLUnixSocket.get_addressc
|
C s¢ z2t t jt j¡| _| j | j¡ | j | j¡ W nj tk
|
rn } z t d|
|
¡ t |fd|W 5 d }~X Y n0 t k
|
r } zt t ||W 5 d }~X Y nX d S )NiÒ rL )rA ÚAF_UNIXÚ SOCK_STREAMr0 rw r1 Úconnectr± rQ r r? r Ú Exceptionr )r7 r r r r r> s ÿ ÿþz MySQLUnixSocket.open_connection)ÚargsÚkwargsr c O s t dt¡ dS )z Switch the socket to use SSL.z2SSL is disabled when using unix socket connectionsN)ÚwarningsÚwarnÚWarning)r7 r¹ rº r r r r¦ s þz MySQLUnixSocket.switch_to_ssl)r° ) r§ r¨ r© rª r r8 r? r> r r¦ Ú __classcell__r r r³ r r¯ ô s þr¯ c sJ e Zd ZdZdeeedd fdd Zed d
|
d Zdd d d Z Z
|
S )ÚMySQLTCPSocketzXMySQL socket class using TCP/IP
|
|
Opens a TCP/IP connection to the MySQL Server.
|
r éê FN)ÚhostÚportÚ
|
force_ipv6r c s& t ¡ || _|| _|| _d| _d S )Nr )r² r8 r5 Ú server_portrà Ú_family)r7 rÁ r rà r³ r r r8 s
|
|
zMySQLTCPSocket.__init__r. c C s | j d| j S )Nú:)r5 rÄ r6 r r r r? ( s zMySQLTCPSocket.get_addressc
|
C s d}zt | j| jdt jt j¡}|D ]6}| jrD|d t jkrD|} q\|d t jkr$|} q\q$| jr~|d dkr~t d| j |d dkr|d }W n< t
|
k
|
rÐ } z t d| ¡ t |fd|W 5 d}~X Y nX |\| _ }}}}z0t | j ||¡| _| j | j¡ | j |¡ W np t
|
k
|
rR } z t d| j| jt |fd|W 5 d}~X Y n2 tk
|
r } ztt||W 5 d}~X Y nX dS )z.Open the TCP/IP connection to the MySQL server)NNNNNr NzNo IPv6 address found for iÓ rL )rA Ú getaddrinfor5 rÄ r¶ ÚSOL_TCPrà ÚAF_INET6ÚAF_INETr rQ r? r rÅ r0 rw r1 r· r¸ r r ) r7 ZaddrinfoZ addrinfosÚinfor ÚsocktypeÚprotorR Zsockaddrr r r r> + sZ ÷
|
û ÿþýþùz MySQLTCPSocket.open_connection)r rÀ F) r§ r¨ r© rª r r« r® r8 r? r> r¾ r r r³ r r¿ s ÿ þ r¿ )-rª r rA r' r» rZ Ú collectionsr r
ÚPROTOCOL_TLSv1ÚPROTOCOL_TLSv1_1ÚPROTOCOL_TLSv1_2r r r ÚPROTOCOL_SSLv23r r Ú ImportErrorÚtypingr r
|
r r r r Ú constantsr Úerrorsr r r Útypesr Úutilsr rQ r r r¬ r« r, r- r¯ r¿ r r r r Ú<module> s@ ýÿý
|
|
%
|
