Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2024-11-23 13:50:11 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[lldb][NFCI] Remove use of ConstString in StructuredData

Browse Source 
The remaining use of ConstString in StructuredData is the Dictionary
class. Internally it's backed by a `std::map<ConstString, ObjectSP>`.
I propose that we replace it with a `llvm::StringMap<ObjectSP>`.

Many StructuredData::Dictionary objects are ephemeral and only exist for
a short amount of time. Many of these Dictionaries are only produced
once and are never used again. That leaves us with a lot of string data
in the ConstString StringPool that is sitting there never to be used
again. Even if the same string is used many times for keys of different
Dictionary objects, that is something we can measure and adjust for
instead of assuming that every key may be reused at some point in the
future.

Quick comparisons of key data is likely not a concern with Dictionary,
but the use of `llvm::StringMap` means that lookups should be fast with
its hashing strategy.

Switching to a llvm::StringMap meant that the iteration order may be
different. To account for this when serializing/dumping the dictionary,
I added some code to sort the output by key before emitting anything.

Differential Revision: https://reviews.llvm.org/D159313
This commit is contained in:
Alex Langford 2023-08-29 17:22:48 -07:00
parent e2d39f799b
commit a5a2a5a3ec
6 changed files with 62 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
lldb/include/lldb/Utility/StructuredData.h
View File

@ -9,10 +9,10 @@
#ifndef LLDB_UTILITY_STRUCTUREDDATA_H
#define LLDB_UTILITY_STRUCTUREDDATA_H
#include "llvm/ADT/StringMap.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/JSON.h"
#include "lldb/Utility/ConstString.h"
#include "lldb/Utility/FileSpec.h"
#include "lldb/Utility/Stream.h"
#include "lldb/lldb-enumerations.h"
@ -423,34 +423,25 @@ public:
size_t GetSize() const { return m_dict.size(); }
void ForEach(std::function<bool(ConstString key, Object *object)> const
void ForEach(std::function<bool(llvm::StringRef key, Object *object)> const
&callback) const {
for (const auto &pair : m_dict) {
if (!callback(pair.first, pair.second.get()))
if (!callback(pair.first(), pair.second.get()))
break;
}
}
ArraySP GetKeys() const {
auto array_sp = std::make_shared<Array>();
collection::const_iterator iter;
for (iter = m_dict.begin(); iter != m_dict.end(); ++iter) {
auto key_object_sp = std::make_shared<String>();
key_object_sp->SetValue(iter->first.AsCString());
for (auto iter = m_dict.begin(); iter != m_dict.end(); ++iter) {
auto key_object_sp = std::make_shared<String>(iter->first());
array_sp->Push(key_object_sp);
}
return array_sp;
}
ObjectSP GetValueForKey(llvm::StringRef key) const {
ObjectSP value_sp;
if (!key.empty()) {
ConstString key_cs(key);
collection::const_iterator iter = m_dict.find(key_cs);
if (iter != m_dict.end())
value_sp = iter->second;
}
return value_sp;
return m_dict.lookup(key);
}
bool GetValueForKeyAsBoolean(llvm::StringRef key, bool &result) const {
@ -539,15 +530,10 @@ public:
return false;
}
bool HasKey(llvm::StringRef key) const {
ConstString key_cs(key);
collection::const_iterator search = m_dict.find(key_cs);
return search != m_dict.end();
}
bool HasKey(llvm::StringRef key) const { return m_dict.contains(key); }
void AddItem(llvm::StringRef key, ObjectSP value_sp) {
ConstString key_cs(key);
m_dict[key_cs] = std::move(value_sp);
m_dict.insert_or_assign(key, std::move(value_sp));
}
template <typename T> void AddIntegerItem(llvm::StringRef key, T value) {
@ -577,8 +563,7 @@ public:
void GetDescription(lldb_private::Stream &s) const override;
protected:
typedef std::map<ConstString, ObjectSP> collection;
collection m_dict;
llvm::StringMap<ObjectSP> m_dict;
};
class Null : public Object {

4
lldb/source/Plugins/Platform/MacOSX/PlatformDarwin.cpp
View File

@ -989,7 +989,7 @@ PlatformDarwin::ExtractAppSpecificInfo(Process &process) {
StructuredData::DictionarySP dict_sp =
std::make_shared<StructuredData::Dictionary>();
auto flatten_asi_dict = [&dict_sp](ConstString key,
auto flatten_asi_dict = [&dict_sp](llvm::StringRef key,
StructuredData::Object *val) -> bool {
if (!val)
return false;
@ -998,7 +998,7 @@ PlatformDarwin::ExtractAppSpecificInfo(Process &process) {
if (!arr || !arr->GetSize())
return false;
dict_sp->AddItem(key.AsCString(), arr->GetItemAtIndex(0));
dict_sp->AddItem(key, arr->GetItemAtIndex(0));
return true;
};

39
lldb/source/Plugins/Process/gdb-remote/ProcessGDBRemote.cpp
View File

@ -1926,24 +1926,23 @@ ThreadSP ProcessGDBRemote::SetThreadStopInfo(
lldb::ThreadSP
ProcessGDBRemote::SetThreadStopInfo(StructuredData::Dictionary *thread_dict) {
static ConstString g_key_tid("tid");
static ConstString g_key_name("name");
static ConstString g_key_reason("reason");
static ConstString g_key_metype("metype");
static ConstString g_key_medata("medata");
static ConstString g_key_qaddr("qaddr");
static ConstString g_key_dispatch_queue_t("dispatch_queue_t");
static ConstString g_key_associated_with_dispatch_queue(
static constexpr llvm::StringLiteral g_key_tid("tid");
static constexpr llvm::StringLiteral g_key_name("name");
static constexpr llvm::StringLiteral g_key_reason("reason");
static constexpr llvm::StringLiteral g_key_metype("metype");
static constexpr llvm::StringLiteral g_key_medata("medata");
static constexpr llvm::StringLiteral g_key_qaddr("qaddr");
static constexpr llvm::StringLiteral g_key_dispatch_queue_t(
"dispatch_queue_t");
static constexpr llvm::StringLiteral g_key_associated_with_dispatch_queue(
"associated_with_dispatch_queue");
static ConstString g_key_queue_name("qname");
static ConstString g_key_queue_kind("qkind");
static ConstString g_key_queue_serial_number("qserialnum");
static ConstString g_key_registers("registers");
static ConstString g_key_memory("memory");
static ConstString g_key_address("address");
static ConstString g_key_bytes("bytes");
static ConstString g_key_description("description");
static ConstString g_key_signal("signal");
static constexpr llvm::StringLiteral g_key_queue_name("qname");
static constexpr llvm::StringLiteral g_key_queue_kind("qkind");
static constexpr llvm::StringLiteral g_key_queue_serial_number("qserialnum");
static constexpr llvm::StringLiteral g_key_registers("registers");
static constexpr llvm::StringLiteral g_key_memory("memory");
static constexpr llvm::StringLiteral g_key_description("description");
static constexpr llvm::StringLiteral g_key_signal("signal");
// Stop with signal and thread info
lldb::tid_t tid = LLDB_INVALID_THREAD_ID;
@ -1971,7 +1970,7 @@ ProcessGDBRemote::SetThreadStopInfo(StructuredData::Dictionary *thread_dict) {
&thread_dispatch_qaddr, &queue_vars_valid,
&associated_with_dispatch_queue, &dispatch_queue_t,
&queue_name, &queue_kind, &queue_serial_number](
ConstString key,
llvm::StringRef key,
StructuredData::Object *object) -> bool {
if (key == g_key_tid) {
// thread in big endian hex
@ -2029,10 +2028,10 @@ ProcessGDBRemote::SetThreadStopInfo(StructuredData::Dictionary *thread_dict) {
if (registers_dict) {
registers_dict->ForEach(
[&expedited_register_map](ConstString key,
[&expedited_register_map](llvm::StringRef key,
StructuredData::Object *object) -> bool {
uint32_t reg;
if (llvm::to_integer(key.AsCString(), reg))
if (llvm::to_integer(key, reg))
expedited_register_map[reg] =
std::string(object->GetStringValue());
return true; // Keep iterating through all array items

6
lldb/source/Plugins/SymbolVendor/MacOSX/SymbolVendorMacOSX.cpp
View File

@ -224,7 +224,7 @@ SymbolVendorMacOSX::CreateInstance(const lldb::ModuleSP &module_sp,
[&module_sp, new_style_source_remapping_dictionary,
original_DBGSourcePath_value,
do_truncate_remapping_names](
ConstString key,
llvm::StringRef key,
StructuredData::Object *object) -> bool {
if (object && object->GetAsString()) {
@ -237,7 +237,7 @@ SymbolVendorMacOSX::CreateInstance(const lldb::ModuleSP &module_sp,
DBGSourcePath = original_DBGSourcePath_value;
}
module_sp->GetSourceMappingList().Append(
key.GetStringRef(), DBGSourcePath, true);
key, DBGSourcePath, true);
// With version 2 of DBGSourcePathRemapping, we
// can chop off the last two filename parts
// from the source remapping and get a more
@ -245,7 +245,7 @@ SymbolVendorMacOSX::CreateInstance(const lldb::ModuleSP &module_sp,
// Add this as another option in addition to
// the full source path remap.
if (do_truncate_remapping_names) {
FileSpec build_path(key.AsCString());
FileSpec build_path(key);
FileSpec source_path(DBGSourcePath.c_str());
build_path.RemoveLastPathComponent();
build_path.RemoveLastPathComponent();

5
lldb/source/Target/Target.cpp
View File

@ -3859,11 +3859,10 @@ void Target::StopHookScripted::GetSubclassDescription(
s.Indent("Args:\n");
s.SetIndentLevel(s.GetIndentLevel() + 4);
auto print_one_element = [&s](ConstString key,
auto print_one_element = [&s](llvm::StringRef key,
StructuredData::Object *object) {
s.Indent();
s.Printf("%s : %s\n", key.GetCString(),
object->GetStringValue().str().c_str());
s.Format("{0} : {1}\n", key, object->GetStringValue());
return true;
};

33
lldb/source/Utility/StructuredData.cpp
View File

@ -162,8 +162,18 @@ void StructuredData::String::Serialize(json::OStream &s) const {
void StructuredData::Dictionary::Serialize(json::OStream &s) const {
s.objectBegin();
for (const auto &pair : m_dict) {
s.attributeBegin(pair.first.GetStringRef());
// To ensure the output format is always stable, we sort the dictionary by key
// first.
using Entry = std::pair<llvm::StringRef, ObjectSP>;
std::vector<Entry> sorted_entries;
for (const auto &pair : m_dict)
sorted_entries.push_back({pair.first(), pair.second});
llvm::sort(sorted_entries);
for (const auto &pair : sorted_entries) {
s.attributeBegin(pair.first);
pair.second->Serialize(s);
s.attributeEnd();
}
@ -228,9 +238,20 @@ void StructuredData::Array::GetDescription(lldb_private::Stream &s) const {
void StructuredData::Dictionary::GetDescription(lldb_private::Stream &s) const {
size_t indentation_level = s.GetIndentLevel();
for (auto iter = m_dict.begin(); iter != m_dict.end(); iter++) {
// To ensure the output format is always stable, we sort the dictionary by key
// first.
using Entry = std::pair<llvm::StringRef, ObjectSP>;
std::vector<Entry> sorted_entries;
for (const auto &pair : m_dict)
sorted_entries.push_back({pair.first(), pair.second});
llvm::sort(sorted_entries);
for (auto iter = sorted_entries.begin(); iter != sorted_entries.end();
iter++) {
// Sanitize.
if (iter->first.IsNull() || iter->first.IsEmpty() || !iter->second)
if (iter->first.empty() || !iter->second)
continue;
// Reset original indentation level.
@ -238,7 +259,7 @@ void StructuredData::Dictionary::GetDescription(lldb_private::Stream &s) const {
s.Indent();
// Print key.
s.Printf("%s:", iter->first.AsCString());
s.Format("{0}:", iter->first);
// Return to new line and increase indentation if value is record type.
// Otherwise add spacing.
@ -252,7 +273,7 @@ void StructuredData::Dictionary::GetDescription(lldb_private::Stream &s) const {
// Print value and new line if now last pair.
iter->second->GetDescription(s);
if (std::next(iter) != m_dict.end())
if (std::next(iter) != sorted_entries.end())
s.EOL();
// Reset indentation level if it was incremented previously.