Core/grin
0
0
Fork 0
mirror of https://github.com/mimblewimble/grin.git synced 2025-01-21 03:21:08 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Verify features in one place; comment on needed Features enum type (#2191)

Browse source
This commit is contained in:
John Tromp 2018-12-22 19:27:54 +01:00 committed by Ignotus Peverell
parent 976bf1dbae
commit 151aa1c842
1 changed files with 23 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
core/src/core/transaction.rs
View file

@ -36,6 +36,14 @@ use std::collections::HashSet;
use std::sync::Arc;
use std::{error, fmt};
/// KernelFeatures needs to be a sumtype / tagged union as
/// pub enum KernelFeatures {
/// COINBASE,
/// PLAIN { fee: u64 },
/// HEIGHTLOCKED { fee: u64, lock_height: u64 },
/// }
/// Similarly for OutputFeatures
///
bitflags! {
/// Options for a kernel's structure or use
#[derive(Serialize, Deserialize)]
@ -253,10 +261,6 @@ impl TxKernel {
/// as a public key and checking the signature verifies with the fee as
/// message.
pub fn verify(&self) -> Result<(), Error> {
if self.is_coinbase() && self.fee != 0 || !self.is_height_locked() && self.lock_height != 0
{
return Err(Error::InvalidKernelFeatures);
}
let secp = static_secp_instance();
let secp = secp.lock();
let sig = &self.excess_sig;
@ -1349,12 +1353,7 @@ impl From<Output> for OutputIdentifier {
}
/// Construct msg from tx fee, lock_height and kernel features.
/// In testnet4 we did not include the kernel features in the message being signed.
/// In mainnet we changed this to include features and we hash (fee || lock_height || features)
/// to produce a 32 byte message to sign.
///
/// testnet4: msg = (fee || lock_height)
/// mainnet: msg = hash(features) for coinbase kernels
/// msg = hash(features) for coinbase kernels
/// hash(features || fee) for plain kernels
/// hash(features || fee || lock_height) for height locked kernels
///
@ -1363,22 +1362,21 @@ pub fn kernel_sig_msg(
lock_height: u64,
features: KernelFeatures,
) -> Result<secp::Message, Error> {
if features.is_coinbase() && fee != 0 || !features.is_height_locked() && lock_height != 0 {
let features_ok = match features {
KernelFeatures::COINBASE => fee == 0 && lock_height == 0,
KernelFeatures::PLAIN => lock_height == 0,
KernelFeatures::HEIGHT_LOCKED => true,
_ => false
};
if !features_ok {
return Err(Error::InvalidKernelFeatures);
}
let msg = if global::is_testnet() {
let mut bytes = [0; 32];
BigEndian::write_u64(&mut bytes[16..24], fee);
BigEndian::write_u64(&mut bytes[24..], lock_height);
secp::Message::from_slice(&bytes)?
} else {
let hash = match features {
KernelFeatures::COINBASE => (features).hash(),
KernelFeatures::COINBASE => features.hash(),
KernelFeatures::PLAIN => (features, fee).hash(),
_ => (features, fee, lock_height).hash(),
};
secp::Message::from_slice(&hash.as_bytes())?
};
let msg = secp::Message::from_slice(&hash.as_bytes())?;
Ok(msg)
}