Mh_S NdZddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZee ZeeZeeZGddejZeeZGddejZeeZGd d ejZeeZd Zd Zd ZdZdZdZdZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)ej*j+j,Z,ge,_-e e,_.ej*j+j/Z0ee e e e e ege0_-ee0_.ej*j+j1Z1eeeeee ge1_-e e1_.ej*j+j2Z3eeeee e e eeeg e3_-e e3_.ej*j+j4Z4eege4_-e e4_.ej*j+j5Z5ge5_-ee5_.ej*j+j6Z6ege6_-e e6_.ej*j+j7Z7ee e ee ge7_-e e7_.ej*j+j8Z8ee e ee ge8_-e e8_.ej*j+j9Z9eege9_-e e9_.ej*j+j:Z:ee ge:_-e e:_.ej*j+j;Z;ege;_-e e;_.ej*j+j<ZZ>ee e ge>_-e e>_.ej*j+j?Z?ege?_-e e?_.ej*j@jAZAeeejBgeA_-eeeA_.ej*j+jCZCe geC_-e eC_.GddZDGddZEdeFddfdZGeHdkr#eIdeGd eId!dSdS)"zvWindows-specific implementation of process utilities with direct WinAPI. This file is meant to be used by process.py N)POINTER)HANDLEHLOCALLPVOIDWORDDWORDBOOLULONGLPCWSTRc$eZdZdefdefdefgZdS)SECURITY_ATTRIBUTESnLengthlpSecurityDescriptorbInheritHandleN)__name__ __module__ __qualname__rrr _fields_g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/IPython/utils/_process_win32_controller.pyr r s-E"'0!4(*HHHrr c~eZdZdefdefdefdefdefdefdefdefd efd efd efd efd efdefdefdefdefdefgZdS) STARTUPINFOcb lpReserved lpDesktoplpTitledwXdwYdwXSizedwYSize dwXCountChars dwYCountCharsdwFillAttributedwFlags wShowWindow cbReserved2 lpReserved2 hStdInput hStdOutput hStdErrorN) rrrrr rrrrrrrrrsu w'g&G$E"E" %( %("E*E"%%'f%v&f%#'HHHrrc*eZdZdefdefdefdefgZdS)PROCESS_INFORMATIONhProcesshThread dwProcessId dwThreadIdN)rrrrrrrrrr-r-2s5V$F#&u%'HHHrr-&miiic"eZdZdZddZddZdS) AvoidUNCPathaA context manager to protect command execution from UNC paths. In the Win32 API, commands can't be invoked with the cwd being a UNC path. This context manager temporarily changes directory to the 'C:' drive on entering, and restores the original working directory on exit. The context manager returns the starting working directory *if* it made a change and None otherwise, so that users can apply the necessary adjustment to their system calls in the event of a change. Examples -------- :: cmd = 'dir' with AvoidUNCPath() as path: if path is not None: cmd = '"pushd %s &&"%s' % (path, cmd) os.system(cmd) returnNctj|_|jd|_|jrtjd|jSdS)Nz\\zC:)osgetcwdpath startswith is_unc_pathchdirselfs r __enter__zAvoidUNCPath.__enter__sNIKK 9//66    HTNNN9 4rcJ|jrtj|jdSdSN)rCr?rDrArFexc_type exc_value tracebacks r__exit__zAvoidUNCPath.__exit__s/   HTY       r)r=N)rrr__doc__rGrNrrrr<r<sF(          rr<cXeZdZdZddZdZdZdZddZd Z d Z d Z d Z d Z dZdS)Win32ShellCommandControlleraRuns a shell command in a 'with' context. This implementation is Win32-specific. Example: # Runs the command interactively with default console stdin/stdout with ShellCommandController('python -i') as scc: scc.run() # Runs the command using the provided functions for stdin/stdout def my_stdout_func(s): # print or save the string 's' write_to_stdout(s) def my_stdin_func(): # If input is available, return it as a string. if input_available(): return get_input() # If no input available, return None after a short delay to # keep from blocking. else: time.sleep(0.01) return None with ShellCommandController('python -i') as scc: scc.run(my_stdout_func, my_stdin_func) Tc"||_||_dS)aUInitializes the shell command controller. The cmd is the program to execute, and mergeout is whether to blend stdout and stderr into one output in stdout. Merging them together in this fashion more reliably keeps stdout and stderr in the correct order especially for interactive shell usage. N)cmdmergeout)rFrSrTs r__init__z$Win32ShellCommandController.__init__s  rc |j}|j}d\|_|_|_d|_ dgdz\}}}}}}t tj  _ d _ d _ fd} | d\}}|rft}tt|ttj|ddt stjn| d\}}| d\}}t%} t'} tj| | _|| _|| _|| _t0| _t4t6z} t9dd|zddd| ddtj| tj|  stjt;|d}t;|d}|t;|d}||_d}||_d}|s ||_d}| |_|rt;||rt;||rt;||rt;||rt;||rt;|nl#|rt;||rt;||rt;||rt;||rt;||rt;|wwxYw|S) NNNNTcttf}ttj|dtj|dtjdstjt ||t dstj|dj|djfS)zCreates a Windows pipe, which consists of two handles. The 'uninherit' parameter controls which handle is not inherited by the child process. rr5)r CreatePipectypesbyrefWinErrorSetHandleInformationHANDLE_FLAG_INHERITvalue) uninherithandlessaAttrs r create_pipez:Win32ShellCommandController.__enter__..create_pipes !((FHH,!&,wqz":":"L44fl66J6JAOO, /++++GI,>/44, /+++qz')999rr)rar5z cmd.exe /c )rSrThstdouthstdinhstderr piProcInfor r[sizeofrrrrDuplicateHandleGetCurrentProcessr\DUPLICATE_SAME_ACCESSr]r-rrr)r*r+STARTF_USESTDHANDLESr%CREATE_SUSPENDEDCREATE_NO_WINDOW CreateProcess CloseHandle)rFrSrT p_hstdout c_hstdout p_hstderr c_hstderrp_hstdinc_hstdinrdrh siStartInfodwCreationFlagsrcs @rrGz%Win32ShellCommandController.__enter__s{h=2B/ dk4<V '59F1H 2Iy)x)**F#]622FN$(F !*.F ' : : : : :$/;#;#;#; Iy @"HH &'8':':I 1 3 3V\)5L5L !4)>@@,!/+++, (3{Q'?'?'?$ 9#.;#;#;#; Hx-..J%--K#];77KN$,K !%.K "$-K !"6K .1AAO "S($o$ [ 9 9L,, .. ( o'''  ! ! !H  " " "I$I&&& #DKH$DLI !(  (DO &H%%% &H%%% 'I&&& 'I&&& 'I&&& 'I&&& &H%%% &H%%% 'I&&& 'I&&& 'I&&& 'I&&&& ' s G,JA)K)c\t}td} |}|t|tj|stj|jt krdSt|ddtj|dstjt|tr| d}t|tstdt|dkrdS||t|dkrt||t|tj|ds,ttkrdStj||jd}t|dk)NrTutf_8z$internal stdin function string error)rGetExitCodeProcessr[r\r]r` STILL_ACTIVE WriteFile isinstanceunicodeencodestr RuntimeErrorlen GetLastError ERROR_NO_DATA)rFhandlehprocessfunc stdout_funcexitCode bytesWrittendatas r _stdin_threadz)Win32ShellCommandController._stdin_threadFs77Qxx - 1466D|)(FL4J4JKK, /+++>\11F Q \22D::, /+++$(( ,{{7++dC(( K"#IJJJ4yyA~~ K   d))q.. s4yy \22D::,$~~66 /+++L.//0d))q..I- 1rcXtjd} td}t||dtj|ds.t }|t krdStj|jd|j}|| dd)NiTrr|replace) r[create_string_bufferrReadFiler\rERROR_BROKEN_PIPEr]r`decode)rFrrr bytesReadless r_stdout_threadz*Win32ShellCommandController._stdout_threadxs*400 /aIFD$ Y//77 ,!^^***F /+++ 1Y_,-A D'9-- . . . /rNcR||||S||jrtdd}g}|r.tj|j|j|jj||f}| tj|j |j |f|js9||}| tj|j |j |ft|jjdkrtj|||D]}|t%|jjt&t(krtj|D]}|||dSdS)aRuns the process, using the provided functions for I/O. The function stdin_func should return strings whenever a character or characters become available. The functions stdout_func and stderr_func are called whenever something is printed to stdout or stderr, respectively. These functions are called from different threads (but not concurrently, because of the GIL). NzqShell command was initiated with merged stdin/stdout, but a separate stderr_func was provided to the run() method)targetargsr9) _run_stdiorTr threadingThreadrrfrhr.appendrrerg ResumeThreadr/r[r]startWaitForSingleObjectINFINITE WAIT_FAILEDjoin)rFr stdin_func stderr_func stdin_threadthreadsthreads rrunzWin32ShellCommandController.runs  :#5+:M??$$ $  "t} " 788 8    :$+43E&*k4?3K *K&9:::L y't/B*., )DFFF G G G} K") NN9+43F.2lK-HJJJ K K K / 0 0J > >/## #  #      F LLNNNN t7 B B  /## #  F KKMMMM  #         $ #rctjtj}t |d}|t krtj|tkrtdddStj d}td}tddt||dtj|dstjt||j}|d d }|d d }tt%|d zd|S) z>Use the raw Win32 handle of sys.stdin to do non-blocking readsd.r{)endNr6r?z    )msvcrt get_osfhandlesysstdinfilenorrr[r] WAIT_TIMEOUTprintrrrr\FlushConsoleInputBufferr`rrepr)rFrresultrrs r_stdin_raw_nonblockz/Win32ShellCommandController._stdin_raw_nonblocks0 %ci&6&6&8&899$VS11 [ /## # | # # #2    4.s33DaI #2    FD# Y//77 (o''' $F + + +:D<<--D<<d++D $t**s" + + + +Krc tjd}|dd}|S#t$r}|jt krYd}~dS|d}~wwxYw)zUse a blocking stdin readr5rrN)rrreadr WindowsErrorwinerrorr)rFrwes r_stdin_raw_blockz,Win32ShellCommandController._stdin_raw_blocksr 9>>!$$D<<d++DK   {m++ttttt  s69 A AAA czt|dtjtjdSzWrites the string to stdoutr{)rfileN)rrstdoutflushrFrs r _stdout_rawz'Win32ShellCommandController._stdout_raw3 aRcj)))) rczt|dtjtjdSr)rrstderrrrs r _stderr_rawz'Win32ShellCommandController._stderr_rawrrc|jr!||j|jS||j|j|jS)zRuns the process using the system standard I/O. IMPORTANT: stdin needs to be asynchronous, so the Python sys.stdin object is not used. Instead, msvcrt.kbhit/getwch are used asynchronously. )rr)rrr)rTrrrrrEs rrz&Win32ShellCommandController._run_stdiosb = 488$*:!%!688 888$*:!%!6"&"244 4rcV|jrt|jd|_|jrt|jd|_|jrt|jd|_|j;t|jjt|jjd|_dSdSrI)rfrqrergrhr.r/rJs rrNz$Win32ShellCommandController.__exit__s ;   $ $ $DK <  % % %DL <  % % %DL ? & 0 1 1 1 / 0 0 0"DOOO ' &r)TrW)rrrrOrUrGrrrrrrrrrNrrrrQrQs6 ! ! ! !^^^@010101d///"1 1 1 1 f>"  444: # # # # #rrQrSr=ct5}|d|d|}t|5}|dddn #1swxYwYddddS#1swxYwYdS)aWin32 version of os.system() that works with network shares. Note that this implementation returns None, as meant for use in IPython. Parameters ---------- cmd : str A command to be executed in the system shell. Returns ------- None : we explicitly do NOT return the subprocess status code, as this utility is meant to be used extensively in IPython, where any return value would trigger : func:`sys.displayhook` calls. Nz"pushd z &&")r<rQr)rSrAsccs rsystemr#s 4   '+ttSS1C ( - -  GGIII               s3A"A  A" A A"A A""A&)A&__main__zTest starting!z python -izTest finished!)JrOr?rrr[rrctypes.wintypesrrrrrr r r LPDWORDLPHANDLE ULONG_PTR Structurer LPSECURITY_ATTRIBUTESr LPSTARTUPINFOr-LPPROCESS_INFORMATIONERROR_HANDLE_EOFrrr_rmrnCREATE_NEW_CONSOLEror~rrrrlENABLE_ECHO_INPUTENABLE_LINE_INPUTENABLE_PROCESSED_INPUTwindllkernel32rargtypesrestype CreateFileW CreateFilerZCreateProcessWrpr}rkrrrGetConsoleModeSetConsoleModerrrjr^rqshell32CommandLineToArgvWc_int LocalFreer<rQrrrrrrrrs '%.. 76?? GENN *****&****  344'''''&"'''& $$ '''''&*'''   344    "}%2    ] # / ufeUFK   ] # . wvu&   &5 !7,AtUFG]  ]+>%w/!M*<"}%2    = ! *VUGV< M " , feWf=  '6!7+'6!5/ -0H$*8 "&m,@ &#-(8"FFH tU}-B!' 6#m$0 x  ]*=& (=(=>$WW-- M " , X  # # # # # # # # Lb#b#b#b#b#b#b#b#J . z E  F; E  r