mirror of
https://github.com/k3s-io/k3s.git
synced 2024-06-07 19:41:36 +00:00
135 lines
3.1 KiB
Go
135 lines
3.1 KiB
Go
// Copyright 2015 flannel 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 subnet
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"regexp"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/flannel-io/flannel/pkg/ip"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
var (
|
|
ErrLeaseTaken = errors.New("subnet: lease already taken")
|
|
ErrNoMoreTries = errors.New("subnet: no more tries")
|
|
subnetRegex = regexp.MustCompile(`(\d+\.\d+.\d+.\d+)-(\d+)`)
|
|
)
|
|
|
|
type LeaseAttrs struct {
|
|
PublicIP ip.IP4
|
|
PublicIPv6 *ip.IP6
|
|
BackendType string `json:",omitempty"`
|
|
BackendData json.RawMessage `json:",omitempty"`
|
|
BackendV6Data json.RawMessage `json:",omitempty"`
|
|
}
|
|
|
|
type Lease struct {
|
|
EnableIPv4 bool
|
|
EnableIPv6 bool
|
|
Subnet ip.IP4Net
|
|
IPv6Subnet ip.IP6Net
|
|
Attrs LeaseAttrs
|
|
Expiration time.Time
|
|
|
|
Asof uint64
|
|
}
|
|
|
|
func (l *Lease) Key() string {
|
|
return MakeSubnetKey(l.Subnet)
|
|
}
|
|
|
|
type (
|
|
EventType int
|
|
|
|
Event struct {
|
|
Type EventType `json:"type"`
|
|
Lease Lease `json:"lease,omitempty"`
|
|
}
|
|
)
|
|
|
|
const (
|
|
EventAdded EventType = iota
|
|
EventRemoved
|
|
)
|
|
|
|
type LeaseWatchResult struct {
|
|
// Either Events or Snapshot will be set. If Events is empty, it means
|
|
// the cursor was out of range and Snapshot contains the current list
|
|
// of items, even if empty.
|
|
Events []Event `json:"events"`
|
|
Snapshot []Lease `json:"snapshot"`
|
|
Cursor interface{} `json:"cursor"`
|
|
}
|
|
|
|
func (et EventType) MarshalJSON() ([]byte, error) {
|
|
s := ""
|
|
|
|
switch et {
|
|
case EventAdded:
|
|
s = "added"
|
|
case EventRemoved:
|
|
s = "removed"
|
|
default:
|
|
return nil, errors.New("bad event type")
|
|
}
|
|
return json.Marshal(s)
|
|
}
|
|
|
|
func (et *EventType) UnmarshalJSON(data []byte) error {
|
|
switch string(data) {
|
|
case "\"added\"":
|
|
*et = EventAdded
|
|
case "\"removed\"":
|
|
*et = EventRemoved
|
|
default:
|
|
fmt.Println(string(data))
|
|
return errors.New("bad event type")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func ParseSubnetKey(s string) *ip.IP4Net {
|
|
if parts := subnetRegex.FindStringSubmatch(s); len(parts) == 3 {
|
|
snIp := net.ParseIP(parts[1]).To4()
|
|
prefixLen, err := strconv.ParseUint(parts[2], 10, 5)
|
|
if snIp != nil && err == nil {
|
|
return &ip.IP4Net{IP: ip.FromIP(snIp), PrefixLen: uint(prefixLen)}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func MakeSubnetKey(sn ip.IP4Net) string {
|
|
return sn.StringSep(".", "-")
|
|
}
|
|
|
|
type Manager interface {
|
|
GetNetworkConfig(ctx context.Context) (*Config, error)
|
|
AcquireLease(ctx context.Context, attrs *LeaseAttrs) (*Lease, error)
|
|
RenewLease(ctx context.Context, lease *Lease) error
|
|
WatchLease(ctx context.Context, sn ip.IP4Net, cursor interface{}) (LeaseWatchResult, error)
|
|
WatchLeases(ctx context.Context, cursor interface{}) (LeaseWatchResult, error)
|
|
|
|
Name() string
|
|
}
|