vacp2p/staking-reward-streamer
1
0
Fork 0
mirror of https://github.com/vacp2p/staking-reward-streamer.git synced 2026-01-08 20:48:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(KarmaRLN): remove userTier, minmalKarma, tierSize, undo changes in Verifier.verifyProof call

Browse Source
This commit is contained in:
Ricardo Guilherme Schmidt
2025-05-28 07:24:24 -03:00
parent 88584bb4b1
commit 67d1211ffc
1 changed files with 10 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
src/rln/KarmaRLN.sol
View File

@@ -2,33 +2,24 @@
pragma solidity 0.8.26;
import "../Karma.sol";
import {IVerifier} from "./IVerifier.sol";
import { IVerifier } from "./IVerifier.sol";
/// @title Rate-Limiting Nullifier registry contract
/// @dev This contract allows you to register RLN commitment and withdraw/slash.
contract KarmaRLN {
/// @dev User metadata struct.
/// @param userAddress: address of depositor;
/// @param userTier: user's message limit (karmaBalance / MINIMAL_KARMA).
struct User {
address userAddress;
uint256 userTier;
uint256 index;
}
/// @dev Minimal membership deposit (stake amount) value - cost of 1 message.
uint256 public immutable MINIMAL_KARMA;
/// @dev Maximal rate.
uint256 public immutable TIER_SIZE;
/// @dev Registry set size (1 << DEPTH).
uint256 public immutable SET_SIZE;
/// @dev Fee percentage.
uint8 public immutable FEE_PERCENTAGE;
/// @dev Current index where identityCommitment will be stored.
uint256 public identityCommitmentIndex;
@@ -44,9 +35,8 @@ contract KarmaRLN {
/// @dev Emmited when a new member registered.
/// @param identityCommitment: `identityCommitment`;
/// @param userTier: user's tier;
/// @param index: idCommitmentIndex value.
event MemberRegistered(uint256 identityCommitment, uint256 userTier, uint256 index);
event MemberRegistered(uint256 identityCommitment, uint256 index);
/// @dev Emmited when a member was withdrawn.
/// @param index: index of `identityCommitment`;
@@ -57,21 +47,10 @@ contract KarmaRLN {
/// @param slasher: address of slasher (msg.sender).
event MemberSlashed(uint256 index, address slasher);
/// @param minimalKarma: minimal membership deposit;
/// @param tierSize: tier sizee;
/// @param depth: depth of the merkle tree;
/// @param _token: address of the ERC20 contract;
/// @param _verifier: address of the Groth16 Verifier.
constructor(
uint256 minimalKarma,
uint256 tierSize,
uint256 depth,
address _token,
address _verifier
) {
MINIMAL_KARMA = minimalKarma;
TIER_SIZE = tierSize;
constructor(uint256 depth, address _verifier, address _token) {
SET_SIZE = 1 << depth;
karma = Karma(_token);
@@ -87,13 +66,10 @@ contract KarmaRLN {
uint256 index = identityCommitmentIndex;
uint256 amount = karma.balanceOf(msg.sender);
require(index < SET_SIZE, "RLN, register: set is full");
require(amount >= MINIMAL_KARMA, "RLN, register: amount is lower than minimal deposit");
require(members[identityCommitment].userAddress == address(0), "RLN, register: idCommitment already registered");
uint256 userTier = amount / TIER_SIZE;
members[identityCommitment] = User(msg.sender, userTier, index);
emit MemberRegistered(identityCommitment, userTier, index);
members[identityCommitment] = User(msg.sender, index);
emit MemberRegistered(identityCommitment, index);
unchecked {
identityCommitmentIndex = index + 1;
@@ -106,7 +82,7 @@ contract KarmaRLN {
function exit(uint256 identityCommitment, uint256[8] calldata proof) external {
User memory member = members[identityCommitment];
require(member.userAddress != address(0), "RLN, withdraw: member doesn't exist");
require(_verifyProof(identityCommitment, member.userAddress, proof), "RLN, withdraw: invalid proof");
require(_verifyProof(identityCommitment, proof), "RLN, withdraw: invalid proof");
delete members[identityCommitment];
emit MemberExited(member.index);
@@ -114,30 +90,25 @@ contract KarmaRLN {
/// @dev Slashes identity with identityCommitment.
/// @param identityCommitment: `identityCommitment`;
/// @param receiver: stake receiver;
/// @param proof: snarkjs's format generated proof (without public inputs) packed consequently.
function slash(uint256 identityCommitment, uint256[8] calldata proof) external {
User memory member = members[identityCommitment];
require(member.userAddress != address(0), "RLN, slash: member doesn't exist");
require(_verifyProof(identityCommitment, proof), "RLN, slash: invalid proof");
karma.slash(member.userAddress);
//karma.slash(member.userAddress);
delete members[identityCommitment];
emit MemberSlashed(member.index, receiver);
emit MemberSlashed(member.index, msg.sender);
}
/// @dev Groth16 proof verification
function _verifyProof(uint256 identityCommitment, uint256[8] calldata proof)
internal
view
returns (bool)
{
function _verifyProof(uint256 identityCommitment, uint256[8] calldata proof) internal view returns (bool) {
return verifier.verifyProof(
[proof[0], proof[1]],
[[proof[2], proof[3]], [proof[4], proof[5]]],
[proof[6], proof[7]],
[identityCommitment]
[identityCommitment, uint256(0)]
);
}
}