123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186 |
- package vxlan
- import (
- "encoding/json"
- "fmt"
- "net"
- "sync"
- "golang.org/x/net/context"
- "github.com/flannel-io/flannel/backend"
- "github.com/flannel-io/flannel/pkg/ip"
- "github.com/flannel-io/flannel/subnet"
- log "k8s.io/klog"
- )
- func init() {
- backend.Register("vxlan", New)
- }
- const (
- defaultVNI = 1
- )
- type VXLANBackend struct {
- subnetMgr subnet.Manager
- extIface *backend.ExternalInterface
- }
- func New(sm subnet.Manager, extIface *backend.ExternalInterface) (backend.Backend, error) {
- backend := &VXLANBackend{
- subnetMgr: sm,
- extIface: extIface,
- }
- return backend, nil
- }
- func newSubnetAttrs(publicIP net.IP, vnid uint16, mac net.HardwareAddr) (*subnet.LeaseAttrs, error) {
- data, err := json.Marshal(&vxlanLeaseAttrs{
- VNI: vnid,
- VtepMAC: hardwareAddr(mac)})
- if err != nil {
- return nil, err
- }
- return &subnet.LeaseAttrs{
- PublicIP: ip.FromIP(publicIP),
- BackendType: "vxlan",
- BackendData: json.RawMessage(data),
- }, nil
- }
- func (be *VXLANBackend) RegisterNetwork(ctx context.Context, wg *sync.WaitGroup, config *subnet.Config) (backend.Network, error) {
-
- cfg := struct {
- VNI int
- Port int
- GBP bool
- Learning bool
- DirectRouting bool
- }{
- VNI: defaultVNI,
- }
- if len(config.Backend) > 0 {
- if err := json.Unmarshal(config.Backend, &cfg); err != nil {
- return nil, fmt.Errorf("error decoding VXLAN backend config: %v", err)
- }
- }
- log.Infof("VXLAN config: VNI=%d Port=%d GBP=%v Learning=%v DirectRouting=%v", cfg.VNI, cfg.Port, cfg.GBP, cfg.Learning, cfg.DirectRouting)
- devAttrs := vxlanDeviceAttrs{
- vni: uint32(cfg.VNI),
- name: fmt.Sprintf("flannel.%v", cfg.VNI),
- vtepIndex: be.extIface.Iface.Index,
- vtepAddr: be.extIface.IfaceAddr,
- vtepPort: cfg.Port,
- gbp: cfg.GBP,
- learning: cfg.Learning,
- }
- dev, err := newVXLANDevice(&devAttrs)
- if err != nil {
- return nil, err
- }
- dev.directRouting = cfg.DirectRouting
- subnetAttrs, err := newSubnetAttrs(be.extIface.ExtAddr, uint16(cfg.VNI), dev.MACAddr())
- if err != nil {
- return nil, err
- }
- lease, err := be.subnetMgr.AcquireLease(ctx, subnetAttrs)
- switch err {
- case nil:
- case context.Canceled, context.DeadlineExceeded:
- return nil, err
- default:
- return nil, fmt.Errorf("failed to acquire lease: %v", err)
- }
-
-
-
- if err := dev.Configure(ip.IP4Net{IP: lease.Subnet.IP, PrefixLen: 32}, config.Network); err != nil {
- return nil, fmt.Errorf("failed to configure interface %s: %s", dev.link.Attrs().Name, err)
- }
- return newNetwork(be.subnetMgr, be.extIface, dev, ip.IP4Net{}, lease)
- }
- type hardwareAddr net.HardwareAddr
- func (hw hardwareAddr) MarshalJSON() ([]byte, error) {
- return []byte(fmt.Sprintf("%q", net.HardwareAddr(hw))), nil
- }
- func (hw *hardwareAddr) UnmarshalJSON(bytes []byte) error {
- if len(bytes) < 2 || bytes[0] != '"' || bytes[len(bytes)-1] != '"' {
- return fmt.Errorf("error parsing hardware addr")
- }
- bytes = bytes[1 : len(bytes)-1]
- mac, err := net.ParseMAC(string(bytes))
- if err != nil {
- return err
- }
- *hw = hardwareAddr(mac)
- return nil
- }
|