| | |
| | | var $db_handle = 0; // Connection handle |
| | | var $db_error = false; |
| | | var $db_error_msg = ''; |
| | | var $debug_mode = false; |
| | | |
| | | var $a_query_results = array('dummy'); |
| | | var $last_res_id = 0; |
| | | |
| | | private $debug_mode = false; |
| | | private $a_query_results = array('dummy'); |
| | | private $last_res_id = 0; |
| | | private $tables; |
| | | |
| | | |
| | |
| | | * @return object PEAR database handle |
| | | * @access private |
| | | */ |
| | | function dsn_connect($dsn) |
| | | private function dsn_connect($dsn) |
| | | { |
| | | // Use persistent connections if available |
| | | $db_options = array( |
| | |
| | | 'emulate_prepared' => $this->debug_mode, |
| | | 'debug' => $this->debug_mode, |
| | | 'debug_handler' => 'mdb2_debug_handler', |
| | | 'portability' => MDB2_PORTABILITY_ALL ^ MDB2_PORTABILITY_EMPTY_TO_NULL); |
| | | 'portability' => MDB2_PORTABILITY_ALL ^ MDB2_PORTABILITY_EMPTY_TO_null); |
| | | |
| | | if ($this->db_provider == 'pgsql') { |
| | | $db_options['disable_smart_seqname'] = true; |
| | |
| | | |
| | | $dbh = MDB2::connect($dsn, $db_options); |
| | | |
| | | if (MDB2::isError($dbh)) |
| | | { |
| | | $this->db_error = TRUE; |
| | | if (MDB2::isError($dbh)) { |
| | | $this->db_error = true; |
| | | $this->db_error_msg = $dbh->getMessage(); |
| | | |
| | | raise_error(array('code' => 500, 'type' => 'db', 'line' => __LINE__, |
| | | 'file' => __FILE__, 'message' => $dbh->getUserInfo()), TRUE, FALSE); |
| | | raise_error(array('code' => 500, 'type' => 'db', |
| | | 'line' => __LINE__, 'file' => __FILE__, |
| | | 'message' => $dbh->getUserInfo()), true, false); |
| | | } |
| | | else if ($this->db_provider=='sqlite') |
| | | { |
| | | else if ($this->db_provider == 'sqlite') { |
| | | $dsn_array = MDB2::parseDSN($dsn); |
| | | if (!filesize($dsn_array['database']) && !empty($this->sqlite_initials)) |
| | | $this->_sqlite_create_database($dbh, $this->sqlite_initials); |
| | |
| | | |
| | | |
| | | /** |
| | | * Connect to appropiate databse |
| | | * depending on the operation |
| | | * Connect to appropiate database depending on the operation |
| | | * |
| | | * @param string Connection mode (r|w) |
| | | * @access public |
| | |
| | | $this->db_mode = $mode; |
| | | |
| | | // Already connected |
| | | if ($this->db_connected) |
| | | { |
| | | if ($this->db_connected) { |
| | | // no replication, current connection is ok |
| | | if ($this->db_dsnw==$this->db_dsnr) |
| | | return; |
| | |
| | | return; |
| | | } |
| | | |
| | | if ($mode=='r') |
| | | $dsn = $this->db_dsnr; |
| | | else |
| | | $dsn = $this->db_dsnw; |
| | | $dsn = ($mode == 'r') ? $this->db_dsnr : $this->db_dsnw; |
| | | |
| | | $this->db_handle = $this->dsn_connect($dsn); |
| | | $this->db_connected = true; |
| | |
| | | * Activate/deactivate debug mode |
| | | * |
| | | * @param boolean True if SQL queries should be logged |
| | | * @access public |
| | | */ |
| | | function set_debug($dbg = true) |
| | | { |
| | | $this->debug_mode = $dbg; |
| | | if ($this->db_connected) |
| | | { |
| | | if ($this->db_connected) { |
| | | $this->db_handle->setOption('debug', $dbg); |
| | | $this->db_handle->setOption('emulate_prepared', $dbg); |
| | | } |
| | |
| | | * Getter for error state |
| | | * |
| | | * @param boolean True on error |
| | | * @access public |
| | | */ |
| | | function is_error() |
| | | { |
| | | return $this->db_error ? $this->db_error_msg : FALSE; |
| | | return $this->db_error ? $this->db_error_msg : false; |
| | | } |
| | | |
| | | |
| | |
| | | * Connection state checker |
| | | * |
| | | * @param boolean True if in connected state |
| | | * @access public |
| | | */ |
| | | function is_connected() |
| | | { |
| | | return PEAR::isError($this->db_handle) ? false : true; |
| | | return PEAR::isError($this->db_handle) ? false : $this->db_connected; |
| | | } |
| | | |
| | | |
| | |
| | | function query() |
| | | { |
| | | if (!$this->is_connected()) |
| | | return NULL; |
| | | return null; |
| | | |
| | | $params = func_get_args(); |
| | | $query = array_shift($params); |
| | |
| | | * @return number Query handle identifier |
| | | * @access private |
| | | */ |
| | | function _query($query, $offset, $numrows, $params) |
| | | private function _query($query, $offset, $numrows, $params) |
| | | { |
| | | // Read or write ? |
| | | if (strtolower(substr(trim($query),0,6))=='select') |
| | | $mode='r'; |
| | | else |
| | | $mode='w'; |
| | | $mode = (strtolower(substr(trim($query),0,6)) == 'select') ? 'r' : 'w'; |
| | | |
| | | $this->db_connect($mode); |
| | | |
| | |
| | | |
| | | if (empty($params)) |
| | | $result = $mode=='r' ? $this->db_handle->query($query) : $this->db_handle->exec($query); |
| | | else |
| | | { |
| | | else { |
| | | $params = (array)$params; |
| | | $q = $this->db_handle->prepare($query, null, $mode=='w' ? MDB2_PREPARE_MANIP : null); |
| | | if ($this->db_handle->isError($q)) |
| | | { |
| | | $this->db_error = TRUE; |
| | | if ($this->db_handle->isError($q)) { |
| | | $this->db_error = true; |
| | | $this->db_error_msg = $q->userinfo; |
| | | |
| | | raise_error(array('code' => 500, 'type' => 'db', |
| | | 'line' => __LINE__, 'file' => __FILE__, |
| | | 'message' => $this->db_error_msg), TRUE, TRUE); |
| | | 'message' => $this->db_error_msg), true, true); |
| | | } |
| | | else |
| | | { |
| | | else { |
| | | $result = $q->execute($params); |
| | | $q->free(); |
| | | } |
| | |
| | | * If no query handle is specified, the last query will be taken as reference |
| | | * |
| | | * @param number Optional query handle identifier |
| | | * @return mixed Number of rows or FALSE on failure |
| | | * @return mixed Number of rows or false on failure |
| | | * @access public |
| | | */ |
| | | function num_rows($res_id=NULL) |
| | | function num_rows($res_id=null) |
| | | { |
| | | if (!$this->db_handle) |
| | | return FALSE; |
| | | return false; |
| | | |
| | | if ($result = $this->_get_result($res_id)) |
| | | return $result->numRows(); |
| | | else |
| | | return FALSE; |
| | | return false; |
| | | } |
| | | |
| | | |
| | |
| | | * Get number of affected rows for the last query |
| | | * |
| | | * @param number Optional query handle identifier |
| | | * @return mixed Number of rows or FALSE on failure |
| | | * @return mixed Number of rows or false on failure |
| | | * @access public |
| | | */ |
| | | function affected_rows($res_id = null) |
| | | { |
| | | if (!$this->db_handle) |
| | | return FALSE; |
| | | return false; |
| | | |
| | | return (int) $this->_get_result($res_id); |
| | | } |
| | |
| | | * For Postgres databases, a sequence name is required |
| | | * |
| | | * @param string Table name (to find the incremented sequence) |
| | | * @return mixed ID or FALSE on failure |
| | | * @return mixed ID or false on failure |
| | | * @access public |
| | | */ |
| | | function insert_id($table = '') |
| | | { |
| | | if (!$this->db_handle || $this->db_mode=='r') |
| | | return FALSE; |
| | | return false; |
| | | |
| | | if ($table) { |
| | | if ($this->db_provider == 'pgsql') |
| | |
| | | * If no query handle is specified, the last query will be taken as reference |
| | | * |
| | | * @param number Optional query handle identifier |
| | | * @return mixed Array with col values or FALSE on failure |
| | | * @return mixed Array with col values or false on failure |
| | | * @access public |
| | | */ |
| | | function fetch_assoc($res_id=NULL) |
| | | function fetch_assoc($res_id=null) |
| | | { |
| | | $result = $this->_get_result($res_id); |
| | | return $this->_fetch_row($result, MDB2_FETCHMODE_ASSOC); |
| | |
| | | * If no query handle is specified, the last query will be taken as reference |
| | | * |
| | | * @param number Optional query handle identifier |
| | | * @return mixed Array with col values or FALSE on failure |
| | | * @return mixed Array with col values or false on failure |
| | | * @access public |
| | | */ |
| | | function fetch_array($res_id=NULL) |
| | | function fetch_array($res_id=null) |
| | | { |
| | | $result = $this->_get_result($res_id); |
| | | return $this->_fetch_row($result, MDB2_FETCHMODE_ORDERED); |
| | |
| | | * |
| | | * @param object Query result handle |
| | | * @param number Fetch mode identifier |
| | | * @return mixed Array with col values or FALSE on failure |
| | | * @return mixed Array with col values or false on failure |
| | | * @access private |
| | | */ |
| | | function _fetch_row($result, $mode) |
| | | private function _fetch_row($result, $mode) |
| | | { |
| | | if ($result === FALSE || PEAR::isError($result) || !$this->is_connected()) |
| | | return FALSE; |
| | | if ($result === false || PEAR::isError($result) || !$this->is_connected()) |
| | | return false; |
| | | |
| | | return $result->fetchRow($mode); |
| | | } |
| | |
| | | * Wrapper for the SHOW TABLES command |
| | | * |
| | | * @return array List of all tables of the current database |
| | | * @access public |
| | | * @since 0.4-beta |
| | | */ |
| | | function list_tables() |
| | | { |
| | |
| | | * Formats input so it can be safely used in a query |
| | | * |
| | | * @param mixed Value to quote |
| | | * @param string Type of data |
| | | * @return string Quoted/converted string for use in query |
| | | * @access public |
| | | */ |
| | | function quote($input, $type = null) |
| | | { |
| | | // handle int directly for better performance |
| | | if ($type == 'integer') |
| | | return intval($input); |
| | | |
| | | // create DB handle if not available |
| | | if (!$this->db_handle) |
| | | $this->db_connect('r'); |
| | | |
| | | // escape pear identifier chars |
| | | $rep_chars = array('?' => '\?', |
| | | '!' => '\!', |
| | | '&' => '\&'); |
| | | |
| | | return $this->db_handle->quote($input, $type); |
| | | } |
| | |
| | | return $this->db_handle->quoteIdentifier($str); |
| | | } |
| | | |
| | | |
| | | /** |
| | | * Escapes a string |
| | | * |
| | |
| | | */ |
| | | function now() |
| | | { |
| | | switch($this->db_provider) |
| | | { |
| | | switch($this->db_provider) { |
| | | case 'mssql': |
| | | case 'sqlsrv': |
| | | return "getdate()"; |
| | |
| | | /** |
| | | * Return list of elements for use with SQL's IN clause |
| | | * |
| | | * @param string Input array |
| | | * @return string Elements list string |
| | | * @param array Input array |
| | | * @param string Type of data |
| | | * @return string Comma-separated list of quoted values for use in query |
| | | * @access public |
| | | */ |
| | | function array2list($arr, $type=null) |
| | |
| | | if (!is_array($arr)) |
| | | return $this->quote($arr, $type); |
| | | |
| | | $res = array(); |
| | | foreach ($arr as $item) |
| | | $res[] = $this->quote($item, $type); |
| | | foreach ($arr as $idx => $item) |
| | | $arr[$idx] = $this->quote($item, $type); |
| | | |
| | | return implode(',', $res); |
| | | return implode(',', $arr); |
| | | } |
| | | |
| | | |
| | |
| | | */ |
| | | function unixtimestamp($field) |
| | | { |
| | | switch($this->db_provider) |
| | | { |
| | | switch($this->db_provider) { |
| | | case 'pgsql': |
| | | return "EXTRACT (EPOCH FROM $field)"; |
| | | break; |
| | | |
| | | case 'mssql': |
| | | case 'sqlsrv': |
| | |
| | | */ |
| | | function fromunixtime($timestamp) |
| | | { |
| | | switch($this->db_provider) |
| | | { |
| | | switch($this->db_provider) { |
| | | case 'mysqli': |
| | | case 'mysql': |
| | | case 'sqlite': |
| | |
| | | function ilike($column, $value) |
| | | { |
| | | // TODO: use MDB2's matchPattern() function |
| | | switch($this->db_provider) |
| | | { |
| | | switch($this->db_provider) { |
| | | case 'pgsql': |
| | | return $this->quote_identifier($column).' ILIKE '.$this->quote($value); |
| | | default: |
| | |
| | | * @return mixed Handle ID |
| | | * @access private |
| | | */ |
| | | function _add_result($res) |
| | | private function _add_result($res) |
| | | { |
| | | // sql error occured |
| | | if (PEAR::isError($res)) |
| | | { |
| | | $this->db_error = TRUE; |
| | | if (PEAR::isError($res)) { |
| | | $this->db_error = true; |
| | | $this->db_error_msg = $res->getMessage(); |
| | | raise_error(array('code' => 500, 'type' => 'db', |
| | | 'line' => __LINE__, 'file' => __FILE__, |
| | | 'message' => $res->getMessage() . " Query: " |
| | | . substr(preg_replace('/[\r\n]+\s*/', ' ', $res->userinfo), 0, 512)), |
| | | TRUE, FALSE); |
| | | true, false); |
| | | } |
| | | |
| | | $res_id = sizeof($this->a_query_results); |
| | |
| | | * If no ID is specified, the last resource handle will be returned |
| | | * |
| | | * @param number Handle ID |
| | | * @return mixed Resource handle or FALSE on failure |
| | | * @return mixed Resource handle or false on failure |
| | | * @access private |
| | | */ |
| | | function _get_result($res_id=NULL) |
| | | private function _get_result($res_id = null) |
| | | { |
| | | if ($res_id==NULL) |
| | | if ($res_id == null) |
| | | $res_id = $this->last_res_id; |
| | | |
| | | if (isset($this->a_query_results[$res_id])) |
| | | if (!PEAR::isError($this->a_query_results[$res_id])) |
| | | return $this->a_query_results[$res_id]; |
| | | |
| | | return FALSE; |
| | | return false; |
| | | } |
| | | |
| | | |
| | |
| | | * @param string File path to use for DB creation |
| | | * @access private |
| | | */ |
| | | function _sqlite_create_database($dbh, $file_name) |
| | | private function _sqlite_create_database($dbh, $file_name) |
| | | { |
| | | if (empty($file_name) || !is_string($file_name)) |
| | | return; |
| | |
| | | if (strlen($data)) |
| | | if (!sqlite_exec($dbh->connection, $data, $error) || MDB2::isError($dbh)) |
| | | raise_error(array('code' => 500, 'type' => 'db', |
| | | 'line' => __LINE__, 'file' => __FILE__, 'message' => $error), TRUE, FALSE); |
| | | 'line' => __LINE__, 'file' => __FILE__, |
| | | 'message' => $error), true, false); |
| | | } |
| | | |
| | | |
| | |
| | | * |
| | | * @access private |
| | | */ |
| | | function _sqlite_prepare() |
| | | private function _sqlite_prepare() |
| | | { |
| | | include_once('include/rcube_sqlite.inc'); |
| | | |
| | | // we emulate via callback some missing MySQL function |
| | | sqlite_create_function($this->db_handle->connection, "from_unixtime", "rcube_sqlite_from_unixtime"); |
| | | sqlite_create_function($this->db_handle->connection, "unix_timestamp", "rcube_sqlite_unix_timestamp"); |
| | | sqlite_create_function($this->db_handle->connection, "now", "rcube_sqlite_now"); |
| | | sqlite_create_function($this->db_handle->connection, "md5", "rcube_sqlite_md5"); |
| | | sqlite_create_function($this->db_handle->connection, |
| | | 'from_unixtime', 'rcube_sqlite_from_unixtime'); |
| | | sqlite_create_function($this->db_handle->connection, |
| | | 'unix_timestamp', 'rcube_sqlite_unix_timestamp'); |
| | | sqlite_create_function($this->db_handle->connection, |
| | | 'now', 'rcube_sqlite_now'); |
| | | sqlite_create_function($this->db_handle->connection, |
| | | 'md5', 'rcube_sqlite_md5'); |
| | | } |
| | | |
| | | |
| | | } // end class rcube_db |
| | | |
| | |
| | | /* this is our own debug handler for the MDB2 connection */ |
| | | function mdb2_debug_handler(&$db, $scope, $message, $context = array()) |
| | | { |
| | | if ($scope != 'prepare') |
| | | { |
| | | if ($scope != 'prepare') { |
| | | $debug_output = $scope . '('.$db->db_index.'): ' . $message; |
| | | write_log('sql', $debug_output); |
| | | } |