fix(sync): add IsBannedAddress check in processing connect event (#1347)

pactus-project · Jun 18, 2024 · e796f2d · e796f2d
1 parent 868a948
commit e796f2d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/sync/sync.go b/sync/sync.go
@@ -340,8 +340,16 @@ func (sync *synchronizer) processStreamMessage(msg *network.StreamMessage) {
 func (sync *synchronizer) processConnectEvent(ce *network.ConnectEvent) {
 	sync.logger.Debug("processing connect event", "pid", ce.PeerID)
 
-	sync.peerSet.UpdateStatus(ce.PeerID, status.StatusConnected)
 	sync.peerSet.UpdateAddress(ce.PeerID, ce.RemoteAddress, ce.Direction)
+
+	if sync.firewall.IsBannedAddress(ce.RemoteAddress) {
+		sync.logger.Debug("Peer is blacklisted", "peer_id", ce.PeerID, "remote_address", ce.RemoteAddress)
+		sync.peerSet.UpdateStatus(ce.PeerID, status.StatusBanned)
+
+		return
+	}
+
+	sync.peerSet.UpdateStatus(ce.PeerID, status.StatusConnected)
 }
 
 func (sync *synchronizer) processProtocolsEvent(pe *network.ProtocolsEvents) {

diff --git a/www/grpc/statik/statik.go b/www/grpc/statik/statik.go