GUACAMOLE-47: Implement support for client hostname/IP token for connections.

This commit is contained in:
Nick Couchman
2017-01-06 19:24:50 -05:00
parent d45580f226
commit 797bb86a43
2 changed files with 60 additions and 0 deletions

View File

@@ -41,6 +41,16 @@ public class APIRequest extends HttpServletRequestWrapper {
*/
private final Map<String, String[]> parameters;
/**
* The remote hostname that initiated the request.
*/
private final String remoteHost;
/**
* The remote ip address that initiated the request.
*/
private final String remoteAddr;
/**
* Wraps the given HttpServletRequest, using the given MultivaluedMap to
* provide all request parameters. All HttpServletRequest functions which
@@ -58,6 +68,29 @@ public class APIRequest extends HttpServletRequestWrapper {
super(request);
// Try a few methods to get client info.
String clientHostname = "";
String clientAddress = "";
if(request.getHeader("X-Guacamole-Client-Hostname") != "") {
this.remoteHost = request.getHeader("X-Guacamole-Client-Hostname");
} else if(request.getHeader("X-Forwarded-For") != "") {
this.remoteHost = request.getHeader("X-Forwarded-For");
} else if(request.getRemoteHost() != "") {
this.remoteHost = request.getRemoteHost();
} else {
this.remoteHost = "";
}
if(request.getHeader("X-Guacamole-Client-IP") != "") {
this.remoteAddr = request.getHeader("X-Guacamole-Client-IP");
} else if(request.getHeader("X-Forwarded-For") != "") {
this.remoteAddr = request.getHeader("X-Forwarded-For");
} else if(request.getRemoteAddr() != "") {
this.remoteAddr = request.getRemoteAddr();
} else {
this.remoteAddr = "";
}
// Copy parameters from given MultivaluedMap
this.parameters = new HashMap<String, String[]>(parameters.size());
for (Map.Entry<String, List<String>> entry : parameters.entrySet()) {
@@ -101,4 +134,14 @@ public class APIRequest extends HttpServletRequestWrapper {
}
@Override
public String getRemoteHost() {
return this.remoteHost;
}
@Override
public String getRemoteAddr() {
return this.remoteAddr;
}
}