mirrors/mumble
2
0
Fork 0
mirror of https://github.com/mumble-voip/mumble.git synced 2025-10-26 11:19:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
615448f304
mumble/src/murmur/ServerDB.cpp
Robert Adam 615448f304 src/murmur/ServerDB.cpp: Use REPLACE instead of INSERT 
In #3803 it is reported that the server regularly crashes because the
MySQL/SQLite part of the database code for group changes doesn't handle
conflicts in primary keys (as the PostgreSQL part does).

This commit makes the SQL statement use REPLACE INTO instead of INSERT
INTO which according to the docs "REPLACE works exactly like INSERT,
except that if an old row in the table has the same value as a new row
for a PRIMARY KEY or a UNIQUE index, the old row is deleted before the
new row is inserted."

It's actually a MySQL extension but according to Wikipedia SQLite
supports it as well.
2020-05-27 18:53:30 +02:00

2210 lines
81 KiB
C++
Raw Blame History

// Copyright 2005-2020 The Mumble Developers. All rights reserved.
// Use of this source code is governed by a BSD-style license
// that can be found in the LICENSE file at the root of the
// Mumble source tree or at <https://www.mumble.info/LICENSE>.
#include <QtCore/QtGlobal>
#ifdef Q_OS_WIN
# include "win.h"
#endif
#include "ServerDB.h"
#include "ACL.h"
#include "Channel.h"
#include "Connection.h"
#include "DBus.h"
#include "Group.h"
#include "Meta.h"
#include "Server.h"
#include "ServerUser.h"
#include "User.h"
#include "PBKDF2.h"
#include "PasswordGenerator.h"
#include <QtCore/QCoreApplication>
#include <QtSql/QSqlError>
#include <QtSql/QSqlQuery>
#ifdef Q_OS_WIN
# include <winsock2.h>
#else
# include <arpa/inet.h>
#endif
#define SQLQUERY(x) ServerDB::query(query, QLatin1String(x), true)
#define SQLDO(x) ServerDB::exec(query, QLatin1String(x), true)
#define SQLMAY(x) ServerDB::exec(query, QLatin1String(x), false, false)
#define SQLPREP(x) ServerDB::prepare(query, QLatin1String(x))
#define SQLEXEC() ServerDB::exec(query)
#define SQLEXECBATCH() ServerDB::execBatch(query)
#define SOFTEXEC() ServerDB::exec(query, QString(), false)
class TransactionHolder {
public:
QSqlQuery *qsqQuery;
TransactionHolder() {
ServerDB::db->transaction();
qsqQuery = new QSqlQuery();
}
~TransactionHolder() {
qsqQuery->clear();
delete qsqQuery;
ServerDB::db->commit();
}
TransactionHolder(const TransactionHolder & other) {
ServerDB::db->transaction();
qsqQuery = other.qsqQuery ? new QSqlQuery(*other.qsqQuery) : 0;
}
};
QSqlDatabase *ServerDB::db = NULL;
Timer ServerDB::tLogClean;
QString ServerDB::qsUpgradeSuffix;
void ServerDB::loadOrSetupMetaPBKDF2IterationCount(QSqlQuery &query) {
if (!Meta::mp.legacyPasswordHash) {
if (Meta::mp.kdfIterations <= 0) {
// Configuration doesn't specify an override, load from db
SQLDO("SELECT `value` FROM `%1meta` WHERE `keystring` = 'pbkdf2_iterations'");
if (query.next()) {
Meta::mp.kdfIterations = query.value(0).toInt();
}
if (Meta::mp.kdfIterations <= 0) {
// Didn't get a valid iteration count from DB, overwrite
Meta::mp.kdfIterations = PBKDF2::benchmark();
qWarning() << "Performed initial PBKDF2 benchmark. Will use" << Meta::mp.kdfIterations << "iterations as default";
SQLPREP("INSERT INTO `%1meta` (`keystring`, `value`) VALUES('pbkdf2_iterations',?)");
query.addBindValue(Meta::mp.kdfIterations);
SQLEXEC();
}
}
if (Meta::mp.kdfIterations < PBKDF2::BENCHMARK_MINIMUM_ITERATION_COUNT) {
qWarning() << "Configured default PBKDF2 iteration count of" << Meta::mp.kdfIterations << "is below minimum recommended value of" << PBKDF2::BENCHMARK_MINIMUM_ITERATION_COUNT << "and could be insecure.";
}
}
}
ServerDB::ServerDB() {
if (Meta::mp.qsDBDriver != QLatin1String("QMYSQL") && Meta::mp.qsDBDriver != QLatin1String("QSQLITE") && Meta::mp.qsDBDriver != QLatin1String("QPSQL")) {
qFatal("ServerDB: invalid DB driver specified: '%s'. Murmur only supports QSQLITE, QMYSQL, and QPSQL.", qPrintable(Meta::mp.qsDBDriver));
}
if (! QSqlDatabase::isDriverAvailable(Meta::mp.qsDBDriver)) {
qFatal("ServerDB: Database driver %s not available", qPrintable(Meta::mp.qsDBDriver));
}
if (db) {
// Don't hide away our previous instance. Fail hard.
qFatal("ServerDB has already been instantiated!");
}
db = new QSqlDatabase(QSqlDatabase::addDatabase(Meta::mp.qsDBDriver));
qsUpgradeSuffix = QString::fromLatin1("_old_%1").arg(QDateTime::currentDateTime().toTime_t());
bool found = false;
if (Meta::mp.qsDBDriver == "QSQLITE") {
if (! Meta::mp.qsDatabase.isEmpty()) {
db->setDatabaseName(Meta::mp.qsDatabase);
found = db->open();
} else {
QStringList datapaths;
int i;
datapaths << Meta::mp.qdBasePath.absolutePath();
datapaths << QDir::currentPath();
datapaths << QCoreApplication::instance()->applicationDirPath();
datapaths << QDir::homePath();
for (i = 0; (i < datapaths.size()) && ! found; i++) {
if (!datapaths[i].isEmpty()) {
QFile f(datapaths[i] + "/murmur.sqlite");
if (f.exists()) {
db->setDatabaseName(f.fileName());
found = db->open();
}
}
}
if (! found) {
for (i = 0; (i < datapaths.size()) && ! found; i++) {
if (!datapaths[i].isEmpty()) {
QFile f(datapaths[i] + "/murmur.sqlite");
db->setDatabaseName(f.fileName());
found = db->open();
}
}
}
}
if (found) {
QFileInfo fi(db->databaseName());
qWarning("ServerDB: Opened SQLite database %s", qPrintable(fi.absoluteFilePath()));
if (! fi.isWritable())
qFatal("ServerDB: Database is not writable");
}
} else {
db->setDatabaseName(Meta::mp.qsDatabase);
db->setHostName(Meta::mp.qsDBHostName);
db->setPort(Meta::mp.iDBPort);
db->setUserName(Meta::mp.qsDBUserName);
db->setPassword(Meta::mp.qsDBPassword);
db->setConnectOptions(Meta::mp.qsDBOpts);
found = db->open();
}
if (! found) {
QSqlError e = db->lastError();
qFatal("ServerDB: Failed initialization: %s",qPrintable(e.text()));
}
// Use SQLite in WAL mode if possible.
if (Meta::mp.qsDBDriver == "QSQLITE") {
if (Meta::mp.iSQLiteWAL == 0) {
qWarning("ServerDB: Using SQLite's default rollback journal.");
} else if (Meta::mp.iSQLiteWAL > 0 && Meta::mp.iSQLiteWAL <= 2) {
QSqlQuery query;
bool hasversion = false;
bool okversion = false;
QString version;
SQLDO("SELECT sqlite_version();");
if (query.next()) {
version = query.value(0).toString();
}
QStringList splitVersion = version.split(QLatin1String("."));
int major = 0, minor = 0;
if (splitVersion.count() >= 3) {
major = splitVersion.at(0).toInt();
minor = splitVersion.at(1).toInt();
hasversion = true;
}
if (major > 3 || (major == 3 && minor >= 7)) {
okversion = true;
}
if (okversion) {
SQLDO("PRAGMA journal_mode=WAL;");
if (Meta::mp.iSQLiteWAL == 1) {
SQLDO("PRAGMA synchronous=NORMAL;");
qWarning("ServerDB: Configured SQLite for journal_mode=WAL, synchronous=NORMAL");
} else if (Meta::mp.iSQLiteWAL == 2) {
SQLDO("PRAGMA synchronous=FULL;");
qWarning("ServerDB: Configured SQLite for journal_mode=WAL, synchronous=FULL");
}
} else if (hasversion) {
qWarning("ServerDB: Unable to use SQLite in WAL mode due to incompatible SQLite version %i.%i", major, minor);
} else {
qWarning("ServerDB: Unable to use SQLite in WAL mode because the SQLite version could not be determined.");
}
} else {
qFatal("ServerDB: Invalid value '%i' for sqlite_wal. Please use 0 (no wal), 1 (wal), 2 (wal with synchronous=full)", Meta::mp.iSQLiteWAL);
}
}
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
int version = 0;
if (Meta::mp.qsDBDriver == "QSQLITE")
SQLDO("CREATE TABLE IF NOT EXISTS `%1meta` (`keystring` TEXT PRIMARY KEY, `value` TEXT)");
else if (Meta::mp.qsDBDriver == "QPSQL")
SQLQUERY("CREATE TABLE IF NOT EXISTS `%1meta` (`keystring` varchar(255) PRIMARY KEY, `value` varchar(255))");
else
SQLDO("CREATE TABLE IF NOT EXISTS `%1meta`(`keystring` varchar(255) PRIMARY KEY, `value` varchar(255)) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("SELECT `value` FROM `%1meta` WHERE `keystring` = 'version'");
if (query.next())
version = query.value(0).toInt();
loadOrSetupMetaPBKDF2IterationCount(query);
if (version < 6) {
if (version > 0) {
qWarning("Renaming old tables...");
SQLQUERY("ALTER TABLE `%1servers` RENAME TO `%1servers%2`");
if (version < 2)
SQLMAY("ALTER TABLE `%1log` RENAME TO `%1slog`");
SQLQUERY("ALTER TABLE `%1slog` RENAME TO `%1slog%2`");
SQLQUERY("ALTER TABLE `%1config` RENAME TO `%1config%2`");
SQLQUERY("ALTER TABLE `%1channels` RENAME TO `%1channels%2`");
if (version < 4)
SQLQUERY("ALTER TABLE `%1players` RENAME TO `%1players%2`");
else
SQLQUERY("ALTER TABLE `%1users` RENAME TO `%1users%2`");
SQLQUERY("ALTER TABLE `%1groups` RENAME TO `%1groups%2`");
SQLQUERY("ALTER TABLE `%1group_members` RENAME TO `%1group_members%2`");
SQLQUERY("ALTER TABLE `%1acl` RENAME TO `%1acl%2`");
SQLQUERY("ALTER TABLE `%1channel_links` RENAME TO `%1channel_links%2`");
SQLQUERY("ALTER TABLE `%1bans` RENAME TO `%1bans%2`");
if (version >= 4) {
SQLQUERY("ALTER TABLE `%1user_info` RENAME TO `%1user_info%2`");
SQLQUERY("ALTER TABLE `%1channel_info` RENAME TO `%1channel_info%2`");
}
}
qWarning("Generating new tables...");
if (Meta::mp.qsDBDriver == "QSQLITE") {
if (version > 0) {
SQLDO("DROP TRIGGER IF EXISTS `%1log_timestamp`");
SQLDO("DROP TRIGGER IF EXISTS `%1log_server_del`");
SQLDO("DROP TRIGGER IF EXISTS `%1slog_timestamp`");
SQLDO("DROP TRIGGER IF EXISTS `%1slog_server_del`");
SQLDO("DROP TRIGGER IF EXISTS `%1config_server_del`");
SQLDO("DROP TRIGGER IF EXISTS `%1channels_parent_del`");
SQLDO("DROP TRIGGER IF EXISTS `%1channels_server_del`");
SQLDO("DROP TRIGGER IF EXISTS `%1channel_info_del_channel`");
SQLDO("DROP TRIGGER IF EXISTS `%1players_server_del`");
SQLDO("DROP TRIGGER IF EXISTS `%1players_update_timestamp`");
SQLDO("DROP TRIGGER IF EXISTS `%1users_server_del`");
SQLDO("DROP TRIGGER IF EXISTS `%1users_update_timestamp`");
SQLDO("DROP TRIGGER IF EXISTS `%1user_info_del_user`");
SQLDO("DROP TRIGGER IF EXISTS `%1groups_del_channel`");
SQLDO("DROP TRIGGER IF EXISTS `%1groups_members_del_group`");
SQLDO("DROP TRIGGER IF EXISTS `%1groups_members_del_player`");
SQLDO("DROP TRIGGER IF EXISTS `%1groups_members_del_user`");
SQLDO("DROP TRIGGER IF EXISTS `%1acl_del_channel`");
SQLDO("DROP TRIGGER IF EXISTS `%1acl_del_player`");
SQLDO("DROP TRIGGER IF EXISTS `%1acl_del_user`");
SQLDO("DROP TRIGGER IF EXISTS `%1channel_links_del_channel`");
SQLDO("DROP TRIGGER IF EXISTS `%1bans_del_server`");
SQLDO("DROP INDEX IF EXISTS `%1log_time`");
SQLDO("DROP INDEX IF EXISTS `%1slog_time`");
SQLDO("DROP INDEX IF EXISTS `%1config_key`");
SQLDO("DROP INDEX IF EXISTS `%1channel_id`");
SQLDO("DROP INDEX IF EXISTS `%1channel_info_id`");
SQLDO("DROP INDEX IF EXISTS `%1players_name`");
SQLDO("DROP INDEX IF EXISTS `%1players_id`");
SQLDO("DROP INDEX IF EXISTS `%1users_name`");
SQLDO("DROP INDEX IF EXISTS `%1users_id`");
SQLDO("DROP INDEX IF EXISTS `%1user_info_id`");
SQLDO("DROP INDEX IF EXISTS `%1groups_name_channels`");
SQLDO("DROP INDEX IF EXISTS `%1acl_channel_pri`");
}
SQLDO("CREATE TABLE `%1servers` (`server_id` INTEGER PRIMARY KEY AUTOINCREMENT)");
SQLDO("CREATE TABLE `%1slog`(`server_id` INTEGER NOT NULL, `msg` TEXT, `msgtime` DATE)");
SQLDO("CREATE INDEX `%1slog_time` ON `%1slog`(`msgtime`)");
SQLDO("CREATE TRIGGER `%1slog_timestamp` AFTER INSERT ON `%1slog` FOR EACH ROW BEGIN UPDATE `%1slog` SET `msgtime` = datetime('now') WHERE rowid = new.rowid; END;");
SQLDO("CREATE TRIGGER `%1slog_server_del` AFTER DELETE ON `%1servers` FOR EACH ROW BEGIN DELETE FROM `%1slog` WHERE `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1config` (`server_id` INTEGER NOT NULL, `key` TEXT, `value` TEXT)");
SQLDO("CREATE UNIQUE INDEX `%1config_key` ON `%1config`(`server_id`, `key`)");
SQLDO("CREATE TRIGGER `%1config_server_del` AFTER DELETE ON `%1servers` FOR EACH ROW BEGIN DELETE FROM `%1config` WHERE `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1channels` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `parent_id` INTEGER, `name` TEXT, `inheritacl` INTEGER)");
SQLDO("CREATE UNIQUE INDEX `%1channel_id` ON `%1channels`(`server_id`, `channel_id`)");
SQLDO("CREATE TRIGGER `%1channels_parent_del` AFTER DELETE ON `%1channels` FOR EACH ROW BEGIN DELETE FROM `%1channels` WHERE `parent_id` = old.`channel_id` AND `server_id` = old.`server_id`; UPDATE `%1users` SET `lastchannel`=0 WHERE `lastchannel` = old.`channel_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TRIGGER `%1channels_server_del` AFTER DELETE ON `%1servers` FOR EACH ROW BEGIN DELETE FROM `%1channels` WHERE `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1channel_info` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `key` INTEGER, `value` TEXT)");
SQLDO("CREATE UNIQUE INDEX `%1channel_info_id` ON `%1channel_info`(`server_id`, `channel_id`, `key`)");
SQLDO("CREATE TRIGGER `%1channel_info_del_channel` AFTER DELETE on `%1channels` FOR EACH ROW BEGIN DELETE FROM `%1channel_info` WHERE `channel_id` = old.`channel_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1users` (`server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `name` TEXT NOT NULL, `pw` TEXT, `salt` TEXT, `kdfiterations` INTEGER, `lastchannel` INTEGER, `texture` BLOB, `last_active` DATE)");
SQLDO("CREATE UNIQUE INDEX `%1users_name` ON `%1users` (`server_id`,`name`)");
SQLDO("CREATE UNIQUE INDEX `%1users_id` ON `%1users` (`server_id`, `user_id`)");
SQLDO("CREATE TRIGGER `%1users_server_del` AFTER DELETE ON `%1servers` FOR EACH ROW BEGIN DELETE FROM `%1users` WHERE `server_id` = old.`server_id`; END;");
SQLDO("CREATE TRIGGER `%1users_update_timestamp` AFTER UPDATE OF `lastchannel` ON `%1users` FOR EACH ROW BEGIN UPDATE `%1users` SET `last_active` = datetime('now') WHERE `user_id` = old.`user_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1user_info` (`server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `key` INTEGER, `value` TEXT)");
SQLDO("CREATE UNIQUE INDEX `%1user_info_id` ON `%1user_info`(`server_id`, `user_id`, `key`)");
SQLDO("CREATE TRIGGER `%1user_info_del_user` AFTER DELETE on `%1users` FOR EACH ROW BEGIN DELETE FROM `%1user_info` WHERE `user_id` = old.`user_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1groups` (`group_id` INTEGER PRIMARY KEY AUTOINCREMENT, `server_id` INTEGER NOT NULL, `name` TEXT, `channel_id` INTEGER NOT NULL, `inherit` INTEGER, `inheritable` INTEGER)");
SQLDO("CREATE UNIQUE INDEX `%1groups_name_channels` ON `%1groups`(`server_id`, `channel_id`, `name`)");
SQLDO("CREATE TRIGGER `%1groups_del_channel` AFTER DELETE ON `%1channels` FOR EACH ROW BEGIN DELETE FROM `%1groups` WHERE `channel_id` = old.`channel_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1group_members` (`group_id` INTEGER NOT NULL, `server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `addit` INTEGER)");
SQLDO("CREATE TRIGGER `%1groups_members_del_group` AFTER DELETE ON `%1groups` FOR EACH ROW BEGIN DELETE FROM `%1group_members` WHERE `group_id` = old.`group_id`; END;");
SQLDO("CREATE TRIGGER `%1groups_members_del_user` AFTER DELETE on `%1users` FOR EACH ROW BEGIN DELETE FROM `%1group_members` WHERE `user_id` = old.`user_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1acl` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `priority` INTEGER, `user_id` INTEGER, `group_name` TEXT, `apply_here` INTEGER, `apply_sub` INTEGER, `grantpriv` INTEGER, `revokepriv` INTEGER)");
SQLDO("CREATE UNIQUE INDEX `%1acl_channel_pri` ON `%1acl`(`server_id`, `channel_id`, `priority`)");
SQLDO("CREATE TRIGGER `%1acl_del_channel` AFTER DELETE ON `%1channels` FOR EACH ROW BEGIN DELETE FROM `%1acl` WHERE `channel_id` = old.`channel_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TRIGGER `%1acl_del_user` AFTER DELETE ON `%1users` FOR EACH ROW BEGIN DELETE FROM `%1acl` WHERE `user_id` = old.`user_id` AND `server_id` = old.`server_id`; END;");
SQLDO("CREATE TABLE `%1channel_links` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `link_id` INTEGER NOT NULL)");
SQLDO("CREATE TRIGGER `%1channel_links_del_channel` AFTER DELETE ON `%1channels` FOR EACH ROW BEGIN DELETE FROM `%1channel_links` WHERE `server_id` = old.`server_id` AND (`channel_id` = old.`channel_id` OR `link_id` = old.`channel_id`); END;");
SQLDO("DELETE FROM `%1channel_links`");
SQLDO("CREATE TABLE `%1bans` (`server_id` INTEGER NOT NULL, `base` BLOB, `mask` INTEGER, `name` TEXT, `hash` TEXT, `reason` TEXT, `start` DATE, `duration` INTEGER)");
SQLDO("CREATE TRIGGER `%1bans_del_server` AFTER DELETE ON `%1servers` FOR EACH ROW BEGIN DELETE FROM `%1bans` WHERE `server_id` = old.`server_id`; END;");
} else if (Meta::mp.qsDBDriver == "QPSQL") {
if (version > 0) {
typedef QPair<QString, QString> qsp;
QList<qsp> qlForeignKeys;
QList<qsp> qlIndexes;
SQLPREP("SELECT TABLE_NAME, CONSTRAINT_NAME FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS WHERE TABLE_SCHEMA=? AND CONSTRAINT_TYPE='FOREIGN KEY'");
query.addBindValue(Meta::mp.qsDatabase);
SQLEXEC();
while (query.next())
qlForeignKeys << qsp(query.value(0).toString(), query.value(1).toString());
foreach(const qsp &key, qlForeignKeys) {
if (key.first.startsWith(Meta::mp.qsDBPrefix))
ServerDB::exec(query, QString::fromLatin1("ALTER TABLE `%1` DROP CONSTRAINT FOREIGN KEY `%2`").arg(key.first).arg(key.second), true);
}
SQLPREP("SELECT TABLE_NAME, CONSTRAINT_NAME FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS WHERE TABLE_SCHEMA=? AND CONSTRAINT_TYPE='PRIMARY KEY'");
query.addBindValue(Meta::mp.qsDatabase);
SQLEXEC();
while (query.next())
qlIndexes << qsp(query.value(0).toString(), query.value(1).toString());
foreach(const qsp &key, qlIndexes) {
if (key.first.startsWith(Meta::mp.qsDBPrefix))
ServerDB::exec(query, QString::fromLatin1("ALTER TABLE `%1` DROP CONSTRAINT PRIMARY KEY `%2`").arg(key.first).arg(key.second), true);
}
SQLQUERY("DROP INDEX IF EXISTS `%1slog_time` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1config_key` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1channel_id` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1channel_info_id` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1players_name` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1players_id` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1users_channel` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1users_name` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1users_id` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1user_info_id` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1groups_name_channels` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1group_members_users` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1acl_channel_pri` CASCADE");
SQLQUERY("DROP INDEX IF EXISTS `%1acl_user` CASCADE");
}
SQLQUERY("CREATE TABLE `%1servers`(`server_id` SERIAL PRIMARY KEY)");
SQLQUERY("CREATE TABLE `%1slog`(`server_id` INTEGER NOT NULL, `msg` TEXT, `msgtime` TIMESTAMP DEFAULT now())");
SQLQUERY("CREATE INDEX `%1slog_time` ON `%1slog`(`msgtime`)");
SQLQUERY("ALTER TABLE `%1slog` ADD CONSTRAINT `%1slog_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1config` (`server_id` INTEGER NOT NULL, `key` varchar(255), `value` TEXT)");
SQLQUERY("CREATE UNIQUE INDEX `%1config_key` ON `%1config`(`server_id`, `key`)");
SQLQUERY("ALTER TABLE `%1config` ADD CONSTRAINT `%1config_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1channels` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `parent_id` INTEGER, `name` varchar(255), `inheritacl` INTEGER)");
SQLQUERY("CREATE UNIQUE INDEX `%1channel_id` ON `%1channels`(`server_id`, `channel_id`)");
SQLQUERY("ALTER TABLE `%1channels` ADD CONSTRAINT `%1channels_parent_del` FOREIGN KEY (`server_id`, `parent_id`) REFERENCES `%1channels`(`server_id`,`channel_id`) ON DELETE CASCADE");
SQLQUERY("ALTER TABLE `%1channels` ADD CONSTRAINT `%1channels_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1channel_info` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `key` INTEGER, `value` TEXT)");
SQLQUERY("CREATE UNIQUE INDEX `%1channel_info_id` ON `%1channel_info`(`server_id`, `channel_id`, `key`)");
SQLQUERY("ALTER TABLE `%1channel_info` ADD CONSTRAINT `%1channel_info_del_channel` FOREIGN KEY (`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`,`channel_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1users` (`server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `name` varchar(255), `pw` varchar(128), `salt` varchar(128), `kdfiterations` INTEGER, `lastchannel` INTEGER, `texture` BYTEA, `last_active` TIMESTAMP)");
SQLQUERY("CREATE INDEX `%1users_channel` ON `%1users`(`server_id`, `lastchannel`)");
SQLQUERY("CREATE UNIQUE INDEX `%1users_name` ON `%1users` (`server_id`,`name`)");
SQLQUERY("CREATE UNIQUE INDEX `%1users_id` ON `%1users` (`server_id`, `user_id`)");
SQLQUERY("ALTER TABLE `%1users` ADD CONSTRAINT `%1users_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1user_info` (`server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `key` INTEGER, `value` TEXT)");
SQLQUERY("CREATE UNIQUE INDEX `%1user_info_id` ON `%1user_info`(`server_id`, `user_id`, `key`)");
SQLQUERY("ALTER TABLE `%1user_info` ADD CONSTRAINT `%1user_info_del_user` FOREIGN KEY (`server_id`, `user_id`) REFERENCES `%1users`(`server_id`,`user_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1groups` (`group_id` SERIAL PRIMARY KEY, `server_id` INTEGER NOT NULL, `name` varchar(255), `channel_id` INTEGER NOT NULL, `inherit` INTEGER, `inheritable` INTEGER)");
SQLQUERY("CREATE UNIQUE INDEX `%1groups_name_channels` ON `%1groups`(`server_id`, `channel_id`, `name`)");
SQLQUERY("ALTER TABLE `%1groups` ADD CONSTRAINT `%1groups_del_channel` FOREIGN KEY (`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`, `channel_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1group_members` (`group_id` INTEGER NOT NULL, `server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `addit` INTEGER)");
SQLQUERY("CREATE INDEX `%1group_members_users` ON `%1group_members`(`server_id`, `user_id`)");
SQLQUERY("ALTER TABLE `%1group_members` ADD CONSTRAINT `%1group_members_del_group` FOREIGN KEY (`group_id`) REFERENCES `%1groups`(`group_id`) ON DELETE CASCADE");
SQLQUERY("ALTER TABLE `%1group_members` ADD CONSTRAINT `%1group_members_del_user` FOREIGN KEY (`server_id`, `user_id`) REFERENCES `%1users`(`server_id`,`user_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1acl` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `priority` INTEGER, `user_id` INTEGER, `group_name` varchar(255), `apply_here` INTEGER, `apply_sub` INTEGER, `grantpriv` INTEGER, `revokepriv` INTEGER)");
SQLQUERY("CREATE UNIQUE INDEX `%1acl_channel_pri` ON `%1acl`(`server_id`, `channel_id`, `priority`)");
SQLQUERY("CREATE INDEX `%1acl_user` ON `%1acl`(`server_id`, `user_id`)");
SQLQUERY("ALTER TABLE `%1acl` ADD CONSTRAINT `%1acl_del_channel` FOREIGN KEY (`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`, `channel_id`) ON DELETE CASCADE");
SQLQUERY("ALTER TABLE `%1acl` ADD CONSTRAINT `%1acl_del_user` FOREIGN KEY (`server_id`, `user_id`) REFERENCES `%1users`(`server_id`, `user_id`) ON DELETE CASCADE");
SQLQUERY("CREATE TABLE `%1channel_links` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `link_id` INTEGER NOT NULL)");
SQLQUERY("ALTER TABLE `%1channel_links` ADD CONSTRAINT `%1channel_links_del_channel` FOREIGN KEY(`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`, `channel_id`) ON DELETE CASCADE");
SQLQUERY("DELETE FROM `%1channel_links`");
SQLQUERY("CREATE TABLE `%1bans` (`server_id` INTEGER NOT NULL, `base` BYTEA, `mask` INTEGER, `name` varchar(255), `hash` CHAR(40), `reason` TEXT, `start` TIMESTAMP, `duration` INTEGER)");
SQLQUERY("ALTER TABLE `%1bans` ADD CONSTRAINT `%1bans_del_server` FOREIGN KEY(`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
} else {
if (version > 0) {
typedef QPair<QString, QString> qsp;
QList<qsp> qlForeignKeys;
QList<qsp> qlIndexes;
SQLPREP("SELECT TABLE_NAME, CONSTRAINT_NAME FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS WHERE TABLE_SCHEMA=? AND CONSTRAINT_TYPE='FOREIGN KEY'");
query.addBindValue(Meta::mp.qsDatabase);
SQLEXEC();
while (query.next())
qlForeignKeys << qsp(query.value(0).toString(), query.value(1).toString());
foreach(const qsp &key, qlForeignKeys) {
if (key.first.startsWith(Meta::mp.qsDBPrefix))
ServerDB::exec(query, QString::fromLatin1("ALTER TABLE `%1` DROP FOREIGN KEY `%2`").arg(key.first).arg(key.second), true);
}
}
SQLDO("CREATE TABLE `%1servers`(`server_id` INTEGER PRIMARY KEY AUTO_INCREMENT) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE TABLE `%1slog`(`server_id` INTEGER NOT NULL, `msg` TEXT, `msgtime` TIMESTAMP) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE INDEX `%1slog_time` ON `%1slog`(`msgtime`)");
SQLDO("ALTER TABLE `%1slog` ADD CONSTRAINT `%1slog_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1config` (`server_id` INTEGER NOT NULL, `key` varchar(255), `value` TEXT) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE UNIQUE INDEX `%1config_key` ON `%1config`(`server_id`, `key`)");
SQLDO("ALTER TABLE `%1config` ADD CONSTRAINT `%1config_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1channels` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `parent_id` INTEGER, `name` varchar(255), `inheritacl` INTEGER) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE UNIQUE INDEX `%1channel_id` ON `%1channels`(`server_id`, `channel_id`)");
SQLDO("ALTER TABLE `%1channels` ADD CONSTRAINT `%1channels_parent_del` FOREIGN KEY (`server_id`, `parent_id`) REFERENCES `%1channels`(`server_id`,`channel_id`) ON DELETE CASCADE");
SQLDO("ALTER TABLE `%1channels` ADD CONSTRAINT `%1channels_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1channel_info` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `key` INTEGER, `value` LONGTEXT) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE UNIQUE INDEX `%1channel_info_id` ON `%1channel_info`(`server_id`, `channel_id`, `key`)");
SQLDO("ALTER TABLE `%1channel_info` ADD CONSTRAINT `%1channel_info_del_channel` FOREIGN KEY (`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`,`channel_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1users` (`server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `name` varchar(255), `pw` varchar(128), `salt` varchar(128), `kdfiterations` INTEGER, `lastchannel` INTEGER, `texture` LONGBLOB, `last_active` TIMESTAMP) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE INDEX `%1users_channel` ON `%1users`(`server_id`, `lastchannel`)");
SQLDO("CREATE UNIQUE INDEX `%1users_name` ON `%1users` (`server_id`,`name`)");
SQLDO("CREATE UNIQUE INDEX `%1users_id` ON `%1users` (`server_id`, `user_id`)");
SQLDO("ALTER TABLE `%1users` ADD CONSTRAINT `%1users_server_del` FOREIGN KEY (`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1user_info` (`server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `key` INTEGER, `value` LONGTEXT) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE UNIQUE INDEX `%1user_info_id` ON `%1user_info`(`server_id`, `user_id`, `key`)");
SQLDO("ALTER TABLE `%1user_info` ADD CONSTRAINT `%1user_info_del_user` FOREIGN KEY (`server_id`, `user_id`) REFERENCES `%1users`(`server_id`,`user_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1groups` (`group_id` INTEGER PRIMARY KEY AUTO_INCREMENT, `server_id` INTEGER NOT NULL, `name` varchar(255), `channel_id` INTEGER NOT NULL, `inherit` INTEGER, `inheritable` INTEGER) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE UNIQUE INDEX `%1groups_name_channels` ON `%1groups`(`server_id`, `channel_id`, `name`)");
SQLDO("ALTER TABLE `%1groups` ADD CONSTRAINT `%1groups_del_channel` FOREIGN KEY (`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`, `channel_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1group_members` (`group_id` INTEGER NOT NULL, `server_id` INTEGER NOT NULL, `user_id` INTEGER NOT NULL, `addit` INTEGER) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE INDEX `%1group_members_users` ON `%1group_members`(`server_id`, `user_id`)");
SQLDO("ALTER TABLE `%1group_members` ADD CONSTRAINT `%1group_members_del_group` FOREIGN KEY (`group_id`) REFERENCES `%1groups`(`group_id`) ON DELETE CASCADE");
SQLDO("ALTER TABLE `%1group_members` ADD CONSTRAINT `%1group_members_del_user` FOREIGN KEY (`server_id`, `user_id`) REFERENCES `%1users`(`server_id`,`user_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1acl` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `priority` INTEGER, `user_id` INTEGER, `group_name` varchar(255), `apply_here` INTEGER, `apply_sub` INTEGER, `grantpriv` INTEGER, `revokepriv` INTEGER) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("CREATE UNIQUE INDEX `%1acl_channel_pri` ON `%1acl`(`server_id`, `channel_id`, `priority`)");
SQLDO("CREATE INDEX `%1acl_user` ON `%1acl`(`server_id`, `user_id`)");
SQLDO("ALTER TABLE `%1acl` ADD CONSTRAINT `%1acl_del_channel` FOREIGN KEY (`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`, `channel_id`) ON DELETE CASCADE");
SQLDO("ALTER TABLE `%1acl` ADD CONSTRAINT `%1acl_del_user` FOREIGN KEY (`server_id`, `user_id`) REFERENCES `%1users`(`server_id`, `user_id`) ON DELETE CASCADE");
SQLDO("CREATE TABLE `%1channel_links` (`server_id` INTEGER NOT NULL, `channel_id` INTEGER NOT NULL, `link_id` INTEGER NOT NULL) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("ALTER TABLE `%1channel_links` ADD CONSTRAINT `%1channel_links_del_channel` FOREIGN KEY(`server_id`, `channel_id`) REFERENCES `%1channels`(`server_id`, `channel_id`) ON DELETE CASCADE");
SQLDO("DELETE FROM `%1channel_links`");
SQLDO("CREATE TABLE `%1bans` (`server_id` INTEGER NOT NULL, `base` BINARY(16), `mask` INTEGER, `name` varchar(255), `hash` CHAR(40), `reason` TEXT, `start` DATETIME, `duration` INTEGER) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_bin");
SQLDO("ALTER TABLE `%1bans` ADD CONSTRAINT `%1bans_del_server` FOREIGN KEY(`server_id`) REFERENCES `%1servers`(`server_id`) ON DELETE CASCADE");
}
if (version == 0) {
SQLDO("INSERT INTO `%1servers` (`server_id`) VALUES(1)");
SQLDO("INSERT INTO `%1meta` (`keystring`, `value`) VALUES('version','6')");
} else {
qWarning("Importing old data...");
if (Meta::mp.qsDBDriver == "QMYSQL")
SQLDO("SET FOREIGN_KEY_CHECKS = 0;");
SQLDO("INSERT INTO `%1servers` (`server_id`) SELECT `server_id` FROM `%1servers%2`");
SQLDO("INSERT INTO `%1slog` (`server_id`, `msg`, `msgtime`) SELECT `server_id`, `msg`, `msgtime` FROM `%1slog%2`");
if (version < 4)
SQLDO("INSERT INTO `%1config` (`server_id`, `key`, `value`) SELECT `server_id`, `keystring`, `value` FROM `%1config%2`");
else
SQLDO("INSERT INTO `%1config` (`server_id`, `key`, `value`) SELECT `server_id`, `key`, `value` FROM `%1config%2`");
SQLDO("INSERT INTO `%1channels` (`server_id`, `channel_id`, `parent_id`, `name`, `inheritacl`) SELECT `server_id`, `channel_id`, `parent_id`, `name`, `inheritacl` FROM `%1channels%2` ORDER BY `parent_id`, `channel_id`");
if (version < 4)
SQLDO("INSERT INTO `%1users` (`server_id`, `user_id`, `name`, `pw`, `lastchannel`, `texture`, `last_active`) SELECT `server_id`, `player_id`, `name`, `pw`, `lastchannel`, `texture`, `last_active` FROM `%1players%2`");
else
SQLDO("INSERT INTO `%1users` (`server_id`, `user_id`, `name`, `pw`, `lastchannel`, `texture`, `last_active`) SELECT `server_id`, `user_id`, `name`, `pw`, `lastchannel`, `texture`, `last_active` FROM `%1users%2`");
SQLDO("INSERT INTO `%1groups` (`group_id`, `server_id`, `name`, `channel_id`, `inherit`, `inheritable`) SELECT `group_id`, `server_id`, `name`, `channel_id`, `inherit`, `inheritable` FROM `%1groups%2`");
if (version < 4)
SQLDO("INSERT INTO `%1group_members` (`group_id`, `server_id`, `user_id`, `addit`) SELECT `group_id`, `server_id`, `player_id`, `addit` FROM `%1group_members%2`");
else
SQLDO("INSERT INTO `%1group_members` (`group_id`, `server_id`, `user_id`, `addit`) SELECT `group_id`, `server_id`, `user_id`, `addit` FROM `%1group_members%2`");
if (version < 4)
SQLDO("INSERT INTO `%1acl` (`server_id`, `channel_id`, `priority`, `user_id`, `group_name`, `apply_here`, `apply_sub`, `grantpriv`, `revokepriv`) SELECT `server_id`, `channel_id`, `priority`, `player_id`, `group_name`, `apply_here`, `apply_sub`, `grantpriv`, `revokepriv` FROM `%1acl%2`");
else
SQLDO("INSERT INTO `%1acl` (`server_id`, `channel_id`, `priority`, `user_id`, `group_name`, `apply_here`, `apply_sub`, `grantpriv`, `revokepriv`) SELECT `server_id`, `channel_id`, `priority`, `user_id`, `group_name`, `apply_here`, `apply_sub`, `grantpriv`, `revokepriv` FROM `%1acl%2`");
SQLDO("INSERT INTO `%1channel_links` (`server_id`, `channel_id`, `link_id`) SELECT `server_id`, `channel_id`, `link_id` FROM `%1channel_links%2`");
if (version < 4) {
QList<QList<QVariant> > ql;
SQLPREP("SELECT `server_id`, `base`, `mask` FROM `%1bans%2`");
SQLEXEC();
while (query.next()) {
QList<QVariant> l;
l << query.value(0);
l << query.value(1);
l << query.value(2);
ql << l;
}
SQLPREP("INSERT INTO `%1bans` (`server_id`, `base`, `mask`) VALUES (?, ?, ?)");
foreach(const QList<QVariant> &l, ql) {
quint32 addr = htonl(l.at(1).toUInt());
const char *ptr = reinterpret_cast<const char *>(&addr);
QByteArray qba(16, 0);
qba[10] = static_cast<char>(-1);
qba[11] = static_cast<char>(-1);
qba[12] = ptr[0];
qba[13] = ptr[1];
qba[14] = ptr[2];
qba[15] = ptr[3];
query.addBindValue(l.at(0));
query.addBindValue(qba);
query.addBindValue(l.at(2).toInt() + 96);
SQLEXEC();
}
} else {
SQLDO("INSERT INTO `%1bans` (`server_id`, `base`, `mask`) SELECT `server_id`, `base`, `mask` FROM `%1bans%2`");
}
if (version < 4)
SQLDO("INSERT INTO `%1user_info` SELECT `server_id`,`player_id`,1,`email` FROM `%1players%2` WHERE `email` IS NOT NULL");
if (version == 3) {
SQLDO("INSERT INTO `%1channel_info` SELECT `server_id`,`channel_id`,0,`description` FROM `%1channels%2` WHERE `description` IS NOT NULL");
}
if (version >= 4) {
SQLDO("INSERT INTO `%1user_info` SELECT * FROM `%1user_info%2`");
SQLDO("INSERT INTO `%1channel_info` SELECT * FROM `%1channel_info%2`");
}
if (Meta::mp.qsDBDriver == "QMYSQL")
SQLDO("SET FOREIGN_KEY_CHECKS = 1;");
qWarning("Removing old tables...");
SQLQUERY("DROP TABLE IF EXISTS `%1slog%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1config%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1channel_info%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1channels%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1user_info%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1users%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1players%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1groups%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1group_members%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1acl%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1channel_links%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1bans%2`");
SQLQUERY("DROP TABLE IF EXISTS `%1servers%2`");
SQLDO("UPDATE `%1meta` SET `value` = '6' WHERE `keystring` = 'version'");
}
}
query.clear();
}
ServerDB::~ServerDB() {
db->close();
delete db;
db = NULL;
}
bool ServerDB::prepare(QSqlQuery &query, const QString &str, bool fatal, bool warn) {
if (! db->isValid()) {
qWarning("SQL [%s] rejected: Database is gone", qPrintable(str));
return false;
}
QString q;
if (str.contains(QLatin1String("%1"))) {
if (str.contains(QLatin1String("%2")))
q = str.arg(Meta::mp.qsDBPrefix, qsUpgradeSuffix);
else
q = str.arg(Meta::mp.qsDBPrefix);
} else {
q = str;
}
if (Meta::mp.qsDBDriver == "QPSQL") {
q.replace("`", "\"");
}
if (query.prepare(q)) {
return true;
} else {
db->close();
if (! db->open()) {
qFatal("Lost connection to SQL Database: Reconnect: %s", qPrintable(db->lastError().text()));
}
query = QSqlQuery();
if (query.prepare(q)) {
qWarning("SQL Connection lost, reconnection OK");
return true;
}
if (fatal) {
*db = QSqlDatabase();
qFatal("SQL Prepare Error [%s]: %s", qPrintable(q), qPrintable(query.lastError().text()));
} else if (warn) {
qDebug("SQL Prepare Error [%s]: %s", qPrintable(q), qPrintable(query.lastError().text()));
}
return false;
}
}
bool ServerDB::query(QSqlQuery &query, const QString &str, bool fatal, bool warn) {
if (! str.isEmpty()) {
if (! db->isValid()) {
qWarning("SQL [%s] rejected: Database is gone", qPrintable(str));
return false;
}
QString q;
if (str.contains(QLatin1String("%1"))) {
if (str.contains(QLatin1String("%2")))
q = str.arg(Meta::mp.qsDBPrefix, qsUpgradeSuffix);
else
q = str.arg(Meta::mp.qsDBPrefix);
} else {
q = str;
}
if (Meta::mp.qsDBDriver == "QPSQL") {
q.replace("`", "\"");
}
if (query.exec(q)) {
return true;
} else {
if (fatal) {
*db = QSqlDatabase();
qFatal("SQL Error [%s]: %s", qPrintable(query.lastQuery()), qPrintable(query.lastError().text()));
} else if (warn) {
qDebug("SQL Error [%s]: %s", qPrintable(query.lastQuery()), qPrintable(query.lastError().text()));
}
return false;
}
} else {
return false;
}
}
bool ServerDB::exec(QSqlQuery &query, const QString &str, bool fatal, bool warn) {
if (! str.isEmpty())
prepare(query, str, fatal, warn);
if (query.exec()) {
return true;
} else {
if (fatal) {
*db = QSqlDatabase();
qFatal("SQL Error [%s]: %s", qPrintable(query.lastQuery()), qPrintable(query.lastError().text()));
} else if (warn) {
qDebug("SQL Error [%s]: %s", qPrintable(query.lastQuery()), qPrintable(query.lastError().text()));
}
return false;
}
}
bool ServerDB::execBatch(QSqlQuery &query, const QString &str, bool fatal) {
if (! str.isEmpty())
prepare(query, str, fatal);
if (query.execBatch()) {
return true;
} else {
if (fatal) {
*db = QSqlDatabase();
qFatal("SQL Error [%s]: %s", qPrintable(query.lastQuery()), qPrintable(query.lastError().text()));
} else
qDebug("SQL Error [%s]: %s", qPrintable(query.lastQuery()), qPrintable(query.lastError().text()));
return false;
}
}
void Server::initialize() {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `channel_id` FROM `%1channels` WHERE `server_id` = ? AND `channel_id` = 0");
query.addBindValue(iServerNum);
SQLEXEC();
if (! query.next()) {
SQLPREP("INSERT INTO `%1channels` (`server_id`, `channel_id`, `parent_id`, `name`) VALUES (?, ?, ?, ?)");
query.addBindValue(iServerNum);
query.addBindValue(0);
query.addBindValue(QVariant());
query.addBindValue(QLatin1String("Root"));
SQLEXEC();
SQLPREP("UPDATE `%1channels` SET `channel_id` = 0 WHERE `server_id` = ? AND `name` = ? AND `parent_id` IS NULL");
query.addBindValue(iServerNum);
query.addBindValue(QLatin1String("Root"));
SQLEXEC();
}
SQLPREP("SELECT `user_id` FROM `%1users` WHERE `server_id` = ? AND `user_id` = 0");
query.addBindValue(iServerNum);
SQLEXEC();
if (! query.next()) {
SQLPREP("INSERT INTO `%1users` (`server_id`, `user_id`, `name`) VALUES (?, ?, ?)");
query.addBindValue(iServerNum);
query.addBindValue(0);
query.addBindValue(QLatin1String("SuperUser"));
SQLEXEC();
SQLPREP("UPDATE `%1users` SET `user_id` = 0 WHERE `server_id` = ? AND `name` = ?");
query.addBindValue(iServerNum);
query.addBindValue(QLatin1String("SuperUser"));
SQLEXEC();
const int passwordLength = 12;
QString pw = PasswordGenerator::generatePassword(passwordLength);
ServerDB::setSUPW(iServerNum, pw);
log(QString("Password for 'SuperUser' set to '%2'").arg(pw));
}
SQLPREP("SELECT COUNT(*) FROM `%1acl` WHERE `server_id`=?");
query.addBindValue(iServerNum);
SQLEXEC();
if (query.next()) {
int c = query.value(0).toInt();
if (c == 0) {
SQLPREP("INSERT INTO `%1acl` (`server_id`, `channel_id`, `priority`, `group_name`, `apply_here`, `apply_sub`, `grantpriv`) VALUES (?,?,?,?,?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(0);
query.addBindValue(1);
query.addBindValue(QLatin1String("admin"));
query.addBindValue(1);
query.addBindValue(1);
query.addBindValue(static_cast<int>(ChanACL::Write));
SQLEXEC();
query.addBindValue(iServerNum);
query.addBindValue(0);
query.addBindValue(2);
query.addBindValue(QLatin1String("auth"));
query.addBindValue(1);
query.addBindValue(1);
query.addBindValue(static_cast<int>(ChanACL::MakeTempChannel));
SQLEXEC();
query.addBindValue(iServerNum);
query.addBindValue(0);
query.addBindValue(3);
query.addBindValue(QLatin1String("all"));
query.addBindValue(1);
query.addBindValue(0);
query.addBindValue(static_cast<int>(ChanACL::SelfRegister));
SQLEXEC();
}
}
SQLPREP("SELECT COUNT(*) FROM `%1groups` WHERE `server_id`=? AND `channel_id`=?");
query.addBindValue(iServerNum);
query.addBindValue(0);
SQLEXEC();
if (query.next()) {
int c = query.value(0).toInt();
if (c == 0) {
SQLPREP("INSERT INTO `%1groups`(`server_id`, `channel_id`, `name`, `inherit`, `inheritable`) VALUES (?,?,?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(0);
query.addBindValue(QLatin1String("admin"));
query.addBindValue(1);
query.addBindValue(1);
SQLEXEC();
}
}
query.clear();
}
int Server::registerUser(const QMap<int, QString> &info) {
const QString &name = info.value(ServerDB::User_Name);
if (name.isEmpty())
return -1;
if (! validateUserName(name))
return -1;
if (getUserID(name) >= 0)
return -1;
qhUserIDCache.remove(name);
int res = -2;
emit registerUserSig(res, info);
if (res != -2) {
qhUserIDCache.remove(name);
}
if (res == -1)
return res;
int id = 0;
{
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
if (res < 0) {
SQLPREP("SELECT MAX(`user_id`)+1 AS id FROM `%1users` WHERE `server_id`=? AND `user_id` < 1000000000");
query.addBindValue(iServerNum);
SQLEXEC();
if (query.next())
id = query.value(0).toInt();
} else {
id = res;
}
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("INSERT INTO `%1users` (`server_id`, `user_id`, `name`) VALUES (:server_id,:user_id,:name) ON CONFLICT (`server_id`, `name`) DO UPDATE SET `user_id` = :u_user_id WHERE `%1users`.`server_id` = :u_server_id AND `%1users`.`name` = :u_name");
query.bindValue(":server_id", iServerNum);
query.bindValue(":user_id", id);
query.bindValue(":name", name);
query.bindValue(":u_server_id", iServerNum);
query.bindValue(":u_user_id", id);
query.bindValue(":u_name", name);
SQLEXEC();
} else {
SQLPREP("REPLACE INTO `%1users` (`server_id`, `user_id`, `name`) VALUES (?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(id);
query.addBindValue(name);
SQLEXEC();
}
}
qhUserNameCache.remove(id);
setInfo(id, info);
return id;
}
bool Server::unregisterUserDB(int id) {
if (id <= 0)
return false;
QMap<int, QString> info = getRegistration(id);
if (info.isEmpty())
return false;
qhUserIDCache.remove(info.value(ServerDB::User_Name));
qhUserNameCache.remove(id);
int res = -2;
emit unregisterUserSig(res, id);
if (res == 0) {
return false;
}
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("DELETE FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
SQLPREP("DELETE FROM `%1user_info` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
return true;
}
QList<UserInfo> Server::getRegisteredUsersEx() {
QMap<int, QString> rpcUsers;
emit getRegisteredUsersSig(QString(), rpcUsers);
QList<UserInfo> users;
QMap<int, QString>::iterator it = rpcUsers.begin();
for (; it != rpcUsers.end(); ++it)
{
users.insert(it.key(), UserInfo(it.key(), it.value()));
}
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `user_id`, `name`, `lastchannel`, `last_active` FROM `%1users` WHERE `server_id` = ?");
query.addBindValue(iServerNum);
SQLEXEC();
while (query.next()) {
UserInfo userinfo;
userinfo.user_id = query.value(0).toInt();
userinfo.name = query.value(1).toString();
userinfo.last_channel = query.value(2).toInt();
userinfo.last_active = QDateTime::fromString(query.value(3).toString(), Qt::ISODate);
userinfo.last_active.setTimeSpec(Qt::UTC);
users << userinfo;
}
return users;
}
QMap<int, QString > Server::getRegisteredUsers(const QString &filter) {
QMap<int, QString > m;
emit getRegisteredUsersSig(filter, m);
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
if (filter.isEmpty()) {
SQLPREP("SELECT `user_id`, `name` FROM `%1users` WHERE `server_id` = ?");
query.addBindValue(iServerNum);
} else {
SQLPREP("SELECT `user_id`, `name` FROM `%1users` WHERE `server_id` = ? AND `name` LIKE ?");
query.addBindValue(iServerNum);
query.addBindValue(filter);
}
SQLEXEC();
while (query.next()) {
int id = query.value(0).toInt();
QString name = query.value(1).toString();
m.insert(id, name);
}
return m;
}
bool Server::isUserId(int id) {
QMap<int, QString> info;
int res = -2;
emit getRegistrationSig(res, id, info);
if (res >= 0)
return (res > 0);
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `user_id` FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
if (query.next()) {
return true;
}
return false;
}
QMap<int, QString> Server::getRegistration(int id) {
QMap<int, QString> info;
int res = -2;
emit getRegistrationSig(res, id, info);
if (res >= 0)
return info;
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `name`, `last_active` FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
if (query.next()) {
info.insert(ServerDB::User_Name, query.value(0).toString());
info.insert(ServerDB::User_LastActive, query.value(1).toString());
SQLPREP("SELECT `key`, `value` FROM `%1user_info` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
while (query.next()) {
const int key = query.value(0).toInt();
if (!info.contains(key))
info.insert(key, query.value(1).toString());
}
}
return info;
}
/// @return UserID of authenticated user, -1 for authentication failures, -2 for unknown user (fallthrough),
/// -3 for authentication failures where the data could (temporarily) not be verified.
int Server::authenticate(QString &name, const QString &password, int sessionId, const QStringList &emails, const QString &certhash, bool bStrongCert, const QList<QSslCertificate> &certs) {
int res = bForceExternalAuth ? -3 : -2;
emit authenticateSig(res, name, sessionId, certs, certhash, bStrongCert, password);
if (res != -2) {
// External authentication handled it. Ignore certificate completely.
if (res != -1) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
int lchan=readLastChannel(res);
if (lchan < 0)
lchan = 0;
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("INSERT INTO `%1users` (`server_id`, `user_id`, `name`, `lastchannel`) VALUES (:server_id,:user_id,:name,:lastchannel) ON CONFLICT (`server_id`, `user_id`) DO UPDATE SET `name` = :u_name, `lastchannel` = :u_lastchannel WHERE `%1users`.`server_id` = :u_server_id AND `%1users`.`user_id` = :u_user_id");
query.bindValue(":server_id", iServerNum);
query.bindValue(":user_id", res);
query.bindValue(":name", name);
query.bindValue(":lastchannel", lchan);
query.bindValue(":u_server_id", iServerNum);
query.bindValue(":u_user_id", res);
query.bindValue(":u_name", name);
query.bindValue(":u_lastchannel", lchan);
SQLEXEC();
} else {
SQLPREP("REPLACE INTO `%1users` (`server_id`, `user_id`, `name`, `lastchannel`) VALUES (?,?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(res);
query.addBindValue(name);
query.addBindValue(lchan);
SQLEXEC();
}
}
if (res >= 0) {
qhUserNameCache.remove(res);
qhUserIDCache.remove(name);
}
return res;
}
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `user_id`,`name`,`pw`, `salt`, `kdfiterations` FROM `%1users` WHERE `server_id` = ? AND LOWER(`name`) = LOWER(?)");
query.addBindValue(iServerNum);
query.addBindValue(name);
SQLEXEC();
if (query.next()) {
const int userId = query.value(0).toInt();
const QString storedPasswordHash = query.value(2).toString();
const QString storedSalt = query.value(3).toString();
const int storedKdfIterations = query.value(4).toInt();
res = -1;
if (!storedPasswordHash.isEmpty()) {
// A user has password authentication enabled if there is a password hash.
if (storedKdfIterations <= 0) {
// If storedKdfIterations is <=0 this means this is an old-style SHA1 hash
// that hasn't been converted yet. Or we are operating in legacy mode.
if (ServerDB::getLegacySHA1Hash(password) == storedPasswordHash) {
name = query.value(1).toString();
res = query.value(0).toInt();
if (! Meta::mp.legacyPasswordHash) {
// Unless disabled upgrade the user password hash
QMap<int, QString> info;
info.insert(ServerDB::User_Password, password);
info.insert(ServerDB::User_KDFIterations, QString::number(Meta::mp.kdfIterations));
if (!setInfo(userId, info)) {
qWarning("ServerDB: Failed to upgrade user account to PBKDF2 hash, rejecting login.");
return -1;
}
}
}
} else {
if (PBKDF2::getHash(storedSalt, password, storedKdfIterations) == storedPasswordHash) {
name = query.value(1).toString();
res = query.value(0).toInt();
if (Meta::mp.legacyPasswordHash) {
// Downgrade the password to the legacy hash
QMap<int, QString> info;
info.insert(ServerDB::User_Password, password);
if (!setInfo(userId, info)) {
qWarning("ServerDB: Failed to downgrade user account to legacy hash, rejecting login.");
return -1;
}
} else if (storedKdfIterations != Meta::mp.kdfIterations) {
// User kdfiterations not equal to the global one. Update it.
QMap<int, QString> info;
info.insert(ServerDB::User_Password, password);
info.insert(ServerDB::User_KDFIterations, QString::number(Meta::mp.kdfIterations));
if (!setInfo(userId, info)) {
qWarning() << "ServerDB: Failed to update user PBKDF2 to new iteration count" << Meta::mp.kdfIterations << ", rejecting login.";
return -1;
}
}
}
}
}
if (userId == 0 && res < 0) {
// For SuperUser only password based authentication is allowed.
// If we couldn't verify the password don't proceed to cert auth
// and instead reject the login attempt.
return -1;
}
}
// No password match. Try cert or email match, but only for non-SuperUser.
if (!certhash.isEmpty() && (res < 0)) {
SQLPREP("SELECT `user_id` FROM `%1user_info` WHERE `server_id` = ? AND `key` = ? AND `value` = ?");
query.addBindValue(iServerNum);
query.addBindValue(ServerDB::User_Hash);
query.addBindValue(certhash);
SQLEXEC();
if (query.next()) {
res = query.value(0).toInt();
} else if (bStrongCert) {
foreach(const QString &email, emails) {
if (! email.isEmpty()) {
query.addBindValue(iServerNum);
query.addBindValue(ServerDB::User_Email);
query.addBindValue(email);
SQLEXEC();
if (query.next()) {
res = query.value(0).toInt();
break;
}
}
}
}
if (res > 0) {
SQLPREP("SELECT `name` FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(res);
SQLEXEC();
if (! query.next()) {
res = -1;
} else {
name = query.value(0).toString();
}
}
}
if (! certhash.isEmpty() && (res > 0)) {
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("INSERT INTO `%1user_info` (`server_id`, `user_id`, `key`, `value`) VALUES (:server_id, :user_id, :key, :value) ON CONFLICT (`server_id`, `user_id`, `key`) DO UPDATE SET `value` = :u_value WHERE `%1user_info`.`server_id` = :u_server_id AND `%1user_info`.`user_id` = :u_user_id AND `%1user_info`.`key` = :u_key");
query.bindValue(":server_id", iServerNum);
query.bindValue(":user_id", res);
query.bindValue(":key", ServerDB::User_Hash);
query.bindValue(":value", certhash);
query.bindValue(":u_server_id", iServerNum);
query.bindValue(":u_user_id", res);
query.bindValue(":u_key", ServerDB::User_Hash);
query.bindValue(":u_value", certhash);
SQLEXEC();
} else {
SQLPREP("REPLACE INTO `%1user_info` (`server_id`, `user_id`, `key`, `value`) VALUES (?, ?, ?, ?)");
query.addBindValue(iServerNum);
query.addBindValue(res);
query.addBindValue(ServerDB::User_Hash);
query.addBindValue(certhash);
SQLEXEC();
}
if (! emails.isEmpty()) {
if (Meta::mp.qsDBDriver == "QPSQL") {
query.bindValue(":server_id", iServerNum);
query.bindValue(":user_id", res);
query.bindValue(":key", ServerDB::User_Email);
query.bindValue(":value", emails.at(0));
query.bindValue(":u_server_id", iServerNum);
query.bindValue(":u_user_id", res);
query.bindValue(":u_key", ServerDB::User_Email);
query.bindValue(":u_value", emails.at(0));
SQLEXEC();
} else {
query.addBindValue(iServerNum);
query.addBindValue(res);
query.addBindValue(ServerDB::User_Email);
query.addBindValue(emails.at(0));
SQLEXEC();
}
}
}
if (res >= 0) {
qhUserNameCache.remove(res);
qhUserIDCache.remove(name);
}
return res;
}
bool Server::setInfo(int id, const QMap<int, QString> &setinfo) {
int res = -2;
QMap<int, QString> info = setinfo;
if (info.contains(ServerDB::User_Name)) {
const QString &uname = info.value(ServerDB::User_Name);
if (uname.isEmpty())
return false;
int idmatch = getUserID(uname);
if ((idmatch >= 0) && (idmatch != id))
return false;
qhUserIDCache.remove(qhUserNameCache.value(id));
qhUserNameCache.remove(id);
qhUserIDCache.remove(info.value(ServerDB::User_Name));
}
emit setInfoSig(res, id, info);
if (res >= 0)
return (res > 0);
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
if (info.contains(ServerDB::User_LastActive)) {
info.remove(ServerDB::User_LastActive);
}
if (info.contains(ServerDB::User_Password)) {
const QString password = info.value(ServerDB::User_Password);
QString passwordHash, salt;
int kdfIterations = -1;
if (Meta::mp.legacyPasswordHash) {
passwordHash = ServerDB::getLegacySHA1Hash(password);
} else {
kdfIterations = Meta::mp.kdfIterations;
if (info.contains(ServerDB::User_KDFIterations)) {
const int targetIterations = info.value(ServerDB::User_KDFIterations).toInt();
if (targetIterations > 0) {
kdfIterations = targetIterations;
}
}
salt = PBKDF2::getSalt();
passwordHash = PBKDF2::getHash(salt, password, kdfIterations);
}
SQLPREP("UPDATE `%1users` SET `pw`=?, `salt`=?, `kdfiterations`=? WHERE `server_id` = ? AND `user_id`=?");
query.addBindValue(passwordHash);
query.addBindValue(salt);
query.addBindValue(kdfIterations);
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
info.remove(ServerDB::User_Password);
}
if (info.contains(ServerDB::User_Name)) {
const QString &name = info.value(ServerDB::User_Name);
SQLPREP("UPDATE `%1users` SET `name`=? WHERE `server_id` = ? AND `user_id`=?");
query.addBindValue(name);
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
info.remove(ServerDB::User_Name);
}
if (! info.isEmpty()) {
QMap<int, QString>::const_iterator i;
QVariantList serverids, userids, keys, values;
for (i = info.constBegin(); i != info.constEnd(); ++i) {
serverids << iServerNum;
userids << id;
keys << i.key();
values << i.value();
}
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("INSERT INTO `%1user_info` (`server_id`, `user_id`, `key`, `value`) VALUES (:server_id, :user_id, :key, :value) ON CONFLICT (`server_id`, `user_id`, `key`) DO UPDATE SET `value` = :u_value WHERE `%1user_info`.`server_id` = :u_server_id AND `%1user_info`.`user_id` = :u_user_id AND `%1user_info`.`key` = :u_key");
query.bindValue(":server_id", serverids);
query.bindValue(":user_id", userids);
query.bindValue(":key", keys);
query.bindValue(":value", values);
query.bindValue(":u_server_id", serverids);
query.bindValue(":u_user_id", userids);
query.bindValue(":u_key", keys);
query.bindValue(":u_value", values);
SQLEXECBATCH();
} else {
SQLPREP("REPLACE INTO `%1user_info` (`server_id`, `user_id`, `key`, `value`) VALUES (?,?,?,?)");
query.addBindValue(serverids);
query.addBindValue(userids);
query.addBindValue(keys);
query.addBindValue(values);
SQLEXECBATCH();
}
}
return true;
}
bool Server::setTexture(int id, const QByteArray &texture) {
if (id <= 0)
return false;
QByteArray tex;
if (texture.size() == 600 * 60 * 4)
tex = qCompress(texture);
else
tex = texture;
foreach(ServerUser *u, qhUsers) {
if (u->iId == id)
hashAssign(u->qbaTexture, u->qbaTextureHash, tex);
}
int res = -2;
emit setTextureSig(res, id, tex);
if (res >= 0)
return (res > 0);
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("UPDATE `%1users` SET `texture`=? WHERE `server_id` = ? AND `user_id`=?");
query.addBindValue(tex, QSql::Binary | QSql::In);
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
return true;
}
void ServerDB::writeSUPW(int srvnum, const QString &pwHash, const QString &saltHash, const QVariant &kdfIterations) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `user_id` FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(srvnum);
query.addBindValue(0);
SQLEXEC();
if (! query.next()) {
SQLPREP("INSERT INTO `%1users` (`server_id`, `user_id`, `name`) VALUES (?, ?, ?)");
query.addBindValue(srvnum);
query.addBindValue(0);
query.addBindValue(QLatin1String("SuperUser"));
SQLEXEC();
}
SQLPREP("UPDATE `%1users` SET `pw`=?, `salt`=?, `kdfiterations`=? WHERE `server_id` = ? AND `user_id`=?");
query.addBindValue(pwHash);
query.addBindValue(saltHash);
query.addBindValue(kdfIterations);
query.addBindValue(srvnum);
query.addBindValue(0);
SQLEXEC();
}
void ServerDB::setSUPW(int srvnum, const QString &pw) {
QString pwHash, saltHash;
if (!Meta::mp.legacyPasswordHash) {
saltHash = PBKDF2::getSalt();
pwHash = PBKDF2::getHash(saltHash, pw, Meta::mp.kdfIterations);
} else {
pwHash = getLegacySHA1Hash(pw);
}
writeSUPW(srvnum, pwHash, saltHash, Meta::mp.kdfIterations);
}
void ServerDB::disableSU(int srvnum) {
writeSUPW(srvnum, QString(), QString(), QVariant()); // NULL, NULL, NULL
}
QString ServerDB::getLegacySHA1Hash(const QString &password) {
QByteArray hash = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1);
return QString::fromLatin1(hash.toHex());
}
QString Server::getUserName(int id) {
if (qhUserNameCache.contains(id))
return qhUserNameCache.value(id);
QString name;
emit idToNameSig(name, id);
if (! name.isEmpty()) {
qhUserIDCache.insert(name, id);
qhUserNameCache.insert(id, name);
return name;
}
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `name` FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
if (query.next()) {
name = query.value(0).toString();
qhUserIDCache.insert(name, id);
qhUserNameCache.insert(id, name);
}
return name;
}
int Server::getUserID(const QString &name) {
if (qhUserIDCache.contains(name))
return qhUserIDCache.value(name);
int id = -2;
emit nameToIdSig(id, name);
if (id != -2) {
qhUserIDCache.insert(name, id);
qhUserNameCache.insert(id, name);
return id;
}
if (!validateUserName(name)) {
return id;
}
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `user_id` FROM `%1users` WHERE `server_id` = ? AND LOWER(`name`) = LOWER(?)");
query.addBindValue(iServerNum);
query.addBindValue(name);
SQLEXEC();
if (query.next()) {
id = query.value(0).toInt();
qhUserIDCache.insert(name, id);
qhUserNameCache.insert(id, name);
}
return id;
}
QByteArray Server::getUserTexture(int id) {
QByteArray qba;
emit idToTextureSig(qba, id);
if (! qba.isNull()) {
return qba;
}
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `texture` FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
if (query.next()) {
qba = query.value(0).toByteArray();
if (! qba.isEmpty())
if (qba.size() == 600 * 60 * 4)
qba = qCompress(qba);
}
return qba;
}
void Server::addLink(Channel *c, Channel *l) {
{
QWriteLocker wl(&qrwlVoiceThread);
c->link(l);
}
if (c->bTemporary || l->bTemporary)
return;
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("INSERT INTO `%1channel_links` (`server_id`, `channel_id`, `link_id`) VALUES (?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
query.addBindValue(l->iId);
SQLEXEC();
query.addBindValue(iServerNum);
query.addBindValue(l->iId);
query.addBindValue(c->iId);
SQLEXEC();
}
void Server::removeLink(Channel *c, Channel *l) {
{
QWriteLocker wl(&qrwlVoiceThread);
c->unlink(l);
}
if (c->bTemporary || l->bTemporary)
return;
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("DELETE FROM `%1channel_links` WHERE `server_id` = ? AND `channel_id` = ? AND `link_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
query.addBindValue(l->iId);
SQLEXEC();
query.addBindValue(iServerNum);
query.addBindValue(l->iId);
query.addBindValue(c->iId);
SQLEXEC();
}
Channel *Server::addChannel(Channel *p, const QString &name, bool temporary, int position, unsigned int maxUsers) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT MAX(`channel_id`)+1 AS id FROM `%1channels` WHERE `server_id`=?");
query.addBindValue(iServerNum);
SQLEXEC();
int id = 0;
if (query.next())
id = query.value(0).toInt();
// Temporary channels might "complicate" this somewhat.
while (qhChannels.contains(id))
++id;
if (! temporary) {
SQLPREP("INSERT INTO `%1channels` (`server_id`, `parent_id`, `channel_id`, `name`) VALUES (?,?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(p->iId);
query.addBindValue(id);
query.addBindValue(name);
SQLEXEC();
// Delete old channel_info rows
SQLPREP("DELETE FROM `%1channel_info` WHERE `server_id` = ? AND `channel_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
// Add channel sorting information
SQLPREP("INSERT INTO `%1channel_info` ( `server_id`, `channel_id`, `key`, `value`) VALUES(?,?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(id);
query.addBindValue(ServerDB::Channel_Position);
query.addBindValue(QVariant(position).toString());
SQLEXEC();
// Update channel maximum users
query.addBindValue(iServerNum);
query.addBindValue(id);
query.addBindValue(ServerDB::Channel_Max_Users);
query.addBindValue(QVariant(maxUsers).toString());
SQLEXEC();
}
Channel *c = new Channel(id, name, p);
c->bTemporary = temporary;
c->iPosition = position;
c->uiMaxUsers = maxUsers;
qhChannels.insert(id, c);
return c;
}
void Server::removeChannelDB(const Channel *c) {
if (! c->bTemporary) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("DELETE FROM `%1channels` WHERE `server_id` = ? AND `channel_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
SQLEXEC();
}
qhChannels.remove(c->iId);
}
void Server::updateChannel(const Channel *c) {
if (c->bTemporary)
return;
TransactionHolder th;
Group *g;
ChanACL *acl;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("UPDATE `%1channels` SET `name` = ?, `parent_id` = ?, `inheritacl` = ? WHERE `server_id` = ? AND `channel_id` = ?");
query.addBindValue(c->qsName);
query.addBindValue(c->cParent ? c->cParent->iId : QVariant());
query.addBindValue(c->bInheritACL ? 1 : 0);
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
SQLEXEC();
// Update channel description information
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("INSERT INTO `%1channel_info` (`server_id`, `channel_id`, `key`, `value`) VALUES (:server_id, :channel_id, :key, :value) ON CONFLICT (`server_id`, `channel_id`, `key`) DO UPDATE SET `value` = :u_value WHERE `%1channel_info`.`server_id` = :u_server_id AND `%1channel_info`.`channel_id` = :u_channel_id AND `%1channel_info`.`key` = :u_key");
query.bindValue(":server_id", iServerNum);
query.bindValue(":channel_id", c->iId);
query.bindValue(":key", ServerDB::Channel_Description);
query.bindValue(":value", c->qsDesc);
query.bindValue(":u_server_id", iServerNum);
query.bindValue(":u_channel_id", c->iId);
query.bindValue(":u_key", ServerDB::Channel_Description);
query.bindValue(":u_value", c->qsDesc);
SQLEXEC();
} else {
SQLPREP("REPLACE INTO `%1channel_info` (`server_id`, `channel_id`, `key`, `value`) VALUES (?, ?, ?, ?)");
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
query.addBindValue(ServerDB::Channel_Description);
query.addBindValue(c->qsDesc);
SQLEXEC();
}
// Update channel position information
if (Meta::mp.qsDBDriver == "QPSQL") {
query.bindValue(":server_id", iServerNum);
query.bindValue(":channel_id", c->iId);
query.bindValue(":key", ServerDB::Channel_Position);
query.bindValue(":value", QVariant(c->iPosition).toString());
query.bindValue(":u_server_id", iServerNum);
query.bindValue(":u_channel_id", c->iId);
query.bindValue(":u_key", ServerDB::Channel_Position);
query.bindValue(":u_value", QVariant(c->iPosition).toString());
SQLEXEC();
} else {
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
query.addBindValue(ServerDB::Channel_Position);
query.addBindValue(QVariant(c->iPosition).toString());
SQLEXEC();
}
// Update channel maximum channels
if (Meta::mp.qsDBDriver == "QPSQL") {
query.bindValue(":server_id", iServerNum);
query.bindValue(":channel_id", c->iId);
query.bindValue(":key", ServerDB::Channel_Max_Users);
query.bindValue(":value", QVariant(c->uiMaxUsers).toString());
query.bindValue(":u_server_id", iServerNum);
query.bindValue(":u_channel_id", c->iId);
query.bindValue(":u_key", ServerDB::Channel_Max_Users);
query.bindValue(":u_value", QVariant(c->uiMaxUsers).toString());
SQLEXEC();
} else {
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
query.addBindValue(ServerDB::Channel_Max_Users);
query.addBindValue(QVariant(c->uiMaxUsers).toString());
SQLEXEC();
}
SQLPREP("DELETE FROM `%1groups` WHERE `server_id` = ? AND `channel_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
SQLEXEC();
SQLPREP("DELETE FROM `%1acl` WHERE `server_id` = ? AND `channel_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(c->iId);
SQLEXEC();
foreach(g, c->qhGroups) {
int id = 0;
int pid;
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("INSERT INTO `%1groups` (`server_id`, `channel_id`, `name`, `inherit`, `inheritable`) VALUES (?,?,?,?,?) RETURNING group_id");
query.addBindValue(iServerNum);
query.addBindValue(g->c->iId);
query.addBindValue(g->qsName);
query.addBindValue(g->bInherit ? 1 : 0);
query.addBindValue(g->bInheritable ? 1 : 0);
SQLEXEC();
if (query.next()) {
id = query.value(0).toInt();
} else {
qFatal("ServerDB: internal query failure: PostgreSQL query did not return the inserted group's group_id");
}
} else {
SQLPREP("REPLACE INTO `%1groups` (`server_id`, `channel_id`, `name`, `inherit`, `inheritable`) VALUES (?,?,?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(g->c->iId);
query.addBindValue(g->qsName);
query.addBindValue(g->bInherit ? 1 : 0);
query.addBindValue(g->bInheritable ? 1 : 0);
SQLEXEC();
id = query.lastInsertId().toInt();
}
foreach(pid, g->qsAdd) {
SQLPREP("INSERT INTO `%1group_members` (`group_id`, `server_id`, `user_id`, `addit`) VALUES (?, ?, ?, ?)");
query.addBindValue(id);
query.addBindValue(iServerNum);
query.addBindValue(pid);
query.addBindValue(1);
SQLEXEC();
}
foreach(pid, g->qsRemove) {
SQLPREP("INSERT INTO `%1group_members` (`group_id`, `server_id`, `user_id`, `addit`) VALUES (?, ?, ?, ?)");
query.addBindValue(id);
query.addBindValue(iServerNum);
query.addBindValue(pid);
query.addBindValue(0);
SQLEXEC();
}
}
int pri = 5;
foreach(acl, c->qlACL) {
SQLPREP("INSERT INTO `%1acl` (`server_id`, `channel_id`, `priority`, `user_id`, `group_name`, `apply_here`, `apply_sub`, `grantpriv`, `revokepriv`) VALUES (?,?,?,?,?,?,?,?,?)");
query.addBindValue(iServerNum);
query.addBindValue(acl->c->iId);
query.addBindValue(pri++);
query.addBindValue((acl->iUserId == -1) ? QVariant() : acl->iUserId);
query.addBindValue((acl->qsGroup.isEmpty()) ? QVariant() : acl->qsGroup);
query.addBindValue(acl->bApplyHere ? 1 : 0);
query.addBindValue(acl->bApplySubs ? 1 : 0);
query.addBindValue(static_cast<int>(acl->pAllow));
query.addBindValue(static_cast<int>(acl->pDeny));
SQLEXEC();
}
}
/** Reads the channel privileges (group and acl) as well as the channel information key/value pairs from the database.
* @param c Channel to fetch information for
*/
void Server::readChannelPrivs(Channel *c) {
TransactionHolder th;
int cid = c->iId;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `key`, `value` FROM `%1channel_info` WHERE `server_id` = ? AND `channel_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(cid);
SQLEXEC();
while (query.next()) {
int key = query.value(0).toInt();
const QString &value = query.value(1).toString();
if (key == ServerDB::Channel_Description) {
hashAssign(c->qsDesc, c->qbaDescHash, value);
} else if (key == ServerDB::Channel_Position) {
c->iPosition = QVariant(value).toInt(); // If the conversion fails it'll return the default value 0
} else if (key == ServerDB::Channel_Max_Users) {
c->uiMaxUsers = QVariant(value).toUInt(); // If the conversion fails it'll return the default value 0
}
}
SQLPREP("SELECT `group_id`, `name`, `inherit`, `inheritable` FROM `%1groups` WHERE `server_id` = ? AND `channel_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(cid);
SQLEXEC();
while (query.next()) {
int gid = query.value(0).toInt();
QString name = query.value(1).toString();
Group *g = new Group(c, name);
g->bInherit = query.value(2).toBool();
g->bInheritable = query.value(3).toBool();
QSqlQuery mem;
mem.prepare(QString::fromLatin1("SELECT user_id, addit FROM %1group_members WHERE group_id = ?").arg(Meta::mp.qsDBPrefix));
mem.addBindValue(gid);
mem.exec();
while (mem.next()) {
int uid = mem.value(0).toInt();
if (mem.value(1).toBool())
g->qsAdd << uid;
else
g->qsRemove << uid;
}
}
SQLPREP("SELECT `user_id`, `group_name`, `apply_here`, `apply_sub`, `grantpriv`, `revokepriv` FROM `%1acl` WHERE `server_id` = ? AND `channel_id` = ? ORDER BY `priority`");
query.addBindValue(iServerNum);
query.addBindValue(cid);
SQLEXEC();
while (query.next()) {
ChanACL *acl = new ChanACL(c);
acl->iUserId = query.value(0).isNull() ? -1 : query.value(0).toInt();
acl->qsGroup = query.value(1).toString();
acl->bApplyHere = query.value(2).toBool();
acl->bApplySubs = query.value(3).toBool();
acl->pAllow = static_cast<ChanACL::Permissions>(query.value(4).toInt());
acl->pDeny = static_cast<ChanACL::Permissions>(query.value(5).toInt());
}
}
void Server::readChannels(Channel *p) {
QList<Channel *> kids;
Channel *c;
QSqlQuery query;
int parentid = -1;
if (p) {
parentid = p->iId;
readChannelPrivs(qhChannels.value(parentid));
}
{
TransactionHolder th;
if (parentid == -1) {
SQLPREP("SELECT `channel_id`, `name`, `inheritacl` FROM `%1channels` WHERE `server_id` = ? AND `parent_id` IS NULL ORDER BY `name`");
query.addBindValue(iServerNum);
} else {
SQLPREP("SELECT `channel_id`, `name`, `inheritacl` FROM `%1channels` WHERE `server_id` = ? AND `parent_id`=? ORDER BY `name`");
query.addBindValue(iServerNum);
query.addBindValue(parentid);
}
SQLEXEC();
while (query.next()) {
c = new Channel(query.value(0).toInt(), query.value(1).toString(), p);
if (! p)
c->setParent(this);
qhChannels.insert(c->iId, c);
c->bInheritACL = query.value(2).toBool();
kids << c;
}
}
query.clear();
foreach(c, kids)
readChannels(c);
}
void Server::readLinks() {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `channel_id`, `link_id` FROM `%1channel_links` WHERE `server_id` = ?");
query.addBindValue(iServerNum);
SQLEXEC();
while (query.next()) {
int cid = query.value(0).toInt();
int lid = query.value(1).toInt();
Channel *c = qhChannels.value(cid);
Channel *l = qhChannels.value(lid);
if (c && l) {
QWriteLocker wl(&qrwlVoiceThread);
c->link(l);
}
}
}
void Server::setLastChannel(const User *p) {
if (p->iId < 0)
return;
if (p->cChannel->bTemporary)
return;
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
if (Meta::mp.qsDBDriver == "QSQLITE") {
SQLPREP("UPDATE `%1users` SET `lastchannel`=? WHERE `server_id` = ? AND `user_id` = ?");
} else {
SQLPREP("UPDATE `%1users` SET `lastchannel`=?, `last_active` = now() WHERE `server_id` = ? AND `user_id` = ?");
}
query.addBindValue(p->cChannel->iId);
query.addBindValue(iServerNum);
query.addBindValue(p->iId);
SQLEXEC();
}
int Server::readLastChannel(int id) {
if (id < 0)
return -1;
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `lastchannel` FROM `%1users` WHERE `server_id` = ? AND `user_id` = ?");
query.addBindValue(iServerNum);
query.addBindValue(id);
SQLEXEC();
if (query.next()) {
int cid = query.value(0).toInt();
if (qhChannels.contains(cid))
return cid;
}
return -1;
}
void Server::dumpChannel(const Channel *c) {
Group *g;
ChanACL *acl;
int pid;
if (c == NULL) {
c = qhChannels.value(0);
}
qWarning("Channel %s (ACLInherit %d)", qPrintable(c->qsName), c->bInheritACL);
qWarning("Description: %s", qPrintable(c->qsDesc));
foreach(g, c->qhGroups) {
qWarning("Group %s (Inh %d Able %d)", qPrintable(g->qsName), g->bInherit, g->bInheritable);
foreach(pid, g->qsAdd)
qWarning("Add %d", pid);
foreach(pid, g->qsRemove)
qWarning("Remove %d", pid);
}
foreach(acl, c->qlACL) {
int allow = static_cast<int>(acl->pAllow);
int deny = static_cast<int>(acl->pDeny);
qWarning("ChanACL Here %d Sub %d Allow %04x Deny %04x ID %d Group %s", acl->bApplyHere, acl->bApplySubs, allow, deny, acl->iUserId, qPrintable(acl->qsGroup));
}
qWarning(" ");
foreach(c, c->qlChannels) {
dumpChannel(c);
}
}
void Server::getBans() {
TransactionHolder th;
qlBans.clear();
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `base`,`mask`,`name`,`hash`,`reason`,`start`,`duration` FROM `%1bans` WHERE `server_id` = ?");
query.addBindValue(iServerNum);
SQLEXEC();
while (query.next()) {
Ban ban;
ban.haAddress = query.value(0).toByteArray();
ban.iMask = query.value(1).toInt();
ban.qsUsername = query.value(2).toString();
ban.qsHash = query.value(3).toString();
ban.qsReason = query.value(4).toString();
ban.qdtStart = query.value(5).toDateTime();
ban.qdtStart.setTimeSpec(Qt::UTC);
ban.iDuration = query.value(6).toInt();
if (ban.isValid())
qlBans << ban;
}
}
void Server::saveBans() {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("DELETE FROM `%1bans` WHERE `server_id` = ? ");
query.addBindValue(iServerNum);
SQLEXEC();
SQLPREP("INSERT INTO `%1bans` (`server_id`, `base`,`mask`,`name`,`hash`,`reason`,`start`,`duration`) VALUES (?,?,?,?,?,?,?,?)");
foreach(const Ban &ban, qlBans) {
query.addBindValue(iServerNum);
query.addBindValue(ban.haAddress.toByteArray());
query.addBindValue(ban.iMask);
query.addBindValue(ban.qsUsername);
query.addBindValue(ban.qsHash);
query.addBindValue(ban.qsReason);
query.addBindValue(ban.qdtStart);
query.addBindValue(ban.iDuration);
SQLEXEC();
}
}
QVariant Server::getConf(const QString &key, QVariant def) {
return ServerDB::getConf(iServerNum, key, def);
}
QVariant ServerDB::getConf(int server_id, const QString &key, QVariant def) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `value` FROM `%1config` WHERE `server_id` = ? AND `key` = ?");
query.addBindValue(server_id);
query.addBindValue(key);
SQLEXEC();
if (query.next()) {
return query.value(0);
}
return def;
}
QMap<QString, QString> ServerDB::getAllConf(int server_id) {
TransactionHolder th;
QMap<QString, QString> map;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `key`, `value` FROM `%1config` WHERE `server_id` = ?");
query.addBindValue(server_id);
SQLEXEC();
while (query.next()) {
map.insert(query.value(0).toString(), query.value(1).toString());
}
return map;
}
void Server::setConf(const QString &key, const QVariant &value) {
ServerDB::setConf(iServerNum, key, value);
}
void Server::dblog(const QString &str) const {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
// Is logging disabled?
if (Meta::mp.iLogDays < 0)
return;
// Once per hour
if (Meta::mp.iLogDays > 0) {
if (ServerDB::tLogClean.isElapsed(3600ULL * 1000000ULL)) {
QString qstr;
if (Meta::mp.qsDBDriver == "QSQLITE") {
qstr = QString::fromLatin1("msgtime < datetime('now','-%1 days')").arg(Meta::mp.iLogDays);
} else if (Meta::mp.qsDBDriver == "QPSQL") {
qstr = QString::fromLatin1("msgtime < now() - INTERVAL '%1 day'").arg(Meta::mp.iLogDays);
} else {
qstr = QString::fromLatin1("msgtime < now() - INTERVAL %1 day").arg(Meta::mp.iLogDays);
}
ServerDB::prepare(query, QString::fromLatin1("DELETE FROM %1slog WHERE ") + qstr);
SQLEXEC();
}
}
SQLPREP("INSERT INTO `%1slog` (`server_id`, `msg`) VALUES(?,?)");
query.addBindValue(iServerNum);
query.addBindValue(str);
SQLEXEC();
}
void ServerDB::wipeLogs() {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLDO("DELETE FROM %1slog");
}
QList<QPair<unsigned int, QString> > ServerDB::getLog(int server_id, unsigned int offs_min, unsigned int offs_max) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("SELECT `msgtime`, `msg` FROM `%1slog` WHERE `server_id` = ? ORDER BY `msgtime` DESC LIMIT ? OFFSET ?");
query.addBindValue(server_id);
query.addBindValue(offs_max);
query.addBindValue(offs_min);
SQLEXEC();
} else {
SQLPREP("SELECT `msgtime`, `msg` FROM `%1slog` WHERE `server_id` = ? ORDER BY `msgtime` DESC LIMIT ?, ?");
query.addBindValue(server_id);
query.addBindValue(offs_min);
query.addBindValue(offs_max);
SQLEXEC();
}
QList<QPair<unsigned int, QString> > ql;
while (query.next()) {
QDateTime qdt = query.value(0).toDateTime();
QString msg = query.value(1).toString();
ql << QPair<unsigned int, QString>(qdt.toLocalTime().toTime_t(), msg);
}
return ql;
}
int ServerDB::getLogLen(int server_id) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT COUNT(`msgtime`) FROM `%1slog` WHERE `server_id` = ?");
query.addBindValue(server_id);
SQLEXEC();
while (query.next()) {
return query.value(0).toInt();
}
return -1;
}
void ServerDB::setConf(int server_id, const QString &k, const QVariant &value) {
TransactionHolder th;
const QString &key = (k == "serverpassword") ? "password" : k;
QSqlQuery &query = *th.qsqQuery;
if (value.isNull() || value.toString().trimmed().isEmpty()) {
SQLPREP("DELETE FROM `%1config` WHERE `server_id` = ? AND `key` = ?");
query.addBindValue(server_id);
query.addBindValue(key);
} else {
if (Meta::mp.qsDBDriver == "QPSQL") {
SQLPREP("INSERT INTO `%1config` (`server_id`, `key`, `value`) VALUES (:server_id, :key, :value) ON CONFLICT (`server_id`, `key`) DO UPDATE SET `value` = :u_value WHERE `%1config`.`server_id` = :u_server_id AND `%1config`.`key` = :u_key");
query.bindValue(":server_id", server_id);
query.bindValue(":key", key);
query.bindValue(":value", value.toString());
query.bindValue(":u_server_id", server_id);
query.bindValue(":u_key", key);
query.bindValue(":u_value", value.toString());
} else {
SQLPREP("REPLACE INTO `%1config` (`server_id`, `key`, `value`) VALUES (?,?,?)");
query.addBindValue(server_id);
query.addBindValue(key);
query.addBindValue(value.toString());
}
}
SQLEXEC();
}
QList<int> ServerDB::getAllServers() {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `server_id` FROM `%1servers`");
SQLEXEC();
QList<int> ql;
while (query.next())
ql << query.value(0).toInt();
return ql;
}
QList<int> ServerDB::getBootServers() {
QList<int> ql = getAllServers();
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
QList<int> bootlist;
foreach(int i, ql) {
SQLPREP("SELECT `value` FROM `%1config` WHERE `server_id` = ? AND `key` = ?");
query.addBindValue(i);
query.addBindValue(QLatin1String("boot"));
SQLEXEC();
if (! query.next() || query.value(0).toBool())
bootlist << i;
}
return bootlist;
}
bool ServerDB::serverExists(int num) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT `server_id` FROM `%1servers` WHERE `server_id` = ?");
query.addBindValue(num);
SQLEXEC();
if (query.next())
return true;
return false;
}
int ServerDB::addServer() {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("SELECT MAX(`server_id`)+1 AS id FROM `%1servers`");
SQLEXEC();
int id = 0;
if (query.next())
id = qMax(1, query.value(0).toInt());
SQLPREP("INSERT INTO `%1servers` (`server_id`) VALUES (?)");
query.addBindValue(id);
SQLEXEC();
return id;
}
void ServerDB::deleteServer(int server_id) {
TransactionHolder th;
QSqlQuery &query = *th.qsqQuery;
SQLPREP("DELETE FROM `%1servers` WHERE `server_id` = ?");
query.addBindValue(server_id);
SQLEXEC();
}
Reference in New Issue View Git Blame Copy Permalink