Add support for indirect relocations in static binary rewriting
[dyninst.git] / dyninstAPI_RT / src / RTstatic_ctors_dtors-x86.c
1 /*
2  * See the dyninst/COPYRIGHT file for copyright information.
3  * 
4  * We provide the Paradyn Tools (below described as "Paradyn")
5  * on an AS IS basis, and do not warrant its validity or performance.
6  * We reserve the right to update, modify, or discontinue this
7  * software at any time.  We shall have no obligation to supply such
8  * updates or modifications or any other form of support to you.
9  * 
10  * By your use of Paradyn, you understand and agree that we (or any
11  * other person or entity with proprietary rights in Paradyn) are
12  * under no obligation to provide either maintenance services,
13  * update services, notices of latent defects, or correction of
14  * defects for Paradyn.
15  * 
16  * This library is free software; you can redistribute it and/or
17  * modify it under the terms of the GNU Lesser General Public
18  * License as published by the Free Software Foundation; either
19  * version 2.1 of the License, or (at your option) any later version.
20  * 
21  * This library is distributed in the hope that it will be useful,
22  * but WITHOUT ANY WARRANTY; without even the implied warranty of
23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
24  * Lesser General Public License for more details.
25  * 
26  * You should have received a copy of the GNU Lesser General Public
27  * License along with this library; if not, write to the Free Software
28  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
29  */
30
31 #if defined(DYNINST_RT_STATIC_LIB)
32 void (*DYNINSTctors_addr)(void);
33 void (*DYNINSTdtors_addr)(void);
34 void *DYNINSTirel_start;
35 void *DYNINSTirel_end;
36
37 #if defined(MUTATEE64)
38 static const unsigned long long CTOR_LIST_TERM = 0x0000000000000000ULL;
39 static const unsigned long long CTOR_LIST_START = 0xffffffffffffffffULL;
40 static const unsigned long long DTOR_LIST_TERM = 0x0000000000000000ULL;
41 static const unsigned long long DTOR_LIST_START = 0xffffffffffffffffULL;
42 #else
43 static const unsigned CTOR_LIST_TERM = 0x00000000;
44 static const unsigned CTOR_LIST_START = 0xffffffff;
45 static const unsigned DTOR_LIST_TERM = 0x00000000;
46 static const unsigned DTOR_LIST_START = 0xffffffff;
47 #endif
48
49 extern void DYNINSTBaseInit();
50
51 typedef struct {
52   long *offset;
53   long info;
54   long (*ptr)(void);
55 } rela_t;
56
57
58 /*
59  * When rewritting a static binary, .ctors and .dtors sections of
60  * instrumentation code needs to be combined with the existing .ctors
61  * and .dtors sections of the static binary.
62  *
63  * The following functions process the .ctors and .dtors sections
64  * that have been rewritten. The rewriter will relocate the 
65  * address of DYNINSTctors_addr and DYNINSTdtors_addr to point to
66  * new .ctors and .dtors sections.
67  */
68
69 void DYNINSTglobal_ctors_handler() {
70     void (**ctors_array)(void) = &DYNINSTctors_addr;
71
72     // Find end of function pointer list
73     void (**tmp_ptr)(void) = ctors_array;
74     unsigned size = 0;
75     while( *tmp_ptr != ( (void(*)(void))CTOR_LIST_TERM ) ) {
76         size++;
77         tmp_ptr++;
78     }
79
80     // Constructors are called in the reverse order that they are listed
81     tmp_ptr = &ctors_array[size-1]; // skip list end
82     while( *tmp_ptr != ( (void(*)(void))CTOR_LIST_START ) ) {
83         (*tmp_ptr)();
84         tmp_ptr--;
85     }
86
87     // This ensures that instrumentation cannot execute until all global
88     // constructors have run
89     DYNINSTBaseInit();
90 }
91
92 void DYNINSTglobal_dtors_handler() {
93     void (**dtors_array)(void) = &DYNINSTdtors_addr;
94
95     // Destructors are called in the forward order that they are listed
96     void (**tmp_ptr)(void) = &dtors_array[1]; // skip list start
97     while( *tmp_ptr != ( (void(*)(void))DTOR_LIST_TERM ) ) {
98         (*tmp_ptr)();
99         tmp_ptr++;
100     }
101 }
102
103 void DYNINSTglobal_irel_handler() {
104   rela_t *rel = 0;
105   for (rel = (rela_t *)(&DYNINSTirel_start); rel != (rela_t *)(&DYNINSTirel_end); ++rel) {
106     long result = 0;
107     if (rel->info != 0x25) continue;
108     result = (rel->ptr());
109     *(rel->offset) = result;
110   }
111 }
112
113 #endif