[Browser] Fix crashing of browser (I hope)

apps/browser/QueryHelper.cpp

@@ -27,9 +27,11 @@ QueryHelper::QueryHelper(QAbstractItemModel *model)
 void QueryHelper::refresh()
 {
     if (!queryThread->isRunning())
+    {
         _model->removeRows(0, _model->rowCount());
         queryThread->start();
         emit started();
+    }
 }
 
 void QueryHelper::terminate()

apps/browser/netutils/QueryClient.cpp

@@ -43,21 +43,41 @@ QueryClient &QueryClient::Get()
 
 map<SystemAddress, QueryData> QueryClient::Query()
 {
-    status = -1;
     map<SystemAddress, QueryData> query;
+    BitStream bs;
+    bs.Write((unsigned char) (ID_MASTER_QUERY));
+    qDebug() << "Locking mutex in QueryClient::Query()";
+    mxServers.lock();
+    status = -1;
+    int attempts = 3;
+    do
+    {
         if (Connect() == IS_NOT_CONNECTED)
+        {
+            qDebug() << "Unlocking mutex in QueryClient::Query()";
+            mxServers.unlock();
             return query;
+        }
 
-    BitStream bs;
-    bs.Write((unsigned char) (ID_MASTER_QUERY));
         int code = peer->Send(&bs, HIGH_PRIORITY, RELIABLE_ORDERED, CHANNEL_MASTER, masterAddr, false);
 
         if (code == 0)
+        {
+            qDebug() << "Unlocking mutex in QueryClient::Query()";
+            mxServers.unlock();
             return query;
+        }
 
         pmq->SetServers(&query);
-    status = GetAnswer();
-
+        status = GetAnswer(ID_MASTER_QUERY);
+        RakSleep(100);
+    }
+    while(status != ID_MASTER_QUERY && attempts-- > 0);
+    if(status != ID_MASTER_QUERY)
+        qDebug() << "Getting query was failed";
+    qDebug() << "Unlocking mutex in QueryClient::Query()";
+    peer->CloseConnection(masterAddr, true);
+    mxServers.unlock();
     qDebug() <<"Answer" << (status == ID_MASTER_QUERY ? "ok." : "wrong.");
 
     return query;
@@ -65,24 +85,40 @@ map<SystemAddress, QueryData> QueryClient::Query()
 
 pair<SystemAddress, QueryData> QueryClient::Update(RakNet::SystemAddress addr)
 {
+    qDebug() << "Locking mutex in QueryClient::Update(RakNet::SystemAddress addr)";
     pair<SystemAddress, QueryData> server;
+    BitStream bs;
+    bs.Write((unsigned char) (ID_MASTER_UPDATE));
+    bs.Write(addr);
+
+    mxServers.lock();
+    status = -1;
+    int attempts = 3;
+    pmu->SetServer(&server);
+    do
+    {
         if (Connect() == IS_NOT_CONNECTED)
         {
-        status = -1;
+            qDebug() << IS_NOT_CONNECTED;
+            qDebug() << "Unlocking mutex in QueryClient::Update(RakNet::SystemAddress addr)";
+            mxServers.unlock();
             return server;
         }
 
-    BitStream bs;
-    bs.Write((unsigned char) (ID_MASTER_UPDATE));
-    bs.Write(addr);
         peer->Send(&bs, HIGH_PRIORITY, RELIABLE_ORDERED, CHANNEL_MASTER, masterAddr, false);
-
-    pmu->SetServer(&server);
-    status = GetAnswer();
+        status = GetAnswer(ID_MASTER_UPDATE);
+        RakSleep(100);
+    }
+    while(status != ID_MASTER_UPDATE && attempts-- > 0);
+    if(status != ID_MASTER_UPDATE)
+        qDebug() << "Getting update was failed";
+    peer->CloseConnection(masterAddr, true);
+    qDebug() << "Unlocking mutex in QueryClient::Update(RakNet::SystemAddress addr)";
+    mxServers.unlock();
     return server;
 }
 
-MASTER_PACKETS QueryClient::GetAnswer()
+MASTER_PACKETS QueryClient::GetAnswer(MASTER_PACKETS waitingPacket)
 {
     RakNet::Packet *packet;
     bool update = true;
@@ -105,20 +141,30 @@ MASTER_PACKETS QueryClient::GetAnswer()
                     update = false;
                     break;
                 case ID_MASTER_QUERY:
+                    qDebug() << "ID_MASTER_QUERY";
+                    if (waitingPacket == ID_MASTER_QUERY)
                         pmq->Read();
+                    else
+                        qDebug() << "Got wrong packet";
                     update = false;
                     id = pid;
                     break;
                 case ID_MASTER_UPDATE:
+                    qDebug() << "ID_MASTER_UPDATE";
+                    if (waitingPacket == ID_MASTER_UPDATE)
                         pmu->Read();
+                    else
+                        qDebug() << "Got wrong packet";
                     update = false;
                     id = pid;
                     break;
                 case ID_MASTER_ANNOUNCE:
+                    qDebug() << "ID_MASTER_ANNOUNCE";
                     update = false;
                     id = pid;
                     break;
                 case ID_CONNECTION_REQUEST_ACCEPTED:
+                    qDebug() << "ID_CONNECTION_REQUEST_ACCEPTED";
                     break;
                 default:
                     break;
@@ -148,8 +194,7 @@ ConnectionState QueryClient::Connect()
             case IS_SILENTLY_DISCONNECTING:
             case IS_DISCONNECTING:
             {
-                qDebug() << "Cannot connect to the master server "<< state;
-                //LOG_MESSAGE_SIMPLE(Log::LOG_WARN, "Cannot connect to master server: %d", masterAddr.ToString());
+                qDebug() << "Cannot connect to the master server. Code:"<< state;
                 return IS_NOT_CONNECTED;
             }
             case IS_PENDING:

apps/browser/netutils/QueryClient.hpp

@@ -10,6 +10,7 @@
 #include <components/openmw-mp/Master/PacketMasterQuery.hpp>
 #include <components/openmw-mp/Master/PacketMasterUpdate.hpp>
 #include <apps/browser/ServerModel.hpp>
+#include <mutex>
 
 class QueryClient
 {
@@ -26,7 +27,7 @@ public:
     int Status();
 private:
     RakNet::ConnectionState Connect();
-    MASTER_PACKETS GetAnswer();
+    MASTER_PACKETS GetAnswer(MASTER_PACKETS packet);
 protected:
     QueryClient();
     ~QueryClient();
@@ -37,6 +38,7 @@ private:
     mwmp::PacketMasterQuery *pmq;
     mwmp::PacketMasterUpdate *pmu;
     std::pair<RakNet::SystemAddress, ServerData> server;
+    std::mutex mxServers;
 
 };