Be friendly to gcc 3.4... good compiler. Nice compiler.
[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 void Interpreter::visitReturnInst(ReturnInst &I) {
436   ExecutionContext &SF = ECStack.back();
437   const Type *RetTy = 0;
438   GenericValue Result;
439
440   // Save away the return value... (if we are not 'ret void')
441   if (I.getNumOperands()) {
442     RetTy  = I.getReturnValue()->getType();
443     Result = getOperandValue(I.getReturnValue(), SF);
444   }
445
446   // Pop the current stack frame... this invalidates SF
447   ECStack.pop_back();
448
449   if (ECStack.empty()) {  // Finished main.  Put result into exit code...
450     if (RetTy && RetTy->isIntegral()) {          // Nonvoid return type?
451       ExitCode = Result.IntVal;   // Capture the exit code of the program
452     } else {
453       ExitCode = 0;
454     }
455   } else {
456     // If we have a previous stack frame, and we have a previous call,
457     // fill in the return value...
458     ExecutionContext &NewSF = ECStack.back();
459     if (NewSF.Caller) {
460       if (NewSF.Caller->getType() != Type::VoidTy)      // Save result...
461         SetValue(NewSF.Caller, Result, NewSF);
462       NewSF.Caller = 0;          // We returned from the call...
463     }
464   }
465 }
466
467 void Interpreter::visitBranchInst(BranchInst &I) {
468   ExecutionContext &SF = ECStack.back();
469   BasicBlock *Dest;
470
471   Dest = I.getSuccessor(0);          // Uncond branches have a fixed dest...
472   if (!I.isUnconditional()) {
473     Value *Cond = I.getCondition();
474     if (getOperandValue(Cond, SF).BoolVal == 0) // If false cond...
475       Dest = I.getSuccessor(1);    
476   }
477   SwitchToNewBasicBlock(Dest, SF);
478 }
479
480 void Interpreter::visitSwitchInst(SwitchInst &I) {
481   ExecutionContext &SF = ECStack.back();
482   GenericValue CondVal = getOperandValue(I.getOperand(0), SF);
483   const Type *ElTy = I.getOperand(0)->getType();
484
485   // Check to see if any of the cases match...
486   BasicBlock *Dest = 0;
487   for (unsigned i = 2, e = I.getNumOperands(); i != e; i += 2)
488     if (executeSetEQInst(CondVal,
489                          getOperandValue(I.getOperand(i), SF), ElTy).BoolVal) {
490       Dest = cast<BasicBlock>(I.getOperand(i+1));
491       break;
492     }
493   
494   if (!Dest) Dest = I.getDefaultDest();   // No cases matched: use default
495   SwitchToNewBasicBlock(Dest, SF);
496 }
497
498 // SwitchToNewBasicBlock - This method is used to jump to a new basic block.
499 // This function handles the actual updating of block and instruction iterators
500 // as well as execution of all of the PHI nodes in the destination block.
501 //
502 // This method does this because all of the PHI nodes must be executed
503 // atomically, reading their inputs before any of the results are updated.  Not
504 // doing this can cause problems if the PHI nodes depend on other PHI nodes for
505 // their inputs.  If the input PHI node is updated before it is read, incorrect
506 // results can happen.  Thus we use a two phase approach.
507 //
508 void Interpreter::SwitchToNewBasicBlock(BasicBlock *Dest, ExecutionContext &SF){
509   BasicBlock *PrevBB = SF.CurBB;      // Remember where we came from...
510   SF.CurBB   = Dest;                  // Update CurBB to branch destination
511   SF.CurInst = SF.CurBB->begin();     // Update new instruction ptr...
512
513   if (!isa<PHINode>(SF.CurInst)) return;  // Nothing fancy to do
514
515   // Loop over all of the PHI nodes in the current block, reading their inputs.
516   std::vector<GenericValue> ResultValues;
517
518   for (; PHINode *PN = dyn_cast<PHINode>(SF.CurInst); ++SF.CurInst) {
519     // Search for the value corresponding to this previous bb...
520     int i = PN->getBasicBlockIndex(PrevBB);
521     assert(i != -1 && "PHINode doesn't contain entry for predecessor??");
522     Value *IncomingValue = PN->getIncomingValue(i);
523     
524     // Save the incoming value for this PHI node...
525     ResultValues.push_back(getOperandValue(IncomingValue, SF));
526   }
527
528   // Now loop over all of the PHI nodes setting their values...
529   SF.CurInst = SF.CurBB->begin();
530   for (unsigned i = 0; PHINode *PN = dyn_cast<PHINode>(SF.CurInst);
531        ++SF.CurInst, ++i)
532     SetValue(PN, ResultValues[i], SF);
533 }
534
535 //===----------------------------------------------------------------------===//
536 //                     Memory Instruction Implementations
537 //===----------------------------------------------------------------------===//
538
539 void Interpreter::visitAllocationInst(AllocationInst &I) {
540   ExecutionContext &SF = ECStack.back();
541
542   const Type *Ty = I.getType()->getElementType();  // Type to be allocated
543
544   // Get the number of elements being allocated by the array...
545   unsigned NumElements = getOperandValue(I.getOperand(0), SF).UIntVal;
546
547   // Allocate enough memory to hold the type...
548   void *Memory = malloc(NumElements * TD.getTypeSize(Ty));
549
550   GenericValue Result = PTOGV(Memory);
551   assert(Result.PointerVal != 0 && "Null pointer returned by malloc!");
552   SetValue(&I, Result, SF);
553
554   if (I.getOpcode() == Instruction::Alloca)
555     ECStack.back().Allocas.add(Memory);
556 }
557
558 void Interpreter::visitFreeInst(FreeInst &I) {
559   ExecutionContext &SF = ECStack.back();
560   assert(isa<PointerType>(I.getOperand(0)->getType()) && "Freeing nonptr?");
561   GenericValue Value = getOperandValue(I.getOperand(0), SF);
562   // TODO: Check to make sure memory is allocated
563   free(GVTOP(Value));   // Free memory
564 }
565
566 // getElementOffset - The workhorse for getelementptr.
567 //
568 GenericValue Interpreter::executeGEPOperation(Value *Ptr, User::op_iterator I,
569                                               User::op_iterator E,
570                                               ExecutionContext &SF) {
571   assert(isa<PointerType>(Ptr->getType()) &&
572          "Cannot getElementOffset of a nonpointer type!");
573
574   PointerTy Total = 0;
575   const Type *Ty = Ptr->getType();
576
577   for (; I != E; ++I) {
578     if (const StructType *STy = dyn_cast<StructType>(Ty)) {
579       const StructLayout *SLO = TD.getStructLayout(STy);
580       
581       // Indices must be ubyte constants...
582       const ConstantUInt *CPU = cast<ConstantUInt>(*I);
583       assert(CPU->getType() == Type::UByteTy);
584       unsigned Index = CPU->getValue();
585       
586       Total += SLO->MemberOffsets[Index];
587       Ty = STy->getElementTypes()[Index];
588     } else if (const SequentialType *ST = cast<SequentialType>(Ty)) {
589       // Get the index number for the array... which must be long type...
590       assert((*I)->getType() == Type::LongTy);
591       unsigned Idx = getOperandValue(*I, SF).LongVal;
592       Ty = ST->getElementType();
593       unsigned Size = TD.getTypeSize(Ty);
594       Total += Size*Idx;
595     }
596   }
597
598   GenericValue Result;
599   Result.PointerVal = getOperandValue(Ptr, SF).PointerVal + Total;
600   return Result;
601 }
602
603 void Interpreter::visitGetElementPtrInst(GetElementPtrInst &I) {
604   ExecutionContext &SF = ECStack.back();
605   SetValue(&I, TheEE->executeGEPOperation(I.getPointerOperand(),
606                                    I.idx_begin(), I.idx_end(), SF), SF);
607 }
608
609 void Interpreter::visitLoadInst(LoadInst &I) {
610   ExecutionContext &SF = ECStack.back();
611   GenericValue SRC = getOperandValue(I.getPointerOperand(), SF);
612   GenericValue *Ptr = (GenericValue*)GVTOP(SRC);
613   GenericValue Result = LoadValueFromMemory(Ptr, I.getType());
614   SetValue(&I, Result, SF);
615 }
616
617 void Interpreter::visitStoreInst(StoreInst &I) {
618   ExecutionContext &SF = ECStack.back();
619   GenericValue Val = getOperandValue(I.getOperand(0), SF);
620   GenericValue SRC = getOperandValue(I.getPointerOperand(), SF);
621   StoreValueToMemory(Val, (GenericValue *)GVTOP(SRC),
622                      I.getOperand(0)->getType());
623 }
624
625 //===----------------------------------------------------------------------===//
626 //                 Miscellaneous Instruction Implementations
627 //===----------------------------------------------------------------------===//
628
629 void Interpreter::visitCallInst(CallInst &I) {
630   ExecutionContext &SF = ECStack.back();
631   SF.Caller = &I;
632   std::vector<GenericValue> ArgVals;
633   ArgVals.reserve(I.getNumOperands()-1);
634   for (unsigned i = 1; i < I.getNumOperands(); ++i) {
635     ArgVals.push_back(getOperandValue(I.getOperand(i), SF));
636     // Promote all integral types whose size is < sizeof(int) into ints.  We do
637     // this by zero or sign extending the value as appropriate according to the
638     // source type.
639     if (I.getOperand(i)->getType()->isIntegral() &&
640         I.getOperand(i)->getType()->getPrimitiveSize() < 4) {
641       const Type *Ty = I.getOperand(i)->getType();
642       if (Ty == Type::ShortTy)
643         ArgVals.back().IntVal = ArgVals.back().ShortVal;
644       else if (Ty == Type::UShortTy)
645         ArgVals.back().UIntVal = ArgVals.back().UShortVal;
646       else if (Ty == Type::SByteTy)
647         ArgVals.back().IntVal = ArgVals.back().SByteVal;
648       else if (Ty == Type::UByteTy)
649         ArgVals.back().UIntVal = ArgVals.back().UByteVal;
650       else if (Ty == Type::BoolTy)
651         ArgVals.back().UIntVal = ArgVals.back().BoolVal;
652       else
653         assert(0 && "Unknown type!");
654     }
655   }
656
657   // To handle indirect calls, we must get the pointer value from the argument 
658   // and treat it as a function pointer.
659   GenericValue SRC = getOperandValue(I.getCalledValue(), SF);  
660   callFunction((Function*)GVTOP(SRC), ArgVals);
661 }
662
663 #define IMPLEMENT_SHIFT(OP, TY) \
664    case Type::TY##TyID: Dest.TY##Val = Src1.TY##Val OP Src2.UByteVal; break
665
666 void Interpreter::visitShl(ShiftInst &I) {
667   ExecutionContext &SF = ECStack.back();
668   const Type *Ty    = I.getOperand(0)->getType();
669   GenericValue Src1 = getOperandValue(I.getOperand(0), SF);
670   GenericValue Src2 = getOperandValue(I.getOperand(1), SF);
671   GenericValue Dest;
672
673   switch (Ty->getPrimitiveID()) {
674     IMPLEMENT_SHIFT(<<, UByte);
675     IMPLEMENT_SHIFT(<<, SByte);
676     IMPLEMENT_SHIFT(<<, UShort);
677     IMPLEMENT_SHIFT(<<, Short);
678     IMPLEMENT_SHIFT(<<, UInt);
679     IMPLEMENT_SHIFT(<<, Int);
680     IMPLEMENT_SHIFT(<<, ULong);
681     IMPLEMENT_SHIFT(<<, Long);
682   default:
683     std::cout << "Unhandled type for Shl instruction: " << *Ty << "\n";
684   }
685   SetValue(&I, Dest, SF);
686 }
687
688 void Interpreter::visitShr(ShiftInst &I) {
689   ExecutionContext &SF = ECStack.back();
690   const Type *Ty    = I.getOperand(0)->getType();
691   GenericValue Src1 = getOperandValue(I.getOperand(0), SF);
692   GenericValue Src2 = getOperandValue(I.getOperand(1), SF);
693   GenericValue Dest;
694
695   switch (Ty->getPrimitiveID()) {
696     IMPLEMENT_SHIFT(>>, UByte);
697     IMPLEMENT_SHIFT(>>, SByte);
698     IMPLEMENT_SHIFT(>>, UShort);
699     IMPLEMENT_SHIFT(>>, Short);
700     IMPLEMENT_SHIFT(>>, UInt);
701     IMPLEMENT_SHIFT(>>, Int);
702     IMPLEMENT_SHIFT(>>, ULong);
703     IMPLEMENT_SHIFT(>>, Long);
704   default:
705     std::cout << "Unhandled type for Shr instruction: " << *Ty << "\n";
706     abort();
707   }
708   SetValue(&I, Dest, SF);
709 }
710
711 #define IMPLEMENT_CAST(DTY, DCTY, STY) \
712    case Type::STY##TyID: Dest.DTY##Val = DCTY Src.STY##Val; break;
713
714 #define IMPLEMENT_CAST_CASE_START(DESTTY, DESTCTY)    \
715   case Type::DESTTY##TyID:                      \
716     switch (SrcTy->getPrimitiveID()) {          \
717       IMPLEMENT_CAST(DESTTY, DESTCTY, Bool);    \
718       IMPLEMENT_CAST(DESTTY, DESTCTY, UByte);   \
719       IMPLEMENT_CAST(DESTTY, DESTCTY, SByte);   \
720       IMPLEMENT_CAST(DESTTY, DESTCTY, UShort);  \
721       IMPLEMENT_CAST(DESTTY, DESTCTY, Short);   \
722       IMPLEMENT_CAST(DESTTY, DESTCTY, UInt);    \
723       IMPLEMENT_CAST(DESTTY, DESTCTY, Int);     \
724       IMPLEMENT_CAST(DESTTY, DESTCTY, ULong);   \
725       IMPLEMENT_CAST(DESTTY, DESTCTY, Long);    \
726       IMPLEMENT_CAST(DESTTY, DESTCTY, Pointer);
727
728 #define IMPLEMENT_CAST_CASE_FP_IMP(DESTTY, DESTCTY) \
729       IMPLEMENT_CAST(DESTTY, DESTCTY, Float);   \
730       IMPLEMENT_CAST(DESTTY, DESTCTY, Double)
731
732 #define IMPLEMENT_CAST_CASE_END()    \
733     default: std::cout << "Unhandled cast: " << SrcTy << " to " << Ty << "\n"; \
734       abort();                                  \
735     }                                           \
736     break
737
738 #define IMPLEMENT_CAST_CASE(DESTTY, DESTCTY) \
739    IMPLEMENT_CAST_CASE_START(DESTTY, DESTCTY);   \
740    IMPLEMENT_CAST_CASE_FP_IMP(DESTTY, DESTCTY); \
741    IMPLEMENT_CAST_CASE_END()
742
743 GenericValue Interpreter::executeCastOperation(Value *SrcVal, const Type *Ty,
744                                                ExecutionContext &SF) {
745   const Type *SrcTy = SrcVal->getType();
746   GenericValue Dest, Src = getOperandValue(SrcVal, SF);
747
748   switch (Ty->getPrimitiveID()) {
749     IMPLEMENT_CAST_CASE(UByte  , (unsigned char));
750     IMPLEMENT_CAST_CASE(SByte  , (  signed char));
751     IMPLEMENT_CAST_CASE(UShort , (unsigned short));
752     IMPLEMENT_CAST_CASE(Short  , (  signed short));
753     IMPLEMENT_CAST_CASE(UInt   , (unsigned int ));
754     IMPLEMENT_CAST_CASE(Int    , (  signed int ));
755     IMPLEMENT_CAST_CASE(ULong  , (uint64_t));
756     IMPLEMENT_CAST_CASE(Long   , ( int64_t));
757     IMPLEMENT_CAST_CASE(Pointer, (PointerTy));
758     IMPLEMENT_CAST_CASE(Float  , (float));
759     IMPLEMENT_CAST_CASE(Double , (double));
760     IMPLEMENT_CAST_CASE(Bool   , (bool));
761   default:
762     std::cout << "Unhandled dest type for cast instruction: " << *Ty << "\n";
763     abort();
764   }
765
766   return Dest;
767 }
768
769 void Interpreter::visitCastInst(CastInst &I) {
770   ExecutionContext &SF = ECStack.back();
771   SetValue(&I, executeCastOperation(I.getOperand(0), I.getType(), SF), SF);
772 }
773
774 void Interpreter::visitVANextInst(VANextInst &I) {
775   ExecutionContext &SF = ECStack.back();
776
777   // Get the incoming valist element.  LLI treats the valist as an integer.
778   GenericValue VAList = getOperandValue(I.getOperand(0), SF);
779   
780   // Move to the next operand.
781   unsigned Argument = VAList.IntVal++;
782   assert(Argument < SF.VarArgs.size() &&
783          "Accessing past the last vararg argument!");
784   SetValue(&I, VAList, SF);
785 }
786
787 //===----------------------------------------------------------------------===//
788 //                        Dispatch and Execution Code
789 //===----------------------------------------------------------------------===//
790
791 //===----------------------------------------------------------------------===//
792 // callFunction - Execute the specified function...
793 //
794 void Interpreter::callFunction(Function *F,
795                                const std::vector<GenericValue> &ArgVals) {
796   assert((ECStack.empty() || ECStack.back().Caller == 0 || 
797           ECStack.back().Caller->getNumOperands()-1 == ArgVals.size()) &&
798          "Incorrect number of arguments passed into function call!");
799   if (F->isExternal()) {
800     GenericValue Result = callExternalFunction(F, ArgVals);
801     const Type *RetTy = F->getReturnType();
802
803     // Copy the result back into the result variable if we are not returning
804     // void.
805     if (RetTy != Type::VoidTy) {
806       if (!ECStack.empty() && ECStack.back().Caller) {
807         ExecutionContext &SF = ECStack.back();
808         SetValue(SF.Caller, Result, SF);
809         SF.Caller = 0;          // We returned from the call...
810       }
811     }
812
813     return;
814   }
815
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   StackFrame.CurBB     = F->begin();
821   StackFrame.CurInst   = StackFrame.CurBB->begin();
822
823   // Run through the function arguments and initialize their values...
824   assert((ArgVals.size() == F->asize() ||
825          (ArgVals.size() > F->asize() && F->getFunctionType()->isVarArg())) &&
826          "Invalid number of values passed to function invocation!");
827
828   // Handle non-varargs arguments...
829   unsigned i = 0;
830   for (Function::aiterator AI = F->abegin(), E = F->aend(); AI != E; ++AI, ++i)
831     SetValue(AI, ArgVals[i], StackFrame);
832
833   // Handle varargs arguments...
834   StackFrame.VarArgs.assign(ArgVals.begin()+i, ArgVals.end());
835 }
836
837 void Interpreter::run() {
838   while (!ECStack.empty()) {
839     // Interpret a single instruction & increment the "PC".
840     ExecutionContext &SF = ECStack.back();  // Current stack frame
841     Instruction &I = *SF.CurInst++;         // Increment before execute
842     
843     // Track the number of dynamic instructions executed.
844     ++NumDynamicInsts;
845
846     visit(I);   // Dispatch to one of the visit* methods...
847   }
848 }