[svn] commit: r1540 - /trunk/src/bin/auth/main.cc

[BIND 10 source code commits]

Author: shane
Date: Thu Mar 18 21:37:55 2010
New Revision: 1540

Log:
Partial fix: prevents infinite loop if client closes connection.
Full fix needs to be to do partial reads on TCP connections.

Modified:
    trunk/src/bin/auth/main.cc

Modified: trunk/src/bin/auth/main.cc
==============================================================================
--- trunk/src/bin/auth/main.cc (original)
+++ trunk/src/bin/auth/main.cc Thu Mar 18 21:37:55 2010
@@ -516,7 +516,14 @@
         if (cc0 < 0) {
             if (verbose_mode) {
                 cerr << "TCP receive error" << endl;
+                close(ts);
+                return;
             }
+        }
+        if (cc0 == 0) {
+            // client closed connection
+            close(ts);
+            return;
         }
         cc += cc0;
     }