2e*~ddlZddlZddlZddlZddlZd dZd dZGddeZGddeZ Gdd eZ y) Nc"tj|tj}|r|tjz}n|tjz}tj|tj|t |tjz S)zl Set the given file-descriptor blocking or non-blocking. Returns the original blocking status. )fcntlF_GETFLos O_NONBLOCKF_SETFLbool)fdblockingold_flagnew_flags ./usr/lib/python3/dist-packages/dockerpty/io.py set_blockingrsd{{2u}}-H o-bmm+ KKEMM8,Hr}},- --cg} tj||||ddS#tj$r/}|j}|tjk(r ggfcYd}~S|d}~wwxYw)z Select the streams from `read_streams` that are ready for reading, and streams from `write_streams` ready for writing. Uses `select.select()` internally but only returns two lists of ready streams. rN)builtin_selectselecterrorerrnoEINTR) read_streams write_streamstimeoutexception_streamsenos rrr+su $$      A      WW  8OG s A!"AA!AA!ceZdZdZ ej ej ejgZdZ dZ dZ d dZ dZ dZdZd Zd Zy ) Streamz Generic Stream class. This is a file-like abstraction on top of os.read() and os.write(), which add consistency to the reading of sockets and files alike. c<||_d|_d|_d|_y)z} Initialize the Stream for the file descriptor `fd`. The `fd` object must have a `fileno()` method. rFN)r bufferclose_requestedclosed)selfr s r__init__zStream.__init__Vs!  $ rc6|jjS)z= Return the fileno() of the file descriptor. )r filenor$s rr'z Stream.filenoas ww~~rct|jdr|jj|yt|j|S)N setblockingT)hasattrr r*rr$values rrzStream.set_blockinghs6 477M * GG   &/ /rc. t|jdr|jj|Stj|jj |S#t $r(}|jtjvr|Yd}~nd}~wwxYw)zU Return `n` bytes of data from the Stream, or None at end of stream. recvN) r+r r/rreadr'EnvironmentErrorrrERRNO_RECOVERABLE)r$nrs rr0z Stream.readosy  477F+77<<?*wwtww~~/33# 77&":"::G;  s0A"-A"" B+BBch|sy|xj|z c_|jt|S)z Write `data` to the Stream. Not all data may be written right away. Use select to find when the stream is writeable, and call do_write() to flush the internal buffer. N)r!do_writelenr$datas rwritez Stream.write~s+ t  4yrc d}t|jdr&|jj|j}n8t j |jj |j}|j|d|_|jr(t|jdk(r|j|S#t$r(}|jtjvr|Yd}~nd}~wwxYw)zZ Flushes as much pending data from the internal write buffer as possible. rsendN)r+r r;r!rr9r'r"r6closer1rrr2)r$writtenrs rr5zStream.do_writes 477F+"ggll4;;7G hhtww~~'7EG"kk'(3 ''C ,<,AJJL# 77&":"::G; !sB?C C4 C//C4c2t|jdkDS)zL Returns True if the stream has data waiting to be written. r)r6r!r(s r needs_writezStream.needs_writes4;;!##rc(d|_|jst|jdk(rfd|_t |j dr|j j ytj |j jyyy)NTrr<) r"r#r6r!r+r r<rr'r(s rr<z Stream.closesc#{{s4;;/14DKtww( )* 5{rcbdjt|j|jS)Nz {cls}({fd}))clsr )formattype__name__r r(s r__repr__zStream.__repr__s&##T (;(;#HHrNi)rE __module__ __qualname____doc__rrEDEADLK EWOULDBLOCKr2r%r'rr0r9r5r?r<rFrrrrEs[      0  0$ +IrrcPeZdZdZdZdZdZd dZdZdZ dZ d Z dd Z d Z y )Demuxera8 Wraps a multiplexed Stream to read in data demultiplexed. Docker multiplexes streams together when there is no PTY attached, by sending an 8-byte header, followed by a chunk of data. The first 4 bytes of the header denote the stream from which the data came (i.e. 0x01 = stdout, 0x02 = stderr). Only the first byte of these initial 4 bytes is used. The next 4 bytes indicate the length of the following chunk of data as an integer in big endian format. This much data must be consumed before the next 8-byte header is read. c ||_d|_y)zA Initialize a new Demuxer reading from `stream`. rN)streamremain)r$rQs rr%zDemuxer.__init__s   rc6|jjS)zn Returns the fileno() of the underlying Stream. This is useful for select() to work. )rQr'r(s rr'zDemuxer.filenos{{!!##rc8|jj|SN)rQrr,s rrzDemuxer.set_blockings{{''..rc|j|}|dkryd}t||kr?|jj|t|z }|s|S||z}t||kr?|S)a Read up to `n` bytes of data from the Stream, after demuxing. Less than `n` bytes of data may be returned depending on the available payload, but the number of bytes returned will never exceed `n`. Because demuxing involves scanning 8-byte headers, the actual amount of data read from the underlying stream may be greater than `n`. rNr)_next_packet_sizer6rQr0)r$r3sizer8nxts rr0z Demuxer.readsr%%a( 19 Dd)d"kk&&tc$i'78Kcz d)d" Krc8|jj|S)z6 Delegates the the underlying Stream. )rQr9r7s rr9z Demuxer.writes {{  &&rcdt|jdr|jjSy)1 Delegates to underlying Stream. r?F)r+rQr?r(s rr?zDemuxer.needs_writes( 4;; .;;**, ,rcdt|jdr|jjSy)r\r5F)r+rQr5r(s rr5zDemuxer.do_write s( 4;; +;;'') )rc6|jjS)r\)rQr<r(s rr<z Demuxer.closes {{  ""rcd}|jdkDr-t||j}|xj|zc_|Sd}t|dkr>|jj dt|z }|sy||z}t|dkr>|yt|dk(r/t j d|\}}t||}||z |_|S)Nrrz>BxxxL)rRminr6rQr0structunpack)r$r3rXr8rY__actuals rrWzDemuxer._next_packet_sizes ;;?q$++&D KK4 K" Dd)a-kk&&q3t9}5cz d)a-|4yA~#]]8T: F1f~$tm  rcbdjt|j|jS)Nz{cls}({stream}))rBrQ)rCrDrErQr(s rrFzDemuxer.__repr__2s- ''DJ,?,?/3{{(< >''- - ww%++%& sAA6A66 B(?B##B(c|j xs |jxr3t|jdxr|jj S)z Returns True if the read stream is done (either it's returned EOF or the pump doesn't have wait_for_output set), and the write side does not have pending bytes to send. r?)rnrmr+rlr?r(s ris_donez Pump.is_done{sH(((4DHH^T^^];\@Z@Z@\] ^rcxdjt|j|j|jS)Nz){cls}(from={from_stream}, to={to_stream}))rBrkrl)rCrDrErkrlr(s rrFz Pump.__repr__s8:AAT ##((nnB& &rN)TTrG) rErHrIrJr%r'rrurwrFrMrrriri7s.  "&!% /$)40^&rri)Trg) rrrrbrrrobjectrrOrirMrrrzsN" .&4sIVsIly