|
9 | 9 | #include "llvm/DebugInfo/BTF/BTFContext.h" |
10 | 10 | #include "llvm/ObjectYAML/YAML.h" |
11 | 11 | #include "llvm/ObjectYAML/yaml2obj.h" |
| 12 | +#include "llvm/Support/SwapByteOrder.h" |
12 | 13 | #include "llvm/Testing/Support/Error.h" |
13 | 14 |
|
14 | 15 | using namespace llvm; |
@@ -112,8 +113,12 @@ struct MockData1 { |
112 | 113 | Yaml << R"( |
113 | 114 | !ELF |
114 | 115 | FileHeader: |
115 | | - Class: ELFCLASS64 |
116 | | - Data: ELFDATA2LSB |
| 116 | + Class: ELFCLASS64)"; |
| 117 | + if (sys::IsBigEndianHost) |
| 118 | + Yaml << "\n Data: ELFDATA2MSB"; |
| 119 | + else |
| 120 | + Yaml << "\n Data: ELFDATA2LSB"; |
| 121 | + Yaml << R"( |
117 | 122 | Type: ET_REL |
118 | 123 | Machine: EM_BPF |
119 | 124 | Sections: |
@@ -162,21 +167,21 @@ TEST(BTFParserTest, simpleCorrectInput) { |
162 | 167 | EXPECT_EQ(BTF.findString(sizeof(MockData1::B::S)), StringRef()); |
163 | 168 |
|
164 | 169 | const BTF::BPFLineInfo *I1 = BTF.findLineInfo({16, 1}); |
165 | | - EXPECT_TRUE(I1); |
| 170 | + ASSERT_TRUE(I1); |
166 | 171 | EXPECT_EQ(I1->getLine(), 7u); |
167 | 172 | EXPECT_EQ(I1->getCol(), 1u); |
168 | 173 | EXPECT_EQ(BTF.findString(I1->FileNameOff), "a.c"); |
169 | 174 | EXPECT_EQ(BTF.findString(I1->LineOff), "first line"); |
170 | 175 |
|
171 | 176 | const BTF::BPFLineInfo *I2 = BTF.findLineInfo({32, 1}); |
172 | | - EXPECT_TRUE(I2); |
| 177 | + ASSERT_TRUE(I2); |
173 | 178 | EXPECT_EQ(I2->getLine(), 14u); |
174 | 179 | EXPECT_EQ(I2->getCol(), 5u); |
175 | 180 | EXPECT_EQ(BTF.findString(I2->FileNameOff), "a.c"); |
176 | 181 | EXPECT_EQ(BTF.findString(I2->LineOff), "second line"); |
177 | 182 |
|
178 | 183 | const BTF::BPFLineInfo *I3 = BTF.findLineInfo({0, 2}); |
179 | | - EXPECT_TRUE(I3); |
| 184 | + ASSERT_TRUE(I3); |
180 | 185 | EXPECT_EQ(I3->getLine(), 42u); |
181 | 186 | EXPECT_EQ(I3->getCol(), 4u); |
182 | 187 | EXPECT_EQ(BTF.findString(I3->FileNameOff), "b.c"); |
|
0 commit comments