snsd0805/Credit-Card-System-smartcontract
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: add the modifier 'onlySelfOrBank' for SBT

Browse Source
This commit is contained in:
Ting-Jun Wang 2023-06-08 19:29:09 +08:00
parent 7b11748282
commit 7051eafe36
Signed by: snsd0805
GPG Key ID: 8DB0D22BC1217D33
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
contracts/SoulboundToken.sol
View File

@ -72,6 +72,13 @@ contract SoulboundToken is ERC721, Ownable {
require(isReliableBank[msg.sender] == true, "Only bank can access this function"); require(isReliableBank[msg.sender] == true, "Only bank can access this function");
_; _;
} }
event HERE(address);
modifier onlySelfOrBank(address addr) {
emit HERE(addr);
require((msg.sender == owner()) || (msg.sender == addr), "Only the owner can access this function.");
_;
}
modifier onlyRegister { modifier onlyRegister {
require(balanceOf(msg.sender)!=0, "Only Register can access this function"); require(balanceOf(msg.sender)!=0, "Only Register can access this function");
@ -106,7 +113,8 @@ contract SoulboundToken is ERC721, Ownable {
return certificates[client]; return certificates[client];
} }
function getAccountNumber(address client) public view onlyBank returns (uint) { function getAccountNumber(address client) public onlySelfOrBank(client) returns (uint) {
return address_to_number[client]; return address_to_number[client];
} }
} }