Function wrapping implementation and Symtab extensions for undefined symbols
[dyninst.git] / dyninstAPI / src / Relocation / Transformers / Modification.h
1 /*
2  * Copyright (c) 1996-2009 Barton P. Miller
3  * 
4  * We provide the Paradyn Parallel Performance Tools (below
5  * described as "Paradyn") on an AS IS basis, and do not warrant its
6  * validity or performance.  We reserve the right to update, modify,
7  * or discontinue this software at any time.  We shall have no
8  * obligation to supply such updates or modifications or any other
9  * form of support to you.
10  * 
11  * By your use of Paradyn, you understand and agree that we (or any
12  * other person or entity with proprietary rights in Paradyn) are
13  * under no obligation to provide either maintenance services,
14  * update services, notices of latent defects, or correction of
15  * defects for Paradyn.
16  * 
17  * This library is free software; you can redistribute it and/or
18  * modify it under the terms of the GNU Lesser General Public
19  * License as published by the Free Software Foundation; either
20  * version 2.1 of the License, or (at your option) any later version.
21  * 
22  * This library is distributed in the hope that it will be useful,
23  * but WITHOUT ANY WARRANTY; without even the implied warranty of
24  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
25  * Lesser General Public License for more details.
26  * 
27  * You should have received a copy of the GNU Lesser General Public
28  * License along with this library; if not, write to the Free Software
29  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
30  */
31
32 #if !defined(_R_T_MODIFICATION_H_)
33 #define _R_T_MODIFICATION_H_
34
35 #include "Transformer.h"
36 #include "../Widgets/Widget.h"
37 #include "dyninstAPI/src/addressSpace.h"
38 #include <list>
39 #include <set>
40 #include <map>
41 #include "../DynInstrumenter.h"
42
43 class block_instance;
44 class instPoint;
45 class func_instance;
46
47 namespace Dyninst {
48 namespace Relocation {
49
50 class Modification : public Transformer {
51   public:
52     // Mimics typedefs in CodeMover.h, but I don't want
53     // to include that file.
54     typedef std::list<RelocBlock *> RelocBlockList;
55     //typedef std::map<Address, RelocBlockList> RelocBlockMap;
56
57     // Block (IDing a call site) -> func
58     typedef Dyninst::PatchAPI::CallModMap CallModMap;
59     // func -> func
60     typedef Dyninst::PatchAPI::FuncModMap FuncModMap;
61     typedef Dyninst::PatchAPI::FuncWrapMap FuncWrapMap;
62
63     virtual bool process(RelocBlock *cur, RelocGraph *);
64
65     Modification(const CallModMap &callRepl,
66                  const FuncModMap &funcRepl,
67                  const FuncWrapMap &funcWrap);
68
69     virtual ~Modification() {};
70
71   private:
72
73     bool replaceCall(RelocBlock *trace, RelocGraph *);
74     bool replaceFunction(RelocBlock *trace, RelocGraph *);
75     bool wrapFunction(RelocBlock *trace, RelocGraph *);
76
77     RelocBlock *makeRelocBlock(block_instance *block, func_instance *func, RelocGraph *cfg);
78
79     const CallModMap &callMods_;
80     const FuncModMap &funcReps_;
81     const FuncWrapMap &funcWraps_;
82
83     struct WrapperPredicate {
84        WrapperPredicate(func_instance *f);
85        func_instance *f_;
86        bool operator()(RelocEdge *e);
87     };
88
89     struct WrapperPatch : public Patch {
90       WrapperPatch(func_instance *func, std::string name) : func_(func), name_(name) {};
91        virtual bool apply(codeGen &gen, CodeBuffer *buf);
92        virtual unsigned estimate(codeGen &) { return 0; }
93        virtual ~WrapperPatch() {};
94
95        func_instance *func_;
96        std::string name_;
97     };
98     
99   };
100    
101 };
102 };
103
104 #endif