/*

 * 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 *

 */

package sun.jvm.hotspot.tools.jcore;

import java.io.*;

import sun.jvm.hotspot.memory.*;

import sun.jvm.hotspot.oops.*;

import sun.jvm.hotspot.debugger.*;

import sun.jvm.hotspot.runtime.*;

import sun.jvm.hotspot.tools.*;

public class ClassDump extends Tool {

    private ClassFilter classFilter;

    private String      outputDirectory;

    public void run() {

        // Ready to go with the database...

        try {

            // load class filters

            String filterClassName = System.getProperty("sun.jvm.hotspot.tools.jcore.filter");

            if (filterClassName != null) {

                try {

                    Class filterClass = Class.forName(filterClassName);

                    classFilter = (ClassFilter) filterClass.newInstance();

                } catch(Exception exp) {

                    System.err.println("Warning: Can not create class filter!");

                }

            }

            outputDirectory = System.getProperty("sun.jvm.hotspot.tools.jcore.outputDir");

            if (outputDirectory == null)

                outputDirectory = ".";

            // walk through the system dictionary

            SystemDictionary dict = VM.getVM().getSystemDictionary();

            dict.classesDo(new SystemDictionary.ClassVisitor() {

                    public void visit(Klass k) {

                        if (k instanceof InstanceKlass) {

                            try {

                                dumpKlass((InstanceKlass) k);

                            } catch (Exception e) {

                                System.out.println(k.getName().asString());

                                e.printStackTrace();

                            }

                        }

                    }

                });

        }

        catch (AddressException e) {

            System.err.println("Error accessing address 0x"

                               + Long.toHexString(e.getAddress()));

            e.printStackTrace();

        }

    }

    public String getName() {

        return "jcore";

    }

    private void dumpKlass(InstanceKlass kls) {

        if (classFilter != null && ! classFilter.canInclude(kls) ) {

            return;

        }

        String klassName = kls.getName().asString();

        klassName = klassName.replace('/', File.separatorChar);

        int index = klassName.lastIndexOf(File.separatorChar);

        File dir = null;

        if (index != -1) {

            String dirName = klassName.substring(0, index);

            dir =  new File(outputDirectory,  dirName);

        } else {

            dir = new File(outputDirectory);

        }

        dir.mkdirs();

        File f = new File(dir, klassName.substring(klassName.lastIndexOf(File.separatorChar) + 1)

                          + ".class");

        try {

            f.createNewFile();

            OutputStream os = new BufferedOutputStream(new FileOutputStream(f));

            try {

                ClassWriter cw = new ClassWriter(kls, os);

                cw.write();

            } finally {

                os.close();

            }

        } catch(IOException exp) {

            exp.printStackTrace();

        }

    }

    public static void main(String[] args) {

        ClassDump cd = new ClassDump();

        cd.start(args);

        cd.stop();

    }

}