mirror of
https://github.com/fosrl/newt.git
synced 2026-02-08 05:56:40 +00:00
119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/fosrl/newt/clients"
|
|
wgnetstack "github.com/fosrl/newt/clients"
|
|
"github.com/fosrl/newt/logger"
|
|
"github.com/fosrl/newt/netstack2"
|
|
"github.com/fosrl/newt/websocket"
|
|
"golang.zx2c4.com/wireguard/tun/netstack"
|
|
|
|
"github.com/fosrl/newt/wgtester"
|
|
)
|
|
|
|
var wgService *clients.WireGuardService
|
|
var wgTesterServer *wgtester.Server
|
|
var ready bool
|
|
|
|
func setupClients(client *websocket.Client) {
|
|
var host = endpoint
|
|
if strings.HasPrefix(host, "http://") {
|
|
host = strings.TrimPrefix(host, "http://")
|
|
} else if strings.HasPrefix(host, "https://") {
|
|
host = strings.TrimPrefix(host, "https://")
|
|
}
|
|
|
|
host = strings.TrimSuffix(host, "/")
|
|
|
|
logger.Info("Setting up clients with netstack2...")
|
|
// Create WireGuard service
|
|
wgService, err = wgnetstack.NewWireGuardService(interfaceName, mtuInt, host, id, client, dns, useNativeInterface)
|
|
if err != nil {
|
|
logger.Fatal("Failed to create WireGuard service: %v", err)
|
|
}
|
|
|
|
// // Set up callback to restart wgtester with netstack when WireGuard is ready
|
|
wgService.SetOnNetstackReady(func(tnet *netstack2.Net) {
|
|
|
|
wgTesterServer = wgtester.NewServerWithNetstack("0.0.0.0", wgService.Port, id, tnet) // TODO: maybe make this the same ip of the wg server?
|
|
err := wgTesterServer.Start()
|
|
if err != nil {
|
|
logger.Error("Failed to start WireGuard tester server: %v", err)
|
|
}
|
|
})
|
|
|
|
wgService.SetOnNetstackClose(func() {
|
|
if wgTesterServer != nil {
|
|
wgTesterServer.Stop()
|
|
wgTesterServer = nil
|
|
}
|
|
})
|
|
|
|
client.OnTokenUpdate(func(token string) {
|
|
wgService.SetToken(token)
|
|
})
|
|
|
|
ready = true
|
|
}
|
|
|
|
func setDownstreamTNetstack(tnet *netstack.Net) {
|
|
if wgService != nil {
|
|
wgService.SetOthertnet(tnet)
|
|
}
|
|
}
|
|
|
|
func closeClients() {
|
|
logger.Info("Closing clients...")
|
|
if wgService != nil {
|
|
wgService.Close()
|
|
wgService = nil
|
|
}
|
|
|
|
if wgTesterServer != nil {
|
|
wgTesterServer.Stop()
|
|
wgTesterServer = nil
|
|
}
|
|
}
|
|
|
|
func clientsHandleNewtConnection(publicKey string, endpoint string) {
|
|
if !ready {
|
|
return
|
|
}
|
|
|
|
// split off the port from the endpoint
|
|
parts := strings.Split(endpoint, ":")
|
|
if len(parts) < 2 {
|
|
logger.Error("Invalid endpoint format: %s", endpoint)
|
|
return
|
|
}
|
|
endpoint = strings.Join(parts[:len(parts)-1], ":")
|
|
|
|
if wgService != nil {
|
|
wgService.StartHolepunch(publicKey, endpoint)
|
|
}
|
|
}
|
|
|
|
func clientsOnConnect() {
|
|
if !ready {
|
|
return
|
|
}
|
|
if wgService != nil {
|
|
wgService.LoadRemoteConfig()
|
|
}
|
|
}
|
|
|
|
// clientsStartDirectRelay starts a direct UDP relay from the main tunnel netstack
|
|
// to the clients' WireGuard, bypassing the proxy for better performance.
|
|
func clientsStartDirectRelay(tunnelIP string) {
|
|
if !ready {
|
|
return
|
|
}
|
|
if wgService != nil {
|
|
if err := wgService.StartDirectUDPRelay(tunnelIP); err != nil {
|
|
logger.Error("Failed to start direct UDP relay: %v", err)
|
|
}
|
|
}
|
|
}
|