140 lines
4.5 KiB
Java
140 lines
4.5 KiB
Java
/*
|
|
* Copyright (c) 2011, 2018, 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.
|
|
*/
|
|
|
|
package vm.mlvm.share;
|
|
|
|
import java.io.File;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
|
|
import nsk.share.ArgumentParser;
|
|
|
|
public abstract class ClassfileGenerator {
|
|
|
|
public static class Klass {
|
|
private String pkgName;
|
|
private String className;
|
|
private byte[] bytes;
|
|
private String mainMethodName;
|
|
private String mainMethodSignature;
|
|
|
|
public Klass(String packageName, String className, String mainMethodName, String mainMethodSignature, byte[] bytes) {
|
|
this.pkgName = packageName;
|
|
this.className = className;
|
|
this.mainMethodName = mainMethodName;
|
|
this.mainMethodSignature = mainMethodSignature;
|
|
this.bytes = bytes;
|
|
}
|
|
|
|
public String getClassName() {
|
|
return (this.pkgName != null ? (this.pkgName + ".") : "") + this.className;
|
|
}
|
|
|
|
public String getSimpleClassName() {
|
|
return this.className;
|
|
}
|
|
|
|
public String getPackageName() {
|
|
return this.pkgName;
|
|
}
|
|
|
|
public String getMainMethodName() {
|
|
return mainMethodName;
|
|
}
|
|
|
|
public String getMainMethodSignature() {
|
|
return mainMethodSignature;
|
|
}
|
|
|
|
public byte[] getBytes() {
|
|
return this.bytes;
|
|
}
|
|
|
|
public void writeClass(String destDir) throws IOException {
|
|
boolean wroteOk = false;
|
|
|
|
File outDir;
|
|
if (this.pkgName != null)
|
|
outDir = new File(destDir, this.pkgName.replace('.', '/'));
|
|
else
|
|
outDir = new File(".");
|
|
|
|
outDir.mkdirs();
|
|
|
|
File outFile = new File(outDir, this.className.concat(".class"));
|
|
FileOutputStream outStream = new FileOutputStream(outFile);
|
|
try {
|
|
outStream.write(this.bytes);
|
|
wroteOk = true;
|
|
} finally {
|
|
outStream.close();
|
|
|
|
if (!wroteOk)
|
|
outFile.delete();
|
|
}
|
|
}
|
|
}
|
|
|
|
protected String fullClassName, pkgName, shortClassName;
|
|
|
|
public void setClassName(String pkgName, String shortClassName) {
|
|
this.pkgName = pkgName;
|
|
this.shortClassName = shortClassName;
|
|
fullClassName = (pkgName == null ? "" : (pkgName.replace('.', '/') + '/'))
|
|
+ shortClassName;
|
|
}
|
|
|
|
public abstract Klass[] generateBytecodes();
|
|
|
|
public static void main(String[] args) {
|
|
try {
|
|
Env.init(new ArgumentParser(args) {
|
|
@Override
|
|
protected boolean checkOption(String option, String value) {
|
|
if (option.equals("d"))
|
|
return true;
|
|
|
|
return super.checkOption(option, value);
|
|
}
|
|
});
|
|
|
|
Class<?> caller = Class.forName(Thread.currentThread()
|
|
.getStackTrace()[2].getClassName());
|
|
ClassfileGenerator gen = (ClassfileGenerator) caller.newInstance();
|
|
gen.setClassName(caller.getPackage().getName(), caller
|
|
.getSimpleName().replaceFirst("^GENERATE_", ""));
|
|
|
|
String destDir = Env.getArgParser().getOptions().getProperty("d");
|
|
|
|
Klass[] klasses = gen.generateBytecodes();
|
|
|
|
for (Klass k : klasses)
|
|
k.writeClass(destDir);
|
|
|
|
} catch (Exception e) {
|
|
Env.complain(e, "Generator caught an error");
|
|
System.exit(1);
|
|
}
|
|
}
|
|
}
|