From 7f3f686b1a53277b6fe714c5c76671cb233112cd Mon Sep 17 00:00:00 2001
From: Scott Haseley <shaseley@buffalo.edu>
Date: Mon, 28 Mar 2016 15:43:39 -0400
Subject: [PATCH] Updated km4 to write to, and check, each page allocated in
 the multi-page allocation. This should find cases where students ignore
 npages and just allocate one page, which would previously work.

---
 kern/test/kmalloctest.c | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/kern/test/kmalloctest.c b/kern/test/kmalloctest.c
index b3bf0cf..d016149 100644
--- a/kern/test/kmalloctest.c
+++ b/kern/test/kmalloctest.c
@@ -324,7 +324,7 @@ kmalloctest4thread(void *sm, unsigned long num)
 	struct semaphore *sem = sm;
 	void *ptrs[NUM_KM4_SIZES];
 	unsigned p, q;
-	unsigned i, j;
+	unsigned i, j, k;
 	uint32_t magic;
 
 	for (i=0; i<NUM_KM4_SIZES; i++) {
@@ -347,13 +347,19 @@ kmalloctest4thread(void *sm, unsigned long num)
 			      num, sizes[p]);
 		}
 
-		// Write to the allocated memory and make sure nothing overwrites it.
-		*(uint32_t *)ptrs[p] = magic;
+		// Write to each page of the allocated memory and make sure nothing
+		// overwrites it.
+		for (k = 0; k < sizes[p]; k++) {
+			*((uint32_t *)ptrs[p] + k*PAGE_SIZE/sizeof(uint32_t)) = magic;
+		}
+
 		for (j = 0; j < ITERATIONS; j++) {
 			random_yielder(4);
-			if ((*(uint32_t *)ptrs[p]) != magic) {
-				panic("km4: expected %u got %u. Your VM is broken!",
-					magic, (*(uint32_t *)ptrs[p]));
+			for (k = 0; k < sizes[p]; k++) {
+				if (*((uint32_t *)ptrs[p] + k*PAGE_SIZE/sizeof(uint32_t)) != magic) {
+					panic("km4: expected %u got %u. Your VM is broken!",
+						magic, (*(uint32_t *)ptrs[p]));
+				}
 			}
 		}
 		magic++;