-
Notifications
You must be signed in to change notification settings - Fork 67
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add support for creating allocations via Filplus EVM Client Contract
- Loading branch information
1 parent
eff5121
commit 2cc97f8
Showing
6 changed files
with
488 additions
and
42 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,272 @@ | ||
| package util | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "fmt" | ||
| "strings" | ||
|
|
||
| mbig "math/big" | ||
|
|
||
| eabi "github.com/ethereum/go-ethereum/accounts/abi" | ||
| "github.com/filecoin-project/go-address" | ||
| "github.com/filecoin-project/go-state-types/abi" | ||
| "github.com/filecoin-project/go-state-types/big" | ||
| "github.com/filecoin-project/go-state-types/builtin" | ||
| verifreg9 "github.com/filecoin-project/go-state-types/builtin/v9/verifreg" | ||
| "github.com/filecoin-project/lotus/api" | ||
| "github.com/filecoin-project/lotus/chain/actors" | ||
| "github.com/filecoin-project/lotus/chain/types" | ||
| "github.com/filecoin-project/lotus/chain/types/ethtypes" | ||
| ) | ||
|
|
||
| // https://github.com/fidlabs/contract-metaallocator/blob/main/src/Client.sol | ||
| const contractABI = `[ | ||
| { | ||
| "type": "function", | ||
| "name": "allowances", | ||
| "inputs": [ | ||
| { | ||
| "name": "client", | ||
| "type": "address", | ||
| "internalType": "address" | ||
| } | ||
| ], | ||
| "outputs": [ | ||
| { | ||
| "name": "allowance", | ||
| "type": "uint256", | ||
| "internalType": "uint256" | ||
| } | ||
| ], | ||
| "stateMutability": "view" | ||
| }, | ||
| { | ||
| "type": "function", | ||
| "name": "transfer", | ||
| "inputs": [ | ||
| { | ||
| "name": "params", | ||
| "type": "tuple", | ||
| "internalType": "struct DataCapTypes.TransferParams", | ||
| "components": [ | ||
| { | ||
| "name": "to", | ||
| "type": "tuple", | ||
| "internalType": "struct CommonTypes.FilAddress", | ||
| "components": [ | ||
| { | ||
| "name": "data", | ||
| "type": "bytes", | ||
| "internalType": "bytes" | ||
| } | ||
| ] | ||
| }, | ||
| { | ||
| "name": "amount", | ||
| "type": "tuple", | ||
| "internalType": "struct CommonTypes.BigInt", | ||
| "components": [ | ||
| { | ||
| "name": "val", | ||
| "type": "bytes", | ||
| "internalType": "bytes" | ||
| }, | ||
| { | ||
| "name": "neg", | ||
| "type": "bool", | ||
| "internalType": "bool" | ||
| } | ||
| ] | ||
| }, | ||
| { | ||
| "name": "operator_data", | ||
| "type": "bytes", | ||
| "internalType": "bytes" | ||
| } | ||
| ] | ||
| } | ||
| ], | ||
| "outputs": [], | ||
| "stateMutability": "nonpayable" | ||
| } | ||
| ]` | ||
|
|
||
| func getAddressAllowanceOnContract(ctx context.Context, api api.Gateway, wallet address.Address, contract address.Address) (*big.Int, error) { | ||
| // Parse the contract ABI | ||
| parsedABI, err := eabi.JSON(strings.NewReader(contractABI)) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("Failed to parse contract ABI: %w", err) | ||
| } | ||
|
|
||
| // Convert from Filecoin to Eth Address | ||
| walletId, err := api.StateLookupID(ctx, wallet, types.EmptyTSK) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| walletEvm, err := ethtypes.EthAddressFromFilecoinAddress(walletId) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Prepare EVM calldata | ||
| calldata, err := parsedABI.Pack("allowances", walletEvm) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to serialize parameters to check allowances: %w", err) | ||
| } | ||
|
|
||
| // Encode EVM calldata as Message parameters | ||
| allowanceParam := abi.CborBytes(calldata) | ||
| allowanceParams, err := actors.SerializeParams(&allowanceParam) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to serialize params: %w", err) | ||
| } | ||
|
|
||
| // Read allowance from the contract | ||
| allowanceMsg := &types.Message{ | ||
| To: contract, | ||
| From: wallet, | ||
| Method: builtin.MethodsEVM.InvokeContract, | ||
| Params: allowanceParams, | ||
| Value: big.Zero(), | ||
| } | ||
| result, err := api.StateCall(ctx, allowanceMsg, types.EmptyTSK) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Decode return value (cbor -> evm ABI -> math/big Int -> filecoin big Int) | ||
| var decodedReturn abi.CborBytes | ||
| decodedReturn.UnmarshalCBOR(bytes.NewReader(result.MsgRct.Return)) | ||
| retValue, err := parsedABI.Unpack("allowances", decodedReturn) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
| allowance, err := big.FromString(retValue[0].(*mbig.Int).String()) | ||
| return &allowance, err | ||
| } | ||
|
|
||
| func buildTransferViaEVMParams(amount *big.Int, receiverParams []byte) ([]byte, error) { | ||
| // Parse the contract's ABI | ||
| parsedABI, err := eabi.JSON(strings.NewReader(contractABI)) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("Failed to parse contract ABI: %w", err) | ||
| } | ||
|
|
||
| // convert amount from Filecoin big.Int to math/big Int | ||
| var amountMbig mbig.Int | ||
| _, success := amountMbig.SetString(amount.String(), 10) | ||
| if !success { | ||
| return nil, fmt.Errorf("failed to serialize the amount") | ||
| } | ||
|
|
||
| // build calldata | ||
| calldata, err := parsedABI.Pack( | ||
| "transfer", | ||
| TransferParams{ | ||
| To: FilAddress{Data: builtin.VerifiedRegistryActorAddr.Bytes()}, | ||
| Amount: BigInt{ | ||
| Neg: amount.LessThan(big.Zero()), | ||
| Val: amountMbig.Bytes(), | ||
| }, | ||
| OperatorData: receiverParams, | ||
| }) | ||
|
|
||
| transferParam := abi.CborBytes(calldata) | ||
| transferParams, err := actors.SerializeParams(&transferParam) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to serialize params: %w", err) | ||
| } | ||
| return transferParams, nil | ||
| } | ||
|
|
||
| func CreateAllocationViaEVMMsg(ctx context.Context, api api.Gateway, infos []PieceInfos, wallet address.Address, contract address.Address, batchSize int) ([]*types.Message, error) { | ||
| // Create allocation requests | ||
| rDataCap, allocationRequests, err := CreateAllocationRequests(ctx, api, infos) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Get datacap balance of the contract | ||
| aDataCap, err := api.StateVerifiedClientStatus(ctx, contract, types.EmptyTSK) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| if aDataCap == nil { | ||
| return nil, fmt.Errorf("contract %s does not have any datacap", wallet) | ||
| } | ||
|
|
||
| // Check that we have enough data cap to make the allocation | ||
| if rDataCap.GreaterThan(big.NewInt(aDataCap.Int64())) { | ||
| return nil, fmt.Errorf("requested datacap %s is greater then the available contract datacap %s", rDataCap, aDataCap) | ||
| } | ||
|
|
||
| // Get datacap allowance of the wallet | ||
| allowance, err := getAddressAllowanceOnContract(ctx, api, wallet, contract) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| // Check that we have enough data cap to make the allocation | ||
| if rDataCap.GreaterThan(*allowance) { | ||
| return nil, fmt.Errorf("requested datacap %s is greater then the available datacap allowance %s", rDataCap, allowance) | ||
| } | ||
|
|
||
| // Batch allocationRequests to create message | ||
| var messages []*types.Message | ||
| for i := 0; i < len(allocationRequests); i += batchSize { | ||
| end := i + batchSize | ||
| if end > len(allocationRequests) { | ||
| end = len(allocationRequests) | ||
| } | ||
| batch := allocationRequests[i:end] | ||
| arequest := &verifreg9.AllocationRequests{ | ||
| Allocations: batch, | ||
| } | ||
| bDataCap := big.NewInt(0) | ||
| for _, bd := range batch { | ||
| bDataCap.Add(big.NewInt(int64(bd.Size)).Int, bDataCap.Int) | ||
| } | ||
|
|
||
| receiverParams, err := actors.SerializeParams(arequest) | ||
| if err != nil { | ||
| return nil, fmt.Errorf("failed to serialize the parameters: %w", err) | ||
| } | ||
|
|
||
| amount := big.Mul(bDataCap, builtin.TokenPrecision) | ||
|
|
||
| transferParams, error := buildTransferViaEVMParams(&amount, receiverParams) | ||
| if error != nil { | ||
| return nil, error | ||
| } | ||
|
|
||
| msg := &types.Message{ | ||
| To: contract, | ||
| From: wallet, | ||
| Method: builtin.MethodsEVM.InvokeContract, | ||
| Params: transferParams, | ||
| Value: big.Zero(), | ||
| } | ||
| messages = append(messages, msg) | ||
| } | ||
| return messages, nil | ||
| } | ||
|
|
||
| // https://github.com/filecoin-project/filecoin-solidity/blob/f655709ab02fcf4b7859f47149f1e0cbfa975041/contracts/v0.8/types/CommonTypes.sol#L86 | ||
| type FilAddress struct { | ||
| Data []byte | ||
| } | ||
|
|
||
| // https://github.com/filecoin-project/filecoin-solidity/blob/f655709ab02fcf4b7859f47149f1e0cbfa975041/contracts/v0.8/types/CommonTypes.sol#L80 | ||
| type BigInt struct { | ||
| Val []byte | ||
| Neg bool | ||
| } | ||
|
|
||
| // https://github.com/filecoin-project/filecoin-solidity/blob/f655709ab02fcf4b7859f47149f1e0cbfa975041/contracts/v0.8/types/DataCapTypes.sol#L52 | ||
| type TransferParams struct { | ||
| To FilAddress | ||
| Amount BigInt | ||
| OperatorData []byte | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.