test/memguard-test.c

   1 /* Run this as: for i in $(seq 0 5); do prem-test $i & done */
   2
   3 #define _GNU_SOURCE         /* See feature_test_macros(7) */
   4 #include <err.h>
   5 #include <sched.h>
   6 #include <stdint.h>
   7 #include <stdio.h>
   8 #include <stdlib.h>
   9 #include <unistd.h>
  10 #include <time.h>
  11
  12 #ifndef SYS_prem_memguard_check
  13 #define SYS_prem_memguard_check 793
  14 #endif
  15
  16 #define RANDOM_ARRAY_SIZE 16777216 // 16M * 8 = 128 MB array
  17
  18 uint64_t random_values[RANDOM_ARRAY_SIZE];
  19 volatile uint64_t sum;
  20
  21 enum prem_phase {
  22         PREM_COMPATIBLE = 0,
  23         PREM_MEMORY     = 1,
  24         PREM_COMPUTE    = 2,
  25 };
  26
  27 void compute_kernel(int time_ms){
  28     struct timespec ts;
  29     uint64_t current_us, end_us;
  30     clock_gettime(CLOCK_MONOTONIC, &ts);
  31     end_us = ts.tv_sec * 1000000 + ts.tv_nsec / 1000 + time_ms * 1000;
  32     do{
  33         clock_gettime(CLOCK_MONOTONIC, &ts);
  34         current_us = ts.tv_sec * 1000000 + ts.tv_nsec / 1000;
  35     }while(current_us < end_us);
  36 }
  37
  38 long memory_kernel(long lines){
  39     for(int i = 0; i < lines; i++){
  40         sum += random_values[(i*8) % RANDOM_ARRAY_SIZE];
  41     }
  42     return sum;
  43 }
  44
  45 void random_memory_kernel(long lines) {
  46     uint32_t index;
  47
  48     for(long i=0; i<lines; i++) {
  49         index = (uint32_t)rand();
  50         index = index % RANDOM_ARRAY_SIZE;
  51         sum += random_values[index];
  52     }
  53 }
  54
  55 long prem_memguard_check(unsigned long phase,
  56                     unsigned long memory_budget,
  57                     unsigned long timeout)
  58 {
  59         return syscall(SYS_prem_memguard_check, phase, memory_budget, timeout);
  60 }
  61
  62 int main(int argc, char *argv[])
  63 {
  64     cpu_set_t set;
  65         int cpu = 0;
  66
  67         if (argc > 1)
  68                 cpu = atoi(argv[1]);
  69
  70         /* Ensure that memory phase starts and ends on the same CPU */
  71         CPU_ZERO(&set);
  72         CPU_SET(cpu, &set);
  73         if (sched_setaffinity(getpid(), sizeof(set), &set) < 0)
  74                 err(1, "sched_setaffinity");
  75         printf("Pinned to CPU %d\n", cpu);
  76
  77     printf("X1\n");
  78     fflush(stdout);
  79         //prem_memguard_check(950000, 1933440000, 1);
  80     prem_memguard_check(950000, 1933440000, 1);
  81     compute_kernel(1000);
  82     prem_memguard_check(0, 0, 1);
  83
  84     printf("X2\n");
  85     fflush(stdout);
  86     //prem_memguard_check(1050000, 2136960000, 1);
  87     prem_memguard_check(1050000, 2136960000, 1);
  88     compute_kernel(1000);
  89     prem_memguard_check(0, 0, 1);
  90
  91     printf("X3\n");
  92     fflush(stdout);
  93     //prem_memguard_check(1000000, 500, 1);
  94     prem_memguard_check(1000000, 5000000, 1);
  95     memory_kernel(100000000);
  96     prem_memguard_check(0, 0, 1);
  97
  98     printf("X4\n");
  99     fflush(stdout);
 100     //prem_memguard_check(1000000, 1500, 1);
 101     prem_memguard_check(1000000, 15000000, 1);
 102     memory_kernel(100000000);
 103     prem_memguard_check(0, 0, 1);
 104
 105     printf("X5\n");
 106     fflush(stdout);
 107     //prem_memguard_check(1000000, 10000, 1);
 108     prem_memguard_check(1000000, 100000, 1);
 109     random_memory_kernel(100000000);
 110     prem_memguard_check(0, 0, 1);
 111
 112     printf("X6\n");
 113     fflush(stdout);
 114     //prem_memguard_check(1000000, 10000000, 1);
 115     prem_memguard_check(1000000, 100000000, 1);
 116     random_memory_kernel(100000000);
 117     prem_memguard_check(0, 0, 1);
 118
 119         return 0;
 120 }