be16f80abc
Any projects which want to use Hush code from now on will need to be licensed as GPLv3 or we will send the lawyers: https://www.softwarefreedom.org/ Notably, Komodo (KMD) is licensed as GPLv2 and is no longer compatible to receive code changes, without causing legal issues. MIT projects, such as Zcash, also cannot pull in changes from the Hush Full Node without permission from The Hush Developers, which may in some circumstances grant an MIT license on a case-by-case basis.
96 lines
3.1 KiB
Python
Executable File
96 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# Copyright (c) 2015 The Bitcoin Core developers
|
|
# Distributed under the GPLv3 software license, see the accompanying
|
|
# file COPYING or https://www.gnu.org/licenses/gpl-3.0.en.html
|
|
|
|
#
|
|
# Test ZMQ interface
|
|
#
|
|
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import assert_equal, bytes_to_hex_str, start_nodes
|
|
|
|
import zmq
|
|
import struct
|
|
|
|
class ZMQTest(BitcoinTestFramework):
|
|
|
|
port = 28332
|
|
|
|
def setup_nodes(self):
|
|
self.zmqContext = zmq.Context()
|
|
self.zmqSubSocket = self.zmqContext.socket(zmq.SUB)
|
|
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashblock")
|
|
self.zmqSubSocket.setsockopt(zmq.SUBSCRIBE, b"hashtx")
|
|
self.zmqSubSocket.connect("tcp://127.0.0.1:%i" % self.port)
|
|
return start_nodes(4, self.options.tmpdir, extra_args=[
|
|
['-zmqpubhashtx=tcp://127.0.0.1:'+str(self.port), '-zmqpubhashblock=tcp://127.0.0.1:'+str(self.port)],
|
|
[],
|
|
[],
|
|
[]
|
|
])
|
|
|
|
def run_test(self):
|
|
self.sync_all()
|
|
|
|
genhashes = self.nodes[0].generate(1)
|
|
self.sync_all()
|
|
|
|
print "listen..."
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
body = msg[1]
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, 0) #must be sequence 0 on hashblock
|
|
blkhash = bytes_to_hex_str(body)
|
|
|
|
assert_equal(genhashes[0], blkhash) #blockhash from generate must be equal to the hash received over zmq
|
|
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
assert_equal(topic, b"hashtx")
|
|
body = msg[1]
|
|
nseq = msg[2]
|
|
[nseq] # hush pyflakes
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, 0) # must be sequence 0 on hashtx
|
|
|
|
n = 10
|
|
genhashes = self.nodes[1].generate(n)
|
|
self.sync_all()
|
|
|
|
zmqHashes = []
|
|
blockcount = 0
|
|
for x in range(0,n*2):
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
body = msg[1]
|
|
if topic == b"hashblock":
|
|
zmqHashes.append(bytes_to_hex_str(body))
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, blockcount+1)
|
|
blockcount += 1
|
|
|
|
for x in range(0,n):
|
|
assert_equal(genhashes[x], zmqHashes[x]) #blockhash from generate must be equal to the hash received over zmq
|
|
|
|
#test tx from a second node
|
|
hashRPC = self.nodes[1].sendtoaddress(self.nodes[0].getnewaddress(), 1.0)
|
|
self.sync_all()
|
|
|
|
# now we should receive a zmq msg because the tx was broadcast
|
|
msg = self.zmqSubSocket.recv_multipart()
|
|
topic = msg[0]
|
|
body = msg[1]
|
|
hashZMQ = ""
|
|
if topic == b"hashtx":
|
|
hashZMQ = bytes_to_hex_str(body)
|
|
msgSequence = struct.unpack('<I', msg[-1])[-1]
|
|
assert_equal(msgSequence, blockcount+1)
|
|
|
|
assert_equal(hashRPC, hashZMQ) #blockhash from generate must be equal to the hash received over zmq
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ZMQTest ().main ()
|