diff --git a/guacamole/src/main/java/org/apache/guacamole/GuacamoleSession.java b/guacamole/src/main/java/org/apache/guacamole/GuacamoleSession.java index 601b2917b..06530c2f1 100644 --- a/guacamole/src/main/java/org/apache/guacamole/GuacamoleSession.java +++ b/guacamole/src/main/java/org/apache/guacamole/GuacamoleSession.java @@ -23,11 +23,11 @@ import java.util.Collections; import java.util.List; import java.util.Map; import java.util.concurrent.ConcurrentHashMap; -import org.apache.guacamole.GuacamoleException; import org.apache.guacamole.environment.Environment; import org.apache.guacamole.net.GuacamoleTunnel; import org.apache.guacamole.net.auth.AuthenticatedUser; import org.apache.guacamole.net.auth.UserContext; +import org.apache.guacamole.tunnel.StreamInterceptingTunnel; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -58,7 +58,8 @@ public class GuacamoleSession { /** * All currently-active tunnels, indexed by tunnel UUID. */ - private final Map tunnels = new ConcurrentHashMap(); + private final Map tunnels = + new ConcurrentHashMap(); /** * The last time this session was accessed. @@ -156,7 +157,7 @@ public class GuacamoleSession { * * @return A map of all active tunnels associated with this session. */ - public Map getTunnels() { + public Map getTunnels() { return tunnels; } @@ -166,7 +167,7 @@ public class GuacamoleSession { * * @param tunnel The tunnel to associate with this session. */ - public void addTunnel(GuacamoleTunnel tunnel) { + public void addTunnel(StreamInterceptingTunnel tunnel) { tunnels.put(tunnel.getUUID().toString(), tunnel); } diff --git a/guacamole/src/main/java/org/apache/guacamole/tunnel/TunnelRequestService.java b/guacamole/src/main/java/org/apache/guacamole/tunnel/TunnelRequestService.java index 3e35dc4b7..80404723a 100644 --- a/guacamole/src/main/java/org/apache/guacamole/tunnel/TunnelRequestService.java +++ b/guacamole/src/main/java/org/apache/guacamole/tunnel/TunnelRequestService.java @@ -23,13 +23,9 @@ import com.google.inject.Inject; import com.google.inject.Singleton; import java.util.List; import org.apache.guacamole.GuacamoleException; -import org.apache.guacamole.GuacamoleException; -import org.apache.guacamole.GuacamoleSecurityException; import org.apache.guacamole.GuacamoleSecurityException; import org.apache.guacamole.GuacamoleSession; import org.apache.guacamole.GuacamoleUnauthorizedException; -import org.apache.guacamole.GuacamoleUnauthorizedException; -import org.apache.guacamole.net.DelegatingGuacamoleTunnel; import org.apache.guacamole.net.GuacamoleTunnel; import org.apache.guacamole.net.auth.Connection; import org.apache.guacamole.net.auth.ConnectionGroup; @@ -242,7 +238,7 @@ public class TunnelRequestService { throws GuacamoleException { // Monitor tunnel closure and data - GuacamoleTunnel monitoredTunnel = new DelegatingGuacamoleTunnel(tunnel) { + StreamInterceptingTunnel monitoredTunnel = new StreamInterceptingTunnel(tunnel) { /** * The time the connection began, measured in milliseconds since