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

and more

Browse source
This commit is contained in:
evilsocket 2018-04-26 12:25:35 +02:00
commit 813ae0a5e9
No known key found for this signature in database
GPG key ID: 1564D7F30393A456
3 changed files with 12 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

17
packets/queue.go
View file

@ -1,7 +1,6 @@
package packets
import (
"bytes"
"fmt"
"net"
"sync"
@ -104,7 +103,7 @@ func (q *Queue) trackProtocols(pkt gopacket.Packet) {
q.Lock()
name := proto.String()
if _, found := q.Protos[name]; found == false {
if _, found := q.Protos[name]; !found {
q.Protos[name] = 1
} else {
q.Protos[name] += 1
@ -126,7 +125,7 @@ func (q *Queue) trackActivity(eth *layers.Ethernet, ip4 *layers.IPv4, address ne
// initialize or update stats
addr := address.String()
if _, found := q.Traffic[addr]; found == false {
if _, found := q.Traffic[addr]; !found {
if isSent {
q.Traffic[addr] = &Traffic{Sent: pktSize}
} else {
@ -165,7 +164,7 @@ func (q *Queue) TrackError() {
func (q *Queue) worker() {
for pkt := range q.srcChannel {
if q.active == false {
if !q.active {
return
}
@ -188,16 +187,16 @@ func (q *Queue) worker() {
// we manage to sniff
// something coming from someone on the LAN
isFromMe := bytes.Compare(q.iface.IP, ip4.SrcIP) == 0
isFromMe := q.iface.IP.Equal(ip4.SrcIP)
isFromLAN := q.iface.Net.Contains(ip4.SrcIP)
if isFromMe == false && isFromLAN {
if !isFromMe && isFromLAN {
q.trackActivity(eth, ip4, ip4.SrcIP, pktSize, true)
}
// something going to someone on the LAN
isToMe := bytes.Compare(q.iface.IP, ip4.DstIP) == 0
isToMe := q.iface.IP.Equal(ip4.DstIP)
isToLAN := q.iface.Net.Contains(ip4.DstIP)
if isToMe == false && isToLAN {
if !isToMe && isToLAN {
q.trackActivity(eth, ip4, ip4.DstIP, pktSize, false)
}
}
@ -208,7 +207,7 @@ func (q *Queue) Send(raw []byte) error {
q.Lock()
defer q.Unlock()
if q.active == false {
if !q.active {
return fmt.Errorf("Packet queue is not active.")
}