Update copyright to LGPL on all files
[dyninst.git] / testsuite / src / dyninst / test1_19.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: test1_19.C,v 1.1 2008/10/30 19:18:08 legendre Exp $
33 /*
34  * #Name: test1_19
35  * #Desc: Mutator Side - oneTimeCode
36  * #Dep: 
37  * #Notes: This test needs to be able to manipulate control flow
38  */
39
40 #include "BPatch.h"
41 #include "BPatch_Vector.h"
42 #include "BPatch_thread.h"
43 #include "BPatch_snippet.h"
44
45 #include "test_lib.h"
46
47 #include "dyninst_comp.h"
48
49 class test1_19_Mutator : public DyninstMutator {
50
51   virtual bool hasCustomExecutionPath() { return true; }
52   virtual test_results_t executeTest();
53 };
54
55 extern "C" DLLEXPORT  TestMutator *test1_19_factory() 
56 {
57   return new test1_19_Mutator();
58 }
59
60 static void test19_oneTimeCodeCallback(BPatch_thread * /*thread*/,
61                                 void *userData,
62                                 void * /*returnValue*/)
63 {
64    *(int *)userData = 1;
65 }
66
67 //
68 // Start Test Case #19 - mutator side (oneTimeCode)
69 //
70
71 test_results_t test1_19_Mutator::executeTest() 
72 {
73     // Avoid a race condition in fast & loose mode
74
75     while (!appThread->isStopped()) 
76         {
77                 BPatch::bpatch->waitForStatusChange();
78     }
79
80     appThread->continueExecution();
81
82     if (waitUntilStopped(BPatch::bpatch, appThread, 19, "oneTimeCode") < 0) 
83         {
84       return FAILED;
85     }
86
87     BPatch_Vector<BPatch_function *> bpfv;
88     char *fn = "test1_19_call1";
89
90     if (NULL == appImage->findFunction(fn, bpfv) || !bpfv.size()
91         || NULL == bpfv[0])
92         {
93       logerror("    Unable to find function %s\n", fn);
94       return FAILED;
95     }
96
97     BPatch_function *call19_1_func = bpfv[0];
98
99     BPatch_Vector<BPatch_snippet *> nullArgs;
100     BPatch_funcCallExpr call19_1Expr(*call19_1_func, nullArgs);
101     checkCost(call19_1Expr);
102
103     appThread->oneTimeCode(call19_1Expr);
104
105     // Let the mutatee run to check the result
106     appThread->continueExecution();
107
108     // Wait for the next test
109
110     if (waitUntilStopped(BPatch::bpatch, appThread, 19, "oneTimeCode") < 0) 
111         {
112       return FAILED;
113     }
114
115     bpfv.clear();
116     char *fn2 = "test1_19_call2";
117
118     if (NULL == appImage->findFunction(fn2, bpfv) || !bpfv.size()
119         || NULL == bpfv[0])
120         {
121       logerror("    Unable to find function %s\n", fn2);
122       return FAILED;
123     }
124
125     BPatch_function *call19_2_func = bpfv[0];
126
127     BPatch_funcCallExpr call19_2Expr(*call19_2_func, nullArgs);
128     checkCost(call19_2Expr);
129
130     int callbackFlag = 0;
131
132     // Register a callback that will set the flag callbackFlag
133     BPatchOneTimeCodeCallback oldCallback = 
134        BPatch::bpatch->registerOneTimeCodeCallback(test19_oneTimeCodeCallback);
135
136     appThread->oneTimeCodeAsync(call19_2Expr, (void *)&callbackFlag);
137
138     while (!appThread->isTerminated() && !appThread->isStopped() )
139     {
140                 BPatch::bpatch->waitForStatusChange();
141     }
142     
143     // Continue mutatee after one-time code runs
144     appThread->continueExecution();
145
146     // Wait for the callback to be called
147     while (!appThread->isTerminated() && !callbackFlag) ;
148
149     // Restore old callback (if there was one)
150         BPatch::bpatch->registerOneTimeCodeCallback(oldCallback);
151
152     return PASSED;
153 } // test1_19_Mutator::executeTest()
154