123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342 |
- <?php
- if (!defined('SMF'))
- die('Hacking attempt...');
- 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_version' => 'smf_db_get_version',
- );
- }
- function smf_db_backup_table($table, $backup_table)
- {
- global $smcFunc, $db_prefix;
- $table = str_replace('{db_prefix}', $db_prefix, $table);
-
- $tables = smf_db_list_tables(false, $backup_table);
- if (!empty($tables))
- $smcFunc['db_query']('', '
- DROP TABLE {raw:backup_table}',
- array(
- 'backup_table' => $backup_table,
- )
- );
-
- $smcFunc['db_query']('', '
- CREATE TABLE {raw:backup_table}
- (
- LIKE {raw:table}
- INCLUDING DEFAULTS
- )',
- array(
- 'backup_table' => $backup_table,
- 'table' => $table,
- )
- );
- $smcFunc['db_query']('', '
- INSERT INTO {raw:backup_table}
- SELECT * FROM {raw:table}',
- array(
- 'backup_table' => $backup_table,
- 'table' => $table,
- )
- );
- }
- function smf_db_optimize_table($table)
- {
- global $smcFunc, $db_prefix;
- $table = str_replace('{db_prefix}', $db_prefix, $table);
- $request = $smcFunc['db_query']('', '
- VACUUM ANALYZE {raw:table}',
- array(
- 'table' => $table,
- )
- );
- if (!$request)
- return -1;
- $row = $smcFunc['db_fetch_assoc']($request);
- $smcFunc['db_free_result']($request);
- if (isset($row['Data_free']))
- return $row['Data_free'] / 1024;
- else
- return 0;
- }
- function smf_db_list_tables($db = false, $filter = false)
- {
- global $smcFunc;
- $request = $smcFunc['db_query']('', '
- SELECT tablename
- FROM pg_tables
- WHERE schemaname = {string:schema_public}' . ($filter == false ? '' : '
- AND tablename LIKE {string:filter}') . '
- ORDER BY tablename',
- array(
- 'schema_public' => 'public',
- 'filter' => $filter,
- )
- );
- $tables = array();
- while ($row = $smcFunc['db_fetch_row']($request))
- $tables[] = $row[0];
- $smcFunc['db_free_result']($request);
- return $tables;
- }
- function smf_db_insert_sql($tableName, $new_table = false)
- {
- global $smcFunc, $db_prefix;
- static $start = 0, $num_rows, $fields, $limit;
- if ($new_table)
- {
- $limit = strstr($tableName, 'log_') !== false ? 500 : 250;
- $start = 0;
- }
- $data = '';
- $tableName = str_replace('{db_prefix}', $db_prefix, $tableName);
-
- $crlf = "\r\n";
- $result = $smcFunc['db_query']('', '
- SELECT *
- FROM ' . $tableName . '
- LIMIT ' . $start . ', ' . $limit,
- array(
- 'security_override' => true,
- )
- );
-
- $num_rows = $smcFunc['db_num_rows']($result);
- if ($num_rows == 0)
- return '';
- if ($new_table)
- {
- $fields = array_keys($smcFunc['db_fetch_assoc']($result));
- $smcFunc['db_data_seek']($result, 0);
- }
-
- $data = '';
- $insert_msg = $crlf . 'INSERT INTO ' . $tableName . $crlf . "\t" . '(' . implode(', ', $fields) . ')' . $crlf . 'VALUES ' . $crlf . "\t";
-
- while ($row = $smcFunc['db_fetch_assoc']($result))
- {
-
- $field_list = array();
- foreach ($row as $key => $item)
- {
-
- if (!isset($item))
- $field_list[] = 'NULL';
- elseif (is_numeric($item) && (int) $item == $item)
- $field_list[] = $item;
- else
- $field_list[] = '\'' . $smcFunc['db_escape_string']($item) . '\'';
- }
-
- $data .= $insert_msg . '(' . implode(', ', $field_list) . ');' . $crlf;
- }
- $smcFunc['db_free_result']($result);
- $data .= $crlf;
- $start += $limit;
- return $data;
- }
- function smf_db_table_sql($tableName)
- {
- global $smcFunc, $db_prefix;
- $tableName = str_replace('{db_prefix}', $db_prefix, $tableName);
-
- $crlf = "\r\n";
-
- $schema_create = 'CREATE TABLE ' . $tableName . ' (' . $crlf;
- $index_create = '';
- $seq_create = '';
-
- $result = $smcFunc['db_query']('', '
- SELECT column_name, column_default, is_nullable, data_type, character_maximum_length
- FROM information_schema.columns
- WHERE table_name = {string:table}
- ORDER BY ordinal_position',
- array(
- 'table' => $tableName,
- )
- );
- while ($row = $smcFunc['db_fetch_assoc']($result))
- {
- if ($row['data_type'] == 'character varying')
- $row['data_type'] = 'varchar';
- elseif ($row['data_type'] == 'character')
- $row['data_type'] = 'char';
- if ($row['character_maximum_length'])
- $row['data_type'] .= '(' . $row['character_maximum_length'] . ')';
-
- $schema_create .= ' "' . $row['column_name'] . '" ' . $row['data_type'] . ($row['is_nullable'] != 'YES' ? ' NOT NULL' : '');
-
- if (trim($row['column_default']) != '')
- {
- $schema_create .= ' default ' . $row['column_default'] . '';
-
- if (preg_match('~nextval\(\'(.+?)\'(.+?)*\)~i', $row['column_default'], $matches) != 0)
- {
-
- $count_req = $smcFunc['db_query']('', '
- SELECT MAX("{raw:column}")
- FROM {raw:table}',
- array(
- 'column' => $row['column_name'],
- 'table' => $tableName,
- )
- );
- list ($max_ind) = $smcFunc['db_fetch_row']($count_req);
- $smcFunc['db_free_result']($count_req);
-
- $seq_create .= 'CREATE SEQUENCE ' . $matches[1] . ' START WITH ' . ($max_ind + 1) . ';' . $crlf . $crlf;
- }
- }
- $schema_create .= ',' . $crlf;
- }
- $smcFunc['db_free_result']($result);
-
- $schema_create = substr($schema_create, 0, -strlen($crlf) - 1);
- $result = $smcFunc['db_query']('', '
- SELECT CASE WHEN i.indisprimary THEN 1 ELSE 0 END AS is_primary, pg_get_indexdef(i.indexrelid) AS inddef
- FROM pg_class AS c
- INNER JOIN pg_index AS i ON (i.indrelid = c.oid)
- INNER JOIN pg_class AS c2 ON (c2.oid = i.indexrelid)
- WHERE c.relname = {string:table}',
- array(
- 'table' => $tableName,
- )
- );
- $indexes = array();
- while ($row = $smcFunc['db_fetch_assoc']($result))
- {
- if ($row['is_primary'])
- {
- if (preg_match('~\(([^\)]+?)\)~i', $row['inddef'], $matches) == 0)
- continue;
- $index_create .= $crlf . 'ALTER TABLE ' . $tableName . ' ADD PRIMARY KEY ("' . $matches[1] . '");';
- }
- else
- $index_create .= $crlf . $row['inddef'] . ';';
- }
- $smcFunc['db_free_result']($result);
-
- $schema_create .= $crlf . ');';
- return $seq_create . $schema_create . $index_create;
- }
- function smf_db_get_version()
- {
- global $smcFunc;
- $request = $smcFunc['db_query']('', '
- SHOW server_version',
- array(
- )
- );
- list ($ver) = $smcFunc['db_fetch_row']($request);
- $smcFunc['db_free_result']($request);
- return $ver;
- }
|