U
|
ý°dU# ã @ s d dl mZ d dlZd dlZd dlZd dlZddlmZ ejrZd dl m
|
Z
|
ddl m Z m Z ejdd d
|
Zejeef Ze d ¡Zd ZG d d d ZdS )é )Ú annotationsNé )ÚProxySchemeUnsupported)ÚLiteralé )Ú_TYPE_PEER_CERT_RETÚ_TYPE_PEER_CERT_RET_DICTÚ_SelfTÚ SSLTransport)ÚboundÚ _ReturnValuei @ c @ sB e Zd ZdZedddddZdnd dd
|
d dd d dZdddddZdddddZddddZ dodd d d d d Z
|
dpddd d"d#d$Z dqd%d&dd'd(d)d*Z drd+ddd,d-d.Z dsd+ddd,d/d0Zdtdddd1d2d&d
|
d
|
d
|
d3d4d5d6Zddd7d8Zddd9d:Zejdud<d=d>d?d@ZejdAdBd>dCd@Zdvd dEd>dFd@Zd
|
ddGdHZdIddJdKZd
|
ddLdMZd
|
ddNdOZdPddQdRZd
|
ddSdTZdUddVdWdXZdUddYdZZ ddd[d\Z dwdd]d d d^d_Z ejd`ddadbdcZ ejddd+ddedfdcZ ejdgdd]d+dhdidcZ dxdjdkd]dldhdmdcZ dS )yr
|
aL
|
The SSLTransport wraps an existing socket and establishes an SSL connection.
|
|
Contrary to Python's implementation of SSLSocket, it allows you to chain
|
multiple TLS connections together. It's particularly useful if you need to
|
implement TLS within TLS.
|
|
The class supports most of the socket API operations.
|
zssl.SSLContextÚNone)Ú ssl_contextÚreturnc C s t | dstddS )zÒ
|
Raises a ProxySchemeUnsupported if the provided ssl_context can't be used
|
for TLS in TLS.
|
|
The only requirement is that the ssl_context provides the 'wrap_bio'
|
methods.
|
Úwrap_biozXTLS in TLS requires SSLContext.wrap_bio() which isn't available on non-native SSLContextN)Úhasattrr )r © r úPd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\urllib3/util/ssltransport.pyÚ$_validate_ssl_context_for_tls_in_tls" s
|
|
ÿz1SSLTransport._validate_ssl_context_for_tls_in_tlsNTz socket.socketz
|
str | NoneÚbool)Úsocketr Úserver_hostnameÚsuppress_ragged_eofsr c C sH t ¡ | _t ¡ | _|| _|| _|j| j| j|d| _| | jj ¡ dS )zV
|
Create an SSLTransport around socket using the provided ssl_context.
|
)r N)
|
ÚsslÚ MemoryBIOÚincomingÚoutgoingr r r ÚsslobjÚ _ssl_io_loopÚ do_handshake)Úselfr r r r r r r Ú__init__2 s
|
|
|
ÿzSSLTransport.__init__r )r r c C s | S ©Nr ©r r r r Ú __enter__I s zSSLTransport.__enter__z
|
typing.Any)Ú_r c G s | ¡ d S r" )Úclose)r r% r r r Ú__exit__L s zSSLTransport.__exit__Úint)r c C s
|
| j ¡ S r" )r Úfilenor# r r r r) O s zSSLTransport.filenoé ztyping.Any | Nonez int | bytes)ÚlenÚbufferr c C s | ||¡S r" )Ú_wrap_ssl_read)r r+ r, r r r ÚreadR s zSSLTransport.readr )ÚbuflenÚflagsr c C s |dkrt d| |¡S )Nr z+non-zero flags not allowed in calls to recv)Ú
|
ValueErrorr- )r r/ r0 r r r ÚrecvU s zSSLTransport.recvÚ _WriteBufferz
|
int | NonezNone | int | bytes)r, Únbytesr0 r c C s, |dkrt d|d kr t|}| ||¡S )Nr z0non-zero flags not allowed in calls to recv_into)r1 r+ r. )r r, r4 r0 r r r Ú recv_intoZ s
|
zSSLTransport.recv_intoÚbytes)Údatar0 r c
|
C sn |dkrt dd}t|H}| d¡2}t|}||k rV| ||d
¡}||7 }q2W 5 Q R X W 5 Q R X d S )Nr z.non-zero flags not allowed in calls to sendallÚB)r1 Ú
|
memoryviewÚcastr+ Úsend)r r7 r0 ÚcountÚviewÚ byte_viewÚamountÚvr r r Úsendallf s zSSLTransport.sendallc C s |dkrt d| | jj|¡S )Nr z+non-zero flags not allowed in calls to send)r1 r r Úwrite)r r7 r0 r r r r; p s zSSLTransport.send)ÚencodingÚerrorsÚnewlineÚstrz1typing.BinaryIO | typing.TextIO | socket.SocketIO)ÚmodeÚ bufferingrC rD rE r c C s t |dddhks"td|dd|k}d|kp6| }|sD|sDtd|k}d} |r\| d7 } |rh| d7 } t | | ¡}
|
| j jd7 _|dkrd }|d
|
k rtj}|d
|
kr¶|s²td |
|
S |rÎ|rÎt |
|
|
|
|¡} n&|ràt |
|
|¡} n|sètt
|
|
|
|¡} |rü| S t | |||¡} || _ | S ) a
|
Python's httpclient uses makefile and buffered io when reading HTTP
|
messages and we need to support it.
|
|
This is unfortunately a copy and paste of socket.py makefile with small
|
changes to point to the socket directly.
|
ÚrÚwÚbz invalid mode z (only r, w, b allowed)Ú r Néÿÿÿÿr z!unbuffered streams must be binary) Úsetr1 ÚAssertionErrorr ÚSocketIOÚ_io_refsÚioÚDEFAULT_BUFFER_SIZEÚBufferedRWPairÚBufferedReaderÚBufferedWriterÚ TextIOWrapperrG ) r rG rH rC rD rE ÚwritingÚreadingÚbinaryÚrawmodeÚrawr, Útextr r r Úmakefileu s@ zSSLTransport.makefilec C s | | jj¡ d S r" )r r Úunwrapr# r r r r_ © s zSSLTransport.unwrapc C s | j ¡ d S r" )r r& r# r r r r& ¬ s zSSLTransport.close.zLiteral[False]z _TYPE_PEER_CERT_RET_DICT | None)Ú binary_formr c C s d S r" r ©r r` r r r Ú getpeercert¯ s zSSLTransport.getpeercertz Literal[True]z bytes | Nonec C s d S r" r ra r r r rb µ s Fr c C s | j |¡S r" )r rb ra r r r rb ¹ s c C s
|
| j ¡ S r" )r Úversionr# r r r rc ¼ s zSSLTransport.versionztuple[str, str, int] | Nonec C s
|
| j ¡ S r" )r Úcipherr# r r r rd ¿ s zSSLTransport.cipherc C s
|
| j ¡ S r" )r Úselected_alpn_protocolr# r r r re  s z#SSLTransport.selected_alpn_protocolc C s
|
| j ¡ S r" )r Úselected_npn_protocolr# r r r rf Å s z"SSLTransport.selected_npn_protocolz!list[tuple[str, str, int]] | Nonec C s
|
| j ¡ S r" )r Úshared_ciphersr# r r r rg È s zSSLTransport.shared_ciphersc C s
|
| j ¡ S r" )r Ú compressionr# r r r rh Ë s zSSLTransport.compressionz float | None)Úvaluer c C s | j |¡ d S r" )r Ú
|
settimeout)r ri r r r rj Î s zSSLTransport.settimeoutc C s
|
| j ¡ S r" )r Ú
|
gettimeoutr# r r r rk Ñ s zSSLTransport.gettimeoutc C s | j ¡ d S r" )r Ú_decref_socketiosr# r r r rl Ô s z SSLTransport._decref_socketioszbytearray | Nonec
|
C s\ z| | jj||¡W S tjk
|
rV } z"|jtjkrD| jrDW Y ¢
|
dS W 5 d }~X Y nX d S )Nr )r r r. r ÚSSLErrorÚerrnoÚ SSL_ERROR_EOFr )r r+ r, Úer r r r- × s
|
zSSLTransport._wrap_ssl_readztyping.Callable[[], None])Úfuncr c C s d S r" r )r rq r r r r á s zSSLTransport._ssl_io_loopz typing.Callable[[bytes], int])rq Úarg1r c C s d S r" r )r rq rr r r r r æ s z/typing.Callable[[int, bytearray | None], bytes])rq rr Úarg2r c C s d S r" r )r rq rr rs r r r r ë s z"typing.Callable[..., _ReturnValue]zNone | bytes | intr c
|
C sî d}d}|râd}z8|dkr*|dkr*| }n |dkr<||}n
|
|||}W n@ t jk
|
r } z |jt jt jfkrr||j}W 5 d}~X Y nX | j ¡ }| j |¡ |dkr®d}q|t jkr| j t
|
¡}|rÖ| j |¡ q| j ¡ qt t|¡S )z>Performs an I/O loop between incoming/outgoing and the socket.TNF)r rm rn ÚSSL_ERROR_WANT_READÚSSL_ERROR_WANT_WRITEr r. r rA r2 Ú SSL_BLOCKSIZEr rB Ú write_eofÚtypingr: r ) r rq rr rs Z should_loopÚretrn rp Úbufr r r r ô s0
|
|
|
)NT)r* N)r* r )Nr )r )r )N).)F)N)NN) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__Ú staticmethodr r! r$ r' r) r. r2 r5 rA r; r^ r_ r& rx Úoverloadrb rc rd re rf rg rh rj rk rl r- r r r r r r
|
sb
|
û ü
|
ýù 4 ÿ
|
ü)Ú
|
__future__r rR r r rx Ú
|
exceptionsr Ú TYPE_CHECKINGZtyping_extensionsr Ússl_r r ÚTypeVarr ÚUnionÚ bytearrayr9 r3 r rv r
|
r r r r Ú<module> s
|
|
