mirror of
https://github.com/netbirdio/netbird.git
synced 2026-04-22 18:26:41 +00:00
ea41cf2d2c
This is experimental and needs more test. the purpose of this change is to validate that a TLS connection stuck using old routes because of keepalive settings on the remote webserver are reset once netbird receives a network map
106 lines
2.9 KiB
Go
106 lines
2.9 KiB
Go
//go:build darwin && !ios
|
|
|
|
package proxy
|
|
|
|
import (
|
|
"fmt"
|
|
"os/exec"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/statemanager"
|
|
)
|
|
|
|
// ShutdownState stores proxy state for cleanup on unclean shutdown.
|
|
type ShutdownState struct {
|
|
ModifiedServices []string `json:"modified_services"`
|
|
}
|
|
|
|
// Name returns the state name for persistence.
|
|
func (s *ShutdownState) Name() string {
|
|
return "proxy_state"
|
|
}
|
|
|
|
// Cleanup restores proxy settings after an unclean shutdown.
|
|
func (s *ShutdownState) Cleanup() error {
|
|
if len(s.ModifiedServices) == 0 {
|
|
return nil
|
|
}
|
|
|
|
log.Infof("cleaning up proxy state for %d services", len(s.ModifiedServices))
|
|
|
|
for _, service := range s.ModifiedServices {
|
|
// Disable web proxy (HTTP)
|
|
cmd := exec.Command(networksetupPath, "-setwebproxystate", service, "off")
|
|
if out, err := cmd.CombinedOutput(); err != nil {
|
|
log.Warnf("cleanup web proxy for %s: %v, output: %s", service, err, out)
|
|
}
|
|
|
|
// Disable secure web proxy (HTTPS)
|
|
cmd = exec.Command(networksetupPath, "-setsecurewebproxystate", service, "off")
|
|
if out, err := cmd.CombinedOutput(); err != nil {
|
|
log.Warnf("cleanup secure web proxy for %s: %v, output: %s", service, err, out)
|
|
}
|
|
|
|
// Disable autoproxy
|
|
cmd = exec.Command(networksetupPath, "-setautoproxystate", service, "off")
|
|
if out, err := cmd.CombinedOutput(); err != nil {
|
|
log.Warnf("cleanup autoproxy for %s: %v, output: %s", service, err, out)
|
|
}
|
|
|
|
log.Debugf("cleaned up proxy for service %s", service)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// RegisterState registers the proxy state with the state manager.
|
|
func RegisterState(stateManager *statemanager.Manager) {
|
|
if stateManager == nil {
|
|
return
|
|
}
|
|
stateManager.RegisterState(&ShutdownState{})
|
|
}
|
|
|
|
// GetProxyState returns the current proxy state from the command line.
|
|
func GetProxyState(service string) (webProxy, secureProxy, autoProxy bool, err error) {
|
|
// Check web proxy state
|
|
cmd := exec.Command(networksetupPath, "-getwebproxy", service)
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
return false, false, false, fmt.Errorf("get web proxy: %w", err)
|
|
}
|
|
webProxy = isProxyEnabled(string(out))
|
|
|
|
// Check secure web proxy state
|
|
cmd = exec.Command(networksetupPath, "-getsecurewebproxy", service)
|
|
out, err = cmd.Output()
|
|
if err != nil {
|
|
return false, false, false, fmt.Errorf("get secure web proxy: %w", err)
|
|
}
|
|
secureProxy = isProxyEnabled(string(out))
|
|
|
|
// Check autoproxy state
|
|
cmd = exec.Command(networksetupPath, "-getautoproxyurl", service)
|
|
out, err = cmd.Output()
|
|
if err != nil {
|
|
return false, false, false, fmt.Errorf("get autoproxy: %w", err)
|
|
}
|
|
autoProxy = isProxyEnabled(string(out))
|
|
|
|
return webProxy, secureProxy, autoProxy, nil
|
|
}
|
|
|
|
func isProxyEnabled(output string) bool {
|
|
return !contains(output, "Enabled: No") && contains(output, "Enabled: Yes")
|
|
}
|
|
|
|
func contains(s, substr string) bool {
|
|
for i := 0; i <= len(s)-len(substr); i++ {
|
|
if s[i:i+len(substr)] == substr {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|