/**
* Copyright (c) 2007-2009 Alysson Bessani, Eduardo Alchieri, Paulo Sousa, and the authors indicated in the @author tags
*
* This file is part of SMaRt.
*
* SMaRt is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* SMaRt is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along with SMaRt. If not, see <http://www.gnu.org/licenses/>.
*/
package bftsmart.communication;
import bftsmart.paxosatwar.messages.MessageFactory;
import bftsmart.paxosatwar.messages.PaxosMessage;
import bftsmart.paxosatwar.roles.Acceptor;
import bftsmart.statemanagment.SMMessage;
import bftsmart.tom.core.TOMLayer;
import bftsmart.tom.core.messages.TOMMessage;
import bftsmart.tom.core.timer.ForwardedMessage;
import bftsmart.tom.leaderchange.LCMessage;
import bftsmart.tom.util.Logger;
import bftsmart.tom.util.TOMUtil;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.ObjectOutputStream;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.util.Arrays;
import java.util.HashMap;
import javax.crypto.BadPaddingException;
import javax.crypto.Cipher;
import javax.crypto.IllegalBlockSizeException;
import javax.crypto.NoSuchPaddingException;
import javax.crypto.spec.SecretKeySpec;
/**
*
* @author edualchieri
*/
public class MessageHandler {
private Acceptor acceptor;
private TOMLayer tomLayer;
private Cipher cipher;
public MessageHandler() {
try {
this.cipher = Cipher.getInstance("DES/ECB/PKCS5Padding");
} catch (NoSuchAlgorithmException ex) {
ex.printStackTrace();
} catch (NoSuchPaddingException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
public void setAcceptor(Acceptor acceptor) {
this.acceptor = acceptor;
}
public void setTOMLayer(TOMLayer tomLayer) {
this.tomLayer = tomLayer;
}
protected void processData(SystemMessage sm) {
if (sm instanceof PaxosMessage) {
int myId = tomLayer.reconfManager.getStaticConf().getProcessId();
PaxosMessage paxosMsg = (PaxosMessage) sm;
if (paxosMsg.authenticated || paxosMsg.getSender() == myId) acceptor.deliver(paxosMsg);
else if (paxosMsg.getPaxosType() == MessageFactory.STRONG && paxosMsg.getMACVector() != null) {
//We are going to verify the MAC vector at the algorithm level
HashMap<Integer, byte[]> macVector = (HashMap<Integer, byte[]>) paxosMsg.getMACVector();
byte[] recvMAC = macVector.get(myId);
PaxosMessage pm = new PaxosMessage(MessageFactory.STRONG,paxosMsg.getNumber(),
paxosMsg.getRound(), paxosMsg.getSender(), paxosMsg.getValue());
ByteArrayOutputStream bOut = new ByteArrayOutputStream(248);
try {
new ObjectOutputStream(bOut).writeObject(pm);
} catch (IOException ex) {
ex.printStackTrace();
}
byte[] data = bOut.toByteArray();
byte[] hash = tomLayer.computeHash(data);
byte[] myMAC = null;
SecretKeySpec key = new SecretKeySpec(tomLayer.getCommunication().getServersConn().getSecretKey(paxosMsg.getSender()).getEncoded(), "DES");
try {
this.cipher.init(Cipher.ENCRYPT_MODE, key);
myMAC = this.cipher.doFinal(hash);
} catch (IllegalBlockSizeException ex) {
ex.printStackTrace();
} catch (InvalidKeyException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} catch (BadPaddingException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
if (recvMAC != null && myMAC != null && Arrays.equals(recvMAC, myMAC))
acceptor.deliver(paxosMsg);
else {
Logger.println("(MessageHandler.processData) WARNING: invalid MAC from " + sm.getSender());
System.out.println("(MessageHandler.processData) WARNING: invalid MAC from " + sm.getSender());
}
} else {
System.out.println("(MessageHandler.processData) Discarding unauthenticated message from " + sm.getSender());
Logger.println("(MessageHandler.processData) Discarding unauthenticated message from " + sm.getSender());
}
} else if (sm.authenticated) {
/*** This is Joao's code, related to leader change */
if (sm instanceof LCMessage) {
LCMessage lcMsg = (LCMessage) sm;
String type = null;
switch(lcMsg.getType()) {
case TOMUtil.STOP:
type = "STOP";
break;
case TOMUtil.STOPDATA:
type = "STOPDATA";
break;
case TOMUtil.SYNC:
type = "SYNC";
break;
default:
type = "LOCAL";
break;
}
System.out.println("(MessageHandler.processData) LC_MSG received: type " + type + ", regency " + lcMsg.getReg() + ", (replica " + lcMsg.getSender() + ")");
if (lcMsg.TRIGGER_LC_LOCALLY) tomLayer.requestsTimer.run_lc_protocol();
else tomLayer.deliverTimeoutRequest(lcMsg);
/**************************************************************/
} else if (sm instanceof ForwardedMessage) {
TOMMessage request = ((ForwardedMessage) sm).getRequest();
tomLayer.requestReceived(request);
/** This is Joao's code, to handle state transfer */
} else if (sm instanceof SMMessage) {
SMMessage smsg = (SMMessage) sm;
String type = null;
switch(smsg.getType()) {
case TOMUtil.SM_REQUEST:
type = "SM_REQUEST";
break;
case TOMUtil.SM_REPLY:
type = "SM_REPLY";
break;
default:
type = "LOCAL";
break;
}
System.out.println("(MessageHandler.processData) SM_MSG received: type " + type + ", regency " + smsg.getRegency() + ", (replica " + smsg.getSender() + ")");
if (smsg.TRIGGER_SM_LOCALLY) {
tomLayer.getStateManager().stateTimeout();
}
else if (smsg.getType() == TOMUtil.SM_REQUEST) {
tomLayer.getStateManager().SMRequestDeliver(smsg);
} else {
tomLayer.getStateManager().SMReplyDeliver(smsg);
}
/******************************************************************/
}
} else {
Logger.println("(MessageHandler.processData) Discarding unauthenticated message from " + sm.getSender());
}
}
protected void verifyPending() {
tomLayer.processOutOfContext();
}
}