package org.bouncycastle.tls.crypto.impl.jcajce;
import java.security.GeneralSecurityException;
import javax.crypto.Cipher;
import javax.crypto.SecretKey;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import org.bouncycastle.tls.crypto.impl.TlsBlockCipherImpl;
/**
* A basic wrapper for a JCE Cipher class to provide the needed block cipher functionality for TLS.
*/
public class JceBlockCipherImpl
implements TlsBlockCipherImpl
{
private final int cipherMode;
private final Cipher cipher;
private final String algorithm;
private SecretKey key;
public JceBlockCipherImpl(Cipher cipher, String algorithm, boolean isEncrypting)
throws GeneralSecurityException
{
this.cipher = cipher;
this.algorithm = algorithm;
this.cipherMode = (isEncrypting) ? Cipher.ENCRYPT_MODE : Cipher.DECRYPT_MODE;
}
public void setKey(byte[] key)
{
this.key = new SecretKeySpec(key, algorithm);
}
public void init(byte[] iv)
{
try
{
cipher.init(cipherMode, key, new IvParameterSpec(iv));
}
catch (GeneralSecurityException e)
{
throw new IllegalStateException(e);
}
}
public int doFinal(byte[] input, int inputOffset, int inputLength, byte[] output, int outputOffset)
{
try
{
return cipher.doFinal(input, inputOffset, inputLength, output, outputOffset);
}
catch (GeneralSecurityException e)
{
throw new IllegalStateException(e);
}
}
public int getBlockSize()
{
return cipher.getBlockSize();
}
}