summaryrefslogtreecommitdiffhomepage
path: root/src/main.go
blob: 74e7ec97c93f893345f4c0783a5fac546f642cdc (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
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)
	log.Println(tun, err)
	if err != nil {
		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()

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