src/core/sqlitestorage.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-2016 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 IdentityId SqliteStorage::createIdentity(UserId user, CoreIdentity &identity)
 380 {
 381     IdentityId identityId;
 382
 383     QSqlDatabase db = logDb();
 384     db.transaction();
 385
 386     {
 387         QSqlQuery query(db);
 388         query.prepare(queryString("insert_identity"));
 389         query.bindValue(":userid", user.toInt());
 390         query.bindValue(":identityname", identity.identityName());
 391         query.bindValue(":realname", identity.realName());
 392         query.bindValue(":awaynick", identity.awayNick());
 393         query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
 394         query.bindValue(":awayreason", identity.awayReason());
 395         query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
 396         query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
 397         query.bindValue(":autoawaytime", identity.autoAwayTime());
 398         query.bindValue(":autoawayreason", identity.autoAwayReason());
 399         query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
 400         query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
 401         query.bindValue(":detachawayreason", identity.detachAwayReason());
 402         query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
 403         query.bindValue(":ident", identity.ident());
 404         query.bindValue(":kickreason", identity.kickReason());
 405         query.bindValue(":partreason", identity.partReason());
 406         query.bindValue(":quitreason", identity.quitReason());
 407 #ifdef HAVE_SSL
 408         query.bindValue(":sslcert", identity.sslCert().toPem());
 409         query.bindValue(":sslkey", identity.sslKey().toPem());
 410 #else
 411         query.bindValue(":sslcert", QByteArray());
 412         query.bindValue(":sslkey", QByteArray());
 413 #endif
 414
 415         lockForWrite();
 416         safeExec(query);
 417
 418         identityId = query.lastInsertId().toInt();
 419         if (!identityId.isValid()) {
 420             watchQuery(query);
 421         }
 422         else {
 423             QSqlQuery deleteNickQuery(db);
 424             deleteNickQuery.prepare(queryString("delete_nicks"));
 425             deleteNickQuery.bindValue(":identityid", identityId.toInt());
 426             safeExec(deleteNickQuery);
 427
 428             QSqlQuery insertNickQuery(db);
 429             insertNickQuery.prepare(queryString("insert_nick"));
 430             foreach(QString nick, identity.nicks()) {
 431                 insertNickQuery.bindValue(":identityid", identityId.toInt());
 432                 insertNickQuery.bindValue(":nick", nick);
 433                 safeExec(insertNickQuery);
 434             }
 435         }
 436         db.commit();
 437     }
 438     unlock();
 439     identity.setId(identityId);
 440     return identityId;
 441 }
 442
 443
 444 bool SqliteStorage::updateIdentity(UserId user, const CoreIdentity &identity)
 445 {
 446     QSqlDatabase db = logDb();
 447     bool error = false;
 448     db.transaction();
 449
 450     {
 451         QSqlQuery checkQuery(db);
 452         checkQuery.prepare(queryString("select_checkidentity"));
 453         checkQuery.bindValue(":identityid", identity.id().toInt());
 454         checkQuery.bindValue(":userid", user.toInt());
 455         lockForRead();
 456         safeExec(checkQuery);
 457
 458         // there should be exactly one identity for the given id and user
 459         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 1);
 460     }
 461     if (error) {
 462         unlock();
 463         return false;
 464     }
 465
 466     {
 467         QSqlQuery query(db);
 468         query.prepare(queryString("update_identity"));
 469         query.bindValue(":identityname", identity.identityName());
 470         query.bindValue(":realname", identity.realName());
 471         query.bindValue(":awaynick", identity.awayNick());
 472         query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
 473         query.bindValue(":awayreason", identity.awayReason());
 474         query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
 475         query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
 476         query.bindValue(":autoawaytime", identity.autoAwayTime());
 477         query.bindValue(":autoawayreason", identity.autoAwayReason());
 478         query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
 479         query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
 480         query.bindValue(":detachawayreason", identity.detachAwayReason());
 481         query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
 482         query.bindValue(":ident", identity.ident());
 483         query.bindValue(":kickreason", identity.kickReason());
 484         query.bindValue(":partreason", identity.partReason());
 485         query.bindValue(":quitreason", identity.quitReason());
 486 #ifdef HAVE_SSL
 487         query.bindValue(":sslcert", identity.sslCert().toPem());
 488         query.bindValue(":sslkey", identity.sslKey().toPem());
 489 #else
 490         query.bindValue(":sslcert", QByteArray());
 491         query.bindValue(":sslkey", QByteArray());
 492 #endif
 493         query.bindValue(":identityid", identity.id().toInt());
 494         safeExec(query);
 495         watchQuery(query);
 496
 497         QSqlQuery deleteNickQuery(db);
 498         deleteNickQuery.prepare(queryString("delete_nicks"));
 499         deleteNickQuery.bindValue(":identityid", identity.id().toInt());
 500         safeExec(deleteNickQuery);
 501         watchQuery(deleteNickQuery);
 502
 503         QSqlQuery insertNickQuery(db);
 504         insertNickQuery.prepare(queryString("insert_nick"));
 505         foreach(QString nick, identity.nicks()) {
 506             insertNickQuery.bindValue(":identityid", identity.id().toInt());
 507             insertNickQuery.bindValue(":nick", nick);
 508             safeExec(insertNickQuery);
 509             watchQuery(insertNickQuery);
 510         }
 511         db.commit();
 512     }
 513     unlock();
 514     return true;
 515 }
 516
 517
 518 void SqliteStorage::removeIdentity(UserId user, IdentityId identityId)
 519 {
 520     QSqlDatabase db = logDb();
 521     db.transaction();
 522
 523     bool error = false;
 524     {
 525         QSqlQuery checkQuery(db);
 526         checkQuery.prepare(queryString("select_checkidentity"));
 527         checkQuery.bindValue(":identityid", identityId.toInt());
 528         checkQuery.bindValue(":userid", user.toInt());
 529         lockForRead();
 530         safeExec(checkQuery);
 531
 532         // there should be exactly one identity for the given id and user
 533         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 1);
 534     }
 535     if (error) {
 536         unlock();
 537         return;
 538     }
 539
 540     {
 541         QSqlQuery deleteNickQuery(db);
 542         deleteNickQuery.prepare(queryString("delete_nicks"));
 543         deleteNickQuery.bindValue(":identityid", identityId.toInt());
 544         safeExec(deleteNickQuery);
 545
 546         QSqlQuery deleteIdentityQuery(db);
 547         deleteIdentityQuery.prepare(queryString("delete_identity"));
 548         deleteIdentityQuery.bindValue(":identityid", identityId.toInt());
 549         deleteIdentityQuery.bindValue(":userid", user.toInt());
 550         safeExec(deleteIdentityQuery);
 551         db.commit();
 552     }
 553     unlock();
 554 }
 555
 556
 557 QList<CoreIdentity> SqliteStorage::identities(UserId user)
 558 {
 559     QList<CoreIdentity> identities;
 560     QSqlDatabase db = logDb();
 561     db.transaction();
 562
 563     {
 564         QSqlQuery query(db);
 565         query.prepare(queryString("select_identities"));
 566         query.bindValue(":userid", user.toInt());
 567
 568         QSqlQuery nickQuery(db);
 569         nickQuery.prepare(queryString("select_nicks"));
 570
 571         lockForRead();
 572         safeExec(query);
 573
 574         while (query.next()) {
 575             CoreIdentity identity(IdentityId(query.value(0).toInt()));
 576
 577             identity.setIdentityName(query.value(1).toString());
 578             identity.setRealName(query.value(2).toString());
 579             identity.setAwayNick(query.value(3).toString());
 580             identity.setAwayNickEnabled(!!query.value(4).toInt());
 581             identity.setAwayReason(query.value(5).toString());
 582             identity.setAwayReasonEnabled(!!query.value(6).toInt());
 583             identity.setAutoAwayEnabled(!!query.value(7).toInt());
 584             identity.setAutoAwayTime(query.value(8).toInt());
 585             identity.setAutoAwayReason(query.value(9).toString());
 586             identity.setAutoAwayReasonEnabled(!!query.value(10).toInt());
 587             identity.setDetachAwayEnabled(!!query.value(11).toInt());
 588             identity.setDetachAwayReason(query.value(12).toString());
 589             identity.setDetachAwayReasonEnabled(!!query.value(13).toInt());
 590             identity.setIdent(query.value(14).toString());
 591             identity.setKickReason(query.value(15).toString());
 592             identity.setPartReason(query.value(16).toString());
 593             identity.setQuitReason(query.value(17).toString());
 594 #ifdef HAVE_SSL
 595             identity.setSslCert(query.value(18).toByteArray());
 596             identity.setSslKey(query.value(19).toByteArray());
 597 #endif
 598
 599             nickQuery.bindValue(":identityid", identity.id().toInt());
 600             QList<QString> nicks;
 601             safeExec(nickQuery);
 602             watchQuery(nickQuery);
 603             while (nickQuery.next()) {
 604                 nicks << nickQuery.value(0).toString();
 605             }
 606             identity.setNicks(nicks);
 607             identities << identity;
 608         }
 609         db.commit();
 610     }
 611     unlock();
 612     return identities;
 613 }
 614
 615
 616 NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info)
 617 {
 618     NetworkId networkId;
 619
 620     QSqlDatabase db = logDb();
 621     db.transaction();
 622
 623     bool error = false;
 624     {
 625         QSqlQuery query(db);
 626         query.prepare(queryString("insert_network"));
 627         query.bindValue(":userid", user.toInt());
 628         bindNetworkInfo(query, info);
 629         lockForWrite();
 630         safeExec(query);
 631         if (!watchQuery(query)) {
 632             db.rollback();
 633             error = true;
 634         }
 635         else {
 636             networkId = query.lastInsertId().toInt();
 637         }
 638     }
 639     if (error) {
 640         unlock();
 641         return NetworkId();
 642     }
 643
 644     {
 645         QSqlQuery insertServersQuery(db);
 646         insertServersQuery.prepare(queryString("insert_server"));
 647         foreach(Network::Server server, info.serverList) {
 648             insertServersQuery.bindValue(":userid", user.toInt());
 649             insertServersQuery.bindValue(":networkid", networkId.toInt());
 650             bindServerInfo(insertServersQuery, server);
 651             safeExec(insertServersQuery);
 652             if (!watchQuery(insertServersQuery)) {
 653                 db.rollback();
 654                 error = true;
 655                 break;
 656             }
 657         }
 658         if (!error)
 659             db.commit();
 660     }
 661     unlock();
 662     if (error)
 663         return NetworkId();
 664     else
 665         return networkId;
 666 }
 667
 668
 669 void SqliteStorage::bindNetworkInfo(QSqlQuery &query, const NetworkInfo &info)
 670 {
 671     query.bindValue(":networkname", info.networkName);
 672     query.bindValue(":identityid", info.identity.toInt());
 673     query.bindValue(":encodingcodec", QString(info.codecForEncoding));
 674     query.bindValue(":decodingcodec", QString(info.codecForDecoding));
 675     query.bindValue(":servercodec", QString(info.codecForServer));
 676     query.bindValue(":userandomserver", info.useRandomServer ? 1 : 0);
 677     query.bindValue(":perform", info.perform.join("\n"));
 678     query.bindValue(":useautoidentify", info.useAutoIdentify ? 1 : 0);
 679     query.bindValue(":autoidentifyservice", info.autoIdentifyService);
 680     query.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
 681     query.bindValue(":usesasl", info.useSasl ? 1 : 0);
 682     query.bindValue(":saslaccount", info.saslAccount);
 683     query.bindValue(":saslpassword", info.saslPassword);
 684     query.bindValue(":useautoreconnect", info.useAutoReconnect ? 1 : 0);
 685     query.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
 686     query.bindValue(":autoreconnectretries", info.autoReconnectRetries);
 687     query.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries ? 1 : 0);
 688     query.bindValue(":rejoinchannels", info.rejoinChannels ? 1 : 0);
 689     // Custom rate limiting
 690     query.bindValue(":usecustomessagerate", info.useCustomMessageRate ? 1 : 0);
 691     query.bindValue(":messagerateburstsize", info.messageRateBurstSize);
 692     query.bindValue(":messageratedelay", info.messageRateDelay);
 693     query.bindValue(":unlimitedmessagerate", info.unlimitedMessageRate ? 1 : 0);
 694     if (info.networkId.isValid())
 695         query.bindValue(":networkid", info.networkId.toInt());
 696 }
 697
 698
 699 void SqliteStorage::bindServerInfo(QSqlQuery &query, const Network::Server &server)
 700 {
 701     query.bindValue(":hostname", server.host);
 702     query.bindValue(":port", server.port);
 703     query.bindValue(":password", server.password);
 704     query.bindValue(":ssl", server.useSsl ? 1 : 0);
 705     query.bindValue(":sslversion", server.sslVersion);
 706     query.bindValue(":useproxy", server.useProxy ? 1 : 0);
 707     query.bindValue(":proxytype", server.proxyType);
 708     query.bindValue(":proxyhost", server.proxyHost);
 709     query.bindValue(":proxyport", server.proxyPort);
 710     query.bindValue(":proxyuser", server.proxyUser);
 711     query.bindValue(":proxypass", server.proxyPass);
 712     query.bindValue(":sslverify", server.sslVerify ? 1 : 0);
 713 }
 714
 715
 716 bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info)
 717 {
 718     QSqlDatabase db = logDb();
 719     bool error = false;
 720     db.transaction();
 721
 722     {
 723         QSqlQuery updateQuery(db);
 724         updateQuery.prepare(queryString("update_network"));
 725         updateQuery.bindValue(":userid", user.toInt());
 726         bindNetworkInfo(updateQuery, info);
 727
 728         lockForWrite();
 729         safeExec(updateQuery);
 730         if (!watchQuery(updateQuery) || updateQuery.numRowsAffected() != 1) {
 731             error = true;
 732             db.rollback();
 733         }
 734     }
 735     if (error) {
 736         unlock();
 737         return false;
 738     }
 739
 740     {
 741         QSqlQuery dropServersQuery(db);
 742         dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
 743         dropServersQuery.bindValue(":networkid", info.networkId.toInt());
 744         safeExec(dropServersQuery);
 745         if (!watchQuery(dropServersQuery)) {
 746             error = true;
 747             db.rollback();
 748         }
 749     }
 750     if (error) {
 751         unlock();
 752         return false;
 753     }
 754
 755     {
 756         QSqlQuery insertServersQuery(db);
 757         insertServersQuery.prepare(queryString("insert_server"));
 758         foreach(Network::Server server, info.serverList) {
 759             insertServersQuery.bindValue(":userid", user.toInt());
 760             insertServersQuery.bindValue(":networkid", info.networkId.toInt());
 761             bindServerInfo(insertServersQuery, server);
 762             safeExec(insertServersQuery);
 763             if (!watchQuery(insertServersQuery)) {
 764                 error = true;
 765                 db.rollback();
 766                 break;
 767             }
 768         }
 769     }
 770
 771     db.commit();
 772     unlock();
 773     return !error;
 774 }
 775
 776
 777 bool SqliteStorage::removeNetwork(UserId user, const NetworkId &networkId)
 778 {
 779     QSqlDatabase db = logDb();
 780     bool error = false;
 781     db.transaction();
 782
 783     {
 784         QSqlQuery deleteNetworkQuery(db);
 785         deleteNetworkQuery.prepare(queryString("delete_network"));
 786         deleteNetworkQuery.bindValue(":networkid", networkId.toInt());
 787         deleteNetworkQuery.bindValue(":userid", user.toInt());
 788         lockForWrite();
 789         safeExec(deleteNetworkQuery);
 790         if (!watchQuery(deleteNetworkQuery) || deleteNetworkQuery.numRowsAffected() != 1) {
 791             error = true;
 792             db.rollback();
 793         }
 794     }
 795     if (error) {
 796         unlock();
 797         return false;
 798     }
 799
 800     {
 801         QSqlQuery deleteBacklogQuery(db);
 802         deleteBacklogQuery.prepare(queryString("delete_backlog_for_network"));
 803         deleteBacklogQuery.bindValue(":networkid", networkId.toInt());
 804         safeExec(deleteBacklogQuery);
 805         if (!watchQuery(deleteBacklogQuery)) {
 806             db.rollback();
 807             error = true;
 808         }
 809     }
 810     if (error) {
 811         unlock();
 812         return false;
 813     }
 814
 815     {
 816         QSqlQuery deleteBuffersQuery(db);
 817         deleteBuffersQuery.prepare(queryString("delete_buffers_for_network"));
 818         deleteBuffersQuery.bindValue(":networkid", networkId.toInt());
 819         safeExec(deleteBuffersQuery);
 820         if (!watchQuery(deleteBuffersQuery)) {
 821             db.rollback();
 822             error = true;
 823         }
 824     }
 825     if (error) {
 826         unlock();
 827         return false;
 828     }
 829
 830     {
 831         QSqlQuery deleteServersQuery(db);
 832         deleteServersQuery.prepare(queryString("delete_ircservers_for_network"));
 833         deleteServersQuery.bindValue(":networkid", networkId.toInt());
 834         safeExec(deleteServersQuery);
 835         if (!watchQuery(deleteServersQuery)) {
 836             db.rollback();
 837             error = true;
 838         }
 839     }
 840     if (error) {
 841         unlock();
 842         return false;
 843     }
 844
 845     db.commit();
 846     unlock();
 847     return true;
 848 }
 849
 850
 851 QList<NetworkInfo> SqliteStorage::networks(UserId user)
 852 {
 853     QList<NetworkInfo> nets;
 854
 855     QSqlDatabase db = logDb();
 856     db.transaction();
 857
 858     {
 859         QSqlQuery networksQuery(db);
 860         networksQuery.prepare(queryString("select_networks_for_user"));
 861         networksQuery.bindValue(":userid", user.toInt());
 862
 863         QSqlQuery serversQuery(db);
 864         serversQuery.prepare(queryString("select_servers_for_network"));
 865
 866         lockForRead();
 867         safeExec(networksQuery);
 868         if (watchQuery(networksQuery)) {
 869             while (networksQuery.next()) {
 870                 NetworkInfo net;
 871                 net.networkId = networksQuery.value(0).toInt();
 872                 net.networkName = networksQuery.value(1).toString();
 873                 net.identity = networksQuery.value(2).toInt();
 874                 net.codecForServer = networksQuery.value(3).toString().toLatin1();
 875                 net.codecForEncoding = networksQuery.value(4).toString().toLatin1();
 876                 net.codecForDecoding = networksQuery.value(5).toString().toLatin1();
 877                 net.useRandomServer = networksQuery.value(6).toInt() == 1 ? true : false;
 878                 net.perform = networksQuery.value(7).toString().split("\n");
 879                 net.useAutoIdentify = networksQuery.value(8).toInt() == 1 ? true : false;
 880                 net.autoIdentifyService = networksQuery.value(9).toString();
 881                 net.autoIdentifyPassword = networksQuery.value(10).toString();
 882                 net.useAutoReconnect = networksQuery.value(11).toInt() == 1 ? true : false;
 883                 net.autoReconnectInterval = networksQuery.value(12).toUInt();
 884                 net.autoReconnectRetries = networksQuery.value(13).toInt();
 885                 net.unlimitedReconnectRetries = networksQuery.value(14).toInt() == 1 ? true : false;
 886                 net.rejoinChannels = networksQuery.value(15).toInt() == 1 ? true : false;
 887                 net.useSasl = networksQuery.value(16).toInt() == 1 ? true : false;
 888                 net.saslAccount = networksQuery.value(17).toString();
 889                 net.saslPassword = networksQuery.value(18).toString();
 890                 // Custom rate limiting
 891                 net.useCustomMessageRate = networksQuery.value(19).toInt() == 1 ? true : false;
 892                 net.messageRateBurstSize = networksQuery.value(20).toUInt();
 893                 net.messageRateDelay = networksQuery.value(21).toUInt();
 894                 net.unlimitedMessageRate = networksQuery.value(22).toInt() == 1 ? true : false;
 895
 896                 serversQuery.bindValue(":networkid", net.networkId.toInt());
 897                 safeExec(serversQuery);
 898                 if (!watchQuery(serversQuery)) {
 899                     nets.clear();
 900                     break;
 901                 }
 902                 else {
 903                     Network::ServerList servers;
 904                     while (serversQuery.next()) {
 905                         Network::Server server;
 906                         server.host = serversQuery.value(0).toString();
 907                         server.port = serversQuery.value(1).toUInt();
 908                         server.password = serversQuery.value(2).toString();
 909                         server.useSsl = serversQuery.value(3).toInt() == 1 ? true : false;
 910                         server.sslVersion = serversQuery.value(4).toInt();
 911                         server.useProxy = serversQuery.value(5).toInt() == 1 ? true : false;
 912                         server.proxyType = serversQuery.value(6).toInt();
 913                         server.proxyHost = serversQuery.value(7).toString();
 914                         server.proxyPort = serversQuery.value(8).toUInt();
 915                         server.proxyUser = serversQuery.value(9).toString();
 916                         server.proxyPass = serversQuery.value(10).toString();
 917                         server.sslVerify = serversQuery.value(11).toInt() == 1 ? true : false;
 918                         servers << server;
 919                     }
 920                     net.serverList = servers;
 921                     nets << net;
 922                 }
 923             }
 924         }
 925     }
 926     db.commit();
 927     unlock();
 928     return nets;
 929 }
 930
 931
 932 QList<NetworkId> SqliteStorage::connectedNetworks(UserId user)
 933 {
 934     QList<NetworkId> connectedNets;
 935
 936     QSqlDatabase db = logDb();
 937     db.transaction();
 938
 939     {
 940         QSqlQuery query(db);
 941         query.prepare(queryString("select_connected_networks"));
 942         query.bindValue(":userid", user.toInt());
 943         lockForRead();
 944         safeExec(query);
 945         watchQuery(query);
 946
 947         while (query.next()) {
 948             connectedNets << query.value(0).toInt();
 949         }
 950         db.commit();
 951     }
 952     unlock();
 953     return connectedNets;
 954 }
 955
 956
 957 void SqliteStorage::setNetworkConnected(UserId user, const NetworkId &networkId, bool isConnected)
 958 {
 959     QSqlDatabase db = logDb();
 960     db.transaction();
 961
 962     {
 963         QSqlQuery query(db);
 964         query.prepare(queryString("update_network_connected"));
 965         query.bindValue(":userid", user.toInt());
 966         query.bindValue(":networkid", networkId.toInt());
 967         query.bindValue(":connected", isConnected ? 1 : 0);
 968
 969         lockForWrite();
 970         safeExec(query);
 971         watchQuery(query);
 972         db.commit();
 973     }
 974     unlock();
 975 }
 976
 977
 978 QHash<QString, QString> SqliteStorage::persistentChannels(UserId user, const NetworkId &networkId)
 979 {
 980     QHash<QString, QString> persistentChans;
 981
 982     QSqlDatabase db = logDb();
 983     db.transaction();
 984     {
 985         QSqlQuery query(db);
 986         query.prepare(queryString("select_persistent_channels"));
 987         query.bindValue(":userid", user.toInt());
 988         query.bindValue(":networkid", networkId.toInt());
 989
 990         lockForRead();
 991         safeExec(query);
 992         watchQuery(query);
 993         while (query.next()) {
 994             persistentChans[query.value(0).toString()] = query.value(1).toString();
 995         }
 996     }
 997     unlock();
 998     return persistentChans;
 999 }
1000
1001
1002 void SqliteStorage::setChannelPersistent(UserId user, const NetworkId &networkId, const QString &channel, bool isJoined)
1003 {
1004     QSqlDatabase db = logDb();
1005     db.transaction();
1006
1007     {
1008         QSqlQuery query(db);
1009         query.prepare(queryString("update_buffer_persistent_channel"));
1010         query.bindValue(":userid", user.toInt());
1011         query.bindValue(":networkid", networkId.toInt());
1012         query.bindValue(":buffercname", channel.toLower());
1013         query.bindValue(":joined", isJoined ? 1 : 0);
1014
1015         lockForWrite();
1016         safeExec(query);
1017         watchQuery(query);
1018         db.commit();
1019     }
1020     unlock();
1021 }
1022
1023
1024 void SqliteStorage::setPersistentChannelKey(UserId user, const NetworkId &networkId, const QString &channel, const QString &key)
1025 {
1026     QSqlDatabase db = logDb();
1027     db.transaction();
1028
1029     {
1030         QSqlQuery query(db);
1031         query.prepare(queryString("update_buffer_set_channel_key"));
1032         query.bindValue(":userid", user.toInt());
1033         query.bindValue(":networkid", networkId.toInt());
1034         query.bindValue(":buffercname", channel.toLower());
1035         query.bindValue(":key", key);
1036
1037         lockForWrite();
1038         safeExec(query);
1039         watchQuery(query);
1040         db.commit();
1041     }
1042     unlock();
1043 }
1044
1045
1046 QString SqliteStorage::awayMessage(UserId user, NetworkId networkId)
1047 {
1048     QSqlDatabase db = logDb();
1049     db.transaction();
1050
1051     QString awayMsg;
1052     {
1053         QSqlQuery query(db);
1054         query.prepare(queryString("select_network_awaymsg"));
1055         query.bindValue(":userid", user.toInt());
1056         query.bindValue(":networkid", networkId.toInt());
1057
1058         lockForRead();
1059         safeExec(query);
1060         watchQuery(query);
1061         if (query.first())
1062             awayMsg = query.value(0).toString();
1063         db.commit();
1064     }
1065     unlock();
1066
1067     return awayMsg;
1068 }
1069
1070
1071 void SqliteStorage::setAwayMessage(UserId user, NetworkId networkId, const QString &awayMsg)
1072 {
1073     QSqlDatabase db = logDb();
1074     db.transaction();
1075
1076     {
1077         QSqlQuery query(db);
1078         query.prepare(queryString("update_network_set_awaymsg"));
1079         query.bindValue(":userid", user.toInt());
1080         query.bindValue(":networkid", networkId.toInt());
1081         query.bindValue(":awaymsg", awayMsg);
1082
1083         lockForWrite();
1084         safeExec(query);
1085         watchQuery(query);
1086         db.commit();
1087     }
1088     unlock();
1089 }
1090
1091
1092 QString SqliteStorage::userModes(UserId user, NetworkId networkId)
1093 {
1094     QSqlDatabase db = logDb();
1095     db.transaction();
1096
1097     QString modes;
1098     {
1099         QSqlQuery query(db);
1100         query.prepare(queryString("select_network_usermode"));
1101         query.bindValue(":userid", user.toInt());
1102         query.bindValue(":networkid", networkId.toInt());
1103
1104         lockForRead();
1105         safeExec(query);
1106         watchQuery(query);
1107         if (query.first())
1108             modes = query.value(0).toString();
1109         db.commit();
1110     }
1111     unlock();
1112
1113     return modes;
1114 }
1115
1116
1117 void SqliteStorage::setUserModes(UserId user, NetworkId networkId, const QString &userModes)
1118 {
1119     QSqlDatabase db = logDb();
1120     db.transaction();
1121
1122     {
1123         QSqlQuery query(db);
1124         query.prepare(queryString("update_network_set_usermode"));
1125         query.bindValue(":userid", user.toInt());
1126         query.bindValue(":networkid", networkId.toInt());
1127         query.bindValue(":usermode", userModes);
1128
1129         lockForWrite();
1130         safeExec(query);
1131         watchQuery(query);
1132         db.commit();
1133     }
1134     unlock();
1135 }
1136
1137
1138 BufferInfo SqliteStorage::bufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer, bool create)
1139 {
1140     QSqlDatabase db = logDb();
1141     db.transaction();
1142
1143     BufferInfo bufferInfo;
1144     {
1145         QSqlQuery query(db);
1146         query.prepare(queryString("select_bufferByName"));
1147         query.bindValue(":networkid", networkId.toInt());
1148         query.bindValue(":userid", user.toInt());
1149         query.bindValue(":buffercname", buffer.toLower());
1150
1151         lockForRead();
1152         safeExec(query);
1153
1154         if (query.first()) {
1155             bufferInfo = BufferInfo(query.value(0).toInt(), networkId, (BufferInfo::Type)query.value(1).toInt(), 0, buffer);
1156             if (query.next()) {
1157                 qCritical() << "SqliteStorage::getBufferInfo(): received more then one Buffer!";
1158                 qCritical() << "         Query:" << query.lastQuery();
1159                 qCritical() << "  bound Values:";
1160                 QList<QVariant> list = query.boundValues().values();
1161                 for (int i = 0; i < list.size(); ++i)
1162                     qCritical() << i << ":" << list.at(i).toString().toLatin1().data();
1163                 Q_ASSERT(false);
1164             }
1165         }
1166         else if (create) {
1167             // let's create the buffer
1168             QSqlQuery createQuery(db);
1169             createQuery.prepare(queryString("insert_buffer"));
1170             createQuery.bindValue(":userid", user.toInt());
1171             createQuery.bindValue(":networkid", networkId.toInt());
1172             createQuery.bindValue(":buffertype", (int)type);
1173             createQuery.bindValue(":buffername", buffer);
1174             createQuery.bindValue(":buffercname", buffer.toLower());
1175             createQuery.bindValue(":joined", type & BufferInfo::ChannelBuffer ? 1 : 0);
1176
1177             unlock();
1178             lockForWrite();
1179             safeExec(createQuery);
1180             watchQuery(createQuery);
1181             bufferInfo = BufferInfo(createQuery.lastInsertId().toInt(), networkId, type, 0, buffer);
1182         }
1183     }
1184     db.commit();
1185     unlock();
1186     return bufferInfo;
1187 }
1188
1189
1190 BufferInfo SqliteStorage::getBufferInfo(UserId user, const BufferId &bufferId)
1191 {
1192     QSqlDatabase db = logDb();
1193     db.transaction();
1194
1195     BufferInfo bufferInfo;
1196     {
1197         QSqlQuery query(db);
1198         query.prepare(queryString("select_buffer_by_id"));
1199         query.bindValue(":userid", user.toInt());
1200         query.bindValue(":bufferid", bufferId.toInt());
1201
1202         lockForRead();
1203         safeExec(query);
1204
1205         if (watchQuery(query) && query.first()) {
1206             bufferInfo = BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), 0, query.value(4).toString());
1207             Q_ASSERT(!query.next());
1208         }
1209         db.commit();
1210     }
1211     unlock();
1212     return bufferInfo;
1213 }
1214
1215
1216 QList<BufferInfo> SqliteStorage::requestBuffers(UserId user)
1217 {
1218     QList<BufferInfo> bufferlist;
1219
1220     QSqlDatabase db = logDb();
1221     db.transaction();
1222
1223     {
1224         QSqlQuery query(db);
1225         query.prepare(queryString("select_buffers"));
1226         query.bindValue(":userid", user.toInt());
1227
1228         lockForRead();
1229         safeExec(query);
1230         watchQuery(query);
1231         while (query.next()) {
1232             bufferlist << BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), query.value(3).toInt(), query.value(4).toString());
1233         }
1234         db.commit();
1235     }
1236     unlock();
1237
1238     return bufferlist;
1239 }
1240
1241
1242 QList<BufferId> SqliteStorage::requestBufferIdsForNetwork(UserId user, NetworkId networkId)
1243 {
1244     QList<BufferId> bufferList;
1245
1246     QSqlDatabase db = logDb();
1247     db.transaction();
1248
1249     {
1250         QSqlQuery query(db);
1251         query.prepare(queryString("select_buffers_for_network"));
1252         query.bindValue(":networkid", networkId.toInt());
1253         query.bindValue(":userid", user.toInt());
1254
1255         lockForRead();
1256         safeExec(query);
1257         watchQuery(query);
1258         while (query.next()) {
1259             bufferList << BufferId(query.value(0).toInt());
1260         }
1261         db.commit();
1262     }
1263     unlock();
1264
1265     return bufferList;
1266 }
1267
1268
1269 bool SqliteStorage::removeBuffer(const UserId &user, const BufferId &bufferId)
1270 {
1271     QSqlDatabase db = logDb();
1272     db.transaction();
1273
1274     bool error = false;
1275     {
1276         QSqlQuery delBufferQuery(db);
1277         delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
1278         delBufferQuery.bindValue(":bufferid", bufferId.toInt());
1279         delBufferQuery.bindValue(":userid", user.toInt());
1280
1281         lockForWrite();
1282         safeExec(delBufferQuery);
1283
1284         error = (!watchQuery(delBufferQuery) || delBufferQuery.numRowsAffected() != 1);
1285     }
1286
1287     if (error) {
1288         db.rollback();
1289         unlock();
1290         return false;
1291     }
1292
1293     {
1294         QSqlQuery delBacklogQuery(db);
1295         delBacklogQuery.prepare(queryString("delete_backlog_for_buffer"));
1296         delBacklogQuery.bindValue(":bufferid", bufferId.toInt());
1297
1298         safeExec(delBacklogQuery);
1299         error = !watchQuery(delBacklogQuery);
1300     }
1301
1302     if (error) {
1303         db.rollback();
1304     }
1305     else {
1306         db.commit();
1307     }
1308     unlock();
1309     return !error;
1310 }
1311
1312
1313 bool SqliteStorage::renameBuffer(const UserId &user, const BufferId &bufferId, const QString &newName)
1314 {
1315     QSqlDatabase db = logDb();
1316     db.transaction();
1317
1318     bool error = false;
1319     {
1320         QSqlQuery query(db);
1321         query.prepare(queryString("update_buffer_name"));
1322         query.bindValue(":buffername", newName);
1323         query.bindValue(":buffercname", newName.toLower());
1324         query.bindValue(":bufferid", bufferId.toInt());
1325         query.bindValue(":userid", user.toInt());
1326
1327         lockForWrite();
1328         safeExec(query);
1329
1330         error = query.lastError().isValid();
1331         // unexepcted error occured (19 == constraint violation)
1332         if (error && query.lastError().number() != 19) {
1333             watchQuery(query);
1334         }
1335         else {
1336             error |= (query.numRowsAffected() != 1);
1337         }
1338     }
1339     if (error) {
1340         db.rollback();
1341     }
1342     else {
1343         db.commit();
1344     }
1345     unlock();
1346     return !error;
1347 }
1348
1349
1350 bool SqliteStorage::mergeBuffersPermanently(const UserId &user, const BufferId &bufferId1, const BufferId &bufferId2)
1351 {
1352     QSqlDatabase db = logDb();
1353     db.transaction();
1354
1355     bool error = false;
1356     {
1357         QSqlQuery checkQuery(db);
1358         checkQuery.prepare(queryString("select_buffers_for_merge"));
1359         checkQuery.bindValue(":oldbufferid", bufferId2.toInt());
1360         checkQuery.bindValue(":newbufferid", bufferId1.toInt());
1361         checkQuery.bindValue(":userid", user.toInt());
1362
1363         lockForRead();
1364         safeExec(checkQuery);
1365         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 2);
1366     }
1367     if (error) {
1368         db.rollback();
1369         unlock();
1370         return false;
1371     }
1372
1373     {
1374         QSqlQuery query(db);
1375         query.prepare(queryString("update_backlog_bufferid"));
1376         query.bindValue(":oldbufferid", bufferId2.toInt());
1377         query.bindValue(":newbufferid", bufferId1.toInt());
1378         safeExec(query);
1379         error = !watchQuery(query);
1380     }
1381     if (error) {
1382         db.rollback();
1383         unlock();
1384         return false;
1385     }
1386
1387     {
1388         QSqlQuery delBufferQuery(db);
1389         delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
1390         delBufferQuery.bindValue(":bufferid", bufferId2.toInt());
1391         delBufferQuery.bindValue(":userid", user.toInt());
1392         safeExec(delBufferQuery);
1393         error = !watchQuery(delBufferQuery);
1394     }
1395
1396     if (error) {
1397         db.rollback();
1398     }
1399     else {
1400         db.commit();
1401     }
1402     unlock();
1403     return !error;
1404 }
1405
1406
1407 void SqliteStorage::setBufferLastSeenMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1408 {
1409     QSqlDatabase db = logDb();
1410     db.transaction();
1411
1412     {
1413         QSqlQuery query(db);
1414         query.prepare(queryString("update_buffer_lastseen"));
1415         query.bindValue(":userid", user.toInt());
1416         query.bindValue(":bufferid", bufferId.toInt());
1417         query.bindValue(":lastseenmsgid", msgId.toInt());
1418
1419         lockForWrite();
1420         safeExec(query);
1421         watchQuery(query);
1422     }
1423     db.commit();
1424     unlock();
1425 }
1426
1427
1428 QHash<BufferId, MsgId> SqliteStorage::bufferLastSeenMsgIds(UserId user)
1429 {
1430     QHash<BufferId, MsgId> lastSeenHash;
1431
1432     QSqlDatabase db = logDb();
1433     db.transaction();
1434
1435     bool error = false;
1436     {
1437         QSqlQuery query(db);
1438         query.prepare(queryString("select_buffer_lastseen_messages"));
1439         query.bindValue(":userid", user.toInt());
1440
1441         lockForRead();
1442         safeExec(query);
1443         error = !watchQuery(query);
1444         if (!error) {
1445             while (query.next()) {
1446                 lastSeenHash[query.value(0).toInt()] = query.value(1).toInt();
1447             }
1448         }
1449     }
1450
1451     db.commit();
1452     unlock();
1453     return lastSeenHash;
1454 }
1455
1456
1457 void SqliteStorage::setBufferMarkerLineMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1458 {
1459     QSqlDatabase db = logDb();
1460     db.transaction();
1461
1462     {
1463         QSqlQuery query(db);
1464         query.prepare(queryString("update_buffer_markerlinemsgid"));
1465         query.bindValue(":userid", user.toInt());
1466         query.bindValue(":bufferid", bufferId.toInt());
1467         query.bindValue(":markerlinemsgid", msgId.toInt());
1468
1469         lockForWrite();
1470         safeExec(query);
1471         watchQuery(query);
1472     }
1473     db.commit();
1474     unlock();
1475 }
1476
1477
1478 QHash<BufferId, MsgId> SqliteStorage::bufferMarkerLineMsgIds(UserId user)
1479 {
1480     QHash<BufferId, MsgId> markerLineHash;
1481
1482     QSqlDatabase db = logDb();
1483     db.transaction();
1484
1485     bool error = false;
1486     {
1487         QSqlQuery query(db);
1488         query.prepare(queryString("select_buffer_markerlinemsgids"));
1489         query.bindValue(":userid", user.toInt());
1490
1491         lockForRead();
1492         safeExec(query);
1493         error = !watchQuery(query);
1494         if (!error) {
1495             while (query.next()) {
1496                 markerLineHash[query.value(0).toInt()] = query.value(1).toInt();
1497             }
1498         }
1499     }
1500
1501     db.commit();
1502     unlock();
1503     return markerLineHash;
1504 }
1505
1506 bool SqliteStorage::logMessage(Message &msg)
1507 {
1508     QSqlDatabase db = logDb();
1509     db.transaction();
1510
1511     bool error = false;
1512     {
1513         QSqlQuery logMessageQuery(db);
1514         logMessageQuery.prepare(queryString("insert_message"));
1515
1516         logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
1517         logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1518         logMessageQuery.bindValue(":type", msg.type());
1519         logMessageQuery.bindValue(":flags", (int)msg.flags());
1520         logMessageQuery.bindValue(":sender", msg.sender());
1521         logMessageQuery.bindValue(":message", msg.contents());
1522
1523         lockForWrite();
1524         safeExec(logMessageQuery);
1525
1526         if (logMessageQuery.lastError().isValid()) {
1527             // constraint violation - must be NOT NULL constraint - probably the sender is missing...
1528             if (logMessageQuery.lastError().number() == 19) {
1529                 QSqlQuery addSenderQuery(db);
1530                 addSenderQuery.prepare(queryString("insert_sender"));
1531                 addSenderQuery.bindValue(":sender", msg.sender());
1532                 safeExec(addSenderQuery);
1533                 safeExec(logMessageQuery);
1534                 error = !watchQuery(logMessageQuery);
1535             }
1536             else {
1537                 watchQuery(logMessageQuery);
1538             }
1539         }
1540         if (!error) {
1541             MsgId msgId = logMessageQuery.lastInsertId().toInt();
1542             if (msgId.isValid()) {
1543                 msg.setMsgId(msgId);
1544             }
1545             else {
1546                 error = true;
1547             }
1548         }
1549     }
1550
1551     if (error) {
1552         db.rollback();
1553     }
1554     else {
1555         db.commit();
1556     }
1557
1558     unlock();
1559     return !error;
1560 }
1561
1562
1563 bool SqliteStorage::logMessages(MessageList &msgs)
1564 {
1565     QSqlDatabase db = logDb();
1566     db.transaction();
1567
1568     {
1569         QSet<QString> senders;
1570         QSqlQuery addSenderQuery(db);
1571         addSenderQuery.prepare(queryString("insert_sender"));
1572         lockForWrite();
1573         for (int i = 0; i < msgs.count(); i++) {
1574             const QString &sender = msgs.at(i).sender();
1575             if (senders.contains(sender))
1576                 continue;
1577             senders << sender;
1578
1579             addSenderQuery.bindValue(":sender", sender);
1580             safeExec(addSenderQuery);
1581         }
1582     }
1583
1584     bool error = false;
1585     {
1586         QSqlQuery logMessageQuery(db);
1587         logMessageQuery.prepare(queryString("insert_message"));
1588         for (int i = 0; i < msgs.count(); i++) {
1589             Message &msg = msgs[i];
1590
1591             logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
1592             logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1593             logMessageQuery.bindValue(":type", msg.type());
1594             logMessageQuery.bindValue(":flags", (int)msg.flags());
1595             logMessageQuery.bindValue(":sender", msg.sender());
1596             logMessageQuery.bindValue(":message", msg.contents());
1597
1598             safeExec(logMessageQuery);
1599             if (!watchQuery(logMessageQuery)) {
1600                 error = true;
1601                 break;
1602             }
1603             else {
1604                 msg.setMsgId(logMessageQuery.lastInsertId().toInt());
1605             }
1606         }
1607     }
1608
1609     if (error) {
1610         db.rollback();
1611         unlock();
1612         // we had a rollback in the db so we need to reset all msgIds
1613         for (int i = 0; i < msgs.count(); i++) {
1614             msgs[i].setMsgId(MsgId());
1615         }
1616     }
1617     else {
1618         db.commit();
1619         unlock();
1620     }
1621     return !error;
1622 }
1623
1624
1625 QList<Message> SqliteStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit)
1626 {
1627     QList<Message> messagelist;
1628
1629     QSqlDatabase db = logDb();
1630     db.transaction();
1631
1632     bool error = false;
1633     BufferInfo bufferInfo;
1634     {
1635         // code dupication from getBufferInfo:
1636         // this is due to the impossibility of nesting transactions and recursive locking
1637         QSqlQuery bufferInfoQuery(db);
1638         bufferInfoQuery.prepare(queryString("select_buffer_by_id"));
1639         bufferInfoQuery.bindValue(":userid", user.toInt());
1640         bufferInfoQuery.bindValue(":bufferid", bufferId.toInt());
1641
1642         lockForRead();
1643         safeExec(bufferInfoQuery);
1644         error = !watchQuery(bufferInfoQuery) || !bufferInfoQuery.first();
1645         if (!error) {
1646             bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), 0, bufferInfoQuery.value(4).toString());
1647             error = !bufferInfo.isValid();
1648         }
1649     }
1650     if (error) {
1651         db.rollback();
1652         unlock();
1653         return messagelist;
1654     }
1655
1656     {
1657         QSqlQuery query(db);
1658         if (last == -1 && first == -1) {
1659             query.prepare(queryString("select_messagesNewestK"));
1660         }
1661         else if (last == -1) {
1662             query.prepare(queryString("select_messagesNewerThan"));
1663             query.bindValue(":firstmsg", first.toInt());
1664         }
1665         else {
1666             query.prepare(queryString("select_messagesRange"));
1667             query.bindValue(":lastmsg", last.toInt());
1668             query.bindValue(":firstmsg", first.toInt());
1669         }
1670         query.bindValue(":bufferid", bufferId.toInt());
1671         query.bindValue(":limit", limit);
1672
1673         safeExec(query);
1674         watchQuery(query);
1675
1676         while (query.next()) {
1677             Message msg(QDateTime::fromTime_t(query.value(1).toInt()),
1678                 bufferInfo,
1679                 (Message::Type)query.value(2).toUInt(),
1680                 query.value(5).toString(),
1681                 query.value(4).toString(),
1682                 (Message::Flags)query.value(3).toUInt());
1683             msg.setMsgId(query.value(0).toInt());
1684             messagelist << msg;
1685         }
1686     }
1687     db.commit();
1688     unlock();
1689
1690     return messagelist;
1691 }
1692
1693
1694 QList<Message> SqliteStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit)
1695 {
1696     QList<Message> messagelist;
1697
1698     QSqlDatabase db = logDb();
1699     db.transaction();
1700
1701     QHash<BufferId, BufferInfo> bufferInfoHash;
1702     {
1703         QSqlQuery bufferInfoQuery(db);
1704         bufferInfoQuery.prepare(queryString("select_buffers"));
1705         bufferInfoQuery.bindValue(":userid", user.toInt());
1706
1707         lockForRead();
1708         safeExec(bufferInfoQuery);
1709         watchQuery(bufferInfoQuery);
1710         while (bufferInfoQuery.next()) {
1711             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());
1712             bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
1713         }
1714
1715         QSqlQuery query(db);
1716         if (last == -1) {
1717             query.prepare(queryString("select_messagesAllNew"));
1718         }
1719         else {
1720             query.prepare(queryString("select_messagesAll"));
1721             query.bindValue(":lastmsg", last.toInt());
1722         }
1723         query.bindValue(":userid", user.toInt());
1724         query.bindValue(":firstmsg", first.toInt());
1725         query.bindValue(":limit", limit);
1726         safeExec(query);
1727
1728         watchQuery(query);
1729
1730         while (query.next()) {
1731             Message msg(QDateTime::fromTime_t(query.value(2).toInt()),
1732                 bufferInfoHash[query.value(1).toInt()],
1733                 (Message::Type)query.value(3).toUInt(),
1734                 query.value(6).toString(),
1735                 query.value(5).toString(),
1736                 (Message::Flags)query.value(4).toUInt());
1737             msg.setMsgId(query.value(0).toInt());
1738             messagelist << msg;
1739         }
1740     }
1741     db.commit();
1742     unlock();
1743     return messagelist;
1744 }
1745
1746
1747 QString SqliteStorage::backlogFile()
1748 {
1749     return Quassel::configDirPath() + "quassel-storage.sqlite";
1750 }
1751
1752
1753 bool SqliteStorage::safeExec(QSqlQuery &query, int retryCount)
1754 {
1755     query.exec();
1756
1757     if (!query.lastError().isValid())
1758         return true;
1759
1760     switch (query.lastError().number()) {
1761     case 5: // SQLITE_BUSY         5   /* The database file is locked */
1762     case 6: // SQLITE_LOCKED       6   /* A table in the database is locked */
1763         if (retryCount < _maxRetryCount)
1764             return safeExec(query, retryCount + 1);
1765     default:
1766         return false;
1767     }
1768 }
1769
1770
1771 // ========================================
1772 //  SqliteMigration
1773 // ========================================
1774 SqliteMigrationReader::SqliteMigrationReader()
1775     : SqliteStorage(),
1776     _maxId(0)
1777 {
1778 }
1779
1780
1781 void SqliteMigrationReader::setMaxId(MigrationObject mo)
1782 {
1783     QString queryString;
1784     switch (mo) {
1785     case Sender:
1786         queryString = "SELECT max(senderid) FROM sender";
1787         break;
1788     case Backlog:
1789         queryString = "SELECT max(messageid) FROM backlog";
1790         break;
1791     default:
1792         _maxId = 0;
1793         return;
1794     }
1795     QSqlQuery query = logDb().exec(queryString);
1796     query.first();
1797     _maxId = query.value(0).toInt();
1798 }
1799
1800
1801 bool SqliteMigrationReader::prepareQuery(MigrationObject mo)
1802 {
1803     setMaxId(mo);
1804
1805     switch (mo) {
1806     case QuasselUser:
1807         newQuery(queryString("migrate_read_quasseluser"), logDb());
1808         break;
1809     case Identity:
1810         newQuery(queryString("migrate_read_identity"), logDb());
1811         break;
1812     case IdentityNick:
1813         newQuery(queryString("migrate_read_identity_nick"), logDb());
1814         break;
1815     case Network:
1816         newQuery(queryString("migrate_read_network"), logDb());
1817         break;
1818     case Buffer:
1819         newQuery(queryString("migrate_read_buffer"), logDb());
1820         break;
1821     case Sender:
1822         newQuery(queryString("migrate_read_sender"), logDb());
1823         bindValue(0, 0);
1824         bindValue(1, stepSize());
1825         break;
1826     case Backlog:
1827         newQuery(queryString("migrate_read_backlog"), logDb());
1828         bindValue(0, 0);
1829         bindValue(1, stepSize());
1830         break;
1831     case IrcServer:
1832         newQuery(queryString("migrate_read_ircserver"), logDb());
1833         break;
1834     case UserSetting:
1835         newQuery(queryString("migrate_read_usersetting"), logDb());
1836         break;
1837     }
1838     return exec();
1839 }
1840
1841
1842 bool SqliteMigrationReader::readMo(QuasselUserMO &user)
1843 {
1844     if (!next())
1845         return false;
1846
1847     user.id = value(0).toInt();
1848     user.username = value(1).toString();
1849     user.password = value(2).toString();
1850     user.hashversion = value(3).toInt();
1851     user.authenticator = value(4).toString();
1852     return true;
1853 }
1854
1855
1856 bool SqliteMigrationReader::readMo(IdentityMO &identity)
1857 {
1858     if (!next())
1859         return false;
1860
1861     identity.id = value(0).toInt();
1862     identity.userid = value(1).toInt();
1863     identity.identityname = value(2).toString();
1864     identity.realname = value(3).toString();
1865     identity.awayNick = value(4).toString();
1866     identity.awayNickEnabled = value(5).toInt() == 1 ? true : false;
1867     identity.awayReason = value(6).toString();
1868     identity.awayReasonEnabled = value(7).toInt() == 1 ? true : false;
1869     identity.autoAwayEnabled = value(8).toInt() == 1 ? true : false;
1870     identity.autoAwayTime = value(9).toInt();
1871     identity.autoAwayReason = value(10).toString();
1872     identity.autoAwayReasonEnabled = value(11).toInt() == 1 ? true : false;
1873     identity.detachAwayEnabled = value(12).toInt() == 1 ? true : false;
1874     identity.detachAwayReason = value(13).toString();
1875     identity.detchAwayReasonEnabled = value(14).toInt() == 1 ? true : false;
1876     identity.ident = value(15).toString();
1877     identity.kickReason = value(16).toString();
1878     identity.partReason = value(17).toString();
1879     identity.quitReason = value(18).toString();
1880     identity.sslCert = value(19).toByteArray();
1881     identity.sslKey = value(20).toByteArray();
1882     return true;
1883 }
1884
1885
1886 bool SqliteMigrationReader::readMo(IdentityNickMO &identityNick)
1887 {
1888     if (!next())
1889         return false;
1890
1891     identityNick.nickid = value(0).toInt();
1892     identityNick.identityId = value(1).toInt();
1893     identityNick.nick = value(2).toString();
1894     return true;
1895 }
1896
1897
1898 bool SqliteMigrationReader::readMo(NetworkMO &network)
1899 {
1900     if (!next())
1901         return false;
1902
1903     network.networkid = value(0).toInt();
1904     network.userid = value(1).toInt();
1905     network.networkname = value(2).toString();
1906     network.identityid = value(3).toInt();
1907     network.encodingcodec = value(4).toString();
1908     network.decodingcodec = value(5).toString();
1909     network.servercodec = value(6).toString();
1910     network.userandomserver = value(7).toInt() == 1 ? true : false;
1911     network.perform = value(8).toString();
1912     network.useautoidentify = value(9).toInt() == 1 ? true : false;
1913     network.autoidentifyservice = value(10).toString();
1914     network.autoidentifypassword = value(11).toString();
1915     network.useautoreconnect = value(12).toInt() == 1 ? true : false;
1916     network.autoreconnectinterval = value(13).toInt();
1917     network.autoreconnectretries = value(14).toInt();
1918     network.unlimitedconnectretries = value(15).toInt() == 1 ? true : false;
1919     network.rejoinchannels = value(16).toInt() == 1 ? true : false;
1920     network.connected = value(17).toInt() == 1 ? true : false;
1921     network.usermode = value(18).toString();
1922     network.awaymessage = value(19).toString();
1923     network.attachperform = value(20).toString();
1924     network.detachperform = value(21).toString();
1925     network.usesasl = value(22).toInt() == 1 ? true : false;
1926     network.saslaccount = value(23).toString();
1927     network.saslpassword = value(24).toString();
1928     // Custom rate limiting
1929     network.usecustommessagerate = value(25).toInt() == 1 ? true : false;
1930     network.messagerateburstsize = value(26).toInt();
1931     network.messageratedelay = value(27).toUInt();
1932     network.unlimitedmessagerate = value(28).toInt() == 1 ? true : false;
1933     return true;
1934 }
1935
1936
1937 bool SqliteMigrationReader::readMo(BufferMO &buffer)
1938 {
1939     if (!next())
1940         return false;
1941
1942     buffer.bufferid = value(0).toInt();
1943     buffer.userid = value(1).toInt();
1944     buffer.groupid = value(2).toInt();
1945     buffer.networkid = value(3).toInt();
1946     buffer.buffername = value(4).toString();
1947     buffer.buffercname = value(5).toString();
1948     buffer.buffertype = value(6).toInt();
1949     buffer.lastmsgid = value(7).toInt();
1950     buffer.lastseenmsgid = value(8).toInt();
1951     buffer.markerlinemsgid = value(9).toInt();
1952     buffer.key = value(10).toString();
1953     buffer.joined = value(11).toInt() == 1 ? true : false;
1954     return true;
1955 }
1956
1957
1958 bool SqliteMigrationReader::readMo(SenderMO &sender)
1959 {
1960     int skipSteps = 0;
1961     while (!next()) {
1962         if (sender.senderId < _maxId) {
1963             bindValue(0, sender.senderId + (skipSteps * stepSize()));
1964             bindValue(1, sender.senderId + ((skipSteps + 1) * stepSize()));
1965             skipSteps++;
1966             if (!exec())
1967                 return false;
1968         }
1969         else {
1970             return false;
1971         }
1972     }
1973
1974     sender.senderId = value(0).toInt();
1975     sender.sender = value(1).toString();
1976     return true;
1977 }
1978
1979
1980 bool SqliteMigrationReader::readMo(BacklogMO &backlog)
1981 {
1982     int skipSteps = 0;
1983     while (!next()) {
1984         if (backlog.messageid < _maxId) {
1985             bindValue(0, backlog.messageid.toInt() + (skipSteps * stepSize()));
1986             bindValue(1, backlog.messageid.toInt() + ((skipSteps + 1) * stepSize()));
1987             skipSteps++;
1988             if (!exec())
1989                 return false;
1990         }
1991         else {
1992             return false;
1993         }
1994     }
1995
1996     backlog.messageid = value(0).toInt();
1997     backlog.time = QDateTime::fromTime_t(value(1).toInt()).toUTC();
1998     backlog.bufferid = value(2).toInt();
1999     backlog.type = value(3).toInt();
2000     backlog.flags = value(4).toInt();
2001     backlog.senderid = value(5).toInt();
2002     backlog.message = value(6).toString();
2003     return true;
2004 }
2005
2006
2007 bool SqliteMigrationReader::readMo(IrcServerMO &ircserver)
2008 {
2009     if (!next())
2010         return false;
2011
2012     ircserver.serverid = value(0).toInt();
2013     ircserver.userid = value(1).toInt();
2014     ircserver.networkid = value(2).toInt();
2015     ircserver.hostname = value(3).toString();
2016     ircserver.port = value(4).toInt();
2017     ircserver.password = value(5).toString();
2018     ircserver.ssl = value(6).toInt() == 1 ? true : false;
2019     ircserver.sslversion = value(7).toInt();
2020     ircserver.useproxy = value(8).toInt() == 1 ? true : false;
2021     ircserver.proxytype = value(9).toInt();
2022     ircserver.proxyhost = value(10).toString();
2023     ircserver.proxyport = value(11).toInt();
2024     ircserver.proxyuser = value(12).toString();
2025     ircserver.proxypass = value(13).toString();
2026     ircserver.sslverify = value(14).toInt() == 1 ? true : false;
2027     return true;
2028 }
2029
2030
2031 bool SqliteMigrationReader::readMo(UserSettingMO &userSetting)
2032 {
2033     if (!next())
2034         return false;
2035
2036     userSetting.userid = value(0).toInt();
2037     userSetting.settingname = value(1).toString();
2038     userSetting.settingvalue = value(2).toByteArray();
2039
2040     return true;
2041 }