72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
package conformance
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/vm"
|
|
"github.com/filecoin-project/lotus/extern/sector-storage/ffiwrapper"
|
|
"github.com/filecoin-project/lotus/lib/blockstore"
|
|
|
|
"github.com/filecoin-project/specs-actors/actors/abi"
|
|
"github.com/filecoin-project/specs-actors/actors/puppet"
|
|
|
|
"github.com/filecoin-project/test-vectors/chaos"
|
|
"github.com/filecoin-project/test-vectors/schema"
|
|
|
|
"github.com/ipfs/go-cid"
|
|
)
|
|
|
|
var (
|
|
// BaseFee to use in the VM.
|
|
// TODO make parametrisable through vector.
|
|
BaseFee = abi.NewTokenAmount(100)
|
|
)
|
|
|
|
type Driver struct {
|
|
ctx context.Context
|
|
vector *schema.TestVector
|
|
}
|
|
|
|
func NewDriver(ctx context.Context, vector *schema.TestVector) *Driver {
|
|
return &Driver{ctx: ctx, vector: vector}
|
|
}
|
|
|
|
// ExecuteMessage executes a conformance test vector message in a temporary VM.
|
|
func (d *Driver) ExecuteMessage(msg *types.Message, preroot cid.Cid, bs blockstore.Blockstore, epoch abi.ChainEpoch) (*vm.ApplyRet, cid.Cid, error) {
|
|
vmOpts := &vm.VMOpts{
|
|
StateBase: preroot,
|
|
Epoch: epoch,
|
|
Rand: &testRand{}, // TODO always succeeds; need more flexibility.
|
|
Bstore: bs,
|
|
Syscalls: mkFakedSigSyscalls(vm.Syscalls(ffiwrapper.ProofVerifier)), // TODO always succeeds; need more flexibility.
|
|
CircSupplyCalc: nil,
|
|
BaseFee: BaseFee,
|
|
}
|
|
|
|
lvm, err := vm.NewVM(vmOpts)
|
|
if err != nil {
|
|
return nil, cid.Undef, err
|
|
}
|
|
|
|
invoker := vm.NewInvoker()
|
|
|
|
// add support for the puppet and chaos actors.
|
|
if puppetOn, ok := d.vector.Selector["puppet_actor"]; ok && puppetOn == "true" {
|
|
invoker.Register(puppet.PuppetActorCodeID, puppet.Actor{}, puppet.State{})
|
|
}
|
|
if chaosOn, ok := d.vector.Selector["chaos_actor"]; ok && chaosOn == "true" {
|
|
invoker.Register(chaos.ChaosActorCodeCID, chaos.Actor{}, chaos.State{})
|
|
}
|
|
|
|
lvm.SetInvoker(invoker)
|
|
|
|
ret, err := lvm.ApplyMessage(d.ctx, msg)
|
|
if err != nil {
|
|
return nil, cid.Undef, err
|
|
}
|
|
|
|
root, err := lvm.Flush(d.ctx)
|
|
return ret, root, err
|
|
}
|