4nk/sdk_client
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add scan key decryption

Browse Source
This commit is contained in:
Alex Silva 2024-03-29 15:26:46 +01:00
parent 98e94799ef
commit c50be51775
1 changed files with 25 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
crates/sp_client/src/user.rs
View File

@ -102,7 +102,8 @@ pub struct User {
recover_data: Vec<u8>,
revoke_data: Option<Vec<u8>>,
sharding: Sharding,
// recovered_spend_key: Option<String>,
//recovered_spend_key: Option<String>,
//recovered_scan_key: Option<String>,
}
impl User {
@ -136,6 +137,7 @@ impl User {
recover_data.extend_from_slice(&entropy_1);
recover_data.extend_from_slice(&entropy_2);
recover_data.extend_from_slice(&entropy_3);
// hash the concatenation
let mut engine = sha256::HashEngine::default();
@ -212,48 +214,56 @@ impl User {
recover_data,
revoke_data: Some(revoke_data),
sharding,
// recovered_spend_key: None,
//recovered_spend_key: None,
//recovered_scan_key: None,
})
}
pub fn login(user_password: String, recover_data: &[u8], sharding: Sharding) -> Result<SecretKey> {
let mut retrieved_key = [0u8; 32];
pub fn login(&self,user_password: String, recover_data: &[u8], sharding: Sharding) -> Result<()> {
let mut retrieved_spend_key = [0u8; 32];
let mut retrieved_scan_key = [0u8; 32];
let mut entropy1 = [0u8; 32];
let mut entropy2 = [0u8; 32];
let mut entropy3 = [0u8; 32];
let mut cipher_scan_key = Vec::with_capacity(32);
let mut part1_ciphertext = Vec::with_capacity(32); // just a guess
let mut reader = recover_data.reader();
reader.read_exact(&mut entropy1)?;
reader.read_exact(&mut entropy2)?;
reader.read_to_end(&mut part1_ciphertext)?;
reader.read_exact(&mut entropy3)?;
reader.read_exact(&mut part1_ciphertext)?;
reader.read_to_end(&mut cipher_scan_key)?;
retrieved_key[..16].copy_from_slice(&Self::recover_part1(&user_password, &entropy1, part1_ciphertext)?);
retrieved_spend_key[..16].copy_from_slice(&Self::recover_key_slice(&user_password, &entropy1, part1_ciphertext)?);
//@todo: get shardings from member managers!
let shardings = sharding.shares_vec.clone(); // temporary
retrieved_key[16..].copy_from_slice(&Self::recover_part2(
retrieved_spend_key[16..].copy_from_slice(&Self::recover_part2(
&user_password,
&entropy2,
shardings,
)?);
let key = SecretKey::from_slice(&retrieved_key)?;
retrieved_scan_key[..32].copy_from_slice(&Self::recover_key_slice(&user_password, &entropy3, cipher_scan_key)?);;
Ok(key)
//@todo: retrieved_scan_key and retrieved_spend_key should be stored somewhere!
Ok(())
}
fn recover_part1(
fn recover_key_slice(
password: &str,
entropy: &[u8],
part1_ciphertext: Vec<u8>,
ciphertext: Vec<u8>,
) -> Result<Vec<u8>> {
let mut engine = sha256::HashEngine::default();
engine.write_all(&password.as_bytes());
engine.write_all(&entropy);
let hash = sha256::Hash::from_engine(engine);
let aes_dec = Aes256Decryption::new(Purpose::Login, part1_ciphertext, hash.to_byte_array().to_vec(), None)?;
let aes_dec = Aes256Decryption::new(Purpose::Login, ciphertext, hash.to_byte_array().to_vec(), None)?;
aes_dec.decrypt_with_key()
}