WIP

libsolidity/analysis/TypeChecker.cpp

@@ -231,6 +231,43 @@ bool TypeChecker::visit(ImportDirective const&)
 	return false;
 }
 
+void TypeChecker::endVisit(ContractDefinition const& _contract)
+{
+	if (
+		ASTPointer<Expression> const baseLocation = _contract.storageBaseLocationExpression()
+	)
+	{
+		if (!*baseLocation->annotation().isPure)
+		{
+			if (auto functionCall = dynamic_cast<FunctionCall const*>(baseLocation.get()))
+				if (
+					auto const* identifier = dynamic_cast<Identifier const*>(&functionCall->expression());
+					identifier && identifier->name() == "erc7201"
+				)
+					return;
+
+			m_errorReporter.typeError(
+				77_error,
+				baseLocation->location(),
+				"The contract base location must be an expression that can be evaluated at compilation time."
+			);
+		}
+		auto const* expressionType = type(*baseLocation);
+		BoolResult result = expressionType->isImplicitlyConvertibleTo(*TypeProvider::uint256());
+		if (!result)
+		{
+			m_errorReporter.typeErrorConcatenateDescriptions(
+				76_error,
+				baseLocation->location(),
+				"Contract storage base location must be "
+				"in range of type uint256. Current type is " +
+				expressionType->humanReadableName(),
+				result.message()
+			);
+		}
+	}
+}
+
 void TypeChecker::endVisit(InheritanceSpecifier const& _inheritance)
 {
 	auto base = dynamic_cast<ContractDefinition const*>(&dereference(_inheritance.name()));

libsolidity/analysis/TypeChecker.h

@@ -157,6 +157,7 @@ class TypeChecker: private ASTConstVisitor
 	void endVisit(ElementaryTypeNameExpression const& _expr) override;
 	void endVisit(Literal const& _literal) override;
 	void endVisit(UsingForDirective const& _usingForDirective) override;
+	void endVisit(ContractDefinition const& _contract) override;
 
 	void checkErrorAndEventParameters(CallableDeclaration const& _callable);
 

libsolidity/ast/AST.h

@@ -572,6 +572,8 @@ class ContractDefinition: public Declaration, public StructurallyDocumented, pub
 	/// Returns the ether receiver function or nullptr if no receive function was specified.
 	FunctionDefinition const* receiveFunction() const;
 
+	ASTPointer<Expression> const& storageBaseLocationExpression() const { return m_storageBaseLocationExpression; }
+
 	std::string fullyQualifiedName() const { return sourceUnitName() + ":" + name(); }
 
 	Type const* type() const override;