From bb1aad091d59e0d7a615c98b6d040e9d03cfb259 Mon Sep 17 00:00:00 2001 From: Dale Curtis Date: Thu, 30 Apr 2020 15:16:31 -0700 Subject: [PATCH 1/2] Add saturated add/sub operations for int64_t. Many places are using their own custom code for handling overflow around timestamps or other int64_t values. There are enough of these now that having some common saturated math functions seems sound. Signed-off-by: Dale Curtis --- libavutil/common.h | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) diff --git a/libavutil/common.h b/libavutil/common.h index 142ff9abe7..e926e7cb02 100644 --- a/libavutil/common.h +++ b/libavutil/common.h @@ -291,6 +291,36 @@ static av_always_inline int av_sat_dsub32_c(int a, int b) return av_sat_sub32(a, av_sat_add32(b, b)); } +/** + * Add two signed 64-bit values with saturation. + * + * @param a one value + * @param b another value + * @return sum with signed saturation + */ +static int64_t av_sat_add64_c(int64_t a, int64_t b) { + if (b >= 0 && a >= INT64_MAX - b) + return INT64_MAX; + if (b <= 0 && a <= INT64_MIN - b) + return INT64_MIN; + return a + b; +} + +/** + * Subtract two signed 64-bit values with saturation. + * + * @param a one value + * @param b another value + * @return difference with signed saturation + */ +static int64_t av_sat_sub64_c(int64_t a, int64_t b) { + if (b <= 0 && a >= INT64_MAX + b) { + return INT64_MAX; + if (b >= 0 && a <= INT64_MIN + b) { + return INT64_MIN; + return a - b; +} + /** * Clip a float value into the amin-amax range. * @param a value to clip @@ -545,6 +575,12 @@ static av_always_inline av_const int av_parity_c(uint32_t v) #ifndef av_sat_dsub32 # define av_sat_dsub32 av_sat_dsub32_c #endif +#ifndef av_sat_add64 +# define av_sat_add64 av_sat_add64_c +#endif +#ifndef av_sat_sub64 +# define av_sat_sub64 av_sat_sub64_c +#endif #ifndef av_clipf # define av_clipf av_clipf_c #endif -- 2.24.1.windows.2