1 //===- llvm/unittest/VMCore/Metadata.cpp - Metadata unit tests ------------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 10 #include "gtest/gtest.h" 11 #include "llvm/Constants.h" 12 #include "llvm/Instructions.h" 13 #include "llvm/LLVMContext.h" 14 #include "llvm/Metadata.h" 15 #include "llvm/Module.h" 16 #include "llvm/Type.h" 17 #include "llvm/Support/raw_ostream.h" 18 #include "llvm/Support/ValueHandle.h" 19 using namespace llvm; 20 21 namespace { 22 23 class MetadataTest : public testing::Test { 24 protected: 25 LLVMContext Context; 26 }; 27 typedef MetadataTest MDStringTest; 28 29 // Test that construction of MDString with different value produces different 30 // MDString objects, even with the same string pointer and nulls in the string. 31 TEST_F(MDStringTest, CreateDifferent) { 32 char x[3] = { 'f', 0, 'A' }; 33 MDString *s1 = MDString::get(Context, StringRef(&x[0], 3)); 34 x[2] = 'B'; 35 MDString *s2 = MDString::get(Context, StringRef(&x[0], 3)); 36 EXPECT_NE(s1, s2); 37 } 38 39 // Test that creation of MDStrings with the same string contents produces the 40 // same MDString object, even with different pointers. 41 TEST_F(MDStringTest, CreateSame) { 42 char x[4] = { 'a', 'b', 'c', 'X' }; 43 char y[4] = { 'a', 'b', 'c', 'Y' }; 44 45 MDString *s1 = MDString::get(Context, StringRef(&x[0], 3)); 46 MDString *s2 = MDString::get(Context, StringRef(&y[0], 3)); 47 EXPECT_EQ(s1, s2); 48 } 49 50 // Test that MDString prints out the string we fed it. 51 TEST_F(MDStringTest, PrintingSimple) { 52 char *str = new char[13]; 53 strncpy(str, "testing 1 2 3", 13); 54 MDString *s = MDString::get(Context, StringRef(str, 13)); 55 strncpy(str, "aaaaaaaaaaaaa", 13); 56 delete[] str; 57 58 std::string Str; 59 raw_string_ostream oss(Str); 60 s->print(oss); 61 EXPECT_STREQ("metadata !\"testing 1 2 3\"", oss.str().c_str()); 62 } 63 64 // Test printing of MDString with non-printable characters. 65 TEST_F(MDStringTest, PrintingComplex) { 66 char str[5] = {0, '\n', '"', '\\', (char)-1}; 67 MDString *s = MDString::get(Context, StringRef(str+0, 5)); 68 std::string Str; 69 raw_string_ostream oss(Str); 70 s->print(oss); 71 EXPECT_STREQ("metadata !\"\\00\\0A\\22\\5C\\FF\"", oss.str().c_str()); 72 } 73 74 typedef MetadataTest MDNodeTest; 75 76 // Test the two constructors, and containing other Constants. 77 TEST_F(MDNodeTest, Simple) { 78 char x[3] = { 'a', 'b', 'c' }; 79 char y[3] = { '1', '2', '3' }; 80 81 MDString *s1 = MDString::get(Context, StringRef(&x[0], 3)); 82 MDString *s2 = MDString::get(Context, StringRef(&y[0], 3)); 83 ConstantInt *CI = ConstantInt::get(getGlobalContext(), APInt(8, 0)); 84 85 std::vector<Value *> V; 86 V.push_back(s1); 87 V.push_back(CI); 88 V.push_back(s2); 89 90 MDNode *n1 = MDNode::get(Context, V); 91 Value *const c1 = n1; 92 MDNode *n2 = MDNode::get(Context, c1); 93 MDNode *n3 = MDNode::get(Context, V); 94 EXPECT_NE(n1, n2); 95 #ifdef ENABLE_MDNODE_UNIQUING 96 EXPECT_EQ(n1, n3); 97 #else 98 (void) n3; 99 #endif 100 101 EXPECT_EQ(3u, n1->getNumOperands()); 102 EXPECT_EQ(s1, n1->getOperand(0)); 103 EXPECT_EQ(CI, n1->getOperand(1)); 104 EXPECT_EQ(s2, n1->getOperand(2)); 105 106 EXPECT_EQ(1u, n2->getNumOperands()); 107 EXPECT_EQ(n1, n2->getOperand(0)); 108 } 109 110 TEST_F(MDNodeTest, Delete) { 111 Constant *C = ConstantInt::get(Type::getInt32Ty(getGlobalContext()), 1); 112 Instruction *I = new BitCastInst(C, Type::getInt32Ty(getGlobalContext())); 113 114 Value *const V = I; 115 MDNode *n = MDNode::get(Context, V); 116 WeakVH wvh = n; 117 118 EXPECT_EQ(n, wvh); 119 120 delete I; 121 } 122 123 TEST(NamedMDNodeTest, Search) { 124 LLVMContext Context; 125 Constant *C = ConstantInt::get(Type::getInt32Ty(Context), 1); 126 Constant *C2 = ConstantInt::get(Type::getInt32Ty(Context), 2); 127 128 Value *const V = C; 129 Value *const V2 = C2; 130 MDNode *n = MDNode::get(Context, V); 131 MDNode *n2 = MDNode::get(Context, V2); 132 133 Module M("MyModule", Context); 134 const char *Name = "llvm.NMD1"; 135 NamedMDNode *NMD = M.getOrInsertNamedMetadata(Name); 136 NMD->addOperand(n); 137 NMD->addOperand(n2); 138 139 std::string Str; 140 raw_string_ostream oss(Str); 141 NMD->print(oss); 142 EXPECT_STREQ("!llvm.NMD1 = !{!0, !1}\n", 143 oss.str().c_str()); 144 } 145 } 146
