mirror of
https://github.com/JakeHillion/object-introspection.git
synced 2024-11-09 21:24:14 +00:00
3d91603c8e
std::basic_string takes three template parameters:
1. CharT
2. Traits
3. Allocator
The Traits parameter was causing issues, as it requires a type which
exposes certain things, e.g. `Traits::value_type`.
We have a few options to resolve this:
1. Remove this parameter, as we do for allocators
Cons: removing a template parameter doesn't work if other
parameters appear after it
2. Stub this parameter, as we do for hashers/comparators
Cons: we need to hardcode an implementation that satisfies the
`Traits::value_type` requirements
3. Leave the parameter as-is
Cons: will not work if a non-standard Traits is used
By using the real implementation of this Traits parameter
(normally `std::char_traits<CharT>`), we get one that we know will
work as long as it is defined in a stdlib header.
Option 3 is what we use in this patch. Instead of adding more
configuration options to the container TOML file format (e.g.
`params_to_keep = [1]`), we add `std::char_traits` as a dummy
container type. Now, whenever `std::char_traits` appears, it will be
left as-is, i.e. not removed, replaced or reverse-engineered.
This is the same approach previously used for Thrift's isset_bitset.
119 lines
3.4 KiB
C++
119 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
#pragma once
|
|
#include <map>
|
|
#include <set>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
extern "C" {
|
|
#include <drgn.h>
|
|
}
|
|
|
|
constexpr int oidMagicId = 0x01DE8;
|
|
|
|
#define LIST_OF_CONTAINER_TYPES \
|
|
X(UNKNOWN_TYPE) \
|
|
X(ARRAY_TYPE) \
|
|
X(SMALL_VEC_TYPE) \
|
|
X(SET_TYPE) \
|
|
X(UNORDERED_SET_TYPE) \
|
|
X(SEQ_TYPE) \
|
|
X(LIST_TYPE) \
|
|
X(STD_MAP_TYPE) \
|
|
X(STD_UNORDERED_MAP_TYPE) \
|
|
X(MAP_SEQ_TYPE) \
|
|
X(BY_MULTI_QRT_TYPE) \
|
|
X(F14_MAP) \
|
|
X(F14_SET) \
|
|
X(FEED_QUICK_HASH_SET) \
|
|
X(FEED_QUICK_HASH_MAP) \
|
|
X(RADIX_TREE_TYPE) \
|
|
X(PAIR_TYPE) \
|
|
X(STRING_TYPE) \
|
|
X(FOLLY_IOBUF_TYPE) \
|
|
X(FOLLY_IOBUFQUEUE_TYPE) \
|
|
X(FB_STRING_TYPE) \
|
|
X(UNIQ_PTR_TYPE) \
|
|
X(SHRD_PTR_TYPE) \
|
|
X(FB_HASH_MAP_TYPE) \
|
|
X(FB_HASH_SET_TYPE) \
|
|
X(FOLLY_OPTIONAL_TYPE) \
|
|
X(OPTIONAL_TYPE) \
|
|
X(TRY_TYPE) \
|
|
X(REF_WRAPPER_TYPE) \
|
|
X(SORTED_VEC_SET_TYPE) \
|
|
X(REPEATED_FIELD_TYPE) \
|
|
X(CAFFE2_BLOB_TYPE) \
|
|
X(MULTI_MAP_TYPE) \
|
|
X(FOLLY_SMALL_HEAP_VECTOR_MAP) \
|
|
X(CONTAINER_ADAPTER_TYPE) \
|
|
X(MICROLIST_TYPE) \
|
|
X(ENUM_MAP_TYPE) \
|
|
X(BOOST_BIMAP_TYPE) \
|
|
X(STD_VARIANT_TYPE) \
|
|
X(DUMMY_TYPE) \
|
|
X(WEAK_PTR_TYPE)
|
|
|
|
enum ContainerTypeEnum {
|
|
#define X(name) name,
|
|
LIST_OF_CONTAINER_TYPES
|
|
#undef X
|
|
};
|
|
|
|
struct RootInfo {
|
|
std::string varName;
|
|
struct drgn_qualified_type type;
|
|
};
|
|
|
|
struct ClassMember {
|
|
std::string typeName;
|
|
std::string varName;
|
|
};
|
|
|
|
struct DrgnClassMemberInfo {
|
|
struct drgn_type* type;
|
|
std::string member_name;
|
|
uint64_t bit_offset;
|
|
uint64_t bit_field_size;
|
|
bool isStubbed;
|
|
};
|
|
|
|
struct TypeHierarchy {
|
|
std::map<struct drgn_type*, std::vector<DrgnClassMemberInfo>> classMembersMap;
|
|
std::map<
|
|
struct drgn_type*,
|
|
std::pair<ContainerTypeEnum, std::vector<struct drgn_qualified_type>>>
|
|
containerTypeMap;
|
|
std::map<struct drgn_type*, struct drgn_type*> typedefMap;
|
|
std::map<std::string, size_t> sizeMap;
|
|
std::set<struct drgn_type*> knownDummyTypeList;
|
|
std::map<struct drgn_type*, struct drgn_type*> pointerToTypeMap;
|
|
std::set<struct drgn_type*> thriftIssetStructTypes;
|
|
std::map<struct drgn_type*, std::vector<struct drgn_type*>> descendantClasses;
|
|
};
|
|
|
|
// Helper for std::variant and std::visit
|
|
// https://en.cppreference.com/w/cpp/utility/variant/visit
|
|
template <class... Ts>
|
|
struct visitor : Ts... {
|
|
using Ts::operator()...;
|
|
};
|
|
|
|
// Type deduction for the helper above
|
|
template <class... Ts>
|
|
visitor(Ts...) -> visitor<Ts...>;
|