10b57cec5SDimitry Andric //===-- RegisterNumber.cpp ------------------------------------------------===//
20b57cec5SDimitry Andric //
30b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
40b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
50b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
60b57cec5SDimitry Andric //
70b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
80b57cec5SDimitry Andric 
90b57cec5SDimitry Andric #include "lldb/Target/RegisterNumber.h"
100b57cec5SDimitry Andric #include "lldb/Target/RegisterContext.h"
110b57cec5SDimitry Andric #include "lldb/Target/Thread.h"
120b57cec5SDimitry Andric 
130b57cec5SDimitry Andric using namespace lldb_private;
140b57cec5SDimitry Andric 
RegisterNumber(lldb_private::Thread & thread,lldb::RegisterKind kind,uint32_t num)150b57cec5SDimitry Andric RegisterNumber::RegisterNumber(lldb_private::Thread &thread,
160b57cec5SDimitry Andric                                lldb::RegisterKind kind, uint32_t num)
170b57cec5SDimitry Andric     : m_reg_ctx_sp(thread.GetRegisterContext()), m_regnum(num), m_kind(kind),
180b57cec5SDimitry Andric       m_kind_regnum_map(), m_name("") {
190b57cec5SDimitry Andric   if (m_reg_ctx_sp.get()) {
200b57cec5SDimitry Andric     const lldb_private::RegisterInfo *reginfo =
210b57cec5SDimitry Andric         m_reg_ctx_sp->GetRegisterInfoAtIndex(
220b57cec5SDimitry Andric             GetAsKind(lldb::eRegisterKindLLDB));
230b57cec5SDimitry Andric     if (reginfo && reginfo->name) {
240b57cec5SDimitry Andric       m_name = reginfo->name;
250b57cec5SDimitry Andric     }
260b57cec5SDimitry Andric   }
270b57cec5SDimitry Andric }
280b57cec5SDimitry Andric 
RegisterNumber()290b57cec5SDimitry Andric RegisterNumber::RegisterNumber() : m_reg_ctx_sp(), m_kind_regnum_map() {}
300b57cec5SDimitry Andric 
init(lldb_private::Thread & thread,lldb::RegisterKind kind,uint32_t num)310b57cec5SDimitry Andric void RegisterNumber::init(lldb_private::Thread &thread, lldb::RegisterKind kind,
320b57cec5SDimitry Andric                           uint32_t num) {
330b57cec5SDimitry Andric   m_reg_ctx_sp = thread.GetRegisterContext();
340b57cec5SDimitry Andric   m_regnum = num;
350b57cec5SDimitry Andric   m_kind = kind;
360b57cec5SDimitry Andric   if (m_reg_ctx_sp.get()) {
370b57cec5SDimitry Andric     const lldb_private::RegisterInfo *reginfo =
380b57cec5SDimitry Andric         m_reg_ctx_sp->GetRegisterInfoAtIndex(
390b57cec5SDimitry Andric             GetAsKind(lldb::eRegisterKindLLDB));
400b57cec5SDimitry Andric     if (reginfo && reginfo->name) {
410b57cec5SDimitry Andric       m_name = reginfo->name;
420b57cec5SDimitry Andric     }
430b57cec5SDimitry Andric   }
440b57cec5SDimitry Andric }
450b57cec5SDimitry Andric 
operator =(const RegisterNumber & rhs)460b57cec5SDimitry Andric const RegisterNumber &RegisterNumber::operator=(const RegisterNumber &rhs) {
470b57cec5SDimitry Andric   m_reg_ctx_sp = rhs.m_reg_ctx_sp;
480b57cec5SDimitry Andric   m_regnum = rhs.m_regnum;
490b57cec5SDimitry Andric   m_kind = rhs.m_kind;
500b57cec5SDimitry Andric   for (auto it : rhs.m_kind_regnum_map)
510b57cec5SDimitry Andric     m_kind_regnum_map[it.first] = it.second;
520b57cec5SDimitry Andric   m_name = rhs.m_name;
530b57cec5SDimitry Andric   return *this;
540b57cec5SDimitry Andric }
550b57cec5SDimitry Andric 
operator ==(RegisterNumber & rhs)560b57cec5SDimitry Andric bool RegisterNumber::operator==(RegisterNumber &rhs) {
570b57cec5SDimitry Andric   if (IsValid() != rhs.IsValid())
580b57cec5SDimitry Andric     return false;
590b57cec5SDimitry Andric 
600b57cec5SDimitry Andric   if (m_kind == rhs.m_kind) {
610b57cec5SDimitry Andric     return m_regnum == rhs.m_regnum;
620b57cec5SDimitry Andric   }
630b57cec5SDimitry Andric 
640b57cec5SDimitry Andric   uint32_t rhs_regnum = rhs.GetAsKind(m_kind);
650b57cec5SDimitry Andric   if (rhs_regnum != LLDB_INVALID_REGNUM) {
660b57cec5SDimitry Andric     return m_regnum == rhs_regnum;
670b57cec5SDimitry Andric   }
680b57cec5SDimitry Andric   uint32_t lhs_regnum = GetAsKind(rhs.m_kind);
690b57cec5SDimitry Andric   { return lhs_regnum == rhs.m_regnum; }
700b57cec5SDimitry Andric   return false;
710b57cec5SDimitry Andric }
720b57cec5SDimitry Andric 
operator !=(RegisterNumber & rhs)730b57cec5SDimitry Andric bool RegisterNumber::operator!=(RegisterNumber &rhs) { return !(*this == rhs); }
740b57cec5SDimitry Andric 
IsValid() const750b57cec5SDimitry Andric bool RegisterNumber::IsValid() const {
760b57cec5SDimitry Andric   return m_reg_ctx_sp.get() && m_kind != lldb::kNumRegisterKinds &&
770b57cec5SDimitry Andric          m_regnum != LLDB_INVALID_REGNUM;
780b57cec5SDimitry Andric }
790b57cec5SDimitry Andric 
GetAsKind(lldb::RegisterKind kind)800b57cec5SDimitry Andric uint32_t RegisterNumber::GetAsKind(lldb::RegisterKind kind) {
810b57cec5SDimitry Andric   if (m_regnum == LLDB_INVALID_REGNUM)
820b57cec5SDimitry Andric     return LLDB_INVALID_REGNUM;
830b57cec5SDimitry Andric 
840b57cec5SDimitry Andric   if (kind == m_kind)
850b57cec5SDimitry Andric     return m_regnum;
860b57cec5SDimitry Andric 
870b57cec5SDimitry Andric   Collection::iterator iter = m_kind_regnum_map.find(kind);
880b57cec5SDimitry Andric   if (iter != m_kind_regnum_map.end()) {
890b57cec5SDimitry Andric     return iter->second;
900b57cec5SDimitry Andric   }
910b57cec5SDimitry Andric   uint32_t output_regnum = LLDB_INVALID_REGNUM;
920b57cec5SDimitry Andric   if (m_reg_ctx_sp &&
930b57cec5SDimitry Andric       m_reg_ctx_sp->ConvertBetweenRegisterKinds(m_kind, m_regnum, kind,
940b57cec5SDimitry Andric                                                 output_regnum) &&
950b57cec5SDimitry Andric       output_regnum != LLDB_INVALID_REGNUM) {
960b57cec5SDimitry Andric     m_kind_regnum_map[kind] = output_regnum;
970b57cec5SDimitry Andric   }
980b57cec5SDimitry Andric   return output_regnum;
990b57cec5SDimitry Andric }
1000b57cec5SDimitry Andric 
GetRegisterNumber() const1010b57cec5SDimitry Andric uint32_t RegisterNumber::GetRegisterNumber() const { return m_regnum; }
1020b57cec5SDimitry Andric 
GetRegisterKind() const1030b57cec5SDimitry Andric lldb::RegisterKind RegisterNumber::GetRegisterKind() const { return m_kind; }
1040b57cec5SDimitry Andric 
GetName()1050b57cec5SDimitry Andric const char *RegisterNumber::GetName() { return m_name; }
1060b57cec5SDimitry Andric