1 //===- ConstantRangeTest.cpp - ConstantRange tests ------------------------===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 9 #include "llvm/IR/DataLayout.h" 10 #include "llvm/IR/GlobalVariable.h" 11 #include "llvm/IR/LLVMContext.h" 12 #include "llvm/IR/Module.h" 13 #include "llvm/IR/Type.h" 14 #include "llvm/TargetParser/Triple.h" 15 #include "llvm/Testing/Support/Error.h" 16 #include "gtest/gtest.h" 17 18 using namespace llvm; 19 20 namespace { 21 22 class DataLayoutTest : public ::testing::Test {}; 23 24 TEST(DataLayout, LayoutStringFormat) { 25 for (StringRef Str : {"", "e", "m:e", "m:e-e"}) 26 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 27 28 for (StringRef Str : {"-", "e-", "-m:e", "m:e--e"}) 29 EXPECT_THAT_EXPECTED( 30 DataLayout::parse(Str), 31 FailedWithMessage("empty specification is not allowed")); 32 } 33 34 TEST(DataLayoutTest, InvalidSpecifier) { 35 EXPECT_THAT_EXPECTED(DataLayout::parse("^"), 36 FailedWithMessage("unknown specifier '^'")); 37 EXPECT_THAT_EXPECTED(DataLayout::parse("I8:8"), 38 FailedWithMessage("unknown specifier 'I'")); 39 EXPECT_THAT_EXPECTED(DataLayout::parse("e-X"), 40 FailedWithMessage("unknown specifier 'X'")); 41 EXPECT_THAT_EXPECTED(DataLayout::parse("p0:32:32-64"), 42 FailedWithMessage("unknown specifier '6'")); 43 } 44 45 TEST(DataLayoutTest, ParseEndianness) { 46 EXPECT_THAT_EXPECTED(DataLayout::parse("e"), Succeeded()); 47 EXPECT_THAT_EXPECTED(DataLayout::parse("E"), Succeeded()); 48 49 for (StringRef Str : {"ee", "e0", "e:0", "E0:E", "El", "E:B"}) 50 EXPECT_THAT_EXPECTED( 51 DataLayout::parse(Str), 52 FailedWithMessage("malformed specification, must be just 'e' or 'E'")); 53 } 54 55 TEST(DataLayoutTest, ParseMangling) { 56 for (StringRef Str : {"m:a", "m:e", "m:l", "m:m", "m:o", "m:w", "m:x"}) 57 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 58 59 for (StringRef Str : {"m", "ms:m", "m:"}) 60 EXPECT_THAT_EXPECTED( 61 DataLayout::parse(Str), 62 FailedWithMessage( 63 "malformed specification, must be of the form \"m:<mangling>\"")); 64 65 for (StringRef Str : {"m:ms", "m:E", "m:0"}) 66 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), 67 FailedWithMessage("unknown mangling mode")); 68 } 69 70 TEST(DataLayoutTest, ParseStackNaturalAlign) { 71 for (StringRef Str : {"S8", "S32768"}) 72 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 73 74 EXPECT_THAT_EXPECTED( 75 DataLayout::parse("S"), 76 FailedWithMessage( 77 "malformed specification, must be of the form \"S<size>\"")); 78 79 for (StringRef Str : {"SX", "S0x20", "S65536"}) 80 EXPECT_THAT_EXPECTED( 81 DataLayout::parse(Str), 82 FailedWithMessage("stack natural alignment must be a 16-bit integer")); 83 84 EXPECT_THAT_EXPECTED( 85 DataLayout::parse("S0"), 86 FailedWithMessage("stack natural alignment must be non-zero")); 87 88 for (StringRef Str : {"S1", "S7", "S24", "S65535"}) 89 EXPECT_THAT_EXPECTED( 90 DataLayout::parse(Str), 91 FailedWithMessage("stack natural alignment must be a power of two " 92 "times the byte width")); 93 } 94 95 TEST(DataLayoutTest, ParseAddrSpace) { 96 for (StringRef Str : {"P0", "A0", "G0", "P1", "A1", "G1", "P16777215", 97 "A16777215", "G16777215"}) 98 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 99 100 for (StringRef Str : {"P", "A", "G"}) 101 EXPECT_THAT_EXPECTED( 102 DataLayout::parse(Str), 103 FailedWithMessage(("malformed specification, must be of the form \"" + 104 Twine(Str.front()) + "<address space>\"") 105 .str())); 106 107 for (StringRef Str : {"Px", "A0x1", "G16777216"}) 108 EXPECT_THAT_EXPECTED( 109 DataLayout::parse(Str), 110 FailedWithMessage("address space must be a 24-bit integer")); 111 } 112 113 TEST(DataLayoutTest, ParseFuncPtrSpec) { 114 for (StringRef Str : {"Fi8", "Fn16", "Fi32768", "Fn32768"}) 115 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 116 117 EXPECT_THAT_EXPECTED( 118 DataLayout::parse("F"), 119 FailedWithMessage( 120 "malformed specification, must be of the form \"F<type><abi>\"")); 121 122 EXPECT_THAT_EXPECTED( 123 DataLayout::parse("FN"), 124 FailedWithMessage("unknown function pointer alignment type 'N'")); 125 EXPECT_THAT_EXPECTED( 126 DataLayout::parse("F32"), 127 FailedWithMessage("unknown function pointer alignment type '3'")); 128 129 for (StringRef Str : {"Fi", "Fn"}) 130 EXPECT_THAT_EXPECTED( 131 DataLayout::parse(Str), 132 FailedWithMessage("ABI alignment component cannot be empty")); 133 134 for (StringRef Str : {"Fii", "Fn32x", "Fi65536", "Fn65536"}) 135 EXPECT_THAT_EXPECTED( 136 DataLayout::parse(Str), 137 FailedWithMessage("ABI alignment must be a 16-bit integer")); 138 139 for (StringRef Str : {"Fi0", "Fn0"}) 140 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), 141 FailedWithMessage("ABI alignment must be non-zero")); 142 143 for (StringRef Str : {"Fi12", "Fn24"}) 144 EXPECT_THAT_EXPECTED( 145 DataLayout::parse(Str), 146 FailedWithMessage( 147 "ABI alignment must be a power of two times the byte width")); 148 } 149 150 class DataLayoutPrimitiveSpecificationTest 151 : public DataLayoutTest, 152 public ::testing::WithParamInterface<char> { 153 char Specifier; 154 155 public: 156 DataLayoutPrimitiveSpecificationTest() : Specifier(GetParam()) {} 157 158 std::string format(StringRef Str) const { 159 std::string Res = Str.str(); 160 std::replace(Res.begin(), Res.end(), '!', Specifier); 161 return Res; 162 } 163 }; 164 165 INSTANTIATE_TEST_SUITE_P(PrmitiveSpecifiers, 166 DataLayoutPrimitiveSpecificationTest, 167 ::testing::Values('i', 'f', 'v')); 168 169 TEST_P(DataLayoutPrimitiveSpecificationTest, ParsePrimitiveSpec) { 170 for (StringRef Str : 171 {"!1:16", "!8:8:8", "!16:32:64", "!16777215:32768:32768"}) 172 EXPECT_THAT_EXPECTED(DataLayout::parse(format(Str)), Succeeded()); 173 174 for (StringRef Str : {"!", "!1", "!32:32:32:32", "!16:32:64:128"}) 175 EXPECT_THAT_EXPECTED( 176 DataLayout::parse(format(Str)), 177 FailedWithMessage(format("malformed specification, must be of the form " 178 "\"!<size>:<abi>[:<pref>]\""))); 179 180 // size 181 for (StringRef Str : {"!:8", "!:16:16", "!:32:64"}) 182 EXPECT_THAT_EXPECTED(DataLayout::parse(format(Str)), 183 FailedWithMessage("size component cannot be empty")); 184 185 for (StringRef Str : 186 {"!0:8", "!0x8:8", "!x:8:8", "!0:16:32", "!16777216:64:64"}) 187 EXPECT_THAT_EXPECTED( 188 DataLayout::parse(format(Str)), 189 FailedWithMessage("size must be a non-zero 24-bit integer")); 190 191 // ABI alignment 192 for (StringRef Str : {"!8:", "!16::16", "!32::64"}) 193 EXPECT_THAT_EXPECTED( 194 DataLayout::parse(format(Str)), 195 FailedWithMessage("ABI alignment component cannot be empty")); 196 197 for (StringRef Str : {"!1:x", "!8:8x:8", "!16:65536:65536"}) 198 EXPECT_THAT_EXPECTED( 199 DataLayout::parse(format(Str)), 200 FailedWithMessage("ABI alignment must be a 16-bit integer")); 201 202 for (StringRef Str : {"!8:0", "!16:0:16", "!32:0:64"}) 203 EXPECT_THAT_EXPECTED(DataLayout::parse(format(Str)), 204 FailedWithMessage("ABI alignment must be non-zero")); 205 206 for (StringRef Str : {"!1:1", "!8:4", "!16:6:16", "!32:24:64"}) 207 EXPECT_THAT_EXPECTED( 208 DataLayout::parse(format(Str)), 209 FailedWithMessage( 210 "ABI alignment must be a power of two times the byte width")); 211 212 // preferred alignment 213 for (StringRef Str : {"!1:8:", "!16:16:", "!64:32:"}) 214 EXPECT_THAT_EXPECTED( 215 DataLayout::parse(format(Str)), 216 FailedWithMessage("preferred alignment component cannot be empty")); 217 218 for (StringRef Str : {"!1:8:x", "!8:8:0x8", "!16:32:65536"}) 219 EXPECT_THAT_EXPECTED( 220 DataLayout::parse(format(Str)), 221 FailedWithMessage("preferred alignment must be a 16-bit integer")); 222 223 for (StringRef Str : {"!8:8:0", "!32:16:0"}) 224 EXPECT_THAT_EXPECTED( 225 DataLayout::parse(format(Str)), 226 FailedWithMessage("preferred alignment must be non-zero")); 227 228 for (StringRef Str : {"!1:8:12", "!8:8:17", "!16:32:40"}) 229 EXPECT_THAT_EXPECTED( 230 DataLayout::parse(format(Str)), 231 FailedWithMessage( 232 "preferred alignment must be a power of two times the byte width")); 233 234 for (StringRef Str : {"!1:16:8", "!64:32:16"}) 235 EXPECT_THAT_EXPECTED( 236 DataLayout::parse(format(Str)), 237 FailedWithMessage( 238 "preferred alignment cannot be less than the ABI alignment")); 239 240 // Additional check for byte-sized integer. 241 if (GetParam() == 'i') { 242 for (StringRef Str : {"!8:16", "!8:16:8", "!8:16:32"}) 243 EXPECT_THAT_EXPECTED(DataLayout::parse(format(Str)), 244 FailedWithMessage("i8 must be 8-bit aligned")); 245 } 246 } 247 248 TEST(DataLayoutTest, ParseAggregateSpec) { 249 for (StringRef Str : {"a:8", "a:0:16", "a0:32:64", "a:32768:32768"}) 250 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 251 252 for (StringRef Str : {"a", "a0", "a:32:32:32", "a0:32:64:128"}) 253 EXPECT_THAT_EXPECTED( 254 DataLayout::parse(Str), 255 FailedWithMessage("malformed specification, must be of the form " 256 "\"a:<abi>[:<pref>]\"")); 257 258 // size 259 for (StringRef Str : {"a1:8", "a0x0:8", "ax:16:32"}) 260 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), 261 FailedWithMessage("size must be zero")); 262 263 // ABI alignment 264 for (StringRef Str : {"a:", "a0:", "a::32"}) 265 EXPECT_THAT_EXPECTED( 266 DataLayout::parse(Str), 267 FailedWithMessage("ABI alignment component cannot be empty")); 268 269 for (StringRef Str : {"a:x", "a0:0x0", "a:65536", "a0:65536:65536"}) 270 EXPECT_THAT_EXPECTED( 271 DataLayout::parse(Str), 272 FailedWithMessage("ABI alignment must be a 16-bit integer")); 273 274 for (StringRef Str : {"a:1", "a:4", "a:9:16", "a0:24:32"}) 275 EXPECT_THAT_EXPECTED( 276 DataLayout::parse(Str), 277 FailedWithMessage( 278 "ABI alignment must be a power of two times the byte width")); 279 280 // preferred alignment 281 for (StringRef Str : {"a:8:", "a0:16:", "a0:0:"}) 282 EXPECT_THAT_EXPECTED( 283 DataLayout::parse(Str), 284 FailedWithMessage("preferred alignment component cannot be empty")); 285 286 for (StringRef Str : {"a:16:x", "a0:8:0x8", "a:16:65536"}) 287 EXPECT_THAT_EXPECTED( 288 DataLayout::parse(Str), 289 FailedWithMessage("preferred alignment must be a 16-bit integer")); 290 291 for (StringRef Str : {"a:0:0", "a0:16:0"}) 292 EXPECT_THAT_EXPECTED( 293 DataLayout::parse(Str), 294 FailedWithMessage("preferred alignment must be non-zero")); 295 296 for (StringRef Str : {"a:8:12", "a:16:17", "a0:32:40"}) 297 EXPECT_THAT_EXPECTED( 298 DataLayout::parse(Str), 299 FailedWithMessage( 300 "preferred alignment must be a power of two times the byte width")); 301 302 for (StringRef Str : {"a:16:8", "a0:32:16"}) 303 EXPECT_THAT_EXPECTED( 304 DataLayout::parse(Str), 305 FailedWithMessage( 306 "preferred alignment cannot be less than the ABI alignment")); 307 } 308 309 TEST(DataLayout, ParsePointerSpec) { 310 for (StringRef Str : 311 {"p:16:8", "p:16:16:64", "p:32:64:64:32", "p0:32:64", "p42:64:32:32", 312 "p16777215:32:32:64:8", "p16777215:16777215:32768:32768:16777215"}) 313 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 314 315 for (StringRef Str : 316 {"p", "p0", "p:32", "p0:32", "p:32:32:32:32:32", "p0:32:32:32:32:32"}) 317 EXPECT_THAT_EXPECTED( 318 DataLayout::parse(Str), 319 FailedWithMessage("malformed specification, must be of the form " 320 "\"p[<n>]:<size>:<abi>[:<pref>[:<idx>]]\"")); 321 322 // address space 323 for (StringRef Str : {"p0x0:32:32", "px:32:32:32", "p16777216:32:32:32:32"}) 324 EXPECT_THAT_EXPECTED( 325 DataLayout::parse(Str), 326 FailedWithMessage("address space must be a 24-bit integer")); 327 328 // pointer size 329 for (StringRef Str : {"p::32", "p0::32"}) 330 EXPECT_THAT_EXPECTED( 331 DataLayout::parse(Str), 332 FailedWithMessage("pointer size component cannot be empty")); 333 334 for (StringRef Str : {"p:0:32", "p0:0x1:32:32", "p42:16777216:32:32:32"}) 335 EXPECT_THAT_EXPECTED( 336 DataLayout::parse(Str), 337 FailedWithMessage("pointer size must be a non-zero 24-bit integer")); 338 339 // ABI alignment 340 for (StringRef Str : {"p:32:", "p0:32::32", "p42:32::32:32"}) 341 EXPECT_THAT_EXPECTED( 342 DataLayout::parse(Str), 343 FailedWithMessage("ABI alignment component cannot be empty")); 344 345 for (StringRef Str : {"p:32:x", "p0:32:0x20:32", "p42:32:65536:32:32"}) 346 EXPECT_THAT_EXPECTED( 347 DataLayout::parse(Str), 348 FailedWithMessage("ABI alignment must be a 16-bit integer")); 349 350 for (StringRef Str : {"p:32:0", "p0:32:0:32", "p42:32:0:32:32"}) 351 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), 352 FailedWithMessage("ABI alignment must be non-zero")); 353 354 for (StringRef Str : {"p:32:4", "p42:32:24:32", "p0:32:65535:32:32"}) 355 EXPECT_THAT_EXPECTED( 356 DataLayout::parse(Str), 357 FailedWithMessage( 358 "ABI alignment must be a power of two times the byte width")); 359 360 // preferred alignment 361 for (StringRef Str : {"p:32:32:", "p0:32:32:", "p42:32:32::32"}) 362 EXPECT_THAT_EXPECTED( 363 DataLayout::parse(Str), 364 FailedWithMessage("preferred alignment component cannot be empty")); 365 366 for (StringRef Str : {"p:32:32:x", "p0:32:32:0x20", "p42:32:32:65536:32"}) 367 EXPECT_THAT_EXPECTED( 368 DataLayout::parse(Str), 369 FailedWithMessage("preferred alignment must be a 16-bit integer")); 370 371 for (StringRef Str : {"p:32:32:0", "p0:32:32:0", "p42:32:32:0:32"}) 372 EXPECT_THAT_EXPECTED( 373 DataLayout::parse(Str), 374 FailedWithMessage("preferred alignment must be non-zero")); 375 376 for (StringRef Str : {"p:32:32:4", "p0:32:32:24", "p42:32:32:65535:32"}) 377 EXPECT_THAT_EXPECTED( 378 DataLayout::parse(Str), 379 FailedWithMessage( 380 "preferred alignment must be a power of two times the byte width")); 381 382 for (StringRef Str : {"p:64:64:32", "p0:16:32:16:16"}) 383 EXPECT_THAT_EXPECTED( 384 DataLayout::parse(Str), 385 FailedWithMessage( 386 "preferred alignment cannot be less than the ABI alignment")); 387 388 // index size 389 for (StringRef Str : {"p:32:32:32:", "p0:32:32:32:"}) 390 EXPECT_THAT_EXPECTED( 391 DataLayout::parse(Str), 392 FailedWithMessage("index size component cannot be empty")); 393 394 for (StringRef Str : 395 {"p:32:32:32:0", "p0:32:32:32:0x20", "p42:32:32:32:16777216"}) 396 EXPECT_THAT_EXPECTED( 397 DataLayout::parse(Str), 398 FailedWithMessage("index size must be a non-zero 24-bit integer")); 399 400 for (StringRef Str : {"p:16:16:16:17", "p0:32:64:64:64", "p42:16:64:64:32"}) 401 EXPECT_THAT_EXPECTED( 402 DataLayout::parse(Str), 403 FailedWithMessage("index size cannot be larger than the pointer size")); 404 } 405 406 TEST(DataLayoutTest, ParseNativeIntegersSpec) { 407 for (StringRef Str : {"n1", "n1:8", "n24:12:16777215"}) 408 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 409 410 for (StringRef Str : {"n", "n1:", "n:8", "n16::32"}) 411 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), 412 FailedWithMessage("size component cannot be empty")); 413 414 for (StringRef Str : {"n0", "n0x8:16", "n8:0", "n16:0:32", "n16777216", 415 "n16:16777216", "n32:64:16777216"}) 416 EXPECT_THAT_EXPECTED( 417 DataLayout::parse(Str), 418 FailedWithMessage("size must be a non-zero 24-bit integer")); 419 } 420 421 TEST(DataLayout, ParseNonIntegralAddrSpace) { 422 for (StringRef Str : {"ni:1", "ni:16777215", "ni:1:16777215"}) 423 EXPECT_THAT_EXPECTED(DataLayout::parse(Str), Succeeded()); 424 425 for (StringRef Str : {"ni", "ni42", "nix"}) 426 EXPECT_THAT_EXPECTED( 427 DataLayout::parse(Str), 428 FailedWithMessage("malformed specification, must be of the form " 429 "\"ni:<address space>[:<address space>]...\"")); 430 431 for (StringRef Str : {"ni:", "ni::42", "ni:42:"}) 432 EXPECT_THAT_EXPECTED( 433 DataLayout::parse(Str), 434 FailedWithMessage("address space component cannot be empty")); 435 436 for (StringRef Str : {"ni:x", "ni:42:0x1", "ni:16777216", "ni:42:16777216"}) 437 EXPECT_THAT_EXPECTED( 438 DataLayout::parse(Str), 439 FailedWithMessage("address space must be a 24-bit integer")); 440 441 for (StringRef Str : {"ni:0", "ni:42:0"}) 442 EXPECT_THAT_EXPECTED( 443 DataLayout::parse(Str), 444 FailedWithMessage("address space 0 cannot be non-integral")); 445 } 446 447 TEST(DataLayout, GetPointerSizeInBits) { 448 std::tuple<StringRef, unsigned, unsigned, unsigned> Cases[] = { 449 {"", 64, 64, 64}, 450 {"p:16:32", 16, 16, 16}, 451 {"p0:32:64", 32, 32, 32}, 452 {"p1:16:32", 64, 16, 64}, 453 {"p1:31:32-p2:15:16:16:14", 64, 31, 15}, 454 }; 455 for (auto [Layout, V0, V1, V2] : Cases) { 456 DataLayout DL = cantFail(DataLayout::parse(Layout)); 457 EXPECT_EQ(DL.getPointerSizeInBits(0), V0) << Layout; 458 EXPECT_EQ(DL.getPointerSizeInBits(1), V1) << Layout; 459 EXPECT_EQ(DL.getPointerSizeInBits(2), V2) << Layout; 460 } 461 } 462 463 TEST(DataLayout, GetPointerSize) { 464 std::tuple<StringRef, unsigned, unsigned, unsigned> Cases[] = { 465 {"", 8, 8, 8}, 466 {"p:16:32", 2, 2, 2}, 467 {"p0:32:64", 4, 4, 4}, 468 {"p1:17:32", 8, 3, 8}, 469 {"p1:31:64-p2:23:8:16:9", 8, 4, 3}, 470 }; 471 for (auto [Layout, V0, V1, V2] : Cases) { 472 DataLayout DL = cantFail(DataLayout::parse(Layout)); 473 EXPECT_EQ(DL.getPointerSize(0), V0) << Layout; 474 EXPECT_EQ(DL.getPointerSize(1), V1) << Layout; 475 EXPECT_EQ(DL.getPointerSize(2), V2) << Layout; 476 } 477 } 478 479 TEST(DataLayout, GetIndexSizeInBits) { 480 std::tuple<StringRef, unsigned, unsigned, unsigned> Cases[] = { 481 {"", 64, 64, 64}, 482 {"p:16:32", 16, 16, 16}, 483 {"p0:32:64", 32, 32, 32}, 484 {"p1:16:32:32:10", 64, 10, 64}, 485 {"p1:31:32:64:20-p2:17:16:16:15", 64, 20, 15}, 486 }; 487 for (auto [Layout, V0, V1, V2] : Cases) { 488 DataLayout DL = cantFail(DataLayout::parse(Layout)); 489 EXPECT_EQ(DL.getIndexSizeInBits(0), V0) << Layout; 490 EXPECT_EQ(DL.getIndexSizeInBits(1), V1) << Layout; 491 EXPECT_EQ(DL.getIndexSizeInBits(2), V2) << Layout; 492 } 493 } 494 495 TEST(DataLayout, GetIndexSize) { 496 std::tuple<StringRef, unsigned, unsigned, unsigned> Cases[] = { 497 {"", 8, 8, 8}, 498 {"p:16:32", 2, 2, 2}, 499 {"p0:27:64", 4, 4, 4}, 500 {"p1:19:32:64:5", 8, 1, 8}, 501 {"p1:33:32:64:23-p2:21:8:16:13", 8, 3, 2}, 502 }; 503 for (auto [Layout, V0, V1, V2] : Cases) { 504 DataLayout DL = cantFail(DataLayout::parse(Layout)); 505 EXPECT_EQ(DL.getIndexSize(0), V0) << Layout; 506 EXPECT_EQ(DL.getIndexSize(1), V1) << Layout; 507 EXPECT_EQ(DL.getIndexSize(2), V2) << Layout; 508 } 509 } 510 511 TEST(DataLayout, GetPointerABIAlignment) { 512 std::tuple<StringRef, unsigned, unsigned, unsigned> Cases[] = { 513 {"", 8, 8, 8}, 514 {"p:16:32", 4, 4, 4}, 515 {"p0:16:32:64", 4, 4, 4}, 516 {"p1:32:16:64", 8, 2, 8}, 517 {"p1:33:16:32:15-p2:23:8:16:9", 8, 2, 1}, 518 }; 519 for (auto [Layout, V0, V1, V2] : Cases) { 520 DataLayout DL = cantFail(DataLayout::parse(Layout)); 521 EXPECT_EQ(DL.getPointerABIAlignment(0).value(), V0) << Layout; 522 EXPECT_EQ(DL.getPointerABIAlignment(1).value(), V1) << Layout; 523 EXPECT_EQ(DL.getPointerABIAlignment(2).value(), V2) << Layout; 524 } 525 } 526 527 TEST(DataLayout, GetPointerPrefAlignment) { 528 std::tuple<StringRef, unsigned, unsigned, unsigned> Cases[] = { 529 {"", 8, 8, 8}, 530 {"p:16:32", 4, 4, 4}, 531 {"p0:8:16:32", 4, 4, 4}, 532 {"p1:32:8:16", 8, 2, 8}, 533 {"p1:33:8:16:31-p2:23:8:32:17", 8, 2, 4}, 534 }; 535 for (auto [Layout, V0, V1, V2] : Cases) { 536 DataLayout DL = cantFail(DataLayout::parse(Layout)); 537 EXPECT_EQ(DL.getPointerPrefAlignment(0).value(), V0) << Layout; 538 EXPECT_EQ(DL.getPointerPrefAlignment(1).value(), V1) << Layout; 539 EXPECT_EQ(DL.getPointerPrefAlignment(2).value(), V2) << Layout; 540 } 541 } 542 543 TEST(DataLayout, IsNonIntegralAddressSpace) { 544 DataLayout Default; 545 EXPECT_THAT(Default.getNonIntegralAddressSpaces(), ::testing::SizeIs(0)); 546 EXPECT_FALSE(Default.isNonIntegralAddressSpace(0)); 547 EXPECT_FALSE(Default.isNonIntegralAddressSpace(1)); 548 549 DataLayout Custom = cantFail(DataLayout::parse("ni:2:16777215")); 550 EXPECT_THAT(Custom.getNonIntegralAddressSpaces(), 551 ::testing::ElementsAreArray({2U, 16777215U})); 552 EXPECT_FALSE(Custom.isNonIntegralAddressSpace(0)); 553 EXPECT_FALSE(Custom.isNonIntegralAddressSpace(1)); 554 EXPECT_TRUE(Custom.isNonIntegralAddressSpace(2)); 555 EXPECT_TRUE(Custom.isNonIntegralAddressSpace(16777215)); 556 } 557 558 TEST(DataLayoutTest, CopyAssignmentInvalidatesStructLayout) { 559 DataLayout DL1 = cantFail(DataLayout::parse("p:32:32")); 560 DataLayout DL2 = cantFail(DataLayout::parse("p:64:64")); 561 562 LLVMContext Ctx; 563 StructType *Ty = StructType::get(PointerType::getUnqual(Ctx)); 564 565 // Initialize struct layout caches. 566 EXPECT_EQ(DL1.getStructLayout(Ty)->getSizeInBits(), 32U); 567 EXPECT_EQ(DL1.getStructLayout(Ty)->getAlignment(), Align(4)); 568 EXPECT_EQ(DL2.getStructLayout(Ty)->getSizeInBits(), 64U); 569 EXPECT_EQ(DL2.getStructLayout(Ty)->getAlignment(), Align(8)); 570 571 // The copy should invalidate DL1's cache. 572 DL1 = DL2; 573 EXPECT_EQ(DL1.getStructLayout(Ty)->getSizeInBits(), 64U); 574 EXPECT_EQ(DL1.getStructLayout(Ty)->getAlignment(), Align(8)); 575 EXPECT_EQ(DL2.getStructLayout(Ty)->getSizeInBits(), 64U); 576 EXPECT_EQ(DL2.getStructLayout(Ty)->getAlignment(), Align(8)); 577 } 578 579 TEST(DataLayoutTest, FunctionPtrAlign) { 580 EXPECT_EQ(MaybeAlign(0), DataLayout("").getFunctionPtrAlign()); 581 EXPECT_EQ(MaybeAlign(1), DataLayout("Fi8").getFunctionPtrAlign()); 582 EXPECT_EQ(MaybeAlign(2), DataLayout("Fi16").getFunctionPtrAlign()); 583 EXPECT_EQ(MaybeAlign(4), DataLayout("Fi32").getFunctionPtrAlign()); 584 EXPECT_EQ(MaybeAlign(8), DataLayout("Fi64").getFunctionPtrAlign()); 585 EXPECT_EQ(MaybeAlign(1), DataLayout("Fn8").getFunctionPtrAlign()); 586 EXPECT_EQ(MaybeAlign(2), DataLayout("Fn16").getFunctionPtrAlign()); 587 EXPECT_EQ(MaybeAlign(4), DataLayout("Fn32").getFunctionPtrAlign()); 588 EXPECT_EQ(MaybeAlign(8), DataLayout("Fn64").getFunctionPtrAlign()); 589 EXPECT_EQ(DataLayout::FunctionPtrAlignType::Independent, 590 DataLayout("").getFunctionPtrAlignType()); 591 EXPECT_EQ(DataLayout::FunctionPtrAlignType::Independent, 592 DataLayout("Fi8").getFunctionPtrAlignType()); 593 EXPECT_EQ(DataLayout::FunctionPtrAlignType::MultipleOfFunctionAlign, 594 DataLayout("Fn8").getFunctionPtrAlignType()); 595 EXPECT_EQ(DataLayout("Fi8"), DataLayout("Fi8")); 596 EXPECT_NE(DataLayout("Fi8"), DataLayout("Fi16")); 597 EXPECT_NE(DataLayout("Fi8"), DataLayout("Fn8")); 598 599 DataLayout a(""), b("Fi8"), c("Fn8"); 600 EXPECT_NE(a, b); 601 EXPECT_NE(a, c); 602 EXPECT_NE(b, c); 603 604 a = b; 605 EXPECT_EQ(a, b); 606 a = c; 607 EXPECT_EQ(a, c); 608 } 609 610 TEST(DataLayoutTest, ValueOrABITypeAlignment) { 611 const DataLayout DL("Fi8"); 612 LLVMContext Context; 613 Type *const FourByteAlignType = Type::getInt32Ty(Context); 614 EXPECT_EQ(Align(16), 615 DL.getValueOrABITypeAlignment(MaybeAlign(16), FourByteAlignType)); 616 EXPECT_EQ(Align(4), 617 DL.getValueOrABITypeAlignment(MaybeAlign(), FourByteAlignType)); 618 } 619 620 TEST(DataLayoutTest, GlobalsAddressSpace) { 621 // When not explicitly defined the globals address space should be zero: 622 EXPECT_EQ(DataLayout("").getDefaultGlobalsAddressSpace(), 0u); 623 EXPECT_EQ(DataLayout("P1-A2").getDefaultGlobalsAddressSpace(), 0u); 624 EXPECT_EQ(DataLayout("G2").getDefaultGlobalsAddressSpace(), 2u); 625 // Check that creating a GlobalVariable without an explicit address space 626 // in a module with a default globals address space respects that default: 627 LLVMContext Context; 628 std::unique_ptr<Module> M(new Module("MyModule", Context)); 629 // Default is globals in address space zero: 630 auto *Int32 = Type::getInt32Ty(Context); 631 auto *DefaultGlobal1 = new GlobalVariable( 632 *M, Int32, false, GlobalValue::ExternalLinkage, nullptr); 633 EXPECT_EQ(DefaultGlobal1->getAddressSpace(), 0u); 634 auto *ExplicitGlobal1 = new GlobalVariable( 635 *M, Int32, false, GlobalValue::ExternalLinkage, nullptr, "", nullptr, 636 GlobalValue::NotThreadLocal, 123); 637 EXPECT_EQ(ExplicitGlobal1->getAddressSpace(), 123u); 638 639 // When using a datalayout with the global address space set to 200, global 640 // variables should default to 200 641 M->setDataLayout("G200"); 642 auto *DefaultGlobal2 = new GlobalVariable( 643 *M, Int32, false, GlobalValue::ExternalLinkage, nullptr); 644 EXPECT_EQ(DefaultGlobal2->getAddressSpace(), 200u); 645 auto *ExplicitGlobal2 = new GlobalVariable( 646 *M, Int32, false, GlobalValue::ExternalLinkage, nullptr, "", nullptr, 647 GlobalValue::NotThreadLocal, 123); 648 EXPECT_EQ(ExplicitGlobal2->getAddressSpace(), 123u); 649 } 650 651 TEST(DataLayoutTest, VectorAlign) { 652 Expected<DataLayout> DL = DataLayout::parse("v64:64"); 653 EXPECT_THAT_EXPECTED(DL, Succeeded()); 654 655 LLVMContext Context; 656 Type *const FloatTy = Type::getFloatTy(Context); 657 Type *const V8F32Ty = FixedVectorType::get(FloatTy, 8); 658 659 // The alignment for a vector type larger than any specified vector type uses 660 // the natural alignment as a fallback. 661 EXPECT_EQ(Align(4 * 8), DL->getABITypeAlign(V8F32Ty)); 662 EXPECT_EQ(Align(4 * 8), DL->getPrefTypeAlign(V8F32Ty)); 663 } 664 665 TEST(DataLayoutTest, UEFI) { 666 Triple TT = Triple("x86_64-unknown-uefi"); 667 668 // Test UEFI X86_64 Mangling Component. 669 EXPECT_STREQ(DataLayout::getManglingComponent(TT), "-m:w"); 670 } 671 672 } // anonymous namespace 673