a ꈾ5[@sxddlZddlZddlZddlZddlZddlZd ddZdddZGdddeZ Gd d d eZ Gd d d eZ dS)NTcCsJt|tj}|r |tj@}n |tjB}t|tj|t|tj@ S)zl Set the given file-descriptor blocking or non-blocking. Returns the original blocking status. )fcntlZF_GETFLos O_NONBLOCKZF_SETFLbool)fdblockingZold_flagZnew_flagr0/usr/lib/python3.9/site-packages/dockerpty/io.py set_blockings  r c Csg}zt||||ddWStjyz}zBtjr<|jn|d}|tjkrbggfWYd}~S|WYd}~n d}~00dS)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_selectselecterrorsixZPY3errnoEINTR)Z read_streamsZ write_streamstimeoutZexception_streamsenorrr r ,s r c@sjeZdZdZejejejgZddZ ddZ ddZ dd d Z d d Z d dZddZddZddZdS)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. cCs||_d|_d|_d|_dS)z} Initialize the Stream for the file descriptor `fd`. The `fd` object must have a `fileno()` method. FN)rbufferclose_requestedclosed)selfrrrr __init__WszStream.__init__cCs |jS)z= Return the fileno() of the file descriptor. )rfilenorrrr rbsz Stream.filenocCs,t|jdr|j|dSt|j|SdS)N setblockingT)hasattrrrr rvaluerrr r is  zStream.set_blockingc Csjz.t|jdr|j|WSt|j|WStyb}z|jtj vrN|WYd}~qd}~00qdS)zU Return `n` bytes of data from the Stream, or None at end of stream. recvN) rrr#rreadrEnvironmentErrorrrERRNO_RECOVERABLE)rnrrrr r$ps  z Stream.readcCs&|sdS|j|7_|t|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)rdo_writelenrdatarrr writes z Stream.writec Cszfd}t|jdr"|j|j}nt|j|j}|j|d|_|jrbt|jdkrb| |WSt y}z|j t j vr|WYd}~qd}~00qdS)zZ Flushes as much pending data from the internal write buffer as possible. rsendN)rrr-rrr,rrr)closer%rrr&)rZwrittenrrrr r(s  zStream.do_writecCst|jdkS)zL Returns True if the stream has data waiting to be written. r)r)rrrrr needs_writeszStream.needs_writecCsLd|_|jsHt|jdkrHd|_t|jdr8|jnt|jdS)NTrr.) rrr)rrrr.rrrrrr r.s   z Stream.closecCsdjt|j|jdS)Nz {cls}({fd}))clsr)formattype__name__rrrrr __repr__szStream.__repr__N)r")r3 __module__ __qualname____doc__rrZEDEADLKZ EWOULDBLOCKr&rrr r$r,r(r/r.r4rrrr rFs    rc@sdeZdZdZddZddZddZdd d Zd d Zd dZ ddZ ddZ dddZ ddZ dS)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. cCs||_d|_dS)zA Initialize a new Demuxer reading from `stream`. rN)streamremain)rr9rrr rszDemuxer.__init__cCs |jS)zn Returns the fileno() of the underlying Stream. This is useful for select() to work. )r9rrrrr rszDemuxer.filenocCs |j|SN)r9r r rrr r szDemuxer.set_blockingr"cCsX||}|dkrdSt}t||krP|j|t|}|sF|S||}q|SdS)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`. rN)_next_packet_sizer binary_typer)r9r$)rr'sizer+nxtrrr r$s   z Demuxer.readcCs |j|S)z6 Delegates the the underlying Stream. )r9r,r*rrr r,sz Demuxer.writecCst|jdr|jSdS)1 Delegates to underlying Stream. r/F)rr9r/rrrr r/s  zDemuxer.needs_writecCst|jdr|jSdS)r@r(F)rr9r(rrrr r( s  zDemuxer.do_writecCs |jS)r@)r9r.rrrr r.sz Demuxer.closercCsd}|jdkr*t||j}|j|8_nvt}t|dkrd|jdt|}|sZdS||}q2|durpdSt|dkrtd|\}}t||}|||_|S)Nrz>BxxxL) r:minrr=r)r9r$structunpack)rr'r>r+r?__actualrrr r<s"       zDemuxer._next_packet_sizecCsdjt|j|jdS)Nz{cls}({stream}))r0r9)r1r2r3r9rrrr r43s zDemuxer.__repr__N)r")r)r3r5r6r7rrr r$r,r/r(r.r<r4rrrr r8s     r8c@sDeZdZdZdddZddZddZdd d Zd d ZddZ dS)Pumpa Stream pump class. A Pump wraps two Streams, reading from one and and writing its data into the other, much like a pipe but manually managed. This abstraction is used to facilitate piping data between the file descriptors associated with the tty and those associated with a container's allocated pty. Pumps are selectable based on the 'read' end of the pipe. TcCs"||_||_d|_||_||_dS)z Initialize a Pump with a Stream to read from and another to write to. `wait_for_output` is a flag that says that we need to wait for EOF on the from_stream in order to consider this pump as "done". FN) from_stream to_streameofwait_for_outputpropagate_close)rrHrIrKrLrrr rFs z Pump.__init__cCs |jS)z Returns the `fileno()` of the reader end of the Pump. This is useful to allow Pumps to function with `select()`. )rHrrrrr rXsz Pump.filenocCs |j|Sr;)rHr r rrr r aszPump.set_blockingr"c CszJ|j|}|dus"t|dkr>d|_|jr8|jWdS|j|WSty~}z|j t j krj|WYd}~n d}~00dS)z Flush `n` bytes of data from the reader Stream to the writer Stream. Returns the number of bytes that were actually flushed. A return value of zero is not an error. If EOF has been reached, `None` is returned. NrT) rHr$r)rJrLrIr.r,OSErrorrZEPIPE)rr'r$rrrr flushds    z Pump.flushcCs&|j s|jo$t|jdo"|j 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/)rKrJrrIr/rrrr is_done|sz Pump.is_donecCsdjt|j|j|jdS)Nz){cls}(from={from_stream}, to={to_stream}))r0rHrI)r1r2r3rHrIrrrr r4s z Pump.__repr__N)TT)r") r3r5r6r7rrr rNrOr4rrrr rG8s    rG)T)r) rrrrCr r rr objectrr8rGrrrr s  v|