Close another memory hole.

[openssl.git] / crypto / rsa / rsa_oaep_test.c

diff --git a/crypto/rsa/rsa_oaep_test.c b/crypto/rsa/rsa_oaep_test.c
index 08eb264dd969f51c08e7c99bde176c8e1f850834..5931c65c2c8787dea8c7cbd7766cfd0eed75ceb4 100644 (file)
--- a/crypto/rsa/rsa_oaep_test.c
+++ b/crypto/rsa/rsa_oaep_test.c
@@ -5,6 +5,7 @@
 
 #include "openssl/e_os.h"
 
+#include <openssl/crypto.h>
 #include <openssl/err.h>
 #ifdef NO_RSA
 int main(int argc, char *argv[])
@@ -215,6 +216,8 @@ int main()
     int clen = 0;
     int num;
 
+    CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ENABLE);
+       
     plen = sizeof(ptext_ex) - 1;
 
     for (v = 0; v < 3; v++)
@@ -296,6 +299,11 @@ int main()
     next:
        RSA_free(key);
        }
+
+    ERR_remove_state(0);
+
+    CRYPTO_mem_leaks_fp(stdout);
+
     return err;
     }
 #endif