Commit | Line | Data |
---|---|---|
88b9d363 | 1 | # Copyright (C) 2015-2022 Free Software Foundation, Inc. |
d11916aa SS |
2 | |
3 | # This program is free software; you can redistribute it and/or modify | |
4 | # it under the terms of the GNU General Public License as published by | |
5 | # the Free Software Foundation; either version 3 of the License, or | |
6 | # (at your option) any later version. | |
7 | # | |
8 | # This program is distributed in the hope that it will be useful, | |
9 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
10 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
11 | # GNU General Public License for more details. | |
12 | # | |
13 | # You should have received a copy of the GNU General Public License | |
14 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
15 | ||
16 | import gdb | |
17 | from gdb.unwinder import Unwinder | |
18 | ||
d11916aa | 19 | |
13123da8 | 20 | class FrameId(object): |
d11916aa SS |
21 | def __init__(self, sp, pc): |
22 | self._sp = sp | |
23 | self._pc = pc | |
24 | ||
25 | @property | |
26 | def sp(self): | |
27 | return self._sp | |
28 | ||
29 | @property | |
30 | def pc(self): | |
31 | return self._pc | |
32 | ||
13123da8 | 33 | |
d11916aa SS |
34 | class TestUnwinder(Unwinder): |
35 | AMD64_RBP = 6 | |
36 | AMD64_RSP = 7 | |
43d5901d | 37 | AMD64_RIP = None |
d11916aa SS |
38 | |
39 | def __init__(self): | |
40 | Unwinder.__init__(self, "test unwinder") | |
41 | self.char_ptr_t = gdb.lookup_type("unsigned char").pointer() | |
42 | self.char_ptr_ptr_t = self.char_ptr_t.pointer() | |
43d5901d AB |
43 | self._last_arch = None |
44 | ||
45 | # Update the register descriptor AMD64_RIP based on ARCH. | |
13123da8 SM |
46 | def _update_register_descriptors(self, arch): |
47 | if self._last_arch != arch: | |
48 | TestUnwinder.AMD64_RIP = arch.registers().find("rip") | |
43d5901d | 49 | self._last_arch = arch |
d11916aa SS |
50 | |
51 | def _read_word(self, address): | |
52 | return address.cast(self.char_ptr_ptr_t).dereference() | |
53 | ||
54 | def __call__(self, pending_frame): | |
55 | """Test unwinder written in Python. | |
56 | ||
57 | This unwinder can unwind the frames that have been deliberately | |
58 | corrupted in a specific way (functions in the accompanying | |
59 | py-unwind.c file do that.) | |
60 | This code is only on AMD64. | |
61 | On AMD64 $RBP points to the innermost frame (unless the code | |
62 | was compiled with -fomit-frame-pointer), which contains the | |
63 | address of the previous frame at offset 0. The functions | |
64 | deliberately corrupt their frames as follows: | |
65 | Before After | |
66 | Corruption: Corruption: | |
67 | +--------------+ +--------------+ | |
68 | RBP-8 | | | Previous RBP | | |
69 | +--------------+ +--------------+ | |
70 | RBP + Previous RBP | | RBP | | |
71 | +--------------+ +--------------+ | |
72 | RBP+8 | Return RIP | | Return RIP | | |
73 | +--------------+ +--------------+ | |
74 | Old SP | | | | | |
75 | ||
76 | This unwinder recognizes the corrupt frames by checking that | |
77 | *RBP == RBP, and restores previous RBP from the word above it. | |
78 | """ | |
87dbc774 AB |
79 | |
80 | # Check that we can access the architecture of the pending | |
81 | # frame, and that this is the same architecture as for the | |
82 | # currently selected inferior. | |
13123da8 SM |
83 | inf_arch = gdb.selected_inferior().architecture() |
84 | frame_arch = pending_frame.architecture() | |
85 | if inf_arch != frame_arch: | |
86 | raise gdb.GdbError("architecture mismatch") | |
87dbc774 | 87 | |
13123da8 | 88 | self._update_register_descriptors(frame_arch) |
43d5901d | 89 | |
d11916aa SS |
90 | try: |
91 | # NOTE: the registers in Unwinder API can be referenced | |
92 | # either by name or by number. The code below uses both | |
93 | # to achieve more coverage. | |
94 | bp = pending_frame.read_register("rbp").cast(self.char_ptr_t) | |
95 | if self._read_word(bp) != bp: | |
96 | return None | |
97 | # Found the frame that the test program has corrupted for us. | |
98 | # The correct BP for the outer frame has been saved one word | |
99 | # above, previous IP and SP are at the expected places. | |
100 | previous_bp = self._read_word(bp - 8) | |
101 | previous_ip = self._read_word(bp + 8) | |
102 | previous_sp = bp + 16 | |
103 | ||
104 | frame_id = FrameId( | |
105 | pending_frame.read_register(TestUnwinder.AMD64_RSP), | |
13123da8 SM |
106 | pending_frame.read_register(TestUnwinder.AMD64_RIP), |
107 | ) | |
d11916aa | 108 | unwind_info = pending_frame.create_unwind_info(frame_id) |
13123da8 | 109 | unwind_info.add_saved_register(TestUnwinder.AMD64_RBP, previous_bp) |
d11916aa SS |
110 | unwind_info.add_saved_register("rip", previous_ip) |
111 | unwind_info.add_saved_register("rsp", previous_sp) | |
112 | return unwind_info | |
113 | except (gdb.error, RuntimeError): | |
114 | return None | |
115 | ||
13123da8 | 116 | |
d11916aa SS |
117 | gdb.unwinder.register_unwinder(None, TestUnwinder(), True) |
118 | print("Python script imported") |