gXLdZddlZddlmZmZddlmZmZmZddl m Z ddl m Z m Z ddlmZdd lmZmZej&eZd Zd Zd Zed Dcgc] }d|zdz  c}Z eZdZdZ dZ!dZ"dZ#dZ$GddZ%GddZ&ycc}w#e$reefZY6wxYw)ze hpack/hpack ~~~~~~~~~~~ Implements the HPACK header compression algorithm as detailed by the IETF. N) HeaderTabletable_entry_size)HPACKDecodingErrorOversizedHeaderListErrorInvalidTableSizeError)HuffmanEncoder) REQUEST_CODESREQUEST_CODES_LENGTH)decode_huffman) HeaderTupleNeverIndexedHeaderTuple@ ict|d}t|d}|s"|jd}|jd}|j||S)zj Provides a header as a unicode string if raw is False, otherwise returns it as a bytestring. rrutf-8)bytesdecode __class__)headerrawnamevalues @/var/www/openai/venv/lib/python3.12/site-packages/hpack/hpack.py_unicode_if_neededr+sR  D &) E {{7# W%   D% ((cPtjd|||dkrtd|z|dks|dkDrtd|zt|}||kr t |gS|g}||z}|dk\r"|j |dzdz|d z}|dk\r"|j |t |S) zn This 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 %sr+Prefix bits must be between 1 and 8, got %s)logdebug ValueError_PREFIX_BIT_MAX_NUMBERS bytearrayappend)integer prefix_bits max_numberelementss rencode_integerr08s II(';?{ 7' A  Q+/ 9K G  )5J'##<:n OOWs]c1 2 MGn  ""rc:|dks|dkDrtd|zt|}d}d}dd|z z } |d|z}||k(r+ ||}|dz }|dk\r ||dz |zz }n |||zz }n|dz }*tj d ||||fS#t$rtd|zwxYw) z This 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%z5Unable to decode HPACK integer representation from %rzDecoded %d, consumed %d bytes)r(r) IndexErrorrr&r')datar-r.indexshiftmasknumber next_bytes rdecode_integerr:ZsQ+/ 9K G  )5J E E Q_ %D a4 Z  K  #y3588Fi500F  II-vu= 5=   Cd J   s 8BBc#Kt|tsJt|jd}|D] }|||f yw)z This converts a dictionary to an iterable of two-tuples. This is a HPACK-specific function because it pulls "special-headers" out first and then emits them. c8t|jd S)N:) _to_bytes startswith)ks rz#_dict_to_iterable..s)A,11$77r)keyN) isinstancedictsortedkeys) header_dictrFrBs r_dict_to_iterablerHsM k4 (( (  7 D;s###sAAc~t|ts t|}t|tr|S|j dS)z" Convert string to bytes. r)rC basestringstrrencode)strings rr>r>s4 fj )V.6JFMM'4JJrcveZdZdZdZedZejdZd dZd dZ dZ d dZ d d Z d 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+ / #%rc.|jjSz> Controls the size of the HPACK header table. rRmaxsizerUs rheader_table_sizezEncoder.header_table_size   (((rc||j_|jjr|jj |yyrQ)rRr[resizedrTr+rVrs rr\zEncoder.header_table_sizes8$)!    $ $  # # * *5 1 %rc g}t|tr t|}|jjr0|j |j d|j_|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. FrrrrzEncoded header block to %s)rCrDrHrRr_r+_encode_table_size_changer indexablelenr>addjoinr&r')rVheadershuffman header_blockr sensitives rrLzEncoder.encodesd gt $'0G    $ $    > > @ A(-D   %FI&+. & 0 00 Vq"1I q *IfQi,@AF   G D Exx -  . =rctjd||||\}}|stnt}|jj ||}|4|j ||||}|s|jj|||S|\} }} | r|j| }|S|j| |||}|s|jj|||S)zQ This function takes a header key-value tuple and serializes it. z7Adding %s to the header table, sensitive:%s, huffman:%s) r&r'INDEX_INCREMENTAL INDEX_NEVERrRsearch_encode_literalre_encode_indexed_encode_indexed_literal) rVto_addrjrhrrindexbitmatchencodedr5perfects rrez Encoder.adds E      e-6$;!!((u5 =**4'JG!!%%dE2N %tW **51G 22uhG!!%%dE2rcJt|d}|dxxdzcc<t|S)zD Encodes a header using the indexed representation. r%rr#)r0r)rVr5fields rrpzEncoder._encode_indexed8s(ua( aDU|rcN|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)rSrLr0rdrfr)rVrrrsrhname_len value_lens rrozEncoder._encode_literal@s %%,,T2D&&--e4E!#d)Q/"3u:q1  QK4 K aLD Lxx uXeI.> F  rc@|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)rlr0ordrSrLrdrfr)rVr5rrsrhprefixr{s rrqzEncoder._encode_indexed_literalUs ( (#E1-F#E1-Fq S]" &&--e4E"3u:q1  aLD Lxxvi(8%@AArcd}|jD])}t|d}|dxxdzcc<|t|z }+g|_|S)zd Produces the encoded form of all header table size change context updates. rr )rTr0r)rVblock size_bytess rrbz!Encoder._encode_table_size_changeksQ 11J' A6J qMT !M U:& &E2#% rN)TF)__name__ __module__ __qualname____doc__rWpropertyr\setterrLrerprorqrbrrrOrOs\ %)) 22 Ob/b *B, rrOczeZdZdZefdZedZejdZd dZ dZ dZ dZ d Z d Zd 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|_yrQ)rrRmax_header_list_sizer[max_allowed_table_size)rVrs rrWzDecoder.__init__s,'M %9!'+&7&7&?&?#rc.|jjSrYrZrUs rr\zDecoder.header_table_sizer]rc&||j_yrQrZr`s rr\zDecoder.header_table_sizes$)!rctjd|t|}g}t|}d}d}||kr||}|dzrdnd} |dzrdnd} |dzrdnd} | r|j ||d\} } nW| r|j ||d\} } n=| r$|r t d |j||d} d} n|j||d\} } | rC|j| |t| z }||jkDrtd |jz|| z }||kr|j |Dcgc]}t||c}Scc}w#t$r t d wxYw) 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#TF@rNz/Table size update not at the start of the blockz.A header list larger than %d has been receivedz"Unable to decode headers as UTF-8.)r&r' memoryviewrd_decode_indexed_decode_literal_indexr_update_encoding_context_decode_literal_no_indexr+rrr_assert_valid_table_sizerUnicodeDecodeError)rVr4rdata_memrgdata_len inflated_size current_indexcurrentindexed literal_indexencoding_updaterconsumedhs rrzDecoder.decodes -&d#t9  h&=)G%nd%G%,dNDM'.nd%O#'#7#7]^,$ #'#=#=]^,$ !,I 88]^,$(#@#@]^,$ v&!16!::  4#<#<<2H112 X %Meh&n %%' K8?@1&q#.@ @@! K$%IJ J Ks*E .EE E EcL|j|jkDr tdy)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)r\rrrUs rrz Decoder._assert_valid_table_size s.  ! !D$?$? ?'E  @rcft|d\}}||jkDr td||_|S)zC Handles a byte that updates the encoding context. rz)Encoder exceeded max allowable table size)r:rrr\)rVr4new_sizers rrz Decoder._update_encoding_contextsB ,D!4( d11 1'; "*rct|d\}}t|jj|}tj d||||fS)zP Decodes a header represented using the indexed representation. r%zDecoded %s, consumed %d)r:r rR get_by_indexr&r')rVr4r5rrs rrzDecoder._decode_indexed!sK)q1xd//<xrc&|j|dS)NF_decode_literalrVr4s rrz Decoder._decode_literal_no_index*s##D%00rc&|j|dS)NTrrs rrzDecoder._decode_literal_index-s##D$//rcd}|r |ddz}d}d}n|d}|dz}d}|dz}|r2t||\}} |jj|d} | }d} nP|dd }t|d \} } || | | z} t| | k7r t d |dd zr t | } | | zdz}|| | zd }t|d \} } || | | z} t| | k7r t d |dd zr t | } || | zz }|r t | | } n t| | } |r|jj| | tjd | ||| |fS)z> Decodes a header represented with a literal. r?r~Fr}rNr%zTruncated header blockr#z/Decoded %s, total consumed %d bytes, indexed %s) r:rRrrdrr rr rer&r')rVr4 should_indextotal_consumed indexed_namerz not_indexable high_byter5rrlengthrrs rrzDecoder._decode_literal0s 7T>LH!MQI$t+LH%,M ,T8 > 7T>"5)E &8++ ,T59F u-F     ! !$ . =     ~%%rNr)rrrrDEFAULT_MAX_HEADER_LIST_SIZErWrr\rrrrrrrrrrrrrysh(-I@8)) **SKj  10I&rr)'rloggingtablerr exceptionsrrrrhr huffman_constantsr r huffman_tabler structr r getLoggerrr& INDEX_NONErmrlranger)rJ NameErrorrKrrrr0r:rHr>rOr)is0rrs 0$*8g!    27q:AAFa<:J ' )#D&R $KZZz@&@&w ;uJsB*B B#"B#