From 7692effba51e09c121f1800266b9637ce49e346e Mon Sep 17 00:00:00 2001
From: Eric Blankenhorn <eric@wolfssl.com>
Date: Tue, 30 Apr 2024 11:41:57 -0500
Subject: [PATCH] Fix state in DoUserAuthRequestPublicKey

---
 src/internal.c | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/src/internal.c b/src/internal.c
index 10bb4a81..e1d7b95e 100644
--- a/src/internal.c
+++ b/src/internal.c
@@ -6783,11 +6783,8 @@ static int DoUserAuthRequestPublicKey(WOLFSSH* ssh, WS_UserAuthData* authData,
                         WLOG(WS_LOG_DEBUG, "DUARPK: user overriding success");
                         authFailure = 1;
                     }
-                    else {
-                        ssh->clientState = CLIENT_USERAUTH_DONE;
-                    }
                 }
-                else {
+                if (!authFailure && !partialSuccess) {
                     ssh->clientState = CLIENT_USERAUTH_DONE;
                 }
             }