Tue, 12 Feb 2019 11:58:44 +0100
8211926: Catastrophic size_t underflow in BitMap::*_large methods
Reviewed-by: kbarrett, stuefe
1 /*
2 * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 *
23 */
25 #ifndef SHARE_VM_UTILITIES_BITMAP_HPP
26 #define SHARE_VM_UTILITIES_BITMAP_HPP
28 #include "memory/allocation.hpp"
29 #include "utilities/top.hpp"
31 // Forward decl;
32 class BitMapClosure;
34 // Operations for bitmaps represented as arrays of unsigned integers.
35 // Bit offsets are numbered from 0 to size-1.
37 class BitMap VALUE_OBJ_CLASS_SPEC {
38 friend class BitMap2D;
40 public:
41 typedef size_t idx_t; // Type used for bit and word indices.
42 typedef uintptr_t bm_word_t; // Element type of array that represents
43 // the bitmap.
45 // Hints for range sizes.
46 typedef enum {
47 unknown_range, small_range, large_range
48 } RangeSizeHint;
50 private:
51 ArrayAllocator<bm_word_t, mtInternal> _map_allocator;
52 bm_word_t* _map; // First word in bitmap
53 idx_t _size; // Size of bitmap (in bits)
55 // Puts the given value at the given offset, using resize() to size
56 // the bitmap appropriately if needed using factor-of-two expansion.
57 void at_put_grow(idx_t index, bool value);
59 // Threshold for performing small range operation, even when large range
60 // operation was requested. Measured in words.
61 static const size_t small_range_words = 32;
63 protected:
64 // Return the position of bit within the word that contains it (e.g., if
65 // bitmap words are 32 bits, return a number 0 <= n <= 31).
66 static idx_t bit_in_word(idx_t bit) { return bit & (BitsPerWord - 1); }
68 // Return a mask that will select the specified bit, when applied to the word
69 // containing the bit.
70 static bm_word_t bit_mask(idx_t bit) { return (bm_word_t)1 << bit_in_word(bit); }
72 // Return the index of the word containing the specified bit.
73 static idx_t word_index(idx_t bit) { return bit >> LogBitsPerWord; }
75 // Return the bit number of the first bit in the specified word.
76 static idx_t bit_index(idx_t word) { return word << LogBitsPerWord; }
78 // Return the array of bitmap words, or a specific word from it.
79 bm_word_t* map() const { return _map; }
80 bm_word_t map(idx_t word) const { return _map[word]; }
82 // Return a pointer to the word containing the specified bit.
83 bm_word_t* word_addr(idx_t bit) const { return map() + word_index(bit); }
85 // Set a word to a specified value or to all ones; clear a word.
86 void set_word (idx_t word, bm_word_t val) { _map[word] = val; }
87 void set_word (idx_t word) { set_word(word, ~(uintptr_t)0); }
88 void clear_word(idx_t word) { _map[word] = 0; }
90 // Utilities for ranges of bits. Ranges are half-open [beg, end).
92 // Ranges within a single word.
93 bm_word_t inverted_bit_mask_for_range(idx_t beg, idx_t end) const;
94 void set_range_within_word (idx_t beg, idx_t end);
95 void clear_range_within_word (idx_t beg, idx_t end);
96 void par_put_range_within_word (idx_t beg, idx_t end, bool value);
98 // Ranges spanning entire words.
99 void set_range_of_words (idx_t beg, idx_t end);
100 void clear_range_of_words (idx_t beg, idx_t end);
101 void set_large_range_of_words (idx_t beg, idx_t end);
102 void clear_large_range_of_words (idx_t beg, idx_t end);
104 static bool is_small_range_of_words(idx_t beg_full_word, idx_t end_full_word);
106 // The index of the first full word in a range.
107 idx_t word_index_round_up(idx_t bit) const;
109 // Verification.
110 inline void verify_index(idx_t index) const NOT_DEBUG_RETURN;
111 inline void verify_range(idx_t beg_index, idx_t end_index) const
112 NOT_DEBUG_RETURN;
114 // Statistics.
115 static idx_t* _pop_count_table;
116 static void init_pop_count_table();
117 static idx_t num_set_bits(bm_word_t w);
118 static idx_t num_set_bits_from_table(unsigned char c);
120 public:
122 // Constructs a bitmap with no map, and size 0.
123 BitMap() : _map(NULL), _size(0), _map_allocator(false) {}
125 // Constructs a bitmap with the given map and size.
126 BitMap(bm_word_t* map, idx_t size_in_bits);
128 // Constructs an empty bitmap of the given size (that is, this clears the
129 // new bitmap). Allocates the map array in resource area if
130 // "in_resource_area" is true, else in the C heap.
131 BitMap(idx_t size_in_bits, bool in_resource_area = true);
133 // Set the map and size.
134 void set_map(bm_word_t* map) { _map = map; }
135 void set_size(idx_t size_in_bits) { _size = size_in_bits; }
137 // Allocates necessary data structure, either in the resource area
138 // or in the C heap, as indicated by "in_resource_area."
139 // Preserves state currently in bit map by copying data.
140 // Zeros any newly-addressable bits.
141 // If "in_resource_area" is false, frees the current map.
142 // (Note that this assumes that all calls to "resize" on the same BitMap
143 // use the same value for "in_resource_area".)
144 void resize(idx_t size_in_bits, bool in_resource_area = true);
146 // Accessing
147 idx_t size() const { return _size; }
148 idx_t size_in_words() const {
149 return word_index(size() + BitsPerWord - 1);
150 }
152 bool at(idx_t index) const {
153 verify_index(index);
154 return (*word_addr(index) & bit_mask(index)) != 0;
155 }
157 // Align bit index up or down to the next bitmap word boundary, or check
158 // alignment.
159 static idx_t word_align_up(idx_t bit) {
160 return align_size_up(bit, BitsPerWord);
161 }
162 static idx_t word_align_down(idx_t bit) {
163 return align_size_down(bit, BitsPerWord);
164 }
165 static bool is_word_aligned(idx_t bit) {
166 return word_align_up(bit) == bit;
167 }
169 // Set or clear the specified bit.
170 inline void set_bit(idx_t bit);
171 inline void clear_bit(idx_t bit);
173 // Atomically set or clear the specified bit.
174 inline bool par_set_bit(idx_t bit);
175 inline bool par_clear_bit(idx_t bit);
177 // Put the given value at the given offset. The parallel version
178 // will CAS the value into the bitmap and is quite a bit slower.
179 // The parallel version also returns a value indicating if the
180 // calling thread was the one that changed the value of the bit.
181 void at_put(idx_t index, bool value);
182 bool par_at_put(idx_t index, bool value);
184 // Update a range of bits. Ranges are half-open [beg, end).
185 void set_range (idx_t beg, idx_t end);
186 void clear_range (idx_t beg, idx_t end);
187 void set_large_range (idx_t beg, idx_t end);
188 void clear_large_range (idx_t beg, idx_t end);
189 void at_put_range(idx_t beg, idx_t end, bool value);
190 void par_at_put_range(idx_t beg, idx_t end, bool value);
191 void at_put_large_range(idx_t beg, idx_t end, bool value);
192 void par_at_put_large_range(idx_t beg, idx_t end, bool value);
194 // Update a range of bits, using a hint about the size. Currently only
195 // inlines the predominant case of a 1-bit range. Works best when hint is a
196 // compile-time constant.
197 void set_range(idx_t beg, idx_t end, RangeSizeHint hint);
198 void clear_range(idx_t beg, idx_t end, RangeSizeHint hint);
199 void par_set_range(idx_t beg, idx_t end, RangeSizeHint hint);
200 void par_clear_range (idx_t beg, idx_t end, RangeSizeHint hint);
202 // Clearing
203 void clear_large();
204 inline void clear();
206 // Iteration support. Returns "true" if the iteration completed, false
207 // if the iteration terminated early (because the closure "blk" returned
208 // false).
209 bool iterate(BitMapClosure* blk, idx_t leftIndex, idx_t rightIndex);
210 bool iterate(BitMapClosure* blk) {
211 // call the version that takes an interval
212 return iterate(blk, 0, size());
213 }
215 // Looking for 1's and 0's at indices equal to or greater than "l_index",
216 // stopping if none has been found before "r_index", and returning
217 // "r_index" (which must be at most "size") in that case.
218 idx_t get_next_one_offset_inline (idx_t l_index, idx_t r_index) const;
219 idx_t get_next_zero_offset_inline(idx_t l_index, idx_t r_index) const;
221 // Like "get_next_one_offset_inline", except requires that "r_index" is
222 // aligned to bitsizeof(bm_word_t).
223 idx_t get_next_one_offset_inline_aligned_right(idx_t l_index,
224 idx_t r_index) const;
226 // Non-inline versionsof the above.
227 idx_t get_next_one_offset (idx_t l_index, idx_t r_index) const;
228 idx_t get_next_zero_offset(idx_t l_index, idx_t r_index) const;
230 idx_t get_next_one_offset(idx_t offset) const {
231 return get_next_one_offset(offset, size());
232 }
233 idx_t get_next_zero_offset(idx_t offset) const {
234 return get_next_zero_offset(offset, size());
235 }
237 // Returns the number of bits set in the bitmap.
238 idx_t count_one_bits() const;
240 // Set operations.
241 void set_union(BitMap bits);
242 void set_difference(BitMap bits);
243 void set_intersection(BitMap bits);
244 // Returns true iff "this" is a superset of "bits".
245 bool contains(const BitMap bits) const;
246 // Returns true iff "this and "bits" have a non-empty intersection.
247 bool intersects(const BitMap bits) const;
249 // Returns result of whether this map changed
250 // during the operation
251 bool set_union_with_result(BitMap bits);
252 bool set_difference_with_result(BitMap bits);
253 bool set_intersection_with_result(BitMap bits);
255 // Requires the submap of "bits" starting at offset to be at least as
256 // large as "this". Modifies "this" to be the intersection of its
257 // current contents and the submap of "bits" starting at "offset" of the
258 // same length as "this."
259 // (For expedience, currently requires the offset to be aligned to the
260 // bitsize of a uintptr_t. This should go away in the future though it
261 // will probably remain a good case to optimize.)
262 void set_intersection_at_offset(BitMap bits, idx_t offset);
264 void set_from(BitMap bits);
266 bool is_same(BitMap bits);
268 // Test if all bits are set or cleared
269 bool is_full() const;
270 bool is_empty() const;
272 void print_on_error(outputStream* st, const char* prefix) const;
274 #ifndef PRODUCT
275 public:
276 // Printing
277 void print_on(outputStream* st) const;
278 #endif
279 };
281 // Convenience class wrapping BitMap which provides multiple bits per slot.
282 class BitMap2D VALUE_OBJ_CLASS_SPEC {
283 public:
284 typedef BitMap::idx_t idx_t; // Type used for bit and word indices.
285 typedef BitMap::bm_word_t bm_word_t; // Element type of array that
286 // represents the bitmap.
287 private:
288 BitMap _map;
289 idx_t _bits_per_slot;
291 idx_t bit_index(idx_t slot_index, idx_t bit_within_slot_index) const {
292 return slot_index * _bits_per_slot + bit_within_slot_index;
293 }
295 void verify_bit_within_slot_index(idx_t index) const {
296 assert(index < _bits_per_slot, "bit_within_slot index out of bounds");
297 }
299 public:
300 // Construction. bits_per_slot must be greater than 0.
301 BitMap2D(bm_word_t* map, idx_t size_in_slots, idx_t bits_per_slot);
303 // Allocates necessary data structure in resource area. bits_per_slot must be greater than 0.
304 BitMap2D(idx_t size_in_slots, idx_t bits_per_slot);
306 idx_t size_in_bits() {
307 return _map.size();
308 }
310 // Returns number of full slots that have been allocated
311 idx_t size_in_slots() {
312 // Round down
313 return _map.size() / _bits_per_slot;
314 }
316 bool is_valid_index(idx_t slot_index, idx_t bit_within_slot_index) {
317 verify_bit_within_slot_index(bit_within_slot_index);
318 return (bit_index(slot_index, bit_within_slot_index) < size_in_bits());
319 }
321 bool at(idx_t slot_index, idx_t bit_within_slot_index) const {
322 verify_bit_within_slot_index(bit_within_slot_index);
323 return _map.at(bit_index(slot_index, bit_within_slot_index));
324 }
326 void set_bit(idx_t slot_index, idx_t bit_within_slot_index) {
327 verify_bit_within_slot_index(bit_within_slot_index);
328 _map.set_bit(bit_index(slot_index, bit_within_slot_index));
329 }
331 void clear_bit(idx_t slot_index, idx_t bit_within_slot_index) {
332 verify_bit_within_slot_index(bit_within_slot_index);
333 _map.clear_bit(bit_index(slot_index, bit_within_slot_index));
334 }
336 void at_put(idx_t slot_index, idx_t bit_within_slot_index, bool value) {
337 verify_bit_within_slot_index(bit_within_slot_index);
338 _map.at_put(bit_index(slot_index, bit_within_slot_index), value);
339 }
341 void at_put_grow(idx_t slot_index, idx_t bit_within_slot_index, bool value) {
342 verify_bit_within_slot_index(bit_within_slot_index);
343 _map.at_put_grow(bit_index(slot_index, bit_within_slot_index), value);
344 }
346 void clear();
347 };
349 // Closure for iterating over BitMaps
351 class BitMapClosure VALUE_OBJ_CLASS_SPEC {
352 public:
353 // Callback when bit in map is set. Should normally return "true";
354 // return of false indicates that the bitmap iteration should terminate.
355 virtual bool do_bit(BitMap::idx_t offset) = 0;
356 };
358 #endif // SHARE_VM_UTILITIES_BITMAP_HPP