github.com/Cloud-Foundations/Dominator@v0.3.4/hypervisor/manager/subnets.go

package manager

import (
	"flag"
	"fmt"
	"net"
	"os"
	"path"
	"sort"

	"github.com/Cloud-Foundations/Dominator/lib/fsutil"
	"github.com/Cloud-Foundations/Dominator/lib/json"
	"github.com/Cloud-Foundations/Dominator/lib/net/util"
	"github.com/Cloud-Foundations/Dominator/lib/srpc"
	proto "github.com/Cloud-Foundations/Dominator/proto/hypervisor"
)

var (
	addHypervisorSubnet = flag.Bool("addHypervisorSubnet", false,
		"If true, automatically add the subnet for the default gateway (deprecated)")
)

func checkSubnetAccess(subnet proto.Subnet,
	authInfo *srpc.AuthInformation) bool {
	if authInfo.HaveMethodAccess {
		return true
	}
	if len(subnet.AllowedUsers) < 1 && len(subnet.AllowedGroups) < 1 {
		return true
	}
	for _, allowedUser := range subnet.AllowedUsers {
		if authInfo.Username == allowedUser {
			return true
		}
	}
	for _, allowedGroup := range subnet.AllowedGroups {
		if _, ok := authInfo.GroupList[allowedGroup]; ok {
			return true
		}
	}
	return false
}

func getHypervisorSubnet() (proto.Subnet, error) {
	defaultRoute, err := util.GetDefaultRoute()
	if err != nil {
		return proto.Subnet{}, err
	}
	resolverConfig, err := util.GetResolverConfiguration()
	if err != nil {
		return proto.Subnet{}, err
	}
	myIP, err := util.GetMyIP()
	if err != nil {
		return proto.Subnet{}, err
	}
	nameservers := make([]net.IP, 0, len(resolverConfig.Nameservers))
	for _, nameserver := range resolverConfig.Nameservers {
		if nameserver[0] == 127 {
			nameservers = append(nameservers, myIP)
		} else {
			nameservers = append(nameservers, nameserver)
		}
	}
	return proto.Subnet{
		Id:                "hypervisor",
		IpGateway:         defaultRoute.Address,
		IpMask:            net.IP(defaultRoute.Mask),
		DomainNameServers: nameservers,
	}, nil
}

func (m *Manager) getBridgeForSubnet(subnet proto.Subnet) (
	string, string, error) {
	mapName := fmt.Sprintf("br@%s", subnet.Id)
	if _, ok := m.BridgeMap[mapName]; ok {
		return mapName, "", nil
	} else if bridge, ok := m.VlanIdToBridge[subnet.VlanId]; ok {
		return bridge, "", nil
	} else if bridge, ok = m.VlanIdToBridge[0]; ok {
		return bridge, fmt.Sprintf(",vlan=%d", subnet.VlanId), nil
	} else {
		return "", "", fmt.Errorf("no usable bridge")
	}
}

// This must be called with the lock held.
func (m *Manager) getMatchingSubnet(ipAddr net.IP) string {
	if len(ipAddr) > 0 {
		for id, subnet := range m.subnets {
			subnetMask := net.IPMask(subnet.IpMask)
			subnetAddr := subnet.IpGateway.Mask(subnetMask)
			if ipAddr.Mask(subnetMask).Equal(subnetAddr) {
				return id
			}
		}
	}
	return ""
}

// This must be called with the lock held.
func (m *Manager) getSubnetAndAuth(subnetId string,
	authInfo *srpc.AuthInformation) (proto.Subnet, error) {
	if len(m.subnets) < 1 {
		return proto.Subnet{}, fmt.Errorf("no subnets exist")
	}
	if subnetId != "" {
		if subnet, ok := m.subnets[subnetId]; !ok {
			return proto.Subnet{}, fmt.Errorf("subnet: %s does not exist",
				subnetId)
		} else if !checkSubnetAccess(subnet, authInfo) {
			return proto.Subnet{}, fmt.Errorf("no access to subnet: %s",
				subnetId)
		} else {
			return subnet, nil
		}
	}
	// Unspecified subnet: try to find one.
	if len(m.subnets) == 1 {
		if subnet, ok := m.subnets["hypervisor"]; !ok {
			return proto.Subnet{}, fmt.Errorf(
				"hypervisor subnet does not exist")
		} else {
			return subnet, nil
		}
	}
	subnetsPermitted := make([]string, 0, 1)
	for _, subnet := range m.subnets {
		if subnet.Id == "hypervisor" {
			continue
		}
		if !checkSubnetAccess(subnet, authInfo) {
			continue
		}
		if len(subnetsPermitted) > 0 {
			return proto.Subnet{}, fmt.Errorf(
				"multiple available subnets: pick one")
		}
		subnetsPermitted = append(subnetsPermitted, subnet.Id)
	}
	if len(subnetsPermitted) < 1 {
		return proto.Subnet{}, fmt.Errorf("no subnets permitted")
	}
	return m.subnets[subnetsPermitted[0]], nil
}

func (m *Manager) listSubnets(doSort bool) []proto.Subnet {
	m.mutex.Lock()
	defer m.mutex.Unlock()
	subnets := make([]proto.Subnet, 0, len(m.subnets))
	if !doSort {
		for _, subnet := range m.subnets {
			subnets = append(subnets, subnet)
		}
		return subnets
	}
	subnetIDs := make([]string, 0, len(m.subnets))
	for subnetID := range m.subnets {
		subnetIDs = append(subnetIDs, subnetID)
	}
	sort.Strings(subnetIDs)
	for _, subnetID := range subnetIDs {
		subnets = append(subnets, m.subnets[subnetID])
	}
	return subnets
}

// This returns with the Manager locked, waiting for existing subnets to be
// drained from the channel by the caller before unlocking.
func (m *Manager) makeSubnetChannel() <-chan proto.Subnet {
	ch := make(chan proto.Subnet, 1)
	m.mutex.Lock()
	m.subnetChannels = append(m.subnetChannels, ch)
	go func() {
		defer m.mutex.Unlock()
		for _, subnet := range m.subnets {
			ch <- subnet
		}
	}()
	return ch
}

func (m *Manager) loadSubnets() error {
	var subnets []proto.Subnet
	err := json.ReadFromFile(path.Join(m.StateDir, "subnets.json"), &subnets)
	if err != nil && !os.IsNotExist(err) {
		return err
	}
	for index := range subnets {
		subnets[index].Shrink()
	}
	m.subnets = make(map[string]proto.Subnet, len(subnets)+1)
	for _, subnet := range subnets {
		m.subnets[subnet.Id] = subnet
	}
	if *addHypervisorSubnet {
		if subnet, err := getHypervisorSubnet(); err != nil {
			return err
		} else {
			m.subnets["hypervisor"] = subnet
		}
	}
	for _, subnet := range m.subnets {
		m.DhcpServer.AddSubnet(subnet)
	}
	return nil
}

func (m *Manager) updateSubnets(request proto.UpdateSubnetsRequest) error {
	for index, subnet := range request.Add {
		if subnet.Id == "hypervisor" {
			return fmt.Errorf("cannot add hypervisor subnet")
		}
		request.Add[index].Shrink()
	}
	for index, subnet := range request.Change {
		if subnet.Id == "hypervisor" {
			return fmt.Errorf("cannot change hypervisor subnet")
		}
		request.Change[index].Shrink()
	}
	for _, subnetId := range request.Delete {
		if subnetId == "hypervisor" {
			return fmt.Errorf("cannot delete hypervisor subnet")
		}
	}
	if err := m.updateSubnetsLocked(request); err != nil {
		return err
	}
	for _, subnet := range request.Add {
		m.DhcpServer.AddSubnet(subnet)
		for _, ch := range m.subnetChannels {
			ch <- subnet
		}
	}
	for _, subnet := range request.Change {
		m.DhcpServer.RemoveSubnet(subnet.Id)
		m.DhcpServer.AddSubnet(subnet)
		// TOOO(rgooch): Design a clean way to send updates to the channels.
	}
	for _, subnetId := range request.Delete {
		m.DhcpServer.RemoveSubnet(subnetId)
		// TOOO(rgooch): Design a clean way to send deletes to the channels.
	}
	return nil
}

func (m *Manager) updateSubnetsLocked(
	request proto.UpdateSubnetsRequest) error {
	m.mutex.Lock()
	defer m.mutex.Unlock()
	for _, subnet := range request.Add {
		if _, ok := m.subnets[subnet.Id]; ok {
			return fmt.Errorf("subnet: %s already exists", subnet.Id)
		}
	}
	for _, subnet := range request.Change {
		if _, ok := m.subnets[subnet.Id]; !ok {
			return fmt.Errorf("subnet: %s does not exist", subnet.Id)
		}
	}
	for _, subnetId := range request.Delete {
		if _, ok := m.subnets[subnetId]; !ok {
			return fmt.Errorf("subnet: %s does not exist", subnetId)
		}
	}
	for _, subnet := range request.Add {
		m.subnets[subnet.Id] = subnet
	}
	for _, subnet := range request.Change {
		m.subnets[subnet.Id] = subnet
	}
	for _, subnetId := range request.Delete {
		delete(m.subnets, subnetId)
	}
	subnetsToWrite := make([]proto.Subnet, 0, len(m.subnets)-1)
	for _, subnet := range m.subnets {
		if subnet.Id != "hypervisor" {
			subnetsToWrite = append(subnetsToWrite, subnet)
		}
	}
	// TODO(rgooch): Should precompute the numFreeAddresses map.
	numFreeAddresses, err := m.computeNumFreeAddressesMap(m.addressPool)
	if err != nil {
		return err
	}
	err = json.WriteToFile(path.Join(m.StateDir, "subnets.json"),
		fsutil.PublicFilePerms, "    ", subnetsToWrite)
	if err != nil {
		return err
	}
	// Make a copy, since these will be sent over a channel.
	registered := make([]proto.Address, 0, len(m.addressPool.Registered))
	for _, addr := range m.addressPool.Registered {
		registered = append(registered, addr)
	}
	m.sendUpdate(
		proto.Update{
			HaveAddressPool:  true,
			AddressPool:      registered,
			NumFreeAddresses: numFreeAddresses,
			HaveSubnets:      true,
			Subnets:          subnetsToWrite,
		})
	return nil
}