fix mpt find

neo-project · May 28, 2020 · f915b56 · f915b56
1 parent 822291d
commit f915b56
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/neo/Cryptography/MPT/MPTTrie.Find.cs b/src/neo/Cryptography/MPT/MPTTrie.Find.cs
@@ -42,7 +42,7 @@ private ReadOnlySpan<byte> Seek(ref MPTNode node, ReadOnlySpan<byte> path, out M
                     {
                         if (path.Length < 1)
                         {
-                            start = extensionNode;
+                            start = extensionNode.Next;
                             return ReadOnlySpan<byte>.Empty;
                         }
                         if (path.StartsWith(extensionNode.Key))
@@ -51,8 +51,8 @@ private ReadOnlySpan<byte> Seek(ref MPTNode node, ReadOnlySpan<byte> path, out M
                         }
                         if (extensionNode.Key.AsSpan().StartsWith(path))
                         {
-                            start = extensionNode;
-                            return extensionNode.Key[path.Length..];
+                            start = extensionNode.Next;
+                            return extensionNode.Key;
                         }
                         break;
                     }
@@ -71,6 +71,7 @@ private ReadOnlySpan<byte> Seek(ref MPTNode node, ReadOnlySpan<byte> path, out M
 
         private IEnumerable<(byte[] Key, byte[] Value)> Travers(MPTNode node, byte[] path)
         {
+            if (node is null) yield break;
             switch (node)
             {
                 case LeafNode leafNode: