diff options
Diffstat (limited to 'shared/src/store.rs')
-rw-r--r-- | shared/src/store.rs | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/shared/src/store.rs b/shared/src/store.rs index 83f1a25..44e5e3d 100644 --- a/shared/src/store.rs +++ b/shared/src/store.rs @@ -1,9 +1,10 @@ use crate::packets::Resource; use anyhow::Result; use redb::{Database, TableDefinition}; +use sha2::{Digest, Sha256}; use std::path::Path; -const T_ENTRIES: TableDefinition<u128, &[u8]> = TableDefinition::new("e"); +const T_ENTRIES: TableDefinition<[u8; 32], &[u8]> = TableDefinition::new("e"); pub struct ResourceStore { db: Database, @@ -22,12 +23,19 @@ impl ResourceStore { None => Ok(None), } } - pub fn set(&self, key: Resource, value: &[u8]) -> Result<()> { + pub fn set(&self, value: &[u8]) -> Result<()> { + let key = sha256(value); let txn = self.db.begin_write()?; let mut ent = txn.open_table(T_ENTRIES)?; - ent.insert(key.0, value)?; + ent.insert(key, value)?; drop(ent); txn.commit()?; Ok(()) } } + +pub fn sha256(x: &[u8]) -> [u8; 32] { + let mut hasher = Sha256::new(); + hasher.update(x); + hasher.finalize().into() +} |