sandboxed-api/sandboxed_api/tools/clang_generator/generator.cc
Christian Blichmann d5ebb81598 clang_generator: Handle intrinsics directly
PiperOrigin-RevId: 435289903
Change-Id: I68a37bb7b25a7b77c046d00a2740aa9de2fcaa89
2022-03-17 02:31:57 -07:00

127 lines
4.6 KiB
C++

// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "sandboxed_api/tools/clang_generator/generator.h"
#include <fstream>
#include <iostream>
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "clang/AST/Type.h"
#include "clang/Format/Format.h"
#include "clang/Lex/PreprocessorOptions.h"
#include "sandboxed_api/tools/clang_generator/diagnostics.h"
#include "sandboxed_api/tools/clang_generator/emitter.h"
#include "sandboxed_api/util/fileops.h"
#include "sandboxed_api/util/status_macros.h"
namespace sapi {
namespace {
// Replaces the file extension of a path name.
std::string ReplaceFileExtension(absl::string_view path,
absl::string_view new_extension) {
auto last_slash = path.rfind('/');
auto pos = path.rfind('.', last_slash);
if (pos != absl::string_view::npos && last_slash != absl::string_view::npos) {
pos += last_slash;
}
return absl::StrCat(path.substr(0, pos), new_extension);
}
} // namespace
std::string GetOutputFilename(absl::string_view source_file) {
return ReplaceFileExtension(source_file, ".sapi.h");
}
bool GeneratorASTVisitor::VisitFunctionDecl(clang::FunctionDecl* decl) {
if (!decl->isCXXClassMember() && // Skip classes
decl->isExternC() && // Skip non external functions
!decl->isTemplated() && // Skip function templates
// Process either all function or just the requested ones
(options_.function_names.empty() ||
options_.function_names.count(ToStringView(decl->getName())) > 0)) {
functions_.push_back(decl);
collector_.CollectRelatedTypes(decl->getDeclaredReturnType());
for (const clang::ParmVarDecl* param : decl->parameters()) {
collector_.CollectRelatedTypes(param->getType());
}
}
return true;
}
void GeneratorASTConsumer::HandleTranslationUnit(clang::ASTContext& context) {
if (!visitor_.TraverseDecl(context.getTranslationUnitDecl())) {
ReportFatalError(context.getDiagnostics(),
context.getTranslationUnitDecl()->getBeginLoc(),
"AST traversal exited early");
}
for (clang::QualType qual : visitor_.collector_.collected()) {
emitter_.CollectType(qual);
}
for (clang::FunctionDecl* func : visitor_.functions_) {
emitter_.CollectFunction(func);
}
}
bool GeneratorFactory::runInvocation(
std::shared_ptr<clang::CompilerInvocation> invocation,
clang::FileManager* files,
std::shared_ptr<clang::PCHContainerOperations> pch_container_ops,
clang::DiagnosticConsumer* diag_consumer) {
auto& options = invocation->getPreprocessorOpts();
// Explicitly ask to define __clang_analyzer__ macro.
options.SetUpStaticAnalyzer = true;
for (const auto& def : {
// Enable code to detect whether it is being SAPI-ized
"__SAPI__",
// TODO(b/222241644): Figure out how to deal with intrinsics properly
"__builtin_ia32_paddsb128=",
"__builtin_ia32_paddsb256=",
"__builtin_ia32_paddsb512=",
"__builtin_ia32_paddsw128=",
"__builtin_ia32_paddsw256=",
"__builtin_ia32_paddsw512=",
"__builtin_ia32_paddusb128=",
"__builtin_ia32_paddusb256=",
"__builtin_ia32_paddusb512=",
"__builtin_ia32_paddusw128=",
"__builtin_ia32_paddusw256=",
"__builtin_ia32_paddusw512=",
"__builtin_ia32_psubsb128=",
"__builtin_ia32_psubsb256=",
"__builtin_ia32_psubsb512=",
"__builtin_ia32_psubsw128=",
"__builtin_ia32_psubsw256=",
"__builtin_ia32_psubsw512=",
"__builtin_ia32_psubusb128=",
"__builtin_ia32_psubusb256=",
"__builtin_ia32_psubusb512=",
"__builtin_ia32_psubusw128=",
"__builtin_ia32_psubusw256=",
"__builtin_ia32_psubusw512=",
}) {
options.addMacroDef(def);
}
return FrontendActionFactory::runInvocation(std::move(invocation), files,
std::move(pch_container_ops),
diag_consumer);
}
} // namespace sapi