1 // icf.h -- Identical Code Folding
2
3 // Copyright (C) 2009-2016 Free Software Foundation, Inc.
4 // Written by Sriraman Tallam <tmsriram@google.com>.
5
6 // This file is part of gold.
7
8 // This program is free software; you can redistribute it and/or modify
9 // it under the terms of the GNU General Public License as published by
10 // the Free Software Foundation; either version 3 of the License, or
11 // (at your option) any later version.
12
13 // This program is distributed in the hope that it will be useful,
14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 // GNU General Public License for more details.
17
18 // You should have received a copy of the GNU General Public License
19 // along with this program; if not, write to the Free Software
20 // Foundation, Inc., 51 Franklin Street - Fifth Floor, Boston,
21 // MA 02110-1301, USA.
22
23 #ifndef GOLD_ICF_H
24 #define GOLD_ICF_H
25
26 #include <vector>
27
28 #include "elfcpp.h"
29 #include "symtab.h"
30 #include "object.h"
31
32 namespace gold
33 {
34
35 class Object;
36 class Input_objects;
37 class Symbol_table;
38
39 class Icf
40 {
41 public:
42 typedef std::vector<Section_id> Sections_reachable_info;
43 typedef std::vector<Symbol*> Symbol_info;
44 typedef std::vector<std::pair<long long, long long> > Addend_info;
45 typedef std::vector<uint64_t> Offset_info;
46 typedef std::vector<unsigned int> Reloc_addend_size_info;
47 typedef Unordered_map<Section_id,
48 unsigned int,
49 Section_id_hash> Uniq_secn_id_map;
50 typedef Unordered_set<Section_id, Section_id_hash> Secn_fptr_taken_set;
51
52 typedef struct
53 {
54 // This stores the section corresponding to the reloc.
55 Sections_reachable_info section_info;
56 // This stores the symbol corresponding to the reloc.
57 Symbol_info symbol_info;
58 // This stores the symbol value and the addend for a reloc.
59 Addend_info addend_info;
60 Offset_info offset_info;
61 Reloc_addend_size_info reloc_addend_size_info;
62 } Reloc_info;
63
64 typedef Unordered_map<Section_id, Reloc_info,
65 Section_id_hash> Reloc_info_list;
66
Icf()67 Icf()
68 : id_section_(), section_id_(), kept_section_id_(),
69 fptr_section_id_(),
70 icf_ready_(false),
71 reloc_info_list_()
72 { }
73
74 // Returns the kept folded identical section corresponding to
75 // dup_obj and dup_shndx.
76 Section_id
77 get_folded_section(Relobj* dup_obj, unsigned int dup_shndx);
78
79 // Forms groups of identical sections where the first member
80 // of each group is the kept section during folding.
81 void
82 find_identical_sections(const Input_objects* input_objects,
83 Symbol_table* symtab);
84
85 // This is set when ICF has been run and the groups of
86 // identical sections have been formed.
87 void
icf_ready()88 icf_ready()
89 { this->icf_ready_ = true; }
90
91 // Returns true if ICF has been run.
92 bool
is_icf_ready()93 is_icf_ready()
94 { return this->icf_ready_; }
95
96 // Unfolds the section denoted by OBJ and SHNDX if folded.
97 void
98 unfold_section(Relobj* obj, unsigned int shndx);
99
100 // Returns the kept section corresponding to the
101 // given section.
102 bool
103 is_section_folded(Relobj* obj, unsigned int shndx);
104
105 // Given an object and a section index, this returns true if the
106 // pointer of the function defined in this section is taken.
107 bool
section_has_function_pointers(Relobj * obj,unsigned int shndx)108 section_has_function_pointers(Relobj* obj, unsigned int shndx)
109 {
110 return (this->fptr_section_id_.find(Section_id(obj, shndx))
111 != this->fptr_section_id_.end());
112 }
113
114 // Records that a pointer of the function defined in this section
115 // is taken.
116 void
set_section_has_function_pointers(Relobj * obj,unsigned int shndx)117 set_section_has_function_pointers(Relobj* obj, unsigned int shndx)
118 {
119 this->fptr_section_id_.insert(Section_id(obj, shndx));
120 }
121
122 // Checks if the section_name should be searched for relocs
123 // corresponding to taken function pointers. Ignores eh_frame
124 // and vtable sections.
125 inline bool
check_section_for_function_pointers(const std::string & section_name,Target * target)126 check_section_for_function_pointers(const std::string& section_name,
127 Target* target)
128 {
129 return (parameters->options().icf_safe_folding()
130 && target->can_check_for_function_pointers()
131 && target->section_may_have_icf_unsafe_pointers(
132 section_name.c_str()));
133 }
134
135 // Returns a map of a section to info (Reloc_info) about its relocations.
136 Reloc_info_list&
reloc_info_list()137 reloc_info_list()
138 { return this->reloc_info_list_; }
139
140 // Returns a mapping of each section to a unique integer.
141 Uniq_secn_id_map&
section_to_int_map()142 section_to_int_map()
143 { return this->section_id_; }
144
145 private:
146
147 // Maps integers to sections.
148 std::vector<Section_id> id_section_;
149 // Does the reverse.
150 Uniq_secn_id_map section_id_;
151 // Given a section id, this maps it to the id of the kept
152 // section. If the id's are the same then this section is
153 // not folded.
154 std::vector<unsigned int> kept_section_id_;
155 // Given a section id, this says if the pointer to this
156 // function is taken in which case it is dangerous to fold
157 // this function.
158 Secn_fptr_taken_set fptr_section_id_;
159 // Flag to indicate if ICF has been run.
160 bool icf_ready_;
161 // This list is populated by gc_process_relocs in gc.h.
162 Reloc_info_list reloc_info_list_;
163 };
164
165 // This function returns true if this section corresponds to a function that
166 // should be considered by icf as a possible candidate for folding. Some
167 // earlier gcc versions, like 4.0.3, put constructors and destructors in
168 // .gnu.linkonce.t sections and hence should be included too.
169 inline bool
is_section_foldable_candidate(const std::string & section_name)170 is_section_foldable_candidate(const std::string& section_name)
171 {
172 const char* section_name_cstr = section_name.c_str();
173 return (is_prefix_of(".text", section_name_cstr)
174 || is_prefix_of(".gnu.linkonce.t", section_name_cstr));
175 }
176
177 } // End of namespace gold.
178
179 #endif
180