github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-08-19 21:13:18 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

new: new limit optional argument for events.show

Browse source
This commit is contained in:
evilsocket 2018-02-08 12:03:40 +01:00
commit 3c2932514a
4 changed files with 37 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

24
modules/events_stream.go
View file

@ -1,6 +1,9 @@
package modules
import (
"strconv"
"github.com/evilsocket/bettercap-ng/core"
"github.com/evilsocket/bettercap-ng/session"
)
@ -34,10 +37,15 @@ func NewEventsStream(s *session.Session) *EventsStream {
return stream.Stop()
}))
stream.AddHandler(session.NewModuleHandler("events.show", "",
stream.AddHandler(session.NewModuleHandler("events.show LIMIT?", "events.show(\\s\\d+)?",
"Show events stream.",
func(args []string) error {
return stream.Show()
limit := -1
if len(args) == 1 {
arg := core.Trim(args[0])
limit, _ = strconv.Atoi(arg)
}
return stream.Show(limit)
}))
stream.AddHandler(session.NewModuleHandler("events.clear", "",
@ -96,8 +104,16 @@ func (s *EventsStream) Start() error {
return nil
}
func (s *EventsStream) Show() error {
for _, e := range s.Session.Events.Sorted() {
func (s *EventsStream) Show(limit int) error {
events := s.Session.Events.Sorted()
num := len(events)
from := 0
if limit > 0 && num > limit {
from = num - limit
}
for _, e := range events[from:num] {
s.View(e, false)
}