undefect. CWE-407 — 63 sites patched across 27 ecosystems
Authors: russell@unturf.com · brackishbert@gmail.com · foxhop.net · TimeHexOn.com Patches, unit tests, benchmarks, whitepaper, and outreach briefs. Public domain — no copyright claimed. Use freely.
This commit is contained in:
commit
0a580b313d
70422 changed files with 17213626 additions and 0 deletions
227
test/jdk/sun/awt/shell/ShellFolderMemoryLeak.java
Normal file
227
test/jdk/sun/awt/shell/ShellFolderMemoryLeak.java
Normal file
|
|
@ -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);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue