After tested under windows.
[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
62     virtual bool process(RelocBlock *cur, RelocGraph *);
63
64     Modification(const CallModMap &callRepl,
65                  const FuncModMap &funcRepl,
66                  const FuncModMap &funcWrap);
67
68     virtual ~Modification() {};
69
70   private:
71
72     bool replaceCall(RelocBlock *trace, RelocGraph *);
73     bool replaceFunction(RelocBlock *trace, RelocGraph *);
74     bool wrapFunction(RelocBlock *trace, RelocGraph *);
75
76     RelocBlock *makeRelocBlock(block_instance *block, func_instance *func, RelocGraph *cfg);
77
78     const CallModMap &callMods_;
79     const FuncModMap &funcReps_;
80     const FuncModMap &funcWraps_;
81
82     struct WrapperPredicate {
83        WrapperPredicate(func_instance *f);
84        func_instance *f_;
85        bool operator()(RelocEdge *e);
86     };
87
88     struct WrapperPatch : public Patch {
89       WrapperPatch(func_instance *func) : func_(func) {};
90        virtual bool apply(codeGen &gen, CodeBuffer *buf);
91        virtual unsigned estimate(codeGen &) { return 0; }
92        virtual ~WrapperPatch() {};
93
94        func_instance *func_;
95     };
96     
97   };
98    
99 };
100 };
101
102 #endif