-
Notifications
You must be signed in to change notification settings - Fork 11
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add node IPAM controller. It supports two modes: * RangeAllocator - classic mode (kubernetes does the same) * CloudAllocator - Talos is responsible for setting PodCIDRs Signed-off-by: Serge Logvinov <[email protected]>
- Loading branch information
1 parent
3b20bb0
commit f2516ce
Showing
33 changed files
with
4,060 additions
and
198 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
250 changes: 250 additions & 0 deletions
250
cmd/talos-cloud-controller-manager/nodeipamcontroller.go
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,250 @@ | ||
| /* | ||
| Copyright 2018 The Kubernetes Authors. | ||
| Licensed under the Apache License, Version 2.0 (the "License"); | ||
| you may not use this file except in compliance with the License. | ||
| You may obtain a copy of the License at | ||
| http://www.apache.org/licenses/LICENSE-2.0 | ||
| Unless required by applicable law or agreed to in writing, software | ||
| distributed under the License is distributed on an "AS IS" BASIS, | ||
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| See the License for the specific language governing permissions and | ||
| limitations under the License. | ||
| */ | ||
|
|
||
| package main // copy from kubernetes/cmd/cloud-controller-manager/nodeipamcontroller.go | ||
|
|
||
| import ( | ||
| "context" | ||
| "errors" | ||
| "fmt" | ||
| "net" | ||
| "strings" | ||
|
|
||
| nodeipamcontrolleroptions "github.com/siderolabs/talos-cloud-controller-manager/cmd/talos-cloud-controller-manager/options" | ||
| nodeipamcontroller "github.com/siderolabs/talos-cloud-controller-manager/pkg/nodeipam" | ||
| nodeipamconfig "github.com/siderolabs/talos-cloud-controller-manager/pkg/nodeipam/config" | ||
| ipam "github.com/siderolabs/talos-cloud-controller-manager/pkg/nodeipam/ipam" | ||
| talosclient "github.com/siderolabs/talos-cloud-controller-manager/pkg/talosclient" | ||
|
|
||
| cloudprovider "k8s.io/cloud-provider" | ||
| "k8s.io/cloud-provider/app" | ||
| cloudcontrollerconfig "k8s.io/cloud-provider/app/config" | ||
| genericcontrollermanager "k8s.io/controller-manager/app" | ||
| "k8s.io/controller-manager/controller" | ||
| "k8s.io/klog/v2" | ||
| netutils "k8s.io/utils/net" | ||
| ) | ||
|
|
||
| const ( | ||
| // defaultNodeMaskCIDRIPv4 is default mask size for IPv4 node cidr. | ||
| defaultNodeMaskCIDRIPv4 = 24 | ||
| // defaultNodeMaskCIDRIPv6 is default mask size for IPv6 node cidr. | ||
| defaultNodeMaskCIDRIPv6 = 80 | ||
| ) | ||
|
|
||
| type nodeIPAMController struct { | ||
| nodeIPAMControllerConfiguration nodeipamconfig.NodeIPAMControllerConfiguration | ||
| nodeIPAMControllerOptions nodeipamcontrolleroptions.NodeIPAMControllerOptions | ||
| } | ||
|
|
||
| func (nodeIpamController *nodeIPAMController) startNodeIpamControllerWrapper( | ||
| initContext app.ControllerInitContext, | ||
| completedConfig *cloudcontrollerconfig.CompletedConfig, | ||
| cloud cloudprovider.Interface, | ||
| ) app.InitFunc { | ||
| klog.V(4).InfoS("nodeIpamController.startNodeIpamControllerWrapper() called") | ||
|
|
||
| allErrors := nodeIpamController.nodeIPAMControllerOptions.Validate() | ||
| if len(allErrors) > 0 { | ||
| klog.Fatal("NodeIPAM controller values are not properly set.") | ||
| } | ||
|
|
||
| nodeIpamController.nodeIPAMControllerOptions.ApplyTo(&nodeIpamController.nodeIPAMControllerConfiguration) //nolint:errcheck | ||
|
|
||
| return func(ctx context.Context, controllerContext genericcontrollermanager.ControllerContext) (controller.Interface, bool, error) { | ||
| return startNodeIpamController(ctx, initContext, completedConfig, nodeIpamController.nodeIPAMControllerConfiguration, controllerContext, cloud) | ||
| } | ||
| } | ||
|
|
||
| func startNodeIpamController( | ||
| ctx context.Context, | ||
| initContext app.ControllerInitContext, | ||
| ccmConfig *cloudcontrollerconfig.CompletedConfig, | ||
| nodeIPAMConfig nodeipamconfig.NodeIPAMControllerConfiguration, | ||
| controllerCtx genericcontrollermanager.ControllerContext, | ||
| cloud cloudprovider.Interface, | ||
| ) (controller.Interface, bool, error) { | ||
| // should we start nodeIPAM | ||
| if !ccmConfig.ComponentConfig.KubeCloudShared.AllocateNodeCIDRs { | ||
| return nil, false, nil | ||
| } | ||
|
|
||
| talos, err := talosclient.New(ctx) | ||
| if err != nil { | ||
| return nil, false, err | ||
| } | ||
|
|
||
| if ccmConfig.ComponentConfig.KubeCloudShared.ClusterCIDR == "" { | ||
| clusterCIDRs, err := talos.GetPodCIDRs(ctx) | ||
| if err != nil { | ||
| return nil, false, err | ||
| } | ||
|
|
||
| ccmConfig.ComponentConfig.KubeCloudShared.ClusterCIDR = strings.Join(clusterCIDRs, ",") | ||
| } | ||
|
|
||
| // failure: bad cidrs in config | ||
| clusterCIDRs, dualStack, err := processCIDRs(ccmConfig.ComponentConfig.KubeCloudShared.ClusterCIDR) | ||
| if err != nil { | ||
| return nil, false, err | ||
| } | ||
|
|
||
| // failure: more than one cidr but they are not configured as dual stack | ||
| if len(clusterCIDRs) > 1 && !dualStack { | ||
| return nil, false, fmt.Errorf("len of ClusterCIDRs==%v and they are not configured as dual stack (at least one from each IPFamily", len(clusterCIDRs)) | ||
| } | ||
|
|
||
| // failure: more than cidrs is not allowed even with dual stack | ||
| if len(clusterCIDRs) > 2 { | ||
| return nil, false, fmt.Errorf("len of clusters is:%v > more than max allowed of 2", len(clusterCIDRs)) | ||
| } | ||
|
|
||
| svcCIDRs, err := talos.GetServiceCIDRs(ctx) | ||
| if err != nil { | ||
| return nil, false, err | ||
| } | ||
|
|
||
| serviceCIDRs, err := netutils.ParseCIDRs(svcCIDRs) | ||
| if err != nil { | ||
| return nil, false, err | ||
| } | ||
|
|
||
| nodeIPAMConfig.ServiceCIDR = svcCIDRs[0] | ||
| if len(svcCIDRs) > 1 { | ||
| nodeIPAMConfig.SecondaryServiceCIDR = svcCIDRs[1] | ||
| } | ||
|
|
||
| nodeCIDRMaskSizes, err := setNodeCIDRMaskSizes(nodeIPAMConfig, clusterCIDRs) | ||
| if err != nil { | ||
| return nil, false, err | ||
| } | ||
|
|
||
| klog.V(4).InfoS("nodeIpamController called", "clusterCIDRs", clusterCIDRs, "serviceCIDRs", serviceCIDRs, "nodeCIDRMaskSizes", nodeCIDRMaskSizes) | ||
|
|
||
| nodeIpamController, err := nodeipamcontroller.NewNodeIpamController( | ||
| ctx, | ||
| controllerCtx.InformerFactory.Core().V1().Nodes(), | ||
| cloud, | ||
| controllerCtx.ClientBuilder.ClientOrDie(initContext.ClientName), | ||
| clusterCIDRs, | ||
| serviceCIDRs, | ||
| nodeCIDRMaskSizes, | ||
| ipam.CIDRAllocatorType(ccmConfig.ComponentConfig.KubeCloudShared.CIDRAllocatorType), | ||
| ) | ||
| if err != nil { | ||
| return nil, true, err | ||
| } | ||
|
|
||
| go nodeIpamController.Run(ctx) | ||
|
|
||
| return nil, true, nil | ||
| } | ||
|
|
||
| // processCIDRs is a helper function that works on a comma separated cidrs and returns | ||
| // a list of typed cidrs | ||
| // a flag if cidrs represents a dual stack | ||
| // error if failed to parse any of the cidrs. | ||
| func processCIDRs(cidrsList string) ([]*net.IPNet, bool, error) { | ||
| cidrsSplit := strings.Split(strings.TrimSpace(cidrsList), ",") | ||
|
|
||
| cidrs, err := netutils.ParseCIDRs(cidrsSplit) | ||
| if err != nil { | ||
| return nil, false, err | ||
| } | ||
|
|
||
| // if cidrs has an error then the previous call will fail | ||
| // safe to ignore error checking on next call | ||
| dualstack, _ := netutils.IsDualStackCIDRs(cidrs) //nolint:errcheck | ||
|
|
||
| return cidrs, dualstack, nil | ||
| } | ||
|
|
||
| // setNodeCIDRMaskSizes returns the IPv4 and IPv6 node cidr mask sizes to the value provided | ||
| // for --node-cidr-mask-size-ipv4 and --node-cidr-mask-size-ipv6 respectively. If value not provided, | ||
| // then it will return default IPv4 and IPv6 cidr mask sizes. | ||
| func setNodeCIDRMaskSizes(cfg nodeipamconfig.NodeIPAMControllerConfiguration, clusterCIDRs []*net.IPNet) ([]int, error) { | ||
| sortedSizes := func(maskSizeIPv4, maskSizeIPv6 int) []int { | ||
| nodeMaskCIDRs := make([]int, len(clusterCIDRs)) | ||
|
|
||
| for idx, clusterCIDR := range clusterCIDRs { | ||
| if netutils.IsIPv6CIDR(clusterCIDR) { | ||
| nodeMaskCIDRs[idx] = maskSizeIPv6 | ||
| } else { | ||
| nodeMaskCIDRs[idx] = maskSizeIPv4 | ||
| } | ||
| } | ||
|
|
||
| return nodeMaskCIDRs | ||
| } | ||
|
|
||
| // --node-cidr-mask-size flag is incompatible with dual stack clusters. | ||
| ipv4Mask, ipv6Mask := defaultNodeMaskCIDRIPv4, defaultNodeMaskCIDRIPv6 | ||
| isDualstack := len(clusterCIDRs) > 1 | ||
|
|
||
| // case one: cluster is dualstack (i.e, more than one cidr) | ||
| if isDualstack { | ||
| // if --node-cidr-mask-size then fail, user must configure the correct dual-stack mask sizes (or use default) | ||
| if cfg.NodeCIDRMaskSize != 0 { | ||
| return nil, errors.New("usage of --node-cidr-mask-size is not allowed with dual-stack clusters") | ||
| } | ||
|
|
||
| if cfg.NodeCIDRMaskSizeIPv4 != 0 { | ||
| ipv4Mask = int(cfg.NodeCIDRMaskSizeIPv4) | ||
| } | ||
|
|
||
| if cfg.NodeCIDRMaskSizeIPv6 != 0 { | ||
| ipv6Mask = int(cfg.NodeCIDRMaskSizeIPv6) | ||
| } | ||
|
|
||
| return sortedSizes(ipv4Mask, ipv6Mask), nil | ||
| } | ||
|
|
||
| maskConfigured := cfg.NodeCIDRMaskSize != 0 | ||
| maskV4Configured := cfg.NodeCIDRMaskSizeIPv4 != 0 | ||
| maskV6Configured := cfg.NodeCIDRMaskSizeIPv6 != 0 | ||
| isSingleStackIPv6 := netutils.IsIPv6CIDR(clusterCIDRs[0]) | ||
|
|
||
| // original flag is set | ||
| if maskConfigured { | ||
| // original mask flag is still the main reference. | ||
| if maskV4Configured || maskV6Configured { | ||
| return nil, errors.New("usage of --node-cidr-mask-size-ipv4 and --node-cidr-mask-size-ipv6 is not allowed if --node-cidr-mask-size is set. For dual-stack clusters please unset it and use IPFamily specific flags") //nolint:lll | ||
| } | ||
|
|
||
| mask := int(cfg.NodeCIDRMaskSize) | ||
|
|
||
| return sortedSizes(mask, mask), nil | ||
| } | ||
|
|
||
| if maskV4Configured { | ||
| if isSingleStackIPv6 { | ||
| return nil, errors.New("usage of --node-cidr-mask-size-ipv4 is not allowed for a single-stack IPv6 cluster") | ||
| } | ||
|
|
||
| ipv4Mask = int(cfg.NodeCIDRMaskSizeIPv4) | ||
| } | ||
|
|
||
| // !maskV4Configured && !maskConfigured && maskV6Configured | ||
| if maskV6Configured { | ||
| if !isSingleStackIPv6 { | ||
| return nil, errors.New("usage of --node-cidr-mask-size-ipv6 is not allowed for a single-stack IPv4 cluster") | ||
| } | ||
|
|
||
| ipv6Mask = int(cfg.NodeCIDRMaskSizeIPv6) | ||
| } | ||
|
|
||
| return sortedSizes(ipv4Mask, ipv6Mask), nil | ||
| } |
Oops, something went wrong.