스크립트

#P2PKH 스크립트 정의 def p2pkh_script(h160): return Script([0x76, 0xa9, h160, 0x88, 0xac] class Script: def __init__(self, cmds=None): if cmds is None: self.cmds = [] else: self.cmds = cmds def __repr__(self): result = [] for cmd in self.cmds: if type(cmd) == int: if OP_CODE_NAMES.get(cmd): name = OP_CODE_NAMES.get(cmd) else: name = 'OP_[{}]'.format(cmd) result.append(name) else: result.append(cmd.hex()) return ' '.join(result) def __add__(self, other): return Script(self.cmds + other.cmds) #스크립트를 원소와 연산자로 나누어 배열에 넣는 함수 @classmethod def parse(cls, s): #스크립트의 길이 length = read_varint(s) #명령어가 저장될 배열 cmds = [] count = 0 while count < length: #1바이트를 읽는다. current = s.read(1) count += 1 #바이트를 10진수로 변경 current_byte = current[0] #만약 1에서 75사이라면 if current_byte >= 1 and current_byte <= 75: #원소를 저장한다. n = current_byte #커멘드에 저장한다 cmds.append(s.read(n)) count += n #76이면 elif current_byte == 76: #op_pushdata1 이다. data_length = little_endian_to_int(s.read(1)) #데이터 길이만큼 커멘드에 저장한다. cmds.append(s.read(data_length)) count += data_length + 1 elif current_byte == 77: # op_pushdata2 이다. data_length = little_endian_to_int(s.read(2)) cmds.append(s.read(data_length)) count += data_length + 2 else: #나머지는 연산자이다. op_code = current_byte #op code를 그대로 커멘드에넣는다. cmds.append(op_code) if count != length: raise SyntaxError('parsing script failed') return cls(cmds) #스크립트 길이를 포함하지 않는 직렬화 def raw_serialize(self): # initialize what we'll send back result = b'' #커멘드를 돌면서 명령어를 넣는다. for cmd in self.cmds: #원소가 만약 숫자이면 연산자이다. if type(cmd) == int: #그대로 붙인다. result += int_to_little_endian(cmd, 1) else: #아니면 원소이다. #길이를 구해서 length = len(cmd) #각 원소의 길이에 따라 코드를 삽입한다. if length < 75: result += int_to_little_endian(length, 1) elif length > 75 and length < 0x100: # pushdata1 result += int_to_little_endian(76, 1) result += int_to_little_endian(length, 1) elif length >= 0x100 and length <= 520: # pushdata2 result += int_to_little_endian(77, 1) result += int_to_little_endian(length, 2) else: raise ValueError('too long an cmd') result += cmd return result #스크립트 길이를 포함하는 직렬화 def serialize(self): #직렬화 데이터를 가져와서 result = self.raw_serialize() total = len(result) #스크립트 길이를 붙이고 리턴한다. return encode_varint(total) + result #스크립트를 검증하는 함수 #op코드 정의와 연산은 다음 소스코드를 참고하기 바란다. #https://github.com/jimmysong/programmingbitcoin/blob/master/code-ch07/op.py def evaluate(self, z): #커멘드를 별도의 공간에 담는다. cmds = self.cmds[:] stack = [] altstack = [] #cmds 변수가 빌 때 까지 while len(cmds) > 0: #일단 하나를 먼저 꺼낸다. cmd = cmds.pop(0) #연산자이면 if type(cmd) == int: #매핑 함수를 가져온다. operation = OP_CODE_FUNCTIONS[cmd] #op_if, op_notif 인 경우 if cmd in (99, 100): #op_if/op_notif는 cmd 배열의 조작을 필요로 한다. if not operation(stack, cmds): LOGGER.info('bad op: {}'.format(OP_CODE_NAMES[cmd])) return False elif cmd in (107, 108): # op_toaltstack/op_fromaltstack은 다른 스택을 사용한다. if not operation(stack, altstack): LOGGER.info('bad op: {}'.format(OP_CODE_NAMES[cmd])) return False elif cmd in (172, 173, 174, 175): #모두가 서명에 관련된 연산자이다. 서명 해시 z를 필요로 한다. if not operation(stack, z): LOGGER.info('bad op: {}'.format(OP_CODE_NAMES[cmd])) return False else: if not operation(stack): LOGGER.info('bad op: {}'.format(OP_CODE_NAMES[cmd])) return False else: stack.append(cmd) if len(stack) == 0: return False if stack.pop() == b'': return False return True