feat: Ripemd_160

Signed-off-by: yHSJ <josh@securitybot.info>

Author: yHSJ

crates/uplc/Cargo.toml

@@ -14,7 +14,7 @@ publish = true
 blst = "0.3.13"
 bumpalo = { version = "3.16.0", features = ["collections"] }
 chumsky = { version = "=1.0.0-alpha.7", features = ["pratt"] }
-cryptoxide = "0.4.4"
+cryptoxide = { version = "0.4.4", features = ["ripemd160"] }
 hamming = "0.1.3"
 minicbor = { version = "0.25.1", features = ["std"] }
 num = "0.4.3"

crates/uplc/src/builtin/default_function.rs

@@ -110,6 +110,7 @@ pub enum DefaultFunction {
     RotateByteString = 83,
     CountSetBits = 84,
     FindFirstSetBit = 85,
+    Ripemd_160 = 86,
 }
 
 impl DefaultFunction {
@@ -201,6 +202,7 @@ impl DefaultFunction {
             DefaultFunction::RotateByteString => 0,
             DefaultFunction::CountSetBits => 0,
             DefaultFunction::FindFirstSetBit => 0,
+            DefaultFunction::Ripemd_160 => 0,
         }
     }
 
@@ -292,6 +294,7 @@ impl DefaultFunction {
             DefaultFunction::RotateByteString => 2,
             DefaultFunction::CountSetBits => 1,
             DefaultFunction::FindFirstSetBit => 1,
+            DefaultFunction::Ripemd_160 => 1,
         }
     }
 }

crates/uplc/src/machine/cost_model/builtin_costs.rs

@@ -104,6 +104,7 @@ pub struct BuiltinCosts {
     rotate_byte_string: TwoArgumentsCosting,
     count_set_bits: OneArgumentCosting,
     find_first_set_bit: OneArgumentCosting,
+    ripemd_160: OneArgumentCosting,
 }
 
 impl Default for BuiltinCosts {
@@ -653,6 +654,13 @@ impl BuiltinCosts {
         )
     }
 
+    pub fn ripemd_160(&self, args: [i64; 1]) -> ExBudget {
+        ExBudget::new(
+            self.ripemd_160.mem.cost(args),
+            self.ripemd_160.cpu.cost(args),
+        )
+    }
+
     pub fn v3() -> Self {
         Self {
             add_integer: TwoArgumentsCosting::new(
@@ -1009,6 +1017,10 @@ impl BuiltinCosts {
                 OneArgumentCosting::constant_cost(1),
                 OneArgumentCosting::linear_cost(106057, 655),
             ),
+            ripemd_160: OneArgumentCosting::new(
+                OneArgumentCosting::constant_cost(3),
+                OneArgumentCosting::linear_cost(1964219, 24520),
+            ),
         }
     }
 }

crates/uplc/src/machine/runtime.rs

@@ -2335,6 +2335,22 @@ impl<'a> Machine<'a> {
                 let result = self.arena.alloc(first_bit);
                 Ok(Value::integer(self.arena, result))
             }
+            DefaultFunction::Ripemd_160 => {
+                use cryptoxide::{digest::Digest, ripemd160::Ripemd160};
+                let input = runtime.args[0].unwrap_byte_string()?;
+                let budget = self
+                    .costs
+                    .builtin_costs
+                    .ripemd_160([cost_model::byte_string_ex_mem(input)]);
+                self.spend_budget(budget)?;
+
+                let mut hasher = Ripemd160::new();
+                hasher.input(input);
+                let result = self.arena.alloc(vec![0; hasher.output_bytes()]);
+                hasher.result(result);
+
+                Ok(Value::byte_string(self.arena, result))
+            }
         }
     }
 }

crates/uplc/src/syn/term.rs

@@ -263,6 +263,7 @@ pub fn builtin_from_str<'a>(arena: &'a Bump, name: &str) -> Option<&'a Term<'a,
         "rotateByteString" => Some(Term::rotate_byte_string(arena)),
         "countSetBits" => Some(Term::count_set_bits(arena)),
         "findFirstSetBit" => Some(Term::find_first_set_bit(arena)),
+        "ripemd_160" => Some(Term::ripemd_160(arena)),
         _ => None,
     }
 }

crates/uplc/src/term.rs

@@ -629,6 +629,11 @@ impl<'a, V> Term<'a, V> {
     pub fn find_first_set_bit(arena: &'a Bump) -> &'a Term<'a, V> {
         let fun = arena.alloc(DefaultFunction::FindFirstSetBit);
 
+        Term::builtin(arena, fun)
+    }
+    pub fn ripemd_160(arena: &'a Bump) -> &'a Term<'a, V> {
+        let fun = arena.alloc(DefaultFunction::Ripemd_160);
+
         Term::builtin(arena, fun)
     }
 }