From b6e3b693832691baeca1e58998e34ce278feb2db Mon Sep 17 00:00:00 2001
From: parketh <yeungparkhay@gmail.com>
Date: Thu, 12 Dec 2024 08:10:39 +0000
Subject: [PATCH] feat: add opcc slashing event (#139)

## Summary

This is a PR to update the BTC Slasher program to support and handle OP
stack consumer chain slashing events.

It accompanies
https://github.com/babylonlabs-io/babylon-contract/pull/92.

## Test plan

Before running integration tests, I needed to set `DOCKER_HOST`:
```bash
export DOCKER_HOST="unix:///Users/parkyeung/.docker/run/docker.sock"
```

To run tests:

```bash
# unit tests
make test

# integration tests
make test-e2e
```
---
 CHANGELOG.md                             |  1 +
 btcstaking-tracker/btcslasher/slasher.go | 66 +++++++++++++++---------
 2 files changed, 44 insertions(+), 23 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 56a9623..d98cdce 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -43,6 +43,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/)
 
 ### Improvements
 
+* [#139](https://github.com/babylonlabs-io/vigilante/pull/139) add opcc slashing event
 * [#136](https://github.com/babylonlabs-io/vigilante/pull/136) rate limit activations
 * [#141](https://github.com/babylonlabs-io/vigilante/pull/141) decrement tracked delegations in atomic slasher
 * [#143](https://github.com/babylonlabs-io/vigilante/pull/143) adds nlreturn linter rule
diff --git a/btcstaking-tracker/btcslasher/slasher.go b/btcstaking-tracker/btcslasher/slasher.go
index 9780e16..ad1ddf2 100644
--- a/btcstaking-tracker/btcslasher/slasher.go
+++ b/btcstaking-tracker/btcslasher/slasher.go
@@ -21,9 +21,10 @@ import (
 )
 
 const (
-	txSubscriberName  = "tx-subscriber"
-	messageActionName = "/babylon.finality.v1.MsgAddFinalitySig"
-	evidenceEventName = "babylon.finality.v1.EventSlashedFinalityProvider.evidence"
+	txSubscriberName          = "tx-subscriber"
+	messageActionName         = "/babylon.finality.v1.MsgAddFinalitySig"
+	consumerMessageActionName = "/babylon.finality.v1.MsgEquivocationEvidence"
+	evidenceEventName         = "babylon.finality.v1.EventSlashedFinalityProvider.evidence"
 )
 
 type BTCSlasher struct {
@@ -43,6 +44,8 @@ type BTCSlasher struct {
 	// channel for finality signature messages, which might include
 	// equivocation evidences
 	finalitySigChan <-chan coretypes.ResultEvent
+	// channel for consumer fp equivocation evidences
+	equivocationEvidenceChan <-chan coretypes.ResultEvent
 	// channel for SKs of slashed finality providers
 	slashedFPSKChan chan *btcec.PrivateKey
 	// channel for receiving the slash result of each BTC delegation
@@ -133,10 +136,18 @@ func (bs *BTCSlasher) Start() error {
 		// start the subscriber to slashing events
 		// NOTE: at this point monitor has already started the Babylon querier routine
 		queryName := fmt.Sprintf("tm.event = 'Tx' AND message.action='%s'", messageActionName)
+		// subscribe to babylon fp slashing events
 		bs.finalitySigChan, startErr = bs.BBNQuerier.Subscribe(txSubscriberName, queryName)
 		if startErr != nil {
 			return
 		}
+		// subscribe to consumer fp slashing events
+		queryName = fmt.Sprintf("tm.event = 'Tx' AND message.action='%s'", consumerMessageActionName)
+		bs.equivocationEvidenceChan, startErr = bs.BBNQuerier.Subscribe(txSubscriberName, queryName)
+		if startErr != nil {
+			return
+		}
+
 		// BTC slasher has started
 		bs.logger.Debugf("slasher routine has started subscribing %s", queryName)
 
@@ -203,6 +214,32 @@ func (bs *BTCSlasher) slashingEnforcer() {
 	}
 }
 
+func (bs *BTCSlasher) handleEvidence(evt *coretypes.ResultEvent, isConsumer bool) {
+	evidence := filterEvidence(evt)
+
+	if evidence == nil {
+		return
+	}
+
+	fpBTCPKHex := evidence.FpBtcPk.MarshalHex()
+	fpType := "babylon"
+	if isConsumer {
+		fpType = "consumer"
+	}
+	bs.logger.Infof("new equivocating %s finality provider %s to be slashed", fpType, fpBTCPKHex)
+	bs.logger.Debugf("found equivocation evidence of %s finality provider %s: %v", fpType, fpBTCPKHex, evidence)
+
+	// extract the SK of the slashed finality provider
+	fpBTCSK, err := evidence.ExtractBTCSK()
+	if err != nil {
+		bs.logger.Errorf("failed to extract BTC SK of the slashed %s finality provider %s: %v", fpType, fpBTCPKHex, err)
+
+		return
+	}
+
+	bs.slashedFPSKChan <- fpBTCSK
+}
+
 // equivocationTracker is a routine to track the equivocation events on Babylon,
 // extract equivocating finality providers' SKs, and sen to slashing enforcer
 // routine
@@ -219,26 +256,9 @@ func (bs *BTCSlasher) equivocationTracker() {
 
 			return
 		case resultEvent := <-bs.finalitySigChan:
-			evidence := filterEvidence(&resultEvent)
-
-			if evidence == nil {
-				// this event does not contain equivocation evidence, skip
-				continue
-			}
-
-			fpBTCPKHex := evidence.FpBtcPk.MarshalHex()
-			bs.logger.Infof("new equivocating finality provider %s to be slashed", fpBTCPKHex)
-			bs.logger.Debugf("found equivocation evidence of finality provider %s: %v", fpBTCPKHex, evidence)
-
-			// extract the SK of the slashed finality provider
-			fpBTCSK, err := evidence.ExtractBTCSK()
-			if err != nil {
-				bs.logger.Errorf("failed to extract BTC SK of the slashed finality provider %s: %v", fpBTCPKHex, err)
-
-				continue
-			}
-
-			bs.slashedFPSKChan <- fpBTCSK
+			bs.handleEvidence(&resultEvent, false)
+		case resultEvent := <-bs.equivocationEvidenceChan:
+			bs.handleEvidence(&resultEvent, true)
 		}
 	}
 }