/*

 * Copyright (C)2021 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

/* Private flag that triggers different TurboJPEG API behavior when fuzzing */

#define TJFLAG_FUZZING  (1 << 30)

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 }

  };

  char arithEnv[16] = "TJ_ARITHMETIC=0";

  char restartEnv[13] = "TJ_RESTART=0";

#if defined(__has_feature) && __has_feature(memory_sanitizer)

  char simdEnv[18] = "JSIMD_FORCENONE=1";

  /* The libjpeg-turbo SIMD extensions produce false positives with

     MemorySanitizer. */

  putenv(simdEnv);

#endif

  putenv(arithEnv);

  putenv(restartEnv);

  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 = tjInitCompress()) == NULL)

    goto bailout;

  for (ti = 0; ti < NUMTESTS; ti++) {

    int flags = TJFLAG_FUZZING | TJFLAG_NOREALLOC, sum = 0, pf = tests[ti].pf;

    unsigned long dstSize = 0, maxBufSize;

    /* Test non-default compression options on specific iterations. */

    if (ti == 0)

      flags |= TJFLAG_BOTTOMUP | TJFLAG_ACCURATEDCT;

    else if (ti == 1 || ti == 3)

      flags |= TJFLAG_PROGRESSIVE;

    if (ti == 2 || ti == 3)

      arithEnv[14] = '1';

    else

      arithEnv[14] = '0';

    if (ti == 1 || ti == 2)

      restartEnv[11] = '2';

    else

      restartEnv[11] = '0';

    /* tjLoadImage() refuses to load images larger than 1 Megapixel when

       FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION is defined (yes, that's a dirty

       hack), so we don't need to check the width and height here. */

    if ((srcBuf = tjLoadImage(filename, &width, 1, &height, &pf,

                              flags)) == NULL)

      continue;

    maxBufSize = tjBufSize(width, height, tests[ti].subsamp);

    if ((dstBuf = (unsigned char *)malloc(maxBufSize)) == NULL)

      goto bailout;

    if ((yuvBuf =

         (unsigned char *)malloc(tjBufSizeYUV2(width, 1, height,

                                               tests[ti].subsamp))) == NULL)

      goto bailout;

    if (tjEncodeYUV3(handle, srcBuf, width, 0, height, pf, yuvBuf, 1,

                     tests[ti].subsamp, flags) == 0 &&

        tjCompressFromYUV(handle, yuvBuf, width, 1, height, tests[ti].subsamp,

                          &dstBuf, &dstSize, tests[ti].quality, flags) == 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;

    tjFree(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);

  tjFree(srcBuf);

  if (fd >= 0) {

    close(fd);

    if (strlen(filename) > 0) unlink(filename);

  }

  if (handle) tjDestroy(handle);

  return 0;

}