fix(autojs): loadDex and loadJar bugs
This commit is contained in:
hyb1996
@@ -1,735 +0,0 @@
|
|||||||
package org.autojs.autojs;
|
|
||||||
|
|
||||||
import org.mozilla.javascript.Context;
|
|
||||||
import org.mozilla.javascript.ContextAction;
|
|
||||||
import org.mozilla.javascript.Function;
|
|
||||||
import org.mozilla.javascript.GeneratedClassLoader;
|
|
||||||
import org.mozilla.javascript.Kit;
|
|
||||||
import org.mozilla.javascript.NativeArray;
|
|
||||||
import org.mozilla.javascript.RhinoException;
|
|
||||||
import org.mozilla.javascript.Script;
|
|
||||||
import org.mozilla.javascript.Scriptable;
|
|
||||||
import org.mozilla.javascript.ScriptableObject;
|
|
||||||
import org.mozilla.javascript.SecurityController;
|
|
||||||
import org.mozilla.javascript.commonjs.module.ModuleScope;
|
|
||||||
import org.mozilla.javascript.commonjs.module.Require;
|
|
||||||
import org.mozilla.javascript.tools.SourceReader;
|
|
||||||
import org.mozilla.javascript.tools.ToolErrorReporter;
|
|
||||||
import org.mozilla.javascript.tools.shell.Global;
|
|
||||||
import org.mozilla.javascript.tools.shell.QuitAction;
|
|
||||||
import org.mozilla.javascript.tools.shell.SecurityProxy;
|
|
||||||
import org.mozilla.javascript.tools.shell.ShellConsole;
|
|
||||||
import org.mozilla.javascript.tools.shell.ShellContextFactory;
|
|
||||||
|
|
||||||
import java.io.File;
|
|
||||||
import java.io.FileNotFoundException;
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.io.InputStream;
|
|
||||||
import java.io.PrintStream;
|
|
||||||
import java.io.UnsupportedEncodingException;
|
|
||||||
import java.lang.ref.ReferenceQueue;
|
|
||||||
import java.lang.ref.SoftReference;
|
|
||||||
import java.net.URI;
|
|
||||||
import java.net.URISyntaxException;
|
|
||||||
import java.nio.charset.Charset;
|
|
||||||
import java.security.MessageDigest;
|
|
||||||
import java.security.NoSuchAlgorithmException;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.Arrays;
|
|
||||||
import java.util.LinkedHashMap;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* The shell program.
|
|
||||||
* <p>
|
|
||||||
* Can execute scripts interactively or in batch mode at the command line.
|
|
||||||
* An example of controlling the JavaScript engine.
|
|
||||||
*
|
|
||||||
* @author Norris Boyd
|
|
||||||
*/
|
|
||||||
public class Main {
|
|
||||||
public static ShellContextFactory
|
|
||||||
shellContextFactory = new ShellContextFactory();
|
|
||||||
|
|
||||||
public static Global global = new Global();
|
|
||||||
static protected ToolErrorReporter errorReporter;
|
|
||||||
static protected int exitCode = 0;
|
|
||||||
static private final int EXITCODE_RUNTIME_ERROR = 3;
|
|
||||||
static private final int EXITCODE_FILE_NOT_FOUND = 4;
|
|
||||||
static boolean processStdin = true;
|
|
||||||
static List<String> fileList = new ArrayList<String>();
|
|
||||||
static List<String> modulePath;
|
|
||||||
static String mainModule;
|
|
||||||
static boolean sandboxed = false;
|
|
||||||
static boolean useRequire = false;
|
|
||||||
static Require require;
|
|
||||||
private static SecurityProxy securityImpl;
|
|
||||||
private final static ScriptCache scriptCache = new ScriptCache(32);
|
|
||||||
|
|
||||||
static {
|
|
||||||
global.initQuitAction(new IProxy(IProxy.SYSTEM_EXIT));
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Proxy class to avoid proliferation of anonymous classes.
|
|
||||||
*/
|
|
||||||
private static class IProxy implements ContextAction, QuitAction {
|
|
||||||
private static final int PROCESS_FILES = 1;
|
|
||||||
private static final int EVAL_INLINE_SCRIPT = 2;
|
|
||||||
private static final int SYSTEM_EXIT = 3;
|
|
||||||
|
|
||||||
private int type;
|
|
||||||
String[] args;
|
|
||||||
String scriptText;
|
|
||||||
|
|
||||||
IProxy(int type) {
|
|
||||||
this.type = type;
|
|
||||||
}
|
|
||||||
|
|
||||||
public Object run(Context cx) {
|
|
||||||
if (useRequire) {
|
|
||||||
require = global.installRequire(cx, modulePath, sandboxed);
|
|
||||||
}
|
|
||||||
if (type == PROCESS_FILES) {
|
|
||||||
processFiles(cx, args);
|
|
||||||
} else if (type == EVAL_INLINE_SCRIPT) {
|
|
||||||
evalInlineScript(cx, scriptText);
|
|
||||||
} else {
|
|
||||||
throw Kit.codeBug();
|
|
||||||
}
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
public void quit(Context cx, int exitCode) {
|
|
||||||
if (type == SYSTEM_EXIT) {
|
|
||||||
System.exit(exitCode);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
throw Kit.codeBug();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Main entry point.
|
|
||||||
* <p>
|
|
||||||
* Process arguments as would a normal Java program. Also
|
|
||||||
* create a new Context and associate it with the current thread.
|
|
||||||
* Then set up the execution environment and begin to
|
|
||||||
* execute scripts.
|
|
||||||
*/
|
|
||||||
public static void main(String args[]) {
|
|
||||||
try {
|
|
||||||
if (Boolean.getBoolean("rhino.use_java_policy_security")) {
|
|
||||||
initJavaPolicySecuritySupport();
|
|
||||||
}
|
|
||||||
} catch (SecurityException ex) {
|
|
||||||
ex.printStackTrace(System.err);
|
|
||||||
}
|
|
||||||
|
|
||||||
int result = exec(args);
|
|
||||||
if (result != 0) {
|
|
||||||
System.exit(result);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Execute the given arguments, but don't System.exit at the end.
|
|
||||||
*/
|
|
||||||
public static int exec(String origArgs[]) {
|
|
||||||
errorReporter = new ToolErrorReporter(false, global.getErr());
|
|
||||||
shellContextFactory.setErrorReporter(errorReporter);
|
|
||||||
String[] args = processOptions(origArgs);
|
|
||||||
if (exitCode > 0) {
|
|
||||||
return exitCode;
|
|
||||||
}
|
|
||||||
if (processStdin) {
|
|
||||||
fileList.add(null);
|
|
||||||
}
|
|
||||||
if (true) {
|
|
||||||
global.init(shellContextFactory);
|
|
||||||
}
|
|
||||||
IProxy iproxy = new IProxy(IProxy.PROCESS_FILES);
|
|
||||||
iproxy.args = args;
|
|
||||||
shellContextFactory.call(iproxy);
|
|
||||||
|
|
||||||
return exitCode;
|
|
||||||
}
|
|
||||||
|
|
||||||
static void processFiles(Context cx, String[] args) {
|
|
||||||
// define "arguments" array in the top-level object:
|
|
||||||
// need to allocate new array since newArray requires instances
|
|
||||||
// of exactly Object[], not ObjectSubclass[]
|
|
||||||
Object[] array = new Object[args.length];
|
|
||||||
System.arraycopy(args, 0, array, 0, args.length);
|
|
||||||
Scriptable argsObj = cx.newArray(global, array);
|
|
||||||
global.defineProperty("arguments", argsObj,
|
|
||||||
ScriptableObject.DONTENUM);
|
|
||||||
|
|
||||||
for (String file : fileList) {
|
|
||||||
try {
|
|
||||||
processSource(cx, file);
|
|
||||||
} catch (IOException ioex) {
|
|
||||||
Context.reportError(ToolErrorReporter.getMessage(
|
|
||||||
"msg.couldnt.read.source", file, ioex.getMessage()));
|
|
||||||
exitCode = EXITCODE_FILE_NOT_FOUND;
|
|
||||||
} catch (RhinoException rex) {
|
|
||||||
ToolErrorReporter.reportException(
|
|
||||||
cx.getErrorReporter(), rex);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
} catch (VirtualMachineError ex) {
|
|
||||||
// Treat StackOverflow and OutOfMemory as runtime errors
|
|
||||||
ex.printStackTrace();
|
|
||||||
String msg = ToolErrorReporter.getMessage(
|
|
||||||
"msg.uncaughtJSException", ex.toString());
|
|
||||||
Context.reportError(msg);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void evalInlineScript(Context cx, String scriptText) {
|
|
||||||
try {
|
|
||||||
Script script = cx.compileString(scriptText, "<command>", 1, null);
|
|
||||||
if (script != null) {
|
|
||||||
script.exec(cx, getShellScope());
|
|
||||||
}
|
|
||||||
} catch (RhinoException rex) {
|
|
||||||
ToolErrorReporter.reportException(
|
|
||||||
cx.getErrorReporter(), rex);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
} catch (VirtualMachineError ex) {
|
|
||||||
// Treat StackOverflow and OutOfMemory as runtime errors
|
|
||||||
ex.printStackTrace();
|
|
||||||
String msg = ToolErrorReporter.getMessage(
|
|
||||||
"msg.uncaughtJSException", ex.toString());
|
|
||||||
Context.reportError(msg);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public static Global getGlobal() {
|
|
||||||
return global;
|
|
||||||
}
|
|
||||||
|
|
||||||
static Scriptable getShellScope() {
|
|
||||||
return getScope(null);
|
|
||||||
}
|
|
||||||
|
|
||||||
static Scriptable getScope(String path) {
|
|
||||||
if (useRequire) {
|
|
||||||
// If CommonJS modules are enabled use a module scope that resolves
|
|
||||||
// relative ids relative to the current URL, file or working directory.
|
|
||||||
URI uri;
|
|
||||||
if (path == null) {
|
|
||||||
// use current directory for shell and -e switch
|
|
||||||
uri = new File(System.getProperty("user.dir")).toURI();
|
|
||||||
} else {
|
|
||||||
// find out whether this is a file path or a URL
|
|
||||||
if (SourceReader.toUrl(path) != null) {
|
|
||||||
try {
|
|
||||||
uri = new URI(path);
|
|
||||||
} catch (URISyntaxException x) {
|
|
||||||
// fall back to file uri
|
|
||||||
uri = new File(path).toURI();
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
uri = new File(path).toURI();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return new ModuleScope(global, uri, null);
|
|
||||||
} else {
|
|
||||||
return global;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Parse arguments.
|
|
||||||
*/
|
|
||||||
public static String[] processOptions(String args[]) {
|
|
||||||
String usageError;
|
|
||||||
goodUsage:
|
|
||||||
for (int i = 0; ; ++i) {
|
|
||||||
if (i == args.length) {
|
|
||||||
return new String[0];
|
|
||||||
}
|
|
||||||
String arg = args[i];
|
|
||||||
if (!arg.startsWith("-")) {
|
|
||||||
processStdin = false;
|
|
||||||
fileList.add(arg);
|
|
||||||
mainModule = arg;
|
|
||||||
String[] result = new String[args.length - i - 1];
|
|
||||||
System.arraycopy(args, i + 1, result, 0, args.length - i - 1);
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
if (arg.equals("-version")) {
|
|
||||||
if (++i == args.length) {
|
|
||||||
usageError = arg;
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
int version;
|
|
||||||
try {
|
|
||||||
version = Integer.parseInt(args[i]);
|
|
||||||
} catch (NumberFormatException ex) {
|
|
||||||
usageError = args[i];
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
if (!Context.isValidLanguageVersion(version)) {
|
|
||||||
usageError = args[i];
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
shellContextFactory.setLanguageVersion(version);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-opt") || arg.equals("-O")) {
|
|
||||||
if (++i == args.length) {
|
|
||||||
usageError = arg;
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
int opt;
|
|
||||||
try {
|
|
||||||
opt = Integer.parseInt(args[i]);
|
|
||||||
} catch (NumberFormatException ex) {
|
|
||||||
usageError = args[i];
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
if (opt == -2) {
|
|
||||||
// Compatibility with Cocoon Rhino fork
|
|
||||||
opt = -1;
|
|
||||||
} else if (!Context.isValidOptimizationLevel(opt)) {
|
|
||||||
usageError = args[i];
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
shellContextFactory.setOptimizationLevel(opt);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-encoding")) {
|
|
||||||
if (++i == args.length) {
|
|
||||||
usageError = arg;
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
String enc = args[i];
|
|
||||||
shellContextFactory.setCharacterEncoding(enc);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-strict")) {
|
|
||||||
shellContextFactory.setStrictMode(true);
|
|
||||||
shellContextFactory.setAllowReservedKeywords(false);
|
|
||||||
errorReporter.setIsReportingWarnings(true);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-fatal-warnings")) {
|
|
||||||
shellContextFactory.setWarningAsError(true);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-e")) {
|
|
||||||
processStdin = false;
|
|
||||||
if (++i == args.length) {
|
|
||||||
usageError = arg;
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
if (false) {
|
|
||||||
global.init(shellContextFactory);
|
|
||||||
}
|
|
||||||
IProxy iproxy = new IProxy(IProxy.EVAL_INLINE_SCRIPT);
|
|
||||||
iproxy.scriptText = args[i];
|
|
||||||
shellContextFactory.call(iproxy);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-require")) {
|
|
||||||
useRequire = true;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-sandbox")) {
|
|
||||||
sandboxed = true;
|
|
||||||
useRequire = true;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-modules")) {
|
|
||||||
if (++i == args.length) {
|
|
||||||
usageError = arg;
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
if (modulePath == null) {
|
|
||||||
modulePath = new ArrayList<String>();
|
|
||||||
}
|
|
||||||
modulePath.add(args[i]);
|
|
||||||
useRequire = true;
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-w")) {
|
|
||||||
errorReporter.setIsReportingWarnings(true);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-f")) {
|
|
||||||
processStdin = false;
|
|
||||||
if (++i == args.length) {
|
|
||||||
usageError = arg;
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
if (args[i].equals("-")) {
|
|
||||||
fileList.add(null);
|
|
||||||
} else {
|
|
||||||
fileList.add(args[i]);
|
|
||||||
mainModule = args[i];
|
|
||||||
}
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-sealedlib")) {
|
|
||||||
global.setSealedStdLib(true);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-debug")) {
|
|
||||||
shellContextFactory.setGeneratingDebug(true);
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if (arg.equals("-?") ||
|
|
||||||
arg.equals("-help")) {
|
|
||||||
// print usage message
|
|
||||||
global.getOut().println(
|
|
||||||
ToolErrorReporter.getMessage("msg.shell.usage", Main.class.getName()));
|
|
||||||
exitCode = 1;
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
usageError = arg;
|
|
||||||
break goodUsage;
|
|
||||||
}
|
|
||||||
// print error and usage message
|
|
||||||
global.getOut().println(
|
|
||||||
ToolErrorReporter.getMessage("msg.shell.invalid", usageError));
|
|
||||||
global.getOut().println(
|
|
||||||
ToolErrorReporter.getMessage("msg.shell.usage", Main.class.getName()));
|
|
||||||
exitCode = 1;
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void initJavaPolicySecuritySupport() {
|
|
||||||
Throwable exObj;
|
|
||||||
try {
|
|
||||||
Class<?> cl = Class.forName
|
|
||||||
("org.mozilla.javascript.tools.shell.JavaPolicySecurity");
|
|
||||||
securityImpl = (SecurityProxy) cl.newInstance();
|
|
||||||
SecurityController.initGlobal(securityImpl);
|
|
||||||
return;
|
|
||||||
} catch (ClassNotFoundException ex) {
|
|
||||||
exObj = ex;
|
|
||||||
} catch (IllegalAccessException ex) {
|
|
||||||
exObj = ex;
|
|
||||||
} catch (InstantiationException ex) {
|
|
||||||
exObj = ex;
|
|
||||||
} catch (LinkageError ex) {
|
|
||||||
exObj = ex;
|
|
||||||
}
|
|
||||||
throw Kit.initCause(new IllegalStateException(
|
|
||||||
"Can not load security support: " + exObj), exObj);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Evaluate JavaScript source.
|
|
||||||
*
|
|
||||||
* @param cx the current context
|
|
||||||
* @param filename the name of the file to compile, or null
|
|
||||||
* for interactive mode.
|
|
||||||
* @throws IOException if the source could not be read
|
|
||||||
* @throws RhinoException thrown during evaluation of source
|
|
||||||
*/
|
|
||||||
public static void processSource(Context cx, String filename)
|
|
||||||
throws IOException {
|
|
||||||
if (filename == null || filename.equals("-")) {
|
|
||||||
Scriptable scope = getShellScope();
|
|
||||||
Charset cs;
|
|
||||||
String charEnc = shellContextFactory.getCharacterEncoding();
|
|
||||||
if (charEnc != null) {
|
|
||||||
cs = Charset.forName(charEnc);
|
|
||||||
} else {
|
|
||||||
cs = Charset.defaultCharset();
|
|
||||||
}
|
|
||||||
ShellConsole console = global.getConsole(cs);
|
|
||||||
if (filename == null) {
|
|
||||||
// print implementation version
|
|
||||||
console.println(cx.getImplementationVersion());
|
|
||||||
}
|
|
||||||
|
|
||||||
int lineno = 1;
|
|
||||||
boolean hitEOF = false;
|
|
||||||
while (!hitEOF) {
|
|
||||||
String[] prompts = global.getPrompts(cx);
|
|
||||||
String prompt = null;
|
|
||||||
if (filename == null)
|
|
||||||
prompt = prompts[0];
|
|
||||||
console.flush();
|
|
||||||
String source = "";
|
|
||||||
|
|
||||||
// Collect lines of source to compile.
|
|
||||||
while (true) {
|
|
||||||
String newline;
|
|
||||||
try {
|
|
||||||
newline = console.readLine(prompt);
|
|
||||||
} catch (IOException ioe) {
|
|
||||||
console.println(ioe.toString());
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
if (newline == null) {
|
|
||||||
hitEOF = true;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
source = source + newline + "\n";
|
|
||||||
lineno++;
|
|
||||||
if (cx.stringIsCompilableUnit(source))
|
|
||||||
break;
|
|
||||||
prompt = prompts[1];
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
Script script = cx.compileString(source, "<stdin>", lineno, null);
|
|
||||||
if (script != null) {
|
|
||||||
Object result = script.exec(cx, scope);
|
|
||||||
// Avoid printing out undefined or function definitions.
|
|
||||||
if (result != Context.getUndefinedValue() &&
|
|
||||||
!(result instanceof Function &&
|
|
||||||
source.trim().startsWith("function"))) {
|
|
||||||
try {
|
|
||||||
console.println(Context.toString(result));
|
|
||||||
} catch (RhinoException rex) {
|
|
||||||
ToolErrorReporter.reportException(
|
|
||||||
cx.getErrorReporter(), rex);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// NativeArray h = global.history;
|
|
||||||
//h.put((int) h.getLength(), h, source);
|
|
||||||
}
|
|
||||||
} catch (RhinoException rex) {
|
|
||||||
ToolErrorReporter.reportException(
|
|
||||||
cx.getErrorReporter(), rex);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
} catch (VirtualMachineError ex) {
|
|
||||||
// Treat StackOverflow and OutOfMemory as runtime errors
|
|
||||||
ex.printStackTrace();
|
|
||||||
String msg = ToolErrorReporter.getMessage(
|
|
||||||
"msg.uncaughtJSException", ex.toString());
|
|
||||||
Context.reportError(msg);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
console.println();
|
|
||||||
console.flush();
|
|
||||||
} else if (useRequire && filename.equals(mainModule)) {
|
|
||||||
require.requireMain(cx, filename);
|
|
||||||
} else {
|
|
||||||
processFile(cx, getScope(filename), filename);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void processFileNoThrow(Context cx, Scriptable scope, String filename) {
|
|
||||||
try {
|
|
||||||
processFile(cx, scope, filename);
|
|
||||||
} catch (IOException ioex) {
|
|
||||||
Context.reportError(ToolErrorReporter.getMessage(
|
|
||||||
"msg.couldnt.read.source", filename, ioex.getMessage()));
|
|
||||||
exitCode = EXITCODE_FILE_NOT_FOUND;
|
|
||||||
} catch (RhinoException rex) {
|
|
||||||
ToolErrorReporter.reportException(
|
|
||||||
cx.getErrorReporter(), rex);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
} catch (VirtualMachineError ex) {
|
|
||||||
// Treat StackOverflow and OutOfMemory as runtime errors
|
|
||||||
ex.printStackTrace();
|
|
||||||
String msg = ToolErrorReporter.getMessage(
|
|
||||||
"msg.uncaughtJSException", ex.toString());
|
|
||||||
Context.reportError(msg);
|
|
||||||
exitCode = EXITCODE_RUNTIME_ERROR;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void processFile(Context cx, Scriptable scope, String filename)
|
|
||||||
throws IOException {
|
|
||||||
if (securityImpl == null) {
|
|
||||||
processFileSecure(cx, scope, filename, null);
|
|
||||||
} else {
|
|
||||||
//securityImpl.callProcessFileSecure(cx, scope, filename);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static void processFileSecure(Context cx, Scriptable scope,
|
|
||||||
String path, Object securityDomain)
|
|
||||||
throws IOException {
|
|
||||||
|
|
||||||
boolean isClass = path.endsWith(".class");
|
|
||||||
Object source = readFileOrUrl(path, !isClass);
|
|
||||||
|
|
||||||
byte[] digest = getDigest(source);
|
|
||||||
String key = path + "_" + cx.getOptimizationLevel();
|
|
||||||
ScriptReference ref = scriptCache.get(key, digest);
|
|
||||||
Script script = ref != null ? ref.get() : null;
|
|
||||||
|
|
||||||
if (script == null) {
|
|
||||||
if (isClass) {
|
|
||||||
script = loadCompiledScript(cx, path, (byte[]) source, securityDomain);
|
|
||||||
} else {
|
|
||||||
String strSrc = (String) source;
|
|
||||||
// Support the executable script #! syntax: If
|
|
||||||
// the first line begins with a '#', treat the whole
|
|
||||||
// line as a comment.
|
|
||||||
if (strSrc.length() > 0 && strSrc.charAt(0) == '#') {
|
|
||||||
for (int i = 1; i != strSrc.length(); ++i) {
|
|
||||||
int c = strSrc.charAt(i);
|
|
||||||
if (c == '\n' || c == '\r') {
|
|
||||||
strSrc = strSrc.substring(i);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
script = cx.compileString(strSrc, path, 1, securityDomain);
|
|
||||||
}
|
|
||||||
scriptCache.put(key, digest, script);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (script != null) {
|
|
||||||
script.exec(cx, scope);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private static byte[] getDigest(Object source) {
|
|
||||||
byte[] bytes, digest = null;
|
|
||||||
|
|
||||||
if (source != null) {
|
|
||||||
if (source instanceof String) {
|
|
||||||
try {
|
|
||||||
bytes = ((String) source).getBytes("UTF-8");
|
|
||||||
} catch (UnsupportedEncodingException ue) {
|
|
||||||
bytes = ((String) source).getBytes();
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
bytes = (byte[]) source;
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
MessageDigest md = MessageDigest.getInstance("MD5");
|
|
||||||
digest = md.digest(bytes);
|
|
||||||
} catch (NoSuchAlgorithmException nsa) {
|
|
||||||
// Should not happen
|
|
||||||
throw new RuntimeException(nsa);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return digest;
|
|
||||||
}
|
|
||||||
|
|
||||||
private static Script loadCompiledScript(Context cx, String path,
|
|
||||||
byte[] data, Object securityDomain)
|
|
||||||
throws FileNotFoundException {
|
|
||||||
if (data == null) {
|
|
||||||
throw new FileNotFoundException(path);
|
|
||||||
}
|
|
||||||
// XXX: For now extract class name of compiled Script from path
|
|
||||||
// instead of parsing class bytes
|
|
||||||
int nameStart = path.lastIndexOf('/');
|
|
||||||
if (nameStart < 0) {
|
|
||||||
nameStart = 0;
|
|
||||||
} else {
|
|
||||||
++nameStart;
|
|
||||||
}
|
|
||||||
int nameEnd = path.lastIndexOf('.');
|
|
||||||
if (nameEnd < nameStart) {
|
|
||||||
// '.' does not exist in path (nameEnd < 0)
|
|
||||||
// or it comes before nameStart
|
|
||||||
nameEnd = path.length();
|
|
||||||
}
|
|
||||||
String name = path.substring(nameStart, nameEnd);
|
|
||||||
try {
|
|
||||||
GeneratedClassLoader loader = SecurityController.createLoader(cx.getApplicationClassLoader(), securityDomain);
|
|
||||||
Class<?> clazz = loader.defineClass(name, data);
|
|
||||||
loader.linkClass(clazz);
|
|
||||||
if (!Script.class.isAssignableFrom(clazz)) {
|
|
||||||
throw Context.reportRuntimeError("msg.must.implement.Script");
|
|
||||||
}
|
|
||||||
return (Script) clazz.newInstance();
|
|
||||||
} catch (IllegalAccessException iaex) {
|
|
||||||
Context.reportError(iaex.toString());
|
|
||||||
throw new RuntimeException(iaex);
|
|
||||||
} catch (InstantiationException inex) {
|
|
||||||
Context.reportError(inex.toString());
|
|
||||||
throw new RuntimeException(inex);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public static InputStream getIn() {
|
|
||||||
return getGlobal().getIn();
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void setIn(InputStream in) {
|
|
||||||
getGlobal().setIn(in);
|
|
||||||
}
|
|
||||||
|
|
||||||
public static PrintStream getOut() {
|
|
||||||
return getGlobal().getOut();
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void setOut(PrintStream out) {
|
|
||||||
getGlobal().setOut(out);
|
|
||||||
}
|
|
||||||
|
|
||||||
public static PrintStream getErr() {
|
|
||||||
return getGlobal().getErr();
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void setErr(PrintStream err) {
|
|
||||||
getGlobal().setErr(err);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Read file or url specified by <tt>path</tt>.
|
|
||||||
*
|
|
||||||
* @return file or url content as <tt>byte[]</tt> or as <tt>String</tt> if
|
|
||||||
* <tt>convertToString</tt> is true.
|
|
||||||
*/
|
|
||||||
private static Object readFileOrUrl(String path, boolean convertToString)
|
|
||||||
throws IOException {
|
|
||||||
return SourceReader.readFileOrUrl(path, convertToString,
|
|
||||||
shellContextFactory.getCharacterEncoding());
|
|
||||||
}
|
|
||||||
|
|
||||||
static class ScriptReference extends SoftReference<Script> {
|
|
||||||
String path;
|
|
||||||
byte[] digest;
|
|
||||||
|
|
||||||
ScriptReference(String path, byte[] digest,
|
|
||||||
Script script, ReferenceQueue<Script> queue) {
|
|
||||||
super(script, queue);
|
|
||||||
this.path = path;
|
|
||||||
this.digest = digest;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
static class ScriptCache extends LinkedHashMap<String, ScriptReference> {
|
|
||||||
ReferenceQueue<Script> queue;
|
|
||||||
int capacity;
|
|
||||||
|
|
||||||
ScriptCache(int capacity) {
|
|
||||||
super(capacity + 1, 2f, true);
|
|
||||||
this.capacity = capacity;
|
|
||||||
queue = new ReferenceQueue<Script>();
|
|
||||||
}
|
|
||||||
|
|
||||||
@Override
|
|
||||||
protected boolean removeEldestEntry(Map.Entry<String, ScriptReference> eldest) {
|
|
||||||
return size() > capacity;
|
|
||||||
}
|
|
||||||
|
|
||||||
ScriptReference get(String path, byte[] digest) {
|
|
||||||
ScriptReference ref;
|
|
||||||
while ((ref = (ScriptReference) queue.poll()) != null) {
|
|
||||||
remove(ref.path);
|
|
||||||
}
|
|
||||||
ref = get(path);
|
|
||||||
if (ref != null && !Arrays.equals(digest, ref.digest)) {
|
|
||||||
remove(ref.path);
|
|
||||||
ref = null;
|
|
||||||
}
|
|
||||||
return ref;
|
|
||||||
}
|
|
||||||
|
|
||||||
void put(String path, byte[] digest, Script script) {
|
|
||||||
put(path, new ScriptReference(path, digest, script, queue));
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,5 +1,7 @@
|
|||||||
package com.stardust.autojs.rhino;
|
package com.stardust.autojs.rhino;
|
||||||
|
|
||||||
|
import android.util.Log;
|
||||||
|
|
||||||
import com.android.dx.command.dexer.Main;
|
import com.android.dx.command.dexer.Main;
|
||||||
|
|
||||||
import net.lingala.zip4j.core.ZipFile;
|
import net.lingala.zip4j.core.ZipFile;
|
||||||
@@ -11,6 +13,7 @@ import org.mozilla.javascript.GeneratedClassLoader;
|
|||||||
|
|
||||||
import java.io.ByteArrayInputStream;
|
import java.io.ByteArrayInputStream;
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
|
import java.io.FileNotFoundException;
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
@@ -24,11 +27,9 @@ import dalvik.system.DexClassLoader;
|
|||||||
public class AndroidClassLoader extends ClassLoader implements GeneratedClassLoader {
|
public class AndroidClassLoader extends ClassLoader implements GeneratedClassLoader {
|
||||||
|
|
||||||
|
|
||||||
|
private static final String LOG_TAG = "AndroidClassLoader";
|
||||||
private final ClassLoader parent;
|
private final ClassLoader parent;
|
||||||
private List<DexClassLoader> mDexClassLoaders;
|
private final List<DexClassLoader> mDexClassLoaders = new ArrayList<>();
|
||||||
private final File dexFile;
|
|
||||||
private final File odexOatFile;
|
|
||||||
private final File classFile;
|
|
||||||
private final File mCacheDir;
|
private final File mCacheDir;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -40,10 +41,6 @@ public class AndroidClassLoader extends ClassLoader implements GeneratedClassLoa
|
|||||||
public AndroidClassLoader(ClassLoader parent, File dir) {
|
public AndroidClassLoader(ClassLoader parent, File dir) {
|
||||||
this.parent = parent;
|
this.parent = parent;
|
||||||
mCacheDir = dir;
|
mCacheDir = dir;
|
||||||
mDexClassLoaders = new ArrayList<>();
|
|
||||||
dexFile = new File(dir, "dex-" + hashCode() + ".jar");
|
|
||||||
odexOatFile = new File(dir, "odex_oat-" + hashCode() + ".tmp");
|
|
||||||
classFile = new File(dir, "class-" + hashCode() + ".jar");
|
|
||||||
dir.mkdirs();
|
dir.mkdirs();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -52,23 +49,37 @@ public class AndroidClassLoader extends ClassLoader implements GeneratedClassLoa
|
|||||||
*/
|
*/
|
||||||
@Override
|
@Override
|
||||||
public Class<?> defineClass(String name, byte[] data) {
|
public Class<?> defineClass(String name, byte[] data) {
|
||||||
|
Log.d(LOG_TAG, "defineClass: name = " + name + " data.length = " + data.length);
|
||||||
|
File classFile = null;
|
||||||
try {
|
try {
|
||||||
|
classFile = createTempClassFile(name);
|
||||||
final ZipFile zipFile = new ZipFile(classFile);
|
final ZipFile zipFile = new ZipFile(classFile);
|
||||||
final ZipParameters parameters = new ZipParameters();
|
final ZipParameters parameters = new ZipParameters();
|
||||||
parameters.setFileNameInZip(name.replace('.', '/') + ".class");
|
parameters.setFileNameInZip(name.replace('.', '/') + ".class");
|
||||||
parameters.setSourceExternalStream(true);
|
parameters.setSourceExternalStream(true);
|
||||||
zipFile.addStream(new ByteArrayInputStream(data), parameters);
|
zipFile.addStream(new ByteArrayInputStream(data), parameters);
|
||||||
return dexJar().loadClass(name);
|
return dexJar(classFile).loadClass(name);
|
||||||
} catch (IOException | ZipException | ClassNotFoundException e) {
|
} catch (IOException | ZipException | ClassNotFoundException e) {
|
||||||
throw new FatalLoadingException(e);
|
throw new FatalLoadingException(e);
|
||||||
} finally {
|
} finally {
|
||||||
dexFile.delete();
|
if (classFile != null) {
|
||||||
odexOatFile.delete();
|
classFile.delete();
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private File createTempClassFile(String name) throws IOException {
|
||||||
|
File file = new File(mCacheDir, name.hashCode() + System.currentTimeMillis() + ".jar");
|
||||||
|
if (!file.exists()) {
|
||||||
|
file.createNewFile();
|
||||||
|
}
|
||||||
|
return file;
|
||||||
|
}
|
||||||
|
|
||||||
public void loadJar(File jar) throws IOException {
|
public void loadJar(File jar) throws IOException {
|
||||||
|
Log.d(LOG_TAG, "loadJar: jar = " + jar);
|
||||||
try {
|
try {
|
||||||
|
final File classFile = new File(mCacheDir, jar.getPath().hashCode() + System.currentTimeMillis() + ".jar");
|
||||||
final ZipFile zipFile = new ZipFile(classFile);
|
final ZipFile zipFile = new ZipFile(classFile);
|
||||||
final ZipFile jarFile = new ZipFile(jar);
|
final ZipFile jarFile = new ZipFile(jar);
|
||||||
//noinspection unchecked
|
//noinspection unchecked
|
||||||
@@ -80,31 +91,32 @@ public class AndroidClassLoader extends ClassLoader implements GeneratedClassLoa
|
|||||||
zipFile.addStream(jarFile.getInputStream(header), parameters);
|
zipFile.addStream(jarFile.getInputStream(header), parameters);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
dexJar();
|
dexJar(classFile);
|
||||||
} catch (ZipException e) {
|
} catch (ZipException e) {
|
||||||
IOException exception = new IOException();
|
throw new IOException(e);
|
||||||
//noinspection UnnecessaryInitCause
|
|
||||||
exception.initCause(e);
|
|
||||||
throw exception;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public DexClassLoader loadDex(File file) throws IOException {
|
public DexClassLoader loadDex(File file) throws FileNotFoundException {
|
||||||
|
Log.d(LOG_TAG, "loadDex: file = " + file);
|
||||||
|
if (!file.exists()) {
|
||||||
|
throw new FileNotFoundException(file.getPath());
|
||||||
|
}
|
||||||
DexClassLoader loader = new DexClassLoader(file.getPath(), mCacheDir.getPath(), null, parent);
|
DexClassLoader loader = new DexClassLoader(file.getPath(), mCacheDir.getPath(), null, parent);
|
||||||
mDexClassLoaders.add(loader);
|
mDexClassLoaders.add(loader);
|
||||||
return loader;
|
return loader;
|
||||||
}
|
}
|
||||||
|
|
||||||
private DexClassLoader dexJar() throws IOException {
|
private DexClassLoader dexJar(File classFile) throws IOException {
|
||||||
if (!classFile.exists()) {
|
|
||||||
classFile.createNewFile();
|
|
||||||
}
|
|
||||||
final Main.Arguments arguments = new Main.Arguments();
|
final Main.Arguments arguments = new Main.Arguments();
|
||||||
arguments.fileNames = new String[]{classFile.getPath()};
|
arguments.fileNames = new String[]{classFile.getPath()};
|
||||||
|
File dexFile = createTempClassFile("dex-" + classFile.getPath());
|
||||||
arguments.outName = dexFile.getPath();
|
arguments.outName = dexFile.getPath();
|
||||||
arguments.jarOutput = true;
|
arguments.jarOutput = true;
|
||||||
Main.run(arguments);
|
Main.run(arguments);
|
||||||
return loadDex(dexFile);
|
DexClassLoader loader = loadDex(dexFile);
|
||||||
|
dexFile.delete();
|
||||||
|
return loader;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
|||||||
Reference in New Issue
Block a user