This commit implements new approach to creating hashsection in rewritten binary....
[dyninst.git] / symtabAPI / src / emitElf.h
1 /*
2  * Copyright (c) 1996-2007 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 #if !defined(_emit_Elf_h_)
33 #define _emit_Elf_h_
34
35 #include "Object.h"
36 #include <vector>
37 using namespace std;
38
39 namespace Dyninst{
40 namespace SymtabAPI{
41
42 class emitElf{
43   public:
44     emitElf(Elf_X &oldElfHandle_, bool isStripped_ = false, int BSSexpandflag = false, void (*)(const char *) = log_msg);
45     ~emitElf();
46     bool createSymbolTables(Symtab *obj, vector<Symbol *>&allSymbols, std::vector<relocationEntry> &relocation_table);
47     bool driver(Symtab *obj, std::string fName);
48  
49   private:
50     Elf_X oldElfHandle;
51     Elf *newElf;
52     Elf *oldElf;
53     
54     //New Section & Program Headers
55     Elf32_Ehdr *newEhdr;
56     Elf32_Ehdr *oldEhdr;
57     
58     Elf32_Phdr *newPhdr;
59     Elf32_Phdr *oldPhdr;
60
61     //important data sections in the
62     //new Elf that need updated
63     Elf_Data *textData;
64     Elf_Data *symStrData;
65     Elf_Data *dynStrData;
66     char *olddynStrData;
67     unsigned olddynStrSize;
68     Elf_Data *symTabData;
69     Elf_Data *hashData;
70     Elf_Data *dynsymData;
71     Elf_Data *dynData;
72     Elf_Data *rodata;
73     Elf_Data *dataData;
74     
75     Elf32_Shdr *textSh;
76     Elf32_Shdr *rodataSh;
77     
78     std::vector<Region *>nonLoadableSecs;
79     std::vector<Region *> newSecs;
80 #if !defined(os_solaris)
81     std::map<unsigned, std::vector<Elf32_Dyn *> > dynamicSecData;
82     std::vector<std::string> DT_NEEDEDEntries;
83     std::vector<std::pair<long, long> > new_dynamic_entries;
84     std::vector<std::string> unversionedNeededEntries;
85 #endif
86
87     // Symbol version table data
88     std::map<std::string, std::map<std::string, unsigned> >verneedEntries;    //verneed entries
89     std::map<std::string, unsigned> verdefEntries;                            //verdef entries
90     std::map<unsigned, std::vector<std::string> > verdauxEntries;
91     std::map<std::string, unsigned> versionNames;
92     std::vector<Elf32_Half> versionSymTable;
93     int curVersionNum, verneednum, verdefnum;
94
95     // Needed when adding a new segment
96     Elf32_Off newSegmentStart;
97     Elf32_Shdr *firstNewLoadSec;// initialize to NULL
98  
99     //text & data segment ends
100     Elf32_Off dataSegEnd, textSegEnd;
101     Elf32_Off dynSegOff, dynSegAddr;
102     unsigned dynSegSize;
103
104     //Section Names for all sections
105     vector<std::string> secNames;
106     unsigned secNameIndex;
107
108     //flags
109     // Expand NOBITS sections within the object file to their size
110     bool BSSExpandFlag;
111     bool addNewSegmentFlag;
112     
113     bool isStripped;
114
115     void (*err_func_)(const char*);
116
117     bool createElfSymbol(Symbol *symbol, unsigned strIndex, vector<Elf32_Sym *> &symbols, bool dynSymFlag = false);
118     void findSegmentEnds();
119     void renameSection(const std::string &oldStr, const std::string &newStr, bool renameAll=true);
120     void fixPhdrs(unsigned &, unsigned &);
121     bool addSectionHeaderTable(Elf32_Shdr *shdr);
122     bool createNonLoadableSections(Elf32_Shdr *& shdr);
123     bool createLoadableSections( Elf32_Shdr* &shdr, unsigned &loadSecTotalSize, unsigned &, dyn_hash_map<std::string,  unsigned>& newIndexMapping, unsigned &sectionNumber);
124     void createRelocationSections(Symtab *obj, std::vector<relocationEntry> &relocation_table, dyn_hash_map<std::string, unsigned> &dynSymNameMapping);
125
126     void updateSymbols(Elf_Data* symtabData,Elf_Data* strData, unsigned long loadSecsSize);
127
128 #if !defined(os_solaris)
129     void updateDynamic(unsigned tag, Elf32_Addr val);
130     void createSymbolVersions(Symtab *obj, Elf32_Half *&symVers, char*&verneedSecData, unsigned &verneedSecSize, char *&verdefSecData, unsigned &verdefSecSize, unsigned &dynSymbolNamesLength, std::vector<std::string> &dynStrs);
131     void createHashSection(Symtab *obj, Elf32_Word *&hashsecData, unsigned &hashsecSize, std::vector<Symbol *>&dynSymbols);
132     void createDynamicSection(void *dynData, unsigned size, Elf32_Dyn *&dynsecData, unsigned &dynsecSize, unsigned &dynSymbolNamesLength, std::vector<std::string> &dynStrs);
133 #endif 
134
135     void log_elferror(void (*err_func)(const char *), const char* msg);
136 };
137
138 } // namespace SymtabAPI
139 } // namespace Dyninst
140
141 #endif