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
6e08d23db9
mumble/src/murmur/ServerDB.cpp

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("INSERT 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