-
Notifications
You must be signed in to change notification settings - Fork 180
/
handler.go
120 lines (104 loc) · 3.4 KB
/
handler.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package wasm
import (
"fmt"
"github.com/gogo/protobuf/proto"
bam "github.com/okex/exchain/libs/cosmos-sdk/baseapp"
sdk "github.com/okex/exchain/libs/cosmos-sdk/types"
sdkerrors "github.com/okex/exchain/libs/cosmos-sdk/types/errors"
sdktypes "github.com/okex/exchain/libs/cosmos-sdk/x/auth/types"
cfg "github.com/okex/exchain/libs/tendermint/config"
"github.com/okex/exchain/libs/tendermint/libs/kv"
types2 "github.com/okex/exchain/libs/tendermint/types"
"github.com/okex/exchain/x/wasm/keeper"
"github.com/okex/exchain/x/wasm/types"
"github.com/okex/exchain/x/wasm/watcher"
)
// NewHandler returns a handler for "wasm" type messages.
func NewHandler(k types.ContractOpsKeeper) sdk.Handler {
msgServer := keeper.NewMsgServerImpl(k)
return func(ctx sdk.Context, msg sdk.Msg) (*sdk.Result, error) {
ctx.SetEventManager(sdk.NewEventManager())
if !types2.HigherThanEarth(ctx.BlockHeight()) {
errMsg := fmt.Sprintf("wasm not support at height %d", ctx.BlockHeight())
return nil, sdkerrors.Wrap(sdkerrors.ErrUnknownRequest, errMsg)
}
var (
res proto.Message
err error
)
// update watcher
defer func() {
// update watchDB when delivering tx
if ctx.IsDeliverWithSerial() || ctx.ParaMsg() != nil {
watcher.Save(err)
}
if err == nil && !ctx.IsCheckTx() {
updateHGU(ctx, msg)
}
}()
switch msg := msg.(type) {
case *MsgStoreCode: //nolint:typecheck
if !ctx.IsCheckTx() {
types2.WasmStoreCode = true
}
res, err = msgServer.StoreCode(sdk.WrapSDKContext(ctx), msg)
case *MsgInstantiateContract:
res, err = msgServer.InstantiateContract(sdk.WrapSDKContext(ctx), msg)
case *MsgExecuteContract:
res, err = msgServer.ExecuteContract(sdk.WrapSDKContext(ctx), msg)
case *MsgMigrateContract:
res, err = msgServer.MigrateContract(sdk.WrapSDKContext(ctx), msg)
case *MsgUpdateAdmin:
res, err = msgServer.UpdateAdmin(sdk.WrapSDKContext(ctx), msg)
case *MsgClearAdmin:
res, err = msgServer.ClearAdmin(sdk.WrapSDKContext(ctx), msg)
default:
errMsg := fmt.Sprintf("unrecognized wasm message type: %T", msg)
return nil, sdkerrors.Wrap(sdkerrors.ErrUnknownRequest, errMsg)
}
ctx.SetEventManager(filterMessageEvents(ctx))
return sdk.WrapServiceResult(ctx, res, err)
}
}
func updateHGU(ctx sdk.Context, msg sdk.Msg) {
if cfg.DynamicConfig.GetMaxGasUsedPerBlock() <= 0 {
return
}
v, ok := msg.(sdktypes.WasmMsgChecker)
if !ok {
return
}
fnSign, deploySize, err := v.FnSignatureInfo()
if err != nil || len(fnSign) <= 0 {
return
}
gc := int64(ctx.GasMeter().GasConsumed())
if deploySize > 0 {
// calculate average gas consume for deploy contract case, The value is too small and need to +1
gc = gc/int64(deploySize) + 1
}
bam.InstanceOfHistoryGasUsedRecordDB().UpdateGasUsed([]byte(fnSign), gc)
}
// filterMessageEvents returns the same events with all of type == EventTypeMessage removed except
// for wasm message types.
// this is so only our top-level message event comes through
func filterMessageEvents(ctx sdk.Context) *sdk.EventManager {
m := sdk.NewEventManager()
for _, e := range ctx.EventManager().Events() {
if e.Type == sdk.EventTypeMessage &&
!hasWasmModuleAttribute(e.Attributes) {
continue
}
m.EmitEvent(e)
}
return m
}
func hasWasmModuleAttribute(attrs []kv.Pair) bool {
for _, a := range attrs {
if sdk.AttributeKeyModule == string(a.Key) &&
types.ModuleName == string(a.Value) {
return true
}
}
return false
}