SCP client continue on to free struct's after failure

examples/scpclient/scpclient.c

@@ -309,35 +309,43 @@ THREAD_RETURN WOLFSSH_THREAD scp_client(void* args)
         }
     } while (ret == WS_WANT_READ || ret == WS_WANT_WRITE ||
                     ret == WS_CHAN_RXD || ret == WS_REKEYING);
-    if (ret != WS_SUCCESS)
-        err_sys("Couldn't copy the file.");
+    if (ret != WS_SUCCESS) {
+        fprintf(stderr, "Couldn't copy the file.");
+        ((func_args*)args)->return_code = 1;
+    }
 
     ret = wolfSSH_shutdown(ssh);
     /* do not continue on with shutdown process if peer already disconnected */
     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.");
+            WLOG(WS_LOG_DEBUG, "Sending the shutdown messages failed.");
         }
-        ret = wolfSSH_worker(ssh, NULL);
-        if (ret != WS_SUCCESS && ret != WS_CHANNEL_CLOSED) {
-            err_sys("Failed to listen for close messages from the peer.");
+        else {
+            ret = wolfSSH_worker(ssh, NULL);
+            if (ret != WS_SUCCESS && ret != WS_CHANNEL_CLOSED) {
+                WLOG(WS_LOG_DEBUG,
+                    "Failed to listen for close messages from the peer.");
+            }
         }
     }
     WCLOSESOCKET(sockFd);
     wolfSSH_free(ssh);
     wolfSSH_CTX_free(ctx);
     if (ret != WS_SUCCESS && ret != WS_SOCKET_ERROR_E &&
             ret != WS_CHANNEL_CLOSED) {
-        err_sys("Closing scp stream failed. Connection could have been closed by peer");
+        WLOG(WS_LOG_DEBUG,
+        "Closing scp stream failed. Connection could have been closed by peer");
     }
 
     ClientFreeBuffers(pubKeyName, privKeyName, NULL);
 #if !defined(WOLFSSH_NO_ECC) && defined(FP_ECC) && defined(HAVE_THREAD_LS)
     wc_ecc_fp_free();  /* free per thread cache */
 #endif
 
+    if (ret != WS_SUCCESS)
+        ((func_args*)args)->return_code = 1;
     return 0;
 }