add local variable

Author: raintung

src/main/java/soot/asm/MethodBuilder.java

@@ -43,8 +43,8 @@
 import soot.tagkit.AnnotationDefaultTag;
 import soot.tagkit.AnnotationTag;
 import soot.tagkit.VisibilityAnnotationTag;
-import soot.tagkit.VisibilityParameterAnnotationTag;
 import soot.tagkit.VisibilityLocalVariableAnnotationTag;
+import soot.tagkit.VisibilityParameterAnnotationTag;
 
 /**
  * Soot method builder.
@@ -96,30 +96,30 @@ public void visitAttribute(Attribute attr) {
   }
 
   @Override
-  public AnnotationVisitor visitLocalVariableAnnotation(final int typeRef, final TypePath typePath,
-			final Label[] start, final Label[] end, final int[] index, final String descriptor, final boolean visible) {
-	final VisibilityAnnotationTag vat = new VisibilityAnnotationTag(
-					visible ? AnnotationConstants.RUNTIME_VISIBLE : AnnotationConstants.RUNTIME_INVISIBLE);
-	if (visible) {
-		if (visibleLocalVarAnnotations == null) {
-			visibleLocalVarAnnotations = new ArrayList<VisibilityAnnotationTag>(2);
-		}
-		visibleLocalVarAnnotations.add(vat);
-	} else {
-		if (invisibleLocalVarAnnotations == null) {
-			invisibleLocalVarAnnotations = new ArrayList<VisibilityAnnotationTag>(2);
-		}
-		invisibleLocalVarAnnotations.add(vat);
-	}
-	return new AnnotationElemBuilder() {
-		@Override
-		public void visitEnd() {
-			AnnotationTag annotTag = new AnnotationTag(desc, elems);
-			vat.addAnnotation(annotTag);
-		}
-	};
+  public AnnotationVisitor visitLocalVariableAnnotation(final int typeRef, final TypePath typePath, final Label[] start,
+      final Label[] end, final int[] index, final String descriptor, final boolean visible) {
+    final VisibilityAnnotationTag vat
+        = new VisibilityAnnotationTag(visible ? AnnotationConstants.RUNTIME_VISIBLE : AnnotationConstants.RUNTIME_INVISIBLE);
+    if (visible) {
+      if (visibleLocalVarAnnotations == null) {
+        visibleLocalVarAnnotations = new ArrayList<VisibilityAnnotationTag>(2);
+      }
+      visibleLocalVarAnnotations.add(vat);
+    } else {
+      if (invisibleLocalVarAnnotations == null) {
+        invisibleLocalVarAnnotations = new ArrayList<VisibilityAnnotationTag>(2);
+      }
+      invisibleLocalVarAnnotations.add(vat);
+    }
+    return new AnnotationElemBuilder() {
+      @Override
+      public void visitEnd() {
+        AnnotationTag annotTag = new AnnotationTag(desc, elems);
+        vat.addAnnotation(annotTag);
+      }
+    };
   }
-  
+
   @Override
   public AnnotationVisitor visitParameterAnnotation(int parameter, final String desc, boolean visible) {
     VisibilityAnnotationTag vat;
@@ -236,22 +236,22 @@ public void visitEnd() {
       }
       method.addTag(tag);
     }
-    if (visibleLocalVarAnnotations != null) { 
-        VisibilityLocalVariableAnnotationTag tag 
-            = new VisibilityLocalVariableAnnotationTag(visibleLocalVarAnnotations.size(), AnnotationConstants.RUNTIME_VISIBLE);
-  	  for (VisibilityAnnotationTag vat : visibleLocalVarAnnotations) {
-  			tag.addVisibilityAnnotation(vat);
-  	  }
-  	  method.addTag(tag);
-  	}
-  	if (invisibleLocalVarAnnotations != null) {
-  	  VisibilityLocalVariableAnnotationTag tag 
-  	      = new VisibilityLocalVariableAnnotationTag(visibleLocalVarAnnotations.size(), AnnotationConstants.RUNTIME_INVISIBLE);
-  	  for (VisibilityAnnotationTag vat : invisibleLocalVarAnnotations) {
-  			tag.addVisibilityAnnotation(vat);
-  	  }
-  	  method.addTag(tag);
-  	}
+    if (visibleLocalVarAnnotations != null) {
+      VisibilityLocalVariableAnnotationTag tag
+          = new VisibilityLocalVariableAnnotationTag(visibleLocalVarAnnotations.size(), AnnotationConstants.RUNTIME_VISIBLE);
+      for (VisibilityAnnotationTag vat : visibleLocalVarAnnotations) {
+        tag.addVisibilityAnnotation(vat);
+      }
+      method.addTag(tag);
+    }
+    if (invisibleLocalVarAnnotations != null) {
+      VisibilityLocalVariableAnnotationTag tag = new VisibilityLocalVariableAnnotationTag(visibleLocalVarAnnotations.size(),
+          AnnotationConstants.RUNTIME_INVISIBLE);
+      for (VisibilityAnnotationTag vat : invisibleLocalVarAnnotations) {
+        tag.addVisibilityAnnotation(vat);
+      }
+      method.addTag(tag);
+    }
     if (method.isConcrete()) {
       method.setSource(
           new AsmMethodSource(maxLocals, instructions, localVariables, tryCatchBlocks, scb.getKlass().moduleName));

src/main/java/soot/tagkit/VisibilityLocalVariableAnnotationTag.java

@@ -1,4 +1,5 @@
 package soot.tagkit;
+
 /*-
  * #%L
  * Soot - a J*va Optimization Framework
@@ -21,62 +22,61 @@
  * #L%
  */
 /**
- * Represents the visibility of an annotation attribute attached to method local variable.
- * Only mark the local variable tag different with{@link soot.tagkit.VisibilityParameterAnnotationTag}
+ * Represents the visibility of an annotation attribute attached to method local variable. Only mark the local variable tag
+ * different with{@link soot.tagkit.VisibilityParameterAnnotationTag}
  * 
  * @author raintung.li
  *
  */
 
 public class VisibilityLocalVariableAnnotationTag extends VisibilityParameterAnnotationTag {
-	
-	public VisibilityLocalVariableAnnotationTag(int num, int kind) {
-		super(num, kind);
-	}
 
-	// should also print here number of annotations and perhaps the annotations themselves
-	public String toString() {
-		int num_var = getVisibilityAnnotations()!=null ? getVisibilityAnnotations().size(): 0;
-		StringBuffer sb = new StringBuffer(
-				"Visibility LocalVariable Annotation: num Annotation: " + num_var + " kind: " + getKind());
-		if (num_var > 0) {
-			for (VisibilityAnnotationTag tag : getVisibilityAnnotations()) {
-				sb.append("\n");
-				if (tag != null) {
-					sb.append(tag.toString());
-				}
-			}
-		}
-		sb.append("\n");
-		return sb.toString();
-	}
-	
+  public VisibilityLocalVariableAnnotationTag(int num, int kind) {
+    super(num, kind);
+  }
+
+  // should also print here number of annotations and perhaps the annotations themselves
+  public String toString() {
+    int num_var = getVisibilityAnnotations() != null ? getVisibilityAnnotations().size() : 0;
+    StringBuffer sb
+        = new StringBuffer("Visibility LocalVariable Annotation: num Annotation: " + num_var + " kind: " + getKind());
+    if (num_var > 0) {
+      for (VisibilityAnnotationTag tag : getVisibilityAnnotations()) {
+        sb.append("\n");
+        if (tag != null) {
+          sb.append(tag.toString());
+        }
+      }
+    }
+    sb.append("\n");
+    return sb.toString();
+  }
+
+  /**
+   * Returns Local Variable tag name
+   * 
+   * @return string
+   */
+  public String getName() {
+    return "VisibilityLocalVariableAnnotationTag";
+  }
+
+  /**
+   * Returns Local Variable tag info
+   * 
+   * @return string
+   */
+  public String getInfo() {
+    return "VisibilityLocalVariableAnnotation";
+  }
 
-	/**
-	 * Returns Local Variable tag name
-	 * 
-	 * @return string
-	 */
-	public String getName() {
-		return "VisibilityLocalVariableAnnotationTag";
-	}
-	
-	/**
-	 * Returns Local Variable tag info
-	 * 
-	 * @return string
-	 */
-	public String getInfo() {
-		return "VisibilityLocalVariableAnnotation";
-	}
-	
-	/**
-	 * VisibilityLocalVariableAnnotationTag not support
-	 * 
-	 * @return 
-	 */
-	public byte[] getValue() {
-		throw new RuntimeException("VisibilityLocalVariableAnnotationTag has no value for bytecode");
-	}
+  /**
+   * VisibilityLocalVariableAnnotationTag not support
+   * 
+   * @return
+   */
+  public byte[] getValue() {
+    throw new RuntimeException("VisibilityLocalVariableAnnotationTag has no value for bytecode");
+  }
 
 }