main.go

package main

import (
	"flag"
	"fmt"
	"os"
)

func main() {
	var (
		port  = flag.String("port", "9060", "Destination Port")
		addr  = flag.String("address", "localhost", "Destination Address")
		rate  = flag.Int("rate", 16, "Packets per second")
		proto = flag.String("protocol", "HEP", "Possible protocols are HEP,IPFIX")
		trans = flag.String("transport", "TLS", "Possible transports are UDP,TCP,TLS")
	)
	flag.Parse()

	if len(*port) < 1 || len(*addr) < 1 || len(*proto) < 1 || len(*trans) < 1 || *rate < 1 {
		fmt.Println("Invalid flags provided!")
		os.Exit(1)
	}

	hammer, err := NewHammer(*proto, *addr, *port, *trans, *rate)
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
	fmt.Printf("Hammer down %s at %s over %s with %d pps\n", *proto, *addr+":"+*port, *trans, *rate)
	hammer.start()
}