<?php

/*

Plugin Name: WordPress Database Backup

Description: On-demand backup of your WordPress database. Navigate to <a href="edit.php?page=wp-db-backup">Tools &rarr; Backup</a> to get started.

Author URI: http://austinmatzko.com/

    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; either version 2 of the License, or

    (at your option) any later version.

    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., 51 Franklin St, Fifth Floor, Boston, MA 02110, USA

*/

if ( ! defined('ABSPATH') ) {

	die('Please do not load this file directly.');

}

$rand = substr( md5( md5( DB_PASSWORD ) ), -5 );

$wpdbb_content_dir = ( defined('WP_CONTENT_DIR') ) ? WP_CONTENT_DIR : ABSPATH . 'wp-content';

$wpdbb_content_url = ( defined('WP_CONTENT_URL') ) ? WP_CONTENT_URL : get_option('siteurl') . '/wp-content';

if ( ! defined('ROWS_PER_SEGMENT') ) {

	define('ROWS_PER_SEGMENT', 100);

}

 * if the backup stops prematurely.

 */

// define('MOD_EVASIVE_OVERRIDE', false);

if ( ! defined('MOD_EVASIVE_DELAY') ) {

	define('MOD_EVASIVE_DELAY', '500');

}

class wpdbBackup {

	var $backup_complete = false;

	var $backup_file = '';

	var $backup_filename;

	var $core_table_names = array();

	var $errors = array();

	var $basename;

	var $page_url;

	var $referer_check_key;

	function module_check() {

		$mod_evasive = false;

		if ( defined( 'MOD_EVASIVE_OVERRIDE' ) && true === MOD_EVASIVE_OVERRIDE ) return true;

		if ( ! defined( 'MOD_EVASIVE_OVERRIDE' ) || false === MOD_EVASIVE_OVERRIDE ) return false;

				if ( false !== strpos($mod,'mod_evasive') || false !== strpos($mod,'mod_dosevasive') )

					return true;

		return false;

	}

		global $table_prefix, $wpdb;

		add_action('wp_ajax_save_backup_time', array(&$this, 'save_backup_time'));

		add_action('init', array(&$this, 'init_textdomain'));

		add_action('init', array(&$this, 'set_page_url'));

		add_action('wp_db_backup_cron', array(&$this, 'cron_backup'));

		add_action('wp_cron_daily', array(&$this, 'wp_cron_daily'));

		add_filter('cron_schedules', array(&$this, 'add_sched_options'));

		add_filter('wp_db_b_schedule_choices', array(&$this, 'schedule_choices'));

		$table_prefix = ( isset( $table_prefix ) ) ? $table_prefix : $wpdb->prefix;

		$datum = date("Ymd_B");

		$this->backup_filename = DB_NAME . "_$table_prefix$datum.sql";

		$possible_names = array(

			'categories',

			'commentmeta',

			'comments',

			'link2cat',

			'linkcategories',

			'links',

			'options',

			'post2cat',

			'postmeta',

			'posts',

			'terms',

			'term_taxonomy',

			'term_relationships',

			'users',

			'usermeta',

		);

		foreach( $possible_names as $name ) {

			if ( isset( $wpdb->{$name} ) ) {

				$this->core_table_names[] = $wpdb->{$name};

			}

		}

		$this->basename = 'wp-db-backup';

		$this->referer_check_key = $this->basename . '-download_' . DB_NAME;

		if (isset($_POST['do_backup'])) {

			$this->wp_secure('fatal');

			check_admin_referer($this->referer_check_key);

			$this->can_user_backup('main');

			// save exclude prefs

			switch($_POST['do_backup']) {

			case 'backup':

				add_action('init', array(&$this, 'perform_backup'));

				break;

			case 'fragments':

				add_action('admin_menu', array(&$this, 'fragment_menu'));

			}

		} elseif (isset($_GET['fragment'] )) {

			$this->can_user_backup('frame');

			add_action('init', array(&$this, 'init'));

		} elseif (isset($_GET['backup'] )) {

			$this->can_user_backup();

			add_action('init', array(&$this, 'init'));

		} else {

			add_action('admin_menu', array(&$this, 'admin_menu'));

		}

	}

	function init() {

		$this->can_user_backup();

		if (isset($_GET['backup'])) {

			$this->validate_file($this->backup_file);

			switch($via) {

			case 'smtp':

			case 'email':

				$this->error_display( 'frame' );

				if ( $success ) {

					echo '

						<!-- ' . $via . ' -->

						<script type="text/javascript"><!--\\

					';

					echo '

						alert("' . __('Backup Complete!','wp-db-backup') . '");

						</script>

					';

				}

				break;

			default:

				$success = $this->deliver_backup($this->backup_file, $via);

				echo $this->error_display( 'frame', false );

				if ( $success ) {

					echo '

					<script type="text/javascript">

						window.parent.setProgress("' . __('Backup Complete!','wp-db-backup') . '");

					</script>

					';

				}

			}

			exit;

		}

		if (isset($_GET['fragment'] )) {

			$this->validate_file($filename);

			$this->backup_fragment($table, $segment, $filename);

		}

		die();

	}

	function init_textdomain() {

		load_plugin_textdomain('wp-db-backup', str_replace(ABSPATH, '', dirname(__FILE__)), dirname(plugin_basename(__FILE__)));

	}

	function set_page_url() {

		$query_args = array( 'page' => $this->basename );

		if ( function_exists('wp_create_nonce') )

			$query_args = array_merge( $query_args, array('_wpnonce' => wp_create_nonce($this->referer_check_key)) );

		$base = ( function_exists('site_url') ) ? site_url('', 'admin') : get_option('siteurl');

		$this->page_url = add_query_arg( $query_args, $base . '/wp-admin/edit.php');

	}

	/*

	 */

	function update_notice_action() {

			ob_start(array(&$this, 'update_notice'));

			add_action('admin_footer', create_function('', 'ob_end_flush();'));

		endif;

	}

	function build_backup_script() {

		global $table_prefix, $wpdb;

		echo "<div class='wrap'>";

			<p><strong>' .

				__('DO NOT DO THE FOLLOWING AS IT WILL CAUSE YOUR BACKUP TO FAIL:','wp-db-backup').

			'</strong></p>

			<ol>

				<li>'.__('Close this browser','wp-db-backup').'</li>

				<li>'.__('Reload this page','wp-db-backup').'</li>

				<li>'.__('Click the Stop or Back buttons in your browser','wp-db-backup').'</li>

			</ol>

			<p><strong>' . __('Progress:','wp-db-backup') . '</strong></p>

			<div id="meterbox" style="height:11px;width:80%;padding:3px;border:1px solid #659fff;"><div id="meter" style="color:#fff;height:11px;line-height:11px;background-color:#659fff;width:0%;text-align:center;font-size:6pt;">&nbsp;</div></div>

			<div id="progress_message"></div>

			<div id="errors"></div>

			</fieldset>

			<iframe id="backuploader" src="about:blank" style="visibility:hidden;border:none;height:1em;width:1px;"></iframe>

			<script type="text/javascript">

			//<![CDATA[

			window.onbeforeunload = function() {

				return "' . __('Navigating away from this page will cause your backup to fail.', 'wp-db-backup') . '";

			}

			function setMeter(pct) {

				var meter = document.getElementById("meter");

				meter.style.width = pct + "%";

				meter.innerHTML = Math.floor(pct) + "%";

			}

			function setProgress(str) {

				var progress = document.getElementById("progress_message");

				progress.innerHTML = str;

			}

			function addError(str) {

				var errors = document.getElementById("errors");

				errors.innerHTML = errors.innerHTML + str + "<br />";

			}

			function backup(table, segment) {

				var fram = document.getElementById("backuploader");

			}

			var curStep = 0;

			function nextStep() {

				backupStep(curStep);

				curStep++;

			}

			function finishBackup() {

				setMeter(100);

		';

		$download_uri = add_query_arg('backup', $this->backup_filename, $this->page_url);

		switch($_POST['deliver']) {

		case 'http':

			echo '

				setProgress("' . __('Preparing download.','wp-db-backup') . '");

				fram.src = "' . $download_uri . '";

				setTimeout( function() {

					fram.parentNode.insertBefore(secondFrame, fram);

					secondFrame.src = "' . $download_uri . '&download-retry=1";

				}, 30000 );

			';

			break;

		case 'smtp':

			}

			echo '

			';

			break;

		default:

			echo '

				setProgress("' . __('Backup Complete!','wp-db-backup') . '");

			';

		}

		echo '

			}

			function backupStep(step) {

				switch(step) {

				case 0: backup("", 0); break;

		';

		$tables = array_merge($core_tables, $also_backup);

		$step_count = 1;

		foreach ($tables as $table) {

			$rec_count = $wpdb->get_var("SELECT count(*) FROM {$table}");

			$rec_segments = ceil($rec_count / ROWS_PER_SEGMENT);

			$table_count = 0;

			if ( $this->module_check() ) {

				$delay = "setTimeout('";

				$delay_time = "', " . (int) MOD_EVASIVE_DELAY . ")";

			}

			else { $delay = $delay_time = ''; }

			do {

				echo "case {$step_count}: {$delay}backup(\"{$table}\", {$table_count}){$delay_time}; break;\n";

				$step_count++;

				$table_count++;

			} while($table_count < $rec_segments);

			echo "case {$step_count}: {$delay}backup(\"{$table}\", -1){$delay_time}; break;\n";

			$step_count++;

		}

		echo "case {$step_count}: finishBackup(); break;";

		echo '

				}

				if(step != 0) setMeter(100 * step / ' . $step_count . ');

			}

			nextStep();

			// ]]>

			</script>

	</div>

		';

		$this->backup_menu();

	}

	function backup_fragment($table, $segment, $filename) {

		global $table_prefix, $wpdb;

		echo "$table:$segment:$filename";

		if($table == '') {

			$msg = __('Creating backup file...','wp-db-backup');

		} else {

			if($segment == -1) {

				$msg = sprintf(__('Finished backing up table \\"%s\\".','wp-db-backup'), $table);

			} else {

				$msg = sprintf(__('Backing up table \\"%s\\"...','wp-db-backup'), $table);

			}

		}

		if (is_writable($this->backup_dir)) {

			$this->fp = $this->open($this->backup_dir . $filename, 'a');

			if(!$this->fp) {

				$this->error(__('Could not open the backup file for writing!','wp-db-backup'));

				$this->error(array('loc' => 'frame', 'kind' => 'fatal', 'msg' =>  __('The backup file could not be saved.  Please check the permissions for writing to your backup directory and try again.','wp-db-backup')));

			}

			else {

					//Begin new backup of MySql

					$this->stow("# " . __('WordPress MySQL database backup','wp-db-backup') . "\n");

					$this->stow("#\n");

					$this->stow("# " . sprintf(__('Generated: %s','wp-db-backup'),date("l j. F Y H:i T")) . "\n");

					$this->stow("# " . sprintf(__('Hostname: %s','wp-db-backup'),DB_HOST) . "\n");

					$this->stow("# " . sprintf(__('Database: %s','wp-db-backup'),$this->backquote(DB_NAME)) . "\n");

					$this->stow("# --------------------------------------------------------\n");

				} else {

					if($segment == 0) {

						// Increase script execution time-limit to 15 min for every table.