aboutsummaryrefslogtreecommitdiff
path: root/backend/unittests/CodeGen/CompilationUnitTest.cpp
blob: a31c9a8a26dc7abac6a734cfef0c9207e13207da (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
// SPDX-License-Identifier: GPL-3.0-or-later

#include "rcc/CodeGen/CompilationUnit.h"

#include "rcc/CodeGen/MachineFunction.h"

#include <cstddef>
#include <iostream>
#include <utility>

namespace {

int fail(const char *Message) {
  std::cerr << Message << '\n';
  return 1;
}

} // namespace

int main() {
  rcc::CompilationUnit Unit;
  auto Empty = Unit.createMachineFunction("");
  if (Empty || Unit.functionCount() != 0)
    return fail("CompilationUnit must reject an empty function name atomically");

  auto First = Unit.createMachineFunction("first");
  if (!First || (*First)->stableID().value() != 1)
    return fail("CompilationUnit must allocate the first stable function identity");

  auto Duplicate = Unit.createMachineFunction("first");
  if (Duplicate || Unit.functionCount() != 1)
    return fail("CompilationUnit must reject duplicate function names atomically");

  auto Second = Unit.createMachineFunction("second");
  if (!Second || (*Second)->stableID().value() != 2)
    return fail("Rejected function creation must not consume a StableID");
  if (Unit.functionCount() != 2 || Unit.functionAt(0) != *First || Unit.functionAt(1) != *Second)
    return fail("CompilationUnit must preserve function document order");
  if (Unit.function("first") != *First || Unit.function("missing") != nullptr)
    return fail("CompilationUnit must provide exact name lookup");
  if (Unit.function((*Second)->stableID()) != *Second ||
      Unit.function(rcc::MachineFunctionStableID::fromValue(99)) != nullptr)
    return fail("CompilationUnit must provide exact StableID lookup");
  if (Unit.functionAt(2) != nullptr)
    return fail("CompilationUnit must reject an out-of-range ordered lookup");

  auto Zero = rcc::GlobalInitializer::zeroFill(4);
  if (!Zero)
    return fail("Global initializer creation must succeed");
  rcc::GlobalObjectDesc Object;
  Object.Symbol = "global";
  Object.ObjectType = rcc::GenericConstantType::byte();
  Object.Size = 4;
  Object.Initializer.emplace(std::move(*Zero));
  auto Global = Unit.addGlobalObject(std::move(Object));
  if (!Global || Unit.globalObjectCount() != 1 || Unit.globalObject("global") != *Global ||
      Unit.globalObjectAt(0) != *Global || Unit.globalObjectAt(1) != nullptr)
    return fail("CompilationUnit must own and index global objects");

  auto ConflictingFunction = Unit.createMachineFunction("global");
  if (ConflictingFunction || Unit.functionCount() != 2)
    return fail("CompilationUnit must reject function/global symbol conflicts atomically");

  rcc::GlobalObjectDesc ConflictingGlobal;
  ConflictingGlobal.Symbol = "first";
  ConflictingGlobal.ObjectType = rcc::GenericConstantType::byte();
  ConflictingGlobal.Size = 1;
  auto Byte = rcc::GlobalInitializer::bytes({std::byte{1}});
  if (!Byte)
    return fail("Byte initializer creation must succeed");
  ConflictingGlobal.Initializer.emplace(std::move(*Byte));
  if (Unit.addGlobalObject(std::move(ConflictingGlobal)) || Unit.globalObjectCount() != 1)
    return fail("CompilationUnit must reject global/function symbol conflicts atomically");

  auto MismatchedBytes = rcc::GlobalInitializer::bytes({std::byte{1}, std::byte{2}});
  if (!MismatchedBytes)
    return fail("Mismatched initializer construction must succeed before object verification");
  rcc::GlobalObjectDesc MismatchedObject;
  MismatchedObject.Symbol = "mismatched";
  MismatchedObject.ObjectType = rcc::GenericConstantType::byte();
  MismatchedObject.Size = 1;
  MismatchedObject.Initializer.emplace(std::move(*MismatchedBytes));
  if (Unit.addGlobalObject(std::move(MismatchedObject)) || Unit.globalObjectCount() != 1)
    return fail("CompilationUnit must reject mismatched global initializer sizes atomically");

  auto Verification = Unit.verify();
  if (!Verification)
    return fail("A valid CompilationUnit must pass verification");

  rcc::CompilationUnit Moved = std::move(Unit);
  if (Moved.functionCount() != 2 || Moved.function("first") != *First ||
      Moved.function("second") != *Second || Moved.globalObject("global") != *Global)
    return fail("Moving a CompilationUnit must preserve owned object addresses and indices");
  Verification = Moved.verify();
  if (!Verification)
    return fail("A moved CompilationUnit must pass verification");
  return 0;
}