wrapper ipvs API as util

kubernetes · Jul 31, 2017 · 4211794 · 4211794
1 parent 9c7426d
commit 4211794
Show file tree

Hide file tree

Showing 9 changed files with 1,535 additions and 0 deletions.
diff --git a/pkg/util/BUILD b/pkg/util/BUILD
@@ -30,6 +30,7 @@ filegroup(
         "//pkg/util/io:all-srcs",
         "//pkg/util/ipconfig:all-srcs",
         "//pkg/util/iptables:all-srcs",
+        "//pkg/util/ipvs:all-srcs",
         "//pkg/util/keymutex:all-srcs",
         "//pkg/util/labels:all-srcs",
         "//pkg/util/limitwriter:all-srcs",

diff --git a/pkg/util/ipvs/BUILD b/pkg/util/ipvs/BUILD
@@ -0,0 +1,56 @@
+package(default_visibility = ["//visibility:public"])
+
+licenses(["notice"])
+
+load(
+    "@io_bazel_rules_go//go:def.bzl",
+    "go_library",
+    "go_test",
+)
+
+go_test(
+    name = "go_default_test",
+    srcs = [
+        "ipvs_linux_test.go",
+        "ipvs_test.go",
+    ],
+    library = ":go_default_library",
+    tags = ["automanaged"],
+    deps = [
+        "//vendor/github.com/docker/libnetwork/ipvs:go_default_library",
+        "//vendor/k8s.io/apimachinery/pkg/util/sets:go_default_library",
+        "//vendor/k8s.io/utils/exec:go_default_library",
+        "//vendor/k8s.io/utils/exec/testing:go_default_library",
+    ],
+)
+
+go_library(
+    name = "go_default_library",
+    srcs = [
+        "ipvs.go",
+        "ipvs_linux.go",
+    ],
+    tags = ["automanaged"],
+    deps = [
+        "//pkg/util/sysctl:go_default_library",
+        "//vendor/github.com/docker/libnetwork/ipvs:go_default_library",
+        "//vendor/github.com/golang/glog:go_default_library",
+        "//vendor/k8s.io/utils/exec:go_default_library",
+    ],
+)
+
+filegroup(
+    name = "package-srcs",
+    srcs = glob(["**"]),
+    tags = ["automanaged"],
+    visibility = ["//visibility:private"],
+)
+
+filegroup(
+    name = "all-srcs",
+    srcs = [
+        ":package-srcs",
+        "//pkg/util/ipvs/testing:all-srcs",
+    ],
+    tags = ["automanaged"],
+)
diff --git a/pkg/util/ipvs/ipvs.go b/pkg/util/ipvs/ipvs.go
@@ -0,0 +1,92 @@
+/*
+Copyright 2017 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 ipvs
+
+import (
+	"net"
+	"strconv"
+)
+
+// Interface is an injectable interface for running ipvs commands.  Implementations must be goroutine-safe.
+type Interface interface {
+	// Flush clears all services in system. return occurred error immediately.
+	Flush() error
+	// EnsureDummyDevice checks if the specified dummy interface is present and, if not, creates it.  If the dummy interface existed, return true.
+	EnsureDummyDevice(dummyDev string) (exist bool, err error)
+	// DeleteDummyDevice deletes the specified dummy interface.  If the dummy interface existed, return error.
+	DeleteDummyDevice(dummyDev string) error
+	// EnsureServiceAddressBind checks if service's address is bound to dummy interface and, if not, binds it. If the address is already bound, return true.
+	EnsureServiceAddressBind(serv *InternalService, dev string) (exist bool, err error)
+	// UnBindServiceAddress checks if service's address is bound to dummy interface and, if so, unbinds it.
+	UnBindServiceAddress(serv *InternalService, dev string) error
+	// AddService creates the specified service.
+	AddService(*InternalService) error
+	// DeleteService deletes the specified service.  If the service does not exist, return error.
+	DeleteService(*InternalService) error
+	// GetService returns the specified service information in the system.
+	GetService(*InternalService) (*InternalService, error)
+	// GetServices lists all services in the system.
+	GetServices() ([]*InternalService, error)
+	// AddDestination creates the specified destination for the specified service.
+	AddDestination(*InternalService, *InternalDestination) error
+	// GetDestinations returns all destinations for the specified service.
+	GetDestinations(*InternalService) ([]*InternalDestination, error)
+	// DeleteDestination deletes the specified destination from the specified service.
+	DeleteDestination(*InternalService, *InternalDestination) error
+}
+
+// InternalService is an internal definition of an IPVS service in its entirety.
+type InternalService struct {
+	Address   net.IP
+	Protocol  string
+	Port      uint16
+	Scheduler string
+	Flags     ServiceFlags
+	Timeout   uint32
+}
+
+// ServiceFlags is used to specify session affinity, ip hash etc.
+type ServiceFlags uint32
+
+const (
+	// FlagPersistent specify IPVS service session affinity
+	FlagPersistent = 0x1
+)
+
+// Equal check the equality of internal service
+func (svc *InternalService) Equal(other *InternalService) bool {
+	return svc.Address.Equal(other.Address) &&
+		svc.Protocol == other.Protocol &&
+		svc.Port == other.Port &&
+		svc.Scheduler == other.Scheduler &&
+		svc.Timeout == other.Timeout
+}
+
+func (svc *InternalService) String() string {
+	return net.JoinHostPort(svc.Address.String(), strconv.Itoa(int(svc.Port))) + "/" + svc.Protocol
+}
+
+// InternalDestination is an internal definition of an IPVS destination in its entirety.
+type InternalDestination struct {
+	Address net.IP
+	Port    uint16
+	Weight  int
+}
+
+func (dest *InternalDestination) String() string {
+	return net.JoinHostPort(dest.Address.String(), strconv.Itoa(int(dest.Port)))
+}