@OOf ddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z Gd d eZ d ed e fdZGddZy))Sequence) NamedTuple)ctx) exceptions) flowfilter)http)version) NO_RESPONSE) RESPONSESc6eZdZUejed<eed<y) BlockSpecmatches status_codeN)__name__ __module__ __qualname__rTFilter__annotations__int[/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/mitmproxy/addons/blocklist.pyr r s   rr optionreturncj|d|dd}}|jj|d}t|dk7r td|\}} t |}t j |}tj|std|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: )rr) lowersplitlen ValueErrorrrparser getr )rsepremparts flow_pattstatusr flow_filters r parse_specr*say&*C IIK  c1 %E 5zQHIIIv@&k ""9-K == %5fX>?? [k BB @5fX>??@s BB2cDeZdZddZdZdZdejddfdZy) 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)Nr2zCannot parse block_list option z: ) r.roptionsr2r*r!r OptionsErrorappend)r/updatedrspeces r configurezBlockList.configure<s 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.rmetadatarr killrResponsemaker MITMPROXY)r/r?r<s rrequestzBlockList.requestHs ==DJJdii JJ D||D!/3 m,##{2IIK$(MM$6$6((8W=N=N2O%7%DM  r)rN) rrrr0r6r>rHTTPFlowrLrrrr,r,*s()   ( DMM d rr,N)collections.abcrtypingr mitmproxyrrrrr mitmproxy.net.http.status_codesr r r r4r*r,rrrrRsL$ 75  CsCyC0**r