+ /* @var $provider_collection \phpbb\auth\provider_collection */

+ /* @var $captcha_factory \phpbb\captcha\factory */

+ /** @var \fastImageSize\fastImageSize */

+ protected $imagesize;

+

+ * @param \fastImageSize\fastImageSize $imagesize fastImageSize class

+ public function __construct(\phpbb\config\config $config, \fastImageSize\fastImageSize $imagesize, $phpbb_root_path, $php_ext, \phpbb\path_helper $path_helper, \phpbb\cache\driver\driver_interface $cache = null)

+ $this->imagesize = $imagesize;

+ // Get image dimensions if they are not set

+ if ($row['avatar_width'] <= 0 || $row['avatar_height'] <= 0)

+ if (($row['avatar_width'] <= 0 || $row['avatar_height'] <= 0) && (($image_data = $this->imagesize->getImageSize($url)) === false))

+ if (!empty($image_data) && ($image_data['width'] <= 0 || $image_data['width'] <= 0))

+ $row['avatar_width'] = ($row['avatar_width'] && $row['avatar_height']) ? $row['avatar_width'] : $image_data['width'];

+ $row['avatar_height'] = ($row['avatar_width'] && $row['avatar_height']) ? $row['avatar_height'] : $image_data['height'];

+ $dims = $this->imagesize->getImageSize($file_path . '/' . $image);

+

+ if ($dims === false)

+ $dims = array(0, 0);

+ $dims = array($dims['width'], $dims['height']);

+ // Get image dimensions

+ if (($width <= 0 || $height <= 0) && (($image_data = $this->imagesize->getImageSize($url)) === false))

+ $error[] = 'UNABLE_GET_IMAGE_SIZE';

+ return false;

+ }

+ if (!empty($image_data) && ($image_data['width'] <= 0 || $image_data['height'] <= 0))

+ {

+ $error[] = 'AVATAR_NO_SIZE';

+ return false;

+ $width = ($width && $height) ? $width : $image_data['width'];

+ $height = ($width && $height) ? $height : $image_data['height'];

+

+ if (!empty($image_data) && (!isset($types[$image_data['type']]) || !in_array($extension, $types[$image_data['type']])))

+ if (!isset($types[$image_data['type']]))

+ $error[] = array('IMAGE_FILETYPE_MISMATCH', $types[$image_data['type']][0], $extension);

+ * @var \phpbb\filesystem\filesystem_interface

+ */

+ protected $filesystem;

+

+ /**

+ * @param \phpbb\filesystem\filesystem_interface phpBB filesystem helper

+ * @param \phpbb\path_helper $path_helper phpBB path helper

+ public function __construct(\phpbb\config\config $config, $phpbb_root_path, $php_ext, \phpbb\filesystem\filesystem_interface $filesystem, \phpbb\path_helper $path_helper, \phpbb\mimetype\guesser $mimetype_guesser, \phpbb\cache\driver\driver_interface $cache = null)

+ $this->filesystem = $filesystem;

+ $upload = new \fileupload($this->filesystem, 'AVATAR_', $this->allowed_extensions, $this->config['avatar_filesize'], $this->config['avatar_min_width'], $this->config['avatar_min_height'], $this->config['avatar_max_width'], $this->config['avatar_max_height'], (isset($this->config['mime_triggers']) ? explode('|', $this->config['mime_triggers']) : false));

+ return (file_exists($this->phpbb_root_path . $this->config['avatar_path']) && $this->filesystem->is_writable($this->phpbb_root_path . $this->config['avatar_path']) && (@ini_get('file_uploads') || strtolower(@ini_get('file_uploads')) == 'on'));

+ strpos($filename, 'url_generator') === 0 ||

+ $query_id = md5($query);

+ if (($result = $this->_read('sql_' . $query_id)) === false)

+ $this->sql_rowset[$query_id] = $result;

+ global $phpbb_filesystem;

+ if ($check && !$phpbb_filesystem->is_writable($this->cache_dir))

+* ACM dummy Caching

+class dummy extends \phpbb\cache\driver\base

+ global $config;

+

+ $config->set('cache_last_gc', time(), false);

+ global $config;

+

+ $config->set('cache_last_gc', time(), false);

+ * @var \phpbb\filesystem\filesystem_interface

+ */

+ protected $filesystem;

+

+ /**

+ global $phpbb_root_path, $phpbb_container;

+

+ $this->cache_dir = !is_null($cache_dir) ? $cache_dir : $phpbb_root_path . 'cache/' . $phpbb_container->getParameter('core.environment') . '/';

+ $this->filesystem = new \phpbb\filesystem\filesystem();

+

+ if (!is_dir($this->cache_dir))

+ {

+ @mkdir($this->cache_dir, 0777, true);

+ }

+ if (!$this->filesystem->is_writable($this->cache_dir))

+ global $config, $phpEx;

+ $config->set('cache_last_gc', time(), false);

+ $query_id = md5($query);

+ if ($this->_write('sql_' . $query_id, $this->sql_rowset[$query_id], $ttl + time(), $query))

+ try

+ $this->filesystem->phpbb_chmod($file, CHMOD_READ | CHMOD_WRITE);

+ }

+ catch (\phpbb\filesystem\exception\filesystem_exception $e)

+ {

+ // Do nothing

+ global $config;

+ // cache has auto GC, no need to have any code here :)

+ $config->set('cache_last_gc', time(), false);

+ $query_id = md5($query);

+ $temp[$query_id] = true;

+ $this->_write('sql_' . $query_id, $this->sql_rowset[$query_id], $ttl);

+ global $config, $db, $user, $request;

+ $this->confirm_id = $request->variable('confirm_id', '');

+ $this->confirm_code = $request->variable('confirm_code', '');

+ $refresh = $request->variable('refresh_vc', false) && $config['confirm_refresh'];

+ global $config, $user, $template, $request, $phpbb_admin_path, $phpEx;

+ $variables .= '&amp;' . rawurlencode($captcha_var) . '=' . $request->variable($captcha_var, (int) $config[$captcha_var]);

+ if (!$user->is_setup())

+ global $request;

+

+ if ($request->variable('confirm_code', false) && $this->solved === 0)

+ global $db, $user, $auth, $template, $phpbb_log, $request;

+ $submit = $request->variable('submit', '');

+ $value = $request->variable($captcha_var, 0);

+ $config->set($captcha_var, $value);

+ $phpbb_log->add('admin', $user->data['user_id'], $user->ip, 'LOG_CONFIG_VISUAL');

+ $var = (isset($_REQUEST[$captcha_var])) ? $request->variable($captcha_var, 0) : $config[$captcha_var];

+ global $config, $request;

+ $config->set($captcha_var, $request->variable($captcha_var, (int) $config[$captcha_var]));

+ global $config, $db, $user, $request;

+ $this->confirm_id = $request->variable('qa_confirm_id', '');

+ $this->answer = $request->variable('qa_answer', '', true);

+ global $phpbb_container;

+ $db_tool = $phpbb_container->get('dbal.tools');

+ global $phpbb_container;

+ $db_tool = $phpbb_container->get('dbal.tools');

+ foreach ($schemas as $table => $schema)

+ global $db, $request;

+ $answer = ($this->question_strict) ? $request->variable('qa_answer', '', true) : utf8_clean_string($request->variable('qa_answer', '', true));

+ global $request;

+

+ if ($request->variable('qa_answer', false) && $this->solved === 0)

+ global $config, $request, $phpbb_log, $template, $user;

+ $submit = $request->variable('submit', false);

+ $question_id = $request->variable('question_id', 0);

+ $action = $request->variable('action', '');

+ $phpbb_log->add('admin', $user->data['user_id'], $user->ip, 'LOG_CONFIG_VISUAL');

+ global $request;

+

+ $answers = $request->variable('answers', '', true);

+ 'question_text' => $request->variable('question_text', '', true),

+ 'strict' => $request->variable('strict', false),

+ 'lang_iso' => $request->variable('lang_iso', ''),

+ global $config, $db, $user, $request;

+ $this->challenge = $request->variable('recaptcha_challenge_field', '');

+ $this->response = $request->variable('recaptcha_response_field', '');

+ global $config, $db, $template, $user, $phpbb_log, $request;

+ $submit = $request->variable('submit', '');

+ $value = $request->variable($captcha_var, '');

+ $config->set($captcha_var, $value);

+ $phpbb_log->add('admin', $user->data['user_id'], $user->ip, 'LOG_CONFIG_VISUAL');

+ $var = (isset($_REQUEST[$captcha_var])) ? $request->variable($captcha_var, '') : ((isset($config[$captcha_var])) ? $config[$captcha_var] : '');

+ "php": ">=5.3.9"

+ /** @var \phpbb\filesystem\filesystem_interface */

+ protected $filesystem;

+

+ function __construct(\phpbb\user $user, \phpbb\db\migrator $migrator, \phpbb\extension\manager $extension_manager, \phpbb\config\config $config, \phpbb\cache\service $cache, \phpbb\log\log $log, \phpbb\filesystem\filesystem_interface $filesystem, $phpbb_root_path)

+ $this->filesystem = $filesystem;

+ $this->migrator->set_output_handler(new \phpbb\db\log_wrapper_migrator_output_handler($this->user, new console_migrator_output_handler($this->user, $output), $this->phpbb_root_path . 'store/migrations_' . time() . '.log', $this->filesystem));

+class exception extends \phpbb\exception\runtime_exception

+ /**

+ * phpBB router

+ * @var \phpbb\routing\router

+ */

+ protected $router;

+

+ * @var \phpbb\filesystem\filesystem_interface The filesystem object

+ * @param \phpbb\routing\router $router phpBB router

+ * @param \phpbb\filesystem\filesystem_interface $filesystem The filesystem object

+ public function __construct(\phpbb\template\template $template, \phpbb\user $user, \phpbb\config\config $config, \phpbb\routing\router $router, \phpbb\symfony_request $symfony_request, \phpbb\request\request_interface $request, \phpbb\filesystem\filesystem_interface $filesystem, $phpbb_root_path, $php_ext)

+ $this->router = $router;

+ $this->router->setContext($context);

+ $route_url = $this->router->generate($route, $params, $reference_type);

+ * Assigns automatic refresh time meta tag in template

+ *

+ * @param int $time time in seconds, when redirection should occur

+ * @param string $url the URL where the user should be redirected

+ * @return null

+ */

+ public function assign_meta_refresh_var($time, $url)

+ {

+ $this->template->assign_vars(array(

+ 'META' => '<meta http-equiv="refresh" content="' . $time . '; url=' . $url . '" />',

+ ));

+ }

+

+ /**

+ public function __construct(ContainerInterface $container, $phpbb_root_path, \phpbb\template\template $template = null)

+ throw new \phpbb\controller\exception('CONTROLLER_NOT_SPECIFIED');

+ throw new \phpbb\controller\exception('CONTROLLER_METHOD_NOT_SPECIFIED');

+ throw new \phpbb\controller\exception('CONTROLLER_SERVICE_UNDEFINED', array($service));

+ throw new \phpbb\controller\exception('CONTROLLER_ARGUMENT_VALUE_MISSING', array($param->getPosition() + 1, get_class($object) . ':' . $method, $param->name));

+ * If $forum_data is not given, forum id will be retrieved via $request->variable()

+ * If $forum_data is not given, forum id will be retrieved via $request->variable()

+ global $user, $phpbb_log;

+

+ $phpbb_log->add('critical', $user->data['user_id'], $user->ip, 'LOG_PLUPLOAD_TIDY_FAILED', false, array(

+ ));

+ if ($query_id)

+ if (!$query_id)

+ if (!$query_id)

+ if ($query_id)

+ $expression = str_replace(array('_', '%'), array("\_", "\%"), $expression);

+ $expression = str_replace(array(chr(0) . "\_", chr(0) . "\%"), array('_', '%'), $expression);

+ $expression = str_replace(array('_', '%'), array("\_", "\%"), $expression);

+ $expression = str_replace(array(chr(0) . "\_", chr(0) . "\%"), array('_', '%'), $expression);

+ $row = mssql_fetch_assoc($result_id);

+ mssql_free_result($result_id);

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ else if (strpos($query, 'SELECT') === 0 && $this->query_result !== true)

+ if (!$query_id || $query_id === true)

+ $row = mssql_fetch_assoc($query_id);

+ if ($query_id === true)

+ {

+ return false;

+ }

+

+ return ($query_id) ? @mssql_data_seek($query_id, $rownum) : false;

+ if ($row = mssql_fetch_assoc($result_id))

+ mssql_free_result($result_id);

+ mssql_free_result($result_id);

+ if ($query_id === true)

+ {

+ return false;

+ }

+

+ return mssql_free_result($query_id);

+ $row = mssql_fetch_assoc($result_id);

+ mssql_free_result($result_id);

+ $row = mssql_fetch_assoc($result_id);

+ mssql_free_result($result_id);

+ while ($row = mssql_fetch_row($result))

+ mssql_free_result($result);

+ if ($result)

+ while ($void = mssql_fetch_assoc($result))

+ {

+ // Take the time spent on parsing rows into account

+ }

+ mssql_free_result($result);

+ $row = odbc_fetch_array($result_id);

+ odbc_free_result($result_id);

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ else if (strpos($query, 'SELECT') === 0)

+ return ($query_id) ? odbc_fetch_array($query_id) : false;

+ if (odbc_fetch_array($result_id))

+ $id = odbc_result($result_id, 1);

+ odbc_free_result($result_id);

+ odbc_free_result($result_id);

+ return odbc_free_result($query_id);

+ if ($result)

+ while ($void = odbc_fetch_array($result))

+ {

+ // Take the time spent on parsing rows into account

+ }

+ odbc_free_result($result);

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ else if (strpos($query, 'SELECT') === 0)

+ if (!$query_id)

+ $row = sqlsrv_fetch_array($query_id, SQLSRV_FETCH_ASSOC);

+ if ($result_id)

+ $row = sqlsrv_fetch_array($result_id);

+ sqlsrv_free_stmt($result_id);

+ return sqlsrv_free_stmt($query_id);

+ sqlsrv_next_result($result);

+ while ($row = sqlsrv_fetch_array($result))

+ sqlsrv_free_stmt($result);

+ if ($result)

+ while ($void = sqlsrv_fetch_array($result))

+ {

+ // Take the time spent on parsing rows into account

+ }

+ sqlsrv_free_stmt($result);

+ if ($result)

+ {

+ $row = mysql_fetch_assoc($result);

+ mysql_free_result($result);

+ $modes = array_map('trim', explode(',', $row['sql_mode']));

+ }

+ else

+ {

+ $modes = array();

+ }

+ if ($result)

+ {

+ $row = mysql_fetch_assoc($result);

+ mysql_free_result($result);

+ $this->sql_server_version = $row['version'];

+ if (!empty($cache) && $use_cache)

+ {

+ $cache->put('mysql_version', $this->sql_server_version);

+ }

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ else if (strpos($query, 'SELECT') === 0)

+ return ($query_id) ? mysql_fetch_assoc($query_id) : false;

+ return mysql_free_result($query_id);

+ while ($row = mysql_fetch_assoc($result))

+ mysql_free_result($result);

+ while ($row = mysql_fetch_assoc($result))

+ mysql_free_result($result);

+ if ($result)

+ while ($void = mysql_fetch_assoc($result))

+ {

+ // Take the time spent on parsing rows into account

+ }

+ mysql_free_result($result);

+ if ($result)

+ $row = mysqli_fetch_assoc($result);

+ mysqli_free_result($result);

+ if ($result)

+ $row = mysqli_fetch_assoc($result);

+ mysqli_free_result($result);

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ if ($query_id)

+ $result = mysqli_fetch_assoc($query_id);

+ return ($query_id) ? @mysqli_data_seek($query_id, $rownum) : false;

+ if (!$query_id)

+ {

+ return false;

+ }

+

+ if ($query_id === true)

+ {

+ return true;

+ }

+

+ return mysqli_free_result($query_id);

+ while ($row = mysqli_fetch_assoc($result))

+ mysqli_free_result($result);

+ while ($row = mysqli_fetch_assoc($result))

+ mysqli_free_result($result);

+ if ($result)

+ while ($void = mysqli_fetch_assoc($result))

+ mysqli_free_result($result);

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ else if (strpos($query, 'SELECT') === 0)

+ if ($query_id)

+ $result = ocifetchinto($query_id, $row, OCI_ASSOC + OCI_RETURN_NULLS);

+ if (!$query_id)

+ if ($stmt)

+ {

+ $success = @ociexecute($stmt, OCI_DEFAULT);

+ if ($success)

+ {

+ $temp_result = ocifetchinto($stmt, $temp_array, OCI_ASSOC + OCI_RETURN_NULLS);

+ ocifreestatement($stmt);

+ if ($temp_result)

+ {

+ return $temp_array['CURRVAL'];

+ }

+ else

+ {

+ return false;

+ }

+ }

+ return ocifreestatement($query_id);

+ if ($result)

+ $success = @ociexecute($result, OCI_DEFAULT);

+ if ($success)

+ {

+ $row = array();

+

+ while (ocifetchinto($result, $row, OCI_ASSOC + OCI_RETURN_NULLS))

+ {

+ // Take the time spent on parsing rows into account

+ }

+ @ocifreestatement($result);

+ }

+ if ($query_id)

+ {

+ $row = pg_fetch_assoc($query_id, null);

+ pg_free_result($query_id);

+ $this->sql_server_version = (!empty($row['version'])) ? trim(substr($row['version'], 10)) : 0;

+ if (!empty($cache) && $use_cache)

+ {

+ $cache->put('pgsql_version', $this->sql_server_version);

+ }

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ else if (strpos($query, 'SELECT') === 0)

+ return ($query_id) ? pg_fetch_assoc($query_id, null) : false;

+ return ($query_id) ? @pg_result_seek($query_id, $rownum) : false;

+ $temp_result = pg_fetch_assoc($temp_q_id, null);

+ pg_free_result($query_id);

+ return pg_free_result($query_id);

+ while ($row = pg_fetch_assoc($result, null))

+ pg_free_result($result);

+ if ($result)

+ while ($void = pg_fetch_assoc($result, null))

+ {

+ // Take the time spent on parsing rows into account

+ }

+ pg_free_result($result);

+ if ($result)

+ {

+ $row = sqlite_fetch_array($result, SQLITE_ASSOC);

+ $this->sql_server_version = (!empty($row['version'])) ? $row['version'] : 0;

+ if (!empty($cache) && $use_cache)

+ {

+ $cache->put('sqlite_version', $this->sql_server_version);

+ }

+ if (!$this->query_result)

+ return false;

+

+ if ($cache && $cache_ttl)

+ $this->query_result = $cache->sql_save($this, $query, $this->query_result, $cache_ttl);

+ return ($query_id) ? sqlite_fetch_array($query_id, SQLITE_ASSOC) : false;

+ return ($query_id) ? @sqlite_seek($query_id, $rownum) : false;

+ if ($result)

+ while ($void = sqlite_fetch_array($result, SQLITE_ASSOC))

+ {

+ // Take the time spent on parsing rows into account

+ }

+ if (!$this->query_result)

+ {

+ return false;

+ }

+

+ if ($result)

+ while ($void = $result->fetchArray(SQLITE3_ASSOC))

+ {

+ // Take the time spent on parsing rows into account

+ }

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+use phpbb\db\extractor\exception\invalid_format_exception;

+use phpbb\db\extractor\exception\extractor_not_initialized_exception;

+

+/**

+ * Abstract base class for database extraction

+ */

+abstract class base_extractor implements extractor_interface

+{

+ /**

+ * @var string phpBB root path

+ */

+ protected $phpbb_root_path;

+

+ /**

+ * @var \phpbb\request\request_interface

+ */

+ protected $request;

+

+ /**

+ * @var \phpbb\db\driver\driver_interface

+ */

+ protected $db;

+

+ /**

+ * @var bool

+ */

+ protected $download;

+

+ /**

+ * @var bool

+ */

+ protected $store;

+

+ /**

+ * @var int

+ */

+ protected $time;

+

+ /**

+ * @var string

+ */

+ protected $format;

+

+ /**

+ * @var resource

+ */

+ protected $fp;

+

+ /**

+ * @var string

+ */

+ protected $write;

+

+ /**

+ * @var string

+ */

+ protected $close;

+

+ /**

+ * @var bool

+ */

+ protected $run_comp;

+

+ /**

+ * @var bool

+ */

+ protected $is_initialized;

+

+ /**

+ * Constructor

+ *

+ * @param string $phpbb_root_path

+ * @param \phpbb\request\request_interface $request

+ * @param \phpbb\db\driver\driver_interface $db

+ */

+ public function __construct($phpbb_root_path, \phpbb\request\request_interface $request, \phpbb\db\driver\driver_interface $db)

+ {

+ $this->phpbb_root_path = $phpbb_root_path;

+ $this->request = $request;

+ $this->db = $db;

+ $this->fp = null;

+

+ $this->is_initialized = false;

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function init_extractor($format, $filename, $time, $download = false, $store = false)

+ {

+ $this->download = $download;

+ $this->store = $store;

+ $this->time = $time;

+ $this->format = $format;

+

+ switch ($format)

+ {

+ case 'text':

+ $ext = '.sql';

+ $open = 'fopen';

+ $this->write = 'fwrite';

+ $this->close = 'fclose';

+ $mimetype = 'text/x-sql';

+ break;

+ case 'bzip2':

+ $ext = '.sql.bz2';

+ $open = 'bzopen';

+ $this->write = 'bzwrite';

+ $this->close = 'bzclose';

+ $mimetype = 'application/x-bzip2';

+ break;

+ case 'gzip':

+ $ext = '.sql.gz';

+ $open = 'gzopen';

+ $this->write = 'gzwrite';

+ $this->close = 'gzclose';

+ $mimetype = 'application/x-gzip';

+ break;

+ default:

+ throw new invalid_format_exception();

+ break;

+ }

+

+ if ($download === true)

+ {

+ $name = $filename . $ext;

+ header('Cache-Control: private, no-cache');

+ header("Content-Type: $mimetype; name=\"$name\"");

+ header("Content-disposition: attachment; filename=$name");

+

+ switch ($format)

+ {

+ case 'bzip2':

+ ob_start();

+ break;

+

+ case 'gzip':

+ if (strpos($this->request->header('Accept-Encoding'), 'gzip') !== false && strpos(strtolower($this->request->header('User-Agent')), 'msie') === false)

+ {

+ ob_start('ob_gzhandler');

+ }

+ else

+ {

+ $this->run_comp = true;

+ }

+ break;

+ }

+ }

+

+ if ($store === true)

+ {

+ $file = $this->phpbb_root_path . 'store/' . $filename . $ext;

+

+ $this->fp = $open($file, 'w');

+

+ if (!$this->fp)

+ {

+ trigger_error('FILE_WRITE_FAIL', E_USER_ERROR);

+ }

+ }

+

+ $this->is_initialized = true;

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_end()

+ {

+ static $close;

+

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ if ($this->store)

+ {

+ if ($close === null)

+ {

+ $close = $this->close;

+ }

+ $close($this->fp);

+ }

+

+ // bzip2 must be written all the way at the end

+ if ($this->download && $this->format === 'bzip2')

+ {

+ $c = ob_get_clean();

+ echo bzcompress($c);

+ }

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function flush($data)

+ {

+ static $write;

+

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ if ($this->store === true)

+ {

+ if ($write === null)

+ {

+ $write = $this->write;

+ }

+ $write($this->fp, $data);

+ }

+

+ if ($this->download === true)

+ {

+ if ($this->format === 'bzip2' || $this->format === 'text' || ($this->format === 'gzip' && !$this->run_comp))

+ {

+ echo $data;

+ }

+

+ // we can write the gzip data as soon as we get it

+ if ($this->format === 'gzip')

+ {

+ if ($this->run_comp)

+ {

+ echo gzencode($data);

+ }

+ else

+ {

+ ob_flush();

+ flush();

+ }

+ }

+ }

+ }

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor\exception;

+

+use phpbb\exception\runtime_exception;

+

+/**

+* This exception is thrown when invalid format is given to the extractor

+*/

+class extractor_not_initialized_exception extends runtime_exception

+{

+

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor\exception;

+

+/**

+* This exception is thrown when invalid format is given to the extractor

+*/

+class invalid_format_exception extends \InvalidArgumentException

+{

+

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+/**

+* Database extractor interface

+*/

+interface extractor_interface

+{

+ /**

+ * Start the extraction of the database

+ *

+ * This function initialize the database extraction. It is required to call this

+ * function before calling any other extractor functions.

+ *

+ * @param string $format

+ * @param string $filename

+ * @param int $time

+ * @param bool $download

+ * @param bool $store

+ * @return null

+ * @throws \phpbb\db\extractor\exception\invalid_format_exception when $format is invalid

+ */

+ public function init_extractor($format, $filename, $time, $download = false, $store = false);

+

+ /**

+ * Writes header comments to the database backup

+ *

+ * @param string $table_prefix prefix of phpBB database tables

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_start($table_prefix);

+

+ /**

+ * Closes file and/or dumps download data

+ *

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_end();

+

+ /**

+ * Extracts database table structure

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_table($table_name);

+

+ /**

+ * Extracts data from database table

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_data($table_name);

+

+ /**

+ * Writes data to file/download content

+ *

+ * @param string $data

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function flush($data);

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+/**

+* A factory which serves the suitable extractor instance for the given dbal

+*/

+class factory

+{

+ /**

+ * @var \phpbb\db\driver\driver_interface

+ */

+ protected $db;

+

+ /**

+ * @var \Symfony\Component\DependencyInjection\ContainerInterface

+ */

+ protected $container;

+

+ /**

+ * Extractor factory constructor

+ *

+ * @param \phpbb\db\driver\driver_interface $db

+ * @param \Symfony\Component\DependencyInjection\ContainerInterface $container

+ */

+ public function __construct(\phpbb\db\driver\driver_interface $db, \Symfony\Component\DependencyInjection\ContainerInterface $container)

+ {

+ $this->db = $db;

+ $this->container = $container;

+ }

+

+ /**

+ * DB extractor factory getter

+ *

+ * @return \phpbb\db\extractor\extractor_interface an appropriate instance of the database extractor for the used database driver

+ * @throws \InvalidArgumentException when the database driver is unknown

+ */

+ public function get()

+ {

+ // Return the appropriate DB extractor

+ if ($this->db instanceof \phpbb\db\driver\mssql || $this->db instanceof \phpbb\db\driver\mssql_base)

+ {

+ return $this->container->get('dbal.extractor.extractors.mssql_extractor');

+ }

+ else if ($this->db instanceof \phpbb\db\driver\mysql_base)

+ {

+ return $this->container->get('dbal.extractor.extractors.mysql_extractor');

+ }

+ else if ($this->db instanceof \phpbb\db\driver\oracle)

+ {

+ return $this->container->get('dbal.extractor.extractors.oracle_extractor');

+ }

+ else if ($this->db instanceof \phpbb\db\driver\postgres)

+ {

+ return $this->container->get('dbal.extractor.extractors.postgres_extractor');

+ }

+ else if ($this->db instanceof \phpbb\db\driver\sqlite)

+ {

+ return $this->container->get('dbal.extractor.extractors.sqlite_extractor');

+ }

+ else if ($this->db instanceof \phpbb\db\driver\sqlite3)

+ {

+ return $this->container->get('dbal.extractor.extractors.sqlite3_extractor');

+ }

+

+ throw new \InvalidArgumentException('Invalid database driver given');

+ }

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+use phpbb\db\extractor\exception\extractor_not_initialized_exception;

+

+class mssql_extractor extends base_extractor

+{

+ /**

+ * Writes closing line(s) to database backup

+ *

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_end()

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $this->flush("COMMIT\nGO\n");

+ parent::write_end();

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_start($table_prefix)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = "--\n";

+ $sql_data .= "-- phpBB Backup Script\n";

+ $sql_data .= "-- Dump of tables for $table_prefix\n";

+ $sql_data .= "-- DATE : " . gmdate("d-m-Y H:i:s", $this->time) . " GMT\n";

+ $sql_data .= "--\n";

+ $sql_data .= "BEGIN TRANSACTION\n";

+ $sql_data .= "GO\n";

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_table($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = '-- Table: ' . $table_name . "\n";

+ $sql_data .= "IF OBJECT_ID(N'$table_name', N'U') IS NOT NULL\n";

+ $sql_data .= "DROP TABLE $table_name;\n";

+ $sql_data .= "GO\n";

+ $sql_data .= "\nCREATE TABLE [$table_name] (\n";

+ $rows = array();

+

+ $text_flag = false;

+

+ $sql = "SELECT COLUMN_NAME, COLUMN_DEFAULT, IS_NULLABLE, DATA_TYPE, CHARACTER_MAXIMUM_LENGTH, COLUMNPROPERTY(object_id(TABLE_NAME), COLUMN_NAME, 'IsIdentity') as IS_IDENTITY

+ FROM INFORMATION_SCHEMA.COLUMNS

+ WHERE TABLE_NAME = '$table_name'";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $line = "\t[{$row['COLUMN_NAME']}] [{$row['DATA_TYPE']}]";

+

+ if ($row['DATA_TYPE'] == 'text')

+ {

+ $text_flag = true;

+ }

+

+ if ($row['IS_IDENTITY'])

+ {

+ $line .= ' IDENTITY (1 , 1)';

+ }

+

+ if ($row['CHARACTER_MAXIMUM_LENGTH'] && $row['DATA_TYPE'] !== 'text')

+ {

+ $line .= ' (' . $row['CHARACTER_MAXIMUM_LENGTH'] . ')';

+ }

+

+ if ($row['IS_NULLABLE'] == 'YES')

+ {

+ $line .= ' NULL';

+ }

+ else

+ {

+ $line .= ' NOT NULL';

+ }

+

+ if ($row['COLUMN_DEFAULT'])

+ {

+ $line .= ' DEFAULT ' . $row['COLUMN_DEFAULT'];

+ }

+

+ $rows[] = $line;

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql_data .= implode(",\n", $rows);

+ $sql_data .= "\n) ON [PRIMARY]";

+

+ if ($text_flag)

+ {

+ $sql_data .= " TEXTIMAGE_ON [PRIMARY]";

+ }

+

+ $sql_data .= "\nGO\n\n";

+ $rows = array();

+

+ $sql = "SELECT CONSTRAINT_NAME, COLUMN_NAME

+ FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE

+ WHERE TABLE_NAME = '$table_name'";

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if (!sizeof($rows))

+ {

+ $sql_data .= "ALTER TABLE [$table_name] WITH NOCHECK ADD\n";

+ $sql_data .= "\tCONSTRAINT [{$row['CONSTRAINT_NAME']}] PRIMARY KEY CLUSTERED \n\t(\n";

+ }

+ $rows[] = "\t\t[{$row['COLUMN_NAME']}]";

+ }

+ if (sizeof($rows))

+ {

+ $sql_data .= implode(",\n", $rows);

+ $sql_data .= "\n\t) ON [PRIMARY] \nGO\n";

+ }

+ $this->db->sql_freeresult($result);

+

+ $index = array();

+ $sql = "EXEC sp_statistics '$table_name'";

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if ($row['TYPE'] == 3)

+ {

+ $index[$row['INDEX_NAME']][] = '[' . $row['COLUMN_NAME'] . ']';

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ foreach ($index as $index_name => $column_name)

+ {

+ $index[$index_name] = implode(', ', $column_name);

+ }

+

+ foreach ($index as $index_name => $columns)

+ {

+ $sql_data .= "\nCREATE INDEX [$index_name] ON [$table_name]($columns) ON [PRIMARY]\nGO\n";

+ }

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_data($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ if ($this->db->get_sql_layer() === 'mssql')

+ {

+ $this->write_data_mssql($table_name);

+ }

+ else if($this->db->get_sql_layer() === 'mssqlnative')

+ {

+ $this->write_data_mssqlnative($table_name);

+ }

+ else

+ {

+ $this->write_data_odbc($table_name);

+ }

+ }

+

+ /**

+ * Extracts data from database table (for MSSQL driver)

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ protected function write_data_mssql($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $ary_type = $ary_name = array();

+ $ident_set = false;

+ $sql_data = '';

+

+ // Grab all of the data from current table.

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = $this->db->sql_query($sql);

+

+ $retrieved_data = mssql_num_rows($result);

+

+ $i_num_fields = mssql_num_fields($result);

+

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $ary_type[$i] = mssql_field_type($result, $i);

+ $ary_name[$i] = mssql_field_name($result, $i);

+ }

+

+ if ($retrieved_data)

+ {

+ $sql = "SELECT 1 as has_identity

+ FROM INFORMATION_SCHEMA.COLUMNS

+ WHERE COLUMNPROPERTY(object_id('$table_name'), COLUMN_NAME, 'IsIdentity') = 1";

+ $result2 = $this->db->sql_query($sql);

+ $row2 = $this->db->sql_fetchrow($result2);

+ if (!empty($row2['has_identity']))

+ {

+ $sql_data .= "\nSET IDENTITY_INSERT $table_name ON\nGO\n";

+ $ident_set = true;

+ }

+ $this->db->sql_freeresult($result2);

+ }

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $schema_vals = $schema_fields = array();

+

+ // Build the SQL statement to recreate the data.

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $str_val = $row[$ary_name[$i]];

+

+ if (preg_match('#char|text|bool|varbinary#i', $ary_type[$i]))

+ {

+ $str_quote = '';

+ $str_empty = "''";

+ $str_val = sanitize_data_mssql(str_replace("'", "''", $str_val));

+ }

+ else if (preg_match('#date|timestamp#i', $ary_type[$i]))

+ {

+ if (empty($str_val))

+ {

+ $str_quote = '';

+ }

+ else

+ {

+ $str_quote = "'";

+ }

+ }

+ else

+ {

+ $str_quote = '';

+ $str_empty = 'NULL';

+ }

+

+ if (empty($str_val) && $str_val !== '0' && !(is_int($str_val) || is_float($str_val)))

+ {

+ $str_val = $str_empty;

+ }

+

+ $schema_vals[$i] = $str_quote . $str_val . $str_quote;

+ $schema_fields[$i] = $ary_name[$i];

+ }

+

+ // Take the ordered fields and their associated data and build it

+ // into a valid sql statement to recreate that field in the data.

+ $sql_data .= "INSERT INTO $table_name (" . implode(', ', $schema_fields) . ') VALUES (' . implode(', ', $schema_vals) . ");\nGO\n";

+

+ $this->flush($sql_data);

+ $sql_data = '';

+ }

+ $this->db->sql_freeresult($result);

+

+ if ($retrieved_data && $ident_set)

+ {

+ $sql_data .= "\nSET IDENTITY_INSERT $table_name OFF\nGO\n";

+ }

+ $this->flush($sql_data);

+ }

+

+ /**

+ * Extracts data from database table (for MSSQL Native driver)

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ protected function write_data_mssqlnative($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $ary_type = $ary_name = array();

+ $ident_set = false;

+ $sql_data = '';

+

+ // Grab all of the data from current table.

+ $sql = "SELECT * FROM $table_name";

+ $this->db->mssqlnative_set_query_options(array('Scrollable' => SQLSRV_CURSOR_STATIC));

+ $result = $this->db->sql_query($sql);

+

+ $retrieved_data = $this->db->mssqlnative_num_rows($result);

+

+ if (!$retrieved_data)

+ {

+ $this->db->sql_freeresult($result);

+ return;

+ }

+

+ $sql = "SELECT COLUMN_NAME, DATA_TYPE

+ FROM INFORMATION_SCHEMA.COLUMNS

+ WHERE INFORMATION_SCHEMA.COLUMNS.TABLE_NAME = '" . $this->db->sql_escape($table_name) . "'";

+ $result_fields = $this->db->sql_query($sql);

+

+ $i_num_fields = 0;

+ while ($row = $this->db->sql_fetchrow($result_fields))

+ {

+ $ary_type[$i_num_fields] = $row['DATA_TYPE'];

+ $ary_name[$i_num_fields] = $row['COLUMN_NAME'];

+ $i_num_fields++;

+ }

+ $this->db->sql_freeresult($result_fields);

+

+ $sql = "SELECT 1 as has_identity

+ FROM INFORMATION_SCHEMA.COLUMNS

+ WHERE COLUMNPROPERTY(object_id('$table_name'), COLUMN_NAME, 'IsIdentity') = 1";

+ $result2 = $this->db->sql_query($sql);

+ $row2 = $this->db->sql_fetchrow($result2);

+

+ if (!empty($row2['has_identity']))

+ {

+ $sql_data .= "\nSET IDENTITY_INSERT $table_name ON\nGO\n";

+ $ident_set = true;

+ }

+ $this->db->sql_freeresult($result2);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $schema_vals = $schema_fields = array();

+

+ // Build the SQL statement to recreate the data.

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $str_val = $row[$ary_name[$i]];

+

+ // defaults to type number - better quote just to be safe, so check for is_int too

+ if (is_int($ary_type[$i]) || preg_match('#char|text|bool|varbinary#i', $ary_type[$i]))

+ {

+ $str_quote = '';

+ $str_empty = "''";

+ $str_val = sanitize_data_mssql(str_replace("'", "''", $str_val));

+ }

+ else if (preg_match('#date|timestamp#i', $ary_type[$i]))

+ {

+ if (empty($str_val))

+ {

+ $str_quote = '';

+ }

+ else

+ {

+ $str_quote = "'";

+ }

+ }

+ else

+ {

+ $str_quote = '';

+ $str_empty = 'NULL';

+ }

+

+ if (empty($str_val) && $str_val !== '0' && !(is_int($str_val) || is_float($str_val)))

+ {

+ $str_val = $str_empty;

+ }

+

+ $schema_vals[$i] = $str_quote . $str_val . $str_quote;

+ $schema_fields[$i] = $ary_name[$i];

+ }

+

+ // Take the ordered fields and their associated data and build it

+ // into a valid sql statement to recreate that field in the data.

+ $sql_data .= "INSERT INTO $table_name (" . implode(', ', $schema_fields) . ') VALUES (' . implode(', ', $schema_vals) . ");\nGO\n";

+

+ $this->flush($sql_data);

+ $sql_data = '';

+ }

+ $this->db->sql_freeresult($result);

+

+ if ($ident_set)

+ {

+ $sql_data .= "\nSET IDENTITY_INSERT $table_name OFF\nGO\n";

+ }

+ $this->flush($sql_data);

+ }

+

+ /**

+ * Extracts data from database table (for ODBC driver)

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ protected function write_data_odbc($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $ary_type = $ary_name = array();

+ $ident_set = false;

+ $sql_data = '';

+

+ // Grab all of the data from current table.

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = $this->db->sql_query($sql);

+

+ $retrieved_data = odbc_num_rows($result);

+

+ if ($retrieved_data)

+ {

+ $sql = "SELECT 1 as has_identity

+ FROM INFORMATION_SCHEMA.COLUMNS

+ WHERE COLUMNPROPERTY(object_id('$table_name'), COLUMN_NAME, 'IsIdentity') = 1";

+ $result2 = $this->db->sql_query($sql);

+ $row2 = $this->db->sql_fetchrow($result2);

+ if (!empty($row2['has_identity']))

+ {

+ $sql_data .= "\nSET IDENTITY_INSERT $table_name ON\nGO\n";

+ $ident_set = true;

+ }

+ $this->db->sql_freeresult($result2);

+ }

+

+ $i_num_fields = odbc_num_fields($result);

+

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $ary_type[$i] = odbc_field_type($result, $i + 1);

+ $ary_name[$i] = odbc_field_name($result, $i + 1);

+ }

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $schema_vals = $schema_fields = array();

+

+ // Build the SQL statement to recreate the data.

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $str_val = $row[$ary_name[$i]];

+

+ if (preg_match('#char|text|bool|varbinary#i', $ary_type[$i]))

+ {

+ $str_quote = '';

+ $str_empty = "''";

+ $str_val = sanitize_data_mssql(str_replace("'", "''", $str_val));

+ }

+ else if (preg_match('#date|timestamp#i', $ary_type[$i]))

+ {

+ if (empty($str_val))

+ {

+ $str_quote = '';

+ }

+ else

+ {

+ $str_quote = "'";

+ }

+ }

+ else

+ {

+ $str_quote = '';

+ $str_empty = 'NULL';

+ }

+

+ if (empty($str_val) && $str_val !== '0' && !(is_int($str_val) || is_float($str_val)))

+ {

+ $str_val = $str_empty;

+ }

+

+ $schema_vals[$i] = $str_quote . $str_val . $str_quote;

+ $schema_fields[$i] = $ary_name[$i];

+ }

+

+ // Take the ordered fields and their associated data and build it

+ // into a valid sql statement to recreate that field in the data.

+ $sql_data .= "INSERT INTO $table_name (" . implode(', ', $schema_fields) . ') VALUES (' . implode(', ', $schema_vals) . ");\nGO\n";

+

+ $this->flush($sql_data);

+

+ $sql_data = '';

+

+ }

+ $this->db->sql_freeresult($result);

+

+ if ($retrieved_data && $ident_set)

+ {

+ $sql_data .= "\nSET IDENTITY_INSERT $table_name OFF\nGO\n";

+ }

+ $this->flush($sql_data);

+ }

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+use phpbb\db\extractor\exception\extractor_not_initialized_exception;

+

+class mysql_extractor extends base_extractor

+{

+ /**

+ * {@inheritdoc}

+ */

+ public function write_start($table_prefix)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = "#\n";

+ $sql_data .= "# phpBB Backup Script\n";

+ $sql_data .= "# Dump of tables for $table_prefix\n";

+ $sql_data .= "# DATE : " . gmdate("d-m-Y H:i:s", $this->time) . " GMT\n";

+ $sql_data .= "#\n";

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_table($table_name)

+ {

+ static $new_extract;

+

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ if ($new_extract === null)

+ {

+ if ($this->db->get_sql_layer() === 'mysqli' || version_compare($this->db->sql_server_info(true), '3.23.20', '>='))

+ {

+ $new_extract = true;

+ }

+ else

+ {

+ $new_extract = false;

+ }

+ }

+

+ if ($new_extract)

+ {

+ $this->new_write_table($table_name);

+ }

+ else

+ {

+ $this->old_write_table($table_name);

+ }

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_data($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ if ($this->db->get_sql_layer() === 'mysqli')

+ {

+ $this->write_data_mysqli($table_name);

+ }

+ else

+ {

+ $this->write_data_mysql($table_name);

+ }

+ }

+

+ /**

+ * Extracts data from database table (for MySQLi driver)

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ protected function write_data_mysqli($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = mysqli_query($this->db->get_db_connect_id(), $sql, MYSQLI_USE_RESULT);

+ if ($result != false)

+ {

+ $fields_cnt = mysqli_num_fields($result);

+

+ // Get field information

+ $field = mysqli_fetch_fields($result);

+ $field_set = array();

+

+ for ($j = 0; $j < $fields_cnt; $j++)

+ {

+ $field_set[] = $field[$j]->name;

+ }

+

+ $search = array("\\", "'", "\x00", "\x0a", "\x0d", "\x1a", '"');

+ $replace = array("\\\\", "\\'", '\0', '\n', '\r', '\Z', '\\"');

+ $fields = implode(', ', $field_set);

+ $sql_data = 'INSERT INTO ' . $table_name . ' (' . $fields . ') VALUES ';

+ $first_set = true;

+ $query_len = 0;

+ $max_len = get_usable_memory();

+

+ while ($row = mysqli_fetch_row($result))

+ {

+ $values = array();

+ if ($first_set)

+ {

+ $query = $sql_data . '(';

+ }

+ else

+ {

+ $query .= ',(';

+ }

+

+ for ($j = 0; $j < $fields_cnt; $j++)

+ {

+ if (!isset($row[$j]) || is_null($row[$j]))

+ {

+ $values[$j] = 'NULL';

+ }

+ else if (($field[$j]->flags & 32768) && !($field[$j]->flags & 1024))

+ {

+ $values[$j] = $row[$j];

+ }

+ else

+ {

+ $values[$j] = "'" . str_replace($search, $replace, $row[$j]) . "'";

+ }

+ }

+ $query .= implode(', ', $values) . ')';

+

+ $query_len += strlen($query);

+ if ($query_len > $max_len)

+ {

+ $this->flush($query . ";\n\n");

+ $query = '';

+ $query_len = 0;

+ $first_set = true;

+ }

+ else

+ {

+ $first_set = false;

+ }

+ }

+ mysqli_free_result($result);

+

+ // check to make sure we have nothing left to flush

+ if (!$first_set && $query)

+ {

+ $this->flush($query . ";\n\n");

+ }

+ }

+ }

+

+ /**

+ * Extracts data from database table (for MySQL driver)

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ protected function write_data_mysql($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = mysql_unbuffered_query($sql, $this->db->get_db_connect_id());

+

+ if ($result != false)

+ {

+ $fields_cnt = mysql_num_fields($result);

+

+ // Get field information

+ $field = array();

+ for ($i = 0; $i < $fields_cnt; $i++)

+ {

+ $field[] = mysql_fetch_field($result, $i);

+ }

+ $field_set = array();

+

+ for ($j = 0; $j < $fields_cnt; $j++)

+ {

+ $field_set[] = $field[$j]->name;

+ }

+

+ $search = array("\\", "'", "\x00", "\x0a", "\x0d", "\x1a", '"');

+ $replace = array("\\\\", "\\'", '\0', '\n', '\r', '\Z', '\\"');

+ $fields = implode(', ', $field_set);

+ $sql_data = 'INSERT INTO ' . $table_name . ' (' . $fields . ') VALUES ';

+ $first_set = true;

+ $query_len = 0;

+ $max_len = get_usable_memory();

+

+ while ($row = mysql_fetch_row($result))

+ {

+ $values = array();

+ if ($first_set)

+ {

+ $query = $sql_data . '(';

+ }

+ else

+ {

+ $query .= ',(';

+ }

+

+ for ($j = 0; $j < $fields_cnt; $j++)

+ {

+ if (!isset($row[$j]) || is_null($row[$j]))

+ {

+ $values[$j] = 'NULL';

+ }

+ else if ($field[$j]->numeric && ($field[$j]->type !== 'timestamp'))

+ {

+ $values[$j] = $row[$j];

+ }

+ else

+ {

+ $values[$j] = "'" . str_replace($search, $replace, $row[$j]) . "'";

+ }

+ }

+ $query .= implode(', ', $values) . ')';

+

+ $query_len += strlen($query);

+ if ($query_len > $max_len)

+ {

+ $this->flush($query . ";\n\n");

+ $query = '';

+ $query_len = 0;

+ $first_set = true;

+ }

+ else

+ {

+ $first_set = false;

+ }

+ }

+ mysql_free_result($result);

+

+ // check to make sure we have nothing left to flush

+ if (!$first_set && $query)

+ {

+ $this->flush($query . ";\n\n");

+ }

+ }

+ }

+

+ /**

+ * Extracts database table structure (for MySQLi or MySQL 3.23.20+)

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ protected function new_write_table($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql = 'SHOW CREATE TABLE ' . $table_name;

+ $result = $this->db->sql_query($sql);

+ $row = $this->db->sql_fetchrow($result);

+

+ $sql_data = '# Table: ' . $table_name . "\n";

+ $sql_data .= "DROP TABLE IF EXISTS $table_name;\n";

+ $this->flush($sql_data . $row['Create Table'] . ";\n\n");

+

+ $this->db->sql_freeresult($result);

+ }

+

+ /**

+ * Extracts database table structure (for MySQL verisons older than 3.23.20)

+ *

+ * @param string $table_name name of the database table

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ protected function old_write_table($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = '# Table: ' . $table_name . "\n";

+ $sql_data .= "DROP TABLE IF EXISTS $table_name;\n";

+ $sql_data .= "CREATE TABLE $table_name(\n";

+ $rows = array();

+

+ $sql = "SHOW FIELDS

+ FROM $table_name";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $line = ' ' . $row['Field'] . ' ' . $row['Type'];

+

+ if (!is_null($row['Default']))

+ {

+ $line .= " DEFAULT '{$row['Default']}'";

+ }

+

+ if ($row['Null'] != 'YES')

+ {

+ $line .= ' NOT NULL';

+ }

+

+ if ($row['Extra'] != '')

+ {

+ $line .= ' ' . $row['Extra'];

+ }

+

+ $rows[] = $line;

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql = "SHOW KEYS

+ FROM $table_name";

+

+ $result = $this->db->sql_query($sql);

+

+ $index = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $kname = $row['Key_name'];

+

+ if ($kname != 'PRIMARY')

+ {

+ if ($row['Non_unique'] == 0)

+ {

+ $kname = "UNIQUE|$kname";

+ }

+ }

+

+ if ($row['Sub_part'])

+ {

+ $row['Column_name'] .= '(' . $row['Sub_part'] . ')';

+ }

+ $index[$kname][] = $row['Column_name'];

+ }

+ $this->db->sql_freeresult($result);

+

+ foreach ($index as $key => $columns)

+ {

+ $line = ' ';

+

+ if ($key == 'PRIMARY')

+ {

+ $line .= 'PRIMARY KEY (' . implode(', ', $columns) . ')';

+ }

+ else if (strpos($key, 'UNIQUE') === 0)

+ {

+ $line .= 'UNIQUE ' . substr($key, 7) . ' (' . implode(', ', $columns) . ')';

+ }

+ else if (strpos($key, 'FULLTEXT') === 0)

+ {

+ $line .= 'FULLTEXT ' . substr($key, 9) . ' (' . implode(', ', $columns) . ')';

+ }

+ else

+ {

+ $line .= "KEY $key (" . implode(', ', $columns) . ')';

+ }

+

+ $rows[] = $line;

+ }

+

+ $sql_data .= implode(",\n", $rows);

+ $sql_data .= "\n);\n\n";

+

+ $this->flush($sql_data);

+ }

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+use phpbb\db\extractor\exception\extractor_not_initialized_exception;

+

+class oracle_extractor extends base_extractor

+{

+ /**

+ * {@inheritdoc}

+ */

+ public function write_table($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = '-- Table: ' . $table_name . "\n";

+ $sql_data .= "DROP TABLE $table_name\n/\n";

+ $sql_data .= "\nCREATE TABLE $table_name (\n";

+

+ $sql = "SELECT COLUMN_NAME, DATA_TYPE, DATA_PRECISION, DATA_LENGTH, NULLABLE, DATA_DEFAULT

+ FROM ALL_TAB_COLS

+ WHERE table_name = '{$table_name}'";

+ $result = $this->db->sql_query($sql);

+

+ $rows = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $line = ' "' . $row['column_name'] . '" ' . $row['data_type'];

+

+ if ($row['data_type'] !== 'CLOB')

+ {

+ if ($row['data_type'] !== 'VARCHAR2' && $row['data_type'] !== 'CHAR')

+ {

+ $line .= '(' . $row['data_precision'] . ')';

+ }

+ else

+ {

+ $line .= '(' . $row['data_length'] . ')';

+ }

+ }

+

+ if (!empty($row['data_default']))

+ {

+ $line .= ' DEFAULT ' . $row['data_default'];

+ }

+

+ if ($row['nullable'] == 'N')

+ {

+ $line .= ' NOT NULL';

+ }

+ $rows[] = $line;

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql = "SELECT A.CONSTRAINT_NAME, A.COLUMN_NAME

+ FROM USER_CONS_COLUMNS A, USER_CONSTRAINTS B

+ WHERE A.CONSTRAINT_NAME = B.CONSTRAINT_NAME

+ AND B.CONSTRAINT_TYPE = 'P'

+ AND A.TABLE_NAME = '{$table_name}'";

+ $result = $this->db->sql_query($sql);

+

+ $primary_key = array();

+ $constraint_name = '';

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $constraint_name = '"' . $row['constraint_name'] . '"';

+ $primary_key[] = '"' . $row['column_name'] . '"';

+ }

+ $this->db->sql_freeresult($result);

+

+ if (sizeof($primary_key))

+ {

+ $rows[] = " CONSTRAINT {$constraint_name} PRIMARY KEY (" . implode(', ', $primary_key) . ')';

+ }

+

+ $sql = "SELECT A.CONSTRAINT_NAME, A.COLUMN_NAME

+ FROM USER_CONS_COLUMNS A, USER_CONSTRAINTS B

+ WHERE A.CONSTRAINT_NAME = B.CONSTRAINT_NAME

+ AND B.CONSTRAINT_TYPE = 'U'

+ AND A.TABLE_NAME = '{$table_name}'";

+ $result = $this->db->sql_query($sql);

+

+ $unique = array();

+ $constraint_name = '';

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $constraint_name = '"' . $row['constraint_name'] . '"';

+ $unique[] = '"' . $row['column_name'] . '"';

+ }

+ $this->db->sql_freeresult($result);

+

+ if (sizeof($unique))

+ {

+ $rows[] = " CONSTRAINT {$constraint_name} UNIQUE (" . implode(', ', $unique) . ')';

+ }

+

+ $sql_data .= implode(",\n", $rows);

+ $sql_data .= "\n)\n/\n";

+

+ $sql = "SELECT A.REFERENCED_NAME, C.*

+ FROM USER_DEPENDENCIES A, USER_TRIGGERS B, USER_SEQUENCES C

+ WHERE A.REFERENCED_TYPE = 'SEQUENCE'

+ AND A.NAME = B.TRIGGER_NAME

+ AND B.TABLE_NAME = '{$table_name}'

+ AND C.SEQUENCE_NAME = A.REFERENCED_NAME";

+ $result = $this->db->sql_query($sql);

+

+ $type = $this->request->variable('type', '');

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $sql_data .= "\nDROP SEQUENCE \"{$row['referenced_name']}\"\n/\n";

+ $sql_data .= "\nCREATE SEQUENCE \"{$row['referenced_name']}\"";

+

+ if ($type == 'full')

+ {

+ $sql_data .= ' START WITH ' . $row['last_number'];

+ }

+

+ $sql_data .= "\n/\n";

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql = "SELECT DESCRIPTION, WHEN_CLAUSE, TRIGGER_BODY

+ FROM USER_TRIGGERS

+ WHERE TABLE_NAME = '{$table_name}'";

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $sql_data .= "\nCREATE OR REPLACE TRIGGER {$row['description']}WHEN ({$row['when_clause']})\n{$row['trigger_body']}\n/\n";

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql = "SELECT A.INDEX_NAME, B.COLUMN_NAME

+ FROM USER_INDEXES A, USER_IND_COLUMNS B

+ WHERE A.UNIQUENESS = 'NONUNIQUE'

+ AND A.INDEX_NAME = B.INDEX_NAME

+ AND B.TABLE_NAME = '{$table_name}'";

+ $result = $this->db->sql_query($sql);

+

+ $index = array();

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $index[$row['index_name']][] = $row['column_name'];

+ }

+

+ foreach ($index as $index_name => $column_names)

+ {

+ $sql_data .= "\nCREATE INDEX $index_name ON $table_name(" . implode(', ', $column_names) . ")\n/\n";

+ }

+ $this->db->sql_freeresult($result);

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_data($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $ary_type = $ary_name = array();

+

+ // Grab all of the data from current table.

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = $this->db->sql_query($sql);

+

+ $i_num_fields = ocinumcols($result);

+

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $ary_type[$i] = ocicolumntype($result, $i + 1);

+ $ary_name[$i] = ocicolumnname($result, $i + 1);

+ }

+

+ $sql_data = '';

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $schema_vals = $schema_fields = array();

+

+ // Build the SQL statement to recreate the data.

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ // Oracle uses uppercase - we use lowercase

+ $str_val = $row[strtolower($ary_name[$i])];

+

+ if (preg_match('#char|text|bool|raw|clob#i', $ary_type[$i]))

+ {

+ $str_quote = '';

+ $str_empty = "''";

+ $str_val = sanitize_data_oracle($str_val);

+ }

+ else if (preg_match('#date|timestamp#i', $ary_type[$i]))

+ {

+ if (empty($str_val))

+ {

+ $str_quote = '';

+ }

+ else

+ {

+ $str_quote = "'";

+ }

+ }

+ else

+ {

+ $str_quote = '';

+ $str_empty = 'NULL';

+ }

+

+ if (empty($str_val) && $str_val !== '0')

+ {

+ $str_val = $str_empty;

+ }

+

+ $schema_vals[$i] = $str_quote . $str_val . $str_quote;

+ $schema_fields[$i] = '"' . $ary_name[$i] . '"';

+ }

+

+ // Take the ordered fields and their associated data and build it

+ // into a valid sql statement to recreate that field in the data.

+ $sql_data = "INSERT INTO $table_name (" . implode(', ', $schema_fields) . ') VALUES (' . implode(', ', $schema_vals) . ")\n/\n";

+

+ $this->flush($sql_data);

+ }

+ $this->db->sql_freeresult($result);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_start($table_prefix)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = "--\n";

+ $sql_data .= "-- phpBB Backup Script\n";

+ $sql_data .= "-- Dump of tables for $table_prefix\n";

+ $sql_data .= "-- DATE : " . gmdate("d-m-Y H:i:s", $this->time) . " GMT\n";

+ $sql_data .= "--\n";

+ $this->flush($sql_data);

+ }

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+use phpbb\db\extractor\exception\extractor_not_initialized_exception;

+

+class postgres_extractor extends base_extractor

+{

+ /**

+ * {@inheritdoc}

+ */

+ public function write_start($table_prefix)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = "--\n";

+ $sql_data .= "-- phpBB Backup Script\n";

+ $sql_data .= "-- Dump of tables for $table_prefix\n";

+ $sql_data .= "-- DATE : " . gmdate("d-m-Y H:i:s", $this->time) . " GMT\n";

+ $sql_data .= "--\n";

+ $sql_data .= "BEGIN TRANSACTION;\n";

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_table($table_name)

+ {

+ static $domains_created = array();

+

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql = "SELECT a.domain_name, a.data_type, a.character_maximum_length, a.domain_default

+ FROM INFORMATION_SCHEMA.domains a, INFORMATION_SCHEMA.column_domain_usage b

+ WHERE a.domain_name = b.domain_name

+ AND b.table_name = '{$table_name}'";

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if (empty($domains_created[$row['domain_name']]))

+ {

+ $domains_created[$row['domain_name']] = true;

+ //$sql_data = "DROP DOMAIN {$row['domain_name']};\n";

+ $sql_data = "CREATE DOMAIN {$row['domain_name']} as {$row['data_type']}";

+ if (!empty($row['character_maximum_length']))

+ {

+ $sql_data .= '(' . $row['character_maximum_length'] . ')';

+ }

+ $sql_data .= ' NOT NULL';

+ if (!empty($row['domain_default']))

+ {

+ $sql_data .= ' DEFAULT ' . $row['domain_default'];

+ }

+ $this->flush($sql_data . ";\n");

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql_data = '-- Table: ' . $table_name . "\n";

+ $sql_data .= "DROP TABLE $table_name;\n";

+ // PGSQL does not "tightly" bind sequences and tables, we must guess...

+ $sql = "SELECT relname

+ FROM pg_class

+ WHERE relkind = 'S'

+ AND relname = '{$table_name}_seq'";

+ $result = $this->db->sql_query($sql);

+ // We don't even care about storing the results. We already know the answer if we get rows back.

+ if ($this->db->sql_fetchrow($result))

+ {

+ $sql_data .= "DROP SEQUENCE {$table_name}_seq;\n";

+ $sql_data .= "CREATE SEQUENCE {$table_name}_seq;\n";

+ }

+ $this->db->sql_freeresult($result);

+

+ $field_query = "SELECT a.attnum, a.attname as field, t.typname as type, a.attlen as length, a.atttypmod as lengthvar, a.attnotnull as notnull

+ FROM pg_class c, pg_attribute a, pg_type t

+ WHERE c.relname = '" . $this->db->sql_escape($table_name) . "'

+ AND a.attnum > 0

+ AND a.attrelid = c.oid

+ AND a.atttypid = t.oid

+ ORDER BY a.attnum";

+ $result = $this->db->sql_query($field_query);

+

+ $sql_data .= "CREATE TABLE $table_name(\n";

+ $lines = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ // Get the data from the table

+ $sql_get_default = "SELECT pg_get_expr(d.adbin, d.adrelid) as rowdefault

+ FROM pg_attrdef d, pg_class c

+ WHERE (c.relname = '" . $this->db->sql_escape($table_name) . "')

+ AND (c.oid = d.adrelid)

+ AND d.adnum = " . $row['attnum'];

+ $def_res = $this->db->sql_query($sql_get_default);

+ $def_row = $this->db->sql_fetchrow($def_res);

+ $this->db->sql_freeresult($def_res);

+

+ if (empty($def_row))

+ {

+ unset($row['rowdefault']);

+ }

+ else

+ {

+ $row['rowdefault'] = $def_row['rowdefault'];

+ }

+

+ if ($row['type'] == 'bpchar')

+ {

+ // Internally stored as bpchar, but isn't accepted in a CREATE TABLE statement.

+ $row['type'] = 'char';

+ }

+

+ $line = ' ' . $row['field'] . ' ' . $row['type'];

+

+ if (strpos($row['type'], 'char') !== false)

+ {

+ if ($row['lengthvar'] > 0)

+ {

+ $line .= '(' . ($row['lengthvar'] - 4) . ')';

+ }

+ }

+

+ if (strpos($row['type'], 'numeric') !== false)

+ {

+ $line .= '(';

+ $line .= sprintf("%s,%s", (($row['lengthvar'] >> 16) & 0xffff), (($row['lengthvar'] - 4) & 0xffff));

+ $line .= ')';

+ }

+

+ if (isset($row['rowdefault']))

+ {

+ $line .= ' DEFAULT ' . $row['rowdefault'];

+ }

+

+ if ($row['notnull'] == 't')

+ {

+ $line .= ' NOT NULL';

+ }

+

+ $lines[] = $line;

+ }

+ $this->db->sql_freeresult($result);

+

+ // Get the listing of primary keys.

+ $sql_pri_keys = "SELECT ic.relname as index_name, bc.relname as tab_name, ta.attname as column_name, i.indisunique as unique_key, i.indisprimary as primary_key

+ FROM pg_class bc, pg_class ic, pg_index i, pg_attribute ta, pg_attribute ia

+ WHERE (bc.oid = i.indrelid)

+ AND (ic.oid = i.indexrelid)

+ AND (ia.attrelid = i.indexrelid)

+ AND (ta.attrelid = bc.oid)

+ AND (bc.relname = '" . $this->db->sql_escape($table_name) . "')

+ AND (ta.attrelid = i.indrelid)

+ AND (ta.attnum = i.indkey[ia.attnum-1])

+ ORDER BY index_name, tab_name, column_name";

+

+ $result = $this->db->sql_query($sql_pri_keys);

+

+ $index_create = $index_rows = $primary_key = array();

+

+ // We do this in two steps. It makes placing the comma easier

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if ($row['primary_key'] == 't')

+ {

+ $primary_key[] = $row['column_name'];

+ $primary_key_name = $row['index_name'];

+ }

+ else

+ {

+ // We have to store this all this info because it is possible to have a multi-column key...

+ // we can loop through it again and build the statement

+ $index_rows[$row['index_name']]['table'] = $table_name;

+ $index_rows[$row['index_name']]['unique'] = ($row['unique_key'] == 't') ? true : false;

+ $index_rows[$row['index_name']]['column_names'][] = $row['column_name'];

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ if (!empty($index_rows))

+ {

+ foreach ($index_rows as $idx_name => $props)

+ {

+ $index_create[] = 'CREATE ' . ($props['unique'] ? 'UNIQUE ' : '') . "INDEX $idx_name ON $table_name (" . implode(', ', $props['column_names']) . ");";

+ }

+ }

+

+ if (!empty($primary_key))

+ {

+ $lines[] = " CONSTRAINT $primary_key_name PRIMARY KEY (" . implode(', ', $primary_key) . ")";

+ }

+

+ // Generate constraint clauses for CHECK constraints

+ $sql_checks = "SELECT conname as index_name, consrc

+ FROM pg_constraint, pg_class bc

+ WHERE conrelid = bc.oid

+ AND bc.relname = '" . $this->db->sql_escape($table_name) . "'

+ AND NOT EXISTS (

+ SELECT *

+ FROM pg_constraint as c, pg_inherits as i

+ WHERE i.inhrelid = pg_constraint.conrelid

+ AND c.conname = pg_constraint.conname

+ AND c.consrc = pg_constraint.consrc

+ AND c.conrelid = i.inhparent

+ )";

+ $result = $this->db->sql_query($sql_checks);

+

+ // Add the constraints to the sql file.

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if (!is_null($row['consrc']))

+ {

+ $lines[] = ' CONSTRAINT ' . $row['index_name'] . ' CHECK ' . $row['consrc'];

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql_data .= implode(", \n", $lines);

+ $sql_data .= "\n);\n";

+

+ if (!empty($index_create))

+ {

+ $sql_data .= implode("\n", $index_create) . "\n\n";

+ }

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_data($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ // Grab all of the data from current table.

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = $this->db->sql_query($sql);

+

+ $i_num_fields = pg_num_fields($result);

+ $seq = '';

+

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $ary_type[] = pg_field_type($result, $i);

+ $ary_name[] = pg_field_name($result, $i);

+

+ $sql = "SELECT pg_get_expr(d.adbin, d.adrelid) as rowdefault

+ FROM pg_attrdef d, pg_class c

+ WHERE (c.relname = '{$table_name}')

+ AND (c.oid = d.adrelid)

+ AND d.adnum = " . strval($i + 1);

+ $result2 = $this->db->sql_query($sql);

+ if ($row = $this->db->sql_fetchrow($result2))

+ {

+ // Determine if we must reset the sequences

+ if (strpos($row['rowdefault'], "nextval('") === 0)

+ {

+ $seq .= "SELECT SETVAL('{$table_name}_seq',(select case when max({$ary_name[$i]})>0 then max({$ary_name[$i]})+1 else 1 end FROM {$table_name}));\n";

+ }

+ }

+ }

+

+ $this->flush("COPY $table_name (" . implode(', ', $ary_name) . ') FROM stdin;' . "\n");

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $schema_vals = array();

+

+ // Build the SQL statement to recreate the data.

+ for ($i = 0; $i < $i_num_fields; $i++)

+ {

+ $str_val = $row[$ary_name[$i]];

+

+ if (preg_match('#char|text|bool|bytea#i', $ary_type[$i]))

+ {

+ $str_val = str_replace(array("\n", "\t", "\r", "\b", "\f", "\v"), array('\n', '\t', '\r', '\b', '\f', '\v'), addslashes($str_val));

+ $str_empty = '';

+ }

+ else

+ {

+ $str_empty = '\N';

+ }

+

+ if (empty($str_val) && $str_val !== '0')

+ {

+ $str_val = $str_empty;

+ }

+

+ $schema_vals[] = $str_val;

+ }

+

+ // Take the ordered fields and their associated data and build it

+ // into a valid sql statement to recreate that field in the data.

+ $this->flush(implode("\t", $schema_vals) . "\n");

+ }

+ $this->db->sql_freeresult($result);

+ $this->flush("\\.\n");

+

+ // Write out the sequence statements

+ $this->flush($seq);

+ }

+

+ /**

+ * Writes closing line(s) to database backup

+ *

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_end()

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $this->flush("COMMIT;\n");

+ parent::write_end();

+ }

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+use phpbb\db\extractor\exception\extractor_not_initialized_exception;

+

+class sqlite3_extractor extends base_extractor

+{

+ /**

+ * {@inheritdoc}

+ */

+ public function write_start($table_prefix)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = "--\n";

+ $sql_data .= "-- phpBB Backup Script\n";

+ $sql_data .= "-- Dump of tables for $table_prefix\n";

+ $sql_data .= "-- DATE : " . gmdate("d-m-Y H:i:s", $this->time) . " GMT\n";

+ $sql_data .= "--\n";

+ $sql_data .= "BEGIN TRANSACTION;\n";

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_table($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = '-- Table: ' . $table_name . "\n";

+ $sql_data .= "DROP TABLE $table_name;\n";

+

+ $sql = "SELECT sql

+ FROM sqlite_master

+ WHERE type = 'table'

+ AND name = '" . $this->db->sql_escape($table_name) . "'

+ ORDER BY name ASC;";

+ $result = $this->db->sql_query($sql);

+ $row = $this->db->sql_fetchrow($result);

+ $this->db->sql_freeresult($result);

+

+ // Create Table

+ $sql_data .= $row['sql'] . ";\n";

+

+ $result = $this->db->sql_query("PRAGMA index_list('" . $this->db->sql_escape($table_name) . "');");

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if (strpos($row['name'], 'autoindex') !== false)

+ {

+ continue;

+ }

+

+ $result2 = $this->db->sql_query("PRAGMA index_info('" . $this->db->sql_escape($row['name']) . "');");

+

+ $fields = array();

+ while ($row2 = $this->db->sql_fetchrow($result2))

+ {

+ $fields[] = $row2['name'];

+ }

+ $this->db->sql_freeresult($result2);

+

+ $sql_data .= 'CREATE ' . ($row['unique'] ? 'UNIQUE ' : '') . 'INDEX ' . $row['name'] . ' ON ' . $table_name . ' (' . implode(', ', $fields) . ");\n";

+ }

+ $this->db->sql_freeresult($result);

+

+ $this->flush($sql_data . "\n");

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_data($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $result = $this->db->sql_query("PRAGMA table_info('" . $this->db->sql_escape($table_name) . "');");

+

+ $col_types = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $col_types[$row['name']] = $row['type'];

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql_insert = 'INSERT INTO ' . $table_name . ' (' . implode(', ', array_keys($col_types)) . ') VALUES (';

+

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ foreach ($row as $column_name => $column_data)

+ {

+ if (is_null($column_data))

+ {

+ $row[$column_name] = 'NULL';

+ }

+ else if ($column_data === '')

+ {

+ $row[$column_name] = "''";

+ }

+ else if (stripos($col_types[$column_name], 'text') !== false || stripos($col_types[$column_name], 'char') !== false || stripos($col_types[$column_name], 'blob') !== false)

+ {

+ $row[$column_name] = sanitize_data_generic(str_replace("'", "''", $column_data));

+ }

+ }

+ $this->flush($sql_insert . implode(', ', $row) . ");\n");

+ }

+ }

+

+ /**

+ * Writes closing line(s) to database backup

+ *

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_end()

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $this->flush("COMMIT;\n");

+ parent::write_end();

+ }

+}

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\extractor;

+

+use phpbb\db\extractor\exception\extractor_not_initialized_exception;

+

+class sqlite_extractor extends base_extractor

+{

+ /**

+ * {@inheritdoc}

+ */

+ public function write_start($table_prefix)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = "--\n";

+ $sql_data .= "-- phpBB Backup Script\n";

+ $sql_data .= "-- Dump of tables for $table_prefix\n";

+ $sql_data .= "-- DATE : " . gmdate("d-m-Y H:i:s", $this->time) . " GMT\n";

+ $sql_data .= "--\n";

+ $sql_data .= "BEGIN TRANSACTION;\n";

+ $this->flush($sql_data);

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_table($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $sql_data = '-- Table: ' . $table_name . "\n";

+ $sql_data .= "DROP TABLE $table_name;\n";

+

+ $sql = "SELECT sql

+ FROM sqlite_master

+ WHERE type = 'table'

+ AND name = '" . $this->db->sql_escape($table_name) . "'

+ ORDER BY type DESC, name;";

+ $result = $this->db->sql_query($sql);

+ $row = $this->db->sql_fetchrow($result);

+ $this->db->sql_freeresult($result);

+

+ // Create Table

+ $sql_data .= $row['sql'] . ";\n";

+

+ $result = $this->db->sql_query("PRAGMA index_list('" . $this->db->sql_escape($table_name) . "');");

+

+ $ar = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $ar[] = $row;

+ }

+ $this->db->sql_freeresult($result);

+

+ foreach ($ar as $value)

+ {

+ if (strpos($value['name'], 'autoindex') !== false)

+ {

+ continue;

+ }

+

+ $result = $this->db->sql_query("PRAGMA index_info('" . $this->db->sql_escape($value['name']) . "');");

+

+ $fields = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $fields[] = $row['name'];

+ }

+ $this->db->sql_freeresult($result);

+

+ $sql_data .= 'CREATE ' . ($value['unique'] ? 'UNIQUE ' : '') . 'INDEX ' . $value['name'] . ' on ' . $table_name . ' (' . implode(', ', $fields) . ");\n";

+ }

+

+ $this->flush($sql_data . "\n");

+ }

+

+ /**

+ * {@inheritdoc}

+ */

+ public function write_data($table_name)

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $col_types = sqlite_fetch_column_types($this->db->get_db_connect_id(), $table_name);

+

+ $sql = "SELECT *

+ FROM $table_name";

+ $result = sqlite_unbuffered_query($this->db->get_db_connect_id(), $sql);

+ $rows = sqlite_fetch_all($result, SQLITE_ASSOC);

+ $sql_insert = 'INSERT INTO ' . $table_name . ' (' . implode(', ', array_keys($col_types)) . ') VALUES (';

+ foreach ($rows as $row)

+ {

+ foreach ($row as $column_name => $column_data)

+ {

+ if (is_null($column_data))

+ {

+ $row[$column_name] = 'NULL';

+ }

+ else if ($column_data == '')

+ {

+ $row[$column_name] = "''";

+ }

+ else if (strpos($col_types[$column_name], 'text') !== false || strpos($col_types[$column_name], 'char') !== false || strpos($col_types[$column_name], 'blob') !== false)

+ {

+ $row[$column_name] = sanitize_data_generic(str_replace("'", "''", $column_data));

+ }

+ }

+ $this->flush($sql_insert . implode(', ', $row) . ");\n");

+ }

+ }

+

+ /**

+ * Writes closing line(s) to database backup

+ *

+ * @return null

+ * @throws \phpbb\db\extractor\exception\extractor_not_initialized_exception when calling this function before init_extractor()

+ */

+ public function write_end()

+ {

+ if (!$this->is_initialized)

+ {

+ throw new extractor_not_initialized_exception();

+ }

+

+ $this->flush("COMMIT;\n");

+ parent::write_end();

+ }

+}

+ * @var \phpbb\filesystem\filesystem_interface

+ */

+ protected $filesystem;

+

+ /**

+ * @param \phpbb\filesystem\filesystem_interface phpBB filesystem object

+ public function __construct(user $user, migrator_output_handler_interface $migrator, $log_file, \phpbb\filesystem\filesystem_interface $filesystem)

+ $this->filesystem = $filesystem;

+ if ($this->filesystem->is_writable(dirname($file)))

+ /* @var $passwords_manager \phpbb\passwords\manager */

+

+ // soon as the \phpbb\db\tools\tools class is capable of properly

+ global $config;

+

+ $config->set('default_style', $default_style);

+ /** @var \phpbb\db\tools\tools_interface */

+ * @param \phpbb\db\tools\tools_interface $db_tools

+ public function __construct(\phpbb\config\config $config, \phpbb\db\driver\driver_interface $db, \phpbb\db\tools\tools_interface $db_tools, $phpbb_root_path, $php_ext, $table_prefix)

+ /* @var $manager \phpbb\profilefields\manager */

+ /** @var \phpbb\db\tools\tools_interface */

+ public function __construct(array $class_names, \phpbb\config\config $config, \phpbb\db\driver\driver_interface $db, \phpbb\db\tools\tools_interface $db_tools, $phpbb_root_path, $php_ext, $table_prefix)

+ global $user, $phpbb_log;

+

+ $phpbb_log->add('admin', $user->data['user_id'], $user->ip, 'LOG_MODULE_ADD', false, array($module_log_name));

+ /** @var \phpbb\db\tools\tools_interface */

+ public function __construct(ContainerInterface $container, \phpbb\config\config $config, \phpbb\db\driver\driver_interface $db, \phpbb\db\tools\tools_interface $db_tools, $migrations_table, $phpbb_root_path, $php_ext, $table_prefix, $tools, \phpbb\db\migration\helper $helper)

+<?php

+/**

+*

+* This file is part of the phpBB Forum Software package.

+*

+* @copyright (c) phpBB Limited <https://www.phpbb.com>

+* @license GNU General Public License, version 2 (GPL-2.0)

+*

+* For full copyright and license information, please see

+* the docs/CREDITS.txt file.

+*

+*/

+

+namespace phpbb\db\tools;

+

+/**

+ * A factory which serves the suitable tools instance for the given dbal

+ */

+class factory

+{

+ /**

+ * @param mixed $db_driver

+ * @param bool $return_statements

+ * @return \phpbb\db\tools\tools_interface

+ */

+ public function get($db_driver, $return_statements = false)

+ {

+ if ($db_driver instanceof \phpbb\db\driver\mssql || $db_driver instanceof \phpbb\db\driver\mssql_base)

+ {

+ return new \phpbb\db\tools\mssql($db_driver, $return_statements);

+ }

+ else if ($db_driver instanceof \phpbb\db\driver\postgres)

+ {

+ return new \phpbb\db\tools\postgres($db_driver, $return_statements);

+ }

+ else if ($db_driver instanceof \phpbb\db\driver\driver_interface)

+ {

+ return new \phpbb\db\tools\tools($db_driver, $return_statements);

+ }

+

+ throw new \InvalidArgumentException('Invalid database driver given');

+ }

+}

+<?php

+/**

+ *

+ * This file is part of the phpBB Forum Software package.

+ *

+ * @copyright (c) phpBB Limited <https://www.phpbb.com>

+ * @license GNU General Public License, version 2 (GPL-2.0)

+ *

+ * For full copyright and license information, please see

+ * the docs/CREDITS.txt file.

+ *

+ */

+

+namespace phpbb\db\tools;

+

+/**

+ * Database Tools for handling cross-db actions such as altering columns, etc.

+ * Currently not supported is returning SQL for creating tables.

+ */

+class mssql extends tools

+{

+ /**

+ * Is the used MS SQL Server a SQL Server 2000?

+ * @var bool

+ */

+ protected $is_sql_server_2000;

+

+ /**

+ * Get the column types for mssql based databases

+ *

+ * @return array

+ */

+ public static function get_dbms_type_map()

+ {

+ return array(

+ 'mssql' => array(

+ 'INT:' => '[int]',

+ 'BINT' => '[float]',

+ 'UINT' => '[int]',

+ 'UINT:' => '[int]',

+ 'TINT:' => '[int]',

+ 'USINT' => '[int]',

+ 'BOOL' => '[int]',

+ 'VCHAR' => '[varchar] (255)',

+ 'VCHAR:' => '[varchar] (%d)',

+ 'CHAR:' => '[char] (%d)',

+ 'XSTEXT' => '[varchar] (1000)',

+ 'STEXT' => '[varchar] (3000)',

+ 'TEXT' => '[varchar] (8000)',

+ 'MTEXT' => '[text]',

+ 'XSTEXT_UNI'=> '[varchar] (100)',

+ 'STEXT_UNI' => '[varchar] (255)',

+ 'TEXT_UNI' => '[varchar] (4000)',

+ 'MTEXT_UNI' => '[text]',

+ 'TIMESTAMP' => '[int]',

+ 'DECIMAL' => '[float]',

+ 'DECIMAL:' => '[float]',

+ 'PDECIMAL' => '[float]',

+ 'PDECIMAL:' => '[float]',

+ 'VCHAR_UNI' => '[varchar] (255)',

+ 'VCHAR_UNI:'=> '[varchar] (%d)',

+ 'VCHAR_CI' => '[varchar] (255)',

+ 'VARBINARY' => '[varchar] (255)',

+ ),

+

+ 'mssqlnative' => array(

+ 'INT:' => '[int]',

+ 'BINT' => '[float]',

+ 'UINT' => '[int]',

+ 'UINT:' => '[int]',

+ 'TINT:' => '[int]',

+ 'USINT' => '[int]',

+ 'BOOL' => '[int]',

+ 'VCHAR' => '[varchar] (255)',

+ 'VCHAR:' => '[varchar] (%d)',

+ 'CHAR:' => '[char] (%d)',

+ 'XSTEXT' => '[varchar] (1000)',

+ 'STEXT' => '[varchar] (3000)',

+ 'TEXT' => '[varchar] (8000)',

+ 'MTEXT' => '[text]',

+ 'XSTEXT_UNI'=> '[varchar] (100)',

+ 'STEXT_UNI' => '[varchar] (255)',

+ 'TEXT_UNI' => '[varchar] (4000)',

+ 'MTEXT_UNI' => '[text]',

+ 'TIMESTAMP' => '[int]',

+ 'DECIMAL' => '[float]',

+ 'DECIMAL:' => '[float]',

+ 'PDECIMAL' => '[float]',

+ 'PDECIMAL:' => '[float]',

+ 'VCHAR_UNI' => '[varchar] (255)',

+ 'VCHAR_UNI:'=> '[varchar] (%d)',

+ 'VCHAR_CI' => '[varchar] (255)',

+ 'VARBINARY' => '[varchar] (255)',

+ ),

+ );

+ }

+

+ /**

+ * Constructor. Set DB Object and set {@link $return_statements return_statements}.

+ *

+ * @param \phpbb\db\driver\driver_interface $db Database connection

+ * @param bool $return_statements True if only statements should be returned and no SQL being executed

+ */

+ public function __construct(\phpbb\db\driver\driver_interface $db, $return_statements = false)

+ {

+ parent::__construct($db, $return_statements);

+

+ // Determine mapping database type

+ switch ($this->db->get_sql_layer())

+ {

+ case 'mssql':

+ case 'mssql_odbc':

+ $this->sql_layer = 'mssql';

+ break;

+

+ case 'mssqlnative':

+ $this->sql_layer = 'mssqlnative';

+ break;

+ }

+

+ $this->dbms_type_map = self::get_dbms_type_map();

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_list_tables()

+ {

+ $sql = "SELECT name

+ FROM sysobjects

+ WHERE type='U'";

+ $result = $this->db->sql_query($sql);

+

+ $tables = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $name = current($row);

+ $tables[$name] = $name;

+ }

+ $this->db->sql_freeresult($result);

+

+ return $tables;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_create_table($table_name, $table_data)

+ {

+ // holds the DDL for a column

+ $columns = $statements = array();

+

+ if ($this->sql_table_exists($table_name))

+ {

+ return $this->_sql_run_sql($statements);

+ }

+

+ // Begin transaction

+ $statements[] = 'begin';

+

+ // Determine if we have created a PRIMARY KEY in the earliest

+ $primary_key_gen = false;

+

+ // Determine if the table requires a sequence

+ $create_sequence = false;

+

+ // Begin table sql statement

+ $table_sql = 'CREATE TABLE [' . $table_name . '] (' . "\n";

+

+ if (!isset($table_data['PRIMARY_KEY']))

+ {

+ $table_data['COLUMNS']['mssqlindex'] = array('UINT', null, 'auto_increment');

+ $table_data['PRIMARY_KEY'] = 'mssqlindex';

+ }

+

+ // Iterate through the columns to create a table

+ foreach ($table_data['COLUMNS'] as $column_name => $column_data)

+ {

+ // here lies an array, filled with information compiled on the column's data

+ $prepared_column = $this->sql_prepare_column_data($table_name, $column_name, $column_data);

+

+ if (isset($prepared_column['auto_increment']) && $prepared_column['auto_increment'] && strlen($column_name) > 26) // "${column_name}_gen"

+ {

+ trigger_error("Index name '${column_name}_gen' on table '$table_name' is too long. The maximum auto increment column length is 26 characters.", E_USER_ERROR);

+ }

+

+ // here we add the definition of the new column to the list of columns

+ $columns[] = "\t [{$column_name}] " . $prepared_column['column_type_sql_default'];

+

+ // see if we have found a primary key set due to a column definition if we have found it, we can stop looking

+ if (!$primary_key_gen)

+ {

+ $primary_key_gen = isset($prepared_column['primary_key_set']) && $prepared_column['primary_key_set'];

+ }

+

+ // create sequence DDL based off of the existance of auto incrementing columns

+ if (!$create_sequence && isset($prepared_column['auto_increment']) && $prepared_column['auto_increment'])

+ {

+ $create_sequence = $column_name;

+ }

+ }

+

+ // this makes up all the columns in the create table statement

+ $table_sql .= implode(",\n", $columns);

+

+ // Close the table for two DBMS and add to the statements

+ $table_sql .= "\n);";

+ $statements[] = $table_sql;

+

+ // we have yet to create a primary key for this table,

+ // this means that we can add the one we really wanted instead

+ if (!$primary_key_gen)

+ {

+ // Write primary key

+ if (isset($table_data['PRIMARY_KEY']))

+ {

+ if (!is_array($table_data['PRIMARY_KEY']))

+ {

+ $table_data['PRIMARY_KEY'] = array($table_data['PRIMARY_KEY']);

+ }

+

+ // We need the data here

+ $old_return_statements = $this->return_statements;

+ $this->return_statements = true;

+

+ $primary_key_stmts = $this->sql_create_primary_key($table_name, $table_data['PRIMARY_KEY']);

+ foreach ($primary_key_stmts as $pk_stmt)

+ {

+ $statements[] = $pk_stmt;

+ }

+

+ $this->return_statements = $old_return_statements;

+ }

+ }

+

+ // Write Keys

+ if (isset($table_data['KEYS']))

+ {

+ foreach ($table_data['KEYS'] as $key_name => $key_data)

+ {

+ if (!is_array($key_data[1]))

+ {

+ $key_data[1] = array($key_data[1]);

+ }

+

+ $old_return_statements = $this->return_statements;

+ $this->return_statements = true;

+

+ $key_stmts = ($key_data[0] == 'UNIQUE') ? $this->sql_create_unique_index($table_name, $key_name, $key_data[1]) : $this->sql_create_index($table_name, $key_name, $key_data[1]);

+

+ foreach ($key_stmts as $key_stmt)

+ {

+ $statements[] = $key_stmt;

+ }

+

+ $this->return_statements = $old_return_statements;

+ }

+ }

+

+ // Commit Transaction

+ $statements[] = 'commit';

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_list_columns($table_name)

+ {

+ $columns = array();

+

+ $sql = "SELECT c.name

+ FROM syscolumns c

+ LEFT JOIN sysobjects o ON c.id = o.id

+ WHERE o.name = '{$table_name}'";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $column = strtolower(current($row));

+ $columns[$column] = $column;

+ }

+ $this->db->sql_freeresult($result);

+

+ return $columns;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_index_exists($table_name, $index_name)

+ {

+ $sql = "EXEC sp_statistics '$table_name'";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if ($row['TYPE'] == 3)

+ {

+ if (strtolower($row['INDEX_NAME']) == strtolower($index_name))

+ {

+ $this->db->sql_freeresult($result);

+ return true;

+ }

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ return false;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_unique_index_exists($table_name, $index_name)

+ {

+ $sql = "EXEC sp_statistics '$table_name'";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ // Usually NON_UNIQUE is the column we want to check, but we allow for both

+ if ($row['TYPE'] == 3)

+ {

+ if (strtolower($row['INDEX_NAME']) == strtolower($index_name))

+ {

+ $this->db->sql_freeresult($result);

+ return true;

+ }

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ return false;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_prepare_column_data($table_name, $column_name, $column_data)

+ {

+ if (strlen($column_name) > 30)

+ {

+ trigger_error("Column name '$column_name' on table '$table_name' is too long. The maximum is 30 characters.", E_USER_ERROR);

+ }

+

+ // Get type

+ list($column_type, ) = $this->get_column_type($column_data[0]);

+

+ // Adjust default value if db-dependent specified

+ if (is_array($column_data[1]))

+ {

+ $column_data[1] = (isset($column_data[1][$this->sql_layer])) ? $column_data[1][$this->sql_layer] : $column_data[1]['default'];

+ }

+

+ $sql = '';

+

+ $return_array = array();

+

+ $sql .= " {$column_type} ";

+ $sql_default = " {$column_type} ";

+

+ // For adding columns we need the default definition

+ if (!is_null($column_data[1]))

+ {

+ // For hexadecimal values do not use single quotes

+ if (strpos($column_data[1], '0x') === 0)

+ {

+ $return_array['default'] = 'DEFAULT (' . $column_data[1] . ') ';

+ $sql_default .= $return_array['default'];

+ }

+ else

+ {

+ $return_array['default'] = 'DEFAULT (' . ((is_numeric($column_data[1])) ? $column_data[1] : "'{$column_data[1]}'") . ') ';

+ $sql_default .= $return_array['default'];

+ }

+ }

+

+ if (isset($column_data[2]) && $column_data[2] == 'auto_increment')

+ {

+ // $sql .= 'IDENTITY (1, 1) ';

+ $sql_default .= 'IDENTITY (1, 1) ';

+ }

+

+ $return_array['textimage'] = $column_type === '[text]';

+

+ if (!is_null($column_data[1]) || (isset($column_data[2]) && $column_data[2] == 'auto_increment'))

+ {

+ $sql .= 'NOT NULL';

+ $sql_default .= 'NOT NULL';

+ }

+ else

+ {

+ $sql .= 'NULL';

+ $sql_default .= 'NULL';

+ }

+

+ $return_array['column_type_sql_default'] = $sql_default;

+

+ $return_array['column_type_sql'] = $sql;

+

+ return $return_array;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_column_add($table_name, $column_name, $column_data, $inline = false)

+ {

+ $column_data = $this->sql_prepare_column_data($table_name, $column_name, $column_data);

+ $statements = array();

+

+ // Does not support AFTER, only through temporary table

+ $statements[] = 'ALTER TABLE [' . $table_name . '] ADD [' . $column_name . '] ' . $column_data['column_type_sql_default'];

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_column_remove($table_name, $column_name, $inline = false)

+ {

+ $statements = array();

+

+ // We need the data here

+ $old_return_statements = $this->return_statements;

+ $this->return_statements = true;

+

+ $indexes = $this->get_existing_indexes($table_name, $column_name);

+ $indexes = array_merge($indexes, $this->get_existing_indexes($table_name, $column_name, true));

+

+ // Drop any indexes

+ $recreate_indexes = array();

+ if (!empty($indexes))

+ {

+ foreach ($indexes as $index_name => $index_data)

+ {

+ $result = $this->sql_index_drop($table_name, $index_name);

+ $statements = array_merge($statements, $result);

+ if (sizeof($index_data) > 1)

+ {

+ // Remove this column from the index and recreate it

+ $recreate_indexes[$index_name] = array_diff($index_data, array($column_name));

+ }

+ }

+ }

+

+ // Drop default value constraint

+ $result = $this->mssql_get_drop_default_constraints_queries($table_name, $column_name);

+ $statements = array_merge($statements, $result);

+

+ // Remove the column

+ $statements[] = 'ALTER TABLE [' . $table_name . '] DROP COLUMN [' . $column_name . ']';

+

+ if (!empty($recreate_indexes))

+ {

+ // Recreate indexes after we removed the column

+ foreach ($recreate_indexes as $index_name => $index_data)

+ {

+ $result = $this->sql_create_index($table_name, $index_name, $index_data);

+ $statements = array_merge($statements, $result);

+ }

+ }

+

+ $this->return_statements = $old_return_statements;

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_index_drop($table_name, $index_name)

+ {

+ $statements = array();

+

+ $statements[] = 'DROP INDEX ' . $table_name . '.' . $index_name;

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_table_drop($table_name)

+ {

+ $statements = array();

+

+ if (!$this->sql_table_exists($table_name))

+ {

+ return $this->_sql_run_sql($statements);

+ }

+

+ // the most basic operation, get rid of the table

+ $statements[] = 'DROP TABLE ' . $table_name;

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_create_primary_key($table_name, $column, $inline = false)

+ {

+ $statements = array();

+

+ $sql = "ALTER TABLE [{$table_name}] WITH NOCHECK ADD ";

+ $sql .= "CONSTRAINT [PK_{$table_name}] PRIMARY KEY CLUSTERED (";

+ $sql .= '[' . implode("],\n\t\t[", $column) . ']';

+ $sql .= ')';

+

+ $statements[] = $sql;

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_create_unique_index($table_name, $index_name, $column)

+ {

+ $statements = array();

+

+ $this->check_index_name_length($table_name, $index_name);

+

+ $statements[] = 'CREATE UNIQUE INDEX [' . $index_name . '] ON [' . $table_name . ']([' . implode('], [', $column) . '])';

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_create_index($table_name, $index_name, $column)

+ {

+ $statements = array();

+

+ $this->check_index_name_length($table_name, $index_name);

+

+ // remove index length

+ $column = preg_replace('#:.*$#', '', $column);

+

+ $statements[] = 'CREATE INDEX [' . $index_name . '] ON [' . $table_name . ']([' . implode('], [', $column) . '])';

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_list_index($table_name)

+ {

+ $index_array = array();

+ $sql = "EXEC sp_statistics '$table_name'";

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if ($row['TYPE'] == 3)

+ {

+ $index_array[] = strtolower($row['INDEX_NAME']);

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ return $index_array;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_column_change($table_name, $column_name, $column_data, $inline = false)

+ {

+ $column_data = $this->sql_prepare_column_data($table_name, $column_name, $column_data);

+ $statements = array();

+

+ // We need the data here

+ $old_return_statements = $this->return_statements;

+ $this->return_statements = true;

+

+ $indexes = $this->get_existing_indexes($table_name, $column_name);

+ $unique_indexes = $this->get_existing_indexes($table_name, $column_name, true);

+

+ // Drop any indexes

+ if (!empty($indexes) || !empty($unique_indexes))

+ {

+ $drop_indexes = array_merge(array_keys($indexes), array_keys($unique_indexes));

+ foreach ($drop_indexes as $index_name)

+ {

+ $result = $this->sql_index_drop($table_name, $index_name);

+ $statements = array_merge($statements, $result);

+ }

+ }

+

+ // Drop default value constraint

+ $result = $this->mssql_get_drop_default_constraints_queries($table_name, $column_name);

+ $statements = array_merge($statements, $result);

+

+ // Change the column

+ $statements[] = 'ALTER TABLE [' . $table_name . '] ALTER COLUMN [' . $column_name . '] ' . $column_data['column_type_sql'];

+

+ if (!empty($column_data['default']))

+ {

+ // Add new default value constraint

+ $statements[] = 'ALTER TABLE [' . $table_name . '] ADD CONSTRAINT [DF_' . $table_name . '_' . $column_name . '_1] ' . $this->db->sql_escape($column_data['default']) . ' FOR [' . $column_name . ']';

+ }

+

+ if (!empty($indexes))

+ {

+ // Recreate indexes after we changed the column

+ foreach ($indexes as $index_name => $index_data)

+ {

+ $result = $this->sql_create_index($table_name, $index_name, $index_data);

+ $statements = array_merge($statements, $result);

+ }

+ }

+

+ if (!empty($unique_indexes))

+ {

+ // Recreate unique indexes after we changed the column

+ foreach ($unique_indexes as $index_name => $index_data)

+ {

+ $result = $this->sql_create_unique_index($table_name, $index_name, $index_data);

+ $statements = array_merge($statements, $result);

+ }

+ }

+

+ $this->return_statements = $old_return_statements;

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * Get queries to drop the default constraints of a column

+ *

+ * We need to drop the default constraints of a column,

+ * before being able to change their type or deleting them.

+ *

+ * @param string $table_name

+ * @param string $column_name

+ * @return array Array with SQL statements

+ */

+ protected function mssql_get_drop_default_constraints_queries($table_name, $column_name)

+ {

+ $statements = array();

+ if ($this->mssql_is_sql_server_2000())

+ {

+ // http://msdn.microsoft.com/en-us/library/aa175912%28v=sql.80%29.aspx

+ // Deprecated in SQL Server 2005

+ $sql = "SELECT so.name AS def_name

+ FROM sysobjects so

+ JOIN sysconstraints sc ON so.id = sc.constid

+ WHERE object_name(so.parent_obj) = '{$table_name}'

+ AND so.xtype = 'D'

+ AND sc.colid = (SELECT colid FROM syscolumns

+ WHERE id = object_id('{$table_name}')

+ AND name = '{$column_name}')";

+ }

+ else

+ {

+ $sql = "SELECT dobj.name AS def_name

+ FROM sys.columns col

+ LEFT OUTER JOIN sys.objects dobj ON (dobj.object_id = col.default_object_id AND dobj.type = 'D')

+ WHERE col.object_id = object_id('{$table_name}')

+ AND col.name = '{$column_name}'

+ AND dobj.name IS NOT NULL";

+ }

+

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $statements[] = 'ALTER TABLE [' . $table_name . '] DROP CONSTRAINT [' . $row['def_name'] . ']';

+ }

+ $this->db->sql_freeresult($result);

+

+ return $statements;

+ }

+

+ /**

+ * Get a list with existing indexes for the column

+ *

+ * @param string $table_name

+ * @param string $column_name

+ * @param bool $unique Should we get unique indexes or normal ones

+ * @return array Array with Index name => columns

+ */

+ public function get_existing_indexes($table_name, $column_name, $unique = false)

+ {

+ $existing_indexes = array();

+ if ($this->mssql_is_sql_server_2000())

+ {

+ // http://msdn.microsoft.com/en-us/library/aa175912%28v=sql.80%29.aspx

+ // Deprecated in SQL Server 2005

+ $sql = "SELECT DISTINCT ix.name AS phpbb_index_name

+ FROM sysindexes ix

+ INNER JOIN sysindexkeys ixc

+ ON ixc.id = ix.id

+ AND ixc.indid = ix.indid

+ INNER JOIN syscolumns cols

+ ON cols.colid = ixc.colid

+ AND cols.id = ix.id

+ WHERE ix.id = object_id('{$table_name}')

+ AND cols.name = '{$column_name}'

+ AND INDEXPROPERTY(ix.id, ix.name, 'IsUnique') = " . ($unique ? '1' : '0');

+ }

+ else

+ {

+ $sql = "SELECT DISTINCT ix.name AS phpbb_index_name

+ FROM sys.indexes ix

+ INNER JOIN sys.index_columns ixc

+ ON ixc.object_id = ix.object_id

+ AND ixc.index_id = ix.index_id

+ INNER JOIN sys.columns cols

+ ON cols.column_id = ixc.column_id

+ AND cols.object_id = ix.object_id

+ WHERE ix.object_id = object_id('{$table_name}')

+ AND cols.name = '{$column_name}'

+ AND ix.is_unique = " . ($unique ? '1' : '0');

+ }

+

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if (!isset($row['is_unique']) || ($unique && $row['is_unique'] == 'UNIQUE') || (!$unique && $row['is_unique'] == 'NONUNIQUE'))

+ {

+ $existing_indexes[$row['phpbb_index_name']] = array();

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ if (empty($existing_indexes))

+ {

+ return array();

+ }

+

+ if ($this->mssql_is_sql_server_2000())

+ {

+ $sql = "SELECT DISTINCT ix.name AS phpbb_index_name, cols.name AS phpbb_column_name

+ FROM sysindexes ix

+ INNER JOIN sysindexkeys ixc

+ ON ixc.id = ix.id

+ AND ixc.indid = ix.indid

+ INNER JOIN syscolumns cols

+ ON cols.colid = ixc.colid

+ AND cols.id = ix.id

+ WHERE ix.id = object_id('{$table_name}')

+ AND " . $this->db->sql_in_set('ix.name', array_keys($existing_indexes));

+ }

+ else

+ {

+ $sql = "SELECT DISTINCT ix.name AS phpbb_index_name, cols.name AS phpbb_column_name

+ FROM sys.indexes ix

+ INNER JOIN sys.index_columns ixc

+ ON ixc.object_id = ix.object_id

+ AND ixc.index_id = ix.index_id

+ INNER JOIN sys.columns cols

+ ON cols.column_id = ixc.column_id

+ AND cols.object_id = ix.object_id

+ WHERE ix.object_id = object_id('{$table_name}')

+ AND " . $this->db->sql_in_set('ix.name', array_keys($existing_indexes));

+ }

+

+ $result = $this->db->sql_query($sql);

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $existing_indexes[$row['phpbb_index_name']][] = $row['phpbb_column_name'];

+ }

+ $this->db->sql_freeresult($result);

+

+ return $existing_indexes;

+ }

+

+ /**

+ * Is the used MS SQL Server a SQL Server 2000?

+ *

+ * @return bool

+ */

+ protected function mssql_is_sql_server_2000()

+ {

+ if ($this->is_sql_server_2000 === null)

+ {

+ $sql = "SELECT CAST(SERVERPROPERTY('productversion') AS VARCHAR(25)) AS mssql_version";

+ $result = $this->db->sql_query($sql);

+ $properties = $this->db->sql_fetchrow($result);

+ $this->db->sql_freeresult($result);

+ $this->is_sql_server_2000 = $properties['mssql_version'][0] == '8';

+ }

+

+ return $this->is_sql_server_2000;

+ }

+

+}

+<?php

+/**

+ *

+ * This file is part of the phpBB Forum Software package.

+ *

+ * @copyright (c) phpBB Limited <https://www.phpbb.com>

+ * @license GNU General Public License, version 2 (GPL-2.0)

+ *

+ * For full copyright and license information, please see

+ * the docs/CREDITS.txt file.

+ *

+ */

+

+namespace phpbb\db\tools;

+

+/**

+ * Database Tools for handling cross-db actions such as altering columns, etc.

+ * Currently not supported is returning SQL for creating tables.

+ */

+class postgres extends tools

+{

+ /**

+ * Get the column types for postgres only

+ *

+ * @return array

+ */

+ public static function get_dbms_type_map()

+ {

+ return array(

+ 'postgres' => array(

+ 'INT:' => 'INT4',

+ 'BINT' => 'INT8',

+ 'UINT' => 'INT4', // unsigned

+ 'UINT:' => 'INT4', // unsigned

+ 'USINT' => 'INT2', // unsigned

+ 'BOOL' => 'INT2', // unsigned

+ 'TINT:' => 'INT2',

+ 'VCHAR' => 'varchar(255)',

+ 'VCHAR:' => 'varchar(%d)',

+ 'CHAR:' => 'char(%d)',

+ 'XSTEXT' => 'varchar(1000)',

+ 'STEXT' => 'varchar(3000)',

+ 'TEXT' => 'varchar(8000)',

+ 'MTEXT' => 'TEXT',

+ 'XSTEXT_UNI'=> 'varchar(100)',

+ 'STEXT_UNI' => 'varchar(255)',

+ 'TEXT_UNI' => 'varchar(4000)',

+ 'MTEXT_UNI' => 'TEXT',

+ 'TIMESTAMP' => 'INT4', // unsigned

+ 'DECIMAL' => 'decimal(5,2)',

+ 'DECIMAL:' => 'decimal(%d,2)',

+ 'PDECIMAL' => 'decimal(6,3)',

+ 'PDECIMAL:' => 'decimal(%d,3)',

+ 'VCHAR_UNI' => 'varchar(255)',

+ 'VCHAR_UNI:'=> 'varchar(%d)',

+ 'VCHAR_CI' => 'varchar_ci',

+ 'VARBINARY' => 'bytea',

+ ),

+ );

+ }

+

+ /**

+ * Constructor. Set DB Object and set {@link $return_statements return_statements}.

+ *

+ * @param \phpbb\db\driver\driver_interface $db Database connection

+ * @param bool $return_statements True if only statements should be returned and no SQL being executed

+ */

+ public function __construct(\phpbb\db\driver\driver_interface $db, $return_statements = false)

+ {

+ parent::__construct($db, $return_statements);

+

+ // Determine mapping database type

+ $this->sql_layer = 'postgres';

+

+ $this->dbms_type_map = self::get_dbms_type_map();

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_list_tables()

+ {

+ $sql = 'SELECT relname

+ FROM pg_stat_user_tables';

+ $result = $this->db->sql_query($sql);

+

+ $tables = array();

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $name = current($row);

+ $tables[$name] = $name;

+ }

+ $this->db->sql_freeresult($result);

+

+ return $tables;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_create_table($table_name, $table_data)

+ {

+ // holds the DDL for a column

+ $columns = $statements = array();

+

+ if ($this->sql_table_exists($table_name))

+ {

+ return $this->_sql_run_sql($statements);

+ }

+

+ // Begin transaction

+ $statements[] = 'begin';

+

+ // Determine if we have created a PRIMARY KEY in the earliest

+ $primary_key_gen = false;

+

+ // Determine if the table requires a sequence

+ $create_sequence = false;

+

+ // Begin table sql statement

+ $table_sql = 'CREATE TABLE ' . $table_name . ' (' . "\n";

+

+ // Iterate through the columns to create a table

+ foreach ($table_data['COLUMNS'] as $column_name => $column_data)

+ {

+ // here lies an array, filled with information compiled on the column's data

+ $prepared_column = $this->sql_prepare_column_data($table_name, $column_name, $column_data);

+

+ if (isset($prepared_column['auto_increment']) && $prepared_column['auto_increment'] && strlen($column_name) > 26) // "${column_name}_gen"

+ {

+ trigger_error("Index name '${column_name}_gen' on table '$table_name' is too long. The maximum auto increment column length is 26 characters.", E_USER_ERROR);

+ }

+

+ // here we add the definition of the new column to the list of columns

+ $columns[] = "\t {$column_name} " . $prepared_column['column_type_sql'];

+

+ // see if we have found a primary key set due to a column definition if we have found it, we can stop looking

+ if (!$primary_key_gen)

+ {

+ $primary_key_gen = isset($prepared_column['primary_key_set']) && $prepared_column['primary_key_set'];

+ }

+

+ // create sequence DDL based off of the existance of auto incrementing columns

+ if (!$create_sequence && isset($prepared_column['auto_increment']) && $prepared_column['auto_increment'])

+ {

+ $create_sequence = $column_name;

+ }

+ }

+

+ // this makes up all the columns in the create table statement

+ $table_sql .= implode(",\n", $columns);

+

+ // we have yet to create a primary key for this table,

+ // this means that we can add the one we really wanted instead

+ if (!$primary_key_gen)

+ {

+ // Write primary key

+ if (isset($table_data['PRIMARY_KEY']))

+ {

+ if (!is_array($table_data['PRIMARY_KEY']))

+ {

+ $table_data['PRIMARY_KEY'] = array($table_data['PRIMARY_KEY']);

+ }

+

+ $table_sql .= ",\n\t PRIMARY KEY (" . implode(', ', $table_data['PRIMARY_KEY']) . ')';

+ }

+ }

+

+ // do we need to add a sequence for auto incrementing columns?

+ if ($create_sequence)

+ {

+ $statements[] = "CREATE SEQUENCE {$table_name}_seq;";

+ }

+

+ // close the table

+ $table_sql .= "\n);";

+ $statements[] = $table_sql;

+

+ // Write Keys

+ if (isset($table_data['KEYS']))

+ {

+ foreach ($table_data['KEYS'] as $key_name => $key_data)

+ {

+ if (!is_array($key_data[1]))

+ {

+ $key_data[1] = array($key_data[1]);

+ }

+

+ $old_return_statements = $this->return_statements;

+ $this->return_statements = true;

+

+ $key_stmts = ($key_data[0] == 'UNIQUE') ? $this->sql_create_unique_index($table_name, $key_name, $key_data[1]) : $this->sql_create_index($table_name, $key_name, $key_data[1]);

+

+ foreach ($key_stmts as $key_stmt)

+ {

+ $statements[] = $key_stmt;

+ }

+

+ $this->return_statements = $old_return_statements;

+ }

+ }

+

+ // Commit Transaction

+ $statements[] = 'commit';

+

+ return $this->_sql_run_sql($statements);

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_list_columns($table_name)

+ {

+ $columns = array();

+

+ $sql = "SELECT a.attname

+ FROM pg_class c, pg_attribute a

+ WHERE c.relname = '{$table_name}'

+ AND a.attnum > 0

+ AND a.attrelid = c.oid";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ $column = strtolower(current($row));

+ $columns[$column] = $column;

+ }

+ $this->db->sql_freeresult($result);

+

+ return $columns;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_index_exists($table_name, $index_name)

+ {

+ $sql = "SELECT ic.relname as index_name

+ FROM pg_class bc, pg_class ic, pg_index i

+ WHERE (bc.oid = i.indrelid)

+ AND (ic.oid = i.indexrelid)

+ AND (bc.relname = '" . $table_name . "')

+ AND (i.indisunique != 't')

+ AND (i.indisprimary != 't')";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ // This DBMS prefixes index names with the table name

+ $row['index_name'] = $this->strip_table_name_from_index_name($table_name, $row['index_name']);

+

+ if (strtolower($row['index_name']) == strtolower($index_name))

+ {

+ $this->db->sql_freeresult($result);

+ return true;

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ return false;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_unique_index_exists($table_name, $index_name)

+ {

+ $sql = "SELECT ic.relname as index_name, i.indisunique

+ FROM pg_class bc, pg_class ic, pg_index i

+ WHERE (bc.oid = i.indrelid)

+ AND (ic.oid = i.indexrelid)

+ AND (bc.relname = '" . $table_name . "')

+ AND (i.indisprimary != 't')";

+ $result = $this->db->sql_query($sql);

+

+ while ($row = $this->db->sql_fetchrow($result))

+ {

+ if ($row['indisunique'] != 't')

+ {

+ continue;

+ }

+

+ // This DBMS prefixes index names with the table name

+ $row['index_name'] = $this->strip_table_name_from_index_name($table_name, $row['index_name']);

+

+ if (strtolower($row['index_name']) == strtolower($index_name))

+ {

+ $this->db->sql_freeresult($result);

+ return true;

+ }

+ }

+ $this->db->sql_freeresult($result);

+

+ return false;

+ }

+

+ /**

+ * Function to prepare some column information for better usage

+ * @access private

+ */

+ function sql_prepare_column_data($table_name, $column_name, $column_data)

+ {

+ if (strlen($column_name) > 30)

+ {

+ trigger_error("Column name '$column_name' on table '$table_name' is too long. The maximum is 30 characters.", E_USER_ERROR);

+ }

+

+ // Get type

+ list($column_type, $orig_column_type) = $this->get_column_type($column_data[0]);

+

+ // Adjust default value if db-dependent specified

+ if (is_array($column_data[1]))

+ {

+ $column_data[1] = (isset($column_data[1][$this->sql_layer])) ? $column_data[1][$this->sql_layer] : $column_data[1]['default'];

+ }

+

+ $sql = " {$column_type} ";

+

+ $return_array = array(

+ 'column_type' => $column_type,

+ 'auto_increment' => false,

+ );

+

+ if (isset($column_data[2]) && $column_data[2] == 'auto_increment')

+ {

+ $default_val = "nextval('{$table_name}_seq')";

+ $return_array['auto_increment'] = true;

+ }

+ else if (!is_null($column_data[1]))

+ {

+ $default_val = "'" . $column_data[1] . "'";

+ $return_array['null'] = 'NOT NULL';

+ $sql .= 'NOT NULL ';

+ }

+ else

+ {

+ // Integers need to have 0 instead of empty string as default

+ if (strpos($column_type, 'INT') === 0)

+ {

+ $default_val = '0';

+ }

+ else

+ {

+ $default_val = "'" . $column_data[1] . "'";

+ }

+ $return_array['null'] = 'NULL';

+ $sql .= 'NULL ';

+ }

+

+ $return_array['default'] = $default_val;

+

+ $sql .= "DEFAULT {$default_val}";

+

+ // Unsigned? Then add a CHECK contraint

+ if (in_array($orig_column_type, $this->unsigned_types))

+ {

+ $return_array['constraint'] = "CHECK ({$column_name} >= 0)";

+ $sql .= " CHECK ({$column_name} >= 0)";

+ }

+

+ $return_array['column_type_sql'] = $sql;

+

+ return $return_array;

+ }

+

+ /**

+ * {@inheritDoc}

+ */

+ function sql_column_add($table_name, $column_name, $column_data, $inline = false)

+ {

+ $column_data = $this->sql_prepare_column_data($table_name, $column_name, $column_data);

+ $statements = array();

+

+ // Does not support AFTER, only through temporary table

+ if (version_compare($this->db->sql_server_info(true), '8.0', '>='))

+ {

+ $statements[] = 'ALTER TABLE ' . $table_name . ' ADD COLUMN "' . $column_name . '" ' . $column_data['column_type_sql'];

+ }

+ else

+ {

+ // old versions cannot add columns with default and null information

+ $statements[] = 'ALTER TABLE ' . $table_name . ' ADD COLUMN "' . $column_name . '" ' . $column_data['column_type'] . ' ' . $column_data['constraint'];

+

+ if (isset($column_data['null']))

+ {

+ if ($column_data['null'] == 'NOT NULL')