Fix PCWdiget implementation on ARMv8
[dyninst.git] / dyninstAPI / src / Relocation / Widgets / PCWidget-aarch64.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     // We want to move a value into LR (x30). 
53     // On ARM, LR is just a normal GPR. We can directly move the value into it
54     codeGen gen(16);
55     gen.applyTemplate(templ);
56     Address origRet = addr() + insn_.size();
57     insnCodeGen::loadImmIntoReg(gen, 30 /* LR */, origRet);
58     buffer.addPIC(gen, tracker(t));
59   }
60   else {
61     IPPatch *newPatch = new IPPatch(IPPatch::Push, addr_, insn_, t->block(), t->func());
62     buffer.addPatch(newPatch, tracker(t));
63   }     
64    
65   return true;
66 }
67
68 bool PCWidget::PCtoReg(const codeGen &templ, const RelocBlock *t, CodeBuffer &buffer) {
69   bool ignored;
70   Register reg = convertRegID(a_.reg(), ignored);
71
72   if(templ.addrSpace()->proc()) {
73     // Move immediate to register?
74     codeGen gen(16);
75     insnCodeGen::loadImmIntoReg(gen, reg, addr_);
76     buffer.addPIC(gen, tracker(t));
77   }
78   else {
79     IPPatch *newPatch = new IPPatch(IPPatch::Reg, addr_, reg, thunkAddr_, insn_, t->block(), t->func());
80     buffer.addPatch(newPatch, tracker(t));
81   }
82   return true;
83 }
84
85 #include "dyninstAPI/src/registerSpace.h"
86 #include "dyninstAPI/src/emit-aarch64.h"
87 bool IPPatch::apply(codeGen &gen, CodeBuffer *) {
88   relocation_cerr << "\t\t IPPatch::apply" << endl;
89   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;
90
91   // We want to generate addr (as modified) into the appropriate location.
92   // TODO get rid of the #ifdef here...
93
94   instPoint *point = gen.point();
95   // If we do not have a point then we have to invent one
96   if (!point || (point->type() != instPoint::PreInsn && point->insnAddr() != addr)) {
97     point = instPoint::preInsn(func, block, addr, insn, true);
98   }
99   assert(point);
100     
101   registerSpace *rs = registerSpace::actualRegSpace(point);
102   gen.setRegisterSpace(rs);
103
104   // Calculate the offset between current PC and original RA
105   EmitterAARCH64* emitter = static_cast<EmitterAARCH64*>(gen.emitter());
106   Address RAOffset = addr - emitter->emitMovePCToReg( 30 /* LR */ , gen) + 4;
107   // Load the offset into a scratch register
108   pdvector<Register> exclude;
109   exclude.push_back(30 /* LR */);
110   Register scratchReg = insnCodeGen::moveValueToReg(gen, RAOffset, &exclude);
111   // Put the original RA into LR
112   insnCodeGen::generateAddSubShifted(gen, insnCodeGen::Add, 0, 0, scratchReg, 30, 30, true);
113   // Do a jump to the actual target (so do not overwrite LR)
114   insnCodeGen::generateBranch(gen, gen.currAddr(), addr, false);
115   return true;
116 }