1 #ifndef _LTTNG_BITFIELD_H
2 #define _LTTNG_BITFIELD_H
7 * Bitfields read/write functions.
9 * Copyright 2010 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
11 * Permission is hereby granted, free of charge, to any person obtaining a copy
12 * of this software and associated documentation files (the "Software"), to
13 * deal in the Software without restriction, including without limitation the
14 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
15 * sell copies of the Software, and to permit persons to whom the Software is
16 * furnished to do so, subject to the following conditions:
18 * The above copyright notice and this permission notice shall be included in
19 * all copies or substantial portions of the Software.
21 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
22 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
23 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
24 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
25 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
26 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
31 #include <stdint.h> /* C99 5.2.4.2 Numerical limits */
32 #include <limits.h> /* C99 5.2.4.2 Numerical limits */
34 #include <common/compat/endian.h> /* Non-standard BIG_ENDIAN, LITTLE_ENDIAN, BYTE_ORDER */
36 /* We can't shift a int from 32 bit, >> 32 and << 32 on int is undefined */
37 #define _piecewise_rshift(_v, _shift) \
39 typeof(_v) ___v = (_v); \
40 typeof(_shift) ___shift = (_shift); \
41 unsigned long sb = (___shift) / (sizeof(___v) * CHAR_BIT - 1); \
42 unsigned long final = (___shift) % (sizeof(___v) * CHAR_BIT - 1); \
45 ___v >>= sizeof(___v) * CHAR_BIT - 1; \
49 #define _piecewise_lshift(_v, _shift) \
51 typeof(_v) ___v = (_v); \
52 typeof(_shift) ___shift = (_shift); \
53 unsigned long sb = (___shift) / (sizeof(___v) * CHAR_BIT - 1); \
54 unsigned long final = (___shift) % (sizeof(___v) * CHAR_BIT - 1); \
57 ___v <<= sizeof(___v) * CHAR_BIT - 1; \
61 #define _is_signed_type(type) ((type) -1 < (type) 0)
63 #define _unsigned_cast(type, v) \
65 (sizeof(v) < sizeof(type)) ? \
66 ((type) (v)) & (~(~(type) 0 << (sizeof(v) * CHAR_BIT))) : \
71 * bitfield_write - write integer to a bitfield in native endianness
73 * Save integer to the bitfield, which starts at the "start" bit, has "len"
75 * The inside of a bitfield is from high bits to low bits.
76 * Uses native endianness.
77 * For unsigned "v", pad MSB with 0 if bitfield is larger than v.
78 * For signed "v", sign-extend v if bitfield is larger than v.
80 * On little endian, bytes are placed from the less significant to the most
81 * significant. Also, consecutive bitfields are placed from lower bits to higher
84 * On big endian, bytes are places from most significant to less significant.
85 * Also, consecutive bitfields are placed from higher to lower bits.
88 #define _bitfield_write_le(_ptr, type, _start, _length, _v) \
90 typeof(_v) __v = (_v); \
91 type *__ptr = (void *) (_ptr); \
92 unsigned long __start = (_start), __length = (_length); \
94 unsigned long ts = sizeof(type) * CHAR_BIT; /* type size */ \
95 unsigned long start_unit, end_unit, this_unit; \
96 unsigned long end, cshift; /* cshift is "complement shift" */ \
101 end = __start + __length; \
102 start_unit = __start / ts; \
103 end_unit = (end + (ts - 1)) / ts; \
105 /* Trim v high bits */ \
106 if (__length < sizeof(__v) * CHAR_BIT) \
107 __v &= ~((~(typeof(__v)) 0) << __length); \
109 /* We can now append v with a simple "or", shift it piece-wise */ \
110 this_unit = start_unit; \
111 if (start_unit == end_unit - 1) { \
112 mask = ~((~(type) 0) << (__start % ts)); \
114 mask |= (~(type) 0) << (end % ts); \
115 cmask = (type) __v << (__start % ts); \
117 __ptr[this_unit] &= mask; \
118 __ptr[this_unit] |= cmask; \
121 if (__start % ts) { \
122 cshift = __start % ts; \
123 mask = ~((~(type) 0) << cshift); \
124 cmask = (type) __v << cshift; \
126 __ptr[this_unit] &= mask; \
127 __ptr[this_unit] |= cmask; \
128 __v = _piecewise_rshift(__v, ts - cshift); \
129 __start += ts - cshift; \
132 for (; this_unit < end_unit - 1; this_unit++) { \
133 __ptr[this_unit] = (type) __v; \
134 __v = _piecewise_rshift(__v, ts); \
138 mask = (~(type) 0) << (end % ts); \
139 cmask = (type) __v; \
141 __ptr[this_unit] &= mask; \
142 __ptr[this_unit] |= cmask; \
144 __ptr[this_unit] = (type) __v; \
147 #define _bitfield_write_be(_ptr, type, _start, _length, _v) \
149 typeof(_v) __v = (_v); \
150 type *__ptr = (void *) (_ptr); \
151 unsigned long __start = (_start), __length = (_length); \
153 unsigned long ts = sizeof(type) * CHAR_BIT; /* type size */ \
154 unsigned long start_unit, end_unit, this_unit; \
155 unsigned long end, cshift; /* cshift is "complement shift" */ \
160 end = __start + __length; \
161 start_unit = __start / ts; \
162 end_unit = (end + (ts - 1)) / ts; \
164 /* Trim v high bits */ \
165 if (__length < sizeof(__v) * CHAR_BIT) \
166 __v &= ~((~(typeof(__v)) 0) << __length); \
168 /* We can now append v with a simple "or", shift it piece-wise */ \
169 this_unit = end_unit - 1; \
170 if (start_unit == end_unit - 1) { \
171 mask = ~((~(type) 0) << ((ts - (end % ts)) % ts)); \
173 mask |= (~((type) 0)) << (ts - (__start % ts)); \
174 cmask = (type) __v << ((ts - (end % ts)) % ts); \
176 __ptr[this_unit] &= mask; \
177 __ptr[this_unit] |= cmask; \
182 mask = ~((~(type) 0) << (ts - cshift)); \
183 cmask = (type) __v << (ts - cshift); \
185 __ptr[this_unit] &= mask; \
186 __ptr[this_unit] |= cmask; \
187 __v = _piecewise_rshift(__v, cshift); \
191 for (; (long) this_unit >= (long) start_unit + 1; this_unit--) { \
192 __ptr[this_unit] = (type) __v; \
193 __v = _piecewise_rshift(__v, ts); \
196 if (__start % ts) { \
197 mask = (~(type) 0) << (ts - (__start % ts)); \
198 cmask = (type) __v; \
200 __ptr[this_unit] &= mask; \
201 __ptr[this_unit] |= cmask; \
203 __ptr[this_unit] = (type) __v; \
207 * bitfield_write - write integer to a bitfield in native endianness
208 * bitfield_write_le - write integer to a bitfield in little endian
209 * bitfield_write_be - write integer to a bitfield in big endian
212 #if (BYTE_ORDER == LITTLE_ENDIAN)
214 #define bitfield_write(ptr, type, _start, _length, _v) \
215 _bitfield_write_le(ptr, type, _start, _length, _v)
217 #define bitfield_write_le(ptr, type, _start, _length, _v) \
218 _bitfield_write_le(ptr, type, _start, _length, _v)
220 #define bitfield_write_be(ptr, type, _start, _length, _v) \
221 _bitfield_write_be(ptr, unsigned char, _start, _length, _v)
223 #elif (BYTE_ORDER == BIG_ENDIAN)
225 #define bitfield_write(ptr, type, _start, _length, _v) \
226 _bitfield_write_be(ptr, type, _start, _length, _v)
228 #define bitfield_write_le(ptr, type, _start, _length, _v) \
229 _bitfield_write_le(ptr, unsigned char, _start, _length, _v)
231 #define bitfield_write_be(ptr, type, _start, _length, _v) \
232 _bitfield_write_be(ptr, type, _start, _length, _v)
234 #else /* (BYTE_ORDER == PDP_ENDIAN) */
236 #error "Byte order not supported"
240 #define _bitfield_read_le(_ptr, type, _start, _length, _vptr) \
242 typeof(*(_vptr)) *__vptr = (_vptr); \
243 typeof(*__vptr) __v; \
244 type *__ptr = (void *) (_ptr); \
245 unsigned long __start = (_start), __length = (_length); \
247 unsigned long ts = sizeof(type) * CHAR_BIT; /* type size */ \
248 unsigned long start_unit, end_unit, this_unit; \
249 unsigned long end, cshift; /* cshift is "complement shift" */ \
256 end = __start + __length; \
257 start_unit = __start / ts; \
258 end_unit = (end + (ts - 1)) / ts; \
260 this_unit = end_unit - 1; \
261 if (_is_signed_type(typeof(__v)) \
262 && (__ptr[this_unit] & ((type) 1 << ((end % ts ? : ts) - 1)))) \
263 __v = ~(typeof(__v)) 0; \
266 if (start_unit == end_unit - 1) { \
267 cmask = __ptr[this_unit]; \
268 cmask >>= (__start % ts); \
269 if ((end - __start) % ts) { \
270 mask = ~((~(type) 0) << (end - __start)); \
273 __v = _piecewise_lshift(__v, end - __start); \
274 __v |= _unsigned_cast(typeof(__v), cmask); \
280 mask = ~((~(type) 0) << cshift); \
281 cmask = __ptr[this_unit]; \
283 __v = _piecewise_lshift(__v, cshift); \
284 __v |= _unsigned_cast(typeof(__v), cmask); \
288 for (; (long) this_unit >= (long) start_unit + 1; this_unit--) { \
289 __v = _piecewise_lshift(__v, ts); \
290 __v |= _unsigned_cast(typeof(__v), __ptr[this_unit]);\
293 if (__start % ts) { \
294 mask = ~((~(type) 0) << (ts - (__start % ts))); \
295 cmask = __ptr[this_unit]; \
296 cmask >>= (__start % ts); \
298 __v = _piecewise_lshift(__v, ts - (__start % ts)); \
299 __v |= _unsigned_cast(typeof(__v), cmask); \
301 __v = _piecewise_lshift(__v, ts); \
302 __v |= _unsigned_cast(typeof(__v), __ptr[this_unit]);\
307 #define _bitfield_read_be(_ptr, type, _start, _length, _vptr) \
309 typeof(*(_vptr)) *__vptr = (_vptr); \
310 typeof(*__vptr) __v; \
311 type *__ptr = (void *) (_ptr); \
312 unsigned long __start = (_start), __length = (_length); \
314 unsigned long ts = sizeof(type) * CHAR_BIT; /* type size */ \
315 unsigned long start_unit, end_unit, this_unit; \
316 unsigned long end, cshift; /* cshift is "complement shift" */ \
323 end = __start + __length; \
324 start_unit = __start / ts; \
325 end_unit = (end + (ts - 1)) / ts; \
327 this_unit = start_unit; \
328 if (_is_signed_type(typeof(__v)) \
329 && (__ptr[this_unit] & ((type) 1 << (ts - (__start % ts) - 1)))) \
330 __v = ~(typeof(__v)) 0; \
333 if (start_unit == end_unit - 1) { \
334 cmask = __ptr[this_unit]; \
335 cmask >>= (ts - (end % ts)) % ts; \
336 if ((end - __start) % ts) { \
337 mask = ~((~(type) 0) << (end - __start)); \
340 __v = _piecewise_lshift(__v, end - __start); \
341 __v |= _unsigned_cast(typeof(__v), cmask); \
345 if (__start % ts) { \
346 cshift = __start % ts; \
347 mask = ~((~(type) 0) << (ts - cshift)); \
348 cmask = __ptr[this_unit]; \
350 __v = _piecewise_lshift(__v, ts - cshift); \
351 __v |= _unsigned_cast(typeof(__v), cmask); \
352 __start += ts - cshift; \
355 for (; this_unit < end_unit - 1; this_unit++) { \
356 __v = _piecewise_lshift(__v, ts); \
357 __v |= _unsigned_cast(typeof(__v), __ptr[this_unit]);\
361 mask = ~((~(type) 0) << (end % ts)); \
362 cmask = __ptr[this_unit]; \
363 cmask >>= ts - (end % ts); \
365 __v = _piecewise_lshift(__v, end % ts); \
366 __v |= _unsigned_cast(typeof(__v), cmask); \
368 __v = _piecewise_lshift(__v, ts); \
369 __v |= _unsigned_cast(typeof(__v), __ptr[this_unit]);\
375 * bitfield_read - read integer from a bitfield in native endianness
376 * bitfield_read_le - read integer from a bitfield in little endian
377 * bitfield_read_be - read integer from a bitfield in big endian
380 #if (BYTE_ORDER == LITTLE_ENDIAN)
382 #define bitfield_read(_ptr, type, _start, _length, _vptr) \
383 _bitfield_read_le(_ptr, type, _start, _length, _vptr)
385 #define bitfield_read_le(_ptr, type, _start, _length, _vptr) \
386 _bitfield_read_le(_ptr, type, _start, _length, _vptr)
388 #define bitfield_read_be(_ptr, type, _start, _length, _vptr) \
389 _bitfield_read_be(_ptr, unsigned char, _start, _length, _vptr)
391 #elif (BYTE_ORDER == BIG_ENDIAN)
393 #define bitfield_read(_ptr, type, _start, _length, _vptr) \
394 _bitfield_read_be(_ptr, type, _start, _length, _vptr)
396 #define bitfield_read_le(_ptr, type, _start, _length, _vptr) \
397 _bitfield_read_le(_ptr, unsigned char, _start, _length, _vptr)
399 #define bitfield_read_be(_ptr, type, _start, _length, _vptr) \
400 _bitfield_read_be(_ptr, type, _start, _length, _vptr)
402 #else /* (BYTE_ORDER == PDP_ENDIAN) */
404 #error "Byte order not supported"
408 #endif /* _LTTNG_BITFIELD_H */
This page took 0.037454 seconds and 5 git commands to generate.