// Copyright 2020 Google LLC

//

// Licensed under the Apache License, Version 2.0 (the "License");

// you may not use this file except in compliance with the License.

// You may obtain a copy of the License at

//

//      http://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software

// distributed under the License is distributed on an "AS IS" BASIS,

// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

// See the License for the specific language governing permissions and

// limitations under the License.

#include <cstddef>

#include <fuzzer/FuzzedDataProvider.h>

#include "spdlog/spdlog.h"

#include "spdlog/sinks/basic_file_sink.h"

#include "spdlog/pattern_formatter.h"

std::string my_formatter_txt = "custom-flag";

class my_formatter_flag : public spdlog::custom_flag_formatter

{

public:

    void format(const spdlog::details::log_msg &, const std::tm &, spdlog::memory_buf_t &dest) override

    {

        dest.append(my_formatter_txt.data(), my_formatter_txt.data() + my_formatter_txt.size());

    }

    std::unique_ptr<custom_flag_formatter> clone() const override

    {

        return spdlog::details::make_unique<my_formatter_flag>();

    }

};

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

  if (size == 0) {

    return 0;

  }

  static std::shared_ptr<spdlog::logger> my_logger;

  if (!my_logger.get()) {

    my_logger = spdlog::basic_logger_mt("basic_logger", "/dev/null");

    spdlog::set_default_logger(my_logger);

  }

  FuzzedDataProvider stream(data, size);

  const unsigned long size_arg = stream.ConsumeIntegral<unsigned long>();

  const unsigned long  int_arg = stream.ConsumeIntegral<unsigned long>();

  const char flag = (char)(stream.ConsumeIntegral<char>());

  const std::string pattern = stream.ConsumeRandomLengthString();

  my_formatter_txt = stream.ConsumeRandomLengthString();

  const std::string string_arg = stream.ConsumeRandomLengthString();

  const std::string format_string = stream.ConsumeRemainingBytesAsString();

  using spdlog::details::make_unique;

  auto formatter = make_unique<spdlog::pattern_formatter>();

  formatter->add_flag<my_formatter_flag>(flag).set_pattern(pattern);

  spdlog::set_formatter(std::move(formatter));

  spdlog::info(format_string.c_str(), size_arg, int_arg, string_arg);

  return 0;

}