summaryrefslogtreecommitdiffhomepage
path: root/src/main.go
blob: 4bece166618144ab7bd1579cbb59ab110f8be2fe (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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package main

import (
	"log"
	"os"
	"runtime"
)

func main() {

	// parse arguments

	var foreground bool
	var interfaceName string
	if len(os.Args) < 2 || len(os.Args) > 3 {
		return
	}

	switch os.Args[1] {

	case "-f", "--foreground":
		foreground = true
		if len(os.Args) != 3 {
			return
		}
		interfaceName = os.Args[2]

	default:
		foreground = false
		if len(os.Args) != 2 {
			return
		}
		interfaceName = os.Args[1]
	}

	// daemonize the process

	if !foreground {
		err := Daemonize()
		if err != nil {
			log.Println("Failed to daemonize:", err)
		}
		return
	}

	// increase number of go workers (for Go <1.5)

	runtime.GOMAXPROCS(runtime.NumCPU())

	// open TUN device

	tun, err := CreateTUN(interfaceName)
	if err != nil {
		log.Println("Failed to create tun device:", err)
		return
	}

	// create wireguard device

	device := NewDevice(tun, LogLevelDebug)

	logInfo := device.log.Info
	logError := device.log.Error
	logInfo.Println("Starting device")

	// start configuration lister

	uapi, err := NewUAPIListener(interfaceName)
	if err != nil {
		logError.Fatal("UAPI listen error:", err)
	}
	defer uapi.Close()

	go func() {
		for {
			conn, err := uapi.Accept()
			if err != nil {
				logError.Fatal("UAPI accept error:", err)
			}
			go ipcHandle(device, conn)
		}
	}()

	device.Wait()
}