U
f|1 @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Zd dl
Zddl m Z ddl
mZ ddlmZmZ ddlmZ ddlmZ eZed ZeeZe d
Z e d Z d d
dddZ d d
dddZ G dd dej!e Z"G dd deZ#G dd deZ$dS ) ) annotationsN )
_TYPE_BODY)HTTPHeaderDict)HTTPSConnection_get_default_user_agent)ConnectionError)BaseHTTPResponseTs ^[!#$%&'*+\-.^_`|~0-9a-z]+$s* [\0\x00\x0a\x0d\r\n]|^[ \r\n\t]|[ \r\n\t]$bytesbool)namereturnc C s t t| S )a5
"An implementation that validates fields according to the definitions in Sections
5.1 and 5.5 of [HTTP] only needs an additional check that field names do not
include uppercase characters." (https://httpwg.org/specs/rfc9113.html#n-field-validity)
`http.client._is_legal_header_name` does not validate the field name according to the
HTTP 1.1 spec, so we do that here, in addition to checking for uppercase characters.
This does not allow for the `:` character in the header name, so should not
be used to validate pseudo-headers.
)r RE_IS_LEGAL_HEADER_NAMEmatch)r
r I/opt/alt/python38/lib/python3.8/site-packages/urllib3/http2/connection.py_is_legal_header_name s r )valuer c C s t t| S )aK
"A field value MUST NOT contain the zero value (ASCII NUL, 0x00), line feed
(ASCII LF, 0x0a), or carriage return (ASCII CR, 0x0d) at any position. A field
value MUST NOT start or end with an ASCII whitespace character (ASCII SP or HTAB,
0x20 or 0x09)." (https://httpwg.org/specs/rfc9113.html#n-field-validity)
)r RE_IS_ILLEGAL_HEADER_VALUEsearch)r r r r _is_illegal_header_value, s r c @ sD e Zd ZdZdZddddZdddd Zd
d d d
dddZdS )
_LockedObjectaX
A wrapper class that hides a specific object behind a lock.
The goal here is to provide a simple way to protect access to an object
that cannot safely be simultaneously accessed from multiple threads. The
intended use of this class is simple: take hold of it with a context
manager, which returns the protected object.
)lock_objr
)objc C s t | _|| _d S N) threadingRLockr r )selfr r r r __init__D s
z_LockedObject.__init__r c C s | j | jS r )r acquirer r r r r __enter__H s
z_LockedObject.__enter__ztype[BaseException] | NonezBaseException | Noneztypes.TracebackType | NoneNone)exc_typeexc_valexc_tbr c C s | j d S r )r release)r r& r' r( r r r __exit__L s z_LockedObject.__exit__N)__name__
__module__ __qualname____doc__ __slots__r r$ r* r r r r r 6 s
r c s e Zd Zd-ddddd fdd
Zd d
d d Zdd
fd
d ZdddddddZddddddZd.dddddZdddddZ d/ddd ddd d d Z
d!d
d"d#Z d0d$d$d$d%ddd&d d'd'd'ddd( d)d*Z dd
fd+d, Z
ZS )1HTTP2ConnectionNstrz
int | Nonez
typing.Anyr% )hostportkwargsr c sV | | _d | _g | _d|ks&d|kr.tdt j||f| | jd k rRtdd S )Nproxy proxy_configz$Proxies aren't supported with HTTP/2z%Tunneling isn't supported with HTTP/2) _new_h2_conn_h2_conn
_h2_stream_headersNotImplementedErrorsuperr _tunnel_host)r r2 r3 r4 __class__r r r V s
zHTTP2Connection.__init__z)_LockedObject[h2.connection.H2Connection]r! c C s t jjdd}tt jj|dS )NT)Z client_side)config)h2r@ ZH2Configurationr
connectionZ H2Connection)r r@ r r r r7 e s z HTTP2Connection._new_h2_connc s@ t | j&}| | }r2| j| W 5 Q R X d S r )r< connectr8 Zinitiate_connection data_to_sendsocksendall)r connrD r> r r rC i s
zHTTP2Connection.connect)methodurlr4 r c K s d|krt dd|kr t d|p&d| _| | d| jkrXd| j d| jpPd }n| j d| jphd }| jd
| jd | f | jd | f | jd
| f | j}| | _
W 5 Q R X dS )zputrequest
This deviates from the HTTPConnection method signature since we never need to override
sending accept-encoding headers or the host header.
skip_hostz`skip_host` isn't supportedskip_accept_encodingz&`skip_accept_encoding` isn't supported/:[z]:i )s :schemes httpss :methods
:authoritys :pathN) r; _request_url_validate_pathr2 r3 r: appendencoder8 Z get_next_available_stream_idr9 )r rH rI r4 authorityrG r r r
putrequestp s
zHTTP2Connection.putrequestz str | bytes)headervaluesr c G s t |tr| n|}| }t|s8tdt| |D ]D}t |trR| n|}t|rptdt| | j||f q| j| q>|
| j W 5 Q R dS t |t
r| }zzt |tr|j | j |dd | }r| j| n@|D ].}|j | j |dd | }r| j| q|
| j W n& tk
rZ tdt| Y nX W 5 Q R X dS )zSend data to the server.
`data` can be: `str`, `bytes`, an iterable, or file-like objects
that support a .read() method.
Nr\ readF)r_ Tz6`data` should be str, bytes, iterable, or file. got %r)r9 r r8 rD rE rF hasattrrc blocksizerW r1 rR Z send_datar_ r TypeErrortype)r rb rG rD chunkr r r send sD
zHTTP2Connection.sendhttpz typing.Mapping[str, str] | None)r2 r3 r^ schemer c C s t dd S )Nz;HTTP/2 does not support setting up a tunnel through a proxy)r; )r r2 r3 r^ rk r r r
set_tunnel s zHTTP2Connection.set_tunnel
HTTP2Responsec C s d }t }| j}d}|s| jd }r||}|D ]}t|tjjrt }|j
D ]6\} }
| dkrtt |
}qV|
| d|
d qVq8t|tjjr||j7 }||j|j q8t|tjjr8d}q8| } r| j| qW 5 Q R X |d k stt||| jt|dS )NFi s :statusasciiT)statusr^ request_urlrb ) bytearrayr8 rE recvZ receive_datarW rA eventsZResponseReceivedr r^ intdecodeaddZ DataReceivedrb Zacknowledge_received_dataZflow_controlled_lengthr] Z StreamEndedrD rF AssertionErrorrm rO r ) r ro rb rG r_ Z
received_datars eventr^ rU r rD r r r getresponse sD
zHTTP2Connection.getresponseT)preload_contentdecode_contentenforce_content_lengthz_TYPE_BODY | Noner ) rH rI bodyr^ rz r{ r| r4 r c K s d|kr| j dk r | j | j | || |p2i }| D ],\} }
| dkr\|
dkr\q r r r 3 s
zHTTP2Connection.close)N)N)NNrj )NN)r+ r, r- r r7 rC rT rZ ra ri rl ry r r
__classcell__r r r> r r0 U s*
0 . "'r0 c s\ e Zd Zdddddddd fd d
Zedd d d
Zdd ddZdd ddZ ZS )rm Frt r r1 r r r% )ro r^ rp rb r{ r c s* t j||ddd ||d || _d| _d S )N zHTTP/2)ro r^ versionversion_stringreasonr{ rp r )r< r _datalength_remaining)r ro r^ rp rb r{ r> r r r F s zHTTP2Response.__init__r! c C s | j S r )r r# r r r rb \ s zHTTP2Response.datac C s d S r r r# r r r get_redirect_location` s z#HTTP2Response.get_redirect_locationc C s d S r r r# r r r r c s zHTTP2Response.close)F) r+ r, r- r propertyrb r r r r r r> r rm D s rm )%
__future__r loggingrer typestypingZ h2.configrA Z
h2.connectionZ h2.eventsZ_base_connectionr _collectionsr rB r r
exceptionsr responser Zorig_HTTPSConnectionTypeVarr
getLoggerr+ logcompiler r r r Genericr r0 rm r r r r s0
p