1 /*
   2  * Copyright (c) 2008, 2017, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /* 
  25  * @test
  26  * @bug 4313887
  27  * @summary Unit test for DELETE_ON_CLOSE open option
  28  * @library /test/lib /java/nio/file
  29  */
  30 
  31 import java.io.IOException;
  32 import java.nio.file.DirectoryStream;
  33 import java.nio.file.Files;
  34 import java.nio.file.OpenOption;
  35 import java.nio.file.Path;
  36 import java.nio.file.Paths;
  37 import java.nio.file.SecureDirectoryStream;
  38 import java.util.HashSet;
  39 import java.util.Set;
  40 
  41 import jdk.test.lib.process.ProcessTools;
  42 
  43 import static java.nio.file.StandardOpenOption.READ;
  44 import static java.nio.file.StandardOpenOption.WRITE;
  45 import static java.nio.file.StandardOpenOption.DELETE_ON_CLOSE;
  46 
  47 public class DeleteOnClose {
  48 
  49     public static void main(String[] args) throws Exception {
  50         if (args.length == 0) {
  51             Path file = Files.createTempFile("blah", "tmp");
  52             ProcessTools.executeTestJava(DeleteOnClose.class.getName(),
  53                                          file.toAbsolutePath().toString())
  54                         .shouldHaveExitValue(0);
  55             runTest(file);
  56         } else {
  57             // open file but do not close it. Its existance will be checked by
  58             // the caller.
  59             Files.newByteChannel(Paths.get(args[0]), READ, WRITE, DELETE_ON_CLOSE);
  60         }
  61     }
  62 
  63     public static void runTest(Path path) throws Exception {
  64         // check temporary file has been deleted after jvm termination
  65         if (Files.exists(path)) {
  66             throw new RuntimeException("Temporary file was not deleted");
  67         }
  68 
  69         // check temporary file has been deleted after closing it
  70         Path file = Files.createTempFile("blep", "tmp");
  71         Files.newByteChannel(file, READ, WRITE, DELETE_ON_CLOSE).close();
  72         if (Files.exists(file))
  73             throw new RuntimeException("Temporary file was not deleted");
  74 
  75         Path dir = Files.createTempDirectory("blah");
  76         try {
  77             // check that DELETE_ON_CLOSE fails when file is a sym link
  78             if (TestUtil.supportsLinks(dir)) {
  79                 file = dir.resolve("foo");
  80                 Files.createFile(file);
  81                 Path link = dir.resolve("link");
  82                 Files.createSymbolicLink(link, file);
  83                 try {
  84                     Files.newByteChannel(link, READ, WRITE, DELETE_ON_CLOSE);
  85                     throw new RuntimeException("IOException expected");
  86                 } catch (IOException ignore) { }
  87             }
  88 
  89             // check that DELETE_ON_CLOSE works with files created via open
  90             // directories
  91             try (DirectoryStream<Path> stream = Files.newDirectoryStream(dir)) {
  92                 if (stream instanceof SecureDirectoryStream) {
  93                     SecureDirectoryStream<Path> secure = (SecureDirectoryStream<Path>)stream;
  94                     file = Paths.get("foo");
  95 
  96                     Set<OpenOption> opts = new HashSet<>();
  97                     opts.add(WRITE);
  98                     opts.add(DELETE_ON_CLOSE);
  99                     secure.newByteChannel(file, opts).close();
 100 
 101                     if (Files.exists(dir.resolve(file)))
 102                         throw new RuntimeException("File not deleted");
 103                 }
 104             }
 105         } finally {
 106             TestUtil.removeAll(dir);
 107         }
 108     }
 109 }