diff --git a/inc/database.php b/inc/database.php index 9aa04f51..40141d7e 100644 --- a/inc/database.php +++ b/inc/database.php @@ -8,30 +8,30 @@ defined('TINYBOARD') or exit; class PreparedQueryDebug { protected $query, $explain_query = false; - + public function __construct($query) { global $pdo, $config; $query = preg_replace("/[\n\t]+/", ' ', $query); - + $this->query = $pdo->prepare($query); if ($config['debug'] && $config['debug_explain'] && preg_match('/^(SELECT|INSERT|UPDATE|DELETE) /i', $query)) $this->explain_query = $pdo->prepare("EXPLAIN $query"); } public function __call($function, $args) { global $config, $debug; - + if ($config['debug'] && $function == 'execute') { if ($this->explain_query) { $this->explain_query->execute() or error(db_error($this->explain_query)); } $start = microtime(true); } - + if ($this->explain_query && $function == 'bindValue') call_user_func_array(array($this->explain_query, $function), $args); - + $return = call_user_func_array(array($this->query, $function), $args); - + if ($config['debug'] && $function == 'execute') { $time = microtime(true) - $start; $debug['sql'][] = array( @@ -42,7 +42,7 @@ class PreparedQueryDebug { ); $debug['time']['db_queries'] += $time; } - + return $return; } } @@ -51,16 +51,16 @@ function sql_open() { global $pdo, $config, $debug; if ($pdo) return true; - - + + if ($config['debug']) $start = microtime(true); - + if (isset($config['db']['server'][0]) && $config['db']['server'][0] == ':') $unix_socket = substr($config['db']['server'], 1); else $unix_socket = false; - + $dsn = $config['db']['type'] . ':' . ($unix_socket ? 'unix_socket=' . $unix_socket : 'host=' . $config['db']['server']) . ';dbname=' . $config['db']['database']; @@ -69,15 +69,16 @@ function sql_open() { try { $options = array( PDO::ATTR_TIMEOUT => $config['db']['timeout'], - PDO::MYSQL_ATTR_USE_BUFFERED_QUERY => true + PDO::MYSQL_ATTR_USE_BUFFERED_QUERY => true, + PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION, // Set a consistent error mode between PHP versions. ); if ($config['db']['persistent']) $options[PDO::ATTR_PERSISTENT] = true; $pdo = new PDO($dsn, $config['db']['user'], $config['db']['password'], $options); - + if ($config['debug']) $debug['time']['db_connect'] = '~' . round((microtime(true) - $start) * 1000, 2) . 'ms'; - + if (mysql_version() >= 50503) query('SET NAMES utf8mb4') or error(db_error()); else @@ -85,11 +86,11 @@ function sql_open() { return $pdo; } catch(PDOException $e) { $message = $e->getMessage(); - + // Remove any sensitive information $message = str_replace($config['db']['user'], 'hidden', $message); $message = str_replace($config['db']['password'], 'hidden', $message); - + // Print error error(_('Database error: ') . $message); } @@ -98,7 +99,7 @@ function sql_open() { // 5.6.10 becomes 50610 function mysql_version() { global $pdo; - + $version = $pdo->getAttribute(PDO::ATTR_SERVER_VERSION); $v = explode('.', $version); if (count($v) != 3) @@ -108,11 +109,11 @@ function mysql_version() { function prepare($query) { global $pdo, $debug, $config; - + $query = preg_replace('/``('.$config['board_regex'].')``/u', '`' . $config['db']['prefix'] . '$1`', $query); - + sql_open(); - + if ($config['debug']) return new PreparedQueryDebug($query); @@ -121,11 +122,11 @@ function prepare($query) { function query($query) { global $pdo, $debug, $config; - + $query = preg_replace('/``('.$config['board_regex'].')``/u', '`' . $config['db']['prefix'] . '$1`', $query); - + sql_open(); - + if ($config['debug']) { if ($config['debug_explain'] && preg_match('/^(SELECT|INSERT|UPDATE|DELETE) /i', $query)) { $explain = $pdo->query("EXPLAIN $query") or error(db_error());