github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-16 10:03:39 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: improved control over events.stream verbosity ( ref #70 )

Browse source
This commit is contained in:
evilsocket 2018-02-24 17:04:16 +01:00
parent e6170263a2
commit 9ee43b6b67
3 changed files with 179 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

55
modules/events_stream.go
View file

@ -1,30 +1,27 @@
package modules
import (
"fmt"
"strconv"
"github.com/bettercap/bettercap/core"
"github.com/bettercap/bettercap/log"
"github.com/bettercap/bettercap/session"
)
type EventsStream struct {
session.SessionModule
filter string
quit chan bool
ignoreList *IgnoreList
quit chan bool
}
func NewEventsStream(s *session.Session) *EventsStream {
stream := &EventsStream{
SessionModule: session.NewSessionModule("events.stream", s),
filter: "",
quit: make(chan bool),
ignoreList: NewIgnoreList(),
}
stream.AddParam(session.NewStringParameter("events.stream.filter",
"",
"",
"If filled, filter events by this prefix type."))
stream.AddHandler(session.NewModuleHandler("events.stream on", "",
"Start events stream.",
func(args []string) error {
@ -48,6 +45,34 @@ func NewEventsStream(s *session.Session) *EventsStream {
return stream.Show(limit)
}))
stream.AddHandler(session.NewModuleHandler("events.ignore FILTER", "events.ignore ([^\\s]+)",
"Events with an identifier matching this filter will not be shown (use multiple times to add more filters).",
func(args []string) error {
return stream.ignoreList.Add(args[0])
}))
stream.AddHandler(session.NewModuleHandler("events.include FILTER", "events.include ([^\\s]+)",
"Used to remove filters passed with the events.ignore command.",
func(args []string) error {
return stream.ignoreList.Remove(args[0])
}))
stream.AddHandler(session.NewModuleHandler("events.filters", "",
"Print the list of filters used to ignore events.",
func(args []string) error {
if stream.ignoreList.Empty() {
fmt.Printf("Ignore filters list is empty.\n")
} else {
stream.ignoreList.RLock()
defer stream.ignoreList.RUnlock()
for _, filter := range stream.ignoreList.Filters() {
fmt.Printf(" '%s'\n", string(filter))
}
}
return nil
}))
stream.AddHandler(session.NewModuleHandler("events.clear", "",
"Clear events stream.",
func(args []string) error {
@ -71,24 +96,20 @@ func (s EventsStream) Author() string {
}
func (s *EventsStream) Configure() error {
var err error
if err, s.filter = s.StringParam("events.stream.filter"); err != nil {
return err
}
return nil
}
func (s *EventsStream) Start() error {
if err := s.Configure(); err != nil {
return err
}
return s.SetRunning(true, func() {
for {
var e session.Event
select {
case e = <-s.Session.Events.NewEvents:
s.View(e, true)
if s.ignoreList.Ignored(e) == false {
s.View(e, true)
} else {
log.Debug("Skipping ignored event %v", e)
}
break
case <-s.quit: