Add ristretto multiply syscall (#12699)

programs/bpf/Cargo.lock

@@ -395,6 +395,19 @@ dependencies = [
  "zeroize",
 ]
 
+[[package]]
+name = "curve25519-dalek"
+version = "3.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c8492de420e9e60bc9a1d66e2dbb91825390b738a388606600663fc529b4b307"
+dependencies = [
+ "byteorder 1.3.4",
+ "digest 0.9.0",
+ "rand_core",
+ "subtle 2.2.2",
+ "zeroize",
+]
+
 [[package]]
 name = "digest"
 version = "0.8.1"
@@ -444,7 +457,7 @@ version = "1.0.0-pre.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "21a8a37f4e8b35af971e6db5e3897e7a6344caa3f92f6544f88125a1f5f0035a"
 dependencies = [
- "curve25519-dalek",
+ "curve25519-dalek 2.1.0",
  "ed25519",
  "rand",
  "serde",
@@ -1738,6 +1751,7 @@ version = "1.4.0"
 dependencies = [
  "bincode",
  "byteorder 1.3.4",
+ "curve25519-dalek 3.0.0",
  "num-derive 0.3.0",
  "num-traits",
  "solana-runtime",
@@ -1912,6 +1926,15 @@ dependencies = [
  "solana-sdk",
 ]
 
+[[package]]
+name = "solana-bpf-rust-ristretto"
+version = "1.4.0"
+dependencies = [
+ "curve25519-dalek 3.0.0",
+ "getrandom",
+ "solana-sdk",
+]
+
 [[package]]
 name = "solana-bpf-rust-sanity"
 version = "1.4.0"
@@ -1952,7 +1975,7 @@ dependencies = [
  "backtrace",
  "bytes 0.4.12",
  "cc",
- "curve25519-dalek",
+ "curve25519-dalek 2.1.0",
  "ed25519-dalek",
  "either",
  "failure",
@@ -2064,7 +2087,7 @@ dependencies = [
  "bv",
  "byteorder 1.3.4",
  "chrono",
- "curve25519-dalek",
+ "curve25519-dalek 2.1.0",
  "digest 0.9.0",
  "ed25519-dalek",
  "generic-array 0.14.3",

programs/bpf/Cargo.toml

@@ -54,6 +54,7 @@ members = [
     "rust/param_passing",
     "rust/param_passing_dep",
     "rust/rand",
+    "rust/ristretto",
     "rust/sanity",
     "rust/sha256",
     "rust/sysval",

programs/bpf/build.rs

@@ -82,6 +82,7 @@ fn main() {
             "panic",
             "param_passing",
             "rand",
+            "ristretto",
             "sanity",
             "sha256",
             "sysval",

programs/bpf/rust/ristretto/Cargo.toml

@@ -0,0 +1,28 @@
+
+# Note: This crate must be built using do.sh
+
+[package]
+name = "solana-bpf-rust-ristretto"
+version = "1.4.0"
+description = "Solana BPF test program written in Rust"
+authors = ["Solana Maintainers <maintainers@solana.foundation>"]
+repository = "https://github.com/solana-labs/solana"
+license = "Apache-2.0"
+homepage = "https://solana.com/"
+edition = "2018"
+
+[dependencies]
+curve25519-dalek = "3"
+getrandom = { version = "0.1.14", features = ["dummy"] }
+solana-sdk = { path = "../../../../sdk/", version = "1.4.0", default-features = false }
+
+[features]
+program = ["solana-sdk/program"]
+default = ["program", "solana-sdk/default"]
+
+[lib]
+name = "solana_bpf_rust_ristretto"
+crate-type = ["cdylib"]
+
+[package.metadata.docs.rs]
+targets = ["x86_64-unknown-linux-gnu"]

programs/bpf/rust/ristretto/Xargo.toml

@@ -0,0 +1,2 @@
+[target.bpfel-unknown-unknown.dependencies.std]
+features = []

programs/bpf/rust/ristretto/src/lib.rs

@@ -0,0 +1,31 @@
+//! @brief Example Rust-based BPF program that performs a ristretto multiply
+
+pub mod ristretto;
+
+extern crate solana_sdk;
+use crate::ristretto::ristretto_mul;
+use curve25519_dalek::{constants::RISTRETTO_BASEPOINT_POINT, scalar::Scalar};
+use solana_sdk::{
+    account_info::AccountInfo, entrypoint, entrypoint::ProgramResult, info, pubkey::Pubkey,
+};
+
+entrypoint!(process_instruction);
+fn process_instruction(
+    _program_id: &Pubkey,
+    _accounts: &[AccountInfo],
+    _instruction_data: &[u8],
+) -> ProgramResult {
+    info!("Ristretto multiply");
+
+    let point = RISTRETTO_BASEPOINT_POINT;
+    let scalar = Scalar::zero();
+    let result = ristretto_mul(&point, &scalar)?;
+    assert_ne!(point, result);
+
+    let point = RISTRETTO_BASEPOINT_POINT;
+    let scalar = Scalar::one();
+    let result = ristretto_mul(&point, &scalar)?;
+    assert_eq!(point, result);
+
+    Ok(())
+}

programs/bpf/rust/ristretto/src/ristretto.rs

@@ -0,0 +1,33 @@
+use curve25519_dalek::{ristretto::RistrettoPoint, scalar::Scalar};
+use solana_sdk::{entrypoint::SUCCESS, program_error::ProgramError};
+
+/// Multiply a ristretto point with a scalar
+///
+/// @param point - Ristretto point
+/// @param scalar - Scalar to mulitply against
+/// @return - result of the multiplication
+#[inline]
+pub fn ristretto_mul(
+    point: &RistrettoPoint,
+    scalar: &Scalar,
+) -> Result<RistrettoPoint, ProgramError> {
+    let mut result = RistrettoPoint::default();
+    let status = unsafe {
+        sol_ristretto_mul(
+            point as *const _ as *const u8,
+            scalar as *const _ as *const u8,
+            &mut result as *const _ as *mut u8,
+        )
+    };
+    match status {
+        SUCCESS => Ok(result),
+        _ => Err(status.into()),
+    }
+}
+extern "C" {
+    fn sol_ristretto_mul(
+        point_addr: *const u8,
+        scalar_addr: *const u8,
+        result_addr: *mut u8,
+    ) -> u64;
+}

programs/bpf/tests/programs.rs

@@ -155,6 +155,7 @@ fn test_program_bpf_sanity() {
             ("solana_bpf_rust_panic", false),
             ("solana_bpf_rust_param_passing", true),
             ("solana_bpf_rust_rand", true),
+            ("solana_bpf_rust_ristretto", true),
             ("solana_bpf_rust_sanity", true),
             ("solana_bpf_rust_sha256", true),
             ("solana_bpf_rust_sysval", true),
@@ -667,9 +668,10 @@ fn assert_instruction_count() {
             ("solana_bpf_rust_external_spend", 538),
             ("solana_bpf_rust_iter", 723),
             ("solana_bpf_rust_many_args", 231),
-            ("solana_bpf_rust_noop", 512),
+            ("solana_bpf_rust_noop", 488),
             ("solana_bpf_rust_param_passing", 46),
-            ("solana_bpf_rust_sanity", 1989),
+            ("solana_bpf_rust_ristretto", 19399),
+            ("solana_bpf_rust_sanity", 1965),
         ]);
     }