lib/CodeGen/MachineInstrAnnot.cpp

   1 //===-- MachineInstrAnnot.cpp ---------------------------------------------===//
   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 defines Annotations used to pass information between code
  11 //  generation phases.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/CodeGen/MachineInstrAnnot.h"
  16 #include "llvm/CodeGen/InstrSelection.h"
  17 #include "llvm/CodeGen/MachineCodeForInstruction.h"
  18 #include "llvm/iOther.h"
  19 #include "llvm/Type.h"
  20
  21
  22 CallArgsDescriptor::CallArgsDescriptor(CallInst* _callInstr,
  23                                        TmpInstruction* _retAddrReg,
  24                                        bool _isVarArgs, bool _noPrototype)
  25   : callInstr(_callInstr),
  26     funcPtr(isa<Function>(_callInstr->getCalledValue())
  27             ? NULL : _callInstr->getCalledValue()),
  28     retAddrReg(_retAddrReg),
  29     isVarArgs(_isVarArgs),
  30     noPrototype(_noPrototype)
  31 {
  32   unsigned int numArgs = callInstr->getNumOperands();
  33   argInfoVec.reserve(numArgs);
  34   assert(callInstr->getOperand(0) == callInstr->getCalledValue()
  35          && "Operand 0 is ignored in the loop below!");
  36   for (unsigned int i=1; i < numArgs; ++i)
  37     argInfoVec.push_back(CallArgInfo(callInstr->getOperand(i)));
  38
  39   // Enter this object in the MachineCodeForInstr object of the CallInst.
  40   // This transfers ownership of this object.
  41   MachineCodeForInstruction::get(callInstr).setCallArgsDescriptor(this);
  42 }
  43
  44
  45 CallInst*
  46 CallArgsDescriptor::getReturnValue() const
  47 {
  48   return (callInstr->getType() == Type::VoidTy? NULL : callInstr);
  49 }
  50
  51
  52 // Mechanism to get the descriptor for a CALL MachineInstr.
  53 // We get the LLVM CallInstr from the ret. addr. register argument
  54 // of the CALL MachineInstr (which is explicit operand #3 for indirect
  55 // calls or the last implicit operand for direct calls).  We then get
  56 // the CallArgsDescriptor from the MachineCodeForInstruction object for
  57 // the CallInstr.
  58 // This is roundabout but avoids adding a new map or annotation just
  59 // to keep track of CallArgsDescriptors.
  60 //
  61 CallArgsDescriptor *CallArgsDescriptor::get(const MachineInstr* MI)
  62 {
  63   const TmpInstruction* retAddrReg =
  64     cast<TmpInstruction>(isa<Function>(MI->getOperand(0).getVRegValue())
  65                          ? MI->getImplicitRef(MI->getNumImplicitRefs()-1)
  66                          : MI->getOperand(2).getVRegValue());
  67
  68   assert(retAddrReg->getNumOperands() == 1 &&
  69          isa<CallInst>(retAddrReg->getOperand(0)) &&
  70          "Location of callInstr arg for CALL instr. changed? FIX THIS CODE!");
  71
  72   const CallInst* callInstr = cast<CallInst>(retAddrReg->getOperand(0));
  73
  74   CallArgsDescriptor* desc =
  75     MachineCodeForInstruction::get(callInstr).getCallArgsDescriptor();
  76   assert(desc->getCallInst()==callInstr && "Incorrect call args descriptor?");
  77   return desc;
  78 }