Add flag to make serialization optional

src/main/java/cmu/pasta/mu2/diff/guidance/DiffMutationReproGuidance.java

@@ -56,14 +56,18 @@ public class DiffMutationReproGuidance extends DiffReproGuidance {
      */
     private static ArraySet runMutants = new ArraySet();
 
+    private boolean serializeIn;
+    private boolean serializeOut;
+
     private File reportFile;
 
-    public DiffMutationReproGuidance(File inputFile, File traceDir, MutationClassLoaders mcls, File resultsDir) throws IOException {
-        super(inputFile, traceDir);
+    public DiffMutationReproGuidance(File inputFile, File traceDir, MutationClassLoaders mcls, boolean serialIn, boolean serialOut, File resultsDir) throws IOException {
+        super(inputFile, traceDir, serialIn, serialOut);
         cclOutcomes = new ArrayList<>();
         MCLs = mcls;
         ind = -1;
-
+        serializeIn = serialIn;
+        serializeOut = serialOut;
         reportFile = new File(resultsDir, "mutate-repro-out.txt");
         this.optLevel = MCLs.getCartographyClassLoader().getOptLevel();
     }
@@ -75,7 +79,6 @@ public void run(TestClass testClass, FrameworkMethod method, Object[] args) thro
 
         recentOutcomes.clear();
         cmpTo = null;
-
         ind++;
 
         // run CCL
@@ -95,7 +98,8 @@ public void run(TestClass testClass, FrameworkMethod method, Object[] args) thro
         // set up info
         cmpTo = new ArrayList<>(recentOutcomes);
         cclOutcomes.add(cmpTo.get(0));
-        byte[] argBytes = Serializer.serialize(args);
+        byte[] argBytes = null;
+        if(serializeIn) argBytes = Serializer.serialize(args);
         recentOutcomes.clear();
 
         for (MutationInstance mutationInstance : MCLs.getCartographyClassLoader().getMutationInstances()) {
@@ -107,7 +111,8 @@ public void run(TestClass testClass, FrameworkMethod method, Object[] args) thro
                 continue;
             }
 
-            MutationRunInfo mri = new MutationRunInfo(MCLs, mutationInstance, testClass, argBytes, args, method);
+            MutationRunInfo mri = new MutationRunInfo(MCLs, mutationInstance, testClass, args, method);
+            if(serializeIn) mri = new MutationRunInfo(MCLs, mutationInstance, testClass, argBytes, args, method);
 
             // run with MCL
             System.out.println("Running Mutant " + mutationInstance);

src/main/java/cmu/pasta/mu2/diff/guidance/DiffReproGuidance.java

@@ -18,20 +18,24 @@ public class DiffReproGuidance extends ReproGuidance implements DiffGuidance {
     private Method compare;
     protected List<Outcome> cmpTo;
     public static final List<Outcome> recentOutcomes = new ArrayList<>();
+    private boolean serializeIn;
+    private boolean serializeOut;
 
-    public DiffReproGuidance(File inputFile, File traceDir) throws IOException {
+    public DiffReproGuidance(File inputFile, File traceDir, boolean serialIn, boolean serialOut) throws IOException {
         super(inputFile, traceDir);
         cmpTo = null;
         recentOutcomes.clear();
+        serializeIn = serialIn;
+        serializeOut = serialOut;
         try {
             compare = Objects.class.getMethod("equals", Object.class, Object.class);
         } catch (NoSuchMethodException e) {
             e.printStackTrace();
         }
     }
 
-    public DiffReproGuidance(File inputFile, File traceDir, List<Outcome> cmpRes) throws IOException {
-        this(inputFile, traceDir);
+    public DiffReproGuidance(File inputFile, File traceDir, List<Outcome> cmpRes, boolean serialIn, boolean serialOut) throws IOException {
+        this(inputFile, traceDir, serialIn, serialOut);
         cmpTo = cmpRes;
     }
 
@@ -50,12 +54,14 @@ public void run(TestClass testClass, FrameworkMethod method, Object[] args) thro
             return;
         }
 
-        // use serialization to load both outputs with the same ClassLoader
-        //TODO may not want serialization for all diff repros
+        // optionally use serialization to load both outputs with the same ClassLoader
         Outcome cmpOut = cmpTo.get(recentOutcomes.size() - 1);
-        ClassLoader cmpCL = compare.getDeclaringClass().getClassLoader();
-        Outcome cmpSerial = new Outcome(Serializer.translate(cmpOut.output, cmpCL), cmpOut.thrown);
-        Outcome outSerial = new Outcome(Serializer.translate(out.output, cmpCL), out.thrown);
+        Outcome cmpSerial = cmpOut, outSerial = out;
+        if(serializeOut) {
+            ClassLoader cmpCL = compare.getDeclaringClass().getClassLoader();
+            cmpSerial = new Outcome(Serializer.translate(cmpOut.output, cmpCL), cmpOut.thrown);
+            outSerial = new Outcome(Serializer.translate(out.output, cmpCL), out.thrown);
+        }
 
         if (!Outcome.same(cmpSerial, outSerial, compare)) {
             throw new DiffException(cmpTo.get(recentOutcomes.size() - 1), out);

src/main/java/cmu/pasta/mu2/diff/plugin/DiffGoal.java

@@ -156,6 +156,12 @@ public class DiffGoal extends AbstractMojo {
     @Parameter(property="optLevel", defaultValue = "EXECUTION")
     private String optLevel;
 
+    @Parameter(property="disableInputSerialization", defaultValue = "false")
+    private boolean disableInputSerialization;
+
+    @Parameter(property="disableOutputSerialization", defaultValue = "false")
+    private boolean disableOutputSerialization;
+
     @Override
     public void execute() throws MojoExecutionException, MojoFailureException {
         ClassLoader loader;
@@ -203,7 +209,7 @@ public void execute() throws MojoExecutionException, MojoFailureException {
             }
             MutationClassLoaders mcl = new MutationClassLoaders(classPath, includes, targetIncludes, ol, baseClassLoader);
             loader = mcl.getCartographyClassLoader();
-            guidance = new MutationGuidance(targetName, mcl, duration, trials, resultsDir, seedsDir, rnd);
+            guidance = new MutationGuidance(targetName, mcl, duration, trials, resultsDir, seedsDir, rnd, !disableInputSerialization, !disableOutputSerialization);
         } catch (DependencyResolutionRequiredException | MalformedURLException e) {
             throw new MojoExecutionException("Could not get project classpath", e);
         } catch (FileNotFoundException e) {

src/main/java/cmu/pasta/mu2/diff/plugin/MutateDiffGoal.java

@@ -71,6 +71,12 @@ public class MutateDiffGoal extends AbstractMojo {
     @Parameter(property="optLevel", defaultValue = "execution")
     private String optLevel;
 
+    @Parameter(property="disableInputSerialization", defaultValue = "false")
+    private boolean disableInputSerialization;
+
+    @Parameter(property="disableOutputSerialization", defaultValue = "false")
+    private boolean disableOutputSerialization;
+
     @Override
     public void execute() throws MojoExecutionException, MojoFailureException {
         Log log = getLog();
@@ -98,7 +104,7 @@ public void execute() throws MojoExecutionException, MojoFailureException {
 
             // Run initial test to compute mutants dynamically
             System.out.println("Starting Initial Run:");
-            DiffMutationReproGuidance dmrg = new DiffMutationReproGuidance(input, null, mcls, resultsDir);
+            DiffMutationReproGuidance dmrg = new DiffMutationReproGuidance(input, null, mcls, !disableInputSerialization, !disableOutputSerialization, resultsDir);
             dmrg.setStopOnFailure(true);
             Result result = GuidedFuzzing.run(testClassName, testMethod, ccl, dmrg, null);
             if (!result.wasSuccessful()) {

src/main/java/cmu/pasta/mu2/diff/plugin/ReproDiffGoal.java

@@ -140,6 +140,12 @@ public class ReproDiffGoal extends AbstractMojo {
     @Parameter(property="dumpArgsDir")
     private String dumpArgsDir;
 
+    @Parameter(property="disableInputSerialization", defaultValue = "false")
+    private boolean disableInputSerialization;
+
+    @Parameter(property="disableOutputSerialization", defaultValue = "false")
+    private boolean disableOutputSerialization;
+
     @Override
     public void execute() throws MojoExecutionException, MojoFailureException {
         ClassLoader loader;
@@ -187,7 +193,7 @@ public void execute() throws MojoExecutionException, MojoFailureException {
         }
 
         try {
-            guidance = new DiffReproGuidance(inputFile, null);
+            guidance = new DiffReproGuidance(inputFile, null, !disableInputSerialization, !disableOutputSerialization);
             result = GuidedFuzzing.run(testClassName, testMethod, loader, guidance, out);
         } catch (ClassNotFoundException e) {
             throw new MojoExecutionException("Could not load test class", e);

src/main/java/cmu/pasta/mu2/fuzz/MutationGuidance.java

@@ -98,15 +98,20 @@ public class MutationGuidance extends ZestGuidance implements DiffGuidance {
 
   private final List<String> exceptions = new ArrayList<>();
 
+  private boolean serializeIn;
+  private boolean serializeOut;
+
   public MutationGuidance(String testName, MutationClassLoaders mutationClassLoaders,
-      Duration duration, Long trials, File outputDirectory, File seedInputDir, Random rand)
+      Duration duration, Long trials, File outputDirectory, File seedInputDir, Random rand, boolean serialIn, boolean serialOut)
       throws IOException {
     super(testName, duration, trials, outputDirectory, seedInputDir, rand);
     this.mutationClassLoaders = mutationClassLoaders;
     this.totalCoverage = new MutationCoverage();
     this.runCoverage = new MutationCoverage();
     this.validCoverage = new MutationCoverage();
     this.optLevel = mutationClassLoaders.getCartographyClassLoader().getOptLevel();
+    serializeIn = serialIn;
+    serializeOut = serialOut;
     try {
       compare = Objects.class.getMethod("equals", Object.class, Object.class);
     } catch (NoSuchMethodException e) {
@@ -160,7 +165,8 @@ public void run(TestClass testClass, FrameworkMethod method, Object[] args) thro
 
     // set up info
     long trialTime = System.currentTimeMillis() - startTime;
-    byte[] argBytes = Serializer.serialize(args);
+    byte[] argBytes = null;
+    if(serializeIn) argBytes = Serializer.serialize(args);
     int run = 1;
 
     for (MutationInstance mutationInstance : getMutationInstances()) {
@@ -176,7 +182,8 @@ public void run(TestClass testClass, FrameworkMethod method, Object[] args) thro
       run += 1;
       mutationInstance.resetTimer();
 
-      MutationRunInfo mri = new MutationRunInfo(mutationClassLoaders, mutationInstance, testClass, argBytes, args, method);
+      MutationRunInfo mri = new MutationRunInfo(mutationClassLoaders, mutationInstance, testClass, args, method);
+      if(serializeIn) mri = new MutationRunInfo(mutationClassLoaders, mutationInstance, testClass, argBytes, args, method);
 
       // run with MCL
       Outcome mclOutcome;
@@ -185,8 +192,10 @@ public void run(TestClass testClass, FrameworkMethod method, Object[] args) thro
         dtr.run();
         if(dtr.getOutput() == null) mclOutcome = new Outcome(null, null);
         else {
-          mclOutcome = new Outcome(Serializer.translate(dtr.getOutput(),
+          if(serializeOut)
+            mclOutcome = new Outcome(Serializer.translate(dtr.getOutput(),
                   mutationClassLoaders.getCartographyClassLoader()), null);
+          else mclOutcome = new Outcome(dtr.getOutput(), null);
         }
       } catch (InstrumentationException e) {
         throw new GuidanceException(e);

src/main/java/cmu/pasta/mu2/fuzz/MutationRunInfo.java

@@ -18,6 +18,13 @@ public class MutationRunInfo {
     public FrameworkMethod method;
     public Object[] args;
 
+    public MutationRunInfo(MutationClassLoaders MCLs, MutationInstance mutationInstance, TestClass testClass, Object[] a, FrameworkMethod fm) throws ClassNotFoundException {
+        MutationClassLoader mcl = MCLs.getMutationClassLoader(mutationInstance);
+        clazz = Class.forName(testClass.getName(), true, mcl);
+        method = fm;
+        args = a;
+    }
+
     public MutationRunInfo(MutationClassLoaders MCLs, MutationInstance mutationInstance, TestClass testClass, byte[] argBytes, Object[] oArgs, FrameworkMethod fm) throws ClassNotFoundException, NoSuchMethodException, IOException {
         // load class with MCL
         MutationClassLoader mcl = MCLs.getMutationClassLoader(mutationInstance);

src/test/java/cmu/pasta/mu2/diff/DiffIT.java

@@ -37,7 +37,7 @@ private static class ProbedMutationGuidance extends MutationGuidance {
         List<Integer> inputHashes = new ArrayList<>();
 
         ProbedMutationGuidance(MutationClassLoaders mcls, long trials, Random rnd) throws IOException {
-            super(null, mcls,null, trials, resultsDir, null, rnd);
+            super(null, mcls,null, trials, resultsDir, null, rnd, true, true);
         }
 
         @Override

src/test/java/cmu/pasta/mu2/fuzz/MutationGuidanceIT.java

@@ -39,7 +39,7 @@ private static class ProbedMutationGuidance extends MutationGuidance {
     List<Integer> inputHashes = new ArrayList<>();
 
     ProbedMutationGuidance(MutationClassLoaders mcls, long trials, Random rnd) throws IOException {
-      super(null, mcls,null, trials, resultsDir, null, rnd);
+      super(null, mcls,null, trials, resultsDir, null, rnd, true, true);
     }
 
     @Override