hostname/net.go

// This creates a simple hello world window
package linuxstatus

import 	(
	// "log"
	"net"
	"sort"
	"strings"

	"go.wit.com/log"
)

func IsIPv6(address string) bool {
	return strings.Count(address, ":") >= 2
}

func (t *IPtype) IsReal() bool {
	if (t.ip.IsPrivate() || t.ip.IsLoopback() || t.ip.IsLinkLocalUnicast()) {
		log.Log(NET, "\t\tIP is Real = false")
		return false
	} else {
		log.Log(NET, "\t\tIP is Real = true")
		return true
	}
}

func IsReal(ip *net.IP) bool {
	if (ip.IsPrivate() || ip.IsLoopback() || ip.IsLinkLocalUnicast()) {
		log.Log(NET, "\t\tIP is Real = false")
		return false
	} else {
		log.Log(NET, "\t\tIP is Real = true")
		return true
	}
}

func renameInterface(i *net.Interface) {
	/*
	/sbin/ip link set eth1 down
	/sbin/ip link set eth1 name eth123
	/sbin/ip link set eth123 up
	*/
}

// Will figure out if an interface was just added
func checkInterface(i net.Interface) {
	val, ok := me.ifmap[i.Index]
	if ! ok {
		log.Log(INFO, i.Name, "is a new network interface. The linux kernel index =", i.Index)
		me.ifmap[i.Index] = new(IFtype)
		me.ifmap[i.Index].gone = false
		me.ifmap[i.Index].iface = &i
		me.changed = true
		if (me.Interfaces != nil) {
			me.Interfaces.AddText(i.Name)
			me.Interfaces.SetText(i.Name)
		}
		return
	}
	me.ifmap[i.Index].gone = false
	log.Log(NET, "me.ifmap[i] does exist. Need to compare everything.", i.Index, i.Name, val.iface.Index, val.iface.Name)
	if (val.iface.Name != i.Name) {
		log.Log(INFO, val.iface.Name, "has changed to it's name to", i.Name)
		me.ifmap[i.Index].iface = &i
		me.changed = true
		if (me.Interfaces != nil) {
			me.Interfaces.AddText(i.Name)
			me.Interfaces.SetText(i.Name)
		}
		return
	}
}

/*
	These are the real IP address you have been
	given from DHCP
*/
func realAAAA() []string {
	var aaaa []string

	for s, t := range me.ipmap {
		if (t.IsReal()) {
			if (t.ipv6) {
				aaaa = append(aaaa, s)
			}
		}
	}
	return aaaa
}

func realA() []string {
	var a []string

	for s, t := range me.ipmap {
		if (t.IsReal()) {
			if (t.ipv4) {
				a = append(a, s)
			}
		}
	}
	return a
}

func checkDNS() (map[string]*IPtype, map[string]*IPtype) {
	var ipv4s map[string]*IPtype
	var ipv6s map[string]*IPtype

	ipv4s = make(map[string]*IPtype)
	ipv6s = make(map[string]*IPtype)

	for s, t := range me.ipmap {
		i := t.iface
		ipt := "IPv4"
		if (t.ipv6) {
			ipt = "IPv6"
		}
		if (t.IsReal()) {
			log.Log(INFO, "\tIP is Real    ", ipt, i.Index, i.Name, s)
			if (t.ipv6) {
				ipv6s[s] = t
			} else {
				ipv4s[s] = t
			}
		} else {
			log.Log(INFO, "\tIP is not Real", ipt, i.Index, i.Name, s)
		}
	}
	return ipv6s, ipv4s
}

// Will figure out if an IP address is new
func checkIP(ip *net.IPNet, i net.Interface) bool {
	log.Log(NET, "\t\taddr.(type) = *net.IPNet")
	log.Log(NET, "\t\taddr.(type) =", ip)
	var realip string
	realip = ip.IP.String()

	val, ok := me.ipmap[realip]
	if ok {
		log.Log(NET, val.ipnet.IP.String(), "is already a defined IP address")
		me.ipmap[realip].gone = false
		return false
	}

	me.ipmap[realip] = new(IPtype)
	me.ipmap[realip].gone = false
	me.ipmap[realip].ipv4 = true
	me.ipmap[realip].ipnet = ip
	me.ipmap[realip].ip = ip.IP
	me.ipmap[realip].iface = &i
	t := "IPv4"
	if (IsIPv6(ip.String())) {
		me.ipmap[realip].ipv6 = true
		me.ipmap[realip].ipv4 = false
		t = "IPv6"
		if (me.IPv6 != nil) {
			me.IPv6.Set(realip)
		}
	} else {
		me.ipmap[realip].ipv6 = false
		me.ipmap[realip].ipv4 = true
		if (me.IPv4 != nil) {
			me.IPv4.Set(realip)
		}
	}
	if (IsReal(&ip.IP)) {
		log.Log(INFO, "\tIP is Real    ", t, i.Index, i.Name, realip)
	} else {
		log.Log(INFO, "\tIP is not Real", t, i.Index, i.Name, realip)
	}
	log.Log(NET, "\t\tIP is IsPrivate() =", ip.IP.IsPrivate())
	log.Log(NET, "\t\tIP is IsLoopback() =", ip.IP.IsLoopback())
	log.Log(NET, "\t\tIP is IsLinkLocalUnicast() =", ip.IP.IsLinkLocalUnicast())
	// log.Log(INFO, "HERE HERE", "realip =", realip, "me.ip[realip]=", me.ipmap[realip])
	return true
}

func scanInterfaces() {
	log.Log(NET, "scanInterfaces() START")
	ifaces, _ := net.Interfaces()
	// me.ifnew = ifaces
	log.Log(NET, SPEW, ifaces)
	for _, i := range ifaces {
		addrs, _ := i.Addrs()
		// log.Log(INFO, "range ifaces = ", i)
		checkInterface(i)
		log.Log(NET, "*net.Interface.Name = ", i.Name, i.Index)
		log.Log(NET, SPEW, i)
		log.Log(NET, SPEW, addrs)
		for _, addr := range addrs {
			log.Log(NET, "\taddr =", addr)
			log.Log(NET, SPEW, addrs)
			ips, _ := net.LookupIP(addr.String())
			log.Log(NET, "\tLookupIP(addr) =", ips)
			switch v := addr.(type) {
			case *net.IPNet:
				if checkIP(v, i) {
					log.Log(NET, "scanInterfaces() IP is new () i =", v.IP.String())
				}
			default:
				log.Log(NET, "\t\taddr.(type) = NO IDEA WHAT TO DO HERE v =", v)
			}
		}
	}
	if deleteChanges() {
		me.changed = true
		log.Log(NET, "deleteChanges() detected network changes")
	}
	updateRealAAAA()
	log.Log(NET, "scanInterfaces() END")
}

// displays the IP address found on your network interfaces
func updateRealAAAA() {
	var all4 []string
	var all6 []string
	for s, t := range me.ipmap {
		if (t.ipv4) {
			all4 = append(all4, s)
			log.Log(NET, "IPv4 =", s)
		} else if (t.ipv6) {
			all6 = append(all6, s)
			log.Log(NET, "IPv6 =", s)
		} else {
			log.Log(NET, "???? =", s)
		}
	}

	// sort and create text
	sort.Strings(all4)
	sort.Strings(all6)
	s4 := strings.Join(all4, "\n")
	s6 := strings.Join(all6, "\n")

	if (me.IPv4.Get() != s4) {
		log.Log(CHANGE, "IPv4 addresses have changed", s4)
		me.IPv4.Set(s4)
		me.changed = true
	}
	if (me.IPv6.Get() != s6) {
		log.Log(CHANGE, "IPv6 addresses have changed", s6)
		me.IPv6.Set(s6)
		me.changed = true
	}
}

// delete network interfaces and ip addresses from the gui
func deleteChanges() bool {
	var changed bool = false
	for i, t := range me.ifmap {
		if (t.gone) {
			log.Log(CHANGE, "DELETE int =", i, "name =", t.name, t.iface)
			delete(me.ifmap, i)
			changed = true
		}
		t.gone = true
	}
	for s, t := range me.ipmap {
		if (t.gone) {
			log.Log(CHANGE, "DELETE name =", s, "IPv4 =", t.ipv4)
			log.Log(CHANGE, "DELETE name =", s, "IPv6 =", t.ipv6)
			log.Log(CHANGE, "DELETE name =", s, "iface =", t.iface)
			log.Log(CHANGE, "DELETE name =", s, "ip =", t.ip)
			delete(me.ipmap, s)
			changed = true
		}
		t.gone = true
	}

	return changed
}

func (ls *LinuxStatus) GetIPv6() []string {
	if ! me.Ready() {return nil}
	tmp := me.workingIPv6.Get()
	return strings.Split(tmp, "\n")
}

func (ls *LinuxStatus) GetIPv4() []string {
	if ! me.Ready() {return nil}
	tmp := me.workingIPv4.Get()
	return strings.Split(tmp, "\n")
}

func (ls *LinuxStatus) GetNameservers() []string {
	if ! me.Ready() {return nil}
	tmp := me.resolver.Get()
	return strings.Split(tmp, "\n")
}