348 lines
12 KiB
Rust
348 lines
12 KiB
Rust
extern crate bincode;
|
|
extern crate solana;
|
|
extern crate solana_program_interface;
|
|
|
|
use std::collections::HashMap;
|
|
#[cfg(feature = "bpf_c")]
|
|
use std::path::Path;
|
|
use std::sync::RwLock;
|
|
use std::thread;
|
|
|
|
use bincode::serialize;
|
|
|
|
use solana::dynamic_program::DynamicProgram;
|
|
#[cfg(feature = "bpf_c")]
|
|
use solana::dynamic_program::ProgramPath;
|
|
use solana::hash::Hash;
|
|
use solana::signature::{Keypair, KeypairUtil};
|
|
use solana::system_program::SystemProgram;
|
|
use solana::system_transaction::SystemTransaction;
|
|
use solana::transaction::Transaction;
|
|
use solana_program_interface::account::{Account, KeyedAccount};
|
|
use solana_program_interface::pubkey::Pubkey;
|
|
|
|
#[cfg(feature = "bpf_c")]
|
|
use solana::tictactoe_program::Command;
|
|
|
|
#[cfg(feature = "bpf_c")]
|
|
#[test]
|
|
fn test_path_create_bpf() {
|
|
let path = ProgramPath::Bpf {}.create("move_funds_c");
|
|
assert_eq!(true, Path::new(&path).exists());
|
|
let path = ProgramPath::Bpf {}.create("tictactoe_c");
|
|
assert_eq!(true, Path::new(&path).exists());
|
|
}
|
|
|
|
#[cfg(feature = "bpf_c")]
|
|
#[test]
|
|
#[ignore]
|
|
fn test_bpf_file_noop_rust() {
|
|
let data: Vec<u8> = vec![0];
|
|
let keys = vec![Pubkey::default(); 2];
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
accounts[0].tokens = 100;
|
|
accounts[1].tokens = 1;
|
|
|
|
{
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
let dp = DynamicProgram::new_bpf_from_file("noop_rust".to_string());
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "bpf_c")]
|
|
#[test]
|
|
fn test_bpf_file_move_funds_c() {
|
|
let data: Vec<u8> = vec![0xa, 0xb, 0xc, 0xd, 0xe, 0xf];
|
|
let keys = vec![Pubkey::new(&[0xAA; 32]), Pubkey::new(&[0xBB; 32])];
|
|
let mut accounts = vec![
|
|
Account::new(0x0123456789abcdef, 4, Pubkey::default()),
|
|
Account::new(1, 8, Pubkey::default()),
|
|
];
|
|
|
|
{
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
let dp = DynamicProgram::new_bpf_from_file("move_funds_c".to_string());
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "bpf_c")]
|
|
fn tictactoe_command(command: Command, accounts: &mut Vec<Account>, player: Pubkey) {
|
|
let p = &command as *const Command as *const u8;
|
|
let data: &[u8] = unsafe { std::slice::from_raw_parts(p, std::mem::size_of::<Command>()) };
|
|
|
|
// Init
|
|
// player_x pub key in keys[2]
|
|
// accounts[0].program_id must be tictactoe
|
|
// accounts[1].userdata must be tictactoe game state
|
|
|
|
let keys = vec![player, Pubkey::default(), player];
|
|
|
|
{
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut *accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
let dp = DynamicProgram::new_bpf_from_file("tictactoe_c".to_string());
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
}
|
|
|
|
#[cfg(feature = "bpf_c")]
|
|
#[test]
|
|
fn test_bpf_file_tictactoe_c() {
|
|
let game_size = 0x78; // corresponds to the C structure size
|
|
let mut accounts = vec![
|
|
Account::new(0, 0, Pubkey::default()),
|
|
Account::new(0, game_size, Pubkey::default()),
|
|
Account::new(0, 0, Pubkey::default()),
|
|
];
|
|
|
|
tictactoe_command(Command::Init, &mut accounts, Pubkey::new(&[0xA; 32]));
|
|
tictactoe_command(
|
|
Command::Join(0xAABBCCDD),
|
|
&mut accounts,
|
|
Pubkey::new(&[0xA; 32]),
|
|
);
|
|
tictactoe_command(Command::Move(1, 1), &mut accounts, Pubkey::new(&[0xA; 32]));
|
|
tictactoe_command(Command::Move(0, 0), &mut accounts, Pubkey::new(&[0xA; 32]));
|
|
tictactoe_command(Command::Move(2, 0), &mut accounts, Pubkey::new(&[0xA; 32]));
|
|
tictactoe_command(Command::Move(0, 2), &mut accounts, Pubkey::new(&[0xA; 32]));
|
|
tictactoe_command(Command::Move(2, 2), &mut accounts, Pubkey::new(&[0xA; 32]));
|
|
tictactoe_command(Command::Move(0, 1), &mut accounts, Pubkey::new(&[0xA; 32]));
|
|
|
|
// validate test
|
|
}
|
|
|
|
#[test]
|
|
fn test_native_file_noop() {
|
|
let data: Vec<u8> = vec![0];
|
|
let keys = vec![Pubkey::default(); 2];
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
accounts[0].tokens = 100;
|
|
accounts[1].tokens = 1;
|
|
|
|
{
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
let dp = DynamicProgram::new_native("noop".to_string());
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_native_file_move_funds_success() {
|
|
let tokens: i64 = 100;
|
|
let data: Vec<u8> = serialize(&tokens).unwrap();
|
|
let keys = vec![Pubkey::default(); 2];
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
accounts[0].tokens = 100;
|
|
accounts[1].tokens = 1;
|
|
|
|
{
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
let dp = DynamicProgram::new_native("move_funds".to_string());
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
assert_eq!(0, accounts[0].tokens);
|
|
assert_eq!(101, accounts[1].tokens);
|
|
}
|
|
|
|
#[test]
|
|
fn test_native_file_move_funds_insufficient_funds() {
|
|
let tokens: i64 = 100;
|
|
let data: Vec<u8> = serialize(&tokens).unwrap();
|
|
let keys = vec![Pubkey::default(); 2];
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
accounts[0].tokens = 10;
|
|
accounts[1].tokens = 1;
|
|
|
|
{
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
let dp = DynamicProgram::new_native("move_funds".to_string());
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
assert_eq!(10, accounts[0].tokens);
|
|
assert_eq!(1, accounts[1].tokens);
|
|
}
|
|
|
|
#[test]
|
|
fn test_program_native_move_funds_succes_many_threads() {
|
|
let num_threads = 42; // number of threads to spawn
|
|
let num_iters = 100; // number of iterations of test in each thread
|
|
let mut threads = Vec::new();
|
|
for _t in 0..num_threads {
|
|
threads.push(thread::spawn(move || {
|
|
for _i in 0..num_iters {
|
|
{
|
|
let tokens: i64 = 100;
|
|
let data: Vec<u8> = serialize(&tokens).unwrap();
|
|
let keys = vec![Pubkey::default(); 2];
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
accounts[0].tokens = 100;
|
|
accounts[1].tokens = 1;
|
|
|
|
{
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
let dp = DynamicProgram::new_native("move_funds".to_string());
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
assert_eq!(0, accounts[0].tokens);
|
|
assert_eq!(101, accounts[1].tokens);
|
|
}
|
|
}
|
|
}));
|
|
}
|
|
|
|
for thread in threads {
|
|
thread.join().unwrap();
|
|
}
|
|
}
|
|
fn process_transaction(
|
|
tx: &Transaction,
|
|
accounts: &mut [Account],
|
|
loaded_programs: &RwLock<HashMap<Pubkey, DynamicProgram>>,
|
|
) {
|
|
let mut refs: Vec<&mut Account> = accounts.iter_mut().collect();
|
|
SystemProgram::process_transaction(&tx, 0, &mut refs[..], loaded_programs)
|
|
}
|
|
|
|
#[test]
|
|
fn test_system_program_load_call() {
|
|
// first load the program
|
|
let loaded_programs = RwLock::new(HashMap::new());
|
|
{
|
|
let from = Keypair::new();
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
let program_id = Pubkey::default(); // same program id for both
|
|
let tx = Transaction::system_load(
|
|
&from,
|
|
Hash::default(),
|
|
0,
|
|
program_id,
|
|
"move_funds".to_string(),
|
|
);
|
|
|
|
process_transaction(&tx, &mut accounts, &loaded_programs);
|
|
}
|
|
// then call the program
|
|
{
|
|
let program_id = Pubkey::default(); // same program id for both
|
|
let keys = vec![Pubkey::default(), Pubkey::default()];
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
accounts[0].tokens = 100;
|
|
accounts[1].tokens = 1;
|
|
let tokens: i64 = 100;
|
|
let data: Vec<u8> = serialize(&tokens).unwrap();
|
|
{
|
|
let hash = loaded_programs.write().unwrap();
|
|
match hash.get(&program_id) {
|
|
Some(dp) => {
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
None => panic!("failed to find program in hash"),
|
|
}
|
|
}
|
|
assert_eq!(0, accounts[0].tokens);
|
|
assert_eq!(101, accounts[1].tokens);
|
|
}
|
|
}
|
|
#[test]
|
|
fn test_system_program_load_call_many_threads() {
|
|
let num_threads = 42;
|
|
let num_iters = 100;
|
|
let mut threads = Vec::new();
|
|
for _t in 0..num_threads {
|
|
threads.push(thread::spawn(move || {
|
|
let _tid = thread::current().id();
|
|
for _i in 0..num_iters {
|
|
// first load the program
|
|
let loaded_programs = RwLock::new(HashMap::new());
|
|
{
|
|
let from = Keypair::new();
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
let program_id = Pubkey::default(); // same program id for both
|
|
let tx = Transaction::system_load(
|
|
&from,
|
|
Hash::default(),
|
|
0,
|
|
program_id,
|
|
"move_funds".to_string(),
|
|
);
|
|
|
|
process_transaction(&tx, &mut accounts, &loaded_programs);
|
|
}
|
|
// then call the program
|
|
{
|
|
let program_id = Pubkey::default(); // same program id for both
|
|
let keys = vec![Pubkey::default(), Pubkey::default()];
|
|
let mut accounts = vec![Account::default(), Account::default()];
|
|
accounts[0].tokens = 100;
|
|
accounts[1].tokens = 1;
|
|
let tokens: i64 = 100;
|
|
let data: Vec<u8> = serialize(&tokens).unwrap();
|
|
{
|
|
let hash = loaded_programs.write().unwrap();
|
|
match hash.get(&program_id) {
|
|
Some(dp) => {
|
|
let mut infos: Vec<_> = (&keys)
|
|
.into_iter()
|
|
.zip(&mut accounts)
|
|
.map(|(key, account)| KeyedAccount { key, account })
|
|
.collect();
|
|
|
|
dp.call(&mut infos, &data);
|
|
}
|
|
None => panic!("failed to find program in hash"),
|
|
}
|
|
}
|
|
assert_eq!(0, accounts[0].tokens);
|
|
assert_eq!(101, accounts[1].tokens);
|
|
}
|
|
}
|
|
}));
|
|
}
|
|
|
|
for thread in threads {
|
|
thread.join().unwrap();
|
|
}
|
|
}
|