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

new: implemented GET /api/events route (ref #5).

Browse source
This commit is contained in:
evilsocket 2018-01-08 05:39:34 +01:00
commit 269d7d845b
14 changed files with 172 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

13
session/targets.go
View file

@ -14,14 +14,16 @@ import (
var log = logging.MustGetLogger("mitm")
type Targets struct {
Session *Session `json:"-"`
Interface *net.Endpoint
Gateway *net.Endpoint
Targets map[string]*net.Endpoint
lock sync.Mutex
}
func NewTargets(iface, gateway *net.Endpoint) *Targets {
func NewTargets(s *Session, iface, gateway *net.Endpoint) *Targets {
return &Targets{
Session: s,
Interface: iface,
Gateway: gateway,
Targets: make(map[string]*net.Endpoint),
@ -33,7 +35,7 @@ func (tp *Targets) Remove(ip, mac string) {
defer tp.lock.Unlock()
if e, found := tp.Targets[mac]; found {
log.Infof("[%slost%s] %s\n", core.RED, core.RESET, e)
tp.Session.Events.Add("target-lost", e)
delete(tp.Targets, mac)
return
}
@ -69,9 +71,14 @@ func (tp *Targets) AddIfNotExist(ip, mac string) *net.Endpoint {
}
e := net.NewEndpoint(ip, mac)
log.Infof("[%snew%s] %s\n", core.GREEN, core.RESET, e)
e.ResolvedCallback = func(e *net.Endpoint) {
tp.Session.Events.Add("target.resolved", e)
}
tp.Targets[mac] = e
tp.Session.Events.Add("target.new", e)
return nil
}