eof: Add peephole optimiser tests for EOF specific optimisations

Author: rodiazet

test/libevmasm/Optimiser.cpp

@@ -1049,6 +1049,142 @@ BOOST_AUTO_TEST_CASE(clear_unreachable_code_eof, *boost::unit_test::precondition
 	}
 }
 
+BOOST_AUTO_TEST_CASE(is_zero_is_zero_rjumpi, *boost::unit_test::precondition(onEOF()))
+{
+	AssemblyItems items{
+		u256(1),
+		Instruction::ISZERO,
+		Instruction::ISZERO,
+		AssemblyItem::conditionalRelativeJumpTo(AssemblyItem(Tag, 1)),
+		u256(0),
+		Instruction::SLOAD,
+		AssemblyItem(Tag, 1),
+	};
+
+	AssemblyItems expectation{
+		u256(1),
+		AssemblyItem::conditionalRelativeJumpTo(AssemblyItem(Tag, 1)),
+		u256(0),
+		Instruction::SLOAD,
+		AssemblyItem(Tag, 1),
+	};
+
+	PeepholeOptimiser peepOpt(items, solidity::test::CommonOptions::get().evmVersion());
+	BOOST_REQUIRE(peepOpt.optimise());
+	BOOST_CHECK_EQUAL_COLLECTIONS(
+		items.begin(), items.end(),
+		expectation.begin(), expectation.end()
+	);
+}
+
+BOOST_AUTO_TEST_CASE(equal_is_zero_rjumpi, *boost::unit_test::precondition(onEOF()))
+{
+	AssemblyItems items{
+		u256(1),
+		u256(2),
+		Instruction::EQ,
+		Instruction::ISZERO,
+		AssemblyItem::conditionalRelativeJumpTo(AssemblyItem(Tag, 1)),
+		u256(0),
+		Instruction::SLOAD,
+		AssemblyItem(Tag, 1),
+	};
+
+	AssemblyItems expectation{
+		u256(1),
+		u256(2),
+		Instruction::SUB,
+		AssemblyItem::conditionalRelativeJumpTo(AssemblyItem(Tag, 1)),
+		u256(0),
+		Instruction::SLOAD,
+		AssemblyItem(Tag, 1),
+	};
+
+	PeepholeOptimiser peepOpt(items, solidity::test::CommonOptions::get().evmVersion());
+	BOOST_REQUIRE(peepOpt.optimise());
+	BOOST_CHECK_EQUAL_COLLECTIONS(
+		items.begin(), items.end(),
+		expectation.begin(), expectation.end()
+	);
+}
+
+BOOST_AUTO_TEST_CASE(double_rjump, *boost::unit_test::precondition(onEOF()))
+{
+	AssemblyItems items{
+		u256(1),
+		AssemblyItem::conditionalRelativeJumpTo(AssemblyItem(Tag, 1)),
+		AssemblyItem::relativeJumpTo(AssemblyItem(Tag, 2)),
+		AssemblyItem(Tag, 1),
+		u256(0),
+		Instruction::SLOAD,
+		AssemblyItem(Tag, 2),
+	};
+
+	AssemblyItems expectation{
+		u256(1),
+		Instruction::ISZERO,
+		AssemblyItem::conditionalRelativeJumpTo(AssemblyItem(Tag, 2)),
+		AssemblyItem(Tag, 1),
+		u256(0),
+		Instruction::SLOAD,
+		AssemblyItem(Tag, 2),
+	};
+
+	PeepholeOptimiser peepOpt(items, solidity::test::CommonOptions::get().evmVersion());
+	BOOST_REQUIRE(peepOpt.optimise());
+	BOOST_CHECK_EQUAL_COLLECTIONS(
+		items.begin(), items.end(),
+		expectation.begin(), expectation.end()
+	);
+}
+
+BOOST_AUTO_TEST_CASE(rjump_to_next, *boost::unit_test::precondition(onEOF()))
+{
+	AssemblyItems items{
+		AssemblyItem::relativeJumpTo(AssemblyItem(Tag, 1)),
+		AssemblyItem(Tag, 1),
+		u256(0),
+		Instruction::SLOAD,
+	};
+
+	AssemblyItems expectation{
+		AssemblyItem(Tag, 1),
+		u256(0),
+		Instruction::SLOAD,
+	};
+
+	PeepholeOptimiser peepOpt(items, solidity::test::CommonOptions::get().evmVersion());
+	BOOST_REQUIRE(peepOpt.optimise());
+	BOOST_CHECK_EQUAL_COLLECTIONS(
+		items.begin(), items.end(),
+		expectation.begin(), expectation.end()
+	);
+}
+
+BOOST_AUTO_TEST_CASE(rjumpi_to_next, *boost::unit_test::precondition(onEOF()))
+{
+	AssemblyItems items{
+		AssemblyItem::conditionalRelativeJumpTo(AssemblyItem(Tag, 1)),
+		AssemblyItem(Tag, 1),
+		u256(0),
+		Instruction::SLOAD,
+	};
+
+	AssemblyItems expectation{
+		Instruction::POP,
+		AssemblyItem(Tag, 1),
+		u256(0),
+		Instruction::SLOAD,
+	};
+
+	PeepholeOptimiser peepOpt(items, solidity::test::CommonOptions::get().evmVersion());
+	BOOST_REQUIRE(peepOpt.optimise());
+	BOOST_CHECK_EQUAL_COLLECTIONS(
+		items.begin(), items.end(),
+		expectation.begin(), expectation.end()
+	);
+}
+
 BOOST_AUTO_TEST_CASE(deduplicateNextTagBlockSize3)
 {
 	AssemblyItems items{