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
99
100
101
102
103
104
105
|
//===-- PrototypeTestGen.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 "utils/LibcTableGenUtil/APIIndexer.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/TableGen/Main.h"
#include "llvm/TableGen/Record.h"
namespace {
llvm::cl::list<std::string>
EntrypointNamesOption("e", llvm::cl::desc("<list of entrypoints>"),
llvm::cl::OneOrMore);
} // anonymous namespace
bool TestGeneratorMain(llvm::raw_ostream &OS, llvm::RecordKeeper &records) {
OS << "#include \"TypeTraits.h\"\n";
llvm_libc::APIIndexer G(records);
std::unordered_set<std::string> headerFileSet;
for (const auto &entrypoint : EntrypointNamesOption) {
auto match = G.FunctionToHeaderMap.find(entrypoint);
if (match == G.FunctionToHeaderMap.end()) {
auto objectMatch = G.ObjectToHeaderMap.find(entrypoint);
if (objectMatch != G.ObjectToHeaderMap.end()) {
headerFileSet.insert(objectMatch->second);
continue;
}
llvm::errs() << "ERROR: entrypoint '" << entrypoint
<< "' could not be found in spec in any public header\n";
return true;
}
headerFileSet.insert(match->second);
}
for (const auto &header : headerFileSet)
OS << "#include <" << header << ">\n";
OS << '\n';
OS << "int main() {\n";
for (const auto &entrypoint : EntrypointNamesOption) {
auto match = G.FunctionSpecMap.find(entrypoint);
if (match == G.FunctionSpecMap.end()) {
auto objectMatch = G.ObjectSpecMap.find(entrypoint);
if (objectMatch != G.ObjectSpecMap.end()) {
auto entrypointPtr = entrypoint + "_ptr";
llvm::Record *objectSpec = G.ObjectSpecMap[entrypoint];
auto objectType = objectSpec->getValueAsString("Type");
// We just make sure that the global object is present.
OS << " " << objectType << " *" << entrypointPtr << " = &"
<< entrypoint << ";\n";
OS << " ++" << entrypointPtr << ";\n"; // To avoid unused var warning.
continue;
}
llvm::errs() << "ERROR: entrypoint '" << entrypoint
<< "' could not be found in spec in any public header\n";
return true;
}
llvm::Record *functionSpec = match->second;
llvm::Record *retValSpec = functionSpec->getValueAsDef("Return");
std::string returnType =
G.getTypeAsString(retValSpec->getValueAsDef("ReturnType"));
// _Noreturn is an indication for the compiler that a function
// doesn't return, and isn't a type understood by c++ templates.
if (llvm::StringRef(returnType).contains("_Noreturn"))
returnType = "void";
OS << " static_assert(__llvm_libc::cpp::IsSame<" << returnType << '(';
auto args = functionSpec->getValueAsListOfDefs("Args");
for (size_t i = 0, size = args.size(); i < size; ++i) {
llvm::Record *argType = args[i]->getValueAsDef("ArgType");
OS << G.getTypeAsString(argType);
if (i < size - 1)
OS << ", ";
}
OS << "), decltype(" << entrypoint << ")>::Value, ";
OS << '"' << entrypoint
<< " prototype in TableGen does not match public header" << '"';
OS << ");\n";
}
OS << '\n';
OS << " return 0;\n";
OS << "}\n\n";
// We provide dummy malloc and free implementations to support the case
// when LLVM libc does to include them.
OS << "void *malloc(size_t) { return nullptr; }\n";
OS << "void free(void *) {}\n";
return false;
}
int main(int argc, char *argv[]) {
llvm::cl::ParseCommandLineOptions(argc, argv);
return TableGenMain(argv[0], TestGeneratorMain);
}
|
