Implement safe API for RIPEMD-160

near · Nov 29, 2021 · 765a66c · 765a66c
1 parent f7fcc8f
commit 765a66c
Show file tree

Hide file tree

Showing 15 changed files with 26 additions and 0 deletions.
diff --git a/examples/callback-results/res/callback_results.wasm b/examples/callback-results/res/callback_results.wasm
diff --git a/examples/cross-contract-high-level/res/cross_contract_high_level.wasm b/examples/cross-contract-high-level/res/cross_contract_high_level.wasm
diff --git a/examples/cross-contract-low-level/res/cross_contract_low_level.wasm b/examples/cross-contract-low-level/res/cross_contract_low_level.wasm
diff --git a/examples/fungible-token/res/defi.wasm b/examples/fungible-token/res/defi.wasm
diff --git a/examples/fungible-token/res/fungible_token.wasm b/examples/fungible-token/res/fungible_token.wasm
diff --git a/examples/lockable-fungible-token/res/lockable_fungible_token.wasm b/examples/lockable-fungible-token/res/lockable_fungible_token.wasm
diff --git a/examples/mission-control/res/mission_control.wasm b/examples/mission-control/res/mission_control.wasm
diff --git a/examples/non-fungible-token/res/approval_receiver.wasm b/examples/non-fungible-token/res/approval_receiver.wasm
diff --git a/examples/non-fungible-token/res/non_fungible_token.wasm b/examples/non-fungible-token/res/non_fungible_token.wasm
diff --git a/examples/non-fungible-token/res/token_receiver.wasm b/examples/non-fungible-token/res/token_receiver.wasm
diff --git a/examples/status-message-collections/res/status_message_collections.wasm b/examples/status-message-collections/res/status_message_collections.wasm
diff --git a/examples/status-message/res/status_message.wasm b/examples/status-message/res/status_message.wasm
diff --git a/examples/test-contract/res/test_contract.wasm b/examples/test-contract/res/test_contract.wasm
diff --git a/near-sdk/src/environment/env.rs b/near-sdk/src/environment/env.rs
@@ -54,6 +54,12 @@ macro_rules! method_into_register {
 //* Note: need specific length functions because const generics don't work with mem::transmute
 //* https://github.com/rust-lang/rust/issues/61956
 
+pub(crate) unsafe fn read_register_fixed_20(register_id: u64) -> [u8; 20] {
+    let mut hash = [MaybeUninit::<u8>::uninit(); 20];
+    sys::read_register(register_id, hash.as_mut_ptr() as _);
+    std::mem::transmute(hash)
+}
+
 pub(crate) unsafe fn read_register_fixed_32(register_id: u64) -> [u8; 32] {
     let mut hash = [MaybeUninit::<u8>::uninit(); 32];
     sys::read_register(register_id, hash.as_mut_ptr() as _);
@@ -279,6 +285,17 @@ pub fn keccak512_hash(value: &[u8]) -> [u8; 64] {
     }
 }
 
+/// Hashes the bytes using the RIPEMD-160 hash function. This returns a 20 byte hash.
+pub fn ripemd160_hash(value: &[u8]) -> [u8; 20] {
+    //* SAFETY: ripemd160 syscall will always generate 20 bytes inside of the atomic op register
+    //*         so the read will have a sufficient buffer of 20, and can transmute from uninit
+    //*         because all bytes are filled. This assumes a valid ripemd160 implementation.
+    unsafe {
+        sys::ripemd160(value.len() as _, value.as_ptr() as _, ATOMIC_OP_REGISTER);
+        read_register_fixed_20(ATOMIC_OP_REGISTER)
+    }
+}
+
 // ################
 // # Promises API #
 // ################
@@ -787,5 +804,10 @@ mod tests {
             .unwrap()
             .as_slice()
         );
+
+        assert_eq!(
+            &super::ripemd160_hash(b"some value"),
+            base64::decode("CfAl/tcE4eysj4iyvaPlaHbaA6w=").unwrap().as_slice()
+        );
     }
 }
diff --git a/near-sdk/src/environment/mock/mocked_blockchain.rs b/near-sdk/src/environment/mock/mocked_blockchain.rs
@@ -191,6 +191,10 @@ mod mock_chain {
         with_mock_interface(|b| b.keccak512(value_len, value_ptr, register_id))
     }
     #[no_mangle]
+    extern "C" fn ripemd160(value_len: u64, value_ptr: u64, register_id: u64) {
+        with_mock_interface(|b| b.ripemd160(value_len, value_ptr, register_id))
+    }
+    #[no_mangle]
     extern "C" fn value_return(value_len: u64, value_ptr: u64) {
         with_mock_interface(|b| b.value_return(value_len, value_ptr))
     }