d:ddlZddlZddlZddlZddlmcmZddlmZe edsdZ Gdde Z e e_ e e_ dZ dd Zdd Zd Zd ZGd de ZdZddej(ej(ddddfdZdej(ej(ddfdZdZgZej2ZdZdZdZdZdZy)Nustr check_outputcd|vr tdtj|dtji|}|j \}}|j }|r0|j d}||d}tj||||S)z/Backport from subprocess module from python 2.7stdoutz3stdout argument not allowed, it will be overridden.argsroutput) ValueError subprocessPopenPIPE communicatepollgetCalledProcessError) popenargskwargsprocessr unused_errretcodecmds H/usr/lib/python3/dist-packages/azurelinuxagent/common/utils/shellutil.pyrrs v 67 7""IP*//PP$002 ,,. **V$C{l//VL L ceZdZddZdZy)rNc.||_||_||_yN) returncoderr )selfrrr s r__init__zCalledProcessError.__init__/s(DODH DKrcNdj|j|jS)Nz/Command '{0}' returned non-zero exit status {1})formatrr)rs r__str__zCalledProcessError.__str__4s txx9 :rr)__name__ __module__ __qualname__r r#rrrr.s  !  :rrct|d S)z9 Return True if the given command is on the path F)run)rs r has_commandr*Cs3 rTc0|g}t|||\}}|S)a) Note: Deprecating in favour of `azurelinuxagent.common.utils.shellutil.run_command` function. Calls run_get_output on 'cmd', returning only the return code. If chk_err=True then errors will be reported in the log. If chk_err=False then errors will be suppressed from the log. )chk_errexpected_errors)run_get_output)rr,r-routs rr)r)Js&!#wXLGS Nrc |g}|rtjd| t|tjtj d}|j \}}t|jt|}|jdk7rf|rVdj||j|}|j|vrtj|ntj||j|fS d|fS#t$rG}|r.tjdj|t|dt|fcYd}~Sd}~wwxYw) a Wrapper for subprocess.check_output. Execute 'cmd'. Returns return code and STDOUT, trapping expected exceptions. Reports exceptions to Error if chk_err parameter is True For new callers, consider using run_command instead as it separates stdout from stderr, returns only stdout on success, logs both outputs and return code on error and raises an exception. NzCommand: [{0}]T)rstderrshellrz1Command: [{0}], return code: [{1}], result: [{2}]0Command [{0}] raised unexpected exception: [{1}])loggerverbose_popenr rSTDOUTr_on_command_completedpid__encode_command_outputrr"infoerror Exceptionr) rr,log_cmdr-rr _msg exceptions rr.r.Ws%(#.#Z__Z=N=NVZ['') gkk*(0    "''-vc73E3Ev'N%%8KK$LL%%%v- - # f9 #  LLL &d9o6 84 ?"" #sCC66 E? __format_command("sort") 'sort' > __format_command(["sort", "-u"]) 'sort -u' > __format_command([["sort"], ["unique", "-n"]]) 'sort | unique -n' rz |  ) isinstancelistjoin)commandrs r__format_commandrIsZ'4 z'!*d3::@sxx}@A Axx  NAsA'c&t||ndddS)zH Encodes the stdout/stderr returned by subprocess.communicate() rzutf-8backslashreplace)encodingerrorsrr s rr;r;s &,#Pb ccrc2eZdZdZedZfdZxZS) CommandErrorzT Exception raised by run_command/run_pipe when the command returns an error ct|trt|dkDr|dn|}dj|||j S)Nrz'{0}' failed: {1} ({2}))rErFlenr"rstrip)rH return_coder1 command_names r _get_messagezCommandError._get_messages>%/%>3wr rOrUrHrrr1)rrHrSrr1 __class__s rr zCommandError.__init__s> i' (A(A';X^(_` %  r)r$r%r&__doc__ staticmethodrUr __classcell__)rXs@rrOrOs(\\rrOc\ |\}}}|rt|}t|}|dk7r<|r"tjdt||||t t|||||S#t$rt $r1}|r)tjdt|t |d}~wwxYw)a Executes the given command_action and returns its stdout. The command_action is a function that executes a command/pipe and returns its exit code, stdout, and stderr. If there are any errors executing the command it raises a RunCommandException; if 'log_error' is True, it also logs details about the error. If encode_output is True the stdout is returned as a string, otherwise it is returned as a bytes object. rz?Command: [{0}], return code: [{1}], stdout: [{2}] stderr: [{3}])rHrSrr1r3N)r;r5r=rIrOr>r)command_actionrH log_error encode_outputrSrr1rBs r __run_commandr`s&4&6# VV ,V4F,V4F !  U$W-  '7'@kbhqwx x    LLLN^_fNgimnwix y sA%A((B+:,B&&B+Fc b  tdfd} t| ||S)aV Executes the given command and returns its stdout. If there are any errors executing the command it raises a RunCommandException; if 'log_error' is True, it also logs details about the error. If encode_output is True the stdout is returned as a string, otherwise it is returned as a bytes object. If track_process is False the command is not added to list of running commands This function is a thin wrapper around Popen/communicate in the subprocess module: * The 'input' parameter corresponds to the same parameter in communicate * The 'stdin' parameter corresponds to the same parameters in Popen * Only one of 'input' and 'stdin' can be specified * The 'stdout' and 'stderr' parameters correspond to the same parameters in Popen, except that they default to subprocess.PIPE instead of None * If the output of the command is redirected using the 'stdout' or 'stderr' parameters (i.e. if the value for these parameters is anything other than the default (subprocess.PIPE)), then the corresponding values returned by this function or the CommandError exception will be empty strings. Note: This is the preferred method to execute shell commands over `azurelinuxagent.common.utils.shellutil.run` function. z4The input and stdin arguments are mutually exclusivecTdx}}4tj}r ttrj n}  }d} rt | d}ntj | d}|j|\}} rt|j|j||fS)NF)stdinrr1r2)input) r rrEstrencoder7r rr9r:r) popen_stdincommunicate_inputrcommand_stdoutcommand_stderrrH encode_inputrdr1rcr track_processs rr]z#run_command..command_actions*.. '  $//K2>:eUXCY _d   K $  WKv]bcG &&wk&Y_glmG)0)<)>AArr]rHr^r_)r r`) rHrdrcrr1r^rkr_rlr]s ````` ` ` r run_commandrns=. U.OPPBB( S\ly zzrcjtdkr tdfd}t|||S)aN Executes the given commands as a pipe and returns its stdout as a string. The pipe is a list of commands, which in turn are a list of strings, e.g. [["sort"], ["uniq", "-n"]] represents 'sort | unique -n' If there are any errors executing the command it raises a RunCommandException; if 'log_error' is True, it also logs details about the error. If encode_output is True the stdout is returned as a string, otherwise it is returned as a bytes object. This function is a thin wrapper around Popen/communicate in the subprocess module: * The 'stdin' parameter is used as input for the first command in the pipe * The 'stdout', and 'stderr' can be used to redirect the output of the pipe * If the output of the pipe is redirected using the 'stdout' or 'stderr' parameters (i.e. if the value for these parameters is anything other than the default (subprocess.PIPE)), then the corresponding values returned by this function or the CommandError exception will be empty strings. z,The pipe must consist of at least 2 commandsc.d} } tjk(rtj}|}n }g}d}|t dz krU|j t ||tj|||j}|dz }|t dz krU|j t || |d}|t |dz kr4||jj|dz }|t |dz kr4||j\}}|D]}t|j|!|jd|j}||j||f||jSS#||jwwxYw)Nr)rcrr1)r rtempfile TemporaryFilerQappendr7rcloserr9r:seekreadr) stderr_filerg popen_stderr processesi pipe_stdout pipe_stderrprocpiper1rcrs rr]z run_pipe..command_actions $ $K(&446 * % IAc$i!m#  Q{:??co!pq'l11 Qc$i!m#   VDG;vVbc dAc)nq((! ##))+Qc)nq(((1|'?'?'A $K! 0%dhh/ 0&  #)..0 Q<**KD&!!#'{&!!#'sBE?A'E?A&E??Frm)rQr r`)rrcrr1r^r_r]s```` rrun_pipers7( 4y1}GHH'$R PYiv wwrcvt|ttfs|f}djtd|DS)a Quote a list or tuple of strings for Unix Shell as words, using the byte-literal single quote. The resulting string is safe for use with ``shell=True`` in ``subprocess``, and in ``os.system``. ``assert shlex.split(ShellQuote(wordList)) == wordList``. See POSIX.1:2013 Vol 3, Chap 2, Sec 2.2.2: http://pubs.opengroup.org/onlinepubs/9699919799/utilities/V3_chap02.html#tag_18_02_02 rDc3^K|]%}dj|jdd'yw)z'{0}''z'\''N)r"replace).0ss r zquote..Ss$TQ #w(?@Ts+-)rEtuplerFrG) word_lists rquoterEs3 i% /L 88DT)TT UUr%AZURE_GUEST_AGENT_PARENT_PROCESS_NAMEAZURE_GUEST_AGENTc>t5i}d|vr|j|dn|jtjt|t <||d<t j|i|}tj|j|cdddS#1swYyxYw)Nenv) _running_commands_lockupdateosenvironrPARENT_PROCESS_NAMEr r _running_commandsrur:)rrrrs rr7r7`s  F? JJve} % JJrzz "$5  u ""D3F3  -s BBBcdt5tj|dddy#1swYyxYwr)rrremove)r:s rr9r9rs( &  %&&&s&/cLt5tddcdddS#1swYyxYw)ar Returns the commands started by run/run_get_output/run_command/run_pipe that are currently running. NOTE: This function is not synchronized with process completion, so the returned array may include processes that have already completed. Also, keep in mind that by the time this function returns additional processes may have started or completed. N)rrr'rrget_running_commandsrws$ $ #$$$s #)TN)TTN) rr rs threadingazurelinuxagent.common.loggercommonr5azurelinuxagent.common.futurerhasattrrr>rr*r)r.rIr;rOr`rrnrrrRLockrrrr7r9rr'rrrs & ...z>* ":Y:+J$6J!  &R&d9""L $4 PZP_P_kp@DTXhl.{bjoojooY^nr@xFV*(*='$& $r