def tx_input(self, msg):
'''
This message is called on tx input message.
'''
if msg.index != self.input_index:
self.set_main_state()
return proto.Failure(message="Input index doesn't correspond with internal state")
print "RECEIVED INPUT", msg
if msg.index == self.signing_index:
# Store message to cache for serializing input in tx_output
self.signing_input = msg
'''
There we have received one input.
'''
if self.signing_index == 0:
'''
If it is first one, we have to prepare
and hash the beginning of the transaction.
'''
if self.input_index == 0:
# First input, let's hash the beginning of tx
self.input_hash = hashlib.sha256(signing.raw_tx_header(self.inputs_count))
self.tx_hash = self.input_hash
#self.tx_hash.update()
#self.input_hash.update(signing.raw_tx_input())
#self.tx_hash.update(signing.raw_tx_input())
# TODO
'''
For every input, hash the input itself.
'''
print "INPUT HASH", self.input_hash.hexdigest()
# TODO
if self.input_index < self.inputs_count - 1:
'''
If this is not the last input, request next input in the row.
'''
self.input_index += 1
return proto.TxRequest(request_type=proto.TXINPUT,
request_index=self.input_index)
'''
We have processed all inputs. Let's request transaction outputs now.
'''
self.output_index = 0
self.output_hash = hashlib.sha256()
return proto.TxRequest(request_type=proto.TXOUTPUT,
request_index=self.output_index)
def tx_input(self, msg):
'''
This message is called on tx input message.
'''
if msg.index != self.input_index:
self.set_main_state()
return proto.Failure(
message="Input index doesn't correspond with internal state")
print "RECEIVED INPUT", msg
if msg.index == self.signing_index:
# Store message to cache for serializing input in tx_output
self.signing_input = msg
'''
There we have received one input.
'''
if self.signing_index == 0:
'''
If it is first one, we have to prepare
and hash the beginning of the transaction.
'''
if self.input_index == 0:
# First input, let's hash the beginning of tx
self.input_hash = hashlib.sha256(
signing.raw_tx_header(self.inputs_count))
self.tx_hash = self.input_hash
#self.tx_hash.update()
#self.input_hash.update(signing.raw_tx_input())
#self.tx_hash.update(signing.raw_tx_input())
# TODO
'''
For every input, hash the input itself.
'''
print "INPUT HASH", self.input_hash.hexdigest()
# TODO
if self.input_index < self.inputs_count - 1:
'''
If this is not the last input, request next input in the row.
'''
self.input_index += 1
return proto.TxRequest(request_type=proto.TXINPUT,
request_index=self.input_index)
'''
We have processed all inputs. Let's request transaction outputs now.
'''
self.output_index = 0
self.output_hash = hashlib.sha256()
return proto.TxRequest(request_type=proto.TXOUTPUT,
request_index=self.output_index)
def tx_output(self, msg):
'''
This message is called on TxInput message, when serialize_output is False.
It does all the hashing for making input signatures.
'''
res = self._check_output_index(msg)
if res is not None:
return res
res = self._check_address_n(msg)
if res is not None:
return res
if self.output_index == 0:
'''
If it is first one, we have to prepare
and hash the middle of the transaction (between inputs and outputs).
'''
# TODO
'''
Let's hash tx output
'''
print "RECEIVED OUTPUT", msg
if self.input_index == 0:
'''
This is first time we're processing this output,
let's display output details on screen
'''
# self.layout.show_transactions()
print "OUTPUT", msg.address, msg.amount
if self.output_index < self.outputs_count - 1:
'''
This was not the last tx output, so request next one.
'''
self.output_index += 1
return proto.TxRequest(request_type=proto.TXOUTPUT,
request_index=self.output_index)
'''
Now we have processed all inputs and outputs. Let's finalize
hash of transaction.
'''
# Now we have hash of all outputs
print "OUTPUT HASH", self.output_hash.hexdigest()
# We also have tx hash now
print "TX HASH", self.tx_hash.hexdigest()
# We want to send header of tx template
serialized_tx = ''
if self.signing_index == 0:
print "!!! SENDING TX HEADER"
serialized_tx += signing.raw_tx_header(self.inputs_count)
'''
Compute signature for current signing index
'''
print "FINISH INPUT SIGNATURE", self.signing_index
# FIXME, TODO, CHECK
start = time.time()
# privkey = self.bip32.get_private_key(self.signing_input.address_n)
(_, signature) = signing.sign_input(self.bip32,
list(self.signing_input.address_n),
hashlib.sha256(self.tx_hash.digest()).digest())
# (_, signature) = self.bip32.sign_input(self.signing_input.address_n,
# hashlib.sha256(self.tx_hash.digest()).digest())
print 'xxxx', time.time() - start
serialized_tx += 'aaaa' + signing.raw_tx_input(self.signing_input, signature) + 'aaaa' # FIXME, TODO, CHECK
if self.signing_index < self.inputs_count - 1:
'''
If we didn't process all signatures yet,
let's restart the signing process
and ask for first input again.
We're also sending signature for now_signed's input
back to the computer.
'''
now_signed = self.signing_index
self.signing_index += 1
self.input_index = 0
self.input_hash = hashlib.sha256()
return proto.TxRequest(request_type=proto.TXINPUT,
request_index=self.input_index,
signed_index=now_signed,
signature=signature,
serialized_tx=serialized_tx)
'''
We signed all inputs, so it looks like we're done!
Let's ask again for all outputs to finalize serialized transaction.
process_message knows that we're in final stage by self.ser_output flag
and will route messages to serialize_output instead to tx_output.
'''
self.output_index = 0 # We need to reset counter
self.ser_output = True
return proto.TxRequest(request_type=proto.TXOUTPUT,
request_index=0,
signed_index=self.signing_index,
signature=signature,
serialized_tx=serialized_tx)
def tx_output(self, msg):
'''
This message is called on TxInput message, when serialize_output is False.
It does all the hashing for making input signatures.
'''
res = self._check_output_index(msg)
if res != None:
return res
res = self._check_address_n(msg)
if res != None:
return res
if self.output_index == 0:
'''
If it is first one, we have to prepare
and hash the middle of the transaction (between inputs and outputs).
'''
# TODO
'''
Let's hash tx output
'''
print "RECEIVED OUTPUT", msg
if self.input_index == 0:
'''
This is first time we're processing this output,
let's display output details on screen
'''
#self.layout.show_transactions()
print "OUTPUT", msg.address, msg.amount
if self.output_index < self.outputs_count - 1:
'''
This was not the last tx output, so request next one.
'''
self.output_index += 1
return proto.TxRequest(request_type=proto.TXOUTPUT,
request_index=self.output_index)
'''
Now we have processed all inputs and outputs. Let's finalize
hash of transaction.
'''
# Now we have hash of all outputs
print "OUTPUT HASH", self.output_hash.hexdigest()
# We also have tx hash now
print "TX HASH", self.tx_hash.hexdigest()
# We want to send header of tx template
serialized_tx = ''
if self.signing_index == 0:
print "!!! SENDING TX HEADER"
serialized_tx += signing.raw_tx_header(self.inputs_count)
'''
Compute signature for current signing index
'''
print "FINISH INPUT SIGNATURE", self.signing_index
# FIXME, TODO, CHECK
start = time.time()
(_, signature) = self.wallet.sign_input(
self.signing_input.address_n,
hashlib.sha256(self.tx_hash.digest()).digest())
print 'xxxx', time.time() - start
serialized_tx += 'aaaa' + signing.raw_tx_input(
self.signing_input, signature) + 'aaaa' # FIXME, TODO, CHECK
if self.signing_index < self.inputs_count - 1:
'''
If we didn't process all signatures yet,
let's restart the signing process
and ask for first input again.
We're also sending signature for now_signed's input
back to the computer.
'''
now_signed = self.signing_index
self.signing_index += 1
self.input_index = 0
self.input_hash = hashlib.sha256()
return proto.TxRequest(request_type=proto.TXINPUT,
request_index=self.input_index,
signed_index=now_signed,
signature=signature,
serialized_tx=serialized_tx)
'''
We signed all inputs, so it looks like we're done!
Let's ask again for all outputs to finalize serialized transaction.
process_message knows that we're in final stage by self.ser_output flag
and will route messages to serialize_output instead to tx_output.
'''
self.output_index = 0 # We need to reset counter
self.ser_output = True
return proto.TxRequest(request_type=proto.TXOUTPUT,
request_index=0,
signed_index=self.signing_index,
signature=signature,
serialized_tx=serialized_tx)
