| //===------------------ RustDemangleTest.cpp ------------------------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "llvm/Demangle/Demangle.h" |
| #include "gmock/gmock.h" |
| #include "gtest/gtest.h" |
| |
| #include <cstdlib> |
| |
| TEST(RustDemangle, Success) { |
| char *Demangled = |
| llvm::rustDemangle("_RNvC1a4main", nullptr, nullptr, nullptr); |
| EXPECT_STREQ(Demangled, "a::main"); |
| std::free(Demangled); |
| |
| // With status. |
| int Status = 0; |
| Demangled = llvm::rustDemangle("_RNvC1a4main", nullptr, nullptr, &Status); |
| EXPECT_EQ(Status, llvm::demangle_success); |
| EXPECT_STREQ(Demangled, "a::main"); |
| std::free(Demangled); |
| |
| // With status and length. |
| size_t N = 0; |
| Demangled = llvm::rustDemangle("_RNvC1a4main", nullptr, &N, &Status); |
| EXPECT_EQ(Status, llvm::demangle_success); |
| EXPECT_EQ(N, 8u); |
| EXPECT_STREQ(Demangled, "a::main"); |
| std::free(Demangled); |
| } |
| |
| TEST(RustDemangle, Invalid) { |
| int Status = 0; |
| char *Demangled = nullptr; |
| |
| // Invalid prefix. |
| Demangled = llvm::rustDemangle("_ABCDEF", nullptr, nullptr, &Status); |
| EXPECT_EQ(Status, llvm::demangle_invalid_mangled_name); |
| EXPECT_EQ(Demangled, nullptr); |
| |
| // Correct prefix but still invalid. |
| Demangled = llvm::rustDemangle("_RRR", nullptr, nullptr, &Status); |
| EXPECT_EQ(Status, llvm::demangle_invalid_mangled_name); |
| EXPECT_EQ(Demangled, nullptr); |
| } |
| |
| TEST(RustDemangle, OutputBufferWithoutLength) { |
| char *Buffer = static_cast<char *>(std::malloc(1024)); |
| ASSERT_NE(Buffer, nullptr); |
| |
| int Status = 0; |
| char *Demangled = |
| llvm::rustDemangle("_RNvC1a4main", Buffer, nullptr, &Status); |
| |
| EXPECT_EQ(Status, llvm::demangle_invalid_args); |
| EXPECT_EQ(Demangled, nullptr); |
| std::free(Buffer); |
| } |
| |
| TEST(RustDemangle, OutputBuffer) { |
| size_t N = 1024; |
| char *Buffer = static_cast<char *>(std::malloc(N)); |
| ASSERT_NE(Buffer, nullptr); |
| |
| int Status = 0; |
| char *Demangled = llvm::rustDemangle("_RNvC1a4main", Buffer, &N, &Status); |
| |
| EXPECT_EQ(Status, llvm::demangle_success); |
| EXPECT_EQ(Demangled, Buffer); |
| EXPECT_STREQ(Demangled, "a::main"); |
| std::free(Demangled); |
| } |
| |
| TEST(RustDemangle, SmallOutputBuffer) { |
| size_t N = 1; |
| char *Buffer = static_cast<char *>(std::malloc(N)); |
| ASSERT_NE(Buffer, nullptr); |
| |
| int Status = 0; |
| char *Demangled = llvm::rustDemangle("_RNvC1a4main", Buffer, &N, &Status); |
| |
| EXPECT_EQ(Status, llvm::demangle_success); |
| EXPECT_STREQ(Demangled, "a::main"); |
| std::free(Demangled); |
| } |