commit 7f73f51aada9004b29e9e70988a4f74c0933e7bb
author Siva Chandra <sivachandra@google.com> Wed Jan 27 11:08:31 2021 -0800
committer Copybara-Service <copybara-worker@google.com> Thu Jan 28 12:33:23 2021 -0800
tree 2fd1b18a397773c10960a7defe81f9f94c480373
parent c4402620a2d382793c1ddf1662d1d7fa9a260e6c

[libc] Include only the relevant header files in the integration test.

GitOrigin-RevId: d90bb66dd931a45cae3399dc76942bf8e412fa7b

config/linux/aarch64/headers.txt

diff --git a/config/linux/aarch64/headers.txt b/config/linux/aarch64/headers.txt
index a40f4b8..6aa6681 100644
--- a/config/linux/aarch64/headers.txt
+++ b/config/linux/aarch64/headers.txt
@@ -1,5 +1,6 @@
 set(PUBLIC_HEADERS
     libc.include.errno
     libc.include.math
+    libc.include.stdlib
     libc.include.sys_syscall
 )

utils/HdrGen/PrototypeTestGen/PrototypeTestGen.cpp

diff --git a/utils/HdrGen/PrototypeTestGen/PrototypeTestGen.cpp b/utils/HdrGen/PrototypeTestGen/PrototypeTestGen.cpp
index 918b520..e6b4353 100644
--- a/utils/HdrGen/PrototypeTestGen/PrototypeTestGen.cpp
+++ b/utils/HdrGen/PrototypeTestGen/PrototypeTestGen.cpp
@@ -24,9 +24,21 @@
 bool TestGeneratorMain(llvm::raw_ostream &OS, llvm::RecordKeeper &records) {
   OS << "#include \"TypeTraits.h\"\n";
   llvm_libc::APIIndexer G(records);
-  for (const auto &header : G.PublicHeaders)
+  std::unordered_set<std::string> headerFileSet;
+  for (const auto &entrypoint : EntrypointNamesOption) {
+    auto match = G.FunctionToHeaderMap.find(entrypoint);
+    if (match == G.FunctionToHeaderMap.end()) {
+      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);