From 59fecec7d0c4aded45300cebd2cf45dd196edec0 Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Sat, 8 Apr 2006 04:09:19 +0000 Subject: [PATCH] use isValidOperands instead of duplicating checks git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@27527 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Bytecode/Reader/Reader.cpp | 37 ++++++++++++++++++++++------------ 1 file changed, 24 insertions(+), 13 deletions(-) diff --git a/lib/Bytecode/Reader/Reader.cpp b/lib/Bytecode/Reader/Reader.cpp index d2acda0ed15..36b2ecb37fd 100644 --- a/lib/Bytecode/Reader/Reader.cpp +++ b/lib/Bytecode/Reader/Reader.cpp @@ -719,19 +719,27 @@ void BytecodeReader::ParseInstruction(std::vector &Oprnds, case Instruction::ExtractElement: { if (Oprnds.size() != 2) throw std::string("Invalid extractelement instruction!"); - Result = new ExtractElementInst(getValue(iType, Oprnds[0]), - getValue(Type::UIntTyID, Oprnds[1])); + Value *V1 = getValue(iType, Oprnds[0]); + Value *V2 = getValue(Type::UIntTyID, Oprnds[1]); + + if (!ExtractElementInst::isValidOperands(V1, V2)) + throw std::string("Invalid extractelement instruction!"); + + Result = new ExtractElementInst(V1, V2); break; } case Instruction::InsertElement: { const PackedType *PackedTy = dyn_cast(InstTy); if (!PackedTy || Oprnds.size() != 3) throw std::string("Invalid insertelement instruction!"); - Result = - new InsertElementInst(getValue(iType, Oprnds[0]), - getValue(getTypeSlot(PackedTy->getElementType()), - Oprnds[1]), - getValue(Type::UIntTyID, Oprnds[2])); + + Value *V1 = getValue(iType, Oprnds[0]); + Value *V2 = getValue(getTypeSlot(PackedTy->getElementType()), Oprnds[1]); + Value *V3 = getValue(Type::UIntTyID, Oprnds[2]); + + if (!InsertElementInst::isValidOperands(V1, V2, V3)) + throw std::string("Invalid insertelement instruction!"); + Result = new InsertElementInst(V1, V2, V3); break; } case Instruction::ShuffleVector: { @@ -1495,22 +1503,25 @@ Value *BytecodeReader::ParseConstantPoolValue(unsigned TypeID) { if (Handler) Handler->handleConstantExpression(Opcode, ArgVec, Result); return Result; } else if (Opcode == Instruction::ExtractElement) { - if (ArgVec.size() != 2) - error("ExtractElement instruction must have two arguments."); + if (ArgVec.size() != 2 || + !ExtractElementInst::isValidOperands(ArgVec[0], ArgVec[1])) + error("Invalid extractelement constand expr arguments"); Constant* Result = ConstantExpr::getExtractElement(ArgVec[0], ArgVec[1]); if (Handler) Handler->handleConstantExpression(Opcode, ArgVec, Result); return Result; } else if (Opcode == Instruction::InsertElement) { - if (ArgVec.size() != 3) - error("InsertElement instruction must have three arguments."); - Constant* Result = + if (ArgVec.size() != 3 || + !InsertElementInst::isValidOperands(ArgVec[0], ArgVec[1], ArgVec[2])) + error("Invalid insertelement constand expr arguments"); + + Constant *Result = ConstantExpr::getInsertElement(ArgVec[0], ArgVec[1], ArgVec[2]); if (Handler) Handler->handleConstantExpression(Opcode, ArgVec, Result); return Result; } else if (Opcode == Instruction::ShuffleVector) { if (ArgVec.size() != 3 || !ShuffleVectorInst::isValidOperands(ArgVec[0], ArgVec[1], ArgVec[2])) - error("shufflevector constant expr must have three arguments."); + error("Invalid shufflevector constant expr arguments."); Constant *Result = ConstantExpr::getShuffleVector(ArgVec[0], ArgVec[1], ArgVec[2]); if (Handler) Handler->handleConstantExpression(Opcode, ArgVec, Result);