Enhance the benchmark code a little. Add image_end symbol.

[jailhouse.git] / inmates / demos / x86 / Membench.c

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