windows build fixes and some more work on symtab serialization
[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, std::vector<relocationEntry> &fbt);
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     Elf_Data *symTabData;
68     Elf_Data *hashData;
69     Elf_Data *dynsymData;
70     Elf_Data *dynData;
71     Elf_Data *rodata;
72     Elf_Data *dataData;
73     
74     Elf32_Shdr *textSh;
75     Elf32_Shdr *rodataSh;
76     
77     std::vector<Region *>nonLoadableSecs;
78     std::vector<Region *> newSecs;
79 #if !defined(os_solaris)
80     std::map<unsigned, std::vector<Elf32_Dyn *> > dynamicSecData;
81     std::vector<std::string> DT_NEEDEDEntries;
82     std::vector<std::string> unversionedNeededEntries;
83 #endif
84
85     // Symbol version table data
86     std::map<std::string, std::map<std::string, unsigned> >verneedEntries;    //verneed entries
87     std::map<std::string, unsigned> verdefEntries;                            //verdef entries
88     std::map<unsigned, std::vector<std::string> > verdauxEntries;
89     std::map<std::string, unsigned> versionNames;
90     std::vector<Elf32_Half> versionSymTable;
91     int curVersionNum, verneednum, verdefnum;
92
93     // Needed when adding a new segment
94     Elf32_Off newSegmentStart;
95     Elf32_Shdr *firstNewLoadSec;// initialize to NULL
96  
97     //text & data segment ends
98     Elf32_Off dataSegEnd, textSegEnd;
99     Elf32_Off dynSegOff, dynSegAddr;
100     unsigned dynSegSize;
101
102     //Section Names for all sections
103     vector<std::string> secNames;
104     unsigned secNameIndex;
105
106     //flags
107     // Expand NOBITS sections within the object file to their size
108     bool BSSExpandFlag;
109     bool addNewSegmentFlag;
110     
111     bool isStripped;
112
113     void (*err_func_)(const char*);
114
115     bool createElfSymbol(Symbol *symbol, std::vector<std::string> &symbolstrs, unsigned &symbolNamesLength, vector<Elf32_Sym *> &symbols, bool dynSymFlag = false);
116     void findSegmentEnds();
117     void renameSection(const std::string &oldStr, const std::string &newStr, bool renameAll=true);
118     void fixPhdrs(unsigned &, unsigned &);
119     bool addSectionHeaderTable(Elf32_Shdr *shdr);
120     bool createNonLoadableSections(Elf32_Shdr *& shdr);
121     bool createLoadableSections( Elf32_Shdr *shdr, unsigned &loadSecTotalSize, unsigned &);
122     void createRelocationSections(Symtab *obj, std::vector<relocationEntry> &relocation_table, std::vector<relocationEntry> &fbt, dyn_hash_map<std::string, unsigned> &dynSymNameMapping);
123
124     void updateSymbols(Elf_Data* symtabData,Elf_Data* strData, unsigned long loadSecsSize);
125
126 #if !defined(os_solaris)
127     void updateDynamic(unsigned tag, Elf32_Addr val);
128     void createSymbolVersions(Symtab *obj, Elf32_Half *&symVers, char*&verneedSecData, unsigned &verneedSecSize, char *&verdefSecData, unsigned &verdefSecSize, unsigned &dynSymbolNamesLength, std::vector<std::string> &dynStrs);
129     void createHashSection(Elf32_Word *&hashsecData, unsigned &hashsecSize, std::vector<Symbol *>&dynSymbols);
130     void createDynamicSection(void *dynData, unsigned size, Elf32_Dyn *&dynsecData, unsigned &dynsecSize, unsigned &dynSymbolNamesLength, std::vector<std::string> &dynStrs);
131 #endif 
132
133     void log_elferror(void (*err_func)(const char *), const char* msg);
134 };
135
136 } // namespace SymtabAPI
137 } // namespace Dyninst
138
139 #endif