Rollup merge of rust-lang#98301 - ortem:pretty-printers-nonzero, r=Mark-Simulacrum

Add GDB/LLDB pretty-printers for NonZero types

Add GDB/LLDB pretty-printers for `NonZero` types.
These pretty-printers were originally implemented for IntelliJ Rust by `@Kobzol` in intellij-rust/intellij-rust#5270.

Part of rust-lang#29392.

Author: matthiaskrgr

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

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.

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

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 ""
 
 

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()

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):

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<T>` and
@@ -153,6 +152,90 @@
 // cdb-check:a_usize          : 0x400 [Type: core::sync::atomic::AtomicUsize]
 // cdb-check:    [<Raw View>]     [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::*;
 

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