blob: 855041db2fa974d1934a70709c6a916f39370673 [file] [log] [blame]
Reid Spencer9bbaa2a2005-04-25 03:59:26 +00001//===- SimplifyLibCalls.cpp - Optimize specific well-known library calls --===//
Reid Spencer39a762d2005-04-25 02:53:12 +00002//
3// The LLVM Compiler Infrastructure
4//
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00005// This file was developed by Reid Spencer and is distributed under the
Reid Spencer9bbaa2a2005-04-25 03:59:26 +00006// University of Illinois Open Source License. See LICENSE.TXT for details.
Reid Spencer39a762d2005-04-25 02:53:12 +00007//
8//===----------------------------------------------------------------------===//
9//
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +000010// This file implements a module pass that applies a variety of small
11// optimizations for calls to specific well-known function calls (e.g. runtime
12// library functions). For example, a call to the function "exit(3)" that
Reid Spencer0b13cda2005-05-21 00:57:44 +000013// occurs within the main() function can be transformed into a simple "return 3"
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +000014// instruction. Any optimization that takes this form (replace call to library
15// function with simpler code that provides the same result) belongs in this
16// file.
Reid Spencer39a762d2005-04-25 02:53:12 +000017//
18//===----------------------------------------------------------------------===//
19
Reid Spencer18b99812005-04-26 23:05:17 +000020#define DEBUG_TYPE "simplify-libcalls"
Reid Spencer2bc7a4f2005-04-26 23:02:16 +000021#include "llvm/Constants.h"
22#include "llvm/DerivedTypes.h"
23#include "llvm/Instructions.h"
Reid Spencer39a762d2005-04-25 02:53:12 +000024#include "llvm/Module.h"
25#include "llvm/Pass.h"
Reid Spencer9bbaa2a2005-04-25 03:59:26 +000026#include "llvm/ADT/hash_map"
Reid Spencer2bc7a4f2005-04-26 23:02:16 +000027#include "llvm/ADT/Statistic.h"
Reid Spencerade18212006-01-19 08:36:56 +000028#include "llvm/Config/config.h"
Reid Spencer2bc7a4f2005-04-26 23:02:16 +000029#include "llvm/Support/Debug.h"
Reid Spencerbb92b4f2005-04-26 19:13:17 +000030#include "llvm/Target/TargetData.h"
Reid Spencer2bc7a4f2005-04-26 23:02:16 +000031#include "llvm/Transforms/IPO.h"
Reid Spencer39a762d2005-04-25 02:53:12 +000032using namespace llvm;
33
34namespace {
Reid Spencer39a762d2005-04-25 02:53:12 +000035
Reid Spencere249a822005-04-27 07:54:40 +000036/// This statistic keeps track of the total number of library calls that have
37/// been simplified regardless of which call it is.
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +000038Statistic<> SimplifiedLibCalls("simplify-libcalls",
Chris Lattner579b20b2005-08-07 20:02:04 +000039 "Number of library calls simplified");
Reid Spencer39a762d2005-04-25 02:53:12 +000040
Reid Spencer7ddcfb32005-04-27 21:29:20 +000041// Forward declarations
Reid Spencere249a822005-04-27 07:54:40 +000042class LibCallOptimization;
43class SimplifyLibCalls;
Reid Spencer7ddcfb32005-04-27 21:29:20 +000044
Chris Lattner33081b42006-01-22 23:10:26 +000045/// This list is populated by the constructor for LibCallOptimization class.
Reid Spencer9fbad132005-05-21 01:27:04 +000046/// Therefore all subclasses are registered here at static initialization time
47/// and this list is what the SimplifyLibCalls pass uses to apply the individual
48/// optimizations to the call sites.
Reid Spencer7ddcfb32005-04-27 21:29:20 +000049/// @brief The list of optimizations deriving from LibCallOptimization
Chris Lattner33081b42006-01-22 23:10:26 +000050static LibCallOptimization *OptList = 0;
Reid Spencer39a762d2005-04-25 02:53:12 +000051
Reid Spencere249a822005-04-27 07:54:40 +000052/// This class is the abstract base class for the set of optimizations that
Reid Spencer7ddcfb32005-04-27 21:29:20 +000053/// corresponds to one library call. The SimplifyLibCalls pass will call the
Reid Spencere249a822005-04-27 07:54:40 +000054/// ValidateCalledFunction method to ask the optimization if a given Function
Reid Spencer7ddcfb32005-04-27 21:29:20 +000055/// is the kind that the optimization can handle. If the subclass returns true,
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +000056/// then SImplifyLibCalls will also call the OptimizeCall method to perform,
Reid Spencer7ddcfb32005-04-27 21:29:20 +000057/// or attempt to perform, the optimization(s) for the library call. Otherwise,
58/// OptimizeCall won't be called. Subclasses are responsible for providing the
59/// name of the library call (strlen, strcpy, etc.) to the LibCallOptimization
60/// constructor. This is used to efficiently select which call instructions to
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +000061/// optimize. The criteria for a "lib call" is "anything with well known
Reid Spencer7ddcfb32005-04-27 21:29:20 +000062/// semantics", typically a library function that is defined by an international
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +000063/// standard. Because the semantics are well known, the optimizations can
Reid Spencer7ddcfb32005-04-27 21:29:20 +000064/// generally short-circuit actually calling the function if there's a simpler
65/// way (e.g. strlen(X) can be reduced to a constant if X is a constant global).
Reid Spencere249a822005-04-27 07:54:40 +000066/// @brief Base class for library call optimizations
Chris Lattner0d4ebfc2006-01-22 22:35:08 +000067class LibCallOptimization {
Chris Lattner33081b42006-01-22 23:10:26 +000068 LibCallOptimization **Prev, *Next;
69 const char *FunctionName; ///< Name of the library call we optimize
70#ifndef NDEBUG
71 Statistic<> occurrences; ///< debug statistic (-debug-only=simplify-libcalls)
72#endif
Jeff Cohen4bc952f2005-04-29 03:05:44 +000073public:
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +000074 /// The \p fname argument must be the name of the library function being
Reid Spencer7ddcfb32005-04-27 21:29:20 +000075 /// optimized by the subclass.
76 /// @brief Constructor that registers the optimization.
Chris Lattner33081b42006-01-22 23:10:26 +000077 LibCallOptimization(const char *FName, const char *Description)
78 : FunctionName(FName)
Reid Spencere95a6472005-04-27 00:05:45 +000079#ifndef NDEBUG
Chris Lattner33081b42006-01-22 23:10:26 +000080 , occurrences("simplify-libcalls", Description)
Reid Spencere95a6472005-04-27 00:05:45 +000081#endif
Reid Spencer39a762d2005-04-25 02:53:12 +000082 {
Chris Lattner33081b42006-01-22 23:10:26 +000083 // Register this optimizer in the list of optimizations.
84 Next = OptList;
85 OptList = this;
86 Prev = &OptList;
87 if (Next) Next->Prev = &Next;
Reid Spencer39a762d2005-04-25 02:53:12 +000088 }
Chris Lattner33081b42006-01-22 23:10:26 +000089
90 /// getNext - All libcall optimizations are chained together into a list,
91 /// return the next one in the list.
92 LibCallOptimization *getNext() { return Next; }
Reid Spencer39a762d2005-04-25 02:53:12 +000093
Reid Spencer7ddcfb32005-04-27 21:29:20 +000094 /// @brief Deregister from the optlist
Chris Lattner33081b42006-01-22 23:10:26 +000095 virtual ~LibCallOptimization() {
96 *Prev = Next;
97 if (Next) Next->Prev = Prev;
98 }
Reid Spencer8ee5aac2005-04-26 03:26:15 +000099
Reid Spencere249a822005-04-27 07:54:40 +0000100 /// The implementation of this function in subclasses should determine if
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000101 /// \p F is suitable for the optimization. This method is called by
102 /// SimplifyLibCalls::runOnModule to short circuit visiting all the call
103 /// sites of such a function if that function is not suitable in the first
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000104 /// place. If the called function is suitabe, this method should return true;
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000105 /// false, otherwise. This function should also perform any lazy
106 /// initialization that the LibCallOptimization needs to do, if its to return
Reid Spencere249a822005-04-27 07:54:40 +0000107 /// true. This avoids doing initialization until the optimizer is actually
108 /// going to be called upon to do some optimization.
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000109 /// @brief Determine if the function is suitable for optimization
Reid Spencere249a822005-04-27 07:54:40 +0000110 virtual bool ValidateCalledFunction(
111 const Function* F, ///< The function that is the target of call sites
112 SimplifyLibCalls& SLC ///< The pass object invoking us
113 ) = 0;
Reid Spencerbb92b4f2005-04-26 19:13:17 +0000114
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000115 /// The implementations of this function in subclasses is the heart of the
116 /// SimplifyLibCalls algorithm. Sublcasses of this class implement
Reid Spencere249a822005-04-27 07:54:40 +0000117 /// OptimizeCall to determine if (a) the conditions are right for optimizing
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000118 /// the call and (b) to perform the optimization. If an action is taken
Reid Spencere249a822005-04-27 07:54:40 +0000119 /// against ci, the subclass is responsible for returning true and ensuring
120 /// that ci is erased from its parent.
Reid Spencere249a822005-04-27 07:54:40 +0000121 /// @brief Optimize a call, if possible.
122 virtual bool OptimizeCall(
123 CallInst* ci, ///< The call instruction that should be optimized.
124 SimplifyLibCalls& SLC ///< The pass object invoking us
125 ) = 0;
Reid Spencerbb92b4f2005-04-26 19:13:17 +0000126
Reid Spencere249a822005-04-27 07:54:40 +0000127 /// @brief Get the name of the library call being optimized
Chris Lattner33081b42006-01-22 23:10:26 +0000128 const char *getFunctionName() const { return FunctionName; }
Reid Spencerbb92b4f2005-04-26 19:13:17 +0000129
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000130 /// @brief Called by SimplifyLibCalls to update the occurrences statistic.
Chris Lattner33081b42006-01-22 23:10:26 +0000131 void succeeded() {
Reid Spencere249a822005-04-27 07:54:40 +0000132#ifndef NDEBUG
Chris Lattner33081b42006-01-22 23:10:26 +0000133 DEBUG(++occurrences);
Reid Spencere249a822005-04-27 07:54:40 +0000134#endif
Chris Lattner33081b42006-01-22 23:10:26 +0000135 }
Reid Spencere249a822005-04-27 07:54:40 +0000136};
137
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000138/// This class is an LLVM Pass that applies each of the LibCallOptimization
Reid Spencere249a822005-04-27 07:54:40 +0000139/// instances to all the call sites in a module, relatively efficiently. The
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000140/// purpose of this pass is to provide optimizations for calls to well-known
Reid Spencere249a822005-04-27 07:54:40 +0000141/// functions with well-known semantics, such as those in the c library. The
Chris Lattner4201cd12005-08-24 17:22:17 +0000142/// class provides the basic infrastructure for handling runOnModule. Whenever
143/// this pass finds a function call, it asks the appropriate optimizer to
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000144/// validate the call (ValidateLibraryCall). If it is validated, then
145/// the OptimizeCall method is also called.
Reid Spencere249a822005-04-27 07:54:40 +0000146/// @brief A ModulePass for optimizing well-known function calls.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000147class SimplifyLibCalls : public ModulePass {
Jeff Cohen4bc952f2005-04-29 03:05:44 +0000148public:
Reid Spencere249a822005-04-27 07:54:40 +0000149 /// We need some target data for accurate signature details that are
150 /// target dependent. So we require target data in our AnalysisUsage.
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000151 /// @brief Require TargetData from AnalysisUsage.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000152 virtual void getAnalysisUsage(AnalysisUsage& Info) const {
Reid Spencere249a822005-04-27 07:54:40 +0000153 // Ask that the TargetData analysis be performed before us so we can use
154 // the target data.
155 Info.addRequired<TargetData>();
156 }
157
158 /// For this pass, process all of the function calls in the module, calling
159 /// ValidateLibraryCall and OptimizeCall as appropriate.
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000160 /// @brief Run all the lib call optimizations on a Module.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000161 virtual bool runOnModule(Module &M) {
Reid Spencere249a822005-04-27 07:54:40 +0000162 reset(M);
163
164 bool result = false;
Chris Lattner33081b42006-01-22 23:10:26 +0000165 hash_map<std::string, LibCallOptimization*> OptznMap;
166 for (LibCallOptimization *Optzn = OptList; Optzn; Optzn = Optzn->getNext())
167 OptznMap[Optzn->getFunctionName()] = Optzn;
Reid Spencere249a822005-04-27 07:54:40 +0000168
169 // The call optimizations can be recursive. That is, the optimization might
170 // generate a call to another function which can also be optimized. This way
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000171 // we make the LibCallOptimization instances very specific to the case they
172 // handle. It also means we need to keep running over the function calls in
Reid Spencere249a822005-04-27 07:54:40 +0000173 // the module until we don't get any more optimizations possible.
174 bool found_optimization = false;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000175 do {
Reid Spencere249a822005-04-27 07:54:40 +0000176 found_optimization = false;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000177 for (Module::iterator FI = M.begin(), FE = M.end(); FI != FE; ++FI) {
Reid Spencere249a822005-04-27 07:54:40 +0000178 // All the "well-known" functions are external and have external linkage
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000179 // because they live in a runtime library somewhere and were (probably)
180 // not compiled by LLVM. So, we only act on external functions that
Reid Spencer38cabd72005-05-03 07:23:44 +0000181 // have external linkage and non-empty uses.
Reid Spencere249a822005-04-27 07:54:40 +0000182 if (!FI->isExternal() || !FI->hasExternalLinkage() || FI->use_empty())
183 continue;
184
185 // Get the optimization class that pertains to this function
Chris Lattner33081b42006-01-22 23:10:26 +0000186 hash_map<std::string, LibCallOptimization*>::iterator OMI =
187 OptznMap.find(FI->getName());
188 if (OMI == OptznMap.end()) continue;
189
190 LibCallOptimization *CO = OMI->second;
Reid Spencere249a822005-04-27 07:54:40 +0000191
192 // Make sure the called function is suitable for the optimization
Chris Lattner33081b42006-01-22 23:10:26 +0000193 if (!CO->ValidateCalledFunction(FI, *this))
Reid Spencere249a822005-04-27 07:54:40 +0000194 continue;
195
196 // Loop over each of the uses of the function
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000197 for (Value::use_iterator UI = FI->use_begin(), UE = FI->use_end();
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000198 UI != UE ; ) {
Reid Spencere249a822005-04-27 07:54:40 +0000199 // If the use of the function is a call instruction
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000200 if (CallInst* CI = dyn_cast<CallInst>(*UI++)) {
Reid Spencere249a822005-04-27 07:54:40 +0000201 // Do the optimization on the LibCallOptimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000202 if (CO->OptimizeCall(CI, *this)) {
Reid Spencere249a822005-04-27 07:54:40 +0000203 ++SimplifiedLibCalls;
204 found_optimization = result = true;
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000205 CO->succeeded();
Reid Spencere249a822005-04-27 07:54:40 +0000206 }
Reid Spencerbb92b4f2005-04-26 19:13:17 +0000207 }
208 }
209 }
Reid Spencere249a822005-04-27 07:54:40 +0000210 } while (found_optimization);
Chris Lattner33081b42006-01-22 23:10:26 +0000211
Reid Spencere249a822005-04-27 07:54:40 +0000212 return result;
213 }
Reid Spencerbb92b4f2005-04-26 19:13:17 +0000214
Reid Spencere249a822005-04-27 07:54:40 +0000215 /// @brief Return the *current* module we're working on.
Reid Spencer93616972005-04-29 09:39:47 +0000216 Module* getModule() const { return M; }
Reid Spencerbb92b4f2005-04-26 19:13:17 +0000217
Reid Spencere249a822005-04-27 07:54:40 +0000218 /// @brief Return the *current* target data for the module we're working on.
Reid Spencer93616972005-04-29 09:39:47 +0000219 TargetData* getTargetData() const { return TD; }
220
221 /// @brief Return the size_t type -- syntactic shortcut
222 const Type* getIntPtrType() const { return TD->getIntPtrType(); }
223
224 /// @brief Return a Function* for the fputc libcall
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000225 Function* get_fputc(const Type* FILEptr_type) {
Reid Spencer93616972005-04-29 09:39:47 +0000226 if (!fputc_func)
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000227 fputc_func = M->getOrInsertFunction("fputc", Type::IntTy, Type::IntTy,
228 FILEptr_type, NULL);
Reid Spencer93616972005-04-29 09:39:47 +0000229 return fputc_func;
230 }
231
232 /// @brief Return a Function* for the fwrite libcall
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000233 Function* get_fwrite(const Type* FILEptr_type) {
Reid Spencer93616972005-04-29 09:39:47 +0000234 if (!fwrite_func)
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000235 fwrite_func = M->getOrInsertFunction("fwrite", TD->getIntPtrType(),
236 PointerType::get(Type::SByteTy),
237 TD->getIntPtrType(),
238 TD->getIntPtrType(),
239 FILEptr_type, NULL);
Reid Spencer93616972005-04-29 09:39:47 +0000240 return fwrite_func;
241 }
242
243 /// @brief Return a Function* for the sqrt libcall
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000244 Function* get_sqrt() {
Reid Spencer93616972005-04-29 09:39:47 +0000245 if (!sqrt_func)
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000246 sqrt_func = M->getOrInsertFunction("sqrt", Type::DoubleTy,
247 Type::DoubleTy, NULL);
Reid Spencer93616972005-04-29 09:39:47 +0000248 return sqrt_func;
249 }
Reid Spencere249a822005-04-27 07:54:40 +0000250
251 /// @brief Return a Function* for the strlen libcall
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000252 Function* get_strcpy() {
Reid Spencer1e520fd2005-05-04 03:20:21 +0000253 if (!strcpy_func)
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000254 strcpy_func = M->getOrInsertFunction("strcpy",
255 PointerType::get(Type::SByteTy),
256 PointerType::get(Type::SByteTy),
257 PointerType::get(Type::SByteTy),
258 NULL);
Reid Spencer1e520fd2005-05-04 03:20:21 +0000259 return strcpy_func;
260 }
261
262 /// @brief Return a Function* for the strlen libcall
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000263 Function* get_strlen() {
Reid Spencere249a822005-04-27 07:54:40 +0000264 if (!strlen_func)
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000265 strlen_func = M->getOrInsertFunction("strlen", TD->getIntPtrType(),
266 PointerType::get(Type::SByteTy),
267 NULL);
Reid Spencere249a822005-04-27 07:54:40 +0000268 return strlen_func;
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000269 }
270
Reid Spencer38cabd72005-05-03 07:23:44 +0000271 /// @brief Return a Function* for the memchr libcall
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000272 Function* get_memchr() {
Reid Spencer38cabd72005-05-03 07:23:44 +0000273 if (!memchr_func)
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000274 memchr_func = M->getOrInsertFunction("memchr",
275 PointerType::get(Type::SByteTy),
276 PointerType::get(Type::SByteTy),
277 Type::IntTy, TD->getIntPtrType(),
278 NULL);
Reid Spencer38cabd72005-05-03 07:23:44 +0000279 return memchr_func;
280 }
281
Reid Spencere249a822005-04-27 07:54:40 +0000282 /// @brief Return a Function* for the memcpy libcall
Chris Lattner4201cd12005-08-24 17:22:17 +0000283 Function* get_memcpy() {
284 if (!memcpy_func) {
285 const Type *SBP = PointerType::get(Type::SByteTy);
286 memcpy_func = M->getOrInsertFunction("llvm.memcpy", Type::VoidTy,SBP, SBP,
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000287 Type::UIntTy, Type::UIntTy, NULL);
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000288 }
Reid Spencere249a822005-04-27 07:54:40 +0000289 return memcpy_func;
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000290 }
Reid Spencer76dab9a2005-04-26 05:24:00 +0000291
Chris Lattner57740402006-01-23 06:24:46 +0000292 Function *getUnaryFloatFunction(const char *Name, Function *&Cache) {
293 if (!Cache)
294 Cache = M->getOrInsertFunction(Name, Type::FloatTy, Type::FloatTy, NULL);
295 return Cache;
Chris Lattner4201cd12005-08-24 17:22:17 +0000296 }
297
Chris Lattner57740402006-01-23 06:24:46 +0000298 Function *get_floorf() { return getUnaryFloatFunction("floorf", floorf_func);}
299 Function *get_ceilf() { return getUnaryFloatFunction( "ceilf", ceilf_func);}
300 Function *get_roundf() { return getUnaryFloatFunction("roundf", roundf_func);}
301 Function *get_rintf() { return getUnaryFloatFunction( "rintf", rintf_func);}
302 Function *get_nearbyintf() { return getUnaryFloatFunction("nearbyintf",
303 nearbyintf_func); }
Reid Spencere249a822005-04-27 07:54:40 +0000304private:
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000305 /// @brief Reset our cached data for a new Module
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000306 void reset(Module& mod) {
Reid Spencere249a822005-04-27 07:54:40 +0000307 M = &mod;
308 TD = &getAnalysis<TargetData>();
Reid Spencer93616972005-04-29 09:39:47 +0000309 fputc_func = 0;
310 fwrite_func = 0;
Reid Spencere249a822005-04-27 07:54:40 +0000311 memcpy_func = 0;
Reid Spencer38cabd72005-05-03 07:23:44 +0000312 memchr_func = 0;
Reid Spencer93616972005-04-29 09:39:47 +0000313 sqrt_func = 0;
Reid Spencer1e520fd2005-05-04 03:20:21 +0000314 strcpy_func = 0;
Reid Spencere249a822005-04-27 07:54:40 +0000315 strlen_func = 0;
Chris Lattner4201cd12005-08-24 17:22:17 +0000316 floorf_func = 0;
Chris Lattner57740402006-01-23 06:24:46 +0000317 ceilf_func = 0;
318 roundf_func = 0;
319 rintf_func = 0;
320 nearbyintf_func = 0;
Reid Spencer76dab9a2005-04-26 05:24:00 +0000321 }
Reid Spencer39a762d2005-04-25 02:53:12 +0000322
Reid Spencere249a822005-04-27 07:54:40 +0000323private:
Chris Lattner57740402006-01-23 06:24:46 +0000324 /// Caches for function pointers.
325 Function *fputc_func, *fwrite_func;
326 Function *memcpy_func, *memchr_func;
327 Function* sqrt_func;
328 Function *strcpy_func, *strlen_func;
329 Function *floorf_func, *ceilf_func, *roundf_func;
330 Function *rintf_func, *nearbyintf_func;
331 Module *M; ///< Cached Module
332 TargetData *TD; ///< Cached TargetData
Reid Spencere249a822005-04-27 07:54:40 +0000333};
334
335// Register the pass
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000336RegisterOpt<SimplifyLibCalls>
Reid Spencere249a822005-04-27 07:54:40 +0000337X("simplify-libcalls","Simplify well-known library calls");
338
339} // anonymous namespace
340
341// The only public symbol in this file which just instantiates the pass object
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000342ModulePass *llvm::createSimplifyLibCallsPass() {
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000343 return new SimplifyLibCalls();
Reid Spencere249a822005-04-27 07:54:40 +0000344}
345
346// Classes below here, in the anonymous namespace, are all subclasses of the
347// LibCallOptimization class, each implementing all optimizations possible for a
348// single well-known library call. Each has a static singleton instance that
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000349// auto registers it into the "optlist" global above.
Reid Spencere249a822005-04-27 07:54:40 +0000350namespace {
351
Reid Spencera7828ba2005-06-18 17:46:28 +0000352// Forward declare utility functions.
Reid Spencer4c444fe2005-04-30 03:17:54 +0000353bool getConstantStringLength(Value* V, uint64_t& len, ConstantArray** A = 0 );
Reid Spencera7828ba2005-06-18 17:46:28 +0000354Value *CastToCStr(Value *V, Instruction &IP);
Reid Spencere249a822005-04-27 07:54:40 +0000355
356/// This LibCallOptimization will find instances of a call to "exit" that occurs
Reid Spencer39a762d2005-04-25 02:53:12 +0000357/// within the "main" function and change it to a simple "ret" instruction with
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000358/// the same value passed to the exit function. When this is done, it splits the
359/// basic block at the exit(3) call and deletes the call instruction.
Reid Spencer39a762d2005-04-25 02:53:12 +0000360/// @brief Replace calls to exit in main with a simple return
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000361struct ExitInMainOptimization : public LibCallOptimization {
Reid Spencer95d8efd2005-05-03 02:54:54 +0000362 ExitInMainOptimization() : LibCallOptimization("exit",
Reid Spencer170ae7f2005-05-07 20:15:59 +0000363 "Number of 'exit' calls simplified") {}
Reid Spencerf2534c72005-04-25 21:11:48 +0000364
365 // Make sure the called function looks like exit (int argument, int return
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000366 // type, external linkage, not varargs).
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000367 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &SLC){
368 return F->arg_size() >= 1 && F->arg_begin()->getType()->isInteger();
Reid Spencerf2534c72005-04-25 21:11:48 +0000369 }
370
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000371 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& SLC) {
Reid Spencerf2534c72005-04-25 21:11:48 +0000372 // To be careful, we check that the call to exit is coming from "main", that
373 // main has external linkage, and the return type of main and the argument
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000374 // to exit have the same type.
Reid Spencerf2534c72005-04-25 21:11:48 +0000375 Function *from = ci->getParent()->getParent();
376 if (from->hasExternalLinkage())
377 if (from->getReturnType() == ci->getOperand(1)->getType())
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000378 if (from->getName() == "main") {
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000379 // Okay, time to actually do the optimization. First, get the basic
Reid Spencerf2534c72005-04-25 21:11:48 +0000380 // block of the call instruction
381 BasicBlock* bb = ci->getParent();
Reid Spencer39a762d2005-04-25 02:53:12 +0000382
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000383 // Create a return instruction that we'll replace the call with.
384 // Note that the argument of the return is the argument of the call
Reid Spencerf2534c72005-04-25 21:11:48 +0000385 // instruction.
386 ReturnInst* ri = new ReturnInst(ci->getOperand(1), ci);
Reid Spencer39a762d2005-04-25 02:53:12 +0000387
Reid Spencerf2534c72005-04-25 21:11:48 +0000388 // Split the block at the call instruction which places it in a new
389 // basic block.
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000390 bb->splitBasicBlock(ci);
Reid Spencer39a762d2005-04-25 02:53:12 +0000391
Reid Spencerf2534c72005-04-25 21:11:48 +0000392 // The block split caused a branch instruction to be inserted into
393 // the end of the original block, right after the return instruction
394 // that we put there. That's not a valid block, so delete the branch
395 // instruction.
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000396 bb->getInstList().pop_back();
Reid Spencer39a762d2005-04-25 02:53:12 +0000397
Reid Spencerf2534c72005-04-25 21:11:48 +0000398 // Now we can finally get rid of the call instruction which now lives
399 // in the new basic block.
400 ci->eraseFromParent();
401
402 // Optimization succeeded, return true.
403 return true;
404 }
405 // We didn't pass the criteria for this optimization so return false
406 return false;
Reid Spencer9bbaa2a2005-04-25 03:59:26 +0000407 }
Reid Spencer39a762d2005-04-25 02:53:12 +0000408} ExitInMainOptimizer;
409
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000410/// This LibCallOptimization will simplify a call to the strcat library
411/// function. The simplification is possible only if the string being
412/// concatenated is a constant array or a constant expression that results in
413/// a constant string. In this case we can replace it with strlen + llvm.memcpy
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000414/// of the constant string. Both of these calls are further reduced, if possible
415/// on subsequent passes.
Reid Spencerf2534c72005-04-25 21:11:48 +0000416/// @brief Simplify the strcat library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000417struct StrCatOptimization : public LibCallOptimization {
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000418public:
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000419 /// @brief Default constructor
Reid Spencer95d8efd2005-05-03 02:54:54 +0000420 StrCatOptimization() : LibCallOptimization("strcat",
Reid Spencer170ae7f2005-05-07 20:15:59 +0000421 "Number of 'strcat' calls simplified") {}
Reid Spencere249a822005-04-27 07:54:40 +0000422
423public:
Reid Spencerf2534c72005-04-25 21:11:48 +0000424
425 /// @brief Make sure that the "strcat" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000426 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Reid Spencerf2534c72005-04-25 21:11:48 +0000427 if (f->getReturnType() == PointerType::get(Type::SByteTy))
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000428 if (f->arg_size() == 2)
Reid Spencerf2534c72005-04-25 21:11:48 +0000429 {
430 Function::const_arg_iterator AI = f->arg_begin();
431 if (AI++->getType() == PointerType::get(Type::SByteTy))
432 if (AI->getType() == PointerType::get(Type::SByteTy))
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000433 {
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000434 // Indicate this is a suitable call type.
Reid Spencerf2534c72005-04-25 21:11:48 +0000435 return true;
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000436 }
Reid Spencerf2534c72005-04-25 21:11:48 +0000437 }
438 return false;
439 }
440
Reid Spencere249a822005-04-27 07:54:40 +0000441 /// @brief Optimize the strcat library function
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000442 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& SLC) {
Reid Spencer08b49402005-04-27 17:46:54 +0000443 // Extract some information from the instruction
444 Module* M = ci->getParent()->getParent()->getParent();
445 Value* dest = ci->getOperand(1);
446 Value* src = ci->getOperand(2);
447
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000448 // Extract the initializer (while making numerous checks) from the
Reid Spencer76dab9a2005-04-26 05:24:00 +0000449 // source operand of the call to strcat. If we get null back, one of
450 // a variety of checks in get_GVInitializer failed
Reid Spencerb4f7b832005-04-26 07:45:18 +0000451 uint64_t len = 0;
Reid Spencer08b49402005-04-27 17:46:54 +0000452 if (!getConstantStringLength(src,len))
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000453 return false;
454
Reid Spencerb4f7b832005-04-26 07:45:18 +0000455 // Handle the simple, do-nothing case
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000456 if (len == 0) {
Reid Spencer08b49402005-04-27 17:46:54 +0000457 ci->replaceAllUsesWith(dest);
Reid Spencer8ee5aac2005-04-26 03:26:15 +0000458 ci->eraseFromParent();
459 return true;
460 }
461
Reid Spencerb4f7b832005-04-26 07:45:18 +0000462 // Increment the length because we actually want to memcpy the null
463 // terminator as well.
464 len++;
Reid Spencerf2534c72005-04-25 21:11:48 +0000465
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000466 // We need to find the end of the destination string. That's where the
467 // memory is to be moved to. We just generate a call to strlen (further
468 // optimized in another pass). Note that the SLC.get_strlen() call
Reid Spencerb4f7b832005-04-26 07:45:18 +0000469 // caches the Function* for us.
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000470 CallInst* strlen_inst =
Reid Spencer08b49402005-04-27 17:46:54 +0000471 new CallInst(SLC.get_strlen(), dest, dest->getName()+".len",ci);
Reid Spencerb4f7b832005-04-26 07:45:18 +0000472
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000473 // Now that we have the destination's length, we must index into the
Reid Spencerb4f7b832005-04-26 07:45:18 +0000474 // destination's pointer to get the actual memcpy destination (end of
475 // the string .. we're concatenating).
476 std::vector<Value*> idx;
477 idx.push_back(strlen_inst);
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000478 GetElementPtrInst* gep =
Reid Spencer08b49402005-04-27 17:46:54 +0000479 new GetElementPtrInst(dest,idx,dest->getName()+".indexed",ci);
Reid Spencerb4f7b832005-04-26 07:45:18 +0000480
481 // We have enough information to now generate the memcpy call to
482 // do the concatenation for us.
483 std::vector<Value*> vals;
484 vals.push_back(gep); // destination
485 vals.push_back(ci->getOperand(2)); // source
Reid Spencer1e520fd2005-05-04 03:20:21 +0000486 vals.push_back(ConstantUInt::get(Type::UIntTy,len)); // length
487 vals.push_back(ConstantUInt::get(Type::UIntTy,1)); // alignment
Reid Spencer08b49402005-04-27 17:46:54 +0000488 new CallInst(SLC.get_memcpy(), vals, "", ci);
Reid Spencerb4f7b832005-04-26 07:45:18 +0000489
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000490 // Finally, substitute the first operand of the strcat call for the
491 // strcat call itself since strcat returns its first operand; and,
Reid Spencerb4f7b832005-04-26 07:45:18 +0000492 // kill the strcat CallInst.
Reid Spencer08b49402005-04-27 17:46:54 +0000493 ci->replaceAllUsesWith(dest);
Reid Spencerb4f7b832005-04-26 07:45:18 +0000494 ci->eraseFromParent();
495 return true;
Reid Spencer9bbaa2a2005-04-25 03:59:26 +0000496 }
497} StrCatOptimizer;
498
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000499/// This LibCallOptimization will simplify a call to the strchr library
Reid Spencer38cabd72005-05-03 07:23:44 +0000500/// function. It optimizes out cases where the arguments are both constant
501/// and the result can be determined statically.
502/// @brief Simplify the strcmp library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000503struct StrChrOptimization : public LibCallOptimization {
Reid Spencer38cabd72005-05-03 07:23:44 +0000504public:
505 StrChrOptimization() : LibCallOptimization("strchr",
Reid Spencer170ae7f2005-05-07 20:15:59 +0000506 "Number of 'strchr' calls simplified") {}
Reid Spencer38cabd72005-05-03 07:23:44 +0000507
508 /// @brief Make sure that the "strchr" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000509 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000510 if (f->getReturnType() == PointerType::get(Type::SByteTy) &&
Reid Spencer38cabd72005-05-03 07:23:44 +0000511 f->arg_size() == 2)
512 return true;
513 return false;
514 }
515
Chris Lattnerf8053ce2005-05-20 22:22:25 +0000516 /// @brief Perform the strchr optimizations
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000517 virtual bool OptimizeCall(CallInst *ci, SimplifyLibCalls &SLC) {
Reid Spencer38cabd72005-05-03 07:23:44 +0000518 // If there aren't three operands, bail
519 if (ci->getNumOperands() != 3)
520 return false;
521
522 // Check that the first argument to strchr is a constant array of sbyte.
523 // If it is, get the length and data, otherwise return false.
524 uint64_t len = 0;
525 ConstantArray* CA;
526 if (!getConstantStringLength(ci->getOperand(1),len,&CA))
527 return false;
528
529 // Check that the second argument to strchr is a constant int, return false
530 // if it isn't
531 ConstantSInt* CSI = dyn_cast<ConstantSInt>(ci->getOperand(2));
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000532 if (!CSI) {
Reid Spencer38cabd72005-05-03 07:23:44 +0000533 // Just lower this to memchr since we know the length of the string as
534 // it is constant.
535 Function* f = SLC.get_memchr();
536 std::vector<Value*> args;
537 args.push_back(ci->getOperand(1));
538 args.push_back(ci->getOperand(2));
539 args.push_back(ConstantUInt::get(SLC.getIntPtrType(),len));
540 ci->replaceAllUsesWith( new CallInst(f,args,ci->getName(),ci));
541 ci->eraseFromParent();
542 return true;
543 }
544
545 // Get the character we're looking for
546 int64_t chr = CSI->getValue();
547
548 // Compute the offset
549 uint64_t offset = 0;
550 bool char_found = false;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000551 for (uint64_t i = 0; i < len; ++i) {
552 if (ConstantSInt* CI = dyn_cast<ConstantSInt>(CA->getOperand(i))) {
Reid Spencer38cabd72005-05-03 07:23:44 +0000553 // Check for the null terminator
554 if (CI->isNullValue())
555 break; // we found end of string
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000556 else if (CI->getValue() == chr) {
Reid Spencer38cabd72005-05-03 07:23:44 +0000557 char_found = true;
558 offset = i;
559 break;
560 }
561 }
562 }
563
564 // strchr(s,c) -> offset_of_in(c,s)
565 // (if c is a constant integer and s is a constant string)
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000566 if (char_found) {
Reid Spencer38cabd72005-05-03 07:23:44 +0000567 std::vector<Value*> indices;
568 indices.push_back(ConstantUInt::get(Type::ULongTy,offset));
569 GetElementPtrInst* GEP = new GetElementPtrInst(ci->getOperand(1),indices,
570 ci->getOperand(1)->getName()+".strchr",ci);
571 ci->replaceAllUsesWith(GEP);
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000572 } else {
Reid Spencer38cabd72005-05-03 07:23:44 +0000573 ci->replaceAllUsesWith(
574 ConstantPointerNull::get(PointerType::get(Type::SByteTy)));
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000575 }
Reid Spencer38cabd72005-05-03 07:23:44 +0000576 ci->eraseFromParent();
577 return true;
578 }
579} StrChrOptimizer;
580
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000581/// This LibCallOptimization will simplify a call to the strcmp library
Reid Spencer4c444fe2005-04-30 03:17:54 +0000582/// function. It optimizes out cases where one or both arguments are constant
583/// and the result can be determined statically.
584/// @brief Simplify the strcmp library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000585struct StrCmpOptimization : public LibCallOptimization {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000586public:
Reid Spencer95d8efd2005-05-03 02:54:54 +0000587 StrCmpOptimization() : LibCallOptimization("strcmp",
Reid Spencer170ae7f2005-05-07 20:15:59 +0000588 "Number of 'strcmp' calls simplified") {}
Reid Spencer4c444fe2005-04-30 03:17:54 +0000589
Chris Lattnerf8053ce2005-05-20 22:22:25 +0000590 /// @brief Make sure that the "strcmp" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000591 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &SLC){
592 return F->getReturnType() == Type::IntTy && F->arg_size() == 2;
Reid Spencer4c444fe2005-04-30 03:17:54 +0000593 }
594
Chris Lattnerf8053ce2005-05-20 22:22:25 +0000595 /// @brief Perform the strcmp optimization
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000596 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& SLC) {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000597 // First, check to see if src and destination are the same. If they are,
Reid Spencer16449a92005-04-30 06:45:47 +0000598 // then the optimization is to replace the CallInst with a constant 0
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000599 // because the call is a no-op.
Reid Spencer4c444fe2005-04-30 03:17:54 +0000600 Value* s1 = ci->getOperand(1);
601 Value* s2 = ci->getOperand(2);
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000602 if (s1 == s2) {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000603 // strcmp(x,x) -> 0
604 ci->replaceAllUsesWith(ConstantInt::get(Type::IntTy,0));
605 ci->eraseFromParent();
606 return true;
607 }
608
609 bool isstr_1 = false;
610 uint64_t len_1 = 0;
611 ConstantArray* A1;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000612 if (getConstantStringLength(s1,len_1,&A1)) {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000613 isstr_1 = true;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000614 if (len_1 == 0) {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000615 // strcmp("",x) -> *x
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000616 LoadInst* load =
Reid Spencera7828ba2005-06-18 17:46:28 +0000617 new LoadInst(CastToCStr(s2,*ci), ci->getName()+".load",ci);
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000618 CastInst* cast =
Reid Spencer4c444fe2005-04-30 03:17:54 +0000619 new CastInst(load,Type::IntTy,ci->getName()+".int",ci);
620 ci->replaceAllUsesWith(cast);
621 ci->eraseFromParent();
622 return true;
623 }
624 }
625
626 bool isstr_2 = false;
627 uint64_t len_2 = 0;
628 ConstantArray* A2;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000629 if (getConstantStringLength(s2, len_2, &A2)) {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000630 isstr_2 = true;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000631 if (len_2 == 0) {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000632 // strcmp(x,"") -> *x
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000633 LoadInst* load =
Reid Spencera7828ba2005-06-18 17:46:28 +0000634 new LoadInst(CastToCStr(s1,*ci),ci->getName()+".val",ci);
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000635 CastInst* cast =
Reid Spencer4c444fe2005-04-30 03:17:54 +0000636 new CastInst(load,Type::IntTy,ci->getName()+".int",ci);
637 ci->replaceAllUsesWith(cast);
638 ci->eraseFromParent();
639 return true;
640 }
641 }
642
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000643 if (isstr_1 && isstr_2) {
Reid Spencer4c444fe2005-04-30 03:17:54 +0000644 // strcmp(x,y) -> cnst (if both x and y are constant strings)
645 std::string str1 = A1->getAsString();
646 std::string str2 = A2->getAsString();
647 int result = strcmp(str1.c_str(), str2.c_str());
648 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,result));
649 ci->eraseFromParent();
650 return true;
651 }
652 return false;
653 }
654} StrCmpOptimizer;
655
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000656/// This LibCallOptimization will simplify a call to the strncmp library
Reid Spencer49fa07042005-05-03 01:43:45 +0000657/// function. It optimizes out cases where one or both arguments are constant
658/// and the result can be determined statically.
659/// @brief Simplify the strncmp library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000660struct StrNCmpOptimization : public LibCallOptimization {
Reid Spencer49fa07042005-05-03 01:43:45 +0000661public:
Reid Spencer95d8efd2005-05-03 02:54:54 +0000662 StrNCmpOptimization() : LibCallOptimization("strncmp",
Reid Spencer170ae7f2005-05-07 20:15:59 +0000663 "Number of 'strncmp' calls simplified") {}
Reid Spencer49fa07042005-05-03 01:43:45 +0000664
Chris Lattnerf8053ce2005-05-20 22:22:25 +0000665 /// @brief Make sure that the "strncmp" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000666 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Reid Spencer49fa07042005-05-03 01:43:45 +0000667 if (f->getReturnType() == Type::IntTy && f->arg_size() == 3)
668 return true;
669 return false;
670 }
671
672 /// @brief Perform the strncpy optimization
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000673 virtual bool OptimizeCall(CallInst *ci, SimplifyLibCalls &SLC) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000674 // First, check to see if src and destination are the same. If they are,
675 // then the optimization is to replace the CallInst with a constant 0
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000676 // because the call is a no-op.
Reid Spencer49fa07042005-05-03 01:43:45 +0000677 Value* s1 = ci->getOperand(1);
678 Value* s2 = ci->getOperand(2);
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000679 if (s1 == s2) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000680 // strncmp(x,x,l) -> 0
681 ci->replaceAllUsesWith(ConstantInt::get(Type::IntTy,0));
682 ci->eraseFromParent();
683 return true;
684 }
685
686 // Check the length argument, if it is Constant zero then the strings are
687 // considered equal.
688 uint64_t len_arg = 0;
689 bool len_arg_is_const = false;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000690 if (ConstantInt* len_CI = dyn_cast<ConstantInt>(ci->getOperand(3))) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000691 len_arg_is_const = true;
692 len_arg = len_CI->getRawValue();
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000693 if (len_arg == 0) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000694 // strncmp(x,y,0) -> 0
695 ci->replaceAllUsesWith(ConstantInt::get(Type::IntTy,0));
696 ci->eraseFromParent();
697 return true;
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000698 }
Reid Spencer49fa07042005-05-03 01:43:45 +0000699 }
700
701 bool isstr_1 = false;
702 uint64_t len_1 = 0;
703 ConstantArray* A1;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000704 if (getConstantStringLength(s1, len_1, &A1)) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000705 isstr_1 = true;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000706 if (len_1 == 0) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000707 // strncmp("",x) -> *x
708 LoadInst* load = new LoadInst(s1,ci->getName()+".load",ci);
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000709 CastInst* cast =
Reid Spencer49fa07042005-05-03 01:43:45 +0000710 new CastInst(load,Type::IntTy,ci->getName()+".int",ci);
711 ci->replaceAllUsesWith(cast);
712 ci->eraseFromParent();
713 return true;
714 }
715 }
716
717 bool isstr_2 = false;
718 uint64_t len_2 = 0;
719 ConstantArray* A2;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000720 if (getConstantStringLength(s2,len_2,&A2)) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000721 isstr_2 = true;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000722 if (len_2 == 0) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000723 // strncmp(x,"") -> *x
724 LoadInst* load = new LoadInst(s2,ci->getName()+".val",ci);
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000725 CastInst* cast =
Reid Spencer49fa07042005-05-03 01:43:45 +0000726 new CastInst(load,Type::IntTy,ci->getName()+".int",ci);
727 ci->replaceAllUsesWith(cast);
728 ci->eraseFromParent();
729 return true;
730 }
731 }
732
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000733 if (isstr_1 && isstr_2 && len_arg_is_const) {
Reid Spencer49fa07042005-05-03 01:43:45 +0000734 // strncmp(x,y,const) -> constant
735 std::string str1 = A1->getAsString();
736 std::string str2 = A2->getAsString();
737 int result = strncmp(str1.c_str(), str2.c_str(), len_arg);
738 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,result));
739 ci->eraseFromParent();
740 return true;
741 }
742 return false;
743 }
744} StrNCmpOptimizer;
745
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000746/// This LibCallOptimization will simplify a call to the strcpy library
747/// function. Two optimizations are possible:
Reid Spencere249a822005-04-27 07:54:40 +0000748/// (1) If src and dest are the same and not volatile, just return dest
749/// (2) If the src is a constant then we can convert to llvm.memmove
750/// @brief Simplify the strcpy library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000751struct StrCpyOptimization : public LibCallOptimization {
Reid Spencere249a822005-04-27 07:54:40 +0000752public:
Reid Spencer95d8efd2005-05-03 02:54:54 +0000753 StrCpyOptimization() : LibCallOptimization("strcpy",
Reid Spencer170ae7f2005-05-07 20:15:59 +0000754 "Number of 'strcpy' calls simplified") {}
Reid Spencere249a822005-04-27 07:54:40 +0000755
756 /// @brief Make sure that the "strcpy" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000757 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Reid Spencere249a822005-04-27 07:54:40 +0000758 if (f->getReturnType() == PointerType::get(Type::SByteTy))
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000759 if (f->arg_size() == 2) {
Reid Spencere249a822005-04-27 07:54:40 +0000760 Function::const_arg_iterator AI = f->arg_begin();
761 if (AI++->getType() == PointerType::get(Type::SByteTy))
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000762 if (AI->getType() == PointerType::get(Type::SByteTy)) {
Reid Spencere249a822005-04-27 07:54:40 +0000763 // Indicate this is a suitable call type.
764 return true;
765 }
766 }
767 return false;
768 }
769
770 /// @brief Perform the strcpy optimization
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000771 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& SLC) {
Reid Spencere249a822005-04-27 07:54:40 +0000772 // First, check to see if src and destination are the same. If they are,
773 // then the optimization is to replace the CallInst with the destination
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000774 // because the call is a no-op. Note that this corresponds to the
Reid Spencere249a822005-04-27 07:54:40 +0000775 // degenerate strcpy(X,X) case which should have "undefined" results
776 // according to the C specification. However, it occurs sometimes and
777 // we optimize it as a no-op.
778 Value* dest = ci->getOperand(1);
779 Value* src = ci->getOperand(2);
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000780 if (dest == src) {
Reid Spencere249a822005-04-27 07:54:40 +0000781 ci->replaceAllUsesWith(dest);
782 ci->eraseFromParent();
783 return true;
784 }
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000785
Reid Spencere249a822005-04-27 07:54:40 +0000786 // Get the length of the constant string referenced by the second operand,
787 // the "src" parameter. Fail the optimization if we can't get the length
788 // (note that getConstantStringLength does lots of checks to make sure this
789 // is valid).
790 uint64_t len = 0;
791 if (!getConstantStringLength(ci->getOperand(2),len))
792 return false;
793
794 // If the constant string's length is zero we can optimize this by just
795 // doing a store of 0 at the first byte of the destination
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000796 if (len == 0) {
Reid Spencere249a822005-04-27 07:54:40 +0000797 new StoreInst(ConstantInt::get(Type::SByteTy,0),ci->getOperand(1),ci);
798 ci->replaceAllUsesWith(dest);
799 ci->eraseFromParent();
800 return true;
801 }
802
803 // Increment the length because we actually want to memcpy the null
804 // terminator as well.
805 len++;
806
807 // Extract some information from the instruction
808 Module* M = ci->getParent()->getParent()->getParent();
809
810 // We have enough information to now generate the memcpy call to
811 // do the concatenation for us.
812 std::vector<Value*> vals;
813 vals.push_back(dest); // destination
814 vals.push_back(src); // source
Reid Spencer1e520fd2005-05-04 03:20:21 +0000815 vals.push_back(ConstantUInt::get(Type::UIntTy,len)); // length
816 vals.push_back(ConstantUInt::get(Type::UIntTy,1)); // alignment
Reid Spencer08b49402005-04-27 17:46:54 +0000817 new CallInst(SLC.get_memcpy(), vals, "", ci);
Reid Spencere249a822005-04-27 07:54:40 +0000818
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000819 // Finally, substitute the first operand of the strcat call for the
820 // strcat call itself since strcat returns its first operand; and,
Reid Spencere249a822005-04-27 07:54:40 +0000821 // kill the strcat CallInst.
822 ci->replaceAllUsesWith(dest);
823 ci->eraseFromParent();
824 return true;
825 }
826} StrCpyOptimizer;
827
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000828/// This LibCallOptimization will simplify a call to the strlen library
829/// function by replacing it with a constant value if the string provided to
Reid Spencer7ddcfb32005-04-27 21:29:20 +0000830/// it is a constant array.
Reid Spencer76dab9a2005-04-26 05:24:00 +0000831/// @brief Simplify the strlen library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +0000832struct StrLenOptimization : public LibCallOptimization {
Reid Spencer95d8efd2005-05-03 02:54:54 +0000833 StrLenOptimization() : LibCallOptimization("strlen",
Reid Spencer170ae7f2005-05-07 20:15:59 +0000834 "Number of 'strlen' calls simplified") {}
Reid Spencer76dab9a2005-04-26 05:24:00 +0000835
836 /// @brief Make sure that the "strlen" function has the right prototype
Reid Spencere249a822005-04-27 07:54:40 +0000837 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC)
Reid Spencer76dab9a2005-04-26 05:24:00 +0000838 {
Reid Spencere249a822005-04-27 07:54:40 +0000839 if (f->getReturnType() == SLC.getTargetData()->getIntPtrType())
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000840 if (f->arg_size() == 1)
Reid Spencer76dab9a2005-04-26 05:24:00 +0000841 if (Function::const_arg_iterator AI = f->arg_begin())
842 if (AI->getType() == PointerType::get(Type::SByteTy))
843 return true;
844 return false;
845 }
846
847 /// @brief Perform the strlen optimization
Reid Spencere249a822005-04-27 07:54:40 +0000848 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& SLC)
Reid Spencer76dab9a2005-04-26 05:24:00 +0000849 {
Reid Spencer170ae7f2005-05-07 20:15:59 +0000850 // Make sure we're dealing with an sbyte* here.
851 Value* str = ci->getOperand(1);
852 if (str->getType() != PointerType::get(Type::SByteTy))
853 return false;
854
855 // Does the call to strlen have exactly one use?
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +0000856 if (ci->hasOneUse())
Reid Spencer170ae7f2005-05-07 20:15:59 +0000857 // Is that single use a binary operator?
858 if (BinaryOperator* bop = dyn_cast<BinaryOperator>(ci->use_back()))
859 // Is it compared against a constant integer?
860 if (ConstantInt* CI = dyn_cast<ConstantInt>(bop->getOperand(1)))
861 {
862 // Get the value the strlen result is compared to
863 uint64_t val = CI->getRawValue();
864
865 // If its compared against length 0 with == or !=
866 if (val == 0 &&
867 (bop->getOpcode() == Instruction::SetEQ ||
868 bop->getOpcode() == Instruction::SetNE))
869 {
870 // strlen(x) != 0 -> *x != 0
871 // strlen(x) == 0 -> *x == 0
872 LoadInst* load = new LoadInst(str,str->getName()+".first",ci);
873 BinaryOperator* rbop = BinaryOperator::create(bop->getOpcode(),
874 load, ConstantSInt::get(Type::SByteTy,0),
875 bop->getName()+".strlen", ci);
876 bop->replaceAllUsesWith(rbop);
877 bop->eraseFromParent();
878 ci->eraseFromParent();
879 return true;
880 }
881 }
882
883 // Get the length of the constant string operand
Reid Spencerb4f7b832005-04-26 07:45:18 +0000884 uint64_t len = 0;
885 if (!getConstantStringLength(ci->getOperand(1),len))
Reid Spencer76dab9a2005-04-26 05:24:00 +0000886 return false;
887
Reid Spencer170ae7f2005-05-07 20:15:59 +0000888 // strlen("xyz") -> 3 (for example)
Chris Lattnere17c5d02005-08-01 16:52:50 +0000889 const Type *Ty = SLC.getTargetData()->getIntPtrType();
890 if (Ty->isSigned())
891 ci->replaceAllUsesWith(ConstantSInt::get(Ty, len));
892 else
893 ci->replaceAllUsesWith(ConstantUInt::get(Ty, len));
894
Reid Spencerb4f7b832005-04-26 07:45:18 +0000895 ci->eraseFromParent();
896 return true;
Reid Spencer76dab9a2005-04-26 05:24:00 +0000897 }
898} StrLenOptimizer;
899
Chris Lattnerc244e7c2005-09-29 04:54:20 +0000900/// IsOnlyUsedInEqualsComparison - Return true if it only matters that the value
901/// is equal or not-equal to zero.
902static bool IsOnlyUsedInEqualsZeroComparison(Instruction *I) {
903 for (Value::use_iterator UI = I->use_begin(), E = I->use_end();
904 UI != E; ++UI) {
905 Instruction *User = cast<Instruction>(*UI);
906 if (User->getOpcode() == Instruction::SetNE ||
907 User->getOpcode() == Instruction::SetEQ) {
908 if (isa<Constant>(User->getOperand(1)) &&
909 cast<Constant>(User->getOperand(1))->isNullValue())
910 continue;
911 } else if (CastInst *CI = dyn_cast<CastInst>(User))
912 if (CI->getType() == Type::BoolTy)
913 continue;
914 // Unknown instruction.
915 return false;
916 }
917 return true;
918}
919
920/// This memcmpOptimization will simplify a call to the memcmp library
921/// function.
922struct memcmpOptimization : public LibCallOptimization {
923 /// @brief Default Constructor
924 memcmpOptimization()
925 : LibCallOptimization("memcmp", "Number of 'memcmp' calls simplified") {}
926
927 /// @brief Make sure that the "memcmp" function has the right prototype
928 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &TD) {
929 Function::const_arg_iterator AI = F->arg_begin();
930 if (F->arg_size() != 3 || !isa<PointerType>(AI->getType())) return false;
931 if (!isa<PointerType>((++AI)->getType())) return false;
932 if (!(++AI)->getType()->isInteger()) return false;
933 if (!F->getReturnType()->isInteger()) return false;
934 return true;
935 }
936
937 /// Because of alignment and instruction information that we don't have, we
938 /// leave the bulk of this to the code generators.
939 ///
940 /// Note that we could do much more if we could force alignment on otherwise
941 /// small aligned allocas, or if we could indicate that loads have a small
942 /// alignment.
943 virtual bool OptimizeCall(CallInst *CI, SimplifyLibCalls &TD) {
944 Value *LHS = CI->getOperand(1), *RHS = CI->getOperand(2);
945
946 // If the two operands are the same, return zero.
947 if (LHS == RHS) {
948 // memcmp(s,s,x) -> 0
949 CI->replaceAllUsesWith(Constant::getNullValue(CI->getType()));
950 CI->eraseFromParent();
951 return true;
952 }
953
954 // Make sure we have a constant length.
955 ConstantInt *LenC = dyn_cast<ConstantInt>(CI->getOperand(3));
956 if (!LenC) return false;
957 uint64_t Len = LenC->getRawValue();
958
959 // If the length is zero, this returns 0.
960 switch (Len) {
961 case 0:
962 // memcmp(s1,s2,0) -> 0
963 CI->replaceAllUsesWith(Constant::getNullValue(CI->getType()));
964 CI->eraseFromParent();
965 return true;
966 case 1: {
967 // memcmp(S1,S2,1) -> *(ubyte*)S1 - *(ubyte*)S2
968 const Type *UCharPtr = PointerType::get(Type::UByteTy);
969 CastInst *Op1Cast = new CastInst(LHS, UCharPtr, LHS->getName(), CI);
970 CastInst *Op2Cast = new CastInst(RHS, UCharPtr, RHS->getName(), CI);
971 Value *S1V = new LoadInst(Op1Cast, LHS->getName()+".val", CI);
972 Value *S2V = new LoadInst(Op2Cast, RHS->getName()+".val", CI);
973 Value *RV = BinaryOperator::createSub(S1V, S2V, CI->getName()+".diff",CI);
974 if (RV->getType() != CI->getType())
975 RV = new CastInst(RV, CI->getType(), RV->getName(), CI);
976 CI->replaceAllUsesWith(RV);
977 CI->eraseFromParent();
978 return true;
979 }
980 case 2:
981 if (IsOnlyUsedInEqualsZeroComparison(CI)) {
982 // TODO: IF both are aligned, use a short load/compare.
983
984 // memcmp(S1,S2,2) -> S1[0]-S2[0] | S1[1]-S2[1] iff only ==/!= 0 matters
985 const Type *UCharPtr = PointerType::get(Type::UByteTy);
986 CastInst *Op1Cast = new CastInst(LHS, UCharPtr, LHS->getName(), CI);
987 CastInst *Op2Cast = new CastInst(RHS, UCharPtr, RHS->getName(), CI);
988 Value *S1V1 = new LoadInst(Op1Cast, LHS->getName()+".val1", CI);
989 Value *S2V1 = new LoadInst(Op2Cast, RHS->getName()+".val1", CI);
990 Value *D1 = BinaryOperator::createSub(S1V1, S2V1,
991 CI->getName()+".d1", CI);
992 Constant *One = ConstantInt::get(Type::IntTy, 1);
993 Value *G1 = new GetElementPtrInst(Op1Cast, One, "next1v", CI);
994 Value *G2 = new GetElementPtrInst(Op2Cast, One, "next2v", CI);
995 Value *S1V2 = new LoadInst(G1, LHS->getName()+".val2", CI);
996 Value *S2V2 = new LoadInst(G1, RHS->getName()+".val2", CI);
997 Value *D2 = BinaryOperator::createSub(S1V2, S2V2,
998 CI->getName()+".d1", CI);
999 Value *Or = BinaryOperator::createOr(D1, D2, CI->getName()+".res", CI);
1000 if (Or->getType() != CI->getType())
1001 Or = new CastInst(Or, CI->getType(), Or->getName(), CI);
1002 CI->replaceAllUsesWith(Or);
1003 CI->eraseFromParent();
1004 return true;
1005 }
1006 break;
1007 default:
1008 break;
1009 }
1010
Chris Lattnerc244e7c2005-09-29 04:54:20 +00001011 return false;
1012 }
1013} memcmpOptimizer;
1014
1015
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001016/// This LibCallOptimization will simplify a call to the memcpy library
1017/// function by expanding it out to a single store of size 0, 1, 2, 4, or 8
Reid Spencer7ddcfb32005-04-27 21:29:20 +00001018/// bytes depending on the length of the string and the alignment. Additional
1019/// optimizations are possible in code generation (sequence of immediate store)
Reid Spencerf2534c72005-04-25 21:11:48 +00001020/// @brief Simplify the memcpy library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001021struct LLVMMemCpyOptimization : public LibCallOptimization {
Reid Spencer7ddcfb32005-04-27 21:29:20 +00001022 /// @brief Default Constructor
Reid Spencer38cabd72005-05-03 07:23:44 +00001023 LLVMMemCpyOptimization() : LibCallOptimization("llvm.memcpy",
Reid Spencer95d8efd2005-05-03 02:54:54 +00001024 "Number of 'llvm.memcpy' calls simplified") {}
1025
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001026protected:
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001027 /// @brief Subclass Constructor
Reid Spencer170ae7f2005-05-07 20:15:59 +00001028 LLVMMemCpyOptimization(const char* fname, const char* desc)
1029 : LibCallOptimization(fname, desc) {}
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001030public:
Reid Spencerf2534c72005-04-25 21:11:48 +00001031
1032 /// @brief Make sure that the "memcpy" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001033 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& TD) {
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001034 // Just make sure this has 4 arguments per LLVM spec.
Reid Spencer2bc7a4f2005-04-26 23:02:16 +00001035 return (f->arg_size() == 4);
Reid Spencerf2534c72005-04-25 21:11:48 +00001036 }
1037
Reid Spencerb4f7b832005-04-26 07:45:18 +00001038 /// Because of alignment and instruction information that we don't have, we
1039 /// leave the bulk of this to the code generators. The optimization here just
1040 /// deals with a few degenerate cases where the length of the string and the
1041 /// alignment match the sizes of our intrinsic types so we can do a load and
1042 /// store instead of the memcpy call.
1043 /// @brief Perform the memcpy optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001044 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& TD) {
Reid Spencer4855ebf2005-04-26 19:55:57 +00001045 // Make sure we have constant int values to work with
1046 ConstantInt* LEN = dyn_cast<ConstantInt>(ci->getOperand(3));
1047 if (!LEN)
1048 return false;
1049 ConstantInt* ALIGN = dyn_cast<ConstantInt>(ci->getOperand(4));
1050 if (!ALIGN)
1051 return false;
1052
1053 // If the length is larger than the alignment, we can't optimize
1054 uint64_t len = LEN->getRawValue();
1055 uint64_t alignment = ALIGN->getRawValue();
Reid Spencer38cabd72005-05-03 07:23:44 +00001056 if (alignment == 0)
1057 alignment = 1; // Alignment 0 is identity for alignment 1
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001058 if (len > alignment)
Reid Spencerb4f7b832005-04-26 07:45:18 +00001059 return false;
1060
Reid Spencer08b49402005-04-27 17:46:54 +00001061 // Get the type we will cast to, based on size of the string
Reid Spencerb4f7b832005-04-26 07:45:18 +00001062 Value* dest = ci->getOperand(1);
1063 Value* src = ci->getOperand(2);
Reid Spencer08b49402005-04-27 17:46:54 +00001064 Type* castType = 0;
Reid Spencerb4f7b832005-04-26 07:45:18 +00001065 switch (len)
1066 {
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001067 case 0:
Reid Spencer93616972005-04-29 09:39:47 +00001068 // memcpy(d,s,0,a) -> noop
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001069 ci->eraseFromParent();
1070 return true;
Reid Spencer08b49402005-04-27 17:46:54 +00001071 case 1: castType = Type::SByteTy; break;
1072 case 2: castType = Type::ShortTy; break;
1073 case 4: castType = Type::IntTy; break;
1074 case 8: castType = Type::LongTy; break;
Reid Spencerb4f7b832005-04-26 07:45:18 +00001075 default:
1076 return false;
1077 }
Reid Spencer08b49402005-04-27 17:46:54 +00001078
1079 // Cast source and dest to the right sized primitive and then load/store
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001080 CastInst* SrcCast =
Reid Spencer08b49402005-04-27 17:46:54 +00001081 new CastInst(src,PointerType::get(castType),src->getName()+".cast",ci);
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001082 CastInst* DestCast =
Reid Spencer08b49402005-04-27 17:46:54 +00001083 new CastInst(dest,PointerType::get(castType),dest->getName()+".cast",ci);
1084 LoadInst* LI = new LoadInst(SrcCast,SrcCast->getName()+".val",ci);
Reid Spencerb4f7b832005-04-26 07:45:18 +00001085 StoreInst* SI = new StoreInst(LI, DestCast, ci);
Reid Spencerb4f7b832005-04-26 07:45:18 +00001086 ci->eraseFromParent();
1087 return true;
Reid Spencerf2534c72005-04-25 21:11:48 +00001088 }
Reid Spencer38cabd72005-05-03 07:23:44 +00001089} LLVMMemCpyOptimizer;
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001090
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001091/// This LibCallOptimization will simplify a call to the memmove library
1092/// function. It is identical to MemCopyOptimization except for the name of
Reid Spencer7ddcfb32005-04-27 21:29:20 +00001093/// the intrinsic.
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001094/// @brief Simplify the memmove library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001095struct LLVMMemMoveOptimization : public LLVMMemCpyOptimization {
Reid Spencer7ddcfb32005-04-27 21:29:20 +00001096 /// @brief Default Constructor
Reid Spencer38cabd72005-05-03 07:23:44 +00001097 LLVMMemMoveOptimization() : LLVMMemCpyOptimization("llvm.memmove",
Reid Spencer95d8efd2005-05-03 02:54:54 +00001098 "Number of 'llvm.memmove' calls simplified") {}
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001099
Reid Spencer38cabd72005-05-03 07:23:44 +00001100} LLVMMemMoveOptimizer;
1101
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001102/// This LibCallOptimization will simplify a call to the memset library
1103/// function by expanding it out to a single store of size 0, 1, 2, 4, or 8
1104/// bytes depending on the length argument.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001105struct LLVMMemSetOptimization : public LibCallOptimization {
Reid Spencer38cabd72005-05-03 07:23:44 +00001106 /// @brief Default Constructor
1107 LLVMMemSetOptimization() : LibCallOptimization("llvm.memset",
Reid Spencer38cabd72005-05-03 07:23:44 +00001108 "Number of 'llvm.memset' calls simplified") {}
Reid Spencer38cabd72005-05-03 07:23:44 +00001109public:
Reid Spencer38cabd72005-05-03 07:23:44 +00001110
1111 /// @brief Make sure that the "memset" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001112 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &TD) {
Reid Spencer38cabd72005-05-03 07:23:44 +00001113 // Just make sure this has 3 arguments per LLVM spec.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001114 return F->arg_size() == 4;
Reid Spencer38cabd72005-05-03 07:23:44 +00001115 }
1116
1117 /// Because of alignment and instruction information that we don't have, we
1118 /// leave the bulk of this to the code generators. The optimization here just
1119 /// deals with a few degenerate cases where the length parameter is constant
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001120 /// and the alignment matches the sizes of our intrinsic types so we can do
Reid Spencer38cabd72005-05-03 07:23:44 +00001121 /// store instead of the memcpy call. Other calls are transformed into the
1122 /// llvm.memset intrinsic.
1123 /// @brief Perform the memset optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001124 virtual bool OptimizeCall(CallInst *ci, SimplifyLibCalls &TD) {
Reid Spencer38cabd72005-05-03 07:23:44 +00001125 // Make sure we have constant int values to work with
1126 ConstantInt* LEN = dyn_cast<ConstantInt>(ci->getOperand(3));
1127 if (!LEN)
1128 return false;
1129 ConstantInt* ALIGN = dyn_cast<ConstantInt>(ci->getOperand(4));
1130 if (!ALIGN)
1131 return false;
1132
1133 // Extract the length and alignment
1134 uint64_t len = LEN->getRawValue();
1135 uint64_t alignment = ALIGN->getRawValue();
1136
1137 // Alignment 0 is identity for alignment 1
1138 if (alignment == 0)
1139 alignment = 1;
1140
1141 // If the length is zero, this is a no-op
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001142 if (len == 0) {
Reid Spencer38cabd72005-05-03 07:23:44 +00001143 // memset(d,c,0,a) -> noop
1144 ci->eraseFromParent();
1145 return true;
1146 }
1147
1148 // If the length is larger than the alignment, we can't optimize
1149 if (len > alignment)
1150 return false;
1151
1152 // Make sure we have a constant ubyte to work with so we can extract
1153 // the value to be filled.
1154 ConstantUInt* FILL = dyn_cast<ConstantUInt>(ci->getOperand(2));
1155 if (!FILL)
1156 return false;
1157 if (FILL->getType() != Type::UByteTy)
1158 return false;
1159
1160 // memset(s,c,n) -> store s, c (for n=1,2,4,8)
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001161
Reid Spencer38cabd72005-05-03 07:23:44 +00001162 // Extract the fill character
1163 uint64_t fill_char = FILL->getValue();
1164 uint64_t fill_value = fill_char;
1165
1166 // Get the type we will cast to, based on size of memory area to fill, and
1167 // and the value we will store there.
1168 Value* dest = ci->getOperand(1);
1169 Type* castType = 0;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001170 switch (len) {
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001171 case 1:
1172 castType = Type::UByteTy;
Reid Spencer38cabd72005-05-03 07:23:44 +00001173 break;
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001174 case 2:
1175 castType = Type::UShortTy;
Reid Spencer38cabd72005-05-03 07:23:44 +00001176 fill_value |= fill_char << 8;
1177 break;
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001178 case 4:
Reid Spencer38cabd72005-05-03 07:23:44 +00001179 castType = Type::UIntTy;
1180 fill_value |= fill_char << 8 | fill_char << 16 | fill_char << 24;
1181 break;
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001182 case 8:
Reid Spencer38cabd72005-05-03 07:23:44 +00001183 castType = Type::ULongTy;
1184 fill_value |= fill_char << 8 | fill_char << 16 | fill_char << 24;
1185 fill_value |= fill_char << 32 | fill_char << 40 | fill_char << 48;
1186 fill_value |= fill_char << 56;
1187 break;
1188 default:
1189 return false;
1190 }
1191
1192 // Cast dest to the right sized primitive and then load/store
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001193 CastInst* DestCast =
Reid Spencer38cabd72005-05-03 07:23:44 +00001194 new CastInst(dest,PointerType::get(castType),dest->getName()+".cast",ci);
1195 new StoreInst(ConstantUInt::get(castType,fill_value),DestCast, ci);
1196 ci->eraseFromParent();
1197 return true;
1198 }
1199} LLVMMemSetOptimizer;
Reid Spencerbb92b4f2005-04-26 19:13:17 +00001200
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001201/// This LibCallOptimization will simplify calls to the "pow" library
1202/// function. It looks for cases where the result of pow is well known and
Reid Spencer93616972005-04-29 09:39:47 +00001203/// substitutes the appropriate value.
1204/// @brief Simplify the pow library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001205struct PowOptimization : public LibCallOptimization {
Reid Spencer93616972005-04-29 09:39:47 +00001206public:
1207 /// @brief Default Constructor
Reid Spencer95d8efd2005-05-03 02:54:54 +00001208 PowOptimization() : LibCallOptimization("pow",
Reid Spencer170ae7f2005-05-07 20:15:59 +00001209 "Number of 'pow' calls simplified") {}
Reid Spencer95d8efd2005-05-03 02:54:54 +00001210
Reid Spencer93616972005-04-29 09:39:47 +00001211 /// @brief Make sure that the "pow" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001212 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Reid Spencer93616972005-04-29 09:39:47 +00001213 // Just make sure this has 2 arguments
1214 return (f->arg_size() == 2);
1215 }
1216
1217 /// @brief Perform the pow optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001218 virtual bool OptimizeCall(CallInst *ci, SimplifyLibCalls &SLC) {
Reid Spencer93616972005-04-29 09:39:47 +00001219 const Type *Ty = cast<Function>(ci->getOperand(0))->getReturnType();
1220 Value* base = ci->getOperand(1);
1221 Value* expn = ci->getOperand(2);
1222 if (ConstantFP *Op1 = dyn_cast<ConstantFP>(base)) {
1223 double Op1V = Op1->getValue();
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001224 if (Op1V == 1.0) {
Reid Spencer93616972005-04-29 09:39:47 +00001225 // pow(1.0,x) -> 1.0
1226 ci->replaceAllUsesWith(ConstantFP::get(Ty,1.0));
1227 ci->eraseFromParent();
1228 return true;
1229 }
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001230 } else if (ConstantFP* Op2 = dyn_cast<ConstantFP>(expn)) {
Reid Spencer93616972005-04-29 09:39:47 +00001231 double Op2V = Op2->getValue();
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001232 if (Op2V == 0.0) {
Reid Spencer93616972005-04-29 09:39:47 +00001233 // pow(x,0.0) -> 1.0
1234 ci->replaceAllUsesWith(ConstantFP::get(Ty,1.0));
1235 ci->eraseFromParent();
1236 return true;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001237 } else if (Op2V == 0.5) {
Reid Spencer93616972005-04-29 09:39:47 +00001238 // pow(x,0.5) -> sqrt(x)
1239 CallInst* sqrt_inst = new CallInst(SLC.get_sqrt(), base,
1240 ci->getName()+".pow",ci);
1241 ci->replaceAllUsesWith(sqrt_inst);
1242 ci->eraseFromParent();
1243 return true;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001244 } else if (Op2V == 1.0) {
Reid Spencer93616972005-04-29 09:39:47 +00001245 // pow(x,1.0) -> x
1246 ci->replaceAllUsesWith(base);
1247 ci->eraseFromParent();
1248 return true;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001249 } else if (Op2V == -1.0) {
Reid Spencer93616972005-04-29 09:39:47 +00001250 // pow(x,-1.0) -> 1.0/x
Chris Lattner4201cd12005-08-24 17:22:17 +00001251 BinaryOperator* div_inst= BinaryOperator::createDiv(
Reid Spencer93616972005-04-29 09:39:47 +00001252 ConstantFP::get(Ty,1.0), base, ci->getName()+".pow", ci);
1253 ci->replaceAllUsesWith(div_inst);
1254 ci->eraseFromParent();
1255 return true;
1256 }
1257 }
1258 return false; // opt failed
1259 }
1260} PowOptimizer;
1261
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001262/// This LibCallOptimization will simplify calls to the "fprintf" library
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001263/// function. It looks for cases where the result of fprintf is not used and the
1264/// operation can be reduced to something simpler.
1265/// @brief Simplify the pow library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001266struct FPrintFOptimization : public LibCallOptimization {
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001267public:
1268 /// @brief Default Constructor
Reid Spencer95d8efd2005-05-03 02:54:54 +00001269 FPrintFOptimization() : LibCallOptimization("fprintf",
Reid Spencer170ae7f2005-05-07 20:15:59 +00001270 "Number of 'fprintf' calls simplified") {}
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001271
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001272 /// @brief Make sure that the "fprintf" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001273 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001274 // Just make sure this has at least 2 arguments
1275 return (f->arg_size() >= 2);
1276 }
1277
1278 /// @brief Perform the fprintf optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001279 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& SLC) {
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001280 // If the call has more than 3 operands, we can't optimize it
1281 if (ci->getNumOperands() > 4 || ci->getNumOperands() <= 2)
1282 return false;
1283
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001284 // If the result of the fprintf call is used, none of these optimizations
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001285 // can be made.
Chris Lattner175463a2005-09-24 22:17:06 +00001286 if (!ci->use_empty())
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001287 return false;
1288
1289 // All the optimizations depend on the length of the second argument and the
1290 // fact that it is a constant string array. Check that now
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001291 uint64_t len = 0;
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001292 ConstantArray* CA = 0;
1293 if (!getConstantStringLength(ci->getOperand(2), len, &CA))
1294 return false;
1295
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001296 if (ci->getNumOperands() == 3) {
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001297 // Make sure there's no % in the constant array
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001298 for (unsigned i = 0; i < len; ++i) {
1299 if (ConstantInt* CI = dyn_cast<ConstantInt>(CA->getOperand(i))) {
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001300 // Check for the null terminator
1301 if (CI->getRawValue() == '%')
1302 return false; // we found end of string
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001303 } else {
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001304 return false;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001305 }
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001306 }
1307
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001308 // fprintf(file,fmt) -> fwrite(fmt,strlen(fmt),file)
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001309 const Type* FILEptr_type = ci->getOperand(1)->getType();
1310 Function* fwrite_func = SLC.get_fwrite(FILEptr_type);
1311 if (!fwrite_func)
1312 return false;
John Criswell4642afd2005-06-29 15:03:18 +00001313
1314 // Make sure that the fprintf() and fwrite() functions both take the
1315 // same type of char pointer.
1316 if (ci->getOperand(2)->getType() !=
1317 fwrite_func->getFunctionType()->getParamType(0))
John Criswell4642afd2005-06-29 15:03:18 +00001318 return false;
John Criswell4642afd2005-06-29 15:03:18 +00001319
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001320 std::vector<Value*> args;
1321 args.push_back(ci->getOperand(2));
1322 args.push_back(ConstantUInt::get(SLC.getIntPtrType(),len));
1323 args.push_back(ConstantUInt::get(SLC.getIntPtrType(),1));
1324 args.push_back(ci->getOperand(1));
Reid Spencer1e520fd2005-05-04 03:20:21 +00001325 new CallInst(fwrite_func,args,ci->getName(),ci);
1326 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,len));
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001327 ci->eraseFromParent();
1328 return true;
1329 }
1330
1331 // The remaining optimizations require the format string to be length 2
1332 // "%s" or "%c".
1333 if (len != 2)
1334 return false;
1335
1336 // The first character has to be a %
1337 if (ConstantInt* CI = dyn_cast<ConstantInt>(CA->getOperand(0)))
1338 if (CI->getRawValue() != '%')
1339 return false;
1340
1341 // Get the second character and switch on its value
1342 ConstantInt* CI = dyn_cast<ConstantInt>(CA->getOperand(1));
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001343 switch (CI->getRawValue()) {
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001344 case 's':
1345 {
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001346 uint64_t len = 0;
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001347 ConstantArray* CA = 0;
1348 if (!getConstantStringLength(ci->getOperand(3), len, &CA))
1349 return false;
1350
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001351 // fprintf(file,"%s",str) -> fwrite(fmt,strlen(fmt),1,file)
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001352 const Type* FILEptr_type = ci->getOperand(1)->getType();
1353 Function* fwrite_func = SLC.get_fwrite(FILEptr_type);
1354 if (!fwrite_func)
1355 return false;
1356 std::vector<Value*> args;
Reid Spencer45bb4af2005-05-21 00:39:30 +00001357 args.push_back(CastToCStr(ci->getOperand(3), *ci));
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001358 args.push_back(ConstantUInt::get(SLC.getIntPtrType(),len));
1359 args.push_back(ConstantUInt::get(SLC.getIntPtrType(),1));
1360 args.push_back(ci->getOperand(1));
Reid Spencer1e520fd2005-05-04 03:20:21 +00001361 new CallInst(fwrite_func,args,ci->getName(),ci);
1362 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,len));
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001363 break;
1364 }
1365 case 'c':
1366 {
1367 ConstantInt* CI = dyn_cast<ConstantInt>(ci->getOperand(3));
1368 if (!CI)
1369 return false;
1370
1371 const Type* FILEptr_type = ci->getOperand(1)->getType();
1372 Function* fputc_func = SLC.get_fputc(FILEptr_type);
1373 if (!fputc_func)
1374 return false;
1375 CastInst* cast = new CastInst(CI,Type::IntTy,CI->getName()+".int",ci);
1376 new CallInst(fputc_func,cast,ci->getOperand(1),"",ci);
Reid Spencer1e520fd2005-05-04 03:20:21 +00001377 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,1));
Reid Spencer2d5c7be2005-05-02 23:59:26 +00001378 break;
1379 }
1380 default:
1381 return false;
1382 }
1383 ci->eraseFromParent();
1384 return true;
1385 }
1386} FPrintFOptimizer;
1387
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001388/// This LibCallOptimization will simplify calls to the "sprintf" library
Reid Spencer1e520fd2005-05-04 03:20:21 +00001389/// function. It looks for cases where the result of sprintf is not used and the
1390/// operation can be reduced to something simpler.
1391/// @brief Simplify the pow library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001392struct SPrintFOptimization : public LibCallOptimization {
Reid Spencer1e520fd2005-05-04 03:20:21 +00001393public:
1394 /// @brief Default Constructor
1395 SPrintFOptimization() : LibCallOptimization("sprintf",
Reid Spencer170ae7f2005-05-07 20:15:59 +00001396 "Number of 'sprintf' calls simplified") {}
Reid Spencer1e520fd2005-05-04 03:20:21 +00001397
Reid Spencer1e520fd2005-05-04 03:20:21 +00001398 /// @brief Make sure that the "fprintf" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001399 virtual bool ValidateCalledFunction(const Function *f, SimplifyLibCalls &SLC){
Reid Spencer1e520fd2005-05-04 03:20:21 +00001400 // Just make sure this has at least 2 arguments
1401 return (f->getReturnType() == Type::IntTy && f->arg_size() >= 2);
1402 }
1403
1404 /// @brief Perform the sprintf optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001405 virtual bool OptimizeCall(CallInst *ci, SimplifyLibCalls &SLC) {
Reid Spencer1e520fd2005-05-04 03:20:21 +00001406 // If the call has more than 3 operands, we can't optimize it
1407 if (ci->getNumOperands() > 4 || ci->getNumOperands() < 3)
1408 return false;
1409
1410 // All the optimizations depend on the length of the second argument and the
1411 // fact that it is a constant string array. Check that now
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001412 uint64_t len = 0;
Reid Spencer1e520fd2005-05-04 03:20:21 +00001413 ConstantArray* CA = 0;
1414 if (!getConstantStringLength(ci->getOperand(2), len, &CA))
1415 return false;
1416
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001417 if (ci->getNumOperands() == 3) {
1418 if (len == 0) {
Reid Spencer1e520fd2005-05-04 03:20:21 +00001419 // If the length is 0, we just need to store a null byte
1420 new StoreInst(ConstantInt::get(Type::SByteTy,0),ci->getOperand(1),ci);
1421 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,0));
1422 ci->eraseFromParent();
1423 return true;
1424 }
1425
1426 // Make sure there's no % in the constant array
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001427 for (unsigned i = 0; i < len; ++i) {
1428 if (ConstantInt* CI = dyn_cast<ConstantInt>(CA->getOperand(i))) {
Reid Spencer1e520fd2005-05-04 03:20:21 +00001429 // Check for the null terminator
1430 if (CI->getRawValue() == '%')
1431 return false; // we found a %, can't optimize
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001432 } else {
Reid Spencer1e520fd2005-05-04 03:20:21 +00001433 return false; // initializer is not constant int, can't optimize
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001434 }
Reid Spencer1e520fd2005-05-04 03:20:21 +00001435 }
1436
1437 // Increment length because we want to copy the null byte too
1438 len++;
1439
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001440 // sprintf(str,fmt) -> llvm.memcpy(str,fmt,strlen(fmt),1)
Reid Spencer1e520fd2005-05-04 03:20:21 +00001441 Function* memcpy_func = SLC.get_memcpy();
1442 if (!memcpy_func)
1443 return false;
1444 std::vector<Value*> args;
1445 args.push_back(ci->getOperand(1));
1446 args.push_back(ci->getOperand(2));
1447 args.push_back(ConstantUInt::get(Type::UIntTy,len));
1448 args.push_back(ConstantUInt::get(Type::UIntTy,1));
1449 new CallInst(memcpy_func,args,"",ci);
1450 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,len));
1451 ci->eraseFromParent();
1452 return true;
1453 }
1454
1455 // The remaining optimizations require the format string to be length 2
1456 // "%s" or "%c".
1457 if (len != 2)
1458 return false;
1459
1460 // The first character has to be a %
1461 if (ConstantInt* CI = dyn_cast<ConstantInt>(CA->getOperand(0)))
1462 if (CI->getRawValue() != '%')
1463 return false;
1464
1465 // Get the second character and switch on its value
1466 ConstantInt* CI = dyn_cast<ConstantInt>(CA->getOperand(1));
Chris Lattner175463a2005-09-24 22:17:06 +00001467 switch (CI->getRawValue()) {
1468 case 's': {
1469 // sprintf(dest,"%s",str) -> llvm.memcpy(dest, str, strlen(str)+1, 1)
1470 Function* strlen_func = SLC.get_strlen();
1471 Function* memcpy_func = SLC.get_memcpy();
1472 if (!strlen_func || !memcpy_func)
Reid Spencer1e520fd2005-05-04 03:20:21 +00001473 return false;
Chris Lattner175463a2005-09-24 22:17:06 +00001474
1475 Value *Len = new CallInst(strlen_func, CastToCStr(ci->getOperand(3), *ci),
1476 ci->getOperand(3)->getName()+".len", ci);
1477 Value *Len1 = BinaryOperator::createAdd(Len,
1478 ConstantInt::get(Len->getType(), 1),
1479 Len->getName()+"1", ci);
1480 if (Len1->getType() != Type::UIntTy)
1481 Len1 = new CastInst(Len1, Type::UIntTy, Len1->getName(), ci);
1482 std::vector<Value*> args;
1483 args.push_back(CastToCStr(ci->getOperand(1), *ci));
1484 args.push_back(CastToCStr(ci->getOperand(3), *ci));
1485 args.push_back(Len1);
1486 args.push_back(ConstantUInt::get(Type::UIntTy,1));
1487 new CallInst(memcpy_func, args, "", ci);
1488
1489 // The strlen result is the unincremented number of bytes in the string.
Chris Lattnerf4877682005-09-25 07:06:48 +00001490 if (!ci->use_empty()) {
1491 if (Len->getType() != ci->getType())
1492 Len = new CastInst(Len, ci->getType(), Len->getName(), ci);
1493 ci->replaceAllUsesWith(Len);
1494 }
Chris Lattner175463a2005-09-24 22:17:06 +00001495 ci->eraseFromParent();
1496 return true;
Reid Spencer1e520fd2005-05-04 03:20:21 +00001497 }
Chris Lattner175463a2005-09-24 22:17:06 +00001498 case 'c': {
1499 // sprintf(dest,"%c",chr) -> store chr, dest
1500 CastInst* cast = new CastInst(ci->getOperand(3),Type::SByteTy,"char",ci);
1501 new StoreInst(cast, ci->getOperand(1), ci);
1502 GetElementPtrInst* gep = new GetElementPtrInst(ci->getOperand(1),
1503 ConstantUInt::get(Type::UIntTy,1),ci->getOperand(1)->getName()+".end",
1504 ci);
1505 new StoreInst(ConstantInt::get(Type::SByteTy,0),gep,ci);
1506 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,1));
1507 ci->eraseFromParent();
1508 return true;
1509 }
1510 }
1511 return false;
Reid Spencer1e520fd2005-05-04 03:20:21 +00001512 }
1513} SPrintFOptimizer;
1514
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001515/// This LibCallOptimization will simplify calls to the "fputs" library
Reid Spencer93616972005-04-29 09:39:47 +00001516/// function. It looks for cases where the result of fputs is not used and the
1517/// operation can be reduced to something simpler.
1518/// @brief Simplify the pow library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001519struct PutsOptimization : public LibCallOptimization {
Reid Spencer93616972005-04-29 09:39:47 +00001520public:
1521 /// @brief Default Constructor
Reid Spencer95d8efd2005-05-03 02:54:54 +00001522 PutsOptimization() : LibCallOptimization("fputs",
Reid Spencer170ae7f2005-05-07 20:15:59 +00001523 "Number of 'fputs' calls simplified") {}
Reid Spencer93616972005-04-29 09:39:47 +00001524
Reid Spencer93616972005-04-29 09:39:47 +00001525 /// @brief Make sure that the "fputs" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001526 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &SLC){
Reid Spencer93616972005-04-29 09:39:47 +00001527 // Just make sure this has 2 arguments
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001528 return F->arg_size() == 2;
Reid Spencer93616972005-04-29 09:39:47 +00001529 }
1530
1531 /// @brief Perform the fputs optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001532 virtual bool OptimizeCall(CallInst* ci, SimplifyLibCalls& SLC) {
Reid Spencer93616972005-04-29 09:39:47 +00001533 // If the result is used, none of these optimizations work
Chris Lattner175463a2005-09-24 22:17:06 +00001534 if (!ci->use_empty())
Reid Spencer93616972005-04-29 09:39:47 +00001535 return false;
1536
1537 // All the optimizations depend on the length of the first argument and the
1538 // fact that it is a constant string array. Check that now
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001539 uint64_t len = 0;
Reid Spencer93616972005-04-29 09:39:47 +00001540 if (!getConstantStringLength(ci->getOperand(1), len))
1541 return false;
1542
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001543 switch (len) {
Reid Spencer93616972005-04-29 09:39:47 +00001544 case 0:
1545 // fputs("",F) -> noop
1546 break;
1547 case 1:
1548 {
1549 // fputs(s,F) -> fputc(s[0],F) (if s is constant and strlen(s) == 1)
Reid Spencer4c444fe2005-04-30 03:17:54 +00001550 const Type* FILEptr_type = ci->getOperand(2)->getType();
1551 Function* fputc_func = SLC.get_fputc(FILEptr_type);
Reid Spencer93616972005-04-29 09:39:47 +00001552 if (!fputc_func)
1553 return false;
1554 LoadInst* loadi = new LoadInst(ci->getOperand(1),
1555 ci->getOperand(1)->getName()+".byte",ci);
1556 CastInst* casti = new CastInst(loadi,Type::IntTy,
1557 loadi->getName()+".int",ci);
1558 new CallInst(fputc_func,casti,ci->getOperand(2),"",ci);
1559 break;
1560 }
1561 default:
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001562 {
Reid Spencer93616972005-04-29 09:39:47 +00001563 // fputs(s,F) -> fwrite(s,1,len,F) (if s is constant and strlen(s) > 1)
Reid Spencer4c444fe2005-04-30 03:17:54 +00001564 const Type* FILEptr_type = ci->getOperand(2)->getType();
1565 Function* fwrite_func = SLC.get_fwrite(FILEptr_type);
Reid Spencer93616972005-04-29 09:39:47 +00001566 if (!fwrite_func)
1567 return false;
1568 std::vector<Value*> parms;
1569 parms.push_back(ci->getOperand(1));
1570 parms.push_back(ConstantUInt::get(SLC.getIntPtrType(),len));
1571 parms.push_back(ConstantUInt::get(SLC.getIntPtrType(),1));
1572 parms.push_back(ci->getOperand(2));
1573 new CallInst(fwrite_func,parms,"",ci);
1574 break;
1575 }
1576 }
1577 ci->eraseFromParent();
1578 return true; // success
1579 }
1580} PutsOptimizer;
1581
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001582/// This LibCallOptimization will simplify calls to the "isdigit" library
Reid Spencer282d0572005-05-04 18:58:28 +00001583/// function. It simply does range checks the parameter explicitly.
1584/// @brief Simplify the isdigit library function.
Chris Lattner5f6035f2005-09-29 06:16:11 +00001585struct isdigitOptimization : public LibCallOptimization {
Reid Spencer282d0572005-05-04 18:58:28 +00001586public:
Chris Lattner5f6035f2005-09-29 06:16:11 +00001587 isdigitOptimization() : LibCallOptimization("isdigit",
Reid Spencer170ae7f2005-05-07 20:15:59 +00001588 "Number of 'isdigit' calls simplified") {}
Reid Spencer282d0572005-05-04 18:58:28 +00001589
Chris Lattner5f6035f2005-09-29 06:16:11 +00001590 /// @brief Make sure that the "isdigit" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001591 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Reid Spencer282d0572005-05-04 18:58:28 +00001592 // Just make sure this has 1 argument
1593 return (f->arg_size() == 1);
1594 }
1595
1596 /// @brief Perform the toascii optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001597 virtual bool OptimizeCall(CallInst *ci, SimplifyLibCalls &SLC) {
1598 if (ConstantInt* CI = dyn_cast<ConstantInt>(ci->getOperand(1))) {
Reid Spencer282d0572005-05-04 18:58:28 +00001599 // isdigit(c) -> 0 or 1, if 'c' is constant
1600 uint64_t val = CI->getRawValue();
1601 if (val >= '0' && val <='9')
1602 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,1));
1603 else
1604 ci->replaceAllUsesWith(ConstantSInt::get(Type::IntTy,0));
1605 ci->eraseFromParent();
1606 return true;
1607 }
1608
1609 // isdigit(c) -> (unsigned)c - '0' <= 9
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001610 CastInst* cast =
Reid Spencer282d0572005-05-04 18:58:28 +00001611 new CastInst(ci->getOperand(1),Type::UIntTy,
1612 ci->getOperand(1)->getName()+".uint",ci);
Chris Lattner4201cd12005-08-24 17:22:17 +00001613 BinaryOperator* sub_inst = BinaryOperator::createSub(cast,
Reid Spencer282d0572005-05-04 18:58:28 +00001614 ConstantUInt::get(Type::UIntTy,0x30),
1615 ci->getOperand(1)->getName()+".sub",ci);
1616 SetCondInst* setcond_inst = new SetCondInst(Instruction::SetLE,sub_inst,
1617 ConstantUInt::get(Type::UIntTy,9),
1618 ci->getOperand(1)->getName()+".cmp",ci);
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001619 CastInst* c2 =
Reid Spencer282d0572005-05-04 18:58:28 +00001620 new CastInst(setcond_inst,Type::IntTy,
1621 ci->getOperand(1)->getName()+".isdigit",ci);
1622 ci->replaceAllUsesWith(c2);
1623 ci->eraseFromParent();
1624 return true;
1625 }
Chris Lattner5f6035f2005-09-29 06:16:11 +00001626} isdigitOptimizer;
1627
Chris Lattner87ef9432005-09-29 06:17:27 +00001628struct isasciiOptimization : public LibCallOptimization {
1629public:
1630 isasciiOptimization()
1631 : LibCallOptimization("isascii", "Number of 'isascii' calls simplified") {}
1632
1633 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &SLC){
1634 return F->arg_size() == 1 && F->arg_begin()->getType()->isInteger() &&
1635 F->getReturnType()->isInteger();
1636 }
1637
1638 /// @brief Perform the isascii optimization.
1639 virtual bool OptimizeCall(CallInst *CI, SimplifyLibCalls &SLC) {
1640 // isascii(c) -> (unsigned)c < 128
1641 Value *V = CI->getOperand(1);
1642 if (V->getType()->isSigned())
1643 V = new CastInst(V, V->getType()->getUnsignedVersion(), V->getName(), CI);
1644 Value *Cmp = BinaryOperator::createSetLT(V, ConstantUInt::get(V->getType(),
1645 128),
1646 V->getName()+".isascii", CI);
1647 if (Cmp->getType() != CI->getType())
1648 Cmp = new CastInst(Cmp, CI->getType(), Cmp->getName(), CI);
1649 CI->replaceAllUsesWith(Cmp);
1650 CI->eraseFromParent();
1651 return true;
1652 }
1653} isasciiOptimizer;
Chris Lattner5f6035f2005-09-29 06:16:11 +00001654
Reid Spencer282d0572005-05-04 18:58:28 +00001655
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001656/// This LibCallOptimization will simplify calls to the "toascii" library
Reid Spencer4c444fe2005-04-30 03:17:54 +00001657/// function. It simply does the corresponding and operation to restrict the
1658/// range of values to the ASCII character set (0-127).
1659/// @brief Simplify the toascii library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001660struct ToAsciiOptimization : public LibCallOptimization {
Reid Spencer4c444fe2005-04-30 03:17:54 +00001661public:
1662 /// @brief Default Constructor
Reid Spencer95d8efd2005-05-03 02:54:54 +00001663 ToAsciiOptimization() : LibCallOptimization("toascii",
Reid Spencer170ae7f2005-05-07 20:15:59 +00001664 "Number of 'toascii' calls simplified") {}
Reid Spencer4c444fe2005-04-30 03:17:54 +00001665
Reid Spencer4c444fe2005-04-30 03:17:54 +00001666 /// @brief Make sure that the "fputs" function has the right prototype
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001667 virtual bool ValidateCalledFunction(const Function* f, SimplifyLibCalls& SLC){
Reid Spencer4c444fe2005-04-30 03:17:54 +00001668 // Just make sure this has 2 arguments
1669 return (f->arg_size() == 1);
1670 }
1671
1672 /// @brief Perform the toascii optimization.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001673 virtual bool OptimizeCall(CallInst *ci, SimplifyLibCalls &SLC) {
Reid Spencer4c444fe2005-04-30 03:17:54 +00001674 // toascii(c) -> (c & 0x7f)
1675 Value* chr = ci->getOperand(1);
Chris Lattner4201cd12005-08-24 17:22:17 +00001676 BinaryOperator* and_inst = BinaryOperator::createAnd(chr,
Reid Spencer4c444fe2005-04-30 03:17:54 +00001677 ConstantInt::get(chr->getType(),0x7F),ci->getName()+".toascii",ci);
1678 ci->replaceAllUsesWith(and_inst);
1679 ci->eraseFromParent();
1680 return true;
1681 }
1682} ToAsciiOptimizer;
1683
Reid Spencerb195fcd2005-05-14 16:42:52 +00001684/// This LibCallOptimization will simplify calls to the "ffs" library
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001685/// calls which find the first set bit in an int, long, or long long. The
Reid Spencerb195fcd2005-05-14 16:42:52 +00001686/// optimization is to compute the result at compile time if the argument is
1687/// a constant.
1688/// @brief Simplify the ffs library function.
Chris Lattner801f4752006-01-17 18:27:17 +00001689struct FFSOptimization : public LibCallOptimization {
Reid Spencerb195fcd2005-05-14 16:42:52 +00001690protected:
1691 /// @brief Subclass Constructor
1692 FFSOptimization(const char* funcName, const char* description)
Chris Lattner801f4752006-01-17 18:27:17 +00001693 : LibCallOptimization(funcName, description) {}
Reid Spencerb195fcd2005-05-14 16:42:52 +00001694
1695public:
1696 /// @brief Default Constructor
1697 FFSOptimization() : LibCallOptimization("ffs",
1698 "Number of 'ffs' calls simplified") {}
1699
Chris Lattner801f4752006-01-17 18:27:17 +00001700 /// @brief Make sure that the "ffs" function has the right prototype
1701 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &SLC){
Reid Spencerb195fcd2005-05-14 16:42:52 +00001702 // Just make sure this has 2 arguments
Chris Lattner801f4752006-01-17 18:27:17 +00001703 return F->arg_size() == 1 && F->getReturnType() == Type::IntTy;
Reid Spencerb195fcd2005-05-14 16:42:52 +00001704 }
1705
1706 /// @brief Perform the ffs optimization.
Chris Lattner801f4752006-01-17 18:27:17 +00001707 virtual bool OptimizeCall(CallInst *TheCall, SimplifyLibCalls &SLC) {
1708 if (ConstantInt *CI = dyn_cast<ConstantInt>(TheCall->getOperand(1))) {
Reid Spencerb195fcd2005-05-14 16:42:52 +00001709 // ffs(cnst) -> bit#
1710 // ffsl(cnst) -> bit#
Reid Spencer17f77842005-05-15 21:19:45 +00001711 // ffsll(cnst) -> bit#
Reid Spencerb195fcd2005-05-14 16:42:52 +00001712 uint64_t val = CI->getRawValue();
Reid Spencer17f77842005-05-15 21:19:45 +00001713 int result = 0;
Chris Lattner801f4752006-01-17 18:27:17 +00001714 if (val) {
1715 ++result;
1716 while ((val & 1) == 0) {
1717 ++result;
1718 val >>= 1;
1719 }
Reid Spencer17f77842005-05-15 21:19:45 +00001720 }
Chris Lattner801f4752006-01-17 18:27:17 +00001721 TheCall->replaceAllUsesWith(ConstantSInt::get(Type::IntTy, result));
1722 TheCall->eraseFromParent();
Reid Spencerb195fcd2005-05-14 16:42:52 +00001723 return true;
1724 }
Reid Spencer17f77842005-05-15 21:19:45 +00001725
Chris Lattner801f4752006-01-17 18:27:17 +00001726 // ffs(x) -> x == 0 ? 0 : llvm.cttz(x)+1
1727 // ffsl(x) -> x == 0 ? 0 : llvm.cttz(x)+1
1728 // ffsll(x) -> x == 0 ? 0 : llvm.cttz(x)+1
1729 const Type *ArgType = TheCall->getOperand(1)->getType();
1730 ArgType = ArgType->getUnsignedVersion();
1731 const char *CTTZName;
1732 switch (ArgType->getTypeID()) {
1733 default: assert(0 && "Unknown unsigned type!");
1734 case Type::UByteTyID : CTTZName = "llvm.cttz.i8" ; break;
1735 case Type::UShortTyID: CTTZName = "llvm.cttz.i16"; break;
1736 case Type::UIntTyID : CTTZName = "llvm.cttz.i32"; break;
1737 case Type::ULongTyID : CTTZName = "llvm.cttz.i64"; break;
1738 }
1739
1740 Function *F = SLC.getModule()->getOrInsertFunction(CTTZName, ArgType,
1741 ArgType, NULL);
1742 Value *V = new CastInst(TheCall->getOperand(1), ArgType, "tmp", TheCall);
1743 Value *V2 = new CallInst(F, V, "tmp", TheCall);
1744 V2 = new CastInst(V2, Type::IntTy, "tmp", TheCall);
1745 V2 = BinaryOperator::createAdd(V2, ConstantSInt::get(Type::IntTy, 1),
1746 "tmp", TheCall);
1747 Value *Cond =
1748 BinaryOperator::createSetEQ(V, Constant::getNullValue(V->getType()),
1749 "tmp", TheCall);
1750 V2 = new SelectInst(Cond, ConstantInt::get(Type::IntTy, 0), V2,
1751 TheCall->getName(), TheCall);
1752 TheCall->replaceAllUsesWith(V2);
1753 TheCall->eraseFromParent();
Reid Spencer17f77842005-05-15 21:19:45 +00001754 return true;
Reid Spencerb195fcd2005-05-14 16:42:52 +00001755 }
1756} FFSOptimizer;
1757
1758/// This LibCallOptimization will simplify calls to the "ffsl" library
1759/// calls. It simply uses FFSOptimization for which the transformation is
1760/// identical.
1761/// @brief Simplify the ffsl library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001762struct FFSLOptimization : public FFSOptimization {
Reid Spencerb195fcd2005-05-14 16:42:52 +00001763public:
1764 /// @brief Default Constructor
1765 FFSLOptimization() : FFSOptimization("ffsl",
1766 "Number of 'ffsl' calls simplified") {}
1767
1768} FFSLOptimizer;
1769
1770/// This LibCallOptimization will simplify calls to the "ffsll" library
1771/// calls. It simply uses FFSOptimization for which the transformation is
1772/// identical.
1773/// @brief Simplify the ffsl library function.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001774struct FFSLLOptimization : public FFSOptimization {
Reid Spencerb195fcd2005-05-14 16:42:52 +00001775public:
1776 /// @brief Default Constructor
1777 FFSLLOptimization() : FFSOptimization("ffsll",
1778 "Number of 'ffsll' calls simplified") {}
1779
1780} FFSLLOptimizer;
1781
Chris Lattner57a28632006-01-23 05:57:36 +00001782/// This optimizes unary functions that take and return doubles.
1783struct UnaryDoubleFPOptimizer : public LibCallOptimization {
1784 UnaryDoubleFPOptimizer(const char *Fn, const char *Desc)
1785 : LibCallOptimization(Fn, Desc) {}
Chris Lattner4201cd12005-08-24 17:22:17 +00001786
Chris Lattner57a28632006-01-23 05:57:36 +00001787 // Make sure that this function has the right prototype
Chris Lattner4201cd12005-08-24 17:22:17 +00001788 virtual bool ValidateCalledFunction(const Function *F, SimplifyLibCalls &SLC){
1789 return F->arg_size() == 1 && F->arg_begin()->getType() == Type::DoubleTy &&
1790 F->getReturnType() == Type::DoubleTy;
1791 }
Chris Lattner57a28632006-01-23 05:57:36 +00001792
1793 /// ShrinkFunctionToFloatVersion - If the input to this function is really a
1794 /// float, strength reduce this to a float version of the function,
1795 /// e.g. floor((double)FLT) -> (double)floorf(FLT). This can only be called
1796 /// when the target supports the destination function and where there can be
1797 /// no precision loss.
1798 static bool ShrinkFunctionToFloatVersion(CallInst *CI, SimplifyLibCalls &SLC,
1799 Function *(SimplifyLibCalls::*FP)()){
Chris Lattner4201cd12005-08-24 17:22:17 +00001800 if (CastInst *Cast = dyn_cast<CastInst>(CI->getOperand(1)))
1801 if (Cast->getOperand(0)->getType() == Type::FloatTy) {
Chris Lattner57a28632006-01-23 05:57:36 +00001802 Value *New = new CallInst((SLC.*FP)(), Cast->getOperand(0),
Chris Lattner4201cd12005-08-24 17:22:17 +00001803 CI->getName(), CI);
1804 New = new CastInst(New, Type::DoubleTy, CI->getName(), CI);
1805 CI->replaceAllUsesWith(New);
1806 CI->eraseFromParent();
1807 if (Cast->use_empty())
1808 Cast->eraseFromParent();
1809 return true;
1810 }
Chris Lattner57a28632006-01-23 05:57:36 +00001811 return false;
Chris Lattner4201cd12005-08-24 17:22:17 +00001812 }
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001813};
1814
Chris Lattner57a28632006-01-23 05:57:36 +00001815
Chris Lattner57a28632006-01-23 05:57:36 +00001816struct FloorOptimization : public UnaryDoubleFPOptimizer {
1817 FloorOptimization()
1818 : UnaryDoubleFPOptimizer("floor", "Number of 'floor' calls simplified") {}
1819
1820 virtual bool OptimizeCall(CallInst *CI, SimplifyLibCalls &SLC) {
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001821#ifdef HAVE_FLOORF
Chris Lattner57a28632006-01-23 05:57:36 +00001822 // If this is a float argument passed in, convert to floorf.
1823 if (ShrinkFunctionToFloatVersion(CI, SLC, &SimplifyLibCalls::get_floorf))
1824 return true;
Reid Spencerade18212006-01-19 08:36:56 +00001825#endif
Chris Lattner57a28632006-01-23 05:57:36 +00001826 return false; // opt failed
1827 }
1828} FloorOptimizer;
Chris Lattner4201cd12005-08-24 17:22:17 +00001829
Chris Lattner57740402006-01-23 06:24:46 +00001830struct CeilOptimization : public UnaryDoubleFPOptimizer {
1831 CeilOptimization()
1832 : UnaryDoubleFPOptimizer("ceil", "Number of 'ceil' calls simplified") {}
1833
1834 virtual bool OptimizeCall(CallInst *CI, SimplifyLibCalls &SLC) {
1835#ifdef HAVE_CEILF
1836 // If this is a float argument passed in, convert to ceilf.
1837 if (ShrinkFunctionToFloatVersion(CI, SLC, &SimplifyLibCalls::get_ceilf))
1838 return true;
1839#endif
1840 return false; // opt failed
1841 }
1842} CeilOptimizer;
Chris Lattner4201cd12005-08-24 17:22:17 +00001843
Chris Lattner57740402006-01-23 06:24:46 +00001844struct RoundOptimization : public UnaryDoubleFPOptimizer {
1845 RoundOptimization()
1846 : UnaryDoubleFPOptimizer("round", "Number of 'round' calls simplified") {}
1847
1848 virtual bool OptimizeCall(CallInst *CI, SimplifyLibCalls &SLC) {
1849#ifdef HAVE_ROUNDF
1850 // If this is a float argument passed in, convert to roundf.
1851 if (ShrinkFunctionToFloatVersion(CI, SLC, &SimplifyLibCalls::get_roundf))
1852 return true;
1853#endif
1854 return false; // opt failed
1855 }
1856} RoundOptimizer;
1857
1858struct RintOptimization : public UnaryDoubleFPOptimizer {
1859 RintOptimization()
1860 : UnaryDoubleFPOptimizer("rint", "Number of 'rint' calls simplified") {}
1861
1862 virtual bool OptimizeCall(CallInst *CI, SimplifyLibCalls &SLC) {
1863#ifdef HAVE_RINTF
1864 // If this is a float argument passed in, convert to rintf.
1865 if (ShrinkFunctionToFloatVersion(CI, SLC, &SimplifyLibCalls::get_rintf))
1866 return true;
1867#endif
1868 return false; // opt failed
1869 }
1870} RintOptimizer;
1871
1872struct NearByIntOptimization : public UnaryDoubleFPOptimizer {
1873 NearByIntOptimization()
1874 : UnaryDoubleFPOptimizer("nearbyint",
1875 "Number of 'nearbyint' calls simplified") {}
1876
1877 virtual bool OptimizeCall(CallInst *CI, SimplifyLibCalls &SLC) {
1878#ifdef HAVE_NEARBYINTF
1879 // If this is a float argument passed in, convert to nearbyintf.
1880 if (ShrinkFunctionToFloatVersion(CI, SLC,&SimplifyLibCalls::get_nearbyintf))
1881 return true;
1882#endif
1883 return false; // opt failed
1884 }
1885} NearByIntOptimizer;
Chris Lattner4201cd12005-08-24 17:22:17 +00001886
Reid Spencer7ddcfb32005-04-27 21:29:20 +00001887/// A function to compute the length of a null-terminated constant array of
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001888/// integers. This function can't rely on the size of the constant array
1889/// because there could be a null terminator in the middle of the array.
1890/// We also have to bail out if we find a non-integer constant initializer
1891/// of one of the elements or if there is no null-terminator. The logic
Reid Spencer7ddcfb32005-04-27 21:29:20 +00001892/// below checks each of these conditions and will return true only if all
1893/// conditions are met. In that case, the \p len parameter is set to the length
1894/// of the null-terminated string. If false is returned, the conditions were
1895/// not met and len is set to 0.
1896/// @brief Get the length of a constant string (null-terminated array).
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001897bool getConstantStringLength(Value *V, uint64_t &len, ConstantArray **CA) {
Reid Spencere249a822005-04-27 07:54:40 +00001898 assert(V != 0 && "Invalid args to getConstantStringLength");
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001899 len = 0; // make sure we initialize this
Reid Spencere249a822005-04-27 07:54:40 +00001900 User* GEP = 0;
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001901 // If the value is not a GEP instruction nor a constant expression with a
1902 // GEP instruction, then return false because ConstantArray can't occur
Reid Spencere249a822005-04-27 07:54:40 +00001903 // any other way
1904 if (GetElementPtrInst* GEPI = dyn_cast<GetElementPtrInst>(V))
1905 GEP = GEPI;
1906 else if (ConstantExpr* CE = dyn_cast<ConstantExpr>(V))
1907 if (CE->getOpcode() == Instruction::GetElementPtr)
1908 GEP = CE;
1909 else
1910 return false;
1911 else
1912 return false;
1913
1914 // Make sure the GEP has exactly three arguments.
1915 if (GEP->getNumOperands() != 3)
1916 return false;
1917
1918 // Check to make sure that the first operand of the GEP is an integer and
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001919 // has value 0 so that we are sure we're indexing into the initializer.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001920 if (ConstantInt* op1 = dyn_cast<ConstantInt>(GEP->getOperand(1))) {
Reid Spencere249a822005-04-27 07:54:40 +00001921 if (!op1->isNullValue())
1922 return false;
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001923 } else
Reid Spencere249a822005-04-27 07:54:40 +00001924 return false;
1925
1926 // Ensure that the second operand is a ConstantInt. If it isn't then this
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001927 // GEP is wonky and we're not really sure what were referencing into and
Reid Spencere249a822005-04-27 07:54:40 +00001928 // better of not optimizing it. While we're at it, get the second index
1929 // value. We'll need this later for indexing the ConstantArray.
1930 uint64_t start_idx = 0;
1931 if (ConstantInt* CI = dyn_cast<ConstantInt>(GEP->getOperand(2)))
1932 start_idx = CI->getRawValue();
1933 else
1934 return false;
1935
1936 // The GEP instruction, constant or instruction, must reference a global
1937 // variable that is a constant and is initialized. The referenced constant
1938 // initializer is the array that we'll use for optimization.
1939 GlobalVariable* GV = dyn_cast<GlobalVariable>(GEP->getOperand(0));
1940 if (!GV || !GV->isConstant() || !GV->hasInitializer())
1941 return false;
1942
1943 // Get the initializer.
1944 Constant* INTLZR = GV->getInitializer();
1945
1946 // Handle the ConstantAggregateZero case
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001947 if (ConstantAggregateZero *CAZ = dyn_cast<ConstantAggregateZero>(INTLZR)) {
Reid Spencere249a822005-04-27 07:54:40 +00001948 // This is a degenerate case. The initializer is constant zero so the
1949 // length of the string must be zero.
1950 len = 0;
1951 return true;
1952 }
1953
1954 // Must be a Constant Array
1955 ConstantArray* A = dyn_cast<ConstantArray>(INTLZR);
1956 if (!A)
1957 return false;
1958
1959 // Get the number of elements in the array
1960 uint64_t max_elems = A->getType()->getNumElements();
1961
1962 // Traverse the constant array from start_idx (derived above) which is
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001963 // the place the GEP refers to in the array.
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001964 for (len = start_idx; len < max_elems; len++) {
1965 if (ConstantInt *CI = dyn_cast<ConstantInt>(A->getOperand(len))) {
Reid Spencere249a822005-04-27 07:54:40 +00001966 // Check for the null terminator
1967 if (CI->isNullValue())
1968 break; // we found end of string
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001969 } else
Reid Spencere249a822005-04-27 07:54:40 +00001970 return false; // This array isn't suitable, non-int initializer
1971 }
Chris Lattner0d4ebfc2006-01-22 22:35:08 +00001972
Reid Spencere249a822005-04-27 07:54:40 +00001973 if (len >= max_elems)
1974 return false; // This array isn't null terminated
1975
1976 // Subtract out the initial value from the length
1977 len -= start_idx;
Reid Spencer4c444fe2005-04-30 03:17:54 +00001978 if (CA)
1979 *CA = A;
Reid Spencere249a822005-04-27 07:54:40 +00001980 return true; // success!
1981}
1982
Reid Spencera7828ba2005-06-18 17:46:28 +00001983/// CastToCStr - Return V if it is an sbyte*, otherwise cast it to sbyte*,
1984/// inserting the cast before IP, and return the cast.
1985/// @brief Cast a value to a "C" string.
1986Value *CastToCStr(Value *V, Instruction &IP) {
1987 const Type *SBPTy = PointerType::get(Type::SByteTy);
1988 if (V->getType() != SBPTy)
1989 return new CastInst(V, SBPTy, V->getName(), &IP);
1990 return V;
1991}
1992
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00001993// TODO:
Reid Spencer649ac282005-04-28 04:40:06 +00001994// Additional cases that we need to add to this file:
1995//
Reid Spencer649ac282005-04-28 04:40:06 +00001996// cbrt:
Reid Spencer649ac282005-04-28 04:40:06 +00001997// * cbrt(expN(X)) -> expN(x/3)
1998// * cbrt(sqrt(x)) -> pow(x,1/6)
1999// * cbrt(sqrt(x)) -> pow(x,1/9)
2000//
Reid Spencer649ac282005-04-28 04:40:06 +00002001// cos, cosf, cosl:
Reid Spencer16983ca2005-04-28 18:05:16 +00002002// * cos(-x) -> cos(x)
Reid Spencer649ac282005-04-28 04:40:06 +00002003//
2004// exp, expf, expl:
Reid Spencer649ac282005-04-28 04:40:06 +00002005// * exp(log(x)) -> x
2006//
Reid Spencer649ac282005-04-28 04:40:06 +00002007// log, logf, logl:
Reid Spencer649ac282005-04-28 04:40:06 +00002008// * log(exp(x)) -> x
2009// * log(x**y) -> y*log(x)
2010// * log(exp(y)) -> y*log(e)
2011// * log(exp2(y)) -> y*log(2)
2012// * log(exp10(y)) -> y*log(10)
2013// * log(sqrt(x)) -> 0.5*log(x)
2014// * log(pow(x,y)) -> y*log(x)
2015//
2016// lround, lroundf, lroundl:
2017// * lround(cnst) -> cnst'
2018//
2019// memcmp:
Reid Spencer649ac282005-04-28 04:40:06 +00002020// * memcmp(x,y,l) -> cnst
2021// (if all arguments are constant and strlen(x) <= l and strlen(y) <= l)
Reid Spencer649ac282005-04-28 04:40:06 +00002022//
Reid Spencer649ac282005-04-28 04:40:06 +00002023// memmove:
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00002024// * memmove(d,s,l,a) -> memcpy(d,s,l,a)
Reid Spencer649ac282005-04-28 04:40:06 +00002025// (if s is a global constant array)
2026//
Reid Spencer649ac282005-04-28 04:40:06 +00002027// pow, powf, powl:
Reid Spencer649ac282005-04-28 04:40:06 +00002028// * pow(exp(x),y) -> exp(x*y)
2029// * pow(sqrt(x),y) -> pow(x,y*0.5)
2030// * pow(pow(x,y),z)-> pow(x,y*z)
2031//
2032// puts:
2033// * puts("") -> fputc("\n",stdout) (how do we get "stdout"?)
2034//
2035// round, roundf, roundl:
2036// * round(cnst) -> cnst'
2037//
2038// signbit:
2039// * signbit(cnst) -> cnst'
2040// * signbit(nncst) -> 0 (if pstv is a non-negative constant)
2041//
Reid Spencer649ac282005-04-28 04:40:06 +00002042// sqrt, sqrtf, sqrtl:
Reid Spencer649ac282005-04-28 04:40:06 +00002043// * sqrt(expN(x)) -> expN(x*0.5)
2044// * sqrt(Nroot(x)) -> pow(x,1/(2*N))
2045// * sqrt(pow(x,y)) -> pow(|x|,y*0.5)
2046//
Reid Spencer170ae7f2005-05-07 20:15:59 +00002047// stpcpy:
2048// * stpcpy(str, "literal") ->
2049// llvm.memcpy(str,"literal",strlen("literal")+1,1)
Reid Spencer38cabd72005-05-03 07:23:44 +00002050// strrchr:
Reid Spencer649ac282005-04-28 04:40:06 +00002051// * strrchr(s,c) -> reverse_offset_of_in(c,s)
2052// (if c is a constant integer and s is a constant string)
2053// * strrchr(s1,0) -> strchr(s1,0)
2054//
Reid Spencer649ac282005-04-28 04:40:06 +00002055// strncat:
2056// * strncat(x,y,0) -> x
2057// * strncat(x,y,0) -> x (if strlen(y) = 0)
2058// * strncat(x,y,l) -> strcat(x,y) (if y and l are constants an l > strlen(y))
2059//
Reid Spencer649ac282005-04-28 04:40:06 +00002060// strncpy:
2061// * strncpy(d,s,0) -> d
2062// * strncpy(d,s,l) -> memcpy(d,s,l,1)
2063// (if s and l are constants)
2064//
2065// strpbrk:
2066// * strpbrk(s,a) -> offset_in_for(s,a)
2067// (if s and a are both constant strings)
2068// * strpbrk(s,"") -> 0
2069// * strpbrk(s,a) -> strchr(s,a[0]) (if a is constant string of length 1)
2070//
2071// strspn, strcspn:
2072// * strspn(s,a) -> const_int (if both args are constant)
2073// * strspn("",a) -> 0
2074// * strspn(s,"") -> 0
2075// * strcspn(s,a) -> const_int (if both args are constant)
2076// * strcspn("",a) -> 0
2077// * strcspn(s,"") -> strlen(a)
2078//
2079// strstr:
2080// * strstr(x,x) -> x
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00002081// * strstr(s1,s2) -> offset_of_s2_in(s1)
Reid Spencer649ac282005-04-28 04:40:06 +00002082// (if s1 and s2 are constant strings)
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00002083//
Reid Spencer649ac282005-04-28 04:40:06 +00002084// tan, tanf, tanl:
Reid Spencer649ac282005-04-28 04:40:06 +00002085// * tan(atan(x)) -> x
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00002086//
Reid Spencer649ac282005-04-28 04:40:06 +00002087// trunc, truncf, truncl:
2088// * trunc(cnst) -> cnst'
2089//
Jeff Cohen5f4ef3c2005-07-27 06:12:32 +00002090//
Reid Spencer39a762d2005-04-25 02:53:12 +00002091}