1/*
2 * Copyright (c) 2008, 2016, 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/* @test
25 * @bug 4313887 6993267
26 * @summary Unit test for Sun-specific ExtendedCopyOption.INTERRUPTIBLE option
27 * @modules jdk.unsupported
28 * @library ..
29 */
30
31import java.nio.file.*;
32import java.io.*;
33import java.util.concurrent.*;
34import com.sun.nio.file.ExtendedCopyOption;
35
36public class InterruptCopy {
37
38    private static final long FILE_SIZE_TO_COPY = 512L * 1024L * 1024L;
39    private static final int DELAY_IN_MS = 500;
40    private static final int DURATION_MAX_IN_MS = 5000;
41
42    public static void main(String[] args) throws Exception {
43        Path dir = TestUtil.createTemporaryDirectory();
44        try {
45            FileStore store = Files.getFileStore(dir);
46            System.out.format("Checking space (%s)\n", store);
47            long usableSpace = store.getUsableSpace();
48            if (usableSpace < 2*FILE_SIZE_TO_COPY) {
49                System.out.println("Insufficient disk space to run test.");
50                return;
51            }
52            doTest(dir);
53        } finally {
54            TestUtil.removeAll(dir);
55        }
56    }
57
58    static void doTest(Path dir) throws Exception {
59        final Path source = dir.resolve("foo");
60        final Path target = dir.resolve("bar");
61
62        // create source file (don't create it as sparse file because we
63        // require the copy to take a long time)
64        System.out.println("Creating source file...");
65        byte[] buf = new byte[32*1024];
66        long total = 0;
67        try (OutputStream out = Files.newOutputStream(source)) {
68            do {
69                out.write(buf);
70                total += buf.length;
71            } while (total < FILE_SIZE_TO_COPY);
72        }
73        System.out.println("Source file created.");
74
75        ScheduledExecutorService pool =
76            Executors.newSingleThreadScheduledExecutor();
77        try {
78            // copy source to target in main thread, interrupting it after a delay
79            final Thread me = Thread.currentThread();
80            Future<?> wakeup = pool.schedule(new Runnable() {
81                public void run() {
82                    me.interrupt();
83                }}, DELAY_IN_MS, TimeUnit.MILLISECONDS);
84            System.out.println("Copying file...");
85            try {
86                long start = System.currentTimeMillis();
87                Files.copy(source, target, ExtendedCopyOption.INTERRUPTIBLE);
88                long duration = System.currentTimeMillis() - start;
89                if (duration > DURATION_MAX_IN_MS)
90                    throw new RuntimeException("Copy was not interrupted");
91            } catch (IOException e) {
92                boolean interrupted = Thread.interrupted();
93                if (!interrupted)
94                    throw new RuntimeException("Interrupt status was not set");
95                System.out.println("Copy failed (this is expected)");
96            }
97            try {
98                wakeup.get();
99            } catch (InterruptedException ignore) { }
100            Thread.interrupted();
101
102            // copy source to target via task in thread pool, interrupting it after
103            // a delay using cancel(true)
104            Future<Void> result = pool.submit(new Callable<Void>() {
105                public Void call() throws IOException {
106                    System.out.println("Copying file...");
107                    Files.copy(source, target, ExtendedCopyOption.INTERRUPTIBLE,
108                        StandardCopyOption.REPLACE_EXISTING);
109                    return null;
110                }
111            });
112            Thread.sleep(DELAY_IN_MS);
113            boolean cancelled = result.cancel(true);
114            if (!cancelled)
115                result.get();
116            System.out.println("Copy cancelled.");
117        } finally {
118            pool.shutdown();
119        }
120    }
121}
122