If image sizes are greater than 32768, fixed point stepping will overflow an int. This CL changes t…

unit_test/scale_test.cc

 /*
  *  Copyright 2011 The LibYuv Project Authors. All rights reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS. All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */
 
 #include <stdlib.h>
 #include <time.h>
 
 #include "libyuv/cpu_id.h"
 #include "libyuv/scale.h"
 #include "../unit_test/unit_test.h"
 
 #define STRINGIZE(line) #line
 #define FILELINESTR(file, line) file ":" STRINGIZE(line)
 
 namespace libyuv {
 
 // Test scaling with C vs Opt and return maximum pixel difference. 0 = exact.
 static int TestFilter(int src_width, int src_height,
                       int dst_width, int dst_height,
                       FilterMode f, int benchmark_iterations,
                       int disable_cpu_flags, int benchmark_cpu_info) {
+  if (src_width > 32768 || src_height > 32768 ||
+      dst_width > 32768 || dst_height > 32768) {
+    printf("Warning - size too large to test.  Skipping\n");
+    return 0;
+  }
   int i, j;
   const int b = 0;  // 128 to test for padding/stride.
   int src_width_uv = (Abs(src_width) + 1) >> 1;
   int src_height_uv = (Abs(src_height) + 1) >> 1;
 
   int64 src_y_plane_size = (Abs(src_width) + b * 2) * (Abs(src_height) + b * 2);
   int64 src_uv_plane_size = (src_width_uv + b * 2) * (src_height_uv + b * 2);
 
   int src_stride_y = b * 2 + Abs(src_width);
   int src_stride_uv = b * 2 + src_width_uv;
@@ skipping 103 matching lines @@
   free_aligned_buffer_page_end(src_v)
 
   return max_diff;
 }
 
 // Test scaling with 8 bit C vs 16 bit C and return maximum pixel difference.
 // 0 = exact.
 static int TestFilter_16(int src_width, int src_height,
                          int dst_width, int dst_height,
                          FilterMode f, int benchmark_iterations) {
+  if (src_width > 32768 || src_height > 32768 ||
+      dst_width > 32768 || dst_height > 32768) {
+    printf("Warning - size too large to test.  Skipping\n");
+    return 0;
+  }
   int i, j;
   const int b = 0;  // 128 to test for padding/stride.
   int src_width_uv = (Abs(src_width) + 1) >> 1;
   int src_height_uv = (Abs(src_height) + 1) >> 1;
 
   int64 src_y_plane_size = (Abs(src_width) + b * 2) *
       (Abs(src_height) + b * 2);
   int64 src_uv_plane_size = (src_width_uv + b * 2) * (src_height_uv + b * 2);
 
   int src_stride_y = b * 2 + Abs(src_width);
@@ skipping 106 matching lines @@
   free_aligned_buffer_page_end(src_y_16)
   free_aligned_buffer_page_end(src_u_16)
   free_aligned_buffer_page_end(src_v_16)
 
   return max_diff;
 }
 
 // The following adjustments in dimensions ensure the scale factor will be
 // exactly achieved.
 // 2 is chroma subsample
-#define DX(x, nom, denom) static_cast<int>((Abs(x) / nom / 2) * nom * 2)
-#define SX(x, nom, denom) static_cast<int>((x / nom / 2) * denom * 2)
+#define DX(x, nom, denom) static_cast<int>(((Abs(x) / nom + 1) / 2) * nom * 2)
+#define SX(x, nom, denom) static_cast<int>(((x / nom + 1) / 2) * denom * 2)
 
 #define TEST_FACTOR1(name, filter, nom, denom, max_diff)                       \
     TEST_F(LibYUVScaleTest, ScaleDownBy##name##_##filter) {                    \
       int diff = TestFilter(SX(benchmark_width_, nom, denom),                  \
                             SX(benchmark_height_, nom, denom),                 \
                             DX(benchmark_width_, nom, denom),                  \
                             DX(benchmark_height_, nom, denom),                 \
                             kFilter##filter, benchmark_iterations_,            \
                             disable_cpu_flags_, benchmark_cpu_info_);          \
       EXPECT_LE(diff, max_diff);                                               \
@@ skipping 66 matching lines @@
 TEST_SCALETO(Scale, 1, 1)
 TEST_SCALETO(Scale, 320, 240)
 TEST_SCALETO(Scale, 352, 288)
 TEST_SCALETO(Scale, 569, 480)
 TEST_SCALETO(Scale, 640, 360)
 TEST_SCALETO(Scale, 1280, 720)
 #undef TEST_SCALETO1
 #undef TEST_SCALETO
 
 }  // namespace libyuv