dnsgo.go

package main

import (
	"flag"
	"fmt"
	"os"
	"time"

	"github.com/grt1st/dnsgo/handles"
	"github.com/grt1st/dnsgo/service"
)

const versionNumber = "1.0.1#20180623"

func main() {
	// lookup
	// localhost
	// log
	version := flag.Bool("version", false, "Show program's version number and exit")
	host := flag.String("host", "localhost", "Address to bind")
	query := flag.Bool("query", false, "Whether to send dns request")
	logFilename := flag.String("log", "", "Filename of log file")
	help := flag.Bool("h", false, "Show usage")
	flag.Usage = func() {
		fmt.Fprintf(os.Stderr, "Usage:\n  %s [Options]\n\nOptions\n", os.Args[0])
		flag.PrintDefaults()
	}

	flag.Parse()

	if *help {
		flag.Usage()
		os.Exit(0)
	}
	if *version {
		fmt.Println(versionNumber)
		os.Exit(0)
	}

	service.Init("./conf/default.conf")
	server := &handles.Server{
		Host:     *host,
		Port:     53,
		RTimeout: 5 * time.Second,
		WTimeout: 5 * time.Second,
	}

	server.Run(*query, *logFilename)
}