Add SignHash stub in wallet

polynetwork · Jun 29, 2022 · 4a14ff6 · 4a14ff6
1 parent f5ae35e
commit 4a14ff6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/wallet/key.go b/wallet/key.go
@@ -23,11 +23,11 @@ import (
 	"fmt"
 	"math/big"
 
-	"github.com/ethereum/go-ethereum/common"
-	"github.com/ethereum/go-ethereum/crypto"
 	"github.com/ethereum/go-ethereum/accounts"
 	"github.com/ethereum/go-ethereum/accounts/keystore"
+	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/ethereum/go-ethereum/crypto"
 )
 
 type KeyProvider struct {
@@ -66,3 +66,11 @@ func (p *KeyProvider) Save(path, passphrase string) (error) {
 	_, err := ks.ImportECDSA(p.priv, passphrase)
 	return err
 }
+
+func (p *KeyProvider) SignHash(account accounts.Account, hash []byte) (sig []byte, err error) {
+	if !bytes.Equal(account.Address.Bytes(), p.Address.Bytes()) {
+		return nil, fmt.Errorf("unexpected account unmatch %s, %s", account.Address.String(), p.Address.String())
+	}
+	return crypto.Sign(hash, p.priv)
+}
+
diff --git a/wallet/wallet.go b/wallet/wallet.go
@@ -83,6 +83,7 @@ type Provider interface {
 	SignTx(account accounts.Account, tx *types.Transaction, chainID *big.Int) (*types.Transaction, error)
 	Init(accounts.Account) error
 	Accounts() []accounts.Account
+    SignHash(accounts.Account,[]byte) ([]byte, error)
 }
 
 func New(config *Config, sdk *eth.SDK) *Wallet {