1 files changed, 16 insertions, 1 deletions
diff --git a/tests/src/vnc-client.c b/tests/src/vnc-client.c
index f70cd568..b8730dcd 100644
--- a/tests/src/vnc-client.c
+++ b/tests/src/vnc-client.c
@@ -68,5 +68,20 @@ main (int argc, char **argv)
     if (n_sent != strlen (buffer))
         return EXIT_FAILURE;
 
-    return EXIT_SUCCESS;
+    while (TRUE)
+    {
+        n_read = g_socket_receive (socket, buffer, 1023, NULL, &error);
+        if (n_read < 0)
+        {
+            g_warning ("Unable to receive on VNC socket: %s", error->message);
+            return EXIT_FAILURE;
+        }
+
+        if (n_read == 0)
+        {
+            status_notify ("VNC-CLIENT DISCONNECTED");
+            return EXIT_SUCCESS;
+        }
+    }
+
 }