完善主程序

2026-06-21 19:13:20 +08:00 · 2021-12-28 21:50:10 +08:00
parent f0956ae742
commit 6ae8e88bd1
18 changed files with 499 additions and 14 deletions
--- a/main.go
+++ b/main.go
@@ -1 +1,146 @@
 package main
+
+import (
+	"flag"
+	"fmt"
+	"os"
+
+	base14 "github.com/fumiama/go-base16384"
+	curve "github.com/fumiama/go-x25519"
+
+	"github.com/fumiama/WireGold/config"
+	"github.com/fumiama/WireGold/gold/link"
+	"github.com/fumiama/WireGold/helper"
+	"github.com/fumiama/WireGold/lower"
+)
+
+const suffix32 = "㴄"
+
+func main() {
+	help := flag.Bool("h", false, "display this help")
+	gen := flag.Bool("g", false, "generate key pair")
+	showp := flag.Bool("p", false, "show my publickey")
+	file := flag.String("c", "config.yaml", "specify conf file")
+	flag.Parse()
+	if *help {
+		displayHelp("")
+	}
+	if *gen {
+		k, err := curve.New(nil)
+		if err != nil {
+			panic(err)
+		}
+		pubk, err := base14.UTF16be2utf8(base14.Encode((*k.Public())[:]))
+		if err != nil {
+			panic(err)
+		}
+		prvk, err := base14.UTF16be2utf8(base14.Encode((*k.Private())[:]))
+		if err != nil {
+			panic(err)
+		}
+		fmt.Println("PublicKey:", helper.BytesToString(pubk[:57]))
+		fmt.Println("PrivateKey:", helper.BytesToString(prvk[:57]))
+		os.Exit(0)
+	}
+	if helper.IsNotExist(*file) {
+		f, err := os.Create(*file)
+		if err != nil {
+			panic(err)
+		}
+		var r string
+		fmt.Print("IP: ")
+		fmt.Scanln(&r)
+		if r == "" {
+			f.Close()
+			os.Remove(*file)
+			fmt.Println("nil ip")
+			return
+		}
+		f.WriteString("IP: " + r + "\n")
+		r = ""
+
+		fmt.Print("SubNet: ")
+		fmt.Scanln(&r)
+		if r == "" {
+			f.Close()
+			os.Remove(*file)
+			fmt.Println("nil subnet")
+			return
+		}
+		f.WriteString("SubNet: " + r + "\n")
+		r = ""
+
+		fmt.Print("PrivateKey: ")
+		fmt.Scanln(&r)
+		if r == "" {
+			f.Close()
+			os.Remove(*file)
+			fmt.Println("nil private key")
+			return
+		}
+		f.WriteString("PrivateKey: " + r + "\n")
+		r = ""
+
+		fmt.Print("EndPoint: ")
+		fmt.Scanln(&r)
+		if r == "" {
+			f.Close()
+			os.Remove(*file)
+			fmt.Println("nil endpoint")
+			return
+		}
+		f.WriteString("EndPoint: " + r + "\n")
+		r = ""
+
+		f.Close()
+	}
+	c := config.Parse(*file)
+	if c.IP == "" {
+		displayHelp("nil ip")
+	}
+	if c.SubNet == "" {
+		displayHelp("nil subnet")
+	}
+	if c.PrivateKey == "" {
+		displayHelp("nil private key")
+	}
+	if c.EndPoint == "" {
+		displayHelp("nil endpoint")
+	}
+	var key [32]byte
+	k, err := base14.UTF82utf16be(helper.StringToBytes(c.PrivateKey + suffix32))
+	if err != nil {
+		panic(err)
+	}
+	n := copy(key[:], base14.Decode(k))
+	if n != 32 {
+		displayHelp("private key length is not 32")
+	}
+
+	if *showp {
+		c := curve.Get(key[:])
+		pubk, err := base14.UTF16be2utf8(base14.Encode((*c.Public())[:]))
+		if err != nil {
+			panic(err)
+		}
+		fmt.Println("PublicKey:", helper.BytesToString(pubk[:57]))
+		os.Exit(0)
+	}
+
+	nic := lower.NewNIC(c.IP, c.SubNet)
+	me := link.NewMe(&key, c.IP+"/32", c.EndPoint, true)
+	nic.Up()
+	defer func() {
+		nic.Stop()
+		nic.Down()
+		nic.Destroy()
+	}()
+
+	nic.Start(&me)
+}
+
+func displayHelp(hint string) {
+	fmt.Println(hint)
+	flag.Usage()
+	os.Exit(0)
+}