blob: 78203faf421a49695399af814c7aef600b84a776 [file] [log] [blame]
Michael J. Spencera55e37f2013-03-01 00:03:36 +00001//===- lib/ReaderWriter/ReaderLinkerScript.cpp ----------------------------===//
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#include "lld/ReaderWriter/ReaderLinkerScript.h"
11
12#include "lld/Core/Error.h"
13#include "lld/Core/File.h"
14#include "lld/Core/LinkerOptions.h"
15#include "lld/ReaderWriter/LinkerScript.h"
16
17using namespace lld;
18using namespace script;
19
20namespace {
21class LinkerScriptFile : public File {
22public:
23 static ErrorOr<std::unique_ptr<LinkerScriptFile> >
24 create(const TargetInfo &ti, std::unique_ptr<llvm::MemoryBuffer> mb) {
25 std::unique_ptr<LinkerScriptFile> file(
26 new LinkerScriptFile(ti, std::move(mb)));
27 file->_script = file->_parser.parse();
28 if (!file->_script)
29 return linker_script_reader_error::parse_error;
30 return std::move(file);
31 }
32
33 virtual Kind kind() const { return kindLinkerScript; }
34
35 static inline bool classof(const File *f) {
36 return f->kind() == kindLinkerScript;
37 }
38
39 virtual void setOrdinalAndIncrement(uint64_t &ordinal) const {
40 _ordinal = ordinal++;
41 }
42
43 virtual const TargetInfo &getTargetInfo() const { return _targetInfo; }
44
45 virtual const atom_collection<DefinedAtom> &defined() const {
46 return _definedAtoms;
47 }
48
49 virtual const atom_collection<UndefinedAtom> &undefined() const {
50 return _undefinedAtoms;
51 }
52
53 virtual const atom_collection<SharedLibraryAtom> &sharedLibrary() const {
54 return _sharedLibraryAtoms;
55 }
56
57 virtual const atom_collection<AbsoluteAtom> &absolute() const {
58 return _absoluteAtoms;
59 }
60
61 const LinkerScript *getScript() {
62 return _script;
63 }
64
65private:
66 LinkerScriptFile(const TargetInfo &ti, std::unique_ptr<llvm::MemoryBuffer> mb)
67 : File(mb->getBufferIdentifier()),
68 _targetInfo(ti),
69 _lexer(std::move(mb)),
70 _parser(_lexer),
71 _script(nullptr) {}
72
73 const TargetInfo &_targetInfo;
74 atom_collection_vector<DefinedAtom> _definedAtoms;
75 atom_collection_vector<UndefinedAtom> _undefinedAtoms;
76 atom_collection_vector<SharedLibraryAtom> _sharedLibraryAtoms;
77 atom_collection_vector<AbsoluteAtom> _absoluteAtoms;
78 Lexer _lexer;
79 Parser _parser;
80 const LinkerScript *_script;
81};
82} // end anon namespace
83
84namespace lld {
85error_code
86ReaderLinkerScript::parseFile(std::unique_ptr<llvm::MemoryBuffer> mb,
87 std::vector<std::unique_ptr<File> > &result) {
88 auto lsf = LinkerScriptFile::create(_targetInfo, std::move(mb));
89 if (!lsf)
90 return lsf;
91 const LinkerScript *ls = (*lsf)->getScript();
92 result.push_back(std::move(*lsf));
93 for (const auto &c : ls->_commands) {
94 if (auto group = dyn_cast<Group>(c))
95 for (const auto &path : group->getPaths()) {
96 auto reader = _getReader(LinkerInput(path._path));
97 if (!reader)
98 return reader;
99 if (error_code ec = reader->readFile(path._path, result))
100 return ec;
101 }
102 }
103 return error_code::success();
104}
105} // end namespace lld