Add GetFullBlocks P2P logic (enable fixing #522)

neo/Network/P2P/Message.cs

@@ -67,6 +67,9 @@ private void DecompressPayload()
                 case MessageCommand.GetBlocks:
                     Payload = decompressed.AsSerializable<GetBlocksPayload>();
                     break;
+                case MessageCommand.GetFullBlocks:
+                    Payload = decompressed.AsSerializable<GetFullBlocksPayload>();
+                    break;
                 case MessageCommand.Headers:
                     Payload = decompressed.AsSerializable<HeadersPayload>();
                     break;

neo/Network/P2P/MessageCommand.cs

@@ -17,6 +17,7 @@ public enum MessageCommand : byte
         Headers = 0x21,
         GetBlocks = 0x24,
         Mempool = 0x25,
+        GetFullBlocks = 0x26,
         Inv = 0x27,
         GetData = 0x28,
         NotFound = 0x2a,

neo/Network/P2P/Payloads/GetFullBlocksPayload.cs

@@ -0,0 +1,37 @@
+using Neo.IO;
+using System;
+using System.IO;
+
+namespace Neo.Network.P2P.Payloads
+{
+    public class GetFullBlocksPayload : ISerializable
+    {
+        public const ushort MaxBlocksCount = 500;
+        public uint IndexStart;
+        public short Count;
+
+        public int Size => sizeof(uint) + sizeof(short);
+
+        public static GetFullBlocksPayload Create(uint index_start, short count = -1)
+        {
+            return new GetFullBlocksPayload
+            {
+                IndexStart = index_start,
+                Count = count
+            };
+        }
+
+        void ISerializable.Deserialize(BinaryReader reader)
+        {
+            IndexStart = reader.ReadUInt32();
+            Count = reader.ReadInt16();
+            if (Count <= 0 || Count > MaxBlocksCount) throw new FormatException();
+        }
+
+        void ISerializable.Serialize(BinaryWriter writer)
+        {
+            writer.Write(IndexStart);
+            writer.Write(Count);
+        }
+    }
+}

neo/Network/P2P/ProtocolHandler.cs

@@ -80,6 +80,9 @@ protected override void OnReceive(object message)
                 case MessageCommand.GetBlocks:
                     OnGetBlocksMessageReceived((GetBlocksPayload)msg.Payload);
                     break;
+                case MessageCommand.GetFullBlocks:
+                    OnGetFullBlocksMessageReceived((GetFullBlocksPayload)msg.Payload);
+                    break;
                 case MessageCommand.GetData:
                     OnGetDataMessageReceived((InvPayload)msg.Payload);
                     break;
@@ -175,6 +178,26 @@ private void OnGetBlocksMessageReceived(GetBlocksPayload payload)
             Context.Parent.Tell(Message.Create(MessageCommand.Inv, InvPayload.Create(InventoryType.Block, hashes.ToArray())));
         }
 
+        private void OnGetFullBlocksMessageReceived(GetFullBlocksPayload payload)
+        {
+            for (uint i = payload.IndexStart; i < payload.IndexStart + payload.Count; i++)
+            {
+                Block block = Blockchain.Singleton.Store.GetBlock(i);
+                if (block == null)
+                    break;
+
+                if (bloom_filter == null)
+                {
+                    Context.Parent.Tell(Message.Create(MessageCommand.Block, block));
+                }
+                else
+                {
+                    BitArray flags = new BitArray(block.Transactions.Select(p => bloom_filter.Test(p)).ToArray());
+                    Context.Parent.Tell(Message.Create(MessageCommand.MerkleBlock, MerkleBlockPayload.Create(block, flags)));
+                }
+            }
+        }
+
         private void OnGetDataMessageReceived(InvPayload payload)
         {
             UInt256[] hashes = payload.Hashes.Where(p => sentHashes.Add(p)).ToArray();