feat: added support for 64 bytes pvk (#97)

klever-io · Feb 6, 2025 · 53b2f66 · 53b2f66
1 parent a9c89eb
commit 53b2f66
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/packages/kos/src/chains/klv/mod.rs b/packages/kos/src/chains/klv/mod.rs
@@ -189,6 +189,36 @@ mod test {
         );
     }
 
+    #[test]
+    fn test_pvk_32() {
+        let pvk = hex::decode("8734062c1158f26a3ca8a4a0da87b527a7c168653f7f4c77045e5cf571497d9d")
+            .unwrap();
+
+        let pbk = crate::chains::klv::KLV {}.get_pbk(pvk).unwrap();
+
+        let address = crate::chains::klv::KLV {}.get_address(pbk.clone()).unwrap();
+        assert_eq!(
+            address,
+            "klv1usdnywjhrlv4tcyu6stxpl6yvhplg35nepljlt4y5r7yppe8er4qujlazy"
+        );
+        assert_eq!(pbk.len(), 32);
+    }
+
+    #[test]
+    fn test_pvk_64() {
+        let pvk = hex::decode("8734062c1158f26a3ca8a4a0da87b527a7c168653f7f4c77045e5cf571497d9de41b323a571fd955e09cd41660ff4465c3f44693c87f2faea4a0fc408727c8ea")
+            .unwrap();
+
+        let pbk = crate::chains::klv::KLV {}.get_pbk(pvk).unwrap();
+
+        let address = crate::chains::klv::KLV {}.get_address(pbk.clone()).unwrap();
+        assert_eq!(
+            address,
+            "klv1usdnywjhrlv4tcyu6stxpl6yvhplg35nepljlt4y5r7yppe8er4qujlazy"
+        );
+        assert_eq!(pbk.len(), 32);
+    }
+
     #[test]
     fn test_sign_raw() {
         let mnemonic =

diff --git a/packages/kos/src/chains/util.rs b/packages/kos/src/chains/util.rs
@@ -11,5 +11,8 @@ pub fn slice_from_vec<const N: usize>(vec: &[u8]) -> Result<[u8; N], ChainError>
 }
 
 pub fn private_key_from_vec<const N: usize>(vec: &[u8]) -> Result<[u8; N], ChainError> {
-    slice_from_vec::<N>(vec).map_err(|_| ChainError::InvalidPrivateKey)
+    // If input is longer than N, take first N bytes
+    let slice = if vec.len() > N { &vec[..N] } else { vec };
+
+    slice_from_vec::<N>(slice).map_err(|_| ChainError::InvalidPrivateKey)
 }