122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao//===- StubFactory.h ------------------------------------------------------===// 222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao// 322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao// The MCLinker Project 422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao// 522add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao// This file is distributed under the University of Illinois Open Source 622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao// License. See LICENSE.TXT for details. 722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao// 822add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao//===----------------------------------------------------------------------===// 937b74a387bb3993387029859c2d9d051c41c724eStephen Hines#ifndef MCLD_LD_STUBFACTORY_H_ 1037b74a387bb3993387029859c2d9d051c41c724eStephen Hines#define MCLD_LD_STUBFACTORY_H_ 1122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 1222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <llvm/Support/DataTypes.h> 1337b74a387bb3993387029859c2d9d051c41c724eStephen Hines 1422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao#include <vector> 1522add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 16d0fbbb227051be16931a1aa9b4a7722ac039c698Shih-wei Liaonamespace mcld { 1722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 1822add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liaoclass BranchIslandFactory; 196f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hinesclass IRBuilder; 20cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hinesclass FragmentRef; 2137b74a387bb3993387029859c2d9d051c41c724eStephen Hinesclass Relocation; 2237b74a387bb3993387029859c2d9d051c41c724eStephen Hinesclass Stub; 2322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 2422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao/** \class StubFactory 2522add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao * \brief the clone factory of Stub 2622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao * 2722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao */ 2837b74a387bb3993387029859c2d9d051c41c724eStephen Hinesclass StubFactory { 2937b74a387bb3993387029859c2d9d051c41c724eStephen Hines public: 3022add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao ~StubFactory(); 3122add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 3222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao /// addPrototype - register a stub prototype 3322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao void addPrototype(Stub* pPrototype); 3422add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 3522add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao /// create - create a stub if needed, otherwise return NULL 3622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao Stub* create(Relocation& pReloc, 3722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao uint64_t pTargetSymValue, 386f75755c9204b1d8817ae5a65a2f7e5af0ec3f70Stephen Hines IRBuilder& pBuilder, 3922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao BranchIslandFactory& pBRIslandFactory); 4022add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 41cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hines Stub* create(FragmentRef& pFragRef, 42cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hines IRBuilder& pBuilder, 43cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hines BranchIslandFactory& pBRIslandFactory); 44cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hines 4537b74a387bb3993387029859c2d9d051c41c724eStephen Hines private: 4622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao /// findPrototype - find if there is a registered stub prototype for the given 4722add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao /// relocation 4822add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao Stub* findPrototype(const Relocation& pReloc, 4922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao const uint64_t pSource, 50cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hines uint64_t pTargetSymValue) const; 51cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hines 52cfcb22478ca64c308df58f9abe6fa2dedb213c16Stephen Hines Stub* findPrototype(const FragmentRef& pFragRef) const; 5322add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 5437b74a387bb3993387029859c2d9d051c41c724eStephen Hines private: 5537b74a387bb3993387029859c2d9d051c41c724eStephen Hines typedef std::vector<Stub*> StubPoolType; 5622add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 5737b74a387bb3993387029859c2d9d051c41c724eStephen Hines private: 5837b74a387bb3993387029859c2d9d051c41c724eStephen Hines StubPoolType m_StubPool; // stub pool 5922add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao}; 6022add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 6137b74a387bb3993387029859c2d9d051c41c724eStephen Hines} // namespace mcld 6222add6ff3426df1a85089fe6a6e1597ee3b6f300Shih-wei Liao 6337b74a387bb3993387029859c2d9d051c41c724eStephen Hines#endif // MCLD_LD_STUBFACTORY_H_ 64