/*
* Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
*
* This file is part of Resin(R) Open Source
*
* Each copy or derived work must preserve the copyright notice and this
* notice unmodified.
*
* Resin Open Source is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Resin Open Source is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
* of NON-INFRINGEMENT. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License
* along with Resin Open Source; if not, write to the
*
* Free Software Foundation, Inc.
* 59 Temple Place, Suite 330
* Boston, MA 02111-1307 USA
*
* @author Scott Ferguson
*/
package com.caucho.server.rewrite;
import java.io.IOException;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.annotation.PostConstruct;
import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import com.caucho.config.ConfigException;
import com.caucho.config.program.ConfigProgram;
import com.caucho.config.program.ContainerProgram;
import com.caucho.server.dispatch.ServletConfigImpl;
import com.caucho.server.webapp.WebApp;
import com.caucho.util.L10N;
public class ProxyRule
extends AbstractRuleWithConditions
{
private static final L10N L = new L10N(ProxyRule.class);
private static final Logger log = Logger.getLogger(ProxyRule.class.getName());
private final WebApp _webApp;
private ServletConfigImpl _servlet;
private String _target;
private ContainerProgram _program = new ContainerProgram();
ProxyRule(RewriteDispatch rewriteDispatch, WebApp webApp)
{
super(rewriteDispatch);
_webApp = webApp;
}
public String getTagName()
{
return "proxy";
}
public void setTarget(String target)
{
_target = target;
}
public void addBuilderProgram(ConfigProgram program)
{
_program.addProgram(program);
}
@Override
public FilterChain dispatch(String uri,
String queryString,
FilterChain accept,
FilterChainMapper next)
throws ServletException
{
if (_target != null)
return new ProxyFilterChain(_servlet.createServletChain(),
_target, queryString);
else
return new ProxyFilterChain(_servlet.createServletChain(),
uri, queryString);
}
@Override
@PostConstruct
public void init()
throws ConfigException
{
super.init();
try {
_servlet = new ServletConfigImpl();
_servlet.setServletName("resin-dispatch-lb");
Class.forName("com.caucho.servlets.HttpProxyServlet");
_servlet.setServletClass("com.caucho.servlets.HttpProxyServlet");
_servlet.setInit(_program);
_webApp.addServlet(_servlet);
}
catch (ServletException ex) {
throw ConfigException.create(ex);
}
catch (ClassNotFoundException e) {
log.log(Level.FINER, e.toString(), e);
throw new ConfigException(L.l("load-balance requires Resin Professional"));
}
}
public static class ProxyFilterChain implements FilterChain {
private final FilterChain _next;
private final String _uri;
private final String _queryString;
ProxyFilterChain(FilterChain next, String uri, String queryString)
{
_next = next;
_uri = uri;
_queryString = queryString;
}
public void doFilter(ServletRequest req,
ServletResponse res)
throws IOException, ServletException
{
_next.doFilter(new ProxyRequest(req, _uri, _queryString), res);
}
}
public static class ProxyRequest extends HttpServletRequestWrapper {
private String _uri;
private String _queryString;
ProxyRequest(ServletRequest req,
String uri,
String queryString)
{
super((HttpServletRequest) req);
_uri = uri;
_queryString = queryString;
}
/**
* Returns the proxy uri
*/
public String getRequestURI()
{
return _uri;
}
/**
* Returns the proxy query string
*/
public String getQueryString()
{
return _queryString;
}
@Override
public Object getAttribute(String name)
{
return null;
}
@Override
public String toString()
{
return getClass().getSimpleName() + "[" + _uri + "?" + _queryString + "]";
}
}
}