Added test mode functionality.

Author: merendamattia

src/main/java/it/unipr/EVMLiSA.java

@@ -149,6 +149,10 @@ public static void setTestMode() {
 		TEST_MODE = true;
 	}
 
+	public static boolean isInTestMode() {
+		return TEST_MODE;
+	}
+
 	/**
 	 * Executes the analysis workflow.
 	 *
@@ -244,7 +248,7 @@ public static void analyzeSetOfContracts(List<SmartContract> contracts) {
 
 		log.info("Finished analysis of {} contracts.", contracts.size());
 
-		Path outputDir = OUTPUT_DIRECTORY_PATH.resolve("contracts-" + System.currentTimeMillis());
+		Path outputDir = OUTPUT_DIRECTORY_PATH.resolve("set-of-contracts");
 		try {
 			Files.createDirectories(outputDir);
 			Files.writeString(
@@ -295,12 +299,12 @@ public static void buildCFG(SmartContract contract) {
 		contract.setExecutionTime(System.currentTimeMillis() - startTime);
 
 		log.info("[OUT] CFG of contract {} built.", contract.getName());
-
+		log.info("[IN] Computing statistics of contract {}.", contract.getName());
 		contract.setStatistics(
 				computeStatistics(checker, lisa, program));
 		contract.setCFG(checker.getComputedCFG());
 
-		log.debug("Contract {} statistics: {}", contract.getAddress(), contract.getStatistics());
+		log.debug("[OUT] Contract {} statistics: {}", contract.getAddress(), contract.getStatistics());
 
 		if (TEST_MODE)
 			return;
@@ -674,6 +678,8 @@ private void setupGlobalOptions(CommandLine cmd) {
 			EVMAbstractState.setUseStorageLive();
 		if (cmd.hasOption("etherscan-api-key"))
 			EVMFrontend.setEtherscanAPIKey(cmd.getOptionValue("etherscan-api-key"));
+		if (cmd.hasOption("test-mode"))
+			EVMLiSA.setTestMode();
 	}
 
 	private Options getOptions() {
@@ -714,6 +720,7 @@ private Options getOptions() {
 				.required(false)
 				.hasArg(true)
 				.build();
+
 		Option abiOption = Option.builder()
 				.longOpt("abi")
 				.desc("ABI of the bytecode to be analyzed (JSON format).")
@@ -798,18 +805,11 @@ private Options getOptions() {
 				.hasArg(true)
 				.build();
 
-		Option crossChainBytecodeDirectoryPathOption = Option.builder()
-				.longOpt("bytecode-directory-path")
-				.desc("Directory path of bytecode files.")
-				.required(false)
-				.hasArg(true)
-				.build();
-
-		Option crossChainAbiDirectoryPathOption = Option.builder()
-				.longOpt("abi-directory-path")
-				.desc("Directory path of abi files.")
+		Option useTestModeOption = Option.builder()
+				.longOpt("test-mode")
+				.desc("Use the test mode (i.e., do not compute functions and events).")
 				.required(false)
-				.hasArg(true)
+				.hasArg(false)
 				.build();
 
 		options.addOption(addressOption);
@@ -829,8 +829,7 @@ private Options getOptions() {
 		options.addOption(outputDirectoryPathOption);
 		options.addOption(etherscanAPIKeyOption);
 		options.addOption(abiOption);
-		options.addOption(crossChainBytecodeDirectoryPathOption);
-		options.addOption(crossChainAbiDirectoryPathOption);
+		options.addOption(useTestModeOption);
 
 		return options;
 	}

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

@@ -1,5 +1,6 @@
 package it.unipr.analysis.contract;
 
+import it.unipr.EVMLiSA;
 import it.unipr.cfg.EVMCFG;
 import it.unipr.cfg.push.Push;
 import it.unipr.frontend.EVMFrontend;
@@ -779,9 +780,25 @@ public JSONObject toJson() {
 		JSONObject jsonObject = new JSONObject();
 
 		jsonObject.put("address", _address != null ? _address : new JSONArray());
-		jsonObject.put("bytecode", _bytecode != null ? _bytecode : new JSONArray());
-		jsonObject.put("mnemonic_bytecode", _mnemonicBytecode != null ? _mnemonicBytecode : new JSONArray());
-		jsonObject.put("abi", _abi != null ? _abi : new JSONArray());
+		if (!EVMLiSA.isInTestMode()) {
+			jsonObject.put("bytecode", _bytecode != null ? _bytecode : new JSONArray());
+			jsonObject.put("mnemonic_bytecode", _mnemonicBytecode != null ? _mnemonicBytecode : new JSONArray());
+			jsonObject.put("abi", _abi != null ? _abi : new JSONArray());
+
+			jsonObject.put("basic_blocks",
+					_basicBlocks != null ? JSONManager.basicBlocksToJson(this) : new JSONArray());
+			JSONArray functionsArray = new JSONArray();
+			if (_functionsSignature != null && !_functionsSignature.isEmpty())
+				for (Signature signature : _functionsSignature)
+					functionsArray.put(signature.toJson());
+			jsonObject.put("functions_signature", !functionsArray.isEmpty() ? functionsArray : new JSONArray());
+
+			JSONArray eventsArray = new JSONArray();
+			if (_eventsSignature != null && !_eventsSignature.isEmpty())
+				for (Signature signature : _eventsSignature)
+					eventsArray.put(signature.toJson());
+			jsonObject.put("events_signature", !eventsArray.isEmpty() ? eventsArray : new JSONArray());
+		}
 
 		jsonObject.put("working_directory", _workingDirectory.toString());
 
@@ -794,26 +811,11 @@ public JSONObject toJson() {
 
 		jsonObject.put("vulnerabilities", _vulnerabilities != null ? _vulnerabilities.toJson() : new JSONArray());
 
-		jsonObject.put("basic_blocks",
-				_basicBlocks != null ? JSONManager.basicBlocksToJson(this) : new JSONArray());
-
 		jsonObject.put("basic_blocks_pc", _basicBlocks != null ? BasicBlock.basicBlocksToLongArrayToString(
 				BasicBlock.basicBlocksToLongArray(_basicBlocks)) : new JSONArray());
 
 		jsonObject.put("execution_time", _executionTime);
 
-		JSONArray functionsArray = new JSONArray();
-		if (_functionsSignature != null && !_functionsSignature.isEmpty())
-			for (Signature signature : _functionsSignature)
-				functionsArray.put(signature.toJson());
-		jsonObject.put("functions_signature", !functionsArray.isEmpty() ? functionsArray : new JSONArray());
-
-		JSONArray eventsArray = new JSONArray();
-		if (_eventsSignature != null && !_eventsSignature.isEmpty())
-			for (Signature signature : _eventsSignature)
-				eventsArray.put(signature.toJson());
-		jsonObject.put("events_signature", !eventsArray.isEmpty() ? eventsArray : new JSONArray());
-
 		return jsonObject;
 	}