summaryrefslogtreecommitdiff
path: root/include/mcld/LD/GNUArchiveReader.h
blob: 8beac2c343f7c931e8d413134633fdd287604f82 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
//===- GNUArchiveReader.h -------------------------------------------------===//
//
//                     The MCLinker Project
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
#ifndef MCLD_GNU_ARCHIVE_READER_H
#define MCLD_GNU_ARCHIVE_READER_H
#ifdef ENABLE_UNITTEST
#include <gtest.h>
#endif

#include "mcld/LD/ArchiveReader.h"
#include "mcld/Support/Path.h"
#include <llvm/ADT/OwningPtr.h>

#include <vector>
#include <string>

namespace llvm
{
class MemoryBuffer;

}

namespace mcld
{
class MCLDInfo;
class Input;
class InputTree;

/** \class GNUArchiveReader
 *  \brief GNUArchiveReader reads GNU archive files.
 */
class GNUArchiveReader : public ArchiveReader
{
private:
  struct ArchiveMemberHeader;
  struct SymbolTableEntry;

public:
  explicit GNUArchiveReader(MCLDInfo &pLDInfo, LDReader::Endian endian)
  : m_pLDInfo(pLDInfo),
    m_endian(endian)
  { }

  ~GNUArchiveReader()
  { }

  /// Read an archive and extract each member in.
  /// Construct the coresponding Input for each member.
  InputTree *readArchive(Input &input);

  bool isMyFormat(Input &input) const;

  LDReader::Endian endian(Input& pFile) const;

private:
  /// set up the archive, including
  /// first, read symbol table
  /// second, read extended file name which is used in thin archive
  InputTree *setupNewArchive(Input &pInput, size_t off);

  /// parse the archive header, and return the member size
  size_t parseMemberHeader(llvm::OwningPtr<llvm::MemoryBuffer> &mapFile,
                   off_t off,
                   std::string *p_Name,
                   off_t *nestedOff,
                   std::string &p_ExtendedName);

  void readSymbolTable(llvm::OwningPtr<llvm::MemoryBuffer> &mapFile,
                      std::vector<SymbolTableEntry> &pSymbolTable,
                      off_t start,
                      size_t size);

private:
  MCLDInfo &m_pLDInfo;
  LDReader::Endian m_endian;
};

} // namespace of mcld

#endif