src/core/sqlitestorage.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2005-07 by the Quassel IRC Team                         *
   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  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "sqlitestorage.h"
  22
  23 #include <QCryptographicHash>
  24 #include <QtSql>
  25
  26 #include "network.h"
  27
  28 SqliteStorage::SqliteStorage(QObject *parent)
  29   : AbstractSqlStorage(parent)
  30 {
  31 }
  32
  33 SqliteStorage::~SqliteStorage() {
  34 }
  35
  36 bool SqliteStorage::isAvailable() const {
  37   if(!QSqlDatabase::isDriverAvailable("QSQLITE")) return false;
  38   return true;
  39 }
  40
  41 QString SqliteStorage::displayName() const {
  42   return QString("SQLite");
  43 }
  44
  45 QString SqliteStorage::description() const {
  46   return tr("SQLite is a file-based database engine that does not require any setup. It is suitable for small and medium-sized "
  47             "databases that do not require access via network. Use SQLite if your Quassel Core should store its data on the same machine "
  48             "it is running on, and if you only expect a few users to use your core.");
  49 }
  50
  51 int SqliteStorage::installedSchemaVersion() {
  52   QSqlQuery query = logDb().exec("SELECT value FROM coreinfo WHERE key = 'schemaversion'");
  53   if(query.first())
  54     return query.value(0).toInt();
  55
  56   // maybe it's really old... (schema version 0)
  57   query = logDb().exec("SELECT MAX(version) FROM coreinfo");
  58   if(query.first())
  59     return query.value(0).toInt();
  60
  61   return AbstractSqlStorage::installedSchemaVersion();
  62 }
  63
  64 UserId SqliteStorage::addUser(const QString &user, const QString &password) {
  65   QByteArray cryptopass = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1);
  66   cryptopass = cryptopass.toHex();
  67
  68   QSqlQuery query(logDb());
  69   query.prepare(queryString("insert_quasseluser"));
  70   query.bindValue(":username", user);
  71   query.bindValue(":password", cryptopass);
  72   query.exec();
  73   if(query.lastError().isValid() && query.lastError().number() == 19) { // user already exists - sadly 19 seems to be the general constraint violation error...
  74     return 0;
  75   }
  76
  77   query.prepare(queryString("select_userid"));
  78   query.bindValue(":username", user);
  79   query.exec();
  80   query.first();
  81   UserId uid = query.value(0).toInt();
  82   emit userAdded(uid, user);
  83   return uid;
  84 }
  85
  86 void SqliteStorage::updateUser(UserId user, const QString &password) {
  87   QByteArray cryptopass = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1);
  88   cryptopass = cryptopass.toHex();
  89
  90   QSqlQuery query(logDb());
  91   query.prepare(queryString("update_userpassword"));
  92   query.bindValue(":userid", user.toInt());
  93   query.bindValue(":password", cryptopass);
  94   query.exec();
  95 }
  96
  97 void SqliteStorage::renameUser(UserId user, const QString &newName) {
  98   QSqlQuery query(logDb());
  99   query.prepare(queryString("update_username"));
 100   query.bindValue(":userid", user.toInt());
 101   query.bindValue(":username", newName);
 102   query.exec();
 103   emit userRenamed(user, newName);
 104 }
 105
 106 UserId SqliteStorage::validateUser(const QString &user, const QString &password) {
 107   QByteArray cryptopass = QCryptographicHash::hash(password.toUtf8(), QCryptographicHash::Sha1);
 108   cryptopass = cryptopass.toHex();
 109
 110   QSqlQuery query(logDb());
 111   query.prepare(queryString("select_authuser"));
 112   query.bindValue(":username", user);
 113   query.bindValue(":password", cryptopass);
 114   query.exec();
 115
 116   if(query.first()) {
 117     return query.value(0).toInt();
 118   } else {
 119     return 0;
 120   }
 121 }
 122
 123 void SqliteStorage::delUser(UserId user) {
 124   QSqlQuery query(logDb());
 125   query.prepare(queryString("delete_backlog_by_uid"));
 126   query.bindValue(":userid", user.toInt());
 127   query.exec();
 128
 129   query.prepare(queryString("delete_buffers_by_uid"));
 130   query.bindValue(":userid", user.toInt());
 131   query.exec();
 132
 133   query.prepare(queryString("delete_networks_by_uid"));
 134   query.bindValue(":userid", user.toInt());
 135   query.exec();
 136
 137   query.prepare(queryString("delete_quasseluser"));
 138   query.bindValue(":userid", user.toInt());
 139   query.exec();
 140   // I hate the lack of foreign keys and on delete cascade... :(
 141   emit userRemoved(user);
 142 }
 143
 144 NetworkId SqliteStorage::createNetwork(UserId user, const NetworkInfo &info) {
 145   NetworkId networkId;
 146   QSqlQuery query(logDb());
 147   query.prepare(queryString("insert_network"));
 148   query.bindValue(":userid", user.toInt());
 149   query.bindValue(":networkname", info.networkName);
 150   query.exec();
 151
 152   networkId = getNetworkId(user, info.networkName);
 153   if(!networkId.isValid()) {
 154     watchQuery(&query);
 155   } else {
 156     updateNetwork(user, info);
 157   }
 158   return networkId;
 159 }
 160
 161 bool SqliteStorage::updateNetwork(UserId user, const NetworkInfo &info) {
 162   if(!isValidNetwork(user, info.networkId))
 163      return false;
 164
 165   QSqlQuery updateQuery(logDb());
 166   updateQuery.prepare(queryString("update_network"));
 167   updateQuery.bindValue(":networkname", info.networkName);
 168   updateQuery.bindValue(":identityid", info.identity.toInt());
 169   updateQuery.bindValue(":usecustomencoding", info.useCustomEncodings ? 1 : 0);
 170   updateQuery.bindValue(":encodingcodec", QString(info.codecForEncoding));
 171   updateQuery.bindValue(":decodingcodec", QString(info.codecForDecoding));
 172   updateQuery.bindValue(":userandomserver", info.useRandomServer ? 1 : 0);
 173   updateQuery.bindValue(":perform", info.perform.join("\n"));
 174   updateQuery.bindValue(":useautoidentify", info.useAutoIdentify ? 1 : 0);
 175   updateQuery.bindValue(":autoidentifyservice", info.autoIdentifyService);
 176   updateQuery.bindValue(":autoidentifypassword", info.autoIdentifyPassword);
 177   updateQuery.bindValue(":useautoreconnect", info.useAutoReconnect ? 1 : 0);
 178   updateQuery.bindValue(":autoreconnectinterval", info.autoReconnectInterval);
 179   updateQuery.bindValue(":autoreconnectretries", info.autoReconnectRetries);
 180   updateQuery.bindValue(":rejoinchannels", info.rejoinChannels ? 1 : 0);
 181   updateQuery.bindValue(":networkid", info.networkId.toInt());
 182   updateQuery.exec();
 183   if(!watchQuery(&updateQuery))
 184     return false;
 185
 186   QSqlQuery dropServersQuery(logDb());
 187   dropServersQuery.prepare("DELETE FROM ircserver WHERE networkid = :networkid");
 188   dropServersQuery.bindValue(":networkid", info.networkId.toInt());
 189   dropServersQuery.exec();
 190   if(!watchQuery(&dropServersQuery))
 191     return false;
 192
 193   QSqlQuery insertServersQuery(logDb());
 194   insertServersQuery.prepare(queryString("insert_server"));
 195   foreach(QVariant server_, info.serverList) {
 196     QVariantMap server = server_.toMap();
 197     insertServersQuery.bindValue(":hostname", server["Host"]);
 198     insertServersQuery.bindValue(":port", server["Port"].toInt());
 199     insertServersQuery.bindValue(":password", server["Password"]);
 200     insertServersQuery.bindValue(":ssl", server["UseSSL"].toBool() ? 1 : 0);
 201     insertServersQuery.bindValue(":userid", user.toInt());
 202     insertServersQuery.bindValue(":networkid", info.networkId.toInt());
 203
 204     insertServersQuery.exec();
 205     if(!watchQuery(&insertServersQuery))
 206       return false;
 207   }
 208
 209   return true;
 210 }
 211
 212 bool SqliteStorage::removeNetwork(UserId user, const NetworkId &networkId) {
 213   if(!isValidNetwork(user, networkId))
 214      return false;
 215
 216   bool withTransaction = logDb().driver()->hasFeature(QSqlDriver::Transactions);
 217   if(withTransaction) {
 218     sync();
 219     if(!logDb().transaction()) {
 220       qWarning() << "SqliteStorage::removeNetwork(): cannot start transaction. continuing with out rollback support!";
 221       withTransaction = false;
 222     }
 223   }
 224
 225   QSqlQuery deleteBacklogQuery(logDb());
 226   deleteBacklogQuery.prepare(queryString("delete_backlog_for_network"));
 227   deleteBacklogQuery.bindValue(":networkid", networkId.toInt());
 228   deleteBacklogQuery.exec();
 229   if(!watchQuery(&deleteBacklogQuery)) {
 230     if(withTransaction)
 231       logDb().rollback();
 232     return false;
 233   }
 234
 235   QSqlQuery deleteBuffersQuery(logDb());
 236   deleteBuffersQuery.prepare(queryString("delete_buffers_for_network"));
 237   deleteBuffersQuery.bindValue(":networkid", networkId.toInt());
 238   deleteBuffersQuery.exec();
 239   if(!watchQuery(&deleteBuffersQuery)) {
 240     if(withTransaction)
 241       logDb().rollback();
 242     return false;
 243   }
 244
 245   QSqlQuery deleteServersQuery(logDb());
 246   deleteServersQuery.prepare(queryString("delete_ircservers_for_network"));
 247   deleteServersQuery.bindValue(":networkid", networkId.toInt());
 248   deleteServersQuery.exec();
 249   if(!watchQuery(&deleteServersQuery)) {
 250     if(withTransaction)
 251       logDb().rollback();
 252     return false;
 253   }
 254
 255   QSqlQuery deleteNetworkQuery(logDb());
 256   deleteNetworkQuery.prepare(queryString("delete_network"));
 257   deleteNetworkQuery.bindValue(":networkid", networkId.toInt());
 258   deleteNetworkQuery.exec();
 259   if(!watchQuery(&deleteNetworkQuery)) {
 260     if(withTransaction)
 261       logDb().rollback();
 262     return false;
 263   }
 264
 265   logDb().commit();
 266   return true;
 267 }
 268
 269 QList<NetworkInfo> SqliteStorage::networks(UserId user) {
 270   QList<NetworkInfo> nets;
 271
 272   QSqlQuery networksQuery(logDb());
 273   networksQuery.prepare(queryString("select_networks_for_user"));
 274   networksQuery.bindValue(":userid", user.toInt());
 275
 276   QSqlQuery serversQuery(logDb());
 277   serversQuery.prepare(queryString("select_servers_for_network"));
 278
 279   networksQuery.exec();
 280   if(!watchQuery(&networksQuery))
 281     return nets;
 282
 283   while(networksQuery.next()) {
 284     NetworkInfo net;
 285     net.networkId = networksQuery.value(0).toInt();
 286     net.networkName = networksQuery.value(1).toString();
 287     net.identity = networksQuery.value(2).toInt();
 288     net.useCustomEncodings = networksQuery.value(3).toInt() == 1 ? true : false;
 289     net.codecForEncoding = networksQuery.value(4).toString().toAscii();
 290     net.codecForDecoding = networksQuery.value(5).toString().toAscii();
 291     net.useRandomServer = networksQuery.value(6).toInt() == 1 ? true : false;
 292     net.perform = networksQuery.value(7).toString().split("\n");
 293     net.useAutoIdentify = networksQuery.value(8).toInt() == 1 ? true : false;
 294     net.autoIdentifyService = networksQuery.value(9).toString();
 295     net.autoIdentifyPassword = networksQuery.value(10).toString();
 296     net.useAutoReconnect = networksQuery.value(11).toInt() == 1 ? true : false;
 297     net.autoReconnectInterval = networksQuery.value(12).toUInt();
 298     net.autoReconnectRetries = networksQuery.value(13).toInt();
 299     net.rejoinChannels = networksQuery.value(14).toInt() == 1 ? true : false;
 300
 301     serversQuery.bindValue(":networkid", net.networkId.toInt());
 302     serversQuery.exec();
 303     if(!watchQuery(&serversQuery))
 304       return nets;
 305
 306     QVariantList servers;
 307     while(serversQuery.next()) {
 308       QVariantMap server;
 309       server["Host"] = serversQuery.value(0).toString();
 310       server["Port"] = serversQuery.value(1).toInt();
 311       server["Password"] = serversQuery.value(2).toString();
 312       server["UseSSL"] = serversQuery.value(3).toInt() == 1 ? true : false;
 313       servers << server;
 314     }
 315     net.serverList = servers;
 316     nets << net;
 317   }
 318   return nets;
 319 }
 320
 321 bool SqliteStorage::isValidNetwork(UserId user, const NetworkId &networkId) {
 322   QSqlQuery query(logDb());
 323   query.prepare(queryString("select_networkExists"));
 324   query.bindValue(":userid", user.toInt());
 325   query.bindValue(":networkid", networkId.toInt());
 326   query.exec();
 327
 328   watchQuery(&query); // there should not occur any errors
 329   if(!query.first())
 330     return false;
 331
 332   Q_ASSERT(!query.next());
 333   return true;
 334 }
 335
 336 NetworkId SqliteStorage::getNetworkId(UserId user, const QString &network) {
 337   QSqlQuery query(logDb());
 338   query.prepare("SELECT networkid FROM network "
 339                 "WHERE userid = :userid AND networkname = :networkname");
 340   query.bindValue(":userid", user.toInt());
 341   query.bindValue(":networkname", network);
 342   query.exec();
 343
 344   if(query.first())
 345     return query.value(0).toInt();
 346   else
 347     return NetworkId();
 348 }
 349
 350 void SqliteStorage::createBuffer(UserId user, const NetworkId &networkId, const QString &buffer) {
 351   QSqlQuery *query = cachedQuery("insert_buffer");
 352   query->bindValue(":userid", user.toInt());
 353   query->bindValue(":networkid", networkId.toInt());
 354   query->bindValue(":buffername", buffer);
 355   query->bindValue(":buffercname", buffer.toLower());
 356   query->exec();
 357
 358   watchQuery(query);
 359 }
 360
 361 BufferInfo SqliteStorage::getBufferInfo(UserId user, const NetworkId &networkId, const QString &buffer) {
 362   QSqlQuery *query = cachedQuery("select_bufferByName");
 363   query->bindValue(":networkid", networkId.toInt());
 364   query->bindValue(":userid", user.toInt());
 365   query->bindValue(":buffercname", buffer.toLower());
 366   query->exec();
 367
 368   if(!query->first()) {
 369     createBuffer(user, networkId, buffer);
 370     query->exec();
 371     if(!query->first()) {
 372       watchQuery(query);
 373       qWarning() << "unable to create BufferInfo for:" << user << networkId << buffer;
 374       return BufferInfo();
 375     }
 376   }
 377
 378   BufferInfo bufferInfo = BufferInfo(query->value(0).toInt(), networkId, 0, buffer);
 379   if(query->next()) {
 380     qWarning() << "SqliteStorage::getBufferInfo(): received more then one Buffer!";
 381     qWarning() << "         Query:" << query->lastQuery();
 382     qWarning() << "  bound Values:" << query->boundValues();
 383     Q_ASSERT(false);
 384   }
 385
 386   return bufferInfo;
 387 }
 388
 389 QList<BufferInfo> SqliteStorage::requestBuffers(UserId user, QDateTime since) {
 390   uint time = 0;
 391   if(since.isValid())
 392     time = since.toTime_t();
 393
 394   QList<BufferInfo> bufferlist;
 395   QSqlQuery query(logDb());
 396   query.prepare(queryString("select_buffers"));
 397   query.bindValue(":userid", user.toInt());
 398   query.bindValue(":time", time);
 399
 400   query.exec();
 401   watchQuery(&query);
 402   while(query.next()) {
 403     bufferlist << BufferInfo(query.value(0).toInt(), query.value(2).toInt(), 0, query.value(1).toString());
 404   }
 405   return bufferlist;
 406 }
 407
 408 MsgId SqliteStorage::logMessage(Message msg) {
 409   QSqlQuery *logMessageQuery = cachedQuery("insert_message");
 410   logMessageQuery->bindValue(":time", msg.timestamp().toTime_t());
 411   logMessageQuery->bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
 412   logMessageQuery->bindValue(":type", msg.type());
 413   logMessageQuery->bindValue(":flags", msg.flags());
 414   logMessageQuery->bindValue(":sender", msg.sender());
 415   logMessageQuery->bindValue(":message", msg.text());
 416   logMessageQuery->exec();
 417
 418   if(logMessageQuery->lastError().isValid()) {
 419     // constraint violation - must be NOT NULL constraint - probably the sender is missing...
 420     if(logMessageQuery->lastError().number() == 19) {
 421       QSqlQuery *addSenderQuery = cachedQuery("insert_sender");
 422       addSenderQuery->bindValue(":sender", msg.sender());
 423       addSenderQuery->exec();
 424       watchQuery(addSenderQuery);
 425       logMessageQuery->exec();
 426       if(!watchQuery(logMessageQuery))
 427         return 0;
 428     } else {
 429       watchQuery(logMessageQuery);
 430     }
 431   }
 432
 433   QSqlQuery *getLastMessageIdQuery = cachedQuery("select_lastMessage");
 434   getLastMessageIdQuery->bindValue(":time", msg.timestamp().toTime_t());
 435   getLastMessageIdQuery->bindValue(":bufferid", msg.bufferInfo().bufferId().toInt());
 436   getLastMessageIdQuery->bindValue(":type", msg.type());
 437   getLastMessageIdQuery->bindValue(":sender", msg.sender());
 438   getLastMessageIdQuery->exec();
 439
 440   if(getLastMessageIdQuery->first()) {
 441     return getLastMessageIdQuery->value(0).toInt();
 442   } else { // somethin went wrong... :(
 443     qDebug() << getLastMessageIdQuery->lastQuery() << "time/bufferid/type/sender:" << msg.timestamp().toTime_t() << msg.bufferInfo().bufferId() << msg.type() << msg.sender();
 444     Q_ASSERT(false);
 445     return 0;
 446   }
 447 }
 448
 449 QList<Message> SqliteStorage::requestMsgs(BufferInfo buffer, int lastmsgs, int offset) {
 450   QList<Message> messagelist;
 451   // we have to determine the real offset first
 452   QSqlQuery *offsetQuery = cachedQuery("select_messagesOffset");
 453   offsetQuery->bindValue(":bufferid", buffer.bufferId().toInt());
 454   offsetQuery->bindValue(":messageid", offset);
 455   offsetQuery->exec();
 456   offsetQuery->first();
 457   offset = offsetQuery->value(0).toInt();
 458
 459   // now let's select the messages
 460   QSqlQuery *msgQuery = cachedQuery("select_messages");
 461   msgQuery->bindValue(":bufferid", buffer.bufferId().toInt());
 462   msgQuery->bindValue(":limit", lastmsgs);
 463   msgQuery->bindValue(":offset", offset);
 464   msgQuery->exec();
 465
 466   watchQuery(msgQuery);
 467
 468   while(msgQuery->next()) {
 469     Message msg(QDateTime::fromTime_t(msgQuery->value(1).toInt()),
 470                 buffer,
 471                 (Message::Type)msgQuery->value(2).toUInt(),
 472                 msgQuery->value(5).toString(),
 473                 msgQuery->value(4).toString(),
 474                 msgQuery->value(3).toUInt());
 475     msg.setMsgId(msgQuery->value(0).toInt());
 476     messagelist << msg;
 477   }
 478   return messagelist;
 479 }
 480
 481
 482 QList<Message> SqliteStorage::requestMsgs(BufferInfo buffer, QDateTime since, int offset) {
 483   QList<Message> messagelist;
 484   // we have to determine the real offset first
 485   QSqlQuery *offsetQuery = cachedQuery("select_messagesSinceOffset");
 486   offsetQuery->bindValue(":bufferid", buffer.bufferId().toInt());
 487   offsetQuery->bindValue(":since", since.toTime_t());
 488   offsetQuery->exec();
 489   offsetQuery->first();
 490   offset = offsetQuery->value(0).toInt();
 491
 492   // now let's select the messages
 493   QSqlQuery *msgQuery = cachedQuery("select_messagesSince");
 494   msgQuery->bindValue(":bufferid", buffer.bufferId().toInt());
 495   msgQuery->bindValue(":since", since.toTime_t());
 496   msgQuery->bindValue(":offset", offset);
 497   msgQuery->exec();
 498
 499   watchQuery(msgQuery);
 500
 501   while(msgQuery->next()) {
 502     Message msg(QDateTime::fromTime_t(msgQuery->value(1).toInt()),
 503                 buffer,
 504                 (Message::Type)msgQuery->value(2).toUInt(),
 505                 msgQuery->value(5).toString(),
 506                 msgQuery->value(4).toString(),
 507                 msgQuery->value(3).toUInt());
 508     msg.setMsgId(msgQuery->value(0).toInt());
 509     messagelist << msg;
 510   }
 511
 512   return messagelist;
 513 }
 514
 515
 516 QList<Message> SqliteStorage::requestMsgRange(BufferInfo buffer, int first, int last) {
 517   QList<Message> messagelist;
 518   QSqlQuery *rangeQuery = cachedQuery("select_messageRange");
 519   rangeQuery->bindValue(":bufferid", buffer.bufferId().toInt());
 520   rangeQuery->bindValue(":firstmsg", first);
 521   rangeQuery->bindValue(":lastmsg", last);
 522   rangeQuery->exec();
 523
 524   watchQuery(rangeQuery);
 525
 526   while(rangeQuery->next()) {
 527     Message msg(QDateTime::fromTime_t(rangeQuery->value(1).toInt()),
 528                 buffer,
 529                 (Message::Type)rangeQuery->value(2).toUInt(),
 530                 rangeQuery->value(5).toString(),
 531                 rangeQuery->value(4).toString(),
 532                 rangeQuery->value(3).toUInt());
 533     msg.setMsgId(rangeQuery->value(0).toInt());
 534     messagelist << msg;
 535   }
 536
 537   return messagelist;
 538 }
 539
 540 QString SqliteStorage::backlogFile() {
 541   // kinda ugly, but I currently see no other way to do that
 542 #ifdef Q_OS_WIN32
 543   QString quasselDir = QDir::homePath() + qgetenv("APPDATA") + "\\quassel\\";
 544 #else
 545   QString quasselDir = QDir::homePath() + "/.quassel/";
 546 #endif
 547
 548   QDir qDir(quasselDir);
 549   if(!qDir.exists(quasselDir))
 550     qDir.mkpath(quasselDir);
 551
 552   return quasselDir + "quassel-storage.sqlite";
 553 }
 554