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             // open file but do not close it. Its existance will be checked by
  52             // the caller.
  53             Path file = Files.createTempFile("blah", "tmp");
  54             Files.newByteChannel(file, READ, WRITE, DELETE_ON_CLOSE);
  55             ProcessTools.executeTestJava(DeleteOnClose.class.getName(),
  56                                          file.toAbsolutePath().toString())
  57                         .outputTo(System.out)
  58                         .errorTo(System.err)
  59                         .shouldHaveExitValue(0);
  60         } else {
  61             runTest(args[0]);
  62         }
  63     }
  64 
  65     public static void runTest(String pathString) throws Exception {
  66         // check temporary file has been deleted after jvm termination
  67         if (Files.exists(Paths.get(pathString))) {
  68             throw new RuntimeException("Temporary file was not deleted");
  69         }
  70 
  71         // check temporary file has been deleted after closing it
  72         Path file = Files.createTempFile("blep", "tmp");
  73         Files.newByteChannel(file, READ, WRITE, DELETE_ON_CLOSE).close();
  74         if (Files.exists(file))
  75             throw new RuntimeException("Temporary file was not deleted");
  76 
  77         Path dir = Files.createTempDirectory("blah");
  78         try {
  79             // check that DELETE_ON_CLOSE fails when file is a sym link
  80             if (TestUtil.supportsLinks(dir)) {
  81                 file = dir.resolve("foo");
  82                 Files.createFile(file);
  83                 Path link = dir.resolve("link");
  84                 Files.createSymbolicLink(link, file);
  85                 try {
  86                     Files.newByteChannel(link, READ, WRITE, DELETE_ON_CLOSE);
  87                     throw new RuntimeException("IOException expected");
  88                 } catch (IOException ignore) { }
  89             }
  90 
  91             // check that DELETE_ON_CLOSE works with files created via open
  92             // directories
  93             try (DirectoryStream<Path> stream = Files.newDirectoryStream(dir)) {
  94                 if (stream instanceof SecureDirectoryStream) {
  95                     SecureDirectoryStream<Path> secure = (SecureDirectoryStream<Path>)stream;
  96                     file = Paths.get("foo");
  97 
  98                     Set<OpenOption> opts = new HashSet<>();
  99                     opts.add(WRITE);
 100                     opts.add(DELETE_ON_CLOSE);
 101                     secure.newByteChannel(file, opts).close();
 102 
 103                     if (Files.exists(dir.resolve(file)))
 104                         throw new RuntimeException("File not deleted");
 105                 }
 106             }
 107         } finally {
 108             TestUtil.removeAll(dir);
 109         }
 110     }
 111 }