lld/COFF/MinGW.h

   1 //===- MinGW.h --------------------------------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef LLD_COFF_MINGW_H
  10 #define LLD_COFF_MINGW_H
  11
  12 #include "Config.h"
  13 #include "Symbols.h"
  14 #include "lld/Common/LLVM.h"
  15 #include "llvm/ADT/ArrayRef.h"
  16 #include "llvm/ADT/DenseSet.h"
  17 #include "llvm/ADT/StringSet.h"
  18 #include "llvm/Option/ArgList.h"
  19 #include <vector>
  20
  21 namespace lld::coff {
  22 class COFFLinkerContext;
  23
  24 // Logic for deciding what symbols to export, when exporting all
  25 // symbols for MinGW.
  26 class AutoExporter {
  27 public:
  28   AutoExporter(COFFLinkerContext &ctx,
  29                const llvm::DenseSet<StringRef> &manualExcludeSymbols);
  30
  31   void addWholeArchive(StringRef path);
  32   void addExcludedSymbol(StringRef symbol);
  33
  34   llvm::StringSet<> excludeSymbols;
  35   llvm::StringSet<> excludeSymbolPrefixes;
  36   llvm::StringSet<> excludeSymbolSuffixes;
  37   llvm::StringSet<> excludeLibs;
  38   llvm::StringSet<> excludeObjects;
  39
  40   const llvm::DenseSet<StringRef> &manualExcludeSymbols;
  41
  42   bool shouldExport(Defined *sym) const;
  43
  44 private:
  45   COFFLinkerContext &ctx;
  46 };
  47
  48 void writeDefFile(COFFLinkerContext &, StringRef name,
  49                   const std::vector<Export> &exports);
  50
  51 // The -wrap option is a feature to rename symbols so that you can write
  52 // wrappers for existing functions. If you pass `-wrap:foo`, all
  53 // occurrences of symbol `foo` are resolved to `__wrap_foo` (so, you are
  54 // expected to write `__wrap_foo` function as a wrapper). The original
  55 // symbol becomes accessible as `__real_foo`, so you can call that from your
  56 // wrapper.
  57 //
  58 // This data structure is instantiated for each -wrap option.
  59 struct WrappedSymbol {
  60   Symbol *sym;
  61   Symbol *real;
  62   Symbol *wrap;
  63 };
  64
  65 std::vector<WrappedSymbol> addWrappedSymbols(COFFLinkerContext &ctx,
  66                                              llvm::opt::InputArgList &args);
  67
  68 void wrapSymbols(COFFLinkerContext &ctx, ArrayRef<WrappedSymbol> wrapped);
  69
  70 } // namespace lld::coff
  71
  72 #endif