com.gargoylesoftware.htmlunit.WebRequest Maven / Gradle / Ivy
Show all versions of htmlunit Show documentation
/*
* Copyright (c) 2002-2021 Gargoyle Software Inc.
*
* 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.gargoylesoftware.htmlunit;
import static java.nio.charset.StandardCharsets.ISO_8859_1;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.Serializable;
import java.net.IDN;
import java.net.MalformedURLException;
import java.net.URL;
import java.nio.charset.Charset;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.regex.Pattern;
import org.apache.http.auth.Credentials;
import org.apache.http.auth.UsernamePasswordCredentials;
import com.gargoylesoftware.htmlunit.util.NameValuePair;
import com.gargoylesoftware.htmlunit.util.UrlUtils;
/**
* Parameter object for making web requests.
*
* @author Brad Clarke
* @author Hans Donner
* @author Ahmed Ashour
* @author Marc Guillemot
* @author Rodney Gitzel
* @author Ronald Brill
* @author Adam Afeltowicz
* @author Joerg Werner
*/
public class WebRequest implements Serializable {
public enum HttpHint {
/** Force to include the charset. */
IncludeCharsetInContentTypeHeader
}
private static final Pattern DOT_PATTERN = Pattern.compile("/\\./");
private static final Pattern DOT_DOT_PATTERN = Pattern.compile("/(?!\\.\\.)[^/]*/\\.\\./");
private static final Pattern REMOVE_DOTS_PATTERN = Pattern.compile("^/(\\.\\.?/)*");
private String url_; // String instead of java.net.URL because "about:blank" URLs don't serialize correctly
private String proxyHost_;
private int proxyPort_;
private boolean isSocksProxy_;
private HttpMethod httpMethod_ = HttpMethod.GET;
private FormEncodingType encodingType_ = FormEncodingType.URL_ENCODED;
private Map additionalHeaders_ = new HashMap<>();
private Credentials urlCredentials_;
private Credentials credentials_;
private int timeout_;
private transient Charset charset_ = ISO_8859_1;
private transient Set httpHints_;
/* These two are mutually exclusive; additionally, requestBody_ should only be set for POST requests. */
private List requestParameters_ = Collections.emptyList();
private String requestBody_;
/**
* Instantiates a {@link WebRequest} for the specified URL.
* @param url the target URL
* @param acceptHeader the accept header to use
* @param acceptEncodingHeader the accept encoding header to use
*/
public WebRequest(final URL url, final String acceptHeader, final String acceptEncodingHeader) {
setUrl(url);
if (acceptHeader != null) {
setAdditionalHeader(HttpHeader.ACCEPT, acceptHeader);
}
if (acceptEncodingHeader != null) {
setAdditionalHeader(HttpHeader.ACCEPT_ENCODING, acceptEncodingHeader);
}
timeout_ = -1;
}
/**
* @return a new request for about:blank
*/
public static WebRequest newAboutBlankRequest() {
return new WebRequest(UrlUtils.URL_ABOUT_BLANK, "*/*", "gzip, deflate");
}
/**
* Instantiates a {@link WebRequest} for the specified URL.
* @param url the target URL
*/
public WebRequest(final URL url) {
this(url, "*/*", "gzip, deflate");
}
/**
* Instantiates a {@link WebRequest} for the specified URL using the specified HTTP submit method.
* @param url the target URL
* @param submitMethod the HTTP submit method to use
*/
public WebRequest(final URL url, final HttpMethod submitMethod) {
this(url);
setHttpMethod(submitMethod);
}
/**
* Returns the target URL.
* @return the target URL
*/
public URL getUrl() {
return UrlUtils.toUrlSafe(url_);
}
/**
* Sets the target URL. The URL may be simplified if needed (for instance eliminating
* irrelevant path portions like "/./").
* @param url the target URL
*/
public void setUrl(URL url) {
if (url == null) {
url_ = null;
return;
}
final String path = url.getPath();
if (path.isEmpty()) {
if (!url.getFile().isEmpty() || url.getProtocol().startsWith("http")) {
url = buildUrlWithNewPath(url, "/");
}
}
else if (path.contains("/.")) {
url = buildUrlWithNewPath(url, removeDots(path));
}
final String idn = IDN.toASCII(url.getHost());
if (!idn.equals(url.getHost())) {
try {
url = UrlUtils.getUrlWithNewHost(url, idn);
}
catch (final Exception e) {
throw new RuntimeException("Cannot change hostname of URL: " + url.toExternalForm(), e);
}
}
url_ = url.toExternalForm();
// http://john.smith:secret@localhost
final String userInfo = url.getUserInfo();
if (userInfo != null) {
final int splitPos = userInfo.indexOf(':');
if (splitPos == -1) {
urlCredentials_ = new UsernamePasswordCredentials(userInfo, "");
}
else {
final String username = userInfo.substring(0, splitPos);
final String password = userInfo.substring(splitPos + 1);
urlCredentials_ = new UsernamePasswordCredentials(username, password);
}
}
}
/*
* Strip a URL string of "/./" and "/../" occurrences.
*
* One trick here is to repeatedly create new matchers on a given
* pattern, so that we can see whether it needs to be re-applied;
* unfortunately .replaceAll() doesn't re-process its own output,
* so if we create a new match with a replacement, it is missed.
*/
private static String removeDots(final String path) {
String newPath = path;
// remove occurrences at the beginning
newPath = REMOVE_DOTS_PATTERN.matcher(newPath).replaceAll("/");
if ("/..".equals(newPath)) {
newPath = "/";
}
// single dots have no effect, so just remove them
while (DOT_PATTERN.matcher(newPath).find()) {
newPath = DOT_PATTERN.matcher(newPath).replaceAll("/");
}
// mid-path double dots should be removed WITH the previous subdirectory and replaced
// with "/" BUT ONLY IF that subdirectory's not also ".." (a regex lookahead helps with this)
while (DOT_DOT_PATTERN.matcher(newPath).find()) {
newPath = DOT_DOT_PATTERN.matcher(newPath).replaceAll("/");
}
return newPath;
}
private static URL buildUrlWithNewPath(URL url, final String newPath) {
try {
url = UrlUtils.getUrlWithNewPath(url, newPath);
}
catch (final Exception e) {
throw new RuntimeException("Cannot change path of URL: " + url.toExternalForm(), e);
}
return url;
}
/**
* Returns the proxy host to use.
* @return the proxy host to use
*/
public String getProxyHost() {
return proxyHost_;
}
/**
* Sets the proxy host to use.
* @param proxyHost the proxy host to use
*/
public void setProxyHost(final String proxyHost) {
proxyHost_ = proxyHost;
}
/**
* Returns the proxy port to use.
* @return the proxy port to use
*/
public int getProxyPort() {
return proxyPort_;
}
/**
* Sets the proxy port to use.
* @param proxyPort the proxy port to use
*/
public void setProxyPort(final int proxyPort) {
proxyPort_ = proxyPort;
}
/**
* Returns whether SOCKS proxy or not.
* @return whether SOCKS proxy or not
*/
public boolean isSocksProxy() {
return isSocksProxy_;
}
/**
* Sets whether SOCKS proxy or not.
* @param isSocksProxy whether SOCKS proxy or not
*/
public void setSocksProxy(final boolean isSocksProxy) {
isSocksProxy_ = isSocksProxy;
}
/**
* @return the timeout to use
*/
public int getTimeout() {
return timeout_;
}
/**
* Sets the timeout to use.
* @param timeout the timeout to use
*/
public void setTimeout(final int timeout) {
timeout_ = timeout;
}
/**
* Returns the form encoding type to use.
* @return the form encoding type to use
*/
public FormEncodingType getEncodingType() {
return encodingType_;
}
/**
* Sets the form encoding type to use.
* @param encodingType the form encoding type to use
*/
public void setEncodingType(final FormEncodingType encodingType) {
encodingType_ = encodingType;
}
/**
* Retrieves the request parameters to use. If set, these request parameters will overwrite any
* request parameters which may be present in the {@link #getUrl() URL}. Should not be used in
* combination with the {@link #setRequestBody(String) request body}.
* @return the request parameters to use
*/
public List getRequestParameters() {
return requestParameters_;
}
/**
* Sets the request parameters to use. If set, these request parameters will overwrite any request
* parameters which may be present in the {@link #getUrl() URL}. Should not be used in combination
* with the {@link #setRequestBody(String) request body}.
* @param requestParameters the request parameters to use
* @throws RuntimeException if the request body has already been set
*/
public void setRequestParameters(final List requestParameters) throws RuntimeException {
if (requestBody_ != null) {
final String msg = "Trying to set the request parameters, but the request body has already been specified;"
+ "the two are mutually exclusive!";
throw new RuntimeException(msg);
}
requestParameters_ = requestParameters;
}
/**
* Returns the body content to be submitted if this is a POST request. Ignored for all other request
* types. Should not be used in combination with {@link #setRequestParameters(List) request parameters}.
* @return the body content to be submitted if this is a POST request
*/
public String getRequestBody() {
return requestBody_;
}
/**
* Sets the body content to be submitted if this is a {@code POST}, {@code PUT} or {@code PATCH} request.
* Ignored for all other request types.
* Should not be used in combination with {@link #setRequestParameters(List) request parameters}.
* @param requestBody the body content to be submitted if this is a {@code POST}, {@code PUT}
* or {@code PATCH} request
* @throws RuntimeException if the request parameters have already been set
* or this is not a {@code POST}, {@code PUT} or {@code PATCH} request.
*/
public void setRequestBody(final String requestBody) throws RuntimeException {
if (requestParameters_ != null && !requestParameters_.isEmpty()) {
final String msg = "Trying to set the request body, but the request parameters have already been specified;"
+ "the two are mutually exclusive!";
throw new RuntimeException(msg);
}
if (httpMethod_ != HttpMethod.POST && httpMethod_ != HttpMethod.PUT && httpMethod_ != HttpMethod.PATCH) {
final String msg = "The request body may only be set for POST, PUT or PATCH requests!";
throw new RuntimeException(msg);
}
requestBody_ = requestBody;
}
/**
* Returns the HTTP submit method to use.
* @return the HTTP submit method to use
*/
public HttpMethod getHttpMethod() {
return httpMethod_;
}
/**
* Sets the HTTP submit method to use.
* @param submitMethod the HTTP submit method to use
*/
public void setHttpMethod(final HttpMethod submitMethod) {
httpMethod_ = submitMethod;
}
/**
* Returns the additional HTTP headers to use.
* @return the additional HTTP headers to use
*/
public Map getAdditionalHeaders() {
return additionalHeaders_;
}
/**
* Sets the additional HTTP headers to use.
* @param additionalHeaders the additional HTTP headers to use
*/
public void setAdditionalHeaders(final Map additionalHeaders) {
additionalHeaders_ = additionalHeaders;
}
/**
* Returns whether the specified header name is already included in the additional HTTP headers.
* @param name the name of the additional HTTP header
* @return true if the specified header name is included in the additional HTTP headers
*/
public boolean isAdditionalHeader(final String name) {
for (final String key : additionalHeaders_.keySet()) {
if (name.equalsIgnoreCase(key)) {
return true;
}
}
return false;
}
/**
* Returns the header value associated with this name.
* @param name the name of the additional HTTP header
* @return the value or null
*/
public String getAdditionalHeader(final String name) {
String newKey = name;
for (final String key : additionalHeaders_.keySet()) {
if (name.equalsIgnoreCase(key)) {
newKey = key;
break;
}
}
return additionalHeaders_.get(newKey);
}
/**
* Sets the referer HTTP header - only if the provided url is valid.
* @param url the url for the referer HTTP header
*/
public void setRefererlHeader(final URL url) {
if (url == null || !url.getProtocol().startsWith("http")) {
return;
}
try {
setAdditionalHeader(HttpHeader.REFERER, UrlUtils.getUrlWithoutRef(url).toExternalForm());
}
catch (final MalformedURLException e) {
// bad luck us the whole url from the pager
}
}
/**
* Sets the specified name/value pair in the additional HTTP headers.
* @param name the name of the additional HTTP header
* @param value the value of the additional HTTP header
*/
public void setAdditionalHeader(final String name, final String value) {
String newKey = name;
for (final String key : additionalHeaders_.keySet()) {
if (name.equalsIgnoreCase(key)) {
newKey = key;
break;
}
}
additionalHeaders_.put(newKey, value);
}
/**
* Removed the specified name/value pair from the additional HTTP headers.
* @param name the name of the additional HTTP header
*/
public void removeAdditionalHeader(String name) {
for (final String key : additionalHeaders_.keySet()) {
if (name.equalsIgnoreCase(key)) {
name = key;
break;
}
}
additionalHeaders_.remove(name);
}
/**
* Returns the credentials to use.
* @return the credentials if set as part of the url
*/
public Credentials getUrlCredentials() {
return urlCredentials_;
}
/**
* Returns the credentials to use.
* @return the credentials if set from the external builder
*/
public Credentials getCredentials() {
return credentials_;
}
/**
* Sets the credentials to use.
* @param credentials the credentials to use
*/
public void setCredentials(final Credentials credentials) {
credentials_ = credentials;
}
/**
* Returns the character set to use to perform the request.
* @return the character set to use to perform the request
*/
public Charset getCharset() {
return charset_;
}
/**
* Sets the character set to use to perform the request. The default value
* is {@link java.nio.charset.StandardCharsets#ISO_8859_1}.
* @param charset the character set to use to perform the request
*/
public void setCharset(final Charset charset) {
charset_ = charset;
}
public boolean hasHint(final HttpHint hint) {
if (httpHints_ == null) {
return false;
}
return httpHints_.contains(hint);
}
public void addHint(final HttpHint hint) {
if (httpHints_ == null) {
httpHints_ = new HashSet<>();
}
httpHints_.add(hint);
}
/**
* Returns a string representation of this object.
* @return a string representation of this object
*/
@Override
public String toString() {
final StringBuilder builder = new StringBuilder(100)
.append(getClass().getSimpleName())
.append("[]");
return builder.toString();
}
private void writeObject(final ObjectOutputStream oos) throws IOException {
oos.defaultWriteObject();
oos.writeObject(charset_ == null ? null : charset_.name());
}
private void readObject(final ObjectInputStream ois) throws ClassNotFoundException, IOException {
ois.defaultReadObject();
final String charsetName = (String) ois.readObject();
if (charsetName != null) {
charset_ = Charset.forName(charsetName);
}
}
}