summaryrefslogtreecommitdiff
path: root/main.go
blob: 904f436dbf011f14728dfcb108ef3ecd64c21502 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
package main

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

	"linum/internal/server"
)

func main() {
	addr := flag.String("addr", ":5353", "listen address")
	upstream := flag.String("upstream", "1.1.1.1:53", "upstream dns server")
	blockfile := flag.String("blocklist", "blocklist.txt", "path to blocklist file")
	flag.Parse()

	srv, err := server.New(server.Config{
		Addr:      *addr,
		Upstream:  *upstream,
		BlockFile: *blockfile,
	})
	if err != nil {
		log.Fatalf("create server: %v", err)
	}

	go func() {
		sig := make(chan os.Signal, 1)
		signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
		<-sig
		log.Println("shutting down")
		os.Exit(0)
	}()

	log.Fatal(srv.ListenAndServe())
}