mirror of
https://github.com/bettercap/bettercap
synced 2025-08-14 02:36:57 -07:00
new: new net.fuzz network fuzzer module
This commit is contained in:
parent
26ebef7d3f
commit
5504e423fe
2 changed files with 160 additions and 0 deletions
|
@ -15,6 +15,12 @@ type Sniffer struct {
|
|||
Stats *SnifferStats
|
||||
Ctx *SnifferContext
|
||||
pktSourceChan chan gopacket.Packet
|
||||
|
||||
fuzzActive bool
|
||||
fuzzSilent bool
|
||||
fuzzLayers []string
|
||||
fuzzRate float64
|
||||
fuzzRatio float64
|
||||
}
|
||||
|
||||
func NewSniffer(s *session.Session) *Sniffer {
|
||||
|
@ -75,6 +81,35 @@ func NewSniffer(s *session.Session) *Sniffer {
|
|||
return sniff.Stop()
|
||||
}))
|
||||
|
||||
sniff.AddHandler(session.NewModuleHandler("net.fuzz on", "",
|
||||
"Enable fuzzing for every sniffed packet containing the sapecified layers.",
|
||||
func(args []string) error {
|
||||
return sniff.StartFuzzing()
|
||||
}))
|
||||
|
||||
sniff.AddHandler(session.NewModuleHandler("net.fuzz off", "",
|
||||
"Disable fuzzing",
|
||||
func(args []string) error {
|
||||
return sniff.StopFuzzing()
|
||||
}))
|
||||
|
||||
sniff.AddParam(session.NewStringParameter("net.fuzz.layers",
|
||||
"Payload",
|
||||
"",
|
||||
"Types of layer to fuzz."))
|
||||
|
||||
sniff.AddParam(session.NewDecimalParameter("net.fuzz.rate",
|
||||
"1.0",
|
||||
"Rate in the [0.0,1.0] interval of packets to fuzz."))
|
||||
|
||||
sniff.AddParam(session.NewDecimalParameter("net.fuzz.ratio",
|
||||
"0.4",
|
||||
"Rate in the [0.0,1.0] interval of bytes to fuzz for each packet."))
|
||||
|
||||
sniff.AddParam(session.NewBoolParameter("net.fuzz.silent",
|
||||
"false",
|
||||
"If true it will not report fuzzed packets."))
|
||||
|
||||
return sniff
|
||||
}
|
||||
|
||||
|
@ -149,6 +184,10 @@ func (s *Sniffer) Start() error {
|
|||
s.Stats.NumLocal++
|
||||
}
|
||||
|
||||
if s.fuzzActive {
|
||||
s.doFuzzing(packet)
|
||||
}
|
||||
|
||||
if s.Ctx.DumpLocal || !isLocal {
|
||||
data := packet.Data()
|
||||
if s.Ctx.Compiled == nil || s.Ctx.Compiled.Match(data) {
|
||||
|
|
121
modules/net_sniff_fuzz.go
Normal file
121
modules/net_sniff_fuzz.go
Normal file
|
@ -0,0 +1,121 @@
|
|||
package modules
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"strings"
|
||||
|
||||
"github.com/google/gopacket"
|
||||
|
||||
"github.com/bettercap/bettercap/log"
|
||||
|
||||
"github.com/evilsocket/islazy/str"
|
||||
"github.com/evilsocket/islazy/tui"
|
||||
)
|
||||
|
||||
var mutators = []func(byte) byte{
|
||||
func(b byte) byte {
|
||||
return byte(rand.Intn(256) & 0xff)
|
||||
},
|
||||
func(b byte) byte {
|
||||
return byte(b << uint(rand.Intn(5)))
|
||||
},
|
||||
func(b byte) byte {
|
||||
return byte(b >> uint(rand.Intn(5)))
|
||||
},
|
||||
}
|
||||
|
||||
func (s *Sniffer) fuzz(data []byte) int {
|
||||
changes := 0
|
||||
for off, b := range data {
|
||||
if rand.Float64() > s.fuzzRatio {
|
||||
continue
|
||||
}
|
||||
|
||||
data[off] = mutators[rand.Intn(len(mutators))](b)
|
||||
changes++
|
||||
}
|
||||
return changes
|
||||
}
|
||||
|
||||
func (s *Sniffer) doFuzzing(pkt gopacket.Packet) {
|
||||
if rand.Float64() > s.fuzzRate {
|
||||
return
|
||||
}
|
||||
|
||||
layersChanged := 0
|
||||
bytesChanged := 0
|
||||
|
||||
for _, fuzzLayerType := range s.fuzzLayers {
|
||||
for _, layer := range pkt.Layers() {
|
||||
if layer.LayerType().String() == fuzzLayerType {
|
||||
fuzzData := layer.LayerContents()
|
||||
changes := s.fuzz(fuzzData)
|
||||
if changes > 0 {
|
||||
layersChanged++
|
||||
bytesChanged += changes
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if bytesChanged > 0 {
|
||||
if s.fuzzSilent {
|
||||
log.Debug("[%s] changed %d bytes in %d layers.", tui.Green("net.fuzz"), bytesChanged, layersChanged)
|
||||
} else {
|
||||
log.Info("[%s] changed %d bytes in %d layers.", tui.Green("net.fuzz"), bytesChanged, layersChanged)
|
||||
}
|
||||
if err := s.Session.Queue.Send(pkt.Data()); err != nil {
|
||||
log.Error("error sending fuzzed packet: %s", err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Sniffer) configureFuzzing() (err error) {
|
||||
layers := ""
|
||||
|
||||
if err, layers = s.StringParam("net.fuzz.layers"); err != nil {
|
||||
return
|
||||
} else {
|
||||
s.fuzzLayers = str.Comma(layers)
|
||||
}
|
||||
|
||||
if err, s.fuzzRate = s.DecParam("net.fuzz.rate"); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if err, s.fuzzRatio = s.DecParam("net.fuzz.ratio"); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
if err, s.fuzzSilent = s.BoolParam("net.fuzz.silent"); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func (s *Sniffer) StartFuzzing() error {
|
||||
if s.fuzzActive {
|
||||
return nil
|
||||
}
|
||||
|
||||
if err := s.configureFuzzing(); err != nil {
|
||||
return err
|
||||
} else if !s.Running() {
|
||||
if err := s.Start(); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
s.fuzzActive = true
|
||||
|
||||
log.Info("[%s] active on layer types %s (rate:%f ratio:%f)", tui.Green("net.fuzz"), strings.Join(s.fuzzLayers, ","), s.fuzzRate, s.fuzzRatio)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Sniffer) StopFuzzing() error {
|
||||
s.fuzzActive = false
|
||||
return nil
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue