Viewing file: DB.php (19.66 KB) -rw-rw-rw- Select action/file-type: (+) | (+) | (+) | Code (+) | Session (+) | (+) | SDB (+) | (+) | (+) | (+) | (+) | (+) |
<?php /* vim: set expandtab tabstop=4 shiftwidth=4 softtabstop=4 foldmethod=marker: */
/** * Storage driver for use against PEAR DB * * PHP versions 4 and 5 * * LICENSE: This source file is subject to version 3.01 of the PHP license * that is available through the world-wide-web at the following URI: * http://www.php.net/license/3_01.txt. If you did not receive a copy of * the PHP License and are unable to obtain it through the web, please * send a note to license@php.net so we can mail you a copy immediately. * * @category Authentication * @package Auth * @author Martin Jansen <mj@php.net> * @author Adam Ashley <aashley@php.net> * @copyright 2001-2006 The PHP Group * @license http://www.php.net/license/3_01.txt PHP License 3.01 * @version CVS: $Id: DB.php,v 1.71 2007/02/02 00:49:55 aashley Exp $ * @link http://pear.php.net/package/Auth */
/** * Include Auth_Container base class */ require_once 'Auth/Container.php'; /** * Include PEAR DB */ require_once 'DB.php';
/** * Storage driver for fetching login data from a database * * This storage driver can use all databases which are supported * by the PEAR DB abstraction layer to fetch login data. * * @category Authentication * @package Auth * @author Martin Jansen <mj@php.net> * @author Adam Ashley <aashley@php.net> * @copyright 2001-2006 The PHP Group * @license http://www.php.net/license/3_01.txt PHP License 3.01 * @version Release: 1.5.0 File: $Revision: 1.71 $ * @link http://pear.php.net/package/Auth */ class Auth_Container_DB extends Auth_Container {
// {{{ properties
/** * Additional options for the storage container * @var array */ var $options = array();
/** * DB object * @var object */ var $db = null; var $dsn = '';
/** * User that is currently selected from the DB. * @var string */ var $activeUser = '';
// }}} // {{{ Auth_Container_DB [constructor]
/** * Constructor of the container class * * Save the initial options passed to the container. Initiation of the DB * connection is no longer performed here and is only done when needed. * * @param string Connection data or DB object * @return object Returns an error object if something went wrong */ function Auth_Container_DB($dsn) { $this->_setDefaults();
if (is_array($dsn)) { $this->_parseOptions($dsn);
if (empty($this->options['dsn'])) { PEAR::raiseError('No connection parameters specified!'); } } else { $this->options['dsn'] = $dsn; } }
// }}} // {{{ _connect()
/** * Connect to database by using the given DSN string * * @access private * @param string DSN string * @return mixed Object on error, otherwise bool */ function _connect($dsn) { $this->log('Auth_Container_DB::_connect() called.', AUTH_LOG_DEBUG);
if (is_string($dsn) || is_array($dsn)) { $this->db = DB::Connect($dsn, $this->options['db_options']); } elseif (is_subclass_of($dsn, 'db_common')) { $this->db = $dsn; } elseif (DB::isError($dsn)) { return PEAR::raiseError($dsn->getMessage(), $dsn->getCode()); } else { return PEAR::raiseError('The given dsn was not valid in file ' . __FILE__ . ' at line ' . __LINE__, 41, PEAR_ERROR_RETURN, null, null ); }
if (DB::isError($this->db) || PEAR::isError($this->db)) { return PEAR::raiseError($this->db->getMessage(), $this->db->getCode()); } else { return true; } }
// }}} // {{{ _prepare()
/** * Prepare database connection * * This function checks if we have already opened a connection to * the database. If that's not the case, a new connection is opened. * * @access private * @return mixed True or a DB error object. */ function _prepare() { if (!DB::isConnection($this->db)) { $res = $this->_connect($this->options['dsn']); if (DB::isError($res) || PEAR::isError($res)) { return $res; } } if ($this->options['auto_quote'] && $this->db->dsn['phptype'] != 'sqlite') { $this->options['final_table'] = $this->db->quoteIdentifier($this->options['table']); $this->options['final_usernamecol'] = $this->db->quoteIdentifier($this->options['usernamecol']); $this->options['final_passwordcol'] = $this->db->quoteIdentifier($this->options['passwordcol']); } else { $this->options['final_table'] = $this->options['table']; $this->options['final_usernamecol'] = $this->options['usernamecol']; $this->options['final_passwordcol'] = $this->options['passwordcol']; } return true; }
// }}} // {{{ query()
/** * Prepare query to the database * * This function checks if we have already opened a connection to * the database. If that's not the case, a new connection is opened. * After that the query is passed to the database. * * @access public * @param string Query string * @return mixed a DB_result object or DB_OK on success, a DB * or PEAR error on failure */ function query($query) { $err = $this->_prepare(); if ($err !== true) { return $err; } return $this->db->query($query); }
// }}} // {{{ _setDefaults()
/** * Set some default options * * @access private * @return void */ function _setDefaults() { $this->options['table'] = 'auth'; $this->options['usernamecol'] = 'username'; $this->options['passwordcol'] = 'password'; $this->options['dsn'] = ''; $this->options['db_fields'] = ''; $this->options['cryptType'] = 'md5'; $this->options['db_options'] = array(); $this->options['db_where'] = ''; $this->options['auto_quote'] = true; }
// }}} // {{{ _parseOptions()
/** * Parse options passed to the container class * * @access private * @param array */ function _parseOptions($array) { foreach ($array as $key => $value) { if (isset($this->options[$key])) { $this->options[$key] = $value; } } }
// }}} // {{{ _quoteDBFields()
/** * Quote the db_fields option to avoid the possibility of SQL injection. * * @access private * @return string A properly quoted string that can be concatenated into a * SELECT clause. */ function _quoteDBFields() { if (isset($this->options['db_fields'])) { if (is_array($this->options['db_fields'])) { if ($this->options['auto_quote']) { $fields = array(); foreach ($this->options['db_fields'] as $field) { $fields[] = $this->db->quoteIdentifier($field); } return implode(', ', $fields); } else { return implode(', ', $this->options['db_fields']); } } else { if (strlen($this->options['db_fields']) > 0) { if ($this->options['auto_quote']) { return $this->db->quoteIdentifier($this->options['db_fields']); } else { return $this->options['db_fields']; } } } }
return ''; } // }}} // {{{ fetchData()
/** * Get user information from database * * This function uses the given username to fetch * the corresponding login data from the database * table. If an account that matches the passed username * and password is found, the function returns true. * Otherwise it returns false. * * @param string Username * @param string Password * @param boolean If true password is secured using a md5 hash * the frontend and auth are responsible for making sure the container supports * challenge response password authentication * @return mixed Error object or boolean */ function fetchData($username, $password, $isChallengeResponse=false) { $this->log('Auth_Container_DB::fetchData() called.', AUTH_LOG_DEBUG); // Prepare for a database query $err = $this->_prepare(); if ($err !== true) { return PEAR::raiseError($err->getMessage(), $err->getCode()); }
// Find if db_fields contains a *, if so assume all columns are selected if (is_string($this->options['db_fields']) && strstr($this->options['db_fields'], '*')) { $sql_from = "*"; } else { $sql_from = $this->options['final_usernamecol']. ", ".$this->options['final_passwordcol'];
if (strlen($fields = $this->_quoteDBFields()) > 0) { $sql_from .= ', '.$fields; } }
$query = "SELECT ".$sql_from. " FROM ".$this->options['final_table']. " WHERE ".$this->options['final_usernamecol']." = ".$this->db->quoteSmart($username);
// check if there is an optional parameter db_where if ($this->options['db_where'] != '') { // there is one, so add it to the query $query .= " AND ".$this->options['db_where']; }
$this->log('Running SQL against DB: '.$query, AUTH_LOG_DEBUG);
$res = $this->db->getRow($query, null, DB_FETCHMODE_ASSOC);
if (DB::isError($res)) { return PEAR::raiseError($res->getMessage(), $res->getCode()); }
if (!is_array($res)) { $this->activeUser = ''; return false; }
// Perform trimming here before the hashihg $password = trim($password, "\r\n"); $res[$this->options['passwordcol']] = trim($res[$this->options['passwordcol']], "\r\n");
// If using Challenge Response md5 the pass with the secret if ($isChallengeResponse) { $res[$this->options['passwordcol']] = md5($res[$this->options['passwordcol']] .$this->_auth_obj->session['loginchallenege']); // UGLY cannot avoid without modifying verifyPassword if ($this->options['cryptType'] == 'md5') { $res[$this->options['passwordcol']] = md5($res[$this->options['passwordcol']]); } //print " Hashed Password [{$res[$this->options['passwordcol']]}]<br/>\n"; }
if ($this->verifyPassword($password, $res[$this->options['passwordcol']], $this->options['cryptType'])) { // Store additional field values in the session foreach ($res as $key => $value) { if ($key == $this->options['passwordcol'] || $key == $this->options['usernamecol']) { continue; } $this->log('Storing additional field: '.$key, AUTH_LOG_DEBUG);
// Use reference to the auth object if exists // This is because the auth session variable can change so a // static call to setAuthData does not make sence $this->_auth_obj->setAuthData($key, $value); } return true; } $this->activeUser = $res[$this->options['usernamecol']]; return false; }
// }}} // {{{ listUsers()
/** * Returns a list of users from the container * * @return mixed * @access public */ function listUsers() { $this->log('Auth_Container_DB::listUsers() called.', AUTH_LOG_DEBUG); $err = $this->_prepare(); if ($err !== true) { return PEAR::raiseError($err->getMessage(), $err->getCode()); }
$retVal = array();
// Find if db_fields contains a *, if so assume all col are selected if ( is_string($this->options['db_fields']) && strstr($this->options['db_fields'], '*')) { $sql_from = "*"; } else { $sql_from = $this->options['final_usernamecol']. ", ".$this->options['final_passwordcol'];
if (strlen($fields = $this->_quoteDBFields()) > 0) { $sql_from .= ', '.$fields; } }
$query = sprintf("SELECT %s FROM %s", $sql_from, $this->options['final_table'] );
// check if there is an optional parameter db_where if ($this->options['db_where'] != '') { // there is one, so add it to the query $query .= " WHERE ".$this->options['db_where']; }
$this->log('Running SQL against DB: '.$query, AUTH_LOG_DEBUG);
$res = $this->db->getAll($query, null, DB_FETCHMODE_ASSOC);
if (DB::isError($res)) { return PEAR::raiseError($res->getMessage(), $res->getCode()); } else { foreach ($res as $user) { $user['username'] = $user[$this->options['usernamecol']]; $retVal[] = $user; } } $this->log('Found '.count($retVal).' users.', AUTH_LOG_DEBUG); return $retVal; }
// }}} // {{{ addUser()
/** * Add user to the storage container * * @access public * @param string Username * @param string Password * @param mixed Additional information that are stored in the DB * * @return mixed True on success, otherwise error object */ function addUser($username, $password, $additional = "") { $this->log('Auth_Container_DB::addUser() called.', AUTH_LOG_DEBUG); $err = $this->_prepare(); if ($err !== true) { return PEAR::raiseError($err->getMessage(), $err->getCode()); }
if ( isset($this->options['cryptType']) && $this->options['cryptType'] == 'none') { $cryptFunction = 'strval'; } elseif ( isset($this->options['cryptType']) && function_exists($this->options['cryptType'])) { $cryptFunction = $this->options['cryptType']; } else { $cryptFunction = 'md5'; }
$password = $cryptFunction($password);
$additional_key = ''; $additional_value = '';
if (is_array($additional)) { foreach ($additional as $key => $value) { if ($this->options['auto_quote']) { $additional_key .= ', ' . $this->db->quoteIdentifier($key); } else { $additional_key .= ', ' . $key; } $additional_value .= ", " . $this->db->quoteSmart($value); } }
$query = sprintf("INSERT INTO %s (%s, %s%s) VALUES (%s, %s%s)", $this->options['final_table'], $this->options['final_usernamecol'], $this->options['final_passwordcol'], $additional_key, $this->db->quoteSmart($username), $this->db->quoteSmart($password), $additional_value );
$this->log('Running SQL against DB: '.$query, AUTH_LOG_DEBUG);
$res = $this->query($query);
if (DB::isError($res)) { return PEAR::raiseError($res->getMessage(), $res->getCode()); } else { return true; } }
// }}} // {{{ removeUser()
/** * Remove user from the storage container * * @access public * @param string Username * * @return mixed True on success, otherwise error object */ function removeUser($username) { $this->log('Auth_Container_DB::removeUser() called.', AUTH_LOG_DEBUG);
$err = $this->_prepare(); if ($err !== true) { return PEAR::raiseError($err->getMessage(), $err->getCode()); }
// check if there is an optional parameter db_where if ($this->options['db_where'] != '') { // there is one, so add it to the query $where = " AND ".$this->options['db_where']; } else { $where = ''; }
$query = sprintf("DELETE FROM %s WHERE %s = %s %s", $this->options['final_table'], $this->options['final_usernamecol'], $this->db->quoteSmart($username), $where );
$this->log('Running SQL against DB: '.$query, AUTH_LOG_DEBUG);
$res = $this->query($query);
if (DB::isError($res)) { return PEAR::raiseError($res->getMessage(), $res->getCode()); } else { return true; } }
// }}} // {{{ changePassword()
/** * Change password for user in the storage container * * @param string Username * @param string The new password (plain text) */ function changePassword($username, $password) { $this->log('Auth_Container_DB::changePassword() called.', AUTH_LOG_DEBUG); $err = $this->_prepare(); if ($err !== true) { return PEAR::raiseError($err->getMessage(), $err->getCode()); }
if ( isset($this->options['cryptType']) && $this->options['cryptType'] == 'none') { $cryptFunction = 'strval'; } elseif ( isset($this->options['cryptType']) && function_exists($this->options['cryptType'])) { $cryptFunction = $this->options['cryptType']; } else { $cryptFunction = 'md5'; }
$password = $cryptFunction($password);
// check if there is an optional parameter db_where if ($this->options['db_where'] != '') { // there is one, so add it to the query $where = " AND ".$this->options['db_where']; } else { $where = ''; }
$query = sprintf("UPDATE %s SET %s = %s WHERE %s = %s %s", $this->options['final_table'], $this->options['final_passwordcol'], $this->db->quoteSmart($password), $this->options['final_usernamecol'], $this->db->quoteSmart($username), $where );
$this->log('Running SQL against DB: '.$query, AUTH_LOG_DEBUG);
$res = $this->query($query);
if (DB::isError($res)) { return PEAR::raiseError($res->getMessage(), $res->getCode()); } else { return true; } }
// }}} // {{{ supportsChallengeResponse()
/** * Determine if this container supports * password authentication with challenge response * * @return bool * @access public */ function supportsChallengeResponse() { return in_array($this->options['cryptType'], array('md5', 'none', '')); }
// }}} // {{{ getCryptType()
/** * Returns the selected crypt type for this container */ function getCryptType() { return($this->options['cryptType']); }
// }}}
} ?>
|