test/jdk/sun/awt/shell/ShellFolderMemoryLeak.java
changeset 47216 71c04702a3d5
parent 40261 86a49ba76f52
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/test/jdk/sun/awt/shell/ShellFolderMemoryLeak.java	Tue Sep 12 19:03:39 2017 +0200
@@ -0,0 +1,227 @@
+/*
+ * Copyright (c) 2015, 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.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+
+/*
+ * @test
+ * @bug 8030099
+ * @summary Memory usage of java process increases
+            after calling Win32ShellFolder:listFiles
+            multiple times on some directory with
+            large number of files/folders
+ * @modules java.desktop/sun.awt.shell
+ * @requires (os.family == "windows")
+ * @run main/timeout=1000 ShellFolderMemoryLeak
+ */
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+import sun.awt.shell.ShellFolder;
+
+public class ShellFolderMemoryLeak {
+
+    private final static String tempDir = System.getProperty("java.io.tmpdir");
+    private static Process process;
+    public static void main(String[] args) throws Exception {
+        if (args.length == 0) {
+            boolean testResultParallel
+                    = createChildProcessWithParallelCollector();
+            String result = "";
+            if (!testResultParallel) {
+                result = "Test failed with Parallel collector";
+            }
+            boolean testResultDefault
+                    = createChildProcessWithDefaultCollector();
+            if (!testResultDefault && !testResultParallel) {
+                result += " and with default collector both.";
+            } else if (!testResultDefault) {
+                result = "Test failed with default collector";
+            }
+            if (!"".equals(result)) {
+                throw new RuntimeException(result);
+            }
+        } else {
+            testListFile(args[args.length - 1]);
+        }
+    }
+
+    public static boolean createChildProcessWithDefaultCollector()
+            throws Exception {
+        String testDirectory = "TestDirectory1";
+        testDirectory = tempDir + testDirectory +File.separator;
+        createTestData(testDirectory);
+        return runProcess("", testDirectory);
+    }
+
+    public static boolean createChildProcessWithParallelCollector()
+            throws Exception {
+        String testDirectory = "TestDirectory2";
+        testDirectory = tempDir + testDirectory +File.separator;
+        createTestData(testDirectory);
+        return runProcess(" -XX:+UseParallelGC", testDirectory);
+    }
+
+    public static boolean runProcess(String arg1, String arg2) throws Exception {
+        String javaPath = System.getProperty("java.home");
+        String classPathDir = System.getProperty("java.class.path");
+
+        //creating java process which run same class with different Xmx value
+        String command = javaPath + File.separator + "bin" + File.separator
+                + "java -Xmx256M" + arg1 + " -cp "
+                + classPathDir
+                + " --add-exports=java.desktop/sun.awt.shell=ALL-UNNAMED"
+                + " ShellFolderMemoryLeak " + arg2;
+        process = Runtime.getRuntime().exec(command);
+        BufferedReader input = null;
+        InputStream errorStream = null;
+        String line = null;
+        try {
+            int exitVal = process.waitFor();
+            input = new BufferedReader(new InputStreamReader(
+                    process.getInputStream()));
+            while ((line = input.readLine()) != null) {
+            }
+            errorStream = process.getErrorStream();
+            if (checkExceptions(errorStream) || exitVal != 0) {
+                return false;
+            }
+        } catch (IllegalThreadStateException e) {
+            throw new RuntimeException(e);
+        } finally {
+            if (input != null) {
+                input.close();
+            }
+            if (errorStream != null) {
+                errorStream.close();
+            }
+            process.destroy();
+        }
+        return true;
+    }
+
+    public static boolean checkExceptions(InputStream in) throws IOException {
+        String tempString;
+        int count = in.available();
+        boolean exception = false;
+        while (count > 0) {
+            byte[] b = new byte[count];
+            in.read(b);
+            tempString = new String(b);
+            if (!exception) {
+                exception = tempString.contains("RunTimeException");
+            }
+            count = in.available();
+        }
+        return exception;
+    }
+
+    private static void createTestData(String testDirectory) {
+        String folder = "folder_";
+        File testFolder = new File(testDirectory);
+        if (testFolder.exists()) {
+            clearTestData(testDirectory);
+        } else {
+            if (testFolder.mkdir()) {
+                for (int inx = 0; inx < 100; inx++) {
+                    new File(testFolder + File.separator + folder + inx).mkdir();
+                }
+            } else {
+                throw new RuntimeException("Failed to create testDirectory");
+            }
+        }
+    }
+
+    public static void deleteDirectory(File file)
+            throws IOException {
+
+        if (file.isDirectory()) {
+            if (file.list().length == 0) {
+                file.delete();
+            } else {
+                String files[] = file.list();
+                for (String temp : files) {
+                    File fileDelete = new File(file, temp);
+                    deleteDirectory(fileDelete);
+                }
+                if (file.list().length == 0) {
+                    file.delete();
+                }
+            }
+        }
+    }
+
+    private static void testListFile(String testDirectory) {
+        try {
+            int mb = 1024 * 1024;
+            ShellFolder folder = ShellFolder.getShellFolder(
+                    new File(testDirectory));
+            Runtime instance = Runtime.getRuntime();
+
+            //Memory used before calling listFiles
+            long startmem = instance.totalMemory() - instance.freeMemory();
+            long start = System.currentTimeMillis();
+            long endmem = 0;
+
+            //Calling listFiles for 5 minutes with sleep of 10 ms.
+            while ((System.currentTimeMillis() - start) < 300000) {
+                try {
+                    folder.listFiles();
+                    Thread.sleep(10);
+                    endmem = instance.totalMemory() - instance.freeMemory();
+                } catch (InterruptedException ex) {
+                    Logger.getLogger(ShellFolderMemoryLeak.class.getName())
+                            .log(Level.SEVERE, "InterruptedException", ex);
+                }
+            }
+
+            //Total increase in memory after 5 minutes
+            long result = (endmem - startmem) / mb;
+
+            if (result > 100) {
+                clearTestData(testDirectory);
+                throw new RuntimeException("Test Failed");
+            }
+            clearTestData(testDirectory);
+        } catch (FileNotFoundException ex) {
+            if(process != null && process.isAlive()) {
+                process.destroy();
+            }
+            Logger.getLogger(ShellFolderMemoryLeak.class.getName())
+                    .log(Level.SEVERE, "File Not Found Exception", ex);
+        }
+    }
+
+    private static void clearTestData(String testDirectory) {
+        File testFolder = new File(testDirectory);
+        try {
+            deleteDirectory(testFolder);
+        } catch (IOException ex) {
+            Logger.getLogger(ShellFolderMemoryLeak.class.getName())
+                    .log(Level.SEVERE, "Unable to delete files", ex);
+        }
+    }
+}