ffmpeg/tests/checkasm/llauddsp.c
Andreas Rheinhardt dff0a1557e tests/checkasm/llauddsp: Avoid UB integer overflow
The only multiplicators used in scalarproduct_and_madd_*
are -1, 0 and +1. Yet it is of type int and the checkasm
test uses the complete range of int for it, leading to overflows
that don't happen for actual users.

Fix this by using a more reasonable range for mul: Given
that it is used in v1[i] += v3[i] * mul with v1 being
a 16bit integer, it makes no sense to use values for mul
that don't fit into 16bit.

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
2024-05-17 13:16:58 +02:00

116 lines
3.5 KiB
C

/*
* Copyright (c) 2016 Alexandra Hájková
*
* This file is part of FFmpeg.
*
* FFmpeg is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* FFmpeg is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with FFmpeg; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include <string.h>
#include "libavutil/common.h"
#include "libavutil/intreadwrite.h"
#include "libavutil/mem.h"
#include "libavutil/mem_internal.h"
#include "libavcodec/lossless_audiodsp.h"
#include "checkasm.h"
#define randomize_buf(buf, len) \
do { \
for (int i = 0; i < len; i++) \
buf[i] = rnd(); \
} while (0)
static void check_scalarproduct_and_madd_int16(LLAudDSPContext *c)
{
#define BUF_SIZE 1088 // multiple of 16
LOCAL_ALIGNED_16(int16_t, v1, [BUF_SIZE]);
LOCAL_ALIGNED_16(int16_t, v2, [BUF_SIZE]);
LOCAL_ALIGNED_16(int16_t, v3, [BUF_SIZE]);
int mul;
declare_func(int32_t, int16_t *, const int16_t *, const int16_t *,
int, int);
randomize_buf(v1, BUF_SIZE);
randomize_buf(v2, BUF_SIZE);
randomize_buf(v3, BUF_SIZE);
mul = (int16_t)rnd();
if (check_func(c->scalarproduct_and_madd_int16,
"scalarproduct_and_madd_int16")) {
LOCAL_ALIGNED_16(int16_t, dst0, [BUF_SIZE]);
LOCAL_ALIGNED_16(int16_t, dst1, [BUF_SIZE]);
int ref, val;
memcpy(dst0, v1, sizeof (*dst0) * BUF_SIZE);
memcpy(dst1, v1, sizeof (*dst1) * BUF_SIZE);
ref = call_ref(dst0, v2, v3, BUF_SIZE, mul);
val = call_new(dst1, v2, v3, BUF_SIZE, mul);
if (memcmp(dst0, dst1, sizeof (*dst0) * BUF_SIZE) != 0 || ref != val)
fail();
bench_new(v1, v2, v3, BUF_SIZE, mul);
}
report("scalarproduct_and_madd_int16");
}
static void check_scalarproduct_and_madd_int32(LLAudDSPContext *c)
{
#define BUF_SIZE 1088 // multiple of 16
LOCAL_ALIGNED_16(int16_t, v1, [BUF_SIZE]);
LOCAL_ALIGNED_16(int32_t, v2, [BUF_SIZE]);
LOCAL_ALIGNED_16(int16_t, v3, [BUF_SIZE]);
int mul;
declare_func(int32_t, int16_t *, const int32_t *, const int16_t *,
int, int);
randomize_buf(v1, BUF_SIZE);
randomize_buf(v2, BUF_SIZE);
randomize_buf(v3, BUF_SIZE);
mul = (int16_t)rnd();
if (check_func(c->scalarproduct_and_madd_int32,
"scalarproduct_and_madd_int32")) {
LOCAL_ALIGNED_16(int16_t, dst0, [BUF_SIZE]);
LOCAL_ALIGNED_16(int16_t, dst1, [BUF_SIZE]);
int ref, val;
memcpy(dst0, v1, sizeof (*dst0) * BUF_SIZE);
memcpy(dst1, v1, sizeof (*dst1) * BUF_SIZE);
ref = call_ref(dst0, v2, v3, BUF_SIZE, mul);
val = call_new(dst1, v2, v3, BUF_SIZE, mul);
if (memcmp(dst0, dst1, sizeof (*dst0) * BUF_SIZE) != 0 || ref != val)
fail();
bench_new(v1, v2, v3, BUF_SIZE, mul);
}
report("scalarproduct_and_madd_int32");
}
void checkasm_check_llauddsp(void)
{
LLAudDSPContext c;
ff_llauddsp_init(&c);
check_scalarproduct_and_madd_int16(&c);
check_scalarproduct_and_madd_int32(&c);
}