Add stub version of unwind support
[oota-llvm.git] / lib / ExecutionEngine / Interpreter / Execution.cpp
1 //===-- Execution.cpp - Implement code to simulate the program ------------===//
2 // 
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file was developed by the LLVM research group and is distributed under
6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
7 // 
8 //===----------------------------------------------------------------------===//
9 // 
10 //  This file contains the actual instruction interpreter.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "Interpreter.h"
15 #include "llvm/Instructions.h"
16 #include "llvm/DerivedTypes.h"
17 #include "llvm/Constants.h"
18 #include "Support/Statistic.h"
19 #include <cmath>  // For fmod
20
21 Interpreter *TheEE = 0;
22
23 namespace {
24   Statistic<> NumDynamicInsts("lli", "Number of dynamic instructions executed");
25 }
26
27 //===----------------------------------------------------------------------===//
28 //                     Value Manipulation code
29 //===----------------------------------------------------------------------===//
30
31 // Operations used by constant expr implementations...
32 static GenericValue executeCastOperation(Value *Src, const Type *DestTy,
33                                          ExecutionContext &SF);
34 static GenericValue executeAddInst(GenericValue Src1, GenericValue Src2, 
35                                    const Type *Ty);
36
37 GenericValue Interpreter::getOperandValue(Value *V, ExecutionContext &SF) {
38   if (ConstantExpr *CE = dyn_cast<ConstantExpr>(V)) {
39     switch (CE->getOpcode()) {
40     case Instruction::Cast:
41       return executeCastOperation(CE->getOperand(0), CE->getType(), SF);
42     case Instruction::GetElementPtr:
43       return TheEE->executeGEPOperation(CE->getOperand(0), CE->op_begin()+1,
44                                         CE->op_end(), SF);
45     case Instruction::Add:
46       return executeAddInst(getOperandValue(CE->getOperand(0), SF),
47                             getOperandValue(CE->getOperand(1), SF),
48                             CE->getType());
49     default:
50       std::cerr << "Unhandled ConstantExpr: " << CE << "\n";
51       abort();
52       return GenericValue();
53     }
54   } else if (Constant *CPV = dyn_cast<Constant>(V)) {
55     return TheEE->getConstantValue(CPV);
56   } else if (GlobalValue *GV = dyn_cast<GlobalValue>(V)) {
57     return PTOGV(TheEE->getPointerToGlobal(GV));
58   } else {
59     return SF.Values[V];
60   }
61 }
62
63 static void SetValue(Value *V, GenericValue Val, ExecutionContext &SF) {
64   SF.Values[V] = Val;
65 }
66
67 //===----------------------------------------------------------------------===//
68 //                    Annotation Wrangling code
69 //===----------------------------------------------------------------------===//
70
71 void Interpreter::initializeExecutionEngine() {
72   TheEE = this;
73 }
74
75 //===----------------------------------------------------------------------===//
76 //                    Binary Instruction Implementations
77 //===----------------------------------------------------------------------===//
78
79 #define IMPLEMENT_BINARY_OPERATOR(OP, TY) \
80    case Type::TY##TyID: Dest.TY##Val = Src1.TY##Val OP Src2.TY##Val; break
81
82 static GenericValue executeAddInst(GenericValue Src1, GenericValue Src2, 
83                                    const Type *Ty) {
84   GenericValue Dest;
85   switch (Ty->getPrimitiveID()) {
86     IMPLEMENT_BINARY_OPERATOR(+, UByte);
87     IMPLEMENT_BINARY_OPERATOR(+, SByte);
88     IMPLEMENT_BINARY_OPERATOR(+, UShort);
89     IMPLEMENT_BINARY_OPERATOR(+, Short);
90     IMPLEMENT_BINARY_OPERATOR(+, UInt);
91     IMPLEMENT_BINARY_OPERATOR(+, Int);
92     IMPLEMENT_BINARY_OPERATOR(+, ULong);
93     IMPLEMENT_BINARY_OPERATOR(+, Long);
94     IMPLEMENT_BINARY_OPERATOR(+, Float);
95     IMPLEMENT_BINARY_OPERATOR(+, Double);
96   default:
97     std::cout << "Unhandled type for Add instruction: " << *Ty << "\n";
98     abort();
99   }
100   return Dest;
101 }
102
103 static GenericValue executeSubInst(GenericValue Src1, GenericValue Src2, 
104                                    const Type *Ty) {
105   GenericValue Dest;
106   switch (Ty->getPrimitiveID()) {
107     IMPLEMENT_BINARY_OPERATOR(-, UByte);
108     IMPLEMENT_BINARY_OPERATOR(-, SByte);
109     IMPLEMENT_BINARY_OPERATOR(-, UShort);
110     IMPLEMENT_BINARY_OPERATOR(-, Short);
111     IMPLEMENT_BINARY_OPERATOR(-, UInt);
112     IMPLEMENT_BINARY_OPERATOR(-, Int);
113     IMPLEMENT_BINARY_OPERATOR(-, ULong);
114     IMPLEMENT_BINARY_OPERATOR(-, Long);
115     IMPLEMENT_BINARY_OPERATOR(-, Float);
116     IMPLEMENT_BINARY_OPERATOR(-, Double);
117   default:
118     std::cout << "Unhandled type for Sub instruction: " << *Ty << "\n";
119     abort();
120   }
121   return Dest;
122 }
123
124 static GenericValue executeMulInst(GenericValue Src1, GenericValue Src2, 
125                                    const Type *Ty) {
126   GenericValue Dest;
127   switch (Ty->getPrimitiveID()) {
128     IMPLEMENT_BINARY_OPERATOR(*, UByte);
129     IMPLEMENT_BINARY_OPERATOR(*, SByte);
130     IMPLEMENT_BINARY_OPERATOR(*, UShort);
131     IMPLEMENT_BINARY_OPERATOR(*, Short);
132     IMPLEMENT_BINARY_OPERATOR(*, UInt);
133     IMPLEMENT_BINARY_OPERATOR(*, Int);
134     IMPLEMENT_BINARY_OPERATOR(*, ULong);
135     IMPLEMENT_BINARY_OPERATOR(*, Long);
136     IMPLEMENT_BINARY_OPERATOR(*, Float);
137     IMPLEMENT_BINARY_OPERATOR(*, Double);
138   default:
139     std::cout << "Unhandled type for Mul instruction: " << Ty << "\n";
140     abort();
141   }
142   return Dest;
143 }
144
145 static GenericValue executeDivInst(GenericValue Src1, GenericValue Src2, 
146                                    const Type *Ty) {
147   GenericValue Dest;
148   switch (Ty->getPrimitiveID()) {
149     IMPLEMENT_BINARY_OPERATOR(/, UByte);
150     IMPLEMENT_BINARY_OPERATOR(/, SByte);
151     IMPLEMENT_BINARY_OPERATOR(/, UShort);
152     IMPLEMENT_BINARY_OPERATOR(/, Short);
153     IMPLEMENT_BINARY_OPERATOR(/, UInt);
154     IMPLEMENT_BINARY_OPERATOR(/, Int);
155     IMPLEMENT_BINARY_OPERATOR(/, ULong);
156     IMPLEMENT_BINARY_OPERATOR(/, Long);
157     IMPLEMENT_BINARY_OPERATOR(/, Float);
158     IMPLEMENT_BINARY_OPERATOR(/, Double);
159   default:
160     std::cout << "Unhandled type for Div instruction: " << *Ty << "\n";
161     abort();
162   }
163   return Dest;
164 }
165
166 static GenericValue executeRemInst(GenericValue Src1, GenericValue Src2, 
167                                    const Type *Ty) {
168   GenericValue Dest;
169   switch (Ty->getPrimitiveID()) {
170     IMPLEMENT_BINARY_OPERATOR(%, UByte);
171     IMPLEMENT_BINARY_OPERATOR(%, SByte);
172     IMPLEMENT_BINARY_OPERATOR(%, UShort);
173     IMPLEMENT_BINARY_OPERATOR(%, Short);
174     IMPLEMENT_BINARY_OPERATOR(%, UInt);
175     IMPLEMENT_BINARY_OPERATOR(%, Int);
176     IMPLEMENT_BINARY_OPERATOR(%, ULong);
177     IMPLEMENT_BINARY_OPERATOR(%, Long);
178   case Type::FloatTyID:
179     Dest.FloatVal = fmod(Src1.FloatVal, Src2.FloatVal);
180     break;
181   case Type::DoubleTyID:
182     Dest.DoubleVal = fmod(Src1.DoubleVal, Src2.DoubleVal);
183     break;
184   default:
185     std::cout << "Unhandled type for Rem instruction: " << *Ty << "\n";
186     abort();
187   }
188   return Dest;
189 }
190
191 static GenericValue executeAndInst(GenericValue Src1, GenericValue Src2, 
192                                    const Type *Ty) {
193   GenericValue Dest;
194   switch (Ty->getPrimitiveID()) {
195     IMPLEMENT_BINARY_OPERATOR(&, Bool);
196     IMPLEMENT_BINARY_OPERATOR(&, UByte);
197     IMPLEMENT_BINARY_OPERATOR(&, SByte);
198     IMPLEMENT_BINARY_OPERATOR(&, UShort);
199     IMPLEMENT_BINARY_OPERATOR(&, Short);
200     IMPLEMENT_BINARY_OPERATOR(&, UInt);
201     IMPLEMENT_BINARY_OPERATOR(&, Int);
202     IMPLEMENT_BINARY_OPERATOR(&, ULong);
203     IMPLEMENT_BINARY_OPERATOR(&, Long);
204   default:
205     std::cout << "Unhandled type for And instruction: " << *Ty << "\n";
206     abort();
207   }
208   return Dest;
209 }
210
211 static GenericValue executeOrInst(GenericValue Src1, GenericValue Src2, 
212                                   const Type *Ty) {
213   GenericValue Dest;
214   switch (Ty->getPrimitiveID()) {
215     IMPLEMENT_BINARY_OPERATOR(|, Bool);
216     IMPLEMENT_BINARY_OPERATOR(|, UByte);
217     IMPLEMENT_BINARY_OPERATOR(|, SByte);
218     IMPLEMENT_BINARY_OPERATOR(|, UShort);
219     IMPLEMENT_BINARY_OPERATOR(|, Short);
220     IMPLEMENT_BINARY_OPERATOR(|, UInt);
221     IMPLEMENT_BINARY_OPERATOR(|, Int);
222     IMPLEMENT_BINARY_OPERATOR(|, ULong);
223     IMPLEMENT_BINARY_OPERATOR(|, Long);
224   default:
225     std::cout << "Unhandled type for Or instruction: " << *Ty << "\n";
226     abort();
227   }
228   return Dest;
229 }
230
231 static GenericValue executeXorInst(GenericValue Src1, GenericValue Src2, 
232                                    const Type *Ty) {
233   GenericValue Dest;
234   switch (Ty->getPrimitiveID()) {
235     IMPLEMENT_BINARY_OPERATOR(^, Bool);
236     IMPLEMENT_BINARY_OPERATOR(^, UByte);
237     IMPLEMENT_BINARY_OPERATOR(^, SByte);
238     IMPLEMENT_BINARY_OPERATOR(^, UShort);
239     IMPLEMENT_BINARY_OPERATOR(^, Short);
240     IMPLEMENT_BINARY_OPERATOR(^, UInt);
241     IMPLEMENT_BINARY_OPERATOR(^, Int);
242     IMPLEMENT_BINARY_OPERATOR(^, ULong);
243     IMPLEMENT_BINARY_OPERATOR(^, Long);
244   default:
245     std::cout << "Unhandled type for Xor instruction: " << *Ty << "\n";
246     abort();
247   }
248   return Dest;
249 }
250
251 #define IMPLEMENT_SETCC(OP, TY) \
252    case Type::TY##TyID: Dest.BoolVal = Src1.TY##Val OP Src2.TY##Val; break
253
254 // Handle pointers specially because they must be compared with only as much
255 // width as the host has.  We _do not_ want to be comparing 64 bit values when
256 // running on a 32-bit target, otherwise the upper 32 bits might mess up
257 // comparisons if they contain garbage.
258 #define IMPLEMENT_POINTERSETCC(OP) \
259    case Type::PointerTyID: \
260         Dest.BoolVal = (void*)(intptr_t)Src1.PointerVal OP \
261                        (void*)(intptr_t)Src2.PointerVal; break
262
263 static GenericValue executeSetEQInst(GenericValue Src1, GenericValue Src2, 
264                                      const Type *Ty) {
265   GenericValue Dest;
266   switch (Ty->getPrimitiveID()) {
267     IMPLEMENT_SETCC(==, UByte);
268     IMPLEMENT_SETCC(==, SByte);
269     IMPLEMENT_SETCC(==, UShort);
270     IMPLEMENT_SETCC(==, Short);
271     IMPLEMENT_SETCC(==, UInt);
272     IMPLEMENT_SETCC(==, Int);
273     IMPLEMENT_SETCC(==, ULong);
274     IMPLEMENT_SETCC(==, Long);
275     IMPLEMENT_SETCC(==, Float);
276     IMPLEMENT_SETCC(==, Double);
277     IMPLEMENT_POINTERSETCC(==);
278   default:
279     std::cout << "Unhandled type for SetEQ instruction: " << *Ty << "\n";
280     abort();
281   }
282   return Dest;
283 }
284
285 static GenericValue executeSetNEInst(GenericValue Src1, GenericValue Src2, 
286                                      const Type *Ty) {
287   GenericValue Dest;
288   switch (Ty->getPrimitiveID()) {
289     IMPLEMENT_SETCC(!=, UByte);
290     IMPLEMENT_SETCC(!=, SByte);
291     IMPLEMENT_SETCC(!=, UShort);
292     IMPLEMENT_SETCC(!=, Short);
293     IMPLEMENT_SETCC(!=, UInt);
294     IMPLEMENT_SETCC(!=, Int);
295     IMPLEMENT_SETCC(!=, ULong);
296     IMPLEMENT_SETCC(!=, Long);
297     IMPLEMENT_SETCC(!=, Float);
298     IMPLEMENT_SETCC(!=, Double);
299     IMPLEMENT_POINTERSETCC(!=);
300
301   default:
302     std::cout << "Unhandled type for SetNE instruction: " << *Ty << "\n";
303     abort();
304   }
305   return Dest;
306 }
307
308 static GenericValue executeSetLEInst(GenericValue Src1, GenericValue Src2, 
309                                      const Type *Ty) {
310   GenericValue Dest;
311   switch (Ty->getPrimitiveID()) {
312     IMPLEMENT_SETCC(<=, UByte);
313     IMPLEMENT_SETCC(<=, SByte);
314     IMPLEMENT_SETCC(<=, UShort);
315     IMPLEMENT_SETCC(<=, Short);
316     IMPLEMENT_SETCC(<=, UInt);
317     IMPLEMENT_SETCC(<=, Int);
318     IMPLEMENT_SETCC(<=, ULong);
319     IMPLEMENT_SETCC(<=, Long);
320     IMPLEMENT_SETCC(<=, Float);
321     IMPLEMENT_SETCC(<=, Double);
322     IMPLEMENT_POINTERSETCC(<=);
323   default:
324     std::cout << "Unhandled type for SetLE instruction: " << Ty << "\n";
325     abort();
326   }
327   return Dest;
328 }
329
330 static GenericValue executeSetGEInst(GenericValue Src1, GenericValue Src2, 
331                                      const Type *Ty) {
332   GenericValue Dest;
333   switch (Ty->getPrimitiveID()) {
334     IMPLEMENT_SETCC(>=, UByte);
335     IMPLEMENT_SETCC(>=, SByte);
336     IMPLEMENT_SETCC(>=, UShort);
337     IMPLEMENT_SETCC(>=, Short);
338     IMPLEMENT_SETCC(>=, UInt);
339     IMPLEMENT_SETCC(>=, Int);
340     IMPLEMENT_SETCC(>=, ULong);
341     IMPLEMENT_SETCC(>=, Long);
342     IMPLEMENT_SETCC(>=, Float);
343     IMPLEMENT_SETCC(>=, Double);
344     IMPLEMENT_POINTERSETCC(>=);
345   default:
346     std::cout << "Unhandled type for SetGE instruction: " << *Ty << "\n";
347     abort();
348   }
349   return Dest;
350 }
351
352 static GenericValue executeSetLTInst(GenericValue Src1, GenericValue Src2, 
353                                      const Type *Ty) {
354   GenericValue Dest;
355   switch (Ty->getPrimitiveID()) {
356     IMPLEMENT_SETCC(<, UByte);
357     IMPLEMENT_SETCC(<, SByte);
358     IMPLEMENT_SETCC(<, UShort);
359     IMPLEMENT_SETCC(<, Short);
360     IMPLEMENT_SETCC(<, UInt);
361     IMPLEMENT_SETCC(<, Int);
362     IMPLEMENT_SETCC(<, ULong);
363     IMPLEMENT_SETCC(<, Long);
364     IMPLEMENT_SETCC(<, Float);
365     IMPLEMENT_SETCC(<, Double);
366     IMPLEMENT_POINTERSETCC(<);
367   default:
368     std::cout << "Unhandled type for SetLT instruction: " << *Ty << "\n";
369     abort();
370   }
371   return Dest;
372 }
373
374 static GenericValue executeSetGTInst(GenericValue Src1, GenericValue Src2, 
375                                      const Type *Ty) {
376   GenericValue Dest;
377   switch (Ty->getPrimitiveID()) {
378     IMPLEMENT_SETCC(>, UByte);
379     IMPLEMENT_SETCC(>, SByte);
380     IMPLEMENT_SETCC(>, UShort);
381     IMPLEMENT_SETCC(>, Short);
382     IMPLEMENT_SETCC(>, UInt);
383     IMPLEMENT_SETCC(>, Int);
384     IMPLEMENT_SETCC(>, ULong);
385     IMPLEMENT_SETCC(>, Long);
386     IMPLEMENT_SETCC(>, Float);
387     IMPLEMENT_SETCC(>, Double);
388     IMPLEMENT_POINTERSETCC(>);
389   default:
390     std::cout << "Unhandled type for SetGT instruction: " << *Ty << "\n";
391     abort();
392   }
393   return Dest;
394 }
395
396 void Interpreter::visitBinaryOperator(BinaryOperator &I) {
397   ExecutionContext &SF = ECStack.back();
398   const Type *Ty    = I.getOperand(0)->getType();
399   GenericValue Src1 = getOperandValue(I.getOperand(0), SF);
400   GenericValue Src2 = getOperandValue(I.getOperand(1), SF);
401   GenericValue R;   // Result
402
403   switch (I.getOpcode()) {
404   case Instruction::Add:   R = executeAddInst  (Src1, Src2, Ty); break;
405   case Instruction::Sub:   R = executeSubInst  (Src1, Src2, Ty); break;
406   case Instruction::Mul:   R = executeMulInst  (Src1, Src2, Ty); break;
407   case Instruction::Div:   R = executeDivInst  (Src1, Src2, Ty); break;
408   case Instruction::Rem:   R = executeRemInst  (Src1, Src2, Ty); break;
409   case Instruction::And:   R = executeAndInst  (Src1, Src2, Ty); break;
410   case Instruction::Or:    R = executeOrInst   (Src1, Src2, Ty); break;
411   case Instruction::Xor:   R = executeXorInst  (Src1, Src2, Ty); break;
412   case Instruction::SetEQ: R = executeSetEQInst(Src1, Src2, Ty); break;
413   case Instruction::SetNE: R = executeSetNEInst(Src1, Src2, Ty); break;
414   case Instruction::SetLE: R = executeSetLEInst(Src1, Src2, Ty); break;
415   case Instruction::SetGE: R = executeSetGEInst(Src1, Src2, Ty); break;
416   case Instruction::SetLT: R = executeSetLTInst(Src1, Src2, Ty); break;
417   case Instruction::SetGT: R = executeSetGTInst(Src1, Src2, Ty); break;
418   default:
419     std::cout << "Don't know how to handle this binary operator!\n-->" << I;
420     abort();
421   }
422
423   SetValue(&I, R, SF);
424 }
425
426 //===----------------------------------------------------------------------===//
427 //                     Terminator Instruction Implementations
428 //===----------------------------------------------------------------------===//
429
430 void Interpreter::exitCalled(GenericValue GV) {
431   ExitCode = GV.SByteVal;
432   ECStack.clear();
433 }
434
435 /// Pop the last stack frame off of ECStack and then copy the result
436 /// back into the result variable if we are not returning void. The
437 /// result variable may be the ExitCode, or the Value of the calling
438 /// CallInst if there was a previous stack frame. This procedure may
439 /// invalidate any ECStack iterators you have.
440 ///
441 void Interpreter::popStackAndReturnValueToCaller (const Type *RetTy,
442                                                   GenericValue Result) {
443   // Pop the current stack frame.
444   ECStack.pop_back();
445
446   if (ECStack.empty()) {  // Finished main.  Put result into exit code... 
447     if (RetTy && RetTy->isIntegral()) {          // Nonvoid return type?       
448       ExitCode = Result.IntVal;   // Capture the exit code of the program 
449     } else { 
450       ExitCode = 0; 
451     } 
452   } else { 
453     // If we have a previous stack frame, and we have a previous call, 
454     // fill in the return value... 
455     ExecutionContext &CallingSF = ECStack.back();
456     if (CallingSF.Caller.getInstruction()) {
457       if (CallingSF.Caller.getType() != Type::VoidTy)      // Save result...
458         SetValue(CallingSF.Caller.getInstruction(), Result, CallingSF);
459       CallingSF.Caller = CallSite();          // We returned from the call...
460     }
461   }
462 }
463
464 void Interpreter::visitReturnInst(ReturnInst &I) {
465   ExecutionContext &SF = ECStack.back();
466   const Type *RetTy = Type::VoidTy;
467   GenericValue Result;
468
469   // Save away the return value... (if we are not 'ret void')
470   if (I.getNumOperands()) {
471     RetTy  = I.getReturnValue()->getType();
472     Result = getOperandValue(I.getReturnValue(), SF);
473   }
474
475   popStackAndReturnValueToCaller(RetTy, Result);
476 }
477
478 void Interpreter::visitUnwindInst(UnwindInst &I) {
479   abort ();
480 }
481
482 void Interpreter::visitBranchInst(BranchInst &I) {
483   ExecutionContext &SF = ECStack.back();
484   BasicBlock *Dest;
485
486   Dest = I.getSuccessor(0);          // Uncond branches have a fixed dest...
487   if (!I.isUnconditional()) {
488     Value *Cond = I.getCondition();
489     if (getOperandValue(Cond, SF).BoolVal == 0) // If false cond...
490       Dest = I.getSuccessor(1);    
491   }
492   SwitchToNewBasicBlock(Dest, SF);
493 }
494
495 void Interpreter::visitSwitchInst(SwitchInst &I) {
496   ExecutionContext &SF = ECStack.back();
497   GenericValue CondVal = getOperandValue(I.getOperand(0), SF);
498   const Type *ElTy = I.getOperand(0)->getType();
499
500   // Check to see if any of the cases match...
501   BasicBlock *Dest = 0;
502   for (unsigned i = 2, e = I.getNumOperands(); i != e; i += 2)
503     if (executeSetEQInst(CondVal,
504                          getOperandValue(I.getOperand(i), SF), ElTy).BoolVal) {
505       Dest = cast<BasicBlock>(I.getOperand(i+1));
506       break;
507     }
508   
509   if (!Dest) Dest = I.getDefaultDest();   // No cases matched: use default
510   SwitchToNewBasicBlock(Dest, SF);
511 }
512
513 // SwitchToNewBasicBlock - This method is used to jump to a new basic block.
514 // This function handles the actual updating of block and instruction iterators
515 // as well as execution of all of the PHI nodes in the destination block.
516 //
517 // This method does this because all of the PHI nodes must be executed
518 // atomically, reading their inputs before any of the results are updated.  Not
519 // doing this can cause problems if the PHI nodes depend on other PHI nodes for
520 // their inputs.  If the input PHI node is updated before it is read, incorrect
521 // results can happen.  Thus we use a two phase approach.
522 //
523 void Interpreter::SwitchToNewBasicBlock(BasicBlock *Dest, ExecutionContext &SF){
524   BasicBlock *PrevBB = SF.CurBB;      // Remember where we came from...
525   SF.CurBB   = Dest;                  // Update CurBB to branch destination
526   SF.CurInst = SF.CurBB->begin();     // Update new instruction ptr...
527
528   if (!isa<PHINode>(SF.CurInst)) return;  // Nothing fancy to do
529
530   // Loop over all of the PHI nodes in the current block, reading their inputs.
531   std::vector<GenericValue> ResultValues;
532
533   for (; PHINode *PN = dyn_cast<PHINode>(SF.CurInst); ++SF.CurInst) {
534     // Search for the value corresponding to this previous bb...
535     int i = PN->getBasicBlockIndex(PrevBB);
536     assert(i != -1 && "PHINode doesn't contain entry for predecessor??");
537     Value *IncomingValue = PN->getIncomingValue(i);
538     
539     // Save the incoming value for this PHI node...
540     ResultValues.push_back(getOperandValue(IncomingValue, SF));
541   }
542
543   // Now loop over all of the PHI nodes setting their values...
544   SF.CurInst = SF.CurBB->begin();
545   for (unsigned i = 0; PHINode *PN = dyn_cast<PHINode>(SF.CurInst);
546        ++SF.CurInst, ++i)
547     SetValue(PN, ResultValues[i], SF);
548 }
549
550 //===----------------------------------------------------------------------===//
551 //                     Memory Instruction Implementations
552 //===----------------------------------------------------------------------===//
553
554 void Interpreter::visitAllocationInst(AllocationInst &I) {
555   ExecutionContext &SF = ECStack.back();
556
557   const Type *Ty = I.getType()->getElementType();  // Type to be allocated
558
559   // Get the number of elements being allocated by the array...
560   unsigned NumElements = getOperandValue(I.getOperand(0), SF).UIntVal;
561
562   // Allocate enough memory to hold the type...
563   void *Memory = malloc(NumElements * TD.getTypeSize(Ty));
564
565   GenericValue Result = PTOGV(Memory);
566   assert(Result.PointerVal != 0 && "Null pointer returned by malloc!");
567   SetValue(&I, Result, SF);
568
569   if (I.getOpcode() == Instruction::Alloca)
570     ECStack.back().Allocas.add(Memory);
571 }
572
573 void Interpreter::visitFreeInst(FreeInst &I) {
574   ExecutionContext &SF = ECStack.back();
575   assert(isa<PointerType>(I.getOperand(0)->getType()) && "Freeing nonptr?");
576   GenericValue Value = getOperandValue(I.getOperand(0), SF);
577   // TODO: Check to make sure memory is allocated
578   free(GVTOP(Value));   // Free memory
579 }
580
581 // getElementOffset - The workhorse for getelementptr.
582 //
583 GenericValue Interpreter::executeGEPOperation(Value *Ptr, User::op_iterator I,
584                                               User::op_iterator E,
585                                               ExecutionContext &SF) {
586   assert(isa<PointerType>(Ptr->getType()) &&
587          "Cannot getElementOffset of a nonpointer type!");
588
589   PointerTy Total = 0;
590   const Type *Ty = Ptr->getType();
591
592   for (; I != E; ++I) {
593     if (const StructType *STy = dyn_cast<StructType>(Ty)) {
594       const StructLayout *SLO = TD.getStructLayout(STy);
595       
596       // Indices must be ubyte constants...
597       const ConstantUInt *CPU = cast<ConstantUInt>(*I);
598       assert(CPU->getType() == Type::UByteTy);
599       unsigned Index = CPU->getValue();
600       
601       Total += SLO->MemberOffsets[Index];
602       Ty = STy->getElementTypes()[Index];
603     } else if (const SequentialType *ST = cast<SequentialType>(Ty)) {
604       // Get the index number for the array... which must be long type...
605       assert((*I)->getType() == Type::LongTy);
606       unsigned Idx = getOperandValue(*I, SF).LongVal;
607       Ty = ST->getElementType();
608       unsigned Size = TD.getTypeSize(Ty);
609       Total += Size*Idx;
610     }
611   }
612
613   GenericValue Result;
614   Result.PointerVal = getOperandValue(Ptr, SF).PointerVal + Total;
615   return Result;
616 }
617
618 void Interpreter::visitGetElementPtrInst(GetElementPtrInst &I) {
619   ExecutionContext &SF = ECStack.back();
620   SetValue(&I, TheEE->executeGEPOperation(I.getPointerOperand(),
621                                    I.idx_begin(), I.idx_end(), SF), SF);
622 }
623
624 void Interpreter::visitLoadInst(LoadInst &I) {
625   ExecutionContext &SF = ECStack.back();
626   GenericValue SRC = getOperandValue(I.getPointerOperand(), SF);
627   GenericValue *Ptr = (GenericValue*)GVTOP(SRC);
628   GenericValue Result = LoadValueFromMemory(Ptr, I.getType());
629   SetValue(&I, Result, SF);
630 }
631
632 void Interpreter::visitStoreInst(StoreInst &I) {
633   ExecutionContext &SF = ECStack.back();
634   GenericValue Val = getOperandValue(I.getOperand(0), SF);
635   GenericValue SRC = getOperandValue(I.getPointerOperand(), SF);
636   StoreValueToMemory(Val, (GenericValue *)GVTOP(SRC),
637                      I.getOperand(0)->getType());
638 }
639
640 //===----------------------------------------------------------------------===//
641 //                 Miscellaneous Instruction Implementations
642 //===----------------------------------------------------------------------===//
643
644 void Interpreter::visitCallSite(CallSite CS) {
645   ExecutionContext &SF = ECStack.back();
646   SF.Caller = CS;
647   std::vector<GenericValue> ArgVals;
648   const unsigned NumArgs = SF.Caller.arg_size();
649   ArgVals.reserve(NumArgs);
650   for (CallSite::arg_iterator i = SF.Caller.arg_begin(),
651          e = SF.Caller.arg_end(); i != e; ++i) {
652     Value *V = *i;
653     ArgVals.push_back(getOperandValue(V, SF));
654     // Promote all integral types whose size is < sizeof(int) into ints.  We do
655     // this by zero or sign extending the value as appropriate according to the
656     // source type.
657     const Type *Ty = V->getType();
658     if (Ty->isIntegral() && Ty->getPrimitiveSize() < 4) {
659       if (Ty == Type::ShortTy)
660         ArgVals.back().IntVal = ArgVals.back().ShortVal;
661       else if (Ty == Type::UShortTy)
662         ArgVals.back().UIntVal = ArgVals.back().UShortVal;
663       else if (Ty == Type::SByteTy)
664         ArgVals.back().IntVal = ArgVals.back().SByteVal;
665       else if (Ty == Type::UByteTy)
666         ArgVals.back().UIntVal = ArgVals.back().UByteVal;
667       else if (Ty == Type::BoolTy)
668         ArgVals.back().UIntVal = ArgVals.back().BoolVal;
669       else
670         assert(0 && "Unknown type!");
671     }
672   }
673
674   // To handle indirect calls, we must get the pointer value from the argument 
675   // and treat it as a function pointer.
676   GenericValue SRC = getOperandValue(SF.Caller.getCalledValue(), SF);  
677   callFunction((Function*)GVTOP(SRC), ArgVals);
678 }
679
680 #define IMPLEMENT_SHIFT(OP, TY) \
681    case Type::TY##TyID: Dest.TY##Val = Src1.TY##Val OP Src2.UByteVal; break
682
683 void Interpreter::visitShl(ShiftInst &I) {
684   ExecutionContext &SF = ECStack.back();
685   const Type *Ty    = I.getOperand(0)->getType();
686   GenericValue Src1 = getOperandValue(I.getOperand(0), SF);
687   GenericValue Src2 = getOperandValue(I.getOperand(1), SF);
688   GenericValue Dest;
689
690   switch (Ty->getPrimitiveID()) {
691     IMPLEMENT_SHIFT(<<, UByte);
692     IMPLEMENT_SHIFT(<<, SByte);
693     IMPLEMENT_SHIFT(<<, UShort);
694     IMPLEMENT_SHIFT(<<, Short);
695     IMPLEMENT_SHIFT(<<, UInt);
696     IMPLEMENT_SHIFT(<<, Int);
697     IMPLEMENT_SHIFT(<<, ULong);
698     IMPLEMENT_SHIFT(<<, Long);
699   default:
700     std::cout << "Unhandled type for Shl instruction: " << *Ty << "\n";
701   }
702   SetValue(&I, Dest, SF);
703 }
704
705 void Interpreter::visitShr(ShiftInst &I) {
706   ExecutionContext &SF = ECStack.back();
707   const Type *Ty    = I.getOperand(0)->getType();
708   GenericValue Src1 = getOperandValue(I.getOperand(0), SF);
709   GenericValue Src2 = getOperandValue(I.getOperand(1), SF);
710   GenericValue Dest;
711
712   switch (Ty->getPrimitiveID()) {
713     IMPLEMENT_SHIFT(>>, UByte);
714     IMPLEMENT_SHIFT(>>, SByte);
715     IMPLEMENT_SHIFT(>>, UShort);
716     IMPLEMENT_SHIFT(>>, Short);
717     IMPLEMENT_SHIFT(>>, UInt);
718     IMPLEMENT_SHIFT(>>, Int);
719     IMPLEMENT_SHIFT(>>, ULong);
720     IMPLEMENT_SHIFT(>>, Long);
721   default:
722     std::cout << "Unhandled type for Shr instruction: " << *Ty << "\n";
723     abort();
724   }
725   SetValue(&I, Dest, SF);
726 }
727
728 #define IMPLEMENT_CAST(DTY, DCTY, STY) \
729    case Type::STY##TyID: Dest.DTY##Val = DCTY Src.STY##Val; break;
730
731 #define IMPLEMENT_CAST_CASE_START(DESTTY, DESTCTY)    \
732   case Type::DESTTY##TyID:                      \
733     switch (SrcTy->getPrimitiveID()) {          \
734       IMPLEMENT_CAST(DESTTY, DESTCTY, Bool);    \
735       IMPLEMENT_CAST(DESTTY, DESTCTY, UByte);   \
736       IMPLEMENT_CAST(DESTTY, DESTCTY, SByte);   \
737       IMPLEMENT_CAST(DESTTY, DESTCTY, UShort);  \
738       IMPLEMENT_CAST(DESTTY, DESTCTY, Short);   \
739       IMPLEMENT_CAST(DESTTY, DESTCTY, UInt);    \
740       IMPLEMENT_CAST(DESTTY, DESTCTY, Int);     \
741       IMPLEMENT_CAST(DESTTY, DESTCTY, ULong);   \
742       IMPLEMENT_CAST(DESTTY, DESTCTY, Long);    \
743       IMPLEMENT_CAST(DESTTY, DESTCTY, Pointer);
744
745 #define IMPLEMENT_CAST_CASE_FP_IMP(DESTTY, DESTCTY) \
746       IMPLEMENT_CAST(DESTTY, DESTCTY, Float);   \
747       IMPLEMENT_CAST(DESTTY, DESTCTY, Double)
748
749 #define IMPLEMENT_CAST_CASE_END()    \
750     default: std::cout << "Unhandled cast: " << SrcTy << " to " << Ty << "\n"; \
751       abort();                                  \
752     }                                           \
753     break
754
755 #define IMPLEMENT_CAST_CASE(DESTTY, DESTCTY) \
756    IMPLEMENT_CAST_CASE_START(DESTTY, DESTCTY);   \
757    IMPLEMENT_CAST_CASE_FP_IMP(DESTTY, DESTCTY); \
758    IMPLEMENT_CAST_CASE_END()
759
760 GenericValue Interpreter::executeCastOperation(Value *SrcVal, const Type *Ty,
761                                                ExecutionContext &SF) {
762   const Type *SrcTy = SrcVal->getType();
763   GenericValue Dest, Src = getOperandValue(SrcVal, SF);
764
765   switch (Ty->getPrimitiveID()) {
766     IMPLEMENT_CAST_CASE(UByte  , (unsigned char));
767     IMPLEMENT_CAST_CASE(SByte  , (  signed char));
768     IMPLEMENT_CAST_CASE(UShort , (unsigned short));
769     IMPLEMENT_CAST_CASE(Short  , (  signed short));
770     IMPLEMENT_CAST_CASE(UInt   , (unsigned int ));
771     IMPLEMENT_CAST_CASE(Int    , (  signed int ));
772     IMPLEMENT_CAST_CASE(ULong  , (uint64_t));
773     IMPLEMENT_CAST_CASE(Long   , ( int64_t));
774     IMPLEMENT_CAST_CASE(Pointer, (PointerTy));
775     IMPLEMENT_CAST_CASE(Float  , (float));
776     IMPLEMENT_CAST_CASE(Double , (double));
777     IMPLEMENT_CAST_CASE(Bool   , (bool));
778   default:
779     std::cout << "Unhandled dest type for cast instruction: " << *Ty << "\n";
780     abort();
781   }
782
783   return Dest;
784 }
785
786 void Interpreter::visitCastInst(CastInst &I) {
787   ExecutionContext &SF = ECStack.back();
788   SetValue(&I, executeCastOperation(I.getOperand(0), I.getType(), SF), SF);
789 }
790
791 void Interpreter::visitVANextInst(VANextInst &I) {
792   ExecutionContext &SF = ECStack.back();
793
794   // Get the incoming valist element.  LLI treats the valist as an integer.
795   GenericValue VAList = getOperandValue(I.getOperand(0), SF);
796   
797   // Move to the next operand.
798   unsigned Argument = VAList.IntVal++;
799   assert(Argument < SF.VarArgs.size() &&
800          "Accessing past the last vararg argument!");
801   SetValue(&I, VAList, SF);
802 }
803
804 //===----------------------------------------------------------------------===//
805 //                        Dispatch and Execution Code
806 //===----------------------------------------------------------------------===//
807
808 //===----------------------------------------------------------------------===//
809 // callFunction - Execute the specified function...
810 //
811 void Interpreter::callFunction(Function *F,
812                                const std::vector<GenericValue> &ArgVals) {
813   assert((ECStack.empty() || ECStack.back().Caller.getInstruction() == 0 || 
814           ECStack.back().Caller.arg_size() == ArgVals.size()) &&
815          "Incorrect number of arguments passed into function call!");
816   // Make a new stack frame... and fill it in.
817   ECStack.push_back(ExecutionContext());
818   ExecutionContext &StackFrame = ECStack.back();
819   StackFrame.CurFunction = F;
820
821   // Special handling for external functions.
822   if (F->isExternal()) {
823     GenericValue Result = callExternalFunction (F, ArgVals);
824     // Simulate a 'ret' instruction of the appropriate type.
825     popStackAndReturnValueToCaller (F->getReturnType (), Result);
826     return;
827   }
828
829   // Get pointers to first LLVM BB & Instruction in function.
830   StackFrame.CurBB     = F->begin();
831   StackFrame.CurInst   = StackFrame.CurBB->begin();
832
833   // Run through the function arguments and initialize their values...
834   assert((ArgVals.size() == F->asize() ||
835          (ArgVals.size() > F->asize() && F->getFunctionType()->isVarArg())) &&
836          "Invalid number of values passed to function invocation!");
837
838   // Handle non-varargs arguments...
839   unsigned i = 0;
840   for (Function::aiterator AI = F->abegin(), E = F->aend(); AI != E; ++AI, ++i)
841     SetValue(AI, ArgVals[i], StackFrame);
842
843   // Handle varargs arguments...
844   StackFrame.VarArgs.assign(ArgVals.begin()+i, ArgVals.end());
845 }
846
847 void Interpreter::run() {
848   while (!ECStack.empty()) {
849     // Interpret a single instruction & increment the "PC".
850     ExecutionContext &SF = ECStack.back();  // Current stack frame
851     Instruction &I = *SF.CurInst++;         // Increment before execute
852     
853     // Track the number of dynamic instructions executed.
854     ++NumDynamicInsts;
855
856     visit(I);   // Dispatch to one of the visit* methods...
857   }
858 }