Merge pull request #64943 from nkcsgexi/block-list

Author: nkcsgexi

include/swift/AST/ASTContext.h

@@ -30,6 +30,7 @@
 #include "swift/Basic/LangOptions.h"
 #include "swift/Basic/Located.h"
 #include "swift/Basic/Malloc.h"
+#include "swift/Basic/BlockList.h"
 #include "swift/SymbolGraphGen/SymbolGraphOptions.h"
 #include "clang/AST/DeclTemplate.h"
 #include "llvm/ADT/ArrayRef.h"
@@ -365,6 +366,8 @@ class ASTContext final {
   /// The Swift module currently being compiled.
   ModuleDecl *MainModule = nullptr;
 
+  /// The block list where we can find special actions based on module name;
+  BlockListStore blockListConfig;
 private:
   /// The current generation number, which reflects the number of
   /// times that external modules have been loaded.

include/swift/Basic/BlockList.h

@@ -0,0 +1,51 @@
+//===--- BlockList.h ---------------------------------------------*- C++ -*-===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2023 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+//
+// This file defines some miscellaneous overloads of hash_value() and
+// simple_display().
+//
+//===----------------------------------------------------------------------===//
+
+#ifndef SWIFT_BASIC_BLOCKLIST_H
+#define SWIFT_BASIC_BLOCKLIST_H
+
+#include "swift/Basic/LLVM.h"
+#include "llvm/ADT/StringRef.h"
+
+namespace swift {
+
+enum class BlockListAction: uint8_t {
+  ShouldUseBinaryModule = 0,
+  ShouldUseTextualModule,
+};
+
+enum class BlockListKeyKind: uint8_t {
+  ModuleName,
+  ProjectName
+};
+
+class BlockListStore {
+public:
+  struct Implementation;
+  bool hasBlockListAction(StringRef key, BlockListKeyKind keyKind,
+                          BlockListAction action);
+  BlockListStore();
+  ~BlockListStore();
+private:
+  friend class ASTContext;
+  void addConfigureFilePath(StringRef path);
+  Implementation &Impl;
+};
+
+} // namespace swift
+
+#endif // SWIFT_BASIC_BLOCKLIST_H

include/swift/Basic/LangOptions.h

@@ -556,6 +556,9 @@ namespace swift {
     /// The model of concurrency to be used.
     ConcurrencyModel ActiveConcurrencyModel = ConcurrencyModel::Standard;
 
+    /// All block list configuration files to be honored in this compilation.
+    std::vector<std::string> BlocklistConfigFilePath;
+
     bool isConcurrencyModelTaskToThread() const {
       return ActiveConcurrencyModel == ConcurrencyModel::TaskToThread;
     }

include/swift/Option/FrontendOptions.td

@@ -240,6 +240,9 @@ def Raccess_note : Separate<["-"], "Raccess-note">,
 def Raccess_note_EQ : Joined<["-"], "Raccess-note=">,
     Alias<Raccess_note>;
 
+def block_list_file
+  : Separate<["-"], "blocklist-file">, MetaVarName<"<path>">,
+    HelpText<"The path to a blocklist configuration file">;
 } // end let Flags = [FrontendOption, NoDriverOption]
 
 def debug_crash_Group : OptionGroup<"<automatic crashing options>">;

lib/AST/ASTContext.cpp

@@ -713,9 +713,14 @@ ASTContext::ASTContext(
   registerNameLookupRequestFunctions(evaluator);
 
   createModuleToExecutablePluginMap();
+  
   // Provide a default OnDiskOutputBackend if user didn't supply one.
   if (!OutputBackend)
     OutputBackend = llvm::makeIntrusiveRefCnt<llvm::vfs::OnDiskOutputBackend>();
+  // Insert all block list config paths.
+  for (auto path: langOpts.BlocklistConfigFilePath) {
+    blockListConfig.addConfigureFilePath(path);
+  }
 }
 
 ASTContext::~ASTContext() {

lib/Basic/BlockList.cpp

@@ -0,0 +1,53 @@
+//===--- BlockList.cpp - BlockList utilities ------------------------------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2023 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/ADT/STLExtras.h"
+#include "llvm/ADT/StringSwitch.h"
+#include "llvm/Support/YAMLParser.h"
+#include "llvm/Support/YAMLTraits.h"
+#include "swift/Basic/BlockList.h"
+#include "swift/Basic/SourceManager.h"
+
+struct swift::BlockListStore::Implementation {
+  llvm::StringMap<std::vector<BlockListAction>> ModuleActionDict;
+  llvm::StringMap<std::vector<BlockListAction>> ProjectActionDict;
+  void addConfigureFilePath(StringRef path);
+  bool hasBlockListAction(StringRef key, BlockListKeyKind keyKind,
+                          BlockListAction action);
+};
+
+swift::BlockListStore::BlockListStore(): Impl(*new Implementation()) {}
+
+swift::BlockListStore::~BlockListStore() { delete &Impl; }
+
+bool swift::BlockListStore::hasBlockListAction(StringRef key,
+    BlockListKeyKind keyKind, BlockListAction action) {
+  return Impl.hasBlockListAction(key, keyKind, action);
+}
+
+void swift::BlockListStore::addConfigureFilePath(StringRef path) {
+  Impl.addConfigureFilePath(path);
+}
+
+bool swift::BlockListStore::Implementation::hasBlockListAction(StringRef key,
+    BlockListKeyKind keyKind, BlockListAction action) {
+  auto *dict = keyKind == BlockListKeyKind::ModuleName ? &ModuleActionDict :
+    &ProjectActionDict;
+  auto it = dict->find(key);
+  if (it == dict->end())
+    return false;
+  return llvm::is_contained(it->second, action);
+}
+
+void swift::BlockListStore::Implementation::addConfigureFilePath(StringRef path) {
+
+}

lib/Basic/CMakeLists.txt

@@ -78,6 +78,7 @@ add_swift_host_library(swiftBasic STATIC
   Unicode.cpp
   UUID.cpp
   Version.cpp
+  BlockList.cpp
 
   ${llvm_revision_inc}
   ${clang_revision_inc}

lib/Frontend/CompilerInvocation.cpp

@@ -1177,6 +1177,9 @@ static bool ParseLangArgs(LangOptions &Opts, ArgList &Args,
 
   Opts.DumpTypeWitnessSystems = Args.hasArg(OPT_dump_type_witness_systems);
 
+  for (auto A : Args.getAllArgValues(options::OPT_block_list_file)) {
+    Opts.BlocklistConfigFilePath.push_back(A);
+  }
   if (const Arg *A = Args.getLastArg(options::OPT_concurrency_model)) {
     Opts.ActiveConcurrencyModel =
         llvm::StringSwitch<ConcurrencyModel>(A->getValue())