SpoofDPI/cmd/spoof-dpi/main.go

package main

import (
	"flag"
	"log"
	"os"
	"os/signal"
	"syscall"

	"github.com/xvzc/SpoofDPI/config"
	"github.com/xvzc/SpoofDPI/proxy"
	"github.com/xvzc/SpoofDPI/util"
)

func main() {
	port := flag.String("port", "8080", "port")
	dns := flag.String("dns", "8.8.8.8", "DNS server")
	debug := flag.Bool("debug", false, "true | false")

	flag.Parse()

	err := config.InitConfig(*port, *dns, *debug)
	if err != nil {
		os.Exit(1)
	}

	util.PrintWelcome()

	err = config.SetOsProxy()
	if err != nil {
		log.Fatal(err)
		os.Exit(1)
	}

	go proxy.Start()

	sigs := make(chan os.Signal, 1)
	done := make(chan bool, 1)

	signal.Notify(
		sigs,
		syscall.SIGKILL,
		syscall.SIGINT,
		syscall.SIGTERM,
		syscall.SIGQUIT,
		syscall.SIGCONT,
		syscall.SIGHUP)

	go func() {
		_ = <-sigs
		done <- true
	}()

	<-done
	err = config.UnSetOsProxy()
	if err != nil {
		log.Fatal(err)
		os.Exit(1)
	}
}