Update copyright to LGPL on all files
[dyninst.git] / dyninstAPI / src / writeBackElf.h
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 /* -*- Mode: C; indent-tabs-mode: true -*- */
33 /* $Id: writeBackElf.h,v 1.21 2007/01/09 02:02:08 giri Exp $ */
34
35 #ifndef writeBackElf__
36 #define writeBackElf__
37
38 #if defined(sparc_sun_solaris2_4) \
39  || defined(i386_unknown_linux2_0) \
40  || defined(x86_64_unknown_linux2_4) /* Blind duplication - Ray */ \
41  || defined(ia64_unknown_linux2_4)
42
43 #include <fcntl.h>
44 #include <stdio.h>
45 #include <stdlib.h>
46 #include <string.h>
47 #include <libelf.h>
48 #if defined(i386_unknown_linux2_0) \
49  || defined(x86_64_unknown_linux2_4) /* Blind duplication - Ray */ \
50  || defined(ia64_unknown_linux2_4)
51
52 #include "linux.h"
53 #endif
54 #include "common/h/Vector.h"
55 #include "dyninstAPI/src/symtab.h"
56 #include "ELF_Section.h"
57 #include <unistd.h>
58 #include "imageUpdate.h"
59
60 class process;
61
62 class writeBackElf{
63
64 private:
65         
66         Elf* newElf;
67         Elf* oldElf;
68
69         int oldfd, newfd;
70         //important data sections in the
71         //new Elf that need updated
72         Elf_Data *textData;
73         Elf_Data *symStrData;
74         Elf_Data *dynStrData;
75         Elf_Data *symTabData;
76         Elf_Data *hashData;
77         Elf_Data *dynsymData;
78         Elf_Data *rodata;
79         Elf_Data *dataData;
80
81         Elf32_Shdr *textSh;
82         Elf32_Shdr *rodataSh;
83         Elf32_Ehdr *newEhdr;
84         Elf32_Phdr *newPhdr;
85
86         //when this is added to dyninst make this a
87         //vector or something nice like that
88         ELF_Section *newSections;
89         unsigned int newSectionsSize;
90
91         unsigned int lastAddr; //last used address in the old Elf
92         unsigned int shiftSize; //shift caused by adding program headers, a multiple of 0x20
93         unsigned int dataStartAddress; 
94         unsigned int insertPoint; //index in newElf of first new section 
95
96         unsigned int startAddr; //start address of .text in newElf
97         unsigned int endAddr; //end address of .text in newElf 
98         unsigned int rodataAddr; //start address of .rodata in newElf 
99         unsigned int rodataSize; //size of .rodata in newElf
100         unsigned int newHeapAddr; //location of the heap in the newElf
101
102         unsigned int oldLastPage;//location of the last page in memory allocated in the oldElf
103         int DEBUG_MSG;
104         int pageSize ; // page size on this system
105         process* mutateeProcess;
106         int mutateeTextSize;
107         unsigned int mutateeTextAddr;
108         unsigned int newHeapAddrIncr;
109         char **newElfNewData_d_buf;
110         int newElfNewData_d_buf_count;
111
112         void updateSymbols(Elf_Data* symtabData,Elf_Data* strData);
113         void updateDynamic(Elf_Data* dynamicData);
114         void driver(); // main processing loop of outputElf()
115         void createSections();
116         void addSectionNames(Elf_Data* newdata, Elf_Data *olddata);
117         void fixPhdrs(Elf32_Phdr*);
118         void parseOldElf();
119
120 public:
121
122         writeBackElf(const char* oldElfName, const char* newElfName,
123                                          int debugOutputFlag=0);
124         ~writeBackElf();
125
126         void registerProcess(process *proc);
127         
128         int addSection(unsigned int addr, void *data, unsigned int dataSize,
129                                                 const char* name, bool loadable=true);
130
131         bool createElf();
132         void compactSections(pdvector<imageUpdate*> imagePatches,
133                                                                 pdvector<imageUpdate*> &newPatches); 
134         void compactLoadableSections(pdvector<imageUpdate*> imagePatches,
135                                                                                   pdvector<imageUpdate*> &newPatches);
136         void alignHighMem(pdvector<imageUpdate*> imagesPatches);
137         Elf* getElf();
138 };
139
140 #endif
141 #endif