diff options
Diffstat (limited to 'lib/Bitcode/Writer/BitcodeWriter.cpp')
| -rw-r--r-- | lib/Bitcode/Writer/BitcodeWriter.cpp | 29 | 
1 files changed, 20 insertions, 9 deletions
diff --git a/lib/Bitcode/Writer/BitcodeWriter.cpp b/lib/Bitcode/Writer/BitcodeWriter.cpp index 037854e26fa4..af0b8acd44ce 100644 --- a/lib/Bitcode/Writer/BitcodeWriter.cpp +++ b/lib/Bitcode/Writer/BitcodeWriter.cpp @@ -19,6 +19,7 @@  #include "llvm/DerivedTypes.h"  #include "llvm/InlineAsm.h"  #include "llvm/Instructions.h" +#include "llvm/LLVMContext.h"  #include "llvm/Metadata.h"  #include "llvm/Module.h"  #include "llvm/Operator.h" @@ -750,10 +751,11 @@ static void WriteConstants(unsigned FirstVal, unsigned LastVal,          assert (0 && "Unknown FP type!");        }      } else if (isa<ConstantArray>(C) && cast<ConstantArray>(C)->isString()) { +      const ConstantArray *CA = cast<ConstantArray>(C);        // Emit constant strings specially. -      unsigned NumOps = C->getNumOperands(); +      unsigned NumOps = CA->getNumOperands();        // If this is a null-terminated string, use the denser CSTRING encoding. -      if (C->getOperand(NumOps-1)->isNullValue()) { +      if (CA->getOperand(NumOps-1)->isNullValue()) {          Code = bitc::CST_CODE_CSTRING;          --NumOps;  // Don't encode the null, which isn't allowed by char6.        } else { @@ -763,7 +765,7 @@ static void WriteConstants(unsigned FirstVal, unsigned LastVal,        bool isCStr7 = Code == bitc::CST_CODE_CSTRING;        bool isCStrChar6 = Code == bitc::CST_CODE_CSTRING;        for (unsigned i = 0; i != NumOps; ++i) { -        unsigned char V = cast<ConstantInt>(C->getOperand(i))->getZExtValue(); +        unsigned char V = cast<ConstantInt>(CA->getOperand(i))->getZExtValue();          Record.push_back(V);          isCStr7 &= (V & 128) == 0;          if (isCStrChar6) @@ -851,6 +853,13 @@ static void WriteConstants(unsigned FirstVal, unsigned LastVal,          Record.push_back(CE->getPredicate());          break;        } +    } else if (const BlockAddress *BA = dyn_cast<BlockAddress>(C)) { +      assert(BA->getFunction() == BA->getBasicBlock()->getParent() && +             "Malformed blockaddress"); +      Code = bitc::CST_CODE_BLOCKADDRESS; +      Record.push_back(VE.getTypeID(BA->getFunction()->getType())); +      Record.push_back(VE.getValueID(BA->getFunction())); +      Record.push_back(VE.getGlobalBasicBlockID(BA->getBasicBlock()));      } else {        llvm_unreachable("Unknown constant!");      } @@ -1000,7 +1009,7 @@ static void WriteInstruction(const Instruction &I, unsigned InstID,    case Instruction::Br:      {        Code = bitc::FUNC_CODE_INST_BR; -      BranchInst &II(cast<BranchInst>(I)); +      BranchInst &II = cast<BranchInst>(I);        Vals.push_back(VE.getValueID(II.getSuccessor(0)));        if (II.isConditional()) {          Vals.push_back(VE.getValueID(II.getSuccessor(1))); @@ -1014,6 +1023,13 @@ static void WriteInstruction(const Instruction &I, unsigned InstID,      for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i)        Vals.push_back(VE.getValueID(I.getOperand(i)));      break; +  case Instruction::IndirectBr: +    Code = bitc::FUNC_CODE_INST_INDIRECTBR; +    Vals.push_back(VE.getTypeID(I.getOperand(0)->getType())); +    for (unsigned i = 0, e = I.getNumOperands(); i != e; ++i) +      Vals.push_back(VE.getValueID(I.getOperand(i))); +    break; +          case Instruction::Invoke: {      const InvokeInst *II = cast<InvokeInst>(&I);      const Value *Callee(II->getCalledValue()); @@ -1054,11 +1070,6 @@ static void WriteInstruction(const Instruction &I, unsigned InstID,        Vals.push_back(VE.getValueID(I.getOperand(i)));      break; -  case Instruction::Free: -    Code = bitc::FUNC_CODE_INST_FREE; -    PushValueAndType(I.getOperand(0), InstID, Vals, VE); -    break; -    case Instruction::Alloca:      Code = bitc::FUNC_CODE_INST_ALLOCA;      Vals.push_back(VE.getTypeID(I.getType()));  | 
