From f5a257c383d4aed8399f126e052590f2540db155 Mon Sep 17 00:00:00 2001
From: Satoshi Yamaguchi <satoshi.yamaguchi@gmail.com>
Date: Wed, 17 Aug 2022 11:56:46 +0900
Subject: [PATCH] Add WOLFSSL_LEAVE to wolfSSL_PEM_read_RSA_PUBKEY and
 wolfSSL_PEM_read_PUBKEY

---
 src/pk.c  | 4 +++-
 src/ssl.c | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/pk.c b/src/pk.c
index 4e3140773..b560b8e58 100644
--- a/src/pk.c
+++ b/src/pk.c
@@ -1588,7 +1588,7 @@ WOLFSSL_RSA *wolfSSL_PEM_read_RSA_PUBKEY(XFILE fp,
     
     /* Validate parameters. */
     if (fp == NULL) {
-        WOLFSSL_MSG("Bad function arguments");
+        WOLFSSL_LEAVE("wolfSSL_PEM_read_RSA_PUBKEY", BAD_FUNC_ARG);
         return NULL;
     }
 
@@ -1609,6 +1609,8 @@ WOLFSSL_RSA *wolfSSL_PEM_read_RSA_PUBKEY(XFILE fp,
     else {
         WOLFSSL_MSG("wolfSSL_PEM_read_PUBKEY failed");
     }
+    
+    WOLFSSL_LEAVE("wolfSSL_PEM_read_RSA_PUBKEY", 0);
 
     return rsa;
 }
diff --git a/src/ssl.c b/src/ssl.c
index 77f941f4f..c4d846480 100644
--- a/src/ssl.c
+++ b/src/ssl.c
@@ -28289,6 +28289,8 @@ WOLFSSL_EVP_PKEY *wolfSSL_PEM_read_PUBKEY(XFILE fp, WOLFSSL_EVP_PKEY **x,
         wolfSSL_BIO_free(bio);
     }
 
+    WOLFSSL_LEAVE("wolfSSL_PEM_read_PUBKEY", 0);
+
     return ret;
 }
 #endif /* NO_FILESYSTEM */