Add zk_token_sdk_enabled feature to gate Zk Token proof program and `…

…sol_zk_token_elgamal_op` syscalls

programs/bpf/Cargo.toml

@@ -86,6 +86,7 @@ members = [
     "rust/sysvar",
     "rust/upgradeable",
     "rust/upgraded",
+    "rust/zk_token_elgamal",
 ]
 
 [package.metadata.docs.rs]

programs/bpf/build.rs

@@ -95,6 +95,7 @@ fn main() {
             "spoof1_system",
             "upgradeable",
             "upgraded",
+            "zk_token_elgamal",
         ];
         for program in rust_programs.iter() {
             println!(

programs/bpf/rust/zk_token_elgamal/Cargo.toml

@@ -0,0 +1,20 @@
+[package]
+name = "solana-bpf-rust-zk_token_elgamal"
+version = "1.10.0"
+description = "Solana BPF test program written in Rust"
+authors = ["Solana Maintainers <[email protected]>"]
+repository = "https://github.com/solana-labs/solana"
+license = "Apache-2.0"
+homepage = "https://solana.com/"
+documentation = "https://docs.rs/solana-bpf-rust-zktoken_crypto"
+edition = "2018"
+
+[dependencies]
+solana-program = { path = "../../../../sdk/program", version = "=1.10.0" }
+solana-zk-token-sdk = { path = "../../../../zk-token-sdk", version = "=1.10.0" }
+
+[lib]
+crate-type = ["cdylib"]
+
+[package.metadata.docs.rs]
+targets = ["x86_64-unknown-linux-gnu"]

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

@@ -0,0 +1,53 @@
+//! @brief zk_token_elgamal syscall tests
+
+extern crate solana_program;
+use {
+    solana_program::{custom_panic_default, msg},
+    solana_zk_token_sdk::zk_token_elgamal::{
+        ops,
+        pod::{ElGamalCiphertext, Zeroable},
+    },
+};
+
+#[no_mangle]
+pub extern "C" fn entrypoint(_input: *mut u8) -> u64 {
+    let zero = ElGamalCiphertext::zeroed();
+
+    msg!("add_to");
+    let one = ops::add_to(&zero, 1).expect("add_to");
+
+    msg!("subtract_from");
+    assert_eq!(zero, ops::subtract_from(&one, 1).expect("subtract_from"));
+
+    msg!("add");
+    assert_eq!(one, ops::add(&zero, &one).expect("add"));
+
+    msg!("subtract");
+    assert_eq!(zero, ops::subtract(&one, &one).expect("subtract"));
+
+    msg!("add_with_lo_hi");
+    assert_eq!(
+        one,
+        ops::add_with_lo_hi(
+            &one,
+            &ElGamalCiphertext::zeroed(),
+            &ElGamalCiphertext::zeroed()
+        )
+        .expect("add_with_lo_hi")
+    );
+
+    msg!("subtract_with_lo_hi");
+    assert_eq!(
+        one,
+        ops::subtract_with_lo_hi(
+            &one,
+            &ElGamalCiphertext::zeroed(),
+            &ElGamalCiphertext::zeroed()
+        )
+        .expect("subtract_with_lo_hi")
+    );
+
+    0
+}
+
+custom_panic_default!();

programs/bpf/tests/programs.rs

@@ -484,6 +484,7 @@ fn test_program_bpf_sanity() {
             ("solana_bpf_rust_sanity", true),
             ("solana_bpf_rust_secp256k1_recover", true),
             ("solana_bpf_rust_sha", true),
+            ("solana_bpf_rust_zk_token_elgamal", true),
         ]);
     }
 

programs/bpf_loader/Cargo.toml

@@ -17,6 +17,7 @@ libsecp256k1 = "0.6.0"
 solana-measure = { path = "../../measure", version = "=1.10.0" }
 solana-program-runtime = { path = "../../program-runtime", version = "=1.10.0" }
 solana-sdk = { path = "../../sdk", version = "=1.10.0" }
+solana-zk-token-sdk = { path = "../../zk-token-sdk", version = "=1.10.0" }
 solana_rbpf = "=0.2.21"
 thiserror = "1.0"