Add syscall tests for curve25519 (#25634)

* add curve25519 syscall tests

* add curve25519 syscall feature set description

* fmt

* rename curve25519 group syscall

* update point to scalar

* resolve conflict
This commit is contained in:
samkim-crypto 2022-06-12 10:18:51 +09:00 committed by GitHub
parent 53695ecd21
commit 697a69cbe8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
13 changed files with 135 additions and 33 deletions

View File

@ -3927,6 +3927,14 @@ dependencies = [
"solana-program 1.11.0",
]
[[package]]
name = "solana-bpf-rust-curve25519"
version = "1.11.0"
dependencies = [
"solana-program 1.11.0",
"solana-zk-token-sdk 1.11.0",
]
[[package]]
name = "solana-bpf-rust-custom-heap"
version = "1.11.0"

View File

@ -50,6 +50,7 @@ members = [
"rust/alloc",
"rust/call_depth",
"rust/caller_access",
"rust/curve25519",
"rust/custom_heap",
"rust/dep_crate",
"rust/deprecated_loader",

View File

@ -62,6 +62,7 @@ fn main() {
"alloc",
"call_depth",
"caller_access",
"curve25519",
"custom_heap",
"dep_crate",
"deprecated_loader",

View File

@ -0,0 +1,20 @@
[package]
name = "solana-bpf-rust-curve25519"
version = "1.11.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/"
documentation = "https://docs.rs/solana-bpf-rust-zktoken_crypto"
edition = "2018"
[dependencies]
solana-program = { path = "../../../../sdk/program", version = "=1.11.0" }
solana-zk-token-sdk = { path = "../../../../zk-token-sdk", version = "=1.11.0" }
[lib]
crate-type = ["cdylib"]
[package.metadata.docs.rs]
targets = ["x86_64-unknown-linux-gnu"]

View File

@ -0,0 +1,71 @@
//! @brief curve25519 syscall tests
extern crate solana_program;
use {
solana_program::{custom_heap_default, custom_panic_default, msg},
solana_zk_token_sdk::curve25519::{edwards, ristretto, scalar},
};
#[no_mangle]
pub extern "C" fn entrypoint(_input: *mut u8) -> u64 {
let scalar_one = scalar::PodScalar([
1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0,
]);
let edwards_identity = edwards::PodEdwardsPoint([
1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0,
]);
let edwards_generator = edwards::PodEdwardsPoint([
88, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102,
102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102, 102,
]);
msg!("validate_edwards");
assert!(edwards::validate_edwards(&edwards_generator));
msg!("add_edwards");
assert_eq!(
edwards_generator,
edwards::add_edwards(&edwards_generator, &edwards_identity).expect("add_edwards")
);
msg!("multiply_edwards");
assert_eq!(
edwards_generator,
edwards::multiply_edwards(&scalar_one, &edwards_generator).expect("multiply_edwards")
);
let ristretto_identity = ristretto::PodRistrettoPoint([
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
0, 0,
]);
let ristretto_generator = ristretto::PodRistrettoPoint([
226, 242, 174, 10, 106, 188, 78, 113, 168, 132, 169, 97, 197, 0, 81, 95, 88, 227, 11, 106,
165, 130, 221, 141, 182, 166, 89, 69, 224, 141, 45, 118,
]);
msg!("validate_ristretto");
assert!(ristretto::validate_ristretto(&ristretto_generator));
msg!("add_ristretto");
assert_eq!(
ristretto_generator,
ristretto::add_ristretto(&ristretto_generator, &ristretto_identity).expect("add_ristretto")
);
msg!("multiply_ristretto");
assert_eq!(
ristretto_generator,
ristretto::multiply_ristretto(&scalar_one, &ristretto_generator)
.expect("multiply_ristretto")
);
0
}
custom_heap_default!();
custom_panic_default!();

View File

@ -500,6 +500,7 @@ fn test_program_bpf_sanity() {
programs.extend_from_slice(&[
("solana_bpf_rust_128bit", true),
("solana_bpf_rust_alloc", true),
("solana_bpf_rust_curve25519", true),
("solana_bpf_rust_custom_heap", true),
("solana_bpf_rust_dep_crate", true),
("solana_bpf_rust_external_spend", false),

View File

@ -243,14 +243,14 @@ pub fn register_syscalls(
register_feature_gated_syscall!(
syscall_registry,
curve25519_syscall_enabled,
b"sol_curve25519_point_validation",
b"sol_curve_validate_point",
SyscallCurvePointValidation::init,
SyscallCurvePointValidation::call,
)?;
register_feature_gated_syscall!(
syscall_registry,
curve25519_syscall_enabled,
b"sol_curve25519_point_validation",
b"sol_curve_group_op",
SyscallCurveGroupOps::init,
SyscallCurveGroupOps::call,
)?;

View File

@ -0,0 +1 @@
/Users/samkim/.cache/solana/v1.27/sbf-tools

View File

@ -65,7 +65,7 @@ define_syscall!(fn sol_log_data(data: *const u8, data_len: u64));
define_syscall!(fn sol_get_processed_sibling_instruction(index: u64, meta: *mut ProcessedSiblingInstruction, program_id: *mut Pubkey, data: *mut u8, accounts: *mut AccountMeta) -> u64);
define_syscall!(fn sol_get_stack_height() -> u64);
define_syscall!(fn sol_curve_validate_point(curve_id: u64, point: *const u8, result: *mut u8) -> u64);
define_syscall!(fn sol_curve_op(curve_id: u64, op_id: u64, left_point: *const u8, right_point: *const u8, result: *mut u8) -> u64);
define_syscall!(fn sol_curve_group_op(curve_id: u64, op_id: u64, left_point: *const u8, right_point: *const u8, result: *mut u8) -> u64);
define_syscall!(fn sol_curve_multiscalar_mul(curve_id: u64, scalars: *const u8, points: *const u8, result: *mut u8) -> u64);
define_syscall!(fn sol_curve_pairing_map(curve_id: u64, point: *const u8, result: *mut u8) -> u64);

View File

@ -472,6 +472,7 @@ lazy_static! {
(stake_merge_with_unmatched_credits_observed::id(), "allow merging active stakes with unmatched credits_observed #18985"),
(gate_large_block::id(), "validator checks block cost against max limit in realtime, reject if exceeds."),
(zk_token_sdk_enabled::id(), "enable Zk Token proof program and syscalls"),
(curve25519_syscall_enabled::id(), "enable curve25519 syscalls"),
(versioned_tx_message_enabled::id(), "enable versioned transaction message processing"),
(libsecp256k1_fail_on_bad_count::id(), "fail libsec256k1_verify if count appears wrong"),
(instructions_sysvar_owned_by_sysvar::id(), "fix owner for instructions sysvar"),

View File

@ -9,6 +9,12 @@
//! zk-token-sdk or curve25519. It should be moved to a more general location in the future.
//!
// Functions are organized by the curve traits, which can be instantiated by multiple curve
// representations. The functions take in a `curve_id` (e.g. `CURVE25519_EDWARDS`) and should run
// the associated functions in the appropriate trait instantiation. The `curve_op` function
// additionally takes in an `op_id` (e.g. `ADD`) that controls which associated functions to run in
// `GroupOperations`.
pub trait PointValidation {
type Point;
@ -77,13 +83,3 @@ pub const CURVE25519_RISTRETTO: u64 = 1;
pub const ADD: u64 = 0;
pub const SUB: u64 = 1;
pub const MUL: u64 = 2;
// Functions are organized by the curve traits, which can be instantiated by multiple curve
// representations. The functions take in a `curve_id` (e.g. `CURVE25519_EDWARDS`) and should run
// the associated functions in the appropriate trait instantiation. The `curve_op` function
// additionally takes in an `op_id` (e.g. `ADD`) that controls which associated functions to run in
// `GroupOperations`.
#[cfg(target_os = "solana")]
pub use solana_program::syscalls::{
sol_curve_multiscalar_mul, sol_curve_op, sol_curve_pairing_map, sol_curve_validate_point,
};

View File

@ -129,15 +129,16 @@ mod target_arch {
mod target_arch {
use {
super::*,
crate::curve25519::curve_syscall_traits::{
sol_curve_op, sol_curve_validate_point, ADD, CURVE25519_EDWARDS, MUL, SUB,
crate::curve25519::{
curve_syscall_traits::{ADD, CURVE25519_EDWARDS, MUL, SUB},
scalar::PodScalar,
},
};
pub fn validate_edwards(point: &PodEdwardsPoint) -> bool {
let mut validate_result = 0u8;
let result = unsafe {
sol_curve_validate_point(
solana_program::syscalls::sol_curve_validate_point(
CURVE25519_EDWARDS,
&point.0 as *const u8,
&mut validate_result,
@ -152,7 +153,7 @@ mod target_arch {
) -> Option<PodEdwardsPoint> {
let mut result_point = PodEdwardsPoint::zeroed();
let result = unsafe {
sol_curve_op(
solana_program::syscalls::sol_curve_group_op(
CURVE25519_EDWARDS,
ADD,
&left_point.0 as *const u8,
@ -174,7 +175,7 @@ mod target_arch {
) -> Option<PodEdwardsPoint> {
let mut result_point = PodEdwardsPoint::zeroed();
let result = unsafe {
sol_curve_op(
solana_program::syscalls::sol_curve_group_op(
CURVE25519_EDWARDS,
SUB,
&left_point.0 as *const u8,
@ -191,16 +192,16 @@ mod target_arch {
}
pub fn multiply_edwards(
left_point: &PodEdwardsPoint,
right_point: &PodEdwardsPoint,
scalar: &PodScalar,
point: &PodEdwardsPoint,
) -> Option<PodEdwardsPoint> {
let mut result_point = PodEdwardsPoint::zeroed();
let result = unsafe {
sol_curve_op(
solana_program::syscalls::sol_curve_group_op(
CURVE25519_EDWARDS,
MUL,
&left_point.0 as *const u8,
&right_point.0 as *const u8,
&scalar.0 as *const u8,
&point.0 as *const u8,
&mut result_point.0 as *mut u8,
)
};

View File

@ -130,15 +130,16 @@ mod target_arch {
mod target_arch {
use {
super::*,
crate::curve25519::curve_syscall_traits::{
sol_curve_op, sol_curve_validate_point, ADD, CURVE25519_RISTRETTO, MUL, SUB,
crate::curve25519::{
curve_syscall_traits::{ADD, CURVE25519_RISTRETTO, MUL, SUB},
scalar::PodScalar,
},
};
pub fn validate_ristretto(point: &PodRistrettoPoint) -> bool {
let mut validate_result = 0u8;
let result = unsafe {
sol_curve_validate_point(
solana_program::syscalls::sol_curve_validate_point(
CURVE25519_RISTRETTO,
&point.0 as *const u8,
&mut validate_result,
@ -154,7 +155,7 @@ mod target_arch {
) -> Option<PodRistrettoPoint> {
let mut result_point = PodRistrettoPoint::zeroed();
let result = unsafe {
sol_curve_op(
solana_program::syscalls::sol_curve_group_op(
CURVE25519_RISTRETTO,
ADD,
&left_point.0 as *const u8,
@ -176,7 +177,7 @@ mod target_arch {
) -> Option<PodRistrettoPoint> {
let mut result_point = PodRistrettoPoint::zeroed();
let result = unsafe {
sol_curve_op(
solana_program::syscalls::sol_curve_group_op(
CURVE25519_RISTRETTO,
SUB,
&left_point.0 as *const u8,
@ -193,16 +194,16 @@ mod target_arch {
}
pub fn multiply_ristretto(
left_point: &PodRistrettoPoint,
right_point: &PodRistrettoPoint,
scalar: &PodScalar,
point: &PodRistrettoPoint,
) -> Option<PodRistrettoPoint> {
let mut result_point = PodRistrettoPoint::zeroed();
let result = unsafe {
sol_curve_op(
solana_program::syscalls::sol_curve_group_op(
CURVE25519_RISTRETTO,
MUL,
&left_point.0 as *const u8,
&right_point.0 as *const u8,
&scalar.0 as *const u8,
&point.0 as *const u8,
&mut result_point.0 as *mut u8,
)
};