From 78046615a085a7e1213efeaa70322e3e4eb44bfa Mon Sep 17 00:00:00 2001
From: Ben Adams <thundercat@illyriad.co.uk>
Date: Thu, 22 Jun 2023 15:50:38 +0100
Subject: [PATCH] Prefer code from code cache for Create (#5854)

* Prefer code from code cache for Create

* Filter code inserts

* Add Note
---
 .../Nethermind.Evm/VirtualMachine.cs          | 12 ++++++-
 .../Nethermind.State/StateProvider.cs         | 33 +++++++++++++++++--
 2 files changed, 42 insertions(+), 3 deletions(-)

diff --git a/src/Nethermind/Nethermind.Evm/VirtualMachine.cs b/src/Nethermind/Nethermind.Evm/VirtualMachine.cs
index 3342381621c..cb07fdbfa0b 100644
--- a/src/Nethermind/Nethermind.Evm/VirtualMachine.cs
+++ b/src/Nethermind/Nethermind.Evm/VirtualMachine.cs
@@ -2444,6 +2444,16 @@ private bool InstructionSelfDestruct<TTracing>(EvmState vmState, ref EvmStack<TT
         }
 
         _state.SubtractFromBalance(env.ExecutingAccount, value, spec);
+
+        ValueKeccak codeHash = ValueKeccak.Compute(initCode);
+        // Prefer code from code cache (e.g. if create from a factory contract or copypasta)
+        if (!_codeCache.TryGet(codeHash, out CodeInfo codeInfo))
+        {
+            codeInfo = new(initCode.ToArray());
+            // Prime the code cache as likely to be used by more txs
+            _codeCache.Set(codeHash, codeInfo);
+        }
+
         ExecutionEnvironment callEnv = new
         (
             txExecutionContext: env.TxExecutionContext,
@@ -2451,7 +2461,7 @@ private bool InstructionSelfDestruct<TTracing>(EvmState vmState, ref EvmStack<TT
             caller: env.ExecutingAccount,
             executingAccount: contractAddress,
             codeSource: null,
-            codeInfo: new CodeInfo(initCode.ToArray()),
+            codeInfo: codeInfo,
             inputData: default,
             transferValue: value,
             value: value
diff --git a/src/Nethermind/Nethermind.State/StateProvider.cs b/src/Nethermind/Nethermind.State/StateProvider.cs
index 7a265c42c89..b4e20f219ea 100644
--- a/src/Nethermind/Nethermind.State/StateProvider.cs
+++ b/src/Nethermind/Nethermind.State/StateProvider.cs
@@ -3,11 +3,13 @@
 
 using System;
 using System.Collections.Generic;
-using System.Runtime.CompilerServices;
+using System.Runtime.InteropServices;
 using Nethermind.Core;
+using Nethermind.Core.Caching;
 using Nethermind.Core.Crypto;
 using Nethermind.Core.Resettables;
 using Nethermind.Core.Specs;
+using Nethermind.Db;
 using Nethermind.Int256;
 using Nethermind.Logging;
 using Nethermind.State.Tracing;
@@ -23,6 +25,11 @@ internal class StateProvider
         private const int StartCapacity = Resettable.StartCapacity;
         private readonly ResettableDictionary<Address, Stack<int>> _intraBlockCache = new();
         private readonly ResettableHashSet<Address> _committedThisRound = new();
+        // Only guarding against hot duplicates so filter doesn't need to be too big
+        // Note:
+        // False negatives are fine as they will just result in a overwrite set
+        // False positives would be problematic as the code _must_ be persisted
+        private readonly LruKeyCache<Keccak> _codeInsertFilter = new(2048, "Code Insert Filter");
 
         private readonly List<Change> _keptInCache = new();
         private readonly ILogger _logger;
@@ -141,7 +148,29 @@ public void InsertCode(Address address, ReadOnlyMemory<byte> code, IReleaseSpec
         {
             _needsStateRootUpdate = true;
             Keccak codeHash = code.Length == 0 ? Keccak.OfAnEmptyString : Keccak.Compute(code.Span);
-            _codeDb[codeHash.Bytes] = code.ToArray();
+
+            // Don't reinsert if already inserted. This can be the case when the same
+            // code is used by multiple deployments. Either from factory contracts (e.g. LPs)
+            // or people copy and pasting popular contracts
+            if (!_codeInsertFilter.Get(codeHash))
+            {
+                if (_codeDb is IDbWithSpan dbWithSpan)
+                {
+                    dbWithSpan.PutSpan(codeHash.Bytes, code.Span);
+                }
+                else if (MemoryMarshal.TryGetArray(code, out ArraySegment<byte> codeArray)
+                        && codeArray.Offset == 0
+                        && codeArray.Count == code.Length)
+                {
+                    _codeDb[codeHash.Bytes] = codeArray.Array;
+                }
+                else
+                {
+                    _codeDb[codeHash.Bytes] = code.ToArray();
+                }
+
+                _codeInsertFilter.Set(codeHash);
+            }
 
             Account? account = GetThroughCache(address);
             if (account is null)