/*
	This file is part of solidity.

	solidity is free software: you can redistribute it and/or modify
	it under the terms of the GNU General Public License as published by
	the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.

	solidity is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU General Public License for more details.

	You should have received a copy of the GNU General Public License
	along with solidity.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <test/tools/fuzzer_common.h>

#include <test/tools/ossfuzz/libFuzzerMutator.h>

#include <test/TestCaseReader.h>

#include <sstream>

using namespace solidity::test::fuzzer;
using namespace solidity::frontend::test;
using namespace std;

extern "C" size_t LLVMFuzzerCustomMutator(
	uint8_t* _data,
	size_t _size,
	size_t _maxSize,
	unsigned int _seed
)
{
	try
	{
		return CustomMutator(_data, _size, _maxSize, _seed).mutate();
	}
	catch (range_error const&)
	{
		return LLVMFuzzerMutate(_data, _size, _maxSize);
	}
}

extern "C" int LLVMFuzzerTestOneInput(uint8_t const* _data, size_t _size)
{
	if (_size <= 600)
	{
		string input(reinterpret_cast<char const*>(_data), _size);
		map<string, string> sourceCode;
		try
		{
			TestCaseReader t = TestCaseReader(std::istringstream(input));
			sourceCode = t.sources().sources;
		}
		catch (runtime_error const&)
		{
			return 0;
		}
		FuzzerUtil::testCompiler(sourceCode, /*optimize=*/false, /*_rand=*/_size);
	}
	return 0;
}