/*
* Copyright (c) 2008, 2011, 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 4313887 6838333 7006126 7023034
* @summary Unit test for Files.createTempXXX
* @library ..
*/
import java.nio.file.*;
import static java.nio.file.StandardOpenOption.*;
import java.nio.file.attribute.*;
import java.io.IOException;
import java.util.Set;
public class TemporaryFiles {
static void checkInDirectory(Path file, Path dir) {
if (dir == null)
dir = Paths.get(System.getProperty("java.io.tmpdir"));
if (!file.getParent().equals(dir))
throw new RuntimeException("Not in expected directory");
}
static void testTempFile(String prefix, String suffix, Path dir)
throws IOException
{
Path file = (dir == null) ?
Files.createTempFile(prefix, suffix) :
Files.createTempFile(dir, prefix, suffix);
try {
// check file name
String name = file.getFileName().toString();
if (prefix != null && !name.startsWith(prefix))
throw new RuntimeException("Should start with " + prefix);
if (suffix == null && !name.endsWith(".tmp"))
throw new RuntimeException("Should end with .tmp");
if (suffix != null && !name.endsWith(suffix))
throw new RuntimeException("Should end with " + suffix);
// check file is in expected directory
checkInDirectory(file, dir);
// check that file can be opened for reading and writing
Files.newByteChannel(file, READ).close();
Files.newByteChannel(file, WRITE).close();
Files.newByteChannel(file, READ,WRITE).close();
// check file permissions are 0600 or more secure
if (Files.getFileStore(file).supportsFileAttributeView("posix")) {
Set<PosixFilePermission> perms = Files.getPosixFilePermissions(file);
perms.remove(PosixFilePermission.OWNER_READ);
perms.remove(PosixFilePermission.OWNER_WRITE);
if (!perms.isEmpty())
throw new RuntimeException("Temporary file is not secure");
}
} finally {
Files.delete(file);
}
}
static void testTempFile(String prefix, String suffix)
throws IOException
{
testTempFile(prefix, suffix, null);
}
static void testTempDirectory(String prefix, Path dir) throws IOException {
Path subdir = (dir == null) ?
Files.createTempDirectory(prefix) :
Files.createTempDirectory(dir, prefix);
try {
// check file name
String name = subdir.getFileName().toString();
if (prefix != null && !name.startsWith(prefix))
throw new RuntimeException("Should start with " + prefix);
// check directory is in expected directory
checkInDirectory(subdir, dir);
// check directory is empty
DirectoryStream<Path> stream = Files.newDirectoryStream(subdir);
try {
if (stream.iterator().hasNext())
throw new RuntimeException("Tempory directory not empty");
} finally {
stream.close();
}
// check that we can create file in directory
Path file = Files.createFile(subdir.resolve("foo"));
try {
Files.newByteChannel(file, READ,WRITE).close();
} finally {
Files.delete(file);
}
// check file permissions are 0700 or more secure
if (Files.getFileStore(subdir).supportsFileAttributeView("posix")) {
Set<PosixFilePermission> perms = Files.getPosixFilePermissions(subdir);
perms.remove(PosixFilePermission.OWNER_READ);
perms.remove(PosixFilePermission.OWNER_WRITE);
perms.remove(PosixFilePermission.OWNER_EXECUTE);
if (!perms.isEmpty())
throw new RuntimeException("Temporary directory is not secure");
}
} finally {
Files.delete(subdir);
}
}
static void testTempDirectory(String prefix) throws IOException {
testTempDirectory(prefix, null);
}
static void testInvalidFileTemp(String prefix, String suffix) throws IOException {
try {
Path file = Files.createTempFile(prefix, suffix);
Files.delete(file);
throw new RuntimeException("IllegalArgumentException expected");
} catch (IllegalArgumentException expected) { }
}
public static void main(String[] args) throws IOException {
// temporary-file directory
testTempFile("blah", ".dat");
testTempFile("blah", null);
testTempFile(null, ".dat");
testTempFile(null, null);
testTempDirectory("blah");
testTempDirectory(null);
// a given directory
Path dir = Files.createTempDirectory("tmpdir");
try {
testTempFile("blah", ".dat", dir);
testTempFile("blah", null, dir);
testTempFile(null, ".dat", dir);
testTempFile(null, null, dir);
testTempDirectory("blah", dir);
testTempDirectory(null, dir);
} finally {
Files.delete(dir);
}
// invalid prefix and suffix
testInvalidFileTemp("../blah", null);
testInvalidFileTemp("dir/blah", null);
testInvalidFileTemp("blah", ".dat/foo");
// nulls
try {
Files.createTempFile("blah", ".tmp", (FileAttribute<?>[])null);
throw new RuntimeException("NullPointerException expected");
} catch (NullPointerException ignore) { }
try {
Files.createTempFile("blah", ".tmp", new FileAttribute<?>[] { null });
throw new RuntimeException("NullPointerException expected");
} catch (NullPointerException ignore) { }
try {
Files.createTempDirectory("blah", (FileAttribute<?>[])null);
throw new RuntimeException("NullPointerException expected");
} catch (NullPointerException ignore) { }
try {
Files.createTempDirectory("blah", new FileAttribute<?>[] { null });
throw new RuntimeException("NullPointerException expected");
} catch (NullPointerException ignore) { }
try {
Files.createTempFile((Path)null, "blah", ".tmp");
throw new RuntimeException("NullPointerException expected");
} catch (NullPointerException ignore) { }
try {
Files.createTempDirectory((Path)null, "blah");
throw new RuntimeException("NullPointerException expected");
} catch (NullPointerException ignore) { }
}
}