Chris@76: <?php
Chris@76: 
Chris@76: /**
Chris@76:  * Simple Machines Forum (SMF)
Chris@76:  *
Chris@76:  * @package SMF
Chris@76:  * @author Simple Machines http://www.simplemachines.org
Chris@76:  * @copyright 2011 Simple Machines
Chris@76:  * @license http://www.simplemachines.org/about/smf/license.php BSD
Chris@76:  *
Chris@76:  * @version 2.0
Chris@76:  */
Chris@76: 
Chris@76: if (!defined('SMF'))
Chris@76: 	die('Hacking attempt...');
Chris@76: 
Chris@76: /*
Chris@76: 	This file has a single job - database backup.
Chris@76: 
Chris@76: 	void DumpDatabase2()
Chris@76: 		- writes all of the database to standard output.
Chris@76: 		- uses gzip compression if compress is set in the URL/post data.
Chris@76: 		- may possibly time out in some cases.
Chris@76: 		- the data dumped depends on whether "struct" and "data" are passed.
Chris@76: 		- requires an administrator and the session hash by post.
Chris@76: 		- is called from ManageMaintenance.php.
Chris@76: 
Chris@76: */
Chris@76: 
Chris@76: // Dumps the database to a file.
Chris@76: function DumpDatabase2()
Chris@76: {
Chris@76: 	global $db_name, $scripturl, $context, $modSettings, $crlf, $smcFunc, $db_prefix;
Chris@76: 
Chris@76: 	// Administrators only!
Chris@76: 	if (!allowedTo('admin_forum'))
Chris@76: 		fatal_lang_error('no_dump_database', 'critical');
Chris@76: 
Chris@76: 	// You can't dump nothing!
Chris@76: 	if (!isset($_REQUEST['struct']) && !isset($_REQUEST['data']))
Chris@76: 		$_REQUEST['data'] = true;
Chris@76: 
Chris@76: 	checkSession('post');
Chris@76: 
Chris@76: 	// We will need this, badly!
Chris@76: 	db_extend();
Chris@76: 
Chris@76: 	// Attempt to stop from dying...
Chris@76: 	@set_time_limit(600);
Chris@76: 	if (@ini_get('memory_limit') < 256)
Chris@76: 		@ini_set('memory_limit', '256M');
Chris@76: 
Chris@76: 	// Start saving the output... (don't do it otherwise for memory reasons.)
Chris@76: 	if (isset($_REQUEST['compress']) && function_exists('gzencode'))
Chris@76: 	{
Chris@76: 		// Make sure we're gzipping output, but then say we're not in the header ^_^.
Chris@76: 		if (empty($modSettings['enableCompressedOutput']))
Chris@76: 			@ob_start('ob_gzhandler');
Chris@76: 		// Try to clean any data already outputted.
Chris@76: 		elseif (ob_get_length() != 0)
Chris@76: 		{
Chris@76: 			ob_end_clean();
Chris@76: 			@ob_start('ob_gzhandler');
Chris@76: 		}
Chris@76: 
Chris@76: 		// Send faked headers so it will just save the compressed output as a gzip.
Chris@76: 		header('Content-Type: application/x-gzip');
Chris@76: 		header('Accept-Ranges: bytes');
Chris@76: 		header('Content-Encoding: none');
Chris@76: 
Chris@76: 		// Gecko browsers... don't like this. (Mozilla, Firefox, etc.)
Chris@76: 		if (!$context['browser']['is_gecko'])
Chris@76: 			header('Content-Transfer-Encoding: binary');
Chris@76: 
Chris@76: 		// The file extension will include .gz...
Chris@76: 		$extension = '.sql.gz';
Chris@76: 	}
Chris@76: 	else
Chris@76: 	{
Chris@76: 		// Get rid of the gzipping alreading being done.
Chris@76: 		if (!empty($modSettings['enableCompressedOutput']))
Chris@76: 			@ob_end_clean();
Chris@76: 		// If we can, clean anything already sent from the output buffer...
Chris@76: 		elseif (function_exists('ob_clean') && ob_get_length() != 0)
Chris@76: 			ob_clean();
Chris@76: 
Chris@76: 		// Tell the client to save this file, even though it's text.
Chris@76: 		header('Content-Type: ' . ($context['browser']['is_ie'] || $context['browser']['is_opera'] ? 'application/octetstream' : 'application/octet-stream'));
Chris@76: 		header('Content-Encoding: none');
Chris@76: 
Chris@76: 		// This time the extension should just be .sql.
Chris@76: 		$extension = '.sql';
Chris@76: 	}
Chris@76: 
Chris@76: 	// This should turn off the session URL parser.
Chris@76: 	$scripturl = '';
Chris@76: 
Chris@76: 	// If this database is flat file and has a handler function pass it to that.
Chris@76: 	if (!empty($smcFunc['db_get_backup']))
Chris@76: 	{
Chris@76: 		$smcFunc['db_get_backup']();
Chris@76: 		exit;
Chris@76: 	}
Chris@76: 
Chris@76: 	// Send the proper headers to let them download this file.
Chris@76: 	header('Content-Disposition: filename="' . $db_name . '-' . (empty($_REQUEST['struct']) ? 'data' : (empty($_REQUEST['data']) ? 'structure' : 'complete')) . '_' . strftime('%Y-%m-%d') . $extension . '"');
Chris@76: 	header('Cache-Control: private');
Chris@76: 	header('Connection: close');
Chris@76: 
Chris@76: 	// This makes things simpler when using it so very very often.
Chris@76: 	$crlf = "\r\n";
Chris@76: 
Chris@76: 	// SQL Dump Header.
Chris@76: 	echo
Chris@76: 		'-- ==========================================================', $crlf,
Chris@76: 		'--', $crlf,
Chris@76: 		'-- Database dump of tables in `', $db_name, '`', $crlf,
Chris@76: 		'-- ', timeformat(time(), false), $crlf,
Chris@76: 		'--', $crlf,
Chris@76: 		'-- ==========================================================', $crlf,
Chris@76: 		$crlf;
Chris@76: 
Chris@76: 	// Get all tables in the database....
Chris@76: 	if (preg_match('~^`(.+?)`\.(.+?)$~', $db_prefix, $match) != 0)
Chris@76: 	{
Chris@76: 		$db = strtr($match[1], array('`' => ''));
Chris@76: 		$dbp = str_replace('_', '\_', $match[2]);
Chris@76: 	}
Chris@76: 	else
Chris@76: 	{
Chris@76: 		$db = false;
Chris@76: 		$dbp = $db_prefix;
Chris@76: 	}
Chris@76: 
Chris@76: 	// Dump each table.
Chris@76: 	$tables = $smcFunc['db_list_tables'](false, $db_prefix . '%');
Chris@76: 	foreach ($tables as $tableName)
Chris@76: 	{
Chris@76: 		if (function_exists('apache_reset_timeout'))
Chris@76: 			@apache_reset_timeout();
Chris@76: 
Chris@76: 		// Are we dumping the structures?
Chris@76: 		if (isset($_REQUEST['struct']))
Chris@76: 		{
Chris@76: 			echo
Chris@76: 				$crlf,
Chris@76: 				'--', $crlf,
Chris@76: 				'-- Table structure for table `', $tableName, '`', $crlf,
Chris@76: 				'--', $crlf,
Chris@76: 				$crlf,
Chris@76: 				$smcFunc['db_table_sql']($tableName), ';', $crlf;
Chris@76: 		}
Chris@76: 
Chris@76: 		// How about the data?
Chris@76: 		if (!isset($_REQUEST['data']) || substr($tableName, -10) == 'log_errors')
Chris@76: 			continue;
Chris@76: 
Chris@76: 		// Are there any rows in this table?
Chris@76: 		$get_rows = $smcFunc['db_insert_sql']($tableName);
Chris@76: 
Chris@76: 		// No rows to get - skip it.
Chris@76: 		if (empty($get_rows))
Chris@76: 			continue;
Chris@76: 
Chris@76: 		echo
Chris@76: 			$crlf,
Chris@76: 			'--', $crlf,
Chris@76: 			'-- Dumping data in `', $tableName, '`', $crlf,
Chris@76: 			'--', $crlf,
Chris@76: 			$crlf,
Chris@76: 			$get_rows,
Chris@76: 			'-- --------------------------------------------------------', $crlf;
Chris@76: 	}
Chris@76: 
Chris@76: 	echo
Chris@76: 		$crlf,
Chris@76: 		'-- Done', $crlf;
Chris@76: 
Chris@76: 	exit;
Chris@76: }
Chris@76: 
Chris@76: ?>