U ~vU"+@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)fdZblockingZold_flagZnew_flagr./usr/lib/python3/dist-packages/dockerpty/io.py set_blockings  r c Cszg}zt||||ddWStjk rt}z6tjr>|jn|d}|tjkr`ggfWY S|W5d}~XYnXdS)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_streamsZtimeoutZexception_streamseZnorrrr ,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. )rfilenorrrrrbsz Stream.filenocCs,t|jdr|j|dSt|j|SdS)N setblockingT)hasattrrrr rvaluerrrr is  zStream.set_blockingc Cshz.t|jdr|j|WSt|j|WStk r`}z|jtj krP|W5d}~XYqXqdS)zU Return `n` bytes of data from the Stream, or None at end of stream. recvN) rrr rreadrEnvironmentErrorrrERRNO_RECOVERABLE)rnrrrrr!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_writelenrdatarrrwrites z Stream.writec Cszfd}t|jdr"|j|j}nt|j|j}|j|d|_|jrbt|jdkrb| |WSt k r}z|j t j kr|W5d}~XYqXqdS)zZ Flushes as much pending data from the internal write buffer as possible. rsendN)rrr*rrr)rrr&closer"rrr#)rZwrittenrrrrr%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+rrrrrrr+s   z Stream.closecCsdjt|j|jdS)Nz {cls}({fd}))clsr)formattype__name__rrrrr__repr__szStream.__repr__N)r)r0 __module__ __qualname____doc__rrZEDEADLKZ EWOULDBLOCKr#rrr r!r)r%r,r+r1rrrrrFs    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)rr6rrrrszDemuxer.__init__cCs |jS)zn Returns the fileno() of the underlying Stream. This is useful for select() to work. )r6rrrrrrszDemuxer.filenocCs |j|SN)r6r rrrrr szDemuxer.set_blockingrcCsX||}|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&r6r!)rr$sizer(nxtrrrr!s   z Demuxer.readcCs |j|S)z6 Delegates the the underlying Stream. )r6r)r'rrrr)sz Demuxer.writecCst|jdr|jSdS)1 Delegates to underlying Stream. r,F)rr6r,rrrrr,s  zDemuxer.needs_writecCst|jdr|jSdS)r=r%F)rr6r%rrrrr% s  zDemuxer.do_writecCs |jS)r=)r6r+rrrrr+sz Demuxer.closercCsd}|jdkr*t||j}|j|8_nvt}t|dkrd|jdt|}|sZdS||}q2|dkrpdSt|dkrtd|\}}t||}|||_|S)Nrz>BxxxL) r7minrr:r&r6r!structZunpack)rr$r;r(r<__Zactualrrrr9s"       zDemuxer._next_packet_sizecCsdjt|j|jdS)Nz{cls}({stream}))r-r6)r.r/r0r6rrrrr13s zDemuxer.__repr__N)r)r)r0r2r3r4rrr r!r)r,r%r+r9r1rrrrr5s     r5c@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)rrCrDrFrGrrrrFs 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()`. )rCrrrrrrXsz Pump.filenocCs |j|Sr8)rCr rrrrr aszPump.set_blockingrc CszJ|j|}|dks"t|dkr>d|_|jr8|jWdS|j|WStk r|}z|j t j krl|W5d}~XYnXdS)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) rCr!r&rErGrDr+r)OSErrorrZEPIPE)rr$r!rrrrflushds    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,)rFrErrDr,rrrris_done|sz Pump.is_donecCsdjt|j|j|jdS)Nz){cls}(from={from_stream}, to={to_stream}))r-rCrD)r.r/r0rCrDrrrrr1s z Pump.__repr__N)TT)r) r0r2r3r4rrr rIrJr1rrrrrB8s    rB)T)r) rrrr@r r rr objectrr5rBrrrrs  v|