test/java/io/Serializable/superclassDataLoss/SuperclassDataLossTest.java

*** 1,7 ****
  /*
!  * Copyright (c) 2000, Oracle and/or its affiliates. All rights reserved.
   * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   *
   * This code is free software; you can redistribute it and/or modify it
   * under the terms of the GNU General Public License version 2 only, as
   * published by the Free Software Foundation.
--- 1,7 ----
  /*
!  * Copyright (c) 2000, 2017, Oracle and/or its affiliates. All rights reserved.
   * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   *
   * This code is free software; you can redistribute it and/or modify it
   * under the terms of the GNU General Public License version 2 only, as
   * published by the Free Software Foundation.
*** 20,57 ****
   * or visit www.oracle.com if you need additional information or have any
   * questions.
   */
  
  /*
   * @bug 4325590
   * @summary Verify that superclass data is not lost when incoming superclass
   *          descriptor is matched with local class that is not a superclass of
   *          the deserialized instance's class.
   */
  
! import java.io.*;
! import java.net.*;
  
  class MixedSuperclassStream extends ObjectInputStream {
!     MixedSuperclassStream(InputStream in) throws IOException { super(in); }
  
      protected Class resolveClass(ObjectStreamClass desc)
          throws IOException, ClassNotFoundException
      {
          // resolve A's classdesc to class != B's superclass
          String name = desc.getName();
          if (name.equals("A")) {
!             return Class.forName(name, true, Test.ldr1);
          } else if (name.equals("B")) {
!             return Class.forName(name, true, Test.ldr2);
          } else {
!             return super.resolveClass(desc);
          }
      }
  }
  
! public class Test {
  
      static URLClassLoader ldr1, ldr2;
      static {
          try {
              ldr1 = new URLClassLoader(new URL[] { new URL("file:cb1.jar") });
--- 20,85 ----
   * or visit www.oracle.com if you need additional information or have any
   * questions.
   */
  
  /*
+  * @test
   * @bug 4325590
+  * @library /lib/testlibrary
+  * @build JarUtils A B
+  * @run main SuperclassDataLossTest
   * @summary Verify that superclass data is not lost when incoming superclass
   *          descriptor is matched with local class that is not a superclass of
   *          the deserialized instance's class.
   */
  
! import java.io.ByteArrayInputStream;
! import java.io.ByteArrayOutputStream;
! import java.io.InputStream;
! import java.io.IOException;
! import java.io.ObjectInputStream;
! import java.io.ObjectOutputStream;
! import java.io.ObjectStreamClass;
! import java.net.URL;
! import java.net.URLClassLoader;
! import java.net.MalformedURLException;
! import java.nio.file.Files;
! import java.nio.file.Path;
! import java.nio.file.Paths;
! import java.nio.file.StandardCopyOption;
  
  class MixedSuperclassStream extends ObjectInputStream {
!     private boolean ldr12A;
! 
!     MixedSuperclassStream(InputStream in, boolean ldr1First) throws IOException {
!         super(in);
!         this.ldr12A = ldr12A;
!     }
  
      protected Class resolveClass(ObjectStreamClass desc)
          throws IOException, ClassNotFoundException
      {
          // resolve A's classdesc to class != B's superclass
          String name = desc.getName();
+         if (ldr12A) {
              if (name.equals("A")) {
!                 return Class.forName(name, true, SuperclassDataLossTest.ldr1);
              } else if (name.equals("B")) {
!                 return Class.forName(name, true, SuperclassDataLossTest.ldr2);
!             }
          } else {
!             if (name.equals("B")) {
!                 return Class.forName(name, true, SuperclassDataLossTest.ldr1);
!             } else if (name.equals("A")) {
!                 return Class.forName(name, true, SuperclassDataLossTest.ldr2);
!             }
          }
+         return super.resolveClass(desc);
      }
  }
  
! public class SuperclassDataLossTest {
  
      static URLClassLoader ldr1, ldr2;
      static {
          try {
              ldr1 = new URLClassLoader(new URL[] { new URL("file:cb1.jar") });
*** 60,79 ****
              throw new Error();
          }
      }
  
      public static void main(String[] args) throws Exception {
          Runnable a = (Runnable) Class.forName("B", true, ldr1).newInstance();
          a.run();
  
          ByteArrayOutputStream bout = new ByteArrayOutputStream();
          ObjectOutputStream oout = new ObjectOutputStream(bout);
          oout.writeObject(a);
          oout.close();
  
          ByteArrayInputStream bin =
              new ByteArrayInputStream(bout.toByteArray());
!         ObjectInputStream oin = new MixedSuperclassStream(bin);
!         a = (Runnable) oin.readObject();
          a.run();
      }
  }
--- 88,125 ----
              throw new Error();
          }
      }
  
      public static void main(String[] args) throws Exception {
+         setup();
+ 
          Runnable a = (Runnable) Class.forName("B", true, ldr1).newInstance();
          a.run();
  
          ByteArrayOutputStream bout = new ByteArrayOutputStream();
          ObjectOutputStream oout = new ObjectOutputStream(bout);
          oout.writeObject(a);
          oout.close();
  
+         test(bout, true);
+         test(bout, false);
+ 
+         ldr1.close();
+         ldr2.close();
+     }
+ 
+     private static void test(ByteArrayOutputStream bout, boolean ldr12A) throws Exception {
          ByteArrayInputStream bin =
              new ByteArrayInputStream(bout.toByteArray());
!         ObjectInputStream oin = new MixedSuperclassStream(bin, ldr12A);
!         Runnable a = (Runnable) oin.readObject();
          a.run();
      }
+ 
+     private static void setup() throws Exception {
+         Path classes = Paths.get(System.getProperty("test.classes", ""));
+         JarUtils.createJarFile(Paths.get("cb1.jar"), classes,
+                 classes.resolve("A.class"), classes.resolve("B.class"));
+         Files.copy(Paths.get("cb1.jar"), Paths.get("cb2.jar"),
+                    StandardCopyOption.REPLACE_EXISTING);
+     }
  }