DB_HOST, 'dbname' => DB_NAME, 'dbuser' => DB_USER, 'dbpassword' => DB_PASSWORD, 'dbcharset' => defined( 'DB_CHARSET' ) ? DB_CHARSET : '', 'dumpfilehandle' => fopen( 'php://output', 'wb' ), 'dumpfile' => NULL, 'dbclientflags' => defined( 'MYSQL_CLIENT_FLAGS' ) ? MYSQL_CLIENT_FLAGS : 0, 'compression' => '' ); $args = wp_parse_args( $args , $default_args ); //set empty host to localhost if ( empty( $args[ 'dbhost' ] ) ) { $args[ 'dbhost' ] = NULL; } //check if port or socket in hostname and set port and socket $args[ 'dbport' ] = NULL; $args[ 'dbsocket' ] = NULL; if ( strstr( $args[ 'dbhost' ], ':' ) ) { $hostparts = explode( ':', $args[ 'dbhost' ], 2 ); $hostparts[ 0 ] = trim( $hostparts[ 0 ] ); $hostparts[ 1 ] = trim( $hostparts[ 1 ] ); if ( empty( $hostparts[ 0 ] ) ) $args[ 'dbhost' ] = NULL; else $args[ 'dbhost' ] = $hostparts[ 0 ]; if ( is_numeric( $hostparts[ 1 ] ) ) $args[ 'dbport' ] = (int) $hostparts[ 1 ]; else $args[ 'dbsocket' ] = $hostparts[ 1 ]; } $this->mysqli = mysqli_init(); if ( ! $this->mysqli ) { throw new BackWPup_MySQLDump_Exception( __( 'Cannot init MySQLi database connection', 'backwpup' ) ); } if ( ! $this->mysqli->options( MYSQLI_OPT_CONNECT_TIMEOUT, 5 ) ) { trigger_error( __( 'Setting of MySQLi connection timeout failed', 'backwpup' ) ); } //connect to Database if ( ! $this->mysqli->real_connect( $args[ 'dbhost' ], $args[ 'dbuser' ], $args[ 'dbpassword' ], $args[ 'dbname' ], $args[ 'dbport' ], $args[ 'dbsocket' ], $args[ 'dbclientflags' ] ) ) { throw new BackWPup_MySQLDump_Exception( sprintf( __( 'Cannot connect to MySQL database %1$d: %2$s', 'backwpup' ), mysqli_connect_errno(), mysqli_connect_error() ) ); } //set charset if ( ! empty( $args[ 'dbcharset' ] ) && method_exists( $this->mysqli, 'set_charset' ) ) { $res = $this->mysqli->set_charset( $args[ 'dbcharset' ] ); if ( ! $res ) { trigger_error( sprintf( _x( 'Cannot set DB charset to %s error: %s','Database Charset', 'backwpup' ), $args[ 'dbcharset' ], $this->mysqli->error ), E_USER_WARNING ); } } //set db name $this->dbname = $args[ 'dbname' ]; //set compression if ( ! empty( $args[ 'compression' ] ) && $args[ 'compression' ] === 'gz' ) { $this->compression = $args[ 'compression' ]; } //open file if set if ( $args[ 'dumpfile' ] ) { if ( substr( strtolower( $args[ 'dumpfile' ] ), -3 ) === '.gz' ) { if ( ! function_exists( 'gzencode' ) ) throw new BackWPup_MySQLDump_Exception( __( 'Functions for gz compression not available', 'backwpup' ) ); $this->compression = 'gz'; $this->handle = fopen( 'compress.zlib://' . $args[ 'dumpfile' ], 'ab' ); } else { $this->compression = ''; $this->handle = fopen( $args[ 'dumpfile' ], 'ab' ); } } else { $this->handle = $args[ 'dumpfilehandle' ]; } //check file handle if ( ! $this->handle ) { throw new BackWPup_MySQLDump_Exception( __( 'Cannot open SQL backup file', 'backwpup' ) ); } //get table info $res = $this->mysqli->query( "SHOW TABLE STATUS FROM `" . $this->dbname . "`" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { throw new BackWPup_MySQLDump_Exception( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW TABLE STATUS FROM `" .$this->dbname . "`" ) ); } else { while ( $tablestatus = $res->fetch_assoc() ) { $this->table_status[ $tablestatus[ 'Name' ] ] = $tablestatus; } $res->close(); } //get table names and types from Database $res = $this->mysqli->query( 'SHOW FULL TABLES FROM `' . $this->dbname . '`' ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { throw new BackWPup_MySQLDump_Exception( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, 'SHOW FULL TABLES FROM `' . $this->dbname . '`' ) ); } else { while ( $table = $res->fetch_array( MYSQLI_NUM ) ) { $this->table_types[ $table[ 0 ] ] = $table[ 1 ]; $this->tables_to_dump[] = $table[ 0 ]; if ( $table[ 1 ] === 'VIEW' ) { $this->views_to_dump[] = $table[ 0 ]; $this->table_status[ $table[ 0 ] ][ 'Rows' ] = 0; } } $res->close(); } } /** * Start the dump */ public function execute() { //increase time limit @set_time_limit( 300 ); //write dump head $this->dump_head(); //write tables foreach( $this->tables_to_dump as $table ) { $this->dump_table_head( $table ); $this->dump_table( $table ); $this->dump_table_footer( $table ); } //write footer $this->dump_footer(); } /** * Write Dump Header * * @param bool $wp_info Dump WordPress info in dump head */ public function dump_head( $wp_info = FALSE ) { // get sql timezone $res = $this->mysqli->query( "SELECT @@time_zone" ); $GLOBALS[ 'wpdb' ]->num_queries ++; $mysqltimezone = $res->fetch_row(); $mysqltimezone = $mysqltimezone[0]; $res->close(); //For SQL always use \n as MySQL wants this on all platforms. $dbdumpheader = "-- ---------------------------------------------------------\n"; $dbdumpheader .= "-- Backup with BackWPup ver.: " . BackWPup::get_plugin_data( 'Version' ) . "\n"; $dbdumpheader .= "-- http://backwpup.com/\n"; if ( $wp_info ) { $dbdumpheader .= "-- Blog Name: " . get_bloginfo( 'name' ) . "\n"; $dbdumpheader .= "-- Blog URL: " . trailingslashit( get_bloginfo( 'url' ) ) . "\n"; $dbdumpheader .= "-- Blog ABSPATH: " . trailingslashit( str_replace( '\\', '/', ABSPATH ) ) . "\n"; $dbdumpheader .= "-- Blog Charset: " . get_bloginfo( 'charset' ) . "\n"; $dbdumpheader .= "-- Table Prefix: " . $GLOBALS[ 'wpdb' ]->prefix . "\n"; } $dbdumpheader .= "-- Database Name: " . $this->dbname . "\n"; $dbdumpheader .= "-- Backup on: " . date( 'Y-m-d H:i.s', current_time( 'timestamp' ) ) . "\n"; $dbdumpheader .= "-- ---------------------------------------------------------\n\n"; //for better import with mysql client $dbdumpheader .= "/*!40101 SET @OLD_CHARACTER_SET_CLIENT=@@CHARACTER_SET_CLIENT */;\n"; $dbdumpheader .= "/*!40101 SET @OLD_CHARACTER_SET_RESULTS=@@CHARACTER_SET_RESULTS */;\n"; $dbdumpheader .= "/*!40101 SET @OLD_COLLATION_CONNECTION=@@COLLATION_CONNECTION */;\n"; $dbdumpheader .= "/*!40101 SET NAMES " . $this->mysqli->character_set_name() . " */;\n"; $dbdumpheader .= "/*!40103 SET @OLD_TIME_ZONE=@@TIME_ZONE */;\n"; $dbdumpheader .= "/*!40103 SET TIME_ZONE='" . $mysqltimezone . "' */;\n"; $dbdumpheader .= "/*!40014 SET @OLD_UNIQUE_CHECKS=@@UNIQUE_CHECKS, UNIQUE_CHECKS=0 */;\n"; $dbdumpheader .= "/*!40014 SET @OLD_FOREIGN_KEY_CHECKS=@@FOREIGN_KEY_CHECKS, FOREIGN_KEY_CHECKS=0 */;\n"; $dbdumpheader .= "/*!40101 SET @OLD_SQL_MODE=@@SQL_MODE, SQL_MODE='NO_AUTO_VALUE_ON_ZERO' */;\n"; $dbdumpheader .= "/*!40111 SET @OLD_SQL_NOTES=@@SQL_NOTES, SQL_NOTES=0 */;\n\n"; $this->write( $dbdumpheader ); } /** * Write Dump Footer with dump of functions and procedures */ public function dump_footer() { //dump Views foreach( $this->views_to_dump as $view ) { $this->dump_view_table_head( $view ); } //dump procedures and functions $this->write( "\n--\n-- Backup routines for database '" . $this->dbname . "'\n--\n" ); //dump Functions $res = $this->mysqli->query( "SHOW FUNCTION STATUS" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW FUNCTION STATUS" ), E_USER_WARNING ); } else { while ( $function_status = $res->fetch_assoc() ) { if ( $this->dbname != $function_status[ 'Db' ] ) { continue; } $res2 = $this->mysqli->query( "SHOW CREATE FUNCTION `" . $function_status[ 'Db' ] . "`.`" . $function_status[ 'Name' ] . "`" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW CREATE FUNCTION `" . $function_status[ 'Db' ] . "`.`" . $function_status[ 'Name' ] . "`" ), E_USER_WARNING ); } else { $create_function = $res2->fetch_assoc(); $res2->close(); $create = "\n--\n-- Function structure for " . $function_status[ 'Name' ] . "\n--\n\n"; $create .= "DROP FUNCTION IF EXISTS `" . $function_status[ 'Name' ] . "`;\n"; $create .= "/*!40101 SET @saved_cs_client = @@character_set_client */;\n"; $create .= "/*!40101 SET character_set_client = '" . $this->mysqli->character_set_name() . "' */;\n"; $create .= $create_function[ 'Create Function' ] . ";\n"; $create .= "/*!40101 SET character_set_client = @saved_cs_client */;\n"; $this->write( $create ); } } $res->close(); } //dump Procedures $res = $this->mysqli->query( "SHOW PROCEDURE STATUS" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW PROCEDURE STATUS" ), E_USER_WARNING ); } else { while ( $procedure_status = $res->fetch_assoc() ) { if ( $this->dbname != $procedure_status[ 'Db' ] ) { continue; } $res2 = $this->mysqli->query( "SHOW CREATE PROCEDURE `" . $procedure_status[ 'Db' ] . "`.`" . $procedure_status[ 'Name' ] . "`" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW CREATE PROCEDURE `" . $procedure_status[ 'Db' ] . "`.`" . $procedure_status[ 'Name' ] . "`" ), E_USER_WARNING ); } else { $create_procedure = $res2->fetch_assoc(); $res2->close(); $create = "\n--\n-- Procedure structure for " . $procedure_status[ 'Name' ] . "\n--\n\n"; $create .= "DROP PROCEDURE IF EXISTS `" . $procedure_status[ 'Name' ] . "`;\n"; $create .= "/*!40101 SET @saved_cs_client = @@character_set_client */;\n"; $create .= "/*!40101 SET character_set_client = '" . $this->mysqli->character_set_name() . "' */;\n"; $create .= $create_procedure[ 'Create Procedure' ] . ";\n"; $create .= "/*!40101 SET character_set_client = @saved_cs_client */;\n"; $this->write( $create ); } } $res->close(); } //dump Trigger $res = $this->mysqli->query( "SHOW TRIGGERS FROM `" . $this->dbname . "`" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW TRIGGERS" ), E_USER_WARNING ); } else { while ( $triggers = $res->fetch_assoc() ) { $res2 = $this->mysqli->query( "SHOW CREATE TRIGGER `" . $this->dbname . "`.`" . $triggers[ 'Trigger' ] . "`" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW CREATE TRIGGER `" . $this->dbname . "`.`" . $triggers[ 'Trigger' ] . "`" ), E_USER_WARNING ); } else { $create_trigger = $res2->fetch_assoc(); $res2->close(); $create = "\n--\n-- Trigger structure for " . $triggers[ 'Trigger' ] . "\n--\n\n"; $create .= "DROP TRIGGER IF EXISTS `" . $triggers[ 'Trigger' ] . "`;\n"; $create .= "DELIMITER //\n"; $create .= $create_trigger[ 'SQL Original Statement' ] . ";\n"; $create .= "//\nDELIMITER ;\n"; $this->write( $create ); } } $res->close(); } //for better import with mysql client $dbdumpfooter = "\n/*!40103 SET TIME_ZONE=@OLD_TIME_ZONE */;\n"; $dbdumpfooter .= "/*!40101 SET SQL_MODE=@OLD_SQL_MODE */;\n"; $dbdumpfooter .= "/*!40014 SET FOREIGN_KEY_CHECKS=@OLD_FOREIGN_KEY_CHECKS */;\n"; $dbdumpfooter .= "/*!40014 SET UNIQUE_CHECKS=@OLD_UNIQUE_CHECKS */;\n"; $dbdumpfooter .= "/*!40101 SET CHARACTER_SET_CLIENT=@OLD_CHARACTER_SET_CLIENT */;\n"; $dbdumpfooter .= "/*!40101 SET CHARACTER_SET_RESULTS=@OLD_CHARACTER_SET_RESULTS */;\n"; $dbdumpfooter .= "/*!40101 SET COLLATION_CONNECTION=@OLD_COLLATION_CONNECTION */;\n"; $dbdumpfooter .= "/*!40111 SET SQL_NOTES=@OLD_SQL_NOTES */;\n"; $dbdumpfooter .= "\n-- Backup completed on " . date( 'Y-m-d H:i:s', current_time( 'timestamp' ) ). "\n"; $this->write( $dbdumpfooter ); } /** * Dump table structure * * @param string $table name of Table to dump * @throws BackWPup_MySQLDump_Exception * @return int Size of table */ public function dump_table_head( $table ) { //dump View if ( $this->table_types[ $table ] === 'VIEW' ) { //Dump the view table structure $fields = array(); $res = $this->mysqli->query( "SELECT * FROM `" . $table . "` LIMIT 1" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SELECT * FROM `" . $table . "` LIMIT 1" ), E_USER_WARNING ); } else { $fields = $res->fetch_fields(); $res->close(); } if ( $res ) { $tablecreate = "\n--\n-- Temporary table structure for view `" . $table . "`\n--\n\n"; $tablecreate .= "DROP TABLE IF EXISTS `" . $table . "`;\n"; $tablecreate .= "/*!50001 DROP VIEW IF EXISTS `" . $table . "`*/;\n"; $tablecreate .= "/*!40101 SET @saved_cs_client = @@character_set_client */;\n"; $tablecreate .= "/*!40101 SET character_set_client = '" . $this->mysqli->character_set_name() . "' */;\n"; $tablecreate .= "CREATE TABLE `" . $table . "` (\n"; foreach( $fields as $field ) { $tablecreate .= " `". $field->orgname . "` tinyint NOT NULL,\n"; } $tablecreate = substr( $tablecreate, 0, -2 ) ."\n"; $tablecreate .= ");\n"; $tablecreate .= "/*!40101 SET character_set_client = @saved_cs_client */;\n"; $this->write( $tablecreate ); } return 0; } //dump normal Table $tablecreate = "\n--\n-- Table structure for `" . $table . "`\n--\n\n"; $tablecreate .= "DROP TABLE IF EXISTS `" . $table . "`;\n"; $tablecreate .= "/*!40101 SET @saved_cs_client = @@character_set_client */;\n"; $tablecreate .= "/*!40101 SET character_set_client = '" . $this->mysqli->character_set_name() . "' */;\n"; //Dump the table structure $res = $this->mysqli->query( "SHOW CREATE TABLE `" . $table . "`" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW CREATE TABLE `" . $table . "`" ), E_USER_WARNING ); } else { $createtable = $res->fetch_assoc(); $res->close(); $tablecreate .= $createtable[ 'Create Table' ] . ";\n"; $tablecreate .= "/*!40101 SET character_set_client = @saved_cs_client */;\n"; $this->write( $tablecreate ); if ( $this->table_status[ $table ][ 'Engine' ] !== 'MyISAM' ) { $this->table_status[ $table ][ 'Rows' ] = '~' . $this->table_status[ $table ][ 'Rows' ]; } if ( $this->table_status[ $table ][ 'Rows' ] !== 0 ) { //Dump Table data $this->write( "\n--\n-- Backup data for table `" . $table . "`\n--\n\nLOCK TABLES `" . $table . "` WRITE;\n/*!40000 ALTER TABLE `" . $table . "` DISABLE KEYS */;\n" ); } return $this->table_status[ $table ][ 'Rows' ]; } return 0; } /** * Dump view structure * * @param string $view name of Table to dump * @throws BackWPup_MySQLDump_Exception * @return int Size of table */ public function dump_view_table_head( $view ) { //Dump the view structure $res = $this->mysqli->query( "SHOW CREATE VIEW `" . $view . "`" ); $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SHOW CREATE VIEW `" . $view . "`" ), E_USER_WARNING ); } else { $createview = $res->fetch_assoc(); $res->close(); $tablecreate = "\n--\n-- View structure for `" . $view . "`\n--\n\n"; $tablecreate .= "DROP TABLE IF EXISTS `" . $view . "`;\n"; $tablecreate .= "DROP VIEW IF EXISTS `" . $view . "`;\n"; $tablecreate .= "/*!40101 SET @saved_cs_client = @@character_set_client */;\n"; $tablecreate .= "/*!40101 SET character_set_client = '" . $this->mysqli->character_set_name() . "' */;\n"; $tablecreate .= $createview[ 'Create View' ] . ";\n"; $tablecreate .= "/*!40101 SET character_set_client = @saved_cs_client */;\n"; $this->write( $tablecreate ); } } /** * Dump table footer * * @param string $table name of Table to dump * @return int Size of table */ public function dump_table_footer( $table ) { if ( $this->table_status[ $table ][ 'Rows' ] !== 0 ) { $this->write( "/*!40000 ALTER TABLE `" . $table . "` ENABLE KEYS */;\nUNLOCK TABLES;\n" ); } } /** * Dump table Data * * @param string $table name of Table to dump * @param int $start Start of lengh paramter * @param int $length how many * @return int done records in this backup * @throws BackWPup_MySQLDump_Exception */ public function dump_table( $table, $start = 0, $length = 0 ) { if ( ! is_numeric( $start ) ) { throw new BackWPup_MySQLDump_Exception( sprintf( __( 'Start for table backup is not correctly set: %1$s', 'backwpup' ), $start ) ); } if ( ! is_numeric( $length ) ) { throw new BackWPup_MySQLDump_Exception( sprintf( __( 'Length for table backup is not correctly set: %1$s', 'backwpup' ), $length ) ); } $done_records = 0; if ( $this->table_types[ $table ] == 'VIEW' ) { return $done_records; } //get data from table if ( $length == 0 && $start == 0 ) { $res = $this->mysqli->query( "SELECT * FROM `" . $table . "` ", MYSQLI_USE_RESULT ); } else { $res = $this->mysqli->query( "SELECT * FROM `" . $table . "` LIMIT " . $start . ", " . $length, MYSQLI_USE_RESULT ); } $GLOBALS[ 'wpdb' ]->num_queries ++; if ( $this->mysqli->error ) { trigger_error( sprintf( __( 'Database error %1$s for query %2$s', 'backwpup' ), $this->mysqli->error, "SELECT * FROM `" . $table . "`" ), E_USER_WARNING ); return 0; } $fieldsarray = array(); $fieldinfo = array(); $fields = $res->fetch_fields(); $i = 0; foreach ( $fields as $filed ) { $fieldsarray[ $i ] = $filed->orgname; $fieldinfo[ $fieldsarray[ $i ] ] = $filed; $i ++; } $dump = ''; while ( $data = $res->fetch_assoc() ) { $values = array(); foreach ( $data as $key => $value ) { if ( is_null( $value ) || ! isset( $value ) ) { // Make Value NULL to string NULL $value = "NULL"; } elseif ( in_array( (int) $fieldinfo[ $key ]->type, array( MYSQLI_TYPE_DECIMAL, MYSQLI_TYPE_TINY, MYSQLI_TYPE_SHORT, MYSQLI_TYPE_LONG, MYSQLI_TYPE_FLOAT, MYSQLI_TYPE_DOUBLE, MYSQLI_TYPE_LONGLONG, MYSQLI_TYPE_INT24 ), true ) ) {//is value numeric no esc $value = empty( $value ) ? 0 : $value; } elseif ( $fieldinfo[ $key ]->flags & MYSQLI_BINARY_FLAG ) {//is value binary $hex = unpack( 'H*', $value ); $value = empty( $value ) ? "''" : "0x$hex[1]"; } else { $value = "'" . $this->mysqli->real_escape_string( $value ) . "'"; } $values[ ] = $value; } //new query in dump on more than 50000 chars. if ( empty( $dump ) ) $dump = "INSERT INTO `" . $table . "` (`" . implode( "`, `", $fieldsarray ) . "`) VALUES \n"; if ( strlen( $dump ) <= 50000 ) { $dump .= "(" . implode( ", ", $values ) . "),\n"; } else { $dump .= "(" . implode( ", ", $values ) . ");\n"; $this->write( $dump ); $dump = ''; } $done_records ++; } if ( ! empty( $dump ) ) { $dump = substr( $dump, 0, -2 ) . ";\n" ; $this->write( $dump ); } $res->close(); return $done_records; } /** * Writes data to handle and compress * * @param $data string to write * @throws BackWPup_MySQLDump_Exception */ private function write( $data ) { $written = fwrite( $this->handle, $data ); if ( ! $written ) throw new BackWPup_MySQLDump_Exception( __( 'Error while writing file!', 'backwpup' ) ); } /** * Closes all confections on shutdown. */ public function __destruct() { //close MySQL connection $this->mysqli->close(); //close file handle if ( is_resource( $this->handle ) ) fclose( $this->handle ); } } /** * Exception Handler */ class BackWPup_MySQLDump_Exception extends Exception { }