wireguard-go/src/main.go

package main

import (
	"fmt"
	"log"
	"os"
	"os/signal"
	"runtime"
)

func printUsage() {
	fmt.Printf("usage:\n")
	fmt.Printf("%s [-f/--foreground] INTERFACE-NAME\n", os.Args[0])
}

func main() {

	// parse arguments

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

	switch os.Args[1] {

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

	default:
		foreground = false
		if len(os.Args) != 2 {
			printUsage()
			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
	}

	// get log level (default: info)

	logLevel := func() int {
		switch os.Getenv("LOG_LEVEL") {
		case "debug":
			return LogLevelDebug
		case "info":
			return LogLevelInfo
		case "error":
			return LogLevelError
		}
		return LogLevelInfo
	}()

	// create wireguard device

	device := NewDevice(tun, logLevel)

	logInfo := device.log.Info
	logError := device.log.Error
	logDebug := device.log.Debug

	logInfo.Println("Device started")
	logDebug.Println("Debug log enabled")

	// start configuration lister

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

	errs := make(chan error)
	term := make(chan os.Signal)
	wait := device.WaitChannel()

	go func() {
		for {
			conn, err := uapi.Accept()
			if err != nil {
				errs <- err
				return
			}
			go ipcHandle(device, conn)
		}
	}()

	logInfo.Println("UAPI listener started")

	// wait for program to terminate

	signal.Notify(term, os.Kill)
	signal.Notify(term, os.Interrupt)

	select {
	case <-wait:
	case <-term:
	case <-errs:
	}

	// clean up UAPI bind

	uapi.Close()

	logInfo.Println("Closing")
}