forked from OSchip/llvm-project
311 lines
9.7 KiB
C++
311 lines
9.7 KiB
C++
//===-- AlphaJITInfo.cpp - Implement the JIT interfaces for the Alpha ---===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the JIT interfaces for the Alpha target.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#define DEBUG_TYPE "jit"
|
|
#include "AlphaJITInfo.h"
|
|
#include "AlphaRelocations.h"
|
|
#include "llvm/Function.h"
|
|
#include "llvm/CodeGen/JITCodeEmitter.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <cstdlib>
|
|
using namespace llvm;
|
|
|
|
#define BUILD_OFormatI(Op, RA, LIT, FUN, RC) \
|
|
((Op << 26) | (RA << 21) | (LIT << 13) | (1 << 12) | (FUN << 5) | (RC))
|
|
#define BUILD_OFormat(Op, RA, RB, FUN, RC) \
|
|
((Op << 26) | (RA << 21) | (RB << 16) | (FUN << 5) | (RC))
|
|
|
|
#define BUILD_LDA(RD, RS, IMM16) \
|
|
((0x08 << 26) | ((RD) << 21) | ((RS) << 16) | ((IMM16) & 65535))
|
|
#define BUILD_LDAH(RD, RS, IMM16) \
|
|
((0x09 << 26) | ((RD) << 21) | ((RS) << 16) | ((IMM16) & 65535))
|
|
|
|
#define BUILD_LDQ(RD, RS, IMM16) \
|
|
((0x29 << 26) | ((RD) << 21) | ((RS) << 16) | ((IMM16) & 0xFFFF))
|
|
|
|
#define BUILD_JMP(RD, RS, IMM16) \
|
|
((0x1A << 26) | ((RD) << 21) | ((RS) << 16) | (0x00 << 14) | ((IMM16) & 0x3FFF))
|
|
#define BUILD_JSR(RD, RS, IMM16) \
|
|
((0x1A << 26) | ((RD) << 21) | ((RS) << 16) | (0x01 << 14) | ((IMM16) & 0x3FFF))
|
|
|
|
#define BUILD_SLLi(RD, RS, IMM8) \
|
|
(BUILD_OFormatI(0x12, RS, IMM8, 0x39, RD))
|
|
|
|
#define BUILD_ORi(RD, RS, IMM8) \
|
|
(BUILD_OFormatI(0x11, RS, IMM8, 0x20, RD))
|
|
|
|
#define BUILD_OR(RD, RS, RT) \
|
|
(BUILD_OFormat(0x11, RS, RT, 0x20, RD))
|
|
|
|
|
|
|
|
static void EmitBranchToAt(void *At, void *To) {
|
|
unsigned long Fn = (unsigned long)To;
|
|
|
|
unsigned *AtI = (unsigned*)At;
|
|
|
|
AtI[0] = BUILD_OR(0, 27, 27);
|
|
|
|
DEBUG(errs() << "Stub targeting " << To << "\n");
|
|
|
|
for (int x = 1; x <= 8; ++x) {
|
|
AtI[2*x - 1] = BUILD_SLLi(27,27,8);
|
|
unsigned d = (Fn >> (64 - 8 * x)) & 0x00FF;
|
|
//DEBUG(errs() << "outputing " << hex << d << dec << "\n");
|
|
AtI[2*x] = BUILD_ORi(27, 27, d);
|
|
}
|
|
AtI[17] = BUILD_JMP(31,27,0); //jump, preserving ra, and setting pv
|
|
AtI[18] = 0x00FFFFFF; //mark this as a stub
|
|
}
|
|
|
|
void AlphaJITInfo::replaceMachineCodeForFunction(void *Old, void *New) {
|
|
//FIXME
|
|
llvm_unreachable(0);
|
|
}
|
|
|
|
static TargetJITInfo::JITCompilerFn JITCompilerFunction;
|
|
//static AlphaJITInfo* AlphaJTI;
|
|
|
|
extern "C" {
|
|
#ifdef __alpha
|
|
|
|
void AlphaCompilationCallbackC(long* oldpv, void* CameFromStub)
|
|
{
|
|
void* Target = JITCompilerFunction(CameFromStub);
|
|
|
|
//rewrite the stub to an unconditional branch
|
|
if (((unsigned*)CameFromStub)[18] == 0x00FFFFFF) {
|
|
DEBUG(errs() << "Came from a stub, rewriting\n");
|
|
EmitBranchToAt(CameFromStub, Target);
|
|
} else {
|
|
DEBUG(errs() << "confused, didn't come from stub at " << CameFromStub
|
|
<< " old jump vector " << oldpv
|
|
<< " new jump vector " << Target << "\n");
|
|
}
|
|
|
|
//Change pv to new Target
|
|
*oldpv = (long)Target;
|
|
}
|
|
|
|
void AlphaCompilationCallback(void);
|
|
|
|
asm(
|
|
".text\n"
|
|
".globl AlphaComilationCallbackC\n"
|
|
".align 4\n"
|
|
".globl AlphaCompilationCallback\n"
|
|
".ent AlphaCompilationCallback\n"
|
|
"AlphaCompilationCallback:\n"
|
|
// //get JIT's GOT
|
|
"ldgp $29, 0($27)\n"
|
|
//Save args, callee saved, and perhaps others?
|
|
//args: $16-$21 $f16-$f21 (12)
|
|
//callee: $9-$14 $f2-$f9 (14)
|
|
//others: fp:$15 ra:$26 pv:$27 (3)
|
|
"lda $30, -232($30)\n"
|
|
"stq $16, 0($30)\n"
|
|
"stq $17, 8($30)\n"
|
|
"stq $18, 16($30)\n"
|
|
"stq $19, 24($30)\n"
|
|
"stq $20, 32($30)\n"
|
|
"stq $21, 40($30)\n"
|
|
"stt $f16, 48($30)\n"
|
|
"stt $f17, 56($30)\n"
|
|
"stt $f18, 64($30)\n"
|
|
"stt $f19, 72($30)\n"
|
|
"stt $f20, 80($30)\n"
|
|
"stt $f21, 88($30)\n"
|
|
"stq $9, 96($30)\n"
|
|
"stq $10, 104($30)\n"
|
|
"stq $11, 112($30)\n"
|
|
"stq $12, 120($30)\n"
|
|
"stq $13, 128($30)\n"
|
|
"stq $14, 136($30)\n"
|
|
"stt $f2, 144($30)\n"
|
|
"stt $f3, 152($30)\n"
|
|
"stt $f4, 160($30)\n"
|
|
"stt $f5, 168($30)\n"
|
|
"stt $f6, 176($30)\n"
|
|
"stt $f7, 184($30)\n"
|
|
"stt $f8, 192($30)\n"
|
|
"stt $f9, 200($30)\n"
|
|
"stq $15, 208($30)\n"
|
|
"stq $26, 216($30)\n"
|
|
"stq $27, 224($30)\n"
|
|
|
|
"addq $30, 224, $16\n" //pass the addr of saved pv as the first arg
|
|
"bis $0, $0, $17\n" //pass the roughly stub addr in second arg
|
|
"jsr $26, AlphaCompilationCallbackC\n" //call without saving ra
|
|
|
|
"ldq $16, 0($30)\n"
|
|
"ldq $17, 8($30)\n"
|
|
"ldq $18, 16($30)\n"
|
|
"ldq $19, 24($30)\n"
|
|
"ldq $20, 32($30)\n"
|
|
"ldq $21, 40($30)\n"
|
|
"ldt $f16, 48($30)\n"
|
|
"ldt $f17, 56($30)\n"
|
|
"ldt $f18, 64($30)\n"
|
|
"ldt $f19, 72($30)\n"
|
|
"ldt $f20, 80($30)\n"
|
|
"ldt $f21, 88($30)\n"
|
|
"ldq $9, 96($30)\n"
|
|
"ldq $10, 104($30)\n"
|
|
"ldq $11, 112($30)\n"
|
|
"ldq $12, 120($30)\n"
|
|
"ldq $13, 128($30)\n"
|
|
"ldq $14, 136($30)\n"
|
|
"ldt $f2, 144($30)\n"
|
|
"ldt $f3, 152($30)\n"
|
|
"ldt $f4, 160($30)\n"
|
|
"ldt $f5, 168($30)\n"
|
|
"ldt $f6, 176($30)\n"
|
|
"ldt $f7, 184($30)\n"
|
|
"ldt $f8, 192($30)\n"
|
|
"ldt $f9, 200($30)\n"
|
|
"ldq $15, 208($30)\n"
|
|
"ldq $26, 216($30)\n"
|
|
"ldq $27, 224($30)\n" //this was updated in the callback with the target
|
|
|
|
"lda $30, 232($30)\n" //restore sp
|
|
"jmp $31, ($27)\n" //jump to the new function
|
|
".end AlphaCompilationCallback\n"
|
|
);
|
|
#else
|
|
void AlphaCompilationCallback() {
|
|
llvm_unreachable("Cannot call AlphaCompilationCallback() on a non-Alpha arch!");
|
|
}
|
|
#endif
|
|
}
|
|
|
|
TargetJITInfo::StubLayout AlphaJITInfo::getStubLayout() {
|
|
// The stub contains 19 4-byte instructions, aligned at 4 bytes:
|
|
// R0 = R27
|
|
// 8 x "R27 <<= 8; R27 |= 8-bits-of-Target" == 16 instructions
|
|
// JMP R27
|
|
// Magic number so the compilation callback can recognize the stub.
|
|
StubLayout Result = {19 * 4, 4};
|
|
return Result;
|
|
}
|
|
|
|
void *AlphaJITInfo::emitFunctionStub(const Function* F, void *Fn,
|
|
JITCodeEmitter &JCE) {
|
|
//assert(Fn == AlphaCompilationCallback && "Where are you going?\n");
|
|
//Do things in a stupid slow way!
|
|
void* Addr = (void*)(intptr_t)JCE.getCurrentPCValue();
|
|
for (int x = 0; x < 19; ++ x)
|
|
JCE.emitWordLE(0);
|
|
EmitBranchToAt(Addr, Fn);
|
|
DEBUG(errs() << "Emitting Stub to " << Fn << " at [" << Addr << "]\n");
|
|
return Addr;
|
|
}
|
|
|
|
TargetJITInfo::LazyResolverFn
|
|
AlphaJITInfo::getLazyResolverFunction(JITCompilerFn F) {
|
|
JITCompilerFunction = F;
|
|
// setZerothGOTEntry((void*)AlphaCompilationCallback);
|
|
return AlphaCompilationCallback;
|
|
}
|
|
|
|
//These describe LDAx
|
|
static const int IMM_LOW = -32768;
|
|
static const int IMM_HIGH = 32767;
|
|
static const int IMM_MULT = 65536;
|
|
|
|
static long getUpper16(long l)
|
|
{
|
|
long y = l / IMM_MULT;
|
|
if (l % IMM_MULT > IMM_HIGH)
|
|
++y;
|
|
if (l % IMM_MULT < IMM_LOW)
|
|
--y;
|
|
assert((short)y == y && "displacement out of range");
|
|
return y;
|
|
}
|
|
|
|
static long getLower16(long l)
|
|
{
|
|
long h = getUpper16(l);
|
|
long y = l - h * IMM_MULT;
|
|
assert(y == (short)y && "Displacement out of range");
|
|
return y;
|
|
}
|
|
|
|
void AlphaJITInfo::relocate(void *Function, MachineRelocation *MR,
|
|
unsigned NumRelocs, unsigned char* GOTBase) {
|
|
for (unsigned i = 0; i != NumRelocs; ++i, ++MR) {
|
|
unsigned *RelocPos = (unsigned*)Function + MR->getMachineCodeOffset()/4;
|
|
long idx = 0;
|
|
bool doCommon = true;
|
|
switch ((Alpha::RelocationType)MR->getRelocationType()) {
|
|
default: llvm_unreachable("Unknown relocation type!");
|
|
case Alpha::reloc_literal:
|
|
//This is a LDQl
|
|
idx = MR->getGOTIndex();
|
|
DEBUG(errs() << "Literal relocation to slot " << idx);
|
|
idx = (idx - GOToffset) * 8;
|
|
DEBUG(errs() << " offset " << idx << "\n");
|
|
break;
|
|
case Alpha::reloc_gprellow:
|
|
idx = (unsigned char*)MR->getResultPointer() - &GOTBase[GOToffset * 8];
|
|
idx = getLower16(idx);
|
|
DEBUG(errs() << "gprellow relocation offset " << idx << "\n");
|
|
DEBUG(errs() << " Pointer is " << (void*)MR->getResultPointer()
|
|
<< " GOT is " << (void*)&GOTBase[GOToffset * 8] << "\n");
|
|
break;
|
|
case Alpha::reloc_gprelhigh:
|
|
idx = (unsigned char*)MR->getResultPointer() - &GOTBase[GOToffset * 8];
|
|
idx = getUpper16(idx);
|
|
DEBUG(errs() << "gprelhigh relocation offset " << idx << "\n");
|
|
DEBUG(errs() << " Pointer is " << (void*)MR->getResultPointer()
|
|
<< " GOT is " << (void*)&GOTBase[GOToffset * 8] << "\n");
|
|
break;
|
|
case Alpha::reloc_gpdist:
|
|
switch (*RelocPos >> 26) {
|
|
case 0x09: //LDAH
|
|
idx = &GOTBase[GOToffset * 8] - (unsigned char*)RelocPos;
|
|
idx = getUpper16(idx);
|
|
DEBUG(errs() << "LDAH: " << idx << "\n");
|
|
//add the relocation to the map
|
|
gpdistmap[std::make_pair(Function, MR->getConstantVal())] = RelocPos;
|
|
break;
|
|
case 0x08: //LDA
|
|
assert(gpdistmap[std::make_pair(Function, MR->getConstantVal())] &&
|
|
"LDAg without seeing LDAHg");
|
|
idx = &GOTBase[GOToffset * 8] -
|
|
(unsigned char*)gpdistmap[std::make_pair(Function, MR->getConstantVal())];
|
|
idx = getLower16(idx);
|
|
DEBUG(errs() << "LDA: " << idx << "\n");
|
|
break;
|
|
default:
|
|
llvm_unreachable("Cannot handle gpdist yet");
|
|
}
|
|
break;
|
|
case Alpha::reloc_bsr: {
|
|
idx = (((unsigned char*)MR->getResultPointer() -
|
|
(unsigned char*)RelocPos) >> 2) + 1; //skip first 2 inst of fun
|
|
*RelocPos |= (idx & ((1 << 21)-1));
|
|
doCommon = false;
|
|
break;
|
|
}
|
|
}
|
|
if (doCommon) {
|
|
short x = (short)idx;
|
|
assert(x == idx);
|
|
*(short*)RelocPos = x;
|
|
}
|
|
}
|
|
}
|