/**
* Send the ConnectInfo associated with this name
*
* @param name the user wanted
*/
private void getConnectInfo(String name, ObjectConnection oc) {
ConnectInfo ci = getConnectInfo(name);
if (ci != null) {
try {
oc.write(ci);
} catch (Exception e) {
}
} else {
try {
oc.write("unknown");
} catch (Exception e) {
}
}
}
/**
* Send the user's startup plugin
*
* @param source the user
*/
private void getStartupPlugin(ObjectConnection oc, String source) {
try {
UserConcept user = store.getUserStore().getUser(source);
oc.write(user.getStartupPlugin());
} catch (Exception e) {
}
}
/**
* Send a message through the network
*
* @param dest the receiver
* @param app the application that need to process this message
* @param data the data to send
*/
public ObjectConnection sendMessageTo(ConnectInfo dest, String app, String data)
throws IOException {
Socket sock = new Socket(dest.hostname, dest.port);
ObjectConnection oc = new ObjectConnection(sock);
Message msg = new Message(myConnectInfo, dest, app, data);
byte[] signature = null;
try {
signature = this.signer.sign(msg);
} catch (SignatureException e) {
Logging.getLogger().warning("Unable to sign: " + e);
}
oc.write(msg);
oc.write(signature);
return oc;
}
/** Send the user list to all users */
public void sendUserList() {
String data = "";
Socket sock = null;
ObjectConnection oc = null;
byte[] signature = {};
/* receiver list */
for (int i = 0; i < this.connections.size(); i++) {
if (((ConnectInfo) this.connections.get(i)).type.equalsIgnoreCase("Client")) {
data = "";
/* users list */
for (int j = 0; j < this.connections.size(); j++) {
if (((ConnectInfo) this.connections.get(j)).type.equalsIgnoreCase("Client"))
data = data + " " + ((ConnectInfo) this.connections.get(j)).getName();
}
try {
sock =
new Socket(
((ConnectInfo) this.connections.get(i)).hostname,
((ConnectInfo) this.connections.get(i)).port);
oc = new ObjectConnection(sock);
Message msg =
new Message(
myConnectInfo,
(ConnectInfo) this.connections.get(i),
"Client",
"USER_LIST " + data);
try {
signature = this.signer.sign(msg);
} catch (SignatureException e) {
Logging.getLogger().warning("Unable to sign: " + e);
}
oc.write(msg);
oc.write(signature);
oc.close();
} catch (IOException ex) {
Logging.getLogger().warning("Unable to connect to host");
continue;
}
}
}
}
/** Send the connected users list */
private void getUserList(ObjectConnection oc) {
Vector v = new Vector();
for (int i = 0; i < this.connections.size(); i++) {
if (((ConnectInfo) this.connections.get(i)).type.equalsIgnoreCase("Client"))
v.addElement(this.connections.get(i));
}
try {
oc.write(v);
} catch (Exception e) {
}
}
/**
* Send a plugin JAR file to the client
*
* @param data the plugin name
*/
private void sendPlugin(ObjectConnection oc, String data) {
DataInputStream dis = null;
try {
dis = new DataInputStream(new FileInputStream(APPLICATIONS_DIRECTORY + data + ".jar"));
byte[] buf = new byte[dis.available()];
dis.readFully(buf);
oc.write(buf);
} catch (Exception e) {
Logging.getLogger().warning("Unable to send the file: " + data + ".jar");
} finally {
if (dis != null) {
try {
dis.close();
} catch (IOException ioe) {
}
}
}
}
/** Handle internal commands */
private void internalCommand(ObjectConnection oc, Message message, String command, String data) {
if (command.equals("CONNECT_DEL")) {
try {
oc.write("OK");
} catch (Exception e) {
}
this.removeConnectInfo(message.getSender());
} else if (command.equals("CONNECT_GET")) {
try {
oc.write("OK");
} catch (Exception e) {
}
this.getConnectInfo(data, oc);
} else if (command.equals("CONNECT_LIST")) {
try {
oc.write("OK");
} catch (Exception e) {
}
this.getUserList(oc);
} else if (command.equals("PLUGIN_LIST")) {
try {
oc.write("OK");
} catch (Exception e) {
}
this.sendPluginList(oc, message.getSender().getName());
} else if (command.equals("PLUGIN_GET")) {
try {
oc.write("OK");
} catch (Exception e) {
}
this.sendPlugin(oc, data);
} else if (command.equals("STARTUP_PLUGINS")) {
try {
oc.write("OK");
} catch (Exception e) {
}
this.getStartupPlugin(oc, message.getSender().getName());
} else {
try {
oc.write("FAILED Unknown command");
} catch (Exception e) {
}
}
}
/**
* Send the plugin list to a client. The list depends of the client's groups
*
* @param source the user that asked this command
*/
private void sendPluginList(ObjectConnection oc, String source) {
Vector plist = new Vector();
Iterator plugins;
String line = "";
try {
UserConcept user = store.getUserStore().getUser(source);
plugins = store.getPluginStore().getAuthorizedPlugins(user);
while (plugins.hasNext()) {
PluginConcept plugin = (PluginConcept) plugins.next();
line = plugin.getName();
line += " " + plugin.getVersion();
plist.add(line);
}
oc.write(plist);
} catch (Exception e) {
Logging.getLogger().warning("Unable to send the plugin list.");
e.printStackTrace();
}
}
/**
* Authenticate users
*
* @param oc the streams
* @param ci the ConnectInfo
*/
private void authentification(ObjectConnection oc, Message message, String data) {
String passwd = null;
String name = null;
String authenticationServer = null;
String hostname = null;
int port = 0;
StringTokenizer stk = new StringTokenizer(data, " ");
try {
passwd = stk.nextToken();
hostname = stk.nextToken();
port = Integer.parseInt(stk.nextToken());
} catch (Exception ex) {
Logging.getLogger().warning("#Err > Incorrect authentication message.");
try {
oc.write("BAD_MESSAGE");
} catch (Exception e) {
}
return;
}
name = message.getSender().getName();
authenticationServer = message.getSender().getAuthenticationServer();
UserConcept user = null;
try {
user = store.getUserStore().getUser(message.getSender().getName());
} catch (Exception e) {
e.printStackTrace();
}
// password ok
if (user != null && store.getUserStore().checkUserPassword(user, passwd)) {
// disconnect already connected user
if (isAlreadyKnown(message.getSender())) {
ConnectInfo oldUser = getCompleteConnectInfo(message.getSender());
try {
ObjectConnection myoc = this.sendMessageTo(oldUser, "Client", "DISCONNECT");
myoc.close();
} catch (Exception e) {
// we can't do much here, the client might have crashed
}
this.removeConnectInfo(oldUser);
}
connections.add(
new ConnectInfo(
name, authenticationServer, hostname, port, user.getPublicKey(), "Client"));
try {
oc.write("AUTH_ACCEPTED " + user.getPrivateKey());
} catch (Exception e) {
e.printStackTrace();
}
this.sendUserList();
} else {
try {
oc.write("NOT_VALID_USER");
} catch (Exception e) {
}
}
}
/**
* Reads messages from the network. A message is either a command for the server or for an
* internal Service.
*
* @param sock the Socket
*/
private void getMessage(Socket sock) {
int i;
boolean alreadyConnected;
boolean isAuthentication;
Message message;
byte[] signature;
String cmd;
String cmdData;
StringTokenizer stk;
ObjectConnection oc = null;
try {
/* streams initialization */
oc = new ObjectConnection(sock);
message = (Message) oc.read();
signature = (byte[]) oc.read();
} catch (Exception ex) {
ex.printStackTrace();
Logging.getLogger().warning("#Err > Unable to read message.");
return;
}
// check wether a user is known or not
alreadyConnected = isAlreadyKnown(message.getSender());
// check if command is authentication
isAuthentication =
message.getApplication().equals("Server")
&& ((String) message.getData()).startsWith("AUTH");
// signature check
if (alreadyConnected && !isAuthentication) {
boolean sigok = false;
try {
ConnectInfo ci = message.getSender();
if (ci.verifier == null) ci = this.getCompleteConnectInfo(ci);
sigok = ci.verifier.verify(message, signature);
} catch (Exception e) {
e.printStackTrace();
}
if (!sigok) {
try {
oc.write("FAILED bad signature");
} catch (Exception e) {
}
Logging.getLogger().warning("#Err > bad signature: " + message.getSender());
return;
}
}
if (message.getApplication().equals("Server")) {
cmd = null;
try {
stk = new StringTokenizer((String) message.getData());
cmd = stk.nextToken();
cmdData = stk.nextToken("\0").substring(1);
} catch (Exception ex) {
if (cmd == null) cmd = "";
cmdData = "";
}
/* if the user asks for authentication, we try to do it and exits this method */
if (cmd.equals("AUTH")) {
try {
oc.write("OK");
} catch (Exception e) {
e.printStackTrace();
}
authentification(oc, message, cmdData);
} else if (!alreadyConnected) {
Logging.getLogger().info("Access denied to " + message.getSender());
try {
oc.write("FAILED No Connection");
} catch (Exception e) {
}
} else {
internalCommand(oc, message, cmd, cmdData);
}
} else if (!alreadyConnected) {
Logging.getLogger().info("Access denied to " + message.getSender());
try {
oc.write("FAILED No Connection");
} catch (Exception e) {
}
} else {
Service s;
/* seek the destination service */
boolean serviceFound = false;
for (i = 0; i < this.services.size(); i++) {
s = (Service) services.get(i);
if (s.getName().equalsIgnoreCase(message.getApplication())) {
serviceFound = true;
UserConcept user = null;
ServiceConcept service = null;
try {
user = store.getUserStore().getUser(message.getSender().getName());
service = store.getServiceStore().getService(message.getReceiver().getName());
} catch (Exception e) {
}
/* tests serviceManager for permissions */
boolean isAutorizedService = false;
try {
isAutorizedService = store.getServiceStore().isAuthorizedService(user, service);
} catch (Exception e) {
}
if (!isAutorizedService) {
Logging.getLogger()
.info(
"#Err > "
+ message.getSender()
+ " : Service denied to "
+ message.getReceiver().getName());
try {
oc.write("FAILED You don't have acces to this service");
} catch (Exception e) {
}
} else {
try {
oc.write("OK");
} catch (Exception e) {
}
serviceFound = true;
s.process(oc, message);
}
break;
}
}
if (!serviceFound) {
try {
oc.write("FAILED unknown");
} catch (Exception e) {
}
Logging.getLogger()
.warning("#Err > Service " + message.getReceiver().getName() + " unknown");
}
}
oc.close();
}
