示例#1
0
func (s *CiliumNetClientSuite) TestGetIPAMConfOK(c *C) {
	cniReq := ipam.IPAMReq{}
	ciliumRoutes := []ipam.Route{
		*ipam.NewRoute(net.IPNet{IP: NodeAddr, Mask: addressing.NodeIPv6Mask}, nil),
		*ipam.NewRoute(addressing.IPv6DefaultRoute, NodeAddr),
	}

	rep := ipam.IPAMConfigRep{
		IPAMConfig: &ipam.IPAMRep{
			IP6: &ipam.IPConfig{
				Gateway: NodeAddr,
				Routes:  ciliumRoutes,
			},
		},
	}

	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		c.Assert(r.Method, Equals, "POST")
		c.Assert(r.URL.Path, Equals, "/allocator/ipam-configuration/"+string(ipam.LibnetworkIPAMType))
		var options ipam.IPAMReq
		err := json.NewDecoder(r.Body).Decode(&options)
		c.Assert(err, IsNil)
		c.Assert(options, DeepEquals, cniReq)
		w.WriteHeader(http.StatusOK)
		err = json.NewEncoder(w).Encode(rep)
		c.Assert(err, Equals, nil)
	}))
	defer server.Close()

	cli := NewTestClient(server.URL, c)

	ipamRep, err := cli.GetIPAMConf(ipam.LibnetworkIPAMType, cniReq)
	c.Assert(err, Equals, nil)
	c.Assert(*ipamRep, DeepEquals, rep)
}
示例#2
0
func (s *DaemonSuite) TestGetIPAMConfOK(c *C) {
	ciliumRoutes := []ipam.Route{
		*ipam.NewRoute(net.IPNet{IP: NodeAddr, Mask: addressing.NodeIPv6Mask}, nil),
		*ipam.NewRoute(addressing.IPv6DefaultRoute, NodeAddr),
	}

	rep := ipam.IPAMConfigRep{
		IPAMConfig: &ipam.IPAMRep{
			IP6: &ipam.IPConfig{
				Gateway: NodeAddr,
				Routes:  ciliumRoutes,
			},
		},
	}

	s.d.OnGetIPAMConf = func(ipamType ipam.IPAMType, options ipam.IPAMReq) (*ipam.IPAMConfigRep, error) {
		c.Assert(ipamType, Equals, ipam.LibnetworkIPAMType)
		c.Assert(options, Equals, ipam.IPAMReq{})
		return &rep, nil
	}

	ipamRep, err := s.c.GetIPAMConf(ipam.LibnetworkIPAMType, ipam.IPAMReq{})
	c.Assert(err, IsNil)
	c.Assert(*ipamRep, DeepEquals, rep)
}
示例#3
0
// getIPAMConfLibnetwork returns the Libnetwork specific IPAM configuration.
func (d *Daemon) getIPAMConfLibnetwork(ln ipam.IPAMReq) (*ipam.IPAMConfigRep, error) {
	if ln.RequestPoolRequest != nil {
		var poolID, pool, gw string

		if ln.RequestPoolRequest.V6 == false {
			poolID = ipam.LibnetworkDefaultPoolV4
			pool = ipam.LibnetworkDummyV4AllocPool
			gw = ipam.LibnetworkDummyV4Gateway
		} else {
			subnetGo := net.IPNet(d.ipamConf.IPAMConfig.Subnet)
			poolID = ipam.LibnetworkDefaultPoolV6
			pool = subnetGo.String()
			gw = d.ipamConf.IPAMConfig.Gateway.String() + "/128"
		}

		return &ipam.IPAMConfigRep{
			RequestPoolResponse: &lnAPI.RequestPoolResponse{
				PoolID: poolID,
				Pool:   pool,
				Data: map[string]string{
					"com.docker.network.gateway": gw,
				},
			},
		}, nil
	}

	ciliumV6Routes := []ipam.Route{}
	for _, r := range d.ipamConf.IPAMConfig.Routes {
		if r.Dst.IP.To4() == nil {
			ciliumRoute := ipam.NewRoute(r.Dst, r.GW)
			ciliumV6Routes = append(ciliumV6Routes, *ciliumRoute)
		}
	}

	rep := &ipam.IPAMConfigRep{
		IPAMConfig: &ipam.IPAMRep{
			IP6: &ipam.IPConfig{
				Gateway: d.ipamConf.IPAMConfig.Gateway,
				Routes:  ciliumV6Routes,
			},
		},
	}

	if d.conf.IPv4Enabled {
		ciliumV4Routes := []ipam.Route{}
		for _, r := range d.ipamConf.IPAMConfig.Routes {
			if r.Dst.IP.To4() != nil {
				ciliumRoute := ipam.NewRoute(r.Dst, r.GW)
				ciliumV4Routes = append(ciliumV4Routes, *ciliumRoute)
			}
		}

		rep.IPAMConfig.IP4 = &ipam.IPConfig{
			Gateway: d.conf.NodeAddress.IPv4Address.IP(),
			Routes:  ciliumV4Routes,
		}
	}

	return rep, nil
}
示例#4
0
// allocateIPCNI allocates an IP for the CNI plugin.
func (d *Daemon) allocateIPCNI(cniReq ipam.IPAMReq) (*ipam.IPAMRep, error) {
	d.ipamConf.AllocatorMutex.Lock()
	defer d.ipamConf.AllocatorMutex.Unlock()

	if cniReq.IP != nil {
		var err error
		if cniReq.IP.To4() != nil {
			if d.conf.IPv4Enabled {
				err = d.ipamConf.IPv4Allocator.Allocate(*cniReq.IP)
			}
		} else {
			err = d.ipamConf.IPv6Allocator.Allocate(*cniReq.IP)
		}
		return nil, err
	}

	ipConf, err := d.ipamConf.IPv6Allocator.AllocateNext()
	if err != nil {
		return nil, err
	}

	v6Routes := []ipam.Route{}
	v4Routes := []ipam.Route{}
	for _, r := range d.ipamConf.IPAMConfig.Routes {
		rt := ipam.NewRoute(r.Dst, r.GW)
		if r.Dst.IP.To4() == nil {
			v6Routes = append(v6Routes, *rt)
		} else {
			v4Routes = append(v4Routes, *rt)
		}
	}

	ipamRep := &ipam.IPAMRep{
		IP6: &ipam.IPConfig{
			Gateway: d.conf.NodeAddress.IPv6Address.IP(),
			IP:      net.IPNet{IP: ipConf, Mask: addressing.ContainerIPv6Mask},
			Routes:  v6Routes,
		},
	}

	if d.ipamConf.IPv4Allocator != nil {
		ip4Conf, err := d.ipamConf.IPv4Allocator.AllocateNext()
		if err != nil {
			return nil, err
		}
		if d.ipamConf.IPv4Allocator != nil {
			ipamRep.IP4 = &ipam.IPConfig{
				Gateway: d.conf.NodeAddress.IPv4Address.IP(),
				IP:      net.IPNet{IP: ip4Conf, Mask: addressing.ContainerIPv4Mask},
				Routes:  v4Routes,
			}
		}
	}
	return ipamRep, nil
}