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::displayName() const
  50 {
  51     // We identify the backend to use for the monolithic core by its displayname.
  52     // so only change this string if you _really_ have to and make sure the core
  53     // setup for the mono client still works ;)
  54     return QString("SQLite");
  55 }
  56
  57
  58 QString SqliteStorage::description() const
  59 {
  60     return tr("SQLite is a file-based database engine that does not require any setup. It is suitable for small and medium-sized "
  61               "databases that do not require access via network. Use SQLite if your Quassel Core should store its data on the same machine "
  62               "it is running on, and if you only expect a few users to use your core.");
  63 }
  64
  65
  66 int SqliteStorage::installedSchemaVersion()
  67 {
  68     // only used when there is a singlethread (during startup)
  69     // so we don't need locking here
  70     QSqlQuery query = logDb().exec("SELECT value FROM coreinfo WHERE key = 'schemaversion'");
  71     if (query.first())
  72         return query.value(0).toInt();
  73
  74     // maybe it's really old... (schema version 0)
  75     query = logDb().exec("SELECT MAX(version) FROM coreinfo");
  76     if (query.first())
  77         return query.value(0).toInt();
  78
  79     return AbstractSqlStorage::installedSchemaVersion();
  80 }
  81
  82
  83 bool SqliteStorage::updateSchemaVersion(int newVersion)
  84 {
  85     // only used when there is a singlethread (during startup)
  86     // so we don't need locking here
  87     QSqlQuery query(logDb());
  88     query.prepare("UPDATE coreinfo SET value = :version WHERE key = 'schemaversion'");
  89     query.bindValue(":version", newVersion);
  90     query.exec();
  91
  92     bool success = true;
  93     if (query.lastError().isValid()) {
  94         qCritical() << "SqliteStorage::updateSchemaVersion(int): Updating schema version failed!";
  95         success = false;
  96     }
  97     return success;
  98 }
  99
 100
 101 bool SqliteStorage::setupSchemaVersion(int version)
 102 {
 103     // only used when there is a singlethread (during startup)
 104     // so we don't need locking here
 105     QSqlQuery query(logDb());
 106     query.prepare("INSERT INTO coreinfo (key, value) VALUES ('schemaversion', :version)");
 107     query.bindValue(":version", version);
 108     query.exec();
 109
 110     bool success = true;
 111     if (query.lastError().isValid()) {
 112         qCritical() << "SqliteStorage::setupSchemaVersion(int): Updating schema version failed!";
 113         success = false;
 114     }
 115     return success;
 116 }
 117
 118
 119 UserId SqliteStorage::addUser(const QString &user, const QString &password)
 120 {
 121     QSqlDatabase db = logDb();
 122     UserId uid;
 123
 124     db.transaction();
 125     // this scope ensures that the query is freed in sqlite before we call unlock()
 126     // this ensures that our thread doesn't hold a internal after unlock is called
 127     // (see sqlites doc on implicit locking for details)
 128     {
 129         QSqlQuery query(db);
 130         query.prepare(queryString("insert_quasseluser"));
 131         query.bindValue(":username", user);
 132         query.bindValue(":password", hashPassword(password));
 133         query.bindValue(":hashversion", Storage::HashVersion::Latest);
 134         lockForWrite();
 135         safeExec(query);
 136         if (query.lastError().isValid() && query.lastError().number() == 19) { // user already exists - sadly 19 seems to be the general constraint violation error...
 137             db.rollback();
 138         }
 139         else {
 140             uid = query.lastInsertId().toInt();
 141             db.commit();
 142         }
 143     }
 144     unlock();
 145
 146     if (uid.isValid())
 147         emit userAdded(uid, user);
 148     return uid;
 149 }
 150
 151
 152 bool SqliteStorage::updateUser(UserId user, const QString &password)
 153 {
 154     QSqlDatabase db = logDb();
 155     bool success = false;
 156
 157     db.transaction();
 158     {
 159         QSqlQuery query(db);
 160         query.prepare(queryString("update_userpassword"));
 161         query.bindValue(":userid", user.toInt());
 162         query.bindValue(":password", hashPassword(password));
 163         query.bindValue(":hashversion", Storage::HashVersion::Latest);
 164         lockForWrite();
 165         safeExec(query);
 166         success = query.numRowsAffected() != 0;
 167         db.commit();
 168     }
 169     unlock();
 170     return success;
 171 }
 172
 173
 174 void SqliteStorage::renameUser(UserId user, const QString &newName)
 175 {
 176     QSqlDatabase db = logDb();
 177     db.transaction();
 178     {
 179         QSqlQuery query(db);
 180         query.prepare(queryString("update_username"));
 181         query.bindValue(":userid", user.toInt());
 182         query.bindValue(":username", newName);
 183         lockForWrite();
 184         safeExec(query);
 185         db.commit();
 186     }
 187     unlock();
 188     emit userRenamed(user, newName);
 189 }
 190
 191
 192 UserId SqliteStorage::validateUser(const QString &user, const QString &password)
 193 {
 194     UserId userId;
 195     QString hashedPassword;
 196     Storage::HashVersion hashVersion = Storage::HashVersion::Latest;
 197
 198     {
 199         QSqlQuery query(logDb());
 200         query.prepare(queryString("select_authuser"));
 201         query.bindValue(":username", user);
 202
 203         lockForRead();
 204         safeExec(query);
 205
 206         if (query.first()) {
 207             userId = query.value(0).toInt();
 208             hashedPassword = query.value(1).toString();
 209             hashVersion = static_cast<Storage::HashVersion>(query.value(2).toInt());
 210         }
 211     }
 212     unlock();
 213
 214     UserId returnUserId;
 215     if (userId != 0 && checkHashedPassword(userId, password, hashedPassword, hashVersion)) {
 216         returnUserId = userId;
 217     }
 218     return returnUserId;
 219 }
 220
 221
 222 UserId SqliteStorage::getUserId(const QString &username)
 223 {
 224     UserId userId;
 225
 226     {
 227         QSqlQuery query(logDb());
 228         query.prepare(queryString("select_userid"));
 229         query.bindValue(":username", username);
 230
 231         lockForRead();
 232         safeExec(query);
 233
 234         if (query.first()) {
 235             userId = query.value(0).toInt();
 236         }
 237     }
 238     unlock();
 239
 240     return userId;
 241 }
 242
 243
 244 UserId SqliteStorage::internalUser()
 245 {
 246     UserId userId;
 247
 248     {
 249         QSqlQuery query(logDb());
 250         query.prepare(queryString("select_internaluser"));
 251         lockForRead();
 252         safeExec(query);
 253
 254         if (query.first()) {
 255             userId = query.value(0).toInt();
 256         }
 257     }
 258     unlock();
 259
 260     return userId;
 261 }
 262
 263
 264 void SqliteStorage::delUser(UserId user)
 265 {
 266     QSqlDatabase db = logDb();
 267     db.transaction();
 268
 269     lockForWrite();
 270     {
 271         QSqlQuery query(db);
 272         query.prepare(queryString("delete_backlog_by_uid"));
 273         query.bindValue(":userid", user.toInt());
 274         safeExec(query);
 275
 276         query.prepare(queryString("delete_buffers_by_uid"));
 277         query.bindValue(":userid", user.toInt());
 278         safeExec(query);
 279
 280         query.prepare(queryString("delete_networks_by_uid"));
 281         query.bindValue(":userid", user.toInt());
 282         safeExec(query);
 283
 284         query.prepare(queryString("delete_quasseluser"));
 285         query.bindValue(":userid", user.toInt());
 286         safeExec(query);
 287         // I hate the lack of foreign keys and on delete cascade... :(
 288         db.commit();
 289     }
 290     unlock();
 291
 292     emit userRemoved(user);
 293 }
 294
 295
 296 void SqliteStorage::setUserSetting(UserId userId, const QString &settingName, const QVariant &data)
 297 {
 298     QByteArray rawData;
 299     QDataStream out(&rawData, QIODevice::WriteOnly);
 300     out.setVersion(QDataStream::Qt_4_2);
 301     out << data;
 302
 303     QSqlDatabase db = logDb();
 304     db.transaction();
 305     {
 306         QSqlQuery query(db);
 307         query.prepare(queryString("insert_user_setting"));
 308         query.bindValue(":userid", userId.toInt());
 309         query.bindValue(":settingname", settingName);
 310         query.bindValue(":settingvalue", rawData);
 311         lockForWrite();
 312         safeExec(query);
 313
 314         if (query.lastError().isValid()) {
 315             QSqlQuery updateQuery(db);
 316             updateQuery.prepare(queryString("update_user_setting"));
 317             updateQuery.bindValue(":userid", userId.toInt());
 318             updateQuery.bindValue(":settingname", settingName);
 319             updateQuery.bindValue(":settingvalue", rawData);
 320             safeExec(updateQuery);
 321         }
 322         db.commit();
 323     }
 324     unlock();
 325 }
 326
 327
 328 QVariant SqliteStorage::getUserSetting(UserId userId, const QString &settingName, const QVariant &defaultData)
 329 {
 330     QVariant data = defaultData;
 331     {
 332         QSqlQuery query(logDb());
 333         query.prepare(queryString("select_user_setting"));
 334         query.bindValue(":userid", userId.toInt());
 335         query.bindValue(":settingname", settingName);
 336         lockForRead();
 337         safeExec(query);
 338
 339         if (query.first()) {
 340             QByteArray rawData = query.value(0).toByteArray();
 341             QDataStream in(&rawData, QIODevice::ReadOnly);
 342             in.setVersion(QDataStream::Qt_4_2);
 343             in >> data;
 344         }
 345     }
 346     unlock();
 347     return data;
 348 }
 349
 350
 351 IdentityId SqliteStorage::createIdentity(UserId user, CoreIdentity &identity)
 352 {
 353     IdentityId identityId;
 354
 355     QSqlDatabase db = logDb();
 356     db.transaction();
 357
 358     {
 359         QSqlQuery query(db);
 360         query.prepare(queryString("insert_identity"));
 361         query.bindValue(":userid", user.toInt());
 362         query.bindValue(":identityname", identity.identityName());
 363         query.bindValue(":realname", identity.realName());
 364         query.bindValue(":awaynick", identity.awayNick());
 365         query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
 366         query.bindValue(":awayreason", identity.awayReason());
 367         query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
 368         query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
 369         query.bindValue(":autoawaytime", identity.autoAwayTime());
 370         query.bindValue(":autoawayreason", identity.autoAwayReason());
 371         query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
 372         query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
 373         query.bindValue(":detachawayreason", identity.detachAwayReason());
 374         query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
 375         query.bindValue(":ident", identity.ident());
 376         query.bindValue(":kickreason", identity.kickReason());
 377         query.bindValue(":partreason", identity.partReason());
 378         query.bindValue(":quitreason", identity.quitReason());
 379 #ifdef HAVE_SSL
 380         query.bindValue(":sslcert", identity.sslCert().toPem());
 381         query.bindValue(":sslkey", identity.sslKey().toPem());
 382 #else
 383         query.bindValue(":sslcert", QByteArray());
 384         query.bindValue(":sslkey", QByteArray());
 385 #endif
 386
 387         lockForWrite();
 388         safeExec(query);
 389
 390         identityId = query.lastInsertId().toInt();
 391         if (!identityId.isValid()) {
 392             watchQuery(query);
 393         }
 394         else {
 395             QSqlQuery deleteNickQuery(db);
 396             deleteNickQuery.prepare(queryString("delete_nicks"));
 397             deleteNickQuery.bindValue(":identityid", identityId.toInt());
 398             safeExec(deleteNickQuery);
 399
 400             QSqlQuery insertNickQuery(db);
 401             insertNickQuery.prepare(queryString("insert_nick"));
 402             foreach(QString nick, identity.nicks()) {
 403                 insertNickQuery.bindValue(":identityid", identityId.toInt());
 404                 insertNickQuery.bindValue(":nick", nick);
 405                 safeExec(insertNickQuery);
 406             }
 407         }
 408         db.commit();
 409     }
 410     unlock();
 411     identity.setId(identityId);
 412     return identityId;
 413 }
 414
 415
 416 bool SqliteStorage::updateIdentity(UserId user, const CoreIdentity &identity)
 417 {
 418     QSqlDatabase db = logDb();
 419     bool error = false;
 420     db.transaction();
 421
 422     {
 423         QSqlQuery checkQuery(db);
 424         checkQuery.prepare(queryString("select_checkidentity"));
 425         checkQuery.bindValue(":identityid", identity.id().toInt());
 426         checkQuery.bindValue(":userid", user.toInt());
 427         lockForRead();
 428         safeExec(checkQuery);
 429
 430         // there should be exactly one identity for the given id and user
 431         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 1);
 432     }
 433     if (error) {
 434         unlock();
 435         return false;
 436     }
 437
 438     {
 439         QSqlQuery query(db);
 440         query.prepare(queryString("update_identity"));
 441         query.bindValue(":identityname", identity.identityName());
 442         query.bindValue(":realname", identity.realName());
 443         query.bindValue(":awaynick", identity.awayNick());
 444         query.bindValue(":awaynickenabled", identity.awayNickEnabled() ? 1 : 0);
 445         query.bindValue(":awayreason", identity.awayReason());
 446         query.bindValue(":awayreasonenabled", identity.awayReasonEnabled() ? 1 : 0);
 447         query.bindValue(":autoawayenabled", identity.awayReasonEnabled() ? 1 : 0);
 448         query.bindValue(":autoawaytime", identity.autoAwayTime());
 449         query.bindValue(":autoawayreason", identity.autoAwayReason());
 450         query.bindValue(":autoawayreasonenabled", identity.autoAwayReasonEnabled() ? 1 : 0);
 451         query.bindValue(":detachawayenabled", identity.detachAwayEnabled() ? 1 : 0);
 452         query.bindValue(":detachawayreason", identity.detachAwayReason());
 453         query.bindValue(":detachawayreasonenabled", identity.detachAwayReasonEnabled() ? 1 : 0);
 454         query.bindValue(":ident", identity.ident());
 455         query.bindValue(":kickreason", identity.kickReason());
 456         query.bindValue(":partreason", identity.partReason());
 457         query.bindValue(":quitreason", identity.quitReason());
 458 #ifdef HAVE_SSL
 459         query.bindValue(":sslcert", identity.sslCert().toPem());
 460         query.bindValue(":sslkey", identity.sslKey().toPem());
 461 #else
 462         query.bindValue(":sslcert", QByteArray());
 463         query.bindValue(":sslkey", QByteArray());
 464 #endif
 465         query.bindValue(":identityid", identity.id().toInt());
 466         safeExec(query);
 467         watchQuery(query);
 468
 469         QSqlQuery deleteNickQuery(db);
 470         deleteNickQuery.prepare(queryString("delete_nicks"));
 471         deleteNickQuery.bindValue(":identityid", identity.id().toInt());
 472         safeExec(deleteNickQuery);
 473         watchQuery(deleteNickQuery);
 474
 475         QSqlQuery insertNickQuery(db);
 476         insertNickQuery.prepare(queryString("insert_nick"));
 477         foreach(QString nick, identity.nicks()) {
 478             insertNickQuery.bindValue(":identityid", identity.id().toInt());
 479             insertNickQuery.bindValue(":nick", nick);
 480             safeExec(insertNickQuery);
 481             watchQuery(insertNickQuery);
 482         }
 483         db.commit();
 484     }
 485     unlock();
 486     return true;
 487 }
 488
 489
 490 void SqliteStorage::removeIdentity(UserId user, IdentityId identityId)
 491 {
 492     QSqlDatabase db = logDb();
 493     db.transaction();
 494
 495     bool error = false;
 496     {
 497         QSqlQuery checkQuery(db);
 498         checkQuery.prepare(queryString("select_checkidentity"));
 499         checkQuery.bindValue(":identityid", identityId.toInt());
 500         checkQuery.bindValue(":userid", user.toInt());
 501         lockForRead();
 502         safeExec(checkQuery);
 503
 504         // there should be exactly one identity for the given id and user
 505         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 1);
 506     }
 507     if (error) {
 508         unlock();
 509         return;
 510     }
 511
 512     {
 513         QSqlQuery deleteNickQuery(db);
 514         deleteNickQuery.prepare(queryString("delete_nicks"));
 515         deleteNickQuery.bindValue(":identityid", identityId.toInt());
 516         safeExec(deleteNickQuery);
 517
 518         QSqlQuery deleteIdentityQuery(db);
 519         deleteIdentityQuery.prepare(queryString("delete_identity"));
 520         deleteIdentityQuery.bindValue(":identityid", identityId.toInt());
 521         deleteIdentityQuery.bindValue(":userid", user.toInt());
 522         safeExec(deleteIdentityQuery);
 523         db.commit();
 524     }
 525     unlock();
 526 }
 527
 528
 529 QList<CoreIdentity> SqliteStorage::identities(UserId user)
 530 {
 531     QList<CoreIdentity> identities;
 532     QSqlDatabase db = logDb();
 533     db.transaction();
 534
 535     {
 536         QSqlQuery query(db);
 537         query.prepare(queryString("select_identities"));
 538         query.bindValue(":userid", user.toInt());
 539
 540         QSqlQuery nickQuery(db);
 541         nickQuery.prepare(queryString("select_nicks"));
 542
 543         lockForRead();
 544         safeExec(query);
 545
 546         while (query.next()) {
 547             CoreIdentity identity(IdentityId(query.value(0).toInt()));
 548
 549             identity.setIdentityName(query.value(1).toString());
 550             identity.setRealName(query.value(2).toString());
 551             identity.setAwayNick(query.value(3).toString());
 552             identity.setAwayNickEnabled(!!query.value(4).toInt());
 553             identity.setAwayReason(query.value(5).toString());
 554             identity.setAwayReasonEnabled(!!query.value(6).toInt());
 555             identity.setAutoAwayEnabled(!!query.value(7).toInt());
 556             identity.setAutoAwayTime(query.value(8).toInt());
 557             identity.setAutoAwayReason(query.value(9).toString());
 558             identity.setAutoAwayReasonEnabled(!!query.value(10).toInt());
 559             identity.setDetachAwayEnabled(!!query.value(11).toInt());
 560             identity.setDetachAwayReason(query.value(12).toString());
 561             identity.setDetachAwayReasonEnabled(!!query.value(13).toInt());
 562             identity.setIdent(query.value(14).toString());
 563             identity.setKickReason(query.value(15).toString());
 564             identity.setPartReason(query.value(16).toString());
 565             identity.setQuitReason(query.value(17).toString());
 566 #ifdef HAVE_SSL
 567             identity.setSslCert(query.value(18).toByteArray());
 568             identity.setSslKey(query.value(19).toByteArray());
 569 #endif
 570
 571             nickQuery.bindValue(":identityid", identity.id().toInt());
 572             QList<QString> nicks;
 573             safeExec(nickQuery);
 574             watchQuery(nickQuery);
 575             while (nickQuery.next()) {
 576                 nicks << nickQuery.value(0).toString();
 577             }
 578             identity.setNicks(nicks);
 579             identities << identity;
 580         }
 581         db.commit();
 582     }
 583     unlock();
 584     return identities;
 585 }
 586
 587
 588 NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info)
 589 {
 590     NetworkId networkId;
 591
 592     QSqlDatabase db = logDb();
 593     db.transaction();
 594
 595     bool error = false;
 596     {
 597         QSqlQuery query(db);
 598         query.prepare(queryString("insert_network"));
 599         query.bindValue(":userid", user.toInt());
 600         bindNetworkInfo(query, info);
 601         lockForWrite();
 602         safeExec(query);
 603         if (!watchQuery(query)) {
 604             db.rollback();
 605             error = true;
 606         }
 607         else {
 608             networkId = query.lastInsertId().toInt();
 609         }
 610     }
 611     if (error) {
 612         unlock();
 613         return NetworkId();
 614     }
 615
 616     {
 617         QSqlQuery insertServersQuery(db);
 618         insertServersQuery.prepare(queryString("insert_server"));
 619         foreach(Network::Server server, info.serverList) {
 620             insertServersQuery.bindValue(":userid", user.toInt());
 621             insertServersQuery.bindValue(":networkid", networkId.toInt());
 622             bindServerInfo(insertServersQuery, server);
 623             safeExec(insertServersQuery);
 624             if (!watchQuery(insertServersQuery)) {
 625                 db.rollback();
 626                 error = true;
 627                 break;
 628             }
 629         }
 630         if (!error)
 631             db.commit();
 632     }
 633     unlock();
 634     if (error)
 635         return NetworkId();
 636     else
 637         return networkId;
 638 }
 639
 640
 641 void SqliteStorage::bindNetworkInfo(QSqlQuery &query, const NetworkInfo &info)
 642 {
 643     query.bindValue(":networkname", info.networkName);
 644     query.bindValue(":identityid", info.identity.toInt());
 645     query.bindValue(":encodingcodec", QString(info.codecForEncoding));
 646     query.bindValue(":decodingcodec", QString(info.codecForDecoding));
 647     query.bindValue(":servercodec", QString(info.codecForServer));
 648     query.bindValue(":userandomserver", info.useRandomServer ? 1 : 0);
 649     query.bindValue(":perform", info.perform.join("\n"));
 650     query.bindValue(":useautoidentify", info.useAutoIdentify ? 1 : 0);
 651     query.bindValue(":autoidentifyservice", info.autoIdentifyService);
 652     query.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
 653     query.bindValue(":usesasl", info.useSasl ? 1 : 0);
 654     query.bindValue(":saslaccount", info.saslAccount);
 655     query.bindValue(":saslpassword", info.saslPassword);
 656     query.bindValue(":useautoreconnect", info.useAutoReconnect ? 1 : 0);
 657     query.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
 658     query.bindValue(":autoreconnectretries", info.autoReconnectRetries);
 659     query.bindValue(":unlimitedconnectretries", info.unlimitedReconnectRetries ? 1 : 0);
 660     query.bindValue(":rejoinchannels", info.rejoinChannels ? 1 : 0);
 661     if (info.networkId.isValid())
 662         query.bindValue(":networkid", info.networkId.toInt());
 663 }
 664
 665
 666 void SqliteStorage::bindServerInfo(QSqlQuery &query, const Network::Server &server)
 667 {
 668     query.bindValue(":hostname", server.host);
 669     query.bindValue(":port", server.port);
 670     query.bindValue(":password", server.password);
 671     query.bindValue(":ssl", server.useSsl ? 1 : 0);
 672     query.bindValue(":sslversion", server.sslVersion);
 673     query.bindValue(":useproxy", server.useProxy ? 1 : 0);
 674     query.bindValue(":proxytype", server.proxyType);
 675     query.bindValue(":proxyhost", server.proxyHost);
 676     query.bindValue(":proxyport", server.proxyPort);
 677     query.bindValue(":proxyuser", server.proxyUser);
 678     query.bindValue(":proxypass", server.proxyPass);
 679     query.bindValue(":sslverify", server.sslVerify ? 1 : 0);
 680 }
 681
 682
 683 bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info)
 684 {
 685     QSqlDatabase db = logDb();
 686     bool error = false;
 687     db.transaction();
 688
 689     {
 690         QSqlQuery updateQuery(db);
 691         updateQuery.prepare(queryString("update_network"));
 692         updateQuery.bindValue(":userid", user.toInt());
 693         bindNetworkInfo(updateQuery, info);
 694
 695         lockForWrite();
 696         safeExec(updateQuery);
 697         if (!watchQuery(updateQuery) || updateQuery.numRowsAffected() != 1) {
 698             error = true;
 699             db.rollback();
 700         }
 701     }
 702     if (error) {
 703         unlock();
 704         return false;
 705     }
 706
 707     {
 708         QSqlQuery dropServersQuery(db);
 709         dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
 710         dropServersQuery.bindValue(":networkid", info.networkId.toInt());
 711         safeExec(dropServersQuery);
 712         if (!watchQuery(dropServersQuery)) {
 713             error = true;
 714             db.rollback();
 715         }
 716     }
 717     if (error) {
 718         unlock();
 719         return false;
 720     }
 721
 722     {
 723         QSqlQuery insertServersQuery(db);
 724         insertServersQuery.prepare(queryString("insert_server"));
 725         foreach(Network::Server server, info.serverList) {
 726             insertServersQuery.bindValue(":userid", user.toInt());
 727             insertServersQuery.bindValue(":networkid", info.networkId.toInt());
 728             bindServerInfo(insertServersQuery, server);
 729             safeExec(insertServersQuery);
 730             if (!watchQuery(insertServersQuery)) {
 731                 error = true;
 732                 db.rollback();
 733                 break;
 734             }
 735         }
 736     }
 737
 738     db.commit();
 739     unlock();
 740     return !error;
 741 }
 742
 743
 744 bool SqliteStorage::removeNetwork(UserId user, const NetworkId &networkId)
 745 {
 746     QSqlDatabase db = logDb();
 747     bool error = false;
 748     db.transaction();
 749
 750     {
 751         QSqlQuery deleteNetworkQuery(db);
 752         deleteNetworkQuery.prepare(queryString("delete_network"));
 753         deleteNetworkQuery.bindValue(":networkid", networkId.toInt());
 754         deleteNetworkQuery.bindValue(":userid", user.toInt());
 755         lockForWrite();
 756         safeExec(deleteNetworkQuery);
 757         if (!watchQuery(deleteNetworkQuery) || deleteNetworkQuery.numRowsAffected() != 1) {
 758             error = true;
 759             db.rollback();
 760         }
 761     }
 762     if (error) {
 763         unlock();
 764         return false;
 765     }
 766
 767     {
 768         QSqlQuery deleteBacklogQuery(db);
 769         deleteBacklogQuery.prepare(queryString("delete_backlog_for_network"));
 770         deleteBacklogQuery.bindValue(":networkid", networkId.toInt());
 771         safeExec(deleteBacklogQuery);
 772         if (!watchQuery(deleteBacklogQuery)) {
 773             db.rollback();
 774             error = true;
 775         }
 776     }
 777     if (error) {
 778         unlock();
 779         return false;
 780     }
 781
 782     {
 783         QSqlQuery deleteBuffersQuery(db);
 784         deleteBuffersQuery.prepare(queryString("delete_buffers_for_network"));
 785         deleteBuffersQuery.bindValue(":networkid", networkId.toInt());
 786         safeExec(deleteBuffersQuery);
 787         if (!watchQuery(deleteBuffersQuery)) {
 788             db.rollback();
 789             error = true;
 790         }
 791     }
 792     if (error) {
 793         unlock();
 794         return false;
 795     }
 796
 797     {
 798         QSqlQuery deleteServersQuery(db);
 799         deleteServersQuery.prepare(queryString("delete_ircservers_for_network"));
 800         deleteServersQuery.bindValue(":networkid", networkId.toInt());
 801         safeExec(deleteServersQuery);
 802         if (!watchQuery(deleteServersQuery)) {
 803             db.rollback();
 804             error = true;
 805         }
 806     }
 807     if (error) {
 808         unlock();
 809         return false;
 810     }
 811
 812     db.commit();
 813     unlock();
 814     return true;
 815 }
 816
 817
 818 QList<NetworkInfo> SqliteStorage::networks(UserId user)
 819 {
 820     QList<NetworkInfo> nets;
 821
 822     QSqlDatabase db = logDb();
 823     db.transaction();
 824
 825     {
 826         QSqlQuery networksQuery(db);
 827         networksQuery.prepare(queryString("select_networks_for_user"));
 828         networksQuery.bindValue(":userid", user.toInt());
 829
 830         QSqlQuery serversQuery(db);
 831         serversQuery.prepare(queryString("select_servers_for_network"));
 832
 833         lockForRead();
 834         safeExec(networksQuery);
 835         if (watchQuery(networksQuery)) {
 836             while (networksQuery.next()) {
 837                 NetworkInfo net;
 838                 net.networkId = networksQuery.value(0).toInt();
 839                 net.networkName = networksQuery.value(1).toString();
 840                 net.identity = networksQuery.value(2).toInt();
 841                 net.codecForServer = networksQuery.value(3).toString().toLatin1();
 842                 net.codecForEncoding = networksQuery.value(4).toString().toLatin1();
 843                 net.codecForDecoding = networksQuery.value(5).toString().toLatin1();
 844                 net.useRandomServer = networksQuery.value(6).toInt() == 1 ? true : false;
 845                 net.perform = networksQuery.value(7).toString().split("\n");
 846                 net.useAutoIdentify = networksQuery.value(8).toInt() == 1 ? true : false;
 847                 net.autoIdentifyService = networksQuery.value(9).toString();
 848                 net.autoIdentifyPassword = networksQuery.value(10).toString();
 849                 net.useAutoReconnect = networksQuery.value(11).toInt() == 1 ? true : false;
 850                 net.autoReconnectInterval = networksQuery.value(12).toUInt();
 851                 net.autoReconnectRetries = networksQuery.value(13).toInt();
 852                 net.unlimitedReconnectRetries = networksQuery.value(14).toInt() == 1 ? true : false;
 853                 net.rejoinChannels = networksQuery.value(15).toInt() == 1 ? true : false;
 854                 net.useSasl = networksQuery.value(16).toInt() == 1 ? true : false;
 855                 net.saslAccount = networksQuery.value(17).toString();
 856                 net.saslPassword = networksQuery.value(18).toString();
 857
 858                 serversQuery.bindValue(":networkid", net.networkId.toInt());
 859                 safeExec(serversQuery);
 860                 if (!watchQuery(serversQuery)) {
 861                     nets.clear();
 862                     break;
 863                 }
 864                 else {
 865                     Network::ServerList servers;
 866                     while (serversQuery.next()) {
 867                         Network::Server server;
 868                         server.host = serversQuery.value(0).toString();
 869                         server.port = serversQuery.value(1).toUInt();
 870                         server.password = serversQuery.value(2).toString();
 871                         server.useSsl = serversQuery.value(3).toInt() == 1 ? true : false;
 872                         server.sslVersion = serversQuery.value(4).toInt();
 873                         server.useProxy = serversQuery.value(5).toInt() == 1 ? true : false;
 874                         server.proxyType = serversQuery.value(6).toInt();
 875                         server.proxyHost = serversQuery.value(7).toString();
 876                         server.proxyPort = serversQuery.value(8).toUInt();
 877                         server.proxyUser = serversQuery.value(9).toString();
 878                         server.proxyPass = serversQuery.value(10).toString();
 879                         server.sslVerify = serversQuery.value(11).toInt() == 1 ? true : false;
 880                         servers << server;
 881                     }
 882                     net.serverList = servers;
 883                     nets << net;
 884                 }
 885             }
 886         }
 887     }
 888     db.commit();
 889     unlock();
 890     return nets;
 891 }
 892
 893
 894 QList<NetworkId> SqliteStorage::connectedNetworks(UserId user)
 895 {
 896     QList<NetworkId> connectedNets;
 897
 898     QSqlDatabase db = logDb();
 899     db.transaction();
 900
 901     {
 902         QSqlQuery query(db);
 903         query.prepare(queryString("select_connected_networks"));
 904         query.bindValue(":userid", user.toInt());
 905         lockForRead();
 906         safeExec(query);
 907         watchQuery(query);
 908
 909         while (query.next()) {
 910             connectedNets << query.value(0).toInt();
 911         }
 912         db.commit();
 913     }
 914     unlock();
 915     return connectedNets;
 916 }
 917
 918
 919 void SqliteStorage::setNetworkConnected(UserId user, const NetworkId &networkId, bool isConnected)
 920 {
 921     QSqlDatabase db = logDb();
 922     db.transaction();
 923
 924     {
 925         QSqlQuery query(db);
 926         query.prepare(queryString("update_network_connected"));
 927         query.bindValue(":userid", user.toInt());
 928         query.bindValue(":networkid", networkId.toInt());
 929         query.bindValue(":connected", isConnected ? 1 : 0);
 930
 931         lockForWrite();
 932         safeExec(query);
 933         watchQuery(query);
 934         db.commit();
 935     }
 936     unlock();
 937 }
 938
 939
 940 QHash<QString, QString> SqliteStorage::persistentChannels(UserId user, const NetworkId &networkId)
 941 {
 942     QHash<QString, QString> persistentChans;
 943
 944     QSqlDatabase db = logDb();
 945     db.transaction();
 946     {
 947         QSqlQuery query(db);
 948         query.prepare(queryString("select_persistent_channels"));
 949         query.bindValue(":userid", user.toInt());
 950         query.bindValue(":networkid", networkId.toInt());
 951
 952         lockForRead();
 953         safeExec(query);
 954         watchQuery(query);
 955         while (query.next()) {
 956             persistentChans[query.value(0).toString()] = query.value(1).toString();
 957         }
 958     }
 959     unlock();
 960     return persistentChans;
 961 }
 962
 963
 964 void SqliteStorage::setChannelPersistent(UserId user, const NetworkId &networkId, const QString &channel, bool isJoined)
 965 {
 966     QSqlDatabase db = logDb();
 967     db.transaction();
 968
 969     {
 970         QSqlQuery query(db);
 971         query.prepare(queryString("update_buffer_persistent_channel"));
 972         query.bindValue(":userid", user.toInt());
 973         query.bindValue(":networkid", networkId.toInt());
 974         query.bindValue(":buffercname", channel.toLower());
 975         query.bindValue(":joined", isJoined ? 1 : 0);
 976
 977         lockForWrite();
 978         safeExec(query);
 979         watchQuery(query);
 980         db.commit();
 981     }
 982     unlock();
 983 }
 984
 985
 986 void SqliteStorage::setPersistentChannelKey(UserId user, const NetworkId &networkId, const QString &channel, const QString &key)
 987 {
 988     QSqlDatabase db = logDb();
 989     db.transaction();
 990
 991     {
 992         QSqlQuery query(db);
 993         query.prepare(queryString("update_buffer_set_channel_key"));
 994         query.bindValue(":userid", user.toInt());
 995         query.bindValue(":networkid", networkId.toInt());
 996         query.bindValue(":buffercname", channel.toLower());
 997         query.bindValue(":key", key);
 998
 999         lockForWrite();
1000         safeExec(query);
1001         watchQuery(query);
1002         db.commit();
1003     }
1004     unlock();
1005 }
1006
1007
1008 QString SqliteStorage::awayMessage(UserId user, NetworkId networkId)
1009 {
1010     QSqlDatabase db = logDb();
1011     db.transaction();
1012
1013     QString awayMsg;
1014     {
1015         QSqlQuery query(db);
1016         query.prepare(queryString("select_network_awaymsg"));
1017         query.bindValue(":userid", user.toInt());
1018         query.bindValue(":networkid", networkId.toInt());
1019
1020         lockForRead();
1021         safeExec(query);
1022         watchQuery(query);
1023         if (query.first())
1024             awayMsg = query.value(0).toString();
1025         db.commit();
1026     }
1027     unlock();
1028
1029     return awayMsg;
1030 }
1031
1032
1033 void SqliteStorage::setAwayMessage(UserId user, NetworkId networkId, const QString &awayMsg)
1034 {
1035     QSqlDatabase db = logDb();
1036     db.transaction();
1037
1038     {
1039         QSqlQuery query(db);
1040         query.prepare(queryString("update_network_set_awaymsg"));
1041         query.bindValue(":userid", user.toInt());
1042         query.bindValue(":networkid", networkId.toInt());
1043         query.bindValue(":awaymsg", awayMsg);
1044
1045         lockForWrite();
1046         safeExec(query);
1047         watchQuery(query);
1048         db.commit();
1049     }
1050     unlock();
1051 }
1052
1053
1054 QString SqliteStorage::userModes(UserId user, NetworkId networkId)
1055 {
1056     QSqlDatabase db = logDb();
1057     db.transaction();
1058
1059     QString modes;
1060     {
1061         QSqlQuery query(db);
1062         query.prepare(queryString("select_network_usermode"));
1063         query.bindValue(":userid", user.toInt());
1064         query.bindValue(":networkid", networkId.toInt());
1065
1066         lockForRead();
1067         safeExec(query);
1068         watchQuery(query);
1069         if (query.first())
1070             modes = query.value(0).toString();
1071         db.commit();
1072     }
1073     unlock();
1074
1075     return modes;
1076 }
1077
1078
1079 void SqliteStorage::setUserModes(UserId user, NetworkId networkId, const QString &userModes)
1080 {
1081     QSqlDatabase db = logDb();
1082     db.transaction();
1083
1084     {
1085         QSqlQuery query(db);
1086         query.prepare(queryString("update_network_set_usermode"));
1087         query.bindValue(":userid", user.toInt());
1088         query.bindValue(":networkid", networkId.toInt());
1089         query.bindValue(":usermode", userModes);
1090
1091         lockForWrite();
1092         safeExec(query);
1093         watchQuery(query);
1094         db.commit();
1095     }
1096     unlock();
1097 }
1098
1099
1100 BufferInfo SqliteStorage::bufferInfo(UserId user, const NetworkId &networkId, BufferInfo::Type type, const QString &buffer, bool create)
1101 {
1102     QSqlDatabase db = logDb();
1103     db.transaction();
1104
1105     BufferInfo bufferInfo;
1106     {
1107         QSqlQuery query(db);
1108         query.prepare(queryString("select_bufferByName"));
1109         query.bindValue(":networkid", networkId.toInt());
1110         query.bindValue(":userid", user.toInt());
1111         query.bindValue(":buffercname", buffer.toLower());
1112
1113         lockForRead();
1114         safeExec(query);
1115
1116         if (query.first()) {
1117             bufferInfo = BufferInfo(query.value(0).toInt(), networkId, (BufferInfo::Type)query.value(1).toInt(), 0, buffer);
1118             if (query.next()) {
1119                 qCritical() << "SqliteStorage::getBufferInfo(): received more then one Buffer!";
1120                 qCritical() << "         Query:" << query.lastQuery();
1121                 qCritical() << "  bound Values:";
1122                 QList<QVariant> list = query.boundValues().values();
1123                 for (int i = 0; i < list.size(); ++i)
1124                     qCritical() << i << ":" << list.at(i).toString().toLatin1().data();
1125                 Q_ASSERT(false);
1126             }
1127         }
1128         else if (create) {
1129             // let's create the buffer
1130             QSqlQuery createQuery(db);
1131             createQuery.prepare(queryString("insert_buffer"));
1132             createQuery.bindValue(":userid", user.toInt());
1133             createQuery.bindValue(":networkid", networkId.toInt());
1134             createQuery.bindValue(":buffertype", (int)type);
1135             createQuery.bindValue(":buffername", buffer);
1136             createQuery.bindValue(":buffercname", buffer.toLower());
1137             createQuery.bindValue(":joined", type & BufferInfo::ChannelBuffer ? 1 : 0);
1138
1139             unlock();
1140             lockForWrite();
1141             safeExec(createQuery);
1142             watchQuery(createQuery);
1143             bufferInfo = BufferInfo(createQuery.lastInsertId().toInt(), networkId, type, 0, buffer);
1144         }
1145     }
1146     db.commit();
1147     unlock();
1148     return bufferInfo;
1149 }
1150
1151
1152 BufferInfo SqliteStorage::getBufferInfo(UserId user, const BufferId &bufferId)
1153 {
1154     QSqlDatabase db = logDb();
1155     db.transaction();
1156
1157     BufferInfo bufferInfo;
1158     {
1159         QSqlQuery query(db);
1160         query.prepare(queryString("select_buffer_by_id"));
1161         query.bindValue(":userid", user.toInt());
1162         query.bindValue(":bufferid", bufferId.toInt());
1163
1164         lockForRead();
1165         safeExec(query);
1166
1167         if (watchQuery(query) && query.first()) {
1168             bufferInfo = BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), 0, query.value(4).toString());
1169             Q_ASSERT(!query.next());
1170         }
1171         db.commit();
1172     }
1173     unlock();
1174     return bufferInfo;
1175 }
1176
1177
1178 QList<BufferInfo> SqliteStorage::requestBuffers(UserId user)
1179 {
1180     QList<BufferInfo> bufferlist;
1181
1182     QSqlDatabase db = logDb();
1183     db.transaction();
1184
1185     {
1186         QSqlQuery query(db);
1187         query.prepare(queryString("select_buffers"));
1188         query.bindValue(":userid", user.toInt());
1189
1190         lockForRead();
1191         safeExec(query);
1192         watchQuery(query);
1193         while (query.next()) {
1194             bufferlist << BufferInfo(query.value(0).toInt(), query.value(1).toInt(), (BufferInfo::Type)query.value(2).toInt(), query.value(3).toInt(), query.value(4).toString());
1195         }
1196         db.commit();
1197     }
1198     unlock();
1199
1200     return bufferlist;
1201 }
1202
1203
1204 QList<BufferId> SqliteStorage::requestBufferIdsForNetwork(UserId user, NetworkId networkId)
1205 {
1206     QList<BufferId> bufferList;
1207
1208     QSqlDatabase db = logDb();
1209     db.transaction();
1210
1211     {
1212         QSqlQuery query(db);
1213         query.prepare(queryString("select_buffers_for_network"));
1214         query.bindValue(":networkid", networkId.toInt());
1215         query.bindValue(":userid", user.toInt());
1216
1217         lockForRead();
1218         safeExec(query);
1219         watchQuery(query);
1220         while (query.next()) {
1221             bufferList << BufferId(query.value(0).toInt());
1222         }
1223         db.commit();
1224     }
1225     unlock();
1226
1227     return bufferList;
1228 }
1229
1230
1231 bool SqliteStorage::removeBuffer(const UserId &user, const BufferId &bufferId)
1232 {
1233     QSqlDatabase db = logDb();
1234     db.transaction();
1235
1236     bool error = false;
1237     {
1238         QSqlQuery delBufferQuery(db);
1239         delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
1240         delBufferQuery.bindValue(":bufferid", bufferId.toInt());
1241         delBufferQuery.bindValue(":userid", user.toInt());
1242
1243         lockForWrite();
1244         safeExec(delBufferQuery);
1245
1246         error = (!watchQuery(delBufferQuery) || delBufferQuery.numRowsAffected() != 1);
1247     }
1248
1249     if (error) {
1250         db.rollback();
1251         unlock();
1252         return false;
1253     }
1254
1255     {
1256         QSqlQuery delBacklogQuery(db);
1257         delBacklogQuery.prepare(queryString("delete_backlog_for_buffer"));
1258         delBacklogQuery.bindValue(":bufferid", bufferId.toInt());
1259
1260         safeExec(delBacklogQuery);
1261         error = !watchQuery(delBacklogQuery);
1262     }
1263
1264     if (error) {
1265         db.rollback();
1266     }
1267     else {
1268         db.commit();
1269     }
1270     unlock();
1271     return !error;
1272 }
1273
1274
1275 bool SqliteStorage::renameBuffer(const UserId &user, const BufferId &bufferId, const QString &newName)
1276 {
1277     QSqlDatabase db = logDb();
1278     db.transaction();
1279
1280     bool error = false;
1281     {
1282         QSqlQuery query(db);
1283         query.prepare(queryString("update_buffer_name"));
1284         query.bindValue(":buffername", newName);
1285         query.bindValue(":buffercname", newName.toLower());
1286         query.bindValue(":bufferid", bufferId.toInt());
1287         query.bindValue(":userid", user.toInt());
1288
1289         lockForWrite();
1290         safeExec(query);
1291
1292         error = query.lastError().isValid();
1293         // unexepcted error occured (19 == constraint violation)
1294         if (error && query.lastError().number() != 19) {
1295             watchQuery(query);
1296         }
1297         else {
1298             error |= (query.numRowsAffected() != 1);
1299         }
1300     }
1301     if (error) {
1302         db.rollback();
1303     }
1304     else {
1305         db.commit();
1306     }
1307     unlock();
1308     return !error;
1309 }
1310
1311
1312 bool SqliteStorage::mergeBuffersPermanently(const UserId &user, const BufferId &bufferId1, const BufferId &bufferId2)
1313 {
1314     QSqlDatabase db = logDb();
1315     db.transaction();
1316
1317     bool error = false;
1318     {
1319         QSqlQuery checkQuery(db);
1320         checkQuery.prepare(queryString("select_buffers_for_merge"));
1321         checkQuery.bindValue(":oldbufferid", bufferId2.toInt());
1322         checkQuery.bindValue(":newbufferid", bufferId1.toInt());
1323         checkQuery.bindValue(":userid", user.toInt());
1324
1325         lockForRead();
1326         safeExec(checkQuery);
1327         error = (!checkQuery.first() || checkQuery.value(0).toInt() != 2);
1328     }
1329     if (error) {
1330         db.rollback();
1331         unlock();
1332         return false;
1333     }
1334
1335     {
1336         QSqlQuery query(db);
1337         query.prepare(queryString("update_backlog_bufferid"));
1338         query.bindValue(":oldbufferid", bufferId2.toInt());
1339         query.bindValue(":newbufferid", bufferId1.toInt());
1340         safeExec(query);
1341         error = !watchQuery(query);
1342     }
1343     if (error) {
1344         db.rollback();
1345         unlock();
1346         return false;
1347     }
1348
1349     {
1350         QSqlQuery delBufferQuery(db);
1351         delBufferQuery.prepare(queryString("delete_buffer_for_bufferid"));
1352         delBufferQuery.bindValue(":bufferid", bufferId2.toInt());
1353         delBufferQuery.bindValue(":userid", user.toInt());
1354         safeExec(delBufferQuery);
1355         error = !watchQuery(delBufferQuery);
1356     }
1357
1358     if (error) {
1359         db.rollback();
1360     }
1361     else {
1362         db.commit();
1363     }
1364     unlock();
1365     return !error;
1366 }
1367
1368
1369 void SqliteStorage::setBufferLastSeenMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1370 {
1371     QSqlDatabase db = logDb();
1372     db.transaction();
1373
1374     {
1375         QSqlQuery query(db);
1376         query.prepare(queryString("update_buffer_lastseen"));
1377         query.bindValue(":userid", user.toInt());
1378         query.bindValue(":bufferid", bufferId.toInt());
1379         query.bindValue(":lastseenmsgid", msgId.toInt());
1380
1381         lockForWrite();
1382         safeExec(query);
1383         watchQuery(query);
1384     }
1385     db.commit();
1386     unlock();
1387 }
1388
1389
1390 QHash<BufferId, MsgId> SqliteStorage::bufferLastSeenMsgIds(UserId user)
1391 {
1392     QHash<BufferId, MsgId> lastSeenHash;
1393
1394     QSqlDatabase db = logDb();
1395     db.transaction();
1396
1397     bool error = false;
1398     {
1399         QSqlQuery query(db);
1400         query.prepare(queryString("select_buffer_lastseen_messages"));
1401         query.bindValue(":userid", user.toInt());
1402
1403         lockForRead();
1404         safeExec(query);
1405         error = !watchQuery(query);
1406         if (!error) {
1407             while (query.next()) {
1408                 lastSeenHash[query.value(0).toInt()] = query.value(1).toInt();
1409             }
1410         }
1411     }
1412
1413     db.commit();
1414     unlock();
1415     return lastSeenHash;
1416 }
1417
1418
1419 void SqliteStorage::setBufferMarkerLineMsg(UserId user, const BufferId &bufferId, const MsgId &msgId)
1420 {
1421     QSqlDatabase db = logDb();
1422     db.transaction();
1423
1424     {
1425         QSqlQuery query(db);
1426         query.prepare(queryString("update_buffer_markerlinemsgid"));
1427         query.bindValue(":userid", user.toInt());
1428         query.bindValue(":bufferid", bufferId.toInt());
1429         query.bindValue(":markerlinemsgid", msgId.toInt());
1430
1431         lockForWrite();
1432         safeExec(query);
1433         watchQuery(query);
1434     }
1435     db.commit();
1436     unlock();
1437 }
1438
1439
1440 QHash<BufferId, MsgId> SqliteStorage::bufferMarkerLineMsgIds(UserId user)
1441 {
1442     QHash<BufferId, MsgId> markerLineHash;
1443
1444     QSqlDatabase db = logDb();
1445     db.transaction();
1446
1447     bool error = false;
1448     {
1449         QSqlQuery query(db);
1450         query.prepare(queryString("select_buffer_markerlinemsgids"));
1451         query.bindValue(":userid", user.toInt());
1452
1453         lockForRead();
1454         safeExec(query);
1455         error = !watchQuery(query);
1456         if (!error) {
1457             while (query.next()) {
1458                 markerLineHash[query.value(0).toInt()] = query.value(1).toInt();
1459             }
1460         }
1461     }
1462
1463     db.commit();
1464     unlock();
1465     return markerLineHash;
1466 }
1467
1468
1469 bool SqliteStorage::logMessage(Message &msg)
1470 {
1471     QSqlDatabase db = logDb();
1472     db.transaction();
1473
1474     bool error = false;
1475     {
1476         QSqlQuery logMessageQuery(db);
1477         logMessageQuery.prepare(queryString("insert_message"));
1478
1479         logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
1480         logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1481         logMessageQuery.bindValue(":type", msg.type());
1482         logMessageQuery.bindValue(":flags", (int)msg.flags());
1483         logMessageQuery.bindValue(":sender", msg.sender());
1484         logMessageQuery.bindValue(":message", msg.contents());
1485
1486         lockForWrite();
1487         safeExec(logMessageQuery);
1488
1489         if (logMessageQuery.lastError().isValid()) {
1490             // constraint violation - must be NOT NULL constraint - probably the sender is missing...
1491             if (logMessageQuery.lastError().number() == 19) {
1492                 QSqlQuery addSenderQuery(db);
1493                 addSenderQuery.prepare(queryString("insert_sender"));
1494                 addSenderQuery.bindValue(":sender", msg.sender());
1495                 safeExec(addSenderQuery);
1496                 safeExec(logMessageQuery);
1497                 error = !watchQuery(logMessageQuery);
1498             }
1499             else {
1500                 watchQuery(logMessageQuery);
1501             }
1502         }
1503         if (!error) {
1504             MsgId msgId = logMessageQuery.lastInsertId().toInt();
1505             if (msgId.isValid()) {
1506                 msg.setMsgId(msgId);
1507             }
1508             else {
1509                 error = true;
1510             }
1511         }
1512     }
1513
1514     if (error) {
1515         db.rollback();
1516     }
1517     else {
1518         db.commit();
1519     }
1520
1521     unlock();
1522     return !error;
1523 }
1524
1525
1526 bool SqliteStorage::logMessages(MessageList &msgs)
1527 {
1528     QSqlDatabase db = logDb();
1529     db.transaction();
1530
1531     {
1532         QSet<QString> senders;
1533         QSqlQuery addSenderQuery(db);
1534         addSenderQuery.prepare(queryString("insert_sender"));
1535         lockForWrite();
1536         for (int i = 0; i < msgs.count(); i++) {
1537             const QString &sender = msgs.at(i).sender();
1538             if (senders.contains(sender))
1539                 continue;
1540             senders << sender;
1541
1542             addSenderQuery.bindValue(":sender", sender);
1543             safeExec(addSenderQuery);
1544         }
1545     }
1546
1547     bool error = false;
1548     {
1549         QSqlQuery logMessageQuery(db);
1550         logMessageQuery.prepare(queryString("insert_message"));
1551         for (int i = 0; i < msgs.count(); i++) {
1552             Message &msg = msgs[i];
1553
1554             logMessageQuery.bindValue(":time", msg.timestamp().toTime_t());
1555             logMessageQuery.bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
1556             logMessageQuery.bindValue(":type", msg.type());
1557             logMessageQuery.bindValue(":flags", (int)msg.flags());
1558             logMessageQuery.bindValue(":sender", msg.sender());
1559             logMessageQuery.bindValue(":message", msg.contents());
1560
1561             safeExec(logMessageQuery);
1562             if (!watchQuery(logMessageQuery)) {
1563                 error = true;
1564                 break;
1565             }
1566             else {
1567                 msg.setMsgId(logMessageQuery.lastInsertId().toInt());
1568             }
1569         }
1570     }
1571
1572     if (error) {
1573         db.rollback();
1574         unlock();
1575         // we had a rollback in the db so we need to reset all msgIds
1576         for (int i = 0; i < msgs.count(); i++) {
1577             msgs[i].setMsgId(MsgId());
1578         }
1579     }
1580     else {
1581         db.commit();
1582         unlock();
1583     }
1584     return !error;
1585 }
1586
1587
1588 QList<Message> SqliteStorage::requestMsgs(UserId user, BufferId bufferId, MsgId first, MsgId last, int limit)
1589 {
1590     QList<Message> messagelist;
1591
1592     QSqlDatabase db = logDb();
1593     db.transaction();
1594
1595     bool error = false;
1596     BufferInfo bufferInfo;
1597     {
1598         // code dupication from getBufferInfo:
1599         // this is due to the impossibility of nesting transactions and recursive locking
1600         QSqlQuery bufferInfoQuery(db);
1601         bufferInfoQuery.prepare(queryString("select_buffer_by_id"));
1602         bufferInfoQuery.bindValue(":userid", user.toInt());
1603         bufferInfoQuery.bindValue(":bufferid", bufferId.toInt());
1604
1605         lockForRead();
1606         safeExec(bufferInfoQuery);
1607         error = !watchQuery(bufferInfoQuery) || !bufferInfoQuery.first();
1608         if (!error) {
1609             bufferInfo = BufferInfo(bufferInfoQuery.value(0).toInt(), bufferInfoQuery.value(1).toInt(), (BufferInfo::Type)bufferInfoQuery.value(2).toInt(), 0, bufferInfoQuery.value(4).toString());
1610             error = !bufferInfo.isValid();
1611         }
1612     }
1613     if (error) {
1614         db.rollback();
1615         unlock();
1616         return messagelist;
1617     }
1618
1619     {
1620         QSqlQuery query(db);
1621         if (last == -1 && first == -1) {
1622             query.prepare(queryString("select_messagesNewestK"));
1623         }
1624         else if (last == -1) {
1625             query.prepare(queryString("select_messagesNewerThan"));
1626             query.bindValue(":firstmsg", first.toInt());
1627         }
1628         else {
1629             query.prepare(queryString("select_messages"));
1630             query.bindValue(":lastmsg", last.toInt());
1631             query.bindValue(":firstmsg", first.toInt());
1632         }
1633         query.bindValue(":bufferid", bufferId.toInt());
1634         query.bindValue(":limit", limit);
1635
1636         safeExec(query);
1637         watchQuery(query);
1638
1639         while (query.next()) {
1640             Message msg(QDateTime::fromTime_t(query.value(1).toInt()),
1641                 bufferInfo,
1642                 (Message::Type)query.value(2).toUInt(),
1643                 query.value(5).toString(),
1644                 query.value(4).toString(),
1645                 (Message::Flags)query.value(3).toUInt());
1646             msg.setMsgId(query.value(0).toInt());
1647             messagelist << msg;
1648         }
1649     }
1650     db.commit();
1651     unlock();
1652
1653     return messagelist;
1654 }
1655
1656
1657 QList<Message> SqliteStorage::requestAllMsgs(UserId user, MsgId first, MsgId last, int limit)
1658 {
1659     QList<Message> messagelist;
1660
1661     QSqlDatabase db = logDb();
1662     db.transaction();
1663
1664     QHash<BufferId, BufferInfo> bufferInfoHash;
1665     {
1666         QSqlQuery bufferInfoQuery(db);
1667         bufferInfoQuery.prepare(queryString("select_buffers"));
1668         bufferInfoQuery.bindValue(":userid", user.toInt());
1669
1670         lockForRead();
1671         safeExec(bufferInfoQuery);
1672         watchQuery(bufferInfoQuery);
1673         while (bufferInfoQuery.next()) {
1674             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());
1675             bufferInfoHash[bufferInfo.bufferId()] = bufferInfo;
1676         }
1677
1678         QSqlQuery query(db);
1679         if (last == -1) {
1680             query.prepare(queryString("select_messagesAllNew"));
1681         }
1682         else {
1683             query.prepare(queryString("select_messagesAll"));
1684             query.bindValue(":lastmsg", last.toInt());
1685         }
1686         query.bindValue(":userid", user.toInt());
1687         query.bindValue(":firstmsg", first.toInt());
1688         query.bindValue(":limit", limit);
1689         safeExec(query);
1690
1691         watchQuery(query);
1692
1693         while (query.next()) {
1694             Message msg(QDateTime::fromTime_t(query.value(2).toInt()),
1695                 bufferInfoHash[query.value(1).toInt()],
1696                 (Message::Type)query.value(3).toUInt(),
1697                 query.value(6).toString(),
1698                 query.value(5).toString(),
1699                 (Message::Flags)query.value(4).toUInt());
1700             msg.setMsgId(query.value(0).toInt());
1701             messagelist << msg;
1702         }
1703     }
1704     db.commit();
1705     unlock();
1706     return messagelist;
1707 }
1708
1709
1710 QString SqliteStorage::backlogFile()
1711 {
1712     return Quassel::configDirPath() + "quassel-storage.sqlite";
1713 }
1714
1715
1716 bool SqliteStorage::safeExec(QSqlQuery &query, int retryCount)
1717 {
1718     query.exec();
1719
1720     if (!query.lastError().isValid())
1721         return true;
1722
1723     switch (query.lastError().number()) {
1724     case 5: // SQLITE_BUSY         5   /* The database file is locked */
1725     case 6: // SQLITE_LOCKED       6   /* A table in the database is locked */
1726         if (retryCount < _maxRetryCount)
1727             return safeExec(query, retryCount + 1);
1728     default:
1729         return false;
1730     }
1731 }
1732
1733
1734 // ========================================
1735 //  SqliteMigration
1736 // ========================================
1737 SqliteMigrationReader::SqliteMigrationReader()
1738     : SqliteStorage(),
1739     _maxId(0)
1740 {
1741 }
1742
1743
1744 void SqliteMigrationReader::setMaxId(MigrationObject mo)
1745 {
1746     QString queryString;
1747     switch (mo) {
1748     case Sender:
1749         queryString = "SELECT max(senderid) FROM sender";
1750         break;
1751     case Backlog:
1752         queryString = "SELECT max(messageid) FROM backlog";
1753         break;
1754     default:
1755         _maxId = 0;
1756         return;
1757     }
1758     QSqlQuery query = logDb().exec(queryString);
1759     query.first();
1760     _maxId = query.value(0).toInt();
1761 }
1762
1763
1764 bool SqliteMigrationReader::prepareQuery(MigrationObject mo)
1765 {
1766     setMaxId(mo);
1767
1768     switch (mo) {
1769     case QuasselUser:
1770         newQuery(queryString("migrate_read_quasseluser"), logDb());
1771         break;
1772     case Identity:
1773         newQuery(queryString("migrate_read_identity"), logDb());
1774         break;
1775     case IdentityNick:
1776         newQuery(queryString("migrate_read_identity_nick"), logDb());
1777         break;
1778     case Network:
1779         newQuery(queryString("migrate_read_network"), logDb());
1780         break;
1781     case Buffer:
1782         newQuery(queryString("migrate_read_buffer"), logDb());
1783         break;
1784     case Sender:
1785         newQuery(queryString("migrate_read_sender"), logDb());
1786         bindValue(0, 0);
1787         bindValue(1, stepSize());
1788         break;
1789     case Backlog:
1790         newQuery(queryString("migrate_read_backlog"), logDb());
1791         bindValue(0, 0);
1792         bindValue(1, stepSize());
1793         break;
1794     case IrcServer:
1795         newQuery(queryString("migrate_read_ircserver"), logDb());
1796         break;
1797     case UserSetting:
1798         newQuery(queryString("migrate_read_usersetting"), logDb());
1799         break;
1800     }
1801     return exec();
1802 }
1803
1804
1805 bool SqliteMigrationReader::readMo(QuasselUserMO &user)
1806 {
1807     if (!next())
1808         return false;
1809
1810     user.id = value(0).toInt();
1811     user.username = value(1).toString();
1812     user.password = value(2).toString();
1813     user.hashversion = value(3).toInt();
1814     return true;
1815 }
1816
1817
1818 bool SqliteMigrationReader::readMo(IdentityMO &identity)
1819 {
1820     if (!next())
1821         return false;
1822
1823     identity.id = value(0).toInt();
1824     identity.userid = value(1).toInt();
1825     identity.identityname = value(2).toString();
1826     identity.realname = value(3).toString();
1827     identity.awayNick = value(4).toString();
1828     identity.awayNickEnabled = value(5).toInt() == 1 ? true : false;
1829     identity.awayReason = value(6).toString();
1830     identity.awayReasonEnabled = value(7).toInt() == 1 ? true : false;
1831     identity.autoAwayEnabled = value(8).toInt() == 1 ? true : false;
1832     identity.autoAwayTime = value(9).toInt();
1833     identity.autoAwayReason = value(10).toString();
1834     identity.autoAwayReasonEnabled = value(11).toInt() == 1 ? true : false;
1835     identity.detachAwayEnabled = value(12).toInt() == 1 ? true : false;
1836     identity.detachAwayReason = value(13).toString();
1837     identity.detchAwayReasonEnabled = value(14).toInt() == 1 ? true : false;
1838     identity.ident = value(15).toString();
1839     identity.kickReason = value(16).toString();
1840     identity.partReason = value(17).toString();
1841     identity.quitReason = value(18).toString();
1842     identity.sslCert = value(19).toByteArray();
1843     identity.sslKey = value(20).toByteArray();
1844     return true;
1845 }
1846
1847
1848 bool SqliteMigrationReader::readMo(IdentityNickMO &identityNick)
1849 {
1850     if (!next())
1851         return false;
1852
1853     identityNick.nickid = value(0).toInt();
1854     identityNick.identityId = value(1).toInt();
1855     identityNick.nick = value(2).toString();
1856     return true;
1857 }
1858
1859
1860 bool SqliteMigrationReader::readMo(NetworkMO &network)
1861 {
1862     if (!next())
1863         return false;
1864
1865     network.networkid = value(0).toInt();
1866     network.userid = value(1).toInt();
1867     network.networkname = value(2).toString();
1868     network.identityid = value(3).toInt();
1869     network.encodingcodec = value(4).toString();
1870     network.decodingcodec = value(5).toString();
1871     network.servercodec = value(6).toString();
1872     network.userandomserver = value(7).toInt() == 1 ? true : false;
1873     network.perform = value(8).toString();
1874     network.useautoidentify = value(9).toInt() == 1 ? true : false;
1875     network.autoidentifyservice = value(10).toString();
1876     network.autoidentifypassword = value(11).toString();
1877     network.useautoreconnect = value(12).toInt() == 1 ? true : false;
1878     network.autoreconnectinterval = value(13).toInt();
1879     network.autoreconnectretries = value(14).toInt();
1880     network.unlimitedconnectretries = value(15).toInt() == 1 ? true : false;
1881     network.rejoinchannels = value(16).toInt() == 1 ? true : false;
1882     network.connected = value(17).toInt() == 1 ? true : false;
1883     network.usermode = value(18).toString();
1884     network.awaymessage = value(19).toString();
1885     network.attachperform = value(20).toString();
1886     network.detachperform = value(21).toString();
1887     network.usesasl = value(22).toInt() == 1 ? true : false;
1888     network.saslaccount = value(23).toString();
1889     network.saslpassword = value(24).toString();
1890     return true;
1891 }
1892
1893
1894 bool SqliteMigrationReader::readMo(BufferMO &buffer)
1895 {
1896     if (!next())
1897         return false;
1898
1899     buffer.bufferid = value(0).toInt();
1900     buffer.userid = value(1).toInt();
1901     buffer.groupid = value(2).toInt();
1902     buffer.networkid = value(3).toInt();
1903     buffer.buffername = value(4).toString();
1904     buffer.buffercname = value(5).toString();
1905     buffer.buffertype = value(6).toInt();
1906     buffer.lastseenmsgid = value(7).toInt();
1907     buffer.markerlinemsgid = value(8).toInt();
1908     buffer.key = value(9).toString();
1909     buffer.joined = value(10).toInt() == 1 ? true : false;
1910     return true;
1911 }
1912
1913
1914 bool SqliteMigrationReader::readMo(SenderMO &sender)
1915 {
1916     int skipSteps = 0;
1917     while (!next()) {
1918         if (sender.senderId < _maxId) {
1919             bindValue(0, sender.senderId + (skipSteps * stepSize()));
1920             bindValue(1, sender.senderId + ((skipSteps + 1) * stepSize()));
1921             skipSteps++;
1922             if (!exec())
1923                 return false;
1924         }
1925         else {
1926             return false;
1927         }
1928     }
1929
1930     sender.senderId = value(0).toInt();
1931     sender.sender = value(1).toString();
1932     return true;
1933 }
1934
1935
1936 bool SqliteMigrationReader::readMo(BacklogMO &backlog)
1937 {
1938     int skipSteps = 0;
1939     while (!next()) {
1940         if (backlog.messageid < _maxId) {
1941             bindValue(0, backlog.messageid.toInt() + (skipSteps * stepSize()));
1942             bindValue(1, backlog.messageid.toInt() + ((skipSteps + 1) * stepSize()));
1943             skipSteps++;
1944             if (!exec())
1945                 return false;
1946         }
1947         else {
1948             return false;
1949         }
1950     }
1951
1952     backlog.messageid = value(0).toInt();
1953     backlog.time = QDateTime::fromTime_t(value(1).toInt()).toUTC();
1954     backlog.bufferid = value(2).toInt();
1955     backlog.type = value(3).toInt();
1956     backlog.flags = value(4).toInt();
1957     backlog.senderid = value(5).toInt();
1958     backlog.message = value(6).toString();
1959     return true;
1960 }
1961
1962
1963 bool SqliteMigrationReader::readMo(IrcServerMO &ircserver)
1964 {
1965     if (!next())
1966         return false;
1967
1968     ircserver.serverid = value(0).toInt();
1969     ircserver.userid = value(1).toInt();
1970     ircserver.networkid = value(2).toInt();
1971     ircserver.hostname = value(3).toString();
1972     ircserver.port = value(4).toInt();
1973     ircserver.password = value(5).toString();
1974     ircserver.ssl = value(6).toInt() == 1 ? true : false;
1975     ircserver.sslversion = value(7).toInt();
1976     ircserver.useproxy = value(8).toInt() == 1 ? true : false;
1977     ircserver.proxytype = value(9).toInt();
1978     ircserver.proxyhost = value(10).toString();
1979     ircserver.proxyport = value(11).toInt();
1980     ircserver.proxyuser = value(12).toString();
1981     ircserver.proxypass = value(13).toString();
1982     ircserver.sslverify = value(14).toInt() == 1 ? true : false;
1983     return true;
1984 }
1985
1986
1987 bool SqliteMigrationReader::readMo(UserSettingMO &userSetting)
1988 {
1989     if (!next())
1990         return false;
1991
1992     userSetting.userid = value(0).toInt();
1993     userSetting.settingname = value(1).toString();
1994     userSetting.settingvalue = value(2).toByteArray();
1995
1996     return true;
1997 }