6h%SSKJr SSKrSSKrSSKrSSKrSSKJrJr SSKJ r SSK J r \R(aSSK r SSK JrJrJrJrJr SSKJrJr \R(aSS KJr \\\\\\44rS rS rSrS rS r \\\ S.r!S\"S'Sr#S r$Sr%Sr&Sr'Sr(S r)Sr*\#\$\%\&\'\(\)\*S.r+S\"S'Sr,S r-S r.S r/\,\-\.\/S.r0S\"S'Sr1S r2\1\2S.r3S\"S'Sr4S r5\4\5S.r6S\"S'Sr7S r8\7\8S.r9S\"S'Sr:S r;\:\;S.rS?\P5rR"S@SA\Q5rS"SBSC\P5rT"SDSE\Q5rU"SFSG\P5rV"SHSI\V5rW"SJSK\Q5rX"SLSM\X5rY\R(a\\R\T-\W-rZ\\S\U-\Y-r[0r\SN\"SO'/r]SP\"SQ'SnSRjr^"SSS/5r_"STSU\_5r`"SVS1\_5raSoSWjrbSXSY/0rcSZ\"S['"S\S]5rd"S^S_5re"S`Sa5rfg)p) annotationsN)ProtocolType)_)attr)encodingerrorpycompatutilwireprototypes)cborutil stringutil)Iteratori)s stream-begins stream-endsencodeddict[bytes, int] STREAM_FLAGS )scommand-requests command-datascommand-responseserror-responses text-outputsprogressssender-protocol-settingssstream-settings FRAME_TYPES)snew continuationsmores have-dataFLAGS_COMMAND_REQUEST)reosFLAGS_COMMAND_DATAFLAGS_COMMAND_RESPONSEFLAGS_SENDER_PROTOCOL_SETTINGSFLAGS_STREAM_ENCODING_SETTINGSzdict[int, dict[bytes, int]]FRAME_TYPE_FLAGSz|)itemsappendgetjoin)mappingvaluekvnamemapflagsvals r?rr@rArCrDrErFr0bytesrOr strmethodrSrGr:rHr2rKrKsq$S!IwwC H''s#K WW# F GG Egg5!G   rHrKc[[[U5-5n[R"S[U55nUSSUSS&[R "SUSXU5 US-U-US'XVSS&U$) z#Assemble a frame into a byte array.z This can be used by user-facing applications and tests for creating frames easily without having to type out a bunch of constants. Request ID and stream IDs are integers. Stream flags, frame type, and flags can be specified by integer or named constant. Flags can be delimited by `|` to bitwise OR them together. If the payload begins with ``cbor:``, the following string will be evaluated as Python literal and the resulting object will be fed into a CBOR encoder. Otherwise, the payload is interpreted as a Python byte string literal.  rrr&scbor:rHN)rCrDrErFr0rO) splitrArrr# startswithr*r streamencoderevalpythonliteral unescapestrr_) sfieldsrCrDrE frametype frameflagsrOfinalstreamflagsflag finalflags validflagss r2makeframefromhumanstringrosG,WWT1 FGMDIII8}H!!$' <   T 2 2  D )  ( K *  N J!),J  &   T* *J #d) #J ' (##((  ! !*">">wqr{"K L  ((1 $  rHcUSSUS--SUS--n[R"SUS5up#nUSnUS -S - nUS -n[XX4Xg5$) zParse a unified framing protocol frame header from a buffer. The header is expected to be in the buffer at offset 0 and the buffer is expected to be large enough to hold a full header. rri@rrXrrr)r[ unpack_fromr6)data framelengthrCrDrE typeflagsrirjs r2 parseheaderrx'svq'C$q'M)EDGO;K'-'9'9&$'J$IQIT!a'IT!J y rHc[[5nURU5nUS:XagU[:wa#[R"[ S5X!4-5e[ U5nURUR5n[U5UR:wa7[R"[ S5UR[U54-5e[URURURURURU5$)zRead a unified framing protocol frame from a file object. Returns a 3-tuple of (type, flags, payload) for the decoded frame or None if no frame is available. May raise if a malformed frame is seen. rNs+received incomplete frame: got %d bytes: %ss'frame length error: expected %d; got %d)rYrZreadintor AbortrrxreadrBrNrKrCrDrErFr0)fhheader readcounthrOs r2 readframer@s( )F F#IA~%%kk < =! "  FAggahhG 7|qxxkk 8 9xxW& '   QZZ!''7 rHstreamc## SU0nU(aX7S'U(aXgS'SR[R"U55nSnSn U(d U [-n O U [-n U(a U [ -n XxX-n U[ U 5- n[ U 5U:XaU[ U5:a U [-n URU[U U S9v U [-(dOMU(asUR[5nSn [ U5[:Xa[n O[n URS 5S:XdeSn URU[U US9v U (ag Mrg 7f) zCreate frames necessary to transmit a request to run a command. This is a generator of bytearrays. Each item represents a frame ready to be sent over the wire to a peer. nameargsredirectrHrTrCrFr0rOFrN)r*rrdFLAG_COMMAND_REQUEST_NEW!FLAG_COMMAND_REQUEST_CONTINUATION FLAG_COMMAND_REQUEST_EXPECT_DATArN FLAG_COMMAND_REQUEST_MORE_FRAMESr_FRAME_TYPE_COMMAND_REQUESTr|DEFAULT_MAX_FRAME_SIZEFLAG_COMMAND_DATA_CONTINUATIONFLAG_COMMAND_DATA_EOSFRAME_TYPE_COMMAND_DATA) rrCcmdargsdatafh maxframesizeredirectruoffsetr0rOdones r2createcommandframesrbsn S>D W $[ 88H))$/ 0D F  - -E 6 6E  5 5E 56#g, w<< 'FSY,> 5 5E-   88 7 :;;56DD4y226-{{1~,,,""#. # 'sEE outputstreamcSR[R"SS055nUR(aUR U5nSnU(dgOSnUR U[ [UUS9$)NrHstatussokTFrCrFr0rOencoded)r*rrdstreamsettingssentencoder_FRAME_TYPE_COMMAND_RESPONSE"FLAG_COMMAND_RESPONSE_CONTINUATION)rrCoverallrs r2createcommandresponseokframerswhhx,,i-?@AG   --(   *0  rHc## UR5nSnX4XB-nU[U5- nU[U5:HnU(a[nO[nUR U[ UUUS:gS9v U(agM\7f)zACreate an empty payload frame representing command end-of-stream.rrHrN)flushrNFLAG_COMMAND_RESPONSE_EOSrr_r)rrCrrOrchunkrr0s r2createcommandresponseeosframesrs llnG F !67#e*W% -E6E.sN     ) sA0A2cpSURURS.S.nSH.n[X$5nUcMXSS[R"U5'M0 SR [ R"U55nUR(aURU5nSnOSnURU[[UUS 9$) Nr)surls mediatype)rlocation)size fullhashes fullhashseedserverdercertsservercadercertsrrHTFr) url mediatypegetattrr bytestrr*rrdrrr_rr)rrClocationruar,rOrs r2$createalternatelocationresponseframersLL",,  D$  5: h..q1 2hhx,,T23G   --(   *0  rHc## SSU0S.nU(aX4SS'SR[R"U55nURU[[ US9v g7f)Nerrormessage)rrrrHr)r*rrdr_rr)rrCmessagermrs r2createcommanderrorresponsersi    A #( Ghhx,,Q/0G   *'  sAAc## [U5[::deSR[R"USU0/S.55nUR U[ SUS9v g7f)NrHmsg)typerrr)rNrr*rrdr_FRAME_TYPE_ERROR_RESPONSE)rrCmsgerrtyperOs r2createerrorframer(sm s8- -- -hh $c]O  G   (  sAAc#*# /nUGH%upVn[U[5(d [S5eUH#n[U[5(aM[S5e UH#n [U [5(aM[S5e URSS5R S5nUV s/sH#oRSS5R S5PM% nn UV s/sH#oRSS5R S5PM% nn SU0n U(aXlS 'U(aX|S 'UR U 5 GM( S R [R"U55n [U 5U:a [S 5eURU[S U S9v gs sn fs sn f7f)arCreate a text output frame to render text to people. ``atoms`` is a 3-tuple of (formatting string, args, labels). The formatting string contains ``%s`` tokens to be replaced by the corresponding indexed entry in ``args``. ``labels`` is an iterable of formatters to be applied at rendering time. In terms of the ``ui`` class, each atom corresponds to a ``ui.write()``. s!must use bytes formatting stringssmust use bytes for argumentssmust use bytes for labelsasciireplacezutf-8strictrrslabelsrHs$cannot encode data in a single framerrN) isinstancerU ValueErrordecoderr(r*rrdrNr_FRAME_TYPE_TEXT_OUTPUT)rrCatomsr atomdicts formattingrlabelsarglabelrr^atomrOs r2createtextoutputframer?sI$) &*e,,AB BCc5)) !@AAEeU++ !=>>  &&w :AA'J GKKd),33G\rSrSrSr\4SSjjrSrSrSr g) bufferingcommandresponseemitterizaHelper object to emit command response frames intelligently. Raw command response data is likely emitted in chunks much smaller than what can fit in a single frame. This class exists to buffer chunks until enough data is available to fit in a single frame. TODO we'll need something like this when compression is supported. So it might make sense to implement this functionality at the stream level. cDXlX lX0l/UlSUlgNr)_stream _requestid_maxsize_chunks _chunkssize)rPrrCrs r2__init__(bufferingcommandresponseemitter.__init__s" #$  rHc# # UcUR5ShvN gURRU5nU(dg[U5UR:a|UR5ShvN SnXX R-nU[U5- nURR UR [[USS9v U[U5:XagMa[U5UR-UR:a:URRU5 U=R[U5- sl gUR5ShvN URRU5 [U5Ul gGNbGNN67f)zSend new data for emission. Is a generator of new frames that were derived from the new input. If the special input ``None`` is received, flushes all buffered data to frames. NrTrFr0rOr) _flushrrrNrr_rrrrrr()rPrurrs r2send$bufferingcommandresponseemitter.sendsM <{{} $ $ ||""4(  t9t}} ${{} $ $Ff}}&<=#e*$ll,,OO6<! -SY&" t9t'' '$-- 7 LL   %   D )   ;;=   D!t9k %* %: !s5FE;AF.E>/CF F 1F>FFc## SRUR5n[U5UR::de/URSS&SUlU(dgUR R UR[[USS9v g7f)NrHrTr) r*rrNrrrr_rrr)rPrOs r2r&bufferingcommandresponseemitter._flushsw((4<<(7|t}},,, Q ll$$ OO.4 %  sBB )rrrrrN)rCrArrAreturnNone) r;r<r=r>r?rrrrrGr:rHr2rrzs5 ;Q!$47 ?%B rHrc\rSrSrSr\R SSj5r\R SSj5r\R SSj5r Sr g) Encoderi9A protocol class for the various encoder implementations.c[eNNotImplementedErrorrPrus r2rEncoder.encode!!rHc[errrPs r2r Encoder.flushrrHc[errrs r2finishEncoder.finishrrHr:NrrU) r;r<r=r>r?abcabstractmethodrrrrGr:rHr2rrsQC"" "" ""rHrc@\rSrSrSr\R SSj5rSrg)Decoderirc[errrs r2rDecoder.decoderrHr:Nr) r;r<r=r>r?rrrrGr:rHr2rrsC""rHrc@\rSrSrSrS SjrS SjrS SjrS SjrSr g) identityencoderiz3Encoder for the "identity" stream encoding profile.cgrr:)rPuis r2ridentityencoder.__init__s rHcU$rr:rs r2ridentityencoder.encode rHcgNrHr:rs r2ridentityencoder.flushrHcgrr:rs r2ridentityencoder.finish r rHr:Nrrr) r;r<r=r>r?rrrrrGr:rHr2rrs= rHrc,\rSrSrSrSSjrSSjrSrg) identitydecoderiz3Decoder for the "identity" stream encoding profile.cPU(a[R"[S55eg)Ns6identity decoder received unexpected additional values)r r{r)rPr extraobjss r2ridentitydecoder.__init__s$ ++KL  rHcU$rr:rs r2ridentitydecoder.decoderrHr:Nr r)r;r<r=r>r?rrrGr:rHr2rrs= rHrc<\rSrSrSSjrS SjrS SjrS SjrSrg) zlibencodericBSSKnX lUR5Ulgr)zlib_zlib compressobj _compressor)rPrrs r2rzlibencoder.__init__s ++-rHc8URRU5$rrcompressrs r2rzlibencoder.encode#((..rHc`URRURR5$r)rrr Z_SYNC_FLUSHrs r2rzlibencoder.flush&s%%%djj&=&=>>rHcrURRURR5nSUlU$r)rrrZ_FINISHrPress r2rzlibencoder.finish+s/$$TZZ%8%89 rH)rrNr r r;r<r=r>rrrrrGr:rHr2rrs. /? rHrc(\rSrSrSSjrSSjrSrg) zlibdecoderi1cSSKnU(a[R"[S55eUR 5Ulg)Nrs2zlib decoder received unexpected additional values)rr r{r decompressobj _decompressor)rPrrrs r2rzlibdecoder.__init__2s5 ++GH "//1rHc8URRU5$rr/ decompressrs r2rzlibdecoder.decode<!!,,T22rHr/Nr rr;r<r=r>rrrGr:rHr2r,r,1s 23rHr,c<\rSrSrSSjrS SjrS SjrS SjrSrg) zstdbaseencoderi@cdSSKJn X lURUS9nUR 5Ulg)Nrzstd)level)r<_zstdZstdCompressorrr)rPr=r<cctxs r2rzstdbaseencoder.__init__As. """/++-rHc8URRU5$rrrs r2rzstdbaseencoder.encodeHr!rHc`URRURR5$r)rrr?COMPRESSOBJ_FLUSH_BLOCKrs r2rzstdbaseencoder.flushKs%%%djj&H&HIIrHcrURRURR5nSUlU$r)rrr?COMPRESSOBJ_FLUSH_FINISHr's r2rzstdbaseencoder.finishQs/$$TZZ%H%HI rH)rr?N)r=rArrrr*r:rHr2r9r9@s./J rHr9c,^\rSrSrSU4SjjrSrU=r$)zstd8mbencoderiWc$>[TU]S5 g)Nr)superr)rPr __class__s r2rzstd8mbencoder.__init__Xs rHr:r r;r<r=r>rrG __classcell__rOs@r2rLrLWs rHrLc(\rSrSrSSjrSSjrSrg)zstdbasedecoderi\cXSSKJn URUS9nUR5Ulg)Nrr;)max_window_size)r>r<ZstdDecompressorr.r/)rP maxwindowsizer<dctxs r2rzstdbasedecoder.__init__]s)$$]$C!//1rHc8URRU5$rr2rs r2rzstdbasedecoder.decodecr5rHr6N)rYrArrrr7r:rHr2rUrU\s 2 3rHrUc,^\rSrSrSU4SjjrSrU=r$)zstd8mbdecoderigcl>U(a[R"[S55e[TU]SS9 g)Ns5zstd8mb decoder received unexpected additional valuesi)rY)r r{rrNr)rPrrrOs r2rzstd8mbdecoder.__init__hs2 ++JK  {3rHr:r rQrSs@r2r_r_gs 44rHr_z&dict[bytes, tuple[EncoderT, DecoderT]]STREAM_ENCODERSz list[bytes]STREAM_ENCODERS_ORDERcd[(agSSKJn UR U(a([ [ 4[S'[RS5 [[4[S'[RS5 [[4[S'[RS5 g![a SnNf=f)Nrr;szstd-8mbszlibidentity) rbr>r< __version__ ImportErrorrLr_rcr(rr,rrr;s r2populatestreamencodersrh}s   (6'G $$$[1 +[9OG  )$3_#EOK   - sB B/.B/cZ\rSrSr%SrS\S'S\S'S S SjjrS SjrS rg )riz5Represents a logical unidirectional series of frames.rArDbool_activecXlX lgr)rDrk)rPrDactives r2rstream.__init__s   rHcvSnUR(dU[-nSUl[XRXRX45$)mCreate a frame to be sent out over this stream. Only returns the frame instance. Does not actually send it. rT)rkSTREAM_FLAG_BEGIN_STREAMr_rD)rPrCrFr0rOrEs r2r_stream.makeframes: || 3 3KDL }}k5  rH)rkrDNFrDrArmrjrr) rCrArFrAr0rArOrUrrY) r;r<r=r>r?__annotations__rr_rGr:rHr2rrsA?M M  &) 25 @E  rHc^^\rSrSr%SrS\S'S S U4SjjjrS SjrS SjrS Sjr S r U=r $) inputstreamiz,Represents a stream used for receiving data.zDecoder | None_decoderc.>[TU]XS9 SUlg)Nrm)rNrrxrPrDrmrOs r2rinputstream.__init__s 1 rHcU[;a"[R"[S5U-5e[US"X5Ulg)zSet the decoder for this stream. Receives the stream profile name and any additional CBOR objects decoded from the stream encoding settings frame payloads. sunknown stream decoder: %srN)rbr r{rrx)rPrrRrs r2 setdecoderinputstream.setdecoders<  &++a =>EF F'-a0? rHc^UR(dU$URRU5$r)rxrrs r2rinputstream.decodes%}}K}}##D))rHcZUR(dgURR5$r)rxrrs r2rinputstream.flushs!}}}}""$$rH)rxrsrtrRrUrrr) r;r<r=r>r?rurr~rrrGrRrSs@r2rwrws+6 @*%%rHrwc^\rSrSr%SrS\S'S\S'S\S'SSU4S jjjrSS jrSS jrSS jr S r SSSjjr SSjr Sr U=r$)riz*Represents a stream used for sending data.rjrzEncoder | None_encoderz bytes | None _encodernamecJ>[TU]XS9 SUlSUlSUlg)NrzF)rNrrrrr{s r2routputstream.__init__s* 1"'  rHcU[;a"[R"[S5U-5e[US"U5UlX lg)zDSet the encoder for this stream. Receives the stream profile name. sunknown stream encoder: %srN)rbr r{rrr)rPrrRs r2 setencoderoutputstream.setencodersB  &++a =>EF F'-a04  rHc^UR(dU$URRU5$r)rrrs r2routputstream.encodes#}}K}}##D))rHcZUR(dgURR5$r)rrrs r2routputstream.flushs}}}}""$$rHc\UR(dgURR5 gr)rrrs r2routputstream.finishs}} rHc"SnUR(dU[-nSUlU(a0UR(d[R"S5eU[ -nU[ :XaU[-(aSUl[XRXbX45$)rprTs@attempting to send encoded frame without sending stream settings) rkrqrr ProgrammingErrorSTREAM_FLAG_ENCODING_APPLIEDFRAME_TYPE_STREAM_SETTINGS!FLAG_STREAM_ENCODING_SETTINGS_EOSr_rD)rPrCrFr0rOrrEs r2r_outputstream.makeframes || 3 3KDL **,,  7 7K 0 099&*D # }}k5  rHcUR(aUR(agSR[R"UR 55nUR U[[U5$)zCreate a stream settings frame for this stream. Returns frame data or None if no stream settings frame is needed or has already been sent. NrH) rrr*rrdrr_rr)rPrCrOs r2makestreamsettingsframe$outputstream.makestreamsettingsframe#sR }} 7 7((8001B1BCD~~  & -    rH)rkrrrrsrtrr) rCrArFrAr0rArOrUrrjrrY)rCrArbytearray | None)r;r<r=r>r?rurrrrrr_rrGrRrSs@r2rrs4!! !* % " " "  "  "  "  " H  rHcrURS-(a#[R"SUR-5eg)NrsAserver should only write to even numbered streams; %d is not even)rDr r)rs r2ensureserverstreamr5s8 $$ 028// B  rHcontentencodingsrezdict[bytes, list[bytes]]DEFAULT_PROTOCOL_SETTINGSc\rSrSr%SrS\S'SSSjjrSSjrSSjrSSjr SS jr SS jr S S!S jjr S"S jr S#SjrS$SjrS%SjrSSjrSSjrSSjrSSjrSSjrSSjrSSjrSrg )& serverreactoriBa~ Holds state of a server handling frame-based protocol requests. This class is the "brain" of the unified frame-based protocol server component. While the protocol is stateless from the perspective of requests/commands, something needs to track which frames have been received, what frames to expect, etc. This class is that thing. Instances are modeled as a state machine of sorts. Instances are also reactionary to external events. The point of this class is to encapsulate the state of the connection and the exchange of frames, not to perform work. Instead, callers tell this class when something occurs, like a frame arriving. If that activity is worthy of a follow-up action (say *run a command*), the return value of that handler will say so. I/O and CPU intensive operations are purposefully delegated outside of this class. Consumers are expected to tell instances when events occur. They do so by calling the various ``on*`` methods. These methods return a 2-tuple describing any follow-up action(s) to take. The first element is the name of an action to perform. The second is a data structure (usually a dict) specific to that action that contains more information. e.g. if the server wants to send frames back to the client, the data structure will contain a reference to those frames. Valid actions that consumers can be instructed to take are: sendframes Indicates that frames should be sent to the client. The ``framegen`` key contains a generator of frames that should be sent. The server assumes that all frames are sent to the client. error Indicates that an error occurred. Consumer should probably abort. runcommand Indicates that the consumer should run a wire protocol command. Details of the command to run are given in the data structure. wantframe Indicates that nothing of interest happened and the server is waiting on more frames from the client before anything interesting can be done. noop Indicates no additional action is required. Known Issues ------------ There are no limits to the number of partially received commands or their size. A malicious client could stream command request data and exhaust the server's memory. Partially received commands are not acted upon when end of input is reached. Should the server error if it receives a partial request? Should the client send a message to abort a partially transmitted request to facilitate graceful shutdown? Active requests that haven't been responded to aren't tracked. This means that if we receive a command and instruct its dispatch, another command with its request ID can come in over the wire and there will be a race between who responds to what. zlist[Iterator[bytearray]]_bufferedframegenscXlX lSUlSUl/Ul0Ul0Ul0Ul[5Ul SUl [[5Ul [5 g)aConstruct a new server reactor. ``deferoutput`` can be used to indicate that no output frames should be instructed to be sent until input has been exhausted. In this mode, events that would normally generate output frames (such as a command response being ready) will instead defer instructing the consumer to send those frames. This is useful for half-duplex transports where the sender cannot receive until all data has been transmitted. initialrN)_ui _deferoutput_state_nextoutgoingstreamidr_incomingstreams_outgoingstreams_receivingcommandsset_activecommands_protocolsettingsdecoderdictr_sendersettingsrh)rPr deferoutputs r2rserverreactor.__init__sh'  %&""$ " ""$ #u(,% $$=> rHcZURS-(d.SUlUR[S5UR-5$URUR;aeUR [ -(d!SUlUR[S55$[UR5URUR'UR [-(aSUl[R"S5eUR [-(aURUR URURURURUR S.nUR#UR5nU(d#[R"SUR-5eU"U5$)zProcess a frame that has been received off the wire. Returns a dict with an ``action`` key that details what action, if any, the consumer should take next. rerroreds/received frame with even numbered stream ID: %dsNreceived frame on unknown inactive stream without beginning of stream flag sets8support for decoding stream payloads not yet implemented)rprotocol-settings-receivingidlecommand-receivingrsunhandled state: %s)rDr_makeerrorresultrrrErqrwrr rSTREAM_FLAG_END_STREAM_onframeinitial_onframeprotocolsettings _onframeidle_onframecommandreceiving_onframeerroredr))rPrKhandlersmeths r2 onframerecvserverreactor.onframerecvsj ~~!$DK((DE..!  >>!6!6 6$$'??( ,,85@4OD ! !%.. 1   ; ;$DK((K    5 5%%enn5,,,0,I,I&&"&"?"?,,  ||DKK((()?$++)MN NE{rHc^^^^[T5 [T[5(a [T5mUUUU4SjnTR U"55$)zSignal that objects are ready to be sent to the client. ``objs`` is an iterable of objects (typically a generator) that will be encoded via CBOR and added to frames, which will be sent to the client. c 3># SnSn[T T5n[T5n[U[R 5(aNU(a[ R""S5eT R%T5nU(aUv ['T TU5v SnSnM{U(a[ R""S5eU(d5T R%T5nU(aUv [)T T5nU(aUv Sn[U[R*5(a'URUR,5HnUv M O[U[R.5(aD[0R2"UR45HnURU5HnUv M M! O:[0R6"U5HnURU5HnUv M M! GM![a= URS5HnUv M U(a[ T T5HnUv M O[ R a7n[T TURUR5HnUv M SnAOySnAf[a8n[T TS[R"U5-SS9HnUv M SnAO9SnAff=f![a$n[T TSU-SS9HnUv M SnAOSnAff=fT R8R;T5 g7f)NFTs%sserverrs:alternatelocationresponse seen after initial output objects(object follows alternatelocationresponse)rnext StopIterationrrr WireprotoCommandErrorrr messageargs Exceptionrr forcebytestrrr alternatelocationresponserrrrencodedresponseruindefinitebytestringresponserstreamencodebytestringfromiterchunksrdrremove) emittedalternatelocationsentemitterorKerobjsrCrPrs r2 sendframes?serverreactor.oncommandresponsereadyobjects..sendframessG$) !5fiHGT A >y I "'KB"Iq15-"& ,#44G# & > >y I "'K!=VY O "'K"&"!^%C%CDD%,\\!&&%9E"'K&:$>FF&.%L%LHH&E*1e)<&+ *=&&.%:%:1%=E)0e)<&+ *=&>{% !(d!3# "4%C"I&E#(K&22!; 199amm"$ " !1! 7 7 :: ) " $ " X!!1 519i"$ "    ' ' 2sK. GA+J! K.BJ!-K..A"J!K.9J! K.AJK.J'-IK. J&.JK.JK.! K+K K. KK.)rrlistiter_handlesendframes)rPrrCrrs```` r2oncommandresponsereadyobjects+serverreactor.oncommandresponsereadyobjectssG 6" dD ! !:Dm 3m 3^%%jl33rHcr^TR(aTR(dS04$U4SjnSSU"504$)zySignals that end of input has been received. No more frames will be received. All pending activity should be completed. noopc3L># TRH nUShvN M gN 7fr)r)genrPs r2makegen)serverreactor.oninputeof..makegenos ../s $" $ sendframesframegen)rr)rPrs` r2 oninputeofserverreactor.oninputeofcs@  (?(?B;      rHcnUR(aURRU5 S04$SSU04$)Nrrr)rrr()rPframegens r2rserverreactor._handlesendframeswsB     # # * *8 4B;  X# rHc^^^^^[T5 UUUU4SjnTRU"55$)Nc3p># [TTTSS9ShvN TRRT5 gN 7f)Nrr)rrr)rrCrPrsr2r/serverreactor.onservererror..sendframess<' 3     ' ' 2  s 64!6rr)rPrrCrrs```` r2 onservererrorserverreactor.onservererrors, 6" 3 3%%jl33rHNcb^^^^^[T5 UUUUU4SjnTRU"55$)zACalled when a command encountered an error before sending output.c3t># [TTTT5ShvN TRRT5 gN 7fr)rrr)rrrCrPrsr2r0serverreactor.oncommanderror..sendframess<1 7D    ' ' 2  s 86!8r)rPrrCrrrs````` r2oncommanderrorserverreactor.oncommanderrors, 6" 3 3%%jl33rHcURnU=RS- sl[U5nX RU'[H4nX0RS;dMUR UR U5 U$ U$)zCreate a stream to be used for sending data to the client. If this is called before protocol settings frames are received, we don't know what stream encodings are supported by the client and we will default to identity. rr)rrrrcrrr)rPrDrgrRs r2makeoutputstreamserverreactor.makeoutputstreamsw-- ""a'"  "*+h'*D++,?@@ TXXt, * rHcSSU04$)Nrrr:)rPrs r2rserverreactor._makeerrorresults    rHc~URUnUS(dSUl[R"S5eURU UR(aSUlOSUlUSR S5 [ R "USR55SnSU;a!SUlUR[S 55$S U;a0US 'XR;deURRU5 S UUSUS URS 5US (aUS R5S.4$SS.4$)N requestdoners,should not be called without requestdone setrrpayloadrrs$command request missing "name" fieldrs runcommandrdata)s requestidscommandrrr) rrr rseekr decodeallgetvaluerrraddr))rPrCentryrequests r2_makeruncommandresult#serverreactor._makeruncommandresultsZ'' 2^$$DK((?   # #I .  " ".DK!DK jq!$$U:%6%?%?%AB1E ' !$DK((9:  ' !!GG  4 4444   + '#G, )$[[56;Gnw002    KO    rHc"SSUR04$)Ns wantframesstate)rrs r2_makewantframeresult"serverreactor._makewantframeresults dkk   rHcUR[-nUR[-nU(a(U(a!SUlUR [ S55$U(d)U(d!SUlUR [ S55$gg)NrsGreceived command request frame with both new and continuation flags setsJreceived command request frame with neither new nor continuation flags set)r0rrrrr)rPrKnew continuations r2_validatecommandrequestframe*serverreactor._validatecommandrequestframeskk44{{%FF <$DK((. <$DK((. (srHcBUR[:Xa2SUl[R"5UlUR U5$UR[:XaSUlURU5$SUlUR[S5UR-5$)NrrrsBexpected sender protocol settings or command request frame; got %d) rF#FRAME_TYPE_SENDER_PROTOCOL_SETTINGSrrbufferingdecoderrrrrrrrPrKs r2rserverreactor._onframeinitials <<> >8DK,4,E,E,GD )007 7 \\7 7!DK$$U+ +%DK((%,,  rHcTURS:XdeURceUR[:wa.SUlUR [ S5UR-5$UR [-nUR [-nU(a(U(a!SUlUR [ S55$U(d(U(d!SUlUR [ S55$URRUR5 U(aUR5$U(deURR5nSUlU(d!SUlUR [ S55$[!U5S:a!SUlUR [ S 55$US nS U;aUS UR"S 'S UlUR5$![aBnSUlUR [ S5[R"U5-5sSnA$SnAff=f) Nrrs/expected sender protocol settings frame; got %dsXsender protocol settings frame cannot have both continuation and end of stream flags setsOsender protocol settings frame must have continuation or end of stream flag sets;error decoding CBOR from sender protocol settings frame: %ss8sender protocol settings frame did not contain CBOR datars=sender protocol settings frame contained multiple CBOR valuesrrr)rrrFrrrr0*FLAG_SENDER_PROTOCOL_SETTINGS_CONTINUATION!FLAG_SENDER_PROTOCOL_SETTINGS_EOSrrOrrrr  getavailablerNr)rPrKmoreeosrdecodedds r2r&serverreactor._onframeprotocolsettingss{{<<<<,,888 <<> >$DK((DE,,  {{GGkk== C$DK((@ C$DK((.    ) ) 0 0 ? ,,. . s//<<>(,%$DK((MN \A $DK((  AJ ! #89:M8ND !4 5 ((**M $DK((R))!,-  s)%G H'%7H"H'"H'cUR[:wa.SUlUR[ S5UR-5$UR U5nU(aU$UR UR;a.SUlUR[ S5UR -5$UR UR;a.SUlUR[ S5UR -5$UR[-nUR[-nUR[-nU(d!SUlUR[ S55$[R"5nURUR 5 USU(+[#U5S.URUR 'U(d"U(dUR%UR 5$U(d U(deSUlUR'5$)Nrs&expected command request frame; got %ds#request with ID %d already receiveds$request with ID %d is already actives3received command request frame without new flag set)rrr expectingdatar)rFrrrrrrCrrr0rrrr bytesiowriterOrjrr )rPrKr(r moreframes expectingdatarOs r2rserverreactor._onframeidleas <<5 5$DK((; >]*** ((**rHcHUR[:XaCUR[-(aUR U5$UR U5nU(aU$UR UR;a.SUlUR[S5UR -5$UR UR;a.SUlUR[S5UR -5$URUR nUR[:XaUR[-n[UR[-5nUS(a!SUlUR[S55$XSS:wa!SUlUR[S55$USRUR 5 U(dS US'U(d"U(dUR#UR 5$UR%5$UR[&:XagUS(d.SUlUR[S 5UR -5$US c[(R*"5US 'UR-X5$SUlUR[S 5UR-5$) Nrs3received frame for request that is still active: %ds4received frame for request that is not receiving: %drsGreceived command request frame when request frames were supposedly doner!s4mismatch between expect data flag and previous framerTsFreceived command data frame for request that is not expecting data: %drs"received unexpected frame type: %d)rFrr0rrrrCrrrrrrrjrr#rOrr rr r"_handlecommanddataframe)rPrKr(rr$r%s r2r&serverreactor._onframecommandreceivingsc <<5 5{{55((//33E:C  ??d22 2$DK((HI//"  ??$"9"9 9$DK((IJ//"  ''8 <<5 5'GGJ /O!OPM^$( ,,0&6 77( ,,MN *  # #EMM 2(,n%m11%//BB,,. . \\4 4)*( ,,.oo &W~%!%g//= =$DK((785<<G rHcUR[:XdeUSRUR5 UR[ -(aUR 5$UR[-(a/USRS5 URUR5$SUl UR[S55$)Nrrrs command data frame without flags)rFrr#rOr0rr rrrrCrrr)rPrKrs r2r(%serverreactor._handlecommanddataframes||6666 gU]]+ ;;7 7,,. . [[0 0 'N   "--eoo> >$DK((+N)OP PrHc6UR[S55$)Nsserver already errored)rrrs r2rserverreactor._onframeerroreds$$Q'@%ABBrH) rrrrrrrrrrrrs)rrjrrrKrK)rCrArHandleSendFramesReturnTrz.tuple[bytes, dict[bytes, Iterator[bytearray]]])rIterator[bytearray]rr/)rrrCrArrUrr/r)rrrCrArrUrr/)rr)rrUr tuple[bytes, dict[bytes, bytes]])rCrA)rr2)r;r<r=r>r?rurrrrrrrrrrr rrrrrr(rrGr:rHr2rrBs>@21!@/bK4!$K4 K4Z ( +    4 4), 438 4  4DH 4 4), 47< 4  4, ) V ,*J+X3+jL\ QCrHrc:\rSrSr%SrS\S'SS SjjrSrg) commandrequestiz&Represents a request to run a command.rUstateNcNXlX lX0lX@lXPlSUlg)Nspending)rCrRrrrr5)rPrCrRrrrs r2rcommandrequest.__init__s$#     rH)rrrRrrCr5NN)rCrArr)r;r<r=r>r?rurrGr:rHr2r4r4s+0 LAE    rHr4c\rSrSr%SrS\S'S\S'SSSjjrSSjrSS jrSS jr SS jr SS jr SS jr SSjr Srg) clientreactoria!Holds state of a client issuing frame-based protocol requests. This is like ``serverreactor`` but for client-side state. Each instance is bound to the lifetime of a connection. For persistent connection transports using e.g. TCP sockets and speaking the raw framing protocol, there will be a single instance for the lifetime of the TCP socket. For transports where there are multiple discrete interactions (say tunneled within in HTTP request), there will be a separate instance for each distinct interaction. Consumers are expected to tell instances when events occur by calling various methods. These methods return a 2-tuple describing any follow-up action(s) to take. The first element is the name of an action to perform. The second is a data structure (usually a dict) specific to that action that contains more information. e.g. if the reactor wants to send frames to the server, the data structure will contain a reference to those frames. Valid actions that consumers can be instructed to take are: noop Indicates no additional action is required. sendframes Indicates that frames should be sent to the server. The ``framegen`` key contains a generator of frames that should be sent. The reactor assumes that all frames in this generator are sent to the server. error Indicates that an error occurred. The ``message`` key contains an error message describing the failure. responsedata Indicates a response to a previously-issued command was received. The ``request`` key contains the ``commandrequest`` instance that represents the request this data is for. The ``data`` key contains the decoded data from the server. ``expectmore`` and ``eos`` evaluate to True when more response data is expected to follow or we're at the end of the response stream, respectively. rj_hasmultiplesend _buffersendsNcXlX lX0lX@lSUlSUlSUlSUl[S5Ul [R"5Ul 0Ul 0Ul0Ul[!5 g)aBCreate a new instance. ``hasmultiplesend`` indicates whether multiple sends are supported by the transport. When True, it is possible to send commands immediately instead of buffering until the caller signals an intent to finish a send operation. ``buffercommands`` indicates whether sends should be buffered until the last request has been issued. ``clientcontentencoders`` is an iterable of content encoders the client will advertise to the server and that the server can use for encoding data. If not defined, the client will not advertise content encoders to the server. TFrN)rr;r<_clientcontentencoders_canissuecommands_cansend_protocolsettingssent_nextrequestidr_outgoingstream collectionsdeque_pendingrequests_activerequestsr_streamsettingsdecodersrh)rPrhasmultiplesend buffersendsclientcontentencoderss r2rclientreactor.__init__9su, /'&;#!% %*"+A + 1 1 3! "')$ rHcUR(d[R"S5eURnU=RS- sl[ XQX#US9nUR (a UR RU5 US04$UR(d[R"S5eUR(dSUlSUlUSSURU504$) zRequest that a command be executed. Receives the command name, a dict of arguments to pass to the command, and an optional file object containing the raw data for the command. Returns a 3-tuple of (request, action, action data). scannot issue new commandsrrrr*sends cannot be performed on this instanceFrr) r?r rrBr4r<rFr(r@r;_makecommandframes)rPrRrrrrCrs r2 callcommandclientreactor.callcommandbs%%(()EF F''  q  T8      ! ! ( ( 1GR' '==,,A(( % ).&!8!8!A rHc^TR(dS04$TR(d[R"S5eTR(dSTlSTlU4SjnSSU"504$)a/Request that all queued commands be sent. If any commands are buffered, this will instruct the caller to send them over the wire. If no commands are buffered it instructs the client to no-op. If instances aren't configured for multiple sends, no new command requests are allowed after this is called. rrOFc3># TR(aGTRR5nTRU5ShvN TR(aMFggN7fr)rFpopleftrP)rrPs r2 makeframes/clientreactor.flushcommands..makeframessH''//779227;;;''';sAAAAArr)rFr@r rr;r?)rPrVs` r2 flushcommandsclientreactor.flushcommandsso$$B; }}((=  $$%*D "!DM <     rHc ## XRUR'SUlUR(dyUR(ahSUlSR [ R"SUR055nURRUR[[US9v [URURURURURUR S9nUShvN SUlgN 7f) zsEmit frames to issue a command request. As a side-effect, update request accounting to reflect its changed state. ssendingTrHrrrNNssent)rGrCr5rAr>r*rrdrCr_rrrrRrrr)rPrrOr(s r2rP clientreactor._makecommandframess3:W../" ))d.I.I)-D &hh%%+T-H-HG&&00!++:7 1 "     LL LL>>%%    sC9D ;D< D cURS-(aSS[S5UR-04$URUR;aSUR[-(dSS[S504$[ UR5URUR'URURnUR[ -(a URUR5UlUR[-(aURUR UR[:XaURU5$URUR;aSS[S5UR-04$URURnSUl[ UR"[$UR&0nUR)UR5nU(d#[*R,"SUR-5eU"X15$) zProcess a frame that has been received off the wire. Returns a 2-tuple of (action, meta) describing further action the caller needs to take as a result of receiving this frame. rrrs.received frame with odd numbered stream ID: %dsEreceived frame on unknown stream without beginning of stream flag sets*received frame for inactive request ID: %ds receivingsunhandled frame type: %d)rDrrrErqrwrrrOrrFr_onstreamsettingsframerCrGr5r_oncommandresponseframer_onerrorresponseframer)r r)rPrKrrrrs r2rclientreactor.onframerecvs >>A KL..)  >>!6!6 6$$'??"AD%5@4OD ! !%.. 1&&u~~6   ; ;"MM%--8EM   5 5%%enn5 <<5 5..u5 5 ??$"6"6 6GH//* &&u7$  ()E)E %t'A'A  ||ELL)((+ell: G##rHcUR[:XdeUR[-nUR[-nU(aU(aSS[ S504$U(dU(dSS[ S504$UR UR;a-[R"5nX@RUR 'URUR nURUR5 U(aS04$U(deUR5nURUR U(dSS[ S504$UR UR R#UR$USUS S5 S04$![a0nSS[ S5[R"U5-04sSnA$SnAff=f![a0nSS[ S 5[R"U5-04sSnA$SnAff=f) NrrsXstream encoding settings frame cannot have both continuation and end of stream flags setsOstream encoding settings frame must have continuation or end of stream flag sets;error decoding CBOR from stream encoding settings frame: %srs8stream encoding settings frame did not contain CBOR datarrs error setting stream decoder: %s)rFrr0*FLAG_STREAM_ENCODING_SETTINGS_CONTINUATIONrrrDrHrrrrOrrrrrr~r)rPrKrrdecoderrrs r2r]$clientreactor._onstreamsettingsframes ||9999{{GGkk== CH  CB!  >>!=!= =//1G;B ( ( 8..u~~>  NN5== ) B;  s&&(  ( ( 8%!    ! !%.. 1 < <'!*gabk {a 2%11!4 5   J =>$11!45  s<F:G F= %F82F=8F= G: %G5/G:5G:cUR[-(aSUlURUR SUUR[ -UR[-UR S.4$)Nsreceiveds responsedata)requests expectmorerr)r0rr5rGrCrrO)rPrrKs r2r^%clientreactor._oncommandresponseframetsb ;;2 2'GM$$W%6%67 #${{-OO &??    rHcSUlURUR [R"UR 5SnSUUSUSS.4$)Nrrrrr)rfrr)r5rGrCrrrO)rPrrKrs r2r_#clientreactor._onerrorresponseframes]"  !2!2 3   u}} -a 0 #7jM   rH) rGr<r?r@r>r;rrBrCrFrArHr)FTN)rIrjrJrjrrr8r0)rr4rr1r.)rr4rKrK)r;r<r=r>r?rurrQrXrPrr]r^r_rGr:rHr2r:r:s,\ !& " '!'! '!  '!R%N B( %( ( TF$PVp   rHr:)r+rr,rArrU)rCrArDrArErArFrAr0rArOrUrrY)rgrUrrY)rurUrr6)rz frame | None)rrrCrArrArr1)rrrCrArr)rrrCrArrArr1)rrrCrArrYr)rrrCrArrUrr1) rrrCrArrUrrUrr1r )rrrr)g __future__rrrDr[typingrri18nr thirdpartyr TYPE_CHECKINGr>r r r r r utilsrrrtuplerUrrYr/rZrrqrrrrurrrrrFRAME_TYPE_PROGRESSrrrrrrrrrrrrrr rrr!rbrr"r#StructARGUMENT_RECORD_HEADERr3rgr6rKr_rorxrrrrrrrrrrrrrrr,r9rLrUr_EncoderTDecoderTrbrcrhrrwrrrrr4r:r:rHr2rvs2#      $E4x 7J0J+K$KL#.),"  "" &*#!3,40*$!D2 !   $(!#' #'  %6 -2 +'"&4 !($ &*" 8 %,( .2*$(!@ -4 0 .2*$(!@ -4 0 5/!7rB')G > 1-  u- dd    @      8?D2N . J JJ  JJZ %(4/  @& &%(&&T:> "-22 ").9>6/ 8 88 8  8vf f Z "h " "h" g  g '* 3' 3g._ 3g34_4 Ok1NBCHOk1NBCH;=7<%'{'..  6"%&"%J^ 6^ B +73 r Cr Cj   J J rH