ADD: verify sig

parami-foundation · Jan 8, 2024 · 91da5c2 · 91da5c2
1 parent ab81266
commit 91da5c2
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/contracts/gptmining/GptMiner.sol b/contracts/gptmining/GptMiner.sol
@@ -4,6 +4,8 @@ pragma solidity >=0.8.2 <0.9.0;
 import "@openzeppelin/contracts/access/Ownable.sol";
 import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
 import "@openzeppelin/contracts/utils/math/Math.sol";
+import "@openzeppelin/contracts/utils/cryptography/ECDSA.sol";
+import "@openzeppelin/contracts/utils/cryptography/MessageHashUtils.sol";
 import "./GPTPower.sol";
 
 contract GPTMiner is Ownable {
@@ -46,6 +48,15 @@ contract GPTMiner is Ownable {
         return Math.min(block.timestamp, periodFinish);
     }
 
+    function genMessageHash(address minerAddress) public pure returns (bytes32){
+        return keccak256(abi.encodePacked("GPTMiner:", minerAddress));
+    }
+
+    function verify(address minerAddress, bytes memory _signature) public pure returns (address) {
+        bytes32 _msgHash = MessageHashUtils.toEthSignedMessageHash(genMessageHash(minerAddress));
+        return ECDSA.recover(_msgHash, _signature);
+    }
+
     function rewardPerToken() public view returns (uint256) {
         if (totalSupply == 0) {
             return rewardPerTokenStored;