Added state mutability support to Signature and updated ABIManager for new constructor

Author: merendamattia

src/main/java/it/unipr/analysis/contract/Signature.java

@@ -17,6 +17,7 @@ public class Signature {
 	private final List<String> _outputParamTypes;
 	private final String _fullSignature;
 	private final String _selector;
+	private final String _stateMutability;
 	private Set<Statement> _entryPoints;
 	private Set<Statement> _exitPoints;
 	private boolean _isProtected;
@@ -31,9 +32,10 @@ public class Signature {
 	 * @param fullSignature    the full signature string
 	 * @param selector         the selector (first 4 bytes of the Keccak-256
 	 *                             hash)
+	 * @param stateMutability  the state mutability of the signature
 	 */
 	public Signature(String name, String type, List<String> inputParamTypes, List<String> outputParamTypes,
-			String fullSignature, String selector) {
+			String fullSignature, String selector, String stateMutability) {
 		this._name = name;
 		this._type = type;
 		this._inputParamTypes = inputParamTypes;
@@ -43,6 +45,23 @@ public Signature(String name, String type, List<String> inputParamTypes, List<St
 		this._entryPoints = new HashSet<>();
 		this._exitPoints = new HashSet<>();
 		this._isProtected = false;
+		this._stateMutability = stateMutability;
+	}
+
+	/**
+	 * Constructs a Signature with the specified properties.
+	 *
+	 * @param name             the name of the function or event
+	 * @param type             the type (e.g., "function", "event")
+	 * @param inputParamTypes  the list of input parameter types
+	 * @param outputParamTypes the list of output parameter types
+	 * @param fullSignature    the full signature string
+	 * @param selector         the selector (first 4 bytes of the Keccak-256
+	 *                             hash)
+	 */
+	public Signature(String name, String type, List<String> inputParamTypes, List<String> outputParamTypes,
+			String fullSignature, String selector) {
+		this(name, type, inputParamTypes, outputParamTypes, fullSignature, selector, "view");
 	}
 
 	/**
@@ -117,6 +136,15 @@ public Set<Statement> getEntryPoints() {
 		return _entryPoints;
 	}
 
+	/**
+	 * Gets the state mutability for this signature.
+	 *
+	 * @return the state mutability
+	 */
+	public String getStateMutability() {
+		return _stateMutability;
+	}
+
 	/**
 	 * Sets the entry points for this signature.
 	 *
@@ -196,12 +224,13 @@ public boolean equals(Object obj) {
 				Objects.equals(_type, other._type) &&
 				Objects.equals(_fullSignature, other._fullSignature) &&
 				Objects.equals(_selector, other._selector) &&
-				_isProtected == other._isProtected;
+				_isProtected == other._isProtected &&
+				Objects.equals(_stateMutability, other._stateMutability);
 	}
 
 	@Override
 	public int hashCode() {
-		return Objects.hash(_name, _type, _fullSignature, _selector, _isProtected);
+		return Objects.hash(_name, _type, _fullSignature, _selector, _isProtected, _stateMutability);
 	}
 
 	/**
@@ -244,6 +273,8 @@ public JSONObject toJson() {
 
 		json.put("is_protected", _isProtected);
 
+		json.put("state_mutability", _stateMutability);
+
 		return json;
 	}
 

src/main/java/it/unipr/utils/ABIManager.java

@@ -72,7 +72,6 @@ public static Set<Signature> parseEventsFromABI(Path abi) {
 	}
 
 	private static Set<Signature> parseABI(Path abi, String type) {
-		Set<String> stateMutabilityForbidden = Set.of("view", "pure");
 		Set<Signature> signatures = new HashSet<>();
 		String abiJson;
 
@@ -88,13 +87,10 @@ private static Set<Signature> parseABI(Path abi, String type) {
 		for (int i = 0; i < abiArray.length(); i++) {
 			JSONObject obj = abiArray.getJSONObject(i);
 			if (obj.has("type") && obj.getString("type").equals(type)) {
-
-				/* We do not need to collect global variables */
-				if (obj.has("stateMutability")
-						&& stateMutabilityForbidden.contains(obj.getString("stateMutability")))
-					continue;
-
 				String functionName = obj.getString("name");
+				String functionStateMutability = "view"; // default
+				if (obj.has("stateMutability"))
+					functionStateMutability = obj.getString("stateMutability");
 				JSONArray inputs = obj.getJSONArray("inputs");
 				JSONArray outputs = obj.optJSONArray("outputs");
 
@@ -122,7 +118,8 @@ private static Set<Signature> parseABI(Path abi, String type) {
 				String fullSignature = signatureBuilder.toString();
 				String selector = getFunctionSelector(fullSignature);
 
-				signatures.add(new Signature(functionName, type, paramTypes, outputTypes, fullSignature, selector));
+				signatures.add(new Signature(functionName, type, paramTypes, outputTypes, fullSignature, selector,
+						functionStateMutability));
 			}
 		}
 		return signatures;