| 1 | /* SPDX-License-Identifier: GPL-2.0 */ |
| 2 | #ifndef _VSTRUCTS_H |
| 3 | #define _VSTRUCTS_H |
| 4 | |
| 5 | #include "util.h" |
| 6 | |
| 7 | /* |
| 8 | * NOTE: we can't differentiate between __le64 and u64 with type_is - this |
| 9 | * assumes u64 is little endian: |
| 10 | */ |
| 11 | #define __vstruct_u64s(_s) \ |
| 12 | ({ \ |
| 13 | ( type_is((_s)->u64s, u64) ? le64_to_cpu((__force __le64) (_s)->u64s) \ |
| 14 | : type_is((_s)->u64s, u32) ? le32_to_cpu((__force __le32) (_s)->u64s) \ |
| 15 | : type_is((_s)->u64s, u16) ? le16_to_cpu((__force __le16) (_s)->u64s) \ |
| 16 | : ((__force u8) ((_s)->u64s))); \ |
| 17 | }) |
| 18 | |
| 19 | #define __vstruct_bytes(_type, _u64s) \ |
| 20 | ({ \ |
| 21 | BUILD_BUG_ON(offsetof(_type, _data) % sizeof(u64)); \ |
| 22 | \ |
| 23 | (size_t) (offsetof(_type, _data) + (_u64s) * sizeof(u64)); \ |
| 24 | }) |
| 25 | |
| 26 | #define vstruct_bytes(_s) \ |
| 27 | __vstruct_bytes(typeof(*(_s)), __vstruct_u64s(_s)) |
| 28 | |
| 29 | #define __vstruct_blocks(_type, _sector_block_bits, _u64s) \ |
| 30 | (round_up(__vstruct_bytes(_type, _u64s), \ |
| 31 | 512 << (_sector_block_bits)) >> (9 + (_sector_block_bits))) |
| 32 | |
| 33 | #define vstruct_blocks(_s, _sector_block_bits) \ |
| 34 | __vstruct_blocks(typeof(*(_s)), _sector_block_bits, __vstruct_u64s(_s)) |
| 35 | |
| 36 | #define vstruct_blocks_plus(_s, _sector_block_bits, _u64s) \ |
| 37 | __vstruct_blocks(typeof(*(_s)), _sector_block_bits, \ |
| 38 | __vstruct_u64s(_s) + (_u64s)) |
| 39 | |
| 40 | #define vstruct_sectors(_s, _sector_block_bits) \ |
| 41 | (round_up(vstruct_bytes(_s), 512 << (_sector_block_bits)) >> 9) |
| 42 | |
| 43 | #define vstruct_next(_s) \ |
| 44 | ((typeof(_s)) ((u64 *) (_s)->_data + __vstruct_u64s(_s))) |
| 45 | #define vstruct_last(_s) \ |
| 46 | ((typeof(&(_s)->start[0])) ((u64 *) (_s)->_data + __vstruct_u64s(_s))) |
| 47 | #define vstruct_end(_s) \ |
| 48 | ((void *) ((u64 *) (_s)->_data + __vstruct_u64s(_s))) |
| 49 | |
| 50 | #define vstruct_for_each(_s, _i) \ |
| 51 | for (typeof(&(_s)->start[0]) _i = (_s)->start; \ |
| 52 | _i < vstruct_last(_s); \ |
| 53 | _i = vstruct_next(_i)) |
| 54 | |
| 55 | #define vstruct_for_each_safe(_s, _i) \ |
| 56 | for (typeof(&(_s)->start[0]) _next, _i = (_s)->start; \ |
| 57 | _i < vstruct_last(_s) && (_next = vstruct_next(_i), true); \ |
| 58 | _i = _next) |
| 59 | |
| 60 | #define vstruct_idx(_s, _idx) \ |
| 61 | ((typeof(&(_s)->start[0])) ((_s)->_data + (_idx))) |
| 62 | |
| 63 | #endif /* _VSTRUCTS_H */ |
| 64 | |