[llvm] b42400c - [Demangle][Rust] Parse array type

[Tomasz Miąsko via llvm-commits]

Author: Tomasz Miąsko
Date: 2021-05-19T00:04:41+02:00
New Revision: b42400ccf3be9c8a422f175e16109e358298973c

URL: https://github.com/llvm/llvm-project/commit/b42400ccf3be9c8a422f175e16109e358298973c
DIFF: https://github.com/llvm/llvm-project/commit/b42400ccf3be9c8a422f175e16109e358298973c.diff

LOG: [Demangle][Rust] Parse array type

Reviewed By: dblaikie

Differential Revision: https://reviews.llvm.org/D102573

Added: 
    

Modified: 
    llvm/lib/Demangle/RustDemangle.cpp
    llvm/test/Demangle/rust.test

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Demangle/RustDemangle.cpp b/llvm/lib/Demangle/RustDemangle.cpp
index 6e2694f46c55..77aa4f7462a4 100644
--- a/llvm/lib/Demangle/RustDemangle.cpp
+++ b/llvm/lib/Demangle/RustDemangle.cpp
@@ -421,14 +421,26 @@ void Demangler::printBasicType(BasicType Type) {
 //          | "D" <dyn-bounds> <lifetime> // dyn Trait<Assoc = X> + Send + 'a
 //          | <backref>                   // backref
 void Demangler::demangleType() {
-  char C = look();
+  size_t Start = Position;
+
+  char C = consume();
   BasicType Type;
-  if (parseBasicType(C, Type)) {
-    consume();
+  if (parseBasicType(C, Type))
     return printBasicType(Type);
-  }
 
-  demanglePath(InType::Yes);
+  switch (C) {
+  case 'A':
+    print("[");
+    demangleType();
+    print("; ");
+    demangleConst();
+    print("]");
+    break;
+  default:
+    Position = Start;
+    demanglePath(InType::Yes);
+    break;
+  }
 }
 
 // <const> = <basic-type> <const-data>

diff  --git a/llvm/test/Demangle/rust.test b/llvm/test/Demangle/rust.test
index 99beacc54b01..388cab1907ff 100644
--- a/llvm/test/Demangle/rust.test
+++ b/llvm/test/Demangle/rust.test
@@ -155,6 +155,11 @@ CHECK: named::<name::Name>
 CHECK: named::<Name<>>
        _RIC5namedIC4NameEE
 
+; Types
+
+CHECK: types::<[u8; 0]>
+       _RIC5typesAhj0_E
+
 ; Integer constants. Test value demangling.
 
 CHECK: integer::<0>