From d6df4470e1153fc01f32691cdff7bc415c73a4f7 Mon Sep 17 00:00:00 2001
From: Peter Surda <surda@economicsofbitcoin.com>
Date: Tue, 23 Jan 2018 15:59:58 +0100
Subject: [PATCH] No connection CPU hog fix

- the previous fix was incomplete, it shouldn't consume excessive
resources now when there are no connections
---
 src/network/asyncore_pollchoose.py | 35 +++++++++++++-----------------
 1 file changed, 15 insertions(+), 20 deletions(-)

diff --git a/src/network/asyncore_pollchoose.py b/src/network/asyncore_pollchoose.py
index de8ededc..f9dc9ee5 100644
--- a/src/network/asyncore_pollchoose.py
+++ b/src/network/asyncore_pollchoose.py
@@ -398,6 +398,8 @@ def loop(timeout=30.0, use_poll=False, map=None, count=None,
          poller=None):
     if map is None:
         map = socket_map
+    if count is None:
+        count =  True
     # code which grants backward compatibility with "use_poll" 
     # argument which should no longer be used in favor of
     # "poller"
@@ -414,27 +416,20 @@ def loop(timeout=30.0, use_poll=False, map=None, count=None,
         elif hasattr(select, 'select'):
             poller = select_poller
 
-    if count is None:
-        while map:
-            # fill buckets first
-            update_sent()
-            update_received()
-            # then poll
-            poller(timeout, map)
+    if timeout == 0:
+        deadline = 0
     else:
-        if timeout == 0:
-            deadline = 0
-        else:
-            deadline = time.time() + timeout
-        while map and count > 0:
-            # fill buckets first
-            update_sent()
-            update_received()
-            subtimeout = deadline - time.time()
-            if subtimeout <= 0:
-                break
-            poller(subtimeout, map)
-            # then poll
+        deadline = time.time() + timeout
+    while count:
+        # fill buckets first
+        update_sent()
+        update_received()
+        subtimeout = deadline - time.time()
+        if subtimeout <= 0:
+            break
+        # then poll
+        poller(subtimeout, map)
+        if type(count) is int:
             count = count - 1
 
 class dispatcher: