kjZddlZddlZddlZddlmZddlmZddlmZejeZ ejdZ ejdZ ejdZ ejdZeej"eZd eezd efd Zd eezd efd Zd ed dfdZy)N)Message)Request)Responses^[!#$%&'*+\-.^_`|~0-9a-zA-Z]+$s^(?:0|[1-9][0-9]*)$z^(?:0|[1-9][0-9]*)$)chunkedcompress,chunkeddeflate,chunked gzip,chunkedcompressdeflategzipidentityvaluereturnct|trttj |}ntt j |}|st d|t|S)zLParse a content-length header value, or raise a ValueError if it is invalid.zinvalid content-length header: ) isinstancestrbool_valid_content_length_strmatch_valid_content_length ValueErrorint)rvalids d/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/mitmproxy/net/http/validate.pyparse_content_lengthr'sX%.44U;<*0078 :5)DEE u:c2|jstd|t|tr|}n|j }|j }t jdd|}|tvrtd|tjt|S)zZParse a transfer-encoding header value, or raise a ValueError if it is invalid or unknown.z"invalid transfer-encoding header: z [\t ]*,[\t ]*,z"unknown transfer-encoding header: ) isasciirrrdecodelowerresub_HTTP_1_1_TRANSFER_ENCODINGStypingcastTransferEncoding)rtes rparse_transfer_encodingr)2s ==?=eYGHH%  \\^ B  #r *B --=eYGHH ;;' ,,rmessagecg}g}|jjD]e\}}tj|st d||j xdk(r|j |Pdk(sU|j |g|r |r t d|rt|dkDrt d||jst d|dd |jt|trDd |jcxkrd ksn|jd k(rt d|dd |jt|d}|xxdk(rnxdk(rnxdk(rnxdk(rnnyxxdk(rnxdk(rnxdk(rnxdk(rnn!t|tr&t d|d}tj |yy|r+t|dkDrt d|t#|dyy)z{ Validate HTTP message headers to avoid request smuggling attacks. Raises a ValueError if they are malformed. zinvalid header name: stransfer-encodingscontent-lengthz>message with both transfer-encoding and content-length headersz$multiple transfer-encoding headers: z"unexpected HTTP transfer-encoding rz for dz for response with status code rrrr r r r r z for requestz!multiple content-length headers: N)headersfields_valid_header_namerrr!appendlen is_http11 http_versionrr status_coder)rr% assert_neverr)r*r(clnamer te_parsedothers rvalidate_headersr=Bs B B--! e!''-4TH=> >jjl% % " % ! b  L   r7Q;CB6JK K  4RUIU7CWCWBXY  gx ( 7&& -# -1D1D1K4RUI=\]d]p]p\qr  ,BqE2 T/2CnT=i&:=gw/$rIs ""#   8 $ RZZ BC" #:;&BJJ'=>>>  )9I)JKe  -3; -3C - K$gK$$K$r