Rafael Espindola | beee25e | 2015-08-14 14:12:54 +0000 | [diff] [blame] | 1 | //===- Driver.h -------------------------------------------------*- C++ -*-===// |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 2 | // |
| 3 | // The LLVM Linker |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | |
| 10 | #ifndef LLD_ELF_DRIVER_H |
| 11 | #define LLD_ELF_DRIVER_H |
| 12 | |
Rui Ueyama | 983ed2b | 2015-10-01 15:23:09 +0000 | [diff] [blame] | 13 | #include "SymbolTable.h" |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 14 | #include "lld/Core/LLVM.h" |
Rui Ueyama | f5c4aca | 2015-09-30 17:06:09 +0000 | [diff] [blame] | 15 | #include "llvm/ADT/StringRef.h" |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 16 | #include "llvm/Option/ArgList.h" |
Rui Ueyama | b694011 | 2016-02-02 22:49:32 +0000 | [diff] [blame^] | 17 | #include "llvm/Support/raw_ostream.h" |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 18 | |
| 19 | namespace lld { |
| 20 | namespace elf2 { |
| 21 | |
Rui Ueyama | 983ed2b | 2015-10-01 15:23:09 +0000 | [diff] [blame] | 22 | extern class LinkerDriver *Driver; |
Rui Ueyama | f5c4aca | 2015-09-30 17:06:09 +0000 | [diff] [blame] | 23 | |
Rui Ueyama | 64cfffd | 2016-01-28 18:40:06 +0000 | [diff] [blame] | 24 | // Entry point of the ELF linker. Returns true on success. |
Rui Ueyama | b694011 | 2016-02-02 22:49:32 +0000 | [diff] [blame^] | 25 | bool link(ArrayRef<const char *> Args, llvm::raw_ostream &Error = llvm::errs()); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 26 | |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 27 | class LinkerDriver { |
| 28 | public: |
Rui Ueyama | 3ce825e | 2015-10-09 21:07:25 +0000 | [diff] [blame] | 29 | void main(ArrayRef<const char *> Args); |
Rui Ueyama | 983ed2b | 2015-10-01 15:23:09 +0000 | [diff] [blame] | 30 | void addFile(StringRef Path); |
Rui Ueyama | 21eecb4 | 2016-02-02 21:13:09 +0000 | [diff] [blame] | 31 | void addLibrary(StringRef Name); |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 32 | |
| 33 | private: |
Rui Ueyama | d441d75 | 2016-01-07 17:54:21 +0000 | [diff] [blame] | 34 | void readConfigs(llvm::opt::InputArgList &Args); |
| 35 | void createFiles(llvm::opt::InputArgList &Args); |
| 36 | template <class ELFT> void link(llvm::opt::InputArgList &Args); |
Denis Protivensky | 1ef7b3f | 2015-10-07 09:13:03 +0000 | [diff] [blame] | 37 | |
Rui Ueyama | a47ee68 | 2015-10-11 01:53:04 +0000 | [diff] [blame] | 38 | llvm::BumpPtrAllocator Alloc; |
Rui Ueyama | 3ce825e | 2015-10-09 21:07:25 +0000 | [diff] [blame] | 39 | bool WholeArchive = false; |
| 40 | std::vector<std::unique_ptr<InputFile>> Files; |
Rui Ueyama | 983ed2b | 2015-10-01 15:23:09 +0000 | [diff] [blame] | 41 | std::vector<std::unique_ptr<MemoryBuffer>> OwningMBs; |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 42 | }; |
| 43 | |
Rui Ueyama | c6e1b97 | 2015-10-11 18:19:01 +0000 | [diff] [blame] | 44 | // Parses command line options. |
| 45 | llvm::opt::InputArgList parseArgs(llvm::BumpPtrAllocator *A, |
| 46 | ArrayRef<const char *> Args); |
| 47 | |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 48 | // Create enum with OPT_xxx values for each option in Options.td |
| 49 | enum { |
| 50 | OPT_INVALID = 0, |
| 51 | #define OPTION(_1, _2, ID, _4, _5, _6, _7, _8, _9, _10, _11) OPT_##ID, |
| 52 | #include "Options.inc" |
| 53 | #undef OPTION |
| 54 | }; |
| 55 | |
Rui Ueyama | f5c4aca | 2015-09-30 17:06:09 +0000 | [diff] [blame] | 56 | // Parses a linker script. Calling this function updates the Symtab and Config. |
Rui Ueyama | a47ee68 | 2015-10-11 01:53:04 +0000 | [diff] [blame] | 57 | void readLinkerScript(llvm::BumpPtrAllocator *A, MemoryBufferRef MB); |
Rui Ueyama | f5c4aca | 2015-09-30 17:06:09 +0000 | [diff] [blame] | 58 | |
Rui Ueyama | 52a1509 | 2015-10-11 03:28:42 +0000 | [diff] [blame] | 59 | std::string findFromSearchPaths(StringRef Path); |
Rui Ueyama | 4b02ca9 | 2015-10-11 03:28:39 +0000 | [diff] [blame] | 60 | std::string searchLibrary(StringRef Path); |
| 61 | std::string buildSysrootedPath(llvm::StringRef Dir, llvm::StringRef File); |
| 62 | |
Michael J. Spencer | 84487f1 | 2015-07-24 21:03:07 +0000 | [diff] [blame] | 63 | } // namespace elf2 |
| 64 | } // namespace lld |
| 65 | |
| 66 | #endif |