99 lines
2.7 KiB
Rust
99 lines
2.7 KiB
Rust
|
//! @brief Example Rust-based BPF program tests loop iteration
|
||
|
|
|
||
|
|
#![no_std]
|
||
|
|
#![allow(unused_attributes)]
|
||
|
|
|
||
|
|
extern crate alloc;
|
||
|
|
#[cfg(not(test))]
|
||
|
|
extern crate solana_sdk_bpf_no_std;
|
||
|
|
extern crate solana_sdk_bpf_utils;
|
||
|
|
|
||
|
|
use alloc::vec::Vec;
|
||
|
|
use solana_sdk_bpf_utils::entrypoint::{SolPubkey};
|
||
|
|
|
||
|
|
pub struct InitPollData<'a> {
|
||
|
|
pub timeout: u32,
|
||
|
|
pub header_len: u32,
|
||
|
|
pub header: &'a [u8],
|
||
|
|
pub option_a_len: u32,
|
||
|
|
pub option_a: &'a [u8],
|
||
|
|
pub option_b_len: u32,
|
||
|
|
pub option_b: &'a [u8],
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct PollData<'a> {
|
||
|
|
pub creator_key: &'a SolPubkey,
|
||
|
|
pub last_block: u64,
|
||
|
|
pub header_len: u32,
|
||
|
|
pub header: &'a [u8],
|
||
|
|
pub option_a: PollOptionData<'a>,
|
||
|
|
pub option_b: PollOptionData<'a>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'a> PollData<'a> {
|
||
|
|
pub fn length(&self) -> usize {
|
||
|
|
(32 + 8 + 4 + self.header_len) as usize + self.option_a.length() + self.option_b.length()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn to_bytes(&self) -> Vec<u8> {
|
||
|
|
let mut bytes = Vec::with_capacity(self.length());
|
||
|
|
bytes.extend_from_slice(self.creator_key);
|
||
|
|
bytes.extend_from_slice(&self.last_block.to_be_bytes());
|
||
|
|
bytes.extend_from_slice(&self.header_len.to_be_bytes());
|
||
|
|
bytes.extend_from_slice(self.header);
|
||
|
|
bytes.extend(self.option_a.to_bytes().into_iter());
|
||
|
|
bytes.extend(self.option_b.to_bytes().into_iter());
|
||
|
|
bytes
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn init(
|
||
|
|
init: InitPollData<'a>,
|
||
|
|
creator_key: &'a SolPubkey,
|
||
|
|
tally_a_key: &'a SolPubkey,
|
||
|
|
tally_b_key: &'a SolPubkey,
|
||
|
|
slot: u64,
|
||
|
|
) -> Self {
|
||
|
|
assert_eq!(init.timeout, 10);
|
||
|
|
Self {
|
||
|
|
creator_key,
|
||
|
|
last_block: slot + init.timeout as u64,
|
||
|
|
header_len: init.header_len,
|
||
|
|
header: init.header,
|
||
|
|
option_a: PollOptionData {
|
||
|
|
text_len: init.option_a_len,
|
||
|
|
text: init.option_a,
|
||
|
|
tally_key: tally_a_key,
|
||
|
|
quantity: 0,
|
||
|
|
},
|
||
|
|
option_b: PollOptionData {
|
||
|
|
text_len: init.option_b_len,
|
||
|
|
text: init.option_b,
|
||
|
|
tally_key: tally_b_key,
|
||
|
|
quantity: 0,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub struct PollOptionData<'a> {
|
||
|
|
pub text_len: u32,
|
||
|
|
pub text: &'a [u8],
|
||
|
|
pub tally_key: &'a SolPubkey,
|
||
|
|
pub quantity: u64,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<'a> PollOptionData<'a> {
|
||
|
|
pub fn length(&self) -> usize {
|
||
|
|
(4 + self.text_len + 32 + 8) as usize
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn to_bytes(&self) -> Vec<u8> {
|
||
|
|
let mut bytes = Vec::with_capacity(self.length());
|
||
|
|
bytes.extend_from_slice(&self.text_len.to_be_bytes());
|
||
|
|
bytes.extend_from_slice(self.text);
|
||
|
|
bytes.extend_from_slice(self.tally_key);
|
||
|
|
bytes.extend_from_slice(&self.quantity.to_be_bytes());
|
||
|
|
bytes
|
||
|
|
}
|
||
|
|
}
|