#include <stdlib.h>
#include <unistd.h>
#include <time.h>
+#include <inttypes.h>
+#include <sys/mman.h>
#ifndef SYS_prem_memguard_check
#define SYS_prem_memguard_check 793
#endif
+/* Return value:
+ * 0-31 - Total time
+ * 32-55 - Sum of memory events
+ * 61 - Memory budget overrun
+ * 62 - Timeout
+ * 63 - Error */
+#define MGRET_TIM_POS 0
+#define MGRET_MEM_POS 32
+#define MGRET_OVER_MEM_POS 61
+#define MGRET_OVER_TIM_POS 62
+
+#define MGRET_TIM_MASK 0xFFFFFFFFlu
+#define MGRET_MEM_MASK 0xFFFFFFlu
+#define MGRET_OVER_MEM_MASK 0x1lu
+#define MGRET_OVER_TIM_MASK 0x1lu
+
#define RANDOM_ARRAY_SIZE 16777216 // 16M * 8 = 128 MB array
uint64_t random_values[RANDOM_ARRAY_SIZE];
-volatile uint64_t sum;
+
enum prem_phase {
PREM_COMPATIBLE = 0,
PREM_COMPUTE = 2,
};
+struct test_case{
+ uint64_t budget_time;
+ uint64_t budget_memory;
+ uint8_t flags;
+ uint64_t measured_time;
+ uint64_t measured_memory;
+ uint8_t status;
+};
+
+
+
+static inline void decode_retval(uint64_t retval, uint64_t * timval, uint64_t * cntval){
+ *timval = (retval >> MGRET_TIM_POS) & MGRET_TIM_MASK;
+ *cntval = (retval >> MGRET_MEM_POS) & MGRET_MEM_MASK;
+}
+
void compute_kernel(int time_ms){
struct timespec ts;
uint64_t current_us, end_us;
clock_gettime(CLOCK_MONOTONIC, &ts);
end_us = ts.tv_sec * 1000000 + ts.tv_nsec / 1000 + time_ms * 1000;
- do{
- clock_gettime(CLOCK_MONOTONIC, &ts);
- current_us = ts.tv_sec * 1000000 + ts.tv_nsec / 1000;
- }while(current_us < end_us);
+ do {
+ clock_gettime(CLOCK_MONOTONIC, &ts);
+ current_us = ts.tv_sec * 1000000 + ts.tv_nsec / 1000;
+ } while (current_us < end_us);
}
long memory_kernel(long lines){
+ uint64_t sum = 0;
+
for(int i = 0; i < lines; i++){
sum += random_values[(i*8) % RANDOM_ARRAY_SIZE];
}
void random_memory_kernel(long lines) {
uint32_t index;
-
+ uint64_t sum = 0;
+
for(long i=0; i<lines; i++) {
index = (uint32_t)rand();
index = index % RANDOM_ARRAY_SIZE;
}
}
-long prem_memguard_check(unsigned long phase,
+/* Memguard flags */
+#define MGF_PERIODIC (1 << 0) /* Chooses between periodic or one-shot budget replenishment */
+#define MGF_MASK_INT (1 << 1) /* Mask (disable) low priority interrupts until next memguard call */
+
+long prem_memguard_check(unsigned long timeout,
unsigned long memory_budget,
- unsigned long timeout)
+ unsigned long flags)
{
- return syscall(SYS_prem_memguard_check, phase, memory_budget, timeout);
+ return syscall(SYS_prem_memguard_check, timeout, memory_budget, flags);
}
+#define MGTEST(timeout_us, mem_budget, flags, code) \
+ do { \
+ long retval = prem_memguard_check(timeout_us, mem_budget, flags); \
+ code; \
+ retval = prem_memguard_check(0, 0, 0); \
+ printf("%-31s flags:%d m_ovr:%ld t_ovr:%ld mem:%d->%ld time:%d->%ld\n", \
+ #code, flags, \
+ (retval >> MGRET_OVER_MEM_POS) & 1, \
+ (retval >> MGRET_OVER_TIM_POS) & 1, \
+ mem_budget, (retval >> MGRET_MEM_POS) & MGRET_MEM_MASK, \
+ timeout_us, (retval >> MGRET_TIM_POS) & MGRET_TIM_MASK); \
+ } while (0)
+
int main(int argc, char *argv[])
{
cpu_set_t set;
- int cpu = 0;
-
- if (argc > 1)
- cpu = atoi(argv[1]);
-
- /* Ensure that memory phase starts and ends on the same CPU */
- CPU_ZERO(&set);
- CPU_SET(cpu, &set);
- if (sched_setaffinity(getpid(), sizeof(set), &set) < 0)
- err(1, "sched_setaffinity");
- printf("Pinned to CPU %d\n", cpu);
-
- printf("X1\n");
- fflush(stdout);
- //prem_memguard_check(950000, 1933440000, 1);
- prem_memguard_check(950000, 1933440000, 1);
- compute_kernel(1000);
- prem_memguard_check(0, 0, 1);
-
- printf("X2\n");
- fflush(stdout);
- //prem_memguard_check(1050000, 2136960000, 1);
- prem_memguard_check(1050000, 2136960000, 1);
- compute_kernel(1000);
- prem_memguard_check(0, 0, 1);
-
- printf("X3\n");
- fflush(stdout);
- //prem_memguard_check(1000000, 500, 1);
- prem_memguard_check(1000000, 5000000, 1);
- memory_kernel(100000000);
- prem_memguard_check(0, 0, 1);
-
- printf("X4\n");
- fflush(stdout);
- //prem_memguard_check(1000000, 1500, 1);
- prem_memguard_check(1000000, 15000000, 1);
- memory_kernel(100000000);
- prem_memguard_check(0, 0, 1);
-
- printf("X5\n");
- fflush(stdout);
- //prem_memguard_check(1000000, 10000, 1);
- prem_memguard_check(1000000, 100000, 1);
- random_memory_kernel(100000000);
- prem_memguard_check(0, 0, 1);
-
- printf("X6\n");
- fflush(stdout);
- //prem_memguard_check(1000000, 10000000, 1);
- prem_memguard_check(1000000, 100000000, 1);
- random_memory_kernel(100000000);
- prem_memguard_check(0, 0, 1);
-
- return 0;
+ int cpu = 0;
+
+ char bla[1];
+
+ for (int i = 0; i < RANDOM_ARRAY_SIZE; i++)
+ random_values[i] = i;
+
+ if (argc > 1)
+ cpu = atoi(argv[1]);
+
+ /* Ensure that memory phase starts and ends on the same CPU */
+ CPU_ZERO(&set);
+ CPU_SET(cpu, &set);
+ if (sched_setaffinity(getpid(), sizeof(set), &set) < 0)
+ err(1, "sched_setaffinity");
+ printf("Pinned to CPU %d\n", cpu);
+
+ volatile long x;
+ MGTEST(100000, 1000, MGF_PERIODIC | MGF_MASK_INT, compute_kernel(1000));
+ MGTEST(1050000, 2136960000, 0, compute_kernel(1000));
+ MGTEST(100000, 500, MGF_PERIODIC, memory_kernel(100000000));
+
+ for (int i = 1; i <= 10; i++)
+ MGTEST(1000000, 500, 0, x=memory_kernel(1000000));
+
+ MGTEST(1000000, 500, 0, x=memory_kernel(100000000));
+ MGTEST(1000000, 100000, 0, random_memory_kernel(100000000));
+ MGTEST(1000000, 100000000, 0, random_memory_kernel(100000000));
+
+ return 0;
}