fosrl/newt
1
0
Fork 0
mirror of https://github.com/fosrl/newt.git synced 2025-05-16 23:20:37 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adjust ws types

Browse source
This commit is contained in:
Owen 2025-02-21 12:44:52 -05:00
parent 45a1ab91d7
commit 56e75902e3
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD
2 changed files with 7 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
main.go
View file

@ -55,7 +55,7 @@ func fixKey(key string) string {
// Decode from base64 // Decode from base64
decoded, err := base64.StdEncoding.DecodeString(key) decoded, err := base64.StdEncoding.DecodeString(key)
if err != nil { if err != nil {
logger.Fatal("Error decoding base64:", err) logger.Fatal("Error decoding base64")
} }
// Convert to hex // Convert to hex

12
wg/wg.go
View file

@ -18,7 +18,6 @@ import (
var ( var (
interfaceName string interfaceName string
listenAddr string
mtuInt int mtuInt int
lastReadings = make(map[string]PeerReading) lastReadings = make(map[string]PeerReading)
mu sync.Mutex mu sync.Mutex
@ -61,7 +60,6 @@ type WireGuardService struct {
key wgtypes.Key key wgtypes.Key
reachableAt string reachableAt string
lastReadings map[string]PeerReading lastReadings map[string]PeerReading
mu sync.Mutex
} }
func NewWireGuardService(interfaceName string, mtu int, reachableAt string, generateAndSaveKeyTo string, wsClient *websocket.Client) (*WireGuardService, error) { func NewWireGuardService(interfaceName string, mtu int, reachableAt string, generateAndSaveKeyTo string, wsClient *websocket.Client) (*WireGuardService, error) {
@ -70,7 +68,7 @@ func NewWireGuardService(interfaceName string, mtu int, reachableAt string, gene
return nil, fmt.Errorf("failed to create WireGuard client: %v", err) return nil, fmt.Errorf("failed to create WireGuard client: %v", err)
} }
key := wgtypes.Key{} var key wgtypes.Key
// if generateAndSaveKeyTo is provided, generate a private key and save it to the file. if the file already exists, load the key from the file // if generateAndSaveKeyTo is provided, generate a private key and save it to the file. if the file already exists, load the key from the file
if _, err := os.Stat(generateAndSaveKeyTo); os.IsNotExist(err) { if _, err := os.Stat(generateAndSaveKeyTo); os.IsNotExist(err) {
// generate a new private key // generate a new private key
@ -105,9 +103,9 @@ func NewWireGuardService(interfaceName string, mtu int, reachableAt string, gene
} }
// Register websocket handlers // Register websocket handlers
wsClient.RegisterHandler("wg/config/receive", service.handleConfig) wsClient.RegisterHandler("newt/wg/receive-config", service.handleConfig)
wsClient.RegisterHandler("wg/peer/add", service.handleAddPeer) wsClient.RegisterHandler("newt/wg/peer/add", service.handleAddPeer)
wsClient.RegisterHandler("wg/peer/remove", service.handleRemovePeer) wsClient.RegisterHandler("newt/wg/peer/remove", service.handleRemovePeer)
// Register connect handler to initiate configuration // Register connect handler to initiate configuration
wsClient.OnConnect(service.loadRemoteConfig) wsClient.OnConnect(service.loadRemoteConfig)
@ -125,7 +123,7 @@ func (s *WireGuardService) loadRemoteConfig() error {
go s.periodicBandwidthCheck() go s.periodicBandwidthCheck()
err := s.client.SendMessage("wg/config/get", body) err := s.client.SendMessage("newt/wg/get-config", body)
if err != nil { if err != nil {
return fmt.Errorf("failed to send config request: %v", err) return fmt.Errorf("failed to send config request: %v", err)
} }