- BUGFIX: Catching DHT exception in case there is a problem

Changelog

@@ -65,6 +65,7 @@
     - BUGFIX: ETA was wrong for torrents with filtered files
     - BUGFIX: Fixed drag'n drop on non-KDE systems
     - BUGFIX: Removed build dependency on Python
+    - BUGFIX: Catching DHT exception in case there is a problem
     - COSMETIC: Redesigned torrent properties a little
     - COSMETIC: Totally redesigned program preferences
     - COSMETIC: Display more logs messages concerning features

TODO

@@ -57,5 +57,6 @@
 - Translations update (IN PROGRESS)
 
 rc6->rc7 changelog:
-- Removed build dependency on Python
+- BUGFIX: Catching DHT exception in case there is a problem
+- BUGFIX: Removed build dependency on Python
 - I18N: Updated Turkish translation

src/GUI.cpp

@@ -884,10 +884,13 @@ void GUI::configureSession(bool deleteOptions) {
   BTSession->setMaxUploadsPerTorrent(options->getMaxUploadsPerTorrent());
   // * DHT
   if(options->isDHTEnabled()) {
-    BTSession->enableDHT(true);
-    downloadingTorrentTab->setInfoBar(tr("DHT support [ON], port: %1").arg(new_listenPort), QString::fromUtf8("blue"));
     // Set DHT Port
     BTSession->setDHTPort(new_listenPort);
+    if(BTSession->enableDHT(true)) {
+      downloadingTorrentTab->setInfoBar(tr("DHT support [ON], port: %1").arg(new_listenPort), QString::fromUtf8("blue"));
+    } else {
+      downloadingTorrentTab->setInfoBar(tr("DHT support [OFF]"), QString::fromUtf8("red"));
+    }
   } else {
     BTSession->enableDHT(false);
     downloadingTorrentTab->setInfoBar(tr("DHT support [OFF]"), QString::fromUtf8("blue"));

src/bittorrent.cpp

@@ -639,7 +639,7 @@ void bittorrent::enableLSD(bool b) {
 }
 
 // Enable DHT
-void bittorrent::enableDHT(bool b) {
+bool bittorrent::enableDHT(bool b) {
   if(b) {
     if(!DHTEnabled) {
       boost::filesystem::ifstream dht_state_file((misc::qBittorrentPath()+QString::fromUtf8("dht_state")).toUtf8().data(), std::ios_base::binary);
@@ -648,12 +648,17 @@ void bittorrent::enableDHT(bool b) {
       try{
         dht_state = bdecode(std::istream_iterator<char>(dht_state_file), std::istream_iterator<char>());
       }catch (std::exception&) {}
+      try {
 	s->start_dht(dht_state);
 	s->add_dht_router(std::make_pair(std::string("router.bittorrent.com"), 6881));
 	s->add_dht_router(std::make_pair(std::string("router.utorrent.com"), 6881));
 	s->add_dht_router(std::make_pair(std::string("router.bitcomet.com"), 6881));
 	DHTEnabled = true;
 	qDebug("DHT enabled");
+       }catch(std::exception e) {
+         qDebug("Could not enable DHT, reason: %s", e.what());
+          return false;
+       }
     }
   } else {
     if(DHTEnabled) {
@@ -662,6 +667,7 @@ void bittorrent::enableDHT(bool b) {
       qDebug("DHT disabled");
     }
   }
+  return true;
 }
 
 void bittorrent::saveTorrentSpeedLimits(QString hash) {

src/bittorrent.h

@@ -135,7 +135,7 @@ class bittorrent : public QObject{
     void enableUPnP(bool b);
     void enableNATPMP(bool b);
     void enableLSD(bool b);
-    void enableDHT(bool b);
+    bool enableDHT(bool b);
 
   protected slots:
     void scanDirectory();