jjejdZddlmZddlZddlmZmZddlmZm Z m Z ddl m Z ddl mZmZdd lmZdd lmZmZmZdd lmZmZerdd lmZej6eZd ZdZdZ dZ!e"dDcgc] }d|zdz  c}Z#dZ$ddZ%ddZ&ddZ' d dZ(d!dZ)GddZ*GddZ+ycc}w)"zL Implements the HPACK header compression algorithm as detailed by RFC 7541. ) annotationsN) TYPE_CHECKINGAny)HPACKDecodingErrorInvalidTableSizeErrorOversizedHeaderListError)HuffmanEncoder) REQUEST_CODESREQUEST_CODES_LENGTH)decode_huffman) HeaderTupleHeaderWeaklyTypedNeverIndexedHeaderTuple) HeaderTabletable_entry_size)Iterable@ ict|d}t|d}|s0|j|jd|jdS|j||S)zj Provides a header as a unicode string if raw is False, otherwise returns it as a bytestring. rrutf-8)bytes __class__decode)headerrawnamevalues T/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/hpack/hpack.py_unicode_if_neededr$%s[  D &) E  G 4ell76KLL   D% ((cXtjd|||dkrd|}t||dks|dkDrd|}t|t|}||kr t |gS|g}||z}|dk\r"|j |dzdz|d z}|dk\r"|j |t |S) zi Encodes an integer according to the wacky integer encoding rules defined in the HPACK spec. zEncoding %d with %d bitsrz'Can only encode positive integers, got r)Prefix bits must be between 1 and 8, got )logdebug ValueError_PREFIX_BIT_MAX_NUMBERS bytearrayappend)integer prefix_bitsmsg max_numberelementss r#encode_integerr72s II(';?{7yAoQ+/9+Go(5J'##|H zG S.3#-.A  S. OOG X r%c~|dks|dkDrd|}t|t|}d}d}dd|z z } |d|z}||k(rD ||}|dz }|dk\r ||dz |zz }n |||zz }n|dz }|tkDrd|}t|Ct j d ||||fS#t$r} d |}t|| d } ~ wwxYw) z Decodes an integer according to the wacky integer encoding rules defined in the HPACK spec. Returns a tuple of the decoded integer and the number of bytes that were consumed from ``data`` in order to get that integer. rr'r(rr)r+z-Variable integer representation is too long: z3Unable to decode HPACK integer representation from NzDecoded %d, consumed %d bytes)r.r/VARINT_MAX_LENGTHr IndexErrorr,r-) datar3r4r5indexshiftmasknumber next_byteerrs r#decode_integerrCQsQ+/9+Go(5J E E Q_ %D/a4 Z  K  #y3588Fi500F ,,J$RC,S11)4II-vu= 5= /CD8L %3./sAB B<&B77B<c#Kt|tsdt|}t|t |j d}|D] }|||f yw)z Converts a dictionary to an iterable of key-value tuples. This is a HPACK-specific function because it pulls "special-headers" out first and then emits them. zheader_dict not a dict, but c8t|jd S)N:) _to_bytes startswith)ks r#z#_dict_to_iterable..s)A,11$77r%)keyN) isinstancedicttype TypeErrorsortedkeys) header_dictr4rQrKs r#_dict_to_iterablerSsh k4 (,T+->,?@n  7 D$;s###$sAActt|}|tur|S|tur t|}|jdS)zB Convert anything to bytes through a UTF-8 encoded string r)rNrstrencode)r"ts r#rGrGs8 U AEz |E  <<  r%ceZdZdZd dZed dZejddZ d ddZdddZ ddZ dddZ ddd Z dd Z y )Encoderzm An HPACK encoder object. This object takes HTTP headers and emits encoded HTTP/2 header blocks. cbt|_ttt|_g|_yN)r header_tabler r r huffman_codertable_size_changesselfs r#__init__zEncoder.__init__s)'M+ / .0r%c.|jjSz> Controls the size of the HPACK header table. r\maxsizer_s r#header_table_sizezEncoder.header_table_size   (((r%c||j_|jjr|jj |yyr[)r\reresizedr^r1r`r"s r#rfzEncoder.header_table_sizes8$)!    $ $  # # * *5 1 %r%c$g}|jjr0|j|jd|j_t |t r t |}n t|}|D]s}d}t |tr|j }nt|dkDr|d}t|dt|df}|j|j|||udj|}tjd||S)a Takes a set of headers and encodes them into a HPACK-encoded header block. :param headers: The headers to encode. Must be either an iterable of tuples, an iterable of :class:`HeaderTuple `, or a ``dict``. If an iterable of tuples, the tuples may be either two-tuples or three-tuples. If they are two-tuples, the tuples must be of the format ``(name, value)``. If they are three-tuples, they must be of the format ``(name, value, sensitive)``, where ``sensitive`` is a boolean value indicating whether the header should be added to header tables anywhere. If not present, ``sensitive`` defaults to ``False``. If an iterable of :class:`HeaderTuple `, the tuples must always be two-tuples. Instead of using ``sensitive`` as a third tuple entry, use :class:`NeverIndexedHeaderTuple ` to request that the field never be indexed. .. warning:: HTTP/2 requires that all special headers (headers whose names begin with ``:`` characters) appear at the *start* of the header block. While this method will ensure that happens for ``dict`` subclasses, callers using any other iterable of tuples **must** ensure they place their special headers at the start of the iterable. For efficiency reasons users should prefer to use iterables of two-tuples: fixing the ordering of dictionary headers is an expensive operation that should be avoided if possible. :param huffman: (optional) Whether to Huffman-encode any header sent as a literal value. Except for use when debugging, it is recommended that this be left enabled. :returns: A bytestring containing the HPACK-encoded header block. Frrrr%zEncoded header block to %s)r\rir1_encode_table_size_changerLrMrSiterr indexablelenrGaddjoinr,r-) r`headershuffman header_block hpack_headersr sensitive new_headerencodeds r#rVzEncoder.encodesp     $ $    > > @ A(-D   % gt $.g6M &!MM$ JFI&+. & 0 00 Vq"1I #F1I. &)0DEJ   Y H I J((<( .8r%c|\}}|s|nd}tjd|||||stnt}|jj ||}|4|j ||||} |s|jj||| S|\} }} | |j| } | S|j| |||} |s|jj||| S)a} Serializes a header key-value tuple. When sensitive is True, the header will not be added to the header table (see https://www.rfc-editor.org/rfc/rfc7541.html#section-7.1.3 for details), furthermore, the header value will be redacted in debug logs, as "SENSITIVE_REDACTED", to prevent accidental exposure of sensitive information. sSENSITIVE_REDACTEDz:Adding %s=%s to the header table, sensitive:%s, huffman:%s) r,r-INDEX_INCREMENTAL INDEX_NEVERr\search_encode_literalrp_encode_indexed_encode_indexed_literal) r`to_addrvrsr!r" display_valueindexbitmatchrxr=perfects r#rpz Encoder.add)s e%.4I  H      -6$;!!((u5 =**4'JG!!%%dE2N %tW  **51G 22uhG!!%%dE2r%cJt|d}|dxxdzcc<t|S)zD Encodes a header using the indexed representation. r+rr))r7r)r`r=fields r#r~zEncoder._encode_indexedas(ua( aDU|r%cN|r6|jj|}|jj|}tt|d}tt|d}|r|dxxdzcc<|dxxdzcc<dj |t ||t ||gS)z Encodes a header with a literal name and literal value. If ``indexing`` is True, the header will be added to the header table: otherwise it will not. r+rr)r%)r]rVr7rorqr)r`r!r"rrsname_len value_lens r#r}zEncoder._encode_literalis %%,,T2D&&--e4E!#d)Q/"3u:q1  QK4 K aLD Lxx uXeI.> F  r%c@|tk7r t|d}n t|d}|dxxt|zcc<|r|jj |}tt |d}|r |dxxdzcc<dj t|t||gS)zv Encodes a header with an indexed name and a literal value and performs incremental indexing. rr+r)r%)rzr7ordr]rVrorqr)r`r=r"rrsprefixrs r#rzEncoder._encode_indexed_literal~s ( (#E1-F#E1-Fq S]" &&--e4E"3u:q1  aLD Lxxvi(8%@AAr%cd}|jD])}t|d}|dxxdzcc<|t|z }+g|_|S)zd Produces the encoded form of all header table size change context updates. r%rr )r^r7r)r`block size_bytesbs r#rlz!Encoder._encode_table_size_changesT 11 Jz1-A aDDLD U1X E #% r%NreturnNonerintr"rrr)T)rrzIterable[HeaderTuple | tuple[bytes | str, bytes | str] | tuple[bytes | str, bytes | str, bool | None]] | dict[bytes | str, bytes | str]rsboolrrF)rztuple[bytes, bytes]rvrrsrrr)r=rrr) r!rr"rrrrsrrr) r=rr"rrrrsrrr)rr)__name__ __module__ __qualname____doc__rapropertyrfsetterrVrpr~r}rrlr%r#rYrYs 0)) 22 $ k2k  k ). kZ6p *B, r%rYceZdZdZefd dZeddZejddZdddZ ddZ ddZ ddZ dd Z dd Zdd Zy )Decodera  An HPACK decoder object. .. versionchanged:: 2.3.0 Added ``max_header_list_size`` argument. :param max_header_list_size: The maximum decompressed size we will allow for any single header block. This is a protection against DoS attacks that attempt to force the application to expand a relatively small amount of data into a really large header list, allowing enormous amounts of memory to be allocated. If this amount of data is exceeded, a `OversizedHeaderListError ` exception will be raised. At this point the connection should be shut down, as the HPACK state will no longer be usable. Defaults to 64kB. :type max_header_list_size: ``int`` cft|_||_|jj|_yr[)rr\max_header_list_sizeremax_allowed_table_size)r`rs r#razDecoder.__init__s,'M %9!'+&7&7&?&?#r%c.|jjSrcrdr_s r#rfzDecoder.header_table_sizergr%c&||j_yr[rdrjs r#rfzDecoder.header_table_sizes$)!r%c tjd|t|}g}t|}d}d}||kr||}t |dz} t |dz} t |dz} | r|j ||d\} } nY| r|j ||d\} } n?| r&|r d}t||j||d} d} n|j||d\} } | rP|j| |t| d| dz }||jkDrd |jd }t||| z }||kr|j |Dcgc]}t||c}Scc}w#t $r}d }t||d}~wwxYw) a Takes an HPACK-encoded header block and decodes it into a header set. :param data: A bytestring representing a complete HPACK-encoded header block. :param raw: (optional) Whether to return the headers as tuples of raw byte strings or to decode them as UTF-8 before returning them. The default value is False, which returns tuples of Unicode strings :returns: A list of two-tuples of ``(name, value)`` representing the HPACK-encoded headers, in the order they were decoded. :raises HPACKDecodingError: If an error is encountered while decoding the header block. z Decoding %srr)@rNz/Table size update not at the start of the blockrzA header list larger than z has been receivedz!Unable to decode headers as UTF-8)r,r- memoryviewror_decode_indexed_decode_literal_indexr_update_encoding_context_decode_literal_no_indexr1rrr _assert_valid_table_sizer$UnicodeDecodeError)r`r<r data_memrrdata_len inflated_size current_indexcurrentindexed literal_indexencoding_updaterconsumedr4hrBs r#rzDecoder.decodes -&d#%'t9  h&=)G7T>*G!40M#7T>2O#'#7#7]^,$ #'#=#=]^,$ !KC,S1188]^,$(#@#@]^,$ v&!1&)VAY!GG  4#<#<<6t7P7P6QQcdC2377 X %M_h&h %%' 38?@1&q#.@ @@! 35C$S)s 2 3s*E' E"E'"E'' F0E>>FcP|j|jkDr d}t|y)zs Check that the table size set by the encoder is lower than the maximum we expect to have. z3Encoder did not shrink table size to within the maxN)rfrr)r`r4s r#rz Decoder._assert_valid_table_size2s.  ! !D$?$? ?GC', , @r%cjt|d\}}||jkDr d}t|||_|S)zC Handles a byte that updates the encoding context. rz)Encoder exceeded max allowable table size)rCrrrf)r`r<new_sizerr4s r#rz Decoder._update_encoding_context;s@ ,D!4( d11 1=C', ,!)r%ct|d\}}t|jj|}tj d||||fS)zP Decodes a header represented using the indexed representation. r+zDecoded %s, consumed %d)rCrr\ get_by_indexr,r-)r`r<r=rrs r#rzDecoder._decode_indexedGsK)q1xd//<xr%c(|j|dS)NF should_index_decode_literalr`r<s r#rz Decoder._decode_literal_no_indexPs##Du#==r%c(|j|dS)NTrrrs r#rzDecoder._decode_literal_indexSs##Dt#< Decodes a header represented with a literal. r?rFrrNr+zTruncated header blockr)z/Decoded %s, total consumed %d bytes, indexed %s)rLrtobytesrrCr\rrorr rrrpr,r-)r`r<rtotal_consumed indexed_namer not_indexable high_byter=rr!lengthr4r"rs r#rzDecoder._decode_literalVs dJ '<<>D 7T>LH!MQI$t+LH T!12M ,T8"5)E &8++ ,T59F u-F     ! !$ . =     ~%%r%N)rrrrrrr)r<rr rrzIterable[HeaderTuple]r)r<bytes | memoryviewrr)r<rrtuple[HeaderTuple, int])r<rrrrr)rrrrDEFAULT_MAX_HEADER_LIST_SIZErarrfrrrrrrrrrr%r#rrsg*4P@8)) **Q3f-  >=O&r%r)rrr rrr)r2rr3rrr0)r<rr3rrztuple[int, int])rRzdict[bytes | str, bytes | str]rz)Iterable[tuple[bytes | str, bytes | str]])r"zbytes | str | Anyrr),r __future__rloggingtypingrr exceptionsrrr rsr huffman_constantsr r huffman_tabler structrrrtablerrcollections.abcr getLoggerrr, INDEX_NONEr{rzr:ranger/rr$r7rCrSrGrYr)is0r#rs#%[[#B)KK0(g!    27q:AAFa<: ' )>/d$ 4$& !}}@C&C&I ;s1B0