2018-03-06 11:18:17 -08:00
|
|
|
//! The `transaction` crate provides functionality for creating log transactions.
|
|
|
|
|
2018-03-07 14:32:22 -08:00
|
|
|
use signature::{KeyPair, KeyPairUtil, PublicKey, Signature, SignatureUtil};
|
2018-03-06 11:18:17 -08:00
|
|
|
use serde::Serialize;
|
|
|
|
use bincode::serialize;
|
2018-03-06 16:36:45 -08:00
|
|
|
use hash::Hash;
|
2018-03-07 20:55:49 -08:00
|
|
|
use chrono::prelude::*;
|
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize, Debug, PartialEq, Eq, Clone)]
|
|
|
|
pub enum Condition {
|
2018-03-07 21:25:45 -08:00
|
|
|
Timestamp(DateTime<Utc>),
|
|
|
|
Signature(PublicKey),
|
2018-03-07 20:55:49 -08:00
|
|
|
}
|
2018-03-06 11:18:17 -08:00
|
|
|
|
|
|
|
#[derive(Serialize, Deserialize, Debug, PartialEq, Eq, Clone)]
|
|
|
|
pub struct Transaction<T> {
|
|
|
|
pub from: PublicKey,
|
|
|
|
pub to: PublicKey,
|
2018-03-07 20:55:49 -08:00
|
|
|
pub if_all: Vec<Condition>,
|
|
|
|
pub unless_any: Vec<Condition>,
|
2018-03-06 13:37:08 -08:00
|
|
|
pub asset: T,
|
2018-03-06 16:36:45 -08:00
|
|
|
pub last_id: Hash,
|
2018-03-06 11:18:17 -08:00
|
|
|
pub sig: Signature,
|
|
|
|
}
|
|
|
|
|
2018-03-06 11:26:39 -08:00
|
|
|
impl<T: Serialize> Transaction<T> {
|
2018-03-07 10:05:06 -08:00
|
|
|
pub fn new(from_keypair: &KeyPair, to: PublicKey, asset: T, last_id: Hash) -> Self {
|
2018-03-06 15:34:14 -08:00
|
|
|
let mut tr = Transaction {
|
2018-03-07 14:32:22 -08:00
|
|
|
from: from_keypair.pubkey(),
|
2018-03-06 11:18:17 -08:00
|
|
|
to,
|
2018-03-07 20:55:49 -08:00
|
|
|
if_all: vec![],
|
|
|
|
unless_any: vec![],
|
|
|
|
asset,
|
|
|
|
last_id,
|
|
|
|
sig: Signature::default(),
|
|
|
|
};
|
|
|
|
tr.sign(from_keypair);
|
|
|
|
tr
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn new_on_date(
|
|
|
|
from_keypair: &KeyPair,
|
|
|
|
to: PublicKey,
|
|
|
|
dt: DateTime<Utc>,
|
|
|
|
asset: T,
|
|
|
|
last_id: Hash,
|
|
|
|
) -> Self {
|
2018-03-07 21:25:45 -08:00
|
|
|
let from = from_keypair.pubkey();
|
2018-03-07 20:55:49 -08:00
|
|
|
let mut tr = Transaction {
|
2018-03-07 21:25:45 -08:00
|
|
|
from,
|
2018-03-07 20:55:49 -08:00
|
|
|
to,
|
2018-03-07 21:25:45 -08:00
|
|
|
if_all: vec![Condition::Timestamp(dt)],
|
|
|
|
unless_any: vec![Condition::Signature(from)],
|
2018-03-06 13:37:08 -08:00
|
|
|
asset,
|
2018-03-06 11:18:17 -08:00
|
|
|
last_id,
|
2018-03-06 15:34:14 -08:00
|
|
|
sig: Signature::default(),
|
|
|
|
};
|
|
|
|
tr.sign(from_keypair);
|
|
|
|
tr
|
2018-03-06 11:18:17 -08:00
|
|
|
}
|
2018-03-06 11:26:39 -08:00
|
|
|
|
2018-03-06 15:34:14 -08:00
|
|
|
fn get_sign_data(&self) -> Vec<u8> {
|
2018-03-07 22:37:16 -08:00
|
|
|
serialize(&(
|
|
|
|
&self.from,
|
|
|
|
&self.to,
|
|
|
|
&self.if_all,
|
|
|
|
&self.unless_any,
|
|
|
|
&self.asset,
|
|
|
|
&self.last_id,
|
|
|
|
)).unwrap()
|
2018-03-06 11:26:39 -08:00
|
|
|
}
|
2018-03-06 11:18:17 -08:00
|
|
|
|
2018-03-07 10:05:06 -08:00
|
|
|
pub fn sign(&mut self, keypair: &KeyPair) {
|
2018-03-06 15:34:14 -08:00
|
|
|
let sign_data = self.get_sign_data();
|
|
|
|
self.sig = Signature::clone_from_slice(keypair.sign(&sign_data).as_ref());
|
|
|
|
}
|
2018-03-06 11:18:17 -08:00
|
|
|
|
2018-03-06 15:34:14 -08:00
|
|
|
pub fn verify(&self) -> bool {
|
2018-03-07 14:32:22 -08:00
|
|
|
self.sig.verify(&self.from, &self.get_sign_data())
|
2018-03-06 15:34:14 -08:00
|
|
|
}
|
2018-03-06 11:18:17 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use super::*;
|
|
|
|
use bincode::{deserialize, serialize};
|
2018-03-06 16:20:37 -08:00
|
|
|
use hash::hash;
|
2018-03-06 15:34:14 -08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_claim() {
|
2018-03-07 14:32:22 -08:00
|
|
|
let keypair = KeyPair::new();
|
2018-03-06 15:34:14 -08:00
|
|
|
let asset = hash(b"hello, world");
|
2018-03-06 16:36:45 -08:00
|
|
|
let zero = Hash::default();
|
2018-03-07 14:32:22 -08:00
|
|
|
let tr0 = Transaction::new(&keypair, keypair.pubkey(), asset, zero);
|
2018-03-06 15:34:14 -08:00
|
|
|
assert!(tr0.verify());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_transfer() {
|
2018-03-06 16:36:45 -08:00
|
|
|
let zero = Hash::default();
|
2018-03-07 14:32:22 -08:00
|
|
|
let keypair0 = KeyPair::new();
|
|
|
|
let keypair1 = KeyPair::new();
|
|
|
|
let pubkey1 = keypair1.pubkey();
|
2018-03-06 15:34:14 -08:00
|
|
|
let asset = hash(b"hello, world");
|
|
|
|
let tr0 = Transaction::new(&keypair0, pubkey1, asset, zero);
|
|
|
|
assert!(tr0.verify());
|
|
|
|
}
|
2018-03-06 11:18:17 -08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_serialize_claim() {
|
2018-03-06 15:34:14 -08:00
|
|
|
let claim0 = Transaction {
|
|
|
|
from: Default::default(),
|
|
|
|
to: Default::default(),
|
2018-03-07 20:55:49 -08:00
|
|
|
if_all: Default::default(),
|
|
|
|
unless_any: Default::default(),
|
2018-03-06 15:34:14 -08:00
|
|
|
asset: 0u8,
|
|
|
|
last_id: Default::default(),
|
|
|
|
sig: Default::default(),
|
|
|
|
};
|
2018-03-06 11:18:17 -08:00
|
|
|
let buf = serialize(&claim0).unwrap();
|
|
|
|
let claim1: Transaction<u8> = deserialize(&buf).unwrap();
|
|
|
|
assert_eq!(claim1, claim0);
|
|
|
|
}
|
2018-03-06 15:34:14 -08:00
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_bad_event_signature() {
|
2018-03-06 16:36:45 -08:00
|
|
|
let zero = Hash::default();
|
2018-03-07 14:32:22 -08:00
|
|
|
let keypair = KeyPair::new();
|
|
|
|
let pubkey = keypair.pubkey();
|
2018-03-06 15:34:14 -08:00
|
|
|
let mut tr = Transaction::new(&keypair, pubkey, hash(b"hello, world"), zero);
|
|
|
|
tr.sign(&keypair);
|
|
|
|
tr.asset = hash(b"goodbye cruel world"); // <-- attack!
|
|
|
|
assert!(!tr.verify());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_hijack_attack() {
|
2018-03-07 14:32:22 -08:00
|
|
|
let keypair0 = KeyPair::new();
|
|
|
|
let keypair1 = KeyPair::new();
|
|
|
|
let thief_keypair = KeyPair::new();
|
|
|
|
let pubkey1 = keypair1.pubkey();
|
2018-03-06 16:36:45 -08:00
|
|
|
let zero = Hash::default();
|
2018-03-06 15:34:14 -08:00
|
|
|
let mut tr = Transaction::new(&keypair0, pubkey1, hash(b"hello, world"), zero);
|
|
|
|
tr.sign(&keypair0);
|
2018-03-07 14:32:22 -08:00
|
|
|
tr.to = thief_keypair.pubkey(); // <-- attack!
|
2018-03-06 15:34:14 -08:00
|
|
|
assert!(!tr.verify());
|
|
|
|
}
|
2018-03-06 11:18:17 -08:00
|
|
|
}
|