github.com/kaisenlinux/docker.io@v0.0.0-20230510090727-ea55db55fac7/libnetwork/cmd/ovrouter/ovrouter.go (about)

     1  package main
     2  
     3  import (
     4  	"fmt"
     5  	"net"
     6  	"os"
     7  	"os/signal"
     8  
     9  	"github.com/docker/docker/pkg/plugingetter"
    10  	"github.com/docker/docker/pkg/reexec"
    11  	"github.com/docker/libnetwork/driverapi"
    12  	"github.com/docker/libnetwork/drivers/overlay"
    13  	"github.com/docker/libnetwork/netlabel"
    14  	"github.com/docker/libnetwork/types"
    15  	"github.com/vishvananda/netlink"
    16  )
    17  
    18  type router struct {
    19  	d driverapi.Driver
    20  }
    21  
    22  type endpoint struct {
    23  	addr *net.IPNet
    24  	mac  net.HardwareAddr
    25  	name string
    26  }
    27  
    28  func (r *router) GetPluginGetter() plugingetter.PluginGetter {
    29  	return nil
    30  }
    31  
    32  func (r *router) RegisterDriver(name string, driver driverapi.Driver, c driverapi.Capability) error {
    33  	r.d = driver
    34  	return nil
    35  }
    36  
    37  func (ep *endpoint) Interface() driverapi.InterfaceInfo {
    38  	return nil
    39  }
    40  
    41  func (ep *endpoint) SetMacAddress(mac net.HardwareAddr) error {
    42  	if ep.mac != nil {
    43  		return types.ForbiddenErrorf("endpoint interface MAC address present (%s). Cannot be modified with %s.", ep.mac, mac)
    44  	}
    45  	if mac == nil {
    46  		return types.BadRequestErrorf("tried to set nil MAC address to endpoint interface")
    47  	}
    48  	ep.mac = types.GetMacCopy(mac)
    49  	return nil
    50  }
    51  
    52  func (ep *endpoint) SetIPAddress(address *net.IPNet) error {
    53  	if address.IP == nil {
    54  		return types.BadRequestErrorf("tried to set nil IP address to endpoint interface")
    55  	}
    56  	if address.IP.To4() == nil {
    57  		return types.NotImplementedErrorf("do not support ipv6 yet")
    58  	}
    59  	if ep.addr != nil {
    60  		return types.ForbiddenErrorf("endpoint interface IP present (%s). Cannot be modified with %s.", ep.addr, address)
    61  	}
    62  	ep.addr = types.GetIPNetCopy(address)
    63  	return nil
    64  }
    65  
    66  func (ep *endpoint) MacAddress() net.HardwareAddr {
    67  	return types.GetMacCopy(ep.mac)
    68  }
    69  
    70  func (ep *endpoint) Address() *net.IPNet {
    71  	return types.GetIPNetCopy(ep.addr)
    72  }
    73  
    74  func (ep *endpoint) AddressIPv6() *net.IPNet {
    75  	return nil
    76  }
    77  
    78  func (ep *endpoint) InterfaceName() driverapi.InterfaceNameInfo {
    79  	return ep
    80  }
    81  
    82  func (ep *endpoint) SetNames(srcName, dstPrefix string) error {
    83  	ep.name = srcName
    84  	return nil
    85  }
    86  
    87  func (ep *endpoint) SetGateway(net.IP) error {
    88  	return nil
    89  }
    90  
    91  func (ep *endpoint) SetGatewayIPv6(net.IP) error {
    92  	return nil
    93  }
    94  
    95  func (ep *endpoint) AddStaticRoute(destination *net.IPNet, routeType int,
    96  	nextHop net.IP) error {
    97  	return nil
    98  }
    99  
   100  func (ep *endpoint) AddTableEntry(tableName string, key string, value []byte) error {
   101  	return nil
   102  }
   103  
   104  func (ep *endpoint) DisableGatewayService() {}
   105  
   106  func main() {
   107  	if reexec.Init() {
   108  		return
   109  	}
   110  
   111  	opt := make(map[string]interface{})
   112  	if len(os.Args) > 1 {
   113  		opt[netlabel.OverlayBindInterface] = os.Args[1]
   114  	}
   115  	if len(os.Args) > 2 {
   116  		opt[netlabel.OverlayNeighborIP] = os.Args[2]
   117  	}
   118  	if len(os.Args) > 3 {
   119  		opt[netlabel.GlobalKVProvider] = os.Args[3]
   120  	}
   121  	if len(os.Args) > 4 {
   122  		opt[netlabel.GlobalKVProviderURL] = os.Args[4]
   123  	}
   124  
   125  	r := &router{}
   126  	if err := overlay.Init(r, opt); err != nil {
   127  		fmt.Printf("Failed to initialize overlay driver: %v\n", err)
   128  		os.Exit(1)
   129  	}
   130  
   131  	if err := r.d.CreateNetwork("testnetwork",
   132  		map[string]interface{}{}, nil, nil, nil); err != nil {
   133  		fmt.Printf("Failed to create network in the driver: %v\n", err)
   134  		os.Exit(1)
   135  	}
   136  
   137  	ep := &endpoint{}
   138  	if err := r.d.CreateEndpoint("testnetwork", "testep",
   139  		ep, map[string]interface{}{}); err != nil {
   140  		fmt.Printf("Failed to create endpoint in the driver: %v\n", err)
   141  		os.Exit(1)
   142  	}
   143  
   144  	if err := r.d.Join("testnetwork", "testep",
   145  		"", ep, map[string]interface{}{}); err != nil {
   146  		fmt.Printf("Failed to join an endpoint in the driver: %v\n", err)
   147  		os.Exit(1)
   148  	}
   149  
   150  	link, err := netlink.LinkByName(ep.name)
   151  	if err != nil {
   152  		fmt.Printf("Failed to find the container interface with name %s: %v\n",
   153  			ep.name, err)
   154  		os.Exit(1)
   155  	}
   156  
   157  	ipAddr := &netlink.Addr{IPNet: ep.addr, Label: ""}
   158  	if err := netlink.AddrAdd(link, ipAddr); err != nil {
   159  		fmt.Printf("Failed to add address to the interface: %v\n", err)
   160  		os.Exit(1)
   161  	}
   162  
   163  	sigCh := make(chan os.Signal, 1)
   164  	signal.Notify(sigCh, os.Interrupt, os.Kill)
   165  
   166  	for {
   167  		select {
   168  		case <-sigCh:
   169  			r.d.Leave("testnetwork", "testep")
   170  			overlay.Fini(r.d)
   171  			os.Exit(0)
   172  		}
   173  	}
   174  }