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-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-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-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> {
|
|
|
|
serialize(&(&self.from, &self.to, &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(),
|
|
|
|
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
|
|
|
}
|