ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-19 13:03:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix for afl-showmap with -Q

Browse Source
This commit is contained in:
van Hauser
2020-04-27 19:30:32 +02:00
parent a46fe3ad43
commit a56354a893
3 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/afl-showmap.c
View File

@ -714,7 +714,6 @@ int main(int argc, char **argv_orig, char **envp) {
case 'Q':
if (fsrv->qemu_mode) { FATAL("Multiple -Q options not supported"); }
if (!mem_limit_given) { fsrv->mem_limit = MEM_LIMIT_QEMU; }
fsrv->qemu_mode = 1;
break;
@ -722,7 +721,6 @@ int main(int argc, char **argv_orig, char **envp) {
case 'U':
if (unicorn_mode) { FATAL("Multiple -U options not supported"); }
if (!mem_limit_given) { fsrv->mem_limit = MEM_LIMIT_UNICORN; }
unicorn_mode = 1;
break;
@ -733,8 +731,6 @@ int main(int argc, char **argv_orig, char **envp) {
fsrv->qemu_mode = 1;
use_wine = 1;
if (!mem_limit_given) { fsrv->mem_limit = 0; }
break;
case 'b':
@ -772,6 +768,9 @@ int main(int argc, char **argv_orig, char **envp) {
if (optind == argc || !out_file) { usage(argv[0]); }
if (fsrv->qemu_mode && !mem_limit_given) { fsrv->mem_limit = MEM_LIMIT_QEMU; }
if (unicorn_mode && !mem_limit_given) { fsrv->mem_limit = MEM_LIMIT_UNICORN; }
check_environment_vars(envp);
sharedmem_t shm = {0};
@ -780,6 +779,14 @@ int main(int argc, char **argv_orig, char **envp) {
set_up_environment(fsrv);
i = 0;
while (argv[i] != NULL && !arg_offset) {
if (strcmp(argv[i], "@@") == 0) arg_offset = i;
i++;
}
fsrv->target_path = find_binary(argv[optind]);
if (!quiet_mode) {
@ -800,12 +807,6 @@ int main(int argc, char **argv_orig, char **envp) {
}
for (i = optind; i < argc; i++) {
if (strcmp(argv[i], "@@") == 0) { arg_offset = i; }
}
if (fsrv->qemu_mode) {
if (use_wine) {
@ -826,6 +827,15 @@ int main(int argc, char **argv_orig, char **envp) {
}
int arg_offset2 = 0;
i = 0;
while (use_argv[i] != NULL && !arg_offset2) {
if (strcmp(use_argv[i], "@@") == 0) { arg_offset2 = i; }
i++;
}
if (in_dir) {
DIR * dir_in, *dir_out;
@ -870,10 +880,11 @@ int main(int argc, char **argv_orig, char **envp) {
fsrv->out_fd = open(stdin_file, O_RDWR | O_CREAT | O_EXCL, 0600);
if (fsrv->out_fd < 0) { PFATAL("Unable to create '%s'", out_file); }
if (arg_offset && argv[arg_offset] != stdin_file) {
if (arg_offset && use_argv[arg_offset] != stdin_file) {
ck_free(argv[arg_offset]);
argv[arg_offset] = strdup(stdin_file);
use_argv[arg_offset2] = argv[arg_offset];
}
@ -888,8 +899,6 @@ int main(int argc, char **argv_orig, char **envp) {
}
SAYF("\n");
SAYF(cMGN "[D]" cRST " %d - %d = %d, %s\n", arg_offset, optind,
arg_offset - optind, infile);
}