You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
231 lines
9.0 KiB
231 lines
9.0 KiB
package main
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"fmt"
|
|
"log"
|
|
"math/big"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/ethereum/go-ethereum/ethdb"
|
|
blockfactory "github.com/harmony-one/harmony/block/factory"
|
|
"github.com/harmony-one/harmony/core"
|
|
core_state "github.com/harmony-one/harmony/core/state"
|
|
"github.com/harmony-one/harmony/core/types"
|
|
"github.com/harmony-one/harmony/core/vm"
|
|
"github.com/harmony-one/harmony/crypto/hash"
|
|
"github.com/harmony-one/harmony/internal/params"
|
|
pkgworker "github.com/harmony-one/harmony/node/worker"
|
|
staking "github.com/harmony-one/harmony/staking/types"
|
|
)
|
|
|
|
const (
|
|
// FaucetContractBinary is binary for faucet contract.
|
|
FaucetContractBinary = "0x60806040526706f05b59d3b2000060015560028054600160a060020a031916331790556101aa806100316000396000f3fe608060405260043610610045577c0100000000000000000000000000000000000000000000000000000000600035046327c78c42811461004a578063b69ef8a81461008c575b600080fd5b34801561005657600080fd5b5061008a6004803603602081101561006d57600080fd5b503573ffffffffffffffffffffffffffffffffffffffff166100b3565b005b34801561009857600080fd5b506100a1610179565b60408051918252519081900360200190f35b60025473ffffffffffffffffffffffffffffffffffffffff1633146100d757600080fd5b600154303110156100e757600080fd5b73ffffffffffffffffffffffffffffffffffffffff811660009081526020819052604090205460ff161561011a57600080fd5b73ffffffffffffffffffffffffffffffffffffffff8116600081815260208190526040808220805460ff1916600190811790915554905181156108fc0292818181858888f19350505050158015610175573d6000803e3d6000fd5b5050565b30319056fea165627a7a723058206b894c1f3badf3b26a7a2768ab8141b1e6fa1c1ddc4622f4f44a7d5041edc9350029"
|
|
)
|
|
|
|
var (
|
|
//FaucetPriKey for the faucet contract Test accounts
|
|
FaucetPriKey, _ = crypto.GenerateKey()
|
|
//FaucetAddress generated via the key.
|
|
FaucetAddress = crypto.PubkeyToAddress(FaucetPriKey.PublicKey)
|
|
//FaucetInitFunds initial funds in facuet contract
|
|
FaucetInitFunds = big.NewInt(8000000000000000000)
|
|
testUserKey, _ = crypto.GenerateKey()
|
|
testUserAddress = crypto.PubkeyToAddress(testUserKey.PublicKey)
|
|
chainConfig = params.TestChainConfig
|
|
blockFactory = blockfactory.ForTest
|
|
// Test transactions
|
|
pendingTxs []*types.Transaction
|
|
database = ethdb.NewMemDatabase()
|
|
gspec = core.Genesis{
|
|
Config: chainConfig,
|
|
Factory: blockFactory,
|
|
Alloc: core.GenesisAlloc{FaucetAddress: {Balance: FaucetInitFunds}},
|
|
ShardID: 0,
|
|
}
|
|
txs []*types.Transaction
|
|
contractworker *pkgworker.Worker
|
|
nonce uint64
|
|
dataEnc []byte
|
|
allRandomUserAddress []common.Address
|
|
allRandomUserKey []*ecdsa.PrivateKey
|
|
faucetContractAddress common.Address
|
|
chain *core.BlockChain
|
|
err error
|
|
state *core_state.DB
|
|
)
|
|
|
|
func init() {
|
|
|
|
firstRandomUserKey, _ := crypto.GenerateKey()
|
|
firstRandomUserAddress := crypto.PubkeyToAddress(firstRandomUserKey.PublicKey)
|
|
|
|
secondRandomUserKey, _ := crypto.GenerateKey()
|
|
secondRandomUserAddress := crypto.PubkeyToAddress(secondRandomUserKey.PublicKey)
|
|
|
|
thirdRandomUserKey, _ := crypto.GenerateKey()
|
|
thirdRandomUserAddress := crypto.PubkeyToAddress(thirdRandomUserKey.PublicKey)
|
|
|
|
//Transactions by first, second and third user with different staking amounts.
|
|
tx1, _ := types.SignTx(types.NewTransaction(0, firstRandomUserAddress, 0, big.NewInt(10), params.TxGas, nil, nil), types.HomesteadSigner{}, FaucetPriKey)
|
|
tx2, _ := types.SignTx(types.NewTransaction(1, secondRandomUserAddress, 0, big.NewInt(20), params.TxGas, nil, nil), types.HomesteadSigner{}, FaucetPriKey)
|
|
tx3, _ := types.SignTx(types.NewTransaction(2, thirdRandomUserAddress, 0, big.NewInt(30), params.TxGas, nil, nil), types.HomesteadSigner{}, FaucetPriKey)
|
|
pendingTxs = append(pendingTxs, tx1)
|
|
pendingTxs = append(pendingTxs, tx2)
|
|
pendingTxs = append(pendingTxs, tx3)
|
|
//tx4, _ := types.SignTx(types.NewTransaction(1, testUserAddress, 0, big.NewInt(1000), params.TxGas, nil, nil), types.HomesteadSigner{}, FaucetPriKey)
|
|
//newTxs = append(newTxs, tx4)
|
|
|
|
}
|
|
|
|
type testWorkerBackend struct {
|
|
db ethdb.Database
|
|
txPool *core.TxPool
|
|
chain *core.BlockChain
|
|
}
|
|
|
|
func fundFaucetContract(chain *core.BlockChain) {
|
|
fmt.Println()
|
|
fmt.Println("--------- Funding addresses for Faucet Contract Call ---------")
|
|
fmt.Println()
|
|
|
|
contractworker = pkgworker.New(params.TestChainConfig, chain, chain.Engine())
|
|
nonce = contractworker.GetCurrentState().GetNonce(crypto.PubkeyToAddress(FaucetPriKey.PublicKey))
|
|
dataEnc = common.FromHex(FaucetContractBinary)
|
|
ftx, _ := types.SignTx(
|
|
types.NewContractCreation(
|
|
nonce, 0, big.NewInt(7000000000000000000),
|
|
params.TxGasContractCreation*10, nil, dataEnc),
|
|
types.HomesteadSigner{}, FaucetPriKey,
|
|
)
|
|
faucetContractAddress = crypto.CreateAddress(FaucetAddress, nonce)
|
|
txs = append(txs, ftx)
|
|
// Funding the user addressed
|
|
for i := 1; i <= 3; i++ {
|
|
randomUserKey, _ := crypto.GenerateKey()
|
|
randomUserAddress := crypto.PubkeyToAddress(randomUserKey.PublicKey)
|
|
amount := i*100000 + 37 // Put different amount in each account.
|
|
tx, _ := types.SignTx(types.NewTransaction(nonce+uint64(i), randomUserAddress, 0, big.NewInt(int64(amount)), params.TxGas, nil, nil), types.HomesteadSigner{}, FaucetPriKey)
|
|
allRandomUserAddress = append(allRandomUserAddress, randomUserAddress)
|
|
allRandomUserKey = append(allRandomUserKey, randomUserKey)
|
|
txs = append(txs, tx)
|
|
}
|
|
amount := 720000
|
|
randomUserKey, _ := crypto.GenerateKey()
|
|
randomUserAddress := crypto.PubkeyToAddress(randomUserKey.PublicKey)
|
|
tx, _ := types.SignTx(types.NewTransaction(nonce+uint64(4), randomUserAddress, 0, big.NewInt(int64(amount)), params.TxGas, nil, nil), types.HomesteadSigner{}, FaucetPriKey)
|
|
txs = append(txs, tx)
|
|
|
|
txmap := make(map[common.Address]types.PoolTransactions)
|
|
for _, tx := range txs {
|
|
txmap[FaucetAddress] = append(txmap[FaucetAddress], tx)
|
|
}
|
|
|
|
err := contractworker.CommitTransactions(
|
|
txmap, testUserAddress,
|
|
)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
block, _ := contractworker.
|
|
FinalizeNewBlock([]byte{}, []byte{}, 0, common.Address{}, nil, nil)
|
|
_, err = chain.InsertChain(types.Blocks{block}, true /* verifyHeaders */)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
state = contractworker.GetCurrentState()
|
|
fmt.Println("Balances before call of faucet contract")
|
|
fmt.Println("contract balance:")
|
|
fmt.Println(state.GetBalance(faucetContractAddress))
|
|
fmt.Println("user address balance")
|
|
fmt.Println(state.GetBalance(allRandomUserAddress[0]))
|
|
fmt.Println()
|
|
fmt.Println("--------- Funding addresses for Faucet Contract Call DONE ---------")
|
|
}
|
|
|
|
func callFaucetContractToFundAnAddress(chain *core.BlockChain) {
|
|
// Send Faucet Contract Transaction ///
|
|
fmt.Println("--------- Now Setting up Faucet Contract Call ---------")
|
|
fmt.Println()
|
|
|
|
paddedAddress := common.LeftPadBytes(allRandomUserAddress[0].Bytes(), 32)
|
|
transferFnSignature := []byte("request(address)")
|
|
|
|
fnHash := hash.Keccak256(transferFnSignature)
|
|
callFuncHex := fnHash[:4]
|
|
|
|
var callEnc []byte
|
|
callEnc = append(callEnc, callFuncHex...)
|
|
callEnc = append(callEnc, paddedAddress...)
|
|
callfaucettx, _ := types.SignTx(types.NewTransaction(nonce+uint64(5), faucetContractAddress, 0, big.NewInt(0), params.TxGasContractCreation*10, nil, callEnc), types.HomesteadSigner{}, FaucetPriKey)
|
|
|
|
txmap := make(map[common.Address]types.PoolTransactions)
|
|
txmap[FaucetAddress] = types.PoolTransactions{callfaucettx}
|
|
|
|
err = contractworker.CommitTransactions(
|
|
txmap, testUserAddress,
|
|
)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
block, _ := contractworker.FinalizeNewBlock(
|
|
[]byte{}, []byte{}, 0, common.Address{}, nil, nil,
|
|
)
|
|
_, err = chain.InsertChain(types.Blocks{block}, true /* verifyHeaders */)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
state = contractworker.GetCurrentState()
|
|
fmt.Println("Balances AFTER CALL of faucet contract")
|
|
fmt.Println("contract balance:")
|
|
fmt.Println(state.GetBalance(faucetContractAddress))
|
|
fmt.Println("user address balance")
|
|
fmt.Println(state.GetBalance(allRandomUserAddress[0]))
|
|
fmt.Println()
|
|
fmt.Println("--------- Faucet Contract Call DONE ---------")
|
|
}
|
|
|
|
func playFaucetContract(chain *core.BlockChain) {
|
|
fundFaucetContract(chain)
|
|
callFaucetContractToFundAnAddress(chain)
|
|
}
|
|
|
|
func main() {
|
|
genesis := gspec.MustCommit(database)
|
|
chain, _ := core.NewBlockChain(database, nil, gspec.Config, chain.Engine(), vm.Config{}, nil)
|
|
|
|
txpool := core.NewTxPool(core.DefaultTxPoolConfig, chainConfig, chain,
|
|
func([]types.RPCTransactionError) {}, func([]staking.RPCTransactionError) {})
|
|
|
|
backend := &testWorkerBackend{
|
|
db: database,
|
|
chain: chain,
|
|
txPool: txpool,
|
|
}
|
|
poolPendingTx := types.PoolTransactions{}
|
|
for _, tx := range pendingTxs {
|
|
poolPendingTx = append(poolPendingTx, tx)
|
|
}
|
|
backend.txPool.AddLocals(poolPendingTx)
|
|
|
|
//// Generate a small n-block chain and an uncle block for it
|
|
n := 3
|
|
if n > 0 {
|
|
blocks, _ := core.GenerateChain(chainConfig, genesis, chain.Engine(), database, n, func(i int, gen *core.BlockGen) {
|
|
gen.SetCoinbase(FaucetAddress)
|
|
gen.SetShardID(0)
|
|
gen.AddTx(pendingTxs[i])
|
|
})
|
|
if _, err := chain.InsertChain(blocks, true /* verifyHeaders */); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
playFaucetContract(chain)
|
|
}
|
|
|