1 //===- Hello.cpp - Example code from "Writing an LLVM Pass" ---------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements two versions of the LLVM "Hello World" pass described
11 // in docs/WritingAnLLVMPass.html
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #define DEBUG_TYPE "hello"
16 #include "llvm/Pass.h"
17 #include "llvm/Function.h"
18 #include "llvm/ADT/StringExtras.h"
19 #include "llvm/Support/Streams.h"
20 #include "llvm/ADT/Statistic.h"
21 using namespace llvm;
22 
23 STATISTIC(HelloCounter, "Counts number of functions greeted");
24 
25 namespace {
26   // Hello - The first implementation, without getAnalysisUsage.
27   struct Hello : public FunctionPass {
28     static const int ID; // Pass identifcation, replacement for typeid
29     Hello() : FunctionPass((intptr_t)&ID) {}
30 
31     virtual bool runOnFunction(Function &F) {
32       HelloCounter++;
33       std::string fname = F.getName();
34       EscapeString(fname);
35       cerr << "Hello: " << fname << "\n";
36       return false;
37     }
38   };
39 
40   const int Hello::ID = 0;
41   RegisterPass<Hello> X("hello", "Hello World Pass");
42 
43   // Hello2 - The second implementation with getAnalysisUsage implemented.
44   struct Hello2 : public FunctionPass {
45     static const int ID; // Pass identifcation, replacement for typeid
46     Hello2() : FunctionPass((intptr_t)&ID) {}
47 
48     virtual bool runOnFunction(Function &F) {
49       HelloCounter++;
50       std::string fname = F.getName();
51       EscapeString(fname);
52       cerr << "Hello: " << fname << "\n";
53       return false;
54     }
55 
56     // We don't modify the program, so we preserve all analyses
57     virtual void getAnalysisUsage(AnalysisUsage &AU) const {
58       AU.setPreservesAll();
59     };
60   };
61   const int Hello2::ID = 0;
62   RegisterPass<Hello2> Y("hello2",
63                         "Hello World Pass (with getAnalysisUsage implemented)");
64 }
65