@OOf(ddlmZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl Z ddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ejXe-Z.Gdde j$j^Z0Gdde!jbZ2Gdd Z3y)!) annotationsN)Sequence) TracebackType)cast)Literal)command)ctx) exceptions)flow)http)io)ConnectionState)Server) UpdateHook)ALERT)Options)commands)events)layers)server)Context)CommandGenerator)HTTPMode) UpstreamMode) asyncio_utilsc8eZdZUdZded<dfd ZddZxZS) MockServerz A mock HTTP "server" that just pretends it received a full HTTP request, which is then processed by the proxy core. http.HTTPFlowr cHt|||j||_yN)super__init__clientr )selfr context __class__s `/var/www/premiumrankchecker/venv/lib/python3.12/site-packages/mitmproxy/addons/clientplayback.pyr"zMockServer.__init__,s '..1 c#VKt|tjr|jjj }t j x|jj_|jj_tjjtjjd|jj|xs |jjj |j|r?tjjtjjd||jjjr]tjjtjjd|jjjtjjtjj!dyt|tjj"tjj$tjj&tjj(tjj*fryt,j/d|yw)N) end_stream replay_flowz Unexpected event during replay: ) isinstancerStartr request raw_contenttimetimestamp_start timestamp_endrr ReceiveHttpRequestHeaderstrailers RequestDataRequestTrailersRequestEndOfMessageResponseHeaders ResponseDataResponseTrailersResponseEndOfMessageResponseProtocolErrorloggerwarning)r$eventcontents r' _handle_eventzMockServer._handle_event0s eV\\ *ii''33G  DII   - 0A0A0O++)) **II%%$+$Ityy/@/@/I/IJ $ + kk--fkk.E.Ea.QRRyy  ))kk--KK//4993D3D3M3MN++))&++*I*I!*LM M   ++ (( ,, 00 11      NN=eWE FsJ'J))r rr%r)rAz events.EventreturnzCommandGenerator[None])__name__ __module__ __qualname____doc____annotations__r"rC __classcell__r&s@r'rr$s "Gr(rcleZdZUded<dfd Zd dZejdf d dZd dZ xZ S) ReplayHandlerzlayers.HttpLayerlayercv|jj}tj|_t ||}t |jj|jjf|_ |jjdk(r6d|j_ |jj|j_|jr|jdj!drlt#j$|jd}t'|t"sJ|j|j(fx|j_|j,_t.|a||jrH|jdj!dr*t3j4|t6j8|_n)t3j4|t6j<|_t?||jA|j:jB|<||_"tGjH|_%y)N)addresshttpsTrz upstream:)& client_conncopyrOPENstaterrr/hostportrschemetls pretty_hostsnimode startswithrparser-rPvia server_connr!r"r HttpLayerrupstreamrN transparentrfork connectionsr asyncioEventdone)r$r optionsr#r%r\r&s r'r"zReplayHandler.__init__Xs!!&&(&++ &'*):):DLL(>(@N1AIIQYYN IIMMO W 7 (OsO?F E>1F 3F4A7F ,AF 3FF F F !F F F )r rrirrDNonerDr)rtstrrprrruzdLiteral[True] | tuple[type[BaseException] | None, BaseException | None, TracebackType | None] | NonerDr)rzcommands.StartHookrDr) rErFrGrIr"rnloggingINFOrsrrJrKs@r'rMrMUsQ $0\\ ; ; ; ;  ;r(rMc8eZdZUdZded<ded<ded<ded <d ed <d Zd ZdZdZddZ dZ dZ e jdddZ e jdddZe jdddZe jdddZy) ClientPlaybackNzasyncio.Task | None playback_taskzhttp.HTTPFlow | Noneinflightz asyncio.Queuequeuerrizset[asyncio.Task] replay_taskscptj|_d|_d|_t |_yr )rfQueuerrtaskrrrms r'r"zClientPlayback.__init__s']]_   Er(ctj|_tj|j d|_y)Nzclient playbackname)r rir create_taskplaybackrrms r'runningzClientPlayback.runnings,{{ *66 MMO"3 r(cK|jr0|jj |jd{yy7#tj$rYywxYwwr )rrrfCancelledErrorrms r'rhzClientPlayback.donesV       % % ' (((( )))  s7'AAAAAAAAAAcK |jjd{|_ |jsJt|j|j}t jj dk(rftj|jd}|jj||j|jjn|jd{|jj#d|_77)#t$rtj!dYGwxYww)Nz!client playback awaiting responserzClient replay has crashed!)rgetrrMrir client_replay_concurrencyrrrnraddadd_done_callbackremove Exceptionr? exception task_done)r$hts r'rzClientPlayback.playbacks"&**.."22DM @}}$}!$-->;;88B>%11 )LA%%))!,''(9(9(@(@A((*$$ JJ " DM#2% @  #=? @sEED ECD0D1D5$EDD=:E<D==Ec|js||jk(ry|jryt|tj r2|j sy|j jy|jyyy)NzCan't replay live flow.zCan't replay intercepted flow.z'Can't replay flow with missing request.z'Can't replay flow with missing content.zCan't replay WebSocket flows.zCan only replay HTTP flows.) liver interceptedr-r HTTPFlowr/r0 websocket)r$fs r'checkzClientPlayback.checksc 66Q$--', ==3 a '99@yy$$,@{{&61r(cr|jdttgd|jdtddy)N client_replayz)Replay client requests from a saved file.rr*zoConcurrency limit on in-flight client replay requests. Currently the only valid values are 1 and -1 (no limit).) add_optionrrrr)r$loaders r'loadzClientPlayback.loads<  SM  7   '  }  r(cd|vrYtjjr? tjtjj}|j|d|vr2tjjdvrt jdyy#t j $r#}t jt|d}~wwxYw)Nrr)rr*zGCurrently the only valid client_replay_concurrency values are -1 and 1.) r rirr read_flows_from_pathsr FlowReadException OptionsErrorr start_replayr)r$updatedflowses r' configurezClientPlayback.configures g %#++*C*C 6001J1JK   e $ &' 1{{44GC --]D 2 // 6 --c!f55 6s-BC (CC zreplay.client.countct|jjtt|jzS)z@ Approximate number of flows queued for replay. )rqsizerrboolrrms r'countzClientPlayback.counts* zz!CT]](;$<<rs" $ 0'&%$"""+203)   8 $.G++.Gb9F,,9xU!U!r(