2018-03-03 13:41:36 -08:00
|
|
|
//! A library for generating the chain's genesis block.
|
|
|
|
|
2018-03-06 11:18:17 -08:00
|
|
|
use event::Event;
|
2018-03-06 15:34:14 -08:00
|
|
|
use transaction::Transaction;
|
2018-03-06 11:48:26 -08:00
|
|
|
use signature::{generate_keypair, get_pubkey, PublicKey};
|
2018-03-06 16:20:37 -08:00
|
|
|
use log::{create_entries, Entry};
|
|
|
|
use hash::{hash, Sha256Hash};
|
2018-03-03 18:57:22 -08:00
|
|
|
use ring::rand::SystemRandom;
|
2018-03-03 13:41:36 -08:00
|
|
|
use ring::signature::Ed25519KeyPair;
|
|
|
|
use untrusted::Input;
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct Creator {
|
|
|
|
pub pubkey: PublicKey,
|
2018-03-05 16:29:32 -08:00
|
|
|
pub tokens: i64,
|
2018-03-03 13:41:36 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
impl Creator {
|
2018-03-05 16:29:32 -08:00
|
|
|
pub fn new(tokens: i64) -> Self {
|
2018-03-03 13:41:36 -08:00
|
|
|
Creator {
|
|
|
|
pubkey: get_pubkey(&generate_keypair()),
|
|
|
|
tokens,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct Genesis {
|
|
|
|
pub pkcs8: Vec<u8>,
|
2018-03-05 16:29:32 -08:00
|
|
|
pub tokens: i64,
|
2018-03-03 13:41:36 -08:00
|
|
|
pub creators: Vec<Creator>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Genesis {
|
2018-03-05 16:29:32 -08:00
|
|
|
pub fn new(tokens: i64, creators: Vec<Creator>) -> Self {
|
2018-03-03 13:41:36 -08:00
|
|
|
let rnd = SystemRandom::new();
|
|
|
|
let pkcs8 = Ed25519KeyPair::generate_pkcs8(&rnd).unwrap().to_vec();
|
2018-03-05 14:34:15 -08:00
|
|
|
println!("{:?}", pkcs8);
|
2018-03-03 13:41:36 -08:00
|
|
|
Genesis {
|
|
|
|
pkcs8,
|
|
|
|
tokens,
|
|
|
|
creators,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-03-05 11:48:09 -08:00
|
|
|
pub fn get_seed(&self) -> Sha256Hash {
|
|
|
|
hash(&self.pkcs8)
|
|
|
|
}
|
|
|
|
|
2018-03-03 23:13:40 -08:00
|
|
|
pub fn get_keypair(&self) -> Ed25519KeyPair {
|
|
|
|
Ed25519KeyPair::from_pkcs8(Input::from(&self.pkcs8)).unwrap()
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn get_pubkey(&self) -> PublicKey {
|
|
|
|
get_pubkey(&self.get_keypair())
|
|
|
|
}
|
|
|
|
|
2018-03-06 13:37:08 -08:00
|
|
|
pub fn create_transaction(&self, asset: i64, to: &PublicKey) -> Event<i64> {
|
2018-03-06 15:34:14 -08:00
|
|
|
let tr = Transaction::new(&self.get_keypair(), *to, asset, self.get_seed());
|
|
|
|
Event::Transaction(tr)
|
2018-03-03 23:13:40 -08:00
|
|
|
}
|
|
|
|
|
2018-03-05 16:29:32 -08:00
|
|
|
pub fn create_events(&self) -> Vec<Event<i64>> {
|
2018-03-03 23:13:40 -08:00
|
|
|
let pubkey = self.get_pubkey();
|
2018-03-03 18:57:22 -08:00
|
|
|
let event0 = Event::Tick;
|
2018-03-03 23:13:40 -08:00
|
|
|
let event1 = self.create_transaction(self.tokens, &pubkey);
|
2018-03-03 18:57:22 -08:00
|
|
|
let mut events = vec![event0, event1];
|
2018-03-03 13:41:36 -08:00
|
|
|
|
2018-03-03 23:13:40 -08:00
|
|
|
for x in &self.creators {
|
|
|
|
let tx = self.create_transaction(x.tokens, &x.pubkey);
|
2018-03-03 13:41:36 -08:00
|
|
|
events.push(tx);
|
|
|
|
}
|
|
|
|
|
|
|
|
events
|
|
|
|
}
|
2018-03-05 12:03:56 -08:00
|
|
|
|
2018-03-05 16:29:32 -08:00
|
|
|
pub fn create_entries(&self) -> Vec<Entry<i64>> {
|
2018-03-05 12:03:56 -08:00
|
|
|
create_entries(&self.get_seed(), self.create_events())
|
|
|
|
}
|
2018-03-03 13:41:36 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
2018-03-05 16:29:32 -08:00
|
|
|
use log::verify_slice_i64;
|
2018-03-03 13:41:36 -08:00
|
|
|
|
|
|
|
#[test]
|
2018-03-03 23:13:40 -08:00
|
|
|
fn test_create_events() {
|
|
|
|
let mut events = Genesis::new(100, vec![]).create_events().into_iter();
|
|
|
|
assert_eq!(events.next().unwrap(), Event::Tick);
|
2018-03-06 10:54:45 -08:00
|
|
|
if let Event::Transaction(Transaction { from, to, .. }) = events.next().unwrap() {
|
2018-03-03 23:13:40 -08:00
|
|
|
assert_eq!(from, to);
|
|
|
|
} else {
|
|
|
|
assert!(false);
|
|
|
|
}
|
|
|
|
assert_eq!(events.next(), None);
|
2018-03-03 13:41:36 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
2018-03-03 23:13:40 -08:00
|
|
|
fn test_create_creator() {
|
2018-03-03 13:41:36 -08:00
|
|
|
assert_eq!(
|
2018-03-04 00:21:40 -08:00
|
|
|
Genesis::new(100, vec![Creator::new(42)])
|
2018-03-03 13:41:36 -08:00
|
|
|
.create_events()
|
|
|
|
.len(),
|
2018-03-03 18:57:22 -08:00
|
|
|
3
|
2018-03-03 13:41:36 -08:00
|
|
|
);
|
|
|
|
}
|
2018-03-05 14:34:15 -08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_verify_entries() {
|
|
|
|
let entries = Genesis::new(100, vec![]).create_entries();
|
2018-03-05 16:29:32 -08:00
|
|
|
assert!(verify_slice_i64(&entries, &entries[0].id));
|
2018-03-05 14:34:15 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_verify_entries_with_transactions() {
|
|
|
|
let entries = Genesis::new(100, vec![Creator::new(42)]).create_entries();
|
2018-03-05 16:29:32 -08:00
|
|
|
assert!(verify_slice_i64(&entries, &entries[0].id));
|
2018-03-05 14:34:15 -08:00
|
|
|
}
|
2018-03-03 13:41:36 -08:00
|
|
|
}
|