update

kanaksad · Jul 23, 2024 · b433ec3 · b433ec3
1 parent 3f50938
commit b433ec3
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 4 deletions.
diff --git a/src/main/java/edu/ucr/cs/riple/taint/ucrtainting/UCRTaintingChecker.java b/src/main/java/edu/ucr/cs/riple/taint/ucrtainting/UCRTaintingChecker.java
@@ -34,12 +34,17 @@
 import com.sun.tools.javac.tree.JCTree;
 import edu.ucr.cs.riple.taint.ucrtainting.handlers.UnannotatedCodeHandler;
 import edu.ucr.cs.riple.taint.ucrtainting.serialization.SerializationService;
+import edu.ucr.cs.riple.taint.ucrtainting.serialization.location.ClassDeclarationLocation;
+import edu.ucr.cs.riple.taint.ucrtainting.serialization.visitors.LocationToJsonVisitor;
+import java.io.FileWriter;
+import java.io.IOException;
 import org.checkerframework.checker.compilermsgs.qual.CompilerMessageKey;
 import org.checkerframework.common.accumulation.AccumulationChecker;
 import org.checkerframework.framework.qual.StubFiles;
 import org.checkerframework.framework.source.SupportedOptions;
 import org.checkerframework.framework.type.AnnotatedTypeMirror;
 import org.checkerframework.javacutil.TreeUtils;
+import org.json.JSONObject;
 
 /** This is the entry point for pluggable type-checking. */
 @StubFiles({
@@ -310,4 +315,18 @@ private boolean shouldBeSkippedForAnnotationMemberSelection(Object source) {
     Symbol.ClassSymbol owner = fieldAccess.sym.enclClass();
     return owner != null && owner.isAnnotationType();
   }
+
+  public void serializeClassDeclaration(ClassDeclarationLocation location, FoundRequired pair) {
+    JSONObject ans = new JSONObject();
+    ans.put("pair", pair.toString());
+    ans.put("location", new LocationToJsonVisitor().visitClassDeclaration(location, null));
+    ans.put("tree", visitor.getCurrentPath().getLeaf().toString());
+    ans.put("file", visitor.getCurrentPath().getCompilationUnit().getSourceFile().getName());
+    // append to file /tmp/class_decl.json with try-resources
+    try (FileWriter file = new FileWriter("/tmp/class_decl.json", true)) {
+      file.write(ans.toString());
+    } catch (IOException e) {
+      System.err.println("Error writing to file");
+    }
+  }
 }
diff --git a/...va/edu/ucr/cs/riple/taint/ucrtainting/serialization/visitors/GenericMethodFixVisitor.java b/...va/edu/ucr/cs/riple/taint/ucrtainting/serialization/visitors/GenericMethodFixVisitor.java
@@ -31,6 +31,7 @@
 import com.sun.tools.javac.util.Context;
 import edu.ucr.cs.riple.taint.ucrtainting.FoundRequired;
 import edu.ucr.cs.riple.taint.ucrtainting.UCRTaintingAnnotatedTypeFactory;
+import edu.ucr.cs.riple.taint.ucrtainting.UCRTaintingChecker;
 import edu.ucr.cs.riple.taint.ucrtainting.serialization.Fix;
 import edu.ucr.cs.riple.taint.ucrtainting.serialization.TypeIndex;
 import edu.ucr.cs.riple.taint.ucrtainting.serialization.location.ClassDeclarationLocation;
@@ -270,7 +271,9 @@ public static Set<Fix> computeFixesOnClassDeclarationForRawType(
       ClassDeclarationLocation classDeclarationLocation =
           new ClassDeclarationLocation(classType, inheritedType);
       classDeclarationLocation.setTypeIndexSet(TypeIndex.setOf(index + 1, 0));
-      return Set.of(new Fix(classDeclarationLocation));
+      ((UCRTaintingChecker) typeFactory.getChecker())
+          .serializeClassDeclaration(classDeclarationLocation, pair);
+      return Set.of();
     } catch (Exception e) {
       return Set.of();
     }

diff --git a/...ava/edu/ucr/cs/riple/taint/ucrtainting/serialization/visitors/SpecializedFixComputer.java b/...ava/edu/ucr/cs/riple/taint/ucrtainting/serialization/visitors/SpecializedFixComputer.java
@@ -32,6 +32,7 @@
 import com.sun.tools.javac.util.Context;
 import edu.ucr.cs.riple.taint.ucrtainting.FoundRequired;
 import edu.ucr.cs.riple.taint.ucrtainting.UCRTaintingAnnotatedTypeFactory;
+import edu.ucr.cs.riple.taint.ucrtainting.UCRTaintingChecker;
 import edu.ucr.cs.riple.taint.ucrtainting.serialization.Fix;
 import edu.ucr.cs.riple.taint.ucrtainting.serialization.TypeIndex;
 import edu.ucr.cs.riple.taint.ucrtainting.serialization.location.ClassDeclarationLocation;
@@ -122,9 +123,9 @@ public Fix buildFixForElement(Element element, FoundRequired pair) {
                 classDeclarationLocation.setTypeIndexSet(
                     untaintedTypeMatchVisitor.visit(annotatedDeclaredType, pair.required, null));
               });
-      return classDeclarationLocation.getTypeIndexSet().isEmpty()
-          ? null
-          : new Fix(classDeclarationLocation);
+      ((UCRTaintingChecker) typeFactory.getChecker())
+          .serializeClassDeclaration(classDeclarationLocation, pair);
+      return null;
     }
     AnnotatedTypeMirror elementAnnotatedType = typeFactory.getAnnotatedType(element);
     // remove redundant indices.