private synchronized void addNewAgent(
int agentId,
SocketChannel socket,
String agentName,
String agentIP,
int agentPort,
int flags) {
if (agentIP.equals(":same")) {
InetAddress agentAddress = socket.socket().getInetAddress();
agentIP = agentAddress.getHostAddress();
}
Log.info(
"New agent id="
+ agentId
+ " name="
+ agentName
+ " address="
+ agentIP
+ ":"
+ agentPort
+ " flags="
+ flags);
AgentInfo agentInfo = new AgentInfo();
agentInfo.agentId = agentId;
agentInfo.flags = flags;
agentInfo.socket = socket;
agentInfo.agentName = agentName;
agentInfo.agentIP = agentIP;
agentInfo.agentPort = agentPort;
agentInfo.outputBuf = new MVByteBuffer(1024);
agentInfo.inputBuf = new MVByteBuffer(1024);
agents.put(socket, agentInfo);
NewAgentMessage newAgentMessage =
new NewAgentMessage(agentId, agentName, agentIP, agentPort, flags);
for (Map.Entry<SocketChannel, AgentInfo> entry : agents.entrySet()) {
if (entry.getKey() == socket) continue;
// Tell other agents about the new one
synchronized (entry.getValue().outputBuf) {
Message.toBytes(newAgentMessage, entry.getValue().outputBuf);
}
// Tell new agent about other agents
NewAgentMessage otherAgentMessage =
new NewAgentMessage(
entry.getValue().agentId,
entry.getValue().agentName,
entry.getValue().agentIP,
entry.getValue().agentPort,
entry.getValue().flags);
synchronized (agentInfo.outputBuf) {
Message.toBytes(otherAgentMessage, agentInfo.outputBuf);
}
}
messageIO.addAgent(agentInfo);
messageIO.outputReady();
}
public boolean handleMessage() throws java.io.IOException {
ByteBuffer buf = ByteBuffer.allocate(4);
int nBytes = ChannelUtil.fillBuffer(buf, agentSocket);
if (nBytes == 0) {
Log.info("DomainServer: agent closed connection " + agentSocket);
return false;
}
if (nBytes < 4) {
Log.error("DomainServer: invalid message " + nBytes);
return false;
}
int msgLen = buf.getInt();
if (msgLen < 0) {
return false;
}
MVByteBuffer buffer = new MVByteBuffer(msgLen);
nBytes = ChannelUtil.fillBuffer(buffer.getNioBuf(), agentSocket);
if (nBytes == 0) {
Log.info("DomainServer: agent closed connection " + agentSocket);
return false;
}
if (nBytes < msgLen) {
Log.error(
"DomainServer: invalid message, expecting "
+ msgLen
+ " got "
+ nBytes
+ " from "
+ agentSocket);
return false;
}
Message message = (Message) MarshallingRuntime.unmarshalObject(buffer);
if (message instanceof AgentHelloMessage) {
// Successfully added agent, clear our socket var
// so we don't close it.
if (handleAgentHello((AgentHelloMessage) message)) agentSocket = null;
return false;
} else if (message instanceof AllocNameMessage)
handleAllocName((AllocNameMessage) message, agentSocket);
return true;
}
public void run() {
try {
while (handleMessage()) {}
} catch (InterruptedIOException ex) {
Log.info("DomainServer: closed connection due to timeout " + agentSocket);
} catch (IOException ex) {
Log.info("DomainServer: agent closed connection " + agentSocket);
} catch (Exception ex) {
Log.exception("DomainServer.SocketHandler: ", ex);
}
try {
if (agentSocket != null) agentSocket.close();
} catch (IOException ex) {
/* ignore */
}
}
private static void saveProcessID(String svrName, String runDir) {
Log.info("Server Name is " + svrName + " Run Dir is " + runDir);
RuntimeMXBean rt = ManagementFactory.getRuntimeMXBean();
String pid = rt.getName();
if (Log.loggingDebug) {
Log.info("PROCESS ID IS " + pid);
Log.info("server name is " + svrName);
}
try {
if (runDir != null) {
File outFile = new File(runDir + "\\" + svrName + ".bat");
PrintWriter out = new PrintWriter(new FileWriter(outFile));
out.println("set pid=" + pid.substring(0, pid.indexOf("@")));
out.close();
}
} catch (IOException e) {
Log.exception("saveProcessID caught exception", e);
}
}
public void handleMessageData(int length, MVByteBuffer messageData, AgentInfo agentInfo) {
if (length == -1 || messageData == null) {
if ((agentInfo.flags & MessageAgent.DOMAIN_FLAG_TRANSIENT) != 0) {
Log.info("Lost connection to '" + agentInfo.agentName + "' (transient)");
agents.remove(agentInfo.socket);
agentNames.remove(agentInfo.agentName);
messageIO.removeAgent(agentInfo);
} else Log.info("Lost connection to '" + agentInfo.agentName + "'");
try {
agentInfo.socket.close();
} catch (java.io.IOException ex) {
Log.exception("close", ex);
}
agentInfo.socket = null;
// ## clear buffers
// ## keep agentInfo? to preserve agentId?
return;
}
Message message = (Message) MarshallingRuntime.unmarshalObject(messageData);
MessageType msgType = message.getMsgType();
if (Log.loggingDebug)
Log.debug(
"handleMessageData from "
+ agentInfo.agentName
+ ","
+ message.getMsgId()
+ " type="
+ msgType.getMsgTypeString()
+ " len="
+ length
+ " class="
+ message.getClass().getName());
try {
if (message instanceof AllocNameMessage)
handleAllocName((AllocNameMessage) message, agentInfo.socket);
else if (message instanceof AwaitPluginDependentsMessage)
handleAwaitPluginDependents((AwaitPluginDependentsMessage) message, agentInfo.socket);
else if (message instanceof PluginAvailableMessage)
handlePluginAvailable((PluginAvailableMessage) message, agentInfo.socket);
else
Log.error(
"Unsupported message from "
+ agentInfo.agentName
+ ","
+ message.getMsgId()
+ " type="
+ msgType.getMsgTypeString()
+ " len="
+ length
+ " class="
+ message.getClass().getName());
} catch (java.io.IOException e) {
Log.error(
"IO error on message from "
+ agentInfo.agentName
+ ","
+ message.getMsgId()
+ " type="
+ msgType.getMsgTypeString()
+ " len="
+ length
+ " class="
+ message.getClass().getName());
}
}
