o b1@sdZddlZddlZddlmZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl mZmZddlmZddlmZmZmZddlmZe Zde_d d ZGd d d ejZd dZddZeeeeGdddej eZ!dS)z< Windows Process Management, used with reactor.spawnProcess N) implementer) _pollingfileerror) BaseProcess) IConsumerIProcessTransport IProducer)quoteArgumentscCst|tjdSN)printsysstdoutflush)msgrA/usr/lib/python3/dist-packages/twisted/internet/_dumbwin32proc.pydebug$src@seZdZddZddZdS)_ReapercCs ||_dSr )proc)selfrrrr__init__*s z_Reaper.__init__cCs@t|jjdtjkr dSt|jj}||j|dSNr) win32eventWaitForSingleObjectrhProcess WAIT_OBJECT_0 win32processGetExitCodeProcess deactivate processEnded)rexitCoderrr checkWork-s z_Reaper.checkWorkN)__name__ __module__ __qualname__rr"rrrrr)s rcCs^t|!}|ddkr|dd}|WdSWddS1s(wYdS)a  Look for a #! line, and return the value following the #! if one exists, or None if this file is not a script. I don't know if there are any conventions for quoting in Windows shebang lines, so this doesn't support any; therefore, you may not pass any arguments to scripts invoked as filters. That's probably wrong, so if somebody knows more about the cultural expectations on Windows, please feel free to fix. This shebang line support was added in support of the CGI tests; appropriately enough, I determined that shebang lines are culturally accepted in the Windows world through this page:: http://www.cgi101.com/learn/connect/winxp.html @param filename: str representing a filename @return: a str representing another filename. z#!i N)openreadreadlinestrip)filenamefexerrr _findShebang9s "r/cCs|jddkS)z Determine if a pywintypes.error is telling us that the given process is 'not a valid win32 application', i.e. not a PE format executable. @param pywinerr: a pywintypes.error instance raised by CreateProcess @return: a boolean r)args)pywinerrrrr_invalidWin32AppTsr3c@seZdZdZdZddZddZddZd d Zd d Z d dZ ddZ ddZ ddZ ddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1efd2d3Zd4S)5Processa A process that integrates with the Twisted event loop. If your subprocess is a python program, you need to: - Run python.exe with the '-u' command line option - this turns on unbuffered I/O. Buffering stdout/err/in can cause problems, see e.g. http://support.microsoft.com/default.aspx?scid=kb;EN-US;q1903 - If you don't want Windows messing with data passed over stdin/out/err, set the pipes to be in binary mode:: import os, sys, mscvrt msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY) msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY) msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY) rc s tj|t|t}d|_t|d\_ }t|d\_ } t|d\} _ t j tj ddt|_| _| _tj_t} t| j | ddtj} tj | _ t| j | ddtj} tj | _ t| j | ddtj} tj | _ tj|pidd Ddd|D}t!|rt"nrt"nfdd } z| Wnat#j$y0}zSt%|st&|t'}|durt&d t(|}|)dt!|}|z| Wnt#j$y%}zt%|rt&d ||ft&|d}~wwWYd}~nd}~wwt| t|t| t*j fd d j+_,t*j fdd j-_.t/j j0_1j,j.j1fD]}2|qmj342t5dS)z- Create a new child process. r rNcSs"i|] \}}t|t|qSrosfsdecode).0keyvaluerrr s"z$Process.__init__..cSsg|]}t|qSrr5)r8xrrr sz$Process.__init__..c s2tj}tddd| \___}dSNr )win32conCREATE_NO_WINDOWr CreateProcessrhThreadpid)flagsdwTid StartupInfocmdlinecommandenvpathrrrdoCreatesz"Process.__init__..doCreatezD%r is neither a Windows executable, nor a script with a shebang linez<%r has an invalid shebang line: %r is not a valid executablecjd|Sr>protochildDataReceiveddatarrrz"Process.__init__..crMNr&rNrQrSrrrTrU)6r _PollingTimerrr win32securitySECURITY_ATTRIBUTESbInheritHandle win32pipe CreatePipehStdoutRhStderrRhStdinWSetNamedPipeHandleState PIPE_NOWAITr STARTUPINFO hStdOutput hStdError hStdInputSTARTF_USESTDHANDLESdwFlagswin32apiGetCurrentProcessDuplicateHandler?DUPLICATE_SAME_ACCESS win32file CloseHandler6environcopyupdateitemsr r7 pywintypesrr3OSErrorr/listinsert_PollableReadPipeoutConnectionLostrerrConnectionLoststderr_PollableWritePipeinConnectionLoststdin_addPollableResourcerOmakeConnectionr)rreactorprotocolrIr1 environmentrKsAttrshStdoutWhStderrWhStdinR currentPidtmprLpwtesheborigcmdpwte2 pipewatcherrrFrr|s            !     zProcess.__init__cCs0|jdur t|dvrt|jddSdS)N)INTTERMKILLr )rCrProcessExitedAlreadyrTerminateProcessr)rsignalIDrrr signalProcesss zProcess.signalProcesscCs|dkr t|St|Sr)r ProcessDoneProcessTerminated)rstatusrrr _getReason s  zProcess._getReasoncC|j|dS)zQ Write data to the process' stdin. @type data: C{bytes} N)r|write)rrRrrrrz Process.writecCr)z[ Write data to the process' stdin. @type seq: C{list} of C{bytes} N)r| writeSequence)rseqrrrrrzProcess.writeSequencecCs |dkr |j|dSt|)a; Similar to L{ITransport.write} but also allows the file descriptor in the child process which will receive the bytes to be specified. This implementation is limited to writing to the child's standard input. @param fd: The file descriptor to which to write. Only stdin (C{0}) is supported. @type fd: C{int} @param data: The bytes to write. @type data: C{bytes} @return: L{None} @raise KeyError: If C{fd} is anything other than the stdin file descriptor (C{0}). rN)r|rKeyError)rfdrRrrr writeToChildszProcess.writeToChildcCsD|dkr |dS|dkr|dS|dkr|dStd)Nrr r&z4Only standard-IO file descriptors available on win32) closeStdin closeStdout closeStderrNotImplementedError)rrrrr closeChildFD7s   zProcess.closeChildFDcCs|jdS)zClose the process' stdin.N)r|closerSrrrrCszProcess.closeStdincC|jdSr )ryrrSrrrrGzProcess.closeStderrcCrr )rrrSrrrrJrzProcess.closeStdoutcCs|||dS)z8 Close the process' stdout, in and err. N)rrrrSrrrloseConnectionMs zProcess.loseConnectioncC|jd|dSr>rOchildConnectionLostconnectionLostNotifyrSrrrrwU  zProcess.outConnectionLostcCrrVrrSrrrrxYrzProcess.errConnectionLostcCrrrrSrrrr{]rzProcess.inConnectionLostcCs|jd7_|dS)zS Will be called 3 times, by stdout/err threads and process handle. r N)closedNotifiesmaybeCallProcessEndedrSrrrras zProcess.connectionLostNotifycCsJ|jdkr!|jr#t|jt|jd|_d|_t|dSdSdS)N)r lostProcessrlrmrrBrrrSrrrrhs  zProcess.maybeCallProcessEndedcCs|j||dSr )r|registerProducer)rproducer streamingrrrrqszProcess.registerProducercCrr )r|unregisterProducerrSrrrrtrzProcess.unregisterProducercC |dSr )_pauserSrrrpauseProducingx zProcess.pauseProducingcCrr )_unpauserSrrrresumeProducing{rzProcess.resumeProducingcCrr )rrSrrr stopProducing~rzProcess.stopProducingcCtd)NzUnimplemented: Process.getHostrrSrrrgetHostzProcess.getHostcCr)NzUnimplemented: Process.getPeerrrSrrrgetPeerrzProcess.getPeerreturncCsd|jjd|jdS)z@ Return a string representation of the process. ) __class__r#rCrSrrr__repr__szProcess.__repr__N)r#r$r%__doc__rrrrrrrrrrrrrwrxr{rrrrrrrrrstrrrrrrr4es8   r4)"rr6r zope.interfacerrrrhr?rrlr[rrXtwisted.internetrrtwisted.internet._baseprocessrtwisted.internet.interfacesrrrtwisted.python.win32r rYPIPE_ATTRS_INHERITABLErZr_PollableResourcerr/r3rWr4rrrrs0