From 4c0c0124ec587266de38cd989ce2358280d60c1d Mon Sep 17 00:00:00 2001
From: JacobBarthelmeh <jacob@wolfssl.com>
Date: Wed, 12 Jul 2023 07:38:45 -0700
Subject: [PATCH] better handling of want read/write cases with SFTP + SSHD

---
 apps/wolfsshd/wolfsshd.c | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/apps/wolfsshd/wolfsshd.c b/apps/wolfsshd/wolfsshd.c
index d039a05a..4991b62d 100644
--- a/apps/wolfsshd/wolfsshd.c
+++ b/apps/wolfsshd/wolfsshd.c
@@ -596,7 +596,8 @@ static int SFTP_Subsystem(WOLFSSHD_CONNECTION* conn, WOLFSSH* ssh,
                 break;
             }
 
-            if (select_ret == WS_SELECT_RECV_READY) {
+            if (ret == WS_WANT_READ || ret == WS_WANT_WRITE ||
+                    select_ret == WS_SELECT_RECV_READY) {
                 ret = wolfSSH_worker(ssh, NULL);
                 error = wolfSSH_get_error(ssh);
                 if (ret == WS_REKEYING) {