Merge pull request #2585 from crytic/dev-storage-location

Add StateVariable location

slither/core/variables/state_variable.py

@@ -12,6 +12,7 @@ class StateVariable(ContractLevel, Variable):
     def __init__(self) -> None:
         super().__init__()
         self._node_initialization: Optional["Node"] = None
+        self._location: Optional[str] = None
 
     def is_declared_by(self, contract: "Contract") -> bool:
         """
@@ -21,6 +22,19 @@ def is_declared_by(self, contract: "Contract") -> bool:
         """
         return self.contract == contract
 
+    def set_location(self, loc: str) -> None:
+        self._location = loc
+
+    @property
+    def location(self) -> Optional[str]:
+        """
+            Variable Location
+            Can be default or transient
+        Returns:
+            (str)
+        """
+        return self._location
+
     # endregion
     ###################################################################################
     ###################################################################################

slither/solc_parsing/variables/state_variable.py

@@ -13,3 +13,18 @@ def underlying_variable(self) -> StateVariable:
         # Todo: Not sure how to overcome this with mypy
         assert isinstance(self._variable, StateVariable)
         return self._variable
+
+    def _analyze_variable_attributes(self, attributes: Dict) -> None:
+        """
+        Variable Location
+        Can be default or transient
+        """
+        if "storageLocation" in attributes:
+            self.underlying_variable.set_location(attributes["storageLocation"])
+        else:
+            # We don't have to support legacy ast
+            # as transient location was added in 0.8.28
+            # and we know it must be default
+            self.underlying_variable.set_location("default")
+
+        super()._analyze_variable_attributes(attributes)