kjn ddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z Gd d eZ d e d e fd ZGddZy))Sequence) NamedTuple)ctx) exceptions) flowfilter)http)version) NO_RESPONSEc6eZdZUejed<eed<y) BlockSpecmatches status_codeN)__name__ __module__ __qualname__rTFilter__annotations__intc/home/ubuntu/.cache/pipx/cdb5059130eadca/lib/python3.12/site-packages/mitmproxy/addons/blocklist.pyr r s   rr optionreturnc|d|dd}}|j|d}t|dk7r td|\}} t|}t j |}t ||S#t$rtd|wxYw)zl Parses strings in the following format, enforces number of segments: /flow-filter/status rNz-Invalid number of parameters (2 are expected)zInvalid HTTP status code: )r r)splitlen ValueErrorrrparser )rsepremparts flow_pattstatusr flow_filters r parse_specr'say&*C IIc1 E 5zQHIIIv@&k ""9-K [k BB @5fX>??@s A))BcDeZdZddZdZdZdejddfdZy) BlockListrNcg|_y)N)items)selfs r__init__zBlockList.__init__(s &( rcB|jdttgdy)N block_lista Block matching requests and return an empty response with the specified HTTP status. Option syntax is "/flow-filter/status-code", where flow-filter describes which requests this rule should be applied to and status-code is the HTTP status code to return for blocked requests. The separator ("/" in the example) can be any character. Setting a non-standard status code of 444 will close the connection without sending a response. ) add_optionrstr)r,loaders rloadzBlockList.load+s"  SM   rcd|vrNg|_tjjD])} t |}|jj|+yy#t $r!}t jd|d||d}~wwxYw)Nr/zCannot parse block_list option z: ) r+roptionsr/r'rr OptionsErrorappend)r,updatedrspeces r configurezBlockList.configure9s 7 "DJ++00 (%f-D  !!$' ( # "$119&A3Gs A A>A99A>flowcz|js|js |jsy|jD]}|j |sd|j d<|j tk(r|jHtjj|j dtji|_y)NT blocklistedServer)headers)responseerrorliver+r metadatarr killrResponsemaker MITMPROXY)r,r<r9s rrequestzBlockList.requestEs ==DJJdii JJ D||D!/3 m,##{2IIK$(MM$6$6((8W=N=N2O%7%DM  r)rN) rrrr-r3r;rHTTPFlowrIrrrr)r)'s()   ( DMM d rr)N)collections.abcrtypingr mitmproxyrrrrr mitmproxy.net.http.status_codesr r r1r'r)rrrrOsI$ 7  CsCyC,**r