1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71

#[cfg(feature = "native")]
use crate::default_context::DefaultContext;
#[cfg(feature = "native")]
use crate::default_signature::private_key::DefaultPrivateKey;
#[cfg(feature = "native")]
use crate::default_signature::DefaultSignature;
use crate::Context;
use crate::Hasher;
use crate::Signature;
#[cfg(feature = "native")]
use crate::Spec;

/// A Transaction object that is compatible with the module-system/sov-default-stf.
#[derive(Debug, PartialEq, Eq, Clone, borsh::BorshDeserialize, borsh::BorshSerialize)]
pub struct Transaction<C: Context> {
    signature: C::Signature,
    pub_key: C::PublicKey,
    runtime_msg: Vec<u8>,
    nonce: u64,
}

impl<C: Context> Transaction<C> {
    pub fn new(msg: Vec<u8>, pub_key: C::PublicKey, signature: C::Signature, nonce: u64) -> Self {
        Self {
            signature,
            runtime_msg: msg,
            pub_key,
            nonce,
        }
    }

    pub fn signature(&self) -> &C::Signature {
        &self.signature
    }

    pub fn pub_key(&self) -> &C::PublicKey {
        &self.pub_key
    }

    pub fn runtime_msg(&self) -> &[u8] {
        &self.runtime_msg
    }

    pub fn nonce(&self) -> u64 {
        self.nonce
    }

    /// Check whether the transaction has been signed correctly.
    pub fn verify(&self) -> anyhow::Result<()> {
        // We check signature against runtime_msg and nonce.
        let mut hasher = C::Hasher::new();
        hasher.update(self.runtime_msg());
        hasher.update(&self.nonce().to_le_bytes());
        let msg_hash = hasher.finalize();
        self.signature().verify(self.pub_key(), msg_hash)?;

        Ok(())
    }
}

#[cfg(feature = "native")]
impl Transaction<DefaultContext> {
    /// Sign the transaction.
    pub fn sign(priv_key: &DefaultPrivateKey, message: &[u8], nonce: u64) -> DefaultSignature {
        let mut hasher = <DefaultContext as Spec>::Hasher::new();
        hasher.update(message);
        hasher.update(&nonce.to_le_bytes());
        let msg_hash = hasher.finalize();
        priv_key.sign(msg_hash)
    }
}