From fa4241f4f80523a78fcf5fc2654faa7fc8135171 Mon Sep 17 00:00:00 2001
From: Dominik Maier <domenukk@gmail.com>
Date: Fri, 12 Jul 2024 12:46:42 +0200
Subject: [PATCH] Make atheris load multiple tokens files (#2382)

---
 fuzzers/libafl_atheris/src/lib.rs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/fuzzers/libafl_atheris/src/lib.rs b/fuzzers/libafl_atheris/src/lib.rs
index 2ca552d22d..d6e4ee2a1b 100644
--- a/fuzzers/libafl_atheris/src/lib.rs
+++ b/fuzzers/libafl_atheris/src/lib.rs
@@ -176,11 +176,11 @@ pub extern "C" fn LLVMFuzzerRunDriver(
         });
 
         // Create a dictionary if not existing
-        if state.metadata_map().get::<Tokens>().is_none() {
-            for tokens_file in &token_files {
-                state.add_metadata(Tokens::from_file(tokens_file)?);
-            }
-        }
+        state.metadata_or_insert_with(|| {
+            Tokens::new()
+                .add_from_files(&token_files)
+                .expect("Could not read tokens files.")
+        });
 
         // A minimization+queue policy to get testcasess from the corpus
         let scheduler =