-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
ZhenQian
committed
Nov 7, 2024
1 parent
76d0623
commit 60c8c07
Showing
7 changed files
with
274 additions
and
6 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,2 +1,3 @@ | ||
| pub mod sui_sign_hash_request; | ||
| pub mod sui_sign_request; | ||
| pub mod sui_signature; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,210 @@ | ||
| use alloc::string::{String, ToString}; | ||
| use alloc::vec::Vec; | ||
| use minicbor::data::{Int, Tag}; | ||
|
|
||
| use crate::cbor::{cbor_array, cbor_map}; | ||
| use crate::crypto_key_path::CryptoKeyPath; | ||
| use crate::impl_template_struct; | ||
| use crate::registry_types::{RegistryType, SUI_SIGN_HASH_REQUEST, UUID}; | ||
| use crate::traits::{MapSize, RegistryItem}; | ||
| use crate::types::Bytes; | ||
|
|
||
| const REQUEST_ID: u8 = 1; | ||
| const MESSAGE_HASH: u8 = 2; | ||
| const DERIVATION_PATHS: u8 = 3; | ||
| const ADDRESSES: u8 = 4; | ||
| const ORIGIN: u8 = 5; | ||
|
|
||
| impl_template_struct!(SuiSignHashRequest { | ||
| request_id: Option<Bytes>, | ||
| message_hash: String, | ||
| derivation_paths: Vec<CryptoKeyPath>, | ||
| addresses: Option<Vec<Bytes>>, | ||
| origin: Option<String> | ||
| }); | ||
|
|
||
| impl RegistryItem for SuiSignHashRequest { | ||
| fn get_registry_type() -> RegistryType<'static> { | ||
| SUI_SIGN_HASH_REQUEST | ||
| } | ||
| } | ||
|
|
||
| impl MapSize for SuiSignHashRequest { | ||
| fn map_size(&self) -> u64 { | ||
| let mut size = 2; | ||
| if self.request_id.is_some() { | ||
| size += 1; | ||
| } | ||
| if self.addresses.is_some() { | ||
| size += 1; | ||
| } | ||
| if self.origin.is_some() { | ||
| size += 1; | ||
| } | ||
| size | ||
| } | ||
| } | ||
|
|
||
| impl<C> minicbor::Encode<C> for SuiSignHashRequest { | ||
| fn encode<W: minicbor::encode::Write>( | ||
| &self, | ||
| e: &mut minicbor::Encoder<W>, | ||
| ctx: &mut C, | ||
| ) -> Result<(), minicbor::encode::Error<W::Error>> { | ||
| e.map(self.map_size())?; | ||
| if let Some(request_id) = self.get_request_id() { | ||
| e.int(Int::from(REQUEST_ID))? | ||
| .tag(Tag::Unassigned(UUID.get_tag()))? | ||
| .bytes(&request_id)?; | ||
| } | ||
| e.int(Int::from(MESSAGE_HASH))? | ||
| .str(&self.get_message_hash())?; | ||
|
|
||
| let derivation_paths = self.get_derivation_paths(); | ||
| if derivation_paths.is_empty() { | ||
| return Err(minicbor::encode::Error::message( | ||
| "derivation paths is invalid", | ||
| )); | ||
| } | ||
| e.int(Int::from(DERIVATION_PATHS))? | ||
| .array(derivation_paths.len() as u64)?; | ||
| for path in derivation_paths { | ||
| e.tag(Tag::Unassigned( | ||
| CryptoKeyPath::get_registry_type().get_tag(), | ||
| ))?; | ||
| CryptoKeyPath::encode(&path, e, ctx)?; | ||
| } | ||
|
|
||
| if let Some(addresses) = self.get_addresses() { | ||
| e.int(Int::from(ADDRESSES))?.array(addresses.len() as u64)?; | ||
| for addr in addresses { | ||
| e.bytes(&addr)?; | ||
| } | ||
| } | ||
|
|
||
| if let Some(origin) = self.get_origin() { | ||
| e.int(Int::from(ORIGIN))?.str(&origin)?; | ||
| } | ||
|
|
||
| Ok(()) | ||
| } | ||
| } | ||
|
|
||
| impl<'b, C> minicbor::Decode<'b, C> for SuiSignHashRequest { | ||
| fn decode(d: &mut minicbor::Decoder<'b>, ctx: &mut C) -> Result<Self, minicbor::decode::Error> { | ||
| let mut result = SuiSignHashRequest::default(); | ||
|
|
||
| cbor_map(d, &mut result, |key, obj, d| { | ||
| let key = | ||
| u8::try_from(key).map_err(|e| minicbor::decode::Error::message(e.to_string()))?; | ||
| match key { | ||
| REQUEST_ID => { | ||
| let tag = d.tag()?; | ||
| if !tag.eq(&Tag::Unassigned(UUID.get_tag())) { | ||
| return Err(minicbor::decode::Error::message("UUID tag is invalid")); | ||
| } | ||
| obj.request_id = Some(d.bytes()?.to_vec()); | ||
| } | ||
| MESSAGE_HASH => { | ||
| obj.message_hash = d.str()?.to_string(); | ||
| } | ||
| DERIVATION_PATHS => { | ||
| cbor_array(d, &mut obj.derivation_paths, |_key, obj, d| { | ||
| let tag = d.tag()?; | ||
| if !tag.eq(&Tag::Unassigned( | ||
| CryptoKeyPath::get_registry_type().get_tag(), | ||
| )) { | ||
| return Err(minicbor::decode::Error::message( | ||
| "CryptoKeyPath tag is invalid", | ||
| )); | ||
| } | ||
| obj.push(CryptoKeyPath::decode(d, ctx)?); | ||
| Ok(()) | ||
| })?; | ||
| } | ||
| ADDRESSES => { | ||
| if obj.addresses.is_none() { | ||
| obj.addresses = Some(Vec::new()) | ||
| } | ||
| cbor_array(d, &mut obj.addresses, |_key, obj, d| { | ||
| match obj { | ||
| Some(v) => v.push(d.bytes()?.to_vec()), | ||
| None => {} | ||
| } | ||
| Ok(()) | ||
| })?; | ||
| } | ||
| ORIGIN => { | ||
| obj.origin = Some(d.str()?.to_string()); | ||
| } | ||
| _ => {} | ||
| } | ||
| Ok(()) | ||
| })?; | ||
| Ok(result) | ||
| } | ||
| } | ||
|
|
||
| #[cfg(test)] | ||
| mod tests { | ||
| use alloc::vec; | ||
| use alloc::vec::Vec; | ||
|
|
||
| use crate::crypto_key_path::PathComponent; | ||
|
|
||
| use super::*; | ||
|
|
||
| #[test] | ||
| fn test_encode() { | ||
| let components = vec![ | ||
| PathComponent::new(Some(44), true).unwrap(), | ||
| PathComponent::new(Some(784), true).unwrap(), | ||
| PathComponent::new(Some(0), true).unwrap(), | ||
| PathComponent::new(Some(0), true).unwrap(), | ||
| PathComponent::new(Some(0), true).unwrap(), | ||
| ]; | ||
| let source_fingerprint = hex::decode("78230804").unwrap().try_into().unwrap(); | ||
| let crypto_key_path = CryptoKeyPath::new(components, Some(source_fingerprint), None); | ||
| let sig = SuiSignHashRequest { | ||
| request_id: Some(hex::decode("9b1deb4d3b7d4bad9bdd2b0d7b3dcb6d").unwrap()), | ||
| message_hash: "00000000000200201ff915a5e9e32fdbe0135535b6c69a00a9809aaf7f7c0275d3239ca79db20d6400081027000000000000020200010101000101020000010000ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec3944093886901a2e3e42930675d9571a467eb5d4b22553c93ccb84e9097972e02c490b4e7a22ab73200000000000020176c4727433105da34209f04ac3f22e192a2573d7948cb2fabde7d13a7f4f149ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec39440938869e803000000000000640000000000000000".to_string(), | ||
| derivation_paths: vec![crypto_key_path], | ||
| addresses: Some(vec![hex::decode("ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec39440938869").unwrap()]), | ||
| origin: Some("Sui Wallet".to_string()) | ||
| }; | ||
| let result: Vec<u8> = sig.try_into().unwrap(); | ||
| let expect_result = "a501d825509b1deb4d3b7d4bad9bdd2b0d7b3dcb6d027901b830303030303030303030303230303230316666393135613565396533326664626530313335353335623663363961303061393830396161663766376330323735643332333963613739646232306436343030303831303237303030303030303030303030303230323030303130313031303030313031303230303030303130303030656265363233653333623733303766313335306638393334626562336662313662616566306663316233663162393238363865656333393434303933383836393031613265336534323933303637356439353731613436376562356434623232353533633933636362383465393039373937326530326334393062346537613232616237333230303030303030303030303032303137366334373237343333313035646133343230396630346163336632326531393261323537336437393438636232666162646537643133613766346631343965626536323365333362373330376631333530663839333462656233666231366261656630666331623366316239323836386565633339343430393338383639653830333030303030303030303030303634303030303030303030303030303030300381d90130a2018a182cf5190310f500f500f500f5021a7823080404815820ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec39440938869056a5375692057616c6c6574"; | ||
|
|
||
| assert_eq!(expect_result, hex::encode(result)); | ||
| } | ||
|
|
||
| #[test] | ||
| fn test_decode() { | ||
| let components = vec![ | ||
| PathComponent::new(Some(44), true).unwrap(), | ||
| PathComponent::new(Some(784), true).unwrap(), | ||
| PathComponent::new(Some(0), true).unwrap(), | ||
| PathComponent::new(Some(0), true).unwrap(), | ||
| PathComponent::new(Some(0), true).unwrap(), | ||
| ]; | ||
| let source_fingerprint = hex::decode("78230804").unwrap().try_into().unwrap(); | ||
| let crypto_key_path = CryptoKeyPath::new(components, Some(source_fingerprint), None); | ||
| let expect_result = SuiSignHashRequest { | ||
| request_id: Some(hex::decode("9b1deb4d3b7d4bad9bdd2b0d7b3dcb6d").unwrap()), | ||
| message_hash:"00000000000200201ff915a5e9e32fdbe0135535b6c69a00a9809aaf7f7c0275d3239ca79db20d6400081027000000000000020200010101000101020000010000ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec3944093886901a2e3e42930675d9571a467eb5d4b22553c93ccb84e9097972e02c490b4e7a22ab73200000000000020176c4727433105da34209f04ac3f22e192a2573d7948cb2fabde7d13a7f4f149ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec39440938869e803000000000000640000000000000000".to_string(), | ||
| derivation_paths: vec![crypto_key_path], | ||
| addresses: Some(vec![hex::decode("ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec39440938869").unwrap()]), | ||
| origin: Some("Sui Wallet".to_string()) | ||
| }; | ||
| let result = SuiSignHashRequest::try_from(hex::decode("a501d825509b1deb4d3b7d4bad9bdd2b0d7b3dcb6d027901b830303030303030303030303230303230316666393135613565396533326664626530313335353335623663363961303061393830396161663766376330323735643332333963613739646232306436343030303831303237303030303030303030303030303230323030303130313031303030313031303230303030303130303030656265363233653333623733303766313335306638393334626562336662313662616566306663316233663162393238363865656333393434303933383836393031613265336534323933303637356439353731613436376562356434623232353533633933636362383465393039373937326530326334393062346537613232616237333230303030303030303030303032303137366334373237343333313035646133343230396630346163336632326531393261323537336437393438636232666162646537643133613766346631343965626536323365333362373330376631333530663839333462656233666231366261656630666331623366316239323836386565633339343430393338383639653830333030303030303030303030303634303030303030303030303030303030300381d90130a2018a182cf5190310f500f500f500f5021a7823080404815820ebe623e33b7307f1350f8934beb3fb16baef0fc1b3f1b92868eec39440938869056a5375692057616c6c6574").unwrap()).unwrap(); | ||
|
|
||
| assert_eq!(expect_result.request_id, result.request_id); | ||
| assert_eq!(expect_result.message_hash, result.message_hash); | ||
| assert_eq!( | ||
| expect_result.derivation_paths[0].get_path(), | ||
| result.derivation_paths[0].get_path() | ||
| ); | ||
| assert_eq!(expect_result.addresses, result.addresses); | ||
| assert_eq!(expect_result.origin, result.origin); | ||
| } | ||
| } |