Bugfix: don't use operator[] if the dictionary doesn't define the key.
[dyninst.git] / dyninstAPI / src / BPatch_libInfo.h
1 /*
2  * Copyright (c) 1996-2004 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  * This license is for research uses.  For such uses, there is no
12  * charge. We define "research use" to mean you may freely use it
13  * inside your organization for whatever purposes you see fit. But you
14  * may not re-distribute Paradyn or parts of Paradyn, in any form
15  * source or binary (including derivatives), electronic or otherwise,
16  * to any other organization or entity without our permission.
17  * 
18  * (for other uses, please contact us at paradyn@cs.wisc.edu)
19  * 
20  * All warranties, including without limitation, any warranty of
21  * merchantability or fitness for a particular purpose, are hereby
22  * excluded.
23  * 
24  * By your use of Paradyn, you understand and agree that we (or any
25  * other person or entity with proprietary rights in Paradyn) are
26  * under no obligation to provide either maintenance services,
27  * update services, notices of latent defects, or correction of
28  * defects for Paradyn.
29  * 
30  * Even if advised of the possibility of such damages, under no
31  * circumstances shall we (or any other person or entity with
32  * proprietary rights in the software licensed hereunder) be liable
33  * to you or any third party for direct, indirect, or consequential
34  * damages of any character regardless of type of action, including,
35  * without limitation, loss of profits, loss of use, loss of good
36  * will, or computer failure or malfunction.  You agree to indemnify
37  * us (and any other person or entity with proprietary rights in the
38  * software licensed hereunder) for any and all liability it may
39  * incur to third parties resulting from your use of Paradyn.
40  */
41
42 #ifndef _BPatch_libInfo_h_
43 #define _BPatch_libInfo_h_
44
45 #include <sys/types.h>
46 #include "dyninstAPI/h/BPatch_process.h"
47 #include "common/h/Dictionary.h"
48 #include "common/h/Types.h"
49 #include "util.h"
50
51 class BPatch_libInfo {
52 public:
53    dictionary_hash<int, BPatch_process *> procsByPid;
54    BPatch_libInfo(): procsByPid(intHash) {};
55 };
56
57 class BPatch_funcMap {
58    dictionary_hash<const int_function*, BPatch_function*> chart;
59    static unsigned hash_bp(const int_function * const &bp ) {
60       return(addrHash4((Address) bp));
61    }
62  public:
63    BPatch_funcMap() : chart(hash_bp) {}
64    ~BPatch_funcMap() { chart.clear(); }
65
66    bool defines(const int_function *func) 
67       { return chart.defines(func); }
68    void add(const int_function *func, BPatch_function *bfunc) 
69       {
70        if( func == NULL ) { return; }
71        if( bfunc == NULL ) { return; }
72        chart[func] = bfunc;
73       }
74    void undefine(const int_function *func) {
75        if (func == NULL) { return; }
76        if (chart.defines(func)) {
77            chart.undef(func);
78        }
79    }
80    BPatch_function *get(const int_function *func) 
81       { 
82       if( chart.defines( func ) ) { return chart.get( func ); }
83       else { return NULL; }
84       }
85    void map(bool (*f)(BPatch_function *, void *), void *data ) {
86       dictionary_hash<const int_function *, BPatch_function *>::iterator iter = 
87          chart.begin();
88       dictionary_hash<const int_function *, BPatch_function *>::iterator end = 
89          chart.end();
90       for (; iter != end; ++iter)
91          if (!f(*iter, data))
92             break;
93    }
94       
95 };
96
97 class BPatch_instpMap {
98   // Note: if we ever have multiple BPatch_points to a single
99   // instPoint (frex, a BPatch_point representing a loop backedge and
100   // a BPatch_point representing the same edge in the CFG), this will
101   // be an instPoint -> vector for BPatch_point map
102    dictionary_hash<const instPoint *, BPatch_point *> chart;
103    static unsigned hash_ip(const instPoint *const &ip) {
104      return (addrHash4((Address)ip));
105    }
106  public:
107    BPatch_instpMap() : chart(hash_ip) {}
108    ~BPatch_instpMap() { chart.clear(); }
109
110    bool defines(instPoint *a) { return chart.defines(a); }
111    void add(instPoint *a, BPatch_point *bp) { chart[a] = bp; }
112    BPatch_point *get(const instPoint *a) { return chart[a]; }
113 };
114
115 #endif /* _BPatch_libInfo_h_ */