1 //===-- DWARFDebugArangeSet.cpp ---------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "DWARFDebugArangeSet.h"
11 
12 #include <assert.h>
13 #include "lldb/Core/Stream.h"
14 #include "SymbolFileDWARF.h"
15 
16 using namespace lldb_private;
17 
18 DWARFDebugArangeSet::DWARFDebugArangeSet() :
19     m_offset(DW_INVALID_OFFSET),
20     m_header(),
21     m_arange_descriptors()
22 {
23     m_header.length     = 0;
24     m_header.version    = 0;
25     m_header.cu_offset  = 0;
26     m_header.addr_size  = 0;
27     m_header.seg_size   = 0;
28 }
29 
30 void
31 DWARFDebugArangeSet::Clear()
32 {
33     m_offset = DW_INVALID_OFFSET;
34     m_header.length     = 0;
35     m_header.version    = 0;
36     m_header.cu_offset  = 0;
37     m_header.addr_size  = 0;
38     m_header.seg_size   = 0;
39     m_arange_descriptors.clear();
40 }
41 
42 void
43 DWARFDebugArangeSet::SetHeader
44 (
45     uint16_t version,
46     uint32_t cu_offset,
47     uint8_t addr_size,
48     uint8_t seg_size
49 )
50 {
51     m_header.version    = version;
52     m_header.cu_offset  = cu_offset;
53     m_header.addr_size  = addr_size;
54     m_header.seg_size   = seg_size;
55 }
56 
57 void
58 DWARFDebugArangeSet::Compact()
59 {
60     if (m_arange_descriptors.empty())
61         return;
62 
63     // Iterate through all arange descriptors and combine any ranges that
64     // overlap or have matching boundaries. The m_arange_descriptors are assumed
65     // to be in ascending order after being built by adding descriptors
66     // using the AddDescriptor method.
67     uint32_t i = 0;
68     while (i + 1 < m_arange_descriptors.size())
69     {
70         if (m_arange_descriptors[i].end_address() >= m_arange_descriptors[i+1].address)
71         {
72             // The current range ends at or exceeds the start of the next address range.
73             // Compute the max end address between the two and use that to make the new
74             // length.
75             const dw_addr_t max_end_addr = std::max(m_arange_descriptors[i].end_address(), m_arange_descriptors[i+1].end_address());
76             m_arange_descriptors[i].length = max_end_addr - m_arange_descriptors[i].address;
77             // Now remove the next entry as it was just combined with the previous one.
78             m_arange_descriptors.erase(m_arange_descriptors.begin()+i+1);
79         }
80         else
81         {
82             // Discontiguous address range, just proceed to the next one.
83             ++i;
84         }
85     }
86 }
87 //----------------------------------------------------------------------
88 // Compare function DWARFDebugArangeSet::Descriptor structures
89 //----------------------------------------------------------------------
90 static bool DescriptorLessThan (const DWARFDebugArangeSet::Descriptor& range1, const DWARFDebugArangeSet::Descriptor& range2)
91 {
92     return range1.address < range2.address;
93 }
94 
95 //----------------------------------------------------------------------
96 // Add a range descriptor and keep things sorted so we can easily
97 // compact the ranges before being saved or used.
98 //----------------------------------------------------------------------
99 void
100 DWARFDebugArangeSet::AddDescriptor(const DWARFDebugArangeSet::Descriptor& range)
101 {
102     if (m_arange_descriptors.empty())
103     {
104         m_arange_descriptors.push_back(range);
105         return;
106     }
107 
108     DescriptorIter end = m_arange_descriptors.end();
109     DescriptorIter pos = lower_bound(m_arange_descriptors.begin(), end, range, DescriptorLessThan);
110     const dw_addr_t range_end_addr = range.end_address();
111     if (pos != end)
112     {
113         const dw_addr_t found_end_addr = pos->end_address();
114         if (range.address < pos->address)
115         {
116             if (range_end_addr < pos->address)
117             {
118                 // Non-contiguous entries, add this one before the found entry
119                 m_arange_descriptors.insert(pos, range);
120             }
121             else if (range_end_addr == pos->address)
122             {
123                 // The top end of 'range' is the lower end of the entry
124                 // pointed to by 'pos'. We can combine range with the
125                 // entry we found by setting the starting address and
126                 // increasing the length since they don't overlap.
127                 pos->address = range.address;
128                 pos->length += range.length;
129             }
130             else
131             {
132                 // We can combine these two and make sure the largest end
133                 // address is used to make end address.
134                 pos->address = range.address;
135                 pos->length = std::max(found_end_addr, range_end_addr) - pos->address;
136             }
137         }
138         else if (range.address == pos->address)
139         {
140             pos->length = std::max(pos->length, range.length);
141         }
142     }
143     else
144     {
145         // NOTE: 'pos' points to entry past the end which is ok for insert,
146         // don't use otherwise!!!
147         const dw_addr_t max_addr = m_arange_descriptors.back().end_address();
148         if (max_addr < range.address)
149         {
150             // Non-contiguous entries, add this one before the found entry
151             m_arange_descriptors.insert(pos, range);
152         }
153         else if (max_addr == range.address)
154         {
155             m_arange_descriptors.back().length += range.length;
156         }
157         else
158         {
159             m_arange_descriptors.back().length = std::max(max_addr, range_end_addr) - m_arange_descriptors.back().address;
160         }
161     }
162 }
163 
164 bool
165 DWARFDebugArangeSet::Extract(const DataExtractor &data, lldb::offset_t *offset_ptr)
166 {
167     if (data.ValidOffset(*offset_ptr))
168     {
169         m_arange_descriptors.clear();
170         m_offset = *offset_ptr;
171 
172         // 7.20 Address Range Table
173         //
174         // Each set of entries in the table of address ranges contained in
175         // the .debug_aranges section begins with a header consisting of: a
176         // 4-byte length containing the length of the set of entries for this
177         // compilation unit, not including the length field itself; a 2-byte
178         // version identifier containing the value 2 for DWARF Version 2; a
179         // 4-byte offset into the.debug_infosection; a 1-byte unsigned integer
180         // containing the size in bytes of an address (or the offset portion of
181         // an address for segmented addressing) on the target system; and a
182         // 1-byte unsigned integer containing the size in bytes of a segment
183         // descriptor on the target system. This header is followed by a series
184         // of tuples. Each tuple consists of an address and a length, each in
185         // the size appropriate for an address on the target architecture.
186         size_t dwarf_offset_size = 4;
187         m_header.length     = data.GetU32(offset_ptr);
188         if (m_header.length == 0xffffffff)
189         {
190             dwarf_offset_size = 8;
191             m_header.length = data.GetU64(offset_ptr);
192         }
193         else if (m_header.length >= 0xffffff00)
194         {
195             // Reserved.
196             return false;
197         }
198         m_header.version    = data.GetU16(offset_ptr);
199         m_header.cu_offset  = data.GetMaxU64(offset_ptr, dwarf_offset_size);
200         m_header.addr_size  = data.GetU8(offset_ptr);
201         m_header.seg_size   = data.GetU8(offset_ptr);
202 
203 
204         // The first tuple following the header in each set begins at an offset
205         // that is a multiple of the size of a single tuple (that is, twice the
206         // size of an address). The header is padded, if necessary, to the
207         // appropriate boundary.
208         const uint32_t header_size = *offset_ptr - m_offset;
209         const uint32_t tuple_size = m_header.addr_size << 1;
210         uint32_t first_tuple_offset = 0;
211         while (first_tuple_offset < header_size)
212             first_tuple_offset += tuple_size;
213 
214         *offset_ptr = m_offset + first_tuple_offset;
215 
216         Descriptor arangeDescriptor;
217 
218         assert(sizeof(arangeDescriptor.address) == sizeof(arangeDescriptor.length));
219         assert(sizeof(arangeDescriptor.address) >= m_header.addr_size);
220 
221         while (data.ValidOffset(*offset_ptr))
222         {
223             arangeDescriptor.address    = data.GetMaxU64(offset_ptr, m_header.addr_size);
224             arangeDescriptor.length     = data.GetMaxU64(offset_ptr, m_header.addr_size);
225 
226             // Each set of tuples is terminated by a 0 for the address and 0
227             // for the length.
228             if (arangeDescriptor.address || arangeDescriptor.length)
229                 m_arange_descriptors.push_back(arangeDescriptor);
230             else
231                 break;  // We are done if we get a zero address and length
232         }
233 
234         return !m_arange_descriptors.empty();
235     }
236     return false;
237 }
238 
239 
240 dw_offset_t
241 DWARFDebugArangeSet::GetOffsetOfNextEntry() const
242 {
243     return m_offset + m_header.length + 4;
244 }
245 
246 
247 void
248 DWARFDebugArangeSet::Dump(Stream *s) const
249 {
250     s->Printf("Address Range Header: length = 0x%8.8x, version = 0x%4.4x, cu_offset = 0x%8.8x, addr_size = 0x%2.2x, seg_size = 0x%2.2x\n",
251         m_header.length ,m_header.version, m_header.cu_offset, m_header.addr_size, m_header.seg_size);
252 
253     const uint32_t hex_width = m_header.addr_size * 2;
254     DescriptorConstIter pos;
255     DescriptorConstIter end = m_arange_descriptors.end();
256     for (pos = m_arange_descriptors.begin(); pos != end; ++pos)
257         s->Printf("[0x%*.*" PRIx64 " - 0x%*.*" PRIx64 ")\n",
258             hex_width, hex_width, pos->address,
259             hex_width, hex_width, pos->end_address());
260 }
261 
262 
263 class DescriptorContainsAddress
264 {
265 public:
266     DescriptorContainsAddress (dw_addr_t address) : m_address(address) {}
267     bool operator() (const DWARFDebugArangeSet::Descriptor& desc) const
268     {
269         return (m_address >= desc.address) && (m_address < (desc.address + desc.length));
270     }
271  private:
272    const dw_addr_t m_address;
273 };
274 
275 dw_offset_t
276 DWARFDebugArangeSet::FindAddress(dw_addr_t address) const
277 {
278     DescriptorConstIter end = m_arange_descriptors.end();
279     DescriptorConstIter pos = std::find_if( m_arange_descriptors.begin(), end,  // Range
280                                             DescriptorContainsAddress(address));// Predicate
281     if (pos != end)
282         return m_header.cu_offset;
283 
284     return DW_INVALID_OFFSET;
285 }
286