* If a global is not a function, just ask the MachineCodeEmitter for the addr
[oota-llvm.git] / lib / Target / SparcV9 / SparcV9CodeEmitter.cpp
1 //===-- SparcV9CodeEmitter.cpp -  --------===//
2 //
3 //
4 //===----------------------------------------------------------------------===//
5
6 #include "llvm/Constants.h"
7 #include "llvm/Function.h"
8 #include "llvm/GlobalVariable.h"
9 #include "llvm/PassManager.h"
10 #include "llvm/CodeGen/MachineCodeEmitter.h"
11 #include "llvm/CodeGen/MachineConstantPool.h"
12 #include "llvm/CodeGen/MachineFunctionInfo.h"
13 #include "llvm/CodeGen/MachineFunctionPass.h"
14 #include "llvm/CodeGen/MachineInstr.h"
15 #include "llvm/Target/TargetMachine.h"
16 #include "llvm/Target/TargetData.h"
17 #include "Support/Statistic.h"
18 #include "Support/hash_set"
19 #include "SparcInternals.h"
20 #include "SparcV9CodeEmitter.h"
21
22 bool UltraSparc::addPassesToEmitMachineCode(PassManager &PM,
23                                             MachineCodeEmitter &MCE) {
24   MachineCodeEmitter *M = &MCE;
25   DEBUG(MachineCodeEmitter::createFilePrinterEmitter(MCE));
26   PM.add(new SparcV9CodeEmitter(*this, *M));
27   PM.add(createMachineCodeDestructionPass()); // Free stuff no longer needed
28   return false;
29 }
30
31 namespace {
32   class JITResolver {
33     SparcV9CodeEmitter &SparcV9;
34     MachineCodeEmitter &MCE;
35
36     // LazyCodeGenMap - Keep track of call sites for functions that are to be
37     // lazily resolved.
38     std::map<uint64_t, Function*> LazyCodeGenMap;
39
40     // LazyResolverMap - Keep track of the lazy resolver created for a
41     // particular function so that we can reuse them if necessary.
42     std::map<Function*, uint64_t> LazyResolverMap;
43   public:
44     JITResolver(SparcV9CodeEmitter &V9,
45                 MachineCodeEmitter &mce) : SparcV9(V9), MCE(mce) {}
46     uint64_t getLazyResolver(Function *F);
47     uint64_t addFunctionReference(uint64_t Address, Function *F);
48
49     // Utility functions for accessing data from static callback
50     uint64_t getCurrentPCValue() {
51       return MCE.getCurrentPCValue();
52     }
53     unsigned getBinaryCodeForInstr(MachineInstr &MI) {
54       return SparcV9.getBinaryCodeForInstr(MI);
55     }
56
57     inline uint64_t insertFarJumpAtAddr(int64_t Value, uint64_t Addr);
58
59   private:
60     uint64_t emitStubForFunction(Function *F);
61     static void CompilationCallback();
62     uint64_t resolveFunctionReference(uint64_t RetAddr);
63
64   };
65
66   JITResolver *TheJITResolver;
67 }
68
69 /// addFunctionReference - This method is called when we need to emit the
70 /// address of a function that has not yet been emitted, so we don't know the
71 /// address.  Instead, we emit a call to the CompilationCallback method, and
72 /// keep track of where we are.
73 ///
74 uint64_t JITResolver::addFunctionReference(uint64_t Address, Function *F) {
75   LazyCodeGenMap[Address] = F;  
76   return (intptr_t)&JITResolver::CompilationCallback;
77 }
78
79 uint64_t JITResolver::resolveFunctionReference(uint64_t RetAddr) {
80   std::map<uint64_t, Function*>::iterator I = LazyCodeGenMap.find(RetAddr);
81   assert(I != LazyCodeGenMap.end() && "Not in map!");
82   Function *F = I->second;
83   LazyCodeGenMap.erase(I);
84   return MCE.forceCompilationOf(F);
85 }
86
87 uint64_t JITResolver::getLazyResolver(Function *F) {
88   std::map<Function*, uint64_t>::iterator I = LazyResolverMap.lower_bound(F);
89   if (I != LazyResolverMap.end() && I->first == F) return I->second;
90   
91 //std::cerr << "Getting lazy resolver for : " << ((Value*)F)->getName() << "\n";
92
93   uint64_t Stub = emitStubForFunction(F);
94   LazyResolverMap.insert(I, std::make_pair(F, Stub));
95   return Stub;
96 }
97
98 uint64_t JITResolver::insertFarJumpAtAddr(int64_t Target, uint64_t Addr) {
99
100   static const unsigned i1 = SparcIntRegClass::i1, i2 = SparcIntRegClass::i2,
101     i7 = SparcIntRegClass::i7,
102     o6 = SparcIntRegClass::o6, g0 = SparcIntRegClass::g0;
103
104   // 
105   // Save %i1, %i2 to the stack so we can form a 64-bit constant in %i2
106   // 
107
108   // stx %i1, [%sp + 2119]       ;; save %i1 to the stack, used as temp
109   MachineInstr *STX = BuildMI(V9::STXi, 3).addReg(i1).addReg(o6).addSImm(2119);
110   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*STX);
111   delete STX;
112   Addr += 4;
113
114   // stx %i2, [%sp + 2127]       ;; save %i2 to the stack
115   STX = BuildMI(V9::STXi, 3).addReg(i2).addReg(o6).addSImm(2127);
116   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*STX);
117   delete STX;
118   Addr += 4;
119
120   //
121   // Get address to branch into %i2, using %i1 as a temporary
122   //
123
124   // sethi %uhi(Target), %i1   ;; get upper 22 bits of Target into %i1
125   MachineInstr *SH = BuildMI(V9::SETHI, 2).addSImm(Target >> 42).addReg(i1);
126   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*SH);
127   delete SH;
128   Addr += 4;
129
130   // or %i1, %ulo(Target), %i1 ;; get 10 lower bits of upper word into %1
131   MachineInstr *OR = BuildMI(V9::ORi, 3)
132     .addReg(i1).addSImm((Target >> 32) & 0x03ff).addReg(i1);
133   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*OR);
134   delete OR;
135   Addr += 4;
136
137   // sllx %i1, 32, %i1            ;; shift those 10 bits to the upper word
138   MachineInstr *SL = BuildMI(V9::SLLXi6, 3).addReg(i1).addSImm(32).addReg(i1);
139   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*SL);
140   delete SL;
141   Addr += 4;
142
143   // sethi %hi(Target), %i2    ;; extract bits 10-31 into the dest reg
144   SH = BuildMI(V9::SETHI, 2).addSImm((Target >> 10) & 0x03fffff).addReg(i2);
145   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*SH);
146   delete SH;
147   Addr += 4;
148
149   // or %i1, %i2, %i2             ;; get upper word (in %i1) into %i2
150   OR = BuildMI(V9::ORr, 3).addReg(i1).addReg(i2).addReg(i2);
151   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*OR);
152   delete OR;
153   Addr += 4;
154
155   // or %i2, %lo(Target), %i2  ;; get lowest 10 bits of Target into %i2
156   OR = BuildMI(V9::ORi, 3).addReg(i2).addSImm(Target & 0x03ff).addReg(i2);
157   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*OR);
158   delete OR;
159   Addr += 4;
160
161   // ldx [%sp + 2119], %i1       ;; restore %i1 -> 2119 = BIAS(2047) + 72
162   MachineInstr *LDX = BuildMI(V9::LDXi, 3).addReg(o6).addSImm(2119).addReg(i1);
163   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*LDX);
164   delete LDX;
165   Addr += 4;
166
167   // jmpl %i2, %g0, %g0          ;; indirect branch on %i2
168   MachineInstr *J = BuildMI(V9::JMPLRETr, 3).addReg(i2).addReg(g0).addReg(g0);
169   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*J);
170   delete J;
171   Addr += 4;
172
173   // ldx [%sp + 2127], %i2       ;; restore %i2 -> 2127 = BIAS(2047) + 80
174   LDX = BuildMI(V9::LDXi, 3).addReg(o6).addSImm(2127).addReg(i2);
175   *((unsigned*)(intptr_t)Addr) = getBinaryCodeForInstr(*LDX);
176   delete LDX;
177   Addr += 4;
178
179   return Addr;
180 }
181
182 void JITResolver::CompilationCallback() {
183   uint64_t CameFrom = (uint64_t)(intptr_t)__builtin_return_address(0);
184   int64_t Target = (int64_t)TheJITResolver->resolveFunctionReference(CameFrom);
185   DEBUG(std::cerr << "In callback! Addr=0x" << std::hex << CameFrom << "\n");
186
187   // Rewrite the call target... so that we don't fault every time we execute
188   // the call.
189 #if 0
190   int64_t RealCallTarget = (int64_t)
191     ((NewVal - TheJITResolver->getCurrentPCValue()) >> 4);
192   if (RealCallTarget >= (1<<22) || RealCallTarget <= -(1<<22)) {
193     std::cerr << "Address out of bounds for 22bit BA: " << RealCallTarget<<"\n";
194     abort();
195   }
196 #endif
197
198   //uint64_t CurrPC    = TheJITResolver->getCurrentPCValue();
199   // we will insert 9 instructions before we do the actual jump
200   //int64_t NewTarget  = (NewVal - 9*4 - InstAddr) >> 2;
201
202   static const unsigned i1 = SparcIntRegClass::i1, i2 = SparcIntRegClass::i2,
203     i7 = SparcIntRegClass::i7, o6 = SparcIntRegClass::o6,
204     o7 = SparcIntRegClass::o7, g0 = SparcIntRegClass::g0;
205
206   // Subtract 4 to overwrite the 'save' that's there now
207   uint64_t InstAddr = CameFrom-4;
208
209   InstAddr = TheJITResolver->insertFarJumpAtAddr(Target, InstAddr);
210
211   // CODE SHOULD NEVER GO PAST THIS LOAD!! The real function should return to
212   // the original caller, not here!!
213
214   // FIXME: add call 0 to make sure?!?
215
216   // =============== THE REAL STUB ENDS HERE =========================
217
218   // What follows below is one-time restore code, because this callback may be
219   // changing registers in unpredictible ways. However, since it is executed
220   // only once per function (after the function is resolved, the callback is no
221   // longer in the path), this has to be done only once.
222   //
223   // Thus, it is after the regular stub code. The call back returns to THIS
224   // point, but every other call to the target function will execute the code
225   // above. Hence, this code is one-time use.
226
227   uint64_t OneTimeRestore = InstAddr;
228
229   // restore %g0, 0, %g0
230   //MachineInstr *R = BuildMI(V9::RESTOREi, 3).addMReg(g0).addSImm(0)
231   //                                          .addMReg(g0, MOTy::Def);
232   //*((unsigned*)(intptr_t)InstAddr)=TheJITResolver->getBinaryCodeForInstr(*R);
233   //delete R;
234
235   // FIXME: BuildMI() above crashes. Encode the instruction directly.
236   // restore %g0, 0, %g0
237   *((unsigned*)(intptr_t)InstAddr) = 0x81e82000U;
238   InstAddr += 4;  
239
240   InstAddr = TheJITResolver->insertFarJumpAtAddr(Target, InstAddr);
241
242   // FIXME: if the target function is close enough to fit into the 19bit disp of
243   // BA, we should use this version, as its much cheaper to generate.
244   /*
245   MachineInstr *MI = BuildMI(V9::BA, 1).addSImm(RealCallTarget);
246   *((unsigned*)(intptr_t)InstAddr) = TheJITResolver->getBinaryCodeForInstr(*MI);
247   delete MI;
248   InstAddr += 4;
249
250   // Add another NOP
251   MachineInstr *Nop = BuildMI(V9::NOP, 0);
252   *((unsigned*)(intptr_t)InstAddr)=TheJITResolver->getBinaryCodeForInstr(*Nop);
253   delete Nop;
254   InstAddr += 4;
255
256   MachineInstr *BA = BuildMI(V9::BA, 1).addSImm(RealCallTarget-2);
257   *((unsigned*)(intptr_t)InstAddr) = TheJITResolver->getBinaryCodeForInstr(*BA);
258   delete BA;
259   */
260
261   // Change the return address to reexecute the call instruction...
262   // The return address is really %o7, but will disappear after this function
263   // returns, and the register windows are rotated away.
264 #if defined(sparc) || defined(__sparc__) || defined(__sparcv9)
265   __asm__ __volatile__ ("or %%g0, %0, %%i7" : : "r" (OneTimeRestore-8));
266 #endif
267 }
268
269 /// emitStubForFunction - This method is used by the JIT when it needs to emit
270 /// the address of a function for a function whose code has not yet been
271 /// generated.  In order to do this, it generates a stub which jumps to the lazy
272 /// function compiler, which will eventually get fixed to call the function
273 /// directly.
274 ///
275 uint64_t JITResolver::emitStubForFunction(Function *F) {
276   MCE.startFunctionStub(*F, 6);
277
278   DEBUG(std::cerr << "Emitting stub at addr: 0x" 
279                   << std::hex << MCE.getCurrentPCValue() << "\n");
280
281   unsigned o6 = SparcIntRegClass::o6;
282   // save %sp, -192, %sp
283   MachineInstr *SV = BuildMI(V9::SAVEi, 3).addReg(o6).addSImm(-192).addReg(o6);
284   SparcV9.emitWord(SparcV9.getBinaryCodeForInstr(*SV));
285   delete SV;
286
287   int64_t CurrPC = MCE.getCurrentPCValue();
288   int64_t Addr = (int64_t)addFunctionReference(CurrPC, F);
289
290   int64_t CallTarget = (Addr-CurrPC) >> 2;
291   if (CallTarget >= (1 << 30) || CallTarget <= -(1 << 30)) {
292     std::cerr << "Call target beyond 30 bit limit of CALL: " 
293               << CallTarget << "\n";
294     abort();
295   }
296   // call CallTarget              ;; invoke the callback
297   MachineInstr *Call = BuildMI(V9::CALL, 1).addSImm(CallTarget);
298   SparcV9.emitWord(SparcV9.getBinaryCodeForInstr(*Call));
299   delete Call;
300   
301   // nop                          ;; call delay slot
302   MachineInstr *Nop = BuildMI(V9::NOP, 0);
303   SparcV9.emitWord(SparcV9.getBinaryCodeForInstr(*Nop));
304   delete Nop;
305
306   SparcV9.emitWord(0xDEADBEEF); // marker so that we know it's really a stub
307   return (intptr_t)MCE.finishFunctionStub(*F);
308 }
309
310
311 SparcV9CodeEmitter::SparcV9CodeEmitter(TargetMachine &tm,
312                                        MachineCodeEmitter &M): TM(tm), MCE(M)
313 {
314   TheJITResolver = new JITResolver(*this, M);
315 }
316
317 SparcV9CodeEmitter::~SparcV9CodeEmitter() {
318   delete TheJITResolver;
319 }
320
321 void SparcV9CodeEmitter::emitWord(unsigned Val) {
322   // Output the constant in big endian byte order...
323   unsigned byteVal;
324   for (int i = 3; i >= 0; --i) {
325     byteVal = Val >> 8*i;
326     MCE.emitByte(byteVal & 255);
327   }
328 }
329
330 bool SparcV9CodeEmitter::isFPInstr(MachineInstr &MI) {
331   for (unsigned i = 0, e = MI.getNumOperands(); i < e; ++i) {
332     const MachineOperand &MO = MI.getOperand(i);
333     if (MO.isPhysicalRegister()) {
334       unsigned fakeReg = MO.getReg(), realReg, regClass, regType;
335       regType = TM.getRegInfo().getRegType(fakeReg);
336       // At least map fakeReg into its class
337       fakeReg = TM.getRegInfo().getClassRegNum(fakeReg, regClass);
338       if (regClass == UltraSparcRegInfo::FPSingleRegType ||
339           regClass == UltraSparcRegInfo::FPDoubleRegType)
340         return true;
341     }
342   }
343   return false;
344 }
345
346 unsigned 
347 SparcV9CodeEmitter::getRealRegNum(unsigned fakeReg, unsigned regClass,
348                                   MachineInstr &MI) {
349   switch (regClass) {
350   case UltraSparcRegInfo::IntRegType: {
351     // Sparc manual, p31
352     static const unsigned IntRegMap[] = {
353       // "o0", "o1", "o2", "o3", "o4", "o5",       "o7",
354       8, 9, 10, 11, 12, 13, 15,
355       // "l0", "l1", "l2", "l3", "l4", "l5", "l6", "l7",
356       16, 17, 18, 19, 20, 21, 22, 23,
357       // "i0", "i1", "i2", "i3", "i4", "i5",  
358       24, 25, 26, 27, 28, 29,
359       // "i6", "i7",
360       30, 31,
361       // "g0", "g1", "g2", "g3", "g4", "g5",  "g6", "g7", 
362       0, 1, 2, 3, 4, 5, 6, 7,
363       // "o6"
364       14
365     }; 
366  
367     return IntRegMap[fakeReg];
368     break;
369   }
370   case UltraSparcRegInfo::FPSingleRegType: {
371     return fakeReg;
372   }
373   case UltraSparcRegInfo::FPDoubleRegType: {
374     return fakeReg;
375   }
376   case UltraSparcRegInfo::FloatCCRegType: {
377     /* These are laid out %fcc0 - %fcc3 => 0 - 3, so are correct */
378     return fakeReg;
379
380   }
381   case UltraSparcRegInfo::IntCCRegType: {
382     static const unsigned FPInstrIntCCReg[]  = { 6 /* xcc */, 4  /* icc */ };
383     static const unsigned IntInstrIntCCReg[] = { 2 /* xcc */, 0  /* icc */ };
384     
385     if (isFPInstr(MI)) {
386       assert(fakeReg < sizeof(FPInstrIntCCReg)/sizeof(FPInstrIntCCReg[0])
387              && "Int CC register out of bounds for FPInstr IntCCReg map");      
388       return FPInstrIntCCReg[fakeReg];
389     } else {
390       assert(fakeReg < sizeof(IntInstrIntCCReg)/sizeof(IntInstrIntCCReg[0])
391              && "Int CC register out of bounds for IntInstr IntCCReg map");
392       return IntInstrIntCCReg[fakeReg];
393     }
394   }
395   default:
396     assert(0 && "Invalid unified register number in getRegType");
397     return fakeReg;
398   }
399 }
400
401 int64_t SparcV9CodeEmitter::getMachineOpValue(MachineInstr &MI,
402                                               MachineOperand &MO) {
403   int64_t rv = 0; // Return value; defaults to 0 for unhandled cases
404                   // or things that get fixed up later by the JIT.
405
406   if (MO.isVirtualRegister()) {
407     std::cerr << "ERROR: virtual register found in machine code.\n";
408     abort();
409   } else if (MO.isPCRelativeDisp()) {
410     DEBUG(std::cerr << "PCRelativeDisp: ");
411     Value *V = MO.getVRegValue();
412     if (BasicBlock *BB = dyn_cast<BasicBlock>(V)) {
413       DEBUG(std::cerr << "Saving reference to BB (VReg)\n");
414       unsigned* CurrPC = (unsigned*)(intptr_t)MCE.getCurrentPCValue();
415       BBRefs.push_back(std::make_pair(BB, std::make_pair(CurrPC, &MI)));
416     } else if (const Constant *C = dyn_cast<Constant>(V)) {
417       if (ConstantMap.find(C) != ConstantMap.end()) {
418         rv = (int64_t)MCE.getConstantPoolEntryAddress(ConstantMap[C]);
419         DEBUG(std::cerr << "const: 0x" << std::hex << rv << "\n");
420       } else {
421         std::cerr << "ERROR: constant not in map:" << MO << "\n";
422         abort();
423       }
424     } else if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) {
425       // same as MO.isGlobalAddress()
426       DEBUG(std::cerr << "GlobalValue: ");
427       // external function calls, etc.?
428       if (Function *F = dyn_cast<Function>(GV)) {
429         DEBUG(std::cerr << "Function: ");
430         if (F->isExternal()) {
431           // Sparc backend broken: this MO should be `ExternalSymbol'
432           rv = (int64_t)MCE.getGlobalValueAddress(F->getName());
433         } else {
434           rv = (int64_t)MCE.getGlobalValueAddress(F);
435         }
436         if (rv == 0) {
437           DEBUG(std::cerr << "not yet generated\n");
438           // Function has not yet been code generated!
439           TheJITResolver->addFunctionReference(MCE.getCurrentPCValue(), F);
440           // Delayed resolution...
441           rv = TheJITResolver->getLazyResolver(F);
442         } else {
443           DEBUG(std::cerr << "already generated: 0x" << std::hex << rv << "\n");
444         }
445       } else {
446         DEBUG(std::cerr << "not a function: " << *GV << "\n");
447         rv = (int64_t)MCE.getGlobalValueAddress(GV);
448       }
449       // The real target of the call is Addr = PC + (rv * 4)
450       // So undo that: give the instruction (Addr - PC) / 4
451       if (MI.getOpcode() == V9::CALL) {
452         int64_t CurrPC = MCE.getCurrentPCValue();
453         DEBUG(std::cerr << "rv addr: 0x" << std::hex << rv << "\n"
454                         << "curr PC: 0x" << CurrPC << "\n");
455         rv = (rv - CurrPC) >> 2;
456         if (rv >= (1<<29) || rv <= -(1<<29)) {
457           std::cerr << "addr out of bounds for the 30-bit call: " << rv << "\n";
458           abort();
459         }
460         DEBUG(std::cerr << "returning addr: 0x" << rv << "\n");
461       }
462     } else {
463       std::cerr << "ERROR: PC relative disp unhandled:" << MO << "\n";
464       abort();
465     }
466   } else if (MO.isPhysicalRegister() ||
467              MO.getType() == MachineOperand::MO_CCRegister)
468   {
469     // This is necessary because the Sparc doesn't actually lay out registers
470     // in the real fashion -- it skips those that it chooses not to allocate,
471     // i.e. those that are the SP, etc.
472     unsigned fakeReg = MO.getReg(), realReg, regClass, regType;
473     regType = TM.getRegInfo().getRegType(fakeReg);
474     // At least map fakeReg into its class
475     fakeReg = TM.getRegInfo().getClassRegNum(fakeReg, regClass);
476     // Find the real register number for use in an instruction
477     /////realReg = getRealRegNum(fakeReg, regClass, MI);
478     realReg = getRealRegNum(fakeReg, regType, MI);
479     DEBUG(std::cerr << MO << ": Reg[" << std::dec << fakeReg << "] = "
480                     << realReg << "\n");
481     rv = realReg;
482   } else if (MO.isImmediate()) {
483     rv = MO.getImmedValue();
484     DEBUG(std::cerr << "immed: " << rv << "\n");
485   } else if (MO.isGlobalAddress()) {
486     DEBUG(std::cerr << "GlobalAddress: not PC-relative\n");
487     rv = (int64_t)
488       (intptr_t)getGlobalAddress(cast<GlobalValue>(MO.getVRegValue()),
489                                  MI, MO.isPCRelative());
490   } else if (MO.isMachineBasicBlock()) {
491     // Duplicate code of the above case for VirtualRegister, BasicBlock... 
492     // It should really hit this case, but Sparc backend uses VRegs instead
493     DEBUG(std::cerr << "Saving reference to MBB\n");
494     BasicBlock *BB = MO.getMachineBasicBlock()->getBasicBlock();
495     unsigned* CurrPC = (unsigned*)(intptr_t)MCE.getCurrentPCValue();
496     BBRefs.push_back(std::make_pair(BB, std::make_pair(CurrPC, &MI)));
497   } else if (MO.isExternalSymbol()) {
498     // Sparc backend doesn't generate this (yet...)
499     std::cerr << "ERROR: External symbol unhandled: " << MO << "\n";
500     abort();
501   } else if (MO.isFrameIndex()) {
502     // Sparc backend doesn't generate this (yet...)
503     int FrameIndex = MO.getFrameIndex();
504     std::cerr << "ERROR: Frame index unhandled.\n";
505     abort();
506   } else if (MO.isConstantPoolIndex()) {
507     // Sparc backend doesn't generate this (yet...)
508     std::cerr << "ERROR: Constant Pool index unhandled.\n";
509     abort();
510   } else {
511     std::cerr << "ERROR: Unknown type of MachineOperand: " << MO << "\n";
512     abort();
513   }
514
515   // Finally, deal with the various bitfield-extracting functions that
516   // are used in SPARC assembly. (Some of these make no sense in combination
517   // with some of the above; we'll trust that the instruction selector
518   // will not produce nonsense, and not check for valid combinations here.)
519   if (MO.opLoBits32()) {          // %lo(val) == %lo() in Sparc ABI doc
520     return rv & 0x03ff;
521   } else if (MO.opHiBits32()) {   // %lm(val) == %hi() in Sparc ABI doc
522     return (rv >> 10) & 0x03fffff;
523   } else if (MO.opLoBits64()) {   // %hm(val) == %ulo() in Sparc ABI doc
524     return (rv >> 32) & 0x03ff;
525   } else if (MO.opHiBits64()) {   // %hh(val) == %uhi() in Sparc ABI doc
526     return rv >> 42;
527   } else {                        // (unadorned) val
528     return rv;
529   }
530 }
531
532 unsigned SparcV9CodeEmitter::getValueBit(int64_t Val, unsigned bit) {
533   Val >>= bit;
534   return (Val & 1);
535 }
536
537 bool SparcV9CodeEmitter::runOnMachineFunction(MachineFunction &MF) {
538   MCE.startFunction(MF);
539   DEBUG(std::cerr << "Starting function " << MF.getFunction()->getName()
540             << ", address: " << "0x" << std::hex 
541             << (long)MCE.getCurrentPCValue() << "\n");
542
543   // The Sparc backend does not use MachineConstantPool;
544   // instead, it has its own constant pool implementation.
545   // We create a new MachineConstantPool here to be compatible with the emitter.
546   MachineConstantPool MCP;
547   const hash_set<const Constant*> &pool = MF.getInfo()->getConstantPoolValues();
548   for (hash_set<const Constant*>::const_iterator I = pool.begin(),
549          E = pool.end();  I != E; ++I)
550   {
551     Constant *C = (Constant*)*I;
552     unsigned idx = MCP.getConstantPoolIndex(C);
553     DEBUG(std::cerr << "Mapping constant 0x" << (intptr_t)C << " to " 
554                     << idx << "\n");
555     ConstantMap[C] = idx;
556   }  
557   MCE.emitConstantPool(&MCP);
558
559   for (MachineFunction::iterator I = MF.begin(), E = MF.end(); I != E; ++I)
560     emitBasicBlock(*I);
561   MCE.finishFunction(MF);
562
563   DEBUG(std::cerr << "Finishing function " << MF.getFunction()->getName() 
564                   << "\n");
565   ConstantMap.clear();
566   for (unsigned i = 0, e = BBRefs.size(); i != e; ++i) {
567     long Location = BBLocations[BBRefs[i].first];
568     unsigned *Ref = BBRefs[i].second.first;
569     MachineInstr *MI = BBRefs[i].second.second;
570     DEBUG(std::cerr << "Fixup @" << std::hex << Ref << " to " << Location
571                     << " in instr: " << std::dec << *MI << "\n");
572   }
573
574   // Resolve branches to BasicBlocks for the entire function
575   for (unsigned i = 0, e = BBRefs.size(); i != e; ++i) {
576     long Location = BBLocations[BBRefs[i].first];
577     unsigned *Ref = BBRefs[i].second.first;
578     MachineInstr *MI = BBRefs[i].second.second;
579     DEBUG(std::cerr << "attempting to resolve BB: " << i << "\n");
580     for (unsigned ii = 0, ee = MI->getNumOperands(); ii != ee; ++ii) {
581       MachineOperand &op = MI->getOperand(ii);
582       if (op.isPCRelativeDisp()) {
583         // the instruction's branch target is made such that it branches to
584         // PC + (br target * 4), so undo that arithmetic here:
585         // Location is the target of the branch
586         // Ref is the location of the instruction, and hence the PC
587         unsigned branchTarget = (Location - (long)Ref) >> 2;
588         // Save the flags.
589         bool loBits32=false, hiBits32=false, loBits64=false, hiBits64=false;   
590         if (op.opLoBits32()) { loBits32=true; }
591         if (op.opHiBits32()) { hiBits32=true; }
592         if (op.opLoBits64()) { loBits64=true; }
593         if (op.opHiBits64()) { hiBits64=true; }
594         MI->SetMachineOperandConst(ii, MachineOperand::MO_SignExtendedImmed,
595                                    branchTarget);
596         if (loBits32) { MI->setOperandLo32(ii); }
597         else if (hiBits32) { MI->setOperandHi32(ii); }
598         else if (loBits64) { MI->setOperandLo64(ii); }
599         else if (hiBits64) { MI->setOperandHi64(ii); }
600         DEBUG(std::cerr << "Rewrote BB ref: ");
601         unsigned fixedInstr = SparcV9CodeEmitter::getBinaryCodeForInstr(*MI);
602         *Ref = fixedInstr;
603         break;
604       }
605     }
606   }
607   BBRefs.clear();
608   BBLocations.clear();
609
610   return false;
611 }
612
613 void SparcV9CodeEmitter::emitBasicBlock(MachineBasicBlock &MBB) {
614   currBB = MBB.getBasicBlock();
615   BBLocations[currBB] = MCE.getCurrentPCValue();
616   for (MachineBasicBlock::iterator I = MBB.begin(), E = MBB.end(); I != E; ++I)
617     emitWord(getBinaryCodeForInstr(**I));
618 }
619
620 void* SparcV9CodeEmitter::getGlobalAddress(GlobalValue *V, MachineInstr &MI,
621                                            bool isPCRelative)
622 {
623   if (isPCRelative) { // must be a call, this is a major hack!
624     // Try looking up the function to see if it is already compiled!
625     if (void *Addr = (void*)(intptr_t)MCE.getGlobalValueAddress(V)) {
626       intptr_t CurByte = MCE.getCurrentPCValue();
627       // The real target of the call is Addr = PC + (target * 4)
628       // CurByte is the PC, Addr we just received
629       return (void*) (((long)Addr - (long)CurByte) >> 2);
630     } else {
631       if (Function *F = dyn_cast<Function>(V)) {
632         // Function has not yet been code generated!
633         TheJITResolver->addFunctionReference(MCE.getCurrentPCValue(),
634                                              cast<Function>(V));
635         // Delayed resolution...
636         return 
637           (void*)(intptr_t)TheJITResolver->getLazyResolver(cast<Function>(V));
638
639       } else if (Constant *C = ConstantPointerRef::get(V)) {
640         if (ConstantMap.find(C) != ConstantMap.end()) {
641           return (void*)
642             (intptr_t)MCE.getConstantPoolEntryAddress(ConstantMap[C]);
643         } else {
644           std::cerr << "Constant: 0x" << std::hex << &*C << std::dec
645                     << ", " << *V << " not found in ConstantMap!\n";
646           abort();
647         }
648       } else {
649         std::cerr << "Unhandled global: " << *V << "\n";
650         abort();
651       }
652     }
653   } else {
654     return (void*)(intptr_t)MCE.getGlobalValueAddress(V);
655   }
656 }
657
658
659 #include "SparcV9CodeEmitter.inc"
660