Add consistency checking to Symtab when adding new Functions/Variables; clean up...
[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(Symbol *sym) :
51     module_(NULL)
52 {
53     assert(sym);
54     module_ = sym->getModule();
55     symbols_.push_back(sym);
56     mangledNames_.push_back(sym->getMangledName());
57     prettyNames_.push_back(sym->getPrettyName());
58     typedNames_.push_back(sym->getTypedName());
59 }
60
61
62 const vector<std::string> &Aggregate::getAllMangledNames() {
63     return mangledNames_;
64 }
65 const vector<std::string> &Aggregate::getAllPrettyNames() {
66     return prettyNames_;
67 }
68 const vector<std::string> &Aggregate::getAllTypedNames() {
69     return typedNames_;
70 }
71
72 bool Aggregate::addSymbol(Symbol *sym) {
73
74     // We keep a "primary" module, which is defined as "anything not DEFAULT_MODULE".
75     if (module_ == NULL) {
76         module_ = sym->getModule();
77     }
78     else if (module_->fileName() == "DEFAULT_MODULE") {
79         module_ = sym->getModule();
80     }
81     // else keep current module.
82
83     symbols_.push_back(sym);
84
85     // We need to add the symbol names (if they aren't there already)
86     // We can have multiple identical names - for example, there are
87     // often two symbols for main (static and dynamic symbol table)
88     
89     bool found = false;
90     for (unsigned j = 0; j < mangledNames_.size(); j++) {
91         if (sym->getMangledName() == mangledNames_[j]) {
92             found = true;
93             break;
94         }
95     }
96     if (!found) mangledNames_.push_back(sym->getMangledName());
97
98     found = false;
99
100     for (unsigned j = 0; j < prettyNames_.size(); j++) {
101         if (sym->getPrettyName() == prettyNames_[j]) {
102             found = true;
103             break;
104         }
105     }
106     if (!found) prettyNames_.push_back(sym->getPrettyName());
107
108     found = false;
109     for (unsigned j = 0; j < typedNames_.size(); j++) {
110         if (sym->getTypedName() == typedNames_[j]) {
111             found = true;
112             break;
113         }
114     }
115     if (!found) typedNames_.push_back(sym->getTypedName());;
116
117     return true;
118 }
119
120 bool Aggregate::removeSymbol(Symbol *sym) {
121     std::vector<Symbol *>::iterator iter;
122     for (iter = symbols_.begin(); iter != symbols_.end(); iter++) {
123         if (*iter == sym) {
124             symbols_.erase(iter);
125             return true;
126         }
127     }
128     // TODO: remove from names. Do we ever call this?
129
130     return false;
131 }
132
133 bool Aggregate::getSymbols(std::vector<Symbol *> &syms) const 
134 {
135     syms = symbols_;
136     return true;
137 }
138
139 Symbol * Aggregate::getFirstSymbol() const
140 {
141     assert( symbols_.size()>0 );
142     return symbols_[0];
143 }
144
145 SYMTAB_EXPORT bool Aggregate::addMangledName(string name, bool isPrimary) 
146  {
147     // Check to see if we're duplicating
148     for (unsigned i = 0; i < mangledNames_.size(); i++) {
149         if (mangledNames_[i] == name)
150             return false;
151     }
152
153     if (isPrimary) {
154         std::vector<std::string>::iterator iter = mangledNames_.begin();
155         mangledNames_.insert(iter, name);
156     }
157     else
158         mangledNames_.push_back(name);
159
160     if (addMangledNameInt(name, isPrimary) == false) return false;
161      return true;
162  }                                                                                                                                      
163  
164  SYMTAB_EXPORT bool Aggregate::addPrettyName(string name, bool isPrimary) 
165  {
166     // Check to see if we're duplicating
167     for (unsigned i = 0; i < prettyNames_.size(); i++) {
168         if (prettyNames_[i] == name)
169             return false;
170     }
171
172     if (isPrimary) {
173         std::vector<std::string>::iterator iter = prettyNames_.begin();
174         prettyNames_.insert(iter, name);
175     }
176     else
177         prettyNames_.push_back(name);
178     if (addPrettyNameInt(name, isPrimary) == false) return false;
179      return true;
180  }                                                                                                                                      
181  
182  SYMTAB_EXPORT bool Aggregate::addTypedName(string name, bool isPrimary) 
183  {
184     // Check to see if we're duplicating
185     for (unsigned i = 0; i < typedNames_.size(); i++) {
186         if (typedNames_[i] == name)
187             return false;
188     }
189     if (addTypedNameInt(name, isPrimary) == false) return false;
190
191     if (isPrimary) {
192         std::vector<std::string>::iterator iter = typedNames_.begin();
193         typedNames_.insert(iter, name);
194     }
195     else
196         typedNames_.push_back(name);
197         return true;
198  }
199
200 bool Aggregate::addMangledNameInt(string name, bool isPrimary) {
201     // Check to see if we're duplicating
202     for (unsigned i = 0; i < mangledNames_.size(); i++) {
203         if (mangledNames_[i] == name)
204             return false;
205     }
206
207     if (isPrimary) {
208         std::vector<std::string>::iterator iter = mangledNames_.begin();
209         mangledNames_.insert(iter, name);
210     }
211     else
212         mangledNames_.push_back(name);
213     return true;
214 }
215
216 bool Aggregate::addPrettyNameInt(string name, bool isPrimary) {
217     // Check to see if we're duplicating
218     for (unsigned i = 0; i < prettyNames_.size(); i++) {
219         if (prettyNames_[i] == name)
220             return false;
221     }
222
223     if (isPrimary) {
224         std::vector<std::string>::iterator iter = prettyNames_.begin();
225         prettyNames_.insert(iter, name);
226     }
227     else
228         prettyNames_.push_back(name);
229     return true;
230 }
231
232 bool Aggregate::addTypedNameInt(string name, bool isPrimary) {
233     // Check to see if we're duplicating
234     for (unsigned i = 0; i < typedNames_.size(); i++) {
235         if (typedNames_[i] == name)
236             return false;
237     }
238
239     if (isPrimary) {
240         std::vector<std::string>::iterator iter = typedNames_.begin();
241         typedNames_.insert(iter, name);
242     }
243     else
244         typedNames_.push_back(name);
245     return true;
246 }