Add output reading
This commit is contained in:
parent
bd9237e8af
commit
d7dccf9c44
@ -107,6 +107,7 @@ int main(int argc, char **argv)
|
||||
|
||||
unsigned long deltas[input_size];
|
||||
u_int32_t inputs[input_size];
|
||||
u_int32_t outputs[input_size];
|
||||
//========= Instrumentation end
|
||||
qemu_init(argc, argv);
|
||||
//========= Instrumentation start
|
||||
@ -123,8 +124,9 @@ int main(int argc, char **argv)
|
||||
|
||||
uint8_t register_in_32b[4];
|
||||
uint8_t reg_tmp_val[4];
|
||||
uint8_t code_output[4];
|
||||
// load input
|
||||
//cpu_physical_memory_rw(target_addr, buffer, read_len, true);
|
||||
// cpu_physical_memory_rw(target_addr, buffer, read_len, true);
|
||||
|
||||
for (u_int32_t i = 0; i < input_size; i++)
|
||||
{
|
||||
@ -169,9 +171,18 @@ int main(int argc, char **argv)
|
||||
vm_start();
|
||||
qemu_main_loop();
|
||||
libafl_qemu_remove_native_breakpoint(end);
|
||||
|
||||
length = libafl_qemu_read_reg(cpu, 11, code_output);
|
||||
if(length != 4) {
|
||||
printf("Error: Could not read register\n");
|
||||
}
|
||||
u_int32_t output = (u_int32_t)code_output[0] | ((u_int32_t)code_output[1] << 8) | ((u_int32_t)code_output[2] << 16) | ((u_int32_t)code_output[3] << 24);
|
||||
printf("Output: %u\n", output);
|
||||
|
||||
unsigned long end_count = icount_get_raw();
|
||||
printf("End: %lu\n", end_count);
|
||||
inputs[i] = i;
|
||||
outputs[i] = output;
|
||||
deltas[i] = end_count - start_count;
|
||||
|
||||
printf("Delta: %lu\n", deltas[i]);
|
||||
@ -182,7 +193,7 @@ int main(int argc, char **argv)
|
||||
FILE *fptr = fopen(output_path, "w");
|
||||
|
||||
for (int i = 0; i < input_size; i++) {
|
||||
fprintf(fptr, "%d,%lu\n", inputs[i], deltas[i]);
|
||||
fprintf(fptr, "%d,%lu,%d\n", inputs[i], deltas[i],outputs[i]);
|
||||
}
|
||||
|
||||
fclose(fptr);
|
||||
|
Loading…
x
Reference in New Issue
Block a user