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