1 /*
   2  * Copyright (c) 2013, 2018, 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.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package jdk.jfr.startupargs;
  27 
  28 import java.io.IOException;
  29 import java.nio.file.DirectoryStream;
  30 import java.nio.file.Files;
  31 import java.nio.file.Path;
  32 import java.nio.file.Paths;
  33 import java.util.Iterator;
  34 import java.util.function.Supplier;
  35 
  36 import jdk.jfr.consumer.RecordingFile;
  37 import jdk.test.lib.Asserts;
  38 import jdk.test.lib.process.OutputAnalyzer;
  39 import jdk.test.lib.process.ProcessTools;
  40 
  41 /*
  42  * @test
  43  * @summary Start a FlightRecording with dumponexit. Verify dump exists.
  44  * @key jfr
  45  * @library /test/lib
  46  * @run main/othervm jdk.jfr.startupargs.TestDumpOnExit
  47  */
  48 public class TestDumpOnExit {
  49 
  50     public static void main(String[] args) throws Exception {
  51         Path dumpPath = Paths.get(".", "dumped.jfr");
  52 
  53         // Test without security manager and a file name relative to current directory
  54         testDumponExit(() -> dumpPath,
  55                 "-Xlog:jfr=trace",
  56                 "-XX:StartFlightRecording=filename=./dumped.jfr,dumponexit=true,settings=profile",
  57                 "jdk.jfr.startupargs.TestDumpOnExit$TestMain"
  58         );
  59         // Test a memory recording without a security manager
  60         testDumponExit(() -> findJFRFileInCurrentDirectory(),
  61                 "-Xlog:jfr=trace",
  62                 "-XX:StartFlightRecording=dumponexit=true,disk=false",
  63                 "jdk.jfr.startupargs.TestDumpOnExit$TestMain"
  64         );
  65 
  66         // Test with security manager and a file name relative to current directory
  67         testDumponExit(() -> dumpPath,
  68                 "-Xlog:jfr=trace",
  69                 "-XX:StartFlightRecording=filename=./dumped.jfr,dumponexit=true,settings=profile",
  70                 "-Djava.security.manager",
  71                 "jdk.jfr.startupargs.TestDumpOnExit$TestMain"
  72         );
  73         // Test with security manager but without a name
  74         testDumponExit(() -> findJFRFileInCurrentDirectory(),
  75                 "-Xlog:jfr=trace",
  76                 "-XX:StartFlightRecording=dumponexit=true,settings=profile",
  77                 "-Djava.security.manager",
  78                 "jdk.jfr.startupargs.TestDumpOnExit$TestMain"
  79         );
  80     }
  81 
  82     private static Path findJFRFileInCurrentDirectory() {
  83         try {
  84             DirectoryStream<Path> ds = Files.newDirectoryStream(Paths.get("."), "*pid-*.jfr");
  85             Iterator<Path> pathIterator = ds.iterator();
  86             if (!pathIterator.hasNext()) {
  87                 throw new RuntimeException("Could not find jfr file in current directory");
  88             }
  89             return pathIterator.next();
  90         } catch (IOException e) {
  91             throw new RuntimeException("Could not list jfr file in current directory");
  92         }
  93     }
  94 
  95     private static void testDumponExit(Supplier<Path> p,String... args) throws Exception, IOException {
  96         ProcessBuilder pb = ProcessTools.createJavaProcessBuilder(true, args);
  97         OutputAnalyzer output = ProcessTools.executeProcess(pb);
  98         System.out.println(output.getOutput());
  99         output.shouldHaveExitValue(0);
 100         Path dump = p.get();
 101         Asserts.assertTrue(Files.isRegularFile(dump), "No recording dumped " + dump);
 102         System.out.println("Dumped recording size=" + Files.size(dump));
 103         Asserts.assertFalse(RecordingFile.readAllEvents(dump).isEmpty(), "No events in dump");
 104     }
 105 
 106     @SuppressWarnings("unused")
 107     private static class TestMain {
 108         public static void main(String[] args) throws Exception {
 109             System.out.println("Hello from test main");
 110         }
 111     }
 112 
 113 }