Add config option to emit try-catch ranges in assembler

Col-E · Sep 4, 2024 · d096b31 · d096b31
1 parent 81e49e6
commit d096b31
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 2 deletions.
diff --git a/...core/src/main/java/software/coley/recaf/services/assembler/AbstractAssemblerPipeline.java b/...core/src/main/java/software/coley/recaf/services/assembler/AbstractAssemblerPipeline.java
@@ -46,6 +46,10 @@ public AbstractAssemblerPipeline(@Nonnull AssemblerPipelineGeneralConfig general
 
 	private void refreshContext() {
 		context = new PrintContext<>(generalConfig.getDisassemblyIndent().getValue());
+
+		// Enable comments that outline where try-catch ranges begin/end.
+		if (pipelineConfig instanceof JvmAssemblerPipelineConfig jvmConfig && jvmConfig.emitTryRangeComments())
+			context.setDebugTryCatchRanges(true);
 	}
 
 	@Nonnull

diff --git a/...f-core/src/main/java/software/coley/recaf/services/assembler/AssemblerPipelineConfig.java b/...f-core/src/main/java/software/coley/recaf/services/assembler/AssemblerPipelineConfig.java
@@ -6,6 +6,8 @@
  * Assembler pipeline config outline.
  *
  * @author Justus Garbe
+ * @see AndroidAssemblerPipelineConfig
+ * @see JvmAssemblerPipelineConfig
  */
 public interface AssemblerPipelineConfig extends ConfigContainer {
 	/**

diff --git a/...src/main/java/software/coley/recaf/services/assembler/AssemblerPipelineGeneralConfig.java b/...src/main/java/software/coley/recaf/services/assembler/AssemblerPipelineGeneralConfig.java
@@ -29,7 +29,7 @@ public AssemblerPipelineGeneralConfig() {
 	}
 
 	/**
-	 * @return Indentation string.
+	 * @return String of a single indentation level.
 	 */
 	@Nonnull
 	public ObservableString getDisassemblyIndent() {

diff --git a/...ore/src/main/java/software/coley/recaf/services/assembler/JvmAssemblerPipelineConfig.java b/...ore/src/main/java/software/coley/recaf/services/assembler/JvmAssemblerPipelineConfig.java
@@ -17,12 +17,15 @@
 public class JvmAssemblerPipelineConfig extends BasicConfigContainer implements ServiceConfig, AssemblerPipelineConfig {
 	private final ObservableBoolean valueAnalysis = new ObservableBoolean(true);
 	private final ObservableBoolean simulateJvmCalls = new ObservableBoolean(true);
+	private final ObservableBoolean tryRangeComments = new ObservableBoolean(true);
+
 
 	@Inject
 	public JvmAssemblerPipelineConfig() {
 		super(ConfigGroups.SERVICE_ASSEMBLER, JvmAssemblerPipeline.SERVICE_ID + CONFIG_SUFFIX);
 		addValue(new BasicConfigValue<>("value-analysis", boolean.class, valueAnalysis));
 		addValue(new BasicConfigValue<>("simulate-jvm-calls", boolean.class, simulateJvmCalls));
+		addValue(new BasicConfigValue<>("try-range-comments", boolean.class, tryRangeComments));
 	}
 
 	@Override
@@ -32,6 +35,13 @@ public boolean isValueAnalysisEnabled() {
 
 	@Override
 	public boolean isSimulatingCommonJvmCalls() {
-		return simulateJvmCalls.hasValue();
+		return simulateJvmCalls.getValue();
+	}
+
+	/**
+	 * @return {@code true} to emit comments for where try-catch ranges start/end/catch.
+	 */
+	public boolean emitTryRangeComments() {
+		return tryRangeComments.getValue();
 	}
 }
diff --git a/recaf-ui/src/main/resources/translations/en_US.lang b/recaf-ui/src/main/resources/translations/en_US.lang
@@ -512,6 +512,7 @@ service.assembler.dalvik-assembler-config.simulate-jvm-calls=Simulate common JVM
 service.assembler.jvm-assembler-config=JVM
 service.assembler.jvm-assembler-config.value-analysis=Enable value analysis
 service.assembler.jvm-assembler-config.simulate-jvm-calls=Simulate common JVM calls
+service.assembler.jvm-assembler-config.try-range-comments=Emit try-catch range comments
 service.assembler.flow-lines-config=Control flow lines
 service.assembler.flow-lines-config.connection-mode=Line mode
 service.assembler.flow-lines-config.render-mode=Render mode