mirror of
https://github.com/netbirdio/netbird.git
synced 2026-05-17 22:29:54 +00:00
1b011a2d85
Add new package ipfwdstate that implements reference counting for IP forwarding state management. This allows multiple usage to safely request IP forwarding without interfering with each other.
52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package ipfwdstate
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/netbirdio/netbird/client/internal/routemanager/systemops"
|
|
)
|
|
|
|
// IPForwardingState is a struct that keeps track of the IP forwarding state.
|
|
// todo: read initial state of the IP forwarding from the system and reset the state based on it
|
|
type IPForwardingState struct {
|
|
enabledCounter int
|
|
}
|
|
|
|
func NewIPForwardingState() *IPForwardingState {
|
|
return &IPForwardingState{}
|
|
}
|
|
|
|
func (f *IPForwardingState) RequestForwarding() error {
|
|
if f.enabledCounter != 0 {
|
|
f.enabledCounter++
|
|
return nil
|
|
}
|
|
|
|
if err := systemops.EnableIPForwarding(); err != nil {
|
|
return fmt.Errorf("failed to enable IP forwarding with sysctl: %w", err)
|
|
}
|
|
f.enabledCounter = 1
|
|
log.Info("IP forwarding enabled")
|
|
|
|
return nil
|
|
}
|
|
|
|
func (f *IPForwardingState) ReleaseForwarding() error {
|
|
if f.enabledCounter == 0 {
|
|
return nil
|
|
}
|
|
|
|
if f.enabledCounter > 1 {
|
|
f.enabledCounter--
|
|
return nil
|
|
}
|
|
|
|
// if failed to disable IP forwarding we anyway decrement the counter
|
|
f.enabledCounter = 0
|
|
|
|
// todo call systemops.DisableIPForwarding()
|
|
return nil
|
|
}
|