Files
sysPass/inc/db.class.php
nuxsmin 7e46e7bc32 * Added database connection port
* Fixed some UI errors
* Fixed error recursion on database connection error (needs more testing)
2015-03-31 00:06:29 +02:00

271 lines
8.4 KiB
PHP

<?php
/**
* sysPass
*
* @author nuxsmin
* @link http://syspass.org
* @copyright 2012-2015 Rubén Domínguez nuxsmin@syspass.org
*
* This file is part of sysPass.
*
* sysPass is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* sysPass 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 General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with sysPass. If not, see <http://www.gnu.org/licenses/>.
*
*/
defined('APP_ROOT') || die(_('No es posible acceder directamente a este archivo'));
/**
* Esta clase es la encargada de realizar las operaciones con la BBDD de sysPass.
*/
class DB
{
static $last_result;
static $affected_rows;
static $lastId;
static $txtError;
static $numError;
static $num_rows;
static $num_fields;
private static $_db;
/**
* Comprobar que la base de datos existe.
*
* @return bool
*/
public static function checkDatabaseExist()
{
if (!self::connection()) {
return false;
}
$query = 'SELECT COUNT(*) '
. 'FROM information_schema.tables'
. " WHERE table_schema='" . SP_Config::getValue("dbname") . "' "
. "AND table_name = 'usrData';";
$resquery = self::$_db->query($query);
if ($resquery) {
$row = $resquery->fetch_row();
}
if (!$resquery || $row[0] == 0) {
return false;
}
return true;
}
/**
* Realizar la conexión con la BBDD.
* Esta función utiliza mysqli para conectar con la base de datos.
* Guarda el objeto creado en la variable $_db de la clase
*
* @return bool
*/
private static function connection()
{
if (is_object(self::$_db)) {
return true;
}
$isInstalled = SP_Config::getValue('installed');
$dbhost = SP_Config::getValue('dbhost');
$dbuser = SP_Config::getValue('dbuser');
$dbpass = SP_Config::getValue('dbpass');
$dbname = SP_Config::getValue('dbname');
$dbport = SP_Config::getValue('dbport', 3306);
if (empty($dbhost) || empty($dbuser) || empty($dbpass) || empty($dbname)) {
if ($isInstalled) {
SP_Init::initError(_('No es posible conectar con la BD'), _('Compruebe los datos de conexión'));
} else {
return false;
}
}
self::$_db = @new mysqli($dbhost, $dbuser, $dbpass, $dbname, $dbport);
if (!is_object(self::$_db) || self::$_db->connect_errno) {
if ($isInstalled) {
if (self::$_db->connect_errno === 1049) {
SP_Config::setValue('installed', '0');
}
SP_Init::initError(_('No es posible conectar con la BD'), 'Error ' . self::$_db->connect_errno . ': ' . self::$_db->connect_error);
} else {
return false;
}
}
if (!self::$_db->set_charset("utf8")) {
SP_Init::initError(_('No es posible conectar con la BD'), 'Error ' . self::$_db->connect_errno . ': ' . self::$_db->connect_error);
}
return true;
}
/**
* Obtener los datos para generar un select.
*
* @param string $tblName con el nombre de la tabla a cunsultar
* @param string $tblColId con el nombre de la columna del tipo Id a mostrar
* @param string $tblColName con el nombre de la columna del tipo Name a mostrar
* @param array $arrFilter con las columnas a filtrar
* @param array $arrOrder con el orden de las columnas
* @return false|array con los valores del select con el Id como clave y el nombre como valor
*/
public static function getValuesForSelect($tblName, $tblColId, $tblColName, $arrFilter = NULL, $arrOrder = NULL)
{
if (!$tblName || !$tblColId || !$tblColName) {
return;
}
$strFilter = (is_array($arrFilter)) ? " WHERE " . implode(" OR ", $arrFilter) : "";
$strOrder = (is_array($arrOrder)) ? " ORDER BY " . implode(",", $arrOrder) : 'ORDER BY ' . $tblColName . ' ASC';
$query = "SELECT $tblColId, $tblColName FROM $tblName $strFilter $strOrder";
$queryRes = self::getResults($query, __FUNCTION__, true);
if ($queryRes === false) {
return false;
}
$arrValues = array();
foreach ($queryRes as $row) {
$arrValues[$row->$tblColId] = $row->$tblColName;
}
return $arrValues;
}
/**
* Obtener los resultados de una consulta.
*
* @param string $query con la consulta a realizar
* @param string $querySource con el nombre de la función que realiza la consulta
* @param bool $retArray devolver un array si la consulta tiene esultados
* @param bool $unbuffered devolver el resultado registro a registro
* @return bool|array devuelve bool si hay un error. Devuelve array con el array de registros devueltos
*/
public static function getResults($query, $querySource, $retArray = false, $unbuffered = false)
{
if ($query) {
self::doQuery($query, $querySource, $unbuffered);
}
if (self::$numError || self::$num_rows === 0) {
return false;
}
if (is_null(self::$numError) && count(self::$last_result) === 0) {
return true;
}
if ($retArray === true && is_object(self::$last_result)) {
return array(self::$last_result);
}
return self::$last_result;
}
/**
* Realizar una consulta a la BBDD.
*
* @param string $query con la consulta a realizar
* @param string $querySource con el nombre de la función que realiza la consulta
* @param bool $unbuffered realizar la consulta para obtener registro a registro
* @return false|int devuelve bool si hay un error. Devuelve int con el número de registros
*/
public static function doQuery($query, $querySource, $unbuffered = false)
{
if (!self::connection() || !is_object(self::$_db)) {
return false;
}
$isSelect = preg_match("/^.*(select|show)\s/i", $query);
// Limpiar valores de caché y errores
self::$last_result = array();
self::$numError = 0;
self::$txtError = '';
// Comprobamos si la consulta debe de ser devuelta completa o por registro
if (!$unbuffered) {
$queryRes = self::$_db->query($query);
} else {
$queryRes = self::$_db->real_query($query);
}
if (!$queryRes) {
self::$numError = self::$_db->errno;
self::$txtError = self::$_db->error;
$message['action'] = $querySource;
$message['text'][] = self::$_db->error . '(' . self::$_db->errno . ')';
$message['text'][] = "SQL: " . self::escape($query);
SP_Log::wrLogInfo($message);
return false;
}
if ($isSelect) {
//self::$num_rows = $queryRes->num_rows;
self::$num_rows = self::$_db->affected_rows;
if (!$unbuffered) {
self::$num_fields = self::$_db->field_count;
if (self::$num_rows === 1) {
self::$last_result = @$queryRes->fetch_object();
} else {
$num_row = 0;
while ($row = @$queryRes->fetch_object()) {
self::$last_result[$num_row] = $row;
$num_row++;
}
}
$queryRes->close();
} else {
self::$last_result = self::$_db->use_result();
}
}
self::$lastId = self::$_db->insert_id;
return self::$num_rows;
}
/**
* Escapar una cadena de texto con funciones de mysqli.
*
* @param string $str con la cadena a escapar
* @return string con la cadena escapada
*/
public static function escape($str)
{
if (self::connection()) {
return self::$_db->real_escape_string(trim($str));
} else {
return $str;
}
}
}