a
XC?h2� �
@ s� d dl Z d dlZd dlmZ d dlmZmZm Z
d dl Z d dl m Z m
Z
mZ d dl mZmZmZ d dl mZmZmZmZmZmZmZ d dl mZmZ d dl m Z m Z m Z m Z zd d l m!Z! W n e"y� Y n0 d d
l m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z, d d l m-Z-m.Z. ej/d e0d
d� e d� e
j/de0dd� e d� ej/de0dd� e d� ej/de0dd� e d� e
j/de0dd� e d� ej/de0dd� e d� e1j2 Z3e1_3dd� e1j4�5� D �Z6e7e1d d�Z8G d d � d e�Z9G d d � d e�Z:G d!d"� d"e�Z;G d#d$� d$e�ZZ>m?Z? d d'l@m@Z@mAZAmBZB d d(l@mCZCmDZD d dl@ZEd dlFZFd dlGZGd dlHZHeIZJd)gZKeLe d*�ZMe-ZNeZOd+d,� ZPd-d.� ZQd/d0� ZRd1d2� ZSed3d4�ZTd5d6� ZUG d7d8� d8ed8d9��ZVG d:d;� d;eVe�ZWG d�d?d@�ZZe2fe[dAeWjYddddddB�dCdD�Z\eZZ]e\Z^G dEdF� dF�Z_dGdH� Z`G dIdJ� dJe@�ZaeaeX_be_eX_cdddAe[e2ddKdKdf dLdM�ZddNdO� ZedPZfdQZgdRdS� ZhdTdU� Zie2dfdVdW�ZjdXdY� ZkdS )Z� N)�
namedtuple)�Enum�IntEnum�IntFlag)�OPENSSL_VERSION_NUMBER�OPENSSL_VERSION_INFO�OPENSSL_VERSION)� _SSLContext� MemoryBIO�
SSLSession)�SSLError�SSLZeroReturnError�SSLWantReadError�SSLWantWriteError�SSLSyscallError� SSLEOFError�SSLCertVerificationError)�txt2obj�nid2obj)� RAND_status�RAND_add�
RAND_bytes�RAND_pseudo_bytes)�RAND_egd)
�HAS_SNI�HAS_ECDH�HAS_NPN�HAS_ALPN� HAS_SSLv2� HAS_SSLv3� HAS_TLSv1� HAS_TLSv1_1� HAS_TLSv1_2� HAS_TLSv1_3)�_DEFAULT_CIPHERS�_OPENSSL_API_VERSION�
_SSLMethodc C s | � d�o| dkS )NZ PROTOCOL_�PROTOCOL_SSLv23��
startswith��name� r, �(/opt/alt/python39/lib64/python3.9/ssl.py�} � r. )�source�Optionsc C s
| � d�S )NZOP_r( r* r, r, r- r. � r/ ZAlertDescriptionc C s
| � d�S )NZALERT_DESCRIPTION_r( r* r, r, r- r. � r/ ZSSLErrorNumberc C s
| � d�S )NZ
SSL_ERROR_r( r* r, r, r- r. � r/ � VerifyFlagsc C s
| � d�S )NZVERIFY_r( r* r, r, r- r. � r/ �
VerifyModec C s
| � d�S )NZCERT_r( r* r, r, r- r. � r/ c C s i | ]\}}||�qS r, r, )�.0r+ �valuer, r, r- �
� r/ r6 ZPROTOCOL_SSLv2c @ s6 e Zd ZejZejZejZ ej
Z ej Z
ejZejZdS )�
TLSVersionN)�__name__�
__module__� __qualname__�_sslZPROTO_MINIMUM_SUPPORTEDZMINIMUM_SUPPORTEDZ PROTO_SSLv3�SSLv3Z PROTO_TLSv1ZTLSv1Z
PROTO_TLSv1_1ZTLSv1_1Z
PROTO_TLSv1_2ZTLSv1_2Z
PROTO_TLSv1_3ZTLSv1_3ZPROTO_MAXIMUM_SUPPORTEDZMAXIMUM_SUPPORTEDr, r, r, r- r7 � s r7 c @ s$ e Zd ZdZdZdZdZdZdZdS )�_TLSContentType� � � � � � N) r8 r9 r: �CHANGE_CIPHER_SPEC�ALERTZ HANDSHAKEZAPPLICATION_DATA�HEADERZINNER_CONTENT_TYPEr, r, r, r- r= � s r= c @ s� e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Z!d Z"d!Z#d"Z$d#S )$�
_TLSAlertTyper �
r> r? r@ � �( �) �* �+ �, �- �. �/ �0 �1 �2 �3 �< �F �G �P �V �Z �d �m �n �o �p �q �r �s �t �x N)%r8 r9 r: Z CLOSE_NOTIFYZUNEXPECTED_MESSAGEZBAD_RECORD_MACZDECRYPTION_FAILEDZRECORD_OVERFLOWZDECOMPRESSION_FAILUREZHANDSHAKE_FAILUREZNO_CERTIFICATEZBAD_CERTIFICATEZUNSUPPORTED_CERTIFICATEZCERTIFICATE_REVOKEDZCERTIFICATE_EXPIREDZCERTIFICATE_UNKNOWNZILLEGAL_PARAMETERZ
UNKNOWN_CAZ
ACCESS_DENIEDZ DECODE_ERRORZ
DECRYPT_ERRORZEXPORT_RESTRICTIONZPROTOCOL_VERSIONZINSUFFICIENT_SECURITYZINTERNAL_ERRORZINAPPROPRIATE_FALLBACKZ
USER_CANCELEDZNO_RENEGOTIATIONZMISSING_EXTENSIONZUNSUPPORTED_EXTENSIONZCERTIFICATE_UNOBTAINABLEZUNRECOGNIZED_NAMEZ BAD_CERTIFICATE_STATUS_RESPONSEZBAD_CERTIFICATE_HASH_VALUEZUNKNOWN_PSK_IDENTITYZCERTIFICATE_REQUIREDZNO_APPLICATION_PROTOCOLr, r, r, r- rG � sD rG c @ sd e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdZdZdZdZdZdZdZdZdS )�_TLSMessageTyper � � � � � � � � � �
� � � r> r? r@ rA � �C �� rC N)r8 r9 r: Z
HELLO_REQUESTZ CLIENT_HELLOZ SERVER_HELLOZHELLO_VERIFY_REQUESTZNEWSESSION_TICKETZEND_OF_EARLY_DATAZHELLO_RETRY_REQUESTZENCRYPTED_EXTENSIONSZ CERTIFICATEZSERVER_KEY_EXCHANGEZCERTIFICATE_REQUESTZ SERVER_DONEZCERTIFICATE_VERIFYZCLIENT_KEY_EXCHANGEZFINISHEDZCERTIFICATE_URLZCERTIFICATE_STATUSZSUPPLEMENTAL_DATAZ
KEY_UPDATEZ
NEXT_PROTOZ MESSAGE_HASHrD r, r, r, r- rf � s, rf �win32)�enum_certificates� enum_crls)�socket� SOCK_STREAM�create_connection)�
SOL_SOCKET�SO_TYPE�
tls-unique� HOSTFLAG_NEVER_CHECK_SUBJECTc C s� | sdS | � d�}|s&| �� |�� kS |dkrsole wildcard without additional labels are not support: {!r}.z.shim_cb)Z sni_callback�callable� TypeError)r� r� r� r, r� r- �set_servername_callback s z"SSLContext.set_servername_callbackc C s` t � }|D ]F}t|d�}t|�dks0t|�dkr8td��|�t|�� |�|� q
| �|� d S )Nr� r r� z)ALPN protocols must be 1 to 255 in length)r� r� r� r r� r� Z_set_alpn_protocols)r� Zalpn_protocolsr� r� r� r, r, r- �set_alpn_protocols! s
z SSLContext.set_alpn_protocolsc C st t � }z Y n0 |tjkrPt}n|tjkr`t}nt}z ||�}W n ty� Y n0 � | |||||�S r� )r7 r� r= rF rE rG rf )Zconn� direction�versionZ content_typeZmsg_type�dataZmsg_enum��callbackr, r- r� � s(
�z'SSLContext._msg_callback..inner)r� r� r� r� �hasattrr� r� )r� r r� r� r� r- r� � s
c s t t� j�S r� )r& r� r� r� r� r, r- r� � s zSSLContext.protocolc s t t� j�S r� )r2 r� � verify_flagsr� r� r, r- r � s zSSLContext.verify_flagsc s t tt�j�| |� d S r� )r� r� r r� r� r� r, r- r � s c s. t � j}z
t|�W S ty( | Y S 0 d S r� )r� � verify_moder3 r� r� r� r, r- r � s
zSSLContext.verify_modec s t tt�j�| |� d S r� )r� r� r r� r� r� r, r- r � s )FTTNN)FNN) r8 r9 r: r� r� r� � PROTOCOL_TLSr� r� r� r� r� r� r� r� r� r� r� r r �propertyr� �setterr� r� r; r� r� r� r r r� r, r, r� r- r� � sh �
�
&%r� )r� r� r� c C s� t | t�st| ��tt�}| tjkr0t|_d|_ |s<|s<|rL|�
|||� n|jt kr`|� | � t
|d�r�tj�d�}|r�tjjs�||_|S )NT�keylog_filename�
SSLKEYLOGFILE)r� r� r� r� r r� r� �
CERT_REQUIREDr �check_hostnamer� � CERT_NONEr� r r� r� r� r� �flags�ignore_environmentr )r� r� r� r� r� �
keylogfiler, r, r- �create_default_context� s
r F)� cert_reqsr
r� �certfile�keyfiler� r� r� c C s� t |t�st|��t| �} |s$d| _|d ur2|| _|rd?� �Z"� fd@dA�Z#edBdC� �Z$edDdE� �Z%� fdFdG�Z&edcdHdI��Z'� fdJdK�Z(dLdM� Z)dNdO� Z*� fdPdQ�Z+edddSdT��Z,edUdV� �Z-� Z.S )e� SSLSocketc O s t | jj� d���d S )NzX does not have a public constructor. Instances are returned by SSLContext.wrap_socket().r r r, r, r- r � s �zSSLSocket.__init__FTNc
sf |� tt�tkrtd��|r8|r(td��|d ur8td��|jrJ|sJtd��t|j|j |j
|� � d�}| j | fi |��} t
t| �jf i |�� |�� }
|�� || _|| _d| _d | _|| _|�|�| _|| _|| _z | �� W n� t �y� } z�| j t j kr�� d} | � � }
| � d� z| �!d�}W n@ t �yb } z&| j t j t j"fv�rJ� d}W Y d } ~ n
d } ~ 0 0 | � |
� |�r�d }t#| j |�}||_$d |_%z | �&� W n t �y� Y n0 z |�W d }nd }0 W Y d } ~ nd } ~ 0 0 d
} | �'|
� | | _(| �rbzH| jj)| || j| | jd �| _|�r<| �� }|d k�r4td
��| �*� W n" t tf�y` | �&� � Y n0 | S )Nz!only stream sockets are supportedz4server_hostname can only be specified in client modez,session can only be specified in client modez'check_hostname requires server_hostname)�family�type�proto�filenoFrg r/ z5Closed before TLS handshake with data in recv buffer.T�r r� � zHdo_handshake_on_connect should not be specified for non-blocking sockets)+Z
getsockoptr} r~ r{ �NotImplementedErrorr� r
�dictr= r> r? r@ r� r� r< r �
gettimeout�detach�_context�_sessionZ_closedr r� r� r� r� r� � getpeernamer� �errnoZENOTCONNZ getblockingZ setblocking�recvZEINVALr �reasonZlibrary�close�
settimeout�
_connected� _wrap_socketr1 )r� r� r� r� r� r� r� r� r� r� Z sock_timeout�eZ connectedZblockingZnotconn_pre_handshake_datarL Z notconn_pre_handshake_data_error�timeoutr� r, r- r� � s�
�
"
�
zSSLSocket._createc C s | j S r� )rG r� r, r, r- r� 8 s zSSLSocket.contextc C s || _ || j_d S r� )rG r r� r r, r, r- r� = s c C s | j d ur| j jS d S r� r r� r, r, r- r� B s
zSSLSocket.sessionc C s || _ | jd ur|| j_d S r� )rH r r� r r, r, r- r� H s
c C s | j d ur| j jS d S r� r! r� r, r, r- r" N s
zSSLSocket.session_reusedc C s t d| jj ��d S )NzCan't dup() %s instances)rC r� r8 r� r, r, r- �dupT s �z
SSLSocket.dupc C s d S r� r, )r� �msgr, r, r- � _checkClosedX s zSSLSocket._checkClosedc C s | j s| �� d S r� )rO rI r� r, r, r- �_check_connected\ s zSSLSocket._check_connectedr# c
C s� | � � | jd u rtd��z*|d ur4| j�||�W S | j�|�W S W n` ty� } zH|jd tkr�| jr�|d ur~W Y d }~dS W Y d }~dS n� W Y d }~n
d }~0 0 d S )Nz'Read on closed or unwrapped SSL socket.r r/ )rU r r� r$ r r� Z
SSL_ERROR_EOFr� )r� r� r% �xr, r, r- r$ d s
zSSLSocket.readc C s& | � � | jd u rtd��| j�|�S )Nz(Write on closed or unwrapped SSL socket.)rU r r� r' r( r, r, r- r' y s
zSSLSocket.writec C s | � � | �� | j�|�S r� )rU rV r r) r* r, r, r- r) � s zSSLSocket.getpeercertc C s* | � � | jd u stjs d S | j�� S d S r� )rU r r; r r+ r� r, r, r- r+ � s z SSLSocket.selected_npn_protocolc C s* | � � | jd u stjs d S | j�� S d S r� )rU r r; r r, r� r, r, r- r, � s z SSLSocket.selected_alpn_protocolc C s$ | � � | jd u rd S | j�� S d S r� )rU r r- r� r, r, r- r- � s
zSSLSocket.cipherc C s$ | � � | jd u rd S | j�� S d S r� )rU r r. r� r, r, r- r. � s
zSSLSocket.shared_ciphersc C s$ | � � | jd u rd S | j�� S d S r� )rU r r/ r� r, r, r- r/ � s
zSSLSocket.compressionr c sF | � � | jd ur4|dkr(td| j ��| j�|�S t� �||�S d S )Nr z3non-zero flags not allowed in calls to send() on %s)rU r r� r� r' r� �send)r� r� r r� r, r- rX � s
�� zSSLSocket.sendc sL | � � | jd ur"td| j ��n&|d u r8t� �||�S t� �|||�S d S )Nz%sendto not allowed on instances of %s)rU r r� r� r� �sendto)r� r� Z
flags_or_addrr� r� r, r- rY � s
�zSSLSocket.sendtoc O s t d| j ��d S )Nz&sendmsg not allowed on instances of %s�rC r� r r, r, r- �sendmsg� s �zSSLSocket.sendmsgc s� | � � | jd ur�|dkr(td| j ��d}t|��f}|�d��<}t|�}||k rn| �||d � �}||7 }qJW d � n1 s�0 Y W d � q�1 s�0 Y nt� � ||�S d S )Nr z6non-zero flags not allowed in calls to sendall() on %s�B)
rU r r� r� �
memoryview�castr� rX r� �sendall)r� r� r r� ZviewZ byte_viewZamountr& r� r, r- r_ � s
��HzSSLSocket.sendallc s, | j d ur| �|||�S t� �|||�S d S r� )r Z_sendfile_use_sendr� �sendfile)r� �file�offsetr� r� r, r- r` � s
zSSLSocket.sendfilec sD | � � | jd ur2|dkr(td| j ��| �|�S t� �||�S d S )Nr z3non-zero flags not allowed in calls to recv() on %s)rU r r� r� r$ r� rK �r� Zbuflenr r� r, r- rK � s
��
zSSLSocket.recvc sj | � � |r |d u r t|�}n |d u r*d}| jd urV|dkrJtd| j ��| �||�S t� �|||�S d S )Nr# r z8non-zero flags not allowed in calls to recv_into() on %s)rU r� r r� r� r$ r� � recv_into�r� r% �nbytesr r� r, r- rd � s
�� zSSLSocket.recv_intoc s4 | � � | jd ur"td| j ��nt� �||�S d S )Nz'recvfrom not allowed on instances of %s)rU r r� r� r� �recvfromrc r� r, r- rg � s
�zSSLSocket.recvfromc s6 | � � | jd ur"td| j ��nt� �|||�S d S )Nz,recvfrom_into not allowed on instances of %s)rU r r� r� r� �
recvfrom_intore r� r, r- rh s
�zSSLSocket.recvfrom_intoc O s t d| j ��d S )Nz&recvmsg not allowed on instances of %srZ r r, r, r- �recvmsg s �zSSLSocket.recvmsgc O s t d| j ��d S )Nz+recvmsg_into not allowed on instances of %srZ r r, r, r- � recvmsg_into s �zSSLSocket.recvmsg_intoc C s$ | � � | jd ur | j�� S dS d S )Nr )rU r r0 r� r, r, r- r0 s
zSSLSocket.pendingc s | � � d | _t� �|� d S r� )rU r r� r2 )r� Zhowr� r, r- r2 s zSSLSocket.shutdownc C s. | j r| j �� }d | _ |S tdt| � ��d S �NzNo SSL wrapper around )r r2 r� r� )r� �sr, r, r- r3 $ s
zSSLSocket.unwrapc C s$ | j r| j �� S tdt| � ��d S rk )r r7 r� r� r� r, r, r- r7 - s
z&SSLSocket.verify_client_post_handshakec s d | _ t� �� d S r� )r r� � _real_closer� r� r, r- rm 4 s zSSLSocket._real_closec C sP | � � | �� }z.|dkr(|r(| �d � | j�� W | �|� n | �|� 0 d S )NrB )rV rE rN r r1 )r� �blockrR r, r, r- r1 8 s
zSSLSocket.do_handshakec s� | j rtd��| js | jd ur&td��| jj| d| j| | jd�| _z@|rVt� � |�}nd }t� �
|� |s~d| _| j r~| � � |W S t
tfy� d | _� Y n0 d S )Nz!can't connect in server-side modez/attempt to connect already-connected SSLSocket!FrA T)r� r� rO r r� rP r� rH r� �
connect_ex�connectr� r1 r� )r� r� ro Zrcr� r, r- �
_real_connectC s* � zSSLSocket._real_connectc C s | � |d� d S )NF�rq �r� r� r, r, r- rp ] s zSSLSocket.connectc C s | � |d�S r� rr rs r, r, r- ro b s zSSLSocket.connect_exc s. t � �� \}}| jj|| j| jdd�}||fS )NT)r� r� r� )r� �acceptr� r� r� r� )r� Znewsockr� r� r, r- rt g s �zSSLSocket.acceptr c C s4 | j d ur| j �|�S |tvr,td�|���d S d S )Nz({0} channel binding type not implemented)r r4 �CHANNEL_BINDING_TYPESr� r� r5 r, r, r- r4 s s
�z SSLSocket.get_channel_bindingc C s | j d ur| j �� S d S d S r� r6 r� r, r, r- r� ~ s
zSSLSocket.version)FTTNNN)N)r# N)F)r )N)r )r N)r# r )Nr )r# r )Nr )F)r )/r8 r9 r: r r� r� r r; r� r r� r" rS rU rV r$ r' r) r+ r, r- r. r/ rX rY r[ r_ r` rK rd rg rh ri rj r0 r2 r3 r7 rm r1 rq rp ro rt r4 r� r� r, r, r� r- r< � s� �_
r< Tc
C sl |r|st d��|r |s t d��t|�}
||
_|r<|
�|� |rL|
�||� | rZ|
�| � |
j| |||d�S )Nz5certfile must be specified for server-side operationsr )r� r� r� r� )r� r� r r� r Z set_ciphersr� ) r� r r r� r � ssl_version�ca_certsr� r� Zciphersr� r, r, r- r� � s"
�r� c C s� ddl m} ddlm} d}d}z |�| d d� �� �d }W n" ty` td| |f ��Y n00 || dd � |�}||d |f|d d
� �S d S ) Nr )�strptime)�timegm) ZJanZFebZMarZAprZMayZJunZJulZAugZSepZOctZNovZDecz %d %H:%M:%S %Y GMTri rg z*time data %r does not match format "%%b%s"rh rl )�timerx Zcalendarry �index�titler� )Z cert_timerx ry ZmonthsZ time_formatZ month_numberZttr, r, r- �cert_time_to_seconds� s
� r} z-----BEGIN CERTIFICATE-----z-----END CERTIFICATE-----c sR t t�| �dd�� tg}|� fdd�tdt� �d�D �7 }|�td � d�|�S )N�ASCII�strictc s g | ]}� ||d � �qS )�@ r, )r4 �i��fr, r- �
� r/ z(DER_cert_to_PEM_cert..r r� �
) r� �base64Zstandard_b64encode�
PEM_HEADER�ranger� r� �
PEM_FOOTERr� )Zder_cert_bytesZssr, r� r- �DER_cert_to_PEM_cert� s
"r� c C s\ | � t�stdt ��| �� �t�s0tdt ��| �� tt�tt� � }t�|� dd��S )Nz(Invalid PEM encoding; must start with %sz&Invalid PEM encoding; must end with %sr~ r )
r) r� r� �strip�endswithr� r� r� Z decodebytesr� )Zpem_cert_string�dr, r, r- �PEM_cert_to_DER_cert� s
��r� c
C s� | \}}|d urt }nt}t|||d�}t| ��D}|�|��}|�d�} W d � n1 s\0 Y W d � n1 sz0 Y t| �S )N)r r� T)r r �_create_stdlib_contextr| r� r) r� )
r� rv rw ZhostZportr r� r� ZsslsockZdercertr, r, r- �get_server_certificate� s �
Fr� c C s t �| d�S )Nz )�_PROTOCOL_NAMESr� )Z
protocol_coder, r, r- �get_protocol_name� s r� )lr� r� � collectionsr �enumr Z_Enumr Z_IntEnumr Z_IntFlagr; r r r r r
r r r
r r r r r r r� r r� r r r r r � ImportErrorr r r r r r r r! r" r# r$ r% � _convert_r8 r& r r' � __members__�itemsr� r8 Z_SSLv2_IF_EXISTSr7 r= rG rf r� rx ry rz r{ r| r} r~ r� r� rJ r� r� Z socket_errorru r ZHAS_NEVER_CHECK_COMMON_NAMEZ_RESTRICTED_SERVER_CIPHERSr� r� r� r� r� r� r� r� r� r� r� r r r Z _create_default_https_contextr� r r; r< r� r� r� r} r� r� r� r� r� r� r, r, r, r- �^ s� $ 0������
)
1# 9� z� #� / >�