45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package database
|
|
|
|
import (
|
|
"fmt"
|
|
"sync/atomic"
|
|
|
|
"github.com/crowdsecurity/crowdsec/pkg/types"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
//we simply append the event to the transaction
|
|
func (c *Context) WriteBanApplication(ban types.BanApplication) error {
|
|
atomic.AddInt32(&c.count, 1)
|
|
|
|
c.lock.Lock()
|
|
defer c.lock.Unlock()
|
|
log.Debugf("Ban application being called : %s %s", ban.Scenario, ban.IpText)
|
|
ret := c.Db.Where(types.BanApplication{IpText: ban.IpText}).Assign(types.BanApplication{Until: ban.Until}).Assign(types.BanApplication{Reason: ban.Reason}).Assign(types.BanApplication{MeasureType: ban.MeasureType}).FirstOrCreate(&ban)
|
|
if ret.Error != nil {
|
|
return fmt.Errorf("failed to write ban record : %v", ret.Error)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Context) WriteSignal(sig types.SignalOccurence) error {
|
|
atomic.AddInt32(&c.count, 1)
|
|
c.lock.Lock()
|
|
defer c.lock.Unlock()
|
|
/*let's ensure we only have one ban active for a given scope*/
|
|
for _, ba := range sig.BanApplications {
|
|
ret := c.Db.Where("ip_text = ?", ba.IpText).Delete(types.BanApplication{})
|
|
if ret.Error != nil {
|
|
log.Errorf("While delete overlaping bans : %s", ret.Error)
|
|
return fmt.Errorf("failed to write signal occurrence : %v", ret.Error)
|
|
}
|
|
}
|
|
/*and add the new one(s)*/
|
|
ret := c.Db.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
|
|
}
|