From 10c40dbc8639acf97b246cfebc2b4bf8c55d564d Mon Sep 17 00:00:00 2001
From: Pierre Ossman <ossman@cendio.se>
Date: Mon, 9 Dec 2019 16:58:07 +0100
Subject: [PATCH] Check the correct stream if there is more data pending

The input stream might no longer be the raw socket, so we need to
query what's currently active. That wrapping stream might have its
own buffering and may have more data even if the socket is drained.
---
 vncviewer/CConn.cxx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vncviewer/CConn.cxx b/vncviewer/CConn.cxx
index 6ba3276b..eba5dd34 100644
--- a/vncviewer/CConn.cxx
+++ b/vncviewer/CConn.cxx
@@ -266,7 +266,7 @@ void CConn::socketEvent(FL_SOCKET fd, void *data)
        // Also check if we need to stop reading and terminate
        if (should_exit())
          break;
-    } while (cc->sock->inStream().checkNoWait(1));
+    } while (cc->getInStream()->checkNoWait(1));
   } catch (rdr::EndOfStream& e) {
     vlog.info("%s", e.str());
     exit_vncviewer();
-- 
2.39.5