1 //===-- ExtractGV.cpp - Global Value extraction pass ----------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This pass extracts global values
12 //===----------------------------------------------------------------------===//
14 #include "llvm/Instructions.h"
15 #include "llvm/LLVMContext.h"
16 #include "llvm/Module.h"
17 #include "llvm/Pass.h"
18 #include "llvm/Constants.h"
19 #include "llvm/Transforms/IPO.h"
20 #include "llvm/ADT/SetVector.h"
25 /// @brief A pass to extract specific functions and their dependencies.
26 class GVExtractorPass : public ModulePass {
27 SetVector<GlobalValue *> Named;
30 static char ID; // Pass identification, replacement for typeid
32 /// FunctionExtractorPass - If deleteFn is true, this pass deletes as the
33 /// specified function. Otherwise, it deletes as much of the module as
34 /// possible, except for the function specified.
36 explicit GVExtractorPass(std::vector<GlobalValue*>& GVs, bool deleteS = true)
37 : ModulePass(ID), Named(GVs.begin(), GVs.end()), deleteStuff(deleteS) {}
39 bool runOnModule(Module &M) {
40 // Visit the global inline asm.
42 M.setModuleInlineAsm("");
44 // For simplicity, just give all GlobalValues ExternalLinkage. A trickier
45 // implementation could figure out which GlobalValues are actually
46 // referenced by the Named set, and which GlobalValues in the rest of
47 // the module are referenced by the NamedSet, and get away with leaving
48 // more internal and private things internal and private. But for now,
49 // be conservative and simple.
51 // Visit the GlobalVariables.
52 for (Module::global_iterator I = M.global_begin(), E = M.global_end();
54 if (I->hasLocalLinkage())
55 I->setVisibility(GlobalValue::HiddenVisibility);
56 I->setLinkage(GlobalValue::ExternalLinkage);
57 if (deleteStuff == (bool)Named.count(I) && !I->isDeclaration())
61 // Visit the Functions.
62 for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I) {
63 if (I->hasLocalLinkage())
64 I->setVisibility(GlobalValue::HiddenVisibility);
65 I->setLinkage(GlobalValue::ExternalLinkage);
66 if (deleteStuff == (bool)Named.count(I) && !I->isDeclaration())
74 char GVExtractorPass::ID = 0;
77 ModulePass *llvm::createGVExtractionPass(std::vector<GlobalValue*>& GVs,
79 return new GVExtractorPass(GVs, deleteFn);