ó ½-'Nc@sˆdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z ddl m Z ddlmZddlmZmZmZeƒZdZd Zd Zd efd „ƒYZdd „Zdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZ dee fd„ƒYZ!defd„ƒYZ"defd„ƒYZ#dS(su Functionality for running arbitrary shell scripts. @var ALL_USERS: A token indicating all users should be allowed. iÿÿÿÿN(tProcessProtocol(tDeferredtfailt maybeDeferred(t ProcessDone(t build_script(t ManagerPlugint SUCCEEDEDtFAILEDifigs</usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bintUnknownUserErrorcBseZRS((t__name__t __module__(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR scCsªd}d}d}|dk r|jdƒ}ytj|ƒ}Wn!tk rctd|ƒ‚nX|j}|j}|j}t j j |ƒsd}qn|||fS(Nsutf-8uUnknown user '%s't/( tNonetencodetpwdtgetpwnamtKeyErrorR tpw_uidtpw_gidtpw_dirtostpathtexists(tusernametuidtgidRt username_strtinfo((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyt get_user_info s      tProcessTimeLimitReachedErrorcBseZdZd„ZRS(s™ Raised when a process has been running for too long. @ivar data: The data that the process printed before reaching the time limit. cCs ||_dS(N(tdata(tselfR((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyt__init__:s(R R t__doc__R!(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR2stProcessFailedErrorcBseZdZd„ZRS(s“Raised when a process exits with a non-0 exit code. @ivar data: The data that the process printed before reaching the time limit. cCs||_||_dS(N(Rt exit_code(R RR$((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR!Es (R R R"R!(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR#>stUnknownInterpreterErrorcBs eZdZd„Zd„ZRS(sˆRaised when the interpreter specified to run a script is invalid. @ivar interpreter: the interpreter specified for the script. cCs#||_tj||jƒƒdS(N(t interpretert ExceptionR!t _get_message(R R&((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR!Ps cCs d|jS(NsUnknown interpreter: '%s'(R&(R ((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR(Ts(R R R"R!R((((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR%Js tScriptRunnerMixincBs/eZdd„Zd„Zd„Zd„ZRS(cCs,|dkrddlm}n||_dS(sn @param process_factory: The L{IReactorProcess} provider to run the process with. iÿÿÿÿ(treactorN(R ttwisted.internetR*tprocess_factory(R R,((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR!Zs cCs(|jjjƒ}|tkp'||kS(N(tregistrytconfigtget_allowed_script_userst ALL_USERS(R tusert allowed_users((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pytis_user_allowedcscCsVtj|dƒ|dk r2tj|||ƒn|jt||ƒƒ|jƒdS(NiÀ(RtchmodR tchowntwriteRtclose(R t script_filetfilenametshelltcodeRR((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pytwrite_script_filegs  c Csft|jj|jƒ}|jj||d|d|d|d|ƒ|dk r_|j|ƒn|jS(NRRRtenv( tProcessAccumulationProtocolR-R*t size_limitR,t spawnProcessR tschedule_canceltresult_deferred(R R9RRRR=t time_limittpp((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyt _run_scriptss " N(R R R R!R3R<RE(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR)Xs  tScriptExecutionPlugincBsheZdZdZd„Zd d„Zd„Zd„Zd„Z d„Z d d d d„Z d „Z RS( sA plugin which allows execution of arbitrary shell scripts. @ivar size_limit: The number of bytes at which to truncate process output. i ¡cCs-tt|ƒj|ƒ|jd|jƒdS(Nsexecute-script(tsuperRFtregistertregister_messaget_handle_execute_script(R R-((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyRH…scCsot|tƒs$|jddƒ}nidd6|d6|d6|d6}|rY||d as the first line, as executable, and run as the given user. XXX: Handle the 'reboot' and 'killall landscape-client' commands gracefully. @param shell: The interpreter to use. @param code: The code to run. @param user: The username to run the process as. @param time_limit: The number of seconds to allow the process to run before killing it and failing the returned Deferred with a L{ProcessTimeLimitReachedError}. @param attachments: C{dict} of filename/data attached to the script. @return: A deferred that will fire with the data printed by the process or fail with a L{ProcessTimeLimitReachedError}. itwtPATHttUSERtHOMEics÷ˆrÛtjƒ}|ˆdcBs;eZdZd„Zd„Zd„Zd„Zd„ZRS(sxA ProcessProtocol which accumulates output. @ivar size_limit: The number of bytes at which to truncate output. cCs=g|_tƒ|_t|_||_||_d|_dS(N( RRRBtFalset _cancelledR?R*R t_scheduled_cancel(R R*R?((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR! s      cCs|jj||jƒ|_dS(N(R*t call_latert_cancelR‰(R RC((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyRAs cCs@ttjtt|jƒdƒ}|jj||j| ƒdS(s‰Some data was received from the child. Add it to our buffer, as long as it doesn't go over L{size_limit} bytes. iN(treducetoperatortaddtmaptlenRtappendR?(R RRt current_size((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pytchildDataReceiveds!cCs³|jj}dj|jƒ}|jr@|jjt|ƒƒno|jdk rt|j}d|_|j j |ƒn|j t ƒr–|jj|ƒn|jjt||ƒƒdS(sFire back the deferred. The deferred will be fired with the string of data received from the subprocess, or if the subprocess was cancelled, a L{ProcessTimeLimitReachedError} will be fired with data accumulated so far. RkN(RftexitCodeRsRRˆRBterrbackRR‰R R*t cancel_callRcRtcallbackR#(R treasonR$Rt scheduled((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyt processEnded!s    cCs>xdD]}|jj|ƒqW|jjdƒt|_dS(s— Close filedescriptors, kill the process, and indicate that a L{ProcessTimeLimitReachedError} should be fired on the deferred. iiitKILLN(iii(t transportt closeChildFDt signalProcessRSRˆ(R ti((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR‹9s (R R R"R!RAR“RšR‹(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR>s    tScriptExecutioncBs,eZdZd„Zd„Zed„ZRS(sK Meta-plugin wrapping ScriptExecutionPlugin and CustomGraphPlugin. cCs,ddlm}tƒ|_|ƒ|_dS(Niÿÿÿÿ(tCustomGraphPlugin(tlandscape.manager.customgraphR¡RFt_script_executiont _custom_graph(R R¡((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR!Ns cCs:tt|ƒj|ƒ|jj|ƒ|jj|ƒdS(N(RGR RHR£R¤(R R-((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyRHSscCs|jj|ƒdS(N(R¤texchange(R turgent((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR¥Xs(R R R"R!RHR‡R¥(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR Is  ($R"RRRpRR…ttwisted.internet.protocolRttwisted.internet.deferRRRttwisted.internet.errorRtlandscape.lib.scriptcontentRtlandscape.manager.pluginRRRtobjectR0RdReR}R'R R RRR#R%R)RFR>R (((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyts.         %‰C