use crate::error::{CasError, CasResult};
use super::cas_hmac::CASHMAC;
use hmac::{Hmac, Mac};
use sha2::{Sha256, Sha512};
type HmacSha256 = Hmac<Sha256>;
type HmacSha512 = Hmac<Sha512>;
pub struct HMAC;
pub struct HMACSha512;
impl CASHMAC for HMAC {
fn sign(key: Vec<u8>, message: Vec<u8>) -> CasResult<Vec<u8>> {
let mut mac = HmacSha256::new_from_slice(&key).map_err(|_| CasError::InvalidKey)?;
mac.update(&message);
Ok(mac.finalize().into_bytes().to_vec())
}
fn verify(key: Vec<u8>, message: Vec<u8>, signature: Vec<u8>) -> CasResult<bool> {
let mut mac = HmacSha256::new_from_slice(&key).map_err(|_| CasError::InvalidKey)?;
mac.update(&message);
Ok(mac.verify_slice(&signature).is_ok())
}
}
impl CASHMAC for HMACSha512 {
fn sign(key: Vec<u8>, message: Vec<u8>) -> CasResult<Vec<u8>> {
let mut mac = HmacSha512::new_from_slice(&key).map_err(|_| CasError::InvalidKey)?;
mac.update(&message);
Ok(mac.finalize().into_bytes().to_vec())
}
fn verify(key: Vec<u8>, message: Vec<u8>, signature: Vec<u8>) -> CasResult<bool> {
let mut mac = HmacSha512::new_from_slice(&key).map_err(|_| CasError::InvalidKey)?;
mac.update(&message);
Ok(mac.verify_slice(&signature).is_ok())
}
}