From aa0ca5a89341f42f902e81dfa57109bf179ecbae Mon Sep 17 00:00:00 2001 From: Alex Peters Date: Mon, 10 Jul 2023 18:06:29 +0200 Subject: [PATCH] Add post vm options --- x/wasm/keeper/keeper_cgo.go | 9 +++++--- x/wasm/keeper/options.go | 26 +++++++++++++++++++--- x/wasm/keeper/options_test.go | 42 ++++++++++++++++++++++++++++++++--- 3 files changed, 68 insertions(+), 9 deletions(-) diff --git a/x/wasm/keeper/keeper_cgo.go b/x/wasm/keeper/keeper_cgo.go index eb322e9cd9..166f9b8918 100644 --- a/x/wasm/keeper/keeper_cgo.go +++ b/x/wasm/keeper/keeper_cgo.go @@ -35,7 +35,6 @@ func NewKeeper( authority string, opts ...Option, ) Keeper { - keeper := &Keeper{ storeKey: storeKey, cdc: cdc, @@ -56,7 +55,8 @@ func NewKeeper( authority: authority, } keeper.wasmVMQueryHandler = DefaultQueryPlugins(bankKeeper, stakingKeeper, distrKeeper, channelKeeper, keeper) - for _, o := range opts { + preOpts, postOpts := splitOpts(opts) + for _, o := range preOpts { o.apply(keeper) } // only set the wasmvm if no one set this in the options @@ -69,7 +69,10 @@ func NewKeeper( } } - // not updateable, yet + for _, o := range postOpts { + o.apply(keeper) + } + // not updatable, yet keeper.wasmVMResponseHandler = NewDefaultWasmVMContractResponseHandler(NewMessageDispatcher(keeper.messenger, keeper)) return *keeper } diff --git a/x/wasm/keeper/options.go b/x/wasm/keeper/options.go index 144ded5167..a79253e75e 100644 --- a/x/wasm/keeper/options.go +++ b/x/wasm/keeper/options.go @@ -16,6 +16,13 @@ func (f optsFn) apply(keeper *Keeper) { f(keeper) } +// option that is applied after keeper is setup with the VM. Used for decorators mainly. +type postOptsFn func(*Keeper) + +func (f postOptsFn) apply(keeper *Keeper) { + f(keeper) +} + // WithWasmEngine is an optional constructor parameter to replace the default wasmVM engine with the // given one. func WithWasmEngine(x types.WasmerEngine) Option { @@ -26,7 +33,7 @@ func WithWasmEngine(x types.WasmerEngine) Option { // WithWasmEngineDecorator is an optional constructor parameter to decorate the default wasmVM engine. func WithWasmEngineDecorator(d func(old types.WasmerEngine) types.WasmerEngine) Option { - return optsFn(func(k *Keeper) { + return postOptsFn(func(k *Keeper) { k.wasmVM = d(k.wasmVM) }) } @@ -42,7 +49,7 @@ func WithMessageHandler(x Messenger) Option { // WithMessageHandlerDecorator is an optional constructor parameter to decorate the wasm handler for wasmVM messages. // This option should not be combined with Option `WithMessageEncoders` or `WithMessageHandler` func WithMessageHandlerDecorator(d func(old Messenger) Messenger) Option { - return optsFn(func(k *Keeper) { + return postOptsFn(func(k *Keeper) { k.messenger = d(k.messenger) }) } @@ -58,7 +65,7 @@ func WithQueryHandler(x WasmVMQueryHandler) Option { // WithQueryHandlerDecorator is an optional constructor parameter to decorate the default wasm query handler for wasmVM requests. // This option should not be combined with Option `WithQueryPlugins` or `WithQueryHandler` func WithQueryHandlerDecorator(d func(old WasmVMQueryHandler) WasmVMQueryHandler) Option { - return optsFn(func(k *Keeper) { + return postOptsFn(func(k *Keeper) { k.wasmVMQueryHandler = d(k.wasmVMQueryHandler) }) } @@ -189,3 +196,16 @@ func asTypeMap(accts []authtypes.AccountI) map[reflect.Type]struct{} { } return m } + +// split into pre and post VM operations +func splitOpts(opts []Option) ([]Option, []Option) { + pre, post := make([]Option, 0), make([]Option, 0) + for _, o := range opts { + if _, ok := o.(postOptsFn); ok { + post = append(post, o) + } else { + pre = append(pre, o) + } + } + return pre, post +} diff --git a/x/wasm/keeper/options_test.go b/x/wasm/keeper/options_test.go index de2f0fde1c..fe6640b35d 100644 --- a/x/wasm/keeper/options_test.go +++ b/x/wasm/keeper/options_test.go @@ -20,8 +20,9 @@ import ( func TestConstructorOptions(t *testing.T) { specs := map[string]struct { - srcOpt Option - verify func(*testing.T, Keeper) + srcOpt Option + verify func(*testing.T, Keeper) + isPostOpt bool }{ "wasm engine": { srcOpt: WithWasmEngine(&wasmtesting.MockWasmer{}), @@ -37,6 +38,7 @@ func TestConstructorOptions(t *testing.T) { verify: func(t *testing.T, k Keeper) { assert.IsType(t, &wasmtesting.MockWasmer{}, k.wasmVM) }, + isPostOpt: true, }, "message handler": { srcOpt: WithMessageHandler(&wasmtesting.MockMessageHandler{}), @@ -58,6 +60,7 @@ func TestConstructorOptions(t *testing.T) { verify: func(t *testing.T, k Keeper) { assert.IsType(t, &wasmtesting.MockMessageHandler{}, k.messenger) }, + isPostOpt: true, }, "query plugins decorator": { srcOpt: WithQueryHandlerDecorator(func(old WasmVMQueryHandler) WasmVMQueryHandler { @@ -67,6 +70,7 @@ func TestConstructorOptions(t *testing.T) { verify: func(t *testing.T, k Keeper) { assert.IsType(t, &wasmtesting.MockQueryHandler{}, k.wasmVMQueryHandler) }, + isPostOpt: true, }, "coin transferrer": { srcOpt: WithCoinTransferrer(&wasmtesting.MockCoinTransferrer{}), @@ -123,7 +127,10 @@ func TestConstructorOptions(t *testing.T) { } for name, spec := range specs { t.Run(name, func(t *testing.T) { - k := NewKeeper(nil, nil, authkeeper.AccountKeeper{}, &bankkeeper.BaseKeeper{}, stakingkeeper.Keeper{}, nil, nil, nil, nil, nil, nil, nil, nil, "tempDir", types.DefaultWasmConfig(), AvailableCapabilities, "", spec.srcOpt) + opt := spec.srcOpt + _, gotPostOptMarker := opt.(postOptsFn) + require.Equal(t, spec.isPostOpt, gotPostOptMarker) + k := NewKeeper(nil, nil, authkeeper.AccountKeeper{}, &bankkeeper.BaseKeeper{}, stakingkeeper.Keeper{}, nil, nil, nil, nil, nil, nil, nil, nil, "tempDir", types.DefaultWasmConfig(), AvailableCapabilities, "", opt) spec.verify(t, k) }) } @@ -133,3 +140,32 @@ func setAPIDefaults() { costHumanize = DefaultGasCostHumanAddress * DefaultGasMultiplier costCanonical = DefaultGasCostCanonicalAddress * DefaultGasMultiplier } + +func TestSplitOpts(t *testing.T) { + a := optsFn(nil) + b := optsFn(nil) + c := postOptsFn(nil) + d := postOptsFn(nil) + specs := map[string]struct { + src []Option + expPre, expPost []Option + }{ + "by type": { + src: []Option{a, c}, + expPre: []Option{a}, + expPost: []Option{c}, + }, + "ordered": { + src: []Option{a, b, c, d}, + expPre: []Option{a, b}, + expPost: []Option{c, d}, + }, + } + for name, spec := range specs { + t.Run(name, func(t *testing.T) { + gotPre, gotPost := splitOpts(spec.src) + assert.Equal(t, spec.expPre, gotPre) + assert.Equal(t, spec.expPost, gotPost) + }) + } +}