phantom/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-01-27 03:33:27 -03:00
Code Issues Projects Releases Packages Wiki Activity Actions

[tests] fix flake8 warnings in script.py and bignum.py

Browse source
This commit is contained in:
John Newbery 2019-10-30 10:13:17 -04:00
parent 6a97e8a060
commit f31fc0e92e
2 changed files with 144 additions and 141 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
test/functional/test_framework/bignum.py
View file

@ -4,25 +4,24 @@
# file COPYING or http://www.opensource.org/licenses/mit-license.php. # file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""Big number routines. """Big number routines.
Functions for converting numbers to MPI format and Bitcoin-specific little
endian format.
This file is copied from python-bitcoinlib. This file is copied from python-bitcoinlib.
""" """
import struct import struct
# generic big endian MPI format
def bn_bytes(v, have_ext=False): def bn_bytes(v, have_ext=False):
ext = 0 ext = 0
if have_ext: if have_ext:
ext = 1 ext = 1
return ((v.bit_length()+7)//8) + ext return (v.bit_length() + 7) // 8 + ext
def bn2bin(v): def bn2bin(v):
s = bytearray() s = bytearray()
i = bn_bytes(v) i = bn_bytes(v)
while i > 0: while i > 0:
s.append((v >> ((i-1) * 8)) & 0xff) s.append((v >> ((i - 1) * 8)) & 0xff)
i -= 1 i -= 1
return s return s

264
test/functional/test_framework/script.py
View file

@ -6,16 +6,21 @@
This file is modified from python-bitcoinlib. This file is modified from python-bitcoinlib.
""" """
from .messages import CTransaction, CTxOut, sha256, hash256, uint256_from_str, ser_uint256, ser_string
import hashlib import hashlib
import struct import struct
from .bignum import bn2vch from .bignum import bn2vch
from .messages import (
CTransaction,
CTxOut,
hash256,
ser_string,
ser_uint256,
sha256,
uint256_from_str,
)
MAX_SCRIPT_ELEMENT_SIZE = 520 MAX_SCRIPT_ELEMENT_SIZE = 520
OPCODE_NAMES = {} OPCODE_NAMES = {}
def hash160(s): def hash160(s):
@ -50,7 +55,7 @@ class CScriptOp(int):
if n == 0: if n == 0:
return OP_0 return OP_0
else: else:
return CScriptOp(OP_1 + n-1) return CScriptOp(OP_1 + n - 1)
def decode_op_n(self): def decode_op_n(self):
"""Decode a small integer opcode, returning an integer""" """Decode a small integer opcode, returning an integer"""
@ -60,7 +65,7 @@ class CScriptOp(int):
if not (self == OP_0 or OP_1 <= self <= OP_16): if not (self == OP_0 or OP_1 <= self <= OP_16):
raise ValueError('op %r is not an OP_N' % self) raise ValueError('op %r is not an OP_N' % self)
return int(self - OP_1+1) return int(self - OP_1 + 1)
def is_small_int(self): def is_small_int(self):
"""Return true if the op pushes a small integer to the stack""" """Return true if the op pushes a small integer to the stack"""
@ -87,7 +92,7 @@ class CScriptOp(int):
return _opcode_instances[n] return _opcode_instances[n]
# Populate opcode instance table # Populate opcode instance table
for n in range(0xff+1): for n in range(0xff + 1):
CScriptOp(n) CScriptOp(n)
@ -100,7 +105,7 @@ OP_PUSHDATA4 = CScriptOp(0x4e)
OP_1NEGATE = CScriptOp(0x4f) OP_1NEGATE = CScriptOp(0x4f)
OP_RESERVED = CScriptOp(0x50) OP_RESERVED = CScriptOp(0x50)
OP_1 = CScriptOp(0x51) OP_1 = CScriptOp(0x51)
OP_TRUE=OP_1 OP_TRUE = OP_1
OP_2 = CScriptOp(0x52) OP_2 = CScriptOp(0x52)
OP_3 = CScriptOp(0x53) OP_3 = CScriptOp(0x53)
OP_4 = CScriptOp(0x54) OP_4 = CScriptOp(0x54)
@ -232,122 +237,122 @@ OP_PUBKEY = CScriptOp(0xfe)
OP_INVALIDOPCODE = CScriptOp(0xff) OP_INVALIDOPCODE = CScriptOp(0xff)
OPCODE_NAMES.update({ OPCODE_NAMES.update({
OP_0 : 'OP_0', OP_0: 'OP_0',
OP_PUSHDATA1 : 'OP_PUSHDATA1', OP_PUSHDATA1: 'OP_PUSHDATA1',
OP_PUSHDATA2 : 'OP_PUSHDATA2', OP_PUSHDATA2: 'OP_PUSHDATA2',
OP_PUSHDATA4 : 'OP_PUSHDATA4', OP_PUSHDATA4: 'OP_PUSHDATA4',
OP_1NEGATE : 'OP_1NEGATE', OP_1NEGATE: 'OP_1NEGATE',
OP_RESERVED : 'OP_RESERVED', OP_RESERVED: 'OP_RESERVED',
OP_1 : 'OP_1', OP_1: 'OP_1',
OP_2 : 'OP_2', OP_2: 'OP_2',
OP_3 : 'OP_3', OP_3: 'OP_3',
OP_4 : 'OP_4', OP_4: 'OP_4',
OP_5 : 'OP_5', OP_5: 'OP_5',
OP_6 : 'OP_6', OP_6: 'OP_6',
OP_7 : 'OP_7', OP_7: 'OP_7',
OP_8 : 'OP_8', OP_8: 'OP_8',
OP_9 : 'OP_9', OP_9: 'OP_9',
OP_10 : 'OP_10', OP_10: 'OP_10',
OP_11 : 'OP_11', OP_11: 'OP_11',
OP_12 : 'OP_12', OP_12: 'OP_12',
OP_13 : 'OP_13', OP_13: 'OP_13',
OP_14 : 'OP_14', OP_14: 'OP_14',
OP_15 : 'OP_15', OP_15: 'OP_15',
OP_16 : 'OP_16', OP_16: 'OP_16',
OP_NOP : 'OP_NOP', OP_NOP: 'OP_NOP',
OP_VER : 'OP_VER', OP_VER: 'OP_VER',
OP_IF : 'OP_IF', OP_IF: 'OP_IF',
OP_NOTIF : 'OP_NOTIF', OP_NOTIF: 'OP_NOTIF',
OP_VERIF : 'OP_VERIF', OP_VERIF: 'OP_VERIF',
OP_VERNOTIF : 'OP_VERNOTIF', OP_VERNOTIF: 'OP_VERNOTIF',
OP_ELSE : 'OP_ELSE', OP_ELSE: 'OP_ELSE',
OP_ENDIF : 'OP_ENDIF', OP_ENDIF: 'OP_ENDIF',
OP_VERIFY : 'OP_VERIFY', OP_VERIFY: 'OP_VERIFY',
OP_RETURN : 'OP_RETURN', OP_RETURN: 'OP_RETURN',
OP_TOALTSTACK : 'OP_TOALTSTACK', OP_TOALTSTACK: 'OP_TOALTSTACK',
OP_FROMALTSTACK : 'OP_FROMALTSTACK', OP_FROMALTSTACK: 'OP_FROMALTSTACK',
OP_2DROP : 'OP_2DROP', OP_2DROP: 'OP_2DROP',
OP_2DUP : 'OP_2DUP', OP_2DUP: 'OP_2DUP',
OP_3DUP : 'OP_3DUP', OP_3DUP: 'OP_3DUP',
OP_2OVER : 'OP_2OVER', OP_2OVER: 'OP_2OVER',
OP_2ROT : 'OP_2ROT', OP_2ROT: 'OP_2ROT',
OP_2SWAP : 'OP_2SWAP', OP_2SWAP: 'OP_2SWAP',
OP_IFDUP : 'OP_IFDUP', OP_IFDUP: 'OP_IFDUP',
OP_DEPTH : 'OP_DEPTH', OP_DEPTH: 'OP_DEPTH',
OP_DROP : 'OP_DROP', OP_DROP: 'OP_DROP',
OP_DUP : 'OP_DUP', OP_DUP: 'OP_DUP',
OP_NIP : 'OP_NIP', OP_NIP: 'OP_NIP',
OP_OVER : 'OP_OVER', OP_OVER: 'OP_OVER',
OP_PICK : 'OP_PICK', OP_PICK: 'OP_PICK',
OP_ROLL : 'OP_ROLL', OP_ROLL: 'OP_ROLL',
OP_ROT : 'OP_ROT', OP_ROT: 'OP_ROT',
OP_SWAP : 'OP_SWAP', OP_SWAP: 'OP_SWAP',
OP_TUCK : 'OP_TUCK', OP_TUCK: 'OP_TUCK',
OP_CAT : 'OP_CAT', OP_CAT: 'OP_CAT',
OP_SUBSTR : 'OP_SUBSTR', OP_SUBSTR: 'OP_SUBSTR',
OP_LEFT : 'OP_LEFT', OP_LEFT: 'OP_LEFT',
OP_RIGHT : 'OP_RIGHT', OP_RIGHT: 'OP_RIGHT',
OP_SIZE : 'OP_SIZE', OP_SIZE: 'OP_SIZE',
OP_INVERT : 'OP_INVERT', OP_INVERT: 'OP_INVERT',
OP_AND : 'OP_AND', OP_AND: 'OP_AND',
OP_OR : 'OP_OR', OP_OR: 'OP_OR',
OP_XOR : 'OP_XOR', OP_XOR: 'OP_XOR',
OP_EQUAL : 'OP_EQUAL', OP_EQUAL: 'OP_EQUAL',
OP_EQUALVERIFY : 'OP_EQUALVERIFY', OP_EQUALVERIFY: 'OP_EQUALVERIFY',
OP_RESERVED1 : 'OP_RESERVED1', OP_RESERVED1: 'OP_RESERVED1',
OP_RESERVED2 : 'OP_RESERVED2', OP_RESERVED2: 'OP_RESERVED2',
OP_1ADD : 'OP_1ADD', OP_1ADD: 'OP_1ADD',
OP_1SUB : 'OP_1SUB', OP_1SUB: 'OP_1SUB',
OP_2MUL : 'OP_2MUL', OP_2MUL: 'OP_2MUL',
OP_2DIV : 'OP_2DIV', OP_2DIV: 'OP_2DIV',
OP_NEGATE : 'OP_NEGATE', OP_NEGATE: 'OP_NEGATE',
OP_ABS : 'OP_ABS', OP_ABS: 'OP_ABS',
OP_NOT : 'OP_NOT', OP_NOT: 'OP_NOT',
OP_0NOTEQUAL : 'OP_0NOTEQUAL', OP_0NOTEQUAL: 'OP_0NOTEQUAL',
OP_ADD : 'OP_ADD', OP_ADD: 'OP_ADD',
OP_SUB : 'OP_SUB', OP_SUB: 'OP_SUB',
OP_MUL : 'OP_MUL', OP_MUL: 'OP_MUL',
OP_DIV : 'OP_DIV', OP_DIV: 'OP_DIV',
OP_MOD : 'OP_MOD', OP_MOD: 'OP_MOD',
OP_LSHIFT : 'OP_LSHIFT', OP_LSHIFT: 'OP_LSHIFT',
OP_RSHIFT : 'OP_RSHIFT', OP_RSHIFT: 'OP_RSHIFT',
OP_BOOLAND : 'OP_BOOLAND', OP_BOOLAND: 'OP_BOOLAND',
OP_BOOLOR : 'OP_BOOLOR', OP_BOOLOR: 'OP_BOOLOR',
OP_NUMEQUAL : 'OP_NUMEQUAL', OP_NUMEQUAL: 'OP_NUMEQUAL',
OP_NUMEQUALVERIFY : 'OP_NUMEQUALVERIFY', OP_NUMEQUALVERIFY: 'OP_NUMEQUALVERIFY',
OP_NUMNOTEQUAL : 'OP_NUMNOTEQUAL', OP_NUMNOTEQUAL: 'OP_NUMNOTEQUAL',
OP_LESSTHAN : 'OP_LESSTHAN', OP_LESSTHAN: 'OP_LESSTHAN',
OP_GREATERTHAN : 'OP_GREATERTHAN', OP_GREATERTHAN: 'OP_GREATERTHAN',
OP_LESSTHANOREQUAL : 'OP_LESSTHANOREQUAL', OP_LESSTHANOREQUAL: 'OP_LESSTHANOREQUAL',
OP_GREATERTHANOREQUAL : 'OP_GREATERTHANOREQUAL', OP_GREATERTHANOREQUAL: 'OP_GREATERTHANOREQUAL',
OP_MIN : 'OP_MIN', OP_MIN: 'OP_MIN',
OP_MAX : 'OP_MAX', OP_MAX: 'OP_MAX',
OP_WITHIN : 'OP_WITHIN', OP_WITHIN: 'OP_WITHIN',
OP_RIPEMD160 : 'OP_RIPEMD160', OP_RIPEMD160: 'OP_RIPEMD160',
OP_SHA1 : 'OP_SHA1', OP_SHA1: 'OP_SHA1',
OP_SHA256 : 'OP_SHA256', OP_SHA256: 'OP_SHA256',
OP_HASH160 : 'OP_HASH160', OP_HASH160: 'OP_HASH160',
OP_HASH256 : 'OP_HASH256', OP_HASH256: 'OP_HASH256',
OP_CODESEPARATOR : 'OP_CODESEPARATOR', OP_CODESEPARATOR: 'OP_CODESEPARATOR',
OP_CHECKSIG : 'OP_CHECKSIG', OP_CHECKSIG: 'OP_CHECKSIG',
OP_CHECKSIGVERIFY : 'OP_CHECKSIGVERIFY', OP_CHECKSIGVERIFY: 'OP_CHECKSIGVERIFY',
OP_CHECKMULTISIG : 'OP_CHECKMULTISIG', OP_CHECKMULTISIG: 'OP_CHECKMULTISIG',
OP_CHECKMULTISIGVERIFY : 'OP_CHECKMULTISIGVERIFY', OP_CHECKMULTISIGVERIFY: 'OP_CHECKMULTISIGVERIFY',
OP_NOP1 : 'OP_NOP1', OP_NOP1: 'OP_NOP1',
OP_CHECKLOCKTIMEVERIFY : 'OP_CHECKLOCKTIMEVERIFY', OP_CHECKLOCKTIMEVERIFY: 'OP_CHECKLOCKTIMEVERIFY',
OP_CHECKSEQUENCEVERIFY : 'OP_CHECKSEQUENCEVERIFY', OP_CHECKSEQUENCEVERIFY: 'OP_CHECKSEQUENCEVERIFY',
OP_NOP4 : 'OP_NOP4', OP_NOP4: 'OP_NOP4',
OP_NOP5 : 'OP_NOP5', OP_NOP5: 'OP_NOP5',
OP_NOP6 : 'OP_NOP6', OP_NOP6: 'OP_NOP6',
OP_NOP7 : 'OP_NOP7', OP_NOP7: 'OP_NOP7',
OP_NOP8 : 'OP_NOP8', OP_NOP8: 'OP_NOP8',
OP_NOP9 : 'OP_NOP9', OP_NOP9: 'OP_NOP9',
OP_NOP10 : 'OP_NOP10', OP_NOP10: 'OP_NOP10',
OP_SMALLINTEGER : 'OP_SMALLINTEGER', OP_SMALLINTEGER: 'OP_SMALLINTEGER',
OP_PUBKEYS : 'OP_PUBKEYS', OP_PUBKEYS: 'OP_PUBKEYS',
OP_PUBKEYHASH : 'OP_PUBKEYHASH', OP_PUBKEYHASH: 'OP_PUBKEYHASH',
OP_PUBKEY : 'OP_PUBKEY', OP_PUBKEY: 'OP_PUBKEY',
OP_INVALIDOPCODE : 'OP_INVALIDOPCODE', OP_INVALIDOPCODE: 'OP_INVALIDOPCODE',
}) })
class CScriptInvalidError(Exception): class CScriptInvalidError(Exception):
@ -392,10 +397,10 @@ class CScriptNum:
if len(value) == 0: if len(value) == 0:
return result return result
for i, byte in enumerate(value): for i, byte in enumerate(value):
result |= int(byte) << 8*i result |= int(byte) << 8 * i
if value[-1] >= 0x80: if value[-1] >= 0x80:
# Mask for all but the highest result bit # Mask for all but the highest result bit
num_mask = (2**(len(value)*8) - 1) >> 1 num_mask = (2**(len(value) * 8) - 1) >> 1
result &= num_mask result &= num_mask
result *= -1 result *= -1
return result return result
@ -493,21 +498,20 @@ class CScript(bytes):
pushdata_type = 'PUSHDATA2' pushdata_type = 'PUSHDATA2'
if i + 1 >= len(self): if i + 1 >= len(self):
raise CScriptInvalidError('PUSHDATA2: missing data length') raise CScriptInvalidError('PUSHDATA2: missing data length')
datasize = self[i] + (self[i+1] << 8) datasize = self[i] + (self[i + 1] << 8)
i += 2 i += 2
elif opcode == OP_PUSHDATA4: elif opcode == OP_PUSHDATA4:
pushdata_type = 'PUSHDATA4' pushdata_type = 'PUSHDATA4'
if i + 3 >= len(self): if i + 3 >= len(self):
raise CScriptInvalidError('PUSHDATA4: missing data length') raise CScriptInvalidError('PUSHDATA4: missing data length')
datasize = self[i] + (self[i+1] << 8) + (self[i+2] << 16) + (self[i+3] << 24) datasize = self[i] + (self[i + 1] << 8) + (self[i + 2] << 16) + (self[i + 3] << 24)
i += 4 i += 4
else: else:
assert False # shouldn't happen assert False # shouldn't happen
data = bytes(self[i:i + datasize])
data = bytes(self[i:i+datasize])
# Check for truncation # Check for truncation
if len(data) < datasize: if len(data) < datasize: