/*
* Copyright 2000-2004 The Apache Software Foundation.
*
* 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 org.apache.jmeter;
import java.awt.event.ActionEvent;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.net.Authenticator;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Locale;
import java.util.Properties;
import java.util.StringTokenizer;
import org.apache.commons.cli.avalon.CLArgsParser;
import org.apache.commons.cli.avalon.CLOption;
import org.apache.commons.cli.avalon.CLOptionDescriptor;
import org.apache.commons.cli.avalon.CLUtil;
import org.apache.jmeter.config.gui.AbstractConfigGui;
import org.apache.jmeter.control.gui.AbstractControllerGui;
import org.apache.jmeter.control.gui.TestPlanGui;
import org.apache.jmeter.control.gui.WorkBenchGui;
import org.apache.jmeter.engine.ClientJMeterEngine;
import org.apache.jmeter.engine.JMeterEngine;
import org.apache.jmeter.engine.RemoteJMeterEngineImpl;
import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jmeter.engine.event.LoopIterationEvent;
import org.apache.jmeter.exceptions.IllegalUserActionException;
import org.apache.jmeter.gui.GuiPackage;
import org.apache.jmeter.gui.action.ActionRouter;
import org.apache.jmeter.gui.action.CheckDirty;
import org.apache.jmeter.gui.action.Load;
import org.apache.jmeter.gui.tree.JMeterTreeListener;
import org.apache.jmeter.gui.tree.JMeterTreeModel;
import org.apache.jmeter.plugin.JMeterPlugin;
import org.apache.jmeter.plugin.PluginManager;
// import org.apache.jmeter.processor.gui.AbstractPostProcessorGui;
// import org.apache.jmeter.processor.gui.AbstractPreProcessorGui;
import org.apache.jmeter.reporters.ResultCollector;
import org.apache.jmeter.reporters.Summariser;
import org.apache.jmeter.samplers.Remoteable;
import org.apache.jmeter.samplers.gui.AbstractSamplerGui;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.services.FileServer;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.testelement.TestListener;
import org.apache.jmeter.testelement.TestPlan;
import org.apache.jmeter.threads.gui.ThreadGroupGui;
import org.apache.jmeter.timers.gui.AbstractTimerGui;
import org.apache.jmeter.util.BeanShellServer;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jmeter.visualizers.gui.AbstractVisualizer;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.gui.ComponentUtil;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.jorphan.util.JOrphanUtils;
import org.apache.log.Logger;
/**
* @author mstover
*/
public class JMeter implements JMeterPlugin {
transient private static Logger log = LoggingManager.getLoggerForClass();
private static final int PROPFILE_OPT = 'p';
private static final int PROPFILE2_OPT = 'q'; // Bug 33920 - additional
// prop files
private static final int TESTFILE_OPT = 't';
private static final int LOGFILE_OPT = 'l';
private static final int NONGUI_OPT = 'n';
private static final int HELP_OPT = 'h';
private static final int VERSION_OPT = 'v';
private static final int SERVER_OPT = 's';
private static final int PROXY_HOST = 'H';
private static final int PROXY_PORT = 'P';
private static final int PROXY_USERNAME = 'u';
private static final int PROXY_PASSWORD = 'a';
private static final int JMETER_PROPERTY = 'J';
private static final int SYSTEM_PROPERTY = 'D';
private static final int LOGLEVEL = 'L';
private static final int REMOTE_OPT = 'r';
private static final int JMETER_HOME_OPT = 'd';
/**
* Define the understood options. Each CLOptionDescriptor contains:
* <ul>
* <li>The "long" version of the option. Eg, "help" means that "--help"
* will be recognised.</li>
* <li>The option flags, governing the option's argument(s).</li>
* <li>The "short" version of the option. Eg, 'h' means that "-h" will be
* recognised.</li>
* <li>A description of the option.</li>
* </ul>
*/
private static final CLOptionDescriptor[] options = new CLOptionDescriptor[] {
new CLOptionDescriptor("help", CLOptionDescriptor.ARGUMENT_DISALLOWED, HELP_OPT,
"print usage information and exit"),
new CLOptionDescriptor("version", CLOptionDescriptor.ARGUMENT_DISALLOWED, VERSION_OPT,
"print the version information and exit"),
new CLOptionDescriptor("propfile", CLOptionDescriptor.ARGUMENT_REQUIRED, PROPFILE_OPT,
"the jmeter property file to use"),
new CLOptionDescriptor("addprop", CLOptionDescriptor.ARGUMENT_REQUIRED
| CLOptionDescriptor.DUPLICATES_ALLOWED, // Bug 33920 -
// allow
// multiple
// props
PROPFILE2_OPT, "additional property file(s)"),
new CLOptionDescriptor("testfile", CLOptionDescriptor.ARGUMENT_REQUIRED, TESTFILE_OPT,
"the jmeter test(.jmx) file to run"),
new CLOptionDescriptor("logfile", CLOptionDescriptor.ARGUMENT_REQUIRED, LOGFILE_OPT,
"the file to log samples to"),
new CLOptionDescriptor("nongui", CLOptionDescriptor.ARGUMENT_DISALLOWED, NONGUI_OPT,
"run JMeter in nongui mode"),
new CLOptionDescriptor("server", CLOptionDescriptor.ARGUMENT_DISALLOWED, SERVER_OPT,
"run the JMeter server"),
new CLOptionDescriptor("proxyHost", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_HOST,
"Set a proxy server for JMeter to use"),
new CLOptionDescriptor("proxyPort", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_PORT,
"Set proxy server port for JMeter to use"),
new CLOptionDescriptor("username", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_USERNAME,
"Set username for proxy server that JMeter is to use"),
new CLOptionDescriptor("password", CLOptionDescriptor.ARGUMENT_REQUIRED, PROXY_PASSWORD,
"Set password for proxy server that JMeter is to use"),
new CLOptionDescriptor("jmeterproperty", CLOptionDescriptor.DUPLICATES_ALLOWED
| CLOptionDescriptor.ARGUMENTS_REQUIRED_2, JMETER_PROPERTY, "Define additional JMeter properties"),
new CLOptionDescriptor("systemproperty", CLOptionDescriptor.DUPLICATES_ALLOWED
| CLOptionDescriptor.ARGUMENTS_REQUIRED_2, SYSTEM_PROPERTY, "Define additional JMeter properties"),
new CLOptionDescriptor("loglevel", CLOptionDescriptor.DUPLICATES_ALLOWED
| CLOptionDescriptor.ARGUMENTS_REQUIRED_2, LOGLEVEL,
"Define loglevel: [category=]level e.g. jorphan=INFO or " + "jmeter.util=DEBUG"),
new CLOptionDescriptor("runremote", CLOptionDescriptor.ARGUMENT_DISALLOWED, REMOTE_OPT,
"Start remote servers from non-gui mode"),
new CLOptionDescriptor("homedir", CLOptionDescriptor.ARGUMENT_REQUIRED, JMETER_HOME_OPT,
"the jmeter home directory to use"), };
public JMeter() {
}
// Hack to allow automated tests to find when test has ended
transient boolean testEnded = false;
private JMeter parent;
/**
* Starts up JMeter in GUI mode
*/
public void startGui(CLOption testFile) {
PluginManager.install(this, true);
JMeterTreeModel treeModel = new JMeterTreeModel();
JMeterTreeListener treeLis = new JMeterTreeListener(treeModel);
treeLis.setActionHandler(ActionRouter.getInstance());
// NOTUSED: GuiPackage guiPack =
GuiPackage.getInstance(treeLis, treeModel);
org.apache.jmeter.gui.MainFrame main = new org.apache.jmeter.gui.MainFrame(ActionRouter.getInstance(),
treeModel, treeLis);
main.setTitle("Apache JMeter");
main.setIconImage(JMeterUtils.getImage("jmeter.jpg").getImage());
ComponentUtil.centerComponentInWindow(main, 80);
main.show();
ActionRouter.getInstance().actionPerformed(new ActionEvent(main, 1, CheckDirty.ADD_ALL));
if (testFile != null) {
try {
File f = new File(testFile.getArgument());
log.info("Loading file: " + f);
FileInputStream reader = new FileInputStream(f);
HashTree tree = SaveService.loadTree(reader);
GuiPackage.getInstance().setTestPlanFile(f.getAbsolutePath());
new Load().insertLoadedTree(1, tree);
} catch (Exception e) {
log.error("Failure loading test file", e);
JMeterUtils.reportErrorToUser(e.toString());
}
}
}
/**
* Takes the command line arguments and uses them to determine how to
* startup JMeter.
*/
public void start(String[] args) {
CLArgsParser parser = new CLArgsParser(args, options);
if (null != parser.getErrorString()) {
System.err.println("Error: " + parser.getErrorString());
System.out.println("Usage");
System.out.println(CLUtil.describeOptions(options).toString());
return;
}
try {
initializeProperties(parser);
setProxy(parser);
log.info("Version " + JMeterUtils.getJMeterVersion());
log.info("java.version=" + System.getProperty("java.version"));
log.info("Locale=" + Locale.getDefault().getDisplayName());
log.info(JMeterUtils.getJMeterCopyright());
updateClassLoader();
if (parser.getArgumentById(VERSION_OPT) != null) {
System.out.println(JMeterUtils.getJMeterCopyright());
System.out.println("Version " + JMeterUtils.getJMeterVersion());
} else if (parser.getArgumentById(HELP_OPT) != null) {
System.out.println(JMeterUtils.getResourceFileAsText("org/apache/jmeter/help.txt"));
} else if (parser.getArgumentById(SERVER_OPT) != null) {
startServer(JMeterUtils.getPropDefault("server_port", 0));
startBSH();
} else if (parser.getArgumentById(NONGUI_OPT) == null) {
startGui(parser.getArgumentById(TESTFILE_OPT));
startBSH();
} else {
startNonGui(parser.getArgumentById(TESTFILE_OPT), parser.getArgumentById(LOGFILE_OPT), parser
.getArgumentById(REMOTE_OPT));
startBSH();
}
} catch (IllegalUserActionException e) {
System.out.println(e.getMessage());
System.out.println("Incorrect Usage");
System.out.println(CLUtil.describeOptions(options).toString());
} catch (Exception e) {
e.printStackTrace();
System.out.println("An error occurred: " + e.getMessage());
System.exit(-1);
}
}
// Update classloader if necessary
private void updateClassLoader() {
String userpath= JMeterUtils.getPropDefault("user.classpath","");
if (userpath.length()> 0){
log.info("user.classpath="+userpath);
StringTokenizer tok = new StringTokenizer(userpath, File.pathSeparator);
while(tok.hasMoreTokens()) {
String path=tok.nextToken();
File f=new File(path);
if (!f.canRead() && !f.isDirectory()) {
log.warn("Can't read "+path);
} else {
URL url;
try {
url = new URL("file","",path);
NewDriver.addURL(url);
} catch (MalformedURLException e) {
log.warn("Can't create URL for "+path+" "+e);
}
}
}
}
}
/**
*
*/
private void startBSH() {
int bshport = JMeterUtils.getPropDefault("beanshell.server.port", 0);
String bshfile = JMeterUtils.getPropDefault("beanshell.server.file", "");
if (bshport > 0) {
log.info("Starting Beanshell server (" + bshport + "," + bshfile + ")");
Runnable t = new BeanShellServer(bshport, bshfile);
t.run();
}
}
/**
* Sets a proxy server for the JVM if the command line arguments are
* specified.
*/
private void setProxy(CLArgsParser parser) throws IllegalUserActionException {
if (parser.getArgumentById(PROXY_USERNAME) != null) {
if (parser.getArgumentById(PROXY_PASSWORD) != null) {
String u, p;
Authenticator.setDefault(new ProxyAuthenticator(u = parser.getArgumentById(PROXY_USERNAME)
.getArgument(), p = parser.getArgumentById(PROXY_PASSWORD).getArgument()));
log.info("Set Proxy login: " + u + "/" + p);
} else {
String u;
Authenticator.setDefault(new ProxyAuthenticator(u = parser.getArgumentById(PROXY_USERNAME)
.getArgument(), ""));
log.info("Set Proxy login: " + u);
}
}
if (parser.getArgumentById(PROXY_HOST) != null && parser.getArgumentById(PROXY_PORT) != null) {
String h, p;
System.setProperty("http.proxyHost", h = parser.getArgumentById(PROXY_HOST).getArgument());
System.setProperty("https.proxyHost", parser.getArgumentById(PROXY_HOST).getArgument());
System.setProperty("http.proxyPort", p = parser.getArgumentById(PROXY_PORT).getArgument());
System.setProperty("https.proxyPort", parser.getArgumentById(PROXY_PORT).getArgument());
log.info("Set http[s].proxyHost: " + h + " Port: " + p);
} else if (parser.getArgumentById(PROXY_HOST) != null || parser.getArgumentById(PROXY_PORT) != null) {
throw new IllegalUserActionException(JMeterUtils.getResString("proxy_cl_error"));
}
}
private void initializeProperties(CLArgsParser parser) {
if (parser.getArgumentById(PROPFILE_OPT) != null) {
JMeterUtils.getProperties(parser.getArgumentById(PROPFILE_OPT).getArgument());
} else {
JMeterUtils.getProperties(NewDriver.getJMeterDir() + File.separator + "bin" + File.separator
+ "jmeter.properties");
}
// Bug 33845 - allow direct override of Home dir
if (parser.getArgumentById(JMETER_HOME_OPT) == null) {
JMeterUtils.setJMeterHome(NewDriver.getJMeterDir());
} else {
JMeterUtils.setJMeterHome(parser.getArgumentById(JMETER_HOME_OPT).getArgument());
}
// Process command line property definitions (can occur multiple times)
Properties jmeterProps = JMeterUtils.getJMeterProperties();
List clOptions = parser.getArguments();
int size = clOptions.size();
for (int i = 0; i < size; i++) {
CLOption option = (CLOption) clOptions.get(i);
String name = option.getArgument(0);
String value = option.getArgument(1);
switch (option.getDescriptor().getId()) {
case PROPFILE2_OPT: // Bug 33920 - allow multiple props
File f = new File(name);
try {
jmeterProps.load(new FileInputStream(f));
} catch (FileNotFoundException e) {
log.warn("Can't find additional property file: " + name, e);
} catch (IOException e) {
log.warn("Error loading additional property file: " + name, e);
}
break;
case SYSTEM_PROPERTY:
if (value.length() > 0) { // Set it
log.info("Setting System property: " + name + "=" + value);
System.getProperties().setProperty(name, value);
} else { // Reset it
log.warn("Removing System property: " + name);
System.getProperties().remove(name);
}
break;
case JMETER_PROPERTY:
if (value.length() > 0) { // Set it
log.info("Setting JMeter property: " + name + "=" + value);
jmeterProps.setProperty(name, value);
} else { // Reset it
log.warn("Removing JMeter property: " + name);
jmeterProps.remove(name);
}
break;
case LOGLEVEL:
if (value.length() > 0) { // Set category
log.info("LogLevel: " + name + "=" + value);
LoggingManager.setPriority(value, name);
} else { // Set root level
log.warn("LogLevel: " + name);
LoggingManager.setPriority(name);
}
break;
}
}
}
public void startServer() {
startServer(0);
}
public void startServer(int port) {
try {
new RemoteJMeterEngineImpl(port);
while (true) {
Thread.sleep(Long.MAX_VALUE);
}
} catch (Exception ex) {
log.error("Giving up, as server failed with:", ex);
System.exit(0);// Give up
}
}
public void startNonGui(CLOption testFile, CLOption logFile, CLOption remoteStart)
throws IllegalUserActionException {
// add a system property so samplers can check to see if JMeter
// is running in NonGui mode
System.setProperty("JMeter.NonGui", "true");
JMeter driver = new JMeter();
driver.parent = this;
PluginManager.install(this, false);
if (testFile == null) {
throw new IllegalUserActionException();
}
if (logFile == null) {
driver.run(testFile.getArgument(), null, remoteStart != null);
} else {
driver.run(testFile.getArgument(), logFile.getArgument(), remoteStart != null);
}
}
private void run(String testFile, String logFile, boolean remoteStart) { // run
// test
// in
// batch
// mode
FileInputStream reader = null;
try {
File f = new File(testFile);
if (!f.exists() || !f.isFile()) {
println("Could not open " + testFile);
return;
}
FileServer.getFileServer().setBasedir(f.getAbsolutePath());
reader = new FileInputStream(f);
log.info("Loading file: " + f);
HashTree tree = SaveService.loadTree(reader);
// Remove the disabled items
// For GUI runs this is done in Start.java
convertSubTree(tree);
if (logFile != null) {
ResultCollector logger = new ResultCollector();
logger.setFilename(logFile);
tree.add(tree.getArray()[0], logger);
}
String summariserName = JMeterUtils.getPropDefault("summariser.name", "");//$NON-NLS-1$
if (summariserName.length() > 0) {
log.info("Creating summariser <" + summariserName + ">");
println("Creating summariser <" + summariserName + ">");
Summariser summer = new Summariser(summariserName);
tree.add(tree.getArray()[0], summer);
}
tree.add(tree.getArray()[0], new ListenToTest(parent));
println("Created the tree successfully");
JMeterEngine engine = null;
if (!remoteStart) {
engine = new StandardJMeterEngine();
engine.configure(tree);
println("Starting the test");
engine.runTest();
} else {
String remote_hosts_string = JMeterUtils.getPropDefault("remote_hosts", "127.0.0.1");
java.util.StringTokenizer st = new java.util.StringTokenizer(remote_hosts_string, ",");
List engines = new LinkedList();
while (st.hasMoreElements()) {
String el = (String) st.nextElement();
println("Configuring remote engine for " + el);
engines.add(doRemoteInit(el.trim(), tree));
}
println("Starting remote engines");
Iterator iter = engines.iterator();
while (iter.hasNext()) {
engine = (JMeterEngine) iter.next();
engine.runTest();
}
println("Remote engines have been started");
}
} catch (Exception e) {
System.out.println("Error in NonGUIDriver " + e.toString());
log.error("", e);
}
}
/**
* Code copied from AbstractAction.java and modified to suit TestElements
*
* @param tree
*/
private void convertSubTree(HashTree tree) {// TODO check build dependencies
Iterator iter = new LinkedList(tree.list()).iterator();
while (iter.hasNext()) {
TestElement item = (TestElement) iter.next();
if (item.isEnabled()) {
// This is done for GUI runs in JMeterTreeModel.addSubTree()
if (item instanceof TestPlan) {
TestPlan tp = (TestPlan) item;
tp.setFunctionalMode(tp.isFunctionalMode());
tp.setSerialized(tp.isSerialized());
}
// TODO handle ReplaceableControllers
// if (item instanceof ReplaceableController)
// {
// System.out.println("Replaceable "+item.getClass().getName());
// HashTree subTree = tree.getTree(item);
//
// if (subTree != null)
// {
// ReplaceableController rc =
// (ReplaceableController) item;//.createTestElement();
// rc.replace(subTree);
// convertSubTree(subTree);
// tree.replace(item, rc.getReplacement());
// }
// }
// else
{
// System.out.println("NonReplaceable
// "+item.getClass().getName());
convertSubTree(tree.getTree(item));
// TestElement testElement = item.createTestElement();
// tree.replace(item, testElement);
}
} else {
// System.out.println("Disabled "+item.getClass().getName());
tree.remove(item);
}
}
}
private JMeterEngine doRemoteInit(String hostName, HashTree testTree) {
JMeterEngine engine = null;
try {
engine = new ClientJMeterEngine(hostName);
} catch (Exception e) {
log.fatalError("Failure connecting to remote host", e);
System.exit(0);
}
engine.configure(testTree);
return engine;
}
/**
* Listen to test and exit program after test completes, after a 5 second
* delay to give listeners a chance to close out their files.
*/
private class ListenToTest implements TestListener, Runnable, Remoteable {
int started = 0;
private JMeter _parent;
private ListenToTest(JMeter parent) {
_parent = parent;
}
public synchronized void testEnded(String host) {
started--;
log.info("Remote host " + host + " finished");
if (started == 0) {
testEnded();
}
}
public void testEnded() {
Thread stopSoon = new Thread(this);
stopSoon.start();
}
public synchronized void testStarted(String host) {
started++;
log.info("Started remote host: " + host);
}
public void testStarted() {
log.info(JMeterUtils.getResString("running_test"));
}
/**
* This is a hack to allow listeners a chance to close their files. Must
* implement a queue for sample responses tied to the engine, and the
* engine won't deliver testEnded signal till all sample responses have
* been delivered. Should also improve performance of remote JMeter
* testing.
*/
public void run() {
println("Tidying up ...");
try {
Thread.sleep(5000);
} catch (InterruptedException e) {
// ignored
}
println("... end of run");
_parent.testEnded = true;
}
/**
* @see TestListener#testIterationStart(LoopIterationEvent)
*/
public void testIterationStart(LoopIterationEvent event) {
// ignored
}
}
private static void println(String str) {
System.out.println(str);
}
private static final String[][] DEFAULT_ICONS = {
{ TestPlanGui.class.getName(), "org/apache/jmeter/images/beaker.gif" },
{ AbstractTimerGui.class.getName(), "org/apache/jmeter/images/timer.gif" },
{ ThreadGroupGui.class.getName(), "org/apache/jmeter/images/thread.gif" },
{ AbstractVisualizer.class.getName(), "org/apache/jmeter/images/meter.png" },
{ AbstractConfigGui.class.getName(), "org/apache/jmeter/images/testtubes.png" },
// Note: these were the original settings (just moved to a static
// array)
// Commented out because there is no such file
// {
// AbstractPreProcessorGui.class.getName(),
// "org/apache/jmeter/images/testtubes.gif" },
// {
// AbstractPostProcessorGui.class.getName(),
// "org/apache/jmeter/images/testtubes.gif" },
{ AbstractControllerGui.class.getName(), "org/apache/jmeter/images/knob.gif" },
{ WorkBenchGui.class.getName(), "org/apache/jmeter/images/clipboard.gif" },
{ AbstractSamplerGui.class.getName(), "org/apache/jmeter/images/pipet.png" }
// AbstractAssertionGUI not defined
};
public String[][] getIconMappings() {
String iconProp = JMeterUtils.getPropDefault("jmeter.icons", "org/apache/jmeter/images/icon.properties");
Properties p = JMeterUtils.loadProperties(iconProp);
if (p == null) {
log.info(iconProp + " not found - using default icon set");
return DEFAULT_ICONS;
}
log.info("Loaded icon properties from " + iconProp);
String[][] iconlist = new String[p.size()][3];
Enumeration pe = p.keys();
int i = 0;
while (pe.hasMoreElements()) {
String key = (String) pe.nextElement();
String icons[] = JOrphanUtils.split(p.getProperty(key), " ");
iconlist[i][0] = key;
iconlist[i][1] = icons[0];
if (icons.length > 1)
iconlist[i][2] = icons[1];
i++;
}
return iconlist;
}
public String[][] getResourceBundles() {
return new String[0][];
}
}