pangolin_mirror/newt
1
0
Fork 0
mirror of https://github.com/fosrl/newt.git synced 2026-03-09 12:16:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'holepunch' into hp-multi-client

Browse Source
This commit is contained in:
Owen
2025-04-01 10:40:38 -04:00
parent 65dc81ca8b c5978d9c4e
commit e47ddaa916
2 changed files with 45 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
wg/wg.go
View File

@@ -176,6 +176,10 @@ func NewWireGuardService(interfaceName string, mtu int, generateAndSaveKeyTo str
func (s *WireGuardService) Close() {
s.wgClient.Close()
// Remove the WireGuard interface
if err := s.removeInterface(); err != nil {
logger.Error("Failed to remove WireGuard interface: %v", err)
}
}
func (s *WireGuardService) SetServerPubKey(serverPubKey string) {
@@ -188,8 +192,16 @@ func (s *WireGuardService) SetToken(token string) {
func (s *WireGuardService) LoadRemoteConfig() error {
err := s.client.SendMessage("newt/wg/get-config", map[string]interface{}{
// get the exising wireguard port
device, err := s.wgClient.Device(s.interfaceName)
if err == nil {
s.Port = uint16(device.ListenPort)
logger.Info("WireGuard interface %s already exists with port %d\n", s.interfaceName, s.Port)
}
err = s.client.SendMessage("newt/wg/get-config", map[string]interface{}{
"publicKey": fmt.Sprintf("%s", s.key.PublicKey().String()),
"port": s.Port,
})
if err != nil {
logger.Error("Failed to send registration message: %v", err)
@@ -885,3 +897,20 @@ func (s *WireGuardService) keepSendingUDPHolePunch(host string) {
}
}
}
func (s *WireGuardService) removeInterface() error {
// Remove the WireGuard interface
link, err := netlink.LinkByName(s.interfaceName)
if err != nil {
return fmt.Errorf("failed to get interface: %v", err)
}
err = netlink.LinkDel(link)
if err != nil {
return fmt.Errorf("failed to delete interface: %v", err)
}
logger.Info("WireGuard interface %s removed successfully", s.interfaceName)
return nil
}