src/core/sqlitestorage.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2018 by the Quassel Project                        *
   3  *   devel@quassel-irc.org                                                 *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) version 3.                                           *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.         *
  19  ***************************************************************************/
  20
  21 #include "sqlitestorage.h"
  22
  23 #include <QtSql>
  24
  25 #include "logger.h"
  26 #include "network.h"
  27 #include "quassel.h"
  28
  29 int SqliteStorage::_maxRetryCount = 150;
  30
  31 SqliteStorage::SqliteStorage(QObject *parent)
  32     : AbstractSqlStorage(parent)
  33 {
  34 }
  35
  36
  37 SqliteStorage::~SqliteStorage()
  38 {
  39 }
  40
  41
  42 bool SqliteStorage::isAvailable() const
  43 {
  44     if (!QSqlDatabase::isDriverAvailable("QSQLITE")) return false;
  45     return true;
  46 }
  47
  48
  49 QString SqliteStorage::backendId() const
  50 {
  51     return QString("SQLite");
  52 }
  53
  54
  55 QString SqliteStorage::displayName() const
  56 {
  57     // Note: Pre-0.13 clients use the displayName property for backend idenfication
  58     // We identify the backend to use for the monolithic core by its displayname.
  59     // so only change this string if you _really_ have to and make sure the core
  60     // setup for the mono client still works ;)
  61     return backendId();
  62 }
  63
  64
  65 QString SqliteStorage::description() const
  66 {
  67     return tr("SQLite is a file-based database engine that does not require any setup. It is suitable for small and medium-sized "
  68               "databases that do not require access via network. Use SQLite if your Quassel Core should store its data on the same machine "
  69               "it is running on, and if you only expect a few users to use your core.");
  70 }
  71
  72
  73 int SqliteStorage::installedSchemaVersion()
  74 {
  75     // only used when there is a singlethread (during startup)
  76     // so we don't need locking here
  77     QSqlQuery query = logDb().exec("SELECT value FROM coreinfo WHERE key = 'schemaversion'");
  78     if (query.first())
  79         return query.value(0).toInt();
  80
  81     // maybe it's really old... (schema version 0)
  82     query = logDb().exec("SELECT MAX(version) FROM coreinfo");
  83     if (query.first())
  84         return query.value(0).toInt();
  85
  86     return AbstractSqlStorage::installedSchemaVersion();
  87 }
  88
  89
  90 bool SqliteStorage::updateSchemaVersion(int newVersion)
  91 {
  92     // only used when there is a singlethread (during startup)
  93     // so we don't need locking here
  94     QSqlQuery query(logDb());
  95     query.prepare("UPDATE coreinfo SET value = :version WHERE key = 'schemaversion'");
  96     query.bindValue(":version", newVersion);
  97     query.exec();
  98
  99     bool success = true;
 100     if (query.lastError().isValid()) {
 101         qCritical() << "SqliteStorage::updateSchemaVersion(int): Updating schema version failed!";
 102         success = false;
 103     }
 104     return success;
 105 }
 106
 107
 108 bool SqliteStorage::setupSchemaVersion(int version)
 109 {
 110     // only used when there is a singlethread (during startup)
 111     // so we don't need locking here
 112     QSqlQuery query(logDb());
 113     query.prepare("INSERT INTO coreinfo (key, value) VALUES ('schemaversion', :version)");
 114     query.bindValue(":version", version);
 115     query.exec();
 116
 117     bool success = true;
 118     if (query.lastError().isValid()) {
 119         qCritical() << "SqliteStorage::setupSchemaVersion(int): Updating schema version failed!";
 120         success = false;
 121     }
 122     return success;
 123 }
 124
 125
 126 UserId SqliteStorage::addUser(const QString &user, const QString &password, const QString &authenticator)
 127 {
 128     QSqlDatabase db = logDb();
 129     UserId uid;
 130
 131     db.transaction();
 132     // this scope ensures that the query is freed in sqlite before we call unlock()
 133     // this ensures that our thread doesn't hold a internal after unlock is called
 134     // (see sqlites doc on implicit locking for details)
 135     {
 136         QSqlQuery query(db);
 137         query.prepare(queryString("insert_quasseluser"));
 138         query.bindValue(":username", user);
 139         query.bindValue(":password", hashPassword(password));
 140         query.bindValue(":hashversion", Storage::HashVersion::Latest);
 141         query.bindValue(":authenticator", authenticator);
 142         lockForWrite();
 143         safeExec(query);
 144         if (query.lastError().isValid() && query.lastError().number() == 19) { // user already exists - sadly 19 seems to be the general constraint violation error...
 145             db.rollback();
 146         }
 147         else {
 148             uid = query.lastInsertId().toInt();
 149             db.commit();
 150         }
 151     }
 152     unlock();
 153
 154     if (uid.isValid())
 155         emit userAdded(uid, user);
 156     return uid;
 157 }
 158
 159
 160 bool SqliteStorage::updateUser(UserId user, const QString &password)
 161 {
 162     QSqlDatabase db = logDb();
 163     bool success = false;
 164
 165     db.transaction();
 166     {
 167         QSqlQuery query(db);
 168         query.prepare(queryString("update_userpassword"));
 169         query.bindValue(":userid", user.toInt());
 170         query.bindValue(":password", hashPassword(password));
 171         query.bindValue(":hashversion", Storage::HashVersion::Latest);
 172         lockForWrite();
 173         safeExec(query);
 174         success = query.numRowsAffected() != 0;
 175         db.commit();
 176     }
 177     unlock();
 178     return success;
 179 }
 180
 181
 182 void SqliteStorage::renameUser(UserId user, const QString &newName)
 183 {
 184     QSqlDatabase db = logDb();
 185     db.transaction();
 186     {
 187         QSqlQuery query(db);
 188         query.prepare(queryString("update_username"));
 189         query.bindValue(":userid", user.toInt());
 190         query.bindValue(":username", newName);
 191         lockForWrite();
 192         safeExec(query);
 193         db.commit();
 194     }
 195     unlock();
 196     emit userRenamed(user, newName);
 197 }
 198
 199
 200 UserId SqliteStorage::validateUser(const QString &user, const QString &password)
 201 {
 202     UserId userId;
 203     QString hashedPassword;
 204     Storage::HashVersion hashVersion = Storage::HashVersion::Latest;
 205
 206     {
 207         QSqlQuery query(logDb());
 208         query.prepare(queryString("select_authuser"));
 209         query.bindValue(":username", user);
 210
 211         lockForRead();
 212         safeExec(query);
 213
 214         if (query.first()) {
 215             userId = query.value(0).toInt();
 216             hashedPassword = query.value(1).toString();
 217             hashVersion = static_cast<Storage::HashVersion>(query.value(2).toInt());
 218         }
 219     }
 220     unlock();
 221
 222     UserId returnUserId;
 223     if (userId != 0 && checkHashedPassword(userId, password, hashedPassword, hashVersion)) {
 224         returnUserId = userId;
 225     }
 226     return returnUserId;
 227 }
 228
 229
 230 UserId SqliteStorage::getUserId(const QString &username)
 231 {
 232     UserId userId;
 233
 234     {
 235         QSqlQuery query(logDb());
 236         query.prepare(queryString("select_userid"));
 237         query.bindValue(":username", username);
 238
 239         lockForRead();
 240         safeExec(query);
 241
 242         if (query.first()) {
 243             userId = query.value(0).toInt();
 244         }
 245     }
 246     unlock();
 247
 248     return userId;
 249 }
 250
 251 QString SqliteStorage::getUserAuthenticator(const UserId userid)
 252 {
 253     QString authenticator = QString("");
 254
 255     {
 256         QSqlQuery query(logDb());
 257         query.prepare(queryString("select_authenticator"));
 258         query.bindValue(":userid", userid.toInt());
 259
 260         lockForRead();
 261         safeExec(query);
 262
 263         if (query.first()) {
 264             authenticator = query.value(0).toString();
 265         }
 266     }
 267     unlock();
 268
 269     return authenticator;
 270 }
 271
 272 UserId SqliteStorage::internalUser()
 273 {
 274     UserId userId;
 275
 276     {
 277         QSqlQuery query(logDb());
 278         query.prepare(queryString("select_internaluser"));
 279         lockForRead();
 280         safeExec(query);
 281
 282         if (query.first()) {
 283             userId = query.value(0).toInt();
 284         }
 285     }
 286     unlock();
 287
 288     return userId;
 289 }
 290
 291
 292 void SqliteStorage::delUser(UserId user)
 293 {
 294     QSqlDatabase db = logDb();
 295     db.transaction();
 296
 297     lockForWrite();
 298     {
 299         QSqlQuery query(db);
 300         query.prepare(queryString("delete_backlog_by_uid"));
 301         query.bindValue(":userid", user.toInt());
 302         safeExec(query);
 303
 304         query.prepare(queryString("delete_buffers_by_uid"));
 305         query.bindValue(":userid", user.toInt());
 306         safeExec(query);
 307
 308         query.prepare(queryString("delete_networks_by_uid"));
 309         query.bindValue(":userid", user.toInt());
 310         safeExec(query);
 311
 312         query.prepare(queryString("delete_quasseluser"));
 313         query.bindValue(":userid", user.toInt());
 314         safeExec(query);
 315         // I hate the lack of foreign keys and on delete cascade... :(
 316         db.commit();
 317     }
 318     unlock();
 319
 320     emit userRemoved(user);
 321 }
 322
 323
 324 void SqliteStorage::setUserSetting(UserId userId, const QString &settingName, const QVariant &data)
 325 {
 326     QByteArray rawData;
 327     QDataStream out(&rawData, QIODevice::WriteOnly);
 328     out.setVersion(QDataStream::Qt_4_2);
 329     out << data;
 330
 331     QSqlDatabase db = logDb();
 332     db.transaction();
 333     {
 334         QSqlQuery query(db);
 335         query.prepare(queryString("insert_user_setting"));
 336         query.bindValue(":userid", userId.toInt());
 337         query.bindValue(":settingname", settingName);
 338         query.bindValue(":settingvalue", rawData);
 339         lockForWrite();
 340         safeExec(query);
 341
 342         if (query.lastError().isValid()) {
 343             QSqlQuery updateQuery(db);
 344             updateQuery.prepare(queryString("update_user_setting"));
 345             updateQuery.bindValue(":userid", userId.toInt());
 346             updateQuery.bindValue(":settingname", settingName);
 347             updateQuery.bindValue(":settingvalue", rawData);
 348             safeExec(updateQuery);
 349         }
 350         db.commit();
 351     }
 352     unlock();
 353 }
 354
 355
 356 QVariant SqliteStorage::getUserSetting(UserId userId, const QString &settingName, const QVariant &defaultData)
 357 {
 358     QVariant data = defaultData;
 359     {
 360         QSqlQuery query(logDb());
 361         query.prepare(queryString("select_user_setting"));
 362         query.bindValue(":userid", userId.toInt());
 363         query.bindValue(":settingname", settingName);
 364         lockForRead();
 365         safeExec(query);
 366
 367         if (query.first()) {
 368             QByteArray rawData = query.value(0).toByteArray();
 369             QDataStream in(&rawData, QIODevice::ReadOnly);
 370             in.setVersion(QDataStream::Qt_4_2);
 371             in >> data;
 372         }
 373     }
 374     unlock();
 375     return data;
 376 }
 377
 378
 379 void SqliteStorage::setCoreState(const QVariantList &data)
 380 {
 381     QByteArray rawData;
 382     QDataStream out(&rawData, QIODevice::WriteOnly);
 383     out.setVersion(QDataStream::Qt_4_2);
 384     out << data;
 385
 386     QSqlDatabase db = logDb();
 387     db.transaction();
 388     {
 389         QSqlQuery query(db);
 390         query.prepare(queryString("insert_core_state"));
 391         query.bindValue(":key", "active_sessions");
 392         query.bindValue(":value", rawData);
 393         lockForWrite();
 394         safeExec(query);
 395
 396         if (query.lastError().isValid()) {
 397             QSqlQuery updateQuery(db);
 398             updateQuery.prepare(queryString("update_core_state"));
 399             updateQuery.bindValue(":key", "active_sessions");
 400             updateQuery.bindValue(":value", rawData);
 401             safeExec(updateQuery);
 402         }
 403         db.commit();
 404     }
 405     unlock();
 406 }
 407
 408
 409 QVariantList SqliteStorage::getCoreState(const QVariantList &defaultData)
 410 {
 411     QVariantList data;
 412     {
 413         QSqlQuery query(logDb());
 414         query.prepare(queryString("select_core_state"));
 415         query.bindValue(":key", "active_sessions");
 416         lockForRead();
 417         safeExec(query);
 418
 419         if (query.first()) {
 420             QByteArray rawData = query.value(0).toByteArray();
 421             QDataStream in(&rawData, QIODevice::ReadOnly);
 422             in.setVersion(QDataStream::Qt_4_2);
 423             in >> data;
 424         } else {
 425             data = defaultData;
 426         }
 427     }
 428     unlock();
 429     return data;
 430 }
 431
 432
 433 IdentityId SqliteStorage::createIdentity(UserId user, CoreIdentity &identity)
 434 {
 435     IdentityId identityId;
 436
 437     QSqlDatabase db = logDb();
 438     db.transaction();
 439
 440     {
 441         QSqlQuery query(db);
 442         query.prepare(queryString("insert_identity"));
 443         query.bindValue(":userid", user.toInt());
 444         query.bindValue(":identityname", identity.identityName());
 445         query.bindValue(":realname", identity.realName());
 446         query.bindValue(":awaynick", identity.awayNick());
 447         query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
 448         query.bindValue(":awayreason", identity.awayReason());
 449         query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
 450         query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
 451         query.bindValue(":autoawaytime", identity.autoAwayTime());
 452         query.bindValue(":autoawayreason", identity.autoAwayReason());
 453         query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
 454         query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
 455         query.bindValue(":detachawayreason", identity.detachAwayReason());
 456         query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
 457         query.bindValue(":ident", identity.ident());
 458         query.bindValue(":kickreason", identity.kickReason());
 459         query.bindValue(":partreason", identity.partReason());
 460         query.bindValue(":quitreason", identity.quitReason());
 461 #ifdef HAVE_SSL
 462         query.bindValue(":sslcert", identity.sslCert().toPem());
 463         query.bindValue(":sslkey", identity.sslKey().toPem());
 464 #else
 465         query.bindValue(":sslcert", QByteArray());
 466         query.bindValue(":sslkey", QByteArray());
 467 #endif
 468
 469         lockForWrite();
 470         safeExec(query);
 471
 472         identityId = query.lastInsertId().toInt();
 473         if (!identityId.isValid()) {
 474             watchQuery(query);
 475         }
 476         else {
 477             QSqlQuery deleteNickQuery(db);
 478             deleteNickQuery.prepare(queryString("delete_nicks"));
 479             deleteNickQuery.bindValue(":identityid", identityId.toInt());
 480             safeExec(deleteNickQuery);
 481
 482             QSqlQuery insertNickQuery(db);
 483             insertNickQuery.prepare(queryString("insert_nick"));
 484             foreach(QString nick, identity.nicks()) {
 485                 insertNickQuery.bindValue(":identityid", identityId.toInt());
 486                 insertNickQuery.bindValue(":nick", nick);
 487                 safeExec(insertNickQuery);
 488             }
 489         }
 490         db.commit();
 491     }
 492     unlock();
 493     identity.setId(identityId);
 494     return identityId;
 495 }
 496
 497
 498 bool SqliteStorage::updateIdentity(UserId user, const CoreIdentity &identity)
 499 {
 500     QSqlDatabase db = logDb();
 501     bool error = false;
 502     db.transaction();
 503
 504     {
 505         QSqlQuery checkQuery(db);
 506         checkQuery.prepare(queryString("select_checkidentity"));
 507         checkQuery.bindValue(":identityid", identity.id().toInt());
 508         checkQuery.bindValue(":userid", user.toInt());
 509         lockForRead();
 510         safeExec(checkQuery);
 511
 512         // there should be exactly one identity for the given id and user
 513         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 1);
 514     }
 515     if (error) {
 516         unlock();
 517         return false;
 518     }
 519
 520     {
 521         QSqlQuery query(db);
 522         query.prepare(queryString("update_identity"));
 523         query.bindValue(":identityname", identity.identityName());
 524         query.bindValue(":realname", identity.realName());
 525         query.bindValue(":awaynick", identity.awayNick());
 526         query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
 527         query.bindValue(":awayreason", identity.awayReason());
 528         query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
 529         query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
 530         query.bindValue(":autoawaytime", identity.autoAwayTime());
 531         query.bindValue(":autoawayreason", identity.autoAwayReason());
 532         query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
 533         query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
 534         query.bindValue(":detachawayreason", identity.detachAwayReason());
 535         query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
 536         query.bindValue(":ident", identity.ident());
 537         query.bindValue(":kickreason", identity.kickReason());
 538         query.bindValue(":partreason", identity.partReason());
 539         query.bindValue(":quitreason", identity.quitReason());
 540 #ifdef HAVE_SSL
 541         query.bindValue(":sslcert", identity.sslCert().toPem());
 542         query.bindValue(":sslkey", identity.sslKey().toPem());
 543 #else
 544         query.bindValue(":sslcert", QByteArray());
 545         query.bindValue(":sslkey", QByteArray());
 546 #endif
 547         query.bindValue(":identityid", identity.id().toInt());
 548         safeExec(query);
 549         watchQuery(query);
 550
 551         QSqlQuery deleteNickQuery(db);
 552         deleteNickQuery.prepare(queryString("delete_nicks"));
 553         deleteNickQuery.bindValue(":identityid", identity.id().toInt());
 554         safeExec(deleteNickQuery);
 555         watchQuery(deleteNickQuery);
 556
 557         QSqlQuery insertNickQuery(db);
 558         insertNickQuery.prepare(queryString("insert_nick"));
 559         foreach(QString nick, identity.nicks()) {
 560             insertNickQuery.bindValue(":identityid", identity.id().toInt());
 561             insertNickQuery.bindValue(":nick", nick);
 562             safeExec(insertNickQuery);
 563             watchQuery(insertNickQuery);
 564         }
 565         db.commit();
 566     }
 567     unlock();
 568     return true;
 569 }
 570
 571
 572 void SqliteStorage::removeIdentity(UserId user, IdentityId identityId)
 573 {
 574     QSqlDatabase db = logDb();
 575     db.transaction();
 576
 577     bool error = false;
 578     {
 579         QSqlQuery checkQuery(db);
 580         checkQuery.prepare(queryString("select_checkidentity"));
 581         checkQuery.bindValue(":identityid", identityId.toInt());
 582         checkQuery.bindValue(":userid", user.toInt());
 583         lockForRead();
 584         safeExec(checkQuery);
 585
 586         // there should be exactly one identity for the given id and user
 587         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 1);
 588     }
 589     if (error) {
 590         unlock();
 591         return;
 592     }
 593
 594     {
 595         QSqlQuery deleteNickQuery(db);
 596         deleteNickQuery.prepare(queryString("delete_nicks"));
 597         deleteNickQuery.bindValue(":identityid", identityId.toInt());
 598         safeExec(deleteNickQuery);
 599
 600         QSqlQuery deleteIdentityQuery(db);
 601         deleteIdentityQuery.prepare(queryString("delete_identity"));
 602         deleteIdentityQuery.bindValue(":identityid", identityId.toInt());
 603         deleteIdentityQuery.bindValue(":userid", user.toInt());
 604         safeExec(deleteIdentityQuery);
 605         db.commit();
 606     }
 607     unlock();
 608 }
 609
 610
 611 QList<CoreIdentity> SqliteStorage::identities(UserId user)
 612 {
 613     QList<CoreIdentity> identities;
 614     QSqlDatabase db = logDb();
 615     db.transaction();
 616
 617     {
 618         QSqlQuery query(db);
 619         query.prepare(queryString("select_identities"));
 620         query.bindValue(":userid", user.toInt());
 621
 622         QSqlQuery nickQuery(db);
 623         nickQuery.prepare(queryString("select_nicks"));
 624
 625         lockForRead();
 626         safeExec(query);
 627
 628         while (query.next()) {
 629             CoreIdentity identity(IdentityId(query.value(0).toInt()));
 630
 631             identity.setIdentityName(query.value(1).toString());
 632             identity.setRealName(query.value(2).toString());
 633             identity.setAwayNick(query.value(3).toString());
 634             identity.setAwayNickEnabled(!!query.value(4).toInt());
 635             identity.setAwayReason(query.value(5).toString());
 636             identity.setAwayReasonEnabled(!!query.value(6).toInt());
 637             identity.setAutoAwayEnabled(!!query.value(7).toInt());
 638             identity.setAutoAwayTime(query.value(8).toInt());
 639             identity.setAutoAwayReason(query.value(9).toString());
 640             identity.setAutoAwayReasonEnabled(!!query.value(10).toInt());
 641             identity.setDetachAwayEnabled(!!query.value(11).toInt());
 642             identity.setDetachAwayReason(query.value(12).toString());
 643             identity.setDetachAwayReasonEnabled(!!query.value(13).toInt());
 644             identity.setIdent(query.value(14).toString());
 645             identity.setKickReason(query.value(15).toString());
 646             identity.setPartReason(query.value(16).toString());
 647             identity.setQuitReason(query.value(17).toString());
 648 #ifdef HAVE_SSL
 649             identity.setSslCert(query.value(18).toByteArray());
 650             identity.setSslKey(query.value(19).toByteArray());
 651 #endif
 652
 653             nickQuery.bindValue(":identityid", identity.id().toInt());
 654             QList<QString> nicks;
 655             safeExec(nickQuery);
 656             watchQuery(nickQuery);
 657             while (nickQuery.next()) {
 658                 nicks << nickQuery.value(0).toString();
 659             }
 660             identity.setNicks(nicks);
 661             identities << identity;
 662         }
 663         db.commit();
 664     }
 665     unlock();
 666     return identities;
 667 }
 668
 669
 670 NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info)
 671 {
 672     NetworkId networkId;
 673
 674     QSqlDatabase db = logDb();
 675     db.transaction();
 676
 677     bool error = false;
 678     {
 679         QSqlQuery query(db);
 680         query.prepare(queryString("insert_network"));
 681         query.bindValue(":userid", user.toInt());
 682         bindNetworkInfo(query, info);
 683         lockForWrite();
 684         safeExec(query);
 685         if (!watchQuery(query)) {
 686             db.rollback();
 687             error = true;
 688         }
 689         else {
 690             networkId = query.lastInsertId().toInt();
 691         }
 692     }
 693     if (error) {
 694         unlock();
 695         return NetworkId();
 696     }
 697
 698     {
 699         QSqlQuery insertServersQuery(db);
 700         insertServersQuery.prepare(queryString("insert_server"));
 701         foreach(Network::Server server, info.serverList) {
 702             insertServersQuery.bindValue(":userid", user.toInt());
 703             insertServersQuery.bindValue(":networkid", networkId.toInt());
 704             bindServerInfo(insertServersQuery, server);
 705             safeExec(insertServersQuery);
 706             if (!watchQuery(insertServersQuery)) {
 707                 db.rollback();
 708                 error = true;
 709                 break;
 710             }
 711         }
 712         if (!error)
 713             db.commit();
 714     }
 715     unlock();
 716     if (error)
 717         return NetworkId();
 718     else
 719         return networkId;
 720 }
 721
 722
 723 void SqliteStorage::bindNetworkInfo(QSqlQuery &query, const NetworkInfo &info)
 724 {
 725     query.bindValue(":networkname", info.networkName);
 726     query.bindValue(":identityid", info.identity.toInt());
 727     query.bindValue(":encodingcodec", QString(info.codecForEncoding));
 728     query.bindValue(":decodingcodec", QString(info.codecForDecoding));
 729     query.bindValue(":servercodec", QString(info.codecForServer));
 730     query.bindValue(":userandomserver", info.useRandomServer ? 1 : 0);
 731     query.bindValue(":perform", info.perform.join("\n"));
 732     query.bindValue(":useautoidentify", info.useAutoIdentify ? 1 : 0);
 733     query.bindValue(":autoidentifyservice", info.autoIdentifyService);
 734     query.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
 735     query.bindValue(":usesasl", info.useSasl ? 1 : 0);
 736     query.bindValue(":saslaccount", info.saslAccount);
 737     query.bindValue(":saslpassword", info.saslPassword);
 738     query.bindValue(":useautoreconnect", info.useAutoReconnect ? 1 : 0);
 739     query.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
 740     query.bindValue(":autoreconnectretries", info.autoReconnectRetries);
 741     query.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries ? 1 : 0);
 742     query.bindValue(":rejoinchannels", info.rejoinChannels ? 1 : 0);
 743     // Custom rate limiting
 744     query.bindValue(":usecustomessagerate", info.useCustomMessageRate ? 1 : 0);
 745     query.bindValue(":messagerateburstsize", info.messageRateBurstSize);
 746     query.bindValue(":messageratedelay", info.messageRateDelay);
 747     query.bindValue(":unlimitedmessagerate", info.unlimitedMessageRate ? 1 : 0);
 748     if (info.networkId.isValid())
 749         query.bindValue(":networkid", info.networkId.toInt());
 750 }
 751
 752
 753 void SqliteStorage::bindServerInfo(QSqlQuery &query, const Network::Server &server)
 754 {
 755     query.bindValue(":hostname", server.host);
 756     query.bindValue(":port", server.port);
 757     query.bindValue(":password", server.password);
 758     query.bindValue(":ssl", server.useSsl ? 1 : 0);
 759     query.bindValue(":sslversion", server.sslVersion);
 760     query.bindValue(":useproxy", server.useProxy ? 1 : 0);
 761     query.bindValue(":proxytype", server.proxyType);
 762     query.bindValue(":proxyhost", server.proxyHost);
 763     query.bindValue(":proxyport", server.proxyPort);
 764     query.bindValue(":proxyuser", server.proxyUser);
 765     query.bindValue(":proxypass", server.proxyPass);
 766     query.bindValue(":sslverify", server.sslVerify ? 1 : 0);
 767 }
 768
 769
 770 bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info)
 771 {
 772     QSqlDatabase db = logDb();
 773     bool error = false;
 774     db.transaction();
 775
 776     {
 777         QSqlQuery updateQuery(db);
 778         updateQuery.prepare(queryString("update_network"));
 779         updateQuery.bindValue(":userid", user.toInt());
 780         bindNetworkInfo(updateQuery, info);
 781
 782         lockForWrite();
 783         safeExec(updateQuery);
 784         if (!watchQuery(updateQuery) || updateQuery.numRowsAffected() != 1) {
 785             error = true;
 786             db.rollback();
 787         }
 788     }
 789     if (error) {
 790         unlock();
 791         return false;
 792     }
 793
 794     {
 795         QSqlQuery dropServersQuery(db);
 796         dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
 797         dropServersQuery.bindValue(":networkid", info.networkId.toInt());
 798         safeExec(dropServersQuery);
 799         if (!watchQuery(dropServersQuery)) {
 800             error = true;
 801             db.rollback();
 802         }
 803     }
 804     if (error) {
 805         unlock();
 806         return false;
 807     }
 808
 809     {
 810         QSqlQuery insertServersQuery(db);
 811         insertServersQuery.prepare(queryString("insert_server"));
 812         foreach(Network::Server server, info.serverList) {
 813             insertServersQuery.bindValue(":userid", user.toInt());
 814             insertServersQuery.bindValue(":networkid", info.networkId.toInt());
 815             bindServerInfo(insertServersQuery, server);
 816             safeExec(insertServersQuery);
 817             if (!watchQuery(insertServersQuery)) {
 818                 error = true;
 819                 db.rollback();
 820                 break;
 821             }
 822         }
 823     }
 824
 825     db.commit();
 826     unlock();
 827     return !error;
 828 }
 829
 830
 831 bool SqliteStorage::removeNetwork(UserId user, const NetworkId &networkId)
 832 {
 833     QSqlDatabase db = logDb();
 834     bool error = false;
 835     db.transaction();
 836
 837     {
 838         QSqlQuery deleteNetworkQuery(db);
 839         deleteNetworkQuery.prepare(queryString("delete_network"));
 840         deleteNetworkQuery.bindValue(":networkid", networkId.toInt());
 841         deleteNetworkQuery.bindValue(":userid", user.toInt());
 842         lockForWrite();
 843         safeExec(deleteNetworkQuery);
 844         if (!watchQuery(deleteNetworkQuery) || deleteNetworkQuery.numRowsAffected() != 1) {
 845             error = true;
 846             db.rollback();
 847         }
 848     }
 849     if (error) {
 850         unlock();
 851         return false;
 852     }
 853
 854     {
 855         QSqlQuery deleteBacklogQuery(db);
 856         deleteBacklogQuery.prepare(queryString("delete_backlog_for_network"));
 857         deleteBacklogQuery.bindValue(":networkid", networkId.toInt());
 858         safeExec(deleteBacklogQuery);
 859         if (!watchQuery(deleteBacklogQuery)) {
 860             db.rollback();
 861             error = true;
 862         }
 863     }
 864     if (error) {
 865         unlock();
 866         return false;
 867     }
 868
 869     {
 870         QSqlQuery deleteBuffersQuery(db);
 871         deleteBuffersQuery.prepare(queryString("delete_buffers_for_network"));
 872         deleteBuffersQuery.bindValue(":networkid", networkId.toInt());
 873         safeExec(deleteBuffersQuery);
 874         if (!watchQuery(deleteBuffersQuery)) {
 875             db.rollback();
 876             error = true;
 877         }
 878     }
 879     if (error) {
 880         unlock();
 881         return false;
 882     }
 883
 884     {
 885         QSqlQuery deleteServersQuery(db);
 886         deleteServersQuery.prepare(queryString("delete_ircservers_for_network"));
 887         deleteServersQuery.bindValue(":networkid", networkId.toInt());
 888         safeExec(deleteServersQuery);
 889         if (!watchQuery(deleteServersQuery)) {
 890             db.rollback();
 891             error = true;
 892         }
 893     }
 894     if (error) {
 895         unlock();
 896         return false;
 897     }
 898
 899     db.commit();
 900     unlock();
 901     return true;
 902 }
 903
 904
 905 QList<NetworkInfo> SqliteStorage::networks(UserId user)
 906 {
 907     QList<NetworkInfo> nets;
 908
 909     QSqlDatabase db = logDb();
 910     db.transaction();
 911
 912     {
 913         QSqlQuery networksQuery(db);
 914         networksQuery.prepare(queryString("select_networks_for_user"));
 915         networksQuery.bindValue(":userid", user.toInt());
 916
 917         QSqlQuery serversQuery(db);
 918         serversQuery.prepare(queryString("select_servers_for_network"));
 919
 920         lockForRead();
 921         safeExec(networksQuery);
 922         if (watchQuery(networksQuery)) {
 923             while (networksQuery.next()) {
 924                 NetworkInfo net;
 925                 net.networkId = networksQuery.value(0).toInt();
 926                 net.networkName = networksQuery.value(1).toString();
 927                 net.identity = networksQuery.value(2).toInt();
 928                 net.codecForServer = networksQuery.value(3).toString().toLatin1();
 929                 net.codecForEncoding = networksQuery.value(4).toString().toLatin1();
 930                 net.codecForDecoding = networksQuery.value(5).toString().toLatin1();
 931                 net.useRandomServer = networksQuery.value(6).toInt() == 1 ? true : false;
 932                 net.perform = networksQuery.value(7).toString().split("\n");
 933                 net.useAutoIdentify = networksQuery.value(8).toInt() == 1 ? true : false;
 934                 net.autoIdentifyService = networksQuery.value(9).toString();
 935                 net.autoIdentifyPassword = networksQuery.value(10).toString();
 936                 net.useAutoReconnect = networksQuery.value(11).toInt() == 1 ? true : false;
 937                 net.autoReconnectInterval = networksQuery.value(12).toUInt();
 938                 net.autoReconnectRetries = networksQuery.value(13).toInt();
 939                 net.unlimitedReconnectRetries = networksQuery.value(14).toInt() == 1 ? true : false;
 940                 net.rejoinChannels = networksQuery.value(15).toInt() == 1 ? true : false;
 941                 net.useSasl = networksQuery.value(16).toInt() == 1 ? true : false;
 942                 net.saslAccount = networksQuery.value(17).toString();
 943                 net.saslPassword = networksQuery.value(18).toString();
 944                 // Custom rate limiting
 945                 net.useCustomMessageRate = networksQuery.value(19).toInt() == 1 ? true : false;
 946                 net.messageRateBurstSize = networksQuery.value(20).toUInt();
 947                 net.messageRateDelay = networksQuery.value(21).toUInt();
 948                 net.unlimitedMessageRate = networksQuery.value(22).toInt() == 1 ? true : false;
 949
 950                 serversQuery.bindValue(":networkid", net.networkId.toInt());
 951                 safeExec(serversQuery);
 952                 if (!watchQuery(serversQuery)) {
 953                     nets.clear();
 954                     break;
 955                 }
 956                 else {
 957                     Network::ServerList servers;
 958                     while (serversQuery.next()) {
 959                         Network::Server server;
 960                         server.host = serversQuery.value(0).toString();
 961                         server.port = serversQuery.value(1).toUInt();
 962                         server.password = serversQuery.value(2).toString();
 963                         server.useSsl = serversQuery.value(3).toInt() == 1 ? true : false;
 964                         server.sslVersion = serversQuery.value(4).toInt();
 965                         server.useProxy = serversQuery.value(5).toInt() == 1 ? true : false;
 966                         server.proxyType = serversQuery.value(6).toInt();
 967                         server.proxyHost = serversQuery.value(7).toString();
 968                         server.proxyPort = serversQuery.value(8).toUInt();
 969                         server.proxyUser = serversQuery.value(9).toString();
 970                         server.proxyPass = serversQuery.value(10).toString();
 971                         server.sslVerify = serversQuery.value(11).toInt() == 1 ? true : false;
 972                         servers << server;
 973                     }
 974                     net.serverList = servers;
 975                     nets << net;
 976                 }
 977             }
 978         }
 979     }
 980     db.commit();
 981     unlock();
 982     return nets;
 983 }
 984
 985
 986 QList<NetworkId> SqliteStorage::connectedNetworks(UserId user)
 987 {
 988     QList<NetworkId> connectedNets;
 989
 990     QSqlDatabase db = logDb();
 991     db.transaction();
 992
 993     {
 994         QSqlQuery query(db);
 995         query.prepare(queryString("select_connected_networks"));
 996         query.bindValue(":userid", user.toInt());
 997         lockForRead();
 998         safeExec(query);
 999         watchQuery(query);
1000
1001         while (query.next()) {
1002             connectedNets << query.value(0).toInt();
1003         }
1004         db.commit();
1005     }
1006     unlock();
1007     return connectedNets;
1008 }
1009
1010
1011 void SqliteStorage::setNetworkConnected(UserId user, const NetworkId &networkId, bool isConnected)
1012 {
1013     QSqlDatabase db = logDb();
1014     db.transaction();
1015
1016     {
1017         QSqlQuery query(db);
1018         query.prepare(queryString("update_network_connected"));
1019         query.bindValue(":userid", user.toInt());
1020         query.bindValue(":networkid", networkId.toInt());
1021         query.bindValue(":connected", isConnected ? 1 : 0);
1022
1023         lockForWrite();
1024         safeExec(query);
1025         watchQuery(query);
1026         db.commit();
1027     }
1028     unlock();
1029 }
1030
1031
1032 QHash<QString, QString> SqliteStorage::persistentChannels(UserId user, const NetworkId &networkId)
1033 {
1034     QHash<QString, QString> persistentChans;
1035
1036     QSqlDatabase db = logDb();
1037     db.transaction();
1038     {
1039         QSqlQuery query(db);
1040         query.prepare(queryString("select_persistent_channels"));
1041         query.bindValue(":userid", user.toInt());
1042         query.bindValue(":networkid", networkId.toInt());
1043
1044         lockForRead();
1045         safeExec(query);
1046         watchQuery(query);
1047         while (query.next()) {
1048             persistentChans[query.value(0).toString()] = query.value(1).toString();
1049         }
1050     }
1051     unlock();
1052     return persistentChans;
1053 }
1054
1055
1056 void SqliteStorage::setChannelPersistent(UserId user, const NetworkId &networkId, const QString &channel, bool isJoined)
1057 {
1058     QSqlDatabase db = logDb();
1059     db.transaction();
1060
1061     {
1062         QSqlQuery query(db);
1063         query.prepare(queryString("update_buffer_persistent_channel"));
1064         query.bindValue(":userid", user.toInt());
1065         query.bindValue(":networkid", networkId.toInt());
1066         query.bindValue(":buffercname", channel.toLower());
1067         query.bindValue(":joined", isJoined ? 1 : 0);
1068
1069         lockForWrite();
1070         safeExec(query);
1071         watchQuery(query);
1072         db.commit();
1073     }
1074     unlock();
1075 }
1076
1077
1078 void SqliteStorage::setPersistentChannelKey(UserId user, const NetworkId &networkId, const QString &channel, const QString &key)
1079 {
1080     QSqlDatabase db = logDb();
1081     db.transaction();
1082
1083     {
1084         QSqlQuery query(db);
1085         query.prepare(queryString("update_buffer_set_channel_key"));
1086         query.bindValue(":userid", user.toInt());
1087         query.bindValue(":networkid", networkId.toInt());
1088         query.bindValue(":buffercname", channel.toLower());
1089         query.bindValue(":key", key);
1090
1091         lockForWrite();
1092         safeExec(query);
1093         watchQuery(query);
1094         db.commit();
1095     }
1096     unlock();
1097 }
1098
1099
1100 QString SqliteStorage::awayMessage(UserId user, NetworkId networkId)
1101 {
1102     QSqlDatabase db = logDb();
1103     db.transaction();
1104
1105     QString awayMsg;
1106     {
1107         QSqlQuery query(db);
1108         query.prepare(queryString("select_network_awaymsg"));
1109         query.bindValue(":userid", user.toInt());
1110         query.bindValue(":networkid", networkId.toInt());
1111
1112         lockForRead();
1113         safeExec(query);
1114         watchQuery(query);
1115         if (query.first())
1116             awayMsg = query.value(0).toString();
1117         db.commit();
1118     }
1119     unlock();
1120
1121     return awayMsg;
1122 }
1123
1124
1125 void SqliteStorage::setAwayMessage(UserId user, NetworkId networkId, const QString &awayMsg)
1126 {
1127     QSqlDatabase db = logDb();
1128     db.transaction();
1129
1130     {
1131         QSqlQuery query(db);
1132         query.prepare(queryString("update_network_set_awaymsg"));
1133         query.bindValue(":userid", user.toInt());
1134         query.bindValue(":networkid", networkId.toInt());
1135         query.bindValue(":awaymsg", awayMsg);
1136
1137         lockForWrite();
1138         safeExec(query);
1139         watchQuery(query);
1140         db.commit();
1141     }
1142     unlock();
1143 }
1144
1145
1146 QString SqliteStorage::userModes(UserId user, NetworkId networkId)
1147 {
1148     QSqlDatabase db = logDb();
1149     db.transaction();
1150
1151     QString modes;
1152     {
1153         QSqlQuery query(db);
1154         query.prepare(queryString("select_network_usermode"));
1155         query.bindValue(":userid", user.toInt());
1156         query.bindValue(":networkid", networkId.toInt());
1157
1158         lockForRead();
1159         safeExec(query);
1160         watchQuery(query);
1161         if (query.first())
1162             modes = query.value(0).toString();
1163         db.commit();
1164     }
1165     unlock();
1166
1167     return modes;
1168 }
1169
1170
1171 void SqliteStorage::setUserModes(UserId user, NetworkId networkId, const QString &userModes)
1172 {
1173     QSqlDatabase db = logDb();
1174     db.transaction();
1175
1176     {
1177         QSqlQuery query(db);
1178         query.prepare(queryString("update_network_set_usermode"));
1179         query.bindValue(":userid", user.toInt());
1180         query.bindValue(":networkid", networkId.toInt());
1181         query.bindValue(":usermode", userModes);
1182
1183         lockForWrite();
1184         safeExec(query);
1185         watchQuery(query);
1186         db.commit();
1187     }
1188     unlock();
1189 }
1190
1191
1192 BufferInfo SqliteStorage::bufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer, bool create)
1193 {
1194     QSqlDatabase db = logDb();
1195     db.transaction();
1196
1197     BufferInfo bufferInfo;
1198     {
1199         QSqlQuery query(db);
1200         query.prepare(queryString("select_bufferByName"));
1201         query.bindValue(":networkid", networkId.toInt());
1202         query.bindValue(":userid", user.toInt());
1203         query.bindValue(":buffercname", buffer.toLower());
1204
1205         lockForRead();
1206         safeExec(query);
1207
1208         if (query.first()) {
1209             bufferInfo = BufferInfo(query.value(0).toInt(), networkId, (BufferInfo::Type)query.value(1).toInt(), 0, buffer);
1210             if (query.next()) {
1211                 qCritical() << "SqliteStorage::getBufferInfo(): received more then one Buffer!";
1212                 qCritical() << "         Query:" << query.lastQuery();
1213                 qCritical() << "  bound Values:";
1214                 QList<QVariant> list = query.boundValues().values();
1215                 for (int i = 0; i < list.size(); ++i)
1216                     qCritical() << i << ":" << list.at(i).toString().toLatin1().data();
1217                 Q_ASSERT(false);
1218             }
1219         }
1220         else if (create) {
1221             // let's create the buffer
1222             QSqlQuery createQuery(db);
1223             createQuery.prepare(queryString("insert_buffer"));
1224             createQuery.bindValue(":userid", user.toInt());
1225             createQuery.bindValue(":networkid", networkId.toInt());
1226             createQuery.bindValue(":buffertype", (int)type);
1227             createQuery.bindValue(":buffername", buffer);
1228             createQuery.bindValue(":buffercname", buffer.toLower());
1229             createQuery.bindValue(":joined", type & BufferInfo::ChannelBuffer ? 1 : 0);
1230
1231             unlock();
1232             lockForWrite();
1233             safeExec(createQuery);
1234             watchQuery(createQuery);
1235             bufferInfo = BufferInfo(createQuery.lastInsertId().toInt(), networkId, type, 0, buffer);
1236         }
1237     }
1238     db.commit();
1239     unlock();
1240     return bufferInfo;
1241 }
1242
1243
1244 BufferInfo SqliteStorage::getBufferInfo(UserId user, const BufferId &bufferId)
1245 {
1246     QSqlDatabase db = logDb();
1247     db.transaction();
1248
1249     BufferInfo bufferInfo;
1250     {
1251         QSqlQuery query(db);
1252         query.prepare(queryString("select_buffer_by_id"));
1253         query.bindValue(":userid", user.toInt());
1254         query.bindValue(":bufferid", bufferId.toInt());
1255
1256         lockForRead();
1257         safeExec(query);
1258
1259         if (watchQuery(query) && query.first()) {
1260             bufferInfo = BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), 0, query.value(4).toString());
1261             Q_ASSERT(!query.next());
1262         }
1263         db.commit();
1264     }
1265     unlock();
1266     return bufferInfo;
1267 }
1268
1269
1270 QList<BufferInfo> SqliteStorage::requestBuffers(UserId user)
1271 {
1272     QList<BufferInfo> bufferlist;
1273
1274     QSqlDatabase db = logDb();
1275     db.transaction();
1276
1277     {
1278         QSqlQuery query(db);
1279         query.prepare(queryString("select_buffers"));
1280         query.bindValue(":userid", user.toInt());
1281
1282         lockForRead();
1283         safeExec(query);
1284         watchQuery(query);
1285         while (query.next()) {
1286             bufferlist << BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), query.value(3).toInt(), query.value(4).toString());
1287         }
1288         db.commit();
1289     }
1290     unlock();
1291
1292     return bufferlist;
1293 }
1294
1295
1296 QList<BufferId> SqliteStorage::requestBufferIdsForNetwork(UserId user, NetworkId networkId)
1297 {
1298     QList<BufferId> bufferList;
1299
1300     QSqlDatabase db = logDb();
1301     db.transaction();
1302
1303     {
1304         QSqlQuery query(db);
1305         query.prepare(queryString("select_buffers_for_network"));
1306         query.bindValue(":networkid", networkId.toInt());
1307         query.bindValue(":userid", user.toInt());
1308
1309         lockForRead();
1310         safeExec(query);
1311         watchQuery(query);
1312         while (query.next()) {
1313             bufferList << BufferId(query.value(0).toInt());
1314         }
1315         db.commit();
1316     }
1317     unlock();
1318
1319     return bufferList;
1320 }
1321
1322
1323 bool SqliteStorage::removeBuffer(const UserId &user, const BufferId &bufferId)
1324 {
1325     QSqlDatabase db = logDb();
1326     db.transaction();
1327
1328     bool error = false;
1329     {
1330         QSqlQuery delBufferQuery(db);
1331         delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
1332         delBufferQuery.bindValue(":bufferid", bufferId.toInt());
1333         delBufferQuery.bindValue(":userid", user.toInt());
1334
1335         lockForWrite();
1336         safeExec(delBufferQuery);
1337
1338         error = (!watchQuery(delBufferQuery) || delBufferQuery.numRowsAffected() != 1);
1339     }
1340
1341     if (error) {
1342         db.rollback();
1343         unlock();
1344         return false;
1345     }
1346
1347     {
1348         QSqlQuery delBacklogQuery(db);
1349         delBacklogQuery.prepare(queryString("delete_backlog_for_buffer"));
1350         delBacklogQuery.bindValue(":bufferid", bufferId.toInt());
1351
1352         safeExec(delBacklogQuery);
1353         error = !watchQuery(delBacklogQuery);
1354     }
1355
1356     if (error) {
1357         db.rollback();
1358     }
1359     else {
1360         db.commit();
1361     }
1362     unlock();
1363     return !error;
1364 }
1365
1366
1367 bool SqliteStorage::renameBuffer(const UserId &user, const BufferId &bufferId, const QString &newName)
1368 {
1369     QSqlDatabase db = logDb();
1370     db.transaction();
1371
1372     bool error = false;
1373     {
1374         QSqlQuery query(db);
1375         query.prepare(queryString("update_buffer_name"));
1376         query.bindValue(":buffername", newName);
1377         query.bindValue(":buffercname", newName.toLower());
1378         query.bindValue(":bufferid", bufferId.toInt());
1379         query.bindValue(":userid", user.toInt());
1380
1381         lockForWrite();
1382         safeExec(query);
1383
1384         error = query.lastError().isValid();
1385         // unexepcted error occured (19 == constraint violation)
1386         if (error && query.lastError().number() != 19) {
1387             watchQuery(query);
1388         }
1389         else {
1390             error |= (query.numRowsAffected() != 1);
1391         }
1392     }
1393     if (error) {
1394         db.rollback();
1395     }
1396     else {
1397         db.commit();
1398     }
1399     unlock();
1400     return !error;
1401 }
1402
1403
1404 bool SqliteStorage::mergeBuffersPermanently(const UserId &user, const BufferId &bufferId1, const BufferId &bufferId2)
1405 {
1406     QSqlDatabase db = logDb();
1407     db.transaction();
1408
1409     bool error = false;
1410     {
1411         QSqlQuery checkQuery(db);
1412         checkQuery.prepare(queryString("select_buffers_for_merge"));
1413         checkQuery.bindValue(":oldbufferid", bufferId2.toInt());
1414         checkQuery.bindValue(":newbufferid", bufferId1.toInt());
1415         checkQuery.bindValue(":userid", user.toInt());
1416
1417         lockForRead();
1418         safeExec(checkQuery);
1419         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 2);
1420     }
1421     if (error) {
1422         db.rollback();
1423         unlock();
1424         return false;
1425     }
1426
1427     {
1428         QSqlQuery query(db);
1429         query.prepare(queryString("update_backlog_bufferid"));
1430         query.bindValue(":oldbufferid", bufferId2.toInt());
1431         query.bindValue(":newbufferid", bufferId1.toInt());
1432         safeExec(query);
1433         error = !watchQuery(query);
1434     }
1435     if (error) {
1436         db.rollback();
1437         unlock();
1438         return false;
1439     }
1440
1441     {
1442         QSqlQuery delBufferQuery(db);
1443         delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
1444         delBufferQuery.bindValue(":bufferid", bufferId2.toInt());
1445         delBufferQuery.bindValue(":userid", user.toInt());
1446         safeExec(delBufferQuery);
1447         error = !watchQuery(delBufferQuery);
1448     }
1449
1450     if (error) {
1451         db.rollback();
1452     }
1453     else {
1454         db.commit();
1455     }
1456     unlock();
1457     return !error;
1458 }
1459
1460
1461 void SqliteStorage::setBufferLastSeenMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1462 {
1463     QSqlDatabase db = logDb();
1464     db.transaction();
1465
1466     {
1467         QSqlQuery query(db);
1468         query.prepare(queryString("update_buffer_lastseen"));
1469         query.bindValue(":userid", user.toInt());
1470         query.bindValue(":bufferid", bufferId.toInt());
1471         query.bindValue(":lastseenmsgid", msgId.toQint64());
1472
1473         lockForWrite();
1474         safeExec(query);
1475         watchQuery(query);
1476     }
1477     db.commit();
1478     unlock();
1479 }
1480
1481
1482 QHash<BufferId, MsgId> SqliteStorage::bufferLastSeenMsgIds(UserId user)
1483 {
1484     QHash<BufferId, MsgId> lastSeenHash;
1485
1486     QSqlDatabase db = logDb();
1487     db.transaction();
1488
1489     bool error = false;
1490     {
1491         QSqlQuery query(db);
1492         query.prepare(queryString("select_buffer_lastseen_messages"));
1493         query.bindValue(":userid", user.toInt());
1494
1495         lockForRead();
1496         safeExec(query);
1497         error = !watchQuery(query);
1498         if (!error) {
1499             while (query.next()) {
1500                 lastSeenHash[query.value(0).toInt()] = query.value(1).toLongLong();
1501             }
1502         }
1503     }
1504
1505     db.commit();
1506     unlock();
1507     return lastSeenHash;
1508 }
1509
1510
1511 void SqliteStorage::setBufferMarkerLineMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1512 {
1513     QSqlDatabase db = logDb();
1514     db.transaction();
1515
1516     {
1517         QSqlQuery query(db);
1518         query.prepare(queryString("update_buffer_markerlinemsgid"));
1519         query.bindValue(":userid", user.toInt());
1520         query.bindValue(":bufferid", bufferId.toInt());
1521         query.bindValue(":markerlinemsgid", msgId.toQint64());
1522
1523         lockForWrite();
1524         safeExec(query);
1525         watchQuery(query);
1526     }
1527     db.commit();
1528     unlock();
1529 }
1530
1531
1532 QHash<BufferId, MsgId> SqliteStorage::bufferMarkerLineMsgIds(UserId user)
1533 {
1534     QHash<BufferId, MsgId> markerLineHash;
1535
1536     QSqlDatabase db = logDb();
1537     db.transaction();
1538
1539     bool error = false;
1540     {
1541         QSqlQuery query(db);
1542         query.prepare(queryString("select_buffer_markerlinemsgids"));
1543         query.bindValue(":userid", user.toInt());
1544
1545         lockForRead();
1546         safeExec(query);
1547         error = !watchQuery(query);
1548         if (!error) {
1549             while (query.next()) {
1550                 markerLineHash[query.value(0).toInt()] = query.value(1).toLongLong();
1551             }
1552         }
1553     }
1554
1555     db.commit();
1556     unlock();
1557     return markerLineHash;
1558 }
1559
1560 void SqliteStorage::setBufferActivity(UserId user, BufferId bufferId, Message::Types bufferActivity)
1561 {
1562     QSqlDatabase db = logDb();
1563     db.transaction();
1564
1565     {
1566         QSqlQuery query(db);
1567         query.prepare(queryString("update_buffer_bufferactivity"));
1568         query.bindValue(":userid", user.toInt());
1569         query.bindValue(":bufferid", bufferId.toInt());
1570         query.bindValue(":bufferactivity", (int) bufferActivity);
1571
1572         lockForWrite();
1573         safeExec(query);
1574         watchQuery(query);
1575     }
1576     db.commit();
1577     unlock();
1578 }
1579
1580
1581 QHash<BufferId, Message::Types> SqliteStorage::bufferActivities(UserId user)
1582 {
1583     QHash<BufferId, Message::Types> bufferActivityHash;
1584
1585     QSqlDatabase db = logDb();
1586     db.transaction();
1587
1588     bool error = false;
1589     {
1590         QSqlQuery query(db);
1591         query.prepare(queryString("select_buffer_bufferactivities"));
1592         query.bindValue(":userid", user.toInt());
1593
1594         lockForRead();
1595         safeExec(query);
1596         error = !watchQuery(query);
1597         if (!error) {
1598             while (query.next()) {
1599                 bufferActivityHash[query.value(0).toInt()] = Message::Types(query.value(1).toInt());
1600             }
1601         }
1602     }
1603
1604     db.commit();
1605     unlock();
1606     return bufferActivityHash;
1607 }
1608
1609
1610 Message::Types SqliteStorage::bufferActivity(BufferId bufferId, MsgId lastSeenMsgId)
1611 {
1612     QSqlDatabase db = logDb();
1613     db.transaction();
1614
1615     Message::Types result = Message::Types(0);
1616     {
1617         QSqlQuery query(db);
1618         query.prepare(queryString("select_buffer_bufferactivity"));
1619         query.bindValue(":bufferid", bufferId.toInt());
1620         query.bindValue(":lastseenmsgid", lastSeenMsgId.toQint64());
1621
1622         lockForRead();
1623         safeExec(query);
1624         if (query.first())
1625             result = Message::Types(query.value(0).toInt());
1626     }
1627
1628     db.commit();
1629     unlock();
1630     return result;
1631 }
1632
1633 QHash<QString, QByteArray> SqliteStorage::bufferCiphers(UserId user, const NetworkId &networkId)
1634 {
1635     QHash<QString, QByteArray> bufferCiphers;
1636
1637     QSqlDatabase db = logDb();
1638     db.transaction();
1639     {
1640         QSqlQuery query(db);
1641         query.prepare(queryString("select_buffer_ciphers"));
1642         query.bindValue(":userid", user.toInt());
1643         query.bindValue(":networkid", networkId.toInt());
1644
1645         lockForRead();
1646         safeExec(query);
1647         watchQuery(query);
1648         while (query.next()) {
1649             bufferCiphers[query.value(0).toString()] = QByteArray::fromHex(query.value(1).toString().toUtf8());
1650         }
1651     }
1652     unlock();
1653     return bufferCiphers;
1654 }
1655
1656 void SqliteStorage::setBufferCipher(UserId user, const NetworkId &networkId, const QString &bufferName, const QByteArray &cipher)
1657 {
1658     QSqlDatabase db = logDb();
1659     db.transaction();
1660
1661     {
1662         QSqlQuery query(db);
1663         query.prepare(queryString("update_buffer_cipher"));
1664         query.bindValue(":userid", user.toInt());
1665         query.bindValue(":networkid", networkId.toInt());
1666         query.bindValue(":buffercname", bufferName.toLower());
1667         query.bindValue(":cipher", QString(cipher.toHex()));
1668
1669         lockForWrite();
1670         safeExec(query);
1671         watchQuery(query);
1672         db.commit();
1673     }
1674     unlock();
1675 }
1676
1677 void SqliteStorage::setHighlightCount(UserId user, BufferId bufferId, int count)
1678 {
1679     QSqlDatabase db = logDb();
1680     db.transaction();
1681
1682     {
1683         QSqlQuery query(db);
1684         query.prepare(queryString("update_buffer_highlightcount"));
1685         query.bindValue(":userid", user.toInt());
1686         query.bindValue(":bufferid", bufferId.toInt());
1687         query.bindValue(":highlightcount", count);
1688
1689         lockForWrite();
1690         safeExec(query);
1691         watchQuery(query);
1692     }
1693     db.commit();
1694     unlock();
1695 }
1696
1697
1698 QHash<BufferId, int> SqliteStorage::highlightCounts(UserId user)
1699 {
1700     QHash<BufferId, int> highlightCountHash;
1701
1702     QSqlDatabase db = logDb();
1703     db.transaction();
1704
1705     bool error = false;
1706     {
1707         QSqlQuery query(db);
1708         query.prepare(queryString("select_buffer_highlightcounts"));
1709         query.bindValue(":userid", user.toInt());
1710
1711         lockForRead();
1712         safeExec(query);
1713         error = !watchQuery(query);
1714         if (!error) {
1715             while (query.next()) {
1716                 highlightCountHash[query.value(0).toInt()] = query.value(1).toInt();
1717             }
1718         }
1719     }
1720
1721     db.commit();
1722     unlock();
1723     return highlightCountHash;
1724 }
1725
1726
1727 int SqliteStorage::highlightCount(BufferId bufferId, MsgId lastSeenMsgId)
1728 {
1729     QSqlDatabase db = logDb();
1730     db.transaction();
1731
1732     int result = 0;
1733     {
1734         QSqlQuery query(db);
1735         query.prepare(queryString("select_buffer_highlightcount"));
1736         query.bindValue(":bufferid", bufferId.toInt());
1737         query.bindValue(":lastseenmsgid", lastSeenMsgId.toQint64());
1738
1739         lockForRead();
1740         safeExec(query);
1741         if (query.first())
1742             result = query.value(0).toInt();
1743     }
1744
1745     db.commit();
1746     unlock();
1747     return result;
1748 }
1749
1750 bool SqliteStorage::logMessage(Message &msg)
1751 {
1752     QSqlDatabase db = logDb();
1753     db.transaction();
1754
1755     bool error = false;
1756     {
1757         QSqlQuery logMessageQuery(db);
1758         logMessageQuery.prepare(queryString("insert_message"));
1759         // Store timestamp in seconds as 64-bit integer
1760         //
1761         // NOTE:  This is a loss of precision.  The database time column would need to store a
1762         // fractional number to support toMSecsSinceEpoch(), or an upgrade step would need to
1763         // convert all past times to milliseconds, multiplying by 1000.
1764 #if QT_VERSION >= 0x050800
1765         logMessageQuery.bindValue(":time", msg.timestamp().toSecsSinceEpoch());
1766 #else
1767         // toSecsSinceEpoch() was added in Qt 5.8.  Manually downconvert to seconds for now.
1768         // See https://doc.qt.io/qt-5/qdatetime.html#toMSecsSinceEpoch
1769         logMessageQuery.bindValue(":time", (qint64)(msg.timestamp().toMSecsSinceEpoch() / 1000));
1770 #endif
1771         logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1772         logMessageQuery.bindValue(":type", msg.type());
1773         logMessageQuery.bindValue(":flags", (int)msg.flags());
1774         logMessageQuery.bindValue(":sender", msg.sender());
1775         logMessageQuery.bindValue(":realname", msg.realName());
1776         logMessageQuery.bindValue(":avatarurl", msg.avatarUrl());
1777         logMessageQuery.bindValue(":senderprefixes", msg.senderPrefixes());
1778         logMessageQuery.bindValue(":message", msg.contents());
1779
1780         lockForWrite();
1781         safeExec(logMessageQuery);
1782
1783         if (logMessageQuery.lastError().isValid()) {
1784             // constraint violation - must be NOT NULL constraint - probably the sender is missing...
1785             if (logMessageQuery.lastError().number() == 19) {
1786                 QSqlQuery addSenderQuery(db);
1787                 addSenderQuery.prepare(queryString("insert_sender"));
1788                 addSenderQuery.bindValue(":sender", msg.sender());
1789                 addSenderQuery.bindValue(":realname", msg.realName());
1790                 addSenderQuery.bindValue(":avatarurl", msg.avatarUrl());
1791                 safeExec(addSenderQuery);
1792                 safeExec(logMessageQuery);
1793                 error = !watchQuery(logMessageQuery);
1794             }
1795             else {
1796                 watchQuery(logMessageQuery);
1797             }
1798         }
1799         if (!error) {
1800             MsgId msgId = logMessageQuery.lastInsertId().toLongLong();
1801             if (msgId.isValid()) {
1802                 msg.setMsgId(msgId);
1803             }
1804             else {
1805                 error = true;
1806             }
1807         }
1808     }
1809
1810     if (error) {
1811         db.rollback();
1812     }
1813     else {
1814         db.commit();
1815     }
1816
1817     unlock();
1818     return !error;
1819 }
1820
1821
1822 bool SqliteStorage::logMessages(MessageList &msgs)
1823 {
1824     QSqlDatabase db = logDb();
1825     db.transaction();
1826
1827     {
1828         QSet<SenderData> senders;
1829         QSqlQuery addSenderQuery(db);
1830         addSenderQuery.prepare(queryString("insert_sender"));
1831         lockForWrite();
1832         for (int i = 0; i < msgs.count(); i++) {
1833             auto &msg = msgs.at(i);
1834             SenderData sender = { msg.sender(), msg.realName(), msg.avatarUrl() };
1835             if (senders.contains(sender))
1836                 continue;
1837             senders << sender;
1838
1839             addSenderQuery.bindValue(":sender", sender.sender);
1840             addSenderQuery.bindValue(":realname", sender.realname);
1841             addSenderQuery.bindValue(":avatarurl", sender.avatarurl);
1842             safeExec(addSenderQuery);
1843         }
1844     }
1845
1846     bool error = false;
1847     {
1848         QSqlQuery logMessageQuery(db);
1849         logMessageQuery.prepare(queryString("insert_message"));
1850         for (int i = 0; i < msgs.count(); i++) {
1851             Message &msg = msgs[i];
1852             // Store timestamp in seconds as 64-bit integer
1853             //
1854             // NOTE:  This is a loss of precision.  The database time column would need to store a
1855             // fractional number to support toMSecsSinceEpoch(), or an upgrade step would need to
1856             // convert all past times to milliseconds, multiplying by 1000.
1857 #if QT_VERSION >= 0x050800
1858             logMessageQuery.bindValue(":time", msg.timestamp().toSecsSinceEpoch());
1859 #else
1860             // toSecsSinceEpoch() was added in Qt 5.8.  Manually downconvert to seconds for now.
1861             // See https://doc.qt.io/qt-5/qdatetime.html#toMSecsSinceEpoch
1862             logMessageQuery.bindValue(":time",
1863                                       (qint64)(msg.timestamp().toMSecsSinceEpoch() / 1000));
1864 #endif
1865             logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1866             logMessageQuery.bindValue(":type", msg.type());
1867             logMessageQuery.bindValue(":flags", (int)msg.flags());
1868             logMessageQuery.bindValue(":sender", msg.sender());
1869             logMessageQuery.bindValue(":realname", msg.realName());
1870             logMessageQuery.bindValue(":avatarurl", msg.avatarUrl());
1871             logMessageQuery.bindValue(":senderprefixes", msg.senderPrefixes());
1872             logMessageQuery.bindValue(":message", msg.contents());
1873
1874             safeExec(logMessageQuery);
1875             if (!watchQuery(logMessageQuery)) {
1876                 error = true;
1877                 break;
1878             }
1879             else {
1880                 msg.setMsgId(logMessageQuery.lastInsertId().toLongLong());
1881             }
1882         }
1883     }
1884
1885     if (error) {
1886         db.rollback();
1887         unlock();
1888         // we had a rollback in the db so we need to reset all msgIds
1889         for (int i = 0; i < msgs.count(); i++) {
1890             msgs[i].setMsgId(MsgId());
1891         }
1892     }
1893     else {
1894         db.commit();
1895         unlock();
1896     }
1897     return !error;
1898 }
1899
1900
1901 QList<Message> SqliteStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit)
1902 {
1903     QList<Message> messagelist;
1904
1905     QSqlDatabase db = logDb();
1906     db.transaction();
1907
1908     bool error = false;
1909     BufferInfo bufferInfo;
1910     {
1911         // code duplication from getBufferInfo:
1912         // this is due to the impossibility of nesting transactions and recursive locking
1913         QSqlQuery bufferInfoQuery(db);
1914         bufferInfoQuery.prepare(queryString("select_buffer_by_id"));
1915         bufferInfoQuery.bindValue(":userid", user.toInt());
1916         bufferInfoQuery.bindValue(":bufferid", bufferId.toInt());
1917
1918         lockForRead();
1919         safeExec(bufferInfoQuery);
1920         error = !watchQuery(bufferInfoQuery) || !bufferInfoQuery.first();
1921         if (!error) {
1922             bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), 0, bufferInfoQuery.value(4).toString());
1923             error = !bufferInfo.isValid();
1924         }
1925     }
1926     if (error) {
1927         db.rollback();
1928         unlock();
1929         return messagelist;
1930     }
1931
1932     {
1933         QSqlQuery query(db);
1934         if (last == -1 && first == -1) {
1935             query.prepare(queryString("select_messagesNewestK"));
1936         }
1937         else if (last == -1) {
1938             query.prepare(queryString("select_messagesNewerThan"));
1939             query.bindValue(":firstmsg", first.toQint64());
1940         }
1941         else {
1942             query.prepare(queryString("select_messagesRange"));
1943             query.bindValue(":lastmsg", last.toQint64());
1944             query.bindValue(":firstmsg", first.toQint64());
1945         }
1946         query.bindValue(":bufferid", bufferId.toInt());
1947         query.bindValue(":limit", limit);
1948
1949         safeExec(query);
1950         watchQuery(query);
1951
1952         while (query.next()) {
1953             Message msg(
1954                 // Read timestamp in seconds as 64-bit integer
1955                 //
1956                 // NOTE:  This is a loss of precision.  The database time column would need to store
1957                 // a fractional number to support fromMSecsSinceEpoch(), or an upgrade step would
1958                 // need to convert all past times to milliseconds, multiplying by 1000.
1959 #if QT_VERSION >= 0x050800
1960                 QDateTime::fromSecsSinceEpoch(query.value(1).toLongLong()),
1961 #else
1962                 // fromSecsSinceEpoch() was added in Qt 5.8.  Manually downconvert to seconds for
1963                 // now.
1964                 // See https://doc.qt.io/qt-5/qdatetime.html#fromMSecsSinceEpoch
1965                 QDateTime::fromMSecsSinceEpoch((qint64)(query.value(1).toLongLong() * 1000)),
1966 #endif
1967                 bufferInfo,
1968                 (Message::Type)query.value(2).toInt(),
1969                 query.value(8).toString(),
1970                 query.value(4).toString(),
1971                 query.value(5).toString(),
1972                 query.value(6).toString(),
1973                 query.value(7).toString(),
1974                 (Message::Flags)query.value(3).toInt());
1975             msg.setMsgId(query.value(0).toLongLong());
1976             messagelist << msg;
1977         }
1978     }
1979     db.commit();
1980     unlock();
1981
1982     return messagelist;
1983 }
1984
1985
1986 QList<Message> SqliteStorage::requestMsgsFiltered(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit, Message::Types type, Message::Flags flags)
1987 {
1988     QList<Message> messagelist;
1989
1990     QSqlDatabase db = logDb();
1991     db.transaction();
1992
1993     bool error = false;
1994     BufferInfo bufferInfo;
1995     {
1996         // code dupication from getBufferInfo:
1997         // this is due to the impossibility of nesting transactions and recursive locking
1998         QSqlQuery bufferInfoQuery(db);
1999         bufferInfoQuery.prepare(queryString("select_buffer_by_id"));
2000         bufferInfoQuery.bindValue(":userid", user.toInt());
2001         bufferInfoQuery.bindValue(":bufferid", bufferId.toInt());
2002
2003         lockForRead();
2004         safeExec(bufferInfoQuery);
2005         error = !watchQuery(bufferInfoQuery) || !bufferInfoQuery.first();
2006         if (!error) {
2007             bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), 0, bufferInfoQuery.value(4).toString());
2008             error = !bufferInfo.isValid();
2009         }
2010     }
2011     if (error) {
2012         db.rollback();
2013         unlock();
2014         return messagelist;
2015     }
2016
2017     {
2018         QSqlQuery query(db);
2019         if (last == -1 && first == -1) {
2020             query.prepare(queryString("select_messagesNewestK_filtered"));
2021         }
2022         else if (last == -1) {
2023             query.prepare(queryString("select_messagesNewerThan_filtered"));
2024             query.bindValue(":firstmsg", first.toQint64());
2025         }
2026         else {
2027             query.prepare(queryString("select_messagesRange_filtered"));
2028             query.bindValue(":lastmsg", last.toQint64());
2029             query.bindValue(":firstmsg", first.toQint64());
2030         }
2031         query.bindValue(":bufferid", bufferId.toInt());
2032         query.bindValue(":limit", limit);
2033         int typeRaw = type;
2034         query.bindValue(":type", typeRaw);
2035         int flagsRaw = flags;
2036         query.bindValue(":flags", flagsRaw);
2037
2038         safeExec(query);
2039         watchQuery(query);
2040
2041         while (query.next()) {
2042             Message msg(
2043                         // Read timestamp in seconds as 64-bit integer
2044                         //
2045                         // NOTE:  This is a loss of precision.  The database time column would need
2046                         // to store a fractional number to support fromMSecsSinceEpoch(), or an
2047                         // upgrade step would need to convert all past times to milliseconds,
2048                         // multiplying by 1000.
2049 #if QT_VERSION >= 0x050800
2050                         QDateTime::fromSecsSinceEpoch(query.value(1).toLongLong()),
2051 #else
2052                         // fromSecsSinceEpoch() was added in Qt 5.8.  Manually downconvert to
2053                         // seconds for now.
2054                         // See https://doc.qt.io/qt-5/qdatetime.html#fromMSecsSinceEpoch
2055                         QDateTime::fromMSecsSinceEpoch(
2056                             (qint64)(query.value(1).toLongLong() * 1000)),
2057 #endif
2058                         bufferInfo,
2059                         (Message::Type)query.value(2).toInt(),
2060                         query.value(8).toString(),
2061                         query.value(4).toString(),
2062                         query.value(5).toString(),
2063                         query.value(6).toString(),
2064                         query.value(7).toString(),
2065                         Message::Flags{query.value(3).toInt()});
2066             msg.setMsgId(query.value(0).toLongLong());
2067             messagelist << msg;
2068         }
2069     }
2070     db.commit();
2071     unlock();
2072
2073     return messagelist;
2074 }
2075
2076
2077 QList<Message> SqliteStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit)
2078 {
2079     QList<Message> messagelist;
2080
2081     QSqlDatabase db = logDb();
2082     db.transaction();
2083
2084     QHash<BufferId, BufferInfo> bufferInfoHash;
2085     {
2086         QSqlQuery bufferInfoQuery(db);
2087         bufferInfoQuery.prepare(queryString("select_buffers"));
2088         bufferInfoQuery.bindValue(":userid", user.toInt());
2089
2090         lockForRead();
2091         safeExec(bufferInfoQuery);
2092         watchQuery(bufferInfoQuery);
2093         while (bufferInfoQuery.next()) {
2094             BufferInfo bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), bufferInfoQuery.value(3).toInt(), bufferInfoQuery.value(4).toString());
2095             bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
2096         }
2097
2098         QSqlQuery query(db);
2099         if (last == -1) {
2100             query.prepare(queryString("select_messagesAllNew"));
2101         }
2102         else {
2103             query.prepare(queryString("select_messagesAll"));
2104             query.bindValue(":lastmsg", last.toQint64());
2105         }
2106         query.bindValue(":userid", user.toInt());
2107         query.bindValue(":firstmsg", first.toQint64());
2108         query.bindValue(":limit", limit);
2109         safeExec(query);
2110
2111         watchQuery(query);
2112
2113         while (query.next()) {
2114             Message msg(
2115                 // Read timestamp in seconds as 64-bit integer
2116                 //
2117                 // NOTE:  This is a loss of precision.  The database time column would need to store
2118                 // a fractional number to support fromMSecsSinceEpoch(), or an upgrade step would
2119                 // need to convert all past times to milliseconds, multiplying by 1000.
2120 #if QT_VERSION >= 0x050800
2121                 QDateTime::fromSecsSinceEpoch(query.value(2).toLongLong()),
2122 #else
2123                 // fromSecsSinceEpoch() was added in Qt 5.8.  Manually downconvert to seconds for
2124                 // now.
2125                 // See https://doc.qt.io/qt-5/qdatetime.html#fromMSecsSinceEpoch
2126                 QDateTime::fromMSecsSinceEpoch((qint64)(query.value(2).toLongLong() * 1000)),
2127 #endif
2128                 bufferInfoHash[query.value(1).toInt()],
2129                 (Message::Type)query.value(3).toInt(),
2130                 query.value(9).toString(),
2131                 query.value(5).toString(),
2132                 query.value(6).toString(),
2133                 query.value(7).toString(),
2134                 query.value(8).toString(),
2135                 (Message::Flags)query.value(4).toInt());
2136             msg.setMsgId(query.value(0).toLongLong());
2137             messagelist << msg;
2138         }
2139     }
2140     db.commit();
2141     unlock();
2142     return messagelist;
2143 }
2144
2145 QList<Message> SqliteStorage::requestAllMsgsFiltered(UserId user, MsgId first, MsgId last, int limit, Message::Types type, Message::Flags flags)
2146 {
2147     QList<Message> messagelist;
2148
2149     QSqlDatabase db = logDb();
2150     db.transaction();
2151
2152     QHash<BufferId, BufferInfo> bufferInfoHash;
2153     {
2154         QSqlQuery bufferInfoQuery(db);
2155         bufferInfoQuery.prepare(queryString("select_buffers"));
2156         bufferInfoQuery.bindValue(":userid", user.toInt());
2157
2158         lockForRead();
2159         safeExec(bufferInfoQuery);
2160         watchQuery(bufferInfoQuery);
2161         while (bufferInfoQuery.next()) {
2162             BufferInfo bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), bufferInfoQuery.value(3).toInt(), bufferInfoQuery.value(4).toString());
2163             bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
2164         }
2165
2166         QSqlQuery query(db);
2167         if (last == -1) {
2168             query.prepare(queryString("select_messagesAllNew_filtered"));
2169         }
2170         else {
2171             query.prepare(queryString("select_messagesAll_filtered"));
2172             query.bindValue(":lastmsg", last.toQint64());
2173         }
2174         query.bindValue(":userid", user.toInt());
2175         query.bindValue(":firstmsg", first.toQint64());
2176         query.bindValue(":limit", limit);
2177         int typeRaw = type;
2178         query.bindValue(":type", typeRaw);
2179         int flagsRaw = flags;
2180         query.bindValue(":flags", flagsRaw);
2181         safeExec(query);
2182
2183         watchQuery(query);
2184
2185         while (query.next()) {
2186             Message msg(
2187                         // Read timestamp in seconds as 64-bit integer
2188                         //
2189                         // NOTE:  This is a loss of precision.  The database time column would need
2190                         // to store a fractional number to support fromMSecsSinceEpoch(), or an
2191                         // upgrade step would need to convert all past times to milliseconds,
2192                         // multiplying by 1000.
2193 #if QT_VERSION >= 0x050800
2194                         QDateTime::fromSecsSinceEpoch(query.value(2).toLongLong()),
2195 #else
2196                         // fromSecsSinceEpoch() was added in Qt 5.8.  Manually downconvert to
2197                         // seconds for now.
2198                         // See https://doc.qt.io/qt-5/qdatetime.html#fromMSecsSinceEpoch
2199                         QDateTime::fromMSecsSinceEpoch(
2200                             (qint64)(query.value(2).toLongLong() * 1000)),
2201 #endif
2202                         bufferInfoHash[query.value(1).toInt()],
2203                         (Message::Type)query.value(3).toInt(),
2204                         query.value(9).toString(),
2205                         query.value(5).toString(),
2206                         query.value(6).toString(),
2207                         query.value(7).toString(),
2208                         query.value(8).toString(),
2209                         Message::Flags{query.value(4).toInt()});
2210             msg.setMsgId(query.value(0).toLongLong());
2211             messagelist << msg;
2212         }
2213     }
2214     db.commit();
2215     unlock();
2216     return messagelist;
2217 }
2218
2219 QMap<UserId, QString> SqliteStorage::getAllAuthUserNames()
2220 {
2221     QMap<UserId, QString> authusernames;
2222
2223     QSqlDatabase db = logDb();
2224     db.transaction();
2225     {
2226         QSqlQuery query(db);
2227         query.prepare(queryString("select_all_authusernames"));
2228
2229         lockForRead();
2230         safeExec(query);
2231         watchQuery(query);
2232         while (query.next()) {
2233             authusernames[query.value(0).toInt()] = query.value(1).toString();
2234         }
2235     }
2236     db.commit();
2237     unlock();
2238     return authusernames;
2239 }
2240
2241
2242 QString SqliteStorage::getAuthUserName(UserId user) {
2243     QString authusername;
2244     QSqlQuery query(logDb());
2245     query.prepare(queryString("select_authusername"));
2246     query.bindValue(":userid", user.toInt());
2247
2248     lockForRead();
2249     safeExec(query);
2250     watchQuery(query);
2251     unlock();
2252
2253     if (query.first()) {
2254         authusername = query.value(0).toString();
2255     }
2256
2257     return authusername;
2258 }
2259
2260
2261 QString SqliteStorage::backlogFile()
2262 {
2263     return Quassel::configDirPath() + "quassel-storage.sqlite";
2264 }
2265
2266
2267 bool SqliteStorage::safeExec(QSqlQuery &query, int retryCount)
2268 {
2269     query.exec();
2270
2271     if (!query.lastError().isValid())
2272         return true;
2273
2274     switch (query.lastError().number()) {
2275     case 5: // SQLITE_BUSY         5   /* The database file is locked */
2276         [[clang::fallthrough]];
2277     case 6: // SQLITE_LOCKED       6   /* A table in the database is locked */
2278         if (retryCount < _maxRetryCount)
2279             return safeExec(query, retryCount + 1);
2280         break;
2281     default:
2282         ;
2283     }
2284     return false;
2285 }
2286
2287
2288 // ========================================
2289 //  SqliteMigration
2290 // ========================================
2291 SqliteMigrationReader::SqliteMigrationReader()
2292     : SqliteStorage(),
2293     _maxId(0)
2294 {
2295 }
2296
2297
2298 void SqliteMigrationReader::setMaxId(MigrationObject mo)
2299 {
2300     QString queryString;
2301     switch (mo) {
2302     case Sender:
2303         queryString = "SELECT max(senderid) FROM sender";
2304         break;
2305     case Backlog:
2306         queryString = "SELECT max(messageid) FROM backlog";
2307         break;
2308     default:
2309         _maxId = 0;
2310         return;
2311     }
2312     QSqlQuery query = logDb().exec(queryString);
2313     query.first();
2314     _maxId = query.value(0).toLongLong();
2315 }
2316
2317
2318 bool SqliteMigrationReader::prepareQuery(MigrationObject mo)
2319 {
2320     setMaxId(mo);
2321
2322     switch (mo) {
2323     case QuasselUser:
2324         newQuery(queryString("migrate_read_quasseluser"), logDb());
2325         break;
2326     case Identity:
2327         newQuery(queryString("migrate_read_identity"), logDb());
2328         break;
2329     case IdentityNick:
2330         newQuery(queryString("migrate_read_identity_nick"), logDb());
2331         break;
2332     case Network:
2333         newQuery(queryString("migrate_read_network"), logDb());
2334         break;
2335     case Buffer:
2336         newQuery(queryString("migrate_read_buffer"), logDb());
2337         break;
2338     case Sender:
2339         newQuery(queryString("migrate_read_sender"), logDb());
2340         bindValue(0, 0);
2341         bindValue(1, stepSize());
2342         break;
2343     case Backlog:
2344         newQuery(queryString("migrate_read_backlog"), logDb());
2345         bindValue(0, 0);
2346         bindValue(1, stepSize());
2347         break;
2348     case IrcServer:
2349         newQuery(queryString("migrate_read_ircserver"), logDb());
2350         break;
2351     case UserSetting:
2352         newQuery(queryString("migrate_read_usersetting"), logDb());
2353         break;
2354     case CoreState:
2355         newQuery(queryString("migrate_read_corestate"), logDb());
2356         break;
2357     }
2358     return exec();
2359 }
2360
2361
2362 bool SqliteMigrationReader::readMo(QuasselUserMO &user)
2363 {
2364     if (!next())
2365         return false;
2366
2367     user.id = value(0).toInt();
2368     user.username = value(1).toString();
2369     user.password = value(2).toString();
2370     user.hashversion = value(3).toInt();
2371     user.authenticator = value(4).toString();
2372     return true;
2373 }
2374
2375
2376 bool SqliteMigrationReader::readMo(IdentityMO &identity)
2377 {
2378     if (!next())
2379         return false;
2380
2381     identity.id = value(0).toInt();
2382     identity.userid = value(1).toInt();
2383     identity.identityname = value(2).toString();
2384     identity.realname = value(3).toString();
2385     identity.awayNick = value(4).toString();
2386     identity.awayNickEnabled = value(5).toInt() == 1 ? true : false;
2387     identity.awayReason = value(6).toString();
2388     identity.awayReasonEnabled = value(7).toInt() == 1 ? true : false;
2389     identity.autoAwayEnabled = value(8).toInt() == 1 ? true : false;
2390     identity.autoAwayTime = value(9).toInt();
2391     identity.autoAwayReason = value(10).toString();
2392     identity.autoAwayReasonEnabled = value(11).toInt() == 1 ? true : false;
2393     identity.detachAwayEnabled = value(12).toInt() == 1 ? true : false;
2394     identity.detachAwayReason = value(13).toString();
2395     identity.detachAwayReasonEnabled = value(14).toInt() == 1 ? true : false;
2396     identity.ident = value(15).toString();
2397     identity.kickReason = value(16).toString();
2398     identity.partReason = value(17).toString();
2399     identity.quitReason = value(18).toString();
2400     identity.sslCert = value(19).toByteArray();
2401     identity.sslKey = value(20).toByteArray();
2402     return true;
2403 }
2404
2405
2406 bool SqliteMigrationReader::readMo(IdentityNickMO &identityNick)
2407 {
2408     if (!next())
2409         return false;
2410
2411     identityNick.nickid = value(0).toInt();
2412     identityNick.identityId = value(1).toInt();
2413     identityNick.nick = value(2).toString();
2414     return true;
2415 }
2416
2417
2418 bool SqliteMigrationReader::readMo(NetworkMO &network)
2419 {
2420     if (!next())
2421         return false;
2422
2423     network.networkid = value(0).toInt();
2424     network.userid = value(1).toInt();
2425     network.networkname = value(2).toString();
2426     network.identityid = value(3).toInt();
2427     network.encodingcodec = value(4).toString();
2428     network.decodingcodec = value(5).toString();
2429     network.servercodec = value(6).toString();
2430     network.userandomserver = value(7).toInt() == 1 ? true : false;
2431     network.perform = value(8).toString();
2432     network.useautoidentify = value(9).toInt() == 1 ? true : false;
2433     network.autoidentifyservice = value(10).toString();
2434     network.autoidentifypassword = value(11).toString();
2435     network.useautoreconnect = value(12).toInt() == 1 ? true : false;
2436     network.autoreconnectinterval = value(13).toInt();
2437     network.autoreconnectretries = value(14).toInt();
2438     network.unlimitedconnectretries = value(15).toInt() == 1 ? true : false;
2439     network.rejoinchannels = value(16).toInt() == 1 ? true : false;
2440     network.connected = value(17).toInt() == 1 ? true : false;
2441     network.usermode = value(18).toString();
2442     network.awaymessage = value(19).toString();
2443     network.attachperform = value(20).toString();
2444     network.detachperform = value(21).toString();
2445     network.usesasl = value(22).toInt() == 1 ? true : false;
2446     network.saslaccount = value(23).toString();
2447     network.saslpassword = value(24).toString();
2448     // Custom rate limiting
2449     network.usecustommessagerate = value(25).toInt() == 1 ? true : false;
2450     network.messagerateburstsize = value(26).toInt();
2451     network.messageratedelay = value(27).toUInt();
2452     network.unlimitedmessagerate = value(28).toInt() == 1 ? true : false;
2453     return true;
2454 }
2455
2456
2457 bool SqliteMigrationReader::readMo(BufferMO &buffer)
2458 {
2459     if (!next())
2460         return false;
2461
2462     buffer.bufferid = value(0).toInt();
2463     buffer.userid = value(1).toInt();
2464     buffer.groupid = value(2).toInt();
2465     buffer.networkid = value(3).toInt();
2466     buffer.buffername = value(4).toString();
2467     buffer.buffercname = value(5).toString();
2468     buffer.buffertype = value(6).toInt();
2469     buffer.lastmsgid = value(7).toLongLong();
2470     buffer.lastseenmsgid = value(8).toLongLong();
2471     buffer.markerlinemsgid = value(9).toLongLong();
2472     buffer.bufferactivity = value(10).toInt();
2473     buffer.highlightcount = value(11).toInt();
2474     buffer.key = value(12).toString();
2475     buffer.joined = value(13).toInt() == 1 ? true : false;
2476     buffer.cipher = value(14).toString();
2477     return true;
2478 }
2479
2480
2481 bool SqliteMigrationReader::readMo(SenderMO &sender)
2482 {
2483     int skipSteps = 0;
2484     while (!next()) {
2485         if (sender.senderId < _maxId) {
2486             bindValue(0, sender.senderId + (skipSteps * stepSize()));
2487             bindValue(1, sender.senderId + ((skipSteps + 1) * stepSize()));
2488             skipSteps++;
2489             if (!exec())
2490                 return false;
2491         }
2492         else {
2493             return false;
2494         }
2495     }
2496
2497     sender.senderId = value(0).toLongLong();
2498     sender.sender = value(1).toString();
2499     sender.realname = value(2).toString();
2500     sender.avatarurl = value(3).toString();
2501     return true;
2502 }
2503
2504
2505 bool SqliteMigrationReader::readMo(BacklogMO &backlog)
2506 {
2507     qint64 skipSteps = 0;
2508     while (!next()) {
2509         if (backlog.messageid < _maxId) {
2510             bindValue(0, backlog.messageid.toQint64() + (skipSteps * stepSize()));
2511             bindValue(1, backlog.messageid.toQint64() + ((skipSteps + 1) * stepSize()));
2512             skipSteps++;
2513             if (!exec())
2514                 return false;
2515         }
2516         else {
2517             return false;
2518         }
2519     }
2520
2521     backlog.messageid = value(0).toLongLong();
2522     // Read timestamp in seconds as 64-bit integer
2523     //
2524     // NOTE:  This is a loss of precision.  The database time column would need to store a
2525     // fractional number to support fromMSecsSinceEpoch(), or an upgrade step would need to convert
2526     // all past times to milliseconds, multiplying by 1000.
2527 #if QT_VERSION >= 0x050800
2528     backlog.time = QDateTime::fromSecsSinceEpoch(value(1).toLongLong()).toUTC();
2529 #else
2530     // fromSecsSinceEpoch() was added in Qt 5.8.  Manually downconvert to seconds for
2531     // now.
2532     // See https://doc.qt.io/qt-5/qdatetime.html#fromMSecsSinceEpoch
2533     backlog.time = QDateTime::fromMSecsSinceEpoch((qint64)(value(1).toLongLong() * 1000)
2534                                                   ).toUTC();
2535 #endif
2536     backlog.bufferid = value(2).toInt();
2537     backlog.type = value(3).toInt();
2538     backlog.flags = value(4).toInt();
2539     backlog.senderid = value(5).toLongLong();
2540     backlog.senderprefixes = value(6).toString();
2541     backlog.message = value(7).toString();
2542     return true;
2543 }
2544
2545
2546 bool SqliteMigrationReader::readMo(IrcServerMO &ircserver)
2547 {
2548     if (!next())
2549         return false;
2550
2551     ircserver.serverid = value(0).toInt();
2552     ircserver.userid = value(1).toInt();
2553     ircserver.networkid = value(2).toInt();
2554     ircserver.hostname = value(3).toString();
2555     ircserver.port = value(4).toInt();
2556     ircserver.password = value(5).toString();
2557     ircserver.ssl = value(6).toInt() == 1 ? true : false;
2558     ircserver.sslversion = value(7).toInt();
2559     ircserver.useproxy = value(8).toInt() == 1 ? true : false;
2560     ircserver.proxytype = value(9).toInt();
2561     ircserver.proxyhost = value(10).toString();
2562     ircserver.proxyport = value(11).toInt();
2563     ircserver.proxyuser = value(12).toString();
2564     ircserver.proxypass = value(13).toString();
2565     ircserver.sslverify = value(14).toInt() == 1 ? true : false;
2566     return true;
2567 }
2568
2569
2570 bool SqliteMigrationReader::readMo(UserSettingMO &userSetting)
2571 {
2572     if (!next())
2573         return false;
2574
2575     userSetting.userid = value(0).toInt();
2576     userSetting.settingname = value(1).toString();
2577     userSetting.settingvalue = value(2).toByteArray();
2578
2579     return true;
2580 }
2581
2582
2583 bool SqliteMigrationReader::readMo(CoreStateMO &coreState)
2584 {
2585     if (!next())
2586         return false;
2587
2588     coreState.key = value(0).toString();
2589     coreState.value = value(1).toByteArray();
2590
2591     return true;
2592 }