StateService: add key length check

Author: ZhangTao1596

src/StateService/MPT/MPTTrie.Delete.cs

@@ -10,7 +10,10 @@ partial class MPTTrie<TKey, TValue>
         public bool Delete(TKey key)
         {
             var path = ToNibbles(key.ToArray());
-            if (path.Length == 0) throw new ArgumentException("could not be empty", nameof(key));
+            if (path.Length == 0)
+                throw new ArgumentException("could not be empty", nameof(key));
+            if (path.Length > MPTNode.MaxKeyLength)
+                throw new ArgumentException("exceeds limit", nameof(key));
             return TryDelete(ref root, path);
         }
 

src/StateService/MPT/MPTTrie.Find.cs

@@ -74,6 +74,8 @@ private ReadOnlySpan<byte> Seek(ref MPTNode node, ReadOnlySpan<byte> path, out M
                 offset = prefix.Length * 2;
                 from = ToNibbles(from.AsSpan());
             }
+            if (path.Length > MPTNode.MaxKeyLength || from.Length > MPTNode.MaxKeyLength)
+                throw new ArgumentException("exceeds limit");
             path = Seek(ref root, path, out MPTNode start).ToArray();
             return Travers(start, path, from, offset)
                 .Select(p => (FromNibbles(p.Key).AsSerializable<TKey>(), p.Value.AsSerializable<TValue>()));

src/StateService/MPT/MPTTrie.Get.cs

@@ -11,7 +11,10 @@ public TValue this[TKey key]
             get
             {
                 var path = ToNibbles(key.ToArray());
-                if (path.Length == 0) throw new ArgumentException("could not be empty", nameof(key));
+                if (path.Length == 0)
+                    throw new ArgumentException("could not be empty", nameof(key));
+                if (path.Length > MPTNode.MaxKeyLength)
+                    throw new ArgumentException("exceeds limit", nameof(key));
                 var result = TryGet(ref root, path, out var value);
                 return result ? value.ToArray().AsSerializable<TValue>() : throw new KeyNotFoundException();
             }
@@ -21,7 +24,10 @@ public bool TryGetValue(TKey key, out TValue value)
         {
             value = default;
             var path = ToNibbles(key.ToArray());
-            if (path.Length == 0) throw new ArgumentException("could not be empty", nameof(key));
+            if (path.Length == 0)
+                throw new ArgumentException("could not be empty", nameof(key));
+            if (path.Length > MPTNode.MaxKeyLength)
+                throw new ArgumentException("exceeds limit", nameof(key));
             var result = TryGet(ref root, path, out var val);
             if (result)
                 val.ToArray().AsSerializable<TValue>();

src/StateService/MPT/MPTTrie.Proof.cs

@@ -13,7 +13,10 @@ partial class MPTTrie<TKey, TValue>
         public bool TryGetProof(TKey key, out HashSet<byte[]> proof)
         {
             var path = ToNibbles(key.ToArray());
-            if (path.Length == 0) throw new ArgumentException("could not be empty", nameof(key));
+            if (path.Length == 0)
+                throw new ArgumentException("could not be empty", nameof(key));
+            if (path.Length > MPTNode.MaxKeyLength)
+                throw new ArgumentException("exceeds limit", nameof(key));
             proof = new HashSet<byte[]>(ByteArrayEqualityComparer.Default);
             return GetProof(ref root, path, proof);
         }