diff --git a/lldb/source/Plugins/Instruction/RISCV/EmulateInstructionRISCV.cpp b/lldb/source/Plugins/Instruction/RISCV/EmulateInstructionRISCV.cpp index 2957cb716041..4f4e6779072d 100644 --- a/lldb/source/Plugins/Instruction/RISCV/EmulateInstructionRISCV.cpp +++ b/lldb/source/Plugins/Instruction/RISCV/EmulateInstructionRISCV.cpp @@ -7,6 +7,8 @@ //===----------------------------------------------------------------------===// #include "EmulateInstructionRISCV.h" +#include "Plugins/Process/Utility/RegisterInfoInterface.h" +#include "Plugins/Process/Utility/RegisterInfoPOSIX_riscv32.h" #include "Plugins/Process/Utility/RegisterInfoPOSIX_riscv64.h" #include "Plugins/Process/Utility/lldb-riscv-register-enums.h" #include "RISCVCInstructions.h" @@ -1837,10 +1839,23 @@ EmulateInstructionRISCV::GetRegisterInfo(RegisterKind reg_kind, } } - RegisterInfoPOSIX_riscv64 reg_info(m_arch, - RegisterInfoPOSIX_riscv64::eRegsetMaskAll); - const RegisterInfo *array = reg_info.GetRegisterInfo(); - const uint32_t length = reg_info.GetRegisterCount(); + std::unique_ptr reg_info; + switch (m_arch.GetTriple().getArch()) { + case llvm::Triple::riscv32: + reg_info = std::make_unique( + m_arch, RegisterInfoPOSIX_riscv32::eRegsetMaskAll); + break; + case llvm::Triple::riscv64: + reg_info = std::make_unique( + m_arch, RegisterInfoPOSIX_riscv64::eRegsetMaskAll); + break; + default: + assert(false && "unsupported triple"); + return {}; + } + + const RegisterInfo *array = reg_info->GetRegisterInfo(); + const uint32_t length = reg_info->GetRegisterCount(); if (reg_index >= length || reg_kind != eRegisterKindLLDB) return {}; diff --git a/lldb/unittests/Instruction/RISCV/TestRISCVEmulator.cpp b/lldb/unittests/Instruction/RISCV/TestRISCVEmulator.cpp index 90d5a7c4f3b9..f713755c77a7 100644 --- a/lldb/unittests/Instruction/RISCV/TestRISCVEmulator.cpp +++ b/lldb/unittests/Instruction/RISCV/TestRISCVEmulator.cpp @@ -16,6 +16,7 @@ #include "lldb/Utility/RegisterValue.h" #include "Plugins/Instruction/RISCV/EmulateInstructionRISCV.h" +#include "Plugins/Process/Utility/RegisterInfoPOSIX_riscv32.h" #include "Plugins/Process/Utility/RegisterInfoPOSIX_riscv64.h" #include "Plugins/Process/Utility/lldb-riscv-register-enums.h" @@ -806,3 +807,19 @@ TEST_F(RISCVEmulatorTester, TestFMV_D_XInst) { ASSERT_TRUE(this->Execute(*decode, false)); ASSERT_EQ(this->fpr.fpr[DecodeRD(FMV_D_XInst)], bits); } + +TEST_F(RISCVEmulatorTester, TestGetRegisterInfoRV64) { + // Test that GetRegisterInfo returns valid register info for riscv64. + auto reg_info = this->GetRegisterInfo(eRegisterKindLLDB, gpr_x1_riscv); + ASSERT_TRUE(reg_info.has_value()); + ASSERT_EQ(reg_info->byte_size, 8u); + ASSERT_STREQ(reg_info->name, "ra"); +} + +TEST_F(RISCVEmulatorTester32, TestGetRegisterInfoRV32) { + // Test that GetRegisterInfo returns valid register info for riscv32. + auto reg_info = this->GetRegisterInfo(eRegisterKindLLDB, gpr_x1_riscv); + ASSERT_TRUE(reg_info.has_value()); + ASSERT_EQ(reg_info->byte_size, 4u); + ASSERT_STREQ(reg_info->name, "ra"); +}