// Copyright (c) the JPEG XL 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.

#include <cstddef>

#include <cstdint>

#include <vector>

#include "lib/jxl/base/span.h"

#include "lib/jxl/color_encoding_internal.h"

#include "lib/jxl/dec_ans.h"

#include "lib/jxl/dec_bit_reader.h"

#include "lib/jxl/fields.h"

#include "lib/jxl/frame_header.h"

#include "lib/jxl/fuzztest.h"

#include "lib/jxl/headers.h"

#include "lib/jxl/image_metadata.h"

#include "lib/jxl/jpeg/jpeg_data.h"

#include "lib/jxl/loop_filter.h"

#include "lib/jxl/modular/encoding/context_predict.h"

#include "lib/jxl/modular/encoding/encoding.h"

#include "lib/jxl/modular/transform/squeeze_params.h"

#include "lib/jxl/modular/transform/transform.h"

#include "lib/jxl/quantizer.h"

namespace {

using ::jxl::BitReader;

using ::jxl::Bytes;

using ::jxl::CodecMetadata;

using ::jxl::CustomTransformData;

using ::jxl::ImageMetadata;

using ::jxl::SizeHeader;

int DoTestOneInput(const uint8_t* data, size_t size) {

  // Global parameters used by some headers.

  CodecMetadata codec_metadata;

  // First byte controls which header to parse.

  if (size == 0) return 0;

  BitReader reader(Bytes(data + 1, size - 1));

#define FUZZER_CASE_HEADER(number, classname, ...) \

  case number: {                                   \

    ::jxl::classname header{__VA_ARGS__};          \

    (void)jxl::Bundle::Read(&reader, &header);     \

    break;                                         \

  }

  switch (data[0]) {

    case 0: {

      SizeHeader size_header;

      (void)jxl::ReadSizeHeader(&reader, &size_header);

      break;

    }

    case 1: {

      ImageMetadata metadata;

      (void)jxl::ReadImageMetadata(&reader, &metadata);

      break;

    }

      FUZZER_CASE_HEADER(2, FrameHeader, &codec_metadata)

      FUZZER_CASE_HEADER(3, jpeg::JPEGData)

      FUZZER_CASE_HEADER(4, AnimationFrame, &codec_metadata)

      FUZZER_CASE_HEADER(5, AnimationHeader)

      FUZZER_CASE_HEADER(6, BitDepth)

      FUZZER_CASE_HEADER(7, BlendingInfo)

      FUZZER_CASE_HEADER(8, ColorEncoding)

      FUZZER_CASE_HEADER(9, CustomTransferFunction)

      FUZZER_CASE_HEADER(10, Customxy)

      FUZZER_CASE_HEADER(11, ExtraChannelInfo)

      FUZZER_CASE_HEADER(12, GroupHeader)

      FUZZER_CASE_HEADER(13, weighted::Header)

      FUZZER_CASE_HEADER(14, LoopFilter)

      FUZZER_CASE_HEADER(15, LZ77Params)

      FUZZER_CASE_HEADER(16, OpsinInverseMatrix)

      FUZZER_CASE_HEADER(17, Passes)

      FUZZER_CASE_HEADER(18, PreviewHeader)

      FUZZER_CASE_HEADER(19, QuantizerParams)

      FUZZER_CASE_HEADER(20, SqueezeParams)

      FUZZER_CASE_HEADER(21, ToneMapping)

      FUZZER_CASE_HEADER(22, Transform)

      FUZZER_CASE_HEADER(23, YCbCrChromaSubsampling)

    default: {

      CustomTransformData transform_data;

      transform_data.nonserialized_xyb_encoded = true;

      (void)jxl::Bundle::Read(&reader, &transform_data);

      break;

    }

  }

  (void)reader.Close();

  return 0;

}

}  // namespace

extern "C" int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {

  return DoTestOneInput(data, size);

}

void TestOneInput(const std::vector<uint8_t>& data) {

  DoTestOneInput(data.data(), data.size());

}

FUZZ_TEST(FieldsFuzzTest, TestOneInput);