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