123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224 |
- <?php
- // Copyright (C) 2010 Combodo SARL
- //
- // This program is free software; you can redistribute it and/or modify
- // it under the terms of the GNU General Public License as published by
- // the Free Software Foundation; version 3 of the License.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU General Public License for more details.
- //
- // You should have received a copy of the GNU General Public License
- // along with this program; if not, write to the Free Software
- // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- /**
- * Heart beat of the application (process asynchron tasks such as broadcasting email)
- *
- * @author Erwan Taloc <erwan.taloc@combodo.com>
- * @author Romain Quetiez <romain.quetiez@combodo.com>
- * @author Denis Flaven <denis.flaven@combodo.com>
- * @license http://www.opensource.org/licenses/gpl-3.0.html LGPL
- */
- if (!defined('__DIR__')) define('__DIR__', dirname(__FILE__));
- require_once(__DIR__.'/../approot.inc.php');
- require_once(APPROOT.'/application/application.inc.php');
- require_once(APPROOT.'/application/nicewebpage.class.inc.php');
- require_once(APPROOT.'/application/webpage.class.inc.php');
- require_once(APPROOT.'/application/clipage.class.inc.php');
- require_once(APPROOT.'/application/startup.inc.php');
- function ReadMandatoryParam($oP, $sParam)
- {
- $sValue = utils::ReadParam($sParam, null, true /* Allow CLI */);
- if (is_null($sValue))
- {
- $oP->p("ERROR: Missing argument '$sParam'\n");
- UsageAndExit($oP);
- }
- return trim($sValue);
- }
- function UsageAndExit($oP)
- {
- $bModeCLI = utils::IsModeCLI();
- if ($bModeCLI)
- {
- $oP->p("USAGE:\n");
- $oP->p("php -q cron.php --auth_user=<login> --auth_pwd=<password> [--param_file=<file>] [--verbose=1]\n");
- }
- else
- {
- $oP->p("Optional parameters: verbose, param_file\n");
- }
- $oP->output();
- exit -2;
- }
- // Known limitation - the background process periodicity is NOT taken into account
- function CronExec($oP, $aBackgroundProcesses, $bVerbose)
- {
- $iStarted = time();
- $iMaxDuration = MetaModel::GetConfig()->Get('cron_max_execution_time');
- $iTimeLimit = $iStarted + $iMaxDuration;
-
- if ($bVerbose)
- {
- $oP->p("Planned duration = $iMaxDuration seconds");
- }
- $iCronSleep = MetaModel::GetConfig()->Get('cron_sleep');
-
- while (time() < $iTimeLimit)
- {
- foreach ($aBackgroundProcesses as $oBackgroundProcess)
- {
- if ($bVerbose)
- {
- $oP->p("Processing asynchronous task: ".get_class($oBackgroundProcess));
- }
- $sMessage = $oBackgroundProcess->Process($iTimeLimit);
- if ($bVerbose && !empty($sMessage))
- {
- $oP->p("Returned: $sMessage");
- }
- }
- if ($bVerbose)
- {
- $oP->p("Sleeping");
- }
- sleep($iCronSleep);
- }
- if ($bVerbose)
- {
- $oP->p("Reached normal execution time limit (exceeded by ".(time()-$iTimeLimit)."s)");
- }
- }
- ////////////////////////////////////////////////////////////////////////////////
- //
- // Main
- //
- if (utils::IsModeCLI())
- {
- $oP = new CLIPage("iTop - CRON");
- }
- else
- {
- $oP = new WebPage("iTop - CRON");
- }
- try
- {
- utils::UseParamFile();
- }
- catch(Exception $e)
- {
- $oP->p("Error: ".$e->GetMessage());
- $oP->output();
- exit -2;
- }
- if (utils::IsModeCLI())
- {
- // Next steps:
- // specific arguments: 'csvfile'
- //
- $sAuthUser = ReadMandatoryParam($oP, 'auth_user');
- $sAuthPwd = ReadMandatoryParam($oP, 'auth_pwd');
- if (UserRights::CheckCredentials($sAuthUser, $sAuthPwd))
- {
- UserRights::Login($sAuthUser); // Login & set the user's language
- }
- else
- {
- $oP->p("Access wrong credentials ('$sAuthUser')");
- $oP->output();
- exit -1;
- }
- }
- else
- {
- $_SESSION['login_mode'] = 'basic';
- require_once(APPROOT.'/application/loginwebpage.class.inc.php');
- LoginWebPage::DoLogin(); // Check user rights and prompt if needed
- }
- if (!UserRights::IsAdministrator())
- {
- $oP->p("Access restricted to administrators");
- $oP->Output();
- exit -1;
- }
- // Enumerate classes implementing BackgroundProcess
- //
- $aBackgroundProcesses = array();
- foreach(get_declared_classes() as $sPHPClass)
- {
- $oRefClass = new ReflectionClass($sPHPClass);
- $oExtensionInstance = null;
- if ($oRefClass->implementsInterface('iBackgroundProcess'))
- {
- if (is_null($oExtensionInstance))
- {
- $oExecInstance = new $sPHPClass;
- }
- $aBackgroundProcesses[$sPHPClass] = $oExecInstance;
- }
- }
- $bVerbose = utils::ReadParam('verbose', false, true /* Allow CLI */);
- if ($bVerbose)
- {
- $aDisplayProcesses = array();
- foreach ($aBackgroundProcesses as $oExecInstance)
- {
- $aDisplayProcesses[] = get_class($oExecInstance);
- }
- $sDisplayProcesses = implode(', ', $aDisplayProcesses);
- $oP->p("Background processes: ".$sDisplayProcesses);
- }
- $sLockName = 'itop.cron.php';
- $oP->p("Starting: ".time());
- $res = CMDBSource::QueryToScalar("SELECT GET_LOCK('$sLockName', 1)");// timeout = 1 second (see also IS_FREE_LOCK)
- if (is_null($res))
- {
- // TODO - Log ?
- $oP->p("ERROR: Failed to acquire the lock '$sLockName'");
- }
- elseif ($res === '1')
- {
- // The current session holds the lock
- try
- {
- CronExec($oP, $aBackgroundProcesses, $bVerbose);
- }
- catch(Exception $e)
- {
- // TODO - Log ?
- $oP->p("ERROR:".$e->GetMessage());
- }
- $res = CMDBSource::QueryToScalar("SELECT RELEASE_LOCK('$sLockName')");
- }
- else
- {
- // Lock already held by another session
- // Exit silently
- $oP->p("Already running...");
- }
- $oP->p("Exiting: ".time());
- $oP->Output();
- ?>
|