kubenetworks/kubevpn

package handler

import (
    "context"
    "net"
    "os"
    "os/signal"
    "syscall"
    "time"

    log "github.com/sirupsen/logrus"
    "k8s.io/apimachinery/pkg/apis/meta/v1"
    "k8s.io/utils/pointer"

    "github.com/wencaiwulue/kubevpn/v2/pkg/config"
)

func (c *ConnectOptions) setupSignalHandler() {
    var stopChan = make(chan os.Signal)
    signal.Notify(stopChan, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM, syscall.SIGQUIT, syscall.SIGKILL)
    select {
    case <-stopChan:
        c.Cleanup()
    case <-c.ctx.Done():
    }
}

func (c *ConnectOptions) Cleanup() {
    if c == nil {
        return
    }

    var inUserDaemon bool
    if c.ctx != nil {
        inUserDaemon = true
    }

    c.once.Do(func() {
        if inUserDaemon {
            log.Info("Performing cleanup operations")
        }
        ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
        defer cancel()
        var ips []net.IP
        if c.localTunIPv4 != nil && c.localTunIPv4.IP != nil {
            ips = append(ips, c.localTunIPv4.IP)
        }
        if c.localTunIPv6 != nil && c.localTunIPv6.IP != nil {
            ips = append(ips, c.localTunIPv6.IP)
        }
        if c.dhcp != nil {
            err := c.dhcp.ReleaseIP(ctx, ips...)
            if err != nil {
                log.Errorf("Failed to release IP to dhcp, err: %v", err)
            }
        }
        if c.clientset != nil {
            _ = c.clientset.CoreV1().Pods(c.Namespace).Delete(ctx, config.CniNetName, v1.DeleteOptions{GracePeriodSeconds: pointer.Int64(0)})
        }
        // leave proxy resources
        err := c.LeaveProxyResources(ctx)
        if err != nil {
            log.Errorf("Leave proxy resources error: %v", err)
        }

        for _, function := range c.getRolloutFunc() {
            if function != nil {
                if err = function(); err != nil {
                    log.Warnf("Rollout function error: %v", err)
                }
            }
        }
        if c.cancel != nil {
            c.cancel()
        }
        if c.dnsConfig != nil {
            if inUserDaemon {
                log.Infof("Clearing DNS settings")
            }
            c.dnsConfig.CancelDNS()
        }
    })
}