iTop/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__));
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');

		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');

		$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';
			$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)

		$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());
}

?>