o j I@sbUddlZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z mZmZejdkr9ddlmZnddlmZeeedd fZzzddlZWn ey[ddlZYnwd ZWn eyjd ZYnwzejd krxdd lmZndd lmZd ZWn eyd ZYnwdZdZdZ dZ!dZ"dZ#Gddde$Z%Gddde Z&Gddde Z'Gddde Z(Gddde d dZ)GdddZ*e*eZ+e*e,d <d!e(d"dfd#d$Z-  d:d%e.dBd&e/d"e0fd'd(Z1Gd)d*d*e Z2ed+e2d,Z3Gd-d.d.eZ4Gd/d0d0e4e e3Z5Gd1d2d2Z6Gd3d4d4e5e'Z7Gd5d6d6e4Z8Gd7d8d8e5d9Z9dS);N)ABCabstractmethod)Executor)AnyFinalGenericProtocol TypedDictTypeVarcast) )Buffer)Unionzmemoryview[int]zmemoryview[bytes]TF)r ZstdDecompressori@iic@seZdZdZdS)TooManyMembersErrorz:A stream concatenated more members than the caller allows.N)__name__ __module__ __qualname____doc__rrL/opt/steambot/venv/lib/python3.10/site-packages/aiohttp/compression_utils.pyr6src@s2eZdZdedefddZd dedefddZd S) ZLibCompressObjProtocoldatareturncCdSNrselfrrrrcompress;z ZLibCompressObjProtocol.compress.modecCrr rr"r%rrrflush<r$zZLibCompressObjProtocol.flushN.)rrrrbytesr#intr'rrrrr:src@sneZdZddededefddZddedefdd Zede fd d Z edefd d Z edefddZ dS)ZLibDecompressObjProtocol.r max_lengthrcCrr rr"rr,rrr decompress@r$z$ZLibDecompressObjProtocol.decompresslengthcCrr rr"r/rrrr'Ar$zZLibDecompressObjProtocol.flushcCrr rr"rrreofCzZLibDecompressObjProtocol.eofcCrr rr1rrrunconsumed_tailFr3z)ZLibDecompressObjProtocol.unconsumed_tailcCrr rr1rrr unused_dataIr3z%ZLibDecompressObjProtocol.unused_dataNr() rrrrr*r)r.r'propertyboolr2r4r5rrrrr+?sr+c@seZdZUeed<eed<eed<eed<eed<      ddeded ed ed ed ed BdefddZ dd ed edefddZ dded edede fddZ dd ededede fddZ d S)ZLibBackendProtocol MAX_WBITS Z_FULL_FLUSH Z_SYNC_FLUSH Z_BEST_SPEEDZ_FINISH.levelmethodwbitsmemLevelstrategyzdictNrcCrr r)r"r>r?r@rArBrCrrr compressobjTszZLibBackendProtocol.compressobjcCrr r)r"r@rCrrr decompressobj]z!ZLibBackendProtocol.decompressobjrcCrr r)r"rr>r@rrrr#arFzZLibBackendProtocol.compressbufsizecCrr r)r"rr@rGrrrr.drFzZLibBackendProtocol.decompress)......)..) rrrr*__annotations__rrrDr+rEr)r#r.rrrrr8Msn     r8c@s&eZdZUeed<eed<eed<dS)CompressObjArgsr@rBr>N)rrrr*rHrrrrrIis  rI)totalc@seZdZdefddZedefddZedefddZ edefd d Z edefd d Z edefd dZ edefddZ dededefddZdededefddZdedededefddZdedededefddZdedefddZdS) ZLibBackendWrapper _zlib_backendcCs ||_dSr )rL)r"rLrrr__init__p zZLibBackendWrapper.__init__rcCst|jddS)Nr undefinedgetattrrLr1rrrnamesszZLibBackendWrapper.namecC|jjSr )rLr9r1rrrr9wzZLibBackendWrapper.MAX_WBITScCrSr )rLr:r1rrrr:{rTzZLibBackendWrapper.Z_FULL_FLUSHcCrSr )rLr;r1rrrr;rTzZLibBackendWrapper.Z_SYNC_FLUSHcCrSr )rLr<r1rrrr<rTzZLibBackendWrapper.Z_BEST_SPEEDcCrSr )rLr=r1rrrr=rTzZLibBackendWrapper.Z_FINISHargskwargscO|jj|i|Sr )rLrDr"rUrVrrrrDzZLibBackendWrapper.compressobjcOrWr )rLrErXrrrrErYz ZLibBackendWrapper.decompressobjrcO|jj|g|Ri|Sr )rLr#r"rrUrVrrrr#zZLibBackendWrapper.compresscOrZr )rLr.r[rrrr.r\zZLibBackendWrapper.decompressattrnamecCs t|j|Sr rP)r"r]rrr __getattr__ zZLibBackendWrapper.__getattr__N)rrrr8rMr6strrRr*r9r:r;r<r=rrrDr+rErr)r#r.r^rrrrrKos&rK ZLibBackendnew_zlib_backendrcCs |t_dSr )rarL)rbrrrset_zlib_backendrNrcencodingsuppress_deflate_headercCs$|dkr dtjS|rtj StjS)Ngzip)rar9)rdrerrrencoding_to_modes rhc@sHeZdZd dededefddZedefddZ edefd d Z d S) MemberDecompressObjProtocol.rr,rcCrr rr-rrrr.r$z&MemberDecompressObjProtocol.decompresscCrr rr1rrrr2r3zMemberDecompressObjProtocol.eofcCrr rr1rrrr5r3z'MemberDecompressObjProtocol.unused_dataNr() rrrrr*r)r.r6r7r2r5rrrrris ri_DecompressObjT)boundc@szeZdZdefdedBdedBfddZeefde dede fd d Z efde dede fd d Z e edefd dZdS)DecompressionBaseHandlerNexecutormax_sync_chunk_sizecCs||_||_dS)z&Base class for decompression handlers.N) _executor_max_sync_chunk_sizer"rmrnrrrrMs z!DecompressionBaseHandler.__init__rr,rcCdSDecompress the given data.Nrr-rrrdecompress_syncr$z(DecompressionBaseHandler.decompress_synccsD|jdurt||jkrt|j|j||IdHS|||Srs)rplenasyncioget_event_looprun_in_executorrorur-rrrr.s   z#DecompressionBaseHandler.decompresscCrr)z7Return True if more output is available by passing b"".Nrr1rrrdata_availabler$z'DecompressionBaseHandler.data_available)rrrMAX_SYNC_CHUNK_SIZErr*rMrZLIB_MAX_LENGTH_UNLIMITEDrr)rur.r6r7rzrrrrrls8     rlc@sZeZdZUdZeed<eed<dZedBed<e defddZ d ed edefd d Z dS) ConcatDecompressionHandlerayHandler for a codec whose streams may concatenate independent members. Concatenated gzip/deflate members and multi-frame zstd (https://datatracker.ietf.org/doc/html/rfc8878#section-3.1.1) decode the same way: a decompressor handles one member, then flags eof and leaves the rest of the input in unused_data, so every member after it needs a fresh one. _unlimited _decompressorN_pending_unused_datarcCrr)z*Return a decompressor for the next member.Nrr1rrr_new_decompressorr$z,ConcatDecompressionHandler._new_decompressorfirstr,c Cst|jj}|g}t|}d}t}|}d} |t|kr|jjr6| d7} | tkr/tdtd||_t}||j krM||}|dkrMt ||d|_ n=t ||t|} |j ||| |} | rn|| |t| 7}|jjr{| t|jj}n | }t |dt}|t|ksd|S)z9Decode the members following the one ``first`` came from.rz Compressed stream has more than z membersN) memoryviewrr5rvMEMBER_WINDOW_MINr2MAX_DECOMPRESS_MEMBERSrrr~r)rminr.appendMEMBER_WINDOW_MAXjoin) r"rr, remainingpartsproducedposwindowbudgetmembersendchunkrrr_decompress_memberssD       "z.ConcatDecompressionHandler._decompress_members) rrrrr*rHrjrr)rrrrrrrr}s  r}c@seZdZddddddefdedBdededBdedBdedBdedBd edBfd d Zd e d e fddZ d e d e fddZ ddedBd e fddZ dS)ZLibCompressorNFrdrer>r@rBrmrnc Csv||_||_|durt||n||_ttj|_i}|j|d<|dur'||d<|dur/||d<|jjdi||_dS)Nr@rBr>r) rorprh_moderKrarLrD _compressor) r"rdrer>r@rBrmrnrVrrrrM!s    zZLibCompressor.__init__rrcCs |j|Sr )rr#r!rrr compress_sync<r_zZLibCompressor.compress_synccsF|jduo t||jk}|rt|j|jj|IdHS||S)aCompress the data and returned the compressed bytes. Note that flush() must be called after the last call to compress() If the data size is large than the max_sync_chunk_size, the compression will be done in the executor. Otherwise, the compression will be done in the event loop. **WARNING: This method is NOT cancellation-safe when used with flush().** If this operation is cancelled, the compressor state may be corrupted. The connection MUST be closed after cancellation to avoid data corruption in subsequent compress operations. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. N) rprvrwget_running_loopryrorr#r)r"rshould_use_executorrrrr#?s    zZLibCompressor.compressr%cCs|j|dur |S|jjS)a Flush the compressor synchronously. **WARNING: This method is NOT cancellation-safe when called after compress().** The flush() operation accesses shared compressor state. If compress() was cancelled, calling flush() may result in corrupted data. The connection MUST be closed after compress() cancellation. For cancellation-safe compression (e.g., WebSocket), the caller MUST wrap compress() + flush() + send operations in a shield and lock to ensure atomicity. N)rr'rLr=r&rrrr'[s  zZLibCompressor.flushr )rrrr{r`r7r*rrMrr)rr#r'rrrrr s4 rc seZdZeZdddefdedBdededBde dBffdd Z d e fd d Z efd e d e d efddZdde d efddZed efddZed efddZZS)ZLibDecompressorNFrdrermrncs<tj||dt|||_ttj|_||_d|_ dS)NrmrnF) superrMrhrrKrarLrr _last_empty)r"rdrermrn __class__rrrMns     zZLibDecompressor.__init__rcCs|jj|jdS)N)r@)rLrErr1rrrr{sz"ZLibDecompressor._new_decompressorrr,cCs~|jdur|jt|}d|_|j|jj||}|jjr(|jjr(|||}|dk|_|jjr=|j |j j kr=| |_|SNr) rr)rr.r4r2r5rrrrLr9rr"rr,resultrrrru~s     z ZLibDecompressor.decompress_syncrr/cCs|dkr |j|S|jS)Nr)rr'r0rrrr's  zZLibDecompressor.flushcCst|jjp|j p|jduSr )r7rr4rrr1rrrrzs zZLibDecompressor.data_availablecCrSr )rr2r1rrrr2rTzZLibDecompressor.eof)r)rrrr|r~r{r`r7rr*rMr+rrr)rur'r6rzr2 __classcell__rrrrrks:  rcsreZdZdefdedBdedBddffdd Zefdedede fd d Z de fd d Z e de fd dZZS)BrotliDecompressorNrmrnrcs0tstdt|_d|_tj||ddS)z)Decompress data using the Brotli library.zIThe brotli decompression is not available. Please install `Brotli` moduleFrN) HAS_BROTLI RuntimeErrorbrotli Decompressor_objrrrMrqrrrrMs zBrotliDecompressor.__init__rr,cCs|t|jdr|tkrtt|j|}n#tt|j||}n|tkr-tt|j|}n tt|j||}|dk|_|S)rtr.r)hasattrrr|r r)r.processrrrrrrus  z"BrotliDecompressor.decompress_synccCs t|jdrtt|jSdS)zFlush the decompressor.r'r)rrr r)r'r1rrrr's zBrotliDecompressor.flushcCs|j o |j Sr )r is_finishedrr1rrrrzsz!BrotliDecompressor.data_available)rrrr{rr*rMr|rr)rur'r6r7rzrrrrrrs* rcseZdZeZdefdedBdedBddffdd Zddd Z e fd e d ede fd d Z de fddZedefddZZS)ZSTDDecompressorNrmrnrcs*tstdtj||d||_dS)NzOThe zstd decompression is not available. Please install `backports.zstd` moduler)HAS_ZSTDrrrMrrrqrrrrMs zZSTDDecompressor.__init__rcCstSr rr1rrrrsz"ZSTDDecompressor._new_decompressorrr,cCsj|tkrtn|}|jdur|j|}d|_|j||}|jjr*|jjr*|||}|jjr3||_|Sr ) r|ZSTD_MAX_LENGTH_UNLIMITEDrrr.r2r5rr)r"rr,zstd_max_lengthrrrrrus    z ZSTDDecompressor.decompress_synccCrrrrr1rrrr' r3zZSTDDecompressor.flushcCs|jj o |jj p|jduSr )r needs_inputr2rr1rrrrz szZSTDDecompressor.data_available)rr)rrrrr~r{rr*rMrr|rr)rur'r6r7rzrrrrrrs.  rr)NF):rwsyszlibabcrrconcurrent.futuresrtypingrrrrr r r version_infocollections.abcrrr) bytearray brotlicffir ImportErrorrcompression.zstdrbackports.zstdrr{r|rrrr ValueErrorrrr+r8rIrKrarHrcr`r7r*rhrirjrlr}rrrrrrrrsv  $         -  #CKA0