From dcc68209f38b59530e0f84f39517a13feffc7d0d Mon Sep 17 00:00:00 2001
From: tb <tb@openbsd.org>
Date: Wed, 29 Dec 2021 22:54:41 +0000
Subject: [PATCH] Check return values of EVP_* API

CID 345158
---
 regress/lib/libcrypto/sha1/sha1test.c | 30 +++++++++++++++++++++------
 1 file changed, 24 insertions(+), 6 deletions(-)

diff --git a/regress/lib/libcrypto/sha1/sha1test.c b/regress/lib/libcrypto/sha1/sha1test.c
index f1cf09d5520..168b422d7aa 100644
--- a/regress/lib/libcrypto/sha1/sha1test.c
+++ b/regress/lib/libcrypto/sha1/sha1test.c
@@ -1,4 +1,4 @@
-/*	$OpenBSD: sha1test.c,v 1.5 2021/12/29 22:53:04 tb Exp $	*/
+/*	$OpenBSD: sha1test.c,v 1.6 2021/12/29 22:54:41 tb Exp $	*/
 /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com)
  * All rights reserved.
  *
@@ -95,7 +95,11 @@ main(int argc, char *argv[])
 	R = ret;
 	i = 1;
 	while (*P != NULL) {
-		EVP_Digest(*P, strlen((char *)*P), md, NULL, EVP_sha1(), NULL);
+		if (!EVP_Digest(*P, strlen((char *)*P), md, NULL, EVP_sha1(),
+		    NULL)) {
+			printf("EVP_Digest failed\n");
+			goto err;
+		}
 		p = pt(md);
 		if (strcmp(p, (char *)*R) != 0) {
 			printf("error calculating SHA1 on '%s'\n", *P);
@@ -109,10 +113,20 @@ main(int argc, char *argv[])
 	}
 
 	memset(buf, 'a', 1000);
-	EVP_DigestInit_ex(c, EVP_sha1(), NULL);
-	for (i = 0; i < 1000; i++)
-		EVP_DigestUpdate(c, buf, 1000);
-	EVP_DigestFinal_ex(c, md, NULL);
+	if (!EVP_DigestInit_ex(c, EVP_sha1(), NULL)) {
+		printf("EVP_DigestInit_ex failed\n");
+		goto err;
+	}
+	for (i = 0; i < 1000; i++) {
+		if (!EVP_DigestUpdate(c, buf, 1000)) {
+			printf("EVP_DigestUpdate failed\n");
+			goto err;
+		}
+	}
+	if (!EVP_DigestFinal_ex(c, md, NULL)) {
+		printf("EVP_DigestFinal_ex failed\n");
+		goto err;
+	}
 	p = pt(md);
 
 	r = bigret;
@@ -125,6 +139,10 @@ main(int argc, char *argv[])
 
 	EVP_MD_CTX_free(c);
 	exit(err);
+
+ err:
+	EVP_MD_CTX_free(c);
+	exit(1);
 }
 
 static char *
-- 
2.20.1