[isoltest] Add support to query balance.

Author: aarlt

test/libsolidity/SemanticTest.cpp

@@ -31,6 +31,7 @@
 #include <optional>
 #include <stdexcept>
 #include <utility>
+#include <functional>
 
 using namespace std;
 using namespace solidity;
@@ -61,6 +62,8 @@ SemanticTest::SemanticTest(
 	m_enforceGasCost(_enforceGasCost),
 	m_enforceGasCostMinValue(_enforceGasCostMinValue)
 {
+	initializeBuiltins();
+
 	string choice = m_reader.stringSetting("compileViaYul", "default");
 	if (choice == "also")
 	{
@@ -122,6 +125,21 @@ SemanticTest::SemanticTest(
 	}
 }
 
+void SemanticTest::initializeBuiltins()
+{
+	m_builtins["account_balance"] = [this](FunctionCall const& _call) -> std::optional<bytes>
+	{
+	  soltestAssert(_call.arguments.parameters.size() == 1, "Account address expected.");
+	  h160 address = h160(_call.arguments.parameters.at(0).rawString);
+	  return util::toBigEndian(SolidityExecutionFramework::balanceAt(address));
+	};
+	m_builtins["contract_balance"] = [this](FunctionCall const& _call) -> std::optional<bytes>
+	{
+	  soltestAssert(_call.arguments.parameters.empty(), "No arguments expected.");
+	  return util::toBigEndian(SolidityExecutionFramework::balanceAt(m_contractAddress));
+	};
+}
+
 TestCase::TestResult SemanticTest::run(ostream& _stream, string const& _linePrefix, bool _formatted)
 {
 	TestResult result = TestResult::Success;

test/libsolidity/SemanticTest.h

@@ -77,7 +77,11 @@ class SemanticTest: public SolidityExecutionFramework, public EVMVersionRestrict
 
 private:
 	TestResult runTest(std::ostream& _stream, std::string const& _linePrefix, bool _formatted, bool _compileViaYul, bool _compileToEwasm);
+
 	bool checkGasCostExpectation(TestFunctionCall& io_test, bool _compileViaYul) const;
+
+	void initializeBuiltins();
+
 	SourceMap m_sources;
 	std::size_t m_lineOffset;
 	std::vector<TestFunctionCall> m_tests;

test/libsolidity/semanticTests/builtins/balance_other_contract.sol

@@ -0,0 +1,23 @@
+contract Other {
+    constructor() payable {
+    }
+    function getAddress() public returns (address) {
+        return address(this);
+    }
+}
+contract ClientReceipt {
+    Other other;
+    constructor() payable {
+        other = new Other{value:500}();
+    }
+    function getAddress() public returns (address) {
+        return other.getAddress();
+    }
+}
+// ====
+// compileViaYul: also
+// ----
+// constructor(), 2000 wei ->
+// contract_balance -> 1500
+// getAddress() -> 0xf01f7809444bd9a93a854361c6fae3f23d9e23db
+// account_balance: 0xf01f7809444bd9a93a854361c6fae3f23d9e23db -> 500

test/libsolidity/semanticTests/builtins/balance_with_balance.sol

@@ -0,0 +1,8 @@
+contract ClientReceipt {
+    constructor() payable {}
+}
+// ====
+// compileViaYul: also
+// ----
+// constructor(), 1000 wei ->
+// contract_balance -> 1000

test/libsolidity/semanticTests/builtins/balance_with_balance2.sol

@@ -0,0 +1,8 @@
+contract ClientReceipt {
+    constructor() payable {}
+}
+// ====
+// compileViaYul: also
+// ----
+// constructor(), 1 ether ->
+// contract_balance -> 1000000000000000000

test/libsolidity/semanticTests/builtins/balance_without_balance.sol

@@ -0,0 +1,7 @@
+contract ClientReceipt {
+}
+// ====
+// compileViaYul: also
+// ----
+// contract_balance -> 0
+// account_balance: 0x0000000000000000000000000000000000000000 -> 0

test/libsolidity/semanticTests/payable/no_nonpayable_circumvention_by_modifier.sol

@@ -8,13 +8,9 @@ contract C {
     function msgvalue() internal returns (uint) {
         return msg.value;
     }
-    // TODO: remove this helper function once isoltest supports balance checking
-    function balance() external returns (uint) {
-        return address(this).balance;
-    }
 }
 // ====
 // compileViaYul: also
 // ----
 // f(), 27 wei -> FAILURE
-// balance() -> 0
+// contract_balance -> 0