40342619e7
- Feat: allow to change VPN type at runtime - Feat: allow to change interface name at runtime - Maint: Add cleanup method to cleanup VPN loop on a vpn shutdown - Change: allow VPN inputs ports only when tunnel is up
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
package vpn
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/qdm12/gluetun/internal/portforward"
|
|
)
|
|
|
|
var (
|
|
errObtainVPNLocalGateway = errors.New("cannot obtain VPN local gateway IP")
|
|
errStartPortForwarding = errors.New("cannot start port forwarding")
|
|
)
|
|
|
|
func (l *Loop) startPortForwarding(ctx context.Context, data tunnelUpData) (err error) {
|
|
if !data.portForwarding {
|
|
return nil
|
|
}
|
|
|
|
// only used for PIA for now
|
|
gateway, err := l.routing.VPNLocalGatewayIP(data.vpnIntf)
|
|
if err != nil {
|
|
return fmt.Errorf("%w: for interface %s: %s", errObtainVPNLocalGateway, data.vpnIntf, err)
|
|
}
|
|
l.logger.Info("VPN gateway IP address: " + gateway.String())
|
|
|
|
pfData := portforward.StartData{
|
|
PortForwarder: data.portForwarder,
|
|
Gateway: gateway,
|
|
ServerName: data.serverName,
|
|
Interface: data.vpnIntf,
|
|
}
|
|
_, err = l.portForward.Start(ctx, pfData)
|
|
if err != nil {
|
|
return fmt.Errorf("%w: %s", errStartPortForwarding, err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (l *Loop) stopPortForwarding(ctx context.Context,
|
|
timeout time.Duration) (err error) {
|
|
if timeout > 0 {
|
|
var cancel context.CancelFunc
|
|
ctx, cancel = context.WithTimeout(ctx, timeout)
|
|
defer cancel()
|
|
}
|
|
|
|
_, err = l.portForward.Stop(ctx)
|
|
return err
|
|
}
|