1 files changed, 6 insertions, 2 deletions

diff --git a/sample.c b/sample.c
index 002c326..20b19c7 100644
--- a/sample.c
+++ b/sample.c
@@ -385,11 +385,15 @@ static int auth_kbdint(SSH_SESSION *session){
                 buffer[sizeof(buffer)-1]=0;
                 if((ptr=strchr(buffer,'\n')))
                     *ptr=0;
-                ssh_userauth_kbdint_setanswer(session,i,buffer);
+                if (ssh_userauth_kbdint_setanswer(session,i,buffer) < 0) {
+                  return SSH_AUTH_ERROR;
+                }
                 memset(buffer,0,strlen(buffer));
             } else {
                 ptr=getpass(prompt);
-                ssh_userauth_kbdint_setanswer(session,i,ptr);
+                if (ssh_userauth_kbdint_setanswer(session,i,ptr) < 0) {
+                  return SSH_AUTH_ERROR;
+                }
             }
         }
         err=ssh_userauth_kbdint(session,NULL,NULL);