/*

 * Copyright (c) 2005, 2006, 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.

 */

/*

 *

 *

 * A test "management tool" used by unit test LocalManagementTest.sh.

 *

 * Usage:    java TestManager <pid> <port>

 *

 * where <pid> is the process-id of the test application, and <port> is

 * TCP port is used to shutdown the application.

 */

import javax.management.MBeanServerConnection;

import javax.management.MBeanServerInvocationHandler;

import javax.management.ObjectName;

import javax.management.remote.JMXServiceURL;

import javax.management.remote.JMXConnectorFactory;

import javax.management.remote.JMXConnector;

import java.lang.management.RuntimeMXBean;

import static java.lang.management.ManagementFactory.*;

import java.net.Socket;

import java.net.InetSocketAddress;

import java.io.File;

import java.io.IOException;

import java.util.Properties;

// Sun specific

import com.sun.tools.attach.VirtualMachine;

// Sun implementation specific

import sun.management.ConnectorAddressLink;

public class TestManager {

    /*

     * Starts the management agent in the target VM

     */

    private static void startManagementAgent(String pid) throws IOException {

        /*

         * JAR file normally in ${java.home}/jre/lib but may be in ${java.home}/lib

         * with development/non-images builds

         */

        String home = System.getProperty("java.home");

        String agent = home + File.separator + "jre" + File.separator + "lib"

                + File.separator + "management-agent.jar";

        File f = new File(agent);

        if (!f.exists()) {

            agent = home + File.separator + "lib" + File.separator +

                "management-agent.jar";

            f = new File(agent);

            if (!f.exists()) {

                throw new RuntimeException("management-agent.jar missing");

            }

        }

        agent = f.getCanonicalPath();

        System.out.println("Loading " + agent + " into target VM ...");

        try {

            VirtualMachine.attach(pid).loadAgent(agent);

        } catch (Exception x) {

            throw new IOException(x.getMessage());

        }

    }

    private static void connect(String pid, String address) throws Exception {

        if (address == null) {

            throw new RuntimeException("Local connector address for " +

                                       pid + " is null");

        }

        System.out.println("Connect to process " + pid + " via: " + address);

        JMXServiceURL url = new JMXServiceURL(address);

        JMXConnector c = JMXConnectorFactory.connect(url);

        MBeanServerConnection server = c.getMBeanServerConnection();

        System.out.println("Connected.");

        RuntimeMXBean rt = newPlatformMXBeanProxy(server,

            RUNTIME_MXBEAN_NAME, RuntimeMXBean.class);

        System.out.println(rt.getName());

        // close the connection

        c.close();

    }

    private final static String LOCAL_CONNECTOR_ADDRESS_PROP =

        "com.sun.management.jmxremote.localConnectorAddress";

    public static void main(String[] args) throws Exception {

        String pid = args[0]; // pid as a string

        VirtualMachine vm = VirtualMachine.attach(pid);

        String agentPropLocalConnectorAddress = (String)

            vm.getAgentProperties().get(LOCAL_CONNECTOR_ADDRESS_PROP);

        int vmid = Integer.parseInt(pid);

        String jvmstatLocalConnectorAddress =

            ConnectorAddressLink.importFrom(vmid);

        if (agentPropLocalConnectorAddress == null &&

            jvmstatLocalConnectorAddress == null) {

            // No JMX Connector address so attach to VM, and load

            // management-agent.jar

            startManagementAgent(pid);

            agentPropLocalConnectorAddress = (String)

                vm.getAgentProperties().get(LOCAL_CONNECTOR_ADDRESS_PROP);

            jvmstatLocalConnectorAddress =

                ConnectorAddressLink.importFrom(vmid);

        }

        // Test address obtained from agent properties

        System.out.println("Testing the connector address from agent properties");

        connect(pid, agentPropLocalConnectorAddress);

        // Test address obtained from jvmstat buffer

        System.out.println("Testing the connector address from jvmstat buffer");

        connect(pid, jvmstatLocalConnectorAddress);

        // Shutdown application

        int port = Integer.parseInt(args[1]);

        System.out.println("Shutdown process via TCP port: " + port);

        Socket s = new Socket();

        s.connect(new InetSocketAddress(port));

        s.close();

    }

}