- HttpSampler is now capable of following several redirects in a row, while still ensuring we won't enter into an infinite loop ( by limiting the maximum number of redirects followed to 10).
- HttpSampler can now follow redirects which switch protocols (http to https or the reverse), change ports, or include parameters in the query string.
- SoapSampler now works.
Also removed a couple of comments (from myself) which no longer make sense.
I'd like one of the developers to consider this for inclusion into the product. Feel free to write back if you have questions / concerns.
There's one important related bug remaining: int Soap Sampler you can't enter a URL containing ${variables}. The solution is long -- requires splitting the URL field into its component fields (protocol, domain, port and path), just as done for the HTTP Sampler -- and unfortunately I don't have that much time now. Sorry.
Salut,
Jordi.
/* * ==================================================================== * The Apache Software License, Version 1.1 * * Copyright (c) 2001 The Apache Software Foundation. All rights * reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in * the documentation and/or other materials provided with the * distribution. * * 3. The end-user documentation included with the redistribution, * if any, must include the following acknowledgment: * "This product includes software developed by the * Apache Software Foundation (http://www.apache.org/)." * Alternately, this acknowledgment may appear in the software itself, * if and wherever such third-party acknowledgments normally appear. * * 4. The names "Apache" and "Apache Software Foundation" and * "Apache JMeter" must not be used to endorse or promote products * derived from this software without prior written permission. For * written permission, please contact [EMAIL PROTECTED] * * 5. Products derived from this software may not be called "Apache", * "Apache JMeter", nor may "Apache" appear in their name, without * prior written permission of the Apache Software Foundation. * * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. * ==================================================================== * * This software consists of voluntary contributions made by many * individuals on behalf of the Apache Software Foundation. For more * information on the Apache Software Foundation, please see * <http://www.apache.org/>. */ package org.apache.jmeter.protocol.http.sampler; import java.io.BufferedInputStream; import java.io.ByteArrayOutputStream; import java.io.FileNotFoundException; import java.io.IOException; import java.net.BindException; import java.net.HttpURLConnection; import java.net.MalformedURLException; import java.net.URL; import java.net.URLConnection; import java.util.ArrayList; import java.util.Collection; import java.util.Iterator;
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.config.ConfigTestElement;
import org.apache.jmeter.protocol.http.control.AuthManager;
import org.apache.jmeter.protocol.http.control.CookieManager;
import org.apache.jmeter.protocol.http.control.Header;
import org.apache.jmeter.protocol.http.control.HeaderManager;
import org.apache.jmeter.protocol.http.util.HTTPArgument;
import org.apache.jmeter.samplers.AbstractSampler;
import org.apache.jmeter.samplers.Entry;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jmeter.util.SSLManager;
import org.apache.log.Hierarchy;
import org.apache.log.Logger;
/****************************************
* A sampler which understands all the parts necessary to read statistics about
* HTTP requests, including cookies and authentication.
*
*@author Michael Stover
*@created $Date: 2002/10/21 17:02:29 $
*@version $Revision: 1.14 $
***************************************/
public class HTTPSampler extends AbstractSampler
{
public final static String HEADERS = "headers";
public final static String HEADER = "header";
public final static String ARGUMENTS = "HTTPsampler.Arguments";
public final static String AUTH_MANAGER = "HTTPSampler.auth_manager";
public final static String COOKIE_MANAGER = "HTTPSampler.cookie_manager";
public final static String HEADER_MANAGER = "HTTPSampler.header_manager";
public final static String MIMETYPE = "HTTPSampler.mimetype";
public final static String DOMAIN = "HTTPSampler.domain";
public final static String PORT = "HTTPSampler.port";
public final static String METHOD = "HTTPSampler.method";
public final static String PATH = "HTTPSampler.path";
public final static String FOLLOW_REDIRECTS = "HTTPSampler.follow_redirects";
public final static String PROTOCOL = "HTTPSampler.protocol";
public final static String URL = "HTTPSampler.URL";
public final static String POST = "POST";
public final static String GET = "GET";
public final static String USE_KEEPALIVE = "HTTPSampler.use_keepalive";
public final static String FILE_NAME = "HTTPSampler.FILE_NAME";
public final static String FILE_FIELD = "HTTPSampler.FILE_FIELD";
public final static String FILE_DATA = "HTTPSampler.FILE_DATA";
public final static String FILE_MIMETYPE = "HTTPSampler.FILE_MIMETYPE";
public final static String CONTENT_TYPE = "HTTPSampler.CONTENT_TYPE";
public final static String NORMAL_FORM = "normal_form";
public final static String MULTIPART_FORM = "multipart_form";
private static final int MAX_REDIRECTS=10;
protected static String encoding = "iso-8859-1";
private static final PostWriter postWriter = new PostWriter();
transient protected HttpURLConnection conn;
private int connectionTries = 0;
public void setFileField(String value)
{
setProperty(FILE_FIELD, value);
}
public String getFileField()
{
return getPropertyAsString(FILE_FIELD);
}
public void setFilename(String value)
{
setProperty(FILE_NAME, value);
}
public String getFilename()
{
return getPropertyAsString(FILE_NAME);
}
public void setProtocol(String value)
{
setProperty(PROTOCOL, value);
}
public String getProtocol()
{
return getPropertyAsString(PROTOCOL);
}
/**
* Sets the Path attribute of the UrlConfig object
*
*@param path The new Path value
*/
public void setPath(String path)
{
if (GET.equals(getMethod()))
{
int index = path.indexOf("?");
if (index > -1)
{
setProperty(PATH, path.substring(0, index));
parseArguments(path.substring(index + 1));
}
else
{
setProperty(PATH, path);
}
}
else
{
setProperty(PATH, path);
}
}
public String getPath()
{
return getPropertyAsString(PATH);
}
public void setFollowRedirects(boolean value)
{
setProperty(FOLLOW_REDIRECTS, new Boolean(value));
}
public boolean getFollowRedirects()
{
return getPropertyAsBoolean(FOLLOW_REDIRECTS);
}
public void setMethod(String value)
{
setProperty(METHOD, value);
}
public String getMethod()
{
return getPropertyAsString(METHOD);
}
public void setUseKeepAlive(boolean value)
{
setProperty(USE_KEEPALIVE, new Boolean(value));
}
public boolean getUseKeepAlive()
{
return getPropertyAsBoolean(USE_KEEPALIVE);
}
public void addEncodedArgument(String name, String value, String metaData)
{
Arguments args = getArguments();
HTTPArgument arg = new HTTPArgument(name, value, metaData, true);
if (arg.getName().equals(arg.getEncodedName())
&& arg.getValue().equals(arg.getEncodedValue()))
{
arg.setAlwaysEncode(false);
}
args.addArgument(arg);
}
public void addArgument(String name, String value)
{
Arguments args = this.getArguments();
args.addArgument(new HTTPArgument(name, value));
}
public void addArgument(String name, String value, String metadata)
{
Arguments args = this.getArguments();
args.addArgument(new HTTPArgument(name, value, metadata));
}
public void setPort(int value)
{
setProperty(PORT, new Integer(value));
}
public int getPort()
{
int port = getPropertyAsInt(PORT);
if (port == 0)
{
if("https".equalsIgnoreCase(getProtocol()))
{
return 443;
}
return 80;
}
return port;
}
public void setDomain(String value)
{
setProperty(DOMAIN, value);
}
public String getDomain()
{
return (String) getProperty(DOMAIN);
}
public void setArguments(Arguments value)
{
setProperty(ARGUMENTS, value);
}
public Arguments getArguments()
{
return (Arguments) getProperty(ARGUMENTS);
}
public void setAuthManager(AuthManager value)
{
setProperty(AUTH_MANAGER, value);
}
public AuthManager getAuthManager()
{
return (AuthManager) getProperty(AUTH_MANAGER);
}
public void setHeaderManager(HeaderManager value)
{
setProperty(HEADER_MANAGER, value);
}
public HeaderManager getHeaderManager()
{
return (HeaderManager) getProperty(HEADER_MANAGER);
}
public void setCookieManager(CookieManager value)
{
setProperty(COOKIE_MANAGER, value);
}
public CookieManager getCookieManager()
{
return (CookieManager) getProperty(COOKIE_MANAGER);
}
public void setMimetype(String value)
{
setProperty(MIMETYPE, value);
}
public String getMimetype()
{
return (String) getProperty(MIMETYPE);
}
protected void addCustomTestElement(TestElement element)
{
if (element instanceof Arguments)
{
if (getProperty(ARGUMENTS) != null)
{
((Arguments) getProperty(ARGUMENTS)).addTestElement(element);
}
else
{
setProperty(ARGUMENTS, element);
}
}
else if (element instanceof AuthManager)
{
if (getProperty(AUTH_MANAGER) != null)
{
((TestElement) getProperty(AUTH_MANAGER)).addTestElement(element);
}
else
{
setProperty(AUTH_MANAGER, element);
}
}
else if (element instanceof CookieManager)
{
if (getProperty(COOKIE_MANAGER) != null)
{
((TestElement) getProperty(COOKIE_MANAGER)).addTestElement(element);
}
else
{
setProperty(COOKIE_MANAGER, element);
}
}
else if (element instanceof HeaderManager)
{
if (getProperty(HEADER_MANAGER) != null)
{
((TestElement) getProperty(HEADER_MANAGER)).addTestElement(element);
}
else
{
setProperty(HEADER_MANAGER, element);
}
}
else if (
element instanceof HTTPSampler || element instanceof ConfigTestElement)
{
this.mergeIn(element);
}
}
/****************************************
* !ToDo (Field description)
***************************************/
protected final static String NON_HTTP_RESPONSE_CODE =
"Non HTTP response code";
/****************************************
* !ToDo (Field description)
***************************************/
protected final static String NON_HTTP_RESPONSE_MESSAGE =
"Non HTTP response message";
transient private static Logger log =
Hierarchy.getDefaultHierarchy().getLoggerFor("jmeter.protocol.http");
/****************************************
* Holds a list of URLs sampled - so we're not flooding stdout with debug
* information
***************************************/
private ArrayList m_sampledURLs = new ArrayList();
/****************************************
* Constructor for the HTTPSampler object
***************************************/
public HTTPSampler()
{
setArguments(new Arguments());
}
public HTTPSampler(URL u)
{
setMethod(GET);
setDomain(u.getHost());
setPath(u.getPath());
setPort(u.getPort());
setProtocol(u.getProtocol());
parseArguments(u.getQuery());
setFollowRedirects(true);
setUseKeepAlive(true);
setArguments(new Arguments());
}
/****************************************
* Do a sampling and return its results.
*
*@param e <code>Entry</code> to be sampled
*@return results of the sampling
*@see org.apache.jmeter.protocol.http.sampler.HTTPSampler.sample(org.apache.jmeter.samplers.Entry,
* int)
***************************************/
public SampleResult sample(Entry e)
{
return sample(0);
}
public SampleResult sample()
{
return sample(0);
}
/**
* !ToDoo (Method description)
*
*@return !ToDo (Return description)
*@exception MalformedURLException !ToDo (Exception description)
*/
public URL getUrl() throws MalformedURLException
{
String pathAndQuery = null;
if (this.getMethod().equals(HTTPSampler.GET)
&& getQueryString().length() > 0)
{
if (this.getPath().indexOf("?") > -1)
{
pathAndQuery = this.getPath() + "&" + getQueryString();
}
else
{
pathAndQuery = this.getPath() + "?" + getQueryString();
}
}
else
{
pathAndQuery = this.getPath();
}
if (!pathAndQuery.startsWith("/"))
{
pathAndQuery = "/" + pathAndQuery;
}
if (getPort() == 0)
{
return new URL(getProtocol(), getDomain(), pathAndQuery);
}
else
{
return new URL(
getProtocol(),
(String) getProperty(HTTPSampler.DOMAIN),
getPort(),
pathAndQuery);
}
}
/**
* Gets the QueryString attribute of the UrlConfig object
*
*@return The QueryString value
*/
public String getQueryString()
{
StringBuffer buf = new StringBuffer();
Iterator iter = getArguments().iterator();
boolean first = true;
while (iter.hasNext())
{
HTTPArgument item = (HTTPArgument) iter.next();
if (!first)
{
buf.append("&");
}
else
{
first = false;
}
buf.append(item.getEncodedName());
if (item.getMetaData() == null)
{
buf.append("=");
}
else
{
buf.append(item.getMetaData());
}
buf.append(item.getEncodedValue());
}
return buf.toString();
}
/****************************************
* Set request headers in preparation to opening a connection
*
*@param connection <code>URLConnection</code> to set headers on
*@exception IOException if an I/O exception occurs
***************************************/
public void setPostHeaders(URLConnection conn) throws IOException
{
postWriter.setHeaders(conn, this);
}
/****************************************
* Send POST data from <code>Entry</code> to the open connection.
*
*@param connection <code>URLConnection</code> of where POST data should
* be sent
*@param url contains the query string for POST
*@exception IOException if an I/O exception occurs
***************************************/
public void sendPostData(URLConnection connection) throws IOException
{
postWriter.sendPostData(connection, this);
}
/****************************************
* Returns a <code>HttpURLConnection</code> with request method(GET or POST),
* headers, cookies, authorization properly set for the URL request
*
*@param u <code>URL</code> of the URL request
*@param url <code>UrlConfig</code> of the URL request
*@return <code>HttpURLConnection</code> of the URL request
*@exception IOException if an I/O Exception occurs
***************************************/
protected HttpURLConnection setupConnection(URL u, String method)
throws IOException
{
HttpURLConnection conn;
// [Jordi <[EMAIL PROTECTED]>]
// I've not been able to find out why we're not using this
// feature of HttpURLConnections and we're doing redirection
// by hand instead. Everything would be so much simpler...
// [/Jordi]
// Mike: answer - it didn't work. Maybe in JDK1.4 it works, but honestly,
// it doesn't seem like they're working on this.
// My longer term plan is to use Apache's home grown HTTP Client, or
// maybe even HTTPUnit's classes. I'm sure both would be better than Sun's
HttpURLConnection.setFollowRedirects(false);
conn = (HttpURLConnection) u.openConnection();
// delegate SSL specific stuff to SSLManager so that compilation still works otherwise.
if ("https".equals(u.getProtocol()))
{
try
{
SSLManager.getInstance().setContext(conn);
}
catch (Exception e)
{
log.warn(
"You may have forgotten to set the ssl.provider property in jmeter.properties",
e);
}
}
// a well-bahaved browser is supposed to send 'Connection: close'
// with the last request to an HTTP server. Instead, most browsers
// leave it to the server to close the connection after their
// timeout period. Leave it to the JMeter user to decide.
if (getUseKeepAlive())
{
conn.setRequestProperty("Connection", "keep-alive");
}
else
{
conn.setRequestProperty("Connection", "close");
}
conn.setRequestMethod(method);
setConnectionHeaders(conn, u, getHeaderManager());
setConnectionCookie(conn, u, getCookieManager());
setConnectionAuthorization(conn, u, getAuthManager());
return conn;
}
//Mark Walsh 2002-08-03, modified to also parse a parameter name value
//string, where string contains only the parameter name and no equal sign.
/**
* This method allows a proxy server to send over the raw text from a browser's
* output stream to be parsed and stored correctly into the UrlConfig object.
*/
public void parseArguments(String queryString)
{
String[] args = JMeterUtils.split(queryString, "&");
for (int i = 0; i < args.length; i++)
{
// need to handle four cases: string contains name=value
// string contains name=
// string contains name
// empty string
// find end of parameter name
int endOfNameIndex = 0;
String metaData = ""; // records the existance of an equal sign
if (args[i].indexOf("=") != -1)
{
endOfNameIndex = args[i].indexOf("="); // case of name=value, name=
metaData = "=";
}
else
{
metaData = "";
if (args[i].length() > 0)
{
endOfNameIndex = args[i].length(); // case name
}
else
{
endOfNameIndex = 0; //case where name value string is empty
}
}
// parse name
String name = ""; // for empty string
if (args[i].length() > 0)
{
name = args[i].substring(0, endOfNameIndex); //for non empty string
}
// parse value
String value = "";
if ((endOfNameIndex + 1) < args[i].length())
{
value = args[i].substring(endOfNameIndex + 1, args[i].length());
}
if (name.length() > 0)
{
// In JDK 1.2, the decode() method has a throws clause:
// "throws Exception". In JDK 1.3, the method does not have
// a throws clause. So, in order to be JDK 1.2 compliant,
// we need to add a try/catch around the method call.
try
{
addEncodedArgument(name, value, metaData);
}
catch (Exception e)
{
log.error(
"UrlConfig:parseArguments(): Unable to parse argument=["
+ value
+ "]");
log.error(
"UrlConfig:parseArguments(): queryString=[" + queryString + "]",
e);
}
}
}
}
/****************************************
* Reads the response from the URL connection
*
*@param conn URL from which to read response
*@return response in <code>String</code>
*@exception IOException if an I/O exception occurs
***************************************/
protected byte[] readResponse(HttpURLConnection conn) throws IOException
{
byte[] buffer = new byte[4096];
BufferedInputStream in;
try
{
in = new BufferedInputStream(conn.getInputStream());
}
catch (Exception e)
{
in = new BufferedInputStream(conn.getErrorStream());
}
java.io.ByteArrayOutputStream w = new ByteArrayOutputStream();
int x = 0;
while ((x = in.read(buffer)) > -1)
{
w.write(buffer, 0, x);
}
in.close();
w.flush();
w.close();
return w.toByteArray();
}
/****************************************
* Gets the ResponseHeaders from the URLConnection, save them to the
* SampleResults object.
*
*@param conn connection from which the headers are read
*@param res where the headers read are stored
***************************************/
protected byte[] getResponseHeaders(HttpURLConnection conn, SampleResult res)
throws IOException
{
StringBuffer headerBuf = new StringBuffer();
headerBuf.append(conn.getHeaderField(0).substring(0, 8));
headerBuf.append(" ");
headerBuf.append(conn.getResponseCode());
headerBuf.append(" ");
headerBuf.append(conn.getResponseMessage());
headerBuf.append("\n");
for (int i = 1; conn.getHeaderFieldKey(i) != null; i++)
{
if (!conn.getHeaderFieldKey(i).equalsIgnoreCase("transfer-encoding"))
{
headerBuf.append(conn.getHeaderFieldKey(i));
headerBuf.append(": ");
headerBuf.append(conn.getHeaderField(i));
headerBuf.append("\n");
}
}
headerBuf.append("\n");
return headerBuf.toString().getBytes("8859_1");
}
/****************************************
* Extracts all the required cookies for that particular URL request and set
* them in the <code>HttpURLConnection</code> passed in
*
*@param conn <code>HttpUrlConnection</code> which represents the URL
* request
*@param u <code>URL</code> of the URL request
*@param cookieManager the <code>CookieManager</code> containing all the
* cookies for this <code>UrlConfig</code>
***************************************/
private void setConnectionCookie(
HttpURLConnection conn,
URL u,
CookieManager cookieManager)
{
if (cookieManager != null)
{
String cookieHeader = cookieManager.getCookieHeaderForURL(u);
if (cookieHeader != null)
{
conn.setRequestProperty("Cookie", cookieHeader);
}
}
}
/****************************************
* Extracts all the required headers for that particular URL request and set
* them in the <code>HttpURLConnection</code> passed in
*
*@param conn <code>HttpUrlConnection</code> which represents the URL
* request
*@param u <code>URL</code> of the URL request
*@param headerManager the <code>HeaderManager</code> containing all the
* cookies for this <code>UrlConfig</code>
***************************************/
private void setConnectionHeaders(
HttpURLConnection conn,
URL u,
HeaderManager headerManager)
{
if (headerManager != null)
{
Collection headers = headerManager.getHeaders();
if (headers != null)
{
Iterator i = headers.iterator();
while (i.hasNext())
{
Header header = (Header) i.next();
conn.setRequestProperty(header.getName(), header.getValue());
}
}
}
}
/****************************************
* Extracts all the required authorization for that particular URL request and
* set them in the <code>HttpURLConnection</code> passed in
*
*@param conn <code>HttpUrlConnection</code> which represents the URL
* request
*@param u <code>URL</code> of the URL request
*@param authManager the <code>AuthManager</code> containing all the cookies
* for this <code>UrlConfig</code>
***************************************/
private void setConnectionAuthorization(
HttpURLConnection conn,
URL u,
AuthManager authManager)
{
if (authManager != null)
{
String authHeader = authManager.getAuthHeaderForURL(u);
if (authHeader != null)
{
conn.setRequestProperty("Authorization", authHeader);
}
}
}
/****************************************
* Get the response code of the URL connection and divide it by 100 thus
* returning 2(for 2xx response codes), 3(for 3xx reponse codes), etc
*
*@param conn <code>HttpURLConnection</code> of URL request
*@param res where all results of sampling will be stored
*@param time time when the URL request was first started
*@return HTTP response code divided by 100
***************************************/
private int getErrorLevel(
HttpURLConnection conn,
SampleResult res,
long time)
throws IOException
{
int errorLevel = 200;
String message = null;
errorLevel = ((HttpURLConnection) conn).getResponseCode();
message = ((HttpURLConnection) conn).getResponseMessage();
res.setResponseCode(String.valueOf(errorLevel));
res.setResponseMessage(message);
return errorLevel;
}
public void removeArguments()
{
setProperty(HTTPSampler.ARGUMENTS, new Arguments());
}
/****************************************
* Follow redirection manually. Normally if the web server does a redirection
* the intermediate page is not returned. Only the resultant page and the
* response code for the page will be returned. With redirection turned off,
* the response code of 3xx will be returned together with a "Location"
* header-value pair to indicate that the "Location" value needs to be followed
* to get the resultant page.
*
*@param conn connection
*@param u
*@param urlConfig !ToDo (Parameter description)
*@exception MalformedURLException if URL is not understood
***************************************/
private void redirectUrl(HttpURLConnection conn, URL u)
throws MalformedURLException
{
String loc = conn.getHeaderField("Location");
if (loc != null)
{
if (loc.indexOf("http") == -1)
{
String tempURL = u.toString();
if (loc.startsWith("/"))
{
int ind = tempURL.indexOf("//") + 2;
loc =
tempURL.substring(0, tempURL.indexOf("/", ind) + 1)
+ loc.substring(1);
}
else
{
loc =
u.toString().substring(0, u.toString().lastIndexOf('/') + 1) + loc;
}
}
}
URL newUrl = new URL(loc);
setMethod(GET);
setProtocol(newUrl.getProtocol());
setDomain(newUrl.getHost());
setPort(newUrl.getPort());
setPath(newUrl.getFile());
removeArguments();
parseArguments(newUrl.getQuery());
}
protected long connect() throws IOException
{
long time = System.currentTimeMillis();
try
{
conn.connect();
}
catch(BindException e)
{
log.debug("Bind exception, try again");
if(connectionTries++ == 10)
{
log.error("Can't connect",e);
throw e;
}
conn.disconnect();
conn = null;
System.gc();
Runtime.getRuntime().runFinalization();
this.setUseKeepAlive(false);
conn = setupConnection(getUrl(),getMethod());
if(getMethod().equals(HTTPSampler.POST))
{
setPostHeaders(conn);
}
time = connect();
}
catch(IOException e)
{
log.debug("Connection failed, giving up");
conn.disconnect();
conn = null;
System.gc();
Runtime.getRuntime().runFinalization();
throw e;
}
return time;
}
/****************************************
* Samples <code>Entry</code> passed in and stores the result in <code>SampleResult</code>
*
*@param e <code>Entry</code> to be sampled
*@param redirects the level of redirection we're processing (0 means
* original request) -- just used to prevent
* an infinite loop.
*@return results of the sampling
***************************************/
private SampleResult sample(int redirects)
{
log.debug("Start : sample2");
long time = System.currentTimeMillis();
SampleResult res = new SampleResult();
URL u = null;
try
{
u = getUrl();
res.setSampleLabel(getName());
// specify the data to the result.
res.setSamplerData(this);
/****************************************
* END - cached logging hack
***************************************/
if (log.isDebugEnabled())
{
log.debug("sample2 : sampling url - " + u);
}
conn = setupConnection(u, getMethod());
if (getProperty(HTTPSampler.METHOD).equals(HTTPSampler.POST))
{
setPostHeaders(conn);
time = connect();
sendPostData(conn);
}
else time = connect();
saveConnectionCookies(conn, u, getCookieManager());
int errorLevel = 0;
try
{
errorLevel = getErrorLevel(conn, res, time);
}
catch (IOException e)
{
time = bundleResponseInResult(time, res, conn);
res.setSuccessful(false);
res.setTime(time);
return res;
}
if (errorLevel / 100 == 2 || errorLevel == 304)
{
time = bundleResponseInResult(time, res, conn);
}
else if (errorLevel / 100 == 3)
{
if (redirects >= MAX_REDIRECTS)
{
throw new IOException("Maximum number of redirects exceeded");
}
if (!getFollowRedirects())
{
time = bundleResponseInResult(time, res, conn);
}
else
{
redirectUrl(conn, u);
time = System.currentTimeMillis() - time;
res = sample(redirects+1);
time += res.getTime();
}
}
else
{
// Could not sample the URL
time = bundleResponseInResult(time, res, conn);
res.setSuccessful(false);
}
res.setTime(time);
log.debug("End : sample2");
return res;
}
catch (IOException ex)
{
log.warn(ex.getMessage(), ex);
res.setDataType(res.TEXT);
res.setResponseData(ex.toString().getBytes());
res.setResponseCode(NON_HTTP_RESPONSE_CODE);
res.setResponseMessage(NON_HTTP_RESPONSE_MESSAGE);
res.setTime(System.currentTimeMillis() - time);
res.setSuccessful(false);
}
finally
{
try
{
// calling disconnect doesn't close the connection immediately, but
// indicates we're through with it. The JVM should close it when
// necessary.
String connection = conn.getHeaderField("Connection");
if (connection == null || connection.equalsIgnoreCase("close"))
conn.disconnect();
}
catch (Exception e)
{
}
}
log.debug("End : sample2");
return res;
}
private long bundleResponseInResult(
long time,
SampleResult res,
HttpURLConnection conn)
throws IOException, FileNotFoundException
{
res.setDataType(res.TEXT);
byte[] ret = readResponse(conn);
byte[] head = getResponseHeaders(conn, res);
time = System.currentTimeMillis() - time;
byte[] complete = new byte[ret.length + head.length];
System.arraycopy(head, 0, complete, 0, head.length);
System.arraycopy(ret, 0, complete, head.length, ret.length);
res.setResponseData(complete);
res.setSuccessful(true);
return time;
}
/****************************************
* From the <code>HttpURLConnection</code>, store all the "set-cookie" key-pair
* values in the cookieManager of the <code>UrlConfig</code>
*
*@param conn <code>HttpUrlConnection</code> which represents the URL
* request
*@param u <code>URL</code> of the URL request
*@param cookieManager the <code>CookieManager</code> containing all the
* cookies for this <code>UrlConfig</code>
***************************************/
private void saveConnectionCookies(
HttpURLConnection conn,
URL u,
CookieManager cookieManager)
{
if (cookieManager != null)
{
for (int i = 1; conn.getHeaderFieldKey(i) != null; i++)
{
if (conn.getHeaderFieldKey(i).equalsIgnoreCase("set-cookie"))
{
cookieManager.addCookieFromHeader(conn.getHeaderField(i), u);
}
}
}
}
public String toString()
{
try
{
return this.getUrl().toString()
+ ((POST.equals(getMethod())) ? "\nQuery Data: " + getQueryString() : "");
}
catch (MalformedURLException e)
{
return "";
}
}
public static class Test extends junit.framework.TestCase
{
public Test(String name)
{
super(name);
}
public void testMakingUrl() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.addArgument("param1", "value1");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1=value1";,
config.getUrl().toString());
}
public void testMakingUrl2() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.addArgument("param1", "value1");
config.setPath("/index.html?p1=p2");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1=value1&p1=p2";,
config.getUrl().toString());
}
public void testMakingUrl3() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.POST);
config.addArgument("param1", "value1");
config.setPath("/index.html?p1=p2");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?p1=p2";,
config.getUrl().toString());
}
// test cases for making Url, and exercise method addArgument(String name,String value,String metadata)
public void testMakingUrl4() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.addArgument("param1", "value1", "=");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1=value1";,
config.getUrl().toString());
}
public void testMakingUrl5() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.addArgument("param1", "", "=");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1=";,
config.getUrl().toString());
}
public void testMakingUrl6() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.addArgument("param1", "", "");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1";,
config.getUrl().toString());
}
// test cases for making Url, and exercise method parseArguments(String queryString)
public void testMakingUrl7() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.parseArguments("param1=value1");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1=value1";,
config.getUrl().toString());
}
public void testMakingUrl8() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.parseArguments("param1=");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1=";,
config.getUrl().toString());
}
public void testMakingUrl9() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.parseArguments("param1");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html?param1";,
config.getUrl().toString());
}
public void testMakingUrl10() throws Exception
{
HTTPSampler config = new HTTPSampler();
config.setProtocol("http");
config.setMethod(HTTPSampler.GET);
config.parseArguments("");
config.setPath("/index.html");
config.setDomain("www.apache.org");
assertEquals(
"http://www.apache.org:80/index.html";,
config.getUrl().toString());
}
}
}
package org.apache.jmeter.protocol.http.sampler;
import java.io.IOException;
import java.io.PrintWriter;
import java.net.HttpURLConnection;
import java.net.URLConnection;
/**
* @author Administrator
*
* To change this generated comment edit the template variable "typecomment":
* Window>Preferences>Java>Templates.
*/
public class SoapSampler extends HTTPSampler
{
public static final String XML_DATA = "HTTPSamper.xml_data";
public void setXmlData(String data)
{
setProperty(XML_DATA,data);
}
public String getXmlData()
{
return getPropertyAsString(XML_DATA);
}
/****************************************
* Set the HTTP request headers in preparation to open the connection
* and sending the POST data:
*
*@param connection <code>URLConnection</code> to set headers on
*@exception IOException if an I/O exception occurs
***************************************/
public void setHeaders(URLConnection connection)
throws IOException
{
((HttpURLConnection)connection).setRequestMethod("POST");
connection.setRequestProperty("Content-length", "" + getXmlData().length());
connection.setRequestProperty("Content-type", "text/xml");
connection.setDoOutput(true);
}
/****************************************
* Send POST data from <code>Entry</code> to the open connection.
*
*@param connection <code>URLConnection</code> of where POST data should
* be sent
*@param url contains the query string for POST
*@exception IOException if an I/O exception occurs
***************************************/
public void sendPostData(URLConnection connection)
throws IOException
{
PrintWriter out = new PrintWriter(connection.getOutputStream());
out.print(getXmlData());
out.close();
}
}
-- To unsubscribe, e-mail: <mailto:jmeter-dev-unsubscribe@;jakarta.apache.org> For additional commands, e-mail: <mailto:jmeter-dev-help@;jakarta.apache.org>
