X-Git-Url: http://rtime.felk.cvut.cz/gitweb/jailhouse.git/blobdiff_plain/1a0880cb726cf46e2d7c311b034090ae6e92771f..6916406a2c23f9d65723d76b2cf2c946e8361277:/inmates/demos/x86/Membench.c

diff --git a/inmates/demos/x86/Membench.c b/inmates/demos/x86/Membench.c
index 4366a30..f13b159 100644
--- a/inmates/demos/x86/Membench.c
+++ b/inmates/demos/x86/Membench.c
@@ -430,6 +430,8 @@ static void run_benchmark(struct cfg *cfg)
 }
 
 #ifdef JAILHOUSE
+const void * image_end;
+
 void inmate_main(void)
 #else //Linux
 int main(int argc, char *argv[])
@@ -476,8 +478,8 @@ int main(int argc, char *argv[])
 	       tsc_freq % 1000);
 
 	u8 * start_memreg = (u8 *) array;
-
-	while ( ((u64) start_memreg )< 0x4000000) {
+	u64 end_memreg_addr = (u64) &image_end;
+	while ( ((u64) start_memreg )< end_memreg_addr) {
 		//printk("%p\n",start_memreg);
 		map_range(start_memreg, HUGE_PAGE_SIZE, MAP_CACHED);
 		start_memreg += HUGE_PAGE_SIZE;
@@ -552,6 +554,6 @@ int main(int argc, char *argv[])
 	return 0;
 #else
 	comm_region->cell_state = JAILHOUSE_CELL_SHUT_DOWN;
-	printk("done!");
+	printk("done!\n");
 #endif
 }