Update copyright to LGPL on all files
[dyninst.git] / testsuite / src / dyninst / snip_ref_shlib_var.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
33 /*
34  * #Name: snip_ref_shlib_var
35  * #Desc: use instrumentation to modify variable in shared library
36  * #Dep: 
37  * #Arch: sparc_sun_solaris2_4_test,i386_unknown_linux2_0_test,alpha_dec_osf4_0_test,ia64_unknown_linux2_4_test,x86_64_unknown_linux2_4_test
38  * #Notes: This test uses libNameA/B 
39  */
40
41 #include "BPatch.h"
42 #include "BPatch_Vector.h"
43 #include "BPatch_thread.h"
44 #include "BPatch_snippet.h"
45
46 #include "test_lib.h"
47 #include "dyninst_comp.h"
48
49 #ifdef os_windows_test
50 #define snprintf _snprintf
51 #endif
52
53
54 class snip_ref_shlib_var_Mutator : public DyninstMutator {
55         const char *libNameAroot;
56         const char *libNameBroot;
57         char libNameA[128], libNameB[128];
58
59         BPatch_snippet *doVarAssign(const char * to, const char *from);
60         public:
61         snip_ref_shlib_var_Mutator();
62         virtual test_results_t executeTest();
63         virtual test_results_t mutatorTest();
64 };
65
66 snip_ref_shlib_var_Mutator::snip_ref_shlib_var_Mutator() : libNameAroot("libtestA"),
67                                                                            libNameBroot("libtestB") 
68 {
69 }
70
71 extern "C" DLLEXPORT  TestMutator *snip_ref_shlib_var_factory() 
72 {
73         return new snip_ref_shlib_var_Mutator();
74 }
75
76 BPatch_snippet * snip_ref_shlib_var_Mutator::doVarAssign(const char *to, const char *from) 
77 {
78         BPatch_variableExpr *to_v = appImage->findVariable(to);
79
80         if (!to_v)
81         {
82                 logerror("%s[%d]:  failed to find var %s\n", FILE__, __LINE__, to);
83                 return NULL;
84         }
85
86         BPatch_variableExpr *from_v = appImage->findVariable(from);
87
88         if (!from_v)
89         {
90                 logerror("%s[%d]:  failed to find var %s\n", FILE__, __LINE__, from);
91                 return NULL;
92         }
93
94         BPatch_snippet *ret;
95         ret = new BPatch_arithExpr(BPatch_assign, *to_v, *from_v);
96         assert(ret);
97         return ret;
98 }
99
100 test_results_t snip_ref_shlib_var_Mutator::mutatorTest() 
101 {
102         //  The check function returns 1 on success (value changed as expected)
103         //  or 0 on failure.
104         const char *check_fname  = "check_snip_ref_shlib_var";
105         const char *inst_func_name = "srsv1";
106         BPatch_Vector<BPatch_function *> funcs;
107
108         appImage->findFunction(inst_func_name, funcs);
109
110         if (!funcs.size())
111         {
112                 logerror("%s[%d]:  failed to find function %s\n", 
113                                 FILE__, __LINE__, inst_func_name);
114                 return FAILED;
115         }
116
117         BPatch_function *inst_func = funcs[0];
118
119         std::vector<BPatch_point *> *pts= inst_func->findPoint(BPatch_entry);
120
121         if (!pts || !pts->size())
122         {
123                 logerror("%s[%d]:   failed to find entry point to %s\n", 
124                                 FILE__, __LINE__, inst_func_name);
125                 return FAILED;
126         }
127
128         BPatch_point *entry_point = (*pts)[0];
129
130         BPatch_Vector<BPatch_snippet *> allInst;
131         BPatch_snippet *snip;
132
133         snip = doVarAssign("gv_srsv1", "snip_ref_shlib_var1");
134         if (NULL == snip) return FAILED;
135         allInst.push_back(snip);
136
137         snip = doVarAssign("gv_srsv2", "snip_ref_shlib_var2");
138         if (NULL == snip) return FAILED;
139         allInst.push_back(snip);
140
141         snip = doVarAssign("gv_srsv3", "snip_ref_shlib_var3");
142         if (NULL == snip) return FAILED;
143         allInst.push_back(snip);
144
145         snip = doVarAssign("gv_srsv4", "snip_ref_shlib_var4");
146         if (NULL == snip) return FAILED;
147         allInst.push_back(snip);
148
149         snip = doVarAssign("gv_srsv5", "snip_ref_shlib_var5");
150         if (NULL == snip) return FAILED;
151         allInst.push_back(snip);
152
153 #if 0
154         snip = doVarAssign("gv_srsv6", "snip_ref_shlib_var6");
155         if (NULL == snip) return FAILED;
156         allInst.push_back(snip);
157 #endif
158
159         BPatch_sequence my_ass(allInst);
160
161         if (!appAddrSpace->insertSnippet(my_ass, *entry_point))
162         {
163                 logerror("%s[%d]:  failed to insert snippet\n", FILE__, __LINE__);
164                 return FAILED;
165         }
166
167
168         return PASSED;
169 }
170
171 test_results_t snip_ref_shlib_var_Mutator::executeTest() 
172 {
173         int pointer_size = 0;
174 #if defined(arch_x86_64_test) || defined(ppc64_linux_test)
175         pointer_size = pointerSize(appImage);
176 #endif
177         strncpy(libNameA, libNameAroot, 128);
178         addLibArchExt(libNameA,128, pointer_size);
179         strncpy(libNameB, libNameBroot, 128);
180         addLibArchExt(libNameB,128, pointer_size);
181
182         char libA[128], libB[128];
183         snprintf(libA, 128, "./%s", libNameA);
184         snprintf(libB, 128, "./%s", libNameB);
185
186         if (! appAddrSpace->loadLibrary(libA)) 
187         {
188                 logerror("**Failed snip_ref_shlib_var (findFunction in module)\n");
189                 logerror("  Mutator couldn't load %s into mutatee\n", libNameA);
190                 return FAILED;
191         }
192
193         if (! appAddrSpace->loadLibrary(libB)) 
194         {
195                 logerror("**Failed snip_ref_shlib_var (findFunction in module)\n");
196                 logerror("  Mutator couldn't load %s into mutatee\n", libNameB);
197                 return FAILED;
198         }
199
200         return mutatorTest();
201 } // snip_ref_shlib_var_Mutator::executeTest()
202