Free resources

wolfSSL · Jul 15, 2024 · 5ea74bc · 5ea74bc
1 parent b0cf4cf
commit 5ea74bc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/apps/wolfsshd/wolfsshd.c b/apps/wolfsshd/wolfsshd.c
@@ -2416,6 +2416,8 @@ static int StartSSHD(int argc, char** argv)
                         == SOCKET_ERROR) {
                         WLOG(WS_LOG_DEBUG, "wolfSSH non-fatal error: "
                                 "ioctlsocket failed");
+                        WCLOSESOCKET(conn->fd);
+                        WFREE(conn, NULL, DYNTYPE_SSHD);
                         continue;
                     }
 #elif defined(WOLFSSL_MDK_ARM) || defined(WOLFSSL_KEIL_TCP_NET) \
@@ -2427,12 +2429,16 @@ static int StartSSHD(int argc, char** argv)
                     if (flags < 0) {
                         WLOG(WS_LOG_DEBUG, "wolfSSH non-fatal error: "
                                 "fcntl get failed");
+                        WCLOSESOCKET(conn->fd);
+                        WFREE(conn, NULL, DYNTYPE_SSHD);
                         continue;
                     }
                     flags = fcntl(conn->fd, F_SETFL, flags | O_NONBLOCK);
                     if (flags < 0) {
                         WLOG(WS_LOG_DEBUG, "wolfSSH non-fatal error: "
                                 "fcntl set failed");
+                        WCLOSESOCKET(conn->fd);
+                        WFREE(conn, NULL, DYNTYPE_SSHD);
                         continue;
                     }
 #endif