kj). &ddlZddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dZ d Z dd ed edzd edzfd Zded dfdZded eeeeeeeeffdZded eeeeffdZdeed efdZdeed efdZdeed efdZy)N)Iterable)Headers)Request)Response)url)validatec|||vrgS||jd}|Dcgc]}|jc}Scc}w)z Retrieve all tokens for a header key. A number of different headers follow a pattern where each header line can containe comma-separated tokens, and headers can be set multiple times. ,)splitstrip)headerskeytokenstokens f/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/mitmproxy/net/http/http1/read.pyget_header_tokensr s?  ' S\   $F'- .eEKKM .. .s9c>d|vrt|d}d|vryd|vry|dvS)z Checks the message to see if the client connection should be closed according to RFC 2616 Section 8.1. If we don't have a Connection header, HTTP 1.1 connections are assumed to be persistent. connectioncloseTz keep-aliveF)zHTTP/1.1sHTTP/1.1zHTTP/2.0sHTTP/2.0)r) http_versionr rs rconnection_closers@w"7L9 f  V #   requestresponsereturnc|s |j}n|j}|jjdk(ryd|jcxkrdkry|jdvryd|jcxkrdkr!nn|jjdk(ry|j d x}rzt j |}|xxd k(ryxd k(ryxd k(ryxd k(ryxxdk(rnxdk(rnxdk(rnxdk(rnn|ry|dk(sd|vrn y}tj||j dx}rt j|S|syy)aS Returns: The expected body length: - a positive integer, if the size is known in advance - None, if the size in unknown in advance (chunked encoding) - -1, if all data should be read until end of stream. Raises: ValueError, if the content-length or transfer-encoding header is invalid HEADrd)i0i+CONNECTztransfer-encodingchunkedzcompress,chunkedzdeflate,chunkedz gzip,chunkedNcompressdeflategzipidentityzcontent-length) r methodupper status_codegetrparse_transfer_encodingtyping assert_neverparse_content_length)rrr te_strteothercls rexpected_http_body_sizer5/sg //"" >>   !V + (&& -# - .   : - (&& -# -'..2F2F2HI2U0011v1  - -f 5TT0T3DTGUTU=i&:=#'77'B##E**[[) **r*,,R00  rrcLtjd|std|y)Ns ^HTTP/\d\.\d$zUnknown HTTP version: )rematch ValueError)rs rraise_if_http_version_unknownr:s* 88%| 41,1ABCC 5rlinecR |j\}}}|dk(s|jdr dd|}}}d\}}n|dk(r(d|d}}}tj|d\}}|st|jdd \}} |j }| j d\}} } d| z}tj|d\}}|xstj|}|sttj|t||||||||fS#t$r} t d || d} ~ wwxYw) N*/r)rsCONNECTT)checks://)maxsplitzBad HTTP request line: ) r startswithrparse_authorityr9lower partition default_portparser:) r;r)targetrscheme authoritypathhostportrest_path_es r_read_request_linerSsCD'+zz|$  T>V..t4&)3tIFJD$ z !&)63tIF,,YdCJD$  !<<<;LFD \\^F"&.."6 Iq%%D D&D!!D&c |jdd}t|dk(r|jd|\}}}t|}t ||||fS#t $r}t d||d}~wwxYw)NrzBad HTTP response line: )r lenappendintr:r9)r;partsrstatus_code_strreasonr+rRs r_read_response_liner\s E 4# u:? LL 05- ov/* %l3 f ,, E3D8<=1DEsA A A1A,,A1linescbg}|D]}|ddvr5|s td|dd|dddz|jzf|d<? |jdd\}}|j}|s t|j||ft |S#t$rtd|wxYw) z Read a set of headers. Stop once a blank line is reached. Returns: A headers object Raises: exceptions.HttpSyntaxException rs zInvalid headersr(rAs :zInvalid header line: )r9r r rWr)r]retr;namevalues r _read_headersrcs&(CC 7f  !2332wqz3r71:#84::<#GHCG C"jjq1 e $,& D%=)C 3< C #8!ABB Cs ABB.ct|d\}}}}}}}t|dd}t||||||||ddtjd S)z Parse an HTTP request head (request line + headers) from an iterable of lines Args: lines: The input lines Returns: The HTTP request object (without body) Raises: ValueError: The input is malformed. rrAN) rMrNr)rJrKrLrr contenttrailerstimestamp_start timestamp_end)rSrcrtime) r]rMrNr)rJrKrLrr s rread_request_headrjspAS aA=D$ 4E!"I&G    !   rc t|d\}}}t|dd}t||||ddtjdS)z Parse an HTTP response head (response line + headers) from an iterable of lines Args: lines: The input lines Returns: The HTTP response object (without body) Raises: ValueError: The input is malformed. rrAN)rr+r[r rerfrgrh)r\rcrri)r]rr+r[r s rread_response_headrlsT)rus $""#"' /.37o o (4o4ZodDD$D #E #E 3UE5% 67#EL -e -eS%.?(@ -%W<T%[WDd5khr