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

new: per ip traffic stats

Browse source
This commit is contained in:
evilsocket 2018-01-13 18:56:27 +01:00
parent 656b633023
commit e7da2d4287
1 changed files with 23 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

24
packets/queue.go
View file

@ -19,6 +19,11 @@ type Activity struct {
Source bool
}
type Traffic struct {
Sent uint64
Received uint64
}
type Queue struct {
iface *bnet.Endpoint
handle *pcap.Handle
@ -32,6 +37,7 @@ type Queue struct {
PktReceived uint64
Errors uint64
Protos map[string]uint64
Traffic map[string]*Traffic
}
func NewQueue(iface *bnet.Endpoint) (*Queue, error) {
@ -48,6 +54,7 @@ func NewQueue(iface *bnet.Endpoint) (*Queue, error) {
PktReceived: 0,
Errors: 0,
Protos: make(map[string]uint64),
Traffic: make(map[string]*Traffic),
Activities: make(chan Activity),
}
@ -67,9 +74,10 @@ func (q *Queue) worker() {
if q.active == false {
return
}
pktSize := uint64(len(pkt.Data()))
q.PktReceived++
q.Received += uint64(len(pkt.Data()))
q.Received += pktSize
// gather protocols stats
pktLayers := pkt.Layers()
@ -99,6 +107,13 @@ func (q *Queue) worker() {
MAC: eth.SrcMAC,
Source: true,
}
addr := ip4.SrcIP.String()
if _, found := q.Traffic[addr]; found == false {
q.Traffic[addr] = &Traffic{}
}
q.Traffic[addr].Sent += pktSize
}
if bytes.Compare(q.iface.IP, ip4.DstIP) != 0 && q.iface.Net.Contains(ip4.DstIP) {
@ -107,6 +122,13 @@ func (q *Queue) worker() {
MAC: eth.SrcMAC,
Source: false,
}
addr := ip4.DstIP.String()
if _, found := q.Traffic[addr]; found == false {
q.Traffic[addr] = &Traffic{}
}
q.Traffic[addr].Received += pktSize
}
}
}