351 lines
8.3 KiB
PHP
351 lines
8.3 KiB
PHP
<?php
|
|
|
|
# Copyright (C) 2006-2011 Tobias Leupold <tobias.leupold@web.de>
|
|
#
|
|
# This file is part of the b8 package
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it
|
|
# under the terms of the GNU Lesser General Public License as published by
|
|
# the Free Software Foundation in version 2.1 of the License.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
# or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
|
|
# License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
/**
|
|
* The MySQL abstraction layer for communicating with the database.
|
|
* Copyright (C) 2009 Oliver Lillie (aka buggedcom)
|
|
* Copyright (C) 2010-2011 Tobias Leupold <tobias.leupold@web.de>
|
|
*
|
|
* @license LGPL
|
|
* @access public
|
|
* @package b8
|
|
* @author Oliver Lillie (aka buggedcom) (original PHP 5 port and optimizations)
|
|
* @author Tobias Leupold
|
|
*/
|
|
|
|
class b8_storage_mysql extends b8_storage_base
|
|
{
|
|
|
|
public $config = array(
|
|
'database' => 'b8_wordlist',
|
|
'table_name' => 'b8_wordlist',
|
|
'host' => 'localhost',
|
|
'user' => FALSE,
|
|
'pass' => FALSE,
|
|
'connection' => NULL
|
|
);
|
|
|
|
public $b8_config = array(
|
|
'degenerator' => NULL,
|
|
'today' => NULL
|
|
);
|
|
|
|
private $_connection = NULL;
|
|
private $_deletes = array();
|
|
private $_puts = array();
|
|
private $_updates = array();
|
|
|
|
const DATABASE_CONNECTION_FAIL = 'DATABASE_CONNECTION_FAIL';
|
|
const DATABASE_CONNECTION_ERROR = 'DATABASE_CONNECTION_ERROR';
|
|
const DATABASE_CONNECTION_BAD_RESOURCE = 'DATABASE_CONNECTION_BAD_RESOURCE';
|
|
const DATABASE_SELECT_ERROR = 'DATABASE_SELECT_ERROR';
|
|
const DATABASE_TABLE_ACCESS_FAIL = 'DATABASE_TABLE_ACCESS_FAIL';
|
|
const DATABASE_WRONG_VERSION = 'DATABASE_WRONG_VERSION';
|
|
|
|
/**
|
|
* Constructs the database layer.
|
|
*
|
|
* @access public
|
|
* @param string $config
|
|
*/
|
|
|
|
function __construct($config, $degenerator, $today)
|
|
{
|
|
|
|
# Pass some variables of the main b8 config to this class
|
|
$this->b8_config['degenerator'] = $degenerator;
|
|
$this->b8_config['today'] = $today;
|
|
|
|
# Validate the config items
|
|
|
|
if(count($config) > 0) {
|
|
|
|
foreach ($config as $name => $value) {
|
|
|
|
switch($name) {
|
|
|
|
case 'table_name':
|
|
case 'host':
|
|
case 'user':
|
|
case 'pass':
|
|
case 'database':
|
|
$this->config[$name] = (string) $value;
|
|
break;
|
|
|
|
case 'connection':
|
|
|
|
if($value !== NULL) {
|
|
|
|
if(is_resource($value) === TRUE) {
|
|
$resource_type = get_resource_type($value);
|
|
$this->config['connection'] = $resource_type !== 'mysql link' && $resource_type !== 'mysql link persistent' ? FALSE : $value;
|
|
}
|
|
|
|
else
|
|
$this->config['connection'] = FALSE;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
/**
|
|
* Closes the database connection.
|
|
*
|
|
* @access public
|
|
* @return void
|
|
*/
|
|
|
|
function __destruct()
|
|
{
|
|
|
|
if($this->_connection === NULL)
|
|
return;
|
|
|
|
# Commit any changes before closing
|
|
$this->_commit();
|
|
|
|
# Just close the connection if no link-resource was passed and b8 created it's own connection
|
|
if($this->config['connection'] === NULL)
|
|
mysql_close($this->_connection);
|
|
|
|
$this->connected = FALSE;
|
|
|
|
}
|
|
|
|
/**
|
|
* Connect to the database and do some checks.
|
|
*
|
|
* @access public
|
|
* @return mixed Returns TRUE on a successful database connection, otherwise returns a constant from b8.
|
|
*/
|
|
|
|
public function connect()
|
|
{
|
|
|
|
# Are we already connected?
|
|
if($this->connected === TRUE)
|
|
return TRUE;
|
|
|
|
# Are we using an existing passed resource?
|
|
if($this->config['connection'] === FALSE) {
|
|
# ... yes we are, but the connection is not a resource, so return an error
|
|
$this->connected = FALSE;
|
|
return self::DATABASE_CONNECTION_BAD_RESOURCE;
|
|
}
|
|
|
|
elseif($this->config['connection'] === NULL) {
|
|
|
|
# ... no we aren't so we have to connect.
|
|
|
|
if($this->_connection = mysql_connect($this->config['host'], $this->config['user'], $this->config['pass'])) {
|
|
if(mysql_select_db($this->config['database'], $this->_connection) === FALSE) {
|
|
$this->connected = FALSE;
|
|
return self::DATABASE_SELECT_ERROR . ": " . mysql_error();
|
|
}
|
|
}
|
|
else {
|
|
$this->connected = FALSE;
|
|
return self::DATABASE_CONNECTION_ERROR;
|
|
}
|
|
|
|
}
|
|
|
|
else {
|
|
# ... yes we are
|
|
$this->_connection = $this->config['connection'];
|
|
}
|
|
|
|
# Just in case ...
|
|
if($this->_connection === NULL) {
|
|
$this->connected = FALSE;
|
|
return self::DATABASE_CONNECTION_FAIL;
|
|
}
|
|
|
|
# Check to see if the wordlist table exists
|
|
if(mysql_query('DESCRIBE ' . $this->config['table_name'], $this->_connection) === FALSE) {
|
|
$this->connected = FALSE;
|
|
return self::DATABASE_TABLE_ACCESS_FAIL . ": " . mysql_error();
|
|
}
|
|
|
|
# Everything is okay and connected
|
|
$this->connected = TRUE;
|
|
|
|
# Let's see if this is a b8 database and the version is okay
|
|
return $this->check_database();
|
|
|
|
}
|
|
|
|
/**
|
|
* Does the actual interaction with the database when fetching data.
|
|
*
|
|
* @access protected
|
|
* @param array $tokens
|
|
* @return mixed Returns an array of the returned data in the format array(token => data) or an empty array if there was no data.
|
|
*/
|
|
|
|
protected function _get_query($tokens)
|
|
{
|
|
|
|
# Construct the query ...
|
|
|
|
if(count($tokens) > 0) {
|
|
|
|
$where = array();
|
|
|
|
foreach ($tokens as $token) {
|
|
$token = mysql_real_escape_string($token, $this->_connection);
|
|
array_push($where, $token);
|
|
}
|
|
|
|
$where = 'token IN ("' . implode('", "', $where) . '")';
|
|
}
|
|
|
|
else {
|
|
$token = mysql_real_escape_string($token, $this->_connection);
|
|
$where = 'token = "' . $token . '"';
|
|
}
|
|
|
|
# ... and fetch the data
|
|
|
|
$result = mysql_query('
|
|
SELECT token, count
|
|
FROM ' . $this->config['table_name'] . '
|
|
WHERE ' . $where . ';
|
|
', $this->_connection);
|
|
|
|
$data = array();
|
|
|
|
while ($row = mysql_fetch_array($result, MYSQL_ASSOC))
|
|
$data[$row['token']] = $row['count'];
|
|
|
|
mysql_free_result($result);
|
|
|
|
return $data;
|
|
|
|
}
|
|
|
|
/**
|
|
* Store a token to the database.
|
|
*
|
|
* @access protected
|
|
* @param string $token
|
|
* @param string $count
|
|
* @return void
|
|
*/
|
|
|
|
protected function _put($token, $count) {
|
|
$token = mysql_real_escape_string($token, $this->_connection);
|
|
$count = mysql_real_escape_string($count, $this->_connection);;
|
|
array_push($this->_puts, '("' . $token . '", "' . $count . '")');
|
|
}
|
|
|
|
/**
|
|
* Update an existing token.
|
|
*
|
|
* @access protected
|
|
* @param string $token
|
|
* @param string $count
|
|
* @return void
|
|
*/
|
|
|
|
protected function _update($token, $count)
|
|
{
|
|
$token = mysql_real_escape_string($token, $this->_connection);
|
|
$count = mysql_real_escape_string($count, $this->_connection);
|
|
array_push($this->_updates, '("' . $token . '", "' . $count . '")');
|
|
}
|
|
|
|
/**
|
|
* Remove a token from the database.
|
|
*
|
|
* @access protected
|
|
* @param string $token
|
|
* @return void
|
|
*/
|
|
|
|
protected function _del($token)
|
|
{
|
|
$token = mysql_real_escape_string($token, $this->_connection);
|
|
array_push($this->_deletes, $token);
|
|
}
|
|
|
|
/**
|
|
* Commits any modification queries.
|
|
*
|
|
* @access protected
|
|
* @return void
|
|
*/
|
|
|
|
protected function _commit()
|
|
{
|
|
|
|
if(count($this->_deletes) > 0) {
|
|
|
|
$result = mysql_query('
|
|
DELETE FROM ' . $this->config['table_name'] . '
|
|
WHERE token IN ("' . implode('", "', $this->_deletes) . '");
|
|
', $this->_connection);
|
|
|
|
if(is_resource($result) === TRUE)
|
|
mysql_free_result($result);
|
|
|
|
$this->_deletes = array();
|
|
|
|
}
|
|
|
|
if(count($this->_puts) > 0) {
|
|
|
|
$result = mysql_query('
|
|
INSERT INTO ' . $this->config['table_name'] . '(token, count)
|
|
VALUES ' . implode(', ', $this->_puts) . ';', $this->_connection);
|
|
|
|
if(is_resource($result) === TRUE)
|
|
mysql_free_result($result);
|
|
|
|
$this->_puts = array();
|
|
|
|
}
|
|
|
|
if(count($this->_updates) > 0) {
|
|
|
|
$result = mysql_query('
|
|
INSERT INTO ' . $this->config['table_name'] . '(token, count)
|
|
VALUES ' . implode(', ', $this->_updates) . '
|
|
ON DUPLICATE KEY UPDATE ' . $this->config['table_name'] . '.count = VALUES(count);', $this->_connection);
|
|
|
|
if(is_resource($result) === TRUE)
|
|
mysql_free_result($result);
|
|
|
|
$this->_updates = array();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
?>
|