sysPass/inc/db.class.php

<?php

/**
 * sysPass
 *
 * @author nuxsmin
 * @link http://syspass.org
 * @copyright 2012-2014 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;
    private static $_db;

    /**
     * @brief 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;
    }

    /**
     * @brief Realizar la conexión con la BBDD
     * @return bool
     *
     * Esta función utiliza mysqli para conectar con la base de datos.
     * Guarda el objeto creado en la variable $_db de la clase
     */
    private static function connection()
    {
        if (self::$_db) {
            return true;
        }

        $dbhost = SP_Config::getValue("dbhost");
        $dbuser = SP_Config::getValue("dbuser");
        $dbpass = SP_Config::getValue("dbpass");
        $dbname = SP_Config::getValue("dbname");

        self::$_db = @new mysqli($dbhost, $dbuser, $dbpass, $dbname);

        if (self::$_db->connect_errno) {
            if (SP_Config::getValue("installed")) {
                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;
            }
        }
        return true;
    }

    /**
     * @brief 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 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;
    }

    /**
     * @brief 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
     * @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)
    {
        if ($query) {
            self::doQuery($query, $querySource);
        }

        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;
    }

    /**
     * @brief 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
     * @return bool|int devuleve bool si hay un error. Devuelve int con el número de registros
     */
    public static function doQuery($query, $querySource)
    {
        if (!self::connection()) {
            return false;
        }

        $isSelect = preg_match("/^.*(select|show)\s/i", $query);

        // Limpiar valores de caché
        self::$last_result = array();

        $queryRes = self::$_db->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) {
            if ($queryRes->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++;
                }
            }

            self::$num_rows = $queryRes->num_rows;

            $queryRes->close();
        }

        self::$lastId = self::$_db->insert_id;
        $numRows = self::$_db->affected_rows;

        return $numRows;
    }

    /**
     * @brief Escapar una cadena de texto
     * @param string $str con la cadena a escapar
     * @return string con la cadena escapada
     *
     * Esta función utiliza mysqli para escapar cadenas de texto.
     */
    public static function escape($str)
    {
        if (self::connection()) {
            return self::$_db->real_escape_string(trim($str));
        } else {
            return $str;
        }
    }
}