[fix] Skip locations without targets

misonijnik · Dec 16, 2022 · 19a57ba · 19a57ba
1 parent 0aa8b15
commit 19a57ba
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/lib/Core/TargetedExecutionManager.cpp b/lib/Core/TargetedExecutionManager.cpp
@@ -241,9 +241,11 @@ TargetedExecutionManager::prepareTargets(KModule *origModule, KModule *kmodule,
   std::unordered_map<KFunction *, ref<TargetForest>> whitelists;
   for (const auto &funcAndPaths : pathForest->layer) {
     auto le = funcAndPaths.first;
-    auto kf = (*loc2targets[le]->begin())->getBlock()->parent;
-    ref<TargetForest> whitelist = new TargetForest(funcAndPaths.second, loc2targets, broken_traces, kf);
-    whitelists[kf] = whitelist;
+    if (loc2targets.count(le)) {
+      auto kf = (*loc2targets.at(le)->begin())->getBlock()->parent;
+      ref<TargetForest> whitelist = new TargetForest(funcAndPaths.second, loc2targets, broken_traces, kf);
+      whitelists[kf] = whitelist;
+    }
   }
 
   for (auto p : loc2targets)