use crate::ecc::keccak256;
use crate::ecc::PublicKey;
use crate::ecc::PublicKeyAddress;
use crate::ecc::SecretKey;
use crate::error::Result;
pub fn sign_raw(sec: SecretKey, msg: &[u8]) -> [u8; 65] {
sign(sec, &hash(msg))
}
pub fn sign(sec: SecretKey, hash: &[u8; 32]) -> [u8; 65] {
let mut sig = sec.sign_hash(hash);
sig[64] += 27;
sig
}
pub fn hash(msg: &[u8]) -> [u8; 32] {
let mut prefix_msg = format!("\x19Ethereum Signed Message:\n{}", msg.len()).into_bytes();
prefix_msg.extend_from_slice(msg);
keccak256(&prefix_msg)
}
pub fn recover(msg: &[u8], sig: impl AsRef<[u8]>) -> Result<PublicKey> {
let sig_byte: [u8; 65] = sig.as_ref().try_into()?;
let hash = hash(msg);
let mut sig712 = sig_byte;
sig712[64] -= 27;
crate::ecc::recover_hash(&hash, &sig712)
}
pub fn verify(msg: &[u8], address: &PublicKeyAddress, sig: impl AsRef<[u8]>) -> bool {
if let Ok(p) = recover(msg, sig) {
p.address() == *address
} else {
false
}
}
#[cfg(test)]
mod test {
use std::str::FromStr;
use super::*;
use crate::ecc::SecretKey;
#[test]
fn test_eip191() {
use hex::FromHex;
let key =
SecretKey::try_from("65860affb4b570dba06db294aa7c676f68e04a5bf2721243ad3cbc05a79c68c0")
.unwrap();
let address =
PublicKeyAddress::from_str("0x11E807fcc88dD319270493fB2e822e388Fe36ab0").unwrap();
let metamask_sig = Vec::from_hex("724fc31d9272b34d8406e2e3a12a182e72510b008de6cc44684577e31e20d9626fb760d6a0badd79a6cf4cd56b2fc0fbd60c438b809aa7d29bfb598c13e7b50e1b").unwrap();
let msg = "test";
let h = self::hash(msg.as_bytes());
let sig = self::sign(key, &h);
assert_eq!(metamask_sig.as_slice(), sig);
let pubkey = self::recover(msg.as_bytes(), sig).unwrap();
assert_eq!(pubkey.address(), address);
assert!(self::verify(msg.as_bytes(), &address, sig));
}
}