fix: rlp issue resolved (#420)

icon-project · Dec 4, 2024 · c72a550 · c72a550
1 parent 3cdf0b3
commit c72a550
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/contracts/soroban/libs/soroban-rlp/src/decoder.rs b/contracts/soroban/libs/soroban-rlp/src/decoder.rs
@@ -131,6 +131,9 @@ pub fn decode_u64(env: &Env, bytes: Bytes) -> u64 {
 }
 
 pub fn decode_u128(env: &Env, bytes: Bytes) -> u128 {
+    if bytes.len() == 1 {
+        return bytes_to_u128(bytes);
+    }
     let decoded = decode(&env, bytes);
     bytes_to_u128(decoded)
 }

diff --git a/contracts/soroban/libs/soroban-rlp/src/utils.rs b/contracts/soroban/libs/soroban-rlp/src/utils.rs
@@ -53,7 +53,7 @@ pub fn bytes_to_u64(bytes: Bytes) -> u64 {
 }
 
 pub fn u128_to_bytes(env: &Env, number: u128) -> Bytes {
-    let mut bytes: Bytes = Bytes::new(&env);
+    let mut bytes = bytes!(&env, 0x00);
     let mut i = 15;
     let mut leading_zero = true;
     while i >= 0 {