feat: add upgrade config for security enhancement (#349)

store/iavlstore.go

@@ -54,6 +54,8 @@ type IavlStore struct {
 	// By default this value should be set the same across all nodes,
 	// so that nodes can know the waypoints their peers store.
 	storeEvery int64
+
+	diff map[string]struct{}
 }
 
 // CONTRACT: tree should be fully loaded.
@@ -63,6 +65,7 @@ func newIAVLStore(tree *iavl.MutableTree, numRecent int64, storeEvery int64) *Ia
 		Tree:       tree,
 		numRecent:  numRecent,
 		storeEvery: storeEvery,
+		diff:       nil,
 	}
 	return st
 }
@@ -71,6 +74,10 @@ func (st *IavlStore) GetImmutableTree() *iavl.ImmutableTree {
 	return st.Tree.ImmutableTree
 }
 
+func (st *IavlStore) GetTree() *iavl.MutableTree {
+	return st.Tree
+}
+
 func (st *IavlStore) SetVersion(version int64) {
 	st.Tree.SetVersion(version)
 }
@@ -147,6 +154,9 @@ func (st *IavlStore) CacheWrapWithTrace(w io.Writer, tc TraceContext) CacheWrap
 // Implements KVStore.
 func (st *IavlStore) Set(key, value []byte) {
 	st.Tree.Set(key, value)
+	if st.diff != nil {
+		st.diff[string(key)] = struct{}{}
+	}
 }
 
 // Implements KVStore.
@@ -155,6 +165,18 @@ func (st *IavlStore) Get(key []byte) (value []byte) {
 	return v
 }
 
+func (st *IavlStore) EnableDiff() {
+	st.diff = map[string]struct{}{}
+}
+
+func (st *IavlStore) GetDiff() map[string]struct{} {
+	return st.diff
+}
+
+func (st *IavlStore) ResetDiff() {
+	st.diff = map[string]struct{}{}
+}
+
 // Implements KVStore.
 func (st *IavlStore) Has(key []byte) (exists bool) {
 	return st.Tree.Has(key)
@@ -163,6 +185,9 @@ func (st *IavlStore) Has(key []byte) (exists bool) {
 // Implements KVStore.
 func (st *IavlStore) Delete(key []byte) {
 	st.Tree.Remove(key)
+	if st.diff != nil {
+		st.diff[string(key)] = struct{}{}
+	}
 }
 
 // Implements KVStore

types/upgrade.go

@@ -24,7 +24,7 @@ const (
 	BEP173                      = "BEP173" // https://github.com/bnb-chain/BEPs/pull/173
 	FixDoubleSignChainId        = "FixDoubleSignChainId"
 	BEP126                      = "BEP126" //https://github.com/binance-chain/BEPs/pull/126
-
+	BEP255                      = "BEP255" // https://github.com/bnb-chain/BEPs/pull/255
 )
 
 var MainNetConfig = UpgradeConfig{