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

Exit syslog acquis only after server is dead ()

Browse source
This commit is contained in:
Shivam Sandbhor 2022-03-01 16:02:28 +05:30 • committed by GitHub
parent a300d5818f
commit c3dbe0080c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions
pkg/acquisition/modules/syslog
internal
syslogserver.go
syslog.go

3
pkg/acquisition/modules/syslog/internal/syslogserver.go
View file

@ -61,7 +61,7 @@ func (s *SyslogServer) StartServer() *tomb.Tomb {
for {
select {
case <-t.Dying():
s.Logger.Info("syslog server tomb is dying")
s.Logger.Info("Syslog server tomb is dying")
err := s.KillServer()
return err
default:
@ -92,5 +92,6 @@ func (s *SyslogServer) KillServer() error {
if err != nil {
return errors.Wrap(err, "could not close UDP connection")
}
close(s.channel)
return nil
}

12
pkg/acquisition/modules/syslog/syslog.go
View file

@ -180,15 +180,15 @@ func (s *SyslogSource) buildLogFromSyslog(ts *time.Time, hostname *string,
}
func (s *SyslogSource) handleSyslogMsg(out chan types.Event, t *tomb.Tomb, c chan syslogserver.SyslogMessage) error {
killed := false
for {
select {
case <-t.Dying():
s.logger.Info("Syslog datasource is dying")
s.serverTomb.Kill(nil)
return s.serverTomb.Wait()
case <-s.serverTomb.Dying():
s.logger.Info("Syslog server is dying, exiting")
return nil
if !killed {
s.logger.Info("Syslog datasource is dying")
s.serverTomb.Kill(nil)
killed = true
}
case <-s.serverTomb.Dead():
s.logger.Info("Syslog server has exited")
return nil