Allow to search in witnesses (#1931)

Co-authored-by: Erik Zhang <erik@neo.org>
neo-project · Sep 14, 2020 · 9a160c4 · 9a160c4
1 parent c18a267
commit 9a160c4
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/neo/Wallets/Wallet.cs b/src/neo/Wallets/Wallet.cs
@@ -371,6 +371,21 @@ public long CalculateNetworkFee(StoreView snapshot, Transaction tx)
             foreach (UInt160 hash in hashes)
             {
                 byte[] witness_script = GetAccount(hash)?.Contract?.Script;
+
+                if (witness_script is null && tx.Witnesses != null)
+                {
+                    // Try to find the script in the witnesses
+
+                    foreach (var witness in tx.Witnesses)
+                    {
+                        if (witness.ScriptHash == hash)
+                        {
+                            witness_script = witness.VerificationScript;
+                            break;
+                        }
+                    }
+                }
+
                 if (witness_script is null)
                 {
                     var contract = snapshot.Contracts.TryGet(hash);