Update copyright to LGPL on all files
[dyninst.git] / valueAdded / sharedMem / src / shmSegment.C
1 /*
2  * Copyright (c) 1996-2009 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 // $Id: shmSegment.C,v 1.1 2006/11/22 21:45:04 bernat Exp $
33 //----------------------------------------------------------------------------
34
35 #include "shmSegment.h"
36 #include "../h/SharedMem.h"
37 #include "sharedMemInternal.h"
38 #include "BPatch_process.h"
39
40 const unsigned ShmSegment::cookie = 0xdeadbeef;
41
42
43 // Cross-platform shmSegment definitions
44 bool ShmSegment::attach(BPatch_process *appProc, Address baseAddr) {
45     BPatch_Vector<BPatch_snippet *>init_args;
46     BPatch_constExpr init_key(GetKey());
47     BPatch_constExpr init_size(GetSize());
48     BPatch_constExpr addr(baseAddr);
49     init_args.push_back(&init_key);
50     init_args.push_back(&init_size);
51     init_args.push_back(&addr);
52
53     assert(appProc->isStopped());
54     BPatch_Vector<BPatch_function *>init_func;
55     if ((NULL == appProc->getImage()->findFunction("RTsharedAttach",
56                                                      init_func)) ||
57         init_func.size() == 0) {
58         fprintf(stderr, "Failed to find shared init function\n");
59         return false;
60     }
61     assert(appProc->isStopped());
62     BPatch_funcCallExpr init_expr(*(init_func[0]), init_args);
63     assert(appProc->isStopped());
64     
65     // We return the address attached to by the child -- set up
66     // argument grabbing
67
68     assert(appProc->isStopped());
69     baseAddrInApplic = (Address) appProc->oneTimeCode(init_expr);
70     assert(appProc->isStopped());
71
72     if (baseAddrInApplic != (Address) -1) {
73         if (baseAddr &&
74             (baseAddrInApplic != baseAddr)) {
75             // We were given a base address, but didn't get it. This is bad.
76             fprintf(stderr, "Error: attached to incorrect address\n");
77             return false;
78         }
79         attached = true;
80         return true;
81     }
82     fprintf(stderr, "Attach failed!\n");
83     assert(appProc->isStopped());
84     
85     return false;
86 }
87
88 bool ShmSegment::detach(BPatch_process *appProc) {
89     BPatch_Vector<BPatch_snippet *>init_args;
90     BPatch_constExpr init_key(GetKey());
91     BPatch_constExpr init_size(GetSize());
92     BPatch_constExpr addr(baseAddrInApplic);
93     init_args.push_back(&init_key);
94     init_args.push_back(&init_size);
95     init_args.push_back(&addr);
96
97     BPatch_Vector<BPatch_function *>init_func;
98     if ((NULL == appProc->getImage()->findFunction("RTsharedDetach",
99                                                      init_func)) ||
100         init_func.size() == 0) {
101         fprintf(stderr, "Failed to find shared init function\n");
102         return false;
103     }
104     BPatch_funcCallExpr init_expr(*(init_func[0]), init_args);
105     
106     // We return the address attached to by the child -- set up
107     // argument grabbing
108
109     baseAddrInApplic = (Address) appProc->oneTimeCode(init_expr);
110     
111     if (baseAddrInApplic != (Address) -1) {
112         attached = true;
113         return true;
114     }
115     return false;
116 }
117
118 Address ShmSegment::malloc(unsigned size) {
119     if (!attached) {
120         // Could try to attach here...
121         return 0;
122     }
123     if (freespace < size) return 0;
124     
125     // Just carve a chunk off and update local variables
126     Address ret = highWaterMark + baseAddrInDaemon;
127     highWaterMark += size;
128     freespace -= size;
129     return ret;
130 }
131
132 void ShmSegment::free(Address /*addr*/) {
133     // We need more sophisticated tracking before this
134     // will work
135 }
136
137
138
139