func (s *IPAddressSuite) assertAddress( c *gc.C, ipAddr *state.IPAddress, addr network.Address, ipState state.AddressState, machineId, ifaceId, subnetId string, ) { c.Assert(ipAddr, gc.NotNil) c.Assert(ipAddr.MachineId(), gc.Equals, machineId) c.Assert(ipAddr.InterfaceId(), gc.Equals, ifaceId) c.Assert(ipAddr.SubnetId(), gc.Equals, subnetId) c.Assert(ipAddr.Value(), gc.Equals, addr.Value) c.Assert(ipAddr.Type(), gc.Equals, addr.Type) c.Assert(ipAddr.Scope(), gc.Equals, addr.Scope) c.Assert(ipAddr.State(), gc.Equals, ipState) c.Assert(ipAddr.Address(), jc.DeepEquals, addr) c.Assert(ipAddr.String(), gc.Equals, addr.String()) c.Assert(ipAddr.Id(), gc.Equals, s.State.EnvironUUID()+":"+addr.Value) c.Assert(ipAddr.InstanceId(), gc.Equals, instance.UnknownId) }
// allocateAddress tries to pick an address out of the given subnet and // allocates it to the container. func (p *ProvisionerAPI) allocateAddress( environ environs.NetworkingEnviron, subnet *state.Subnet, host, container *state.Machine, instId instance.Id, macAddress string, ) (*state.IPAddress, error) { hostname := containerHostname(container.Tag()) if !environs.AddressAllocationEnabled() { // Even if the address allocation feature flag is not enabled, we might // be running on MAAS 1.8+ with devices support, which we can use to // register containers getting IPs via DHCP. However, most of the usual // allocation code can be bypassed, we just need the parent instance ID // and a MAC address (no subnet or IP address). allocatedAddress := network.Address{} err := environ.AllocateAddress(instId, network.AnySubnet, &allocatedAddress, macAddress, hostname) if err != nil { // Not using MAAS 1.8+ or some other error. return nil, errors.Trace(err) } logger.Infof( "allocated address %q on instance %q for container %q", allocatedAddress.String(), instId, hostname, ) // Add the address to state, so we can look it up later by MAC address. stateAddr, err := p.st.AddIPAddress(allocatedAddress, string(network.AnySubnet)) if err != nil { return nil, errors.Annotatef(err, "failed to save address %q", allocatedAddress) } err = p.setAllocatedOrRelease(stateAddr, environ, instId, container, network.AnySubnet, macAddress) if err != nil { return nil, errors.Trace(err) } return stateAddr, nil } subnetId := network.Id(subnet.ProviderId()) for { addr, err := subnet.PickNewAddress() if err != nil { return nil, err } netAddr := addr.Address() logger.Tracef("picked new address %q on subnet %q", addr.String(), subnetId) // Attempt to allocate with environ. err = environ.AllocateAddress(instId, subnetId, &netAddr, macAddress, hostname) if err != nil { logger.Warningf( "allocating address %q on instance %q and subnet %q failed: %v (retrying)", addr.String(), instId, subnetId, err, ) // It's as good as unavailable for us, so mark it as // such. err = setAddrState(addr, state.AddressStateUnavailable) if err != nil { logger.Warningf( "cannot set address %q to %q: %v (ignoring and retrying)", addr.String(), state.AddressStateUnavailable, err, ) continue } logger.Tracef( "setting address %q to %q and retrying", addr.String(), state.AddressStateUnavailable, ) continue } logger.Infof( "allocated address %q on instance %q and subnet %q", addr.String(), instId, subnetId, ) err = p.setAllocatedOrRelease(addr, environ, instId, container, subnetId, macAddress) if err != nil { // Something went wrong - retry. continue } return addr, nil } }