Add filterload wire message

Author: cpacia

bitcoin/messages.py

@@ -33,6 +33,7 @@
 from bitcoin.core import *
 from bitcoin.core.serialize import *
 from bitcoin.net import *
+from bitcoin.bloom import CBloomFilter
 import bitcoin
 
 MSG_TX = 1
@@ -505,6 +506,28 @@ def msg_ser(self, f):
     def __repr__(self):
         return "msg_mempool()"
 
+
+class msg_filterload(MsgSerializable):
+    command = b"filterload"
+
+    def __init__(self, protover=PROTO_VERSION, filter=None):
+        super(msg_filterload, self).__init__(protover)
+        self.protover = protover
+        self.filter = filter
+
+    @classmethod
+    def msg_deser(cls, f, protover=PROTO_VERSION):
+        c = cls()
+        c.filter = CBloomFilter.stream_deserialize(f)
+        return c
+
+    def msg_ser(self, f):
+        self.filter.stream_serialize(f)
+
+    def __repr__(self):
+        return "msg_filterload(vData=%i nHashFunctions=%i nTweak=%i nFlags=%i" % (self.filter.vData, self.filter.nHashFunctions, self.filter.nTweak, self.filter.nFlags)
+
+
 msg_classes = [msg_version, msg_verack, msg_addr, msg_alert, msg_inv,
                msg_getdata, msg_notfound, msg_getblocks, msg_getheaders,
                msg_headers, msg_tx, msg_block, msg_getaddr, msg_ping,
@@ -536,6 +559,7 @@ def __repr__(self):
         'msg_ping',
         'msg_pong',
         'msg_mempool',
+        'msg_filterload',
         'msg_classes',
         'messagemap',
 )

bitcoin/tests/test_messages.py

@@ -13,8 +13,8 @@
 
 from bitcoin.messages import msg_version, msg_verack, msg_addr, msg_alert, \
     msg_inv, msg_getdata, msg_getblocks, msg_getheaders, msg_headers, msg_tx, \
-    msg_block, msg_merkleblock, msg_getaddr, msg_ping, msg_pong, msg_mempool, MsgSerializable, \
-    msg_notfound, msg_reject
+    msg_block, msg_merkleblock, msg_getaddr, msg_ping, msg_pong, msg_mempool, msg_filterload, \
+    MsgSerializable, msg_notfound, msg_reject
 
 import sys
 if sys.version > '3':
@@ -122,6 +122,11 @@ def test_serialization(self):
         super(Test_msg_mempool, self).serialization_test(msg_mempool)
 
 
+class Test_msg_filterload(MessageTestCase):
+    def test_serialization(self):
+        super(Test_msg_filterload, self).serialization_test(msg_filterload)
+
+
 class Test_messages(unittest.TestCase):
     verackbytes = b'\xf9\xbe\xb4\xd9verack\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00]\xf6\xe0\xe2'