--- /dev/null 2017-11-09 09:38:01.297999907 +0100 +++ new/src/jdk.jfr/share/classes/jdk/jfr/internal/ShutdownHook.java 2018-04-09 16:05:02.015115633 +0200 @@ -0,0 +1,108 @@ +/* + * Copyright (c) 2016, 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. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * 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. + */ + +package jdk.jfr.internal; + +import java.io.IOException; +import java.nio.file.Paths; +import java.security.AccessControlContext; +import java.security.AccessController; +import java.security.PrivilegedActionException; +import java.security.PrivilegedExceptionAction; +import java.time.LocalDateTime; + +import jdk.jfr.RecordingState; + +/** + * Class responsible for dumping recordings on exit + * + */ +final class ShutdownHook implements Runnable { + private final PlatformRecorder recorder; + Object tlabDummyObject; + + ShutdownHook(PlatformRecorder recorder) { + this.recorder = recorder; + } + + @Override + public void run() { + // this allocation is done in order to fetch a new TLAB before + // starting any "real" operations. In low memory situations, + // we would like to take an OOM as early as possible. + tlabDummyObject = new Object(); + + for (PlatformRecording recording : recorder.getRecordings()) { + if (recording.getDumpOnExit() && recording.getState() == RecordingState.RUNNING) { + dump(recording); + } + } + recorder.destroy(); + } + + private void dump(PlatformRecording recording) { + try { + WriteableUserPath dest = recording.getDestination(); + if (dest == null) { + dest = makeDumpOnExitPath(recording); + } + if (dest != null) { + recording.stop("Dump on exit", dest); + } + } catch (Exception e) { + Logger.log(LogTag.JFR, LogLevel.DEBUG, () -> "Could not dump recording " + recording.getName() + " on exit."); + } + } + + private WriteableUserPath makeDumpOnExitPath(PlatformRecording recording) { + try { + String pid = JVM.getJVM().getPid(); + String dfText = Repository.REPO_DATE_FORMAT.format(LocalDateTime.now()); + String name = "hotspot-" + "pid-" + pid + "-id-" + recording.getId() + "-" + dfText + ".jfr"; + AccessControlContext acc = recording.getNoDestinationDumpOnExitAccessControlContext(); + return AccessController.doPrivileged(new PrivilegedExceptionAction() { + @Override + public WriteableUserPath run() throws Exception { + return new WriteableUserPath(Paths.get(".", name)); + } + }, acc); + } catch (PrivilegedActionException e) { + Throwable t = e.getCause(); + if (t instanceof SecurityException) { + Logger.log(LogTag.JFR, LogLevel.WARN, "Not allowed to create dump path for recording " + recording.getId() + " on exit. " + e.getMessage()); + } + if (t instanceof IOException) { + Logger.log(LogTag.JFR, LogLevel.WARN, "Could not dump " + recording.getId() + " on exit. " + e.getMessage()); + } + return null; + } + } + + static final class ExceptionHandler implements Thread.UncaughtExceptionHandler { + public void uncaughtException(Thread t, Throwable e) { + JVM.getJVM().uncaughtException(t, e); + } + } +}