SpoofDPI/cmd/spoof-dpi/main.go
2022-01-04 17:38:36 +09:00

61 lines
888 B
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)
}
}