Преглед на файлове

Handle closed connection properly on tor_resolve

svn:r2066
Nick Mathewson преди 21 години
родител
ревизия
6d53717215
променени са 1 файла, в които са добавени 5 реда и са изтрити 1 реда
  1. 5 1
      contrib/tor_resolve

+ 5 - 1
contrib/tor_resolve

@@ -97,7 +97,11 @@ def resolve(hostname, sockshost, socksport, socksver=4):
     answer = s.recv(8)
     result = parse(answer)
     while result is None:
-        answer += s.recv(1)
+        more = s.recv(1)
+        if not more:
+            print "Connection closed; dying."
+            return None
+        answer += more
         result = parse(answer)
     print "Got answer",result
     m = s.recv(1)