chore: make wireguard work over webrtc

2026-05-02 15:16:38 +00:00 · 2021-11-16 14:36:24 +01:00
parent f0a0888046
commit b35dcd21df
8 changed files with 113 additions and 80 deletions
--- a/browser/client/client_js.go
+++ b/browser/client/client_js.go
@@ -0,0 +1,122 @@
+package main
+
+import (
+	"context"
+	"encoding/hex"
+	"fmt"
+	"github.com/wiretrustee/wiretrustee/browser/conn"
+	"github.com/wiretrustee/wiretrustee/signal/client"
+	"golang.zx2c4.com/wireguard/device"
+	"golang.zx2c4.com/wireguard/tun/netstack"
+	"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
+	"log"
+	"net"
+	"syscall/js"
+	"time"
+)
+
+func handleError(err error) {
+	fmt.Printf("Unexpected error. Check console.")
+	panic(err)
+}
+
+func getElementByID(id string) js.Value {
+	return js.Global().Get("document").Call("getElementById", id)
+}
+
+func main() {
+
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Minute)
+	defer cancel()
+
+	connectToSignal := func(key wgtypes.Key, remoteKey wgtypes.Key) {
+		signalClient, err := client.NewWebsocketClient(ctx, "ws://localhost:80/signal", key)
+		if err != nil {
+			return
+		}
+
+		time.Sleep(5 * time.Second)
+
+		tun, _, err := netstack.CreateNetTUN(
+			[]net.IP{net.ParseIP("10.100.0.1")},
+			[]net.IP{net.ParseIP("8.8.8.8")},
+			1420)
+
+		b := conn.NewWebRTCBind("chann-1", signalClient, key.PublicKey().String(), remoteKey.String())
+		dev := device.NewDevice(tun, b, device.NewLogger(device.LogLevelVerbose, ""))
+
+		err = dev.IpcSet(fmt.Sprintf("private_key=%s\npublic_key=%s\npersistent_keepalive_interval=10\nendpoint=webrtc://datachannel\nallowed_ip=0.0.0.0/0",
+			hex.EncodeToString(key[:]),
+			hex.EncodeToString(remoteKey[:]),
+		))
+		log.Println("4")
+
+		if err != nil {
+			log.Panic(err)
+		}
+		err = dev.Up()
+		if err != nil {
+			log.Panic(err)
+		}
+
+		log.Printf("device started")
+
+		/*client := http.Client{
+			Transport: &http.Transport{
+				DialContext: tnet.DialContext,
+			},
+		}
+		time.Sleep(2 * time.Second)
+		req, _ := http.NewRequest("GET", "https://httpbin.org/ip", nil)
+		req.Header.Set("js.fetch:mode", "no-cors")
+		resp, err := client.Do(req)
+		if err != nil {
+			log.Panic(err)
+		}
+		body, err := io.ReadAll(resp.Body)
+		if err != nil {
+			log.Panic(err)
+		}
+		log.Printf(string(body))
+		log.Printf(resp.Status)*/
+
+		select {}
+	}
+
+	js.Global().Set("generateWireguardKey", js.FuncOf(func(this js.Value, args []js.Value) interface{} {
+
+		key, err := wgtypes.GenerateKey()
+		if err != nil {
+			return nil
+		}
+
+		js.Global().Get("document").Call("getElementById", "wgPrivateKey").Set("value", key.String())
+
+		log.Printf("Wireguard Public key %s", key.PublicKey().String())
+		js.Global().Get("document").Call("getElementById", "publicKey").Set("value", key.PublicKey().String())
+
+		return nil
+	}))
+
+	js.Global().Set("connect", js.FuncOf(func(this js.Value, args []js.Value) interface{} {
+
+		wgPrivateKey := js.Global().Get("document").Call("getElementById", "wgPrivateKey").Get("value").String()
+		key, err := wgtypes.ParseKey(wgPrivateKey)
+		if err != nil {
+			return err
+		}
+
+		remotePublicKey := js.Global().Get("document").Call("getElementById", "peerKey").Get("value").String()
+		remoteKey, err := wgtypes.ParseKey(remotePublicKey)
+		if err != nil {
+			return err
+		}
+
+		log.Printf("Remote Wireguard Public key %s", remoteKey.String())
+		log.Printf("Our Wireguard Public key %s", key.PublicKey().String())
+		go connectToSignal(key, remoteKey)
+		return nil
+	}))
+
+	select {}
+}