1 /*
   2  * Copyright (c) 2004, 2015, 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 4997445
  26  * @summary Test that with server=y, when VM runs to System.exit() no error happens
  27  * @library /lib/testlibrary
  28  * @modules java.management
  29  *          jdk.jdi
  30  * @build jdk.testlibrary.* VMConnection RunToExit Exit0
  31  * @run driver RunToExit
  32  */
  33 import java.net.ServerSocket;
  34 import com.sun.jdi.Bootstrap;
  35 import com.sun.jdi.VirtualMachine;
  36 import com.sun.jdi.event.*;
  37 import com.sun.jdi.connect.Connector;
  38 import com.sun.jdi.connect.AttachingConnector;
  39 import java.net.ConnectException;
  40 import java.util.Map;
  41 import java.util.List;
  42 import java.util.Iterator;
  43 import java.util.concurrent.TimeUnit;
  44 import java.util.stream.Collectors;
  45 import jdk.testlibrary.ProcessTools;
  46 
  47 public class RunToExit {
  48 
  49     /* Increment this when ERROR: seen */
  50     static volatile int error_seen = 0;
  51     static volatile boolean ready = false;
  52 
  53     /*
  54      * Find a connector by name
  55      */
  56     private static Connector findConnector(String name) {
  57         List connectors = Bootstrap.virtualMachineManager().allConnectors();
  58         Iterator iter = connectors.iterator();
  59         while (iter.hasNext()) {
  60             Connector connector = (Connector)iter.next();
  61             if (connector.name().equals(name)) {
  62                 return connector;
  63             }
  64         }
  65         return null;
  66     }
  67 
  68     /*
  69      * Launch a server debuggee with the given address
  70      */
  71     private static Process launch(String address, String class_name) throws Exception {
  72         String args[] = new String[]{
  73             "-agentlib:jdwp=transport=dt_socket,server=y,suspend=y,address="
  74                 + address,
  75             class_name
  76         };
  77         args = VMConnection.insertDebuggeeVMOptions(args);
  78 
  79         ProcessBuilder launcher = ProcessTools.createJavaProcessBuilder(args);
  80 
  81         System.out.println(launcher.command().stream().collect(Collectors.joining(" ", "Starting: ", "")));
  82 
  83         Process p = ProcessTools.startProcess(
  84             class_name,
  85             launcher,
  86             RunToExit::checkForError,
  87             RunToExit::isTransportListening,
  88             0,
  89             TimeUnit.NANOSECONDS
  90         );
  91 
  92         return p;
  93     }
  94 
  95     private static boolean isTransportListening(String line) {
  96         return line.startsWith("Listening for transport dt_socket");
  97     }
  98 
  99     private static void checkForError(String line) {
 100         if (line.contains("ERROR:")) {
 101             error_seen++;
 102         }
 103     }
 104 
 105     /*
 106      * - pick a TCP port
 107      * - Launch a server debuggee: server=y,suspend=y,address=${port}
 108      * - run it to VM death
 109      * - verify we saw no error
 110      */
 111     public static void main(String args[]) throws Exception {
 112         // find a free port
 113         ServerSocket ss = new ServerSocket(0);
 114         int port = ss.getLocalPort();
 115         ss.close();
 116 
 117         String address = String.valueOf(port);
 118 
 119         // launch the server debuggee
 120         Process process = launch(address, "Exit0");
 121 
 122         // attach to server debuggee and resume it so it can exit
 123         AttachingConnector conn = (AttachingConnector)findConnector("com.sun.jdi.SocketAttach");
 124         Map conn_args = conn.defaultArguments();
 125         Connector.IntegerArgument port_arg =
 126             (Connector.IntegerArgument)conn_args.get("port");
 127         port_arg.setValue(port);
 128 
 129         System.out.println("Connection arguments: " + conn_args);
 130 
 131         VirtualMachine vm = null;
 132         while (vm == null) {
 133             try {
 134                 vm = conn.attach(conn_args);
 135             } catch (ConnectException e) {
 136                 e.printStackTrace(System.out);
 137                 System.out.println("--- Debugee not ready. Retrying in 500ms. ---");
 138                 Thread.sleep(500);
 139             }
 140         }
 141 
 142         // The first event is always a VMStartEvent, and it is always in
 143         // an EventSet by itself.  Wait for it.
 144         EventSet evtSet = vm.eventQueue().remove();
 145         for (Event event: evtSet) {
 146             if (event instanceof VMStartEvent) {
 147                 break;
 148             }
 149             throw new RuntimeException("Test failed - debuggee did not start properly");
 150         }
 151         vm.eventRequestManager().deleteAllBreakpoints();
 152         vm.resume();
 153 
 154         int exitCode = process.waitFor();
 155 
 156         // if the server debuggee ran cleanly, we assume we were clean
 157         if (exitCode == 0 && error_seen == 0) {
 158             System.out.println("Test passed - server debuggee cleanly terminated");
 159         } else {
 160             throw new RuntimeException("Test failed - server debuggee generated an error when it terminated, " +
 161                 "exit code was " + exitCode + ", " + error_seen + " error(s) seen in debugee output.");
 162         }
 163     }
 164 }