ITSSTOOLS/datamodels/2.x/itop-backup/ajax.backup.php

<?php
// Copyright (C) 2014 Combodo SARL
//
//   This file is part of iTop.
//
//   iTop is free software; you can redistribute it and/or modify	
//   it under the terms of the GNU Affero General Public License as published by
//   the Free Software Foundation, either version 3 of the License, or
//   (at your option) any later version.
//
//   iTop 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 Affero General Public License for more details.
//
//   You should have received a copy of the GNU Affero General Public License
//   along with iTop. If not, see <http://www.gnu.org/licenses/>

/**
 * Backup from an interactive session
 *
 * @copyright   Copyright (C) 2013 Combodo SARL
 * @license     http://opensource.org/licenses/AGPL-3.0
 */

if (!defined('__DIR__')) define('__DIR__', dirname(__FILE__));
if (!defined('APPROOT')) require_once(__DIR__.'/../../approot.inc.php');
require_once(APPROOT.'/application/application.inc.php');
require_once(APPROOT.'/application/webpage.class.inc.php');
require_once(APPROOT.'/application/ajaxwebpage.class.inc.php');

require_once(APPROOT.'core/mutex.class.inc.php');

try
{
	$sOperation = utils::ReadParam('operation', '');

	switch($sOperation)
	{
		case 'backup':
		require_once(APPROOT.'/application/startup.inc.php');
		require_once(APPROOT.'/application/loginwebpage.class.inc.php');
		LoginWebPage::DoLogin(true); // Check user rights and prompt if needed (must be admin)

		$oPage = new ajax_page("");
		$oPage->no_cache();
		$oPage->SetContentType('text/html');

		if (utils::GetConfig()->Get('demo_mode'))
		{
			$oPage->add("<div data-error-stimulus=\"Error\">Sorry, iTop is in <b>demonstration mode</b>: the feature is disabled.</div>");
		}
		else
		{
			try
			{
				set_time_limit(0);
				$oBB = new BackupExec(APPROOT.'data/backups/manual/', 0 /*iRetentionCount*/);
				$sRes = $oBB->Process(time() + 36000); // 10 hours to complete should be sufficient!
			}
			catch (Exception $e)
			{
				$oPage->p('Error: '.$e->getMessage());
			}
		}
		$oPage->output();
		break;

		case 'restore_get_token':
		require_once(APPROOT.'/application/startup.inc.php');
		require_once(APPROOT.'/application/loginwebpage.class.inc.php');
		LoginWebPage::DoLogin(true); // Check user rights and prompt if needed (must be admin)

		$oPage = new ajax_page("");
		$oPage->no_cache();
		$oPage->SetContentType('text/html');

		$sEnvironment = utils::ReadParam('environment', 'production', false, 'raw_data');
		$oRestoreMutex = new iTopMutex('restore.'.$sEnvironment);
		if ($oRestoreMutex->TryLock())
		{
			$oRestoreMutex->Unlock();
			$sFile = utils::ReadParam('file', '', false, 'raw_data');
			$sToken = str_replace(' ', '', (string)microtime());
			$sTokenFile = APPROOT.'/data/restore.'.$sToken.'.tok';
			file_put_contents($sTokenFile, $sFile);
	
			$oPage->add_ready_script(
<<<EOF
	$("#restore_token").val('$sToken');
EOF
			);
		}
		else
		{
			$oPage->p(Dict::S('bkp-restore-running'));
		}
		$oPage->output();
		break;


		case 'restore_exec':
		require_once(APPROOT."setup/runtimeenv.class.inc.php");
		require_once(APPROOT.'/application/utils.inc.php');
		require_once(APPROOT.'/setup/backup.class.inc.php');
		require_once(dirname(__FILE__).'/dbrestore.class.inc.php');

		IssueLog::Enable(APPROOT.'log/error.log');

		$oPage = new ajax_page("");
		$oPage->no_cache();
		$oPage->SetContentType('text/html');

		if (utils::GetConfig()->Get('demo_mode'))
		{
			$oPage->add("<div data-error-stimulus=\"Error\">Sorry, iTop is in <b>demonstration mode</b>: the feature is disabled.</div>");
		}
		else
		{
			$sEnvironment = utils::ReadParam('environment', 'production', false, 'raw_data');
			$oRestoreMutex = new iTopMutex('restore.'.$sEnvironment);
			$oRestoreMutex->Lock();
			try
			{
				set_time_limit(0);
	
				// Get the file and destroy the token (single usage)
				$sToken = utils::ReadParam('token', '', false, 'raw_data');
				$sTokenFile = APPROOT.'/data/restore.'.$sToken.'.tok';
				if (!is_file($sTokenFile))
				{
					throw new Exception("Error: missing token file: '$sTokenFile'");
				}
				$sFile = file_get_contents($sTokenFile);
				unlink($sTokenFile);
	
				$sMySQLBinDir = utils::ReadParam('mysql_bindir', '', false, 'raw_data');
				$sDBHost = utils::ReadParam('db_host', '', false, 'raw_data');
				$sDBUser = utils::ReadParam('db_user', '', false, 'raw_data');
				$sDBPwd = utils::ReadParam('db_pwd', '', false, 'raw_data');
				$sDBName = utils::ReadParam('db_name', '', false, 'raw_data');
				$sDBSubName = utils::ReadParam('db_subname', '', false, 'raw_data');
	
				$oDBRS = new DBRestore($sDBHost, $sDBUser, $sDBPwd, $sDBName, $sDBSubName);
				$oDBRS->SetMySQLBinDir($sMySQLBinDir);
	
				$sBackupDir = APPROOT.'data/backups/';
				$sBackupFile = $sBackupDir.$sFile;
				$sRes = $oDBRS->RestoreFromZip($sBackupFile, $sEnvironment);
	
				$oRestoreMutex->Unlock();
			}
			catch (Exception $e)
			{
				$oRestoreMutex->Unlock();
				$oPage->p('Error: '.$e->getMessage());
			}
		}
		$oPage->output();
		break;

		case 'download':
		require_once(APPROOT.'/application/startup.inc.php');
		require_once(APPROOT.'/application/loginwebpage.class.inc.php');
		LoginWebPage::DoLogin(true); // Check user rights and prompt if needed (must be admin)

		if (utils::GetConfig()->Get('demo_mode'))
		{
			throw new Exception('iTop is in demonstration mode: the feature is disabled');
		}
		$sFile = utils::ReadParam('file', '', false, 'raw_data');
		$oBackup = new DBBackupScheduled();
		$sBackupDir = APPROOT.'data/backups/';
		$oBackup->DownloadBackup($sBackupDir.$sFile);
		break;
	}
}
catch (Exception $e)
{
	IssueLog::Error($e->getMessage());
}

?>