Support address.code in sema

src/sema/ast.rs

@@ -1586,6 +1586,7 @@ pub enum StringLocation<T> {
 
 #[derive(PartialEq, Eq, Clone, Copy, Debug)]
 pub enum Builtin {
+    ContractCode,
     GetAddress,
     Balance,
     PayableSend,

src/sema/expression/member_access.rs

@@ -347,37 +347,52 @@ pub(super) fn member_access(
             }
             _ => {}
         },
-        Type::Address(_) => {
-            if id.name == "balance" {
-                if ns.target.is_polkadot() {
-                    let mut is_this = false;
-
-                    if let Expression::Cast { expr: this, .. } = &expr {
-                        if let Expression::Builtin {
-                            kind: Builtin::GetAddress,
-                            ..
-                        } = this.as_ref()
-                        {
-                            is_this = true;
-                        }
+        Type::Address(_) if id.name == "balance" => {
+            if ns.target.is_polkadot() {
+                let mut is_this = false;
+
+                if let Expression::Cast { expr: this, .. } = &expr {
+                    if let Expression::Builtin {
+                        kind: Builtin::GetAddress,
+                        ..
+                    } = this.as_ref()
+                    {
+                        is_this = true;
                     }
+                }
 
-                    if !is_this {
-                        diagnostics.push(Diagnostic::error(
-                            expr.loc(),
-                            "polkadot can only retrieve balance of this, like 'address(this).balance'".to_string(),
-                        ));
-                        return Err(());
-                    }
+                if !is_this {
+                    diagnostics.push(Diagnostic::error(
+                        expr.loc(),
+                        "polkadot can only retrieve balance of this, like 'address(this).balance'"
+                            .to_string(),
+                    ));
+                    return Err(());
                 }
-                used_variable(ns, &expr, symtable);
-                return Ok(Expression::Builtin {
-                    loc: *loc,
-                    tys: vec![Type::Value],
-                    kind: Builtin::Balance,
-                    args: vec![expr],
-                });
             }
+            used_variable(ns, &expr, symtable);
+            return Ok(Expression::Builtin {
+                loc: *loc,
+                tys: vec![Type::Value],
+                kind: Builtin::Balance,
+                args: vec![expr],
+            });
+        }
+        Type::Address(_) if id.name == "code" => {
+            if ns.target != Target::EVM {
+                diagnostics.push(Diagnostic::error(
+                    expr.loc(),
+                    format!("'address.code' is not supported on {}", ns.target),
+                ));
+                return Err(());
+            }
+            used_variable(ns, &expr, symtable);
+            return Ok(Expression::Builtin {
+                loc: *loc,
+                tys: vec![Type::Slice(Type::Uint(8).into())],
+                kind: Builtin::ContractCode,
+                args: vec![expr],
+            });
         }
         Type::Contract(ref_contract_no) => {
             let mut name_matches = 0;

tests/contract_testcases/evm/address_code.sol

@@ -0,0 +1,10 @@
+contract UpgradeableProxy {
+    function _setImplementation(
+        address newImplementation
+    ) public pure returns (uint) {
+        return newImplementation.code;
+    }
+}
+
+// ---- Expect: diagnostics ----
+// error: 5:9-38: conversion from uint8 slice to uint256 not possible

tests/contract_testcases/polkadot/builtins/address_code_01.sol

@@ -0,0 +1,10 @@
+contract UpgradeableProxy {
+    function _setImplementation(
+        address newImplementation
+    ) public pure returns (bytes) {
+        return newImplementation.code;
+    }
+}
+
+// ---- Expect: diagnostics ----
+// error: 5:16-33: 'address.code' is not supported on Polkadot

tests/contract_testcases/polkadot/builtins/address_code_02.sol

@@ -0,0 +1,8 @@
+contract UpgradeableProxy {
+    function _setImplementation(address newImplementation) public view {
+        assert(newImplementation.code.length != 0);
+    }
+}
+
+// ---- Expect: diagnostics ----
+// error: 3:16-33: 'address.code' is not supported on Polkadot

tests/evm.rs

@@ -249,7 +249,7 @@ fn ethereum_solidity_tests() {
         })
         .sum();
 
-    assert_eq!(errors, 1012);
+    assert_eq!(errors, 1010);
 }
 
 fn set_file_contents(source: &str, path: &Path) -> (FileResolver, Vec<String>) {