diff options
| author | van Hauser <vh@thc.org> | 2020-08-31 10:57:01 +0200 |
|---|---|---|
| committer | van Hauser <vh@thc.org> | 2020-08-31 10:57:01 +0200 |
| commit | 4566bcf122c251c023abce0683666921bd4df755 (patch) | |
| tree | 2c8897ddf6d7e3b883e539dd49cddd9b00fd48de /llvm_mode/afl-clang-fast.c | |
| parent | ca0105ddf6bd3c28a7ea4f9a963f820b1eb55ac6 (diff) | |
| download | afl++-4566bcf122c251c023abce0683666921bd4df755.tar.gz | |
code-format
Diffstat (limited to 'llvm_mode/afl-clang-fast.c')
| -rw-r--r-- | llvm_mode/afl-clang-fast.c | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/llvm_mode/afl-clang-fast.c b/llvm_mode/afl-clang-fast.c index 6ea98111..173dc268 100644 --- a/llvm_mode/afl-clang-fast.c +++ b/llvm_mode/afl-clang-fast.c @@ -812,16 +812,24 @@ int main(int argc, char **argv, char **envp) { ptr += strlen("ngram"); while (*ptr && (*ptr < '0' || *ptr > '9')) { + ptr++; + } + if (!*ptr) { + ptr = getenv("AFL_LLVM_NGRAM_SIZE"); if (!ptr || !*ptr) { + FATAL( "you must set the NGRAM size with (e.g. for value 2) " "AFL_LLVM_INSTRUMENT=ngram-2"); + } + } + ngram_size = atoi(ptr); if (ngram_size < 2 || ngram_size > NGRAM_SIZE_MAX) FATAL( |