diff --git a/bolt/include/bolt/Core/MCPlusBuilder.h b/bolt/include/bolt/Core/MCPlusBuilder.h --- a/bolt/include/bolt/Core/MCPlusBuilder.h +++ b/bolt/include/bolt/Core/MCPlusBuilder.h @@ -74,9 +74,10 @@ AllocatorIdTy MaxAllocatorId = 0; /// We encode Index and Value into a 64-bit immediate operand value. - static int64_t encodeAnnotationImm(unsigned Index, int64_t Value) { - assert(Index < 256 && "annotation index max value exceeded"); - assert((Value == (Value << 8) >> 8) && "annotation value out of range"); + static int64_t encodeAnnotationImm(uint8_t Index, int64_t Value) { + if (LLVM_UNLIKELY(Value != + llvm::SignExtend64<56>(Value & 0xff'ffff'ffff'ffff))) + llvm::report_fatal_error("annotation value out of range"); Value &= 0xff'ffff'ffff'ffff; Value |= (int64_t)Index << 56; @@ -85,14 +86,13 @@ } /// Extract annotation index from immediate operand value. - static unsigned extractAnnotationIndex(int64_t ImmValue) { + static uint8_t extractAnnotationIndex(int64_t ImmValue) { return ImmValue >> 56; } /// Extract annotation value from immediate operand value. static int64_t extractAnnotationValue(int64_t ImmValue) { - ImmValue &= 0xff'ffff'ffff'ffff; - return (ImmValue << 8) >> 8; + return llvm::SignExtend64<56>(ImmValue & 0xff'ffff'ffff'ffffULL); } MCInst *getAnnotationInst(const MCInst &Inst) const { diff --git a/bolt/unittests/Core/MCPlusBuilder.cpp b/bolt/unittests/Core/MCPlusBuilder.cpp --- a/bolt/unittests/Core/MCPlusBuilder.cpp +++ b/bolt/unittests/Core/MCPlusBuilder.cpp @@ -109,4 +109,32 @@ testRegAliases(Triple::x86_64, X86::AX, AliasesAX, AliasesAXCount, true); } +TEST_P(MCPlusBuilderTester, Annotation) { + if (GetParam() != Triple::x86_64) + GTEST_SKIP(); + MCInst Inst; + bool Success = BC->MIB->createCall(Inst, BC->Ctx->createNamedTempSymbol(), + BC->Ctx.get()); + ASSERT_TRUE(Success); + MCSymbol *LPSymbol = BC->Ctx->createNamedTempSymbol("LP"); + uint64_t Value = INT32_MIN; + // Test encodeAnnotationImm using this indirect way + BC->MIB->addEHInfo(Inst, MCPlus::MCLandingPad(LPSymbol, Value)); + // Round-trip encoding-decoding check for negative values + Optional EHInfo = BC->MIB->getEHInfo(Inst); + ASSERT_TRUE(EHInfo.hasValue()); + MCPlus::MCLandingPad LP = EHInfo.getValue(); + uint64_t DecodedValue = LP.second; + ASSERT_EQ(Value, DecodedValue); + + // Large int64 should trigger an out of range assertion + Value = 0x1FF'FFFF'FFFF'FFFFULL; + MCInst Inst2; + bool Success2 = BC->MIB->createCall(Inst2, BC->Ctx->createNamedTempSymbol(), + BC->Ctx.get()); + ASSERT_TRUE(Success2); + ASSERT_DEATH(BC->MIB->addEHInfo(Inst2, MCPlus::MCLandingPad(LPSymbol, Value)), + "annotation value out of range"); +} + #endif // X86_AVAILABLE