kc3-lang/libjpeg-turbo/fuzz/compress_yuv.cc

• Show log

  Commit

• Hash : b4336c3a
  Author :
  Date : 2024-08-13T15:41:54
  

  Work around valgrind/MSan SIMD false positives
  
  Referring to
  https://sourceforge.net/p/libjpeg-turbo/bugs/48,
  https://sourceforge.net/p/libjpeg-turbo/bugs/82,
   #15, #238, #253, and #619,
  valgrind and MSan have failed to properly detect data initialization by
  libjpeg-turbo's x86 SIMD extensions for the entire 14 years that
  libjpeg-turbo has been a project, resulting in false positives unless
  libjpeg-turbo is built with WITH_SIMD=0 or run with JSIMD_FORCENONE=1.
  This commit introduces a new C preprocessor macro (ZERO_BUFFERS) that,
  if set, causes libjpeg-turbo to zero certain buffers in order to work
  around the specific valgrind/MSan test failures caused by the
  aforementioned false positives.  This allows us to more closely
  approximate the production configuration of libjpeg-turbo when testing
  with valgrind or MSan.
  
  Closes #781
  

Download

• fuzz/compress_yuv.cc

• /*
   * Copyright (C)2021-2023 D. R. Commander.  All Rights Reserved.
   *
   * Redistribution and use in source and binary forms, with or without
   * modification, are permitted provided that the following conditions are met:
   *
   * - Redistributions of source code must retain the above copyright notice,
   *   this list of conditions and the following disclaimer.
   * - Redistributions in binary form must reproduce the above copyright notice,
   *   this list of conditions and the following disclaimer in the documentation
   *   and/or other materials provided with the distribution.
   * - Neither the name of the libjpeg-turbo Project nor the names of its
   *   contributors may be used to endorse or promote products derived from this
   *   software without specific prior written permission.
   *
   * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS",
   * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
   * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
   * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
   * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
   * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
   * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
   * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
   * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
   * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
   * POSSIBILITY OF SUCH DAMAGE.
   */
  
  #include <turbojpeg.h>
  #include <stdio.h>
  #include <stdlib.h>
  #include <stdint.h>
  #include <string.h>
  #include <unistd.h>
  
  
  #define NUMTESTS  6
  
  
  struct test {
    enum TJPF pf;
    enum TJSAMP subsamp;
    int quality;
  };
  
  
  extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
  {
    tjhandle handle = NULL;
    unsigned char *srcBuf = NULL, *dstBuf = NULL, *yuvBuf = NULL;
    int width = 0, height = 0, fd = -1, i, ti;
    char filename[FILENAME_MAX] = { 0 };
    struct test tests[NUMTESTS] = {
      { TJPF_XBGR, TJSAMP_444, 100 },
      { TJPF_XRGB, TJSAMP_422, 90 },
      { TJPF_BGR, TJSAMP_420, 80 },
      { TJPF_RGB, TJSAMP_411, 70 },
      { TJPF_BGR, TJSAMP_GRAY, 60 },
      { TJPF_GRAY, TJSAMP_GRAY, 50 }
    };
  
    snprintf(filename, FILENAME_MAX, "/tmp/libjpeg-turbo_compress_yuv_fuzz.XXXXXX");
    if ((fd = mkstemp(filename)) < 0 || write(fd, data, size) < 0)
      goto bailout;
  
    if ((handle = tj3Init(TJINIT_COMPRESS)) == NULL)
      goto bailout;
  
    for (ti = 0; ti < NUMTESTS; ti++) {
      int sum = 0, pf = tests[ti].pf;
      size_t dstSize = 0, maxBufSize;
  
      /* Test non-default compression options on specific iterations. */
      tj3Set(handle, TJPARAM_BOTTOMUP, ti == 0);
      tj3Set(handle, TJPARAM_FASTDCT, ti == 1);
      tj3Set(handle, TJPARAM_OPTIMIZE, ti == 4);
      tj3Set(handle, TJPARAM_PROGRESSIVE, ti == 1 || ti == 3);
      tj3Set(handle, TJPARAM_ARITHMETIC, ti == 2 || ti == 3);
      tj3Set(handle, TJPARAM_NOREALLOC, 1);
      tj3Set(handle, TJPARAM_RESTARTBLOCKS, ti == 3 || ti == 4 ? 4 : 0);
  
      tj3Set(handle, TJPARAM_MAXPIXELS, 1048576);
      /* tj3LoadImage8() will refuse to load images larger than 1 Megapixel, so
         we don't need to check the width and height here. */
      if ((srcBuf = tj3LoadImage8(handle, filename, &width, 1, &height,
                                  &pf)) == NULL)
        continue;
  
      maxBufSize = tj3JPEGBufSize(width, height, tests[ti].subsamp);
      if ((dstBuf = (unsigned char *)malloc(maxBufSize)) == NULL)
        goto bailout;
      if ((yuvBuf =
           (unsigned char *)malloc(tj3YUVBufSize(width, 1, height,
                                                 tests[ti].subsamp))) == NULL)
        goto bailout;
  
      tj3Set(handle, TJPARAM_SUBSAMP, tests[ti].subsamp);
      tj3Set(handle, TJPARAM_QUALITY, tests[ti].quality);
      if (tj3EncodeYUV8(handle, srcBuf, width, 0, height, pf, yuvBuf, 1) == 0 &&
          tj3CompressFromYUV8(handle, yuvBuf, width, 1, height, &dstBuf,
                              &dstSize) == 0) {
        /* Touch all of the output pixels in order to catch uninitialized reads
           when using MemorySanitizer. */
        for (i = 0; i < dstSize; i++)
          sum += dstBuf[i];
      }
  
      free(dstBuf);
      dstBuf = NULL;
      free(yuvBuf);
      yuvBuf = NULL;
      tj3Free(srcBuf);
      srcBuf = NULL;
  
      /* Prevent the code above from being optimized out.  This test should never
         be true, but the compiler doesn't know that. */
      if (sum > 255 * maxBufSize)
        goto bailout;
    }
  
  bailout:
    free(dstBuf);
    free(yuvBuf);
    tj3Free(srcBuf);
    if (fd >= 0) {
      close(fd);
      if (strlen(filename) > 0) unlink(filename);
    }
    tj3Destroy(handle);
    return 0;
  }
  

Download