checkpoint for merge
[dyninst.git] / symtabAPI / src / Aggregate.C
1 /*
2  * Copyright (c) 1996-2007 Barton P. Miller
3  * 
4  * We provide the Paradyn Parallel Performance Tools (below
5  * described as "Paradyn") on an AS IS basis, and do not warrant its
6  * validity or performance.  We reserve the right to update, modify,
7  * or discontinue this software at any time.  We shall have no
8  * obligation to supply such updates or modifications or any other
9  * form of support to you.
10  * 
11  * By your use of Paradyn, you understand and agree that we (or any
12  * other person or entity with proprietary rights in Paradyn) are
13  * under no obligation to provide either maintenance services,
14  * update services, notices of latent defects, or correction of
15  * defects for Paradyn.
16  * 
17  * This library is free software; you can redistribute it and/or
18  * modify it under the terms of the GNU Lesser General Public
19  * License as published by the Free Software Foundation; either
20  * version 2.1 of the License, or (at your option) any later version.
21  * 
22  * This library is distributed in the hope that it will be useful,
23  * but WITHOUT ANY WARRANTY; without even the implied warranty of
24  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
25  * Lesser General Public License for more details.
26  * 
27  * You should have received a copy of the GNU Lesser General Public
28  * License along with this library; if not, write to the Free Software
29  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
30  */
31
32 #include "Annotatable.h"
33 #include "common/h/serialize.h"
34
35 #include "Symtab.h"
36 #include "symutil.h"
37 #include "Module.h"
38 #include "Collections.h"
39
40 #include "symtabAPI/src/Object.h"
41
42
43 #include "Aggregate.h"
44 #include "Symbol.h"
45
46 using namespace std;
47 using namespace Dyninst;
48 using namespace Dyninst::SymtabAPI;
49
50 Aggregate::Aggregate() : module_(NULL)
51 {}
52
53 Offset Aggregate::getAddress() const {
54     return symbols_[0]->getAddr();
55 }
56
57 Module *Aggregate::getModule() const {
58     return module_;
59 }
60
61 const vector<std::string> &Aggregate::getAllMangledNames() {
62     return mangledNames_;
63 }
64 const vector<std::string> &Aggregate::getAllPrettyNames() {
65     return prettyNames_;
66 }
67 const vector<std::string> &Aggregate::getAllTypedNames() {
68     return typedNames_;
69 }
70
71 bool Aggregate::addSymbol(Symbol *sym) {
72
73     // We keep a "primary" module, which is defined as "anything not DEFAULT_MODULE".
74     if (module_ == NULL) {
75         module_ = sym->getModule();
76     }
77     else if (module_->fileName() == "DEFAULT_MODULE") {
78         module_ = sym->getModule();
79     }
80     // else keep current module.
81
82     symbols_.push_back(sym);
83
84     // We need to add the symbol names (if they aren't there already)
85     // We can have multiple identical names - for example, there are
86     // often two symbols for main (static and dynamic symbol table)
87     
88     bool found = false;
89     for (unsigned j = 0; j < mangledNames_.size(); j++) {
90         if (sym->getMangledName() == mangledNames_[j]) {
91             found = true;
92             break;
93         }
94     }
95     if (!found) mangledNames_.push_back(sym->getMangledName());
96
97     found = false;
98
99     for (unsigned j = 0; j < prettyNames_.size(); j++) {
100         if (sym->getPrettyName() == prettyNames_[j]) {
101             found = true;
102             break;
103         }
104     }
105     if (!found) prettyNames_.push_back(sym->getPrettyName());
106
107     found = false;
108     for (unsigned j = 0; j < typedNames_.size(); j++) {
109         if (sym->getTypedName() == typedNames_[j]) {
110             found = true;
111             break;
112         }
113     }
114     if (!found) typedNames_.push_back(sym->getTypedName());;
115
116     return true;
117 }
118
119 bool Aggregate::removeSymbol(Symbol *sym) {
120     std::vector<Symbol *>::iterator iter;
121     for (iter = symbols_.begin(); iter != symbols_.end(); iter++) {
122         if (*iter == sym) {
123             symbols_.erase(iter);
124             return true;
125         }
126     }
127     // TODO: remove from names. Do we ever call this?
128
129     return false;
130 }
131
132 bool Aggregate::getAllSymbols(std::vector<Symbol *> &syms) const {
133     syms = symbols_;
134     return true;
135 }
136
137 Symbol * Aggregate::getFirstSymbol() const
138 {
139     assert( symbols_.size()>0 );
140     return symbols_[0];
141 }
142
143 SYMTAB_EXPORT bool Aggregate::addMangledName(string name, bool isPrimary) 
144  {
145     // Check to see if we're duplicating
146     for (unsigned i = 0; i < mangledNames_.size(); i++) {
147         if (mangledNames_[i] == name)
148             return false;
149     }
150
151     if (isPrimary) {
152         std::vector<std::string>::iterator iter = mangledNames_.begin();
153         mangledNames_.insert(iter, name);
154     }
155     else
156         mangledNames_.push_back(name);
157
158     if (addMangledNameInt(name, isPrimary) == false) return false;
159      return true;
160  }                                                                                                                                      
161  
162  SYMTAB_EXPORT bool Aggregate::addPrettyName(string name, bool isPrimary) 
163  {
164     // Check to see if we're duplicating
165     for (unsigned i = 0; i < prettyNames_.size(); i++) {
166         if (prettyNames_[i] == name)
167             return false;
168     }
169
170     if (isPrimary) {
171         std::vector<std::string>::iterator iter = prettyNames_.begin();
172         prettyNames_.insert(iter, name);
173     }
174     else
175         prettyNames_.push_back(name);
176     if (addPrettyNameInt(name, isPrimary) == false) return false;
177      return true;
178  }                                                                                                                                      
179  
180  SYMTAB_EXPORT bool Aggregate::addTypedName(string name, bool isPrimary) 
181  {
182     // Check to see if we're duplicating
183     for (unsigned i = 0; i < typedNames_.size(); i++) {
184         if (typedNames_[i] == name)
185             return false;
186     }
187     if (addTypedNameInt(name, isPrimary) == false) return false;
188
189     if (isPrimary) {
190         std::vector<std::string>::iterator iter = typedNames_.begin();
191         typedNames_.insert(iter, name);
192     }
193     else
194         typedNames_.push_back(name);
195         return true;
196  }
197
198 bool Aggregate::addMangledNameInt(string name, bool isPrimary) {
199     // Check to see if we're duplicating
200     for (unsigned i = 0; i < mangledNames_.size(); i++) {
201         if (mangledNames_[i] == name)
202             return false;
203     }
204
205     if (isPrimary) {
206         std::vector<std::string>::iterator iter = mangledNames_.begin();
207         mangledNames_.insert(iter, name);
208     }
209     else
210         mangledNames_.push_back(name);
211     return true;
212 }
213
214 bool Aggregate::addPrettyNameInt(string name, bool isPrimary) {
215     // Check to see if we're duplicating
216     for (unsigned i = 0; i < prettyNames_.size(); i++) {
217         if (prettyNames_[i] == name)
218             return false;
219     }
220
221     if (isPrimary) {
222         std::vector<std::string>::iterator iter = prettyNames_.begin();
223         prettyNames_.insert(iter, name);
224     }
225     else
226         prettyNames_.push_back(name);
227     return true;
228 }
229
230 bool Aggregate::addTypedNameInt(string name, bool isPrimary) {
231     // Check to see if we're duplicating
232     for (unsigned i = 0; i < typedNames_.size(); i++) {
233         if (typedNames_[i] == name)
234             return false;
235     }
236
237     if (isPrimary) {
238         std::vector<std::string>::iterator iter = typedNames_.begin();
239         typedNames_.insert(iter, name);
240     }
241     else
242         typedNames_.push_back(name);
243     return true;
244 }
245