import.php

   1 <?php
   2 /* vim: set expandtab sw=4 ts=4 sts=4: */
   3 /**
   4  * Core script for import, this is just the glue around all other stuff
   5  *
   6  * @uses    PMA_Bookmark_getList()
   7  * @version $Id$
   8  * @package phpMyAdmin
   9  */
  10
  11 /**
  12  * Get the variables sent or posted to this script and a core script
  13  */
  14 require_once './libraries/common.inc.php';
  15 //require_once './libraries/display_import_functions.lib.php';
  16 $GLOBALS['js_include'][] = 'functions.js';
  17 $GLOBALS['js_include'][] = 'mootools.js';
  18
  19 // reset import messages for ajax request
  20 $_SESSION['Import_message']['message'] = null;
  21 $_SESSION['Import_message']['go_back_url'] = null;
  22 // default values
  23 $GLOBALS['reload'] = false;
  24
  25 // Are we just executing plain query or sql file? (eg. non import, but query box/window run)
  26 if (!empty($sql_query)) {
  27     // run SQL query
  28     $import_text = $sql_query;
  29     $import_type = 'query';
  30     $format = 'sql';
  31
  32     // refresh left frame on changes in table or db structure
  33     if (preg_match('/^(CREATE|ALTER|DROP)\s+(VIEW|TABLE|DATABASE|SCHEMA)\s+/i', $sql_query)) {
  34         $GLOBALS['reload'] = true;
  35     }
  36
  37     $sql_query = '';
  38 } elseif (!empty($sql_localfile)) {
  39     // run SQL file on server
  40     $local_import_file = $sql_localfile;
  41     $import_type = 'queryfile';
  42     $format = 'sql';
  43     unset($sql_localfile);
  44 } elseif (!empty($sql_file)) {
  45     // run uploaded SQL file
  46     $import_file = $sql_file;
  47     $import_type = 'queryfile';
  48     $format = 'sql';
  49     unset($sql_file);
  50 } elseif (!empty($id_bookmark)) {
  51     // run bookmark
  52     $import_type = 'query';
  53     $format = 'sql';
  54 }
  55
  56 // If we didn't get any parameters, either user called this directly, or
  57 // upload limit has been reached, let's assume the second possibility.
  58 ;
  59 if ($_POST == array() && $_GET == array()) {
  60     require_once './libraries/header.inc.php';
  61     $message = PMA_Message::error('strUploadLimit');
  62     $message->addParam('[a@./Documentation.html#faq1_16@_blank]');
  63     $message->addParam('[/a]');
  64
  65     // so we can obtain the message
  66     $_SESSION['Import_message']['message'] = $message->getDisplay();
  67     $_SESSION['Import_message']['go_back_url'] = $goto;
  68
  69     $message->display();
  70     require './libraries/footer.inc.php';
  71 }
  72
  73 // Check needed parameters
  74 PMA_checkParameters(array('import_type', 'format'));
  75
  76 // We don't want anything special in format
  77 $format = PMA_securePath($format);
  78
  79 // Import functions
  80 require_once './libraries/import.lib.php';
  81
  82 // Create error and goto url
  83 if ($import_type == 'table') {
  84     $err_url = 'tbl_import.php?' . PMA_generate_common_url($db, $table);
  85     $_SESSION['Import_message']['go_back_url'] = $err_url;
  86     $goto = 'tbl_import.php';
  87 } elseif ($import_type == 'database') {
  88     $err_url = 'db_import.php?' . PMA_generate_common_url($db);
  89     $_SESSION['Import_message']['go_back_url'] = $err_url;
  90     $goto = 'db_import.php';
  91 } elseif ($import_type == 'server') {
  92     $err_url = 'server_import.php?' . PMA_generate_common_url();
  93     $_SESSION['Import_message']['go_back_url'] = $err_url;
  94     $goto = 'server_import.php';
  95 } else {
  96     if (empty($goto) || !preg_match('@^(server|db|tbl)(_[a-z]*)*\.php$@i', $goto)) {
  97         if (strlen($table) && strlen($db)) {
  98             $goto = 'tbl_structure.php';
  99         } elseif (strlen($db)) {
 100             $goto = 'db_structure.php';
 101         } else {
 102             $goto = 'server_sql.php';
 103         }
 104     }
 105     if (strlen($table) && strlen($db)) {
 106         $common = PMA_generate_common_url($db, $table);
 107     } elseif (strlen($db)) {
 108         $common = PMA_generate_common_url($db);
 109     } else {
 110         $common = PMA_generate_common_url();
 111     }
 112     $err_url  = $goto
 113               . '?' . $common
 114               . (preg_match('@^tbl_[a-z]*\.php$@', $goto) ? '&amp;table=' . urlencode($table) : '');
 115     $_SESSION['Import_message']['go_back_url'] = $err_url;
 116 }
 117
 118
 119 if (strlen($db)) {
 120     PMA_DBI_select_db($db);
 121 }
 122
 123 @set_time_limit($cfg['ExecTimeLimit']);
 124 if (!empty($cfg['MemoryLimit'])) {
 125     @ini_set('memory_limit', $cfg['MemoryLimit']);
 126 }
 127
 128 $timestamp = time();
 129 if (isset($allow_interrupt)) {
 130     $maximum_time = ini_get('max_execution_time');
 131 } else {
 132     $maximum_time = 0;
 133 }
 134
 135 // set default values
 136 $timeout_passed = FALSE;
 137 $error = FALSE;
 138 $read_multiply = 1;
 139 $finished = FALSE;
 140 $offset = 0;
 141 $max_sql_len = 0;
 142 $file_to_unlink = '';
 143 $sql_query = '';
 144 $sql_query_disabled = FALSE;
 145 $go_sql = FALSE;
 146 $executed_queries = 0;
 147 $run_query = TRUE;
 148 $charset_conversion = FALSE;
 149 $reset_charset = FALSE;
 150 $bookmark_created = FALSE;
 151
 152 // Bookmark Support: get a query back from bookmark if required
 153 if (!empty($id_bookmark)) {
 154     $id_bookmark = (int)$id_bookmark;
 155     require_once './libraries/bookmark.lib.php';
 156     switch ($action_bookmark) {
 157         case 0: // bookmarked query that have to be run
 158             $import_text = PMA_Bookmark_get($db, $id_bookmark, 'id', isset($action_bookmark_all));
 159             if (isset($bookmark_variable) && !empty($bookmark_variable)) {
 160                 $import_text = preg_replace('|/\*(.*)\[VARIABLE\](.*)\*/|imsU', '${1}' . PMA_sqlAddslashes($bookmark_variable) . '${2}', $import_text);
 161             }
 162
 163             // refresh left frame on changes in table or db structure
 164             if (preg_match('/^(CREATE|ALTER|DROP)\s+(VIEW|TABLE|DATABASE|SCHEMA)\s+/i', $import_text)) {
 165                 $GLOBALS['reload'] = true;
 166             }
 167
 168             break;
 169         case 1: // bookmarked query that have to be displayed
 170             $import_text = PMA_Bookmark_get($db, $id_bookmark);
 171             $run_query = FALSE;
 172             break;
 173         case 2: // bookmarked query that have to be deleted
 174             $import_text = PMA_Bookmark_get($db, $id_bookmark);
 175             PMA_Bookmark_delete($db, $id_bookmark);
 176             $run_query = FALSE;
 177             $error = TRUE; // this is kind of hack to skip processing the query
 178             break;
 179     }
 180 } // end bookmarks reading
 181
 182 // Do no run query if we show PHP code
 183 if (isset($GLOBALS['show_as_php'])) {
 184     $run_query = FALSE;
 185     $go_sql = TRUE;
 186 }
 187
 188 // Store the query as a bookmark before executing it if bookmarklabel was given
 189 if (!empty($bkm_label) && !empty($import_text)) {
 190     require_once './libraries/bookmark.lib.php';
 191     $bfields = array(
 192                  'dbase' => $db,
 193                  'user'  => $cfg['Bookmark']['user'],
 194                  'query' => urlencode($import_text),
 195                  'label' => $bkm_label
 196     );
 197
 198     // Should we replace bookmark?
 199     if (isset($bkm_replace)) {
 200         $bookmarks = PMA_Bookmark_getList($db);
 201         foreach ($bookmarks as $key => $val) {
 202             if ($val == $bkm_label) {
 203                 PMA_Bookmark_delete($db, $key);
 204             }
 205         }
 206     }
 207
 208     PMA_Bookmark_save($bfields, isset($bkm_all_users));
 209
 210     $bookmark_created = TRUE;
 211 } // end store bookmarks
 212
 213 // We can not read all at once, otherwise we can run out of memory
 214 $memory_limit = trim(@ini_get('memory_limit'));
 215 // 2 MB as default
 216 if (empty($memory_limit)) {
 217     $memory_limit = 2 * 1024 * 1024;
 218 }
 219 // In case no memory limit we work on 10MB chunks
 220 if ($memory_limit == -1) {
 221     $memory_limit = 10 * 1024 * 1024;
 222 }
 223
 224 // Calculate value of the limit
 225 if (strtolower(substr($memory_limit, -1)) == 'm') {
 226     $memory_limit = (int)substr($memory_limit, 0, -1) * 1024 * 1024;
 227 } elseif (strtolower(substr($memory_limit, -1)) == 'k') {
 228     $memory_limit = (int)substr($memory_limit, 0, -1) * 1024;
 229 } elseif (strtolower(substr($memory_limit, -1)) == 'g') {
 230     $memory_limit = (int)substr($memory_limit, 0, -1) * 1024 * 1024 * 1024;
 231 } else {
 232     $memory_limit = (int)$memory_limit;
 233 }
 234
 235 $read_limit = $memory_limit / 8; // Just to be sure, there might be lot of memory needed for uncompression
 236
 237 // handle filenames
 238 if (!empty($local_import_file) && !empty($cfg['UploadDir'])) {
 239
 240     // sanitize $local_import_file as it comes from a POST
 241     $local_import_file = PMA_securePath($local_import_file);
 242
 243     $import_file  = PMA_userDir($cfg['UploadDir']) . $local_import_file;
 244 } elseif (empty($import_file) || !is_uploaded_file($import_file))  {
 245     $import_file  = 'none';
 246 }
 247
 248 // Do we have file to import?
 249
 250 if ($import_file != 'none' && !$error) {
 251     // work around open_basedir and other limitations
 252     $open_basedir = @ini_get('open_basedir');
 253
 254     // If we are on a server with open_basedir, we must move the file
 255     // before opening it. The doc explains how to create the "./tmp"
 256     // directory
 257
 258     if (!empty($open_basedir)) {
 259
 260         $tmp_subdir = (PMA_IS_WINDOWS ? '.\\tmp\\' : './tmp/');
 261
 262         if (is_writable($tmp_subdir)) {
 263
 264
 265             $import_file_new = $tmp_subdir . basename($import_file);
 266             if (move_uploaded_file($import_file, $import_file_new)) {
 267                 $import_file = $import_file_new;
 268                 $file_to_unlink = $import_file_new;
 269             }
 270
 271             $size = filesize($import_file);
 272         }
 273     }
 274
 275     /**
 276      *  Handle file compression
 277      *  @todo duplicate code exists in File.class.php
 278      */
 279     $compression = PMA_detectCompression($import_file);
 280     if ($compression === FALSE) {
 281         $message = PMA_Message::error('strFileCouldNotBeRead');
 282         $error = TRUE;
 283     } else {
 284         switch ($compression) {
 285             case 'application/bzip2':
 286                 if ($cfg['BZipDump'] && @function_exists('bzopen')) {
 287                     $import_handle = @bzopen($import_file, 'r');
 288                 } else {
 289                     $message = PMA_Message::error('strUnsupportedCompressionDetected');
 290                     $message->addParam($compression);
 291                     $error = TRUE;
 292                 }
 293                 break;
 294             case 'application/gzip':
 295                 if ($cfg['GZipDump'] && @function_exists('gzopen')) {
 296                     $import_handle = @gzopen($import_file, 'r');
 297                 } else {
 298                     $message = PMA_Message::error('strUnsupportedCompressionDetected');
 299                     $message->addParam($compression);
 300                     $error = TRUE;
 301                 }
 302                 break;
 303             case 'application/zip':
 304                 if ($cfg['ZipDump'] && @function_exists('zip_open')) {
 305                     /**
 306                      * Load interface for zip extension.
 307                      */
 308                     include_once './libraries/zip_extension.lib.php';
 309                     $result = PMA_getZipContents($import_file);
 310                     if (! empty($result['error'])) {
 311                         $message = PMA_Message::rawError($result['error']);
 312                         $error = TRUE;
 313                     } else {
 314                         $import_text = $result['data'];
 315                     }
 316                 } else {
 317                     $message = PMA_Message::error('strUnsupportedCompressionDetected');
 318                     $message->addParam($compression);
 319                     $error = TRUE;
 320                 }
 321                 break;
 322             case 'none':
 323                 $import_handle = @fopen($import_file, 'r');
 324                 break;
 325             default:
 326                 $message = PMA_Message::error('strUnsupportedCompressionDetected');
 327                 $message->addParam($compression);
 328                 $error = TRUE;
 329                 break;
 330         }
 331     }
 332     // use isset() because zip compression type does not use a handle
 333     if (!$error && isset($import_handle) && $import_handle === FALSE) {
 334         $message = PMA_Message::error('strFileCouldNotBeRead');
 335         $error = TRUE;
 336     }
 337 } elseif (!$error) {
 338     if (!isset($import_text) || empty($import_text)) {
 339         $message = PMA_Message::error('strNoDataReceived');
 340         $error = TRUE;
 341     }
 342 }
 343
 344 // so we can obtain the message
 345 //$_SESSION['Import_message'] = $message->getDisplay();
 346
 347 // Convert the file's charset if necessary
 348 if ($cfg['AllowAnywhereRecoding'] && isset($charset_of_file)) {
 349     if ($charset_of_file != $charset) {
 350         $charset_conversion = TRUE;
 351     }
 352 } elseif (isset($charset_of_file) && $charset_of_file != 'utf8') {
 353     PMA_DBI_query('SET NAMES \'' . $charset_of_file . '\'');
 354     // We can not show query in this case, it is in different charset
 355     $sql_query_disabled = TRUE;
 356     $reset_charset = TRUE;
 357 }
 358
 359 // Something to skip?
 360 if (!$error && isset($skip)) {
 361     $original_skip = $skip;
 362     while ($skip > 0) {
 363         PMA_importGetNextChunk($skip < $read_limit ? $skip : $read_limit);
 364         $read_multiply = 1; // Disable read progresivity, otherwise we eat all memory!
 365         $skip -= $read_limit;
 366     }
 367     unset($skip);
 368 }
 369
 370 if (!$error) {
 371     // Check for file existance
 372     if (!file_exists('./libraries/import/' . $format . '.php')) {
 373         $error = TRUE;
 374         $message = PMA_Message::error('strCanNotLoadImportPlugins');
 375     } else {
 376         // Do the real import
 377         $plugin_param = $import_type;
 378         require './libraries/import/' . $format . '.php';
 379     }
 380 }
 381
 382 if (! $error && FALSE !== $import_handle && NULL !== $import_handle) {
 383     fclose($import_handle);
 384 }
 385
 386 // Cleanup temporary file
 387 if ($file_to_unlink != '') {
 388     unlink($file_to_unlink);
 389 }
 390
 391 // Reset charset back, if we did some changes
 392 if ($reset_charset) {
 393     PMA_DBI_query('SET CHARACTER SET utf8');
 394     PMA_DBI_query('SET SESSION collation_connection =\'' . $collation_connection . '\'');
 395 }
 396
 397 // Show correct message
 398 if (!empty($id_bookmark) && $action_bookmark == 2) {
 399     $message = PMA_Message::success('strBookmarkDeleted');
 400     $display_query = $import_text;
 401     $error = FALSE; // unset error marker, it was used just to skip processing
 402 } elseif (!empty($id_bookmark) && $action_bookmark == 1) {
 403     $message = PMA_Message::notice('strShowingBookmark');
 404 } elseif ($bookmark_created) {
 405     $special_message = '[br]' . sprintf($strBookmarkCreated, htmlspecialchars($bkm_label));
 406 } elseif ($finished && !$error) {
 407     if ($import_type == 'query') {
 408         $message = PMA_Message::success();
 409     } else {
 410         if ($import_notice) {
 411             $message = PMA_Message::success('<em>'.$GLOBALS['strImportSuccessfullyFinished'].'</em>');
 412             $message->addParam($executed_queries);
 413
 414             $message->addString($import_notice);
 415         } else {
 416             $message = PMA_Message::success('strImportSuccessfullyFinished');
 417             $message->addParam($executed_queries);
 418         }
 419     }
 420 }
 421
 422 // Did we hit timeout? Tell it user.
 423 if ($timeout_passed) {
 424     $message = PMA_Message::error('strTimeoutPassed');
 425     if ($offset == 0 || (isset($original_skip) && $original_skip == $offset)) {
 426         $message->addString('strTimeoutNothingParsed');
 427     }
 428 }
 429
 430 // if there is any message, copy it into $_SESSION as well, so we can obtain it by AJAX call
 431 if (isset($message)) {
 432     $_SESSION['Import_message']['message'] = $message->getDisplay();
 433 //  $_SESSION['Import_message']['go_back_url'] = $goto.'?'.  PMA_generate_common_url();
 434 }
 435 // Parse and analyze the query, for correct db and table name
 436 // in case of a query typed in the query window
 437 // (but if the query is too large, in case of an imported file, the parser
 438 //  can choke on it so avoid parsing)
 439 if (strlen($sql_query) <= $GLOBALS['cfg']['MaxCharactersInDisplayedSQL']) {
 440     require_once './libraries/parse_analyze.lib.php';
 441 }
 442
 443 // There was an error?
 444 if (isset($my_die)) {
 445     foreach ($my_die AS $key => $die) {
 446         PMA_mysqlDie($die['error'], $die['sql'], '', $err_url, $error);
 447     }
 448 }
 449
 450 // we want to see the results of the last query that returned at least a row
 451 if (! empty($last_query_with_results)) {
 452     // but we want to show intermediate results too
 453     $disp_query = $sql_query;
 454     $disp_message = $strSuccess;
 455     $sql_query = $last_query_with_results;
 456     $go_sql = true;
 457 }
 458
 459 if ($go_sql) {
 460     require './sql.php';
 461 } else {
 462     $active_page = $goto;
 463     require './' . $goto;
 464 }
 465 exit();
 466 ?>