fix

ten-protocol · Mar 15, 2024 · 18ed245 · 18ed245
1 parent ebfed3e
commit 18ed245
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 13 deletions.
diff --git a/go/enclave/vkhandler/vk_handler_test.go b/go/enclave/vkhandler/vk_handler_test.go
@@ -45,18 +45,6 @@ type MessageWithSignatureType struct {
 	SignatureType viewingkey.SignatureType
 }
 
-func TestAttack(t *testing.T) {
-	userPrivKey, _, userID, userAddress := generateRandomUserKeys()
-	fmt.Println("userAddress: ", userAddress.Hex())
-
-	PersonalSignMessageHash := accounts.TextHash([]byte(viewingkey.GeneratePersonalSignMessage(userID, chainID, viewingkey.PersonalSignMessageSupportedVersions[0])))
-	signature, _ := crypto.Sign(PersonalSignMessageHash, userPrivKey)
-	addr1, err := viewingkey.CheckSignature(userID, signature, chainID, viewingkey.PersonalSign)
-	fmt.Println("addr1: ", addr1.Hex(), "err: ", err)
-	addr2, err := viewingkey.CheckSignature(userID, signature, chainID, viewingkey.EIP712Signature)
-	fmt.Println("addr2: ", addr2.Hex(), "err: ", err)
-}
-
 func TestCheckSignature(t *testing.T) {
 	userPrivKey, _, userID, userAddress := generateRandomUserKeys()
 

diff --git a/tools/walletextension/common/common.go b/tools/walletextension/common/common.go
@@ -59,7 +59,7 @@ func CreateEncClient(
 		PrivateKey:              privateKey,
 		PublicKey:               PrivateKeyToCompressedPubKey(privateKey),
 		SignatureWithAccountKey: signature,
-		SignatureType:           viewingkey.SignatureType(signatureType),
+		SignatureType:           signatureType,
 	}
 	encClient, err := rpc.NewEncNetworkClient(hostRPCBindAddr, vk, logger)
 	if err != nil {