mirror of
				https://github.com/ethereum/solidity
				synced 2023-10-03 13:03:40 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			131 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			131 lines
		
	
	
		
			4.4 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
| 	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/>.
 | |
| */
 | |
| // SPDX-License-Identifier: GPL-3.0
 | |
| 
 | |
| #include <test/libyul/YulOptimizerTest.h>
 | |
| #include <test/libyul/YulOptimizerTestCommon.h>
 | |
| 
 | |
| #include <test/libsolidity/util/SoltestErrors.h>
 | |
| #include <test/libyul/Common.h>
 | |
| #include <test/Common.h>
 | |
| 
 | |
| #include <libyul/Object.h>
 | |
| #include <libyul/optimiser/ReasoningBasedSimplifier.h>
 | |
| #include <libyul/AsmPrinter.h>
 | |
| 
 | |
| #include <liblangutil/SourceReferenceFormatter.h>
 | |
| 
 | |
| #include <libsolutil/AnsiColorized.h>
 | |
| 
 | |
| #include <fstream>
 | |
| 
 | |
| using namespace solidity;
 | |
| using namespace solidity::util;
 | |
| using namespace solidity::langutil;
 | |
| using namespace solidity::yul;
 | |
| using namespace solidity::yul::test;
 | |
| using namespace solidity::frontend;
 | |
| using namespace solidity::frontend::test;
 | |
| using namespace std;
 | |
| 
 | |
| YulOptimizerTest::YulOptimizerTest(string const& _filename):
 | |
| 	EVMVersionRestrictedTestCase(_filename)
 | |
| {
 | |
| 	boost::filesystem::path path(_filename);
 | |
| 
 | |
| 	if (path.empty() || std::next(path.begin()) == path.end() || std::next(std::next(path.begin())) == path.end())
 | |
| 		BOOST_THROW_EXCEPTION(runtime_error("Filename path has to contain a directory: \"" + _filename + "\"."));
 | |
| 	m_optimizerStep = std::prev(std::prev(path.end()))->string();
 | |
| 
 | |
| 	if (m_optimizerStep == "reasoningBasedSimplifier" && (
 | |
| 		solidity::test::CommonOptions::get().disableSMT ||
 | |
| 		ReasoningBasedSimplifier::invalidInCurrentEnvironment()
 | |
| 	))
 | |
| 		m_shouldRun = false;
 | |
| 
 | |
| 	m_source = m_reader.source();
 | |
| 
 | |
| 	auto dialectName = m_reader.stringSetting("dialect", "evm");
 | |
| 	m_dialect = &dialect(dialectName, solidity::test::CommonOptions::get().evmVersion());
 | |
| 
 | |
| 	m_expectation = m_reader.simpleExpectations();
 | |
| }
 | |
| 
 | |
| TestCase::TestResult YulOptimizerTest::run(ostream& _stream, string const& _linePrefix, bool const _formatted)
 | |
| {
 | |
| 	std::tie(m_object, m_analysisInfo) = parse(_stream, _linePrefix, _formatted, m_source);
 | |
| 	if (!m_object)
 | |
| 		return TestResult::FatalError;
 | |
| 
 | |
| 	soltestAssert(m_dialect, "Dialect not set.");
 | |
| 
 | |
| 	m_object->analysisInfo = m_analysisInfo;
 | |
| 	YulOptimizerTestCommon tester(m_object, *m_dialect);
 | |
| 	tester.setStep(m_optimizerStep);
 | |
| 
 | |
| 	if (!tester.runStep())
 | |
| 	{
 | |
| 		AnsiColorized(_stream, _formatted, {formatting::BOLD, formatting::RED}) << _linePrefix << "Invalid optimizer step: " << m_optimizerStep << endl;
 | |
| 		return TestResult::FatalError;
 | |
| 	}
 | |
| 
 | |
| 	auto const printed = (m_object->subObjects.empty() ? AsmPrinter{ *m_dialect }(*m_object->code) : m_object->toString(m_dialect));
 | |
| 
 | |
| 	// Re-parse new code for compilability
 | |
| 	// TODO: support for wordSizeTransform which needs different input and output dialects
 | |
| 	if (m_optimizerStep != "wordSizeTransform" && !std::get<0>(parse(_stream, _linePrefix, _formatted, printed)))
 | |
| 	{
 | |
| 		util::AnsiColorized(_stream, _formatted, {util::formatting::BOLD, util::formatting::CYAN})
 | |
| 			<< _linePrefix << "Result after the optimiser:" << endl;
 | |
| 		printIndented(_stream, printed, _linePrefix + "  ");
 | |
| 		return TestResult::FatalError;
 | |
| 	}
 | |
| 
 | |
| 	m_obtainedResult = "step: " + m_optimizerStep + "\n\n" + printed + "\n";
 | |
| 
 | |
| 	return checkResult(_stream, _linePrefix, _formatted);
 | |
| }
 | |
| 
 | |
| std::pair<std::shared_ptr<Object>, std::shared_ptr<AsmAnalysisInfo>> YulOptimizerTest::parse(
 | |
| 	ostream& _stream,
 | |
| 	string const& _linePrefix,
 | |
| 	bool const _formatted,
 | |
| 	string const& _source
 | |
| )
 | |
| {
 | |
| 	ErrorList errors;
 | |
| 	soltestAssert(m_dialect, "");
 | |
| 	shared_ptr<Object> object;
 | |
| 	shared_ptr<AsmAnalysisInfo> analysisInfo;
 | |
| 	std::tie(object, analysisInfo) = yul::test::parse(_source, *m_dialect, errors);
 | |
| 	if (!object || !analysisInfo || !Error::containsOnlyWarnings(errors))
 | |
| 	{
 | |
| 		AnsiColorized(_stream, _formatted, {formatting::BOLD, formatting::RED}) << _linePrefix << "Error parsing source." << endl;
 | |
| 		printErrors(_stream, errors);
 | |
| 		return {};
 | |
| 	}
 | |
| 	return {std::move(object), std::move(analysisInfo)};
 | |
| }
 | |
| 
 | |
| void YulOptimizerTest::printErrors(ostream& _stream, ErrorList const& _errors)
 | |
| {
 | |
| 	SourceReferenceFormatter formatter(_stream, true, false);
 | |
| 
 | |
| 	for (auto const& error: _errors)
 | |
| 		formatter.printErrorInformation(*error);
 | |
| }
 |