add SignMessage test

AAStarCommunity · Jul 20, 2024 · cced50f · cced50f
1 parent bf73032
commit cced50f
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/internal/common_util/crypto.go b/internal/common_util/crypto.go
@@ -5,6 +5,7 @@ import (
 	"encoding/hex"
 	"github.com/ethereum/go-ethereum/accounts"
 	"github.com/ethereum/go-ethereum/crypto"
+	"github.com/ethereum/go-ethereum/common/hexutil"
 )
 
 func EthereumSignHexStr(msg string, privateKey *ecdsa.PrivateKey) (string, error) {
@@ -18,6 +19,20 @@ func EthereumSignHexStr(msg string, privateKey *ecdsa.PrivateKey) (string, error
 		return EncodeToHexStringWithPrefix(hash), nil
 	}
 }
+
+func SignMessage(message string, privateKey *ecdsa.PrivateKey) (string, error) {
+	messageHash := accounts.TextHash([]byte(message))
+
+	signature, err := crypto.Sign(messageHash, privateKey)
+	if err != nil {
+		return "", err
+	}
+
+	signature[crypto.RecoveryIDOffset] += 27
+
+	return hexutil.Encode(signature), nil
+}
+
 func DecodeStringWithPrefix(data string) ([]byte, error) {
 	if data[:2] == "0x" {
 		data = data[2:]

diff --git a/plugins/passkey_relay_party/tx_signature.go b/plugins/passkey_relay_party/tx_signature.go
@@ -89,7 +89,7 @@ func (relay *RelayParty) finishTxSignature(ctx *gin.Context) {
 		response.GetResponse().FailCode(ctx, 403, "SignIn failed: "+err.Error())
 		return
 	}
-	signHexStr, err := common_util.EthereumSignHexStr(signPayment.TxData, privateKey)
+	signHexStr, err := common_util.SignMessage(signPayment.TxData, privateKey)
 	txSigRlt := seedworks.TxSignatureResult{
 		Code:    200,
 		TxData:  signPayment.TxData,