Current File : //usr/local/lib/python3.8/asyncio/__pycache__/sslproto.cpython-38.pyc
U
paJj @ s d dl Z d dlZz d dlZW n ek
r4 dZY nX ddlmZ ddlmZ ddlmZ ddlmZ ddl m
Z
dd Z d
Z d Z
d Zd
ZG dd deZG dd dejejZG dd dejZdS ) N ) base_events) constants) protocols)
transports)loggerc C s" | r t dt }|s d|_|S )Nz(Server side SSL needs a valid SSLContextF)
ValueErrorsslZcreate_default_contextZcheck_hostname) server_sideserver_hostname
sslcontext r
,/usr/local/lib/python3.8/asyncio/sslproto.py_create_transport_context s r Z UNWRAPPEDZ DO_HANDSHAKEZWRAPPEDZSHUTDOWNc @ s~ e Zd ZdZdZdddZedd Zedd Zed
d Z ed d
Z
dddZ d ddZ dd Z
d ddZd ddZdS ) _SSLPipea An SSL "Pipe".
An SSL pipe allows you to communicate with an SSL/TLS protocol instance
through memory buffers. It can be used to implement a security layer for an
existing connection where you don't have access to the connection's file
descriptor, or for some reason you don't want to use it.
An SSL pipe can be in "wrapped" and "unwrapped" mode. In unwrapped mode,
data is passed through untransformed. In wrapped mode, application level
data is encrypted to SSL record level data and vice versa. The SSL record
level is the lowest level in the SSL protocol suite and is what travels
as-is over the wire.
An SslPipe initially is in "unwrapped" mode. To start SSL, call
do_handshake(). To shutdown SSL again, call unwrap().
i Nc C sH || _ || _|| _t| _t | _t | _d| _ d| _
d| _ d| _ dS )a
The *context* argument specifies the ssl.SSLContext to use.
The *server_side* argument indicates whether this is a server side or
client side transport.
The optional *server_hostname* argument can be used to specify the
hostname you are connecting to. You may only specify this parameter if
the _ssl module supports Server Name Indication (SNI).
NF)
_context _server_side_server_hostname
_UNWRAPPED_stater Z MemoryBIO _incoming _outgoing_sslobj
_need_ssldata
_handshake_cb _shutdown_cb)selfcontextr
r r
r
r __init__8 s
z_SSLPipe.__init__c C s | j S )z*The SSL context passed to the constructor.)r r r
r
r r N s z_SSLPipe.contextc C s | j S )z^The internal ssl.SSLObject instance.
Return None if the pipe is not wrapped.
)r r r
r
r
ssl_objectS s z_SSLPipe.ssl_objectc C s | j S )zgWhether more record level data is needed to complete a handshake
that is currently in progress.)r r r
r
r need_ssldata[ s z_SSLPipe.need_ssldatac C s
| j tkS )zj
Whether a security layer is currently in effect.
Return False during handshake.
)r _WRAPPEDr r
r
r wrappeda s z_SSLPipe.wrappedc C sb | j tkrtd| jj| j| j| j| jd| _ t
| _ || _ | j ddd\}}t
|dks^t|S )aL Start the SSL handshake.
Return a list of ssldata. A ssldata element is a list of buffers
The optional *callback* argument can be used to install a callback that
will be called when the handshake is complete. The callback will be
called with None if successful, else an exception instance.
z"handshake in progress or completed)r
r T)only_handshaker )r r RuntimeErrorr Zwrap_bior r r r r
_DO_HANDSHAKEr feed_ssldatalenAssertionErrorr callbackssldataappdatar
r
r do_handshakej s
z_SSLPipe.do_handshakec C sj | j tkrtd| j tkr$td| j ttfks6tt| _ || _| d\}}|g ksf|dgksft|S )a1 Start the SSL shutdown sequence.
Return a list of ssldata. A ssldata element is a list of buffers
The optional *callback* argument can be used to install a callback that
will be called when the shutdown is complete. The callback will be
called without arguments.
zno security layer presentzshutdown in progressr$ ) r r r& _SHUTDOWNr" r' r* r r( r+ r
r
r shutdown s
z_SSLPipe.shutdownc C s2 | j | d\}}|g ks.|dgks.tdS )zSend a potentially "ragged" EOF.
This method will raise an SSL_ERROR_EOF exception if the EOF is
unexpected.
r$ N)r Z write_eofr( r* )r r- r. r
r
r feed_eof s
z_SSLPipe.feed_eofFc
C s | j tkr"|r|g}ng }g |fS d| _|r8| j| g }g }z| j tkrz| j t| _ | j rl| d |rz||fW S | j tkr| j
| j }| | |sqqnJ| j t
kr| j d| _t| _ | jr| n| j tkr| | j
W nz tjtjfk
rl } zRt|dd}|tjtjtjfkrP| j tkrN| j rN| | |tjk| _W 5 d}~X Y nX | jjr| | j
||fS )a Feed SSL record level data into the pipe.
The data must be a bytes instance. It is OK to send an empty bytes
instance. This can be used to get ssldata for a handshake initiated by
this endpoint.
Return a (ssldata, appdata) tuple. The ssldata element is a list of
buffers containing SSL data that needs to be sent to the remote SSL.
The appdata element is a list of buffers containing plaintext data that
needs to be forwarded to the application. The appdata list may contain
an empty buffer indicating an SSL "close_notify" alert. This alert must
be acknowledged by calling shutdown().
FNerrno)r r r r writer' r r/ r" r readmax_sizeappendr0 Zunwrapr r SSLErrorCertificateErrorgetattrSSL_ERROR_WANT_READSSL_ERROR_WANT_WRITESSL_ERROR_SYSCALLr pending)r datar% r. r- chunkexc exc_errnor
r
r r( sZ
z_SSLPipe.feed_ssldatar c
C s4 d| krt |ks n t| jtkrT|t |k rD||d g}ng }|t |fS g }t|}d| _z(|t |k r|| j||d 7 }W nh tj k
r } zHt
|dd}|j dkrtj }|_
|tj tjtjfkrڂ |tj k| _W 5 d}~X Y nX | jjr|| j |t |ks,| jr`q,q`||fS )a Feed plaintext data into the pipe.
Return an (ssldata, offset) tuple. The ssldata element is a list of
buffers containing record level data that needs to be sent to the
remote SSL instance. The offset is the number of plaintext bytes that
were processed, which may be less than the length of data.
NOTE: In case of short writes, this call MUST be retried with the SAME
buffer passed into the *data* argument (i.e. the id() must be the
same). This is an OpenSSL requirement. A further particularity is that
a short write will always have offset == 0, because the _ssl module
does not enable partial writes. And even though the offset is zero,
there will still be encrypted data in ssldata.
r NFr3 ZPROTOCOL_IS_SHUTDOWN)r) r* r r
memoryviewr r r4 r r8 r: reasonr; r3 r< r= r r> r7 r5 )r r? offsetr- viewrA rB r
r
r feed_appdata s6
z_SSLPipe.feed_appdata)N)N)N)F)r )__name__
__module__ __qualname____doc__r6 r propertyr r r! r# r/ r1 r2 r( rG r
r
r
r r $ s
Kr c @ s e Zd ZejjZdd Zd"ddZdd Z dd Z
d
d Z d d
Z e
jfddZdd Zdd Zdd Zd#ddZdd Zedd Zd d Zd d Zd d! ZdS )$_SSLProtocolTransportc C s || _ || _d| _d S )NF)_loop
_ssl_protocol_closed)r loopZ ssl_protocolr
r
r r ! s z _SSLProtocolTransport.__init__Nc C s | j ||S )z#Get optional transport information.)rO _get_extra_infor namedefaultr
r
r get_extra_info' s z$_SSLProtocolTransport.get_extra_infoc C s | j | d S N)rO _set_app_protocol)r protocolr
r
r set_protocol+ s z"_SSLProtocolTransport.set_protocolc C s | j jS rW )rO
_app_protocolr r
r
r get_protocol. s z"_SSLProtocolTransport.get_protocolc C s | j S rW )rP r r
r
r
is_closing1 s z _SSLProtocolTransport.is_closingc C s d| _ | j dS )a
Close the transport.
Buffered data will be flushed asynchronously. No more data
will be received. After all buffered data is flushed, the
protocol's connection_lost() method will (eventually) called
with None as its argument.
TN)rP rO _start_shutdownr r
r
r close4 s z_SSLProtocolTransport.closec C s& | j s"|d| t| d | d S )Nzunclosed transport )source)rP ResourceWarningr_ )r _warnr
r
r __del__? s z _SSLProtocolTransport.__del__c C s | j j}|d krtd| S )Nz*SSL transport has not been initialized yet)rO
_transportr&
is_reading)r trr
r
r re D s z _SSLProtocolTransport.is_readingc C s | j j dS )zPause the receiving end.
No data will be passed to the protocol's data_received()
method until resume_reading() is called.
N)rO rd
pause_readingr r
r
r rg J s z#_SSLProtocolTransport.pause_readingc C s | j j dS )zResume the receiving end.
Data received will once again be passed to the protocol's
data_received() method.
N)rO rd resume_readingr r
r
r rh R s z$_SSLProtocolTransport.resume_readingc C s | j j|| dS )a Set the high- and low-water limits for write flow control.
These two values control when to call the protocol's
pause_writing() and resume_writing() methods. If specified,
the low-water limit must be less than or equal to the
high-water limit. Neither value can be negative.
The defaults are implementation-specific. If only the
high-water limit is given, the low-water limit defaults to an
implementation-specific value less than or equal to the
high-water limit. Setting high to zero forces low to zero as
well, and causes pause_writing() to be called whenever the
buffer becomes non-empty. Setting low to zero causes
resume_writing() to be called only once the buffer is empty.
Use of zero for either limit is generally sub-optimal as it
reduces opportunities for doing I/O and computation
concurrently.
N)rO rd set_write_buffer_limits)r highlowr
r
r ri Z s z-_SSLProtocolTransport.set_write_buffer_limitsc C s | j j S )z,Return the current size of the write buffer.)rO rd get_write_buffer_sizer r
r
r rl o s z+_SSLProtocolTransport.get_write_buffer_sizec C s
| j jjS rW )rO rd _protocol_pausedr r
r
r rm s s z&_SSLProtocolTransport._protocol_pausedc C s< t |tttfs$tdt|j |s,dS | j| dS )zWrite some data bytes to the transport.
This does not block; it buffers the data and arranges for it
to be sent out asynchronously.
z+data: expecting a bytes-like instance, got N)
isinstancebytes bytearrayrC TypeErrortyperH rO _write_appdatar r? r
r
r r4 x s
z_SSLProtocolTransport.writec C s dS )zAReturn True if this transport supports write_eof(), False if not.Fr
r r
r
r
can_write_eof s z#_SSLProtocolTransport.can_write_eofc C s | j d| _dS )zClose the transport immediately.
Buffered data will be lost. No more data will be received.
The protocol's connection_lost() method will (eventually) be
called with None as its argument.
TN)rO _abortrP r r
r
r abort s
z_SSLProtocolTransport.abort)N)NN)rH rI rJ r Z
_SendfileModeZFALLBACKZ_sendfile_compatibler rV rZ r\ r] r_ warningswarnrc re rg rh ri rl rL rm r4 ru rw r
r
r
r rM s$
rM c @ s e Zd ZdZd,ddZdd Zd-d d
Zd d Zd
d Zdd Z dd Z
dd Z dd Z d.ddZ
dd Zdd Zd d Zd d Zd!d" Zd#d$ Zd/d&d'Zd(d) Zd*d+ ZdS )0 SSLProtocolzSSL protocol.
Implementation of SSL on top of a socket using incoming and outgoing
buffers which are ssl.MemoryBIO objects.
FNTc C s t d krtd|d kr tj}n|dkr6td| |sDt||}|| _|rZ|sZ|| _nd | _|| _t |d| _
t | _
d| _|| _|| _| | t| j| | _d | _d| _d| _d| _d | _|| _|| _d S )Nz stdlib ssl module not availabler z7ssl_handshake_timeout should be a positive number, got )r F)r r& r ZSSL_HANDSHAKE_TIMEOUTr r r r _sslcontextdict_extra collectionsdeque_write_backlog_write_buffer_size_waiterrN rX rM _app_transport_sslpipe_session_established
_in_handshake _in_shutdownrd _call_connection_made_ssl_handshake_timeout) r rQ app_protocolr waiterr
r Zcall_connection_madeZssl_handshake_timeoutr
r
r r s@
zSSLProtocol.__init__c C s || _ t|tj| _d S rW )r[ rn r ZBufferedProtocol_app_protocol_is_buffer)r r r
r
r rX s
z SSLProtocol._set_app_protocolc C sD | j d krd S | j s:|d k r.| j | n | j d d | _ d S rW )r cancelled
set_exception
set_resultr rA r
r
r _wakeup_waiter s
zSSLProtocol._wakeup_waiterc C s& || _ t| j| j| j| _| dS )zXCalled when the low-level connection is made.
Start the SSL handshake.
N)rd r r{ r r r _start_handshake)r transportr
r
r connection_made s zSSLProtocol.connection_madec C sn | j r d| _ | j| jj| n| jdk r2d| j_d| _d| _t| ddrT| j
| | d| _d| _ dS )zCalled when the low-level connection is lost or closed.
The argument is an exception object or None (the latter
meaning a regular EOF is received or the connection was
aborted or closed).
FNT_handshake_timeout_handle)
r rN call_soonr[ connection_lostr rP rd r: r cancelr r r r
r
r r s
zSSLProtocol.connection_lostc C s | j dS )z\Called when the low-level transport's buffer goes over
the high-water mark.
N)r[
pause_writingr r
r
r r s zSSLProtocol.pause_writingc C s | j dS )z^Called when the low-level transport's buffer drains below
the low-water mark.
N)r[ resume_writingr r
r
r r s zSSLProtocol.resume_writingc C s" | j dkrdS z| j |\}}W nL ttfk
r< Y n4 tk
rn } z| |d W Y dS d}~X Y nX |D ]}| j| qt|D ]}|rz&| jrt
| j | n | j | W nP ttfk
r Y n8 tk
r
} z| |d W Y dS d}~X Y nX q|
q qdS )zXCalled when some SSL data is received.
The argument is a bytes object.
NzSSL error in data receivedz/application protocol failed to receive SSL data)r r(
SystemExitKeyboardInterrupt
BaseException _fatal_errorrd r4 r r Z _feed_data_to_buffered_protor[
data_receivedr^ )r r? r- r. er@ exr
r
r r s<
zSSLProtocol.data_receivedc C sT zB| j rtd| | t | js@| j
}|r@t d W 5 | j X dS )a Called when the other end of the low-level stream
is half-closed.
If this returns a false value (including None), the transport
will close itself. If it returns a true value, closing the
transport is up to the protocol.
z%r received EOFz?returning true from eof_received() has no effect when using sslN) rd r_ rN get_debugr debugr ConnectionResetErrorr r[ eof_receivedwarning)r Z keep_openr
r
r r - s
zSSLProtocol.eof_receivedc C s4 || j kr| j | S | jd k r,| j||S |S d S rW )r} rd rV rS r
r
r rR C s
zSSLProtocol._get_extra_infoc C s. | j r
d S | jr| nd| _ | d d S )NTr$ )r r rv rs r r
r
r r^ K s
zSSLProtocol._start_shutdownc C s. | j |df | jt|7 _| d S )Nr )r r7 r r) _process_write_backlogrt r
r
r rs T s zSSLProtocol._write_appdatac C s\ | j r$td| | j | _nd | _d| _| jd | j | j
| j | _ |
d S )Nz%r starts SSL handshakeT)r$ r )rN r r r time_handshake_start_timer r r7 Z
call_laterr _check_handshake_timeoutr r r r
r
r r Y s
z SSLProtocol._start_handshakec C s* | j dkr&d| j d}| t| d S )NTz$SSL handshake is taking longer than z! seconds: aborting the connection)r r r ConnectionAbortedError)r msgr
r
r r h s
z$SSLProtocol._check_handshake_timeoutc
C s d| _ | j | jj}z|d k r&|| }W nb ttfk
rJ Y nJ tk
r } z,t |t
j rld}nd}| || W Y d S d }~X Y nX | j
r| j
| j }td| |d | jj|| | |d | jr| j| j | d| _ | j
| j d S )NFz1SSL handshake failed on verifying the certificatezSSL handshake failedz %r: SSL handshake took %.1f msg @@)peercertcipher compressionr T) r r r r r Z getpeercertr r r rn r r9 r rN r r r r r r} updater r r r[ r r r r r r )r Z
handshake_excZsslobjr rA r Zdtr
r
r _on_handshake_completeq s8
z"SSLProtocol._on_handshake_completec
C sP | j d ks| jd krd S ztt| jD ]}| jd \}}|rR| j||\}}n*|rj| j| j}d}n| j| j }d}|D ]}| j
| q|t|k r||f| jd< | jj st | j j
r| j q| jd= | jt|8 _q(W n^ ttfk
r Y nD tk
rJ } z$| jr.| | n | |d W 5 d }~X Y nX d S )Nr r z Fatal error on SSL transport)rd r ranger) r rG r/ r r1 _finalizer4 r! r* Z_pausedrh r r r r r r )r ir? rE r- r@ rA r
r
r r s<
z"SSLProtocol._process_write_backlogFatal error on transportc C sV t |tr(| j r@tjd| |dd n| j||| j| d | jrR| j| d S )Nz%r: %sT)exc_info)message exceptionr rY ) rn OSErrorrN r r r Zcall_exception_handlerrd Z _force_close)r rA r r
r
r r s
zSSLProtocol._fatal_errorc C s d | _ | jd k r| j d S rW )r rd r_ r r
r
r r s
zSSLProtocol._finalizec C s( z| jd k r| j W 5 | X d S rW )r rd rw r r
r
r rv s
zSSLProtocol._abort)FNTN)N)N)r )rH rI rJ rK r rX r r r r r r r rR r^ rs r r r r r r rv r
r
r
r rz s0
.
&
)+
rz )r~ rx r ImportError r r r r logr r r r' r" r0 objectr Z_FlowControlMixinZ TransportrM ZProtocolrz r
r
r
r <module> s*
yx
Mini Shell