From 72281ecda814ad952b868ac61f018507916dd6da Mon Sep 17 00:00:00 2001
From: Russ Cox <rsc@swtch.com>
Date: Mon, 24 Sep 2007 22:29:16 -0400
Subject: venti/randtest: compute totalbytes correctly

---
 src/cmd/venti/randtest.c | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/cmd/venti/randtest.c b/src/cmd/venti/randtest.c
index 2a1fa6ee..7d94eef2 100644
--- a/src/cmd/venti/randtest.c
+++ b/src/cmd/venti/randtest.c
@@ -96,8 +96,9 @@ run(void (*fn)(char*, char*), Channel *c)
 	buf = vtmalloc(blocksize);
 	cur = 0;
 	packets = totalbytes/blocksize;
-	if(maxpackets == 0)
-		maxpackets = packets;
+	if(maxpackets > 0 && maxpackets < packets)
+		packets = maxpackets;
+	totalbytes = (vlong)packets * blocksize;
 	order = vtmalloc(packets*sizeof order[0]);
 	for(i=0; i<packets; i++)
 		order[i] = i;
@@ -109,7 +110,7 @@ run(void (*fn)(char*, char*), Channel *c)
 			order[j] = t;
 		}
 	}
-	for(i=0; i<packets && i<maxpackets; i++){
+	for(i=0; i<packets; i++){
 		memmove(buf, template, blocksize);
 		*(uint*)buf = order[i];
 		if(c){
-- 
cgit v1.2.3