[isoltest] Add support to query balance.

Author: aarlt

test/libsolidity/SemanticTest.cpp

@@ -49,13 +49,19 @@ SemanticTest::SemanticTest(string const& _filename, langutil::EVMVersion _evmVer
 	m_enforceViaYul(enforceViaYul)
 {
 	using namespace std::placeholders;
-	m_builtins
-		= {{"smoke",
-			{
-				{"test0()", std::bind(&SemanticTest::builtinSmokeTest, this, _1)},
-				{"test1(uint256)", std::bind(&SemanticTest::builtinSmokeTest, this, _1)},
-				{"test2(uint256,uint256)", std::bind(&SemanticTest::builtinSmokeTest, this, _1)},
-			}}};
+	m_builtins = {
+		{"smoke",
+		 {
+			 {"test0()", std::bind(&SemanticTest::builtinSmokeTest, this, _1)},
+			 {"test1(uint256)", std::bind(&SemanticTest::builtinSmokeTest, this, _1)},
+			 {"test2(uint256,uint256)", std::bind(&SemanticTest::builtinSmokeTest, this, _1)},
+		 }},
+		{"contract",
+		 {
+			 {"balance()", std::bind(&SemanticTest::builtinContractBalance, this, _1)},
+			 {"balance(uint256)", std::bind(&SemanticTest::builtinContractBalance, this, _1)},
+		 }},
+	};
 
 	string choice = m_reader.stringSetting("compileViaYul", "default");
 	if (choice == "also")
@@ -401,3 +407,16 @@ bytes SemanticTest::builtinSmokeTest(FunctionCall const& call)
 		result += util::toBigEndian(u256{call.arguments.parameters.at(i).rawBytes});
 	return result;
 }
+
+bytes SemanticTest::builtinContractBalance(FunctionCall const& call)
+{
+	assert(call.arguments.parameters.size() <= 1);
+	if (call.arguments.parameters.empty())
+		return util::toBigEndian(SolidityExecutionFramework::balanceAt(m_contractAddress));
+	else
+	{
+		h160 address{util::fromHex(call.arguments.parameters.at(0).rawString)};
+		return util::toBigEndian(SolidityExecutionFramework::balanceAt(address));
+	}
+	return bytes{};
+}

test/libsolidity/SemanticTest.h

@@ -62,6 +62,7 @@ class SemanticTest: public SolidityExecutionFramework, public EVMVersionRestrict
 private:
 	// builtin functions
 	bytes builtinSmokeTest(FunctionCall const& call);
+	bytes builtinContractBalance(FunctionCall const& call);
 
 	TestResult runTest(std::ostream& _stream, std::string const& _linePrefix, bool _formatted, bool _compileViaYul, bool _compileToEwasm);
 	SourceMap m_sources;

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(), 1000 wei ->
+// contract.balance() -> 500
+// getAddress() -> 0xF01F7809444BD9A93A854361C6FAE3F23D9E23DB
+// contract.balance(uint256): 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_without_balance.sol

@@ -0,0 +1,7 @@
+contract ClientReceipt {
+}
+// ====
+// compileViaYul: also
+// ----
+// contract.balance() -> 0
+// contract.balance(uint256): 0x0000000000000000000000000000000000000000 -> 0