public void visit(JobInitMessage message) {
LOGGER.debug("Process JobInitMessage: " + message);
initSessionContext_.acquireWriteLock();
try {
sessionId_ =
initSessionContext_.tryAllocFreeSlot(
new InitSessionObject(peerAddress_, localSourceJobId_, data_));
} catch (SessionRuntimeException e) {
endWithError(e, ErrorNotificationMethod.REMOTE);
return;
} finally {
initSessionContext_.releaseWriteLock();
}
String peerKeyId = networkMonitor_.getPeerPublicKeyId(peerAddress_);
try {
Pair<KeyAgreement, DiffieHellmanInitPackage> firstStep =
DiffieHellmanProtocol.firstStepDHKeyAgreement();
EncryptedObject encryptedData = encryptionAPI_.encrypt(firstStep.getSecond(), peerKeyId);
initSessionContext_
.tryGetInitSessionObject(sessionId_)
.setKeyAgreement(firstStep.getFirst());
Message initSessionMessage =
new InitSessionMessage(myAddress_, peerAddress_, sessionId_, getJobId(), encryptedData);
networkQueue_.add(initSessionMessage);
} catch (CryptoException e) {
endWithErrorAndClear(e, ErrorNotificationMethod.LOCAL);
}
}
public void visit(InitSessionMessage message) {
LOGGER.debug("Process InitSessionMessage: " + message);
peerAddress_ = message.getSourceAddress();
remoteSourceJobId_ = message.getSourceJobId();
sessionId_ = message.getSessionId();
if (!peerAddress_.equals(myAddress_)) {
initSessionContext_.acquireWriteLock();
try {
initSessionContext_.allocFreeSlot(sessionId_, new InitSessionObject(peerAddress_));
} catch (SessionRuntimeException e) {
endWithError(e, ErrorNotificationMethod.REMOTE);
return;
} finally {
initSessionContext_.releaseWriteLock();
}
}
try {
DiffieHellmanInitPackage diffieHellmanInitPackage =
(DiffieHellmanInitPackage)
encryptionAPI_.decrypt(message.getEncryptedData(), privateKeyPeerId_);
Pair<KeyAgreement, DiffieHellmanResponsePackage> secondStep =
DiffieHellmanProtocol.secondStepDHKeyAgreement(diffieHellmanInitPackage);
initSessionContext_
.tryGetInitSessionObject(sessionId_)
.setSessionKey(DiffieHellmanProtocol.fourthStepDHKeyAgreement(secondStep.getFirst()));
String peerKeyId = networkMonitor_.getPeerPublicKeyId(peerAddress_);
EncryptedObject encryptedData = encryptionAPI_.encrypt(secondStep.getSecond(), peerKeyId);
Message initSessionResponseMessage =
new InitSessionResponseMessage(
remoteSourceJobId_,
myAddress_,
peerAddress_,
sessionId_,
getJobId(),
encryptedData);
networkQueue_.add(initSessionResponseMessage);
endWithSuccess();
} catch (CryptoException e) {
endWithErrorAndClear(e, ErrorNotificationMethod.REMOTE);
}
}