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 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());
}
}