-'Nc@sVdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZmZddlmZddlmZddlmZmZdd lmZdd lmZmZdd lmZdd lmZdd l m!Z!m"Z"m#Z#ddl$m%Z%ddl&m'Z'm(Z(m)Z)ddl*m+Z+dZ,dZ-dZ.dZ/de0fdYZ1de0fdYZ2de0fdYZ3de4fdYZ5de5fdYZ6de5fdYZ7d e5fd!YZ8d"efd#YZ9d$e4fd%YZ:d&efd'YZ;d(Z<d)efd*YZ=e!e#d+d,d-d.e#d/d,d-d.e#d0d,d-d.e#d1d,d-d.e#d2d,d-d.e#d3d,d-d.e#d4d,d-d5e#d6d,d-d.e#d7d,d-d.e"d8d,d-d9g Z>d:Z?ej@dd;ZBdS(<sSee L{WatchDog}. The WatchDog must run as root, because it spawns the Landscape Manager. The main C{landscape-client} program uses this watchdog. iN(twarningtinfoterror(treactor(tDeferredtsucceed(tProcessProtocol(tProcessExitedAlready(tServicet Application(tstartApplication(t Configurationt init_logging(tgather_results(t log_failure(t BootstrapListt BootstrapFiletBootstrapDirectory(t rotate_logs(tRemoteBrokerConnectortRemoteMonitorConnectortRemoteManagerConnector(tTwistedReactori iit DaemonErrorcBseZdZRS(s(One of the daemons could not be started.(t__name__t __module__t__doc__(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR(st TimeoutErrorcBseZdZRS(sSomething took too long.(RRR(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR,stExecutableNotFoundErrorcBseZdZRS(sAn executable was not found.(RRR(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR0stDaemoncBseZdZdZdZdZeeddZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdZRS(sfA Landscape daemon which can be started and tracked. This class should be subclassed to specify individual daemon. @cvar program: The name of the executable program that will start this daemon. @cvar username: The name of the user to switch to, by default. @cvar service: The DBus service name that the program will be expected to listen on. @cvar max_retries: The maximum number of retries before giving up when trying to connect to the watched daemon. @cvar factor: The factor by which the delay between subsequent connection attempts will increase. t landscapeig?cCs||_||_tjj|_tjdkrtj|j }|j |_ |j |_ |j|jd<|j |jd<|j |jdscSstS(N(tFalse(tx((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR_s(R"tconnectt max_retriestfactorR7t addCallbackt addErrback(R9R^RItkwargsRYt connected((R^R9s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt_connect_and_calls   cCs |jdS(Ntexit(Ri(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt request_exitscCs |jdS(Ntping(Ri(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt is_runningscCs |jstdS|jjS(sN Return a Deferred which will fire when the process has died. N(R4RR1twait(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRns  cCs |jstdS|jjS(s Wait for the process to die for C{GRACEFUL_WAIT_PERIOD}. If it hasn't died by that point, send it a SIGTERM. If it doesn't die for C{SIGKILL_DELAY}, N(R4RR1t wait_or_die(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRos  cCs t|_dS(sCalled by the watchdog when starting to shut us down. It will prevent our L{WatchedProcessProtocol} to restart the process when it exits. N(R`R8(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pytprepare_for_shutdownscCs|jS(s>Return a boolean indicating if the daemon should be restarted.(R8(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt allow_restartscCs|jjdS(N(R4R(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRsN(RRRR+RcRdRR`R1R>RHRWRPRiRkRmRnRoRpRqR(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR4s"         tBrokercBseZdZRS(slandscape-broker(RRRE(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRrstMonitorcBseZdZRS(slandscape-monitor(RRRE(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRsstManagercBseZdZdZRS(slandscape-managertroot(RRRER+(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRtsRQcBsqeZdZeZdZdZedZdZdZ dZ dZ dZ d Z d ZRS( su A process-watching protocol which sends any of its output to the log file and restarts it when it dies. cCs(||_d|_d|_d|_dS(N(tdaemonR1t _wait_resultt_delayed_really_killt_delayed_terminate(R9Rv((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR>s   cCs|j|jS(N(t _terminateRn(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRXs cCs|jdk r{|r/td|jjfny|jjtjWntk rYq{Xt j t |j }||_ ndS(Ns%s didn't exit. Sending SIGTERM(t transportR1RRvREt signalProcesstsignaltSIGTERMRRt callLatert SIGKILL_DELAYt _really_killRx(R9twarntdelayed((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRzs cCsLy|jjtjWntk r*nXtd|jjd|_ dS(Ns %s didn't die. Sending SIGKILL.( R{R|R}tSIGKILLRRRvRER1Rx(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRs  cCsA|jdk r=y|jjtjWq=tk r9q=XndS(N(R{R1R|R}tSIGUSR1R(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR s  cCs/|jjdkrtdSt|_|jS(N(R{tpidR1RRRw(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRns  cCs(tjt|jdt|_|jS(NR(RRtGRACEFUL_WAIT_PERIODRzR7RyRn(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRos cCstjj|dS(N(RBtstdouttwrite(R9tdata((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt outReceivedscCstjj|dS(N(RBtstderrR(R9R((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt errReceived scCs|jdk r|jjn|jdk rM|jjrM|jjn|jdk ro|jjdn|jjr|jj ndS(s"The process has ended; restart it.N( RxR1tcancelRytactiveRwtcallbackRvRqRW(R9treason((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt processEnded#s(RRRR`t_killedR>RXRzRRRnRoRRR(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRQs        tWatchDogcBskeZdZeed d d d d dZdZdZdZ dZ dZ dZ dZ RS( sj The Landscape WatchDog starts all other landscape daemons and ensures that they are working. c Csat}|dkr'tttg}n|dkrft|krftt||d|d|j}n|dkrt|krtt||d|d|j}n|dkrt|krtt||d|d|j}n||_ ||_ ||_ t d|j |j |j g|_ ||_d|_t|_tjtj|ji|_dS(NR;R<(RR1RrRsRtRR<RRtbrokertmonitortmanagertfiltertdaemonsRt _checkingR`t _stoppingR}Rt_notify_rotate_logst_ping_failures( R9RR;R<RRRtenabled_daemonsttwisted_reactor((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR>6s0     $   cCsbg}x=|jD]2}|j}|j|d|j|qWd}t|j|S(s6Return a list of any daemons that are already running.cSs ||fS(N((Rmtd((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR_\scSs%g|D]}|dr|d^qS(Nii((trRa((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pytgot_all_results_s(RRmReRRR (R9tresultsRvtresultR((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt check_runningTs  cCs,x|jD]}|jq W|jdS(sm Start all daemons. The broker will be started first, and no other daemons will be started before it is running and responding to DBUS messages. @return: A deferred which fires when all services have successfully started. If a daemon could not be started, the deferred will fail with L{DaemonError}. N(RRWtstart_monitoring(R9Rv((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRWcs cCs|jjd|j|_dS(s;Start monitoring processes which have already been started.iN(RRt_checkR(R9((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRqscs| rj rtdjfjkrDdjRRWRRRRkR(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR0s        tWatchDogConfigurationcBseZdZdZRS(cCsdtt|j}|jddddd|jddddd |jd dddd |S( Ns--daemontactiont store_truethelpsFork and run in the background.s --pid-filettypetstrsThe file to write the PID to.s--monitor-onlysbDon't enable management features. This is useful if you want to run the client as a non-root user.(tsuperRt make_parsert add_option(R9tparser((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRscCs)ttg}|js%|jtn|S(N(RrRst monitor_onlyRRRt(R9R((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pytget_enabled_daemonss  (RRRR(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRs cCstjrtjdntjtjrBtjdntjdtjdtj}xVtdD]H}ytj||Wqqt k r}|j t j krqqqXqqWtj |dS(Nii?s /dev/nulli( R$tforkt_exittsetsidtumasktopentO_RDWRtrangetdup2tOSErrorterrnotEBADFtclose(tnulltite((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyt daemonizes    tWatchDogServicecBs5eZdZdZdZdZdZRS(cCs>||_td|j d|d|j|_d|_dS(NR;R<Ri(R3RRvRtwatchdogt exit_code(R9R<((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR>s  csztjtjdjjdjjjj}fd}fd}|j ||j ||S(Nt data_pathtlog_dircs_|r>tddjd|Dd_tjdSjtdjjS(Ns4ERROR: The following daemons are already running: %ss, css|]}|jVqdS(N(RE(t.0Ra((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pys sisWatchdog watching for daemons.( RRDRRtcrasht _daemonizeRRRW(trunning_daemons(R9(s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pytstart_if_not_runnings    cs$t|dd_tjdS(NsUnknown error occurred!i(RRRR(tfailure(R9(s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pytdies  ( Rt startServicetbootstrap_listt bootstrapR3RRRRReRf(R9RRR((R9s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRs     cCsa|jjr]t|jjr]t|jjd}|jttj|j q]ndS(Ntw( R3RvRtpid_fileRRRR$tgetpidR(R9tstream((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRs   csVtdtjtjtjtjjj}|jfd|S(NsStopping client...cs jS(N(t _remove_pid(R(R9(s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR_s( RRt stopServiceR}tSIGINTtSIG_IGNRRkR(R9tdone((R9s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRs   cCs~|jj}|dk rztj|tjrzt|}|j}|j|t tj krztj |qzndS(N( R3RR1R$taccesstW_OKRtreadRRRtunlink(R9RRR((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRs !   (RRR>RRRR(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyRs     s $data_pathRRuis$data_path/packages$data_path/package/hash-ids$data_path/package/binariess$data_path/package/upgrade-tools$data_path/messagess$data_path/socketsis$data_path/custom-graph-scriptss$log_dirs$data_path/package/databaseicCsUxNtjjD]=}|jds@|jds@|dkrtj|=qqWdS(sTUnset dangerous environment variables. In particular unset all variables beginning with DEBIAN_ or DEBCONF_, to avoid any problems when landscape-client is invoked from its postinst script. Some environment variables may be set which would affect *other* maintainer scripts which landscape-client invokes (via smart). tDEBIAN_tDEBCONF_tLANDSCAPE_ATTACHMENTStMAILN(RR(R$R%tkeyst startswith(tkey((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pytclean_environment.s  cCstt}|j|ytjdj}Wntk rStjdnXt j d|fkr|tjdnt |dt d}t |}|j||d krddlm}n|jdt|t|j|jS( sStart the watchdog. This is the topmost function that kicks off the Landscape client. It cleans up the environment, loads the configuration, and starts the reactor. @param args: Command line arguments, including the program name as the first element. @param reactor: The reactor to use. If none is specified, the global reactor is used. @raise SystemExit: if command line arguments are bad, or when landscape- client is not running as 'root' or 'landscape'. Rs#The 'landscape' user doesn't exist!is:landscape-client can only be run as 'root' or 'landscape'.Rslandscape-clienti(RN(RRtloadR)R*R,tKeyErrorRBRjR$R(R R RtsetServiceParentR1ttwisted.internetRRR R`trunR(RIRR<t landscape_uidt applicationtwatchdog_service((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pyR=s$         (CRR$RRBR)R}RMtloggingRRRRRttwisted.internet.deferRRttwisted.internet.protocolRttwisted.internet.errorRttwisted.application.serviceRR ttwisted.application.appR tlandscape.deploymentR R tlandscape.lib.twisted_utilR tlandscape.lib.logRtlandscape.lib.bootstrapRRRt landscape.logRtlandscape.broker.ampRRRtlandscape.reactorRRRORNRt ExceptionRRRtobjectRRrRsRtRQRRRRRRRCR1R(((s6/usr/lib/python2.7/dist-packages/landscape/watchdog.pytsj      O @