nbiW.dZddlmZddlZddlmZmZmZddlm Z ddl m Z m Z ddl mZdd lmZmZdd lmZmZejeZd Zd Zd ZdedDZeefZdZ dZ!dZ"dZ#dZ$dZ%GddZ&GddZ'dS)ze hpack/hpack ~~~~~~~~~~~ Implements the HPACK header compression algorithm as detailed by the IETF. ) annotationsN)HPACKDecodingErrorInvalidTableSizeErrorOversizedHeaderListError)HuffmanEncoder) REQUEST_CODESREQUEST_CODES_LENGTH)decode_huffman) HeaderTupleNeverIndexedHeaderTuple) HeaderTabletable_entry_size@cg|] }d|zdz  S)r).0is _/var/www/html/mdtn/previsions/meteo_cartes/venv/lib/python3.11/site-packages/jh2/hpack/hpack.py r s 888!AqDA:888 ict|d}t|d}|s*|d}|d}|||S)zj Provides a header as a unicode string if raw is False, otherwise returns it as a bytestring. rrutf-8)bytesdecode __class__)headerrawnamevalues r_unicode_if_neededr%*sf   D &)  E &{{7## W%%   D% ( ((rctd|||dkrtd|z|dks|dkrtd|zt|}||krt |gS|g}||z}|dkr&||dzdz|d z}|dk&||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_integerr67s II(';???{{DwNOOOQ+//FTUUU(5J'###<:nn OOWs]c1 2 2 2 MGnn    """rct|dks|dkrtd|zt|}d}d}dd|z z } |d|z}||kr/ ||}|dz }|dkr ||dz |zz }n |||zz }n|dz }.n #t$rtd |zwxYwtd ||||fS) 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(rTr)r+z5Unable to decode HPACK integer representation from %rzDecoded %d, consumed %d bytes)r.r/ IndexErrorrr,r-)datar3r4indexshiftmasknumber next_bytes rdecode_integerr@UsQ+//FTUUU(5J E E A O $D a4 Z    K  ##y3588FFi500F       Cd J    II-vu=== 5=s AA::Bc#Kt|tsJt|d}|D]}|||fVdS)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. cHt|d S)N:) _to_bytes startswith)ks rz#_dict_to_iterable..s ! 8O8OPT8U8U4Ur)keyN) isinstancedictsortedkeys) header_dictrLrHs r_dict_to_iterablerN|sr k4 ( (((( +""$$*U*U V V VD$$;s######$$rct|tst|}t|tr|n|dS)z" Convert string to bytes. r)rI basestringstrrencode)strings rrDrDsF fj ) )V.. J66FMM'4J4JJrczeZdZdZdZedZejdZddZddZ d Z dd Z dd Z d Z d S)Encoderzm An HPACK encoder object. This object takes HTTP headers and emits encoded HTTP/2 header blocks. cxt|_ttt|_g|_dSN)r header_tablerr r huffman_codertable_size_changesselfs r__init__zEncoder.__init__s/'MM+M;OPP"$rc|jjSz> Controls the size of the HPACK header table. rXmaxsizer[s rheader_table_sizezEncoder.header_table_size  ((rcn||j_|jjr|j|dSdSrW)rXraresizedrZr1r\r$s rrbzEncoder.header_table_sizesB$)!   $ 2  # * *5 1 1 1 1 1 2 2rTcXg}t|trt|}|jjr3||d|j_|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)rIrJrNrXrer1_encode_table_size_changer indexablelenrDaddjoinr,r-)r\headershuffman header_blockr! sensitives rrRzEncoder.encodes*d gt $ $ 1'00G   $ .    > > @ @ A A A(-D  % F FFI&+.. & & 00 Vq"1I q **IfQi,@,@AF   G D D E E E Exx --  . ===rFctd||||\}}|stnt}|j||}|7|||||}|s|j|||S|\} }} | r|| }n5| | |||}|s|j|||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_NEVERrXsearch_encode_literalrk_encode_indexed_encode_indexed_literal) r\to_addrprnr#r$indexbitmatchencodedr;perfects rrkz Encoder.adds E        e-6F$$;!((u55 =**4'JJG 3!%%dE222N %tW  3**511GG225%7SSG 3!%%dE222rc`t|d}|dxxdzcc<t|S)zD Encodes a header using the indexed representation. r+rr))r6r)r\r;fields rrvzEncoder._encode_indexed*s5ua(( aDU||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. r+rr)r)rYrRr6rjrlr)r\r#r$ryrnname_len value_lens rruzEncoder._encode_literal2s  5%,,T22D&--e44E!#d))Q//"3u::q11  ! QKKK4 KKK aLLLD LLLxx5??D% :J:JERSSSrc|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. rr+r)r)rrr6ordrYrRrjrlr)r\r;r$ryrnprefixrs rrwzEncoder._encode_indexed_literalEs ( ( (#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 )rZr6r)r\block size_bytess rrhz!Encoder._encode_table_size_change[sa 1 ' 'J' A66J qMMMT !MMM U:&& &EE"$ rN)TF)__name__ __module__ __qualname____doc__r]propertyrbsetterrRrkrvrurwrhrrrrUrUs %%% ))X) 222 OOOOb----^TTTT&BBBB,     rrUc~eZdZdZefdZedZejdZddZ dZ dZ d Z d Z d Zd Zd S)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`` c\t|_||_|jj|_dSrW)rrXmax_header_list_sizeramax_allowed_table_size)r\rs rr]zDecoder.__init__s.'MM %9!'+&7&?###rc|jjSr_r`r[s rrbzDecoder.header_table_sizercrc||j_dSrWr`rfs rrbzDecoder.header_table_sizes$)!!!rFc*td|t|}g}t|}d}d}||kr ||}|dzrdnd} |dzrdnd} |dzrdnd} | r!|||d\} } nv| r!|||d\} } nS| r1|rt d |||d} d} n |||d\} } | rC| | |t| z }||j krtd |j z|| z }||k |  fd |DS#t$rt 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 receivedc0g|]}t|Sr)r%)rhr"s rrz"Decoder.decode..s$@@@1&q#..@@@rz"Unable to decode headers as UTF-8.)r,r- memoryviewrj_decode_indexed_decode_literal_indexr_update_encoding_context_decode_literal_no_indexr1rrr_assert_valid_table_sizeUnicodeDecodeError)r\r:r"data_memrmdata_len inflated_size current_indexcurrentindexed literal_indexencoding_updater!consumeds ` rrzDecoder.decodes2 -&&&d##t99  h&&=)G%n7dd%G%,dN=DDM'.n?dd%O #'#7#78P#Q#Q  #'#=#=h}~~>V#W#W  ,I 88-..9QRR$(#@#@]^^,$$  v&&&!16!::  4#<<<2H34 X %MYh&&b %%''' K@@@@@@@ @! K K K$%IJJ J Ks * E88FcD|j|jkrtddS)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)rbrrr[s rrz Decoder._assert_valid_table_sizes3  !D$? ? ?'E  @ ?rcnt|d\}}||jkrtd||_|S)zC Handles a byte that updates the encoding context. rz)Encoder exceeded max allowable table size)r@rrrb)r\r:new_sizers rrz Decoder._update_encoding_contextsB ,D!44( d1 1 1'(STT T!)rct|d\}}t|j|}td||||fS)zP Decodes a header represented using the indexed representation. r+zDecoded %s, consumed %d)r@r rX get_by_indexr,r-)r\r:r;rr!s rrzDecoder._decode_indexed sT)q11xd/<>>xrc.||dS)NF_decode_literalr\r:s rrz Decoder._decode_literal_no_indexs##D%000rc.||dS)NTrrs rrzDecoder._decode_literal_indexs##D$///rcFd}|r|ddz}d}d}n|d}|dz}d}|dz}|r8t||\}} |j|d} | }d} nn|dd }t|d \} } || | | z} t| | krt d |dd zrt | } | | zdz}|| | zd }t|d \} } || | | z} t| | krt d |dd zrt | } || | zz }|rt | | } nt| | } |r|j| | t d | ||| |fS)z> Decodes a header represented with a literal. r?rFrrNr+zTruncated header blockr)z/Decoded %s, total consumed %d bytes, indexed %s) r@rXrrjrr r r rkr,r-)r\r: should_indextotal_consumed indexed_namer not_indexable high_byter;rr#lengthr$r!s rrzDecoder._decode_literals  -7T>LH!MMQI$t+LH%,M  3,T8<> > 7T> *"5))E &8++  .,T599FF u--F  /   ! !$ . . . =       ~%%rNr)rrrrDEFAULT_MAX_HEADER_LIST_SIZEr]rrbrrrrrrrrrrrrris*-I@@@@8))X) ***MKMKMKMK^      111000I&I&I&I&I&rr)(r __future__rlogging exceptionsrrrrnrhuffman_constantsr r huffman_tabler structr r tablerr getLoggerrr, INDEX_NONErsrrranger/rQrrPrr%r6r@rNrDrUrrrrrs#""""" $#####BBBBBBBB))))))8888888800000000g!!    98uuQxx8885\ % ) ) )###<$$$N $ $ $KKKTTTTTTTTny&y&y&y&y&y&y&y&y&y&r