/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.cxf.common.util;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
import javax.tools.Diagnostic;
import javax.tools.DiagnosticListener;
import javax.tools.JavaCompiler;
import javax.tools.JavaCompiler.CompilationTask;
import javax.tools.JavaFileManager;
import javax.tools.JavaFileObject;
import javax.tools.StandardJavaFileManager;
import javax.tools.ToolProvider;
import org.apache.cxf.helpers.FileUtils;
public class Compiler {
private long maxMemory = Runtime.getRuntime().maxMemory();
private boolean verbose;
private String target;
private String outputDir;
private String classPath;
private String encoding;
private boolean forceFork = Boolean.getBoolean(Compiler.class.getName() + "-fork");
private File classpathTmpFile;
private List<String> errors = new LinkedList<String>();
private List<String> warnings = new LinkedList<String>();
public Compiler() {
}
public List<String> getErrors() {
return errors;
}
public List<String> getWarnings() {
return warnings;
}
public void setMaxMemory(long l) {
maxMemory = l;
}
public void setVerbose(boolean b) {
verbose = b;
}
public void setTarget(String s) {
target = s;
}
public void setOutputDir(File s) {
if (s != null) {
outputDir = s.getAbsolutePath().replace(File.pathSeparatorChar, '/');
} else {
outputDir = null;
}
}
public void setOutputDir(String s) {
outputDir = s.replace(File.pathSeparatorChar, '/');
}
public void setClassPath(String s) {
classPath = StringUtils.isEmpty(s) ? null : s;
}
protected void addArgs(List<String> list) {
if (verbose) {
list.add("-verbose");
}
if (!StringUtils.isEmpty(encoding)) {
list.add("-encoding");
list.add(encoding);
}
if (!StringUtils.isEmpty(target)) {
list.add("-target");
list.add(target);
list.add("-source");
list.add(target);
}
if (!StringUtils.isEmpty(outputDir)) {
list.add("-d");
list.add(outputDir);
}
if (StringUtils.isEmpty(classPath)) {
String javaClasspath = SystemPropertyAction.getProperty("java.class.path");
boolean classpathSetted = javaClasspath != null ? true : false;
if (!classpathSetted) {
File f = new File(getClass().getClassLoader().getResource(".").getFile());
f = new File(f, "../lib");
if (f.exists() && f.isDirectory()) {
list.add("-extdirs");
list.add(f.toString());
}
} else {
list.add("-classpath");
list.add(javaClasspath);
}
} else {
list.add("-classpath");
list.add(classPath);
}
}
public boolean compileFiles(File[] files) {
List<String> f = new ArrayList<String>(files.length);
for (File file : files) {
f.add(file.getAbsolutePath());
}
return compileFiles(f.toArray(new String[files.length]));
}
public boolean compileFiles(List<File> files) {
List<String> f = new ArrayList<String>(files.size());
for (File file : files) {
f.add(file.getAbsolutePath());
}
return compileFiles(f.toArray(new String[files.size()]));
}
public boolean compileFiles(String[] files) {
String endorsed = SystemPropertyAction.getProperty("java.endorsed.dirs");
if (!forceFork) {
return useJava6Compiler(files);
}
List<String> list = new ArrayList<String>();
// Start of honoring java.home for used javac
String fsep = File.separator;
String javacstr = "javac";
String platformjavacname = "javac";
if (SystemPropertyAction.getProperty("os.name").toLowerCase().indexOf("windows") > -1) {
platformjavacname = "javac.exe";
}
if (new File(SystemPropertyAction.getProperty("java.home") + fsep + platformjavacname).exists()) {
// check if java.home is jdk home
javacstr = SystemPropertyAction.getProperty("java.home") + fsep + platformjavacname;
} else if (new File(SystemPropertyAction.getProperty("java.home") + fsep + ".." + fsep + "bin" + fsep
+ platformjavacname).exists()) {
// check if java.home is jre home
javacstr = SystemPropertyAction.getProperty("java.home") + fsep + ".." + fsep + "bin" + fsep
+ platformjavacname;
}
list.add(javacstr);
// End of honoring java.home for used javac
if (!StringUtils.isEmpty(endorsed)) {
list.add("-endorseddirs");
list.add(endorsed);
}
//fix for CXF-2081, set maximum heap of this VM to javac.
list.add("-J-Xmx" + maxMemory);
addArgs(list);
int classpathIdx = list.indexOf("-classpath");
String classpath = list.get(classpathIdx + 1);
checkLongClasspath(classpath, list, classpathIdx);
int idx = list.size();
list.addAll(Arrays.asList(files));
return internalCompile(list.toArray(new String[list.size()]), idx);
}
protected boolean useJava6Compiler(String[] files) {
JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
Iterable<? extends JavaFileObject> fileList = fileManager.getJavaFileObjectsFromStrings(Arrays
.asList(files));
return internalJava6Compile(compiler, wrapJavaFileManager(fileManager), setupDiagnosticListener(),
fileList);
}
protected JavaFileManager wrapJavaFileManager(StandardJavaFileManager standardJavaFileManger) {
return standardJavaFileManger;
}
protected DiagnosticListener<JavaFileObject> setupDiagnosticListener() {
return new DiagnosticListener<JavaFileObject>() {
public void report(Diagnostic<? extends JavaFileObject> diagnostic) {
switch (diagnostic.getKind()) {
case ERROR:
errors.add(diagnostic.toString());
if (verbose) {
System.err.println(diagnostic.toString());
}
break;
case WARNING:
case MANDATORY_WARNING:
warnings.add(diagnostic.toString());
if (verbose) {
System.err.println(diagnostic.toString());
}
break;
default:
break;
}
}
};
}
protected boolean internalJava6Compile(JavaCompiler compiler, JavaFileManager fileManager,
DiagnosticListener<JavaFileObject> listener,
Iterable<? extends JavaFileObject> fileList) {
List<String> args = new ArrayList<String>();
addArgs(args);
CompilationTask task = compiler.getTask(null, fileManager, listener, args, null, fileList);
Boolean ret = task.call();
try {
fileManager.close();
} catch (IOException e) {
System.err.print("[ERROR] IOException during compiling.");
e.printStackTrace();
}
return ret;
}
public boolean internalCompile(String[] args, int sourceFileIndex) {
Process p = null;
String cmdArray[] = null;
File tmpFile = null;
try {
if (isLongCommandLines(args) && sourceFileIndex >= 0) {
PrintWriter out = null;
tmpFile = FileUtils.createTempFile("cxf-compiler", null);
out = new PrintWriter(new FileWriter(tmpFile));
for (int i = sourceFileIndex; i < args.length; i++) {
if (args[i].indexOf(" ") > -1) {
args[i] = args[i].replace(File.separatorChar, '/');
//
// javac gives an error if you use forward slashes
// with package-info.java. Refer to:
// http://bugs.sun.com/bugdatabase/view_bug.do?bug_id=6198196
//
if (args[i].indexOf("package-info.java") > -1
&& SystemPropertyAction.getProperty("os.name")
.toLowerCase().indexOf("windows") > -1) {
out.println("\"" + args[i].replaceAll("/", "\\\\\\\\") + "\"");
} else {
out.println("\"" + args[i] + "\"");
}
} else {
out.println(args[i]);
}
}
out.flush();
out.close();
cmdArray = new String[sourceFileIndex + 1];
System.arraycopy(args, 0, cmdArray, 0, sourceFileIndex);
cmdArray[sourceFileIndex] = "@" + tmpFile;
} else {
cmdArray = new String[args.length];
System.arraycopy(args, 0, cmdArray, 0, args.length);
}
if (SystemPropertyAction.getProperty("os.name").toLowerCase().indexOf("windows") > -1) {
for (int i = 0; i < cmdArray.length; i++) {
if (cmdArray[i].indexOf("package-info") == -1) {
cmdArray[i] = cmdArray[i].replace('\\', '/');
}
}
}
p = Runtime.getRuntime().exec(cmdArray);
if (p.getErrorStream() != null) {
StreamPrinter errorStreamPrinter = new StreamPrinter(p.getErrorStream(), "", System.out);
errorStreamPrinter.start();
}
if (p.getInputStream() != null) {
StreamPrinter infoStreamPrinter = new StreamPrinter(p.getInputStream(), "[INFO]", System.out);
infoStreamPrinter.start();
}
return p.waitFor() == 0 ? true : false;
} catch (SecurityException e) {
System.err.println("[ERROR] SecurityException during exec() of compiler \"" + args[0] + "\".");
} catch (InterruptedException e) {
// ignore
} catch (IOException e) {
System.err.print("[ERROR] IOException during exec() of compiler \"" + args[0] + "\"");
System.err.println(". Check your path environment variable.");
} finally {
if (tmpFile != null && tmpFile.exists()) {
FileUtils.delete(tmpFile);
}
if (classpathTmpFile != null && classpathTmpFile.exists()) {
FileUtils.delete(classpathTmpFile);
}
}
return false;
}
private boolean isLongCommandLines(String args[]) {
StringBuilder strBuffer = new StringBuilder();
for (int i = 0; i < args.length; i++) {
strBuffer.append(args[i]);
}
return strBuffer.toString().length() > 4096 ? true : false;
}
private boolean isLongClasspath(String classpath) {
return classpath.length() > 2048 ? true : false;
}
private void checkLongClasspath(String classpath, List<String> list, int classpathIdx) {
if (isLongClasspath(classpath)) {
PrintWriter out = null;
try {
classpathTmpFile = FileUtils.createTempFile("cxf-compiler-classpath", null);
out = new PrintWriter(new FileWriter(classpathTmpFile));
out.println(classpath);
out.flush();
out.close();
list.set(classpathIdx + 1, "@" + classpathTmpFile);
} catch (IOException e) {
System.err.print("[ERROR] can't write long classpath to @argfile");
}
}
}
public void setEncoding(String string) {
encoding = string;
}
}