<?php
/* vim: set expandtab sw=4 ts=4 sts=4: */
/**
 * Main export handling code
 *
 * @package PhpMyAdmin
 */

/**
 * Get the variables sent or posted to this script and a core script
 */
if (!defined('TESTSUITE')) {
    /**
     * If we are sending the export file (as opposed to just displaying it
     * as text), we have to bypass the usual PMA_Response mechanism
     */
    if (isset($_POST['output_format']) && $_POST['output_format'] == 'sendit') {
        define('PMA_BYPASS_GET_INSTANCE', 1);
    }
    include_once 'libraries/common.inc.php';
    include_once 'libraries/zip.lib.php';
    include_once 'libraries/plugin_interface.lib.php';
    include_once 'libraries/export.lib.php';
    
    //check if it's the GET request to check export time out
    if (isset($_GET['check_time_out'])) {
        if (isset($_SESSION['pma_export_error'])) {
            $err = $_SESSION['pma_export_error'];
            unset($_SESSION['pma_export_error']);
            echo $err;
        } else {
            echo "success";
        }
        exit;
    }
    /**
     * Sets globals from $_POST
     *
     * - Please keep the parameters in order of their appearance in the form
     * - Some of these parameters are not used, as the code below directly
     *   verifies from the superglobal $_POST or $_REQUEST
     */
    $post_params = array(
            'db',
            'table',
            'single_table',
            'export_type',
            'export_method',
            'quick_or_custom',
            'db_select',
            'table_select',
            'limit_to',
            'limit_from',
            'allrows',
            'output_format',
            'filename_template',
            'remember_template',
            'charset_of_file',
            'compression',
            'what',
            'htmlword_structure_or_data',
            'htmlword_null',
            'htmlword_columns',
            'mediawiki_structure_or_data',
            'mediawiki_caption',
            'pdf_report_title',
            'pdf_structure_or_data',
            'odt_structure_or_data',
            'odt_relation',
            'odt_comments',
            'odt_mime',
            'odt_columns',
            'odt_null',
            'codegen_structure_or_data',
            'codegen_format',
            'excel_null',
            'excel_removeCRLF',
            'excel_columns',
            'excel_edition',
            'excel_structure_or_data',
            'yaml_structure_or_data',
            'ods_null',
            'ods_structure_or_data',
            'ods_columns',
            'json_structure_or_data',
            'xml_structure_or_data',
            'xml_export_functions',
            'xml_export_procedures',
            'xml_export_tables',
            'xml_export_triggers',
            'xml_export_views',
            'xml_export_contents',
            'texytext_structure_or_data',
            'texytext_columns',
            'texytext_null',
            'phparray_structure_or_data',
            'sql_include_comments',
            'sql_header_comment',
            'sql_dates',
            'sql_relation',
            'sql_mime',
            'sql_use_transaction',
            'sql_disable_fk',
            'sql_compatibility',
            'sql_structure_or_data',
            'sql_create_database',
            'sql_drop_table',
            'sql_procedure_function',
            'sql_create_table_statements',
            'sql_if_not_exists',
            'sql_auto_increment',
            'sql_backquotes',
            'sql_truncate',
            'sql_delayed',
            'sql_ignore',
            'sql_type',
            'sql_insert_syntax',
            'sql_max_query_size',
            'sql_hex_for_blob',
            'sql_utc_time',
            'csv_separator',
            'csv_enclosed',
            'csv_escaped',
            'csv_terminated',
            'csv_null',
            'csv_removeCRLF',
            'csv_columns',
            'csv_structure_or_data',
            // csv_replace should have been here but we use it directly from $_POST
            'latex_caption',
            'latex_structure_or_data',
            'latex_structure_caption',
            'latex_structure_continued_caption',
            'latex_structure_label',
            'latex_relation',
            'latex_comments',
            'latex_mime',
            'latex_columns',
            'latex_data_caption',
            'latex_data_continued_caption',
            'latex_data_label',
            'latex_null'
    );

    foreach ($post_params as $one_post_param) {
        if (isset($_POST[$one_post_param])) {
            $GLOBALS[$one_post_param] = $_POST[$one_post_param];
        }
    }

    // sanitize this parameter which will be used below in a file inclusion
    $what = PMA_securePath($what);

    PMA_Util::checkParameters(array('what', 'export_type'));

    // export class instance, not array of properties, as before
    $export_plugin = PMA_getPlugin(
        "export",
        $what,
        'libraries/plugins/export/',
        array(
            'export_type' => $export_type,
            'single_table' => isset($single_table)
        )
    );

    // Backward compatibility
    $type = $what;

    // Check export type
    if (! isset($export_plugin)) {
        PMA_fatalError(__('Bad type!'));
    }

    /**
     * valid compression methods
     */
    $compression_methods = array(
        'zip',
        'gzip',
        'bzip2',
    );

    /**
     * init and variable checking
     */
    $compression = false;
    $onserver = false;
    $save_on_server = false;
    $buffer_needed = false;

    // Is it a quick or custom export?
    if ($_REQUEST['quick_or_custom'] == 'quick') {
        $quick_export = true;
    } else {
        $quick_export = false;
    }

    if ($_REQUEST['output_format'] == 'astext') {
        $asfile = false;
    } else {
        $asfile = true;
        if (in_array($_REQUEST['compression'], $compression_methods)) {
            $compression = $_REQUEST['compression'];
            $buffer_needed = true;
        }
        if (($quick_export && ! empty($_REQUEST['quick_export_onserver']))
            || (! $quick_export && ! empty($_REQUEST['onserver']))
        ) {
            if ($quick_export) {
                $onserver = $_REQUEST['quick_export_onserver'];
            } else {
                $onserver = $_REQUEST['onserver'];
            }
            // Will we save dump on server?
            $save_on_server = ! empty($cfg['SaveDir']) && $onserver;
        }
    }

    // Generate error url and check for needed variables
    if ($export_type == 'server') {
        $err_url = 'server_export.php?' . PMA_URL_getCommon();
    } elseif ($export_type == 'database' && strlen($db)) {
        $err_url = 'db_export.php?' . PMA_URL_getCommon($db);
        // Check if we have something to export
        if (isset($table_select)) {
            $tables = $table_select;
        } else {
            $tables = array();
        }
    } elseif ($export_type == 'table' && strlen($db) && strlen($table)) {
        $err_url = 'tbl_export.php?' . PMA_URL_getCommon($db, $table);
    } else {
        PMA_fatalError(__('Bad parameters!'));
    }

    /**
     * Increase time limit for script execution and initializes some variables
     */
    @set_time_limit($cfg['ExecTimeLimit']);
    if (! empty($cfg['MemoryLimit'])) {
        @ini_set('memory_limit', $cfg['MemoryLimit']);
    }
    register_shutdown_function('PMA_shutdown');
    // Start with empty buffer
    $dump_buffer = '';
    $dump_buffer_len = 0;

    // We send fake headers to avoid browser timeout when buffering
    $time_start = time();
}

// Defines the default <CR><LF> format.
// For SQL always use \n as MySQL wants this on all platforms.
if (!defined('TESTSUITE')) {
    if ($what == 'sql') {
        $crlf = "\n";
    } else {
        $crlf = PMA_Util::whichCrlf();
    }

    $output_kanji_conversion = function_exists('PMA_Kanji_strConv')
        && $type != 'xls';

    // Do we need to convert charset?
    $output_charset_conversion = $asfile
        && $GLOBALS['PMA_recoding_engine'] != PMA_CHARSET_NONE
        && isset($charset_of_file) && $charset_of_file != 'utf-8'
        && $type != 'xls';

    // Use on the fly compression?
    $GLOBALS['onfly_compression'] = $GLOBALS['cfg']['CompressOnFly']
        && ($compression == 'gzip' || $compression == 'bzip2');
    if ($GLOBALS['onfly_compression']) {
        $GLOBALS['memory_limit'] = PMA_getMemoryLimitForExport();
    }

    // Generate filename and mime type if needed
    if ($asfile) {
        if (empty($remember_template)) {
            $remember_template = '';
        }
        list($filename, $mime_type) = PMA_getExportFilenameAndMimetype(
            $export_type, $remember_template, $export_plugin, $compression,
            $filename_template
        );
    }

    // Open file on server if needed
    if ($save_on_server) {
        list($save_filename, $message, $file_handle) = PMA_openExportFile(
            $filename, $quick_export
        );

        // problem opening export file on server?
        if (! empty($message)) {
            if ($export_type == 'server') {
                $active_page = 'server_export.php';
                include 'server_export.php';
            } elseif ($export_type == 'database') {
                $active_page = 'db_export.php';
                include 'db_export.php';
            } else {
                $active_page = 'tbl_export.php';
                include 'tbl_export.php';
            }
            exit();
        }
    }

    /**
     * Send headers depending on whether the user chose to download a dump file
     * or not
     */
    if (! $save_on_server) {
        if ($asfile) {
            // Download
            // (avoid rewriting data containing HTML with anchors and forms;
            // this was reported to happen under Plesk)
            @ini_set('url_rewriter.tags', '');
            $filename = PMA_sanitizeFilename($filename);

            PMA_downloadHeader($filename, $mime_type);
        } else {
            // HTML
            if ($export_type == 'database') {
                $num_tables = count($tables);
                if ($num_tables == 0) {
                    $message = PMA_Message::error(
                        __('No tables found in database.')
                    );
                    $active_page = 'db_export.php';
                    include 'db_export.php';
                    exit();
                }
            }
            list($html, $back_button) = PMA_getHtmlForDisplayedExportHeader(
                $export_type, $db, $table
            );
            echo $html;
            unset($html);
        } // end download
    }

    // Fake loop just to allow skip of remain of this code by break, I'd really
    // need exceptions here :-)
    do {

        // Add possibly some comments to export
        if (! $export_plugin->exportHeader($db)) {
            break;
        }

        // Will we need relation & co. setup?
        $do_relation = isset($GLOBALS[$what . '_relation']);
        $do_comments = isset($GLOBALS[$what . '_include_comments']);
        $do_mime     = isset($GLOBALS[$what . '_mime']);
        if ($do_relation || $do_comments || $do_mime) {
            $cfgRelation = PMA_getRelationsParam();
        }
        if ($do_mime) {
            include_once 'libraries/transformations.lib.php';
        }

        // Include dates in export?
        $do_dates = isset($GLOBALS[$what . '_dates']);

        $whatStrucOrData = $GLOBALS[$what . '_structure_or_data'];

        /**
         * Builds the dump
         */
        // Gets the number of tables if a dump of a database has been required
        if ($export_type == 'server') {
            if (isset($db_select)) {
                $tmp_select = implode($db_select, '|');
                $tmp_select = '|' . $tmp_select . '|';
            }
            // Walk over databases
            foreach ($GLOBALS['pma']->databases as $current_db) {
                if (isset($tmp_select)
                    && strpos(' ' . $tmp_select, '|' . $current_db . '|')
                ) {
                    if (! $export_plugin->exportDBHeader($current_db)) {
                        break 2;
                    }
                    if (! $export_plugin->exportDBCreate($current_db)) {
                        break 2;
                    }
                    if (method_exists($export_plugin, 'exportRoutines')
                        && strpos($whatStrucOrData, 'structure') !== false
                        && isset($GLOBALS['sql_procedure_function'])
                    ) {
                        $export_plugin->exportRoutines($current_db);
                    }

                    $tables = $GLOBALS['dbi']->getTables($current_db);
                    $views = array();
                    foreach ($tables as $table) {
                        // if this is a view, collect it for later;
                        // views must be exported after the tables
                        $is_view = PMA_Table::isView($current_db, $table);
                        if ($is_view) {
                            $views[] = $table;
                        }
                        if ($whatStrucOrData == 'structure'
                            || $whatStrucOrData == 'structure_and_data'
                        ) {
                            // for a view, export a stand-in definition of the table
                            // to resolve view dependencies
                            if (! $export_plugin->exportStructure(
                                $current_db, $table, $crlf, $err_url,
                                $is_view ? 'stand_in' : 'create_table', $export_type,
                                $do_relation, $do_comments, $do_mime, $do_dates
                            )) {
                                break 3;
                            }
                        }
                        // if this is a view or a merge table, don't export data
                        if (($whatStrucOrData == 'data'
                            || $whatStrucOrData == 'structure_and_data')
                            && ! ($is_view || PMA_Table::isMerge($current_db, $table))
                        ) {
                            $local_query  = 'SELECT * FROM '
                                . PMA_Util::backquote($current_db)
                                . '.' . PMA_Util::backquote($table);
                            if (! $export_plugin->exportData(
                                $current_db, $table, $crlf, $err_url, $local_query
                            )) {
                                break 3;
                            }
                        }
                        // now export the triggers (needs to be done after the data
                        // because triggers can modify already imported tables)
                        if ($whatStrucOrData == 'structure'
                            || $whatStrucOrData == 'structure_and_data'
                        ) {
                            if (! $export_plugin->exportStructure(
                                $current_db, $table, $crlf, $err_url,
                                'triggers', $export_type,
                                $do_relation, $do_comments, $do_mime, $do_dates
                            )) {
                                break 2;
                            }
                        }
                    }
                    foreach ($views as $view) {
                        // no data export for a view
                        if ($whatStrucOrData == 'structure'
                            || $whatStrucOrData == 'structure_and_data'
                        ) {
                            if (! $export_plugin->exportStructure(
                                $current_db, $view, $crlf, $err_url,
                                'create_view', $export_type,
                                $do_relation, $do_comments, $do_mime, $do_dates
                            )) {
                                break 3;
                            }
                        }
                    }
                    if (! $export_plugin->exportDBFooter($current_db)) {
                        break 2;
                    }
                }
            }
        } elseif ($export_type == 'database') {
            if (! $export_plugin->exportDBHeader($db)) {
                break;
            }
            if (! $export_plugin->exportDBCreate($db)) {
                break;
            }

            if (method_exists($export_plugin, 'exportRoutines')
                && strpos($GLOBALS['sql_structure_or_data'], 'structure') !== false
                && isset($GLOBALS['sql_procedure_function'])
            ) {
                $export_plugin->exportRoutines($db);
            }

            $i = 0;
            $views = array();

            // $tables contains the choices from the user (via $table_select)
            foreach ($tables as $table) {
                // if this is a view, collect it for later;
                // views must be exported after the tables
                $is_view = PMA_Table::isView($db, $table);
                if ($is_view) {
                    $views[] = $table;
                }
                if ($whatStrucOrData == 'structure'
                    || $whatStrucOrData == 'structure_and_data'
                ) {
                    // for a view, export a stand-in definition of the table
                    // to resolve view dependencies
                    if (! $export_plugin->exportStructure(
                        $db, $table, $crlf, $err_url,
                        $is_view ? 'stand_in' : 'create_table', $export_type,
                        $do_relation, $do_comments, $do_mime, $do_dates
                    )) {
                        break 2;
                    }
                }
                // if this is a view or a merge table, don't export data
                if (($whatStrucOrData == 'data'
                    || $whatStrucOrData == 'structure_and_data')
                    && ! ($is_view || PMA_Table::isMerge($db, $table))
                ) {
                    $local_query  = 'SELECT * FROM ' . PMA_Util::backquote($db)
                        . '.' . PMA_Util::backquote($table);
                    if (! $export_plugin->exportData(
                        $db, $table, $crlf, $err_url, $local_query
                    )) {
                        break 2;
                    }
                }
                // now export the triggers (needs to be done after the data because
                // triggers can modify already imported tables)
                if ($whatStrucOrData == 'structure'
                    || $whatStrucOrData == 'structure_and_data'
                ) {
                    if (! $export_plugin->exportStructure(
                        $db, $table, $crlf, $err_url,
                        'triggers', $export_type,
                        $do_relation, $do_comments, $do_mime, $do_dates
                    )) {
                        break 2;
                    }
                }
            }
            foreach ($views as $view) {
                // no data export for a view
                if ($whatStrucOrData == 'structure'
                    || $whatStrucOrData == 'structure_and_data'
                ) {
                    if (! $export_plugin->exportStructure(
                        $db, $view, $crlf, $err_url,
                        'create_view', $export_type,
                        $do_relation, $do_comments, $do_mime, $do_dates
                    )) {
                        break 2;
                    }
                }
            }

            if (! $export_plugin->exportDBFooter($db)) {
                break;
            }
        } else {
            if (! $export_plugin->exportDBHeader($db)) {
                break;
            }
            // We export just one table
            // $allrows comes from the form when "Dump all rows" has been selected
            if (isset($allrows)
                && $allrows == '0'
                && $limit_to > 0
                && $limit_from >= 0
            ) {
                $add_query  = ' LIMIT '
                            . (($limit_from > 0) ? $limit_from . ', ' : '')
                            . $limit_to;
            } else {
                $add_query  = '';
            }

            $is_view = PMA_Table::isView($db, $table);
            if ($whatStrucOrData == 'structure'
                || $whatStrucOrData == 'structure_and_data'
            ) {
                if (! $export_plugin->exportStructure(
                    $db, $table, $crlf, $err_url,
                    $is_view ? 'create_view' : 'create_table', $export_type,
                    $do_relation, $do_comments, $do_mime, $do_dates
                )) {
                    break;
                }
            }
            // If this is an export of a single view, we have to export data;
            // for example, a PDF report
            // if it is a merge table, no data is exported
            if (($whatStrucOrData == 'data'
                || $whatStrucOrData == 'structure_and_data')
                && ! PMA_Table::isMerge($db, $table)
            ) {
                if (! empty($sql_query)) {
                    // only preg_replace if needed
                    if (! empty($add_query)) {
                        // remove trailing semicolon before adding a LIMIT
                        $sql_query = preg_replace('%;\s*$%', '', $sql_query);
                    }
                    $local_query = $sql_query . $add_query;
                    $GLOBALS['dbi']->selectDb($db);
                } else {
                    $local_query  = 'SELECT * FROM ' . PMA_Util::backquote($db)
                        . '.' . PMA_Util::backquote($table) . $add_query;
                }
                if (! $export_plugin->exportData(
                    $db, $table, $crlf, $err_url, $local_query
                )) {
                    break;
                }
            }
            // now export the triggers (needs to be done after the data because
            // triggers can modify already imported tables)
            if ($whatStrucOrData == 'structure'
                || $whatStrucOrData == 'structure_and_data'
            ) {
                if (! $export_plugin->exportStructure(
                    $db, $table, $crlf, $err_url,
                    'triggers', $export_type,
                    $do_relation, $do_comments, $do_mime, $do_dates
                )) {
                    break 2;
                }
            }
            if (! $export_plugin->exportDBFooter($db)) {
                break;
            }
        }
        if (! $export_plugin->exportFooter()) {
            break;
        }

    } while (false);
    // End of fake loop

    if ($save_on_server && ! empty($message)) {
        if ($export_type == 'server') {
            $active_page = 'server_export.php';
            include 'server_export.php';
        } elseif ($export_type == 'database') {
            $active_page = 'db_export.php';
            include 'db_export.php';
        } else {
            $active_page = 'tbl_export.php';
            include 'tbl_export.php';
        }
        exit();
    }

    /**
     * Send the dump as a file...
     */
    if (! empty($asfile)) {
        // Convert the charset if required.
        if ($output_charset_conversion) {
            $dump_buffer = PMA_convertString(
                'utf-8',
                $GLOBALS['charset_of_file'],
                $dump_buffer
            );
        }

        // Compression needed?
        if ($compression) {
            $dump_buffer = PMA_compressExport($dump_buffer, $compression);
        }

        /* If we saved on server, we have to close file now */
        if ($save_on_server) {
            $message = PMA_closeExportFile(
                $file_handle, $dump_buffer, $save_filename
            );
            if ($export_type == 'server') {
                $active_page = 'server_export.php';
                include_once 'server_export.php';
            } elseif ($export_type == 'database') {
                $active_page = 'db_export.php';
                include_once 'db_export.php';
            } else {
                $active_page = 'tbl_export.php';
                include_once 'tbl_export.php';
            }
            exit();
        } else {
            echo $dump_buffer;
        }
    } else {
        echo PMA_getHtmlForDisplayedExportFooter($back_button);
    } // end if
}
?>