ó ÍÿŠTc@sêdZddlZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZmZm Z e!ƒZ"d Z#dZ$dZ%de&fd„ƒYZ'dd„Z)de&fd„ƒYZ*de&fd„ƒYZ+de&fd„ƒYZ,de!fd„ƒYZ-dee-fd„ƒYZ.defd„ƒYZ/defd „ƒYZ0dS(!su Functionality for running arbitrary shell scripts. @var ALL_USERS: A token indicating all users should be allowed. iÿÿÿÿN(tProcessProtocol(tDeferredtfailtinlineCallbackst returnValuetsucceed(t ProcessDone(tVERSION(t UBUNTU_PATH(t build_script(t fetch_asynct HTTPCodeError(tPersist(tencode_if_needed(t ManagerPlugint SUCCEEDEDtFAILEDifigihtUnknownUserErrorcBseZRS((t__name__t __module__(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR scCs§d}d}d}|dk ršt|ƒ}ytj|ƒ}Wn!tk r`td|ƒ‚nX|j}|j}|j}t j j |ƒsšd}qšn|||fS(NuUnknown user '%s't/( tNoneR tpwdtgetpwnamtKeyErrorRtpw_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(RRt__doc__R((((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR%6stProcessFailedErrorcBseZdZd„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(R&t exit_code(R'R&R+((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR(Is (RRR)R((((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR*BstUnknownInterpreterErrorcBs 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(Ts cCs d|jS(NsUnknown interpreter: '%s'(R-(R'((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR/Xs(RRR)R(R/(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR,Ns tScriptRunnerMixincBs5eZdZdd„Zd„Zd„Zd„ZRS(sb @param process_factory: The L{IReactorProcess} provider to run the process with. cCs,|dkrddlm}n||_dS(Niÿÿÿÿ(treactor(Rttwisted.internetR1tprocess_factory(R'R3((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR(bs 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_allowedgscCsVtj|dƒ|dk r2tj|||ƒn|jt||ƒƒ|jƒdS(NiÀ(RtchmodRtchowntwriteR tclose(R't script_filetfilenametshelltcodeR R!((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pytwrite_script_fileks  c Csœ|tjƒkrd}n|tjƒkr6d}nt|jj|jƒ}|jj ||d|d|d|d|ƒ|dk r•|j |ƒn|j S(NR R!Rtenv( RtgetuidRtgetgidtProcessAccumulationProtocolR4R1t size_limitR3t spawnProcesstschedule_canceltresult_deferred(R'R@R R!RRDt time_limittpp((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyt _run_scriptws   " N(RRR)RR(R:RCRN(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR0\s    tScriptExecutionPlugincBszeZdZdZd„Zd d„Zd„Zd„Zd„Z d„Z e d„ƒZ d d d d d „Z d „ZRS( 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(tsuperROtregistertregister_messaget_handle_execute_script(R'R4((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyRQscCsut|tƒs$|jddƒ}nidd6|d6|d6|d6}|rY||dR( R'RbR R!t computer_idt root_pathtattachment_dirRwR@t attachment_idR&t full_filenamet attachment((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyt_save_attachmentsÏs0#      c s†tjj|jƒdƒs,tt|ƒƒSt|ƒ\‰‰‰tjƒ\}‰tj |dƒ}ˆj |ˆ||ˆˆƒit d6|p”dd6ˆp¡dd6‰|r¾ˆj |ƒntj dƒ} |r3tdtjjˆjjjd ƒƒ} | jd ƒ} | jd ƒ} ˆj|ˆˆ| ƒ} n td ƒ} ‡‡‡‡‡‡‡fd †} | j| ƒ| jˆjˆˆ| ƒS(s8 Run a script based on a shell and the code. A file will be written with #! 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}. itwtPATHttUSERtHOMEiR@sbroker.bpicklet registrations secure-idcs5|dk r|ˆdxdD]}|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 signalProcessR]R¡(R'ti((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR¤hs (RRR)R(RJR¬R³R¤(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyRG5s    tScriptExecutioncBs,eZdZd„Zd„Zed„ZRS(sK Meta-plugin wrapping ScriptExecutionPlugin and CustomGraphPlugin. cCs,ddlm}tƒ|_|ƒ|_dS(Niÿÿÿÿ(tCustomGraphPlugin(tlandscape.manager.customgraphRºROt_script_executiont _custom_graph(R'Rº((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR(}s cCs:tt|ƒj|ƒ|jj|ƒ|jj|ƒdS(N(RPR¹RQR¼R½(R'R4((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyRQ‚scCs|jj|ƒdS(N(R½texchange(R'turgent((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR¾‡s(RRR)R(RQR R¾(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyR¹xs  (1R)RRR{R¦Rttwisted.internet.protocolRttwisted.internet.deferRRRRRttwisted.internet.errorRt landscapeRtlandscape.constantsRtlandscape.lib.scriptcontentR tlandscape.lib.fetchR R tlandscape.lib.persistR tlandscape.lib.encodingR tlandscape.manager.pluginRRRtobjectR7RpRqRrR.RRR$R%R*R,R0RORGR¹(((sE/usr/lib/python2.7/dist-packages/landscape/manager/scriptexecution.pyts8     (    +®C