Prevent crashing if the Resolver jams

[netvis.git] / source / resolver.cpp

diff --git a/source/resolver.cpp b/source/resolver.cpp
index d8dc41d81f989ca6a973a1321b348a8eb6d6aeb5..702ce9670d59100d649a47dd29a7df060dd557b2 100644 (file)
--- a/source/resolver.cpp
+++ b/source/resolver.cpp
@@ -34,6 +34,24 @@ void Resolver::push(Host *h)
        in_queue.push_back(h);
 }
 
+void Resolver::cancel(Host *h)
+{
+       MutexLock l_(mutex);
+
+       if(h==current)
+       {
+               current=0;
+               return;
+       }
+
+       for(list<Host *>::iterator i=in_queue.begin(); i!=in_queue.end(); ++i)
+               if(*i==h)
+               {
+                       in_queue.erase(i);
+                       break;
+               }
+}
+
 void Resolver::tick()
 {
        MutexLock l_(mutex);
@@ -49,24 +67,24 @@ void Resolver::main()
        {
                while(!done)
                {
-                       Host *host;
+                       sockaddr_in addr;
+                       addr.sin_family = AF_INET;
                        {
                                MutexLock l_(mutex);
                                if(in_queue.empty())
                                        break;
-                               host = in_queue.front();
+                               current = in_queue.front();
                                in_queue.erase(in_queue.begin());
+                               addr.sin_addr.s_addr = htonl(current->get_address());
                        }
                        
-                       sockaddr_in addr;
-                       addr.sin_family = AF_INET;
-                       addr.sin_addr.s_addr = htonl(host->get_address());
                        char buf[128];
                        int err = getnameinfo(reinterpret_cast<sockaddr *>(&addr), sizeof(addr), buf, sizeof(buf), 0, 0, NI_NOFQDN|NI_NAMEREQD);
                        if(err==0)
                        {
                                MutexLock l_(mutex);
-                               out_queue.push_back(Result(host, buf));
+                               if(current)
+                                       out_queue.push_back(Result(current, buf));
                        }
                }