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 1749603 2016-06-21 20:50:19Z 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("<"); 202 break; 203 case '>': 204 buf.append(">"); 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}