| /* |
| * Copyright (c) 2012, 2018, 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 4244896 |
| * @summary Test for the various platform specific implementations of |
| * destroyForcibly. |
| */ |
| |
| import java.io.*; |
| import java.util.ArrayList; |
| import java.util.concurrent.TimeUnit; |
| |
| abstract class ProcessTest implements Runnable { |
| ProcessBuilder bldr; |
| Process p; |
| |
| public Process killProc(boolean force) throws Exception { |
| if (force) { |
| p.destroyForcibly(); |
| } else { |
| p.destroy(); |
| } |
| return p; |
| } |
| |
| public boolean isAlive() { |
| return p.isAlive(); |
| } |
| |
| public void run() { |
| try { |
| String line; |
| BufferedReader is = |
| new BufferedReader(new InputStreamReader(p.getInputStream())); |
| while ((line = is.readLine()) != null) |
| System.err.println("ProcessTrap: " + line); |
| } catch(IOException e) { |
| if (!e.getMessage().matches("[Ss]tream [Cc]losed")) { |
| throw new RuntimeException(e); |
| } |
| } |
| } |
| |
| public abstract void runTest() throws Exception; |
| } |
| |
| class UnixTest extends ProcessTest { |
| public UnixTest(File script) throws IOException { |
| script.deleteOnExit(); |
| createScript(script); |
| bldr = new ProcessBuilder(script.getCanonicalPath()); |
| bldr.redirectErrorStream(true); |
| bldr.directory(new File(".")); |
| p = bldr.start(); |
| } |
| |
| void createScript(File processTrapScript) throws IOException { |
| processTrapScript.deleteOnExit(); |
| FileWriter fstream = new FileWriter(processTrapScript); |
| try (BufferedWriter out = new BufferedWriter(fstream)) { |
| out.write("#!/bin/bash\n" + |
| "echo \\\"ProcessTrap.sh started: trapping SIGTERM/SIGINT\\\"\n" + |
| "trap bashtrap SIGTERM SIGINT\n" + |
| "bashtrap()\n" + |
| "{\n" + |
| " echo \\\"SIGTERM/SIGINT detected!\\\"\n" + |
| "}\n" + |
| "\n" + |
| "while :\n" + |
| "do\n" + |
| " sleep 1;\n" + |
| "done\n"); |
| } |
| processTrapScript.setExecutable(true, true); |
| } |
| |
| @Override |
| public void runTest() throws Exception { |
| killProc(false); |
| Thread.sleep(1000); |
| if (!p.isAlive()) |
| throw new RuntimeException("Process terminated prematurely."); |
| killProc(true).waitFor(); |
| if (p.isAlive()) |
| throw new RuntimeException("Problem terminating the process."); |
| } |
| } |
| |
| class MacTest extends UnixTest { |
| public MacTest(File script) throws IOException { |
| super(script); |
| } |
| |
| @Override |
| public void runTest() throws Exception { |
| // On Mac, it appears that when we close the processes streams |
| // after a destroy() call, the process terminates with a |
| // SIGPIPE even if it was trapping the SIGTERM, so as with |
| // windows, we skip the trap test and go straight to destroyForcibly(). |
| killProc(true).waitFor(); |
| if (p.isAlive()) |
| throw new RuntimeException("Problem terminating the process."); |
| } |
| } |
| |
| class WindowsTest extends ProcessTest { |
| public WindowsTest() throws IOException { |
| bldr = new ProcessBuilder("ftp"); |
| bldr.redirectErrorStream(true); |
| bldr.directory(new File(".")); |
| p = bldr.start(); |
| } |
| |
| @Override |
| public void runTest() throws Exception { |
| killProc(true).waitFor(); |
| } |
| } |
| |
| public class DestroyTest { |
| |
| public static ProcessTest getTest() throws Exception { |
| String osName = System.getProperty("os.name"); |
| if (osName.startsWith("Windows")) { |
| return new WindowsTest(); |
| } else { |
| File userDir = new File(System.getProperty("user.dir", ".")); |
| File tempFile = File.createTempFile("ProcessTrap-", ".sh", userDir); |
| if (osName.startsWith("Linux") == true |
| || osName.equals("SunOS") |
| || osName.equals("AIX")) { |
| return new UnixTest(tempFile); |
| } else if (osName.startsWith("Mac OS")) { |
| return new MacTest(tempFile); |
| } |
| } |
| return null; |
| } |
| |
| public static void main(String args[]) throws Exception { |
| ProcessTest test = getTest(); |
| if (test == null) { |
| throw new RuntimeException("Unrecognised OS"); |
| } else { |
| new Thread(test).start(); |
| Thread.sleep(1000); |
| test.runTest(); |
| } |
| } |
| } |
| |