[lld] r192082 - [lld] Add FileArchive
Shankar Easwaran
shankare at codeaurora.org
Sun Oct 6 19:55:42 PDT 2013
Author: shankare
Date: Sun Oct 6 21:55:42 2013
New Revision: 192082
URL: http://llvm.org/viewvc/llvm-project?rev=192082&view=rev
Log:
[lld] Add FileArchive
Fix adding a missed file in the previous commit.
Added:
lld/trunk/include/lld/ReaderWriter/FileArchive.h
Added: lld/trunk/include/lld/ReaderWriter/FileArchive.h
URL: http://llvm.org/viewvc/llvm-project/lld/trunk/include/lld/ReaderWriter/FileArchive.h?rev=192082&view=auto
==============================================================================
--- lld/trunk/include/lld/ReaderWriter/FileArchive.h (added)
+++ lld/trunk/include/lld/ReaderWriter/FileArchive.h Sun Oct 6 21:55:42 2013
@@ -0,0 +1,192 @@
+//===- lld/ReaderWriter/FileArchive.h - Archive Library File -----------===//
+//
+// The LLVM Linker
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===---------------------------------------------------------------------===//
+#ifndef LLD_FILEARCHIVE_H
+#define LLD_FILEARCHIVE_H
+
+#include "lld/Core/ArchiveLibraryFile.h"
+
+#include "llvm/ADT/Hashing.h"
+#include "llvm/Object/ObjectFile.h"
+#include "llvm/Object/Archive.h"
+#include "llvm/Support/MemoryBuffer.h"
+
+#include <unordered_map>
+
+namespace lld {
+
+/// \brief The FileArchive class represents an Archive Library file
+class FileArchive : public ArchiveLibraryFile {
+public:
+
+ virtual ~FileArchive() { }
+
+ /// \brief Check if any member of the archive contains an Atom with the
+ /// specified name and return the File object for that member, or nullptr.
+ virtual const File *find(StringRef name, bool dataSymbolOnly) const {
+ auto member = _symbolMemberMap.find(name);
+ if (member == _symbolMemberMap.end())
+ return nullptr;
+
+ llvm::object::Archive::child_iterator ci = member->second;
+
+ if (dataSymbolOnly) {
+ OwningPtr<MemoryBuffer> buff;
+ if (ci->getMemoryBuffer(buff, true))
+ return nullptr;
+ if (isDataSymbol(buff.take(), name))
+ return nullptr;
+ }
+
+ std::vector<std::unique_ptr<File>> result;
+
+ OwningPtr<MemoryBuffer> buff;
+ if (ci->getMemoryBuffer(buff, true))
+ return nullptr;
+ if (_context.logInputFiles())
+ llvm::outs() << buff->getBufferIdentifier() << "\n";
+ std::unique_ptr<MemoryBuffer> mb(buff.take());
+ if (_context.getDefaultReader().parseFile(mb, result))
+ return nullptr;
+
+ assert(result.size() == 1);
+
+ result[0]->setOrdinalAndIncrement(_curChildOrd);
+
+ // give up the pointer so that this object no longer manages it
+ return result[0].release();
+ }
+
+ /// \brief Load all members of the archive ?
+ virtual bool isWholeArchive() const { return _isWholeArchive; }
+
+ /// \brief parse each member
+ virtual error_code parseAllMembers(std::vector<std::unique_ptr<File>> &result) const {
+ for (auto mf = _archive->begin_children(),
+ me = _archive->end_children(); mf != me; ++mf) {
+ OwningPtr<MemoryBuffer> buff;
+ error_code ec;
+ if ((ec = mf->getMemoryBuffer(buff, true)))
+ return ec;
+ if (_context.logInputFiles())
+ llvm::outs() << buff->getBufferIdentifier() << "\n";
+ std::unique_ptr<MemoryBuffer> mbc(buff.take());
+ if ((ec = _context.getDefaultReader().parseFile(mbc, result)))
+ return ec;
+ }
+ for (auto &file : result)
+ file->setOrdinalAndIncrement(_curChildOrd);
+ return error_code::success();
+ }
+
+ /// \brief members
+
+ virtual void setOrdinalAndIncrement(uint64_t &ordinal) const {
+ _ordinal = ordinal++;
+ _curChildOrd = _ordinal;
+ // Leave space in ordinal range for all children
+ for (auto mf = _archive->begin_children(),
+ me = _archive->end_children(); mf != me; ++mf) {
+ ordinal++;
+ }
+ }
+
+ virtual const atom_collection<DefinedAtom> &defined() const {
+ return _definedAtoms;
+ }
+
+ virtual const atom_collection<UndefinedAtom> &undefined() const {
+ return _undefinedAtoms;
+ }
+
+ virtual const atom_collection<SharedLibraryAtom> &sharedLibrary() const {
+ return _sharedLibraryAtoms;
+ }
+
+ virtual const atom_collection<AbsoluteAtom> &absolute() const {
+ return _absoluteAtoms;
+ }
+
+protected:
+ error_code isDataSymbol(MemoryBuffer *mb, StringRef symbol) const {
+ std::unique_ptr<llvm::object::ObjectFile>
+ obj(llvm::object::ObjectFile::createObjectFile(mb));
+ error_code ec;
+ llvm::object::SymbolRef::Type symtype;
+ uint32_t symflags;
+ llvm::object::symbol_iterator ibegin = obj->begin_symbols();
+ llvm::object::symbol_iterator iend = obj->end_symbols();
+ StringRef symbolname;
+
+ for (llvm::object::symbol_iterator i = ibegin; i != iend; i.increment(ec)) {
+ if (ec) return ec;
+
+ // Get symbol name
+ if ((ec = (i->getName(symbolname)))) return ec;
+
+ if (symbolname != symbol)
+ continue;
+
+ // Get symbol flags
+ if ((ec = (i->getFlags(symflags)))) return ec;
+
+ if (symflags <= llvm::object::SymbolRef::SF_Undefined)
+ continue;
+
+ // Get Symbol Type
+ if ((ec = (i->getType(symtype)))) return ec;
+
+ if (symtype == llvm::object::SymbolRef::ST_Data) {
+ return error_code::success();
+ }
+ }
+ return llvm::object::object_error::parse_failed;
+ }
+
+private:
+ std::unique_ptr<llvm::object::Archive> _archive;
+ atom_collection_vector<DefinedAtom> _definedAtoms;
+ atom_collection_vector<UndefinedAtom> _undefinedAtoms;
+ atom_collection_vector<SharedLibraryAtom> _sharedLibraryAtoms;
+ atom_collection_vector<AbsoluteAtom> _absoluteAtoms;
+ mutable uint64_t _curChildOrd;
+ bool _isWholeArchive;
+
+public:
+ /// only subclasses of ArchiveLibraryFile can be instantiated
+ FileArchive(const LinkingContext &context,
+ std::unique_ptr<MemoryBuffer> mb, error_code &ec,
+ bool isWholeArchive)
+ : ArchiveLibraryFile(context, mb->getBufferIdentifier()),
+ _isWholeArchive(isWholeArchive) {
+ std::unique_ptr<llvm::object::Archive> archive_obj(
+ new llvm::object::Archive(mb.release(), ec));
+ if (ec)
+ return;
+ _archive.swap(archive_obj);
+
+ // Cache symbols.
+ for (auto i = _archive->begin_symbols(), e = _archive->end_symbols();
+ i != e; ++i) {
+ StringRef name;
+ llvm::object::Archive::child_iterator member;
+ if ((ec = i->getName(name)))
+ return;
+ if ((ec = i->getMember(member)))
+ return;
+ _symbolMemberMap[name] = member;
+ }
+ }
+
+ std::unordered_map<StringRef,
+ llvm::object::Archive::child_iterator> _symbolMemberMap;
+}; // class FileArchive
+
+} // end namespace lld
+
+#endif // LLD_FILEARCHIVE_H
More information about the llvm-commits
mailing list