func main() {
listen = flag.String("listen", "127.0.0.1:4443", "Listen on this address.")
quite = flag.Bool("quite", false, "Quite mode.")
statsonly = flag.Bool(
"statsonly", false, "Only prints stats on stdout and discard data.",
)
flag.Parse()
conn, err := net.ListenPacket("udp4", *listen)
if err != nil {
fmt.Println(err)
return
}
if !*quite {
if *statsonly {
fmt.Printf(
"UDP: Server started on %s in statsonly mode.\n", *listen,
)
} else {
fmt.Printf("UDP: Server started on %s.\n", *listen)
}
}
obytes := make([]byte, 64*1024)
start := time.Now()
var bcount, count time.Duration
for {
bytes := obytes
n, _, err := conn.ReadFrom(bytes)
if err != nil {
fmt.Println(err)
continue
}
if *statsonly {
count += 1
bcount += time.Duration(n)
now := time.Now()
diff := now.Sub(start)
if diff > 1e9 {
fmt.Printf(
"bps = %sps, pps = %d.\n",
gutils.FormatBytes(float64(bcount)), count,
)
start = now
bcount, count = 0, 0
}
} else {
bytes = bytes[:n]
fmt.Printf(string(bytes))
}
}
}
func UDPReader(conn net.PacketConn) {
obytes := make([]byte, 64*1024)
start := time.Now()
var bcount, count time.Duration
fmt.Printf("UDP:%s -> REDIS:%s@%s.\n", *listen, key, *redis)
for {
bytes := obytes
n, _, err := conn.ReadFrom(bytes)
if err != nil {
fmt.Println(err)
continue
}
count += 1
bcount += time.Duration(n)
now := time.Now()
diff := now.Sub(start)
if diff > 1e9 {
fmt.Printf(
"bps = %s, pps = %d.\n", gutils.FormatBytes(float64(bcount)),
count,
)
start = now
bcount, count = 0, 0
}
}
}
func main() {
server = flag.String("server", "localhost:4443", "Server to flood.")
file = flag.String("file", "/dev/stdin", "Content to send to server.")
rate = flag.Int("rate", 0, "Max bytes per second to attempt.")
max = flag.Int("max", -1, "Number of packets to send.")
nuance = flag.Int("nuance", 180000, "Nuance.")
flag.Parse()
bytes, err := ioutil.ReadFile(*file)
if err != nil {
fmt.Println(err)
return
}
conn, err := net.DialTimeout("udp", *server, 1e9)
if err != nil {
fmt.Printf("Error establishing connection to host: %s\n", err)
return
}
start := time.Now()
start0 := time.Now()
var sleep, bcount, count time.Duration
if *rate > 0 {
sleep = time.Duration(1e9 / *rate - *nuance)
}
fmt.Printf("Starting to flood %s.\n", *server)
for {
n, err := conn.Write(bytes)
if err != nil {
fmt.Println("Error sending data:", err)
}
if *rate > 0 {
now := time.Now()
diff := now.Sub(start0)
// fmt.Printf("diff=%d\n", diff)
if diff < sleep {
<-time.After(sleep - diff)
}
start0 = time.Now()
}
count += 1
bcount += time.Duration(n)
now := time.Now()
diff := now.Sub(start)
if diff > 1e9 {
fmt.Printf(
"bps = %sps, pps = %d.\n",
gutils.FormatBytes(float64(bcount)), count,
)
start = now
bcount, count = 0, 0
}
if *max > -1 {
*max -= 1
if *max == 0 {
return
}
}
}
}