From 0bc536edf354abad0f0fa2c59be5610293a2d47b Mon Sep 17 00:00:00 2001 From: Artem Mukhin Date: Mon, 20 Jun 2022 19:07:10 +0200 Subject: [PATCH] Add GDB/LLDB pretty-printers for NonZero types --- src/etc/gdb_lookup.py | 3 + src/etc/gdb_providers.py | 11 ++++ src/etc/lldb_commands | 1 + src/etc/lldb_lookup.py | 3 + src/etc/lldb_providers.py | 8 +++ src/etc/rust_types.py | 3 + src/test/debuginfo/numeric-types.rs | 85 +++++++++++++++++++++++++++- src/tools/compiletest/src/runtest.rs | 1 + 8 files changed, 114 insertions(+), 1 deletion(-) diff --git a/src/etc/gdb_lookup.py b/src/etc/gdb_lookup.py index 292e91b4d5d61..8171cb4e9a68a 100644 --- a/src/etc/gdb_lookup.py +++ b/src/etc/gdb_lookup.py @@ -89,4 +89,7 @@ def lookup(valobj): if rust_type == RustType.STD_REF_CELL: return StdRefCellProvider(valobj) + if rust_type == RustType.STD_NONZERO_NUMBER: + return StdNonZeroNumberProvider(valobj) + return None diff --git a/src/etc/gdb_providers.py b/src/etc/gdb_providers.py index 0a52b8c976f6a..c351c3450f582 100644 --- a/src/etc/gdb_providers.py +++ b/src/etc/gdb_providers.py @@ -231,6 +231,17 @@ def children(self): yield "borrow", self.borrow +class StdNonZeroNumberProvider: + def __init__(self, valobj): + fields = valobj.type.fields() + assert len(fields) == 1 + field = list(fields)[0] + self.value = str(valobj[field.name]) + + def to_string(self): + return self.value + + # Yields children (in a provider's sense of the word) for a BTreeMap. def children_of_btree_map(map): # Yields each key/value pair in the node and in any child nodes. diff --git a/src/etc/lldb_commands b/src/etc/lldb_commands index 4a1204ccc4be7..ed66ecf30729e 100644 --- a/src/etc/lldb_commands +++ b/src/etc/lldb_commands @@ -15,4 +15,5 @@ type summary add -F lldb_lookup.summary_lookup -e -x -h "^(core::([a-z_]+::)+)C type summary add -F lldb_lookup.summary_lookup -e -x -h "^(core::([a-z_]+::)+)Ref<.+>$" --category Rust type summary add -F lldb_lookup.summary_lookup -e -x -h "^(core::([a-z_]+::)+)RefMut<.+>$" --category Rust type summary add -F lldb_lookup.summary_lookup -e -x -h "^(core::([a-z_]+::)+)RefCell<.+>$" --category Rust +type summary add -F lldb_lookup.summary_lookup -e -x -h "^core::num::([a-z_]+::)*NonZero.+$" --category Rust type category enable Rust diff --git a/src/etc/lldb_lookup.py b/src/etc/lldb_lookup.py index 3cee51982ba9f..bca9c2ae192a7 100644 --- a/src/etc/lldb_lookup.py +++ b/src/etc/lldb_lookup.py @@ -55,6 +55,9 @@ def summary_lookup(valobj, dict): if rust_type == RustType.STD_REF_CELL: return StdRefSummaryProvider(valobj, dict) + if rust_type == RustType.STD_NONZERO_NUMBER: + return StdNonZeroNumberSummaryProvider(valobj, dict) + return "" diff --git a/src/etc/lldb_providers.py b/src/etc/lldb_providers.py index 35ac07f0db763..8a9927e7d96d7 100644 --- a/src/etc/lldb_providers.py +++ b/src/etc/lldb_providers.py @@ -739,3 +739,11 @@ def update(self): def has_children(self): # type: () -> bool return True + + +def StdNonZeroNumberSummaryProvider(valobj, _dict): + # type: (SBValue, dict) -> str + objtype = valobj.GetType() + field = objtype.GetFieldAtIndex(0) + element = valobj.GetChildMemberWithName(field.name) + return element.GetValue() diff --git a/src/etc/rust_types.py b/src/etc/rust_types.py index bbc945a7ddab0..bf512bc99b8f2 100644 --- a/src/etc/rust_types.py +++ b/src/etc/rust_types.py @@ -31,6 +31,7 @@ class RustType(object): STD_REF = "StdRef" STD_REF_MUT = "StdRefMut" STD_REF_CELL = "StdRefCell" + STD_NONZERO_NUMBER = "StdNonZeroNumber" STD_STRING_REGEX = re.compile(r"^(alloc::(\w+::)+)String$") @@ -49,6 +50,7 @@ class RustType(object): STD_REF_REGEX = re.compile(r"^(core::(\w+::)+)Ref<.+>$") STD_REF_MUT_REGEX = re.compile(r"^(core::(\w+::)+)RefMut<.+>$") STD_REF_CELL_REGEX = re.compile(r"^(core::(\w+::)+)RefCell<.+>$") +STD_NONZERO_NUMBER_REGEX = re.compile(r"^core::num::([a-z_]+::)*NonZero.+$") TUPLE_ITEM_REGEX = re.compile(r"__\d+$") @@ -72,6 +74,7 @@ class RustType(object): RustType.STD_REF_MUT: STD_REF_MUT_REGEX, RustType.STD_REF_CELL: STD_REF_CELL_REGEX, RustType.STD_CELL: STD_CELL_REGEX, + RustType.STD_NONZERO_NUMBER: STD_NONZERO_NUMBER_REGEX, } def is_tuple_fields(fields): diff --git a/src/test/debuginfo/numeric-types.rs b/src/test/debuginfo/numeric-types.rs index 2eae9239b6118..b32ccdfd99930 100644 --- a/src/test/debuginfo/numeric-types.rs +++ b/src/test/debuginfo/numeric-types.rs @@ -1,4 +1,3 @@ -// only-cdb // compile-flags:-g // Tests the visualizations for `NonZero{I,U}{8,16,32,64,128,size}`, `Wrapping` and @@ -153,6 +152,90 @@ // cdb-check:a_usize : 0x400 [Type: core::sync::atomic::AtomicUsize] // cdb-check: [] [Type: core::sync::atomic::AtomicUsize] + +// === GDB TESTS =================================================================================== + +// gdb-command:run + +// gdb-command:print nz_i8 +// gdbg-check:[...]$1 = 11 + +// gdb-command:print nz_i16 +// gdbg-check:[...]$2 = 22 + +// gdb-command:print nz_i32 +// gdbg-check:[...]$3 = 33 + +// gdb-command:print nz_i64 +// gdbg-check:[...]$4 = 44 + +// gdb-command:print nz_i128 +// gdbg-check:[...]$5 = 55 + +// gdb-command:print nz_isize +// gdbg-check:[...]$6 = 66 + +// gdb-command:print nz_u8 +// gdbg-check:[...]$7 = 77 + +// gdb-command:print nz_u16 +// gdbg-check:[...]$8 = 88 + +// gdb-command:print nz_u32 +// gdbg-check:[...]$9 = 99 + +// gdb-command:print nz_u64 +// gdbg-check:[...]$10 = 100 + +// gdb-command:print nz_u128 +// gdbg-check:[...]$11 = 111 + +// gdb-command:print nz_usize +// gdbg-check:[...]$12 = 122 + + + +// === LLDB TESTS ================================================================================== + +// lldb-command:run + +// lldb-command:print nz_i8 +// lldb-check:[...]$0 = 11 + +// lldb-command:print nz_i16 +// lldb-check:[...]$1 = 22 + +// lldb-command:print nz_i32 +// lldb-check:[...]$2 = 33 + +// lldb-command:print nz_i64 +// lldb-check:[...]$3 = 44 + +// lldb-command:print nz_i128 +// lldb-check:[...]$4 = 55 + +// lldb-command:print nz_isize +// lldb-check:[...]$5 = 66 + +// lldb-command:print nz_u8 +// lldb-check:[...]$6 = 77 + +// lldb-command:print nz_u16 +// lldb-check:[...]$7 = 88 + +// lldb-command:print nz_u32 +// lldb-check:[...]$8 = 99 + +// lldb-command:print nz_u64 +// lldb-check:[...]$9 = 100 + +// lldb-command:print nz_u128 +// lldb-check:[...]$10 = 111 + +// lldb-command:print nz_usize +// lldb-check:[...]$11 = 122 + + use std::num::*; use std::sync::atomic::*; diff --git a/src/tools/compiletest/src/runtest.rs b/src/tools/compiletest/src/runtest.rs index b758bb9cf6790..70da954a54827 100644 --- a/src/tools/compiletest/src/runtest.rs +++ b/src/tools/compiletest/src/runtest.rs @@ -1094,6 +1094,7 @@ impl<'test> TestCx<'test> { "^(core::([a-z_]+::)+)Ref<.+>$", "^(core::([a-z_]+::)+)RefMut<.+>$", "^(core::([a-z_]+::)+)RefCell<.+>$", + "^core::num::([a-z_]+::)*NonZero.+$", ]; script_str