Hash Time Lock Contracts (#168)

tuxedo-core/src/verifier.rs

@@ -10,9 +10,11 @@ use scale_info::TypeInfo;
 use serde::{Deserialize, Serialize};
 use sp_std::fmt::Debug;
 
+mod htlc;
 mod multi_signature;
 mod simple_signature;
 
+pub use htlc::{BlakeTwoHashLock, TimeLock};
 pub use multi_signature::ThresholdMultiSignature;
 pub use simple_signature::{Sr25519Signature, P2PKH};
 

tuxedo-core/src/verifier/htlc.rs

@@ -0,0 +1,386 @@
+//! This module contains `Verifier` implementations related to Hash Time Lock Contracts.
+//! It contains a simple hash lock, a simple time lock, and a hash time lock.
+//!
+//! These could be used as the base of an atomic swap protocol with a similarly expressive
+//! utxo chain like Bitcoin. For atomic swaps with less expressive counter party chains,
+//! such as Monero, see the Farcaster protocol.
+
+use super::Verifier;
+use parity_scale_codec::{Decode, Encode};
+use scale_info::TypeInfo;
+use serde::{Deserialize, Serialize};
+use sp_core::{
+    sr25519::{Public, Signature},
+    H256,
+};
+use sp_runtime::traits::{BlakeTwo256, Hash};
+use sp_std::vec::Vec;
+
+/// Allows UTXOs to be spent after a certain block height has been reached.
+/// This is useful for locking up tokens as a future investment. Timelocking
+/// also form the basis of timeout paths in swapping protocols.
+///
+/// This verifier is unlike many others because it requires some environmental information,
+/// namely the current block number. So there is a decision to be made:
+/// * Allow the verifier to take come config and grab that info by calling a function given in the config.
+///   This is what we do with constraint checker.
+/// * Modify the `Verifier` trait to pass along the block number.
+///
+/// On the one hand the block number seems like a pretty fundamental and basic thing to add. On the other
+/// hand there could be many more things to pass. For example, the timestamp.
+/// However any more complex information would require coupling with Constraint Checkers and it is not
+/// easy to red state like in accounts.
+///
+/// Decision: I will add block number to the signature. And remain open to adding more blockchain-level
+/// fundamental things. Especially if they are available in bitcoin script.
+///
+/// Regarding the verifier constraint checker separation, perhaps the right line to be drawn is
+/// that verifiers are useful in a lot of places, but perhaps not expressive enough in others.
+/// When they are not expressive enough, just use `UpForGrabs` and rely on the constraint checker,
+#[derive(Serialize, Deserialize, Encode, Decode, Debug, PartialEq, Eq, Clone, TypeInfo)]
+pub struct TimeLock {
+    pub unlock_block_height: u32,
+}
+
+impl Verifier for TimeLock {
+    type Redeemer = ();
+    fn verify(&self, _: &[u8], block_height: u32, _: &()) -> bool {
+        block_height >= self.unlock_block_height
+    }
+}
+
+/// Allows UTXOs to be spent when a preimage to a recorded hash is provided.
+/// This could be used as a puzzle (although a partial preimage search would be better)
+/// or a means of sharing a password, or as part of a simple atomic swapping protocol.
+#[derive(Serialize, Deserialize, Encode, Decode, Debug, PartialEq, Eq, Clone, TypeInfo)]
+pub struct BlakeTwoHashLock {
+    pub hash_lock: H256,
+}
+
+impl BlakeTwoHashLock {
+    pub fn new_from_secret(secret: Vec<u8>) -> Self {
+        Self {
+            hash_lock: BlakeTwo256::hash(&secret),
+        }
+    }
+}
+
+impl Verifier for BlakeTwoHashLock {
+    type Redeemer = Vec<u8>;
+    fn verify(&self, _: &[u8], _: u32, secret: &Self::Redeemer) -> bool {
+        BlakeTwo256::hash(secret) == self.hash_lock
+    }
+}
+
+/// Allows a UTXO to be spent, and therefore acknowledged by an intended recipient by revealing
+/// a hash preimage. After an initial claim period elapses on chain, the UTXO can also be spent
+/// by the refunder. In practice, the refunder is often the same address initially funded the HTLC.
+///
+/// The receiver and refunder are specified as a simple public keys for simplicity. It would be
+/// interesting to use public key hash, or better yet, simply abstract this over some opaque
+/// inner verifier for maximum composability.
+///
+/// After the time refund path opens, the happy path remains open. This is for compatibility with
+/// bitcoin, but may not be desired in all cases.
+#[derive(Serialize, Deserialize, Encode, Decode, Debug, PartialEq, Eq, Clone, TypeInfo)]
+pub struct HashTimeLockContract {
+    /// The hash whose preimage must be revealed (along with the recipient's signature) to spend the UTXO.
+    pub hash_lock: H256,
+    /// The pubkey that is intended to receive and acknowledge receipt of the funds.
+    pub recipient_pubkey: Public,
+    /// The time (as a block height) when the refund path opens up.
+    pub claim_period_end: u32,
+    /// The address who can spend the coins without revealing the preimage after the claim period has ended.
+    pub refunder_pubkey: Public,
+}
+
+/// This is the redeemer information needed to spend a `HashTimeLockContract` verifier.
+///
+/// The `HashTimeLockContract` has two spend paths, and therefore this enum has two variants.
+/// The variant selects the spend path and contains the corresponding witness data.
+#[derive(Serialize, Deserialize, Encode, Decode, Debug, PartialEq, Eq, Clone)]
+pub enum HtlcSpendPath {
+    /// The primary spend path is for the recipient to claim the UTXO by revealing the
+    /// hash preimage as well as a signature.
+    Claim {
+        secret: Vec<u8>,
+        signature: Signature,
+    },
+    /// The secondary spend path is for the original owner to refund their money to their private
+    /// ownership. This path is not enabled until the enough time has elapsed. Once the time has
+    /// elapsed, only the refunder's signature is required.
+    Refund { signature: Signature },
+}
+
+impl Verifier for HashTimeLockContract {
+    type Redeemer = HtlcSpendPath;
+
+    fn verify(&self, simplified_tx: &[u8], block_height: u32, spend_path: &HtlcSpendPath) -> bool {
+        match spend_path {
+            HtlcSpendPath::Claim { secret, signature } => {
+                // Claims are valid as long as the secret is correct and the receiver signature is correct.
+                BlakeTwo256::hash(secret) == self.hash_lock
+                    && sp_io::crypto::sr25519_verify(
+                        signature,
+                        simplified_tx,
+                        &self.recipient_pubkey,
+                    )
+            }
+            HtlcSpendPath::Refund { signature } => {
+                // Check that the time has elapsed
+                block_height >= self.claim_period_end
+
+                &&
+
+                // Check that the refunder has signed properly
+                sp_io::crypto::sr25519_verify(
+                    signature,
+                    simplified_tx,
+                    &self.refunder_pubkey,
+                )
+            }
+        }
+    }
+}
+
+#[cfg(test)]
+mod test {
+    use super::*;
+    use sp_core::{sr25519::Pair, Pair as _};
+
+    fn bad_sig() -> Signature {
+        Signature::from_slice(
+            b"bogus_signature_bogus_signature_bogus_signature_bogus_signature!".as_slice(),
+        )
+        .expect("Should be able to create a bogus signature.")
+    }
+
+    #[test]
+    fn time_lock_too_soon() {
+        let time_lock = TimeLock {
+            unlock_block_height: 100,
+        };
+        assert!(!time_lock.verify(&[], 10, &()));
+    }
+
+    #[test]
+    fn time_lock_exactly_on_time() {
+        let time_lock = TimeLock {
+            unlock_block_height: 100,
+        };
+        assert!(time_lock.verify(&[], 100, &()));
+    }
+
+    #[test]
+    fn time_lock_past_threshold() {
+        let time_lock = TimeLock {
+            unlock_block_height: 100,
+        };
+        assert!(time_lock.verify(&[], 200, &()));
+    }
+
+    #[test]
+    fn hash_lock_correct_secret() {
+        let secret = "htlc ftw";
+
+        let hash_lock = BlakeTwoHashLock::new_from_secret(secret.encode());
+        assert!(hash_lock.verify(&[], 0, &secret.encode()));
+    }
+
+    #[test]
+    fn hash_lock_wrong_secret() {
+        let secret = "htlc ftw";
+        let incorrect = "there is no second best";
+
+        let hash_lock = BlakeTwoHashLock::new_from_secret(secret.encode());
+        assert!(!hash_lock.verify(&[], 0, &incorrect.encode()));
+    }
+
+    #[test]
+    fn htlc_claim_success() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let simplified_tx = b"hello world".as_slice();
+        let recipient_sig = recipient_pair.sign(simplified_tx);
+        let redeemer = HtlcSpendPath::Claim {
+            secret,
+            signature: recipient_sig,
+        };
+
+        assert!(htlc.verify(&simplified_tx, 0, &redeemer));
+    }
+
+    #[test]
+    fn htlc_claim_wrong_secret() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let incorrect_secret = "there is no second best".encode();
+
+        let simplified_tx = b"hello world".as_slice();
+        let recipient_sig = recipient_pair.sign(simplified_tx);
+        let redeemer = HtlcSpendPath::Claim {
+            secret: incorrect_secret,
+            signature: recipient_sig,
+        };
+
+        assert!(!htlc.verify(&simplified_tx, 0, &redeemer));
+    }
+
+    #[test]
+    fn htlc_claim_bogus_signature() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let simplified_tx = b"hello world".as_slice();
+        let redeemer = HtlcSpendPath::Claim {
+            secret,
+            signature: bad_sig(),
+        };
+
+        assert!(!htlc.verify(&simplified_tx, 0, &redeemer));
+    }
+
+    #[test]
+    fn htlc_claim_fails_when_signature_is_from_refunder() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let simplified_tx = b"hello world".as_slice();
+        let refunder_sig = refunder_pair.sign(simplified_tx);
+        let redeemer = HtlcSpendPath::Claim {
+            secret,
+            signature: refunder_sig,
+        };
+
+        assert!(!htlc.verify(&simplified_tx, 0, &redeemer));
+    }
+
+    #[test]
+    fn htlc_refund_success() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let simplified_tx = b"hello world".as_slice();
+        let refunder_sig = refunder_pair.sign(simplified_tx);
+        let redeemer = HtlcSpendPath::Refund {
+            signature: refunder_sig,
+        };
+
+        assert!(htlc.verify(&simplified_tx, 2 * THRESHOLD, &redeemer));
+    }
+
+    #[test]
+    fn htlc_refund_too_early() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let simplified_tx = b"hello world".as_slice();
+        let refunder_sig = refunder_pair.sign(simplified_tx);
+        let redeemer = HtlcSpendPath::Refund {
+            signature: refunder_sig,
+        };
+
+        assert!(!htlc.verify(&simplified_tx, 0, &redeemer));
+    }
+
+    #[test]
+    fn htlc_refund_bogus_sig() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let simplified_tx = b"hello world".as_slice();
+        let redeemer = HtlcSpendPath::Refund {
+            signature: bad_sig(),
+        };
+
+        assert!(!htlc.verify(&simplified_tx, 2 * THRESHOLD, &redeemer));
+    }
+
+    #[test]
+    fn htlc_refund_fails_when_signature_is_from_recipient() {
+        const THRESHOLD: u32 = 100;
+        let secret = "htlc ftw".encode();
+        let recipient_pair = Pair::from_seed(&[0u8; 32]);
+        let refunder_pair = Pair::from_seed(&[1u8; 32]);
+
+        let htlc = HashTimeLockContract {
+            hash_lock: BlakeTwo256::hash(&secret),
+            recipient_pubkey: recipient_pair.public(),
+            claim_period_end: THRESHOLD,
+            refunder_pubkey: refunder_pair.public(),
+        };
+
+        let simplified_tx = b"hello world".as_slice();
+        let recipient_sig = recipient_pair.sign(simplified_tx);
+        let redeemer = HtlcSpendPath::Refund {
+            signature: recipient_sig,
+        };
+
+        assert!(!htlc.verify(&simplified_tx, 2 * THRESHOLD, &redeemer));
+    }
+}