blob: 5e63ef92fac3eca2a4dc455cca4af32fe1c07d36 [file] [log] [blame]
//===-- StructuredData.h ----------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#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/FileSpec.h"
#include "lldb/Utility/Stream.h"
#include "lldb/lldb-enumerations.h"
#include <cassert>
#include <cstddef>
#include <cstdint>
#include <functional>
#include <map>
#include <memory>
#include <optional>
#include <string>
#include <type_traits>
#include <utility>
#include <variant>
#include <vector>
namespace lldb_private {
class Status;
}
namespace lldb_private {
/// \class StructuredData StructuredData.h "lldb/Utility/StructuredData.h"
/// A class which can hold structured data
///
/// The StructuredData class is designed to hold the data from a JSON or plist
/// style file -- a serialized data structure with dictionaries (maps,
/// hashes), arrays, and concrete values like integers, floating point
/// numbers, strings, booleans.
///
/// StructuredData does not presuppose any knowledge of the schema for the
/// data it is holding; it can parse JSON data, for instance, and other parts
/// of lldb can iterate through the parsed data set to find keys and values
/// that may be present.
class StructuredData {
template <typename N> class Integer;
public:
class Object;
class Array;
using UnsignedInteger = Integer<uint64_t>;
using SignedInteger = Integer<int64_t>;
class Float;
class Boolean;
class String;
class Dictionary;
class Generic;
typedef std::shared_ptr<Object> ObjectSP;
typedef std::shared_ptr<Array> ArraySP;
typedef std::shared_ptr<UnsignedInteger> UnsignedIntegerSP;
typedef std::shared_ptr<SignedInteger> SignedIntegerSP;
typedef std::shared_ptr<Float> FloatSP;
typedef std::shared_ptr<Boolean> BooleanSP;
typedef std::shared_ptr<String> StringSP;
typedef std::shared_ptr<Dictionary> DictionarySP;
typedef std::shared_ptr<Generic> GenericSP;
typedef std::variant<UnsignedIntegerSP, SignedIntegerSP> IntegerSP;
class Object : public std::enable_shared_from_this<Object> {
public:
Object(lldb::StructuredDataType t = lldb::eStructuredDataTypeInvalid)
: m_type(t) {}
virtual ~Object() = default;
virtual bool IsValid() const { return true; }
virtual void Clear() { m_type = lldb::eStructuredDataTypeInvalid; }
lldb::StructuredDataType GetType() const { return m_type; }
void SetType(lldb::StructuredDataType t) { m_type = t; }
Array *GetAsArray() {
return ((m_type == lldb::eStructuredDataTypeArray)
? static_cast<Array *>(this)
: nullptr);
}
Dictionary *GetAsDictionary() {
return ((m_type == lldb::eStructuredDataTypeDictionary)
? static_cast<Dictionary *>(this)
: nullptr);
}
UnsignedInteger *GetAsUnsignedInteger() {
// NOTE: For backward compatibility, eStructuredDataTypeInteger is
// the same as eStructuredDataTypeUnsignedInteger.
return ((m_type == lldb::eStructuredDataTypeInteger ||
m_type == lldb::eStructuredDataTypeUnsignedInteger)
? static_cast<UnsignedInteger *>(this)
: nullptr);
}
SignedInteger *GetAsSignedInteger() {
return ((m_type == lldb::eStructuredDataTypeSignedInteger)
? static_cast<SignedInteger *>(this)
: nullptr);
}
uint64_t GetUnsignedIntegerValue(uint64_t fail_value = 0) {
UnsignedInteger *integer = GetAsUnsignedInteger();
return ((integer != nullptr) ? integer->GetValue() : fail_value);
}
int64_t GetSignedIntegerValue(int64_t fail_value = 0) {
SignedInteger *integer = GetAsSignedInteger();
return ((integer != nullptr) ? integer->GetValue() : fail_value);
}
Float *GetAsFloat() {
return ((m_type == lldb::eStructuredDataTypeFloat)
? static_cast<Float *>(this)
: nullptr);
}
double GetFloatValue(double fail_value = 0.0) {
Float *f = GetAsFloat();
return ((f != nullptr) ? f->GetValue() : fail_value);
}
Boolean *GetAsBoolean() {
return ((m_type == lldb::eStructuredDataTypeBoolean)
? static_cast<Boolean *>(this)
: nullptr);
}
bool GetBooleanValue(bool fail_value = false) {
Boolean *b = GetAsBoolean();
return ((b != nullptr) ? b->GetValue() : fail_value);
}
String *GetAsString() {
return ((m_type == lldb::eStructuredDataTypeString)
? static_cast<String *>(this)
: nullptr);
}
llvm::StringRef GetStringValue(const char *fail_value = nullptr) {
String *s = GetAsString();
if (s)
return s->GetValue();
return fail_value;
}
Generic *GetAsGeneric() {
return ((m_type == lldb::eStructuredDataTypeGeneric)
? static_cast<Generic *>(this)
: nullptr);
}
ObjectSP GetObjectForDotSeparatedPath(llvm::StringRef path);
void DumpToStdout(bool pretty_print = true) const;
virtual void Serialize(llvm::json::OStream &s) const = 0;
void Dump(lldb_private::Stream &s, bool pretty_print = true) const {
llvm::json::OStream jso(s.AsRawOstream(), pretty_print ? 2 : 0);
Serialize(jso);
}
virtual void GetDescription(lldb_private::Stream &s) const {
s.IndentMore();
Dump(s, false);
s.IndentLess();
}
private:
lldb::StructuredDataType m_type;
};
class Array : public Object {
public:
Array() : Object(lldb::eStructuredDataTypeArray) {}
~Array() override = default;
bool
ForEach(std::function<bool(Object *object)> const &foreach_callback) const {
for (const auto &object_sp : m_items) {
if (!foreach_callback(object_sp.get()))
return false;
}
return true;
}
size_t GetSize() const { return m_items.size(); }
ObjectSP operator[](size_t idx) {
if (idx < m_items.size())
return m_items[idx];
return ObjectSP();
}
ObjectSP GetItemAtIndex(size_t idx) const {
assert(idx < GetSize());
if (idx < m_items.size())
return m_items[idx];
return ObjectSP();
}
template <class IntType>
std::optional<IntType> GetItemAtIndexAsInteger(size_t idx) const {
if (auto item_sp = GetItemAtIndex(idx)) {
if constexpr (std::numeric_limits<IntType>::is_signed) {
if (auto *signed_value = item_sp->GetAsSignedInteger())
return static_cast<IntType>(signed_value->GetValue());
} else {
if (auto *unsigned_value = item_sp->GetAsUnsignedInteger())
return static_cast<IntType>(unsigned_value->GetValue());
}
}
return {};
}
std::optional<llvm::StringRef> GetItemAtIndexAsString(size_t idx) const {
if (auto item_sp = GetItemAtIndex(idx)) {
if (auto *string_value = item_sp->GetAsString())
return string_value->GetValue();
}
return {};
}
/// Retrieves the element at index \a idx from a StructuredData::Array if it
/// is a Dictionary.
///
/// \param[in] idx
/// The index of the element to retrieve.
///
/// \return
/// If the element at index \a idx is a Dictionary, this method returns a
/// valid pointer to the Dictionary wrapped in a std::optional. If the
/// element is not a Dictionary or the index is invalid, this returns
/// std::nullopt. Note that the underlying Dictionary pointer is never
/// nullptr.
std::optional<Dictionary *> GetItemAtIndexAsDictionary(size_t idx) const {
if (auto item_sp = GetItemAtIndex(idx)) {
if (auto *dict = item_sp->GetAsDictionary())
return dict;
}
return {};
}
void Push(const ObjectSP &item) { m_items.push_back(item); }
void AddItem(const ObjectSP &item) { m_items.push_back(item); }
template <typename T> void AddIntegerItem(T value) {
static_assert(std::is_integral<T>::value ||
std::is_floating_point<T>::value,
"value type should be integral");
if constexpr (std::numeric_limits<T>::is_signed)
AddItem(std::make_shared<SignedInteger>(value));
else
AddItem(std::make_shared<UnsignedInteger>(value));
}
void AddFloatItem(double value) { AddItem(std::make_shared<Float>(value)); }
void AddStringItem(llvm::StringRef value) {
AddItem(std::make_shared<String>(std::move(value)));
}
void AddBooleanItem(bool value) {
AddItem(std::make_shared<Boolean>(value));
}
void Serialize(llvm::json::OStream &s) const override;
void GetDescription(lldb_private::Stream &s) const override;
protected:
typedef std::vector<ObjectSP> collection;
collection m_items;
};
private:
template <typename N> class Integer : public Object {
static_assert(std::is_integral<N>::value, "N must be an integral type");
public:
Integer(N i = 0)
: Object(std::numeric_limits<N>::is_signed
? lldb::eStructuredDataTypeSignedInteger
: lldb::eStructuredDataTypeUnsignedInteger),
m_value(i) {}
~Integer() override = default;
void SetValue(N value) { m_value = value; }
N GetValue() { return m_value; }
void Serialize(llvm::json::OStream &s) const override {
s.value(static_cast<N>(m_value));
}
void GetDescription(lldb_private::Stream &s) const override {
s.Printf(std::numeric_limits<N>::is_signed ? "%" PRId64 : "%" PRIu64,
static_cast<N>(m_value));
}
protected:
N m_value;
};
public:
class Float : public Object {
public:
Float(double d = 0.0)
: Object(lldb::eStructuredDataTypeFloat), m_value(d) {}
~Float() override = default;
void SetValue(double value) { m_value = value; }
double GetValue() { return m_value; }
void Serialize(llvm::json::OStream &s) const override;
void GetDescription(lldb_private::Stream &s) const override;
protected:
double m_value;
};
class Boolean : public Object {
public:
Boolean(bool b = false)
: Object(lldb::eStructuredDataTypeBoolean), m_value(b) {}
~Boolean() override = default;
void SetValue(bool value) { m_value = value; }
bool GetValue() { return m_value; }
void Serialize(llvm::json::OStream &s) const override;
void GetDescription(lldb_private::Stream &s) const override;
protected:
bool m_value;
};
class String : public Object {
public:
String() : Object(lldb::eStructuredDataTypeString) {}
explicit String(llvm::StringRef S)
: Object(lldb::eStructuredDataTypeString), m_value(S) {}
void SetValue(llvm::StringRef S) { m_value = std::string(S); }
llvm::StringRef GetValue() { return m_value; }
void Serialize(llvm::json::OStream &s) const override;
void GetDescription(lldb_private::Stream &s) const override;
protected:
std::string m_value;
};
class Dictionary : public Object {
public:
Dictionary() : Object(lldb::eStructuredDataTypeDictionary) {}
Dictionary(ObjectSP obj_sp) : Object(lldb::eStructuredDataTypeDictionary) {
if (!obj_sp || obj_sp->GetType() != lldb::eStructuredDataTypeDictionary) {
SetType(lldb::eStructuredDataTypeInvalid);
return;
}
Dictionary *dict = obj_sp->GetAsDictionary();
m_dict = dict->m_dict;
}
~Dictionary() override = default;
size_t GetSize() const { return m_dict.size(); }
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()))
break;
}
}
ArraySP GetKeys() const {
auto array_sp = std::make_shared<Array>();
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 {
return m_dict.lookup(key);
}
bool GetValueForKeyAsBoolean(llvm::StringRef key, bool &result) const {
bool success = false;
ObjectSP value_sp = GetValueForKey(key);
if (value_sp.get()) {
Boolean *result_ptr = value_sp->GetAsBoolean();
if (result_ptr) {
result = result_ptr->GetValue();
success = true;
}
}
return success;
}
template <class IntType>
bool GetValueForKeyAsInteger(llvm::StringRef key, IntType &result) const {
ObjectSP value_sp = GetValueForKey(key);
if (value_sp) {
if constexpr (std::numeric_limits<IntType>::is_signed) {
if (auto signed_value = value_sp->GetAsSignedInteger()) {
result = static_cast<IntType>(signed_value->GetValue());
return true;
}
} else {
if (auto unsigned_value = value_sp->GetAsUnsignedInteger()) {
result = static_cast<IntType>(unsigned_value->GetValue());
return true;
}
}
}
return false;
}
template <class IntType>
bool GetValueForKeyAsInteger(llvm::StringRef key, IntType &result,
IntType default_val) const {
bool success = GetValueForKeyAsInteger<IntType>(key, result);
if (!success)
result = default_val;
return success;
}
bool GetValueForKeyAsString(llvm::StringRef key,
llvm::StringRef &result) const {
ObjectSP value_sp = GetValueForKey(key);
if (value_sp.get()) {
if (auto string_value = value_sp->GetAsString()) {
result = string_value->GetValue();
return true;
}
}
return false;
}
bool GetValueForKeyAsString(llvm::StringRef key, llvm::StringRef &result,
const char *default_val) const {
bool success = GetValueForKeyAsString(key, result);
if (!success) {
if (default_val)
result = default_val;
else
result = llvm::StringRef();
}
return success;
}
bool GetValueForKeyAsDictionary(llvm::StringRef key,
Dictionary *&result) const {
result = nullptr;
ObjectSP value_sp = GetValueForKey(key);
if (value_sp.get()) {
result = value_sp->GetAsDictionary();
return (result != nullptr);
}
return false;
}
bool GetValueForKeyAsArray(llvm::StringRef key, Array *&result) const {
result = nullptr;
ObjectSP value_sp = GetValueForKey(key);
if (value_sp.get()) {
result = value_sp->GetAsArray();
return (result != nullptr);
}
return false;
}
bool HasKey(llvm::StringRef key) const { return m_dict.contains(key); }
void AddItem(llvm::StringRef key, ObjectSP value_sp) {
m_dict.insert_or_assign(key, std::move(value_sp));
}
template <typename T> void AddIntegerItem(llvm::StringRef key, T value) {
static_assert(std::is_integral<T>::value ||
std::is_floating_point<T>::value,
"value type should be integral");
if constexpr (std::numeric_limits<T>::is_signed)
AddItem(key, std::make_shared<SignedInteger>(value));
else
AddItem(key, std::make_shared<UnsignedInteger>(value));
}
void AddFloatItem(llvm::StringRef key, double value) {
AddItem(key, std::make_shared<Float>(value));
}
void AddStringItem(llvm::StringRef key, llvm::StringRef value) {
AddItem(key, std::make_shared<String>(std::move(value)));
}
void AddBooleanItem(llvm::StringRef key, bool value) {
AddItem(key, std::make_shared<Boolean>(value));
}
void Serialize(llvm::json::OStream &s) const override;
void GetDescription(lldb_private::Stream &s) const override;
protected:
llvm::StringMap<ObjectSP> m_dict;
};
class Null : public Object {
public:
Null() : Object(lldb::eStructuredDataTypeNull) {}
~Null() override = default;
bool IsValid() const override { return false; }
void Serialize(llvm::json::OStream &s) const override;
void GetDescription(lldb_private::Stream &s) const override;
};
class Generic : public Object {
public:
explicit Generic(void *object = nullptr)
: Object(lldb::eStructuredDataTypeGeneric), m_object(object) {}
void SetValue(void *value) { m_object = value; }
void *GetValue() const { return m_object; }
bool IsValid() const override { return m_object != nullptr; }
void Serialize(llvm::json::OStream &s) const override;
void GetDescription(lldb_private::Stream &s) const override;
private:
void *m_object;
};
static ObjectSP ParseJSON(llvm::StringRef json_text);
static ObjectSP ParseJSONFromFile(const FileSpec &file, Status &error);
static bool IsRecordType(const ObjectSP object_sp);
};
} // namespace lldb_private
#endif // LLDB_UTILITY_STRUCTUREDDATA_H