mirror of
https://github.com/RPCS3/llvm.git
synced 2024-12-30 16:34:03 +00:00
667d4b8de6
and extern_weak_odr. These are the same as the non-odr versions, except that they indicate that the global will only be overridden by an *equivalent* global. In C, a function with weak linkage can be overridden by a function which behaves completely differently. This means that IP passes have to skip weak functions, since any deductions made from the function definition might be wrong, since the definition could be replaced by something completely different at link time. This is not allowed in C++, thanks to the ODR (One-Definition-Rule): if a function is replaced by another at link-time, then the new function must be the same as the original function. If a language knows that a function or other global can only be overridden by an equivalent global, it can give it the weak_odr linkage type, and the optimizers will understand that it is alright to make deductions based on the function body. The code generators on the other hand map weak and weak_odr linkage to the same thing. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@66339 91177308-0d34-0410-b5e6-96231b3b80d8
90 lines
3.3 KiB
C++
90 lines
3.3 KiB
C++
//===-- IndMemRemoval.cpp - Remove indirect allocations and frees ----------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This pass finds places where memory allocation functions may escape into
|
|
// indirect land. Some transforms are much easier (aka possible) only if free
|
|
// or malloc are not called indirectly.
|
|
// Thus find places where the address of memory functions are taken and construct
|
|
// bounce functions with direct calls of those functions.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#define DEBUG_TYPE "indmemrem"
|
|
#include "llvm/Transforms/IPO.h"
|
|
#include "llvm/Pass.h"
|
|
#include "llvm/Module.h"
|
|
#include "llvm/Instructions.h"
|
|
#include "llvm/Type.h"
|
|
#include "llvm/DerivedTypes.h"
|
|
#include "llvm/ADT/Statistic.h"
|
|
#include "llvm/Support/Compiler.h"
|
|
using namespace llvm;
|
|
|
|
STATISTIC(NumBounceSites, "Number of sites modified");
|
|
STATISTIC(NumBounce , "Number of bounce functions created");
|
|
|
|
namespace {
|
|
class VISIBILITY_HIDDEN IndMemRemPass : public ModulePass {
|
|
public:
|
|
static char ID; // Pass identification, replacement for typeid
|
|
IndMemRemPass() : ModulePass(&ID) {}
|
|
|
|
virtual bool runOnModule(Module &M);
|
|
};
|
|
} // end anonymous namespace
|
|
|
|
char IndMemRemPass::ID = 0;
|
|
static RegisterPass<IndMemRemPass>
|
|
X("indmemrem","Indirect Malloc and Free Removal");
|
|
|
|
bool IndMemRemPass::runOnModule(Module &M) {
|
|
// In theory, all direct calls of malloc and free should be promoted
|
|
// to intrinsics. Therefore, this goes through and finds where the
|
|
// address of free or malloc are taken and replaces those with bounce
|
|
// functions, ensuring that all malloc and free that might happen
|
|
// happen through intrinsics.
|
|
bool changed = false;
|
|
if (Function* F = M.getFunction("free")) {
|
|
if (F->isDeclaration() && F->arg_size() == 1 && !F->use_empty()) {
|
|
Function* FN = Function::Create(F->getFunctionType(),
|
|
GlobalValue::LinkOnceAnyLinkage,
|
|
"free_llvm_bounce", &M);
|
|
BasicBlock* bb = BasicBlock::Create("entry",FN);
|
|
Instruction* R = ReturnInst::Create(bb);
|
|
new FreeInst(FN->arg_begin(), R);
|
|
++NumBounce;
|
|
NumBounceSites += F->getNumUses();
|
|
F->replaceAllUsesWith(FN);
|
|
changed = true;
|
|
}
|
|
}
|
|
if (Function* F = M.getFunction("malloc")) {
|
|
if (F->isDeclaration() && F->arg_size() == 1 && !F->use_empty()) {
|
|
Function* FN = Function::Create(F->getFunctionType(),
|
|
GlobalValue::LinkOnceAnyLinkage,
|
|
"malloc_llvm_bounce", &M);
|
|
FN->setDoesNotAlias(0);
|
|
BasicBlock* bb = BasicBlock::Create("entry",FN);
|
|
Instruction* c = CastInst::CreateIntegerCast(
|
|
FN->arg_begin(), Type::Int32Ty, false, "c", bb);
|
|
Instruction* a = new MallocInst(Type::Int8Ty, c, "m", bb);
|
|
ReturnInst::Create(a, bb);
|
|
++NumBounce;
|
|
NumBounceSites += F->getNumUses();
|
|
F->replaceAllUsesWith(FN);
|
|
changed = true;
|
|
}
|
|
}
|
|
return changed;
|
|
}
|
|
|
|
ModulePass *llvm::createIndMemRemPass() {
|
|
return new IndMemRemPass();
|
|
}
|