Message ID | 20221215125227.1165037-2-gwan-gyeong.mun@intel.com (mailing list archive) |
---|---|
State | New, archived |
Headers | show |
Series | Fixes integer overflow or integer truncation issues in page lookups, ttm place configuration and scatterlist creation | expand |
On 15.12.2022 13:52, Gwan-gyeong Mun wrote: > From: Kees Cook <keescook@chromium.org> > > Implement a robust overflows_type() macro to test if a variable or > constant value would overflow another variable or type. This can be > used as a constant expression for static_assert() (which requires a > constant expression[1][2]) when used on constant values. This must be > constructed manually, since __builtin_add_overflow() does not produce > a constant expression[3]. > > Additionally adds castable_to_type(), similar to __same_type(), but for > checking if a constant value would overflow if cast to a given type. > > Add unit tests for overflows_type(), __same_type(), and castable_to_type() > to the existing KUnit "overflow" test: > > [16:03:33] ================== overflow (21 subtests) ================== > ... > [16:03:33] [PASSED] overflows_type_test > [16:03:33] [PASSED] same_type_test > [16:03:33] [PASSED] castable_to_type_test > [16:03:33] ==================== [PASSED] overflow ===================== > [16:03:33] ============================================================ > [16:03:33] Testing complete. Ran 21 tests: passed: 21 > [16:03:33] Elapsed time: 24.022s total, 0.002s configuring, 22.598s building, 0.767s running > > [1] https://en.cppreference.com/w/c/language/_Static_assert > [2] C11 standard (ISO/IEC 9899:2011): 6.7.10 Static assertions > [3] https://gcc.gnu.org/onlinedocs/gcc/Integer-Overflow-Builtins.html > 6.56 Built-in Functions to Perform Arithmetic with Overflow Checking > Built-in Function: bool __builtin_add_overflow (type1 a, type2 b, > > Cc: Luc Van Oostenryck <luc.vanoostenryck@gmail.com> > Cc: Nathan Chancellor <nathan@kernel.org> > Cc: Nick Desaulniers <ndesaulniers@google.com> > Cc: Tom Rix <trix@redhat.com> > Cc: Daniel Latypov <dlatypov@google.com> > Cc: Vitor Massaru Iha <vitor@massaru.org> > Cc: "Gustavo A. R. Silva" <gustavoars@kernel.org> > Cc: Jani Nikula <jani.nikula@intel.com> > Cc: Mauro Carvalho Chehab <mchehab@kernel.org> > Cc: linux-hardening@vger.kernel.org > Cc: llvm@lists.linux.dev > Co-developed-by: Gwan-gyeong Mun <gwan-gyeong.mun@intel.com> > Signed-off-by: Gwan-gyeong Mun <gwan-gyeong.mun@intel.com> > Signed-off-by: Kees Cook <keescook@chromium.org> > Link: https://lore.kernel.org/r/20221024201125.1416422-1-gwan-gyeong.mun@intel.com > --- > drivers/gpu/drm/i915/i915_user_extensions.c | 2 +- > drivers/gpu/drm/i915/i915_utils.h | 4 - > include/linux/compiler.h | 1 + > include/linux/overflow.h | 48 +++ > lib/Makefile | 1 + > lib/overflow_kunit.c | 381 ++++++++++++++++++++ > 6 files changed, 432 insertions(+), 5 deletions(-) > > diff --git a/drivers/gpu/drm/i915/i915_user_extensions.c b/drivers/gpu/drm/i915/i915_user_extensions.c > index c822d0aafd2d..e3f808372c47 100644 > --- a/drivers/gpu/drm/i915/i915_user_extensions.c > +++ b/drivers/gpu/drm/i915/i915_user_extensions.c > @@ -51,7 +51,7 @@ int i915_user_extensions(struct i915_user_extension __user *ext, > return err; > > if (get_user(next, &ext->next_extension) || > - overflows_type(next, ext)) > + overflows_type(next, uintptr_t)) > return -EFAULT; > > ext = u64_to_user_ptr(next); > diff --git a/drivers/gpu/drm/i915/i915_utils.h b/drivers/gpu/drm/i915/i915_utils.h > index b64192d9c7da..2c430c0c3bad 100644 > --- a/drivers/gpu/drm/i915/i915_utils.h > +++ b/drivers/gpu/drm/i915/i915_utils.h > @@ -111,10 +111,6 @@ bool i915_error_injected(void); > #define range_overflows_end_t(type, start, size, max) \ > range_overflows_end((type)(start), (type)(size), (type)(max)) > > -/* Note we don't consider signbits :| */ > -#define overflows_type(x, T) \ > - (sizeof(x) > sizeof(T) && (x) >> BITS_PER_TYPE(T)) > - > #define ptr_mask_bits(ptr, n) ({ \ > unsigned long __v = (unsigned long)(ptr); \ > (typeof(ptr))(__v & -BIT(n)); \ > diff --git a/include/linux/compiler.h b/include/linux/compiler.h > index 973a1bfd7ef5..947a60b801db 100644 > --- a/include/linux/compiler.h > +++ b/include/linux/compiler.h > @@ -236,6 +236,7 @@ static inline void *offset_to_ptr(const int *off) > * bool and also pointer types. > */ > #define is_signed_type(type) (((type)(-1)) < (__force type)1) > +#define is_unsigned_type(type) (!is_signed_type(type)) > > /* > * This is needed in functions which generate the stack canary, see > diff --git a/include/linux/overflow.h b/include/linux/overflow.h > index 1d3be1a2204c..16ae8d912390 100644 > --- a/include/linux/overflow.h > +++ b/include/linux/overflow.h > @@ -128,6 +128,54 @@ static inline bool __must_check __must_check_overflow(bool overflow) > (*_d >> _to_shift) != _a); \ > })) > > +#define __overflows_type_constexpr(x, T) ( \ > + is_unsigned_type(typeof(x)) ? \ > + (x) > type_max(typeof(T)) ? 1 : 0 \ > + : is_unsigned_type(typeof(T)) ? \ > + (x) < 0 || (x) > type_max(typeof(T)) ? 1 : 0 \ > + : (x) < type_min(typeof(T)) || \ > + (x) > type_max(typeof(T)) ? 1 : 0) Syntactically seems fine, but I am little bit lost with indentation. Shouldn't the last two lines be one tab to the left? Up to you. Btw, maybe extra helper in_range/between: in_range(x, type_min(T), type_max(T)) :) The rest looks OK Reviewed-by: Andrzej Hajda <andrzej.hajda@intel.com> Regards Andrzej > + > +#define __overflows_type(x, T) ({ \ > + typeof(T) v = 0; \ > + check_add_overflow((x), v, &v); \ > +}) > + > +/** > + * overflows_type - helper for checking the overflows between value, variables, > + * or data type > + * > + * @n: source constant value or variable to be checked > + * @T: destination variable or data type proposed to store @x > + * > + * Compares the @x expression for whether or not it can safely fit in > + * the storage of the type in @T. @x and @T can have different types. > + * If @x is a constant expression, this will also resolve to a constant > + * expression. > + * > + * Returns: true if overflow can occur, false otherwise. > + */ > +#define overflows_type(n, T) \ > + __builtin_choose_expr(__is_constexpr(n), \ > + __overflows_type_constexpr(n, T), \ > + __overflows_type(n, T)) > + > +/** > + * castable_to_type - like __same_type(), but also allows for casted literals > + * > + * @n: variable or constant value > + * @T: variable or data type > + * > + * Unlike the __same_type() macro, this allows a constant value as the > + * first argument. If this value would not overflow into an assignment > + * of the second argument's type, it returns true. Otherwise, this falls > + * back to __same_type(). > + */ > +#define castable_to_type(n, T) \ > + __builtin_choose_expr(__is_constexpr(n), \ > + !__overflows_type_constexpr(n, T), \ > + __same_type(n, T)) > + > /** > * size_mul() - Calculate size_t multiplication with saturation at SIZE_MAX > * @factor1: first factor > diff --git a/lib/Makefile b/lib/Makefile > index 59bd7c2f793a..889f96b10f12 100644 > --- a/lib/Makefile > +++ b/lib/Makefile > @@ -377,6 +377,7 @@ obj-$(CONFIG_CMDLINE_KUNIT_TEST) += cmdline_kunit.o > obj-$(CONFIG_SLUB_KUNIT_TEST) += slub_kunit.o > obj-$(CONFIG_MEMCPY_KUNIT_TEST) += memcpy_kunit.o > obj-$(CONFIG_IS_SIGNED_TYPE_KUNIT_TEST) += is_signed_type_kunit.o > +CFLAGS_overflow_kunit.o = $(call cc-disable-warning, tautological-constant-out-of-range-compare) > obj-$(CONFIG_OVERFLOW_KUNIT_TEST) += overflow_kunit.o > CFLAGS_stackinit_kunit.o += $(call cc-disable-warning, switch-unreachable) > obj-$(CONFIG_STACKINIT_KUNIT_TEST) += stackinit_kunit.o > diff --git a/lib/overflow_kunit.c b/lib/overflow_kunit.c > index b8556a2e7bb1..dcd3ba102db6 100644 > --- a/lib/overflow_kunit.c > +++ b/lib/overflow_kunit.c > @@ -736,6 +736,384 @@ static void overflow_size_helpers_test(struct kunit *test) > #undef check_one_size_helper > } > > +static void overflows_type_test(struct kunit *test) > +{ > + int count = 0; > + unsigned int var; > + > +#define __TEST_OVERFLOWS_TYPE(func, arg1, arg2, of) do { \ > + bool __of = func(arg1, arg2); \ > + KUNIT_EXPECT_EQ_MSG(test, __of, of, \ > + "expected " #func "(" #arg1 ", " #arg2 " to%s overflow\n",\ > + of ? "" : " not"); \ > + count++; \ > +} while (0) > + > +/* Args are: first type, second type, value, overflow expected */ > +#define TEST_OVERFLOWS_TYPE(__t1, __t2, v, of) do { \ > + __t1 t1 = (v); \ > + __t2 t2; \ > + __TEST_OVERFLOWS_TYPE(__overflows_type, t1, t2, of); \ > + __TEST_OVERFLOWS_TYPE(__overflows_type, t1, __t2, of); \ > + __TEST_OVERFLOWS_TYPE(__overflows_type_constexpr, t1, t2, of); \ > + __TEST_OVERFLOWS_TYPE(__overflows_type_constexpr, t1, __t2, of);\ > +} while (0) > + > + TEST_OVERFLOWS_TYPE(u8, u8, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(u8, u16, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(u8, s8, U8_MAX, true); > + TEST_OVERFLOWS_TYPE(u8, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(u8, s8, (u8)S8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u8, s16, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(s8, u8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s8, u8, -1, true); > + TEST_OVERFLOWS_TYPE(s8, u8, S8_MIN, true); > + TEST_OVERFLOWS_TYPE(s8, u16, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s8, u16, -1, true); > + TEST_OVERFLOWS_TYPE(s8, u16, S8_MIN, true); > + TEST_OVERFLOWS_TYPE(s8, u32, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s8, u32, -1, true); > + TEST_OVERFLOWS_TYPE(s8, u32, S8_MIN, true); > +#if BITS_PER_LONG == 64 > + TEST_OVERFLOWS_TYPE(s8, u64, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s8, u64, -1, true); > + TEST_OVERFLOWS_TYPE(s8, u64, S8_MIN, true); > +#endif > + TEST_OVERFLOWS_TYPE(s8, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s8, s8, S8_MIN, false); > + TEST_OVERFLOWS_TYPE(s8, s16, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s8, s16, S8_MIN, false); > + TEST_OVERFLOWS_TYPE(u16, u8, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(u16, u8, (u16)U8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u16, u8, U16_MAX, true); > + TEST_OVERFLOWS_TYPE(u16, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(u16, s8, (u16)S8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u16, s8, U16_MAX, true); > + TEST_OVERFLOWS_TYPE(u16, s16, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(u16, s16, (u16)S16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u16, s16, U16_MAX, true); > + TEST_OVERFLOWS_TYPE(u16, u32, U16_MAX, false); > + TEST_OVERFLOWS_TYPE(u16, s32, U16_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, u8, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, u8, (s16)U8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s16, u8, -1, true); > + TEST_OVERFLOWS_TYPE(s16, u8, S16_MIN, true); > + TEST_OVERFLOWS_TYPE(s16, u16, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, u16, -1, true); > + TEST_OVERFLOWS_TYPE(s16, u16, S16_MIN, true); > + TEST_OVERFLOWS_TYPE(s16, u32, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, u32, -1, true); > + TEST_OVERFLOWS_TYPE(s16, u32, S16_MIN, true); > +#if BITS_PER_LONG == 64 > + TEST_OVERFLOWS_TYPE(s16, u64, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, u64, -1, true); > + TEST_OVERFLOWS_TYPE(s16, u64, S16_MIN, true); > +#endif > + TEST_OVERFLOWS_TYPE(s16, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, s8, S8_MIN, false); > + TEST_OVERFLOWS_TYPE(s16, s8, (s16)S8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s16, s8, (s16)S8_MIN - 1, true); > + TEST_OVERFLOWS_TYPE(s16, s8, S16_MAX, true); > + TEST_OVERFLOWS_TYPE(s16, s8, S16_MIN, true); > + TEST_OVERFLOWS_TYPE(s16, s16, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, s16, S16_MIN, false); > + TEST_OVERFLOWS_TYPE(s16, s32, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(s16, s32, S16_MIN, false); > + TEST_OVERFLOWS_TYPE(u32, u8, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(u32, u8, (u32)U8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u32, u8, U32_MAX, true); > + TEST_OVERFLOWS_TYPE(u32, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(u32, s8, (u32)S8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u32, s8, U32_MAX, true); > + TEST_OVERFLOWS_TYPE(u32, u16, U16_MAX, false); > + TEST_OVERFLOWS_TYPE(u32, u16, U16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u32, u16, U32_MAX, true); > + TEST_OVERFLOWS_TYPE(u32, s16, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(u32, s16, (u32)S16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u32, s16, U32_MAX, true); > + TEST_OVERFLOWS_TYPE(u32, u32, U32_MAX, false); > + TEST_OVERFLOWS_TYPE(u32, s32, S32_MAX, false); > + TEST_OVERFLOWS_TYPE(u32, s32, U32_MAX, true); > + TEST_OVERFLOWS_TYPE(u32, s32, (u32)S32_MAX + 1, true); > +#if BITS_PER_LONG == 64 > + TEST_OVERFLOWS_TYPE(u32, u64, U32_MAX, false); > + TEST_OVERFLOWS_TYPE(u32, s64, U32_MAX, false); > +#endif > + TEST_OVERFLOWS_TYPE(s32, u8, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, u8, (s32)U8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s32, u16, S32_MAX, true); > + TEST_OVERFLOWS_TYPE(s32, u8, -1, true); > + TEST_OVERFLOWS_TYPE(s32, u8, S32_MIN, true); > + TEST_OVERFLOWS_TYPE(s32, u16, U16_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, u16, (s32)U16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s32, u16, S32_MAX, true); > + TEST_OVERFLOWS_TYPE(s32, u16, -1, true); > + TEST_OVERFLOWS_TYPE(s32, u16, S32_MIN, true); > + TEST_OVERFLOWS_TYPE(s32, u32, S32_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, u32, -1, true); > + TEST_OVERFLOWS_TYPE(s32, u32, S32_MIN, true); > +#if BITS_PER_LONG == 64 > + TEST_OVERFLOWS_TYPE(s32, u64, S32_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, u64, -1, true); > + TEST_OVERFLOWS_TYPE(s32, u64, S32_MIN, true); > +#endif > + TEST_OVERFLOWS_TYPE(s32, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, s8, S8_MIN, false); > + TEST_OVERFLOWS_TYPE(s32, s8, (s32)S8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s32, s8, (s32)S8_MIN - 1, true); > + TEST_OVERFLOWS_TYPE(s32, s8, S32_MAX, true); > + TEST_OVERFLOWS_TYPE(s32, s8, S32_MIN, true); > + TEST_OVERFLOWS_TYPE(s32, s16, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, s16, S16_MIN, false); > + TEST_OVERFLOWS_TYPE(s32, s16, (s32)S16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s32, s16, (s32)S16_MIN - 1, true); > + TEST_OVERFLOWS_TYPE(s32, s16, S32_MAX, true); > + TEST_OVERFLOWS_TYPE(s32, s16, S32_MIN, true); > + TEST_OVERFLOWS_TYPE(s32, s32, S32_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, s32, S32_MIN, false); > +#if BITS_PER_LONG == 64 > + TEST_OVERFLOWS_TYPE(s32, s64, S32_MAX, false); > + TEST_OVERFLOWS_TYPE(s32, s64, S32_MIN, false); > + TEST_OVERFLOWS_TYPE(u64, u8, U64_MAX, true); > + TEST_OVERFLOWS_TYPE(u64, u8, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, u8, (u64)U8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u64, u16, U64_MAX, true); > + TEST_OVERFLOWS_TYPE(u64, u16, U16_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, u16, (u64)U16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u64, u32, U64_MAX, true); > + TEST_OVERFLOWS_TYPE(u64, u32, U32_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, u32, (u64)U32_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u64, u64, U64_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, s8, (u64)S8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u64, s8, U64_MAX, true); > + TEST_OVERFLOWS_TYPE(u64, s16, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, s16, (u64)S16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u64, s16, U64_MAX, true); > + TEST_OVERFLOWS_TYPE(u64, s32, S32_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, s32, (u64)S32_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(u64, s32, U64_MAX, true); > + TEST_OVERFLOWS_TYPE(u64, s64, S64_MAX, false); > + TEST_OVERFLOWS_TYPE(u64, s64, U64_MAX, true); > + TEST_OVERFLOWS_TYPE(u64, s64, (u64)S64_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s64, u8, S64_MAX, true); > + TEST_OVERFLOWS_TYPE(s64, u8, S64_MIN, true); > + TEST_OVERFLOWS_TYPE(s64, u8, -1, true); > + TEST_OVERFLOWS_TYPE(s64, u8, U8_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, u8, (s64)U8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s64, u16, S64_MAX, true); > + TEST_OVERFLOWS_TYPE(s64, u16, S64_MIN, true); > + TEST_OVERFLOWS_TYPE(s64, u16, -1, true); > + TEST_OVERFLOWS_TYPE(s64, u16, U16_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, u16, (s64)U16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s64, u32, S64_MAX, true); > + TEST_OVERFLOWS_TYPE(s64, u32, S64_MIN, true); > + TEST_OVERFLOWS_TYPE(s64, u32, -1, true); > + TEST_OVERFLOWS_TYPE(s64, u32, U32_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, u32, (s64)U32_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s64, u64, S64_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, u64, S64_MIN, true); > + TEST_OVERFLOWS_TYPE(s64, u64, -1, true); > + TEST_OVERFLOWS_TYPE(s64, s8, S8_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, s8, S8_MIN, false); > + TEST_OVERFLOWS_TYPE(s64, s8, (s64)S8_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s64, s8, (s64)S8_MIN - 1, true); > + TEST_OVERFLOWS_TYPE(s64, s8, S64_MAX, true); > + TEST_OVERFLOWS_TYPE(s64, s16, S16_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, s16, S16_MIN, false); > + TEST_OVERFLOWS_TYPE(s64, s16, (s64)S16_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s64, s16, (s64)S16_MIN - 1, true); > + TEST_OVERFLOWS_TYPE(s64, s16, S64_MAX, true); > + TEST_OVERFLOWS_TYPE(s64, s32, S32_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, s32, S32_MIN, false); > + TEST_OVERFLOWS_TYPE(s64, s32, (s64)S32_MAX + 1, true); > + TEST_OVERFLOWS_TYPE(s64, s32, (s64)S32_MIN - 1, true); > + TEST_OVERFLOWS_TYPE(s64, s32, S64_MAX, true); > + TEST_OVERFLOWS_TYPE(s64, s64, S64_MAX, false); > + TEST_OVERFLOWS_TYPE(s64, s64, S64_MIN, false); > +#endif > + > + /* Check for macro side-effects. */ > + var = INT_MAX - 1; > + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, false); > + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, false); > + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, true); > + var = INT_MAX - 1; > + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, false); > + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, false); > + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, true); > + > + kunit_info(test, "%d overflows_type() tests finished\n", count); > +#undef TEST_OVERFLOWS_TYPE > +#undef __TEST_OVERFLOWS_TYPE > +} > + > +static void same_type_test(struct kunit *test) > +{ > + int count = 0; > + int var; > + > +#define TEST_SAME_TYPE(t1, t2, same) do { \ > + typeof(t1) __t1h = type_max(t1); \ > + typeof(t1) __t1l = type_min(t1); \ > + typeof(t2) __t2h = type_max(t2); \ > + typeof(t2) __t2l = type_min(t2); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(t1, __t1h)); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(t1, __t1l)); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(__t1h, t1)); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(__t1l, t1)); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(t2, __t2h)); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(t2, __t2l)); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(__t2h, t2)); \ > + KUNIT_EXPECT_EQ(test, true, __same_type(__t2l, t2)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(t1, t2)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(t2, __t1h)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(t2, __t1l)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(__t1h, t2)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(__t1l, t2)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(t1, __t2h)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(t1, __t2l)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(__t2h, t1)); \ > + KUNIT_EXPECT_EQ(test, same, __same_type(__t2l, t1)); \ > +} while (0) > + > +#if BITS_PER_LONG == 64 > +# define TEST_SAME_TYPE64(base, t, m) TEST_SAME_TYPE(base, t, m) > +#else > +# define TEST_SAME_TYPE64(base, t, m) do { } while (0) > +#endif > + > +#define TEST_TYPE_SETS(base, mu8, mu16, mu32, ms8, ms16, ms32, mu64, ms64) \ > +do { \ > + TEST_SAME_TYPE(base, u8, mu8); \ > + TEST_SAME_TYPE(base, u16, mu16); \ > + TEST_SAME_TYPE(base, u32, mu32); \ > + TEST_SAME_TYPE(base, s8, ms8); \ > + TEST_SAME_TYPE(base, s16, ms16); \ > + TEST_SAME_TYPE(base, s32, ms32); \ > + TEST_SAME_TYPE64(base, u64, mu64); \ > + TEST_SAME_TYPE64(base, s64, ms64); \ > +} while (0) > + > + TEST_TYPE_SETS(u8, true, false, false, false, false, false, false, false); > + TEST_TYPE_SETS(u16, false, true, false, false, false, false, false, false); > + TEST_TYPE_SETS(u32, false, false, true, false, false, false, false, false); > + TEST_TYPE_SETS(s8, false, false, false, true, false, false, false, false); > + TEST_TYPE_SETS(s16, false, false, false, false, true, false, false, false); > + TEST_TYPE_SETS(s32, false, false, false, false, false, true, false, false); > +#if BITS_PER_LONG == 64 > + TEST_TYPE_SETS(u64, false, false, false, false, false, false, true, false); > + TEST_TYPE_SETS(s64, false, false, false, false, false, false, false, true); > +#endif > + > + /* Check for macro side-effects. */ > + var = 4; > + KUNIT_EXPECT_EQ(test, var, 4); > + KUNIT_EXPECT_TRUE(test, __same_type(var++, int)); > + KUNIT_EXPECT_EQ(test, var, 4); > + KUNIT_EXPECT_TRUE(test, __same_type(int, var++)); > + KUNIT_EXPECT_EQ(test, var, 4); > + KUNIT_EXPECT_TRUE(test, __same_type(var++, var++)); > + KUNIT_EXPECT_EQ(test, var, 4); > + > + kunit_info(test, "%d __same_type() tests finished\n", count); > + > +#undef TEST_TYPE_SETS > +#undef TEST_SAME_TYPE64 > +#undef TEST_SAME_TYPE > +} > + > +static void castable_to_type_test(struct kunit *test) > +{ > + int count = 0; > + > +#define TEST_CASTABLE_TO_TYPE(arg1, arg2, pass) do { \ > + bool __pass = castable_to_type(arg1, arg2); \ > + KUNIT_EXPECT_EQ_MSG(test, __pass, pass, \ > + "expected castable_to_type(" #arg1 ", " #arg2 ") to%s pass\n",\ > + pass ? "" : " not"); \ > + count++; \ > +} while (0) > + > + TEST_CASTABLE_TO_TYPE(16, u8, true); > + TEST_CASTABLE_TO_TYPE(16, u16, true); > + TEST_CASTABLE_TO_TYPE(16, u32, true); > + TEST_CASTABLE_TO_TYPE(16, s8, true); > + TEST_CASTABLE_TO_TYPE(16, s16, true); > + TEST_CASTABLE_TO_TYPE(16, s32, true); > + TEST_CASTABLE_TO_TYPE(-16, s8, true); > + TEST_CASTABLE_TO_TYPE(-16, s16, true); > + TEST_CASTABLE_TO_TYPE(-16, s32, true); > +#if BITS_PER_LONG == 64 > + TEST_CASTABLE_TO_TYPE(16, u64, true); > + TEST_CASTABLE_TO_TYPE(-16, s64, true); > +#endif > + > +#define TEST_CASTABLE_TO_TYPE_VAR(width) do { \ > + u ## width u ## width ## var = 0; \ > + s ## width s ## width ## var = 0; \ > + \ > + /* Constant expressions that fit types. */ \ > + TEST_CASTABLE_TO_TYPE(type_max(u ## width), u ## width, true); \ > + TEST_CASTABLE_TO_TYPE(type_min(u ## width), u ## width, true); \ > + TEST_CASTABLE_TO_TYPE(type_max(u ## width), u ## width ## var, true); \ > + TEST_CASTABLE_TO_TYPE(type_min(u ## width), u ## width ## var, true); \ > + TEST_CASTABLE_TO_TYPE(type_max(s ## width), s ## width, true); \ > + TEST_CASTABLE_TO_TYPE(type_min(s ## width), s ## width, true); \ > + TEST_CASTABLE_TO_TYPE(type_max(s ## width), s ## width ## var, true); \ > + TEST_CASTABLE_TO_TYPE(type_min(u ## width), s ## width ## var, true); \ > + /* Constant expressions that do not fit types. */ \ > + TEST_CASTABLE_TO_TYPE(type_max(u ## width), s ## width, false); \ > + TEST_CASTABLE_TO_TYPE(type_max(u ## width), s ## width ## var, false); \ > + TEST_CASTABLE_TO_TYPE(type_min(s ## width), u ## width, false); \ > + TEST_CASTABLE_TO_TYPE(type_min(s ## width), u ## width ## var, false); \ > + /* Non-constant expression with mismatched type. */ \ > + TEST_CASTABLE_TO_TYPE(s ## width ## var, u ## width, false); \ > + TEST_CASTABLE_TO_TYPE(u ## width ## var, s ## width, false); \ > +} while (0) > + > +#define TEST_CASTABLE_TO_TYPE_RANGE(width) do { \ > + unsigned long big = U ## width ## _MAX; \ > + signed long small = S ## width ## _MIN; \ > + u ## width u ## width ## var = 0; \ > + s ## width s ## width ## var = 0; \ > + \ > + /* Constant expression in range. */ \ > + TEST_CASTABLE_TO_TYPE(U ## width ## _MAX, u ## width, true); \ > + TEST_CASTABLE_TO_TYPE(U ## width ## _MAX, u ## width ## var, true); \ > + TEST_CASTABLE_TO_TYPE(S ## width ## _MIN, s ## width, true); \ > + TEST_CASTABLE_TO_TYPE(S ## width ## _MIN, s ## width ## var, true); \ > + /* Constant expression out of range. */ \ > + TEST_CASTABLE_TO_TYPE((unsigned long)U ## width ## _MAX + 1, u ## width, false); \ > + TEST_CASTABLE_TO_TYPE((unsigned long)U ## width ## _MAX + 1, u ## width ## var, false); \ > + TEST_CASTABLE_TO_TYPE((signed long)S ## width ## _MIN - 1, s ## width, false); \ > + TEST_CASTABLE_TO_TYPE((signed long)S ## width ## _MIN - 1, s ## width ## var, false); \ > + /* Non-constant expression with mismatched type. */ \ > + TEST_CASTABLE_TO_TYPE(big, u ## width, false); \ > + TEST_CASTABLE_TO_TYPE(big, u ## width ## var, false); \ > + TEST_CASTABLE_TO_TYPE(small, s ## width, false); \ > + TEST_CASTABLE_TO_TYPE(small, s ## width ## var, false); \ > +} while (0) > + > + TEST_CASTABLE_TO_TYPE_VAR(8); > + TEST_CASTABLE_TO_TYPE_VAR(16); > + TEST_CASTABLE_TO_TYPE_VAR(32); > +#if BITS_PER_LONG == 64 > + TEST_CASTABLE_TO_TYPE_VAR(64); > +#endif > + > + TEST_CASTABLE_TO_TYPE_RANGE(8); > + TEST_CASTABLE_TO_TYPE_RANGE(16); > +#if BITS_PER_LONG == 64 > + TEST_CASTABLE_TO_TYPE_RANGE(32); > +#endif > + kunit_info(test, "%d castable_to_type() tests finished\n", count); > + > +#undef TEST_CASTABLE_TO_TYPE_RANGE > +#undef TEST_CASTABLE_TO_TYPE_VAR > +#undef TEST_CASTABLE_TO_TYPE > +} > + > static struct kunit_case overflow_test_cases[] = { > KUNIT_CASE(u8_u8__u8_overflow_test), > KUNIT_CASE(s8_s8__s8_overflow_test), > @@ -755,6 +1133,9 @@ static struct kunit_case overflow_test_cases[] = { > KUNIT_CASE(shift_nonsense_test), > KUNIT_CASE(overflow_allocation_test), > KUNIT_CASE(overflow_size_helpers_test), > + KUNIT_CASE(overflows_type_test), > + KUNIT_CASE(same_type_test), > + KUNIT_CASE(castable_to_type_test), > {} > }; >
On 12/15/22 5:09 PM, Andrzej Hajda wrote: > > > On 15.12.2022 13:52, Gwan-gyeong Mun wrote: >> From: Kees Cook <keescook@chromium.org> >> >> Implement a robust overflows_type() macro to test if a variable or >> constant value would overflow another variable or type. This can be >> used as a constant expression for static_assert() (which requires a >> constant expression[1][2]) when used on constant values. This must be >> constructed manually, since __builtin_add_overflow() does not produce >> a constant expression[3]. >> >> Additionally adds castable_to_type(), similar to __same_type(), but for >> checking if a constant value would overflow if cast to a given type. >> >> Add unit tests for overflows_type(), __same_type(), and >> castable_to_type() >> to the existing KUnit "overflow" test: >> >> [16:03:33] ================== overflow (21 subtests) ================== >> ... >> [16:03:33] [PASSED] overflows_type_test >> [16:03:33] [PASSED] same_type_test >> [16:03:33] [PASSED] castable_to_type_test >> [16:03:33] ==================== [PASSED] overflow ===================== >> [16:03:33] ============================================================ >> [16:03:33] Testing complete. Ran 21 tests: passed: 21 >> [16:03:33] Elapsed time: 24.022s total, 0.002s configuring, 22.598s >> building, 0.767s running >> >> [1] https://en.cppreference.com/w/c/language/_Static_assert >> [2] C11 standard (ISO/IEC 9899:2011): 6.7.10 Static assertions >> [3] https://gcc.gnu.org/onlinedocs/gcc/Integer-Overflow-Builtins.html >> 6.56 Built-in Functions to Perform Arithmetic with Overflow Checking >> Built-in Function: bool __builtin_add_overflow (type1 a, type2 b, >> >> Cc: Luc Van Oostenryck <luc.vanoostenryck@gmail.com> >> Cc: Nathan Chancellor <nathan@kernel.org> >> Cc: Nick Desaulniers <ndesaulniers@google.com> >> Cc: Tom Rix <trix@redhat.com> >> Cc: Daniel Latypov <dlatypov@google.com> >> Cc: Vitor Massaru Iha <vitor@massaru.org> >> Cc: "Gustavo A. R. Silva" <gustavoars@kernel.org> >> Cc: Jani Nikula <jani.nikula@intel.com> >> Cc: Mauro Carvalho Chehab <mchehab@kernel.org> >> Cc: linux-hardening@vger.kernel.org >> Cc: llvm@lists.linux.dev >> Co-developed-by: Gwan-gyeong Mun <gwan-gyeong.mun@intel.com> >> Signed-off-by: Gwan-gyeong Mun <gwan-gyeong.mun@intel.com> >> Signed-off-by: Kees Cook <keescook@chromium.org> >> Link: >> https://lore.kernel.org/r/20221024201125.1416422-1-gwan-gyeong.mun@intel.com >> >> --- >> drivers/gpu/drm/i915/i915_user_extensions.c | 2 +- >> drivers/gpu/drm/i915/i915_utils.h | 4 - >> include/linux/compiler.h | 1 + >> include/linux/overflow.h | 48 +++ >> lib/Makefile | 1 + >> lib/overflow_kunit.c | 381 ++++++++++++++++++++ >> 6 files changed, 432 insertions(+), 5 deletions(-) >> >> diff --git a/drivers/gpu/drm/i915/i915_user_extensions.c >> b/drivers/gpu/drm/i915/i915_user_extensions.c >> index c822d0aafd2d..e3f808372c47 100644 >> --- a/drivers/gpu/drm/i915/i915_user_extensions.c >> +++ b/drivers/gpu/drm/i915/i915_user_extensions.c >> @@ -51,7 +51,7 @@ int i915_user_extensions(struct i915_user_extension >> __user *ext, >> return err; >> if (get_user(next, &ext->next_extension) || >> - overflows_type(next, ext)) >> + overflows_type(next, uintptr_t)) >> return -EFAULT; >> ext = u64_to_user_ptr(next); >> diff --git a/drivers/gpu/drm/i915/i915_utils.h >> b/drivers/gpu/drm/i915/i915_utils.h >> index b64192d9c7da..2c430c0c3bad 100644 >> --- a/drivers/gpu/drm/i915/i915_utils.h >> +++ b/drivers/gpu/drm/i915/i915_utils.h >> @@ -111,10 +111,6 @@ bool i915_error_injected(void); >> #define range_overflows_end_t(type, start, size, max) \ >> range_overflows_end((type)(start), (type)(size), (type)(max)) >> -/* Note we don't consider signbits :| */ >> -#define overflows_type(x, T) \ >> - (sizeof(x) > sizeof(T) && (x) >> BITS_PER_TYPE(T)) >> - >> #define ptr_mask_bits(ptr, n) ({ \ >> unsigned long __v = (unsigned long)(ptr); \ >> (typeof(ptr))(__v & -BIT(n)); \ >> diff --git a/include/linux/compiler.h b/include/linux/compiler.h >> index 973a1bfd7ef5..947a60b801db 100644 >> --- a/include/linux/compiler.h >> +++ b/include/linux/compiler.h >> @@ -236,6 +236,7 @@ static inline void *offset_to_ptr(const int *off) >> * bool and also pointer types. >> */ >> #define is_signed_type(type) (((type)(-1)) < (__force type)1) >> +#define is_unsigned_type(type) (!is_signed_type(type)) >> /* >> * This is needed in functions which generate the stack canary, see >> diff --git a/include/linux/overflow.h b/include/linux/overflow.h >> index 1d3be1a2204c..16ae8d912390 100644 >> --- a/include/linux/overflow.h >> +++ b/include/linux/overflow.h >> @@ -128,6 +128,54 @@ static inline bool __must_check >> __must_check_overflow(bool overflow) >> (*_d >> _to_shift) != _a); \ >> })) >> +#define __overflows_type_constexpr(x, T) ( \ >> + is_unsigned_type(typeof(x)) ? \ >> + (x) > type_max(typeof(T)) ? 1 : 0 \ >> + : is_unsigned_type(typeof(T)) ? \ >> + (x) < 0 || (x) > type_max(typeof(T)) ? 1 : 0 \ >> + : (x) < type_min(typeof(T)) || \ >> + (x) > type_max(typeof(T)) ? 1 : 0) > > Syntactically seems fine, but I am little bit lost with indentation. > Shouldn't the last two lines be one tab to the left? Up to you. > Btw, maybe extra helper in_range/between: in_range(x, type_min(T), > type_max(T)) :) > The rest looks OK > Hi Andrzej, As you commented, the indent part seems to need an update. Now this patch [1] has been merged into linux-next as described in the cover letter. So it seems good to fix the indent as a separate update. And the implementations of in_range exist in several places with the same name. [2] In order to avoid name conflicts or parameter mismatches, it is thought that the existing code should be modified first. This part also seems to need to be applied as a separate patch after this patch. [1] https://git.kernel.org/pub/scm/linux/kernel/git/next/linux-next.git/commit/?id=4b21d25bf519c9487935a664886956bb18f04f6d [2] [arch/arm/mm/pageattr.c] static bool in_range(unsigned long start, unsigned long size, unsigned long range_start, unsigned long range_end) [drivers/gpu/drm/arm/display/include/malidp_utils.h] static inline bool in_range(struct malidp_range *rg, u32 v) [drivers/gpu/drm/msm/adreno/a6xx_gmu.c] static int in_range(u32 addr, u32 start, u32 size) [drivers/net/ethernet/chelsio/cxgb3/cxgb3_main.c] static int in_range(int val, int lo, int hi) [drivers/virt/acrn/ioreq.c] static bool in_range(struct acrn_ioreq_range *range, struct acrn_io_request *req) [fs/btrfs/misc.h] #define in_range(b, first, len) ((b) >= (first) && (b) < (first) + (len)) [fs/ext2/balloc.c] #define in_range(b, first, len) ((b) >= (first) && (b) <= (first) + (len) - 1) [fs/ext4/ext4.h] #define in_range(b, first, len) ((b) >= (first) && (b) <= (first) + (len) - 1) [fs/ufs/util.h] #define in_range(b,first,len) ((b)>=(first)&&(b)<(first)+(len)) [net/tipc/core.h] static inline int in_range(u16 val, u16 min, u16 max) [tools/testing/selftests/bpf/progs/get_branch_snapshot.c] static inline bool in_range(__u64 val) Br, G.G. > Reviewed-by: Andrzej Hajda <andrzej.hajda@intel.com> > > Regards > Andrzej > >> + >> +#define __overflows_type(x, T) ({ \ >> + typeof(T) v = 0; \ >> + check_add_overflow((x), v, &v); \ >> +}) >> + >> +/** >> + * overflows_type - helper for checking the overflows between value, >> variables, >> + * or data type >> + * >> + * @n: source constant value or variable to be checked >> + * @T: destination variable or data type proposed to store @x >> + * >> + * Compares the @x expression for whether or not it can safely fit in >> + * the storage of the type in @T. @x and @T can have different types. >> + * If @x is a constant expression, this will also resolve to a constant >> + * expression. >> + * >> + * Returns: true if overflow can occur, false otherwise. >> + */ >> +#define overflows_type(n, T) \ >> + __builtin_choose_expr(__is_constexpr(n), \ >> + __overflows_type_constexpr(n, T), \ >> + __overflows_type(n, T)) >> + >> +/** >> + * castable_to_type - like __same_type(), but also allows for casted >> literals >> + * >> + * @n: variable or constant value >> + * @T: variable or data type >> + * >> + * Unlike the __same_type() macro, this allows a constant value as the >> + * first argument. If this value would not overflow into an assignment >> + * of the second argument's type, it returns true. Otherwise, this falls >> + * back to __same_type(). >> + */ >> +#define castable_to_type(n, T) \ >> + __builtin_choose_expr(__is_constexpr(n), \ >> + !__overflows_type_constexpr(n, T), \ >> + __same_type(n, T)) >> + >> /** >> * size_mul() - Calculate size_t multiplication with saturation at >> SIZE_MAX >> * @factor1: first factor >> diff --git a/lib/Makefile b/lib/Makefile >> index 59bd7c2f793a..889f96b10f12 100644 >> --- a/lib/Makefile >> +++ b/lib/Makefile >> @@ -377,6 +377,7 @@ obj-$(CONFIG_CMDLINE_KUNIT_TEST) += cmdline_kunit.o >> obj-$(CONFIG_SLUB_KUNIT_TEST) += slub_kunit.o >> obj-$(CONFIG_MEMCPY_KUNIT_TEST) += memcpy_kunit.o >> obj-$(CONFIG_IS_SIGNED_TYPE_KUNIT_TEST) += is_signed_type_kunit.o >> +CFLAGS_overflow_kunit.o = $(call cc-disable-warning, >> tautological-constant-out-of-range-compare) >> obj-$(CONFIG_OVERFLOW_KUNIT_TEST) += overflow_kunit.o >> CFLAGS_stackinit_kunit.o += $(call cc-disable-warning, >> switch-unreachable) >> obj-$(CONFIG_STACKINIT_KUNIT_TEST) += stackinit_kunit.o >> diff --git a/lib/overflow_kunit.c b/lib/overflow_kunit.c >> index b8556a2e7bb1..dcd3ba102db6 100644 >> --- a/lib/overflow_kunit.c >> +++ b/lib/overflow_kunit.c >> @@ -736,6 +736,384 @@ static void overflow_size_helpers_test(struct >> kunit *test) >> #undef check_one_size_helper >> } >> +static void overflows_type_test(struct kunit *test) >> +{ >> + int count = 0; >> + unsigned int var; >> + >> +#define __TEST_OVERFLOWS_TYPE(func, arg1, arg2, of) do { \ >> + bool __of = func(arg1, arg2); \ >> + KUNIT_EXPECT_EQ_MSG(test, __of, of, \ >> + "expected " #func "(" #arg1 ", " #arg2 " to%s overflow\n",\ >> + of ? "" : " not"); \ >> + count++; \ >> +} while (0) >> + >> +/* Args are: first type, second type, value, overflow expected */ >> +#define TEST_OVERFLOWS_TYPE(__t1, __t2, v, of) do { \ >> + __t1 t1 = (v); \ >> + __t2 t2; \ >> + __TEST_OVERFLOWS_TYPE(__overflows_type, t1, t2, of); \ >> + __TEST_OVERFLOWS_TYPE(__overflows_type, t1, __t2, of); \ >> + __TEST_OVERFLOWS_TYPE(__overflows_type_constexpr, t1, t2, of); \ >> + __TEST_OVERFLOWS_TYPE(__overflows_type_constexpr, t1, __t2, of);\ >> +} while (0) >> + >> + TEST_OVERFLOWS_TYPE(u8, u8, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u8, u16, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u8, s8, U8_MAX, true); >> + TEST_OVERFLOWS_TYPE(u8, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u8, s8, (u8)S8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u8, s16, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s8, u8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s8, u8, -1, true); >> + TEST_OVERFLOWS_TYPE(s8, u8, S8_MIN, true); >> + TEST_OVERFLOWS_TYPE(s8, u16, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s8, u16, -1, true); >> + TEST_OVERFLOWS_TYPE(s8, u16, S8_MIN, true); >> + TEST_OVERFLOWS_TYPE(s8, u32, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s8, u32, -1, true); >> + TEST_OVERFLOWS_TYPE(s8, u32, S8_MIN, true); >> +#if BITS_PER_LONG == 64 >> + TEST_OVERFLOWS_TYPE(s8, u64, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s8, u64, -1, true); >> + TEST_OVERFLOWS_TYPE(s8, u64, S8_MIN, true); >> +#endif >> + TEST_OVERFLOWS_TYPE(s8, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s8, s8, S8_MIN, false); >> + TEST_OVERFLOWS_TYPE(s8, s16, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s8, s16, S8_MIN, false); >> + TEST_OVERFLOWS_TYPE(u16, u8, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u16, u8, (u16)U8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u16, u8, U16_MAX, true); >> + TEST_OVERFLOWS_TYPE(u16, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u16, s8, (u16)S8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u16, s8, U16_MAX, true); >> + TEST_OVERFLOWS_TYPE(u16, s16, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(u16, s16, (u16)S16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u16, s16, U16_MAX, true); >> + TEST_OVERFLOWS_TYPE(u16, u32, U16_MAX, false); >> + TEST_OVERFLOWS_TYPE(u16, s32, U16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, u8, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, u8, (s16)U8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s16, u8, -1, true); >> + TEST_OVERFLOWS_TYPE(s16, u8, S16_MIN, true); >> + TEST_OVERFLOWS_TYPE(s16, u16, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, u16, -1, true); >> + TEST_OVERFLOWS_TYPE(s16, u16, S16_MIN, true); >> + TEST_OVERFLOWS_TYPE(s16, u32, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, u32, -1, true); >> + TEST_OVERFLOWS_TYPE(s16, u32, S16_MIN, true); >> +#if BITS_PER_LONG == 64 >> + TEST_OVERFLOWS_TYPE(s16, u64, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, u64, -1, true); >> + TEST_OVERFLOWS_TYPE(s16, u64, S16_MIN, true); >> +#endif >> + TEST_OVERFLOWS_TYPE(s16, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, s8, S8_MIN, false); >> + TEST_OVERFLOWS_TYPE(s16, s8, (s16)S8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s16, s8, (s16)S8_MIN - 1, true); >> + TEST_OVERFLOWS_TYPE(s16, s8, S16_MAX, true); >> + TEST_OVERFLOWS_TYPE(s16, s8, S16_MIN, true); >> + TEST_OVERFLOWS_TYPE(s16, s16, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, s16, S16_MIN, false); >> + TEST_OVERFLOWS_TYPE(s16, s32, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s16, s32, S16_MIN, false); >> + TEST_OVERFLOWS_TYPE(u32, u8, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u32, u8, (u32)U8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u32, u8, U32_MAX, true); >> + TEST_OVERFLOWS_TYPE(u32, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u32, s8, (u32)S8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u32, s8, U32_MAX, true); >> + TEST_OVERFLOWS_TYPE(u32, u16, U16_MAX, false); >> + TEST_OVERFLOWS_TYPE(u32, u16, U16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u32, u16, U32_MAX, true); >> + TEST_OVERFLOWS_TYPE(u32, s16, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(u32, s16, (u32)S16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u32, s16, U32_MAX, true); >> + TEST_OVERFLOWS_TYPE(u32, u32, U32_MAX, false); >> + TEST_OVERFLOWS_TYPE(u32, s32, S32_MAX, false); >> + TEST_OVERFLOWS_TYPE(u32, s32, U32_MAX, true); >> + TEST_OVERFLOWS_TYPE(u32, s32, (u32)S32_MAX + 1, true); >> +#if BITS_PER_LONG == 64 >> + TEST_OVERFLOWS_TYPE(u32, u64, U32_MAX, false); >> + TEST_OVERFLOWS_TYPE(u32, s64, U32_MAX, false); >> +#endif >> + TEST_OVERFLOWS_TYPE(s32, u8, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, u8, (s32)U8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s32, u16, S32_MAX, true); >> + TEST_OVERFLOWS_TYPE(s32, u8, -1, true); >> + TEST_OVERFLOWS_TYPE(s32, u8, S32_MIN, true); >> + TEST_OVERFLOWS_TYPE(s32, u16, U16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, u16, (s32)U16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s32, u16, S32_MAX, true); >> + TEST_OVERFLOWS_TYPE(s32, u16, -1, true); >> + TEST_OVERFLOWS_TYPE(s32, u16, S32_MIN, true); >> + TEST_OVERFLOWS_TYPE(s32, u32, S32_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, u32, -1, true); >> + TEST_OVERFLOWS_TYPE(s32, u32, S32_MIN, true); >> +#if BITS_PER_LONG == 64 >> + TEST_OVERFLOWS_TYPE(s32, u64, S32_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, u64, -1, true); >> + TEST_OVERFLOWS_TYPE(s32, u64, S32_MIN, true); >> +#endif >> + TEST_OVERFLOWS_TYPE(s32, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, s8, S8_MIN, false); >> + TEST_OVERFLOWS_TYPE(s32, s8, (s32)S8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s32, s8, (s32)S8_MIN - 1, true); >> + TEST_OVERFLOWS_TYPE(s32, s8, S32_MAX, true); >> + TEST_OVERFLOWS_TYPE(s32, s8, S32_MIN, true); >> + TEST_OVERFLOWS_TYPE(s32, s16, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, s16, S16_MIN, false); >> + TEST_OVERFLOWS_TYPE(s32, s16, (s32)S16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s32, s16, (s32)S16_MIN - 1, true); >> + TEST_OVERFLOWS_TYPE(s32, s16, S32_MAX, true); >> + TEST_OVERFLOWS_TYPE(s32, s16, S32_MIN, true); >> + TEST_OVERFLOWS_TYPE(s32, s32, S32_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, s32, S32_MIN, false); >> +#if BITS_PER_LONG == 64 >> + TEST_OVERFLOWS_TYPE(s32, s64, S32_MAX, false); >> + TEST_OVERFLOWS_TYPE(s32, s64, S32_MIN, false); >> + TEST_OVERFLOWS_TYPE(u64, u8, U64_MAX, true); >> + TEST_OVERFLOWS_TYPE(u64, u8, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, u8, (u64)U8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u64, u16, U64_MAX, true); >> + TEST_OVERFLOWS_TYPE(u64, u16, U16_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, u16, (u64)U16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u64, u32, U64_MAX, true); >> + TEST_OVERFLOWS_TYPE(u64, u32, U32_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, u32, (u64)U32_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u64, u64, U64_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, s8, (u64)S8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u64, s8, U64_MAX, true); >> + TEST_OVERFLOWS_TYPE(u64, s16, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, s16, (u64)S16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u64, s16, U64_MAX, true); >> + TEST_OVERFLOWS_TYPE(u64, s32, S32_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, s32, (u64)S32_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(u64, s32, U64_MAX, true); >> + TEST_OVERFLOWS_TYPE(u64, s64, S64_MAX, false); >> + TEST_OVERFLOWS_TYPE(u64, s64, U64_MAX, true); >> + TEST_OVERFLOWS_TYPE(u64, s64, (u64)S64_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s64, u8, S64_MAX, true); >> + TEST_OVERFLOWS_TYPE(s64, u8, S64_MIN, true); >> + TEST_OVERFLOWS_TYPE(s64, u8, -1, true); >> + TEST_OVERFLOWS_TYPE(s64, u8, U8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, u8, (s64)U8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s64, u16, S64_MAX, true); >> + TEST_OVERFLOWS_TYPE(s64, u16, S64_MIN, true); >> + TEST_OVERFLOWS_TYPE(s64, u16, -1, true); >> + TEST_OVERFLOWS_TYPE(s64, u16, U16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, u16, (s64)U16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s64, u32, S64_MAX, true); >> + TEST_OVERFLOWS_TYPE(s64, u32, S64_MIN, true); >> + TEST_OVERFLOWS_TYPE(s64, u32, -1, true); >> + TEST_OVERFLOWS_TYPE(s64, u32, U32_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, u32, (s64)U32_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s64, u64, S64_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, u64, S64_MIN, true); >> + TEST_OVERFLOWS_TYPE(s64, u64, -1, true); >> + TEST_OVERFLOWS_TYPE(s64, s8, S8_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, s8, S8_MIN, false); >> + TEST_OVERFLOWS_TYPE(s64, s8, (s64)S8_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s64, s8, (s64)S8_MIN - 1, true); >> + TEST_OVERFLOWS_TYPE(s64, s8, S64_MAX, true); >> + TEST_OVERFLOWS_TYPE(s64, s16, S16_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, s16, S16_MIN, false); >> + TEST_OVERFLOWS_TYPE(s64, s16, (s64)S16_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s64, s16, (s64)S16_MIN - 1, true); >> + TEST_OVERFLOWS_TYPE(s64, s16, S64_MAX, true); >> + TEST_OVERFLOWS_TYPE(s64, s32, S32_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, s32, S32_MIN, false); >> + TEST_OVERFLOWS_TYPE(s64, s32, (s64)S32_MAX + 1, true); >> + TEST_OVERFLOWS_TYPE(s64, s32, (s64)S32_MIN - 1, true); >> + TEST_OVERFLOWS_TYPE(s64, s32, S64_MAX, true); >> + TEST_OVERFLOWS_TYPE(s64, s64, S64_MAX, false); >> + TEST_OVERFLOWS_TYPE(s64, s64, S64_MIN, false); >> +#endif >> + >> + /* Check for macro side-effects. */ >> + var = INT_MAX - 1; >> + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, false); >> + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, false); >> + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, true); >> + var = INT_MAX - 1; >> + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, false); >> + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, false); >> + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, true); >> + >> + kunit_info(test, "%d overflows_type() tests finished\n", count); >> +#undef TEST_OVERFLOWS_TYPE >> +#undef __TEST_OVERFLOWS_TYPE >> +} >> + >> +static void same_type_test(struct kunit *test) >> +{ >> + int count = 0; >> + int var; >> + >> +#define TEST_SAME_TYPE(t1, t2, same) do { \ >> + typeof(t1) __t1h = type_max(t1); \ >> + typeof(t1) __t1l = type_min(t1); \ >> + typeof(t2) __t2h = type_max(t2); \ >> + typeof(t2) __t2l = type_min(t2); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(t1, __t1h)); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(t1, __t1l)); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(__t1h, t1)); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(__t1l, t1)); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(t2, __t2h)); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(t2, __t2l)); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(__t2h, t2)); \ >> + KUNIT_EXPECT_EQ(test, true, __same_type(__t2l, t2)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(t1, t2)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(t2, __t1h)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(t2, __t1l)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(__t1h, t2)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(__t1l, t2)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(t1, __t2h)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(t1, __t2l)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(__t2h, t1)); \ >> + KUNIT_EXPECT_EQ(test, same, __same_type(__t2l, t1)); \ >> +} while (0) >> + >> +#if BITS_PER_LONG == 64 >> +# define TEST_SAME_TYPE64(base, t, m) TEST_SAME_TYPE(base, t, m) >> +#else >> +# define TEST_SAME_TYPE64(base, t, m) do { } while (0) >> +#endif >> + >> +#define TEST_TYPE_SETS(base, mu8, mu16, mu32, ms8, ms16, ms32, mu64, >> ms64) \ >> +do { \ >> + TEST_SAME_TYPE(base, u8, mu8); \ >> + TEST_SAME_TYPE(base, u16, mu16); \ >> + TEST_SAME_TYPE(base, u32, mu32); \ >> + TEST_SAME_TYPE(base, s8, ms8); \ >> + TEST_SAME_TYPE(base, s16, ms16); \ >> + TEST_SAME_TYPE(base, s32, ms32); \ >> + TEST_SAME_TYPE64(base, u64, mu64); \ >> + TEST_SAME_TYPE64(base, s64, ms64); \ >> +} while (0) >> + >> + TEST_TYPE_SETS(u8, true, false, false, false, false, false, >> false, false); >> + TEST_TYPE_SETS(u16, false, true, false, false, false, false, >> false, false); >> + TEST_TYPE_SETS(u32, false, false, true, false, false, false, >> false, false); >> + TEST_TYPE_SETS(s8, false, false, false, true, false, false, >> false, false); >> + TEST_TYPE_SETS(s16, false, false, false, false, true, false, >> false, false); >> + TEST_TYPE_SETS(s32, false, false, false, false, false, true, >> false, false); >> +#if BITS_PER_LONG == 64 >> + TEST_TYPE_SETS(u64, false, false, false, false, false, false, >> true, false); >> + TEST_TYPE_SETS(s64, false, false, false, false, false, false, >> false, true); >> +#endif >> + >> + /* Check for macro side-effects. */ >> + var = 4; >> + KUNIT_EXPECT_EQ(test, var, 4); >> + KUNIT_EXPECT_TRUE(test, __same_type(var++, int)); >> + KUNIT_EXPECT_EQ(test, var, 4); >> + KUNIT_EXPECT_TRUE(test, __same_type(int, var++)); >> + KUNIT_EXPECT_EQ(test, var, 4); >> + KUNIT_EXPECT_TRUE(test, __same_type(var++, var++)); >> + KUNIT_EXPECT_EQ(test, var, 4); >> + >> + kunit_info(test, "%d __same_type() tests finished\n", count); >> + >> +#undef TEST_TYPE_SETS >> +#undef TEST_SAME_TYPE64 >> +#undef TEST_SAME_TYPE >> +} >> + >> +static void castable_to_type_test(struct kunit *test) >> +{ >> + int count = 0; >> + >> +#define TEST_CASTABLE_TO_TYPE(arg1, arg2, pass) do { \ >> + bool __pass = castable_to_type(arg1, arg2); \ >> + KUNIT_EXPECT_EQ_MSG(test, __pass, pass, \ >> + "expected castable_to_type(" #arg1 ", " #arg2 ") to%s pass\n",\ >> + pass ? "" : " not"); \ >> + count++; \ >> +} while (0) >> + >> + TEST_CASTABLE_TO_TYPE(16, u8, true); >> + TEST_CASTABLE_TO_TYPE(16, u16, true); >> + TEST_CASTABLE_TO_TYPE(16, u32, true); >> + TEST_CASTABLE_TO_TYPE(16, s8, true); >> + TEST_CASTABLE_TO_TYPE(16, s16, true); >> + TEST_CASTABLE_TO_TYPE(16, s32, true); >> + TEST_CASTABLE_TO_TYPE(-16, s8, true); >> + TEST_CASTABLE_TO_TYPE(-16, s16, true); >> + TEST_CASTABLE_TO_TYPE(-16, s32, true); >> +#if BITS_PER_LONG == 64 >> + TEST_CASTABLE_TO_TYPE(16, u64, true); >> + TEST_CASTABLE_TO_TYPE(-16, s64, true); >> +#endif >> + >> +#define TEST_CASTABLE_TO_TYPE_VAR(width) do { \ >> + u ## width u ## width ## var = 0; \ >> + s ## width s ## width ## var = 0; \ >> + \ >> + /* Constant expressions that fit types. */ \ >> + TEST_CASTABLE_TO_TYPE(type_max(u ## width), u ## width, >> true); \ >> + TEST_CASTABLE_TO_TYPE(type_min(u ## width), u ## width, >> true); \ >> + TEST_CASTABLE_TO_TYPE(type_max(u ## width), u ## width ## var, >> true); \ >> + TEST_CASTABLE_TO_TYPE(type_min(u ## width), u ## width ## var, >> true); \ >> + TEST_CASTABLE_TO_TYPE(type_max(s ## width), s ## width, >> true); \ >> + TEST_CASTABLE_TO_TYPE(type_min(s ## width), s ## width, >> true); \ >> + TEST_CASTABLE_TO_TYPE(type_max(s ## width), s ## width ## var, >> true); \ >> + TEST_CASTABLE_TO_TYPE(type_min(u ## width), s ## width ## var, >> true); \ >> + /* Constant expressions that do not fit types. */ \ >> + TEST_CASTABLE_TO_TYPE(type_max(u ## width), s ## width, >> false); \ >> + TEST_CASTABLE_TO_TYPE(type_max(u ## width), s ## width ## var, >> false); \ >> + TEST_CASTABLE_TO_TYPE(type_min(s ## width), u ## width, >> false); \ >> + TEST_CASTABLE_TO_TYPE(type_min(s ## width), u ## width ## var, >> false); \ >> + /* Non-constant expression with mismatched type. */ \ >> + TEST_CASTABLE_TO_TYPE(s ## width ## var, u ## width, >> false); \ >> + TEST_CASTABLE_TO_TYPE(u ## width ## var, s ## width, >> false); \ >> +} while (0) >> + >> +#define TEST_CASTABLE_TO_TYPE_RANGE(width) do { \ >> + unsigned long big = U ## width ## _MAX; \ >> + signed long small = S ## width ## _MIN; \ >> + u ## width u ## width ## var = 0; \ >> + s ## width s ## width ## var = 0; \ >> + \ >> + /* Constant expression in range. */ \ >> + TEST_CASTABLE_TO_TYPE(U ## width ## _MAX, u ## width, >> true); \ >> + TEST_CASTABLE_TO_TYPE(U ## width ## _MAX, u ## width ## var, >> true); \ >> + TEST_CASTABLE_TO_TYPE(S ## width ## _MIN, s ## width, >> true); \ >> + TEST_CASTABLE_TO_TYPE(S ## width ## _MIN, s ## width ## var, >> true); \ >> + /* Constant expression out of range. */ \ >> + TEST_CASTABLE_TO_TYPE((unsigned long)U ## width ## _MAX + 1, u ## >> width, false); \ >> + TEST_CASTABLE_TO_TYPE((unsigned long)U ## width ## _MAX + 1, u ## >> width ## var, false); \ >> + TEST_CASTABLE_TO_TYPE((signed long)S ## width ## _MIN - 1, s ## >> width, false); \ >> + TEST_CASTABLE_TO_TYPE((signed long)S ## width ## _MIN - 1, s ## >> width ## var, false); \ >> + /* Non-constant expression with mismatched type. */ \ >> + TEST_CASTABLE_TO_TYPE(big, u ## width, false); \ >> + TEST_CASTABLE_TO_TYPE(big, u ## width ## var, false); \ >> + TEST_CASTABLE_TO_TYPE(small, s ## width, false); \ >> + TEST_CASTABLE_TO_TYPE(small, s ## width ## var, false); \ >> +} while (0) >> + >> + TEST_CASTABLE_TO_TYPE_VAR(8); >> + TEST_CASTABLE_TO_TYPE_VAR(16); >> + TEST_CASTABLE_TO_TYPE_VAR(32); >> +#if BITS_PER_LONG == 64 >> + TEST_CASTABLE_TO_TYPE_VAR(64); >> +#endif >> + >> + TEST_CASTABLE_TO_TYPE_RANGE(8); >> + TEST_CASTABLE_TO_TYPE_RANGE(16); >> +#if BITS_PER_LONG == 64 >> + TEST_CASTABLE_TO_TYPE_RANGE(32); >> +#endif >> + kunit_info(test, "%d castable_to_type() tests finished\n", count); >> + >> +#undef TEST_CASTABLE_TO_TYPE_RANGE >> +#undef TEST_CASTABLE_TO_TYPE_VAR >> +#undef TEST_CASTABLE_TO_TYPE >> +} >> + >> static struct kunit_case overflow_test_cases[] = { >> KUNIT_CASE(u8_u8__u8_overflow_test), >> KUNIT_CASE(s8_s8__s8_overflow_test), >> @@ -755,6 +1133,9 @@ static struct kunit_case overflow_test_cases[] = { >> KUNIT_CASE(shift_nonsense_test), >> KUNIT_CASE(overflow_allocation_test), >> KUNIT_CASE(overflow_size_helpers_test), >> + KUNIT_CASE(overflows_type_test), >> + KUNIT_CASE(same_type_test), >> + KUNIT_CASE(castable_to_type_test), >> {} >> }; >
diff --git a/drivers/gpu/drm/i915/i915_user_extensions.c b/drivers/gpu/drm/i915/i915_user_extensions.c index c822d0aafd2d..e3f808372c47 100644 --- a/drivers/gpu/drm/i915/i915_user_extensions.c +++ b/drivers/gpu/drm/i915/i915_user_extensions.c @@ -51,7 +51,7 @@ int i915_user_extensions(struct i915_user_extension __user *ext, return err; if (get_user(next, &ext->next_extension) || - overflows_type(next, ext)) + overflows_type(next, uintptr_t)) return -EFAULT; ext = u64_to_user_ptr(next); diff --git a/drivers/gpu/drm/i915/i915_utils.h b/drivers/gpu/drm/i915/i915_utils.h index b64192d9c7da..2c430c0c3bad 100644 --- a/drivers/gpu/drm/i915/i915_utils.h +++ b/drivers/gpu/drm/i915/i915_utils.h @@ -111,10 +111,6 @@ bool i915_error_injected(void); #define range_overflows_end_t(type, start, size, max) \ range_overflows_end((type)(start), (type)(size), (type)(max)) -/* Note we don't consider signbits :| */ -#define overflows_type(x, T) \ - (sizeof(x) > sizeof(T) && (x) >> BITS_PER_TYPE(T)) - #define ptr_mask_bits(ptr, n) ({ \ unsigned long __v = (unsigned long)(ptr); \ (typeof(ptr))(__v & -BIT(n)); \ diff --git a/include/linux/compiler.h b/include/linux/compiler.h index 973a1bfd7ef5..947a60b801db 100644 --- a/include/linux/compiler.h +++ b/include/linux/compiler.h @@ -236,6 +236,7 @@ static inline void *offset_to_ptr(const int *off) * bool and also pointer types. */ #define is_signed_type(type) (((type)(-1)) < (__force type)1) +#define is_unsigned_type(type) (!is_signed_type(type)) /* * This is needed in functions which generate the stack canary, see diff --git a/include/linux/overflow.h b/include/linux/overflow.h index 1d3be1a2204c..16ae8d912390 100644 --- a/include/linux/overflow.h +++ b/include/linux/overflow.h @@ -128,6 +128,54 @@ static inline bool __must_check __must_check_overflow(bool overflow) (*_d >> _to_shift) != _a); \ })) +#define __overflows_type_constexpr(x, T) ( \ + is_unsigned_type(typeof(x)) ? \ + (x) > type_max(typeof(T)) ? 1 : 0 \ + : is_unsigned_type(typeof(T)) ? \ + (x) < 0 || (x) > type_max(typeof(T)) ? 1 : 0 \ + : (x) < type_min(typeof(T)) || \ + (x) > type_max(typeof(T)) ? 1 : 0) + +#define __overflows_type(x, T) ({ \ + typeof(T) v = 0; \ + check_add_overflow((x), v, &v); \ +}) + +/** + * overflows_type - helper for checking the overflows between value, variables, + * or data type + * + * @n: source constant value or variable to be checked + * @T: destination variable or data type proposed to store @x + * + * Compares the @x expression for whether or not it can safely fit in + * the storage of the type in @T. @x and @T can have different types. + * If @x is a constant expression, this will also resolve to a constant + * expression. + * + * Returns: true if overflow can occur, false otherwise. + */ +#define overflows_type(n, T) \ + __builtin_choose_expr(__is_constexpr(n), \ + __overflows_type_constexpr(n, T), \ + __overflows_type(n, T)) + +/** + * castable_to_type - like __same_type(), but also allows for casted literals + * + * @n: variable or constant value + * @T: variable or data type + * + * Unlike the __same_type() macro, this allows a constant value as the + * first argument. If this value would not overflow into an assignment + * of the second argument's type, it returns true. Otherwise, this falls + * back to __same_type(). + */ +#define castable_to_type(n, T) \ + __builtin_choose_expr(__is_constexpr(n), \ + !__overflows_type_constexpr(n, T), \ + __same_type(n, T)) + /** * size_mul() - Calculate size_t multiplication with saturation at SIZE_MAX * @factor1: first factor diff --git a/lib/Makefile b/lib/Makefile index 59bd7c2f793a..889f96b10f12 100644 --- a/lib/Makefile +++ b/lib/Makefile @@ -377,6 +377,7 @@ obj-$(CONFIG_CMDLINE_KUNIT_TEST) += cmdline_kunit.o obj-$(CONFIG_SLUB_KUNIT_TEST) += slub_kunit.o obj-$(CONFIG_MEMCPY_KUNIT_TEST) += memcpy_kunit.o obj-$(CONFIG_IS_SIGNED_TYPE_KUNIT_TEST) += is_signed_type_kunit.o +CFLAGS_overflow_kunit.o = $(call cc-disable-warning, tautological-constant-out-of-range-compare) obj-$(CONFIG_OVERFLOW_KUNIT_TEST) += overflow_kunit.o CFLAGS_stackinit_kunit.o += $(call cc-disable-warning, switch-unreachable) obj-$(CONFIG_STACKINIT_KUNIT_TEST) += stackinit_kunit.o diff --git a/lib/overflow_kunit.c b/lib/overflow_kunit.c index b8556a2e7bb1..dcd3ba102db6 100644 --- a/lib/overflow_kunit.c +++ b/lib/overflow_kunit.c @@ -736,6 +736,384 @@ static void overflow_size_helpers_test(struct kunit *test) #undef check_one_size_helper } +static void overflows_type_test(struct kunit *test) +{ + int count = 0; + unsigned int var; + +#define __TEST_OVERFLOWS_TYPE(func, arg1, arg2, of) do { \ + bool __of = func(arg1, arg2); \ + KUNIT_EXPECT_EQ_MSG(test, __of, of, \ + "expected " #func "(" #arg1 ", " #arg2 " to%s overflow\n",\ + of ? "" : " not"); \ + count++; \ +} while (0) + +/* Args are: first type, second type, value, overflow expected */ +#define TEST_OVERFLOWS_TYPE(__t1, __t2, v, of) do { \ + __t1 t1 = (v); \ + __t2 t2; \ + __TEST_OVERFLOWS_TYPE(__overflows_type, t1, t2, of); \ + __TEST_OVERFLOWS_TYPE(__overflows_type, t1, __t2, of); \ + __TEST_OVERFLOWS_TYPE(__overflows_type_constexpr, t1, t2, of); \ + __TEST_OVERFLOWS_TYPE(__overflows_type_constexpr, t1, __t2, of);\ +} while (0) + + TEST_OVERFLOWS_TYPE(u8, u8, U8_MAX, false); + TEST_OVERFLOWS_TYPE(u8, u16, U8_MAX, false); + TEST_OVERFLOWS_TYPE(u8, s8, U8_MAX, true); + TEST_OVERFLOWS_TYPE(u8, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(u8, s8, (u8)S8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u8, s16, U8_MAX, false); + TEST_OVERFLOWS_TYPE(s8, u8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s8, u8, -1, true); + TEST_OVERFLOWS_TYPE(s8, u8, S8_MIN, true); + TEST_OVERFLOWS_TYPE(s8, u16, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s8, u16, -1, true); + TEST_OVERFLOWS_TYPE(s8, u16, S8_MIN, true); + TEST_OVERFLOWS_TYPE(s8, u32, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s8, u32, -1, true); + TEST_OVERFLOWS_TYPE(s8, u32, S8_MIN, true); +#if BITS_PER_LONG == 64 + TEST_OVERFLOWS_TYPE(s8, u64, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s8, u64, -1, true); + TEST_OVERFLOWS_TYPE(s8, u64, S8_MIN, true); +#endif + TEST_OVERFLOWS_TYPE(s8, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s8, s8, S8_MIN, false); + TEST_OVERFLOWS_TYPE(s8, s16, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s8, s16, S8_MIN, false); + TEST_OVERFLOWS_TYPE(u16, u8, U8_MAX, false); + TEST_OVERFLOWS_TYPE(u16, u8, (u16)U8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u16, u8, U16_MAX, true); + TEST_OVERFLOWS_TYPE(u16, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(u16, s8, (u16)S8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u16, s8, U16_MAX, true); + TEST_OVERFLOWS_TYPE(u16, s16, S16_MAX, false); + TEST_OVERFLOWS_TYPE(u16, s16, (u16)S16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u16, s16, U16_MAX, true); + TEST_OVERFLOWS_TYPE(u16, u32, U16_MAX, false); + TEST_OVERFLOWS_TYPE(u16, s32, U16_MAX, false); + TEST_OVERFLOWS_TYPE(s16, u8, U8_MAX, false); + TEST_OVERFLOWS_TYPE(s16, u8, (s16)U8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s16, u8, -1, true); + TEST_OVERFLOWS_TYPE(s16, u8, S16_MIN, true); + TEST_OVERFLOWS_TYPE(s16, u16, S16_MAX, false); + TEST_OVERFLOWS_TYPE(s16, u16, -1, true); + TEST_OVERFLOWS_TYPE(s16, u16, S16_MIN, true); + TEST_OVERFLOWS_TYPE(s16, u32, S16_MAX, false); + TEST_OVERFLOWS_TYPE(s16, u32, -1, true); + TEST_OVERFLOWS_TYPE(s16, u32, S16_MIN, true); +#if BITS_PER_LONG == 64 + TEST_OVERFLOWS_TYPE(s16, u64, S16_MAX, false); + TEST_OVERFLOWS_TYPE(s16, u64, -1, true); + TEST_OVERFLOWS_TYPE(s16, u64, S16_MIN, true); +#endif + TEST_OVERFLOWS_TYPE(s16, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s16, s8, S8_MIN, false); + TEST_OVERFLOWS_TYPE(s16, s8, (s16)S8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s16, s8, (s16)S8_MIN - 1, true); + TEST_OVERFLOWS_TYPE(s16, s8, S16_MAX, true); + TEST_OVERFLOWS_TYPE(s16, s8, S16_MIN, true); + TEST_OVERFLOWS_TYPE(s16, s16, S16_MAX, false); + TEST_OVERFLOWS_TYPE(s16, s16, S16_MIN, false); + TEST_OVERFLOWS_TYPE(s16, s32, S16_MAX, false); + TEST_OVERFLOWS_TYPE(s16, s32, S16_MIN, false); + TEST_OVERFLOWS_TYPE(u32, u8, U8_MAX, false); + TEST_OVERFLOWS_TYPE(u32, u8, (u32)U8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u32, u8, U32_MAX, true); + TEST_OVERFLOWS_TYPE(u32, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(u32, s8, (u32)S8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u32, s8, U32_MAX, true); + TEST_OVERFLOWS_TYPE(u32, u16, U16_MAX, false); + TEST_OVERFLOWS_TYPE(u32, u16, U16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u32, u16, U32_MAX, true); + TEST_OVERFLOWS_TYPE(u32, s16, S16_MAX, false); + TEST_OVERFLOWS_TYPE(u32, s16, (u32)S16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u32, s16, U32_MAX, true); + TEST_OVERFLOWS_TYPE(u32, u32, U32_MAX, false); + TEST_OVERFLOWS_TYPE(u32, s32, S32_MAX, false); + TEST_OVERFLOWS_TYPE(u32, s32, U32_MAX, true); + TEST_OVERFLOWS_TYPE(u32, s32, (u32)S32_MAX + 1, true); +#if BITS_PER_LONG == 64 + TEST_OVERFLOWS_TYPE(u32, u64, U32_MAX, false); + TEST_OVERFLOWS_TYPE(u32, s64, U32_MAX, false); +#endif + TEST_OVERFLOWS_TYPE(s32, u8, U8_MAX, false); + TEST_OVERFLOWS_TYPE(s32, u8, (s32)U8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s32, u16, S32_MAX, true); + TEST_OVERFLOWS_TYPE(s32, u8, -1, true); + TEST_OVERFLOWS_TYPE(s32, u8, S32_MIN, true); + TEST_OVERFLOWS_TYPE(s32, u16, U16_MAX, false); + TEST_OVERFLOWS_TYPE(s32, u16, (s32)U16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s32, u16, S32_MAX, true); + TEST_OVERFLOWS_TYPE(s32, u16, -1, true); + TEST_OVERFLOWS_TYPE(s32, u16, S32_MIN, true); + TEST_OVERFLOWS_TYPE(s32, u32, S32_MAX, false); + TEST_OVERFLOWS_TYPE(s32, u32, -1, true); + TEST_OVERFLOWS_TYPE(s32, u32, S32_MIN, true); +#if BITS_PER_LONG == 64 + TEST_OVERFLOWS_TYPE(s32, u64, S32_MAX, false); + TEST_OVERFLOWS_TYPE(s32, u64, -1, true); + TEST_OVERFLOWS_TYPE(s32, u64, S32_MIN, true); +#endif + TEST_OVERFLOWS_TYPE(s32, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s32, s8, S8_MIN, false); + TEST_OVERFLOWS_TYPE(s32, s8, (s32)S8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s32, s8, (s32)S8_MIN - 1, true); + TEST_OVERFLOWS_TYPE(s32, s8, S32_MAX, true); + TEST_OVERFLOWS_TYPE(s32, s8, S32_MIN, true); + TEST_OVERFLOWS_TYPE(s32, s16, S16_MAX, false); + TEST_OVERFLOWS_TYPE(s32, s16, S16_MIN, false); + TEST_OVERFLOWS_TYPE(s32, s16, (s32)S16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s32, s16, (s32)S16_MIN - 1, true); + TEST_OVERFLOWS_TYPE(s32, s16, S32_MAX, true); + TEST_OVERFLOWS_TYPE(s32, s16, S32_MIN, true); + TEST_OVERFLOWS_TYPE(s32, s32, S32_MAX, false); + TEST_OVERFLOWS_TYPE(s32, s32, S32_MIN, false); +#if BITS_PER_LONG == 64 + TEST_OVERFLOWS_TYPE(s32, s64, S32_MAX, false); + TEST_OVERFLOWS_TYPE(s32, s64, S32_MIN, false); + TEST_OVERFLOWS_TYPE(u64, u8, U64_MAX, true); + TEST_OVERFLOWS_TYPE(u64, u8, U8_MAX, false); + TEST_OVERFLOWS_TYPE(u64, u8, (u64)U8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u64, u16, U64_MAX, true); + TEST_OVERFLOWS_TYPE(u64, u16, U16_MAX, false); + TEST_OVERFLOWS_TYPE(u64, u16, (u64)U16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u64, u32, U64_MAX, true); + TEST_OVERFLOWS_TYPE(u64, u32, U32_MAX, false); + TEST_OVERFLOWS_TYPE(u64, u32, (u64)U32_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u64, u64, U64_MAX, false); + TEST_OVERFLOWS_TYPE(u64, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(u64, s8, (u64)S8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u64, s8, U64_MAX, true); + TEST_OVERFLOWS_TYPE(u64, s16, S16_MAX, false); + TEST_OVERFLOWS_TYPE(u64, s16, (u64)S16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u64, s16, U64_MAX, true); + TEST_OVERFLOWS_TYPE(u64, s32, S32_MAX, false); + TEST_OVERFLOWS_TYPE(u64, s32, (u64)S32_MAX + 1, true); + TEST_OVERFLOWS_TYPE(u64, s32, U64_MAX, true); + TEST_OVERFLOWS_TYPE(u64, s64, S64_MAX, false); + TEST_OVERFLOWS_TYPE(u64, s64, U64_MAX, true); + TEST_OVERFLOWS_TYPE(u64, s64, (u64)S64_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s64, u8, S64_MAX, true); + TEST_OVERFLOWS_TYPE(s64, u8, S64_MIN, true); + TEST_OVERFLOWS_TYPE(s64, u8, -1, true); + TEST_OVERFLOWS_TYPE(s64, u8, U8_MAX, false); + TEST_OVERFLOWS_TYPE(s64, u8, (s64)U8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s64, u16, S64_MAX, true); + TEST_OVERFLOWS_TYPE(s64, u16, S64_MIN, true); + TEST_OVERFLOWS_TYPE(s64, u16, -1, true); + TEST_OVERFLOWS_TYPE(s64, u16, U16_MAX, false); + TEST_OVERFLOWS_TYPE(s64, u16, (s64)U16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s64, u32, S64_MAX, true); + TEST_OVERFLOWS_TYPE(s64, u32, S64_MIN, true); + TEST_OVERFLOWS_TYPE(s64, u32, -1, true); + TEST_OVERFLOWS_TYPE(s64, u32, U32_MAX, false); + TEST_OVERFLOWS_TYPE(s64, u32, (s64)U32_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s64, u64, S64_MAX, false); + TEST_OVERFLOWS_TYPE(s64, u64, S64_MIN, true); + TEST_OVERFLOWS_TYPE(s64, u64, -1, true); + TEST_OVERFLOWS_TYPE(s64, s8, S8_MAX, false); + TEST_OVERFLOWS_TYPE(s64, s8, S8_MIN, false); + TEST_OVERFLOWS_TYPE(s64, s8, (s64)S8_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s64, s8, (s64)S8_MIN - 1, true); + TEST_OVERFLOWS_TYPE(s64, s8, S64_MAX, true); + TEST_OVERFLOWS_TYPE(s64, s16, S16_MAX, false); + TEST_OVERFLOWS_TYPE(s64, s16, S16_MIN, false); + TEST_OVERFLOWS_TYPE(s64, s16, (s64)S16_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s64, s16, (s64)S16_MIN - 1, true); + TEST_OVERFLOWS_TYPE(s64, s16, S64_MAX, true); + TEST_OVERFLOWS_TYPE(s64, s32, S32_MAX, false); + TEST_OVERFLOWS_TYPE(s64, s32, S32_MIN, false); + TEST_OVERFLOWS_TYPE(s64, s32, (s64)S32_MAX + 1, true); + TEST_OVERFLOWS_TYPE(s64, s32, (s64)S32_MIN - 1, true); + TEST_OVERFLOWS_TYPE(s64, s32, S64_MAX, true); + TEST_OVERFLOWS_TYPE(s64, s64, S64_MAX, false); + TEST_OVERFLOWS_TYPE(s64, s64, S64_MIN, false); +#endif + + /* Check for macro side-effects. */ + var = INT_MAX - 1; + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, false); + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, false); + __TEST_OVERFLOWS_TYPE(__overflows_type, var++, int, true); + var = INT_MAX - 1; + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, false); + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, false); + __TEST_OVERFLOWS_TYPE(overflows_type, var++, int, true); + + kunit_info(test, "%d overflows_type() tests finished\n", count); +#undef TEST_OVERFLOWS_TYPE +#undef __TEST_OVERFLOWS_TYPE +} + +static void same_type_test(struct kunit *test) +{ + int count = 0; + int var; + +#define TEST_SAME_TYPE(t1, t2, same) do { \ + typeof(t1) __t1h = type_max(t1); \ + typeof(t1) __t1l = type_min(t1); \ + typeof(t2) __t2h = type_max(t2); \ + typeof(t2) __t2l = type_min(t2); \ + KUNIT_EXPECT_EQ(test, true, __same_type(t1, __t1h)); \ + KUNIT_EXPECT_EQ(test, true, __same_type(t1, __t1l)); \ + KUNIT_EXPECT_EQ(test, true, __same_type(__t1h, t1)); \ + KUNIT_EXPECT_EQ(test, true, __same_type(__t1l, t1)); \ + KUNIT_EXPECT_EQ(test, true, __same_type(t2, __t2h)); \ + KUNIT_EXPECT_EQ(test, true, __same_type(t2, __t2l)); \ + KUNIT_EXPECT_EQ(test, true, __same_type(__t2h, t2)); \ + KUNIT_EXPECT_EQ(test, true, __same_type(__t2l, t2)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(t1, t2)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(t2, __t1h)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(t2, __t1l)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(__t1h, t2)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(__t1l, t2)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(t1, __t2h)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(t1, __t2l)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(__t2h, t1)); \ + KUNIT_EXPECT_EQ(test, same, __same_type(__t2l, t1)); \ +} while (0) + +#if BITS_PER_LONG == 64 +# define TEST_SAME_TYPE64(base, t, m) TEST_SAME_TYPE(base, t, m) +#else +# define TEST_SAME_TYPE64(base, t, m) do { } while (0) +#endif + +#define TEST_TYPE_SETS(base, mu8, mu16, mu32, ms8, ms16, ms32, mu64, ms64) \ +do { \ + TEST_SAME_TYPE(base, u8, mu8); \ + TEST_SAME_TYPE(base, u16, mu16); \ + TEST_SAME_TYPE(base, u32, mu32); \ + TEST_SAME_TYPE(base, s8, ms8); \ + TEST_SAME_TYPE(base, s16, ms16); \ + TEST_SAME_TYPE(base, s32, ms32); \ + TEST_SAME_TYPE64(base, u64, mu64); \ + TEST_SAME_TYPE64(base, s64, ms64); \ +} while (0) + + TEST_TYPE_SETS(u8, true, false, false, false, false, false, false, false); + TEST_TYPE_SETS(u16, false, true, false, false, false, false, false, false); + TEST_TYPE_SETS(u32, false, false, true, false, false, false, false, false); + TEST_TYPE_SETS(s8, false, false, false, true, false, false, false, false); + TEST_TYPE_SETS(s16, false, false, false, false, true, false, false, false); + TEST_TYPE_SETS(s32, false, false, false, false, false, true, false, false); +#if BITS_PER_LONG == 64 + TEST_TYPE_SETS(u64, false, false, false, false, false, false, true, false); + TEST_TYPE_SETS(s64, false, false, false, false, false, false, false, true); +#endif + + /* Check for macro side-effects. */ + var = 4; + KUNIT_EXPECT_EQ(test, var, 4); + KUNIT_EXPECT_TRUE(test, __same_type(var++, int)); + KUNIT_EXPECT_EQ(test, var, 4); + KUNIT_EXPECT_TRUE(test, __same_type(int, var++)); + KUNIT_EXPECT_EQ(test, var, 4); + KUNIT_EXPECT_TRUE(test, __same_type(var++, var++)); + KUNIT_EXPECT_EQ(test, var, 4); + + kunit_info(test, "%d __same_type() tests finished\n", count); + +#undef TEST_TYPE_SETS +#undef TEST_SAME_TYPE64 +#undef TEST_SAME_TYPE +} + +static void castable_to_type_test(struct kunit *test) +{ + int count = 0; + +#define TEST_CASTABLE_TO_TYPE(arg1, arg2, pass) do { \ + bool __pass = castable_to_type(arg1, arg2); \ + KUNIT_EXPECT_EQ_MSG(test, __pass, pass, \ + "expected castable_to_type(" #arg1 ", " #arg2 ") to%s pass\n",\ + pass ? "" : " not"); \ + count++; \ +} while (0) + + TEST_CASTABLE_TO_TYPE(16, u8, true); + TEST_CASTABLE_TO_TYPE(16, u16, true); + TEST_CASTABLE_TO_TYPE(16, u32, true); + TEST_CASTABLE_TO_TYPE(16, s8, true); + TEST_CASTABLE_TO_TYPE(16, s16, true); + TEST_CASTABLE_TO_TYPE(16, s32, true); + TEST_CASTABLE_TO_TYPE(-16, s8, true); + TEST_CASTABLE_TO_TYPE(-16, s16, true); + TEST_CASTABLE_TO_TYPE(-16, s32, true); +#if BITS_PER_LONG == 64 + TEST_CASTABLE_TO_TYPE(16, u64, true); + TEST_CASTABLE_TO_TYPE(-16, s64, true); +#endif + +#define TEST_CASTABLE_TO_TYPE_VAR(width) do { \ + u ## width u ## width ## var = 0; \ + s ## width s ## width ## var = 0; \ + \ + /* Constant expressions that fit types. */ \ + TEST_CASTABLE_TO_TYPE(type_max(u ## width), u ## width, true); \ + TEST_CASTABLE_TO_TYPE(type_min(u ## width), u ## width, true); \ + TEST_CASTABLE_TO_TYPE(type_max(u ## width), u ## width ## var, true); \ + TEST_CASTABLE_TO_TYPE(type_min(u ## width), u ## width ## var, true); \ + TEST_CASTABLE_TO_TYPE(type_max(s ## width), s ## width, true); \ + TEST_CASTABLE_TO_TYPE(type_min(s ## width), s ## width, true); \ + TEST_CASTABLE_TO_TYPE(type_max(s ## width), s ## width ## var, true); \ + TEST_CASTABLE_TO_TYPE(type_min(u ## width), s ## width ## var, true); \ + /* Constant expressions that do not fit types. */ \ + TEST_CASTABLE_TO_TYPE(type_max(u ## width), s ## width, false); \ + TEST_CASTABLE_TO_TYPE(type_max(u ## width), s ## width ## var, false); \ + TEST_CASTABLE_TO_TYPE(type_min(s ## width), u ## width, false); \ + TEST_CASTABLE_TO_TYPE(type_min(s ## width), u ## width ## var, false); \ + /* Non-constant expression with mismatched type. */ \ + TEST_CASTABLE_TO_TYPE(s ## width ## var, u ## width, false); \ + TEST_CASTABLE_TO_TYPE(u ## width ## var, s ## width, false); \ +} while (0) + +#define TEST_CASTABLE_TO_TYPE_RANGE(width) do { \ + unsigned long big = U ## width ## _MAX; \ + signed long small = S ## width ## _MIN; \ + u ## width u ## width ## var = 0; \ + s ## width s ## width ## var = 0; \ + \ + /* Constant expression in range. */ \ + TEST_CASTABLE_TO_TYPE(U ## width ## _MAX, u ## width, true); \ + TEST_CASTABLE_TO_TYPE(U ## width ## _MAX, u ## width ## var, true); \ + TEST_CASTABLE_TO_TYPE(S ## width ## _MIN, s ## width, true); \ + TEST_CASTABLE_TO_TYPE(S ## width ## _MIN, s ## width ## var, true); \ + /* Constant expression out of range. */ \ + TEST_CASTABLE_TO_TYPE((unsigned long)U ## width ## _MAX + 1, u ## width, false); \ + TEST_CASTABLE_TO_TYPE((unsigned long)U ## width ## _MAX + 1, u ## width ## var, false); \ + TEST_CASTABLE_TO_TYPE((signed long)S ## width ## _MIN - 1, s ## width, false); \ + TEST_CASTABLE_TO_TYPE((signed long)S ## width ## _MIN - 1, s ## width ## var, false); \ + /* Non-constant expression with mismatched type. */ \ + TEST_CASTABLE_TO_TYPE(big, u ## width, false); \ + TEST_CASTABLE_TO_TYPE(big, u ## width ## var, false); \ + TEST_CASTABLE_TO_TYPE(small, s ## width, false); \ + TEST_CASTABLE_TO_TYPE(small, s ## width ## var, false); \ +} while (0) + + TEST_CASTABLE_TO_TYPE_VAR(8); + TEST_CASTABLE_TO_TYPE_VAR(16); + TEST_CASTABLE_TO_TYPE_VAR(32); +#if BITS_PER_LONG == 64 + TEST_CASTABLE_TO_TYPE_VAR(64); +#endif + + TEST_CASTABLE_TO_TYPE_RANGE(8); + TEST_CASTABLE_TO_TYPE_RANGE(16); +#if BITS_PER_LONG == 64 + TEST_CASTABLE_TO_TYPE_RANGE(32); +#endif + kunit_info(test, "%d castable_to_type() tests finished\n", count); + +#undef TEST_CASTABLE_TO_TYPE_RANGE +#undef TEST_CASTABLE_TO_TYPE_VAR +#undef TEST_CASTABLE_TO_TYPE +} + static struct kunit_case overflow_test_cases[] = { KUNIT_CASE(u8_u8__u8_overflow_test), KUNIT_CASE(s8_s8__s8_overflow_test), @@ -755,6 +1133,9 @@ static struct kunit_case overflow_test_cases[] = { KUNIT_CASE(shift_nonsense_test), KUNIT_CASE(overflow_allocation_test), KUNIT_CASE(overflow_size_helpers_test), + KUNIT_CASE(overflows_type_test), + KUNIT_CASE(same_type_test), + KUNIT_CASE(castable_to_type_test), {} };