iCdZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZddlmZmZmZddlmZmZej,dj.Zdeddfd ZdZej6eZd ZGd d ej>Z Gd dejBZ"GddZ#Gdde ejHZ$Gdde Z%GddZ&y)zKBi-directional asyncio.Transport implementations based on file descriptors.N)Thread)AnyClassVarSequence) JsonObjectget_intargsreturnc0t|dk7r tdy)Nrzprctl() failed) sys_prctlOSError)r s 4/usr/lib/python3/dist-packages/cockpit/transports.pyprctlrs $1&''ic eZdZUdZeeed<ejed<ejed<ded<eed<eed<e ed <e ed <e ed <d Z e ed < d.dejdejdededdf fd Z d/dZ de fdZd/dZd/dZd/dZd/dZd0d1dZde fdZd/dZdefd Zd2d!Zd3d4d"Zd/d#Zd/d$Zd/d%Zd&eddfd'Zd&eddfd(Zd/d)Zdej@fd*Z!de fd+Z"dej@ddfd,Z#d/d-Z$xZ%S)5 _Transporti BLOCK_SIZE_loop _protocolzcollections.deque[bytes] | None_queue_in_fd_out_fd_closing _is_reading_eofF _eio_is_eofNloopprotocolin_fdout_fdextrazdict[str, object] | Nonec~t||||_||_tj d||||d|_d|_d|_d|_ ||_ ||_ tj|d||k7rtj|d|jj||jy)Nz/Created transport %s for protocol %s, fds %d %dF)super__init__rrloggerdebugrrrrrros set_blockingconnection_maderesume_reading)selfrrr r!r" __class__s rr%z_Transport.__init__9s  ! FhX]_ef       u% U? OOFE * &&t, rr ctjd||j|j t j |jt j}|dk7r;tjdt||jj|ytjd|j!|jj#}|s|j%yy#t$rYyt$rL}|jr |jtjk(rd}n|j|Yd}~yYd}~d}~wwxYw)NzRead ready on %s %s %drz read %d bytesz got EOF)r&r'rrr(readrrBlockingIOErrorrrerrnoEIOabortlen data_received _close_reader eof_receivedclose)r,dataexc keep_opens r _read_readyz_Transport._read_readyTs -tT^^T[[Q 774;; (=(=>D 3; LL*CI 6 NN ( ( . LL %    335I %   CII$: 3  s.C// E:E=E  Ec|jSN)rr,s r is_readingz_Transport.is_readingmrc2|jd|_yN) pause_readingrr?s rr6z_Transport._close_readerps  rcv|jr-|jj|jd|_yyNF)rr remove_readerrr?s rrEz_Transport.pause_readingts.    JJ $ $T[[ 1$D  rc|jsH|jdk7r8|jj|j|jd|_yyy)NrDT)rrr add_readerr<r?s rr+z_Transport.resume_readingysFDKK2$5 JJ ! !$++t/?/? @#D %6rcyr>r?s r_closez_Transport._closes rcd|_|j|j|jj |j j ||jyNT)rr6_remove_write_queuer call_soonrconnection_lostrM)r,r:s rr3z_Transport.abortsH     " T^^;;SA rctr>NotImplementedErrorr?s r can_write_eofz_Transport.can_write_eof!!rc|jrJd|_|j'tjd||j ytjd|y)NTz%s got EOF. closing backend.z,%s got EOF. bytes in queue, deferring close)rrr&r'_write_eof_nowr?s r write_eofz_Transport.write_eofsI99} ;;  LL8$ ?    ! LLG NrcT|jytd|jDS)Nrc32K|]}t|ywr>)r4).0blocks r z3_Transport.get_write_buffer_size..s7%3u:7s)rsumr?s rget_write_buffer_sizez _Transport.get_write_buffer_sizes$ ;; 74;;777rcy)N)rrrLr?s rget_write_buffer_limitsz"_Transport.get_write_buffer_limitssrc*||dk(sJ||dk(sJyyNrrL)r,highlows rset_write_buffer_limitsz"_Transport.set_write_buffer_limitss(|tqy(({cQh&&h{rctr>rTr?s rrYz_Transport._write_eof_nowrWrc@tjd||jJ tj|j |j}tjd||r|jj}t||kDr4tjd|jj||dn'|t|z}tjd||r|jsttjd|j|jr%tjd|j|jr|jyyy#t $rd}Y*t$r}|j|Yd}~yd}~wwxYw)Nz%s _write_readyrz, successfully wrote %d bytes from the queuez incomplete block. Stop.z& removed complete block. %d remains.z%s queue drained.z'%s queue drained. closing backend now.)r&r'rr(writevrr0rr3popleftr4 appendleftrPrrYr)r,n_bytesr:r^s r _write_readyz_Transport._write_readysC &-{{&&& ii dkk:G  CWMKK'')E5zG# 9: &&uWX7 s5z !G LLA7 K{{ LL, -  $ $ &yy FG##%}}  % G  JJsO  s*E,, F:FFFc|jG|jj|jj |j d|_yyr>)rrresume_writingr remove_writerrr?s rrPz_Transport._remove_write_queues> ;; " NN ) ) + JJ $ $T\\ 2DK #rr9c,tjd||j|jJ|jj |j|j tj|f|_|jjy)Nz!%s creating write queue for fd %s) r&r'rrr add_writerro collectionsdequer pause_writingr,r9s r_create_write_queuez_Transport._create_write_queuesi 8$ M{{""" dllD,=,=>!''0  $$&rc|jr!tjd|jy|jrJ|j |j j |t|j tkDrPdj|j }|j j|j j |y tj|j|}|t|k7r|j!||dyy#t$rd}Y1t$r}|j|Yd}~yd}~wwxYw)Nz+ignoring write() to closing transport fd %irr)rr&r'rrrappendr4IOV_MAXjoinclearr(writer0rr3ry)r,r9all_datarnr:s rrz_Transport.writes == LLF U 99} ;; " KK  t $4;;')88DKK0 !!# ""8,  hht||T2G c$i   $ $T'(^ 4  G  JJsO  s D E"E*EEc|jryd|_|j|jy|jyrO)rr6rr3r?s rr8z_Transport.closes5 ==    ;; "  rc|jSr>)rr?s r get_protocolz_Transport.get_protocols ~~rc|jSr>)rr?s r is_closingz_Transport.is_closings }}rctr>rT)r,rs r set_protocolz_Transport.set_protocolrWrc$|jyr>)rMr?s r__del__z_Transport.__del__ s  r)rDrDNr Nr>)r:zException | Noner N)r ztuple[int, int])NN)rf int | Nonergrr N)&__name__ __module__ __qualname__rrint__annotations__asyncioAbstractEventLoopProtocolboolrr%r<r@r6rEr+rMr3rVrZrarcrhrYrorPbytesryrr8 BaseProtocolrrrr __classcell__r-s@rrr*sf +J +  $ $$ -- K LN JK 1359 00"+++.3 62 D % $ "t"O8s8 '"B ''$'5%5D5@ g22D"W%9%9"d"rrceZdZdZddZy)SubprocessProtocolzBAn extension to asyncio.Protocol for use with SubprocessTransport.Nct)z"Called when subprocess has exited.rTr?s rprocess_exitedz!SubprocessProtocol.process_exiteds!!rr)rrr__doc__rrLrrrrs L"rrceZdZdefdZy) WindowSizevaluecHt|d|_t|d|_y)Nrowscols)r rr)r,rs rr%zWindowSize.__init__sE6* E6* rN)rrrrr%rLrrrrs+j+rrc JeZdZUdZdZded<dZded<dZded<ded <d d d ed e fdZ d&dZ d ddde jdedee dedddef fdZded dfdZd efdZd'dZd efdZd(dZded e j2fd Zd!ej8d dfd"Zd'd#Zd'd$Zd'd%Z xZ!S))SubprocessTransportaA bi-directional transport speaking with stdin/out of a subprocess. Note: this is not really a normal SubprocessTransport. Although it implements the entire API of asyncio.SubprocessTransport, it is not designed to be used with asyncio.SubprocessProtocol objects. Instead, it pair with normal Protocol objects which also implement the SubprocessProtocol defined in this module (which only has a process_exited() method). Whatever the protocol writes is sent to stdin, and whatever comes from stdout is given to the Protocol via the .data_received() function. If stderr is configured as a pipe, the transport will separately collect data from it, making it available via the .get_stderr() method. Nr_pty_fdzsubprocess.Popen[bytes] | None_process _returncodezSpooler | None_stderrFresetrr ct|j,|jj|jdSy)Nrreplace)errors)rgetdecode)r,rs r get_stderrzSubprocessTransport.get_stderr0s5 << #<<##%#077y7I IrcLdtdtddffd d fd }d fd } tjjjj |y#t tf$r,tdj|d jYywxYw) Npidstatusr cP|jk(sJtjtsJ t j |_tjdj jsjjyy#t$r |_YZwxYw)NzProcess exited with status %d) r isinstancerrr(waitstatus_to_exitcoder ValueErrorr&r'rr)rrprocessr,s r child_exitedz4SubprocessTransport.watch_exit..child_exited7s'++% %%dnn.@A AA *#%#<#.pidfd_readyDsB**W[[!4KC JJ $ $U + HHUO f %rctjjd\}}jj ||yre)r(rrrcall_soon_threadsafe)rrrrr,s rwaitpid_threadz6SubprocessTransport.watch_exit..waitpid_threadJs1**W[[!4KC JJ + +L#v Frzcockpit-waitpid-T)nametargetdaemonr) rr( pidfd_openrrrJAttributeErrorrrstart)r,rrrrrs`` @@r watch_exitzSubprocessTransport.watch_exit6s 0c 03 04 0 & &  G  fMM'++.E JJ ! !% 5( f *7;;-8X\ ] c c e fs;A((8B#"B#)ptywindowrrr rrzWindowSize | Nonekwargsc dfd }rtj\|_}||j|||d<t j |f|||dd||_tj||j|j} } d|_nt j |ftjtj|d||_|j jsJ|j jsJ|j jj} |j jj} |j j4t||j jj|_nd|_t |E||| | |j%|j y)Nctttjr&t j dt jdyyre)r SET_PDEATHSIGsignalSIGTERMfcntlioctltermios TIOCSCTTY)rsr preexec_fnz0SubprocessTransport.__init__..preexec_fn`s/ - 0 Aw00!4rstderrT)stdinstdoutrstart_new_session)rrrr)r(openptyrset_window_size subprocessPopenrr8rPIPErrfilenorSpoolerrr$r%r) r,rrr rrrr session_fdr r!r-s ` rr%zSubprocessTransport.__init__Vs| 5 ')zz| $DL*!$$V,)F8 &,,Tf3=j8BVZf^dfDM HHZ  LL$,,6E#D ',,TNQ[Q`Q`8BNFLNDM==&& &&=='' ''MM((//1E]]((//1F ==   +"4)=)=)D)D)FGDLDL x7  &rsizec (|jJtdt|jd}tdt|jd}t j |jtjtjd||y)Nri2H4x) rmaxminrrrrr TIOCSWINSZstructpack)r,rrrs rrz#SubprocessTransport.set_window_sizesh||'''1c$))Y/01c$))Y/0 DLL'"4"4fkk&$PT6UVrcN|jJ|jjduSr>)rrr?s rrVz!SubprocessTransport.can_write_eofs'}}(((}}""$..rc|jJ|jjJ|jjjd|_yrC)rrr8rr?s rrYz"SubprocessTransport._write_eof_nowsE}}(((}}""... !!# rcJ|jJ|jjSr>)rrr?s rget_pidzSubprocessTransport.get_pids"}}(((}}   rc|jSr>)rr?s rget_returncodez"SubprocessTransport.get_returncoderArfdctr>rT)r,rs rget_pipe_transportz&SubprocessTransport.get_pipe_transportrWrsigc|jJ|j,tjd||jjy t j |jj|tjd||jjy#t$r.tjd||jjYywxYw)Nz3won't attempt %s to process %i. It exited already.zsent %s to process %iz3can't send %s to process %i. It's exited just now.)rrr&r'rr(killProcessLookupError)r,rs r send_signalzSubprocessTransport.send_signals}}(((    ' LLNPSUYUbUbUfUf g  h GGDMM%%s + LL0#t}}7H7H I! h LLNPSUYUbUbUfUf g hsAB4CCcB|jtjyr>)rrrr?s r terminatezSubprocessTransport.terminate (rcB|jtjyr>)rrSIGKILLr?s rrzSubprocessTransport.killrrc|j&tj|jd|_|j]|jj5|jjjd|j_ |j yy#t $r-tjd|jjYywxYw)Nzcan't kill %i due to EPERM) rr(r8rrrPermissionErrorr&r'rr?s rrMzSubprocessTransport._closes << # HHT\\ "DL == $}}"". ##))+&* # N % # N 94==;L;LM Ns B3CC)rzsubprocess.Popen[bytes]r Nr)r r)"rrrrrrrrrstrrrrrrrrr%rrrVrYrrr TransportrrSignalsrrrrMrrs@rrrs !G\ 15H.5 $K$ */4C fJ#/3 ,'00,'-,' },'  ,' - ,','\WJW4W /t/ !! "S"W->->"hv~~h$h0)) Nrrc reZdZdZd dej dej dedeffd Zde fdZ d d Z xZ S) StdioTransportzA bi-directional transport that corresponds to stdin/out. Can talk to just about anything: - files - pipes - character devices (including terminals) - sockets rrrrc*t|||||yr>)r$r%)r,rrrrr-s rr%zStdioTransport.__init__s x7rr cyrGrLr?s rrVzStdioTransport.can_write_eofsrctd)NzCan't write EOF to stdout) RuntimeErrorr?s rrYzStdioTransport._write_eof_nows677r)rrr) rrrrrrrrr%rrVrYrrs@rrrsH8W668'BRBR8[^8lo8t8rrceZdZUdZej ed<eed<ded<dej defdZdd Z d e fd Z d dde d e fdZ ddZddZy )rzConsumes data from an fd, storing it in a buffer. This makes a copy of the fd, so you don't have to worry about holding it open. r_fdz list[bytes] _contentsrrc||_d|_g|_tj||_tj |jd|j |j|jy)NrDF)rr r r(dupr)rJr<)r,rrs rr%zSpooler.__init__sP 66": %( $"2"23rr Nc tj|jd}|dk7r|j j |y|jy#t$rYyt$rd}YIwxYw)Ni r)r(r/r r0rr r{r8rxs rr<zSpooler._read_readys` 77488T*D 3; NN ! !$ ' JJL   D s A A, A,+A,ct|jdk(rytj|jgggdgggfk7S)NrDFr)r selectr?s r _is_readyzSpooler._is_readys6 88r>}}dhhZR3B|CCrFrrc|jr!|j|jr!dj|j}|rg|_|S)Nr)rr<r}r )r,rresults rrz Spooler.getsGnn    nn$..) DN rc|jdk7rL|jj|jtj|jd|_yyrC)r rrHr(r8r?s rr8z Spooler.closes? 88r> JJ $ $TXX . HHTXX DH rc$|jyr>)r8r?s rrzSpooler.__del__s  rr)rrrrrrrrr%r<rrrrr8rrLrrrrsn  $ $$ H4W664C4 D4D $)DU rr)'rrructypesr1rloggingr(rrrrr threadingrtypingrrrjsonutilrr CDLLrr rr getLoggerrr&r|rrrrrrrrrLrrrs R  **) FKK  # # (((     8 $ a""aH"))"++ mN*g&A&AmN`8Z8(88r