{-# LANGUAGE LambdaCase #-}
{-|
Module      : Haskoin.Test.Script
Copyright   : No rights reserved
License     : MIT
Maintainer  : jprupp@protonmail.ch
Stability   : experimental
Portability : POSIX
-}
module Haskoin.Util.Arbitrary.Script where

import           Crypto.Secp256k1
import qualified Data.ByteString                as B
import           Data.Maybe
import           Data.Word
import           Haskoin.Address
import           Haskoin.Constants
import           Haskoin.Keys.Common
import           Haskoin.Script
import           Haskoin.Transaction.Common
import           Haskoin.Util
import           Haskoin.Util.Arbitrary.Address
import           Haskoin.Util.Arbitrary.Crypto
import           Haskoin.Util.Arbitrary.Keys
import           Haskoin.Util.Arbitrary.Util
import           Test.QuickCheck

-- | Arbitrary 'Script' with random script ops.
arbitraryScript :: Gen Script
arbitraryScript :: Gen Script
arbitraryScript = [ScriptOp] -> Script
Script ([ScriptOp] -> Script) -> Gen [ScriptOp] -> Gen Script
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ScriptOp -> Gen [ScriptOp]
forall a. Gen a -> Gen [a]
listOf Gen ScriptOp
arbitraryScriptOp

-- | Arbitrary 'ScriptOp' (push operations have random data).
arbitraryScriptOp :: Gen ScriptOp
arbitraryScriptOp :: Gen ScriptOp
arbitraryScriptOp =
    [Gen ScriptOp] -> Gen ScriptOp
forall a. [Gen a] -> Gen a
oneof
          -- Pushing Data
        [ ByteString -> ScriptOp
opPushData (ByteString -> ScriptOp) -> Gen ByteString -> Gen ScriptOp
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ByteString
arbitraryBS1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_0
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1NEGATE
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RESERVED
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_3
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_4
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_5
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_6
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_7
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_8
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_9
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_10
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_11
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_12
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_13
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_14
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_15
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_16
        -- Flow control
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VER
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_IF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOTIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VERIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VERNOTIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ELSE
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ENDIF
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_VERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RETURN
        -- Stack operations
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_TOALTSTACK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_FROMALTSTACK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_IFDUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DEPTH
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DROP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NIP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_OVER
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_PICK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ROLL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ROT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SWAP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_TUCK
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2DROP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2DUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_3DUP
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2OVER
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2ROT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2SWAP
        -- Splice
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CAT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SUBSTR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LEFT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RIGHT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SIZE
        -- Bitwise logic
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_INVERT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_AND
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_OR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_XOR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_EQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_EQUALVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RESERVED1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RESERVED2
        -- Arithmetic
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1ADD
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_1SUB
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2MUL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_2DIV
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NEGATE
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ABS
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_0NOTEQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_ADD
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SUB
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MUL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_DIV
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MOD
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LSHIFT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RSHIFT
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_BOOLAND
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_BOOLOR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NUMEQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NUMEQUALVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NUMNOTEQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LESSTHAN
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_GREATERTHAN
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_LESSTHANOREQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_GREATERTHANOREQUAL
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MIN
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_MAX
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_WITHIN
        -- Crypto
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_RIPEMD160
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SHA1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_SHA256
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_HASH160
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_HASH256
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CODESEPARATOR
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKSIG
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKSIGVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKMULTISIG
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKMULTISIGVERIFY
        -- Expansion
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP1
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKLOCKTIMEVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKSEQUENCEVERIFY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP4
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP5
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP6
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP7
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP8
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP9
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_NOP10
        -- Bitcoin Cash Nov 2018 hard fork
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKDATASIG
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_CHECKDATASIGVERIFY
        -- Bitcoin Cash May 2020 hard fork
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_REVERSEBYTES
        -- Other
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_PUBKEYHASH
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return ScriptOp
OP_PUBKEY
        , ScriptOp -> Gen ScriptOp
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOp -> Gen ScriptOp) -> ScriptOp -> Gen ScriptOp
forall a b. (a -> b) -> a -> b
$ Word8 -> ScriptOp
OP_INVALIDOPCODE 0xff
        ]

-- | Arbtirary 'ScriptOp' with a value in @[OP_1 .. OP_16]@.
arbitraryIntScriptOp :: Gen ScriptOp
arbitraryIntScriptOp :: Gen ScriptOp
arbitraryIntScriptOp =
    [ScriptOp] -> Gen ScriptOp
forall a. [a] -> Gen a
elements
        [ ScriptOp
OP_1,  ScriptOp
OP_2,  ScriptOp
OP_3,  ScriptOp
OP_4
        , ScriptOp
OP_5,  ScriptOp
OP_6,  ScriptOp
OP_7,  ScriptOp
OP_8
        , ScriptOp
OP_9,  ScriptOp
OP_10, ScriptOp
OP_11, ScriptOp
OP_12
        , ScriptOp
OP_13, ScriptOp
OP_14, ScriptOp
OP_15, ScriptOp
OP_16
        ]

-- | Arbitrary 'PushDataType'.
arbitraryPushDataType :: Gen PushDataType
arbitraryPushDataType :: Gen PushDataType
arbitraryPushDataType = [PushDataType] -> Gen PushDataType
forall a. [a] -> Gen a
elements [PushDataType
OPCODE, PushDataType
OPDATA1, PushDataType
OPDATA2, PushDataType
OPDATA4]

-- | Arbitrary 'SigHash' (including invalid/unknown sighash codes).
arbitrarySigHash :: Gen SigHash
arbitrarySigHash :: Gen SigHash
arbitrarySigHash = Word32 -> SigHash
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word32 -> SigHash) -> Gen Word32 -> Gen SigHash
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Gen Word32
forall a. Arbitrary a => Gen a
arbitrary :: Gen Word32)

-- | Arbitrary valid 'SigHash'.
arbitraryValidSigHash :: Network -> Gen SigHash
arbitraryValidSigHash :: Network -> Gen SigHash
arbitraryValidSigHash net :: Network
net = do
    SigHash
sh <- [SigHash] -> Gen SigHash
forall a. [a] -> Gen a
elements [SigHash
sigHashAll, SigHash
sigHashNone, SigHash
sigHashSingle]
    SigHash -> SigHash
f1 <-
        [SigHash -> SigHash] -> Gen (SigHash -> SigHash)
forall a. [a] -> Gen a
elements ([SigHash -> SigHash] -> Gen (SigHash -> SigHash))
-> [SigHash -> SigHash] -> Gen (SigHash -> SigHash)
forall a b. (a -> b) -> a -> b
$
        if Maybe Word32 -> Bool
forall a. Maybe a -> Bool
isJust (Network -> Maybe Word32
getSigHashForkId Network
net)
            then [SigHash -> SigHash
forall a. a -> a
id, SigHash -> SigHash
setForkIdFlag]
            else [SigHash -> SigHash
forall a. a -> a
id]
    SigHash -> SigHash
f2 <- [SigHash -> SigHash] -> Gen (SigHash -> SigHash)
forall a. [a] -> Gen a
elements [SigHash -> SigHash
forall a. a -> a
id, SigHash -> SigHash
setAnyoneCanPayFlag]
    SigHash -> Gen SigHash
forall (m :: * -> *) a. Monad m => a -> m a
return (SigHash -> Gen SigHash) -> SigHash -> Gen SigHash
forall a b. (a -> b) -> a -> b
$ SigHash -> SigHash
f1 (SigHash -> SigHash) -> SigHash -> SigHash
forall a b. (a -> b) -> a -> b
$ SigHash -> SigHash
f2 SigHash
sh

arbitrarySigHashFlag :: Gen SigHashFlag
arbitrarySigHashFlag :: Gen SigHashFlag
arbitrarySigHashFlag =
    [SigHashFlag] -> Gen SigHashFlag
forall a. [a] -> Gen a
elements
        [ SigHashFlag
SIGHASH_ALL
        , SigHashFlag
SIGHASH_NONE
        , SigHashFlag
SIGHASH_SINGLE
        , SigHashFlag
SIGHASH_FORKID
        , SigHashFlag
SIGHASH_ANYONECANPAY
        ]

-- | Arbitrary message hash, private key and corresponding 'TxSignature'. The
-- signature is generated deterministically using a random message and a random
-- private key.
arbitraryTxSignature :: Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature :: Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature net :: Network
net = do
    (m :: Hash256
m, key :: SecKey
key, sig :: Sig
sig) <- Gen (Hash256, SecKey, Sig)
arbitrarySignature
    SigHash
sh <- (Word8 -> SigHash
forall a b. (Integral a, Num b) => a -> b
fromIntegral (Word8 -> SigHash) -> Gen Word8 -> Gen SigHash
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> (Gen Word8
forall a. Arbitrary a => Gen a
arbitrary :: Gen Word8)) Gen SigHash -> (SigHash -> Bool) -> Gen SigHash
forall a. Gen a -> (a -> Bool) -> Gen a
`suchThat` SigHash -> Bool
filterBad
    let txsig :: TxSignature
txsig = Sig -> SigHash -> TxSignature
TxSignature Sig
sig SigHash
sh
    (TxHash, SecKey, TxSignature) -> Gen (TxHash, SecKey, TxSignature)
forall (m :: * -> *) a. Monad m => a -> m a
return (Hash256 -> TxHash
TxHash Hash256
m, SecKey
key, TxSignature
txsig)
  where
    filterBad :: SigHash -> Bool
filterBad sh :: SigHash
sh = Bool -> Bool
not (Bool -> Bool) -> Bool -> Bool
forall a b. (a -> b) -> a -> b
$
        SigHash -> Bool
isSigHashUnknown SigHash
sh Bool -> Bool -> Bool
||
        Maybe Word32 -> Bool
forall a. Maybe a -> Bool
isNothing (Network -> Maybe Word32
getSigHashForkId Network
net) Bool -> Bool -> Bool
&& SigHash -> Bool
hasForkIdFlag SigHash
sh

-- | Arbitrary transaction signature that could also be empty.
arbitraryTxSignatureEmpty :: Network -> Gen TxSignature
arbitraryTxSignatureEmpty :: Network -> Gen TxSignature
arbitraryTxSignatureEmpty net :: Network
net =
    [(Int, Gen TxSignature)] -> Gen TxSignature
forall a. [(Int, Gen a)] -> Gen a
frequency [ (1, TxSignature -> Gen TxSignature
forall (m :: * -> *) a. Monad m => a -> m a
return TxSignature
TxSignatureEmpty)
              , (10, (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ((TxHash, SecKey, TxSignature) -> TxSignature)
-> Gen (TxHash, SecKey, TxSignature) -> Gen TxSignature
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net)
              ]

-- | Arbitrary m of n parameters.
arbitraryMSParam :: Gen (Int, Int)
arbitraryMSParam :: Gen (Int, Int)
arbitraryMSParam = do
    Int
m <- (Int, Int) -> Gen Int
forall a. Random a => (a, a) -> Gen a
choose (1,16)
    Int
n <- (Int, Int) -> Gen Int
forall a. Random a => (a, a) -> Gen a
choose (Int
m,16)
    (Int, Int) -> Gen (Int, Int)
forall (m :: * -> *) a. Monad m => a -> m a
return (Int
m, Int
n)

-- | Arbitrary 'ScriptOutput' (Can by any valid type).
arbitraryScriptOutput :: Network -> Gen ScriptOutput
arbitraryScriptOutput :: Network -> Gen ScriptOutput
arbitraryScriptOutput net :: Network
net =
    [Gen ScriptOutput] -> Gen ScriptOutput
forall a. [Gen a] -> Gen a
oneof ([Gen ScriptOutput] -> Gen ScriptOutput)
-> [Gen ScriptOutput] -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$
    [ Gen ScriptOutput
arbitraryPKOutput
    , Gen ScriptOutput
arbitraryPKHashOutput
    , Gen ScriptOutput
arbitraryMSOutput
    , Gen ScriptOutput
arbitrarySHOutput
    , Gen ScriptOutput
arbitraryDCOutput
    ] [Gen ScriptOutput] -> [Gen ScriptOutput] -> [Gen ScriptOutput]
forall a. [a] -> [a] -> [a]
++
    if Network -> Bool
getSegWit Network
net
        then [ Gen ScriptOutput
arbitraryWPKHashOutput
             , Gen ScriptOutput
arbitraryWSHOutput
             , Gen ScriptOutput
arbitraryWitOutput
             ]
        else []

-- | Arbitrary 'ScriptOutput' of type 'PayPK', 'PayPKHash' or 'PayMS'
-- (Not 'PayScriptHash', 'DataCarrier', or SegWit)
arbitrarySimpleOutput :: Gen ScriptOutput
arbitrarySimpleOutput :: Gen ScriptOutput
arbitrarySimpleOutput =
    [Gen ScriptOutput] -> Gen ScriptOutput
forall a. [Gen a] -> Gen a
oneof
        [ Gen ScriptOutput
arbitraryPKOutput
        , Gen ScriptOutput
arbitraryPKHashOutput
        , Gen ScriptOutput
arbitraryMSOutput
        ]

-- | Arbitrary 'ScriptOutput' of type 'PayPK'
arbitraryPKOutput :: Gen ScriptOutput
arbitraryPKOutput :: Gen ScriptOutput
arbitraryPKOutput =  PubKeyI -> ScriptOutput
PayPK (PubKeyI -> ScriptOutput)
-> ((SecKeyI, PubKeyI) -> PubKeyI)
-> (SecKeyI, PubKeyI)
-> ScriptOutput
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ((SecKeyI, PubKeyI) -> ScriptOutput)
-> Gen (SecKeyI, PubKeyI) -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (SecKeyI, PubKeyI)
arbitraryKeyPair

-- | Arbitrary 'ScriptOutput' of type 'PayPKHash'
arbitraryPKHashOutput :: Gen ScriptOutput
arbitraryPKHashOutput :: Gen ScriptOutput
arbitraryPKHashOutput = Hash160 -> ScriptOutput
PayPKHash (Hash160 -> ScriptOutput) -> Gen Hash160 -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Hash160
arbitraryHash160

-- | Arbitrary 'PayWitnessPKHash' output.
arbitraryWPKHashOutput :: Gen ScriptOutput
arbitraryWPKHashOutput :: Gen ScriptOutput
arbitraryWPKHashOutput = Hash160 -> ScriptOutput
PayWitnessPKHash (Hash160 -> ScriptOutput) -> Gen Hash160 -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Hash160
arbitraryHash160

-- | Arbitrary 'PayWitnessScriptHash' output.
arbitraryWSHOutput :: Gen ScriptOutput
arbitraryWSHOutput :: Gen ScriptOutput
arbitraryWSHOutput = Hash256 -> ScriptOutput
PayWitnessScriptHash (Hash256 -> ScriptOutput) -> Gen Hash256 -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Hash256
arbitraryHash256

arbitraryWitOutput :: Gen ScriptOutput
arbitraryWitOutput :: Gen ScriptOutput
arbitraryWitOutput = do
    Word8
ver <- (Word8, Word8) -> Gen Word8
forall a. Random a => (a, a) -> Gen a
choose (0, 16)
    Int
len <- (Int, Int) -> Gen Int
forall a. Random a => (a, a) -> Gen a
choose (2, 40)
    [Word8]
ws <- Int -> Gen Word8 -> Gen [Word8]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
len Gen Word8
forall a. Arbitrary a => Gen a
arbitrary
    let bs :: ByteString
bs = [Word8] -> ByteString
B.pack [Word8]
ws
    ScriptOutput -> Gen ScriptOutput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOutput -> Gen ScriptOutput)
-> ScriptOutput -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$ Word8 -> ByteString -> ScriptOutput
PayWitness Word8
ver ByteString
bs

-- | Arbitrary 'ScriptOutput' of type 'PayMS'.
arbitraryMSOutput :: Gen ScriptOutput
arbitraryMSOutput :: Gen ScriptOutput
arbitraryMSOutput = do
    (m :: Int
m, n :: Int
n) <- Gen (Int, Int)
arbitraryMSParam
    [PubKeyI]
keys <- ((SecKeyI, PubKeyI) -> PubKeyI)
-> [(SecKeyI, PubKeyI)] -> [PubKeyI]
forall a b. (a -> b) -> [a] -> [b]
map (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ([(SecKeyI, PubKeyI)] -> [PubKeyI])
-> Gen [(SecKeyI, PubKeyI)] -> Gen [PubKeyI]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int -> Gen (SecKeyI, PubKeyI) -> Gen [(SecKeyI, PubKeyI)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
n Gen (SecKeyI, PubKeyI)
arbitraryKeyPair
    ScriptOutput -> Gen ScriptOutput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOutput -> Gen ScriptOutput)
-> ScriptOutput -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$ [PubKeyI] -> Int -> ScriptOutput
PayMulSig [PubKeyI]
keys Int
m

-- | Arbitrary 'ScriptOutput' of type 'PayMS', only using compressed keys.
arbitraryMSOutputC :: Gen ScriptOutput
arbitraryMSOutputC :: Gen ScriptOutput
arbitraryMSOutputC = do
    (m :: Int
m, n :: Int
n) <- Gen (Int, Int)
arbitraryMSParam
    [PubKeyI]
keys <-
        ((SecKeyI, PubKeyI) -> PubKeyI)
-> [(SecKeyI, PubKeyI)] -> [PubKeyI]
forall a b. (a -> b) -> [a] -> [b]
map (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ([(SecKeyI, PubKeyI)] -> [PubKeyI])
-> Gen [(SecKeyI, PubKeyI)] -> Gen [PubKeyI]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$>
        Int -> Gen (SecKeyI, PubKeyI) -> Gen [(SecKeyI, PubKeyI)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
n (Gen (SecKeyI, PubKeyI)
arbitraryKeyPair Gen (SecKeyI, PubKeyI)
-> ((SecKeyI, PubKeyI) -> Bool) -> Gen (SecKeyI, PubKeyI)
forall a. Gen a -> (a -> Bool) -> Gen a
`suchThat` (PubKeyI -> Bool
pubKeyCompressed (PubKeyI -> Bool)
-> ((SecKeyI, PubKeyI) -> PubKeyI) -> (SecKeyI, PubKeyI) -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd))
    ScriptOutput -> Gen ScriptOutput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptOutput -> Gen ScriptOutput)
-> ScriptOutput -> Gen ScriptOutput
forall a b. (a -> b) -> a -> b
$ [PubKeyI] -> Int -> ScriptOutput
PayMulSig [PubKeyI]
keys Int
m

-- | Arbitrary 'ScriptOutput' of type 'PayScriptHash'.
arbitrarySHOutput :: Gen ScriptOutput
arbitrarySHOutput :: Gen ScriptOutput
arbitrarySHOutput = Hash160 -> ScriptOutput
PayScriptHash (Hash160 -> ScriptOutput)
-> (Address -> Hash160) -> Address -> ScriptOutput
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Address -> Hash160
getAddrHash160 (Address -> ScriptOutput) -> Gen Address -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen Address
arbitraryScriptAddress

-- | Arbitrary 'ScriptOutput' of type 'DataCarrier'.
arbitraryDCOutput :: Gen ScriptOutput
arbitraryDCOutput :: Gen ScriptOutput
arbitraryDCOutput = ByteString -> ScriptOutput
DataCarrier (ByteString -> ScriptOutput) -> Gen ByteString -> Gen ScriptOutput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ByteString
arbitraryBS1

-- | Arbitrary 'ScriptInput'.
arbitraryScriptInput :: Network -> Gen ScriptInput
arbitraryScriptInput :: Network -> Gen ScriptInput
arbitraryScriptInput net :: Network
net =
    [Gen ScriptInput] -> Gen ScriptInput
forall a. [Gen a] -> Gen a
oneof
        [ Network -> Gen ScriptInput
arbitraryPKInput Network
net
        , Network -> Gen ScriptInput
arbitraryPKHashInput Network
net
        , Network -> Gen ScriptInput
arbitraryMSInput Network
net
        , Network -> Gen ScriptInput
arbitrarySHInput Network
net
        ]

-- | Arbitrary 'ScriptInput' of type 'SpendPK', 'SpendPKHash' or 'SpendMulSig'
-- (not 'ScriptHashInput')
arbitrarySimpleInput :: Network -> Gen ScriptInput
arbitrarySimpleInput :: Network -> Gen ScriptInput
arbitrarySimpleInput net :: Network
net =
    [Gen ScriptInput] -> Gen ScriptInput
forall a. [Gen a] -> Gen a
oneof
        [ Network -> Gen ScriptInput
arbitraryPKInput Network
net
        , Network -> Gen ScriptInput
arbitraryPKHashInput Network
net
        , Network -> Gen ScriptInput
arbitraryMSInput Network
net
        ]

-- | Arbitrary 'ScriptInput' of type 'SpendPK'.
arbitraryPKInput :: Network -> Gen ScriptInput
arbitraryPKInput :: Network -> Gen ScriptInput
arbitraryPKInput net :: Network
net = SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput)
-> (TxSignature -> SimpleInput) -> TxSignature -> ScriptInput
forall b c a. (b -> c) -> (a -> b) -> a -> c
. TxSignature -> SimpleInput
SpendPK (TxSignature -> ScriptInput) -> Gen TxSignature -> Gen ScriptInput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net

-- | Arbitrary 'ScriptInput' of type 'SpendPK'.
arbitraryPKHashInput :: Network -> Gen ScriptInput
arbitraryPKHashInput :: Network -> Gen ScriptInput
arbitraryPKHashInput net :: Network
net = do
    TxSignature
sig <- Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net
    PubKeyI
key <- (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ((SecKeyI, PubKeyI) -> PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (SecKeyI, PubKeyI)
arbitraryKeyPair
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ TxSignature -> PubKeyI -> SimpleInput
SpendPKHash TxSignature
sig PubKeyI
key

-- | Like 'arbitraryPKHashInput' without empty signatures.
arbitraryPKHashInputFull :: Network -> Gen ScriptInput
arbitraryPKHashInputFull :: Network -> Gen ScriptInput
arbitraryPKHashInputFull net :: Network
net = do
    TxSignature
sig <- (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ((TxHash, SecKey, TxSignature) -> TxSignature)
-> Gen (TxHash, SecKey, TxSignature) -> Gen TxSignature
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net
    PubKeyI
key <- (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd ((SecKeyI, PubKeyI) -> PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (SecKeyI, PubKeyI)
arbitraryKeyPair
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ TxSignature -> PubKeyI -> SimpleInput
SpendPKHash TxSignature
sig PubKeyI
key

-- | Like above but only compressed.
arbitraryPKHashInputFullC :: Network -> Gen ScriptInput
arbitraryPKHashInputFullC :: Network -> Gen ScriptInput
arbitraryPKHashInputFullC net :: Network
net = do
    TxSignature
sig <- (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ((TxHash, SecKey, TxSignature) -> TxSignature)
-> Gen (TxHash, SecKey, TxSignature) -> Gen TxSignature
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net
    PubKeyI
key <- ((SecKeyI, PubKeyI) -> PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd (Gen (SecKeyI, PubKeyI) -> Gen PubKeyI)
-> Gen (SecKeyI, PubKeyI) -> Gen PubKeyI
forall a b. (a -> b) -> a -> b
$ Gen (SecKeyI, PubKeyI)
arbitraryKeyPair Gen (SecKeyI, PubKeyI)
-> ((SecKeyI, PubKeyI) -> Bool) -> Gen (SecKeyI, PubKeyI)
forall a. Gen a -> (a -> Bool) -> Gen a
`suchThat` (PubKeyI -> Bool
pubKeyCompressed (PubKeyI -> Bool)
-> ((SecKeyI, PubKeyI) -> PubKeyI) -> (SecKeyI, PubKeyI) -> Bool
forall b c a. (b -> c) -> (a -> b) -> a -> c
. (SecKeyI, PubKeyI) -> PubKeyI
forall a b. (a, b) -> b
snd)
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ TxSignature -> PubKeyI -> SimpleInput
SpendPKHash TxSignature
sig PubKeyI
key

-- | Arbitrary 'ScriptInput' of type 'SpendMulSig'.
arbitraryMSInput :: Network -> Gen ScriptInput
arbitraryMSInput :: Network -> Gen ScriptInput
arbitraryMSInput net :: Network
net = do
    Int
m    <- (Int, Int) -> Int
forall a b. (a, b) -> a
fst ((Int, Int) -> Int) -> Gen (Int, Int) -> Gen Int
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen (Int, Int)
arbitraryMSParam
    [TxSignature]
sigs <- Int -> Gen TxSignature -> Gen [TxSignature]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net)
    ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptInput
RegularInput (SimpleInput -> ScriptInput) -> SimpleInput -> ScriptInput
forall a b. (a -> b) -> a -> b
$ [TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs

-- | Arbitrary 'ScriptInput' of type 'ScriptHashInput'.
arbitrarySHInput :: Network -> Gen ScriptInput
arbitrarySHInput :: Network -> Gen ScriptInput
arbitrarySHInput net :: Network
net = do
    ScriptInput
i <- Network -> Gen ScriptInput
arbitrarySimpleInput Network
net
    SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput (ScriptInput -> SimpleInput
getRegularInput ScriptInput
i) (ScriptOutput -> ScriptInput)
-> Gen ScriptOutput -> Gen ScriptInput
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Gen ScriptOutput
arbitrarySimpleOutput

-- | Arbitrary 'ScriptInput' of type 'ScriptHashInput' containing a
-- 'RedeemScript' of type 'PayMulSig' and an input of type 'SpendMulSig'.
arbitraryMulSigSHInput :: Network -> Gen ScriptInput
arbitraryMulSigSHInput :: Network -> Gen ScriptInput
arbitraryMulSigSHInput net :: Network
net =
    Gen ScriptOutput
arbitraryMSOutput Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig _ m :: Int
m) -> do
            [TxSignature]
sigs <- Int -> Gen TxSignature -> Gen [TxSignature]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net)
            ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        _ -> Gen ScriptInput
forall a. HasCallStack => a
undefined

-- | Arbitrary 'ScriptInput' of type 'ScriptHashInput' containing a
-- 'RedeemScript' of type 'PayMulSig' and an input of type 'SpendMulSig'.
arbitraryMulSigSHInputC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputC net :: Network
net =
    Gen ScriptOutput
arbitraryMSOutputC Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig _ m :: Int
m) -> do
          [TxSignature]
sigs <- Int -> Gen TxSignature -> Gen [TxSignature]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen TxSignature
arbitraryTxSignatureEmpty Network
net)
          ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        _ -> Gen ScriptInput
forall a. HasCallStack => a
undefined

-- | Like 'arbitraryMulSigSHCInput' with no empty signatures.
arbitraryMulSigSHInputFull :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFull :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFull net :: Network
net =
    Gen ScriptOutput
arbitraryMSOutput Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig _ m :: Int
m) -> do
            [TxSignature]
sigs <- ((TxHash, SecKey, TxSignature) -> TxSignature)
-> [(TxHash, SecKey, TxSignature)] -> [TxSignature]
forall a b. (a -> b) -> [a] -> [b]
map (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ([(TxHash, SecKey, TxSignature)] -> [TxSignature])
-> Gen [(TxHash, SecKey, TxSignature)] -> Gen [TxSignature]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int
-> Gen (TxHash, SecKey, TxSignature)
-> Gen [(TxHash, SecKey, TxSignature)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net)
            ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        _ -> Gen ScriptInput
forall a. HasCallStack => a
undefined

-- | Like 'arbitraryMulSigSHCInput' with no empty signatures.
arbitraryMulSigSHInputFullC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFullC :: Network -> Gen ScriptInput
arbitraryMulSigSHInputFullC net :: Network
net =
    Gen ScriptOutput
arbitraryMSOutputC Gen ScriptOutput
-> (ScriptOutput -> Gen ScriptInput) -> Gen ScriptInput
forall (m :: * -> *) a b. Monad m => m a -> (a -> m b) -> m b
>>= \case
        rdm :: ScriptOutput
rdm@(PayMulSig _ m :: Int
m) -> do
            [TxSignature]
sigs <- ((TxHash, SecKey, TxSignature) -> TxSignature)
-> [(TxHash, SecKey, TxSignature)] -> [TxSignature]
forall a b. (a -> b) -> [a] -> [b]
map (TxHash, SecKey, TxSignature) -> TxSignature
forall a b c. (a, b, c) -> c
lst3 ([(TxHash, SecKey, TxSignature)] -> [TxSignature])
-> Gen [(TxHash, SecKey, TxSignature)] -> Gen [TxSignature]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> Int
-> Gen (TxHash, SecKey, TxSignature)
-> Gen [(TxHash, SecKey, TxSignature)]
forall a. Int -> Gen a -> Gen [a]
vectorOf Int
m (Network -> Gen (TxHash, SecKey, TxSignature)
arbitraryTxSignature Network
net)
            ScriptInput -> Gen ScriptInput
forall (m :: * -> *) a. Monad m => a -> m a
return (ScriptInput -> Gen ScriptInput) -> ScriptInput -> Gen ScriptInput
forall a b. (a -> b) -> a -> b
$ SimpleInput -> ScriptOutput -> ScriptInput
ScriptHashInput ([TxSignature] -> SimpleInput
SpendMulSig [TxSignature]
sigs) ScriptOutput
rdm
        _ -> Gen ScriptInput
forall a. HasCallStack => a
undefined