mirror of
https://github.com/gyurix1968/guacamole-client.git
synced 2025-09-06 13:17:41 +00:00
GUAC-1427: Copy parameter values from request to new object since we can't keep the original request around.
This commit is contained in:
@@ -22,8 +22,11 @@
|
|||||||
|
|
||||||
package org.glyptodon.guacamole.net.basic;
|
package org.glyptodon.guacamole.net.basic;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
|
import java.util.HashMap;
|
||||||
import java.util.List;
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletRequest;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@@ -34,33 +37,54 @@ import javax.servlet.http.HttpServletRequest;
|
|||||||
public class HTTPTunnelRequest extends TunnelRequest {
|
public class HTTPTunnelRequest extends TunnelRequest {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* The wrapped HttpServletRequest.
|
* A copy of the parameters obtained from the HttpServletRequest used to
|
||||||
|
* construct the HttpTunnelRequest.
|
||||||
*/
|
*/
|
||||||
private final HttpServletRequest request;
|
private final Map<String, List<String>> parameterMap =
|
||||||
|
new HashMap<String, List<String>>();
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Creates a TunnelRequest implementation which delegates parameter and
|
* Creates a HTTPTunnelRequest which copies and exposes the parameters
|
||||||
* session retrieval to the given HttpServletRequest.
|
* from the given HttpServletRequest.
|
||||||
*
|
*
|
||||||
* @param request The HttpServletRequest to wrap.
|
* @param request
|
||||||
|
* The HttpServletRequest to copy parameter values from.
|
||||||
*/
|
*/
|
||||||
|
@SuppressWarnings("unchecked") // getParameterMap() is defined as returning Map<String, String[]>
|
||||||
public HTTPTunnelRequest(HttpServletRequest request) {
|
public HTTPTunnelRequest(HttpServletRequest request) {
|
||||||
this.request = request;
|
|
||||||
|
// For each parameter
|
||||||
|
for (Map.Entry<String, String[]> mapEntry : ((Map<String, String[]>)
|
||||||
|
request.getParameterMap()).entrySet()) {
|
||||||
|
|
||||||
|
// Get parameter name and corresponding values
|
||||||
|
String parameterName = mapEntry.getKey();
|
||||||
|
List<String> parameterValues = Arrays.asList(mapEntry.getValue());
|
||||||
|
|
||||||
|
// Store copy of all values in our own map
|
||||||
|
parameterMap.put(
|
||||||
|
parameterName,
|
||||||
|
new ArrayList<String>(parameterValues)
|
||||||
|
);
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public String getParameter(String name) {
|
public String getParameter(String name) {
|
||||||
return request.getParameter(name);
|
List<String> values = getParameterValues(name);
|
||||||
|
|
||||||
|
// Return the first value from the list if available
|
||||||
|
if (values != null && !values.isEmpty())
|
||||||
|
return values.get(0);
|
||||||
|
|
||||||
|
return null;
|
||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public List<String> getParameterValues(String name) {
|
public List<String> getParameterValues(String name) {
|
||||||
|
return parameterMap.get(name);
|
||||||
String[] values = request.getParameterValues(name);
|
|
||||||
if (values == null)
|
|
||||||
return null;
|
|
||||||
|
|
||||||
return Arrays.asList(values);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@@ -24,11 +24,10 @@ package org.glyptodon.guacamole.net.basic.websocket.jetty8;
|
|||||||
|
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
import javax.servlet.http.HttpServletRequest;
|
|
||||||
import org.glyptodon.guacamole.GuacamoleException;
|
import org.glyptodon.guacamole.GuacamoleException;
|
||||||
import org.glyptodon.guacamole.net.GuacamoleTunnel;
|
import org.glyptodon.guacamole.net.GuacamoleTunnel;
|
||||||
import org.glyptodon.guacamole.net.basic.TunnelRequestService;
|
import org.glyptodon.guacamole.net.basic.TunnelRequestService;
|
||||||
import org.glyptodon.guacamole.net.basic.HTTPTunnelRequest;
|
import org.glyptodon.guacamole.net.basic.TunnelRequest;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Tunnel servlet implementation which uses WebSocket as a tunnel backend,
|
* Tunnel servlet implementation which uses WebSocket as a tunnel backend,
|
||||||
@@ -45,9 +44,9 @@ public class BasicGuacamoleWebSocketTunnelServlet extends GuacamoleWebSocketTunn
|
|||||||
private TunnelRequestService tunnelRequestService;
|
private TunnelRequestService tunnelRequestService;
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected GuacamoleTunnel doConnect(HttpServletRequest request)
|
protected GuacamoleTunnel doConnect(TunnelRequest request)
|
||||||
throws GuacamoleException {
|
throws GuacamoleException {
|
||||||
return tunnelRequestService.createTunnel(new HTTPTunnelRequest(request));
|
return tunnelRequestService.createTunnel(request);
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@@ -33,6 +33,8 @@ import org.eclipse.jetty.websocket.WebSocket.Connection;
|
|||||||
import org.eclipse.jetty.websocket.WebSocketServlet;
|
import org.eclipse.jetty.websocket.WebSocketServlet;
|
||||||
import org.glyptodon.guacamole.GuacamoleClientException;
|
import org.glyptodon.guacamole.GuacamoleClientException;
|
||||||
import org.glyptodon.guacamole.GuacamoleConnectionClosedException;
|
import org.glyptodon.guacamole.GuacamoleConnectionClosedException;
|
||||||
|
import org.glyptodon.guacamole.net.basic.HTTPTunnelRequest;
|
||||||
|
import org.glyptodon.guacamole.net.basic.TunnelRequest;
|
||||||
import org.glyptodon.guacamole.protocol.GuacamoleStatus;
|
import org.glyptodon.guacamole.protocol.GuacamoleStatus;
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
import org.slf4j.LoggerFactory;
|
import org.slf4j.LoggerFactory;
|
||||||
@@ -70,7 +72,9 @@ public abstract class GuacamoleWebSocketTunnelServlet extends WebSocketServlet {
|
|||||||
}
|
}
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
public WebSocket doWebSocketConnect(final HttpServletRequest request, String protocol) {
|
public WebSocket doWebSocketConnect(HttpServletRequest request, String protocol) {
|
||||||
|
|
||||||
|
final TunnelRequest tunnelRequest = new HTTPTunnelRequest(request);
|
||||||
|
|
||||||
// Return new WebSocket which communicates through tunnel
|
// Return new WebSocket which communicates through tunnel
|
||||||
return new WebSocket.OnTextMessage() {
|
return new WebSocket.OnTextMessage() {
|
||||||
@@ -109,7 +113,7 @@ public abstract class GuacamoleWebSocketTunnelServlet extends WebSocketServlet {
|
|||||||
public void onOpen(final Connection connection) {
|
public void onOpen(final Connection connection) {
|
||||||
|
|
||||||
try {
|
try {
|
||||||
tunnel = doConnect(request);
|
tunnel = doConnect(tunnelRequest);
|
||||||
}
|
}
|
||||||
catch (GuacamoleException e) {
|
catch (GuacamoleException e) {
|
||||||
logger.error("Creation of WebSocket tunnel to guacd failed: {}", e.getMessage());
|
logger.error("Creation of WebSocket tunnel to guacd failed: {}", e.getMessage());
|
||||||
@@ -209,7 +213,7 @@ public abstract class GuacamoleWebSocketTunnelServlet extends WebSocketServlet {
|
|||||||
* result of this connection request (whether some sort of credentials must
|
* result of this connection request (whether some sort of credentials must
|
||||||
* be specified, for example).
|
* be specified, for example).
|
||||||
*
|
*
|
||||||
* @param request The HttpServletRequest associated with the connection
|
* @param request The TunnelRequest associated with the connection
|
||||||
* request received. Any parameters specified along with
|
* request received. Any parameters specified along with
|
||||||
* the connection request can be read from this object.
|
* the connection request can be read from this object.
|
||||||
* @return A newly constructed GuacamoleTunnel if successful,
|
* @return A newly constructed GuacamoleTunnel if successful,
|
||||||
@@ -218,7 +222,7 @@ public abstract class GuacamoleWebSocketTunnelServlet extends WebSocketServlet {
|
|||||||
* GuacamoleTunnel, or if the conditions
|
* GuacamoleTunnel, or if the conditions
|
||||||
* required for connection are not met.
|
* required for connection are not met.
|
||||||
*/
|
*/
|
||||||
protected abstract GuacamoleTunnel doConnect(HttpServletRequest request)
|
protected abstract GuacamoleTunnel doConnect(TunnelRequest request)
|
||||||
throws GuacamoleException;
|
throws GuacamoleException;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@@ -24,11 +24,10 @@ package org.glyptodon.guacamole.net.basic.websocket.tomcat;
|
|||||||
|
|
||||||
import com.google.inject.Inject;
|
import com.google.inject.Inject;
|
||||||
import com.google.inject.Singleton;
|
import com.google.inject.Singleton;
|
||||||
import javax.servlet.http.HttpServletRequest;
|
|
||||||
import org.glyptodon.guacamole.GuacamoleException;
|
import org.glyptodon.guacamole.GuacamoleException;
|
||||||
import org.glyptodon.guacamole.net.GuacamoleTunnel;
|
import org.glyptodon.guacamole.net.GuacamoleTunnel;
|
||||||
import org.glyptodon.guacamole.net.basic.TunnelRequestService;
|
import org.glyptodon.guacamole.net.basic.TunnelRequestService;
|
||||||
import org.glyptodon.guacamole.net.basic.HTTPTunnelRequest;
|
import org.glyptodon.guacamole.net.basic.TunnelRequest;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Tunnel servlet implementation which uses WebSocket as a tunnel backend,
|
* Tunnel servlet implementation which uses WebSocket as a tunnel backend,
|
||||||
@@ -45,9 +44,9 @@ public class BasicGuacamoleWebSocketTunnelServlet extends GuacamoleWebSocketTunn
|
|||||||
private TunnelRequestService tunnelRequestService;
|
private TunnelRequestService tunnelRequestService;
|
||||||
|
|
||||||
@Override
|
@Override
|
||||||
protected GuacamoleTunnel doConnect(HttpServletRequest request)
|
protected GuacamoleTunnel doConnect(TunnelRequest request)
|
||||||
throws GuacamoleException {
|
throws GuacamoleException {
|
||||||
return tunnelRequestService.createTunnel(new HTTPTunnelRequest(request));
|
return tunnelRequestService.createTunnel(request);
|
||||||
};
|
};
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@@ -38,6 +38,8 @@ import org.apache.catalina.websocket.WebSocketServlet;
|
|||||||
import org.apache.catalina.websocket.WsOutbound;
|
import org.apache.catalina.websocket.WsOutbound;
|
||||||
import org.glyptodon.guacamole.GuacamoleClientException;
|
import org.glyptodon.guacamole.GuacamoleClientException;
|
||||||
import org.glyptodon.guacamole.GuacamoleConnectionClosedException;
|
import org.glyptodon.guacamole.GuacamoleConnectionClosedException;
|
||||||
|
import org.glyptodon.guacamole.net.basic.HTTPTunnelRequest;
|
||||||
|
import org.glyptodon.guacamole.net.basic.TunnelRequest;
|
||||||
import org.glyptodon.guacamole.protocol.GuacamoleStatus;
|
import org.glyptodon.guacamole.protocol.GuacamoleStatus;
|
||||||
import org.slf4j.Logger;
|
import org.slf4j.Logger;
|
||||||
import org.slf4j.LoggerFactory;
|
import org.slf4j.LoggerFactory;
|
||||||
@@ -93,7 +95,9 @@ public abstract class GuacamoleWebSocketTunnelServlet extends WebSocketServlet {
|
|||||||
|
|
||||||
@Override
|
@Override
|
||||||
public StreamInbound createWebSocketInbound(String protocol,
|
public StreamInbound createWebSocketInbound(String protocol,
|
||||||
final HttpServletRequest request) {
|
HttpServletRequest request) {
|
||||||
|
|
||||||
|
final TunnelRequest tunnelRequest = new HTTPTunnelRequest(request);
|
||||||
|
|
||||||
// Return new WebSocket which communicates through tunnel
|
// Return new WebSocket which communicates through tunnel
|
||||||
return new StreamInbound() {
|
return new StreamInbound() {
|
||||||
@@ -137,7 +141,7 @@ public abstract class GuacamoleWebSocketTunnelServlet extends WebSocketServlet {
|
|||||||
public void onOpen(final WsOutbound outbound) {
|
public void onOpen(final WsOutbound outbound) {
|
||||||
|
|
||||||
try {
|
try {
|
||||||
tunnel = doConnect(request);
|
tunnel = doConnect(tunnelRequest);
|
||||||
}
|
}
|
||||||
catch (GuacamoleException e) {
|
catch (GuacamoleException e) {
|
||||||
logger.error("Creation of WebSocket tunnel to guacd failed: {}", e.getMessage());
|
logger.error("Creation of WebSocket tunnel to guacd failed: {}", e.getMessage());
|
||||||
@@ -242,16 +246,20 @@ public abstract class GuacamoleWebSocketTunnelServlet extends WebSocketServlet {
|
|||||||
* result of this connection request (whether some sort of credentials must
|
* result of this connection request (whether some sort of credentials must
|
||||||
* be specified, for example).
|
* be specified, for example).
|
||||||
*
|
*
|
||||||
* @param request The HttpServletRequest associated with the connection
|
* @param request
|
||||||
* request received. Any parameters specified along with
|
* The TunnelRequest associated with the connection request received.
|
||||||
* the connection request can be read from this object.
|
* Any parameters specified along with the connection request can be
|
||||||
* @return A newly constructed GuacamoleTunnel if successful,
|
* read from this object.
|
||||||
* null otherwise.
|
*
|
||||||
* @throws GuacamoleException If an error occurs while constructing the
|
* @return
|
||||||
* GuacamoleTunnel, or if the conditions
|
* A newly constructed GuacamoleTunnel if successful, null otherwise.
|
||||||
* required for connection are not met.
|
*
|
||||||
|
* @throws GuacamoleException
|
||||||
|
* If an error occurs while constructing the GuacamoleTunnel, or if the
|
||||||
|
* conditions required for connection are not met.
|
||||||
*/
|
*/
|
||||||
protected abstract GuacamoleTunnel doConnect(HttpServletRequest request) throws GuacamoleException;
|
protected abstract GuacamoleTunnel doConnect(TunnelRequest request)
|
||||||
|
throws GuacamoleException;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user