U
|
[ ±dtF ã @ sþ d Z ddlmZ ddlmZmZmZmZmZm Z m
|
Z
|
m Z m Z ddl Z ddlmZmZmZ ddlmZmZmZmZmZ ddlmZ dd lmZ dd
|
lm Z m Z m Z m Z m Z m!Z! d Z"G d d d Z#ee$ e e e
|
e% dddZ&ee$ e e e e dddZ'dS )zR
|
wsproto/handshake
|
~~~~~~~~~~~~~~~~~~
|
|
An implementation of WebSocket handshakes.
|
é )Údeque) ÚcastÚDequeÚDictÚ GeneratorÚIterableÚListÚOptionalÚSequenceÚUnionNé )Ú
|
ConnectionÚConnectionStateÚConnectionType)ÚAcceptConnectionÚEventÚRejectConnectionÚ
|
RejectDataÚRequest)Ú Extension)ÚHeaders)Úgenerate_accept_tokenÚgenerate_nonceÚLocalProtocolErrorÚnormed_header_dictÚRemoteProtocolErrorÚsplit_comma_headers 13c @ s e Zd ZdZeddddZeedddZee e
|
dd d
|
Z e e eef dd d d ZeedddZe e ddddZeeddf dddZejedddZeedddZeedddZ e edd d Z eed d d Z ej edd!d"Z!edd#d$Z"dS )%Ú H11Handshakez4A Handshake implementation for HTTP/1.1 connections.N)Úconnection_typeÚreturnc C sV |t jk| _tj| _| jr*t tj¡| _nt tj ¡| _d | _
|
t | _ d | _ d | _d S ©N)r ÚCLIENTÚclientr Ú
|
CONNECTINGÚ_stateÚh11r Ú_h11_connectionÚSERVERÚ _connectionr Ú_eventsÚ_initiating_requestÚ_nonce)Úselfr © r- úHd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\wsproto/handshake.pyÚ__init__* s zH11Handshake.__init__)r c C s | j S r )r$ ©r, r- r- r. Ústate8 s zH11Handshake.statec C s | j S )zÙReturn the established connection.
|
|
This will either return the connection or raise a
|
LocalProtocolError if the connection has not yet been
|
established.
|
|
:rtype: h11.Connection
|
)r( r0 r- r- r. Ú
|
connection< s
|
zH11Handshake.connection)ÚheadersÚpathr c C s> | j rtdtjd||d}t tj¡}| | |¡¡ dS )zóInitiate an upgrade connection.
|
|
This should be used if the request has already be received and
|
parsed.
|
|
:param list headers: HTTP headers represented as a list of 2-tuples.
|
:param str path: A URL path.
|
z?Cannot initiate an upgrade connection when acting as the clientó GET©ÚmethodÚtargetr3 N)r" r r% r r r! Ú receive_dataÚsend)r, r3 r4 Zupgrade_requestZ
|
h11_clientr- r- r. Úinitiate_upgrade_connectionH s ÿ z(H11Handshake.initiate_upgrade_connection)Úeventr c C s d}t |tr || |¡7 }n^t |tr8|| |¡7 }nDt |trR|| |¡7 }n*t |trl|| |¡7 }nt d| d|S )a
|
Send an event to the remote.
|
|
This will return the bytes to send based on the event or raise
|
a LocalProtocolError if the event is not valid given the
|
state.
|
|
:returns: Data to send to the WebSocket peer.
|
:rtype: bytes
|
ó zEvent z$ cannot be sent during the handshake)
|
Ú
|
isinstancer Ú_initiate_connectionr Ú_acceptr Ú_rejectr Ú_send_reject_datar ©r, r< Údatar- r- r. r: [ s
|
|
|
|
|
|
ÿzH11Handshake.send)rD r c C sl | j |p d¡ z| j ¡ }W n$ tjk
|
rB tdt dY nX t|tjsd|tjksd|tj krhqh| j
|
rHt|tj rÂ|j dkr| j | |¡¡ n&| j tt|j|j dd¡ tj| _nt|tjrötj| _| j tt|j|j dd¡ nPt|tjr | j t|jdd¡ n*t|tjrf| j tddd¡ tj| _qt|tjr| j | |¡¡ qd S )
|
zåReceive data from the remote.
|
|
A list of events that the remote peer triggered by sending
|
this data can be retrieved with :meth:`events`.
|
|
:param bytes data: Data received from the WebSocket peer.
|
r= zBad HTTP message©Z
|
event_hintée F)r3 Ú status_codeÚhas_bodyT)rD Ú body_finishedN) r& r9 Z
|
next_eventr% r r r> ZConnectionClosedZ NEED_DATAZPAUSEDr" ÚInformationalResponserG r) ÚappendÚ _establish_client_connectionÚlistr3 r ÚCLOSEDr$ ÚResponseÚ REJECTINGÚDatar rD Ú EndOfMessager Ú_process_connection_request)r, rD r< r- r- r. r9 t sZ ÿ
|
ÿþý
|
ýÿ
|
ýÿ ÿ
|
zH11Handshake.receive_datac c s | j r| j ¡ V q dS )z¢Return a generator that provides any events that have been generated
|
by protocol activity.
|
|
:returns: a generator that yields H11 events.
|
N)r) Úpopleftr0 r- r- r. Úevents¬ s zH11Handshake.eventsc C s¨ |j dkrtdt dd }g }d }d }g }d}d }g } |jD ]¢\}
|
} |
|
¡ }
|
|
|
dkr`t| }nr|
|
dkrv| d¡}q>n\|
|
dkr| t| ¡ q>nB|
|
d kr| }n4|
|
d
|
kr¸| t| ¡ q>n|
|
d krÆ| }n |
|
d krÒ| }| |
|
| f¡ q>|d ksþt d d |D s tdt d|t
|
kr:tdtdt
|
fg|r.dnddd|d krRtdt d| ¡ dkrntdt d|d krtdt dt || |||j d¡d| _ | j S )Nr5 zRequest method must be GETrE r= ó
|
connections hostÚidnaó sec-websocket-extensionsó sec-websocket-keyó sec-websocket-protocols sec-websocket-versionó upgradec s s | ]}| ¡ d kV qdS ©ÚupgradeN©Úlower©Ú.0Útokenr- r- r. Ú <genexpr>Ú s z;H11Handshake._process_connection_request.<locals>.<genexpr>ú%Missing header, 'Connection: Upgrade'z'Missing header, 'Sec-WebSocket-Version'ó Sec-WebSocket-Versioniª i )r3 rG z#Missing header, 'Sec-WebSocket-Key'ó websocketú$Missing header, 'Upgrade: WebSocket'zMissing header, 'Host'Úascii)Ú
|
extensionsÚ extra_headersÚhostÚ subprotocolsr8 )r7 r r r3 r_ r ÚdecodeÚextendrK ÚanyÚWEBSOCKET_VERSIONr r8 r* ) r, r< Úconnection_tokensri rk Úkeyrl r] Úversionr3 ÚnameÚvaluer- r- r. rS · s
|
ÿ
|
|
ÿ
|
ÿ
|
þþ
|
ÿ ÿ
|
ÿ
|
ûz(H11Handshake._process_connection_requestc C sð | j d k stt| j j}|d }t|}ddd|fg}|jd k rv|j| j jkr`td|j | d|j d¡f¡ |j
|
r¨t t t t | j j
|
|j
|
}|r¨| d|f¡ tjd ||j d
|
}t| jrÊtjntj|j
|
| _tj| _| j |¡pîd S ) NrY ©ó Upgrades WebSocket©s
|
Connectionrw s Sec-WebSocket-Acceptzunexpected subprotocol ó Sec-WebSocket-Protocolrh ó Sec-WebSocket-ExtensionsrF ©rG r3 r= ) r* ÚAssertionErrorr rj r Ú subprotocolrl r rK Úencoderi Úserver_extensions_handshaker r
|
Ústrr% rJ r r" r r! r' r( r ÚOPENr$ r& r: )r, r< Zrequest_headersÚnonceÚ accept_tokenr3 ÚacceptsÚresponser- r- r. r@ þ s> ý
|
ÿþ ÿþzH11Handshake._acceptc C s | j tjkrtd| j t|j}|js4| d¡ tj |j
|
|d}| j |¡pRd}tj | _|js|| j t ¡ ¡pvd7 }tj| _|S )Nz)Connection cannot be rejected in state %s)s content-lengthó 0r{ r= )r1 r r# r rM r3 rH rK r% rO rG r& r: rP r$ rR rN )r, r< r3 r
rD r- r- r. rA % s ÿ
|
|
zH11Handshake._rejectc C s` | j tjkr td| j | j tj|jd¡p4d}|j r\|| j t
|
¡ ¡pPd7 }tj | _ |S )Nz$Cannot send rejection data in state )rD r= ) r1 r rP r r& r: r% rQ rD rI rR rN r$ rC r- r- r. rB 6 s
|
ÿz H11Handshake._send_reject_data)Úrequestr c
|
C s || _ t | _d|j d¡fddd| jfdtfg}|jrT| dd |j¡ d ¡f¡ |j rði }|j D ] }t
|
|t svt | ¡ ||j< qdg }| ¡ D ]D\}}| d ¡}t
|
|tr¾|rÖ| |¡ q| d
|
|| d ¡f ¡ q|rð| d d |¡f¡ tjd |j d ¡||j d} | j | ¡p dS )Ns HostrW rv rx s Sec-WebSocket-Keyre ry z, rh ó %s; %srz ó , r5 r6 r= )r* r r+ rk r~ rp rl rK Újoinri r> r r| Úofferrt ÚitemsÚboolr% r r8 rj r& r: )
|
r, r r3 ZoffersÚeri rt ÚparamsZbnamer] r- r- r. r? D sD ûþÿ
|
|
|
|
ýz!H11Handshake._initiate_connectionc C s | j d k st| jd k s td }d }g }d }d}g }|jD ]z\}} | ¡ }|dkr^t| }q:nH|dkrrt| }q:n4|dkr| }q:n$|dkr| d¡}q:n|dkr¦| }q:| || f¡ q:|d ksÐtdd |D sÞt d
|
t
|
d | ¡ d krøt d t
|
d t | j}
|
||
|
krt dt
|
d |d k rF|| j j krFt d| t
|
d t |ttt | j j} t| jrntjntj| | jjd | _tj| _ t | ||dS )Nr= rV rX s sec-websocket-acceptrZ rh r[ c s s | ]}| ¡ d kV qdS r\ r^ r` r- r- r. rc s z<H11Handshake._establish_client_connection.<locals>.<genexpr>rd rE rf rg zBad accept tokenzunrecognized subprotocol r )ri rj r} ) r* r| r+ r3 r_ r rm rK ro r r r rl Úclient_extensions_handshaker r
|
r ri r r" r r! r' r& Z trailing_datar( r r r$ r ) r, r< Úacceptrq r r} r] r3 rt ru r ri r- r- r. rL o s|
|
ÿ ÿ ÿ
|
|
|
þ ÿ
|
ý ÿz)H11Handshake._establish_client_connectionc C s d | jj| j| j¡S )Nz{}(client={}, state={}))ÚformatÚ __class__Ú__name__r" r1 r0 r- r- r. Ú__repr__° s
|
ÿzH11Handshake.__repr__)#r Ú
|
__module__Ú __qualname__Ú__doc__r r/ Úpropertyr r1 r r r2 r r Úbytesr r; r r: r9 r rU r% r rS r r@ r rA r rB r? rJ rL r r- r- r- r. r ' s0
|
þ 8 þ G',þ Ar )Ú requestedÚ supportedr c
|
C sì i }| D ]h}| dd¡d ¡ }|D ]J}|j|kr$| |¡}t|trV|rnd||j< q$|dk r$| d¡||j< q$q|règ }| ¡ D ]Z\}}| d¡} t|tr²|s¦t| | ¡ q|dkrÊ| d| ¡ q| d | |f ¡ qd
|
|
|¡S dS ) zAgree on the extensions to use returning an appropriate header value.
|
|
This returns None if there are no agreed extensions
|
ú;r r TNrh r= s %sr r ) ÚsplitÚstriprt r r> r r~ r r| rK r )
|
r r r r rt Ú extensionr ri r Z
|
name_bytesr- r- r. r ¶ s.
|
|
|
|
|
|
r )Úacceptedr r c C sf g }| D ]X}| dd¡d ¡ }|D ]&}|j|kr$| |¡ | |¡ qq$td| t dq|S )Nr r r zunrecognized extension rE )r r rt ÚfinalizerK r r )r¡ r ri r rt r r- r- r. r Ú s
|
|
|
ÿr )(r Ú collectionsr Útypingr r r r r r r r
|
r r% r2 r r r rU r r r r r ri r r Z utilitiesr r r r r r rp r r r r r r- r- r- r. Ú<module> s* ,
|
þ % þ
|
