Edit File by line

<?php

[0] Fix | Delete

/**

[1] Fix | Delete

* Define database parameters here

[2] Fix | Delete

[3] Fix | Delete

$upload_dir = wp_upload_dir();

[4] Fix | Delete

$backup_dirname = $upload_dir['basedir'].'/wp-file-manager-pro/fm_backup';

[5] Fix | Delete

define("BACKUP_DIR", $backup_dirname);

[6] Fix | Delete

define("CHARSET", 'utf8');

[7] Fix | Delete

define("DISABLE_FOREIGN_KEY_CHECKS", true);

[8] Fix | Delete

[9] Fix | Delete

/**

[10] Fix | Delete

* The Restore_Database class

[11] Fix | Delete

[12] Fix | Delete

class Restore_Database {

[13] Fix | Delete

/**

[14] Fix | Delete

* Host where the database is located

[15] Fix | Delete

[16] Fix | Delete

var $host;

[17] Fix | Delete

[18] Fix | Delete

/**

[19] Fix | Delete

* Username used to connect to database

[20] Fix | Delete

[21] Fix | Delete

var $username;

[22] Fix | Delete

[23] Fix | Delete

/**

[24] Fix | Delete

* Password used to connect to database

[25] Fix | Delete

[26] Fix | Delete

var $passwd;

[27] Fix | Delete

[28] Fix | Delete

/**

[29] Fix | Delete

* Database to backup

[30] Fix | Delete

[31] Fix | Delete

var $dbName;

[32] Fix | Delete

[33] Fix | Delete

/**

[34] Fix | Delete

* Database charset

[35] Fix | Delete

[36] Fix | Delete

var $charset;

[37] Fix | Delete

[38] Fix | Delete

/**

[39] Fix | Delete

* Database connection

[40] Fix | Delete

[41] Fix | Delete

var $conn;

[42] Fix | Delete

[43] Fix | Delete

/**

[44] Fix | Delete

* Disable foreign key checks

[45] Fix | Delete

[46] Fix | Delete

var $disableForeignKeyChecks;

[47] Fix | Delete

[48] Fix | Delete

/**

[49] Fix | Delete

* Constructor initializes database

[50] Fix | Delete

[51] Fix | Delete

function __construct($filename) {

[52] Fix | Delete

$this->host = DB_HOST;

[53] Fix | Delete

$this->username = DB_USER;

[54] Fix | Delete

$this->passwd = DB_PASSWORD;

[55] Fix | Delete

$this->dbName = DB_NAME;

[56] Fix | Delete

$this->charset = DB_CHARSET;

[57] Fix | Delete

$this->disableForeignKeyChecks = defined('DISABLE_FOREIGN_KEY_CHECKS') ? DISABLE_FOREIGN_KEY_CHECKS : true;

[58] Fix | Delete

$this->conn = $this->initializeDatabase();

[59] Fix | Delete

$this->backupDir = defined('BACKUP_DIR') ? BACKUP_DIR : '.';

[60] Fix | Delete

$this->backupFile = $filename;

[61] Fix | Delete

}

[62] Fix | Delete

[63] Fix | Delete

/**

[64] Fix | Delete

* Destructor re-enables foreign key checks

[65] Fix | Delete

[66] Fix | Delete

function __destructor() {

[67] Fix | Delete

/**

[68] Fix | Delete

* Re-enable foreign key checks

[69] Fix | Delete

[70] Fix | Delete

if ($this->disableForeignKeyChecks === true) {

[71] Fix | Delete

mysqli_query($this->conn, 'SET foreign_key_checks = 1');

[72] Fix | Delete

}

[73] Fix | Delete

}

[74] Fix | Delete

[75] Fix | Delete

protected function initializeDatabase() {

[76] Fix | Delete

try {

[77] Fix | Delete

$conn = mysqli_connect($this->host, $this->username, $this->passwd, $this->dbName);

[78] Fix | Delete

if (mysqli_connect_errno()) {

[79] Fix | Delete

throw new Exception('ERROR connecting database: ' . mysqli_connect_error());

[80] Fix | Delete

die();

[81] Fix | Delete

}

[82] Fix | Delete

if (!mysqli_set_charset($conn, $this->charset)) {

[83] Fix | Delete

mysqli_query($conn, 'SET NAMES '.$this->charset);

[84] Fix | Delete

}

[85] Fix | Delete

[86] Fix | Delete

/**

[87] Fix | Delete

* Disable foreign key checks

[88] Fix | Delete

[89] Fix | Delete

if ($this->disableForeignKeyChecks === true) {

[90] Fix | Delete

mysqli_query($conn, 'SET foreign_key_checks = 0');

[91] Fix | Delete

}

[92] Fix | Delete

[93] Fix | Delete

} catch (Exception $e) {

[94] Fix | Delete

print_r($e->getMessage());

[95] Fix | Delete

die();

[96] Fix | Delete

}

[97] Fix | Delete

[98] Fix | Delete

return $conn;

[99] Fix | Delete

}

[100] Fix | Delete

[101] Fix | Delete

/**

[102] Fix | Delete

* Backup the whole database or just some tables

[103] Fix | Delete

* Use '*' for whole database or 'table1 table2 table3...'

[104] Fix | Delete

* @param string $tables

[105] Fix | Delete

[106] Fix | Delete

public function restoreDb() {

[107] Fix | Delete

try {

[108] Fix | Delete

$sql = '';

[109] Fix | Delete

$multiLineComment = false;

[110] Fix | Delete

[111] Fix | Delete

$backupDir = $this->backupDir;

[112] Fix | Delete

$backupFile = $this->backupFile;

[113] Fix | Delete

[114] Fix | Delete

/**

[115] Fix | Delete

* Gunzip file if gzipped

[116] Fix | Delete

[117] Fix | Delete

$backupFileIsGzipped = substr($backupFile, -3, 3) == '.gz' ? true : false;

[118] Fix | Delete

[119] Fix | Delete

if ($backupFileIsGzipped) {

[120] Fix | Delete

if (!$backupFile = $this->gunzipBackupFile()) {

[121] Fix | Delete

throw new Exception("ERROR: couldn't gunzip backup file " . $backupDir . '/' . $backupFile);

[122] Fix | Delete

}

[123] Fix | Delete

}

[124] Fix | Delete

[125] Fix | Delete

/**

[126] Fix | Delete

* Read backup file line by line

[127] Fix | Delete

[128] Fix | Delete

$handle = fopen($backupDir . '/' . $backupFile, "r");

[129] Fix | Delete

if ($handle) {

[130] Fix | Delete

while (($line = fgets($handle)) !== false) {

[131] Fix | Delete

$line = ltrim(rtrim($line));

[132] Fix | Delete

if (strlen($line) > 1) { // avoid blank lines

[133] Fix | Delete

$lineIsComment = false;

[134] Fix | Delete

if (preg_match('/^\/\*/', $line)) {

[135] Fix | Delete

$multiLineComment = true;

[136] Fix | Delete

$lineIsComment = true;

[137] Fix | Delete

}

[138] Fix | Delete

if ($multiLineComment or preg_match('/^\/\//', $line)) {

[139] Fix | Delete

$lineIsComment = true;

[140] Fix | Delete

}

[141] Fix | Delete

if (!$lineIsComment) {

[142] Fix | Delete

$sql .= $line;

[143] Fix | Delete

if (preg_match('/;$/', $line)) {

[144] Fix | Delete

[145] Fix | Delete

mysqli_query($this->conn, "SET sql_mode = ''");

[146] Fix | Delete

// execute query

[147] Fix | Delete

if(mysqli_query($this->conn, $sql)) {

[148] Fix | Delete

if (preg_match('/^CREATE TABLE `([^`]+)`/i', $sql, $tableName)) {

[149] Fix | Delete

$this->obfPrint("Table succesfully created: `" . $tableName[1] . "`");

[150] Fix | Delete

}

[151] Fix | Delete

$sql = '';

[152] Fix | Delete

} else {

[153] Fix | Delete

throw new Exception("ERROR: SQL execution error: " . mysqli_error($this->conn));

[154] Fix | Delete

}

[155] Fix | Delete

}

[156] Fix | Delete

} else if (preg_match('/\*\/$/', $line)) {

[157] Fix | Delete

$multiLineComment = false;

[158] Fix | Delete

}

[159] Fix | Delete

}

[160] Fix | Delete

}

[161] Fix | Delete

fclose($handle);

[162] Fix | Delete

} else {

[163] Fix | Delete

throw new Exception("ERROR: couldn't open backup file " . $backupDir . '/' . $backupFile);

[164] Fix | Delete

}

[165] Fix | Delete

} catch (Exception $e) {

[166] Fix | Delete

print_r($e->getMessage());

[167] Fix | Delete

return false;

[168] Fix | Delete

}

[169] Fix | Delete

[170] Fix | Delete

if ($backupFileIsGzipped) {

[171] Fix | Delete

unlink($backupDir . '/' . $backupFile);

[172] Fix | Delete

}

[173] Fix | Delete

return true;

[174] Fix | Delete

}

[175] Fix | Delete

[176] Fix | Delete

[177] Fix | Delete

* Gunzip backup file

[178] Fix | Delete

[179] Fix | Delete

* @return string New filename (without .gz appended and without backup directory) if success, or false if operation fails

[180] Fix | Delete

[181] Fix | Delete

protected function gunzipBackupFile() {

[182] Fix | Delete

// Raising this value may increase performance

[183] Fix | Delete

$bufferSize = 4096; // read 4kb at a time

[184] Fix | Delete

$error = false;

[185] Fix | Delete

[186] Fix | Delete

$source = $this->backupDir . '/' . $this->backupFile;

[187] Fix | Delete

$dest = $this->backupDir . '/' . date("Ymd_His", time()) . '_' . substr($this->backupFile, 0, -3);

[188] Fix | Delete

[189] Fix | Delete

$this->obfPrint('Gunzipping backup file ' . $source . '... ', 1, 1);

[190] Fix | Delete

[191] Fix | Delete

// Remove $dest file if exists

[192] Fix | Delete

if (file_exists($dest)) {

[193] Fix | Delete

if (!unlink($dest)) {

[194] Fix | Delete

return false;

[195] Fix | Delete

}

[196] Fix | Delete

}

[197] Fix | Delete

[198] Fix | Delete

// Open gzipped and destination files in binary mode

[199] Fix | Delete

if (!$srcFile = gzopen($this->backupDir . '/' . $this->backupFile, 'rb')) {

[200] Fix | Delete

return false;

[201] Fix | Delete

}

[202] Fix | Delete

if (!$dstFile = fopen($dest, 'wb')) {

[203] Fix | Delete

return false;

[204] Fix | Delete

}

[205] Fix | Delete

[206] Fix | Delete

while (!gzeof($srcFile)) {

[207] Fix | Delete

// Read buffer-size bytes

[208] Fix | Delete

// Both fwrite and gzread are binary-safe

[209] Fix | Delete

if(!fwrite($dstFile, gzread($srcFile, $bufferSize))) {

[210] Fix | Delete

return false;

[211] Fix | Delete

}

[212] Fix | Delete

}

[213] Fix | Delete

[214] Fix | Delete

fclose($dstFile);

[215] Fix | Delete

gzclose($srcFile);

[216] Fix | Delete

[217] Fix | Delete

// Return backup filename excluding backup directory

[218] Fix | Delete

return str_replace($this->backupDir . '/', '', $dest);

[219] Fix | Delete

}

[220] Fix | Delete

[221] Fix | Delete

/**

[222] Fix | Delete

* Prints message forcing output buffer flush

[223] Fix | Delete

[224] Fix | Delete

[225] Fix | Delete

public function obfPrint ($msg = '', $lineBreaksBefore = 0, $lineBreaksAfter = 1) {

[226] Fix | Delete

if (!$msg) {

[227] Fix | Delete

return false;

[228] Fix | Delete

}

[229] Fix | Delete

[230] Fix | Delete

$msg = date("Y-m-d H:i:s") . ' - ' . $msg;

[231] Fix | Delete

$output = '';

[232] Fix | Delete

[233] Fix | Delete

if (php_sapi_name() != "cli") {

[234] Fix | Delete

$lineBreak = "<br />";

[235] Fix | Delete

} else {

[236] Fix | Delete

$lineBreak = "\n";

[237] Fix | Delete

}

[238] Fix | Delete

[239] Fix | Delete

if ($lineBreaksBefore > 0) {

[240] Fix | Delete

for ($i = 1; $i <= $lineBreaksBefore; $i++) {

[241] Fix | Delete

$output .= $lineBreak;

[242] Fix | Delete

}

[243] Fix | Delete

}

[244] Fix | Delete

[245] Fix | Delete

$output .= $msg;

[246] Fix | Delete

[247] Fix | Delete

if ($lineBreaksAfter > 0) {

[248] Fix | Delete

for ($i = 1; $i <= $lineBreaksAfter; $i++) {

[249] Fix | Delete

$output .= $lineBreak;

[250] Fix | Delete

}

[251] Fix | Delete

}

[252] Fix | Delete

[253] Fix | Delete

if (php_sapi_name() == "cli") {

[254] Fix | Delete

$output .= "\n";

[255] Fix | Delete

}

[256] Fix | Delete

[257] Fix | Delete

if (php_sapi_name() != "cli") {

[258] Fix | Delete

ob_flush();

[259] Fix | Delete

}

[260] Fix | Delete

[261] Fix | Delete

flush();

[262] Fix | Delete

}

[263] Fix | Delete

}

[264] Fix | Delete

[265] Fix | Delete