From 89851077da34f052819b2da026bf881f3ae0eac7 Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Fri, 29 Mar 2002 03:43:24 +0000 Subject: [PATCH] Simplify code a bit by using Module::getOrInsertFunction git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@2032 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Transforms/Instrumentation/TraceValues.cpp | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/lib/Transforms/Instrumentation/TraceValues.cpp b/lib/Transforms/Instrumentation/TraceValues.cpp index 9bbf7e580ea..e064caa8176 100644 --- a/lib/Transforms/Instrumentation/TraceValues.cpp +++ b/lib/Transforms/Instrumentation/TraceValues.cpp @@ -14,7 +14,6 @@ #include "llvm/iOther.h" #include "llvm/Function.h" #include "llvm/Module.h" -#include "llvm/SymbolTable.h" #include "llvm/Pass.h" #include "llvm/Assembly/Writer.h" #include "Support/StringExtras.h" @@ -67,20 +66,12 @@ Pass *createTraceValuesPassForBasicBlocks() { // Trace BB's and methods // Add a prototype for printf if it is not already in the program. // bool InsertTraceCode::doInitialization(Module *M) { - SymbolTable *ST = M->getSymbolTable(); const Type *SBP = PointerType::get(Type::SByteTy); const MethodType *MTy = MethodType::get(Type::IntTy, vector(1, SBP), true); - if (Value *Func = ST->lookup(PointerType::get(MTy), "printf")) { - PrintfFunc = cast(Func); - return false; - } - - // Create a new method and add it to the module - PrintfFunc = new Function(MTy, false, "printf"); - M->getFunctionList().push_back(PrintfFunc); - return true; + PrintfFunc = M->getOrInsertFunction("printf", MTy); + return false; } -- 2.34.1