| /* |
| * Copyright (C) 2010 The Android Open Source Project |
| * |
| * Licensed 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 com.android.monkeyrunner; |
| |
| import com.google.common.base.Predicate; |
| import com.google.common.base.Predicates; |
| import com.google.common.collect.ImmutableMap; |
| |
| import com.android.chimpchat.ChimpChat; |
| |
| import org.python.util.PythonInterpreter; |
| |
| import java.io.File; |
| import java.io.IOException; |
| import java.net.MalformedURLException; |
| import java.net.URL; |
| import java.net.URLClassLoader; |
| import java.util.Enumeration; |
| import java.util.Map; |
| import java.util.TreeMap; |
| import java.util.jar.Attributes; |
| import java.util.jar.JarFile; |
| import java.util.jar.Manifest; |
| import java.util.logging.Formatter; |
| import java.util.logging.Handler; |
| import java.util.logging.Level; |
| import java.util.logging.LogManager; |
| import java.util.logging.Logger; |
| |
| /** |
| * MonkeyRunner is a host side application to control a monkey instance on a |
| * device. MonkeyRunner provides some useful helper functions to control the |
| * device as well as various other methods to help script tests. This class bootstraps |
| * MonkeyRunner. |
| */ |
| public class MonkeyRunnerStarter { |
| private static final Logger LOG = Logger.getLogger(MonkeyRunnerStarter.class.getName()); |
| private static final String MONKEY_RUNNER_MAIN_MANIFEST_NAME = "MonkeyRunnerStartupRunner"; |
| |
| private final ChimpChat chimp; |
| private final MonkeyRunnerOptions options; |
| |
| public MonkeyRunnerStarter(MonkeyRunnerOptions options) { |
| Map<String, String> chimp_options = new TreeMap<String, String>(); |
| chimp_options.put("backend", options.getBackendName()); |
| this.options = options; |
| this.chimp = ChimpChat.getInstance(chimp_options); |
| MonkeyRunner.setChimpChat(chimp); |
| } |
| |
| |
| |
| private int run() { |
| // This system property gets set by the included starter script |
| String monkeyRunnerPath = System.getProperty("com.android.monkeyrunner.bindir") + |
| File.separator + "monkeyrunner"; |
| |
| Map<String, Predicate<PythonInterpreter>> plugins = handlePlugins(); |
| if (options.getScriptFile() == null) { |
| ScriptRunner.console(monkeyRunnerPath); |
| chimp.shutdown(); |
| return 0; |
| } else { |
| int error = ScriptRunner.run(monkeyRunnerPath, options.getScriptFile().getAbsolutePath(), |
| options.getArguments(), plugins); |
| chimp.shutdown(); |
| return error; |
| } |
| } |
| |
| private Predicate<PythonInterpreter> handlePlugin(File f) { |
| JarFile jarFile; |
| try { |
| jarFile = new JarFile(f); |
| } catch (IOException e) { |
| LOG.log(Level.SEVERE, "Unable to open plugin file. Is it a jar file? " + |
| f.getAbsolutePath(), e); |
| return Predicates.alwaysFalse(); |
| } |
| Manifest manifest; |
| try { |
| manifest = jarFile.getManifest(); |
| } catch (IOException e) { |
| LOG.log(Level.SEVERE, "Unable to get manifest file from jar: " + |
| f.getAbsolutePath(), e); |
| return Predicates.alwaysFalse(); |
| } |
| Attributes mainAttributes = manifest.getMainAttributes(); |
| String pluginClass = mainAttributes.getValue(MONKEY_RUNNER_MAIN_MANIFEST_NAME); |
| if (pluginClass == null) { |
| // No main in this plugin, so it always succeeds. |
| return Predicates.alwaysTrue(); |
| } |
| URL url; |
| try { |
| url = f.toURI().toURL(); |
| } catch (MalformedURLException e) { |
| LOG.log(Level.SEVERE, "Unable to convert file to url " + f.getAbsolutePath(), |
| e); |
| return Predicates.alwaysFalse(); |
| } |
| URLClassLoader classLoader = new URLClassLoader(new URL[] { url }, |
| ClassLoader.getSystemClassLoader()); |
| Class<?> clz; |
| try { |
| clz = Class.forName(pluginClass, true, classLoader); |
| } catch (ClassNotFoundException e) { |
| LOG.log(Level.SEVERE, "Unable to load the specified plugin: " + pluginClass, e); |
| return Predicates.alwaysFalse(); |
| } |
| Object loadedObject; |
| try { |
| loadedObject = clz.newInstance(); |
| } catch (InstantiationException e) { |
| LOG.log(Level.SEVERE, "Unable to load the specified plugin: " + pluginClass, e); |
| return Predicates.alwaysFalse(); |
| } catch (IllegalAccessException e) { |
| LOG.log(Level.SEVERE, "Unable to load the specified plugin " + |
| "(did you make it public?): " + pluginClass, e); |
| return Predicates.alwaysFalse(); |
| } |
| // Cast it to the right type |
| if (loadedObject instanceof Runnable) { |
| final Runnable run = (Runnable) loadedObject; |
| return new Predicate<PythonInterpreter>() { |
| public boolean apply(PythonInterpreter i) { |
| run.run(); |
| return true; |
| } |
| }; |
| } else if (loadedObject instanceof Predicate<?>) { |
| return (Predicate<PythonInterpreter>) loadedObject; |
| } else { |
| LOG.severe("Unable to coerce object into correct type: " + pluginClass); |
| return Predicates.alwaysFalse(); |
| } |
| } |
| |
| private Map<String, Predicate<PythonInterpreter>> handlePlugins() { |
| ImmutableMap.Builder<String, Predicate<PythonInterpreter>> builder = ImmutableMap.builder(); |
| for (File f : options.getPlugins()) { |
| builder.put(f.getAbsolutePath(), handlePlugin(f)); |
| } |
| return builder.build(); |
| } |
| |
| /* Similar to above, when this fails, it no longer throws a |
| * runtime exception, but merely will log the failure. |
| */ |
| |
| |
| private static final void replaceAllLogFormatters(Formatter form, Level level) { |
| LogManager mgr = LogManager.getLogManager(); |
| Enumeration<String> loggerNames = mgr.getLoggerNames(); |
| while (loggerNames.hasMoreElements()) { |
| String loggerName = loggerNames.nextElement(); |
| Logger logger = mgr.getLogger(loggerName); |
| for (Handler handler : logger.getHandlers()) { |
| handler.setFormatter(form); |
| handler.setLevel(level); |
| } |
| } |
| } |
| |
| public static void main(String[] args) { |
| MonkeyRunnerOptions options = MonkeyRunnerOptions.processOptions(args); |
| |
| if (options == null) { |
| return; |
| } |
| |
| // logging property files are difficult |
| replaceAllLogFormatters(MonkeyFormatter.DEFAULT_INSTANCE, options.getLogLevel()); |
| |
| MonkeyRunnerStarter runner = new MonkeyRunnerStarter(options); |
| int error = runner.run(); |
| |
| // This will kill any background threads as well. |
| System.exit(error); |
| } |
| } |