1 | /* SPDX-License-Identifier: GPL-2.0 */ |
2 | #ifndef _LINUX_MINMAX_H |
3 | #define _LINUX_MINMAX_H |
4 | |
5 | #include <linux/build_bug.h> |
6 | #include <linux/compiler.h> |
7 | #include <linux/const.h> |
8 | #include <linux/types.h> |
9 | |
10 | /* |
11 | * min()/max()/clamp() macros must accomplish three things: |
12 | * |
13 | * - Avoid multiple evaluations of the arguments (so side-effects like |
14 | * "x++" happen only once) when non-constant. |
15 | * - Retain result as a constant expressions when called with only |
16 | * constant expressions (to avoid tripping VLA warnings in stack |
17 | * allocation usage). |
18 | * - Perform signed v unsigned type-checking (to generate compile |
19 | * errors instead of nasty runtime surprises). |
20 | * - Unsigned char/short are always promoted to signed int and can be |
21 | * compared against signed or unsigned arguments. |
22 | * - Unsigned arguments can be compared against non-negative signed constants. |
23 | * - Comparison of a signed argument against an unsigned constant fails |
24 | * even if the constant is below __INT_MAX__ and could be cast to int. |
25 | */ |
26 | #define __typecheck(x, y) \ |
27 | (!!(sizeof((typeof(x) *)1 == (typeof(y) *)1))) |
28 | |
29 | /* is_signed_type() isn't a constexpr for pointer types */ |
30 | #define __is_signed(x) \ |
31 | __builtin_choose_expr(__is_constexpr(is_signed_type(typeof(x))), \ |
32 | is_signed_type(typeof(x)), 0) |
33 | |
34 | /* True for a non-negative signed int constant */ |
35 | #define __is_noneg_int(x) \ |
36 | (__builtin_choose_expr(__is_constexpr(x) && __is_signed(x), x, -1) >= 0) |
37 | |
38 | #define __types_ok(x, y) \ |
39 | (__is_signed(x) == __is_signed(y) || \ |
40 | __is_signed((x) + 0) == __is_signed((y) + 0) || \ |
41 | __is_noneg_int(x) || __is_noneg_int(y)) |
42 | |
43 | #define __cmp_op_min < |
44 | #define __cmp_op_max > |
45 | |
46 | #define __cmp(op, x, y) ((x) __cmp_op_##op (y) ? (x) : (y)) |
47 | |
48 | #define __cmp_once(op, x, y, unique_x, unique_y) ({ \ |
49 | typeof(x) unique_x = (x); \ |
50 | typeof(y) unique_y = (y); \ |
51 | static_assert(__types_ok(x, y), \ |
52 | #op "(" #x ", " #y ") signedness error, fix types or consider u" #op "() before " #op "_t()"); \ |
53 | __cmp(op, unique_x, unique_y); }) |
54 | |
55 | #define __careful_cmp(op, x, y) \ |
56 | __builtin_choose_expr(__is_constexpr((x) - (y)), \ |
57 | __cmp(op, x, y), \ |
58 | __cmp_once(op, x, y, __UNIQUE_ID(__x), __UNIQUE_ID(__y))) |
59 | |
60 | #define __clamp(val, lo, hi) \ |
61 | ((val) >= (hi) ? (hi) : ((val) <= (lo) ? (lo) : (val))) |
62 | |
63 | #define __clamp_once(val, lo, hi, unique_val, unique_lo, unique_hi) ({ \ |
64 | typeof(val) unique_val = (val); \ |
65 | typeof(lo) unique_lo = (lo); \ |
66 | typeof(hi) unique_hi = (hi); \ |
67 | static_assert(__builtin_choose_expr(__is_constexpr((lo) > (hi)), \ |
68 | (lo) <= (hi), true), \ |
69 | "clamp() low limit " #lo " greater than high limit " #hi); \ |
70 | static_assert(__types_ok(val, lo), "clamp() 'lo' signedness error"); \ |
71 | static_assert(__types_ok(val, hi), "clamp() 'hi' signedness error"); \ |
72 | __clamp(unique_val, unique_lo, unique_hi); }) |
73 | |
74 | #define __careful_clamp(val, lo, hi) ({ \ |
75 | __builtin_choose_expr(__is_constexpr((val) - (lo) + (hi)), \ |
76 | __clamp(val, lo, hi), \ |
77 | __clamp_once(val, lo, hi, __UNIQUE_ID(__val), \ |
78 | __UNIQUE_ID(__lo), __UNIQUE_ID(__hi))); }) |
79 | |
80 | /** |
81 | * min - return minimum of two values of the same or compatible types |
82 | * @x: first value |
83 | * @y: second value |
84 | */ |
85 | #define min(x, y) __careful_cmp(min, x, y) |
86 | |
87 | /** |
88 | * max - return maximum of two values of the same or compatible types |
89 | * @x: first value |
90 | * @y: second value |
91 | */ |
92 | #define max(x, y) __careful_cmp(max, x, y) |
93 | |
94 | /** |
95 | * umin - return minimum of two non-negative values |
96 | * Signed types are zero extended to match a larger unsigned type. |
97 | * @x: first value |
98 | * @y: second value |
99 | */ |
100 | #define umin(x, y) \ |
101 | __careful_cmp(min, (x) + 0u + 0ul + 0ull, (y) + 0u + 0ul + 0ull) |
102 | |
103 | /** |
104 | * umax - return maximum of two non-negative values |
105 | * @x: first value |
106 | * @y: second value |
107 | */ |
108 | #define umax(x, y) \ |
109 | __careful_cmp(max, (x) + 0u + 0ul + 0ull, (y) + 0u + 0ul + 0ull) |
110 | |
111 | /** |
112 | * min3 - return minimum of three values |
113 | * @x: first value |
114 | * @y: second value |
115 | * @z: third value |
116 | */ |
117 | #define min3(x, y, z) min((typeof(x))min(x, y), z) |
118 | |
119 | /** |
120 | * max3 - return maximum of three values |
121 | * @x: first value |
122 | * @y: second value |
123 | * @z: third value |
124 | */ |
125 | #define max3(x, y, z) max((typeof(x))max(x, y), z) |
126 | |
127 | /** |
128 | * min_not_zero - return the minimum that is _not_ zero, unless both are zero |
129 | * @x: value1 |
130 | * @y: value2 |
131 | */ |
132 | #define min_not_zero(x, y) ({ \ |
133 | typeof(x) __x = (x); \ |
134 | typeof(y) __y = (y); \ |
135 | __x == 0 ? __y : ((__y == 0) ? __x : min(__x, __y)); }) |
136 | |
137 | /** |
138 | * clamp - return a value clamped to a given range with strict typechecking |
139 | * @val: current value |
140 | * @lo: lowest allowable value |
141 | * @hi: highest allowable value |
142 | * |
143 | * This macro does strict typechecking of @lo/@hi to make sure they are of the |
144 | * same type as @val. See the unnecessary pointer comparisons. |
145 | */ |
146 | #define clamp(val, lo, hi) __careful_clamp(val, lo, hi) |
147 | |
148 | /* |
149 | * ..and if you can't take the strict |
150 | * types, you can specify one yourself. |
151 | * |
152 | * Or not use min/max/clamp at all, of course. |
153 | */ |
154 | |
155 | /** |
156 | * min_t - return minimum of two values, using the specified type |
157 | * @type: data type to use |
158 | * @x: first value |
159 | * @y: second value |
160 | */ |
161 | #define min_t(type, x, y) __careful_cmp(min, (type)(x), (type)(y)) |
162 | |
163 | /** |
164 | * max_t - return maximum of two values, using the specified type |
165 | * @type: data type to use |
166 | * @x: first value |
167 | * @y: second value |
168 | */ |
169 | #define max_t(type, x, y) __careful_cmp(max, (type)(x), (type)(y)) |
170 | |
171 | /* |
172 | * Do not check the array parameter using __must_be_array(). |
173 | * In the following legit use-case where the "array" passed is a simple pointer, |
174 | * __must_be_array() will return a failure. |
175 | * --- 8< --- |
176 | * int *buff |
177 | * ... |
178 | * min = min_array(buff, nb_items); |
179 | * --- 8< --- |
180 | * |
181 | * The first typeof(&(array)[0]) is needed in order to support arrays of both |
182 | * 'int *buff' and 'int buff[N]' types. |
183 | * |
184 | * The array can be an array of const items. |
185 | * typeof() keeps the const qualifier. Use __unqual_scalar_typeof() in order |
186 | * to discard the const qualifier for the __element variable. |
187 | */ |
188 | #define __minmax_array(op, array, len) ({ \ |
189 | typeof(&(array)[0]) __array = (array); \ |
190 | typeof(len) __len = (len); \ |
191 | __unqual_scalar_typeof(__array[0]) __element = __array[--__len];\ |
192 | while (__len--) \ |
193 | __element = op(__element, __array[__len]); \ |
194 | __element; }) |
195 | |
196 | /** |
197 | * min_array - return minimum of values present in an array |
198 | * @array: array |
199 | * @len: array length |
200 | * |
201 | * Note that @len must not be zero (empty array). |
202 | */ |
203 | #define min_array(array, len) __minmax_array(min, array, len) |
204 | |
205 | /** |
206 | * max_array - return maximum of values present in an array |
207 | * @array: array |
208 | * @len: array length |
209 | * |
210 | * Note that @len must not be zero (empty array). |
211 | */ |
212 | #define max_array(array, len) __minmax_array(max, array, len) |
213 | |
214 | /** |
215 | * clamp_t - return a value clamped to a given range using a given type |
216 | * @type: the type of variable to use |
217 | * @val: current value |
218 | * @lo: minimum allowable value |
219 | * @hi: maximum allowable value |
220 | * |
221 | * This macro does no typechecking and uses temporary variables of type |
222 | * @type to make all the comparisons. |
223 | */ |
224 | #define clamp_t(type, val, lo, hi) __careful_clamp((type)(val), (type)(lo), (type)(hi)) |
225 | |
226 | /** |
227 | * clamp_val - return a value clamped to a given range using val's type |
228 | * @val: current value |
229 | * @lo: minimum allowable value |
230 | * @hi: maximum allowable value |
231 | * |
232 | * This macro does no typechecking and uses temporary variables of whatever |
233 | * type the input argument @val is. This is useful when @val is an unsigned |
234 | * type and @lo and @hi are literals that will otherwise be assigned a signed |
235 | * integer type. |
236 | */ |
237 | #define clamp_val(val, lo, hi) clamp_t(typeof(val), val, lo, hi) |
238 | |
239 | static inline bool in_range64(u64 val, u64 start, u64 len) |
240 | { |
241 | return (val - start) < len; |
242 | } |
243 | |
244 | static inline bool in_range32(u32 val, u32 start, u32 len) |
245 | { |
246 | return (val - start) < len; |
247 | } |
248 | |
249 | /** |
250 | * in_range - Determine if a value lies within a range. |
251 | * @val: Value to test. |
252 | * @start: First value in range. |
253 | * @len: Number of values in range. |
254 | * |
255 | * This is more efficient than "if (start <= val && val < (start + len))". |
256 | * It also gives a different answer if @start + @len overflows the size of |
257 | * the type by a sufficient amount to encompass @val. Decide for yourself |
258 | * which behaviour you want, or prove that start + len never overflow. |
259 | * Do not blindly replace one form with the other. |
260 | */ |
261 | #define in_range(val, start, len) \ |
262 | ((sizeof(start) | sizeof(len) | sizeof(val)) <= sizeof(u32) ? \ |
263 | in_range32(val, start, len) : in_range64(val, start, len)) |
264 | |
265 | /** |
266 | * swap - swap values of @a and @b |
267 | * @a: first value |
268 | * @b: second value |
269 | */ |
270 | #define swap(a, b) \ |
271 | do { typeof(a) __tmp = (a); (a) = (b); (b) = __tmp; } while (0) |
272 | |
273 | #endif /* _LINUX_MINMAX_H */ |
274 | |