kubevpn/pkg/dhcp/server.go

package dhcp

import (
	"context"
	"net"
	"sync"

	"k8s.io/client-go/kubernetes"

	"github.com/wencaiwulue/kubevpn/v2/pkg/dhcp/rpc"
	plog "github.com/wencaiwulue/kubevpn/v2/pkg/log"
)

type Server struct {
	rpc.UnimplementedDHCPServer

	sync.Mutex
	clientset *kubernetes.Clientset
}

func NewServer(clientset *kubernetes.Clientset) *Server {
	return &Server{
		clientset: clientset,
	}
}

func (s *Server) RentIP(ctx context.Context, req *rpc.RentIPRequest) (*rpc.RentIPResponse, error) {
	s.Lock()
	defer s.Unlock()

	plog.G(ctx).Infof("Handling rent IP request, pod name: %s, ns: %s", req.PodName, req.PodNamespace)
	mapInterface := s.clientset.CoreV1().ConfigMaps(req.PodNamespace)
	manager := NewDHCPManager(mapInterface, req.PodNamespace)
	v4, v6, err := manager.RentIP(ctx)
	if err != nil {
		plog.G(ctx).Errorf("Failed to rent IP: %v", err)
		return nil, err
	}
	// todo patch annotation
	resp := &rpc.RentIPResponse{
		IPv4CIDR: v4.String(),
		IPv6CIDR: v6.String(),
	}
	return resp, nil
}

func (s *Server) ReleaseIP(ctx context.Context, req *rpc.ReleaseIPRequest) (*rpc.ReleaseIPResponse, error) {
	s.Lock()
	defer s.Unlock()

	plog.G(ctx).Infof("Handling release IP request, pod name: %s, ns: %s, IPv4: %s, IPv6: %s", req.PodName, req.PodNamespace, req.IPv4CIDR, req.IPv6CIDR)
	var ips []net.IP
	for _, ipStr := range []string{req.IPv4CIDR, req.IPv6CIDR} {
		ip, _, err := net.ParseCIDR(ipStr)
		if err != nil {
			plog.G(ctx).Errorf("IP %s is invailed: %v", ipStr, err)
			continue
		}
		ips = append(ips, ip)
	}

	mapInterface := s.clientset.CoreV1().ConfigMaps(req.PodNamespace)
	manager := NewDHCPManager(mapInterface, req.PodNamespace)
	if err := manager.ReleaseIP(ctx, ips...); err != nil {
		plog.G(ctx).Errorf("Failed to release IP: %v", err)
		return nil, err
	}
	return &rpc.ReleaseIPResponse{}, nil
}