Move architecture dependent code in PCWidget.C to three separate architecture specifi...
[dyninst.git] / dyninstAPI / src / Relocation / Widgets / PCWidget-x86.C
1 /*
2  * See the dyninst/COPYRIGHT file for copyright information.
3  * 
4  * We provide the Paradyn Tools (below described as "Paradyn")
5  * on an AS IS basis, and do not warrant its validity or performance.
6  * We reserve the right to update, modify, or discontinue this
7  * software at any time.  We shall have no obligation to supply such
8  * updates or modifications or any other form of support to you.
9  * 
10  * By your use of Paradyn, you understand and agree that we (or any
11  * other person or entity with proprietary rights in Paradyn) are
12  * under no obligation to provide either maintenance services,
13  * update services, notices of latent defects, or correction of
14  * defects for Paradyn.
15  * 
16  * This library is free software; you can redistribute it and/or
17  * modify it under the terms of the GNU Lesser General Public
18  * License as published by the Free Software Foundation; either
19  * version 2.1 of the License, or (at your option) any later version.
20  * 
21  * This library is distributed in the hope that it will be useful,
22  * but WITHOUT ANY WARRANTY; without even the implied warranty of
23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
24  * Lesser General Public License for more details.
25  * 
26  * You should have received a copy of the GNU Lesser General Public
27  * License along with this library; if not, write to the Free Software
28  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
29  */
30
31 #include "PCWidget.h"
32 #include "instructionAPI/h/Instruction.h"
33 #include "../dyninstAPI/src/debug.h"
34 #include "../CFG/RelocBlock.h"
35 #include "../CodeBuffer.h"
36 #include "../CodeTracker.h"
37 #include "dyninstAPI/src/function.h"
38
39 #include "dyninstAPI/src/addressSpace.h" // For determining which type of getPC to emit
40 #include "dyninstAPI/src/RegisterConversion.h"
41 #include "dyninstAPI/src/registerSpace.h"
42
43 #include "dyninstAPI/src/emitter.h"
44
45 using namespace Dyninst;
46 using namespace Relocation;
47 using namespace InstructionAPI;
48
49 bool PCWidget::PCtoReturnAddr(const codeGen &templ, const RelocBlock *t, CodeBuffer &buffer) {
50   if(templ.addrSpace()->proc()) {
51     std::vector<unsigned char> newInsn;
52     if (templ.getArch() == Arch_x86_64) {
53         codeGen gen(16);
54         Address RIP = addr_ + insn_.size();
55         insnCodeGen::generatePush64(gen, RIP);
56         buffer.addPIC(gen, tracker(t));
57     } else if (templ.getArch() == Arch_x86) {
58         newInsn.push_back(0x68); // push
59         Address EIP = addr_ + insn_.size();
60         unsigned char *tmp = (unsigned char *) &EIP;
61         newInsn.insert(newInsn.end(), tmp, tmp+sizeof(unsigned int));
62         buffer.addPIC(newInsn, tracker(t));
63     }  else {
64         assert(!"Unimplemented architecture!");
65     }
66   }
67   else {
68     IPPatch *newPatch = new IPPatch(IPPatch::Push, addr_, insn_, t->block(), t->func());
69     buffer.addPatch(newPatch, tracker(t));
70   }     
71    
72   return true;
73 }
74
75 bool PCWidget::PCtoReg(const codeGen &templ, const RelocBlock *t, CodeBuffer &buffer) {
76   bool ignored;
77   Register reg = convertRegID(a_.reg(), ignored);
78
79   if(templ.addrSpace()->proc()) {
80     std::vector<unsigned char> newInsn;
81     newInsn.push_back(static_cast<unsigned char>(0xb8 + reg));
82     // MOV family, destination of the register encoded by
83     // 'reg', source is an Iv immediate
84      
85     Address EIP = addr_ + insn_.size();
86     unsigned char *tmp = (unsigned char *) &EIP;
87     newInsn.insert(newInsn.end(), tmp, tmp + sizeof(unsigned int));
88     buffer.addPIC(newInsn, tracker(t));          
89   }
90   else {
91     IPPatch *newPatch = new IPPatch(IPPatch::Reg, addr_, reg, thunkAddr_, insn_, t->block(), t->func());
92     buffer.addPatch(newPatch, tracker(t));
93   }
94   return true;
95 }
96
97 #include "dyninstAPI/src/registerSpace.h"
98 #include "dyninstAPI/src/inst-x86.h"
99
100 bool IPPatch::apply(codeGen &gen, CodeBuffer *) {
101   relocation_cerr << "\t\t IPPatch::apply" << endl;
102   relocation_cerr << "\t\t\t Generating IPPatch for target address " << std::hex << addr << ", CodeGen current address " << std::hex << gen.currAddr() << " and register number " << reg << endl;
103
104   // We want to generate addr (as modified) into the appropriate location.
105   // TODO get rid of the #ifdef here...
106
107
108   // Emit a call to the next instruction to get current PC
109   // This is necessary for PIC code
110   GET_PTR(newInsn, gen); 
111   *newInsn = 0xE8;
112   newInsn++;
113   unsigned int *temp = (uint32_t *) newInsn;
114   *temp = 0;
115   newInsn += sizeof(uint32_t);
116   SET_PTR(newInsn, gen);
117   // Compensating PC on stack to the original location
118   Address offset = addr - gen.currAddr() + insn.size();
119   REGET_PTR(newInsn, gen);
120   *newInsn = 0x81;
121   newInsn++;
122   *newInsn = 0x04;
123   newInsn++;
124   *newInsn = 0x24;
125   newInsn++;
126   temp =  (uint32_t *) newInsn;
127   *temp = offset;
128   newInsn += sizeof(uint32_t);    
129
130   if (type == Reg) {
131     assert(reg != (Register) -1);
132     // pop...
133     *newInsn++ = static_cast<unsigned char>(0x58 + reg); // POP family
134   }
135   SET_PTR(newInsn, gen);
136   return true;
137 }
138