From 9f2d737045121ef9eba0a64a7e769a05118e3d03 Mon Sep 17 00:00:00 2001
From: Tamas Blummer <tamas.blummer@gmail.com>
Date: Mon, 28 May 2018 15:24:35 +0200
Subject: [PATCH] add witness inv types

---
 src/network/message_blockdata.rs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/network/message_blockdata.rs b/src/network/message_blockdata.rs
index 556b3c3f..4f2d7c97 100644
--- a/src/network/message_blockdata.rs
+++ b/src/network/message_blockdata.rs
@@ -31,7 +31,11 @@ pub enum InvType {
     /// Transaction
     Transaction,
     /// Block
-    Block
+    Block,
+    /// Witness Block
+    WitnessBlock,
+    /// Witness Transaction
+    WitnessTransaction
 }
 
 // Some simple messages
@@ -103,7 +107,9 @@ impl<S: SimpleEncoder> ConsensusEncodable<S> for Inventory {
         try!(match self.inv_type {
             InvType::Error => 0u32, 
             InvType::Transaction => 1,
-            InvType::Block => 2
+            InvType::Block => 2,
+            InvType::WitnessBlock => 0x40000002,
+            InvType::WitnessTransaction => 0x40000001
         }.consensus_encode(s));
         self.hash.consensus_encode(s)
     }