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.SetText(realip)
		}
	} else {
		me.ipmap[realip].ipv6 = false
		me.ipmap[realip].ipv4 = true
		if me.IPv4 != nil {
			me.IPv4.SetText(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.String() != s4 {
		log.Log(CHANGE, "IPv4 addresses have changed", s4)
		me.IPv4.SetText(s4)
		me.changed = true
	}
	if me.IPv6.String() != s6 {
		log.Log(CHANGE, "IPv6 addresses have changed", s6)
		me.IPv6.SetText(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.String()
	return strings.Split(tmp, "\n")
}

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

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