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.toInt());
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).toInt();
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.toInt());
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).toInt();
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.toInt());
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.toInt());
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
1760         logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
1761         logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1762         logMessageQuery.bindValue(":type", msg.type());
1763         logMessageQuery.bindValue(":flags", (int)msg.flags());
1764         logMessageQuery.bindValue(":sender", msg.sender());
1765         logMessageQuery.bindValue(":realname", msg.realName());
1766         logMessageQuery.bindValue(":avatarurl", msg.avatarUrl());
1767         logMessageQuery.bindValue(":senderprefixes", msg.senderPrefixes());
1768         logMessageQuery.bindValue(":message", msg.contents());
1769
1770         lockForWrite();
1771         safeExec(logMessageQuery);
1772
1773         if (logMessageQuery.lastError().isValid()) {
1774             // constraint violation - must be NOT NULL constraint - probably the sender is missing...
1775             if (logMessageQuery.lastError().number() == 19) {
1776                 QSqlQuery addSenderQuery(db);
1777                 addSenderQuery.prepare(queryString("insert_sender"));
1778                 addSenderQuery.bindValue(":sender", msg.sender());
1779                 addSenderQuery.bindValue(":realname", msg.realName());
1780                 addSenderQuery.bindValue(":avatarurl", msg.avatarUrl());
1781                 safeExec(addSenderQuery);
1782                 safeExec(logMessageQuery);
1783                 error = !watchQuery(logMessageQuery);
1784             }
1785             else {
1786                 watchQuery(logMessageQuery);
1787             }
1788         }
1789         if (!error) {
1790             MsgId msgId = logMessageQuery.lastInsertId().toInt();
1791             if (msgId.isValid()) {
1792                 msg.setMsgId(msgId);
1793             }
1794             else {
1795                 error = true;
1796             }
1797         }
1798     }
1799
1800     if (error) {
1801         db.rollback();
1802     }
1803     else {
1804         db.commit();
1805     }
1806
1807     unlock();
1808     return !error;
1809 }
1810
1811
1812 bool SqliteStorage::logMessages(MessageList &msgs)
1813 {
1814     QSqlDatabase db = logDb();
1815     db.transaction();
1816
1817     {
1818         QSet<SenderData> senders;
1819         QSqlQuery addSenderQuery(db);
1820         addSenderQuery.prepare(queryString("insert_sender"));
1821         lockForWrite();
1822         for (int i = 0; i < msgs.count(); i++) {
1823             auto &msg = msgs.at(i);
1824             SenderData sender = { msg.sender(), msg.realName(), msg.avatarUrl() };
1825             if (senders.contains(sender))
1826                 continue;
1827             senders << sender;
1828
1829             addSenderQuery.bindValue(":sender", sender.sender);
1830             addSenderQuery.bindValue(":realname", sender.realname);
1831             addSenderQuery.bindValue(":avatarurl", sender.avatarurl);
1832             safeExec(addSenderQuery);
1833         }
1834     }
1835
1836     bool error = false;
1837     {
1838         QSqlQuery logMessageQuery(db);
1839         logMessageQuery.prepare(queryString("insert_message"));
1840         for (int i = 0; i < msgs.count(); i++) {
1841             Message &msg = msgs[i];
1842
1843             logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
1844             logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1845             logMessageQuery.bindValue(":type", msg.type());
1846             logMessageQuery.bindValue(":flags", (int)msg.flags());
1847             logMessageQuery.bindValue(":sender", msg.sender());
1848             logMessageQuery.bindValue(":realname", msg.realName());
1849             logMessageQuery.bindValue(":avatarurl", msg.avatarUrl());
1850             logMessageQuery.bindValue(":senderprefixes", msg.senderPrefixes());
1851             logMessageQuery.bindValue(":message", msg.contents());
1852
1853             safeExec(logMessageQuery);
1854             if (!watchQuery(logMessageQuery)) {
1855                 error = true;
1856                 break;
1857             }
1858             else {
1859                 msg.setMsgId(logMessageQuery.lastInsertId().toInt());
1860             }
1861         }
1862     }
1863
1864     if (error) {
1865         db.rollback();
1866         unlock();
1867         // we had a rollback in the db so we need to reset all msgIds
1868         for (int i = 0; i < msgs.count(); i++) {
1869             msgs[i].setMsgId(MsgId());
1870         }
1871     }
1872     else {
1873         db.commit();
1874         unlock();
1875     }
1876     return !error;
1877 }
1878
1879
1880 QList<Message> SqliteStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit)
1881 {
1882     QList<Message> messagelist;
1883
1884     QSqlDatabase db = logDb();
1885     db.transaction();
1886
1887     bool error = false;
1888     BufferInfo bufferInfo;
1889     {
1890         // code dupication from getBufferInfo:
1891         // this is due to the impossibility of nesting transactions and recursive locking
1892         QSqlQuery bufferInfoQuery(db);
1893         bufferInfoQuery.prepare(queryString("select_buffer_by_id"));
1894         bufferInfoQuery.bindValue(":userid", user.toInt());
1895         bufferInfoQuery.bindValue(":bufferid", bufferId.toInt());
1896
1897         lockForRead();
1898         safeExec(bufferInfoQuery);
1899         error = !watchQuery(bufferInfoQuery) || !bufferInfoQuery.first();
1900         if (!error) {
1901             bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), 0, bufferInfoQuery.value(4).toString());
1902             error = !bufferInfo.isValid();
1903         }
1904     }
1905     if (error) {
1906         db.rollback();
1907         unlock();
1908         return messagelist;
1909     }
1910
1911     {
1912         QSqlQuery query(db);
1913         if (last == -1 && first == -1) {
1914             query.prepare(queryString("select_messagesNewestK"));
1915         }
1916         else if (last == -1) {
1917             query.prepare(queryString("select_messagesNewerThan"));
1918             query.bindValue(":firstmsg", first.toInt());
1919         }
1920         else {
1921             query.prepare(queryString("select_messagesRange"));
1922             query.bindValue(":lastmsg", last.toInt());
1923             query.bindValue(":firstmsg", first.toInt());
1924         }
1925         query.bindValue(":bufferid", bufferId.toInt());
1926         query.bindValue(":limit", limit);
1927
1928         safeExec(query);
1929         watchQuery(query);
1930
1931         while (query.next()) {
1932             Message msg(QDateTime::fromTime_t(query.value(1).toInt()),
1933                 bufferInfo,
1934                 (Message::Type)query.value(2).toUInt(),
1935                 query.value(8).toString(),
1936                 query.value(4).toString(),
1937                 query.value(5).toString(),
1938                 query.value(6).toString(),
1939                 query.value(7).toString(),
1940                 (Message::Flags)query.value(3).toUInt());
1941             msg.setMsgId(query.value(0).toInt());
1942             messagelist << msg;
1943         }
1944     }
1945     db.commit();
1946     unlock();
1947
1948     return messagelist;
1949 }
1950
1951
1952 QList<Message> SqliteStorage::requestMsgsFiltered(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit, Message::Types type, Message::Flags flags)
1953 {
1954     QList<Message> messagelist;
1955
1956     QSqlDatabase db = logDb();
1957     db.transaction();
1958
1959     bool error = false;
1960     BufferInfo bufferInfo;
1961     {
1962         // code dupication from getBufferInfo:
1963         // this is due to the impossibility of nesting transactions and recursive locking
1964         QSqlQuery bufferInfoQuery(db);
1965         bufferInfoQuery.prepare(queryString("select_buffer_by_id"));
1966         bufferInfoQuery.bindValue(":userid", user.toInt());
1967         bufferInfoQuery.bindValue(":bufferid", bufferId.toInt());
1968
1969         lockForRead();
1970         safeExec(bufferInfoQuery);
1971         error = !watchQuery(bufferInfoQuery) || !bufferInfoQuery.first();
1972         if (!error) {
1973             bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), 0, bufferInfoQuery.value(4).toString());
1974             error = !bufferInfo.isValid();
1975         }
1976     }
1977     if (error) {
1978         db.rollback();
1979         unlock();
1980         return messagelist;
1981     }
1982
1983     {
1984         QSqlQuery query(db);
1985         if (last == -1 && first == -1) {
1986             query.prepare(queryString("select_messagesNewestK_filtered"));
1987         }
1988         else if (last == -1) {
1989             query.prepare(queryString("select_messagesNewerThan_filtered"));
1990             query.bindValue(":firstmsg", first.toInt());
1991         }
1992         else {
1993             query.prepare(queryString("select_messagesRange_filtered"));
1994             query.bindValue(":lastmsg", last.toInt());
1995             query.bindValue(":firstmsg", first.toInt());
1996         }
1997         query.bindValue(":bufferid", bufferId.toInt());
1998         query.bindValue(":limit", limit);
1999         int typeRaw = type;
2000         query.bindValue(":type", typeRaw);
2001         int flagsRaw = flags;
2002         query.bindValue(":flags", flagsRaw);
2003
2004         safeExec(query);
2005         watchQuery(query);
2006
2007         while (query.next()) {
2008             Message msg(QDateTime::fromTime_t(query.value(1).toInt()),
2009                         bufferInfo,
2010                         (Message::Type)query.value(2).toUInt(),
2011                         query.value(8).toString(),
2012                         query.value(4).toString(),
2013                         query.value(5).toString(),
2014                         query.value(6).toString(),
2015                         query.value(7).toString(),
2016                         Message::Flags{query.value(3).toInt()});
2017             msg.setMsgId(query.value(0).toInt());
2018             messagelist << msg;
2019         }
2020     }
2021     db.commit();
2022     unlock();
2023
2024     return messagelist;
2025 }
2026
2027
2028 QList<Message> SqliteStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit)
2029 {
2030     QList<Message> messagelist;
2031
2032     QSqlDatabase db = logDb();
2033     db.transaction();
2034
2035     QHash<BufferId, BufferInfo> bufferInfoHash;
2036     {
2037         QSqlQuery bufferInfoQuery(db);
2038         bufferInfoQuery.prepare(queryString("select_buffers"));
2039         bufferInfoQuery.bindValue(":userid", user.toInt());
2040
2041         lockForRead();
2042         safeExec(bufferInfoQuery);
2043         watchQuery(bufferInfoQuery);
2044         while (bufferInfoQuery.next()) {
2045             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());
2046             bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
2047         }
2048
2049         QSqlQuery query(db);
2050         if (last == -1) {
2051             query.prepare(queryString("select_messagesAllNew"));
2052         }
2053         else {
2054             query.prepare(queryString("select_messagesAll"));
2055             query.bindValue(":lastmsg", last.toInt());
2056         }
2057         query.bindValue(":userid", user.toInt());
2058         query.bindValue(":firstmsg", first.toInt());
2059         query.bindValue(":limit", limit);
2060         safeExec(query);
2061
2062         watchQuery(query);
2063
2064         while (query.next()) {
2065             Message msg(QDateTime::fromTime_t(query.value(2).toInt()),
2066                 bufferInfoHash[query.value(1).toInt()],
2067                 (Message::Type)query.value(3).toUInt(),
2068                 query.value(9).toString(),
2069                 query.value(5).toString(),
2070                 query.value(6).toString(),
2071                 query.value(7).toString(),
2072                 query.value(8).toString(),
2073                 (Message::Flags)query.value(4).toUInt());
2074             msg.setMsgId(query.value(0).toInt());
2075             messagelist << msg;
2076         }
2077     }
2078     db.commit();
2079     unlock();
2080     return messagelist;
2081 }
2082
2083 QList<Message> SqliteStorage::requestAllMsgsFiltered(UserId user, MsgId first, MsgId last, int limit, Message::Types type, Message::Flags flags)
2084 {
2085     QList<Message> messagelist;
2086
2087     QSqlDatabase db = logDb();
2088     db.transaction();
2089
2090     QHash<BufferId, BufferInfo> bufferInfoHash;
2091     {
2092         QSqlQuery bufferInfoQuery(db);
2093         bufferInfoQuery.prepare(queryString("select_buffers"));
2094         bufferInfoQuery.bindValue(":userid", user.toInt());
2095
2096         lockForRead();
2097         safeExec(bufferInfoQuery);
2098         watchQuery(bufferInfoQuery);
2099         while (bufferInfoQuery.next()) {
2100             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());
2101             bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
2102         }
2103
2104         QSqlQuery query(db);
2105         if (last == -1) {
2106             query.prepare(queryString("select_messagesAllNew_filtered"));
2107         }
2108         else {
2109             query.prepare(queryString("select_messagesAll_filtered"));
2110             query.bindValue(":lastmsg", last.toInt());
2111         }
2112         query.bindValue(":userid", user.toInt());
2113         query.bindValue(":firstmsg", first.toInt());
2114         query.bindValue(":limit", limit);
2115         int typeRaw = type;
2116         query.bindValue(":type", typeRaw);
2117         int flagsRaw = flags;
2118         query.bindValue(":flags", flagsRaw);
2119         safeExec(query);
2120
2121         watchQuery(query);
2122
2123         while (query.next()) {
2124             Message msg(QDateTime::fromTime_t(query.value(2).toInt()),
2125                         bufferInfoHash[query.value(1).toInt()],
2126                         (Message::Type)query.value(3).toUInt(),
2127                         query.value(9).toString(),
2128                         query.value(5).toString(),
2129                         query.value(6).toString(),
2130                         query.value(7).toString(),
2131                         query.value(8).toString(),
2132                         Message::Flags{query.value(4).toInt()});
2133             msg.setMsgId(query.value(0).toInt());
2134             messagelist << msg;
2135         }
2136     }
2137     db.commit();
2138     unlock();
2139     return messagelist;
2140 }
2141
2142 QMap<UserId, QString> SqliteStorage::getAllAuthUserNames()
2143 {
2144     QMap<UserId, QString> authusernames;
2145
2146     QSqlDatabase db = logDb();
2147     db.transaction();
2148     {
2149         QSqlQuery query(db);
2150         query.prepare(queryString("select_all_authusernames"));
2151
2152         lockForRead();
2153         safeExec(query);
2154         watchQuery(query);
2155         while (query.next()) {
2156             authusernames[query.value(0).toInt()] = query.value(1).toString();
2157         }
2158     }
2159     db.commit();
2160     unlock();
2161     return authusernames;
2162 }
2163
2164
2165 QString SqliteStorage::getAuthUserName(UserId user) {
2166     QString authusername;
2167     QSqlQuery query(logDb());
2168     query.prepare(queryString("select_authusername"));
2169     query.bindValue(":userid", user.toInt());
2170
2171     lockForRead();
2172     safeExec(query);
2173     watchQuery(query);
2174     unlock();
2175
2176     if (query.first()) {
2177         authusername = query.value(0).toString();
2178     }
2179
2180     return authusername;
2181 }
2182
2183
2184 QString SqliteStorage::backlogFile()
2185 {
2186     return Quassel::configDirPath() + "quassel-storage.sqlite";
2187 }
2188
2189
2190 bool SqliteStorage::safeExec(QSqlQuery &query, int retryCount)
2191 {
2192     query.exec();
2193
2194     if (!query.lastError().isValid())
2195         return true;
2196
2197     switch (query.lastError().number()) {
2198     case 5: // SQLITE_BUSY         5   /* The database file is locked */
2199         [[clang::fallthrough]];
2200     case 6: // SQLITE_LOCKED       6   /* A table in the database is locked */
2201         if (retryCount < _maxRetryCount)
2202             return safeExec(query, retryCount + 1);
2203         break;
2204     default:
2205         ;
2206     }
2207     return false;
2208 }
2209
2210
2211 // ========================================
2212 //  SqliteMigration
2213 // ========================================
2214 SqliteMigrationReader::SqliteMigrationReader()
2215     : SqliteStorage(),
2216     _maxId(0)
2217 {
2218 }
2219
2220
2221 void SqliteMigrationReader::setMaxId(MigrationObject mo)
2222 {
2223     QString queryString;
2224     switch (mo) {
2225     case Sender:
2226         queryString = "SELECT max(senderid) FROM sender";
2227         break;
2228     case Backlog:
2229         queryString = "SELECT max(messageid) FROM backlog";
2230         break;
2231     default:
2232         _maxId = 0;
2233         return;
2234     }
2235     QSqlQuery query = logDb().exec(queryString);
2236     query.first();
2237     _maxId = query.value(0).toInt();
2238 }
2239
2240
2241 bool SqliteMigrationReader::prepareQuery(MigrationObject mo)
2242 {
2243     setMaxId(mo);
2244
2245     switch (mo) {
2246     case QuasselUser:
2247         newQuery(queryString("migrate_read_quasseluser"), logDb());
2248         break;
2249     case Identity:
2250         newQuery(queryString("migrate_read_identity"), logDb());
2251         break;
2252     case IdentityNick:
2253         newQuery(queryString("migrate_read_identity_nick"), logDb());
2254         break;
2255     case Network:
2256         newQuery(queryString("migrate_read_network"), logDb());
2257         break;
2258     case Buffer:
2259         newQuery(queryString("migrate_read_buffer"), logDb());
2260         break;
2261     case Sender:
2262         newQuery(queryString("migrate_read_sender"), logDb());
2263         bindValue(0, 0);
2264         bindValue(1, stepSize());
2265         break;
2266     case Backlog:
2267         newQuery(queryString("migrate_read_backlog"), logDb());
2268         bindValue(0, 0);
2269         bindValue(1, stepSize());
2270         break;
2271     case IrcServer:
2272         newQuery(queryString("migrate_read_ircserver"), logDb());
2273         break;
2274     case UserSetting:
2275         newQuery(queryString("migrate_read_usersetting"), logDb());
2276         break;
2277     case CoreState:
2278         newQuery(queryString("migrate_read_corestate"), logDb());
2279         break;
2280     }
2281     return exec();
2282 }
2283
2284
2285 bool SqliteMigrationReader::readMo(QuasselUserMO &user)
2286 {
2287     if (!next())
2288         return false;
2289
2290     user.id = value(0).toInt();
2291     user.username = value(1).toString();
2292     user.password = value(2).toString();
2293     user.hashversion = value(3).toInt();
2294     user.authenticator = value(4).toString();
2295     return true;
2296 }
2297
2298
2299 bool SqliteMigrationReader::readMo(IdentityMO &identity)
2300 {
2301     if (!next())
2302         return false;
2303
2304     identity.id = value(0).toInt();
2305     identity.userid = value(1).toInt();
2306     identity.identityname = value(2).toString();
2307     identity.realname = value(3).toString();
2308     identity.awayNick = value(4).toString();
2309     identity.awayNickEnabled = value(5).toInt() == 1 ? true : false;
2310     identity.awayReason = value(6).toString();
2311     identity.awayReasonEnabled = value(7).toInt() == 1 ? true : false;
2312     identity.autoAwayEnabled = value(8).toInt() == 1 ? true : false;
2313     identity.autoAwayTime = value(9).toInt();
2314     identity.autoAwayReason = value(10).toString();
2315     identity.autoAwayReasonEnabled = value(11).toInt() == 1 ? true : false;
2316     identity.detachAwayEnabled = value(12).toInt() == 1 ? true : false;
2317     identity.detachAwayReason = value(13).toString();
2318     identity.detchAwayReasonEnabled = value(14).toInt() == 1 ? true : false;
2319     identity.ident = value(15).toString();
2320     identity.kickReason = value(16).toString();
2321     identity.partReason = value(17).toString();
2322     identity.quitReason = value(18).toString();
2323     identity.sslCert = value(19).toByteArray();
2324     identity.sslKey = value(20).toByteArray();
2325     return true;
2326 }
2327
2328
2329 bool SqliteMigrationReader::readMo(IdentityNickMO &identityNick)
2330 {
2331     if (!next())
2332         return false;
2333
2334     identityNick.nickid = value(0).toInt();
2335     identityNick.identityId = value(1).toInt();
2336     identityNick.nick = value(2).toString();
2337     return true;
2338 }
2339
2340
2341 bool SqliteMigrationReader::readMo(NetworkMO &network)
2342 {
2343     if (!next())
2344         return false;
2345
2346     network.networkid = value(0).toInt();
2347     network.userid = value(1).toInt();
2348     network.networkname = value(2).toString();
2349     network.identityid = value(3).toInt();
2350     network.encodingcodec = value(4).toString();
2351     network.decodingcodec = value(5).toString();
2352     network.servercodec = value(6).toString();
2353     network.userandomserver = value(7).toInt() == 1 ? true : false;
2354     network.perform = value(8).toString();
2355     network.useautoidentify = value(9).toInt() == 1 ? true : false;
2356     network.autoidentifyservice = value(10).toString();
2357     network.autoidentifypassword = value(11).toString();
2358     network.useautoreconnect = value(12).toInt() == 1 ? true : false;
2359     network.autoreconnectinterval = value(13).toInt();
2360     network.autoreconnectretries = value(14).toInt();
2361     network.unlimitedconnectretries = value(15).toInt() == 1 ? true : false;
2362     network.rejoinchannels = value(16).toInt() == 1 ? true : false;
2363     network.connected = value(17).toInt() == 1 ? true : false;
2364     network.usermode = value(18).toString();
2365     network.awaymessage = value(19).toString();
2366     network.attachperform = value(20).toString();
2367     network.detachperform = value(21).toString();
2368     network.usesasl = value(22).toInt() == 1 ? true : false;
2369     network.saslaccount = value(23).toString();
2370     network.saslpassword = value(24).toString();
2371     // Custom rate limiting
2372     network.usecustommessagerate = value(25).toInt() == 1 ? true : false;
2373     network.messagerateburstsize = value(26).toInt();
2374     network.messageratedelay = value(27).toUInt();
2375     network.unlimitedmessagerate = value(28).toInt() == 1 ? true : false;
2376     return true;
2377 }
2378
2379
2380 bool SqliteMigrationReader::readMo(BufferMO &buffer)
2381 {
2382     if (!next())
2383         return false;
2384
2385     buffer.bufferid = value(0).toInt();
2386     buffer.userid = value(1).toInt();
2387     buffer.groupid = value(2).toInt();
2388     buffer.networkid = value(3).toInt();
2389     buffer.buffername = value(4).toString();
2390     buffer.buffercname = value(5).toString();
2391     buffer.buffertype = value(6).toInt();
2392     buffer.lastmsgid = value(7).toInt();
2393     buffer.lastseenmsgid = value(8).toInt();
2394     buffer.markerlinemsgid = value(9).toInt();
2395     buffer.bufferactivity = value(10).toInt();
2396     buffer.highlightcount = value(11).toInt();
2397     buffer.key = value(12).toString();
2398     buffer.joined = value(13).toInt() == 1 ? true : false;
2399     buffer.cipher = value(14).toString();
2400     return true;
2401 }
2402
2403
2404 bool SqliteMigrationReader::readMo(SenderMO &sender)
2405 {
2406     int skipSteps = 0;
2407     while (!next()) {
2408         if (sender.senderId < _maxId) {
2409             bindValue(0, sender.senderId + (skipSteps * stepSize()));
2410             bindValue(1, sender.senderId + ((skipSteps + 1) * stepSize()));
2411             skipSteps++;
2412             if (!exec())
2413                 return false;
2414         }
2415         else {
2416             return false;
2417         }
2418     }
2419
2420     sender.senderId = value(0).toInt();
2421     sender.sender = value(1).toString();
2422     sender.realname = value(2).toString();
2423     sender.avatarurl = value(3).toString();
2424     return true;
2425 }
2426
2427
2428 bool SqliteMigrationReader::readMo(BacklogMO &backlog)
2429 {
2430     int skipSteps = 0;
2431     while (!next()) {
2432         if (backlog.messageid < _maxId) {
2433             bindValue(0, backlog.messageid.toInt() + (skipSteps * stepSize()));
2434             bindValue(1, backlog.messageid.toInt() + ((skipSteps + 1) * stepSize()));
2435             skipSteps++;
2436             if (!exec())
2437                 return false;
2438         }
2439         else {
2440             return false;
2441         }
2442     }
2443
2444     backlog.messageid = value(0).toInt();
2445     backlog.time = QDateTime::fromTime_t(value(1).toInt()).toUTC();
2446     backlog.bufferid = value(2).toInt();
2447     backlog.type = value(3).toInt();
2448     backlog.flags = value(4).toInt();
2449     backlog.senderid = value(5).toInt();
2450     backlog.senderprefixes = value(6).toString();
2451     backlog.message = value(7).toString();
2452     return true;
2453 }
2454
2455
2456 bool SqliteMigrationReader::readMo(IrcServerMO &ircserver)
2457 {
2458     if (!next())
2459         return false;
2460
2461     ircserver.serverid = value(0).toInt();
2462     ircserver.userid = value(1).toInt();
2463     ircserver.networkid = value(2).toInt();
2464     ircserver.hostname = value(3).toString();
2465     ircserver.port = value(4).toInt();
2466     ircserver.password = value(5).toString();
2467     ircserver.ssl = value(6).toInt() == 1 ? true : false;
2468     ircserver.sslversion = value(7).toInt();
2469     ircserver.useproxy = value(8).toInt() == 1 ? true : false;
2470     ircserver.proxytype = value(9).toInt();
2471     ircserver.proxyhost = value(10).toString();
2472     ircserver.proxyport = value(11).toInt();
2473     ircserver.proxyuser = value(12).toString();
2474     ircserver.proxypass = value(13).toString();
2475     ircserver.sslverify = value(14).toInt() == 1 ? true : false;
2476     return true;
2477 }
2478
2479
2480 bool SqliteMigrationReader::readMo(UserSettingMO &userSetting)
2481 {
2482     if (!next())
2483         return false;
2484
2485     userSetting.userid = value(0).toInt();
2486     userSetting.settingname = value(1).toString();
2487     userSetting.settingvalue = value(2).toByteArray();
2488
2489     return true;
2490 }
2491
2492
2493 bool SqliteMigrationReader::readMo(CoreStateMO &coreState)
2494 {
2495     if (!next())
2496         return false;
2497
2498     coreState.key = value(0).toString();
2499     coreState.value = value(1).toByteArray();
2500
2501     return true;
2502 }