src/core/postgresqlstorage.cpp

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