[<prev] [next>] [<thread-prev] [thread-next>] [day] [month] [year] [list]
Message-Id: <20210801194000.3646303-8-liambeguin@gmail.com>
Date: Sun, 1 Aug 2021 15:39:54 -0400
From: Liam Beguin <liambeguin@...il.com>
To: liambeguin@...il.com, peda@...ntia.se, jic23@...nel.org,
lars@...afoo.de, pmeerw@...erw.net
Cc: linux-kernel@...r.kernel.org, linux-iio@...r.kernel.org,
devicetree@...r.kernel.org, robh+dt@...nel.org
Subject: [PATCH v7 07/13] iio: afe: rescale: reduce risk of integer overflow
From: Liam Beguin <lvb@...hos.com>
Reduce the risk of integer overflow by doing the scale calculation with
64bit integers and looking for a Greatest Common Divider for both parts
of the fractional value when required.
Signed-off-by: Liam Beguin <lvb@...hos.com>
---
drivers/iio/afe/iio-rescale.c | 15 ++++++++++++---
1 file changed, 12 insertions(+), 3 deletions(-)
diff --git a/drivers/iio/afe/iio-rescale.c b/drivers/iio/afe/iio-rescale.c
index 623744da269a..abd7ad73d1ce 100644
--- a/drivers/iio/afe/iio-rescale.c
+++ b/drivers/iio/afe/iio-rescale.c
@@ -21,15 +21,24 @@
int rescale_process_scale(struct rescale *rescale, int scale_type,
int *val, int *val2)
{
- unsigned long long tmp;
+ s64 tmp, tmp2;
+ u32 factor;
u32 mult;
u32 rem;
u32 neg;
switch (scale_type) {
case IIO_VAL_FRACTIONAL:
- *val *= rescale->numerator;
- *val2 *= rescale->denominator;
+ if (check_mul_overflow(*val, rescale->numerator, (s32 *)&tmp) ||
+ check_mul_overflow(*val2, rescale->denominator, (s32 *)&tmp2)) {
+ tmp = (s64)*val * rescale->numerator;
+ tmp2 = (s64)*val2 * rescale->denominator;
+ factor = gcd(abs(tmp), abs(tmp2));
+ tmp = div_s64(tmp, factor);
+ tmp2 = div_s64(tmp2, factor);
+ }
+ *val = tmp;
+ *val2 = tmp2;
return scale_type;
case IIO_VAL_INT:
*val *= rescale->numerator;
--
2.30.1.489.g328c10930387
Powered by blists - more mailing lists