uqbar-dao/nectar
1
1
Fork 0
mirror of https://github.com/uqbar-dao/nectar.git synced 2024-12-20 07:01:40 +03:00
Code Issues Packages Projects Releases Wiki Activity

state: do not hash process_ids for keys

Browse Source
This commit is contained in:
bitful-pannul 2023-12-15 20:26:50 -03:00
parent ee4aa1d282
commit fe1e4c7d1a
2 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/state.rs
View File

@ -32,8 +32,8 @@ pub async fn load_state(
let db = DB::open_default(state_path).unwrap();
let mut process_map: ProcessMap = HashMap::new();
let kernel_id = KERNEL_PROCESS_ID.to_hash();
match db.get(kernel_id) {
let kernel_id = process_to_vec(KERNEL_PROCESS_ID.clone());
match db.get(&kernel_id) {
Ok(Some(value)) => {
process_map = bincode::deserialize::<ProcessMap>(&value).unwrap();
}
@ -47,7 +47,7 @@ pub async fn load_state(
.await
.unwrap();
db.put(kernel_id, bincode::serialize(&process_map).unwrap())
db.put(&kernel_id, bincode::serialize(&process_map).unwrap())
.unwrap();
}
Err(e) => {
@ -145,8 +145,8 @@ async fn handle_request(
let (ipc, bytes) = match action {
StateAction::SetState(process_id) => {
let key = process_id.to_hash();
// TODO consistency with to_stirngs
let key = process_to_vec(process_id);
let Some(ref payload) = payload else {
return Err(StateError::BadBytes {
action: "SetState".into(),
@ -157,7 +157,7 @@ async fn handle_request(
(serde_json::to_vec(&StateResponse::SetState).unwrap(), None)
}
StateAction::GetState(process_id) => {
let key = process_id.to_hash();
let key = process_to_vec(process_id.clone());
match db.get(key) {
Ok(Some(value)) => (
serde_json::to_vec(&StateResponse::GetState).unwrap(),
@ -178,7 +178,7 @@ async fn handle_request(
}
}
StateAction::DeleteState(process_id) => {
let key = process_id.to_hash();
let key = process_to_vec(process_id);
match db.delete(key) {
Ok(_) => (
serde_json::to_vec(&StateResponse::DeleteState).unwrap(),
@ -623,6 +623,10 @@ fn make_error_message(our_name: String, km: &KernelMessage, error: StateError) -
}
}
fn process_to_vec(process: ProcessId) -> Vec<u8> {
process.to_string().as_bytes().to_vec()
}
impl From<std::io::Error> for VfsError {
fn from(err: std::io::Error) -> Self {
VfsError::IOError {

7
src/types.rs
View File

@ -153,13 +153,6 @@ impl ProcessId {
pub fn publisher(&self) -> &str {
&self.publisher_node
}
pub fn to_hash(&self) -> [u8; 32] {
let mut hasher = blake3::Hasher::new();
hasher.update(self.process_name.as_bytes());
hasher.update(self.package_name.as_bytes());
hasher.update(self.publisher_node.as_bytes());
hasher.finalize().into()
}
pub fn en_wit(&self) -> wit::ProcessId {
wit::ProcessId {
process_name: self.process_name.clone(),