2013-09-10 15:49:43 +02:00
|
|
|
/*
|
|
|
|
Minetest
|
|
|
|
Copyright (C) 2013 celeron55, Perttu Ahola <celeron55@gmail.com>
|
|
|
|
|
|
|
|
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; either version 2.1 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
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.,
|
|
|
|
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
|
|
*/
|
|
|
|
|
2012-10-22 23:18:44 +02:00
|
|
|
/*
|
2014-11-16 21:31:57 +01:00
|
|
|
SQLite format specification:
|
|
|
|
blocks:
|
|
|
|
(PK) INT id
|
|
|
|
BLOB data
|
2012-10-22 23:18:44 +02:00
|
|
|
*/
|
|
|
|
|
2013-09-10 15:49:43 +02:00
|
|
|
|
|
|
|
#include "database-sqlite3.h"
|
|
|
|
|
2012-10-22 23:18:44 +02:00
|
|
|
#include "log.h"
|
2014-09-12 00:22:05 +02:00
|
|
|
#include "filesys.h"
|
2014-11-16 21:31:57 +01:00
|
|
|
#include "exceptions.h"
|
|
|
|
#include "settings.h"
|
2015-12-23 11:38:50 +01:00
|
|
|
#include "porting.h"
|
2014-11-16 21:31:57 +01:00
|
|
|
#include "util/string.h"
|
2017-04-23 14:35:08 +02:00
|
|
|
#include "remoteplayer.h"
|
2020-04-11 09:59:09 +02:00
|
|
|
#include "server/player_sao.h"
|
2014-11-16 21:31:57 +01:00
|
|
|
|
|
|
|
#include <cassert>
|
|
|
|
|
2015-12-23 11:38:50 +01:00
|
|
|
// When to print messages when the database is being held locked by another process
|
|
|
|
// Note: I've seen occasional delays of over 250ms while running minetestmapper.
|
|
|
|
#define BUSY_INFO_TRESHOLD 100 // Print first informational message after 100ms.
|
|
|
|
#define BUSY_WARNING_TRESHOLD 250 // Print warning message after 250ms. Lag is increased.
|
|
|
|
#define BUSY_ERROR_TRESHOLD 1000 // Print error message after 1000ms. Significant lag.
|
|
|
|
#define BUSY_FATAL_TRESHOLD 3000 // Allow SQLITE_BUSY to be returned, which will cause a minetest crash.
|
|
|
|
#define BUSY_ERROR_INTERVAL 10000 // Safety net: report again every 10 seconds
|
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
|
2015-12-28 19:34:16 +01:00
|
|
|
#define SQLRES(s, r, m) \
|
2014-11-16 21:31:57 +01:00
|
|
|
if ((s) != (r)) { \
|
2016-05-17 09:59:02 +02:00
|
|
|
throw DatabaseException(std::string(m) + ": " +\
|
2014-11-16 21:31:57 +01:00
|
|
|
sqlite3_errmsg(m_database)); \
|
|
|
|
}
|
2015-12-28 19:34:16 +01:00
|
|
|
#define SQLOK(s, m) SQLRES(s, SQLITE_OK, m)
|
2014-11-16 21:31:57 +01:00
|
|
|
|
|
|
|
#define PREPARE_STATEMENT(name, query) \
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLOK(sqlite3_prepare_v2(m_database, query, -1, &m_stmt_##name, NULL),\
|
|
|
|
"Failed to prepare query '" query "'")
|
2014-11-16 21:31:57 +01:00
|
|
|
|
2016-07-08 06:24:22 +02:00
|
|
|
#define SQLOK_ERRSTREAM(s, m) \
|
|
|
|
if ((s) != SQLITE_OK) { \
|
|
|
|
errorstream << (m) << ": " \
|
|
|
|
<< sqlite3_errmsg(m_database) << std::endl; \
|
|
|
|
}
|
|
|
|
|
|
|
|
#define FINALIZE_STATEMENT(statement) SQLOK_ERRSTREAM(sqlite3_finalize(statement), \
|
|
|
|
"Failed to finalize " #statement)
|
2012-10-22 23:18:44 +02:00
|
|
|
|
2015-12-23 11:38:50 +01:00
|
|
|
int Database_SQLite3::busyHandler(void *data, int count)
|
|
|
|
{
|
|
|
|
s64 &first_time = reinterpret_cast<s64 *>(data)[0];
|
|
|
|
s64 &prev_time = reinterpret_cast<s64 *>(data)[1];
|
2016-03-06 20:31:16 +01:00
|
|
|
s64 cur_time = porting::getTimeMs();
|
2015-12-23 11:38:50 +01:00
|
|
|
|
|
|
|
if (count == 0) {
|
|
|
|
first_time = cur_time;
|
|
|
|
prev_time = first_time;
|
|
|
|
} else {
|
|
|
|
while (cur_time < prev_time)
|
|
|
|
cur_time += s64(1)<<32;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cur_time - first_time < BUSY_INFO_TRESHOLD) {
|
|
|
|
; // do nothing
|
|
|
|
} else if (cur_time - first_time >= BUSY_INFO_TRESHOLD &&
|
|
|
|
prev_time - first_time < BUSY_INFO_TRESHOLD) {
|
|
|
|
infostream << "SQLite3 database has been locked for "
|
|
|
|
<< cur_time - first_time << " ms." << std::endl;
|
|
|
|
} else if (cur_time - first_time >= BUSY_WARNING_TRESHOLD &&
|
|
|
|
prev_time - first_time < BUSY_WARNING_TRESHOLD) {
|
2015-12-28 19:34:16 +01:00
|
|
|
warningstream << "SQLite3 database has been locked for "
|
2015-12-23 11:38:50 +01:00
|
|
|
<< cur_time - first_time << " ms." << std::endl;
|
|
|
|
} else if (cur_time - first_time >= BUSY_ERROR_TRESHOLD &&
|
|
|
|
prev_time - first_time < BUSY_ERROR_TRESHOLD) {
|
|
|
|
errorstream << "SQLite3 database has been locked for "
|
|
|
|
<< cur_time - first_time << " ms; this causes lag." << std::endl;
|
|
|
|
} else if (cur_time - first_time >= BUSY_FATAL_TRESHOLD &&
|
|
|
|
prev_time - first_time < BUSY_FATAL_TRESHOLD) {
|
2015-12-28 19:34:16 +01:00
|
|
|
errorstream << "SQLite3 database has been locked for "
|
2015-12-23 11:38:50 +01:00
|
|
|
<< cur_time - first_time << " ms - giving up!" << std::endl;
|
|
|
|
} else if ((cur_time - first_time) / BUSY_ERROR_INTERVAL !=
|
|
|
|
(prev_time - first_time) / BUSY_ERROR_INTERVAL) {
|
|
|
|
// Safety net: keep reporting every BUSY_ERROR_INTERVAL
|
|
|
|
errorstream << "SQLite3 database has been locked for "
|
|
|
|
<< (cur_time - first_time) / 1000 << " seconds!" << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
prev_time = cur_time;
|
|
|
|
|
|
|
|
// Make sqlite transaction fail if delay exceeds BUSY_FATAL_TRESHOLD
|
|
|
|
return cur_time - first_time < BUSY_FATAL_TRESHOLD;
|
|
|
|
}
|
|
|
|
|
2012-10-22 23:18:44 +02:00
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
Database_SQLite3::Database_SQLite3(const std::string &savedir, const std::string &dbname) :
|
2014-11-16 21:31:57 +01:00
|
|
|
m_savedir(savedir),
|
2017-06-16 11:25:52 +02:00
|
|
|
m_dbname(dbname)
|
2012-10-22 23:18:44 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
void Database_SQLite3::beginSave()
|
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLRES(sqlite3_step(m_stmt_begin), SQLITE_DONE,
|
|
|
|
"Failed to start SQLite3 transaction");
|
2014-11-16 21:31:57 +01:00
|
|
|
sqlite3_reset(m_stmt_begin);
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
void Database_SQLite3::endSave()
|
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLRES(sqlite3_step(m_stmt_end), SQLITE_DONE,
|
|
|
|
"Failed to commit SQLite3 transaction");
|
2014-11-16 21:31:57 +01:00
|
|
|
sqlite3_reset(m_stmt_end);
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
void Database_SQLite3::openDatabase()
|
2012-10-22 23:18:44 +02:00
|
|
|
{
|
2014-11-16 21:31:57 +01:00
|
|
|
if (m_database) return;
|
2012-10-22 23:18:44 +02:00
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
std::string dbp = m_savedir + DIR_DELIM + m_dbname + ".sqlite";
|
2014-06-23 21:27:54 +02:00
|
|
|
|
|
|
|
// Open the database connection
|
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
if (!fs::CreateAllDirs(m_savedir)) {
|
|
|
|
infostream << "Database_SQLite3: Failed to create directory \""
|
|
|
|
<< m_savedir << "\"" << std::endl;
|
|
|
|
throw FileNotGoodException("Failed to create database "
|
|
|
|
"save directory");
|
|
|
|
}
|
2014-06-23 21:27:54 +02:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
bool needs_create = !fs::PathExists(dbp);
|
2014-06-23 21:27:54 +02:00
|
|
|
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLOK(sqlite3_open_v2(dbp.c_str(), &m_database,
|
|
|
|
SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL),
|
|
|
|
std::string("Failed to open SQLite3 database file ") + dbp);
|
2014-06-23 21:27:54 +02:00
|
|
|
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLOK(sqlite3_busy_handler(m_database, Database_SQLite3::busyHandler,
|
|
|
|
m_busy_handler_data), "Failed to set SQLite3 busy handler");
|
2015-12-23 11:38:50 +01:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
if (needs_create) {
|
2014-06-23 21:27:54 +02:00
|
|
|
createDatabase();
|
2014-11-16 21:31:57 +01:00
|
|
|
}
|
2014-06-23 21:27:54 +02:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
std::string query_str = std::string("PRAGMA synchronous = ")
|
2014-06-23 21:27:54 +02:00
|
|
|
+ itos(g_settings->getU16("sqlite_synchronous"));
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLOK(sqlite3_exec(m_database, query_str.c_str(), NULL, NULL, NULL),
|
|
|
|
"Failed to modify sqlite3 synchronous mode");
|
2017-04-23 14:35:08 +02:00
|
|
|
SQLOK(sqlite3_exec(m_database, "PRAGMA foreign_keys = ON", NULL, NULL, NULL),
|
|
|
|
"Failed to enable sqlite3 foreign key support");
|
2014-11-16 21:31:57 +01:00
|
|
|
}
|
2014-06-23 21:27:54 +02:00
|
|
|
|
2020-07-07 18:50:52 +02:00
|
|
|
void Database_SQLite3::verifyDatabase()
|
2014-11-16 21:31:57 +01:00
|
|
|
{
|
|
|
|
if (m_initialized) return;
|
|
|
|
|
|
|
|
openDatabase();
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
PREPARE_STATEMENT(begin, "BEGIN;");
|
|
|
|
PREPARE_STATEMENT(end, "COMMIT;");
|
|
|
|
|
|
|
|
initStatements();
|
|
|
|
|
|
|
|
m_initialized = true;
|
|
|
|
}
|
|
|
|
|
|
|
|
Database_SQLite3::~Database_SQLite3()
|
|
|
|
{
|
|
|
|
FINALIZE_STATEMENT(m_stmt_begin)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_end)
|
|
|
|
|
|
|
|
SQLOK_ERRSTREAM(sqlite3_close(m_database), "Failed to close database");
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Map database
|
|
|
|
*/
|
|
|
|
|
|
|
|
MapDatabaseSQLite3::MapDatabaseSQLite3(const std::string &savedir):
|
|
|
|
Database_SQLite3(savedir, "map"),
|
2017-06-16 11:25:52 +02:00
|
|
|
MapDatabase()
|
2017-04-23 14:35:08 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
MapDatabaseSQLite3::~MapDatabaseSQLite3()
|
|
|
|
{
|
|
|
|
FINALIZE_STATEMENT(m_stmt_read)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_write)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_list)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_delete)
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void MapDatabaseSQLite3::createDatabase()
|
|
|
|
{
|
|
|
|
assert(m_database); // Pre-condition
|
|
|
|
|
|
|
|
SQLOK(sqlite3_exec(m_database,
|
|
|
|
"CREATE TABLE IF NOT EXISTS `blocks` (\n"
|
|
|
|
" `pos` INT PRIMARY KEY,\n"
|
|
|
|
" `data` BLOB\n"
|
|
|
|
");\n",
|
|
|
|
NULL, NULL, NULL),
|
|
|
|
"Failed to create database table");
|
|
|
|
}
|
|
|
|
|
|
|
|
void MapDatabaseSQLite3::initStatements()
|
|
|
|
{
|
2014-11-16 21:31:57 +01:00
|
|
|
PREPARE_STATEMENT(read, "SELECT `data` FROM `blocks` WHERE `pos` = ? LIMIT 1");
|
2014-07-07 22:57:11 +02:00
|
|
|
#ifdef __ANDROID__
|
2014-11-16 21:31:57 +01:00
|
|
|
PREPARE_STATEMENT(write, "INSERT INTO `blocks` (`pos`, `data`) VALUES (?, ?)");
|
2014-07-07 22:57:11 +02:00
|
|
|
#else
|
2014-11-16 21:31:57 +01:00
|
|
|
PREPARE_STATEMENT(write, "REPLACE INTO `blocks` (`pos`, `data`) VALUES (?, ?)");
|
2014-07-07 22:57:11 +02:00
|
|
|
#endif
|
2014-11-16 21:31:57 +01:00
|
|
|
PREPARE_STATEMENT(delete, "DELETE FROM `blocks` WHERE `pos` = ?");
|
|
|
|
PREPARE_STATEMENT(list, "SELECT `pos` FROM `blocks`");
|
2014-06-23 21:27:54 +02:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
verbosestream << "ServerMap: SQLite3 database opened." << std::endl;
|
|
|
|
}
|
2014-06-23 21:27:54 +02:00
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
inline void MapDatabaseSQLite3::bindPos(sqlite3_stmt *stmt, const v3s16 &pos, int index)
|
2014-11-16 21:31:57 +01:00
|
|
|
{
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLOK(sqlite3_bind_int64(stmt, index, getBlockAsInteger(pos)),
|
|
|
|
"Internal error: failed to bind query at " __FILE__ ":" TOSTRING(__LINE__));
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
bool MapDatabaseSQLite3::deleteBlock(const v3s16 &pos)
|
2015-01-15 22:20:05 +01:00
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2015-01-15 22:20:05 +01:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
bindPos(m_stmt_delete, pos);
|
|
|
|
|
|
|
|
bool good = sqlite3_step(m_stmt_delete) == SQLITE_DONE;
|
|
|
|
sqlite3_reset(m_stmt_delete);
|
2015-01-15 22:20:05 +01:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
if (!good) {
|
2015-10-14 07:26:03 +02:00
|
|
|
warningstream << "deleteBlock: Block failed to delete "
|
2014-11-16 21:31:57 +01:00
|
|
|
<< PP(pos) << ": " << sqlite3_errmsg(m_database) << std::endl;
|
2015-01-15 22:20:05 +01:00
|
|
|
}
|
2014-11-16 21:31:57 +01:00
|
|
|
return good;
|
2015-01-15 22:20:05 +01:00
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
bool MapDatabaseSQLite3::saveBlock(const v3s16 &pos, const std::string &data)
|
2012-10-22 23:18:44 +02:00
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2014-06-22 02:31:24 +02:00
|
|
|
|
2014-07-07 22:57:11 +02:00
|
|
|
#ifdef __ANDROID__
|
|
|
|
/**
|
2014-11-16 21:31:57 +01:00
|
|
|
* Note: For some unknown reason SQLite3 fails to REPLACE blocks on Android,
|
|
|
|
* deleting them and then inserting works.
|
2014-07-07 22:57:11 +02:00
|
|
|
*/
|
2014-11-16 21:31:57 +01:00
|
|
|
bindPos(m_stmt_read, pos);
|
2014-07-07 22:57:11 +02:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
if (sqlite3_step(m_stmt_read) == SQLITE_ROW) {
|
|
|
|
deleteBlock(pos);
|
2014-07-07 22:57:11 +02:00
|
|
|
}
|
2014-11-16 21:31:57 +01:00
|
|
|
sqlite3_reset(m_stmt_read);
|
2014-07-07 22:57:11 +02:00
|
|
|
#endif
|
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
bindPos(m_stmt_write, pos);
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLOK(sqlite3_bind_blob(m_stmt_write, 2, data.data(), data.size(), NULL),
|
|
|
|
"Internal error: failed to bind query at " __FILE__ ":" TOSTRING(__LINE__));
|
2014-06-22 02:31:24 +02:00
|
|
|
|
2015-12-28 19:34:16 +01:00
|
|
|
SQLRES(sqlite3_step(m_stmt_write), SQLITE_DONE, "Failed to save block")
|
2014-11-16 21:31:57 +01:00
|
|
|
sqlite3_reset(m_stmt_write);
|
2014-07-07 22:57:11 +02:00
|
|
|
|
2014-07-07 07:20:25 +02:00
|
|
|
return true;
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
void MapDatabaseSQLite3::loadBlock(const v3s16 &pos, std::string *block)
|
2012-10-22 23:18:44 +02:00
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2013-11-17 16:57:45 +01:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
bindPos(m_stmt_read, pos);
|
2013-11-17 16:57:45 +01:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
if (sqlite3_step(m_stmt_read) != SQLITE_ROW) {
|
|
|
|
sqlite3_reset(m_stmt_read);
|
2016-05-14 12:23:15 +02:00
|
|
|
return;
|
2014-11-16 21:31:57 +01:00
|
|
|
}
|
2016-05-14 12:23:15 +02:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
const char *data = (const char *) sqlite3_column_blob(m_stmt_read, 0);
|
|
|
|
size_t len = sqlite3_column_bytes(m_stmt_read, 0);
|
2013-11-17 16:57:45 +01:00
|
|
|
|
2021-09-11 21:06:57 +02:00
|
|
|
if (data)
|
|
|
|
block->assign(data, len);
|
|
|
|
else
|
|
|
|
block->clear();
|
2013-11-17 16:57:45 +01:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
sqlite3_step(m_stmt_read);
|
|
|
|
// We should never get more than 1 row, so ok to reset
|
|
|
|
sqlite3_reset(m_stmt_read);
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
void MapDatabaseSQLite3::listAllLoadableBlocks(std::vector<v3s16> &dst)
|
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2017-04-23 14:35:08 +02:00
|
|
|
|
|
|
|
while (sqlite3_step(m_stmt_list) == SQLITE_ROW)
|
|
|
|
dst.push_back(getIntegerAsBlock(sqlite3_column_int64(m_stmt_list, 0)));
|
|
|
|
|
|
|
|
sqlite3_reset(m_stmt_list);
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Player Database
|
|
|
|
*/
|
|
|
|
|
|
|
|
PlayerDatabaseSQLite3::PlayerDatabaseSQLite3(const std::string &savedir):
|
|
|
|
Database_SQLite3(savedir, "players"),
|
2017-06-16 11:25:52 +02:00
|
|
|
PlayerDatabase()
|
2017-04-23 14:35:08 +02:00
|
|
|
{
|
|
|
|
}
|
2017-06-16 11:25:52 +02:00
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
PlayerDatabaseSQLite3::~PlayerDatabaseSQLite3()
|
|
|
|
{
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_load)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_add)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_update)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_remove)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_list)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_add_inventory)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_add_inventory_items)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_remove_inventory)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_remove_inventory_items)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_load_inventory)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_load_inventory_items)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_metadata_load)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_metadata_add)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_player_metadata_remove)
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
void PlayerDatabaseSQLite3::createDatabase()
|
2012-10-22 23:18:44 +02:00
|
|
|
{
|
2015-03-06 11:21:51 +01:00
|
|
|
assert(m_database); // Pre-condition
|
2017-04-23 14:35:08 +02:00
|
|
|
|
2014-11-16 21:31:57 +01:00
|
|
|
SQLOK(sqlite3_exec(m_database,
|
2017-04-23 14:35:08 +02:00
|
|
|
"CREATE TABLE IF NOT EXISTS `player` ("
|
|
|
|
"`name` VARCHAR(50) NOT NULL,"
|
|
|
|
"`pitch` NUMERIC(11, 4) NOT NULL,"
|
|
|
|
"`yaw` NUMERIC(11, 4) NOT NULL,"
|
|
|
|
"`posX` NUMERIC(11, 4) NOT NULL,"
|
|
|
|
"`posY` NUMERIC(11, 4) NOT NULL,"
|
|
|
|
"`posZ` NUMERIC(11, 4) NOT NULL,"
|
|
|
|
"`hp` INT NOT NULL,"
|
|
|
|
"`breath` INT NOT NULL,"
|
|
|
|
"`creation_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,"
|
|
|
|
"`modification_date` DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,"
|
|
|
|
"PRIMARY KEY (`name`));",
|
2015-12-28 19:34:16 +01:00
|
|
|
NULL, NULL, NULL),
|
2017-04-23 14:35:08 +02:00
|
|
|
"Failed to create player table");
|
|
|
|
|
|
|
|
SQLOK(sqlite3_exec(m_database,
|
|
|
|
"CREATE TABLE IF NOT EXISTS `player_metadata` ("
|
|
|
|
" `player` VARCHAR(50) NOT NULL,"
|
|
|
|
" `metadata` VARCHAR(256) NOT NULL,"
|
|
|
|
" `value` TEXT,"
|
|
|
|
" PRIMARY KEY(`player`, `metadata`),"
|
|
|
|
" FOREIGN KEY (`player`) REFERENCES player (`name`) ON DELETE CASCADE );",
|
|
|
|
NULL, NULL, NULL),
|
|
|
|
"Failed to create player metadata table");
|
|
|
|
|
|
|
|
SQLOK(sqlite3_exec(m_database,
|
|
|
|
"CREATE TABLE IF NOT EXISTS `player_inventories` ("
|
|
|
|
" `player` VARCHAR(50) NOT NULL,"
|
|
|
|
" `inv_id` INT NOT NULL,"
|
|
|
|
" `inv_width` INT NOT NULL,"
|
|
|
|
" `inv_name` TEXT NOT NULL DEFAULT '',"
|
|
|
|
" `inv_size` INT NOT NULL,"
|
|
|
|
" PRIMARY KEY(player, inv_id),"
|
|
|
|
" FOREIGN KEY (`player`) REFERENCES player (`name`) ON DELETE CASCADE );",
|
|
|
|
NULL, NULL, NULL),
|
|
|
|
"Failed to create player inventory table");
|
|
|
|
|
|
|
|
SQLOK(sqlite3_exec(m_database,
|
|
|
|
"CREATE TABLE `player_inventory_items` ("
|
|
|
|
" `player` VARCHAR(50) NOT NULL,"
|
|
|
|
" `inv_id` INT NOT NULL,"
|
|
|
|
" `slot_id` INT NOT NULL,"
|
|
|
|
" `item` TEXT NOT NULL DEFAULT '',"
|
|
|
|
" PRIMARY KEY(player, inv_id, slot_id),"
|
|
|
|
" FOREIGN KEY (`player`) REFERENCES player (`name`) ON DELETE CASCADE );",
|
|
|
|
NULL, NULL, NULL),
|
|
|
|
"Failed to create player inventory items table");
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
void PlayerDatabaseSQLite3::initStatements()
|
|
|
|
{
|
|
|
|
PREPARE_STATEMENT(player_load, "SELECT `pitch`, `yaw`, `posX`, `posY`, `posZ`, `hp`, "
|
|
|
|
"`breath`"
|
|
|
|
"FROM `player` WHERE `name` = ?")
|
|
|
|
PREPARE_STATEMENT(player_add, "INSERT INTO `player` (`name`, `pitch`, `yaw`, `posX`, "
|
|
|
|
"`posY`, `posZ`, `hp`, `breath`) VALUES (?, ?, ?, ?, ?, ?, ?, ?)")
|
|
|
|
PREPARE_STATEMENT(player_update, "UPDATE `player` SET `pitch` = ?, `yaw` = ?, "
|
|
|
|
"`posX` = ?, `posY` = ?, `posZ` = ?, `hp` = ?, `breath` = ?, "
|
|
|
|
"`modification_date` = CURRENT_TIMESTAMP WHERE `name` = ?")
|
|
|
|
PREPARE_STATEMENT(player_remove, "DELETE FROM `player` WHERE `name` = ?")
|
|
|
|
PREPARE_STATEMENT(player_list, "SELECT `name` FROM `player`")
|
|
|
|
|
|
|
|
PREPARE_STATEMENT(player_add_inventory, "INSERT INTO `player_inventories` "
|
|
|
|
"(`player`, `inv_id`, `inv_width`, `inv_name`, `inv_size`) VALUES (?, ?, ?, ?, ?)")
|
|
|
|
PREPARE_STATEMENT(player_add_inventory_items, "INSERT INTO `player_inventory_items` "
|
|
|
|
"(`player`, `inv_id`, `slot_id`, `item`) VALUES (?, ?, ?, ?)")
|
|
|
|
PREPARE_STATEMENT(player_remove_inventory, "DELETE FROM `player_inventories` "
|
|
|
|
"WHERE `player` = ?")
|
|
|
|
PREPARE_STATEMENT(player_remove_inventory_items, "DELETE FROM `player_inventory_items` "
|
|
|
|
"WHERE `player` = ?")
|
|
|
|
PREPARE_STATEMENT(player_load_inventory, "SELECT `inv_id`, `inv_width`, `inv_name`, "
|
|
|
|
"`inv_size` FROM `player_inventories` WHERE `player` = ? ORDER BY inv_id")
|
|
|
|
PREPARE_STATEMENT(player_load_inventory_items, "SELECT `slot_id`, `item` "
|
|
|
|
"FROM `player_inventory_items` WHERE `player` = ? AND `inv_id` = ?")
|
|
|
|
|
|
|
|
PREPARE_STATEMENT(player_metadata_load, "SELECT `metadata`, `value` FROM "
|
|
|
|
"`player_metadata` WHERE `player` = ?")
|
|
|
|
PREPARE_STATEMENT(player_metadata_add, "INSERT INTO `player_metadata` "
|
|
|
|
"(`player`, `metadata`, `value`) VALUES (?, ?, ?)")
|
|
|
|
PREPARE_STATEMENT(player_metadata_remove, "DELETE FROM `player_metadata` "
|
|
|
|
"WHERE `player` = ?")
|
|
|
|
verbosestream << "ServerEnvironment: SQLite3 database opened (players)." << std::endl;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool PlayerDatabaseSQLite3::playerDataExists(const std::string &name)
|
2012-10-22 23:18:44 +02:00
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2017-04-23 14:35:08 +02:00
|
|
|
str_to_sqlite(m_stmt_player_load, 1, name);
|
|
|
|
bool res = (sqlite3_step(m_stmt_player_load) == SQLITE_ROW);
|
|
|
|
sqlite3_reset(m_stmt_player_load);
|
|
|
|
return res;
|
|
|
|
}
|
2014-06-22 02:31:24 +02:00
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
void PlayerDatabaseSQLite3::savePlayer(RemotePlayer *player)
|
|
|
|
{
|
|
|
|
PlayerSAO* sao = player->getPlayerSAO();
|
|
|
|
sanity_check(sao);
|
|
|
|
|
|
|
|
const v3f &pos = sao->getBasePosition();
|
|
|
|
// Begin save in brace is mandatory
|
|
|
|
if (!playerDataExists(player->getName())) {
|
|
|
|
beginSave();
|
|
|
|
str_to_sqlite(m_stmt_player_add, 1, player->getName());
|
2018-11-28 09:38:50 +01:00
|
|
|
double_to_sqlite(m_stmt_player_add, 2, sao->getLookPitch());
|
|
|
|
double_to_sqlite(m_stmt_player_add, 3, sao->getRotation().Y);
|
2017-04-23 14:35:08 +02:00
|
|
|
double_to_sqlite(m_stmt_player_add, 4, pos.X);
|
|
|
|
double_to_sqlite(m_stmt_player_add, 5, pos.Y);
|
|
|
|
double_to_sqlite(m_stmt_player_add, 6, pos.Z);
|
|
|
|
int64_to_sqlite(m_stmt_player_add, 7, sao->getHP());
|
|
|
|
int64_to_sqlite(m_stmt_player_add, 8, sao->getBreath());
|
|
|
|
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_add), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_add);
|
|
|
|
} else {
|
|
|
|
beginSave();
|
2018-11-28 09:38:50 +01:00
|
|
|
double_to_sqlite(m_stmt_player_update, 1, sao->getLookPitch());
|
|
|
|
double_to_sqlite(m_stmt_player_update, 2, sao->getRotation().Y);
|
2017-04-23 14:35:08 +02:00
|
|
|
double_to_sqlite(m_stmt_player_update, 3, pos.X);
|
|
|
|
double_to_sqlite(m_stmt_player_update, 4, pos.Y);
|
|
|
|
double_to_sqlite(m_stmt_player_update, 5, pos.Z);
|
|
|
|
int64_to_sqlite(m_stmt_player_update, 6, sao->getHP());
|
|
|
|
int64_to_sqlite(m_stmt_player_update, 7, sao->getBreath());
|
|
|
|
str_to_sqlite(m_stmt_player_update, 8, player->getName());
|
|
|
|
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_update), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_update);
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
2017-04-23 14:35:08 +02:00
|
|
|
|
|
|
|
// Write player inventories
|
|
|
|
str_to_sqlite(m_stmt_player_remove_inventory, 1, player->getName());
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_remove_inventory), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_remove_inventory);
|
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_player_remove_inventory_items, 1, player->getName());
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_remove_inventory_items), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_remove_inventory_items);
|
|
|
|
|
|
|
|
std::vector<const InventoryList*> inventory_lists = sao->getInventory()->getLists();
|
2021-09-11 21:06:57 +02:00
|
|
|
std::ostringstream oss;
|
2017-04-23 14:35:08 +02:00
|
|
|
for (u16 i = 0; i < inventory_lists.size(); i++) {
|
|
|
|
const InventoryList* list = inventory_lists[i];
|
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_player_add_inventory, 1, player->getName());
|
|
|
|
int_to_sqlite(m_stmt_player_add_inventory, 2, i);
|
|
|
|
int_to_sqlite(m_stmt_player_add_inventory, 3, list->getWidth());
|
|
|
|
str_to_sqlite(m_stmt_player_add_inventory, 4, list->getName());
|
|
|
|
int_to_sqlite(m_stmt_player_add_inventory, 5, list->getSize());
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_add_inventory), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_add_inventory);
|
|
|
|
|
|
|
|
for (u32 j = 0; j < list->getSize(); j++) {
|
2021-09-11 21:06:57 +02:00
|
|
|
oss.str("");
|
|
|
|
oss.clear();
|
|
|
|
list->getItem(j).serialize(oss);
|
|
|
|
std::string itemStr = oss.str();
|
2017-04-23 14:35:08 +02:00
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_player_add_inventory_items, 1, player->getName());
|
|
|
|
int_to_sqlite(m_stmt_player_add_inventory_items, 2, i);
|
|
|
|
int_to_sqlite(m_stmt_player_add_inventory_items, 3, j);
|
|
|
|
str_to_sqlite(m_stmt_player_add_inventory_items, 4, itemStr);
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_add_inventory_items), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_add_inventory_items);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_player_metadata_remove, 1, player->getName());
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_metadata_remove), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_metadata_remove);
|
|
|
|
|
2018-04-06 10:52:29 +02:00
|
|
|
const StringMap &attrs = sao->getMeta().getStrings();
|
2017-08-17 23:02:50 +02:00
|
|
|
for (const auto &attr : attrs) {
|
2017-04-23 14:35:08 +02:00
|
|
|
str_to_sqlite(m_stmt_player_metadata_add, 1, player->getName());
|
2017-08-17 23:02:50 +02:00
|
|
|
str_to_sqlite(m_stmt_player_metadata_add, 2, attr.first);
|
|
|
|
str_to_sqlite(m_stmt_player_metadata_add, 3, attr.second);
|
2017-04-23 14:35:08 +02:00
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_metadata_add), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_metadata_add);
|
|
|
|
}
|
|
|
|
|
|
|
|
endSave();
|
2019-01-04 10:20:04 +01:00
|
|
|
|
2019-01-04 11:33:04 +01:00
|
|
|
player->onSuccessfulSave();
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
bool PlayerDatabaseSQLite3::loadPlayer(RemotePlayer *player, PlayerSAO *sao)
|
2012-10-22 23:18:44 +02:00
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2014-11-16 21:31:57 +01:00
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
str_to_sqlite(m_stmt_player_load, 1, player->getName());
|
|
|
|
if (sqlite3_step(m_stmt_player_load) != SQLITE_ROW) {
|
|
|
|
sqlite3_reset(m_stmt_player_load);
|
|
|
|
return false;
|
|
|
|
}
|
2018-11-28 09:38:50 +01:00
|
|
|
sao->setLookPitch(sqlite_to_float(m_stmt_player_load, 0));
|
|
|
|
sao->setPlayerYaw(sqlite_to_float(m_stmt_player_load, 1));
|
2017-04-23 14:35:08 +02:00
|
|
|
sao->setBasePosition(sqlite_to_v3f(m_stmt_player_load, 2));
|
2019-02-11 00:03:26 +01:00
|
|
|
sao->setHPRaw((u16) MYMIN(sqlite_to_int(m_stmt_player_load, 5), U16_MAX));
|
2017-04-23 14:35:08 +02:00
|
|
|
sao->setBreath((u16) MYMIN(sqlite_to_int(m_stmt_player_load, 6), U16_MAX), false);
|
|
|
|
sqlite3_reset(m_stmt_player_load);
|
|
|
|
|
|
|
|
// Load inventory
|
|
|
|
str_to_sqlite(m_stmt_player_load_inventory, 1, player->getName());
|
|
|
|
while (sqlite3_step(m_stmt_player_load_inventory) == SQLITE_ROW) {
|
|
|
|
InventoryList *invList = player->inventory.addList(
|
|
|
|
sqlite_to_string(m_stmt_player_load_inventory, 2),
|
|
|
|
sqlite_to_uint(m_stmt_player_load_inventory, 3));
|
|
|
|
invList->setWidth(sqlite_to_uint(m_stmt_player_load_inventory, 1));
|
|
|
|
|
|
|
|
u32 invId = sqlite_to_uint(m_stmt_player_load_inventory, 0);
|
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_player_load_inventory_items, 1, player->getName());
|
|
|
|
int_to_sqlite(m_stmt_player_load_inventory_items, 2, invId);
|
|
|
|
while (sqlite3_step(m_stmt_player_load_inventory_items) == SQLITE_ROW) {
|
|
|
|
const std::string itemStr = sqlite_to_string(m_stmt_player_load_inventory_items, 1);
|
|
|
|
if (itemStr.length() > 0) {
|
|
|
|
ItemStack stack;
|
|
|
|
stack.deSerialize(itemStr);
|
2017-06-25 11:39:39 +02:00
|
|
|
invList->changeItem(sqlite_to_uint(m_stmt_player_load_inventory_items, 0), stack);
|
2017-04-23 14:35:08 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
sqlite3_reset(m_stmt_player_load_inventory_items);
|
|
|
|
}
|
|
|
|
|
|
|
|
sqlite3_reset(m_stmt_player_load_inventory);
|
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_player_metadata_load, 1, sao->getPlayer()->getName());
|
|
|
|
while (sqlite3_step(m_stmt_player_metadata_load) == SQLITE_ROW) {
|
|
|
|
std::string attr = sqlite_to_string(m_stmt_player_metadata_load, 0);
|
|
|
|
std::string value = sqlite_to_string(m_stmt_player_metadata_load, 1);
|
|
|
|
|
2018-04-06 10:52:29 +02:00
|
|
|
sao->getMeta().setString(attr, value);
|
2017-04-23 14:35:08 +02:00
|
|
|
}
|
2018-04-06 10:52:29 +02:00
|
|
|
sao->getMeta().setModified(false);
|
2017-04-23 14:35:08 +02:00
|
|
|
sqlite3_reset(m_stmt_player_metadata_load);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool PlayerDatabaseSQLite3::removePlayer(const std::string &name)
|
|
|
|
{
|
|
|
|
if (!playerDataExists(name))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_player_remove, 1, name);
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_player_remove), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_player_remove);
|
|
|
|
return true;
|
2012-10-22 23:18:44 +02:00
|
|
|
}
|
2014-11-16 21:31:57 +01:00
|
|
|
|
2017-04-23 14:35:08 +02:00
|
|
|
void PlayerDatabaseSQLite3::listPlayers(std::vector<std::string> &res)
|
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2017-04-23 14:35:08 +02:00
|
|
|
|
|
|
|
while (sqlite3_step(m_stmt_player_list) == SQLITE_ROW)
|
|
|
|
res.push_back(sqlite_to_string(m_stmt_player_list, 0));
|
|
|
|
|
|
|
|
sqlite3_reset(m_stmt_player_list);
|
|
|
|
}
|
2018-08-05 13:13:38 +02:00
|
|
|
|
|
|
|
/*
|
|
|
|
* Auth database
|
|
|
|
*/
|
|
|
|
|
|
|
|
AuthDatabaseSQLite3::AuthDatabaseSQLite3(const std::string &savedir) :
|
|
|
|
Database_SQLite3(savedir, "auth"), AuthDatabase()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
AuthDatabaseSQLite3::~AuthDatabaseSQLite3()
|
|
|
|
{
|
|
|
|
FINALIZE_STATEMENT(m_stmt_read)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_write)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_create)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_delete)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_list_names)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_read_privs)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_write_privs)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_delete_privs)
|
|
|
|
FINALIZE_STATEMENT(m_stmt_last_insert_rowid)
|
|
|
|
}
|
|
|
|
|
|
|
|
void AuthDatabaseSQLite3::createDatabase()
|
|
|
|
{
|
|
|
|
assert(m_database); // Pre-condition
|
|
|
|
|
|
|
|
SQLOK(sqlite3_exec(m_database,
|
|
|
|
"CREATE TABLE IF NOT EXISTS `auth` ("
|
|
|
|
"`id` INTEGER PRIMARY KEY AUTOINCREMENT,"
|
|
|
|
"`name` VARCHAR(32) UNIQUE,"
|
|
|
|
"`password` VARCHAR(512),"
|
|
|
|
"`last_login` INTEGER"
|
|
|
|
");",
|
|
|
|
NULL, NULL, NULL),
|
|
|
|
"Failed to create auth table");
|
|
|
|
|
|
|
|
SQLOK(sqlite3_exec(m_database,
|
|
|
|
"CREATE TABLE IF NOT EXISTS `user_privileges` ("
|
|
|
|
"`id` INTEGER,"
|
|
|
|
"`privilege` VARCHAR(32),"
|
|
|
|
"PRIMARY KEY (id, privilege)"
|
|
|
|
"CONSTRAINT fk_id FOREIGN KEY (id) REFERENCES auth (id) ON DELETE CASCADE"
|
|
|
|
");",
|
|
|
|
NULL, NULL, NULL),
|
|
|
|
"Failed to create auth privileges table");
|
|
|
|
}
|
|
|
|
|
|
|
|
void AuthDatabaseSQLite3::initStatements()
|
|
|
|
{
|
|
|
|
PREPARE_STATEMENT(read, "SELECT id, name, password, last_login FROM auth WHERE name = ?");
|
|
|
|
PREPARE_STATEMENT(write, "UPDATE auth set name = ?, password = ?, last_login = ? WHERE id = ?");
|
|
|
|
PREPARE_STATEMENT(create, "INSERT INTO auth (name, password, last_login) VALUES (?, ?, ?)");
|
|
|
|
PREPARE_STATEMENT(delete, "DELETE FROM auth WHERE name = ?");
|
|
|
|
|
|
|
|
PREPARE_STATEMENT(list_names, "SELECT name FROM auth ORDER BY name DESC");
|
|
|
|
|
|
|
|
PREPARE_STATEMENT(read_privs, "SELECT privilege FROM user_privileges WHERE id = ?");
|
|
|
|
PREPARE_STATEMENT(write_privs, "INSERT OR IGNORE INTO user_privileges (id, privilege) VALUES (?, ?)");
|
|
|
|
PREPARE_STATEMENT(delete_privs, "DELETE FROM user_privileges WHERE id = ?");
|
|
|
|
|
|
|
|
PREPARE_STATEMENT(last_insert_rowid, "SELECT last_insert_rowid()");
|
|
|
|
}
|
|
|
|
|
|
|
|
bool AuthDatabaseSQLite3::getAuth(const std::string &name, AuthEntry &res)
|
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2018-08-05 13:13:38 +02:00
|
|
|
str_to_sqlite(m_stmt_read, 1, name);
|
|
|
|
if (sqlite3_step(m_stmt_read) != SQLITE_ROW) {
|
|
|
|
sqlite3_reset(m_stmt_read);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
res.id = sqlite_to_uint(m_stmt_read, 0);
|
|
|
|
res.name = sqlite_to_string(m_stmt_read, 1);
|
|
|
|
res.password = sqlite_to_string(m_stmt_read, 2);
|
|
|
|
res.last_login = sqlite_to_int64(m_stmt_read, 3);
|
|
|
|
sqlite3_reset(m_stmt_read);
|
|
|
|
|
|
|
|
int64_to_sqlite(m_stmt_read_privs, 1, res.id);
|
|
|
|
while (sqlite3_step(m_stmt_read_privs) == SQLITE_ROW) {
|
|
|
|
res.privileges.emplace_back(sqlite_to_string(m_stmt_read_privs, 0));
|
|
|
|
}
|
|
|
|
sqlite3_reset(m_stmt_read_privs);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool AuthDatabaseSQLite3::saveAuth(const AuthEntry &authEntry)
|
|
|
|
{
|
|
|
|
beginSave();
|
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_write, 1, authEntry.name);
|
|
|
|
str_to_sqlite(m_stmt_write, 2, authEntry.password);
|
|
|
|
int64_to_sqlite(m_stmt_write, 3, authEntry.last_login);
|
|
|
|
int64_to_sqlite(m_stmt_write, 4, authEntry.id);
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_write), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_write);
|
|
|
|
|
|
|
|
writePrivileges(authEntry);
|
|
|
|
|
|
|
|
endSave();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool AuthDatabaseSQLite3::createAuth(AuthEntry &authEntry)
|
|
|
|
{
|
|
|
|
beginSave();
|
|
|
|
|
|
|
|
// id autoincrements
|
|
|
|
str_to_sqlite(m_stmt_create, 1, authEntry.name);
|
|
|
|
str_to_sqlite(m_stmt_create, 2, authEntry.password);
|
|
|
|
int64_to_sqlite(m_stmt_create, 3, authEntry.last_login);
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_create), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_create);
|
|
|
|
|
|
|
|
// obtain id and write back to original authEntry
|
|
|
|
sqlite3_step(m_stmt_last_insert_rowid);
|
|
|
|
authEntry.id = sqlite_to_uint(m_stmt_last_insert_rowid, 0);
|
|
|
|
sqlite3_reset(m_stmt_last_insert_rowid);
|
|
|
|
|
|
|
|
writePrivileges(authEntry);
|
|
|
|
|
|
|
|
endSave();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool AuthDatabaseSQLite3::deleteAuth(const std::string &name)
|
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2018-08-05 13:13:38 +02:00
|
|
|
|
|
|
|
str_to_sqlite(m_stmt_delete, 1, name);
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_delete), SQLITE_DONE);
|
|
|
|
int changes = sqlite3_changes(m_database);
|
|
|
|
sqlite3_reset(m_stmt_delete);
|
|
|
|
|
|
|
|
// privileges deleted by foreign key on delete cascade
|
|
|
|
|
|
|
|
return changes > 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
void AuthDatabaseSQLite3::listNames(std::vector<std::string> &res)
|
|
|
|
{
|
2020-07-07 18:50:52 +02:00
|
|
|
verifyDatabase();
|
2018-08-05 13:13:38 +02:00
|
|
|
|
|
|
|
while (sqlite3_step(m_stmt_list_names) == SQLITE_ROW) {
|
|
|
|
res.push_back(sqlite_to_string(m_stmt_list_names, 0));
|
|
|
|
}
|
|
|
|
sqlite3_reset(m_stmt_list_names);
|
|
|
|
}
|
|
|
|
|
|
|
|
void AuthDatabaseSQLite3::reload()
|
|
|
|
{
|
|
|
|
// noop for SQLite
|
|
|
|
}
|
|
|
|
|
|
|
|
void AuthDatabaseSQLite3::writePrivileges(const AuthEntry &authEntry)
|
|
|
|
{
|
|
|
|
int64_to_sqlite(m_stmt_delete_privs, 1, authEntry.id);
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_delete_privs), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_delete_privs);
|
|
|
|
for (const std::string &privilege : authEntry.privileges) {
|
|
|
|
int64_to_sqlite(m_stmt_write_privs, 1, authEntry.id);
|
|
|
|
str_to_sqlite(m_stmt_write_privs, 2, privilege);
|
|
|
|
sqlite3_vrfy(sqlite3_step(m_stmt_write_privs), SQLITE_DONE);
|
|
|
|
sqlite3_reset(m_stmt_write_privs);
|
|
|
|
}
|
|
|
|
}
|