    001/*
    002 * Licensed to the Apache Software Foundation (ASF) under one or more
    003 * contributor license agreements.  See the NOTICE file distributed with
    004 * this work for additional information regarding copyright ownership.
    005 * The ASF licenses this file to You under the Apache License, Version 2.0
    006 * (the "License"); you may not use this file except in compliance with
    007 * the License.  You may obtain a copy of the License at
    008 *
    009 *      http://www.apache.org/licenses/LICENSE-2.0
    010 *
    011 *  Unless required by applicable law or agreed to in writing, software
    012 *  distributed under the License is distributed on an "AS IS" BASIS,
    013 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    014 *  See the License for the specific language governing permissions and
    015 *  limitations under the License.
    016 *
    017 */
    018package org.apache.bcel.util;
    019
    020import java.io.File;
    021import java.io.FileOutputStream;
    022import java.io.IOException;
    023import java.io.PrintWriter;
    024import java.util.HashSet;
    025import java.util.Set;
    026
    027import org.apache.bcel.Const;
    028import org.apache.bcel.Constants;
    029import org.apache.bcel.classfile.Attribute;
    030import org.apache.bcel.classfile.ClassParser;
    031import org.apache.bcel.classfile.ConstantPool;
    032import org.apache.bcel.classfile.JavaClass;
    033import org.apache.bcel.classfile.Method;
    034import org.apache.bcel.classfile.Utility;
    035
    036/**
    037 * Read class file(s) and convert them into HTML files.
    038 *
    039 * Given a JavaClass object "class" that is in package "package" five files
    040 * will be created in the specified directory.
    041 *
    042 * <OL>
    043 * <LI> "package"."class".html as the main file which defines the frames for
    044 * the following subfiles.
    045 * <LI>  "package"."class"_attributes.html contains all (known) attributes found in the file
    046 * <LI>  "package"."class"_cp.html contains the constant pool
    047 * <LI>  "package"."class"_code.html contains the byte code
    048 * <LI>  "package"."class"_methods.html contains references to all methods and fields of the class
    049 * </OL>
    050 *
    051 * All subfiles reference each other appropriately, e.g. clicking on a
    052 * method in the Method's frame will jump to the appropriate method in
    053 * the Code frame.
    054 *
    055 * @version $Id: Class2HTML.java 1806200 2017-08-25 16:33:06Z ggregory $
    056 */
    057public class Class2HTML implements Constants {
    058
    059    private final JavaClass java_class; // current class object
    060    private final String dir;
    061    private static String class_package; // name of package, unclean to make it static, but ...
    062    private static String class_name; // name of current class, dito
    063    private static ConstantPool constant_pool;
    064    private static final Set<String> basic_types = new HashSet<>();
    065
    066    static {
    067        basic_types.add("int");
    068        basic_types.add("short");
    069        basic_types.add("boolean");
    070        basic_types.add("void");
    071        basic_types.add("char");
    072        basic_types.add("byte");
    073        basic_types.add("long");
    074        basic_types.add("double");
    075        basic_types.add("float");
    076    }
    077
    078    /**
    079     * Write contents of the given JavaClass into HTML files.
    080     *
    081     * @param java_class The class to write
    082     * @param dir The directory to put the files in
    083     */
    084    public Class2HTML(final JavaClass java_class, final String dir) throws IOException {
    085        final Method[] methods = java_class.getMethods();
    086        this.java_class = java_class;
    087        this.dir = dir;
    088        class_name = java_class.getClassName(); // Remember full name
    089        constant_pool = java_class.getConstantPool();
    090        // Get package name by tacking off everything after the last `.'
    091        final int index = class_name.lastIndexOf('.');
    092        if (index > -1) {
    093            class_package = class_name.substring(0, index);
    094        } else {
    095            class_package = ""; // default package
    096        }
    097        final ConstantHTML constant_html = new ConstantHTML(dir, class_name, class_package, methods,
    098                constant_pool);
    099        /* Attributes can't be written in one step, so we just open a file
    100         * which will be written consequently.
    101         */
    102        final AttributeHTML attribute_html = new AttributeHTML(dir, class_name, constant_pool,
    103                constant_html);
    104        new MethodHTML(dir, class_name, methods, java_class.getFields(),
    105                constant_html, attribute_html);
    106        // Write main file (with frames, yuk)
    107        writeMainHTML(attribute_html);
    108        new CodeHTML(dir, class_name, methods, constant_pool, constant_html);
    109        attribute_html.close();
    110    }
    111
    112
    113    public static void main( final String[] argv ) throws IOException {
    114        final String[] file_name = new String[argv.length];
    115        int files = 0;
    116        ClassParser parser = null;
    117        JavaClass java_class = null;
    118        String zip_file = null;
    119        final char sep = File.separatorChar;
    120        String dir = "." + sep; // Where to store HTML files
    121        /* Parse command line arguments.
    122         */
    123        for (int i = 0; i < argv.length; i++) {
    124            if (argv[i].charAt(0) == '-') { // command line switch
    125                if (argv[i].equals("-d")) { // Specify target directory, default '.'
    126                    dir = argv[++i];
    127                    if (!dir.endsWith("" + sep)) {
    128                        dir = dir + sep;
    129                    }
    130                    final File store = new File(dir);
    131                    if (!store.isDirectory()) {
    132                        final boolean created = store.mkdirs(); // Create target directory if necessary
    133                        if (!created) {
    134                            if (!store.isDirectory()) {
    135                                System.out.println("Tried to create the directory " + dir + " but failed");
    136                            }
    137                        }
    138                    }
    139                } else if (argv[i].equals("-zip")) {
    140                    zip_file = argv[++i];
    141                } else {
    142                    System.out.println("Unknown option " + argv[i]);
    143                }
    144            } else {
    145                file_name[files++] = argv[i];
    146            }
    147        }
    148        if (files == 0) {
    149            System.err.println("Class2HTML: No input files specified.");
    150        } else { // Loop through files ...
    151            for (int i = 0; i < files; i++) {
    152                System.out.print("Processing " + file_name[i] + "...");
    153                if (zip_file == null) {
    154                    parser = new ClassParser(file_name[i]); // Create parser object from file
    155                } else {
    156                    parser = new ClassParser(zip_file, file_name[i]); // Create parser object from zip file
    157                }
    158                java_class = parser.parse();
    159                new Class2HTML(java_class, dir);
    160                System.out.println("Done.");
    161            }
    162        }
    163    }
    164
    165
    166    /**
    167     * Utility method that converts a class reference in the constant pool,
    168     * i.e., an index to a string.
    169     */
    170    static String referenceClass( final int index ) {
    171        String str = constant_pool.getConstantString(index, Const.CONSTANT_Class);
    172        str = Utility.compactClassName(str);
    173        str = Utility.compactClassName(str, class_package + ".", true);
    174        return "<A HREF=\"" + class_name + "_cp.html#cp" + index + "\" TARGET=ConstantPool>" + str
    175                + "</A>";
    176    }
    177
    178
    179    static String referenceType( final String type ) {
    180        String short_type = Utility.compactClassName(type);
    181        short_type = Utility.compactClassName(short_type, class_package + ".", true);
    182        final int index = type.indexOf('['); // Type is an array?
    183        String base_type = type;
    184        if (index > -1) {
    185            base_type = type.substring(0, index); // Tack of the `['
    186        }
    187        // test for basic type
    188        if (basic_types.contains(base_type)) {
    189            return "<FONT COLOR=\"#00FF00\">" + type + "</FONT>";
    190        }
    191        return "<A HREF=\"" + base_type + ".html\" TARGET=_top>" + short_type + "</A>";
    192    }
    193
    194
    195    static String toHTML( final String str ) {
    196        final StringBuilder buf = new StringBuilder();
    197        for (int i = 0; i < str.length(); i++) {
    198            char ch;
    199            switch (ch = str.charAt(i)) {
    200                case '<':
    201                    buf.append("&lt;");
    202                    break;
    203                case '>':
    204                    buf.append("&gt;");
    205                    break;
    206                case '\n':
    207                    buf.append("\\n");
    208                    break;
    209                case '\r':
    210                    buf.append("\\r");
    211                    break;
    212                default:
    213                    buf.append(ch);
    214            }
    215        }
    216        return buf.toString();
    217    }
    218
    219
    220    private void writeMainHTML( final AttributeHTML attribute_html ) throws IOException {
    221        try (PrintWriter file = new PrintWriter(new FileOutputStream(dir + class_name + ".html"))) {
    222            file.println("<HTML>\n" + "<HEAD><TITLE>Documentation for " + class_name + "</TITLE>" + "</HEAD>\n"
    223                    + "<FRAMESET BORDER=1 cols=\"30%,*\">\n" + "<FRAMESET BORDER=1 rows=\"80%,*\">\n"
    224                    + "<FRAME NAME=\"ConstantPool\" SRC=\"" + class_name + "_cp.html" + "\"\n MARGINWIDTH=\"0\" "
    225                    + "MARGINHEIGHT=\"0\" FRAMEBORDER=\"1\" SCROLLING=\"AUTO\">\n" + "<FRAME NAME=\"Attributes\" SRC=\""
    226                    + class_name + "_attributes.html" + "\"\n MARGINWIDTH=\"0\" "
    227                    + "MARGINHEIGHT=\"0\" FRAMEBORDER=\"1\" SCROLLING=\"AUTO\">\n" + "</FRAMESET>\n"
    228                    + "<FRAMESET BORDER=1 rows=\"80%,*\">\n" + "<FRAME NAME=\"Code\" SRC=\"" + class_name
    229                    + "_code.html\"\n MARGINWIDTH=0 " + "MARGINHEIGHT=0 FRAMEBORDER=1 SCROLLING=\"AUTO\">\n"
    230                    + "<FRAME NAME=\"Methods\" SRC=\"" + class_name + "_methods.html\"\n MARGINWIDTH=0 "
    231                    + "MARGINHEIGHT=0 FRAMEBORDER=1 SCROLLING=\"AUTO\">\n" + "</FRAMESET></FRAMESET></HTML>");
    232        }
    233        final Attribute[] attributes = java_class.getAttributes();
    234        for (int i = 0; i < attributes.length; i++) {
    235            attribute_html.writeAttribute(attributes[i], "class" + i);
    236        }
    237    }
    238}