org.eclipse.jetty.ee8.nested.Dispatcher Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jetty-ee8-nested Show documentation
Show all versions of jetty-ee8-nested Show documentation
The jetty core handler adapted to legacy ee8 handler.
The newest version!
//
// ========================================================================
// Copyright (c) 1995 Mort Bay Consulting Pty Ltd and others.
//
// This program and the accompanying materials are made available under the
// terms of the Eclipse Public License v. 2.0 which is available at
// https://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0
// which is available at https://www.apache.org/licenses/LICENSE-2.0.
//
// SPDX-License-Identifier: EPL-2.0 OR Apache-2.0
// ========================================================================
//
package org.eclipse.jetty.ee8.nested;
import java.io.IOException;
import java.util.HashSet;
import java.util.Objects;
import java.util.Set;
import javax.servlet.DispatcherType;
import javax.servlet.RequestDispatcher;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletMapping;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.eclipse.jetty.http.HttpException;
import org.eclipse.jetty.http.HttpURI;
import org.eclipse.jetty.http.UriCompliance;
import org.eclipse.jetty.util.Attributes;
import org.eclipse.jetty.util.Fields;
import org.eclipse.jetty.util.URIUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class Dispatcher implements RequestDispatcher {
private static final Logger LOG = LoggerFactory.getLogger(Dispatcher.class);
/**
* Dispatch include attribute names
*/
public static final String __INCLUDE_PREFIX = "javax.servlet.include.";
/**
* Dispatch include attribute names
*/
public static final String __FORWARD_PREFIX = "javax.servlet.forward.";
private final ContextHandler _contextHandler;
private final HttpURI _uri;
private final String _pathInContext;
private final String _named;
public Dispatcher(ContextHandler contextHandler, HttpURI uri, String pathInContext) {
_contextHandler = contextHandler;
_uri = uri.asImmutable();
_pathInContext = pathInContext;
_named = null;
}
public Dispatcher(ContextHandler contextHandler, String name) throws IllegalStateException {
_contextHandler = contextHandler;
_uri = null;
_pathInContext = null;
_named = name;
}
public void error(ServletRequest request, ServletResponse response) throws ServletException, IOException {
forward(request, response, DispatcherType.ERROR);
}
@Override
public void include(ServletRequest request, ServletResponse response) throws ServletException, IOException {
Request baseRequest = Objects.requireNonNull(Request.getBaseRequest(request));
if (!(request instanceof HttpServletRequest))
request = new ServletRequestHttpWrapper(request);
if (!(response instanceof HttpServletResponse))
response = new ServletResponseHttpWrapper(response);
final DispatcherType old_type = baseRequest.getDispatcherType();
final Attributes old_attr = baseRequest.getAttributes();
final Fields old_query_fields = baseRequest.getQueryFields();
final ContextHandler.APIContext old_context = baseRequest.getContext();
final ServletPathMapping old_mapping = baseRequest.getServletPathMapping();
try {
baseRequest.setDispatcherType(DispatcherType.INCLUDE);
baseRequest.getResponse().include();
if (_named != null) {
_contextHandler.handle(_named, baseRequest, (HttpServletRequest) request, (HttpServletResponse) response);
} else {
Objects.requireNonNull(_uri);
// Check any URI violations against the compliance for this request
checkUriViolations(_uri, baseRequest);
IncludeAttributes attr = new IncludeAttributes(old_attr, baseRequest, old_context, old_mapping, _uri.getPath(), _pathInContext, _uri.getQuery());
if (attr._query != null)
baseRequest.mergeQueryParameters(baseRequest.getQueryString(), attr._query);
baseRequest.setAttributes(attr);
_contextHandler.handle(_pathInContext, baseRequest, (HttpServletRequest) request, (HttpServletResponse) response);
}
} finally {
baseRequest.setAttributes(old_attr);
baseRequest.getResponse().included();
baseRequest.setQueryFields(old_query_fields);
baseRequest.resetParameters();
baseRequest.setDispatcherType(old_type);
}
}
@Override
public void forward(ServletRequest request, ServletResponse response) throws ServletException, IOException {
forward(request, response, DispatcherType.FORWARD);
}
protected void forward(ServletRequest request, ServletResponse response, DispatcherType dispatch) throws ServletException, IOException {
Request baseRequest = Objects.requireNonNull(Request.getBaseRequest(request));
Response baseResponse = baseRequest.getResponse();
baseResponse.resetForForward();
if (!(request instanceof HttpServletRequest))
request = new ServletRequestHttpWrapper(request);
if (!(response instanceof HttpServletResponse))
response = new ServletResponseHttpWrapper(response);
final HttpURI old_uri = baseRequest.getHttpURI();
final ContextHandler.APIContext old_context = baseRequest.getContext();
final String old_path_in_context = baseRequest.getPathInContext();
final ServletPathMapping old_mapping = baseRequest.getServletPathMapping();
final ServletPathMapping source_mapping = baseRequest.findServletPathMapping();
final Fields old_query_params = baseRequest.getQueryFields();
final Attributes old_attr = baseRequest.getAttributes();
final DispatcherType old_type = baseRequest.getDispatcherType();
try {
baseRequest.setDispatcherType(dispatch);
if (_named != null) {
_contextHandler.handle(_named, baseRequest, (HttpServletRequest) request, (HttpServletResponse) response);
} else {
Objects.requireNonNull(_uri);
// Check any URI violations against the compliance for this request
checkUriViolations(_uri, baseRequest);
// If we have already been forwarded previously, then keep using the established
// original value. Otherwise, this is the first forward, and we need to establish the values.
// Note: the established value on the original request for pathInfo and
// for queryString is allowed to be null, but cannot be null for the other values.
// Note: the pathInfo is passed as the pathInContext since it is only used when there is
// no mapping, and when there is no mapping the pathInfo is the pathInContext.
if (old_attr.getAttribute(FORWARD_REQUEST_URI) == null)
baseRequest.setAttributes(new ForwardAttributes(old_attr, old_uri.getPath(), baseRequest.getContextPath(), baseRequest.getPathInContext(), source_mapping, old_uri.getQuery()));
String query = _uri.getQuery();
if (query == null)
query = old_uri.getQuery();
String decodedPathInContext = URIUtil.decodePath(_pathInContext);
baseRequest.setHttpURI(HttpURI.build(old_uri, _uri.getPath(), _uri.getParam(), query));
baseRequest.setContext(_contextHandler.getServletContext(), decodedPathInContext);
if (_uri.getQuery() != null || old_uri.getQuery() != null) {
try {
baseRequest.mergeQueryParameters(old_uri.getQuery(), _uri.getQuery());
} catch (Throwable e) {
// Only throw HttpException if not in Error Dispatch Mode
// This allows application ErrorPageErrorHandler to handle HttpException messages
if (e instanceof HttpException && dispatch == DispatcherType.ERROR) {
LOG.warn("Ignoring Original Bad Request Query String: {}", old_uri, e);
} else {
throw e;
}
}
}
_contextHandler.handle(decodedPathInContext, baseRequest, (HttpServletRequest) request, (HttpServletResponse) response);
// If we are not async and not closed already, then close via the possibly wrapped response.
if (!baseRequest.getHttpChannelState().isAsync() && !baseResponse.getHttpOutput().isClosed()) {
try {
response.getOutputStream().close();
} catch (IllegalStateException e) {
response.getWriter().close();
}
}
}
} finally {
baseRequest.setHttpURI(old_uri);
baseRequest.setContext(old_context, old_path_in_context);
baseRequest.setServletPathMapping(old_mapping);
baseRequest.setQueryFields(old_query_params);
baseRequest.resetParameters();
baseRequest.setAttributes(old_attr);
baseRequest.setDispatcherType(old_type);
}
}
private static void checkUriViolations(HttpURI uri, Request baseRequest) {
if (uri.hasViolations()) {
HttpChannel channel = baseRequest.getHttpChannel();
UriCompliance compliance = channel == null || channel.getHttpConfiguration() == null ? null : channel.getHttpConfiguration().getUriCompliance();
String illegalState = UriCompliance.checkUriCompliance(compliance, uri, org.eclipse.jetty.server.HttpChannel.from(baseRequest.getCoreRequest()).getComplianceViolationListener());
if (illegalState != null)
throw new IllegalStateException(illegalState);
}
}
@Override
public String toString() {
return String.format("Dispatcher@0x%x{%s,%s}", hashCode(), _named, _uri);
}
private class ForwardAttributes extends Attributes.Wrapper {
private final String _requestURI;
private final String _contextPath;
private final String _servletPath;
private final String _pathInfo;
private final ServletPathMapping _servletPathMapping;
private final String _query;
public ForwardAttributes(Attributes attributes, String requestURI, String contextPath, String pathInContext, ServletPathMapping mapping, String query) {
super(attributes);
_requestURI = requestURI;
_contextPath = contextPath;
_servletPathMapping = mapping;
_query = query;
_pathInfo = _servletPathMapping == null ? pathInContext : _servletPathMapping.getPathInfo();
_servletPath = _servletPathMapping == null ? null : _servletPathMapping.getServletPath();
}
@Override
public Object getAttribute(String key) {
if (Dispatcher.this._named == null) {
switch(key) {
case FORWARD_PATH_INFO:
return _pathInfo;
case FORWARD_REQUEST_URI:
return _requestURI;
case FORWARD_SERVLET_PATH:
return _servletPath;
case FORWARD_CONTEXT_PATH:
return _contextPath;
case FORWARD_QUERY_STRING:
return _query;
case FORWARD_MAPPING:
return _servletPathMapping;
default:
break;
}
}
// If we are forwarded then we hide include attributes
if (key.startsWith(__INCLUDE_PREFIX))
return null;
return getWrapped().getAttribute(key);
}
@Override
public Set getAttributeNameSet() {
HashSet set = new HashSet<>();
for (String name : getWrapped().getAttributeNameSet()) {
if (!name.startsWith(__INCLUDE_PREFIX) && !name.startsWith(__FORWARD_PREFIX))
set.add(name);
}
if (_named == null) {
if (_pathInfo != null)
set.add(FORWARD_PATH_INFO);
if (_requestURI != null)
set.add(FORWARD_REQUEST_URI);
if (_servletPath != null)
set.add(FORWARD_SERVLET_PATH);
if (_contextPath != null)
set.add(FORWARD_CONTEXT_PATH);
if (_servletPathMapping != null)
set.add(FORWARD_MAPPING);
if (_query != null)
set.add(FORWARD_QUERY_STRING);
}
return set;
}
@Override
public Object setAttribute(String key, Object value) {
// Allow any attribute to be set, even if a reserved name. If a reserved
// name is set here, it will be hidden by this class during the forward,
// but revealed after the forward is complete just as if the reserved name
// attribute had be set by the application before the forward.
return getWrapped().setAttribute(key, value);
}
@Override
public String toString() {
return "FORWARD+" + getWrapped().toString();
}
@Override
public void clearAttributes() {
throw new IllegalStateException();
}
@Override
public Object removeAttribute(String name) {
return setAttribute(name, null);
}
}
/**
* Attributes Wrapper to provide the {@link DispatcherType#INCLUDE} attributes.
*
* The source {@link ContextHandler.APIContext} and
* {@link ServletPathMapping} instances are also retained by this wrapper so they
* may be used by {@link Request#getContextPath()}, {@link Request#getServletPath()},
* {@link Request#getPathInfo()} and {@link Request#getHttpServletMapping()}.
*/
class IncludeAttributes extends Attributes.Wrapper {
private final Request _baseRequest;
private final ContextHandler.APIContext _sourceContext;
private final ServletPathMapping _sourceMapping;
private final String _requestURI;
private final String _pathInContext;
private final String _query;
public IncludeAttributes(Attributes attributes, Request baseRequest, ContextHandler.APIContext sourceContext, ServletPathMapping sourceMapping, String requestURI, String pathInContext, String query) {
super(attributes);
_baseRequest = baseRequest;
_sourceMapping = sourceMapping;
_requestURI = requestURI;
_sourceContext = sourceContext;
_pathInContext = pathInContext;
_query = query;
}
ContextHandler.APIContext getSourceContext() {
return _sourceContext;
}
ServletPathMapping getSourceMapping() {
return _sourceMapping;
}
@Override
public Object getAttribute(String key) {
if (Dispatcher.this._named == null) {
switch(key) {
case INCLUDE_PATH_INFO:
{
ServletPathMapping mapping = _baseRequest.getServletPathMapping();
return mapping == null ? _pathInContext : mapping.getPathInfo();
}
case INCLUDE_SERVLET_PATH:
{
ServletPathMapping mapping = _baseRequest.getServletPathMapping();
return mapping == null ? null : mapping.getServletPath();
}
case INCLUDE_CONTEXT_PATH:
{
ContextHandler.APIContext context = _baseRequest.getContext();
return context == null ? null : context.getContextHandler().getRequestContextPath();
}
case INCLUDE_QUERY_STRING:
return _query;
case INCLUDE_REQUEST_URI:
return _requestURI;
case INCLUDE_MAPPING:
return _baseRequest.getServletPathMapping();
default:
break;
}
}
return getWrapped().getAttribute(key);
}
@Override
public Set getAttributeNameSet() {
HashSet set = new HashSet<>();
for (String name : getWrapped().getAttributeNameSet()) {
if (!name.startsWith(__INCLUDE_PREFIX))
set.add(name);
}
// We can't assign these in the constructor because the ServletPathMapping hasn't been set by the ServletHandler.
String pathInfo = (String) getAttribute(INCLUDE_PATH_INFO);
String servletPath = (String) getAttribute(INCLUDE_SERVLET_PATH);
String contextPath = (String) getAttribute(INCLUDE_CONTEXT_PATH);
HttpServletMapping includeMapping = (HttpServletMapping) getAttribute(INCLUDE_MAPPING);
if (_named == null) {
if (pathInfo != null)
set.add(INCLUDE_PATH_INFO);
if (_requestURI != null)
set.add(INCLUDE_REQUEST_URI);
if (servletPath != null)
set.add(INCLUDE_SERVLET_PATH);
if (contextPath != null)
set.add(INCLUDE_CONTEXT_PATH);
if (includeMapping != null)
set.add(INCLUDE_MAPPING);
if (_query != null)
set.add(INCLUDE_QUERY_STRING);
}
return set;
}
@Override
public Object setAttribute(String key, Object value) {
// Allow any attribute to be set, even if a reserved name. If a reserved
// name is set here, it will be revealed after the include is complete.
return getWrapped().setAttribute(key, value);
}
@Override
public String toString() {
return "INCLUDE+" + getWrapped().toString();
}
@Override
public void clearAttributes() {
throw new IllegalStateException();
}
@Override
public Object removeAttribute(String name) {
return setAttribute(name, null);
}
}
}