Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
14 | pmbaty | 1 | //===- Endian.h - Utilities for IO with endian specific data ----*- C++ -*-===// |
2 | // |
||
3 | // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
||
4 | // See https://llvm.org/LICENSE.txt for license information. |
||
5 | // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
||
6 | // |
||
7 | //===----------------------------------------------------------------------===// |
||
8 | // |
||
9 | // This file declares generic functions to read and write endian specific data. |
||
10 | // |
||
11 | //===----------------------------------------------------------------------===// |
||
12 | |||
13 | #ifndef LLVM_SUPPORT_ENDIAN_H |
||
14 | #define LLVM_SUPPORT_ENDIAN_H |
||
15 | |||
16 | #include "llvm/Support/Compiler.h" |
||
17 | #include "llvm/Support/SwapByteOrder.h" |
||
18 | #include <cassert> |
||
19 | #include <cstddef> |
||
20 | #include <cstdint> |
||
21 | #include <cstring> |
||
22 | #include <type_traits> |
||
23 | |||
24 | namespace llvm { |
||
25 | namespace support { |
||
26 | |||
27 | enum endianness {big, little, native}; |
||
28 | |||
29 | // These are named values for common alignments. |
||
30 | enum {aligned = 0, unaligned = 1}; |
||
31 | |||
32 | namespace detail { |
||
33 | |||
34 | /// ::value is either alignment, or alignof(T) if alignment is 0. |
||
35 | template<class T, int alignment> |
||
36 | struct PickAlignment { |
||
37 | enum { value = alignment == 0 ? alignof(T) : alignment }; |
||
38 | }; |
||
39 | |||
40 | } // end namespace detail |
||
41 | |||
42 | namespace endian { |
||
43 | |||
44 | constexpr endianness system_endianness() { |
||
45 | return sys::IsBigEndianHost ? big : little; |
||
46 | } |
||
47 | |||
48 | template <typename value_type> |
||
49 | inline value_type byte_swap(value_type value, endianness endian) { |
||
50 | if ((endian != native) && (endian != system_endianness())) |
||
51 | sys::swapByteOrder(value); |
||
52 | return value; |
||
53 | } |
||
54 | |||
55 | /// Swap the bytes of value to match the given endianness. |
||
56 | template<typename value_type, endianness endian> |
||
57 | inline value_type byte_swap(value_type value) { |
||
58 | return byte_swap(value, endian); |
||
59 | } |
||
60 | |||
61 | /// Read a value of a particular endianness from memory. |
||
62 | template <typename value_type, std::size_t alignment> |
||
63 | inline value_type read(const void *memory, endianness endian) { |
||
64 | value_type ret; |
||
65 | |||
66 | memcpy(&ret, |
||
67 | LLVM_ASSUME_ALIGNED( |
||
68 | memory, (detail::PickAlignment<value_type, alignment>::value)), |
||
69 | sizeof(value_type)); |
||
70 | return byte_swap<value_type>(ret, endian); |
||
71 | } |
||
72 | |||
73 | template<typename value_type, |
||
74 | endianness endian, |
||
75 | std::size_t alignment> |
||
76 | inline value_type read(const void *memory) { |
||
77 | return read<value_type, alignment>(memory, endian); |
||
78 | } |
||
79 | |||
80 | /// Read a value of a particular endianness from a buffer, and increment the |
||
81 | /// buffer past that value. |
||
82 | template <typename value_type, std::size_t alignment, typename CharT> |
||
83 | inline value_type readNext(const CharT *&memory, endianness endian) { |
||
84 | value_type ret = read<value_type, alignment>(memory, endian); |
||
85 | memory += sizeof(value_type); |
||
86 | return ret; |
||
87 | } |
||
88 | |||
89 | template<typename value_type, endianness endian, std::size_t alignment, |
||
90 | typename CharT> |
||
91 | inline value_type readNext(const CharT *&memory) { |
||
92 | return readNext<value_type, alignment, CharT>(memory, endian); |
||
93 | } |
||
94 | |||
95 | /// Write a value to memory with a particular endianness. |
||
96 | template <typename value_type, std::size_t alignment> |
||
97 | inline void write(void *memory, value_type value, endianness endian) { |
||
98 | value = byte_swap<value_type>(value, endian); |
||
99 | memcpy(LLVM_ASSUME_ALIGNED( |
||
100 | memory, (detail::PickAlignment<value_type, alignment>::value)), |
||
101 | &value, sizeof(value_type)); |
||
102 | } |
||
103 | |||
104 | template<typename value_type, |
||
105 | endianness endian, |
||
106 | std::size_t alignment> |
||
107 | inline void write(void *memory, value_type value) { |
||
108 | write<value_type, alignment>(memory, value, endian); |
||
109 | } |
||
110 | |||
111 | template <typename value_type> |
||
112 | using make_unsigned_t = std::make_unsigned_t<value_type>; |
||
113 | |||
114 | /// Read a value of a particular endianness from memory, for a location |
||
115 | /// that starts at the given bit offset within the first byte. |
||
116 | template <typename value_type, endianness endian, std::size_t alignment> |
||
117 | inline value_type readAtBitAlignment(const void *memory, uint64_t startBit) { |
||
118 | assert(startBit < 8); |
||
119 | if (startBit == 0) |
||
120 | return read<value_type, endian, alignment>(memory); |
||
121 | else { |
||
122 | // Read two values and compose the result from them. |
||
123 | value_type val[2]; |
||
124 | memcpy(&val[0], |
||
125 | LLVM_ASSUME_ALIGNED( |
||
126 | memory, (detail::PickAlignment<value_type, alignment>::value)), |
||
127 | sizeof(value_type) * 2); |
||
128 | val[0] = byte_swap<value_type, endian>(val[0]); |
||
129 | val[1] = byte_swap<value_type, endian>(val[1]); |
||
130 | |||
131 | // Shift bits from the lower value into place. |
||
132 | make_unsigned_t<value_type> lowerVal = val[0] >> startBit; |
||
133 | // Mask off upper bits after right shift in case of signed type. |
||
134 | make_unsigned_t<value_type> numBitsFirstVal = |
||
135 | (sizeof(value_type) * 8) - startBit; |
||
136 | lowerVal &= ((make_unsigned_t<value_type>)1 << numBitsFirstVal) - 1; |
||
137 | |||
138 | // Get the bits from the upper value. |
||
139 | make_unsigned_t<value_type> upperVal = |
||
140 | val[1] & (((make_unsigned_t<value_type>)1 << startBit) - 1); |
||
141 | // Shift them in to place. |
||
142 | upperVal <<= numBitsFirstVal; |
||
143 | |||
144 | return lowerVal | upperVal; |
||
145 | } |
||
146 | } |
||
147 | |||
148 | /// Write a value to memory with a particular endianness, for a location |
||
149 | /// that starts at the given bit offset within the first byte. |
||
150 | template <typename value_type, endianness endian, std::size_t alignment> |
||
151 | inline void writeAtBitAlignment(void *memory, value_type value, |
||
152 | uint64_t startBit) { |
||
153 | assert(startBit < 8); |
||
154 | if (startBit == 0) |
||
155 | write<value_type, endian, alignment>(memory, value); |
||
156 | else { |
||
157 | // Read two values and shift the result into them. |
||
158 | value_type val[2]; |
||
159 | memcpy(&val[0], |
||
160 | LLVM_ASSUME_ALIGNED( |
||
161 | memory, (detail::PickAlignment<value_type, alignment>::value)), |
||
162 | sizeof(value_type) * 2); |
||
163 | val[0] = byte_swap<value_type, endian>(val[0]); |
||
164 | val[1] = byte_swap<value_type, endian>(val[1]); |
||
165 | |||
166 | // Mask off any existing bits in the upper part of the lower value that |
||
167 | // we want to replace. |
||
168 | val[0] &= ((make_unsigned_t<value_type>)1 << startBit) - 1; |
||
169 | make_unsigned_t<value_type> numBitsFirstVal = |
||
170 | (sizeof(value_type) * 8) - startBit; |
||
171 | make_unsigned_t<value_type> lowerVal = value; |
||
172 | if (startBit > 0) { |
||
173 | // Mask off the upper bits in the new value that are not going to go into |
||
174 | // the lower value. This avoids a left shift of a negative value, which |
||
175 | // is undefined behavior. |
||
176 | lowerVal &= (((make_unsigned_t<value_type>)1 << numBitsFirstVal) - 1); |
||
177 | // Now shift the new bits into place |
||
178 | lowerVal <<= startBit; |
||
179 | } |
||
180 | val[0] |= lowerVal; |
||
181 | |||
182 | // Mask off any existing bits in the lower part of the upper value that |
||
183 | // we want to replace. |
||
184 | val[1] &= ~(((make_unsigned_t<value_type>)1 << startBit) - 1); |
||
185 | // Next shift the bits that go into the upper value into position. |
||
186 | make_unsigned_t<value_type> upperVal = value >> numBitsFirstVal; |
||
187 | // Mask off upper bits after right shift in case of signed type. |
||
188 | upperVal &= ((make_unsigned_t<value_type>)1 << startBit) - 1; |
||
189 | val[1] |= upperVal; |
||
190 | |||
191 | // Finally, rewrite values. |
||
192 | val[0] = byte_swap<value_type, endian>(val[0]); |
||
193 | val[1] = byte_swap<value_type, endian>(val[1]); |
||
194 | memcpy(LLVM_ASSUME_ALIGNED( |
||
195 | memory, (detail::PickAlignment<value_type, alignment>::value)), |
||
196 | &val[0], sizeof(value_type) * 2); |
||
197 | } |
||
198 | } |
||
199 | |||
200 | } // end namespace endian |
||
201 | |||
202 | namespace detail { |
||
203 | |||
204 | template <typename ValueType, endianness Endian, std::size_t Alignment, |
||
205 | std::size_t ALIGN = PickAlignment<ValueType, Alignment>::value> |
||
206 | struct packed_endian_specific_integral { |
||
207 | using value_type = ValueType; |
||
208 | static constexpr endianness endian = Endian; |
||
209 | static constexpr std::size_t alignment = Alignment; |
||
210 | |||
211 | packed_endian_specific_integral() = default; |
||
212 | |||
213 | explicit packed_endian_specific_integral(value_type val) { *this = val; } |
||
214 | |||
215 | operator value_type() const { |
||
216 | return endian::read<value_type, endian, alignment>( |
||
217 | (const void*)Value.buffer); |
||
218 | } |
||
219 | |||
220 | void operator=(value_type newValue) { |
||
221 | endian::write<value_type, endian, alignment>( |
||
222 | (void*)Value.buffer, newValue); |
||
223 | } |
||
224 | |||
225 | packed_endian_specific_integral &operator+=(value_type newValue) { |
||
226 | *this = *this + newValue; |
||
227 | return *this; |
||
228 | } |
||
229 | |||
230 | packed_endian_specific_integral &operator-=(value_type newValue) { |
||
231 | *this = *this - newValue; |
||
232 | return *this; |
||
233 | } |
||
234 | |||
235 | packed_endian_specific_integral &operator|=(value_type newValue) { |
||
236 | *this = *this | newValue; |
||
237 | return *this; |
||
238 | } |
||
239 | |||
240 | packed_endian_specific_integral &operator&=(value_type newValue) { |
||
241 | *this = *this & newValue; |
||
242 | return *this; |
||
243 | } |
||
244 | |||
245 | private: |
||
246 | struct { |
||
247 | alignas(ALIGN) char buffer[sizeof(value_type)]; |
||
248 | } Value; |
||
249 | |||
250 | public: |
||
251 | struct ref { |
||
252 | explicit ref(void *Ptr) : Ptr(Ptr) {} |
||
253 | |||
254 | operator value_type() const { |
||
255 | return endian::read<value_type, endian, alignment>(Ptr); |
||
256 | } |
||
257 | |||
258 | void operator=(value_type NewValue) { |
||
259 | endian::write<value_type, endian, alignment>(Ptr, NewValue); |
||
260 | } |
||
261 | |||
262 | private: |
||
263 | void *Ptr; |
||
264 | }; |
||
265 | }; |
||
266 | |||
267 | } // end namespace detail |
||
268 | |||
269 | using ulittle16_t = |
||
270 | detail::packed_endian_specific_integral<uint16_t, little, unaligned>; |
||
271 | using ulittle32_t = |
||
272 | detail::packed_endian_specific_integral<uint32_t, little, unaligned>; |
||
273 | using ulittle64_t = |
||
274 | detail::packed_endian_specific_integral<uint64_t, little, unaligned>; |
||
275 | |||
276 | using little16_t = |
||
277 | detail::packed_endian_specific_integral<int16_t, little, unaligned>; |
||
278 | using little32_t = |
||
279 | detail::packed_endian_specific_integral<int32_t, little, unaligned>; |
||
280 | using little64_t = |
||
281 | detail::packed_endian_specific_integral<int64_t, little, unaligned>; |
||
282 | |||
283 | using aligned_ulittle16_t = |
||
284 | detail::packed_endian_specific_integral<uint16_t, little, aligned>; |
||
285 | using aligned_ulittle32_t = |
||
286 | detail::packed_endian_specific_integral<uint32_t, little, aligned>; |
||
287 | using aligned_ulittle64_t = |
||
288 | detail::packed_endian_specific_integral<uint64_t, little, aligned>; |
||
289 | |||
290 | using aligned_little16_t = |
||
291 | detail::packed_endian_specific_integral<int16_t, little, aligned>; |
||
292 | using aligned_little32_t = |
||
293 | detail::packed_endian_specific_integral<int32_t, little, aligned>; |
||
294 | using aligned_little64_t = |
||
295 | detail::packed_endian_specific_integral<int64_t, little, aligned>; |
||
296 | |||
297 | using ubig16_t = |
||
298 | detail::packed_endian_specific_integral<uint16_t, big, unaligned>; |
||
299 | using ubig32_t = |
||
300 | detail::packed_endian_specific_integral<uint32_t, big, unaligned>; |
||
301 | using ubig64_t = |
||
302 | detail::packed_endian_specific_integral<uint64_t, big, unaligned>; |
||
303 | |||
304 | using big16_t = |
||
305 | detail::packed_endian_specific_integral<int16_t, big, unaligned>; |
||
306 | using big32_t = |
||
307 | detail::packed_endian_specific_integral<int32_t, big, unaligned>; |
||
308 | using big64_t = |
||
309 | detail::packed_endian_specific_integral<int64_t, big, unaligned>; |
||
310 | |||
311 | using aligned_ubig16_t = |
||
312 | detail::packed_endian_specific_integral<uint16_t, big, aligned>; |
||
313 | using aligned_ubig32_t = |
||
314 | detail::packed_endian_specific_integral<uint32_t, big, aligned>; |
||
315 | using aligned_ubig64_t = |
||
316 | detail::packed_endian_specific_integral<uint64_t, big, aligned>; |
||
317 | |||
318 | using aligned_big16_t = |
||
319 | detail::packed_endian_specific_integral<int16_t, big, aligned>; |
||
320 | using aligned_big32_t = |
||
321 | detail::packed_endian_specific_integral<int32_t, big, aligned>; |
||
322 | using aligned_big64_t = |
||
323 | detail::packed_endian_specific_integral<int64_t, big, aligned>; |
||
324 | |||
325 | using unaligned_uint16_t = |
||
326 | detail::packed_endian_specific_integral<uint16_t, native, unaligned>; |
||
327 | using unaligned_uint32_t = |
||
328 | detail::packed_endian_specific_integral<uint32_t, native, unaligned>; |
||
329 | using unaligned_uint64_t = |
||
330 | detail::packed_endian_specific_integral<uint64_t, native, unaligned>; |
||
331 | |||
332 | using unaligned_int16_t = |
||
333 | detail::packed_endian_specific_integral<int16_t, native, unaligned>; |
||
334 | using unaligned_int32_t = |
||
335 | detail::packed_endian_specific_integral<int32_t, native, unaligned>; |
||
336 | using unaligned_int64_t = |
||
337 | detail::packed_endian_specific_integral<int64_t, native, unaligned>; |
||
338 | |||
339 | template <typename T> |
||
340 | using little_t = detail::packed_endian_specific_integral<T, little, unaligned>; |
||
341 | template <typename T> |
||
342 | using big_t = detail::packed_endian_specific_integral<T, big, unaligned>; |
||
343 | |||
344 | template <typename T> |
||
345 | using aligned_little_t = |
||
346 | detail::packed_endian_specific_integral<T, little, aligned>; |
||
347 | template <typename T> |
||
348 | using aligned_big_t = detail::packed_endian_specific_integral<T, big, aligned>; |
||
349 | |||
350 | namespace endian { |
||
351 | |||
352 | template <typename T> inline T read(const void *P, endianness E) { |
||
353 | return read<T, unaligned>(P, E); |
||
354 | } |
||
355 | |||
356 | template <typename T, endianness E> inline T read(const void *P) { |
||
357 | return *(const detail::packed_endian_specific_integral<T, E, unaligned> *)P; |
||
358 | } |
||
359 | |||
360 | inline uint16_t read16(const void *P, endianness E) { |
||
361 | return read<uint16_t>(P, E); |
||
362 | } |
||
363 | inline uint32_t read32(const void *P, endianness E) { |
||
364 | return read<uint32_t>(P, E); |
||
365 | } |
||
366 | inline uint64_t read64(const void *P, endianness E) { |
||
367 | return read<uint64_t>(P, E); |
||
368 | } |
||
369 | |||
370 | template <endianness E> inline uint16_t read16(const void *P) { |
||
371 | return read<uint16_t, E>(P); |
||
372 | } |
||
373 | template <endianness E> inline uint32_t read32(const void *P) { |
||
374 | return read<uint32_t, E>(P); |
||
375 | } |
||
376 | template <endianness E> inline uint64_t read64(const void *P) { |
||
377 | return read<uint64_t, E>(P); |
||
378 | } |
||
379 | |||
380 | inline uint16_t read16le(const void *P) { return read16<little>(P); } |
||
381 | inline uint32_t read32le(const void *P) { return read32<little>(P); } |
||
382 | inline uint64_t read64le(const void *P) { return read64<little>(P); } |
||
383 | inline uint16_t read16be(const void *P) { return read16<big>(P); } |
||
384 | inline uint32_t read32be(const void *P) { return read32<big>(P); } |
||
385 | inline uint64_t read64be(const void *P) { return read64<big>(P); } |
||
386 | |||
387 | template <typename T> inline void write(void *P, T V, endianness E) { |
||
388 | write<T, unaligned>(P, V, E); |
||
389 | } |
||
390 | |||
391 | template <typename T, endianness E> inline void write(void *P, T V) { |
||
392 | *(detail::packed_endian_specific_integral<T, E, unaligned> *)P = V; |
||
393 | } |
||
394 | |||
395 | inline void write16(void *P, uint16_t V, endianness E) { |
||
396 | write<uint16_t>(P, V, E); |
||
397 | } |
||
398 | inline void write32(void *P, uint32_t V, endianness E) { |
||
399 | write<uint32_t>(P, V, E); |
||
400 | } |
||
401 | inline void write64(void *P, uint64_t V, endianness E) { |
||
402 | write<uint64_t>(P, V, E); |
||
403 | } |
||
404 | |||
405 | template <endianness E> inline void write16(void *P, uint16_t V) { |
||
406 | write<uint16_t, E>(P, V); |
||
407 | } |
||
408 | template <endianness E> inline void write32(void *P, uint32_t V) { |
||
409 | write<uint32_t, E>(P, V); |
||
410 | } |
||
411 | template <endianness E> inline void write64(void *P, uint64_t V) { |
||
412 | write<uint64_t, E>(P, V); |
||
413 | } |
||
414 | |||
415 | inline void write16le(void *P, uint16_t V) { write16<little>(P, V); } |
||
416 | inline void write32le(void *P, uint32_t V) { write32<little>(P, V); } |
||
417 | inline void write64le(void *P, uint64_t V) { write64<little>(P, V); } |
||
418 | inline void write16be(void *P, uint16_t V) { write16<big>(P, V); } |
||
419 | inline void write32be(void *P, uint32_t V) { write32<big>(P, V); } |
||
420 | inline void write64be(void *P, uint64_t V) { write64<big>(P, V); } |
||
421 | |||
422 | } // end namespace endian |
||
423 | |||
424 | } // end namespace support |
||
425 | } // end namespace llvm |
||
426 | |||
427 | #endif // LLVM_SUPPORT_ENDIAN_H |