com.github.zhangxd1989.basetool.crypto.digest.mac.DefaultHMacEngine Maven / Gradle / Ivy
package com.github.zhangxd1989.basetool.crypto.digest.mac;
import com.github.zhangxd1989.basetool.crypto.CryptoException;
import com.github.zhangxd1989.basetool.crypto.SecureUtil;
import com.github.zhangxd1989.basetool.io.IoUtil;
import javax.crypto.Mac;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import java.io.IOException;
import java.io.InputStream;
/**
* 默认的HMAC算法实现引擎，使用{@link Mac} 实现摘要
* 当引入BouncyCastle库时自动使用其作为Provider
*
* @author sheldon
*/
public class DefaultHMacEngine implements MacEngine {
private Mac mac;
/**
* 构造
*
* @param algorithm 算法
* @param key 密钥
*/
public DefaultHMacEngine(String algorithm, byte[] key) {
init(algorithm, key);
}
/**
* 构造
*
* @param algorithm 算法
* @param key 密钥
*/
public DefaultHMacEngine(String algorithm, SecretKey key) {
init(algorithm, key);
}
/**
* 初始化
*
* @param algorithm 算法
* @param key 密钥
* @return this
*/
public DefaultHMacEngine init(String algorithm, byte[] key) {
return init(algorithm, (null == key) ? null : new SecretKeySpec(key, algorithm));
}
/**
* 初始化
*
* @param algorithm 算法
* @param key 密钥 {@link SecretKey}
* @return this
* @throws CryptoException Cause by IOException
*/
public DefaultHMacEngine init(String algorithm, SecretKey key) {
try {
mac = SecureUtil.createMac(algorithm);
if (null == key) {
key = SecureUtil.generateKey(algorithm);
}
mac.init(key);
} catch (Exception e) {
throw new CryptoException(e);
}
return this;
}
@Override
public byte[] digest(InputStream data, int bufferLength) {
if (bufferLength < 1) {
bufferLength = IoUtil.DEFAULT_BUFFER_SIZE;
}
byte[] buffer = new byte[bufferLength];
byte[] result;
try {
int read = data.read(buffer, 0, bufferLength);
while (read > -1) {
mac.update(buffer, 0, read);
read = data.read(buffer, 0, bufferLength);
}
result = mac.doFinal();
} catch (IOException e) {
throw new CryptoException(e);
} finally {
mac.reset();
}
return result;
}
/**
* 获得 {@link Mac}
*
* @return {@link Mac}
*/
public Mac getMac() {
return mac;
}
}