123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419 |
- // Copyright (c) 2013-2017 The btcsuite developers
- // Use of this source code is governed by an ISC
- // license that can be found in the LICENSE file.
- package txscript
- import (
- "testing"
- "github.com/pkt-cash/pktd/chaincfg/chainhash"
- "github.com/pkt-cash/pktd/txscript/opcode"
- "github.com/pkt-cash/pktd/txscript/txscripterr"
- "github.com/pkt-cash/pktd/wire"
- )
- // TestBadPC sets the pc to a deliberately bad result then confirms that Step()
- // and Disasm fail correctly.
- func TestBadPC(t *testing.T) {
- tests := []struct {
- script, off int
- }{
- {script: 2, off: 0},
- {script: 0, off: 2},
- }
- // tx with almost empty scripts.
- tx := &wire.MsgTx{
- Version: 1,
- TxIn: []*wire.TxIn{
- {
- PreviousOutPoint: wire.OutPoint{
- Hash: chainhash.Hash([32]byte{
- 0xc9, 0x97, 0xa5, 0xe5,
- 0x6e, 0x10, 0x41, 0x02,
- 0xfa, 0x20, 0x9c, 0x6a,
- 0x85, 0x2d, 0xd9, 0x06,
- 0x60, 0xa2, 0x0b, 0x2d,
- 0x9c, 0x35, 0x24, 0x23,
- 0xed, 0xce, 0x25, 0x85,
- 0x7f, 0xcd, 0x37, 0x04,
- }),
- Index: 0,
- },
- SignatureScript: mustParseShortForm("NOP"),
- Sequence: 4294967295,
- },
- },
- TxOut: []*wire.TxOut{{
- Value: 1000000000,
- PkScript: nil,
- }},
- LockTime: 0,
- }
- pkScript := mustParseShortForm("NOP")
- for _, test := range tests {
- vm, err := NewEngine(pkScript, tx, 0, 0, nil, nil, -1)
- if err != nil {
- t.Errorf("Failed to create script: %v", err)
- }
- // set to after all scripts
- vm.scriptIdx = test.script
- vm.scriptOff = test.off
- _, err = vm.Step()
- if err == nil {
- t.Errorf("Step with invalid pc (%v) succeeds!", test)
- continue
- }
- _, err = vm.DisasmPC()
- if err == nil {
- t.Errorf("DisasmPC with invalid pc (%v) succeeds!",
- test)
- }
- }
- }
- // TestCheckErrorCondition tests the execute early test in CheckErrorCondition()
- // since most code paths are tested elsewhere.
- func TestCheckErrorCondition(t *testing.T) {
- // tx with almost empty scripts.
- tx := &wire.MsgTx{
- Version: 1,
- TxIn: []*wire.TxIn{{
- PreviousOutPoint: wire.OutPoint{
- Hash: chainhash.Hash([32]byte{
- 0xc9, 0x97, 0xa5, 0xe5,
- 0x6e, 0x10, 0x41, 0x02,
- 0xfa, 0x20, 0x9c, 0x6a,
- 0x85, 0x2d, 0xd9, 0x06,
- 0x60, 0xa2, 0x0b, 0x2d,
- 0x9c, 0x35, 0x24, 0x23,
- 0xed, 0xce, 0x25, 0x85,
- 0x7f, 0xcd, 0x37, 0x04,
- }),
- Index: 0,
- },
- SignatureScript: nil,
- Sequence: 4294967295,
- }},
- TxOut: []*wire.TxOut{{
- Value: 1000000000,
- PkScript: nil,
- }},
- LockTime: 0,
- }
- pkScript := mustParseShortForm("NOP NOP NOP NOP NOP NOP NOP NOP NOP" +
- " NOP TRUE")
- vm, err := NewEngine(pkScript, tx, 0, 0, nil, nil, 0)
- if err != nil {
- t.Errorf("failed to create script: %v", err)
- }
- for i := 0; i < len(pkScript)-1; i++ {
- done, err := vm.Step()
- if err != nil {
- t.Fatalf("failed to step %dth time: %v", i, err)
- }
- if done {
- t.Fatalf("finshed early on %dth time", i)
- }
- err = vm.CheckErrorCondition(false)
- if !txscripterr.ErrScriptUnfinished.Is(err) {
- t.Fatalf("got unexepected error %v on %dth iteration",
- err, i)
- }
- }
- done, err := vm.Step()
- if err != nil {
- t.Fatalf("final step failed %v", err)
- }
- if !done {
- t.Fatalf("final step isn't done!")
- }
- err = vm.CheckErrorCondition(false)
- if err != nil {
- t.Errorf("unexpected error %v on final check", err)
- }
- }
- // TestInvalidFlagCombinations ensures the script engine returns the expected
- // error when disallowed flag combinations are specified.
- func TestInvalidFlagCombinations(t *testing.T) {
- tests := []ScriptFlags{
- ScriptVerifyCleanStack,
- }
- // tx with almost empty scripts.
- tx := &wire.MsgTx{
- Version: 1,
- TxIn: []*wire.TxIn{
- {
- PreviousOutPoint: wire.OutPoint{
- Hash: chainhash.Hash([32]byte{
- 0xc9, 0x97, 0xa5, 0xe5,
- 0x6e, 0x10, 0x41, 0x02,
- 0xfa, 0x20, 0x9c, 0x6a,
- 0x85, 0x2d, 0xd9, 0x06,
- 0x60, 0xa2, 0x0b, 0x2d,
- 0x9c, 0x35, 0x24, 0x23,
- 0xed, 0xce, 0x25, 0x85,
- 0x7f, 0xcd, 0x37, 0x04,
- }),
- Index: 0,
- },
- SignatureScript: []uint8{opcode.OP_NOP},
- Sequence: 4294967295,
- },
- },
- TxOut: []*wire.TxOut{
- {
- Value: 1000000000,
- PkScript: nil,
- },
- },
- LockTime: 0,
- }
- pkScript := []byte{opcode.OP_NOP}
- for i, test := range tests {
- _, err := NewEngine(pkScript, tx, 0, test, nil, nil, -1)
- if !txscripterr.ErrInvalidFlags.Is(err) {
- t.Fatalf("TestInvalidFlagCombinations #%d unexpected "+
- "error: %v", i, err)
- }
- }
- }
- // TestCheckPubKeyEncoding ensures the internal checkPubKeyEncoding function
- // works as expected.
- func TestCheckPubKeyEncoding(t *testing.T) {
- tests := []struct {
- name string
- key []byte
- isValid bool
- }{
- {
- name: "uncompressed ok",
- key: hexToBytes("0411db93e1dcdb8a016b49840f8c53bc1eb68" +
- "a382e97b1482ecad7b148a6909a5cb2e0eaddfb84ccf" +
- "9744464f82e160bfa9b8b64f9d4c03f999b8643f656b" +
- "412a3"),
- isValid: true,
- },
- {
- name: "compressed ok",
- key: hexToBytes("02ce0b14fb842b1ba549fdd675c98075f12e9" +
- "c510f8ef52bd021a9a1f4809d3b4d"),
- isValid: true,
- },
- {
- name: "compressed ok",
- key: hexToBytes("032689c7c2dab13309fb143e0e8fe39634252" +
- "1887e976690b6b47f5b2a4b7d448e"),
- isValid: true,
- },
- {
- name: "hybrid",
- key: hexToBytes("0679be667ef9dcbbac55a06295ce870b07029" +
- "bfcdb2dce28d959f2815b16f81798483ada7726a3c46" +
- "55da4fbfc0e1108a8fd17b448a68554199c47d08ffb1" +
- "0d4b8"),
- isValid: false,
- },
- {
- name: "empty",
- key: nil,
- isValid: false,
- },
- }
- vm := Engine{flags: ScriptVerifyStrictEncoding}
- for _, test := range tests {
- err := vm.checkPubKeyEncoding(test.key)
- if err != nil && test.isValid {
- t.Errorf("checkSignatureEncoding test '%s' failed "+
- "when it should have succeeded: %v", test.name,
- err)
- } else if err == nil && !test.isValid {
- t.Errorf("checkSignatureEncooding test '%s' succeeded "+
- "when it should have failed", test.name)
- }
- }
- }
- // TestCheckSignatureEncoding ensures the internal checkSignatureEncoding
- // function works as expected.
- func TestCheckSignatureEncoding(t *testing.T) {
- tests := []struct {
- name string
- sig []byte
- isValid bool
- }{
- {
- name: "valid signature",
- sig: hexToBytes("304402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: true,
- },
- {
- name: "empty.",
- sig: nil,
- isValid: false,
- },
- {
- name: "bad magic",
- sig: hexToBytes("314402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "bad 1st int marker magic",
- sig: hexToBytes("304403204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "bad 2nd int marker",
- sig: hexToBytes("304402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41032018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "short len",
- sig: hexToBytes("304302204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "long len",
- sig: hexToBytes("304502204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "long X",
- sig: hexToBytes("304402424e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "long Y",
- sig: hexToBytes("304402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022118152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "short Y",
- sig: hexToBytes("304402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41021918152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "trailing crap",
- sig: hexToBytes("304402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d0901"),
- isValid: false,
- },
- {
- name: "X == N ",
- sig: hexToBytes("30440220fffffffffffffffffffffffffffff" +
- "ffebaaedce6af48a03bbfd25e8cd0364141022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "X == N ",
- sig: hexToBytes("30440220fffffffffffffffffffffffffffff" +
- "ffebaaedce6af48a03bbfd25e8cd0364142022018152" +
- "2ec8eca07de4860a4acdd12909d831cc56cbbac46220" +
- "82221a8768d1d09"),
- isValid: false,
- },
- {
- name: "Y == N",
- sig: hexToBytes("304402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd410220fffff" +
- "ffffffffffffffffffffffffffebaaedce6af48a03bb" +
- "fd25e8cd0364141"),
- isValid: false,
- },
- {
- name: "Y > N",
- sig: hexToBytes("304402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd410220fffff" +
- "ffffffffffffffffffffffffffebaaedce6af48a03bb" +
- "fd25e8cd0364142"),
- isValid: false,
- },
- {
- name: "0 len X",
- sig: hexToBytes("302402000220181522ec8eca07de4860a4acd" +
- "d12909d831cc56cbbac4622082221a8768d1d09"),
- isValid: false,
- },
- {
- name: "0 len Y",
- sig: hexToBytes("302402204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd410200"),
- isValid: false,
- },
- {
- name: "extra R padding",
- sig: hexToBytes("30450221004e45e16932b8af514961a1d3a1a" +
- "25fdf3f4f7732e9d624c6c61548ab5fb8cd410220181" +
- "522ec8eca07de4860a4acdd12909d831cc56cbbac462" +
- "2082221a8768d1d09"),
- isValid: false,
- },
- {
- name: "extra S padding",
- sig: hexToBytes("304502204e45e16932b8af514961a1d3a1a25" +
- "fdf3f4f7732e9d624c6c61548ab5fb8cd41022100181" +
- "522ec8eca07de4860a4acdd12909d831cc56cbbac462" +
- "2082221a8768d1d09"),
- isValid: false,
- },
- }
- vm := Engine{flags: ScriptVerifyStrictEncoding}
- for _, test := range tests {
- err := vm.checkSignatureEncoding(test.sig)
- if err != nil && test.isValid {
- t.Errorf("checkSignatureEncoding test '%s' failed "+
- "when it should have succeeded: %v", test.name,
- err)
- } else if err == nil && !test.isValid {
- t.Errorf("checkSignatureEncooding test '%s' succeeded "+
- "when it should have failed", test.name)
- }
- }
- }
|