Merge pull request #710 from JacobBarthelmeh/scp_example

add check for channel closed error
wolfSSL · Jun 17, 2024 · 9ad1d1c · 9ad1d1c
2 parents 524b5c6 + 0a0bf3f
commit 9ad1d1c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/examples/scpclient/scpclient.c b/examples/scpclient/scpclient.c
@@ -313,7 +313,9 @@ THREAD_RETURN WOLFSSH_THREAD scp_client(void* args)
 
     ret = wolfSSH_shutdown(ssh);
     /* do not continue on with shutdown process if peer already disconnected */
-    if (ret != WS_SOCKET_ERROR_E && wolfSSH_get_error(ssh) != WS_SOCKET_ERROR_E) {
+    if (ret != WS_CHANNEL_CLOSED && ret != WS_SOCKET_ERROR_E &&
+            wolfSSH_get_error(ssh) != WS_SOCKET_ERROR_E &&
+            wolfSSH_get_error(ssh) != WS_CHANNEL_CLOSED) {
         if (ret != WS_SUCCESS) {
             err_sys("Sending the shutdown messages failed.");
         }

diff --git a/src/ssh.c b/src/ssh.c
@@ -1000,7 +1000,7 @@ int wolfSSH_shutdown(WOLFSSH* ssh)
 
     if (ssh != NULL && ssh->channelList == NULL) {
         WLOG(WS_LOG_DEBUG, "channel list was already removed");
-        ret = WS_SUCCESS;
+        ret = WS_CHANNEL_CLOSED;
     }
 
     WLOG(WS_LOG_DEBUG, "Leaving wolfSSH_shutdown(), ret = %d", ret);