Minor bug fix (for cases when race between thread and main thread is causing server._running to not be set to True)

stamparm · stamparm · commit 36f3ab579885 · 2014-01-09T15:46:55.000+01:00
diff --git a/lib/request/dns.py b/lib/request/dns.py
@@ -65,6 +65,7 @@ def __init__(self):
         self._socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
         self._socket.bind(("", 53))
         self._running = False
+        self._initialized = False
 
     def pop(self, prefix=None, suffix=None):
         """
@@ -91,6 +92,7 @@ def run(self):
         def _():
             try:
                 self._running = True
+                self._initialized = True
 
                 while True:
                     data, addr = self._socket.recvfrom(1024)
@@ -116,6 +118,9 @@ def _():
         server = DNSServer()
         server.run()
 
+        while not server._initialized:
+            time.sleep(0.1)
+
         while server._running:
             while True:
                 _ = server.pop()
