69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"fmt"
|
||
|
|
"github.com/nmiculinic/wg-quick-go"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
"io/ioutil"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
func printHelp() {
|
||
|
|
fmt.Println("wg-quick [-iface=wg0] [ up | down ] config_file")
|
||
|
|
os.Exit(1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
flag.String("iface", "", "interface")
|
||
|
|
flag.Parse()
|
||
|
|
args := flag.Args()
|
||
|
|
if len(args) != 2 {
|
||
|
|
printHelp()
|
||
|
|
}
|
||
|
|
|
||
|
|
iface := flag.Lookup("iface").Value.String()
|
||
|
|
log := logrus.WithField("iface", iface)
|
||
|
|
|
||
|
|
cfg := args[1]
|
||
|
|
|
||
|
|
_, err := os.Stat(cfg)
|
||
|
|
switch {
|
||
|
|
case err == nil:
|
||
|
|
case os.IsNotExist(err):
|
||
|
|
if iface == "" {
|
||
|
|
iface = cfg
|
||
|
|
log = logrus.WithField("iface", iface)
|
||
|
|
}
|
||
|
|
cfg = "/etc/wireguard/" + cfg + ".conf"
|
||
|
|
_, err = os.Stat(cfg)
|
||
|
|
if err != nil {
|
||
|
|
log.WithError(err).Errorln("cannot find config file")
|
||
|
|
printHelp()
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
logrus.WithError(err).Errorln("error while reading config file")
|
||
|
|
printHelp()
|
||
|
|
}
|
||
|
|
|
||
|
|
b, err := ioutil.ReadFile(cfg)
|
||
|
|
if err != nil {
|
||
|
|
logrus.WithError(err).Fatalln("cannot read file")
|
||
|
|
}
|
||
|
|
c := &wgquick.Config{}
|
||
|
|
if err := c.UnmarshalText(b); err != nil {
|
||
|
|
logrus.WithError(err).Fatalln("cannot parse config file")
|
||
|
|
}
|
||
|
|
|
||
|
|
switch args[0] {
|
||
|
|
case "up":
|
||
|
|
if err := wgquick.Up(c, iface, log); err != nil {
|
||
|
|
logrus.WithError(err).Errorln("cannot up interface")
|
||
|
|
}
|
||
|
|
case "down":
|
||
|
|
if err := wgquick.Down(c, iface, log); err != nil {
|
||
|
|
logrus.WithError(err).Errorln("cannot down interface")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|