1 //===- COFFImportFile.h - COFF short import file implementation -*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // COFF short import file is a special kind of file which contains
11 // only symbol names for DLL-exported symbols. This class implements
12 // exporting of Symbols to create libraries and a SymbolicFile
13 // interface for the file type.
14 //
15 //===----------------------------------------------------------------------===//
16 
17 #ifndef LLVM_OBJECT_COFF_IMPORT_FILE_H
18 #define LLVM_OBJECT_COFF_IMPORT_FILE_H
19 
20 #include "llvm/ADT/ArrayRef.h"
21 #include "llvm/Object/COFF.h"
22 #include "llvm/Object/IRObjectFile.h"
23 #include "llvm/Object/ObjectFile.h"
24 #include "llvm/Object/SymbolicFile.h"
25 #include "llvm/Support/MemoryBuffer.h"
26 #include "llvm/Support/raw_ostream.h"
27 
28 namespace llvm {
29 namespace object {
30 
31 class COFFImportFile : public SymbolicFile {
32 public:
COFFImportFile(MemoryBufferRef Source)33   COFFImportFile(MemoryBufferRef Source)
34       : SymbolicFile(ID_COFFImportFile, Source) {}
35 
classof(Binary const * V)36   static bool classof(Binary const *V) { return V->isCOFFImportFile(); }
37 
moveSymbolNext(DataRefImpl & Symb)38   void moveSymbolNext(DataRefImpl &Symb) const override { ++Symb.p; }
39 
printSymbolName(raw_ostream & OS,DataRefImpl Symb)40   std::error_code printSymbolName(raw_ostream &OS,
41                                   DataRefImpl Symb) const override {
42     if (Symb.p == 0)
43       OS << "__imp_";
44     OS << StringRef(Data.getBufferStart() + sizeof(coff_import_header));
45     return std::error_code();
46   }
47 
getSymbolFlags(DataRefImpl Symb)48   uint32_t getSymbolFlags(DataRefImpl Symb) const override {
49     return SymbolRef::SF_Global;
50   }
51 
symbol_begin()52   basic_symbol_iterator symbol_begin() const override {
53     return BasicSymbolRef(DataRefImpl(), this);
54   }
55 
symbol_end()56   basic_symbol_iterator symbol_end() const override {
57     DataRefImpl Symb;
58     Symb.p = isData() ? 1 : 2;
59     return BasicSymbolRef(Symb, this);
60   }
61 
getCOFFImportHeader()62   const coff_import_header *getCOFFImportHeader() const {
63     return reinterpret_cast<const object::coff_import_header *>(
64         Data.getBufferStart());
65   }
66 
67 private:
isData()68   bool isData() const {
69     return getCOFFImportHeader()->getType() == COFF::IMPORT_DATA;
70   }
71 };
72 
73 struct COFFShortExport {
74   std::string Name;
75   std::string ExtName;
76   std::string SymbolName;
77   std::string AliasTarget;
78 
79   uint16_t Ordinal = 0;
80   bool Noname = false;
81   bool Data = false;
82   bool Private = false;
83   bool Constant = false;
84 
85   friend bool operator==(const COFFShortExport &L, const COFFShortExport &R) {
86     return L.Name == R.Name && L.ExtName == R.ExtName &&
87             L.Ordinal == R.Ordinal && L.Noname == R.Noname &&
88             L.Data == R.Data && L.Private == R.Private;
89   }
90 
91   friend bool operator!=(const COFFShortExport &L, const COFFShortExport &R) {
92     return !(L == R);
93   }
94 };
95 
96 Error writeImportLibrary(StringRef ImportName, StringRef Path,
97                          ArrayRef<COFFShortExport> Exports,
98                          COFF::MachineTypes Machine, bool MinGW);
99 
100 } // namespace object
101 } // namespace llvm
102 
103 #endif
104