bio chain test: deduplicate chain walking code

1 files changed, 40 insertions, 52 deletions

diff --git a/regress/lib/libcrypto/bio/bio_chain.c b/regress/lib/libcrypto/bio/bio_chain.c
index 810395676f7..de6fc3af2a5 100644
--- a/regress/lib/libcrypto/bio/bio_chain.c
+++ b/regress/lib/libcrypto/bio/bio_chain.c
@@ -1,4 +1,4 @@
-/*	$OpenBSD: bio_chain.c,v 1.12 2022/12/10 10:42:13 tb Exp $	*/
+/*	$OpenBSD: bio_chain.c,v 1.13 2022/12/10 10:45:39 tb Exp $	*/
 /*
  * Copyright (c) 2022 Theo Buehler <tb@openbsd.org>
  *
@@ -192,78 +192,66 @@ bio_chain_pop_test(void)
 	return failed;
 }
 
-static int
-walk_forward(BIO *start, BIO *end, size_t expected_len, size_t i, size_t j,
-    const char *fn, const char *description)
+static void
+walk(BIO *(*step)(BIO *), BIO *start, BIO **end, size_t *len)
 {
-	BIO *prev, *next;
-	size_t len;
-	int ret = 0;
-
-	prev = NULL;
-	next = start;
-	len = 0;
+	BIO *current = NULL;
+	BIO *next = start;
 
+	*len = 0;
 	while (next != NULL) {
-		prev = next;
-		next = BIO_next(prev);
-		len++;
+		current = next;
+		next = step(current);
+		(*len)++;
 	}
+	*end = current;
+}
 
-	if (prev != end) {
-		fprintf(stderr, "%s case (%zu, %zu) %s has unexpected end\n",
-		    fn, i, j, description);
-		goto err;
+static int
+walk_report(BIO *last, BIO *expected_last, size_t len, size_t expected_len,
+    size_t i, size_t j, const char *fn, const char *description,
+    const char *direction, const char *last_name)
+{
+	if (last != expected_last) {
+		fprintf(stderr, "%s case (%zu, %zu) %s has unexpected %s\n",
+		    fn, i, j, description, last_name);
+		return 0;
 	}
 
 	if (len != expected_len) {
 		fprintf(stderr, "%s case (%zu, %zu) %s length "
-		    "(walking forward) want: %zu, got %zu\n",
-		    fn, i, j, description, expected_len, len);
-		goto err;
+		    "(walking %s) want: %zu, got %zu\n",
+		    fn, i, j, description, direction, expected_len, len);
+		return 0;
 	}
 
-	ret = 1;
-
- err:
-	return ret;
+	return 1;
 }
 
 static int
-walk_backward(BIO *start, BIO *end, size_t expected_len, size_t i, size_t j,
-    const char *fn, const char *description)
+walk_forward(BIO *start, BIO *expected_end, size_t expected_len,
+    size_t i, size_t j, const char *fn, const char *description)
 {
-	BIO *prev, *next;
+	BIO *end;
 	size_t len;
-	int ret = 0;
-
-	next = NULL;
-	prev = end;
-	len = 0;
 
-	while (prev != NULL) {
-		next = prev;
-		prev = BIO_prev(prev);
-		len++;
-	}
+	walk(BIO_next, start, &end, &len);
 
-	if (next != start) {
-		fprintf(stderr, "%s case (%zu, %zu) %s has unexpected start\n",
-		    fn, i, j, description);
-		goto err;
-	}
+	return walk_report(end, expected_end, len, expected_len,
+	    i, j, fn, description, "forward", "end");
+}
 
-	if (len != expected_len) {
-		fprintf(stderr, "%s case (%zu, %zu) %s length "
-		    "(walking backward) want: %zu, got %zu\n",
-		    fn, i, j, description, expected_len, len);
-		goto err;
-	}
+static int
+walk_backward(BIO *expected_start, BIO *end, size_t expected_len,
+    size_t i, size_t j, const char *fn, const char *description)
+{
+	BIO *start;
+	size_t len;
 
-	ret = 1;
+	walk(BIO_prev, end, &start, &len);
 
- err:
-	return ret;
+	return walk_report(start, expected_start, len, expected_len,
+	    i, j, fn, description, "backward", "start");
 }
 
 static int