From f95e0995ccc5a41b6fd5a91c13d18524a46e85ec Mon Sep 17 00:00:00 2001
From: John Safranek <john@wolfssl.com>
Date: Thu, 7 Oct 2021 15:28:40 -0700
Subject: [PATCH] After sending the shutdown messages, client needs to wait for
 the server to respond before closing its socket.

---
 examples/client/client.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/examples/client/client.c b/examples/client/client.c
index b3b8e8d5..610f2b81 100644
--- a/examples/client/client.c
+++ b/examples/client/client.c
@@ -1142,6 +1142,13 @@ THREAD_RETURN WOLFSSH_THREAD client_test(void* args)
 #endif
     }
     ret = wolfSSH_shutdown(ssh);
+    if (ret != WS_SUCCESS) {
+        err_sys("Sending the shutdown messages failed.");
+    }
+    ret = wolfSSH_worker(ssh, NULL);
+    if (ret != WS_SUCCESS) {
+        err_sys("Failed to listen for close messages from the peer.");
+    }
     WCLOSESOCKET(sockFd);
     wolfSSH_free(ssh);
     wolfSSH_CTX_free(ctx);