Current File : /home/admin/public_html/old/happyezystyle/board/Sources/DbExtra-sqlite.php
<?php
/**********************************************************************************
* DbExtra-sqlite.php *
***********************************************************************************
* SMF: Simple Machines Forum *
* Open-Source Project Inspired by Zef Hemel (zef@zefhemel.com) *
* =============================================================================== *
* Software Version: SMF 2.0 RC1 *
* Software by: Simple Machines (http://www.simplemachines.org) *
* Copyright 2006-2009 by: Simple Machines LLC (http://www.simplemachines.org) *
* 2001-2006 by: Lewis Media (http://www.lewismedia.com) *
* Support, News, Updates at: http://www.simplemachines.org *
***********************************************************************************
* This program is free software; you may redistribute it and/or modify it under *
* the terms of the provided license as published by Simple Machines LLC. *
* *
* This program is distributed in the hope that it is and will be useful, but *
* WITHOUT ANY WARRANTIES; without even any implied warranty of MERCHANTABILITY *
* or FITNESS FOR A PARTICULAR PURPOSE. *
* *
* See the "license.txt" file for details of the Simple Machines license. *
* The latest version can always be found at http://www.simplemachines.org. *
**********************************************************************************/
if (!defined('SMF'))
die('Hacking attempt...');
/* This file contains rarely used extended database functionality.
// !!!
string db_insert_sql(string table_name)
- gets all the necessary INSERTs for the table named table_name.
- goes in 250 row segments.
- returns the query to insert the data back in.
- returns an empty string if the table was empty.
string db_table_sql(string table_name)
- dumps the CREATE for the specified table. (by table_name.)
- returns the CREATE statement.
*/
// Add the file functions to the $smcFunc array.
function db_extra_init()
{
global $smcFunc;
if (!isset($smcFunc['db_backup_table']) || $smcFunc['db_backup_table'] != 'smf_db_backup_table')
$smcFunc += array(
'db_backup_table' => 'smf_db_backup_table',
'db_optimize_table' => 'smf_db_optimize_table',
'db_insert_sql' => 'smf_db_insert_sql',
'db_table_sql' => 'smf_db_table_sql',
'db_list_tables' => 'smf_db_list_tables',
'db_get_backup' => 'smf_db_get_backup',
'db_get_version' => 'smf_db_get_version',
);
}
// Backup $table to $backup_table.
function smf_db_backup_table($table, $backup_table)
{
global $smcFunc, $db_prefix;
$table = str_replace('{db_prefix}', $db_prefix, $table);
$result = $smcFunc['db_query']('', '
SHOW CREATE TABLE {raw:table}',
array(
'table' => $table,
)
);
list (, $create) = $smcFunc['db_fetch_row']($result);
$smcFunc['db_free_result']($result);
$create = preg_split('/[\n\r]/', $create);
$auto_inc = '';
// Default engine type.
$engine = 'MyISAM';
$charset = '';
$collate = '';
foreach ($create as $k => $l)
{
// Get the name of the auto_increment column.
if (strpos($l, 'auto_increment'))
$auto_inc = trim($l);
// For the engine type, see if we can work out what it is.
if (strpos($l, 'ENGINE') !== false || strpos($l, 'TYPE') !== false)
{
// Extract the engine type.
preg_match('~(ENGINE|TYPE)=(\w+)(\sDEFAULT)?(\sCHARSET=(\w+))?(\sCOLLATE=(\w+))?~', $l, $match);
if (!empty($match[1]))
$engine = $match[1];
if (!empty($match[2]))
$engine = $match[2];
if (!empty($match[5]))
$charset = $match[5];
if (!empty($match[7]))
$collate = $match[7];
}
// Skip everything but keys...
if (strpos($l, 'KEY') === false)
unset($create[$k]);
}
if (!empty($create))
$create = '(
' . implode('
', $create) . ')';
else
$create = '';
$smcFunc['db_query']('', '
DROP TABLE IF EXISTS {raw:backup_table}',
array(
'backup_table' => $backup_table,
)
);
$request = $smcFunc['db_query']('', '
CREATE TABLE {raw:backup_table} {raw:create}
TYPE={raw:engine}' . (empty($charset) ? '' : ' CHARACTER SET {raw:charset}' . (empty($collate) ? '' : ' COLLATE {raw:collate}')) . '
SELECT *
FROM {raw:table}',
array(
'backup_table' => $backup_table,
'table' => $table,
'create' => $create,
'engine' => $engine,
'charset' => empty($charset) ? '' : $charset,
'collate' => empty($collate) ? '' : $collate,
)
);
if ($auto_inc != '')
{
if (preg_match('~\`(.+?)\`\s~', $auto_inc, $match) != 0 && substr($auto_inc, -1, 1) == ',')
$auto_inc = substr($auto_inc, 0, -1);
$smcFunc['db_query']('', '
ALTER TABLE {raw:backup_table}
CHANGE COLUMN {raw:column_detail} {raw:auto_inc}',
array(
'backup_table' => $backup_table,
'column_detail' => $match[1],
'auto_inc' => $auto_inc,
)
);
}
return $request;
}
// Optimize a table - return data freed!
function smf_db_optimize_table($table)
{
global $smcFunc, $db_prefix;
$table = str_replace('{db_prefix}', $db_prefix, $table);
$request = $smcFunc['db_query']('', '
VACUUM {raw:table}',
array(
'table' => $table,
)
);
if (!$request)
return -1;
$row = $smcFunc['db_fetch_assoc']($request);
$smcFunc['db_free_result']($request);
// The function returns nothing.
return 0;
}
// List all the tables in the database.
function smf_db_list_tables($db = false, $filter = false)
{
global $smcFunc;
$filter = $filter == false ? '' : ' AND name LIKE \'' . $filter . '\'';
$request = $smcFunc['db_query']('', '
SELECT name
FROM sqlite_master
WHERE type = {string:type}
{raw:filter}
ORDER BY name',
array(
'type' => 'table',
'filter' => $filter,
)
);
$tables = array();
while ($row = $smcFunc['db_fetch_row']($request))
$tables[] = $row[0];
$smcFunc['db_free_result']($request);
return $tables;
}
// Get the content (INSERTs) for a table.
function smf_db_insert_sql($tableName)
{
global $smcFunc, $db_prefix;
$tableName = str_replace('{db_prefix}', $db_prefix, $tableName);
// This will be handy...
$crlf = "\r\n";
// Get everything from the table.
$result = $smcFunc['db_query']('', '
SELECT *
FROM {raw:table}',
array(
'table' => $tableName,
)
);
// The number of rows, just for record keeping and breaking INSERTs up.
$num_rows = $smcFunc['db_num_rows']($result);
$current_row = 0;
if ($num_rows == 0)
return '';
$fields = array_keys($smcFunc['db_fetch_assoc']($result));
// SQLite fetches an array so we need to filter out the numberic index for the columns.
foreach ($fields as $key => $name)
if (is_numeric($name))
unset($fields[$key]);
$smcFunc['db_data_seek']($result, 0);
// Start it off with the basic INSERT INTO.
$data = 'BEGIN TRANSACTION;' . $crlf;
// Loop through each row.
while ($row = $smcFunc['db_fetch_row']($result))
{
$current_row++;
// Get the fields in this row...
$field_list = array();
for ($j = 0; $j < $smcFunc['db_num_fields']($result); $j++)
{
// Try to figure out the type of each field. (NULL, number, or 'string'.)
if (!isset($row[$j]))
$field_list[] = 'NULL';
elseif (is_numeric($row[$j]))
$field_list[] = $row[$j];
else
$field_list[] = '\'' . $smcFunc['db_escape_string']($row[$j]) . '\'';
}
$data .= 'INSERT INTO ' . $tableName . ' (' . implode(', ', $fields) . ') VALUES (' . implode(', ', $field_list) . ');' . $crlf;
}
$smcFunc['db_free_result']($result);
// Return an empty string if there were no rows.
return $num_rows == 0 ? '' : $data . 'COMMIT;' . $crlf;
}
// Get the schema (CREATE) for a table.
function smf_db_table_sql($tableName)
{
global $smcFunc, $db_prefix;
$tableName = str_replace('{db_prefix}', $db_prefix, $tableName);
// This will be needed...
$crlf = "\r\n";
// Start the create table...
$schema_create = '';
$index_create = '';
$seq_create = '';
// Lets get the create statement directly from SQLite.
$result = $smcFunc['db_query']('', '
SELECT sql
FROM sqlite_master
WHERE type = {string:type}
AND name = {string:table_name}',
array(
'type' => 'table',
'table_name' => $tableName,
)
);
list ($schema_create) = $smcFunc['db_fetch_row']($result);
$smcFunc['db_free_result']($result);
// Now the indexes.
$result = $smcFunc['db_query']('', '
SELECT sql
FROM sqlite_master
WHERE type = {string:type}
AND tbl_name = {string:table_name}',
array(
'type' => 'index',
'table_name' => $tableName,
)
);
$indexes = array();
while ($row = $smcFunc['db_fetch_assoc']($result))
if (trim($row['sql']) != '')
$indexes[] = $row['sql'];
$smcFunc['db_free_result']($result);
$index_create .= implode(';' . $crlf, $indexes);
$schema_create = empty($indexes) ? rtrim($schema_create) : $schema_create . ';' . $crlf . $crlf;
return $seq_create . $schema_create . $index_create;
}
// Get the version number.
function smf_db_get_version()
{
return sqlite_libversion();
}
// Simple return the database - and die!
function smf_db_get_backup()
{
global $db_name;
$db_file = substr($db_name, -3) === '.db' ? $db_name : $db_name . '.db';
// Add more info if zipped...
$ext = '';
if (isset($_REQUEST['compress']) && function_exists('gzencode'))
$ext = '.gz';
// Do the remaining headers.
header('Content-Disposition: attachment; filename="' . $db_file . $ext . '"');
header('Cache-Control: private');
header('Connection: close');
// Literally dump the contents. Try reading the file first.
if (@readfile($db_file) == null)
echo file_get_contents($db_file);
obExit(false);
}
?>