Aleksander Machniak
2014-09-22 000f6ee3041c0e587c4a54daa95dfc421f7bd2d4
Merge branch 'oracle'
5 files modified
507 ■■■■■ changed files
installer/check.php 1 ●●●● patch | view | raw | blame | history
program/include/rcmail_install.php 1 ●●●● patch | view | raw | blame | history
program/lib/Roundcube/rcube_db.php 104 ●●●● patch | view | raw | blame | history
program/lib/Roundcube/rcube_db_oracle.php 398 ●●●●● patch | view | raw | blame | history
program/lib/Roundcube/rcube_imap_cache.php 3 ●●●●● patch | view | raw | blame | history
installer/check.php
@@ -71,6 +71,7 @@
    'DOM'       => 'http://www.php.net/manual/en/book.dom.php',
    'Intl'      => 'http://www.php.net/manual/en/book.intl.php',
    'Exif'      => 'http://www.php.net/manual/en/book.exif.php',
    'oci8'      => 'http://www.php.net/manual/en/book.oci8.php',
    'PDO'       => 'http://www.php.net/manual/en/book.pdo.php',
    'LDAP'      => 'http://www.php.net/manual/en/book.ldap.php',
    'pdo_mysql'   => 'http://www.php.net/manual/en/ref.pdo-mysql.php',
program/include/rcmail_install.php
@@ -55,6 +55,7 @@
    'SQLite (v2)'         => 'pdo_sqlite2',
    'SQL Server (SQLSRV)' => 'pdo_sqlsrv',
    'SQL Server (DBLIB)'  => 'pdo_dblib',
    'Oracle'              => 'oci8',
  );
program/lib/Roundcube/rcube_db.php
@@ -70,6 +70,7 @@
            'dblib'   => 'mssql',
            'mysqli'  => 'mysql',
            'oci'     => 'oracle',
            'oci8'    => 'oracle',
        );
        $driver = isset($driver_map[$driver]) ? $driver_map[$driver] : $driver;
@@ -130,6 +131,20 @@
            return $this->dbh;
        }
        // connect to database
        if ($dbh = $this->conn_create($dsn)) {
            $this->dbh          = $dbh;
            $this->dbhs[$mode]  = $dbh;
            $this->db_mode      = $mode;
            $this->db_connected = true;
        }
    }
    /**
     * Create PDO connection
     */
    protected function conn_create($dsn)
    {
        // Get database specific connection options
        $dsn_string  = $this->dsn_string($dsn);
        $dsn_options = $this->dsn_options($dsn);
@@ -151,6 +166,8 @@
            // don't throw exceptions or warnings
            $dbh->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_SILENT);
            $this->conn_configure($dsn, $dbh);
        }
        catch (Exception $e) {
            $this->db_error     = true;
@@ -163,11 +180,7 @@
            return null;
        }
        $this->dbh          = $dbh;
        $this->dbhs[$mode]  = $dbh;
        $this->db_mode      = $mode;
        $this->db_connected = true;
        $this->conn_configure($dsn, $dbh);
        return $dbh;
    }
    /**
@@ -442,8 +455,16 @@
        // log query
        $this->debug($query);
        return $this->query_execute($query);
    }
    /**
     * Query execution
     */
    protected function query_execute($query)
    {
        // destroy reference to previous result, required for SQLite driver (#1488874)
        $this->last_result = null;
        $this->last_result  = null;
        $this->db_error_msg = null;
        // send query
@@ -453,9 +474,7 @@
            $result = $this->handle_error($query);
        }
        $this->last_result = $result;
        return $result;
        return $this->last_result = $result;
    }
    /**
@@ -537,7 +556,9 @@
    public function affected_rows($result = null)
    {
        if ($result || ($result === null && ($result = $this->last_result))) {
            return $result->rowCount();
            if ($result !== true) {
                return $result->rowCount();
            }
        }
        return 0;
@@ -553,7 +574,7 @@
     */
    public function num_rows($result = null)
    {
        if ($result || ($result === null && ($result = $this->last_result))) {
        if (($result || ($result === null && ($result = $this->last_result))) && $result !== true) {
            // repeat query with SELECT COUNT(*) ...
            if (preg_match('/^SELECT\s+(?:ALL\s+|DISTINCT\s+)?(?:.*?)\s+FROM\s+(.*)$/ims', $result->queryString, $m)) {
                $query = $this->dbh->query('SELECT COUNT(*) FROM ' . $m[1], PDO::FETCH_NUM);
@@ -629,7 +650,9 @@
    protected function _fetch_row($result, $mode)
    {
        if ($result || ($result === null && ($result = $this->last_result))) {
            return $result->fetch($mode);
            if ($result !== true) {
                return $result->fetch($mode);
            }
        }
        return false;
@@ -699,6 +722,63 @@
    }
    /**
     * Start transaction
     *
     * @return bool True on success, False on failure
     */
    public function startTransaction()
    {
        $this->db_connect('w', true);
        // check connection before proceeding
        if (!$this->is_connected()) {
            return $this->last_result = false;
        }
        $this->debug('BEGIN TRANSACTION');
        return $this->last_result = $this->dbh->beginTransaction();
    }
    /**
     * Commit transaction
     *
     * @return bool True on success, False on failure
     */
    public function endTransaction()
    {
        $this->db_connect('w', true);
        // check connection before proceeding
        if (!$this->is_connected()) {
            return $this->last_result = false;
        }
        $this->debug('COMMIT TRANSACTION');
        return $this->last_result = $this->dbh->commit();
    }
    /**
     * Rollback transaction
     *
     * @return bool True on success, False on failure
     */
    public function rollbackTransaction()
    {
        $this->db_connect('w', true);
        // check connection before proceeding
        if (!$this->is_connected()) {
            return $this->last_result = false;
        }
        $this->debug('ROLLBACK TRANSACTION');
        return $this->last_result = $this->dbh->rollBack();
    }
    /**
     * Formats input so it can be safely used in a query
     *
     * @param mixed  $input Value to quote
program/lib/Roundcube/rcube_db_oracle.php
@@ -10,8 +10,8 @@
 | See the README file for a full license statement.                     |
 |                                                                       |
 | PURPOSE:                                                              |
 |   Database wrapper class that implements PHP PDO functions            |
 |   for Oracle database                                                 |
 |   Database wrapper class that implements database functions           |
 |   for Oracle database using OCI8 extension                            |
 +-----------------------------------------------------------------------+
 | Author: Aleksander Machniak <machniak@kolabsys.com>                   |
 +-----------------------------------------------------------------------+
@@ -19,7 +19,6 @@
/**
 * Database independent query interface
 * This is a wrapper for the PHP PDO
 *
 * @package    Framework
 * @subpackage Database
@@ -27,6 +26,49 @@
class rcube_db_oracle extends rcube_db
{
    public $db_provider = 'oracle';
    /**
     * Create connection instance
     */
    protected function conn_create($dsn)
    {
        // Get database specific connection options
        $dsn_options = $this->dsn_options($dsn);
        $function = $this->db_pconn ? 'oci_pconnect' : 'oci_connect';
        if (!function_exists($function)) {
            $this->db_error     = true;
            $this->db_error_msg = 'OCI8 extension not loaded. See http://php.net/manual/en/book.oci8.php';
            rcube::raise_error(array('code' => 500, 'type' => 'db',
                'line' => __LINE__, 'file' => __FILE__,
                'message' => $this->db_error_msg), true, false);
            return;
        }
        // connect
        $dbh = @$function($dsn['username'], $dsn['password'], $dsn_options['database'], $dsn_options['charset']);
        if (!$dbh) {
            $error              = oci_error();
            $this->db_error     = true;
            $this->db_error_msg = $error['message'];
            rcube::raise_error(array('code' => 500, 'type' => 'db',
                'line' => __LINE__, 'file' => __FILE__,
                'message' => $this->db_error_msg), true, false);
            return;
        }
        // configure session
        $this->conn_configure($dsn, $dbh);
        return $dbh;
    }
    /**
     * Driver-specific configuration of database connection
@@ -36,8 +78,135 @@
     */
    protected function conn_configure($dsn, $dbh)
    {
        $dbh->query("ALTER SESSION SET nls_date_format = 'YYYY-MM-DD'");
        $dbh->query("ALTER SESSION SET nls_timestamp_format = 'YYYY-MM-DD HH24:MI:SS'");
        $init_queries = array(
            "ALTER SESSION SET nls_date_format = 'YYYY-MM-DD'",
            "ALTER SESSION SET nls_timestamp_format = 'YYYY-MM-DD HH24:MI:SS'",
        );
        foreach ($init_queries as $query) {
            $stmt = oci_parse($dbh, $query);
            oci_execute($stmt);
        }
    }
    /**
     * Connection state checker
     *
     * @return boolean True if in connected state
     */
    public function is_connected()
    {
        return empty($this->dbh) ? false : $this->db_connected;
    }
    /**
     * Execute a SQL query with limits
     *
     * @param string $query   SQL query to execute
     * @param int    $offset  Offset for LIMIT statement
     * @param int    $numrows Number of rows for LIMIT statement
     * @param array  $params  Values to be inserted in query
     *
     * @return PDOStatement|bool Query handle or False on error
     */
    protected function _query($query, $offset, $numrows, $params)
    {
        $query = ltrim($query);
        $this->db_connect($this->dsn_select($query), true);
        // check connection before proceeding
        if (!$this->is_connected()) {
            return $this->last_result = false;
        }
        if ($numrows || $offset) {
            $query = $this->set_limit($query, $numrows, $offset);
        }
        // replace self::DEFAULT_QUOTE with driver-specific quoting
        $query = $this->query_parse($query);
        // Because in Roundcube we mostly use queries that are
        // executed only once, we will not use prepared queries
        $pos  = 0;
        $idx  = 0;
        $args = array();
        if (count($params)) {
            while ($pos = strpos($query, '?', $pos)) {
                if ($query[$pos+1] == '?') {  // skip escaped '?'
                    $pos += 2;
                }
                else {
                    $val = $this->quote($params[$idx++]);
                    // long strings are not allowed inline, need to be parametrized
                    if (strlen($val) > 4000) {
                        $key = ':param' . (count($args) + 1);
                        $args[$key] = $params[$idx-1];
                        $val = $key;
                    }
                    unset($params[$idx-1]);
                    $query = substr_replace($query, $val, $pos, 1);
                    $pos += strlen($val);
                }
            }
        }
        // replace escaped '?' back to normal, see self::quote()
        $query = str_replace('??', '?', $query);
        $query = rtrim($query, " \t\n\r\0\x0B;");
        // log query
        $this->debug($query);
        // destroy reference to previous result
        $this->last_result  = null;
        $this->db_error_msg = null;
        // prepare query
        $result = @oci_parse($this->dbh, $query);
        $mode   = $this->in_transaction ? OCI_NO_AUTO_COMMIT : OCI_COMMIT_ON_SUCCESS;
        if ($result) {
            foreach ($args as $param => $arg) {
                oci_bind_by_name($result, $param, $args[$param], -1, SQLT_LNG);
            }
        }
        // execute query
        if (!$result || !@oci_execute($result, $mode)) {
            $result = $this->handle_error($query, $result);
        }
        return $this->last_result = $result;
    }
    /**
     * Helper method to handle DB errors.
     * This by default logs the error but could be overriden by a driver implementation
     *
     * @param string Query that triggered the error
     * @return mixed Result to be stored and returned
     */
    protected function handle_error($query, $result = null)
    {
        $error = oci_error(is_resource($result) ? $result : $this->dbh);
        // @TODO: Find error codes for key errors
        if (empty($this->options['ignore_key_errors']) || !in_array($error['code'], array('23000', '23505'))) {
            $this->db_error = true;
            $this->db_error_msg = sprintf('[%s] %s', $error['code'], $error['message']);
            rcube::raise_error(array('code' => 500, 'type' => 'db',
                'line' => __LINE__, 'file' => __FILE__,
                'message' => $this->db_error_msg . " (SQL Query: $query)"
                ), true, false);
        }
        return false;
    }
    /**
@@ -54,9 +223,83 @@
        }
        $sequence = $this->quote_identifier($this->sequence_name($table));
        $result   = $dbh->query("SELECT $sequence.currval FROM dual");
        $result   = $this->query("SELECT $sequence.currval FROM dual");
        $result   = $this->fetch_array($result);
        return $result ? $result->fetchColumn() : false;
        return $result[0] ?: false;
    }
    /**
     * Get number of affected rows for the last query
     *
     * @param mixed $result Optional query handle
     *
     * @return int Number of (matching) rows
     */
    public function affected_rows($result = null)
    {
        if ($result || ($result === null && ($result = $this->last_result))) {
            return oci_num_rows($result);
        }
        return 0;
    }
    /**
     * Get number of rows for a SQL query
     * If no query handle is specified, the last query will be taken as reference
     *
     * @param mixed $result Optional query handle
     * @return mixed   Number of rows or false on failure
     * @deprecated This method shows very poor performance and should be avoided.
     */
    public function num_rows($result = null)
    {
        // not implemented
        return false;
    }
    /**
     * Get an associative array for one row
     * If no query handle is specified, the last query will be taken as reference
     *
     * @param mixed $result Optional query handle
     *
     * @return mixed Array with col values or false on failure
     */
    public function fetch_assoc($result = null)
    {
        return $this->_fetch_row($result, OCI_ASSOC);
    }
    /**
     * Get an index array for one row
     * If no query handle is specified, the last query will be taken as reference
     *
     * @param mixed $result Optional query handle
     *
     * @return mixed Array with col values or false on failure
     */
    public function fetch_array($result = null)
    {
        return $this->_fetch_row($result, OCI_NUM);
    }
    /**
     * Get col values for a result row
     *
     * @param mixed $result Optional query handle
     * @param int   $mode   Fetch mode identifier
     *
     * @return mixed Array with col values or false on failure
     */
    protected function _fetch_row($result, $mode)
    {
        if ($result || ($result === null && ($result = $this->last_result))) {
            return oci_fetch_array($result, $mode + OCI_RETURN_NULLS + OCI_RETURN_LOBS);
        }
        return false;
    }
    /**
@@ -177,22 +420,12 @@
        // @TODO: Oracle 12g has better OFFSET support
        $orderby = stristr($query, 'ORDER BY');
        $select  = substr($query, 0, stripos($query, 'FROM'));
        $offset += 1;
        if ($orderby !== false) {
            $query = trim(substr($query, 0, -1 * strlen($orderby)));
        if (!$offset) {
            $query = "SELECT * FROM ($query) a WHERE rownum <= $end";
        }
        else {
            // it shouldn't happen, paging without sorting has not much sense
            // @FIXME: I don't know how to build paging query without ORDER BY
            $orderby = "ORDER BY 1";
            $query = "SELECT * FROM (SELECT a.*, rownum as rn FROM ($query) a WHERE rownum <= $end) b WHERE rn > $offset";
        }
        $query = preg_replace('/^SELECT\s/i', '', $query);
        $query = "$select FROM (SELECT ROW_NUMBER() OVER ($orderby) AS row_number, $query)"
            . " WHERE row_number BETWEEN $offset AND $end";
        return $query;
    }
@@ -232,12 +465,11 @@
    }
    /**
     * Returns PDO DSN string from DSN array
     * Returns connection options from DSN array
     */
    protected function dsn_string($dsn)
    protected function dsn_options($dsn)
    {
        $params = array();
        $result = 'oci:';
        if ($dsn['hostspec']) {
            $host = $dsn['hostspec'];
@@ -245,19 +477,123 @@
                $host .= ':' . $dsn['port'];
            }
            $dsn['database'] = $host . '/' . $dsn['database'];
        }
        if ($dsn['database']) {
            $params[] = 'dbname=' . $dsn['database'];
            $params['database'] = $host . '/' . $dsn['database'];
        }
        $params['charset'] = 'UTF8';
        if (!empty($params)) {
            $result .= implode(';', $params);
        return $params;
    }
    /**
     * Execute the given SQL script
     *
     * @param string SQL queries to execute
     *
     * @return boolen True on success, False on error
     */
    public function exec_script($sql)
    {
        $sql  = $this->fix_table_names($sql);
        $buff = '';
        $body = false;
        foreach (explode("\n", $sql) as $line) {
            $tok = strtolower(trim($line));
            if (preg_match('/^--/', $line) || $tok == '') {
                continue;
            }
            $buff .= $line . "\n";
            // detect PL/SQL function bodies, don't break on semicolon
            if ($body && $tok == 'end;') {
                $body = false;
            }
            else if (!$body && $tok == 'begin') {
                $body = true;
            }
            if (!$body && substr($tok, -1) == ';') {
                $this->query($buff);
                $buff = '';
                if ($this->db_error) {
                    break;
                }
            }
        }
        return $result;
        return !$this->db_error;
    }
    /**
     * Start transaction
     *
     * @return bool True on success, False on failure
     */
    public function startTransaction()
    {
        $this->db_connect('w', true);
        // check connection before proceeding
        if (!$this->is_connected()) {
            return $this->last_result = false;
        }
        $this->debug('BEGIN TRANSACTION');
        return $this->last_result = $this->in_transaction = true;
    }
    /**
     * Commit transaction
     *
     * @return bool True on success, False on failure
     */
    public function endTransaction()
    {
        $this->db_connect('w', true);
        // check connection before proceeding
        if (!$this->is_connected()) {
            return $this->last_result = false;
        }
        $this->debug('COMMIT TRANSACTION');
        if ($result = @oci_commit($this->dbh)) {
            $this->in_transaction = true;
        }
        else {
            $this->handle_error('COMMIT');
        }
        return $this->last_result = $result;
    }
    /**
     * Rollback transaction
     *
     * @return bool True on success, False on failure
     */
    public function rollbackTransaction()
    {
        $this->db_connect('w', true);
        // check connection before proceeding
        if (!$this->is_connected()) {
            return $this->last_result = false;
        }
        $this->debug('ROLLBACK TRANSACTION');
        if ($result = @oci_rollback($this->dbh)) {
            $this->in_transaction = false;
        }
        else {
            $this->handle_error('ROLLBACK');
        }
        return $this->last_result = $this->dbh->rollBack();
    }
}
program/lib/Roundcube/rcube_imap_cache.php
@@ -562,6 +562,8 @@
            }
        }
        $binary_check = $this->db->db_provider == 'oracle' ? "BITAND(`flags`, %d)" : "(`flags` & %d)";
        $this->db->query(
            "UPDATE {$this->messages_table}"
            ." SET `expires` = ". ($this->ttl ? $this->db->now($this->ttl) : 'NULL')
@@ -570,6 +572,7 @@
                ." AND `mailbox` = ?"
                .(!empty($uids) ? " AND `uid` IN (".$this->db->array2list($uids, 'integer').")" : "")
                ." AND (`flags` & $idx) ".($enabled ? "= 0" : "= $idx"),
                ." AND " . sprintf($binary_check, $idx) . ($enabled ? " = 0" : " = $idx"),
            $this->userid, $mailbox);
    }