added CalculateHash helper

neo-project · Sep 3, 2020 · 4cea6d0 · 4cea6d0
1 parent 89842e1
commit 4cea6d0
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 3 deletions.
diff --git a/src/neo/Network/P2P/Helper.cs b/src/neo/Network/P2P/Helper.cs
@@ -1,3 +1,4 @@
+using Neo.Cryptography;
 using Neo.Network.P2P.Payloads;
 using System.IO;
 
@@ -21,5 +22,15 @@ public static byte[] GetHashData(this IVerifiable verifiable, uint magic)
                 return ms.ToArray();
             }
         }
+
+        public static UInt256 CalculateHash(this IVerifiable verifiable)
+        {
+            return CalculateHash(verifiable, ProtocolSettings.Default.Magic);
+        }
+
+        public static UInt256 CalculateHash(this IVerifiable verifiable, uint magic)
+        {
+            return new UInt256(Crypto.Hash256(verifiable.GetHashData(magic)));
+        }
     }
 }
diff --git a/src/neo/Network/P2P/Payloads/BlockBase.cs b/src/neo/Network/P2P/Payloads/BlockBase.cs
@@ -26,7 +26,7 @@ public UInt256 Hash
             {
                 if (_hash == null)
                 {
-                    _hash = new UInt256(Crypto.Hash256(this.GetHashData()));
+                    _hash = this.CalculateHash();
                 }
                 return _hash;
             }

diff --git a/src/neo/Network/P2P/Payloads/ConsensusPayload.cs b/src/neo/Network/P2P/Payloads/ConsensusPayload.cs
@@ -45,7 +45,7 @@ public UInt256 Hash
             {
                 if (_hash == null)
                 {
-                    _hash = new UInt256(Crypto.Hash256(this.GetHashData()));
+                    _hash = this.CalculateHash();
                 }
                 return _hash;
             }

diff --git a/src/neo/Network/P2P/Payloads/Transaction.cs b/src/neo/Network/P2P/Payloads/Transaction.cs
@@ -62,7 +62,7 @@ public UInt256 Hash
             {
                 if (_hash == null)
                 {
-                    _hash = new UInt256(Crypto.Hash256(this.GetHashData()));
+                    _hash = this.CalculateHash();
                 }
                 return _hash;
             }