| // |
| // ======================================================================== |
| // Copyright (c) 1995-2016 Mort Bay Consulting Pty. Ltd. |
| // ------------------------------------------------------------------------ |
| // All rights reserved. This program and the accompanying materials |
| // are made available under the terms of the Eclipse Public License v1.0 |
| // and Apache License v2.0 which accompanies this distribution. |
| // |
| // The Eclipse Public License is available at |
| // http://www.eclipse.org/legal/epl-v10.html |
| // |
| // The Apache License v2.0 is available at |
| // http://www.opensource.org/licenses/apache2.0.php |
| // |
| // You may elect to redistribute this code under either of these licenses. |
| // ======================================================================== |
| // |
| |
| package org.eclipse.jetty.rhttp.gateway; |
| |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.util.HashMap; |
| import java.util.Map; |
| import java.util.concurrent.TimeUnit; |
| |
| import javax.servlet.http.HttpServletRequest; |
| import javax.servlet.http.HttpServletResponse; |
| |
| import org.eclipse.jetty.client.Address; |
| import org.eclipse.jetty.client.ContentExchange; |
| import org.eclipse.jetty.client.HttpClient; |
| import org.eclipse.jetty.client.HttpExchange; |
| import org.eclipse.jetty.io.Buffer; |
| import org.eclipse.jetty.io.ByteArrayBuffer; |
| import org.eclipse.jetty.rhttp.client.JettyClient; |
| import org.eclipse.jetty.rhttp.client.RHTTPClient; |
| import org.eclipse.jetty.rhttp.client.RHTTPListener; |
| import org.eclipse.jetty.rhttp.client.RHTTPRequest; |
| import org.eclipse.jetty.rhttp.client.RHTTPResponse; |
| import org.eclipse.jetty.server.Connector; |
| import org.eclipse.jetty.server.nio.SelectChannelConnector; |
| import org.eclipse.jetty.util.log.Log; |
| import org.eclipse.jetty.util.log.Logger; |
| |
| /** |
| * <p>This class combines a gateway server and a gateway client to obtain the functionality of a simple proxy server.</p> |
| * <p>This gateway proxy server starts on port 8080 and can be set as http proxy in browsers such as Firefox, and used |
| * to browse the internet.</p> |
| * <p>Its functionality is limited (for example, it only supports http, and not https).</p> |
| * @version $Revision$ $Date$ |
| */ |
| public class GatewayProxyServer |
| { |
| private static final Logger logger = Log.getLogger(GatewayProxyServer.class.toString()); |
| |
| public static void main(String[] args) throws Exception |
| { |
| GatewayServer server = new GatewayServer(); |
| |
| Connector plainConnector = new SelectChannelConnector(); |
| plainConnector.setPort(8080); |
| server.addConnector(plainConnector); |
| |
| ((StandardGateway)server.getGateway()).setExternalTimeout(180000); |
| ((StandardGateway)server.getGateway()).setGatewayTimeout(20000); |
| server.setTargetIdRetriever(new ProxyTargetIdRetriever()); |
| server.start(); |
| |
| HttpClient httpClient = new HttpClient(); |
| httpClient.setConnectorType(HttpClient.CONNECTOR_SOCKET); |
| httpClient.start(); |
| |
| RHTTPClient client = new JettyClient(httpClient, new Address("localhost", plainConnector.getPort()), server.getContext().getContextPath() + "/gw", "proxy"); |
| client.addListener(new ProxyListener(httpClient, client)); |
| client.connect(); |
| |
| Runtime.getRuntime().addShutdownHook(new Shutdown(server, httpClient, client)); |
| logger.info("{} started", GatewayProxyServer.class.getSimpleName()); |
| } |
| |
| private static class Shutdown extends Thread |
| { |
| private final GatewayServer server; |
| private final HttpClient httpClient; |
| private final RHTTPClient client; |
| |
| public Shutdown(GatewayServer server, HttpClient httpClient, RHTTPClient client) |
| { |
| this.server = server; |
| this.httpClient = httpClient; |
| this.client = client; |
| } |
| |
| @Override |
| public void run() |
| { |
| try |
| { |
| client.disconnect(); |
| httpClient.stop(); |
| server.stop(); |
| logger.info("{} stopped", GatewayProxyServer.class.getSimpleName()); |
| } |
| catch (Exception x) |
| { |
| logger.debug("Exception while stopping " + GatewayProxyServer.class.getSimpleName(), x); |
| } |
| } |
| } |
| |
| private static class ProxyListener implements RHTTPListener |
| { |
| private final HttpClient httpClient; |
| private final RHTTPClient client; |
| |
| private ProxyListener(HttpClient httpClient, RHTTPClient client) |
| { |
| this.httpClient = httpClient; |
| this.client = client; |
| } |
| |
| public void onRequest(RHTTPRequest request) throws Exception |
| { |
| ProxyExchange exchange = new ProxyExchange(); |
| Address address = Address.from(request.getHeaders().get("Host")); |
| if (address.getPort() == 0) address = new Address(address.getHost(), 80); |
| exchange.setAddress(address); |
| exchange.setMethod(request.getMethod()); |
| exchange.setURI(request.getURI()); |
| for (Map.Entry<String, String> header : request.getHeaders().entrySet()) |
| exchange.setRequestHeader(header.getKey(), header.getValue()); |
| exchange.setRequestContent(new ByteArrayBuffer(request.getBody())); |
| int status = syncSend(exchange); |
| if (status == HttpExchange.STATUS_COMPLETED) |
| { |
| int statusCode = exchange.getResponseStatus(); |
| String statusMessage = exchange.getResponseMessage(); |
| Map<String, String> responseHeaders = exchange.getResponseHeaders(); |
| byte[] responseBody = exchange.getResponseBody(); |
| RHTTPResponse response = new RHTTPResponse(request.getId(), statusCode, statusMessage, responseHeaders, responseBody); |
| client.deliver(response); |
| } |
| else |
| { |
| int statusCode = HttpServletResponse.SC_SERVICE_UNAVAILABLE; |
| String statusMessage = "Gateway error"; |
| HashMap<String, String> responseHeaders = new HashMap<String, String>(); |
| responseHeaders.put("Connection", "close"); |
| byte[] responseBody = new byte[0]; |
| RHTTPResponse response = new RHTTPResponse(request.getId(), statusCode, statusMessage, responseHeaders, responseBody); |
| client.deliver(response); |
| } |
| } |
| |
| private int syncSend(ProxyExchange exchange) throws Exception |
| { |
| long start = System.nanoTime(); |
| httpClient.send(exchange); |
| int status = exchange.waitForDone(); |
| long end = System.nanoTime(); |
| long millis = TimeUnit.NANOSECONDS.toMillis(end - start); |
| long micros = TimeUnit.NANOSECONDS.toMicros(end - start - TimeUnit.MILLISECONDS.toNanos(millis)); |
| logger.debug("Proxied request took {}.{} ms", millis, micros); |
| return status; |
| } |
| } |
| |
| private static class ProxyExchange extends ContentExchange |
| { |
| private String responseMessage; |
| private Map<String, String> responseHeaders = new HashMap<String, String>(); |
| private ByteArrayOutputStream responseBody = new ByteArrayOutputStream(); |
| |
| private ProxyExchange() |
| { |
| super(true); |
| } |
| |
| public String getResponseMessage() |
| { |
| return responseMessage; |
| } |
| |
| public Map<String, String> getResponseHeaders() |
| { |
| return responseHeaders; |
| } |
| |
| public byte[] getResponseBody() |
| { |
| return responseBody.toByteArray(); |
| } |
| |
| @Override |
| protected void onResponseStatus(Buffer version, int code, Buffer message) throws IOException |
| { |
| super.onResponseStatus(version, code, message); |
| this.responseMessage = message.toString("UTF-8"); |
| } |
| |
| @Override |
| protected void onResponseHeader(Buffer nameBuffer, Buffer valueBuffer) throws IOException |
| { |
| super.onResponseHeader(nameBuffer, valueBuffer); |
| String name = nameBuffer.toString("UTF-8"); |
| String value = valueBuffer.toString("UTF-8"); |
| // Skip chunked header, since we read the whole body and will not re-chunk it |
| if (!name.equalsIgnoreCase("Transfer-Encoding") || !value.equalsIgnoreCase("chunked")) |
| responseHeaders.put(name, value); |
| } |
| |
| @Override |
| protected void onResponseContent(Buffer buffer) throws IOException |
| { |
| responseBody.write(buffer.asArray()); |
| super.onResponseContent(buffer); |
| } |
| } |
| |
| public static class ProxyTargetIdRetriever implements TargetIdRetriever |
| { |
| public String retrieveTargetId(HttpServletRequest httpRequest) |
| { |
| return "proxy"; |
| } |
| } |
| } |