231 lines
6.2 KiB
Rust
231 lines
6.2 KiB
Rust
//! Example SBF program using Poseidon syscall
|
|
|
|
use solana_program::{
|
|
custom_heap_default, custom_panic_default, msg,
|
|
poseidon::{hashv, Endianness, Parameters, PoseidonSyscallError},
|
|
};
|
|
|
|
fn test_poseidon_input_ones_twos() -> Result<(), PoseidonSyscallError> {
|
|
let input1 = [1u8; 32];
|
|
let input2 = [2u8; 32];
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[&input1, &input2],
|
|
)?;
|
|
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
13, 84, 225, 147, 143, 138, 140, 28, 125, 235, 94, 3, 85, 242, 99, 25, 32, 123, 132,
|
|
254, 156, 162, 206, 27, 38, 231, 53, 200, 41, 130, 25, 144
|
|
]
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::LittleEndian,
|
|
&[&input1, &input2],
|
|
)?;
|
|
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
144, 25, 130, 41, 200, 53, 231, 38, 27, 206, 162, 156, 254, 132, 123, 32, 25, 99, 242,
|
|
85, 3, 94, 235, 125, 28, 140, 138, 143, 147, 225, 84, 13
|
|
],
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
fn test_poseidon_input_one() -> Result<(), PoseidonSyscallError> {
|
|
let input = [
|
|
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, 1,
|
|
];
|
|
|
|
let hash = hashv(Parameters::Bn254X5, Endianness::BigEndian, &[&input])?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
41, 23, 97, 0, 234, 169, 98, 189, 193, 254, 108, 101, 77, 106, 60, 19, 14, 150, 164,
|
|
209, 22, 139, 51, 132, 139, 137, 125, 197, 2, 130, 1, 51,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[&input, &input],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
0, 122, 243, 70, 226, 211, 4, 39, 158, 121, 224, 169, 243, 2, 63, 119, 18, 148, 167,
|
|
138, 203, 112, 231, 63, 144, 175, 226, 124, 173, 64, 30, 129,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[&input, &input, &input],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
2, 192, 6, 110, 16, 167, 42, 189, 43, 51, 195, 178, 20, 203, 62, 129, 188, 177, 182,
|
|
227, 9, 97, 205, 35, 194, 2, 177, 134, 115, 191, 37, 67,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[&input, &input, &input, &input],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
8, 44, 156, 55, 10, 13, 36, 244, 65, 111, 188, 65, 74, 55, 104, 31, 120, 68, 45, 39,
|
|
216, 99, 133, 153, 28, 23, 214, 252, 12, 75, 125, 113,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[&input, &input, &input, &input, &input],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
16, 56, 150, 5, 174, 104, 141, 79, 20, 219, 133, 49, 34, 196, 125, 102, 168, 3, 199,
|
|
43, 65, 88, 156, 177, 191, 134, 135, 65, 178, 6, 185, 187,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[&input, &input, &input, &input, &input, &input],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
42, 115, 246, 121, 50, 140, 62, 171, 114, 74, 163, 229, 189, 191, 80, 179, 144, 53,
|
|
215, 114, 159, 19, 91, 151, 9, 137, 15, 133, 197, 220, 94, 118,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[&input, &input, &input, &input, &input, &input, &input],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
34, 118, 49, 10, 167, 243, 52, 58, 40, 66, 20, 19, 157, 157, 169, 89, 190, 42, 49, 178,
|
|
199, 8, 165, 248, 25, 84, 178, 101, 229, 58, 48, 184,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[
|
|
&input, &input, &input, &input, &input, &input, &input, &input,
|
|
],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
23, 126, 20, 83, 196, 70, 225, 176, 125, 43, 66, 51, 66, 81, 71, 9, 92, 79, 202, 187,
|
|
35, 61, 35, 11, 109, 70, 162, 20, 217, 91, 40, 132,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[
|
|
&input, &input, &input, &input, &input, &input, &input, &input, &input,
|
|
],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
14, 143, 238, 47, 228, 157, 163, 15, 222, 235, 72, 196, 46, 187, 68, 204, 110, 231, 5,
|
|
95, 97, 251, 202, 94, 49, 59, 138, 95, 202, 131, 76, 71,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[
|
|
&input, &input, &input, &input, &input, &input, &input, &input, &input, &input,
|
|
],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
46, 196, 198, 94, 99, 120, 171, 140, 115, 48, 133, 79, 74, 112, 119, 193, 255, 146, 96,
|
|
228, 72, 133, 196, 184, 29, 209, 49, 173, 58, 134, 205, 150,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[
|
|
&input, &input, &input, &input, &input, &input, &input, &input, &input, &input, &input,
|
|
],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
0, 113, 61, 65, 236, 166, 53, 241, 23, 212, 236, 188, 235, 95, 58, 102, 220, 65, 66,
|
|
235, 112, 181, 103, 101, 188, 53, 143, 27, 236, 64, 187, 155,
|
|
],
|
|
);
|
|
|
|
let hash = hashv(
|
|
Parameters::Bn254X5,
|
|
Endianness::BigEndian,
|
|
&[
|
|
&input, &input, &input, &input, &input, &input, &input, &input, &input, &input, &input,
|
|
&input,
|
|
],
|
|
)?;
|
|
assert_eq!(
|
|
hash.to_bytes(),
|
|
[
|
|
20, 57, 11, 224, 186, 239, 36, 155, 212, 124, 101, 221, 172, 101, 194, 229, 46, 133,
|
|
19, 192, 129, 193, 205, 114, 201, 128, 6, 9, 142, 154, 143, 190,
|
|
],
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[no_mangle]
|
|
pub extern "C" fn entrypoint(_input: *mut u8) -> u64 {
|
|
msg!("poseidon_hash");
|
|
|
|
if let Err(e) = test_poseidon_input_ones_twos() {
|
|
return e.into();
|
|
}
|
|
if let Err(e) = test_poseidon_input_one() {
|
|
return e.into();
|
|
}
|
|
|
|
0
|
|
}
|
|
|
|
custom_heap_default!();
|
|
custom_panic_default!();
|