1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
//===---------------------ProcessStructReader.h ------------------*- C++ -*-===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef utility_ProcessStructReader_h_
#define utility_ProcessStructReader_h_
#include "lldb/lldb-defines.h"
#include "lldb/lldb-types.h"
#include "lldb/Core/ConstString.h"
#include "lldb/Core/DataExtractor.h"
#include "lldb/Core/Error.h"
#include "lldb/Symbol/CompilerType.h"
#include "lldb/Target/Process.h"
#include <initializer_list>
#include <map>
#include <string>
namespace lldb_private {
class ProcessStructReader
{
protected:
struct FieldImpl
{
CompilerType type;
size_t offset;
size_t size;
};
std::map<ConstString, FieldImpl> m_fields;
DataExtractor m_data;
lldb::ByteOrder m_byte_order;
size_t m_addr_byte_size;
public:
ProcessStructReader (Process *process, lldb::addr_t base_addr, CompilerType struct_type)
{
if (!process)
return;
if (base_addr == 0 || base_addr == LLDB_INVALID_ADDRESS)
return;
m_byte_order = process->GetByteOrder();
m_addr_byte_size = process->GetAddressByteSize();
for (size_t idx = 0; idx < struct_type.GetNumFields(); idx++)
{
std::string name;
uint64_t bit_offset;
uint32_t bitfield_bit_size;
bool is_bitfield;
CompilerType field_type = struct_type.GetFieldAtIndex(idx,name,&bit_offset,&bitfield_bit_size,&is_bitfield);
// no support for bitfields in here (yet)
if (is_bitfield)
return;
auto size = field_type.GetByteSize(nullptr);
// no support for things larger than a uint64_t (yet)
if (size > 8)
return;
ConstString const_name = ConstString(name.c_str());
size_t byte_index = static_cast<size_t>(bit_offset / 8);
m_fields[const_name] = FieldImpl{field_type, byte_index, static_cast<size_t>(size)};
}
size_t total_size = struct_type.GetByteSize(nullptr);
lldb::DataBufferSP buffer_sp(new DataBufferHeap(total_size,0));
Error error;
process->ReadMemoryFromInferior(base_addr,
buffer_sp->GetBytes(),
total_size,
error);
if (error.Fail())
return;
m_data = DataExtractor(buffer_sp,m_byte_order,m_addr_byte_size);
}
template<typename RetType>
RetType
GetField (ConstString name, RetType fail_value = RetType())
{
auto iter = m_fields.find(name), end = m_fields.end();
if (iter == end)
return fail_value;
auto size = iter->second.size;
if (sizeof(RetType) < size)
return fail_value;
lldb::offset_t offset = iter->second.offset;
if (offset + size > m_data.GetByteSize())
return fail_value;
return (RetType)(m_data.GetMaxU64(&offset, size));
}
size_t
GetOffsetOf(ConstString name, size_t fail_value = SIZE_MAX)
{
auto iter = m_fields.find(name), end = m_fields.end();
if (iter == end)
return fail_value;
return iter->second.offset;
}
};
}
#endif // utility_ProcessStructReader_h_
|