SYS-OSS/FRET-qemu
4
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fallback on start cpustate when current_cpu is null

Browse Source
This commit is contained in:
Andrea Fioraldi 2021-10-04 10:42:37 +02:00
parent 78936b7cc7
commit 048ae77625
2 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
cpu.c
View File

@ -63,6 +63,8 @@ struct libafl_hook {
struct libafl_hook* libafl_qemu_hooks = NULL; struct libafl_hook* libafl_qemu_hooks = NULL;
CPUArchState *libafl_qemu_env;
void libafl_helper_table_add(TCGHelperInfo* info); void libafl_helper_table_add(TCGHelperInfo* info);
static GByteArray *libafl_qemu_mem_buf = NULL; static GByteArray *libafl_qemu_mem_buf = NULL;
@ -79,7 +81,10 @@ int libafl_qemu_write_reg(int reg, uint8_t* val)
{ {
CPUState *cpu = current_cpu; CPUState *cpu = current_cpu;
if (!cpu) { if (!cpu) {
return 0; cpu = env_cpu(libafl_qemu_env);
if (!cpu) {
return 0;
}
} }
CPUClass *cc = CPU_GET_CLASS(cpu); CPUClass *cc = CPU_GET_CLASS(cpu);
@ -93,7 +98,10 @@ int libafl_qemu_read_reg(int reg, uint8_t* val)
{ {
CPUState *cpu = current_cpu; CPUState *cpu = current_cpu;
if (!cpu) { if (!cpu) {
return 0; cpu = env_cpu(libafl_qemu_env);
if (!cpu) {
return 0;
}
} }
if (libafl_qemu_mem_buf == NULL) { if (libafl_qemu_mem_buf == NULL) {
@ -116,7 +124,10 @@ int libafl_qemu_num_regs(void)
{ {
CPUState *cpu = current_cpu; CPUState *cpu = current_cpu;
if (!cpu) { if (!cpu) {
return 0; cpu = env_cpu(libafl_qemu_env);
if (!cpu) {
return 0;
}
} }
CPUClass *cc = CPU_GET_CLASS(cpu); CPUClass *cc = CPU_GET_CLASS(cpu);

2
linux-user/main.c
View File

@ -630,7 +630,7 @@ uint64_t libafl_load_addr(void);
int libafl_qemu_main(void); int libafl_qemu_main(void);
int libafl_qemu_run(void); int libafl_qemu_run(void);
static CPUArchState *libafl_qemu_env; extern CPUArchState *libafl_qemu_env;
struct image_info libafl_image_info; struct image_info libafl_image_info;