Do similar setup for lshift and rshift benchmarks.

This prevents realloc from unnecessarily impacting the lshift benchmarks.

1 files changed, 13 insertions, 1 deletions

diff --git a/src/regress/lib/libcrypto/bn/bn_shift.c b/src/regress/lib/libcrypto/bn/bn_shift.c
index 5ba5f5403e..fd5184523b 100644
--- a/src/regress/lib/libcrypto/bn/bn_shift.c
+++ b/src/regress/lib/libcrypto/bn/bn_shift.c
@@ -1,4 +1,4 @@
-/*      $OpenBSD: bn_shift.c,v 1.3 2022/12/18 00:22:07 tb Exp $ */
+/*      $OpenBSD: bn_shift.c,v 1.4 2022/12/23 02:12:11 jsing Exp $ */
 /*
  * Copyright (c) 2022 Joel Sing <jsing@openbsd.org>
  *
@@ -319,6 +319,9 @@ benchmark_bn_lshift1(BIGNUM *bn)
 {
         int i;
 
+        if (!BN_set_bit(bn, 8192))
+                errx(1, "BN_set_bit");
+
         if (!BN_one(bn))
                 errx(1, "BN_one");
 
@@ -333,6 +336,9 @@ benchmark_bn_lshift(BIGNUM *bn, int n)
 {
         int i;
 
+        if (!BN_set_bit(bn, 8192 * n))
+                errx(1, "BN_set_bit");
+
         if (!BN_one(bn))
                 errx(1, "BN_one");
 
@@ -371,6 +377,9 @@ benchmark_bn_rshift1(BIGNUM *bn)
 {
         int i;
 
+        if (!BN_one(bn))
+                errx(1, "BN_one");
+
         if (!BN_set_bit(bn, 8192))
                 errx(1, "BN_set_bit");
 
@@ -385,6 +394,9 @@ benchmark_bn_rshift(BIGNUM *bn, int n)
 {
         int i;
 
+        if (!BN_one(bn))
+                errx(1, "BN_one");
+
         if (!BN_set_bit(bn, 8192 * n))
                 errx(1, "BN_set_bit");