mirror of
https://github.com/maride/afl-transmit.git
synced 2024-10-18 11:06:34 +00:00
50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"fmt"
|
|
"github.com/maride/afl-transmit/net"
|
|
"github.com/maride/afl-transmit/stats"
|
|
"github.com/maride/afl-transmit/watchdog"
|
|
"log"
|
|
)
|
|
|
|
var (
|
|
outputDirectory string
|
|
)
|
|
|
|
func main() {
|
|
// Register flags
|
|
watchdog.RegisterWatchdogFlags()
|
|
net.RegisterSenderFlags()
|
|
net.RegisterListenFlags()
|
|
RegisterGlobalFlags()
|
|
flag.Parse()
|
|
|
|
// Check if we have the only required argument present - outputDirectory
|
|
if outputDirectory == "" {
|
|
fmt.Println("Please specify fuzzer-directory. See help (--help) for details.")
|
|
return
|
|
}
|
|
|
|
// Read peers file
|
|
net.ReadPeers()
|
|
|
|
// Start watchdog for local afl instances
|
|
go watchdog.WatchFuzzers(outputDirectory)
|
|
|
|
// Start stat printer
|
|
go stats.PrintStats()
|
|
|
|
// Listen for incoming connections
|
|
listenErr := net.Listen(outputDirectory)
|
|
if listenErr != nil {
|
|
log.Println(listenErr)
|
|
}
|
|
}
|
|
|
|
// Registers flags which are required by multiple modules and need to be handled here
|
|
func RegisterGlobalFlags() {
|
|
flag.StringVar(&outputDirectory, "fuzzer-directory", "", "The output directory of the fuzzer(s)")
|
|
}
|