blob: 2fcc6a6cfcd1180b0b9209149d53a1930f6bf03e (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
package de.bwlehrpool.bwlp_guac;
import org.apache.guacamole.GuacamoleException;
import org.apache.guacamole.net.auth.AbstractUserContext;
import org.apache.guacamole.net.auth.AuthenticatedUser;
import org.apache.guacamole.net.auth.AuthenticationProvider;
import org.apache.guacamole.net.auth.Connection;
import org.apache.guacamole.net.auth.Directory;
import org.apache.guacamole.net.auth.User;
import org.apache.guacamole.net.auth.UserContext;
import org.apache.guacamole.net.auth.credentials.GuacamoleCredentialsException;
import org.apache.guacamole.net.auth.permission.ObjectPermissionSet;
import org.apache.guacamole.net.auth.simple.SimpleConnection;
import org.apache.guacamole.net.auth.simple.SimpleDirectory;
import org.apache.guacamole.net.auth.simple.SimpleObjectPermissionSet;
import org.apache.guacamole.net.auth.simple.SimpleUser;
import org.apache.guacamole.protocol.GuacamoleConfiguration;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class BwlpUserContext extends AbstractUserContext {
private static final Logger LOGGER = LoggerFactory.getLogger(BwlpUserContext.class);
private static final SimpleConnection FAKE = new SimpleConnection("FAKE", "FAKE", new GuacamoleConfiguration());
static {
FAKE.setParentIdentifier(DEFAULT_ROOT_CONNECTION_GROUP);
}
private final AuthenticatedUser authUser;
private final UserContext originalContext;
private final Integer groupid;
/**
* The Directory with access to all connections within the root group associated
* with this UserContext.
*/
private Directory<Connection> connectionDirectory;
public BwlpUserContext(AuthenticatedUser authenticatedUser, UserContext context, int groupid)
throws GuacamoleCredentialsException {
authUser = authenticatedUser;
originalContext = context;
this.groupid = groupid;
// OK
addConn();
}
public BwlpUserContext(AuthenticatedUser authenticatedUser, UserContext context, WrappedConnection exConn) {
authUser = authenticatedUser;
originalContext = context;
this.groupid = -1;
connectionDirectory = new SimpleDirectory<Connection>(exConn);
}
private void addConn() throws GuacamoleCredentialsException {
WrappedConnection connection = ConnectionManager.getForUser(authUser.getCredentials().getUsername(), groupid);
if (connection != null) {
connection.setContext(this);
connectionDirectory = new SimpleDirectory<Connection>(connection);
} else {
connectionDirectory = new SimpleDirectory<Connection>();
}
}
public User self() {
return new SimpleUser(authUser.getCredentials().getUsername()) {
@Override
public ObjectPermissionSet getConnectionGroupPermissions() throws GuacamoleException {
return new SimpleObjectPermissionSet(getConnectionDirectory().getIdentifiers());
}
@Override
public ObjectPermissionSet getConnectionPermissions() throws GuacamoleException {
return new SimpleObjectPermissionSet(getConnectionGroupDirectory().getIdentifiers());
}
};
}
@Override
public Object getResource() throws GuacamoleException {
return null;
}
public AuthenticationProvider getAuthenticationProvider() {
return originalContext.getAuthenticationProvider();
}
@Override
public Directory<Connection> getConnectionDirectory() throws GuacamoleException {
return connectionDirectory;
}
public void clearConnections() {
connectionDirectory = new SimpleDirectory<Connection>();
}
public boolean hasValidConnection() {
boolean ok = false;
try {
synchronized (this) {
for (String id : connectionDirectory.getIdentifiers()) {
Connection con = connectionDirectory.get(id);
if (con instanceof WrappedConnection) {
LOGGER.info("Checking connection for " + con.getIdentifier());
if (((WrappedConnection) con).checkConnection(3)) {
ok = true;
}
} else {
}
}
}
} catch (Exception e) {
LOGGER.warn("hasValidConnection", e);
}
if (!ok) connectionDirectory = new SimpleDirectory<Connection>();
return ok;
}
public UserContext getOriginalContext() {
return originalContext;
}
}
|