/*

 * Copyright (c) 2005, 2014, 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 sun.tools.jconsole;

import java.util.*;

import java.io.IOException;

import java.io.File;

import com.sun.tools.attach.VirtualMachine;

import com.sun.tools.attach.VirtualMachineDescriptor;

import com.sun.tools.attach.AttachNotSupportedException;

import jdk.internal.agent.ConnectorAddressLink;

import sun.jvmstat.monitor.HostIdentifier;

import sun.jvmstat.monitor.MonitoredHost;

import sun.jvmstat.monitor.MonitoredVm;

import sun.jvmstat.monitor.MonitoredVmUtil;

import sun.jvmstat.monitor.MonitorException;

import sun.jvmstat.monitor.VmIdentifier;

public class LocalVirtualMachine {

    private String address;

    private String commandLine;

    private String displayName;

    private int vmid;

    private boolean isAttachSupported;

    public LocalVirtualMachine(int vmid, String commandLine, boolean canAttach, String connectorAddress) {

        this.vmid = vmid;

        this.commandLine = commandLine;

        this.address = connectorAddress;

        this.isAttachSupported = canAttach;

        this.displayName = getDisplayName(commandLine);

    }

    private static String getDisplayName(String commandLine) {

        // trim the pathname of jar file if it's a jar

        String[] res = commandLine.split(" ", 2);

        if (res[0].endsWith(".jar")) {

           File jarfile = new File(res[0]);

           String displayName = jarfile.getName();

           if (res.length == 2) {

               displayName += " " + res[1];

           }

           return displayName;

        }

        return commandLine;

    }

    public int vmid() {

        return vmid;

    }

    public boolean isManageable() {

        return (address != null);

    }

    public boolean isAttachable() {

        return isAttachSupported;

    }

    public void startManagementAgent() throws IOException {

        if (address != null) {

            // already started

            return;

        }

        if (!isAttachable()) {

            throw new IOException("This virtual machine \"" + vmid +

                "\" does not support dynamic attach.");

        }

        loadManagementAgent();

        // fails to load or start the management agent

        if (address == null) {

            // should never reach here

            throw new IOException("Fails to find connector address");

        }

    }

    public String connectorAddress() {

        // return null if not available or no JMX agent

        return address;

    }

    public String displayName() {

        return displayName;

    }

    public String toString() {

        return commandLine;

    }

    // This method returns the list of all virtual machines currently

    // running on the machine

    public static Map<Integer, LocalVirtualMachine> getAllVirtualMachines() {

        Map<Integer, LocalVirtualMachine> map =

            new HashMap<Integer, LocalVirtualMachine>();

        getMonitoredVMs(map);

        getAttachableVMs(map);

        return map;

    }

    private static void getMonitoredVMs(Map<Integer, LocalVirtualMachine> map) {

        MonitoredHost host;

        Set<Integer> vms;

        try {

            host = MonitoredHost.getMonitoredHost(new HostIdentifier((String)null));

            vms = host.activeVms();

        } catch (java.net.URISyntaxException | MonitorException x) {

            throw new InternalError(x.getMessage(), x);

        }

        for (Object vmid: vms) {

            if (vmid instanceof Integer) {

                int pid = ((Integer) vmid).intValue();

                String name = vmid.toString(); // default to pid if name not available

                boolean attachable = false;

                String address = null;

                try {

                     MonitoredVm mvm = host.getMonitoredVm(new VmIdentifier(name));

                     // use the command line as the display name

                     name =  MonitoredVmUtil.commandLine(mvm);

                     attachable = MonitoredVmUtil.isAttachable(mvm);

                     address = ConnectorAddressLink.importFrom(pid);

                     mvm.detach();

                } catch (Exception x) {

                     // ignore

                }

                map.put((Integer) vmid,

                        new LocalVirtualMachine(pid, name, attachable, address));

            }

        }

    }

    private static final String LOCAL_CONNECTOR_ADDRESS_PROP =

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

    private static void getAttachableVMs(Map<Integer, LocalVirtualMachine> map) {

        List<VirtualMachineDescriptor> vms = VirtualMachine.list();

        for (VirtualMachineDescriptor vmd : vms) {

            try {

                Integer vmid = Integer.valueOf(vmd.id());

                if (!map.containsKey(vmid)) {

                    boolean attachable = false;

                    String address = null;

                    try {

                        VirtualMachine vm = VirtualMachine.attach(vmd);

                        attachable = true;

                        Properties agentProps = vm.getAgentProperties();

                        address = (String) agentProps.get(LOCAL_CONNECTOR_ADDRESS_PROP);

                        vm.detach();

                    } catch (AttachNotSupportedException x) {

                        // not attachable

                    } catch (IOException x) {

                        // ignore

                    }

                    map.put(vmid, new LocalVirtualMachine(vmid.intValue(),

                                                          vmd.displayName(),

                                                          attachable,

                                                          address));

                }

            } catch (NumberFormatException e) {

                // do not support vmid different than pid

            }

        }

    }

    public static LocalVirtualMachine getLocalVirtualMachine(int vmid) {

        Map<Integer, LocalVirtualMachine> map = getAllVirtualMachines();

        LocalVirtualMachine lvm = map.get(vmid);

        if (lvm == null) {

            // Check if the VM is attachable but not included in the list

            // if it's running with a different security context.

            // For example, Windows services running

            // local SYSTEM account are attachable if you have Adminstrator

            // privileges.

            boolean attachable = false;

            String address = null;

            String name = String.valueOf(vmid); // default display name to pid

            try {

                VirtualMachine vm = VirtualMachine.attach(name);

                attachable = true;

                Properties agentProps = vm.getAgentProperties();

                address = (String) agentProps.get(LOCAL_CONNECTOR_ADDRESS_PROP);

                vm.detach();

                lvm = new LocalVirtualMachine(vmid, name, attachable, address);

            } catch (AttachNotSupportedException x) {

                // not attachable

                if (JConsole.isDebug()) {

                    x.printStackTrace();

                }

            } catch (IOException x) {

                // ignore

                if (JConsole.isDebug()) {

                    x.printStackTrace();

                }

            }

        }

        return lvm;

    }

    // load the management agent into the target VM

    private void loadManagementAgent() throws IOException {

        VirtualMachine vm = null;

        String name = String.valueOf(vmid);

        try {

            vm = VirtualMachine.attach(name);

        } catch (AttachNotSupportedException x) {

            IOException ioe = new IOException(x.getMessage());

            ioe.initCause(x);

            throw ioe;

        }

        vm.startLocalManagementAgent();

        // get the connector address

        Properties agentProps = vm.getAgentProperties();

        address = (String) agentProps.get(LOCAL_CONNECTOR_ADDRESS_PROP);

        vm.detach();

    }

}