1 //===- lib/MC/MCModule.cpp - MCModule implementation ----------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/MC/MCModule.h"
11 #include "llvm/MC/MCAtom.h"
12 #include "llvm/MC/MCFunction.h"
17 static bool AtomComp(const MCAtom *L, uint64_t Addr) {
18 return L->getEndAddr() < Addr;
21 void MCModule::map(MCAtom *NewAtom) {
22 uint64_t Begin = NewAtom->Begin,
25 assert(Begin < End && "Creating MCAtom with endpoints reversed?");
27 // Check for atoms already covering this range.
28 AtomListTy::iterator I = std::lower_bound(atom_begin(), atom_end(),
30 assert((I == atom_end() || (*I)->getBeginAddr() > End)
31 && "Offset range already occupied!");
33 // Insert the new atom to the list.
34 Atoms.insert(I, NewAtom);
37 MCTextAtom *MCModule::createTextAtom(uint64_t Begin, uint64_t End) {
38 MCTextAtom *NewAtom = new MCTextAtom(this, Begin, End);
43 MCDataAtom *MCModule::createDataAtom(uint64_t Begin, uint64_t End) {
44 MCDataAtom *NewAtom = new MCDataAtom(this, Begin, End);
49 // remap - Update the interval mapping for an atom.
50 void MCModule::remap(MCAtom *Atom, uint64_t NewBegin, uint64_t NewEnd) {
51 // Find and erase the old mapping.
52 AtomListTy::iterator I = std::lower_bound(atom_begin(), atom_end(),
53 Atom->Begin, AtomComp);
54 assert(I != atom_end() && "Atom offset not found in module!");
55 assert(*I == Atom && "Previous atom mapping was invalid!");
58 // Insert the new mapping.
59 AtomListTy::iterator NewI = std::lower_bound(atom_begin(), atom_end(),
61 Atoms.insert(NewI, Atom);
63 // Update the atom internal bounds.
64 Atom->Begin = NewBegin;
68 const MCAtom *MCModule::findAtomContaining(uint64_t Addr) const {
69 AtomListTy::const_iterator I = std::lower_bound(atom_begin(), atom_end(),
71 if (I != atom_end() && (*I)->getBeginAddr() <= Addr)
76 MCAtom *MCModule::findAtomContaining(uint64_t Addr) {
77 AtomListTy::iterator I = std::lower_bound(atom_begin(), atom_end(),
79 if (I != atom_end() && (*I)->getBeginAddr() <= Addr)
84 MCFunction *MCModule::createFunction(const StringRef &Name) {
85 Functions.push_back(new MCFunction(Name));
86 return Functions.back();
89 MCModule::~MCModule() {
90 for (AtomListTy::iterator AI = atom_begin(),
94 for (FunctionListTy::iterator FI = func_begin(),