0ct0pu5/crowdsec
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

dedup bans

Browse source
This commit is contained in:
Thibault bui Koechlin 2020-07-20 17:42:29 +02:00
parent 66161bc8ae
commit edc24c8410
1 changed files with 14 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
pkg/database/write.go
View file

@ -26,12 +26,20 @@ func (c *Context) WriteSignal(sig types.SignalOccurence) error {
atomic.AddInt32(&c.count, 1) atomic.AddInt32(&c.count, 1)
c.lock.Lock() c.lock.Lock()
defer c.lock.Unlock() defer c.lock.Unlock()
//log.Debugf("Ban signal being called : %s %s", sig.Scenario, sig.Source.Ip.String()) /*let's ensure we only have one ban active for a given scope*/
ret := c.tx.Create(&sig) for _, ba := range sig.BanApplications {
//sig.Scenario = sig.Scenario ret := c.tx.Where("ip_text = ?", ba.IpText).Delete(types.BanApplication{})
if ret.Error != nil { if ret.Error != nil {
log.Errorf("FAILED : %+v \n", ret.Error) log.Errorf("While delete overlaping bans : %s", ret.Error)
return fmt.Errorf("failed to write signal occurrence : %v", ret.Error) return fmt.Errorf("failed to write signal occurrence : %v", ret.Error)
}
} }
/*and add the new one(s)*/
ret := c.tx.Create(&sig)
if ret.Error != nil {
log.Errorf("While creating new bans : %s", ret.Error)
return fmt.Errorf("failed to write signal occurrence : %s", ret.Error)
}
return nil return nil
} }