ei_TdZddlmZddlZddlmZmZddlmZm Z m Z ddl m Z ddl mZmZdd lmZdd lmZmZmZdd lmZmZerdd lmZejeZd ZdZdZ de!dDZ"dZ#d.dZ$d/dZ%d0d"Z&d1d&Z'd2d)Z(Gd*d+Z)Gd,d-Z*dS)3zL 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@cg|] }d|zdz  S)r).0is O/home/div/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hpack/hpack.py rs :::AAFa<::: iheaderrrawboolreturnrct|d}t|d}|s<||d|dS|||S)zj Provides a header as a unicode string if raw is False, otherwise returns it as a bytestring. rrutf-8)bytes __class__decode)r r!namevalues r_unicode_if_neededr+#ss   D &)  E M G 4 4ell76K6KLLL   D% ( ((rintegerint prefix_bits bytearrayctd|||dkrd|}t||dks|dkrd|}t|t|}||krt |gS|g}||z}|dkr&||dzdz|d z}|dk&||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_NUMBERSr/append)r,r.msg max_numberelementss rencode_integerr>0s II(';???{{AAAooQ+//G+GGoo(5J'###|H zG S..3#-...A  S.. OOG X  rdatar&tuple[int, int]c|dks|dkrd|}t|t|}d}d}dd|z z } |d|z}||kr/ ||}|dz }|dkr ||dz |zz }n |||zz }n|dz }.n'#t$r} d |}t|| d } ~ wwxYwtd ||||fS) 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. rr1r2rTr3r5z3Unable to decode HPACK integer representation from NzDecoded %d, consumed %d bytes)r8r9 IndexErrorrr6r7) r?r.r;r<indexshiftmasknumber next_byteerrs rdecode_integerrJOs&Q+//G+GGoo(5J E E Q_ %D/a4 Z    K  ##y3588FFi500F   ///LDLL %%3./II-vu=== 5=sAA<< B BB  header_dictdict[bytes | str, bytes | str])Iterable[tuple[bytes | str, bytes | str]]c#Kt|ts!dt|}t|t |d}|D]}|||fVdS)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 cHt|d S)N:) _to_bytes startswith)ks rz#_dict_to_iterable..s)A,,11$777r)keyN) isinstancedicttype TypeErrorsortedkeys)rKr;r[rUs r_dict_to_iterabler\vs k4 ( (@T+->->@@nn  7 7   D$$;s######$$rr*bytes | str | Anyct|}|tur|S|turt|}|dS)zB Convert anything to bytes through a UTF-8 encoded string r%)rXr&strencode)r*ts rrQrQsC U AEzz ||E  <<  rceZdZdZddZeddZejddZ d d!dZd"d#dZ d$dZ d"d%dZ d"d&dZ d'dZ dS)(Encoderzm An HPACK encoder object. This object takes HTTP headers and emits encoded HTTP/2 header blocks. r#Nonecxt|_ttt|_g|_dSN)r header_tabler r r huffman_codertable_size_changesselfs r__init__zEncoder.__init__s6'MM+ /  .0rr-c|jjSz> Controls the size of the HPACK header table. rgmaxsizerjs rheader_table_sizezEncoder.header_table_size  ((rr*cn||j_|jjr|j|dSdSrf)rgrpresizedrir:rkr*s rrqzEncoder.header_table_sizesB$)!   $ 2  # * *5 1 1 1 1 1 2 2rTheadersIterable[HeaderTuple | tuple[bytes | str, bytes | str] | tuple[bytes | str, bytes | str, bool | None]] | dict[bytes | str, bytes | str]huffmanr"r&czg}|jjr3||d|j_t |t rt |}n t|}|D]}d}t |tr |j }nt|dkr|d}t|dt|df}|| |||d |}td||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)rgrtr:_encode_table_size_changerVrWr\iterr indexablelenrQaddjoinr6r7) rkrvrx header_block hpack_headersr sensitive new_headerencodeds rr`zEncoder.encodes=p    $ .    > > @ @ A A A(-D  % gt $ $ *.g66MM &!MMM$ J JFI&+.. & & 00 Vq"1I #F1I.. &)0D0DEJ   Y H H I I I I((<(( .888rFto_addtuple[bytes, bytes]rctd||||\}}|stnt}|j||}|7|||||}|s|j|||S|\} }} | r|| }n5| | |||}|s|j|||S)z6 Serializes a header key-value tuple. z7Adding %s to the header table, sensitive:%s, huffman:%s) r6r7INDEX_INCREMENTAL INDEX_NEVERrgsearch_encode_literalr~_encode_indexed_encode_indexed_literal) rkrrrxr)r*indexbitmatchrrDperfects rr~z Encoder.adds E        e-6F$$;!((u55 =**4'JJG 3!%%dE222N %tW  3**511GG22uhG 3!%%dE222rrDc`t|d}|dxxdzcc<t|S)zD Encodes a header using the indexed representation. r5rr3)r>r&)rkrDfields rrzEncoder._encode_indexedMs5ua(( aDU||rr)rc|r4|j|}|j|}tt|d}tt|d}|r |dxxdzcc<|dxxdzcc<d|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. r5rr3r)rhr`r>r}rr&)rkr)r*rrxname_len value_lens rrzEncoder._encode_literalUs  5%,,T22D&--e44E!#d))Q//"3u::q11  ! QKKK4 KKK aLLLD LLLxx uXeI.>.> F   rc|tkrt|d}nt|d}|dxxt|zcc<|r|j|}tt |d}|r|dxxdzcc<dt|t||gS)zv Encodes a header with an indexed name and a literal value and performs incremental indexing. rr5r3r)rr>ordrhr`r}rr&)rkrDr*rrxprefixrs rrzEncoder._encode_indexed_literaljs ( ( (#E1--FF#E1--Fq S]]"  5&--e44E"3u::q11  ! aLLLD LLLxxvi(8(8%@AAArcd}|jD]4}t|d}|dxxdzcc<|t|z }5g|_|S)zd Produces the encoded form of all header table size change context updates. rr )rir>r&)rkblock size_bytesbs rrzz!Encoder._encode_table_size_changes^ 1  Jz1--A aDDDDLDDD U1XX EE"$ rNr#rdr#r-r*r-r#rd)T)rvrwrxr"r#r&F)rrrr"rxr"r#r&)rDr-r#r&) r)r&r*r&rr&rxr"r#r&) rDr-r*r&rr&rxr"r#r&)r#r&)__name__ __module__ __qualname____doc__rlpropertyrqsetterr`r~rrrrzrrrrcrcs 0000)))X) 2222 $ kkkkkZ/////b     *BBBBB,      rrcceZdZdZefddZeddZejdd ZdddZ ddZ d dZ d!dZ d!dZ d!dZd"dZdS)#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`` max_header_list_sizer-r#rdc\t|_||_|jj|_dSrf)rrgrrpmax_allowed_table_size)rkrs rrlzDecoder.__init__s.'MM %9!'+&7&?###rc|jjSrnrorjs rrqzDecoder.header_table_sizerrrr*c||j_dSrfrorus rrqzDecoder.header_table_sizes$)!!!rFr?r&r!r"Iterable[HeaderTuple]ctd|t|}g}t|}d}d}||kr@||}t |dz} t |dz} t |dz} | r!|||d\} } nx| r!|||d\} } nU| r3|rd}t||||d} d} n | ||d\} } | rY| | |t| d| dz }||j krd |j d }t||| z }||k@| fd |DS#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 %srr3@rNz/Table size update not at the start of the blockrzA header list larger than z has been receivedc0g|]}t|Sr)r+)rhr!s rrz"Decoder.decode..s$@@@1&q#..@@@rz!Unable to decode headers as UTF-8)r6r7 memoryviewr}r"_decode_indexed_decode_literal_indexr_update_encoding_context_decode_literal_no_indexr:rrr _assert_valid_table_sizeUnicodeDecodeError)rkr?r!data_memrvdata_len inflated_size current_indexcurrentindexed literal_indexencoding_updater consumedr;rIs ` rr(zDecoder.decodesQ -&&&d##%'t99  h&&=)G7T>**G!400M#7T>22O #'#7#7]^^,$$  #'#=#=]^^,$$ ! 2KC,S11188]^^,$(#@#@]^^,$$  8v&&&!1&)VAY!G!GG  4#<<<dt7PdddC23777 X %M_h&&h %%''' 3@@@@@@@ @! 3 3 35C$S))s 2 3s F++ G 5GG cH|j|jkrd}t|dS)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)rqrr)rkr;s rrz Decoder._assert_valid_table_sizes1  !D$? ? ?GC',, , @ ?rcrt|d\}}||jkrd}t|||_|S)zC Handles a byte that updates the encoding context. rz)Encoder exceeded max allowable table size)rJrrrq)rkr?new_sizerr;s rrz Decoder._update_encoding_context'sF ,D!44( d1 1 1=C',, ,!)rtuple[HeaderTuple, int]ct|d\}}t|j|}td||||fS)zP Decodes a header represented using the indexed representation. r5zDecoded %s, consumed %d)rJrrg get_by_indexr6r7)rkr?rDrr s rrzDecoder._decode_indexed3sT)q11xd/<>>xrc0||dS)NF should_index_decode_literalrkr?s rrz Decoder._decode_literal_no_index<s##Du#===rc0||dS)NTrrrs rrzDecoder._decode_literal_index?s##Dt#<< Decodes a header represented with a literal. r?rFrrNr5zTruncated header blockr3z/Decoded %s, total consumed %d bytes, indexed %s) r"rJrgrr}rr rrr~r6r7)rkr?rtotal_consumed indexed_namer not_indexable high_byterDrr)lengthr;r*r s rrzDecoder._decode_literalBs,  37T>LH!MMQI$t+LH T!122M  3,T8< *"5))E &8++  .,T599FF u--F  /   ! !$ . . . =       ~%%rN)rr-r#rdrrr)r?r&r!r"r#rr)r?r&r#r-)r?r&r#r)r?r&rr"r#r)rrrrDEFAULT_MAX_HEADER_LIST_SIZErlrrqrr(rrrrrrrrrrrs#*4P@@@@@8)))X) ****Q3Q3Q3Q3Q3f----        >>>>====L&L&L&L&L&L&rr)r rr!r"r#r)r,r-r.r-r#r/)r?r&r.r-r#r@)rKrLr#rM)r*r]r#r&)+r __future__rloggingtypingrr exceptionsrrr rxr huffman_constantsr r huffman_tabler structrrrtablerrcollections.abcr getLoggerrr6 INDEX_NONErrranger9rr+r>rJr\rQrcrrrrrs#"""""%%%%%%%%[[[[[[[[[[######BBBBBBBB))))))KKKKKKKKKK00000000)((((((g!!    ;:q::: ' ) ) ) )>$$$$N$$$$& ! ! ! !vvvvvvvvr@&@&@&@&@&@&@&@&@&@&r