From 88e1e0f90eda8c562f5dcf5eaafc45cf2de3522f Mon Sep 17 00:00:00 2001
From: Martin Nowack <m.nowack@imperial.ac.uk>
Date: Thu, 6 Sep 2018 11:29:25 +0100
Subject: Unify the error message if that function has not been found.

---
 lib/Module/KModule.cpp    | 3 ++-
 lib/Module/ModuleUtil.cpp | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

(limited to 'lib/Module')

diff --git a/lib/Module/KModule.cpp b/lib/Module/KModule.cpp
index 07f3572e..49a1ba69 100644
--- a/lib/Module/KModule.cpp
+++ b/lib/Module/KModule.cpp
@@ -153,7 +153,8 @@ injectStaticConstructorsAndDestructors(Module *m,
 
   Function *mainFn = m->getFunction(entryFunction);
   if (!mainFn)
-    klee_error("Could not find main() function.");
+    klee_error("Entry function '%s' not found in module.",
+               entryFunction.str().c_str());
 
   if (ctors) {
 #if LLVM_VERSION_CODE >= LLVM_VERSION(3, 8)
diff --git a/lib/Module/ModuleUtil.cpp b/lib/Module/ModuleUtil.cpp
index 5ca0a55b..5f9a12b7 100644
--- a/lib/Module/ModuleUtil.cpp
+++ b/lib/Module/ModuleUtil.cpp
@@ -223,7 +223,8 @@ klee::linkModules(std::vector<std::unique_ptr<llvm::Module>> &modules,
 
   // fail if not found
   if (!composite) {
-    errorMsg = "'" + entryFunction.str() + "' function not found in module.";
+    errorMsg =
+        "Entry function '" + entryFunction.str() + "' not found in module.";
     return nullptr;
   }
 
-- 
cgit 1.4.1