/*

 * Copyright (c) 2002, 2004, 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.

 *

 */

import com.sun.jdi.*;

import com.sun.jdi.connect.*;

import java.util.Map;

import java.util.List;

import java.util.Iterator;

import java.io.IOException;

public class sagclient {

    static AttachingConnector myCoreConn;

    static AttachingConnector myPIDConn;

    static AttachingConnector myDbgSvrConn;

    static VirtualMachine vm;

    static VirtualMachineManager vmmgr;

    public static void println(String msg) {

        System.out.println("jj: " + msg);

    }

    public static void main(String args[]) {

        vmmgr = Bootstrap.virtualMachineManager();

        List attachingConnectors = vmmgr.attachingConnectors();

        if (attachingConnectors.isEmpty()) {

            System.err.println( "ERROR: No attaching connectors");

            return;

        }

        Iterator myIt = attachingConnectors.iterator();

        while (myIt.hasNext()) {

            AttachingConnector tmpCon = (AttachingConnector)myIt.next();

            if (tmpCon.name().equals(

                "sun.jvm.hotspot.jdi.SACoreAttachingConnector")) {

                myCoreConn = tmpCon;

            } else if (tmpCon.name().equals(

                "sun.jvm.hotspot.jdi.SAPIDAttachingConnector")) {

                myPIDConn = tmpCon;

            } else if (tmpCon.name().equals(

                "sun.jvm.hotspot.jdi.SADebugServerAttachingConnector")) {

                myDbgSvrConn = tmpCon;

            }

        }

        String execPath = null;

        String pidText = null;

        String coreFilename = null;

        String debugServer = null;

        int pid = 0;

        switch (args.length) {

        case (0):

            break;

        case (1):

            // If all numbers, it is a PID to attach to

            // Else, it is a pathname to a .../bin/java for a core file.

            try {

                pidText = args[0];

                pid = Integer.parseInt(pidText);

                System.out.println( "pid: " + pid);

                vm = attachPID(pid);

            } catch (NumberFormatException e) {

                System.out.println("trying remote server ..");

                debugServer = args[0];

                System.out.println( "remote server: " + debugServer);

                vm = attachDebugServer(debugServer);

            }

            break;

        case (2):

            execPath = args[0];

            coreFilename = args[1];

            System.out.println( "jdk: " + execPath);

            System.out.println( "core: " + coreFilename);

            vm = attachCore(coreFilename, execPath);

            break;

        }

        if (vm != null) {

            System.out.println("sagclient: attached ok!");

            sagdoit mine = new sagdoit(vm);

            mine.doAll();

            vm.dispose();

        }

    }

    private static VirtualMachine attachCore(String coreFilename, String execPath) {

        Map connArgs = myCoreConn.defaultArguments();

        System.out.println("connArgs = " + connArgs);

        VirtualMachine vm;

        Connector.StringArgument connArg = (Connector.StringArgument)connArgs.get("core");

        connArg.setValue(coreFilename);

        connArg =  (Connector.StringArgument)connArgs.get("javaExecutable");

        connArg.setValue(execPath);

        try {

            vm = myCoreConn.attach(connArgs);

        } catch (IOException ee) {

            System.err.println("ERROR: myCoreConn.attach got IO Exception:" + ee);

            vm = null;

        } catch (IllegalConnectorArgumentsException ee) {

            System.err.println("ERROR: myCoreConn.attach got illegal args exception:" + ee);

            vm = null;

        }

        return vm;

   }

   private static VirtualMachine attachPID(int pid) {

        Map connArgs = myPIDConn.defaultArguments();

        System.out.println("connArgs = " + connArgs);

        VirtualMachine vm;

        Connector.StringArgument connArg = (Connector.StringArgument)connArgs.get("pid");

        connArg.setValue(Integer.toString(pid));

        try {

            vm = myPIDConn.attach(connArgs);

        } catch (IOException ee) {

            System.err.println("ERROR: myPIDConn.attach got IO Exception:" + ee);

            vm = null;

        } catch (IllegalConnectorArgumentsException ee) {

            System.err.println("ERROR: myPIDConn.attach got illegal args exception:" + ee);

            vm = null;

        }

        return vm;

   }

   private static VirtualMachine attachDebugServer(String debugServer) {

        Map connArgs = myDbgSvrConn.defaultArguments();

        System.out.println("connArgs = " + connArgs);

        VirtualMachine vm;

        Connector.StringArgument connArg = (Connector.StringArgument)connArgs.get("debugServerName");

        connArg.setValue(debugServer);

        try {

            vm = myDbgSvrConn.attach(connArgs);

        } catch (IOException ee) {

            System.err.println("ERROR: myDbgSvrConn.attach got IO Exception:" + ee);

            vm = null;

        } catch (IllegalConnectorArgumentsException ee) {

            System.err.println("ERROR: myDbgSvrConn.attach got illegal args exception:" + ee);

            vm = null;

        }

        return vm;

   }

}