Initial commit after restructure

process/iphelper/get.go

@@ -0,0 +1,220 @@
+// +build windows
+
+package iphelper
+
+import (
+	"fmt"
+	"net"
+	"sync"
+)
+
+var (
+	tcp4Connections []*connectionEntry
+	tcp4Listeners   []*connectionEntry
+	tcp6Connections []*connectionEntry
+	tcp6Listeners   []*connectionEntry
+
+	udp4Connections []*connectionEntry
+	udp4Listeners   []*connectionEntry
+	udp6Connections []*connectionEntry
+	udp6Listeners   []*connectionEntry
+
+	ipHelper *IPHelper
+	lock     sync.RWMutex
+)
+
+func checkIPHelper() (err error) {
+	if ipHelper == nil {
+		ipHelper, err = New()
+		return err
+	}
+	return nil
+}
+
+func GetTCP4PacketInfo(localIP net.IP, localPort uint16, remoteIP net.IP, remotePort uint16, pktDirection bool) (pid int, direction bool, err error) {
+
+	// search
+	pid, direction = search(tcp4Connections, tcp4Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return
+	}
+
+	// if unable to find, refresh
+	lock.Lock()
+	err = checkIPHelper()
+	if err == nil {
+		tcp4Connections, tcp4Listeners, err = ipHelper.GetTables(TCP, IPv4)
+	}
+	lock.Unlock()
+	if err != nil {
+		return -1, direction, err
+	}
+
+	// search
+	pid, direction = search(tcp4Connections, tcp4Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return
+	}
+
+	return -1, direction, nil
+}
+
+func GetTCP6PacketInfo(localIP net.IP, localPort uint16, remoteIP net.IP, remotePort uint16, pktDirection bool) (pid int, direction bool, err error) {
+
+	// search
+	pid, direction = search(tcp6Connections, tcp6Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return
+	}
+
+	// if unable to find, refresh
+	lock.Lock()
+	err = checkIPHelper()
+	if err == nil {
+		tcp6Connections, tcp6Listeners, err = ipHelper.GetTables(TCP, IPv6)
+	}
+	lock.Unlock()
+	if err != nil {
+		return -1, direction, err
+	}
+
+	// search
+	pid, direction = search(tcp6Connections, tcp6Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return
+	}
+
+	return -1, direction, nil
+}
+
+func GetUDP4PacketInfo(localIP net.IP, localPort uint16, remoteIP net.IP, remotePort uint16, pktDirection bool) (pid int, direction bool, err error) {
+
+	// search
+	pid, _ = search(udp4Connections, udp4Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return pid, pktDirection, nil
+	}
+
+	// if unable to find, refresh
+	lock.Lock()
+	err = checkIPHelper()
+	if err == nil {
+		udp4Connections, udp4Listeners, err = ipHelper.GetTables(UDP, IPv4)
+	}
+	lock.Unlock()
+	if err != nil {
+		return -1, pktDirection, err
+	}
+
+	// search
+	pid, _ = search(udp4Connections, udp4Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return pid, pktDirection, nil
+	}
+
+	return -1, pktDirection, nil
+}
+
+func GetUDP6PacketInfo(localIP net.IP, localPort uint16, remoteIP net.IP, remotePort uint16, pktDirection bool) (pid int, direction bool, err error) {
+
+	// search
+	pid, _ = search(udp6Connections, udp6Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return pid, pktDirection, nil
+	}
+
+	// if unable to find, refresh
+	lock.Lock()
+	err = checkIPHelper()
+	if err == nil {
+		udp6Connections, udp6Listeners, err = ipHelper.GetTables(UDP, IPv6)
+	}
+	lock.Unlock()
+	if err != nil {
+		return -1, pktDirection, err
+	}
+
+	// search
+	pid, _ = search(udp6Connections, udp6Listeners, localIP, remoteIP, localPort, remotePort, pktDirection)
+	if pid >= 0 {
+		return pid, pktDirection, nil
+	}
+
+	return -1, pktDirection, nil
+}
+
+func search(connections, listeners []*connectionEntry, localIP, remoteIP net.IP, localPort, remotePort uint16, pktDirection bool) (pid int, direction bool) {
+	lock.RLock()
+	defer lock.RUnlock()
+
+	if pktDirection {
+		// inbound
+		pid = searchListeners(listeners, localIP, localPort)
+		if pid >= 0 {
+			return pid, true
+		}
+		pid = searchConnections(connections, localIP, remoteIP, localPort, remotePort)
+		if pid >= 0 {
+			return pid, false
+		}
+	} else {
+		// outbound
+		pid = searchConnections(connections, localIP, remoteIP, localPort, remotePort)
+		if pid >= 0 {
+			return pid, false
+		}
+		pid = searchListeners(listeners, localIP, localPort)
+		if pid >= 0 {
+			return pid, true
+		}
+	}
+
+	return -1, pktDirection
+}
+
+func searchConnections(list []*connectionEntry, localIP, remoteIP net.IP, localPort, remotePort uint16) (pid int) {
+
+	for _, entry := range list {
+		if localPort == entry.localPort &&
+			remotePort == entry.remotePort &&
+			remoteIP.Equal(entry.remoteIP) &&
+			localIP.Equal(entry.localIP) {
+			return entry.pid
+		}
+	}
+
+	return -1
+}
+
+func searchListeners(list []*connectionEntry, localIP net.IP, localPort uint16) (pid int) {
+
+	for _, entry := range list {
+		if localPort == entry.localPort &&
+			entry.localIP == nil || // nil IP means zero IP, see tables.go
+			localIP.Equal(entry.localIP) {
+			return entry.pid
+		}
+	}
+
+	return -1
+}
+
+func GetActiveConnectionIDs() (connections []string) {
+	lock.Lock()
+	defer lock.Unlock()
+
+	for _, entry := range tcp4Connections {
+		connections = append(connections, fmt.Sprintf("%d-%s-%d-%s-%d", TCP, entry.localIP, entry.localPort, entry.remoteIP, entry.remotePort))
+	}
+	for _, entry := range tcp6Connections {
+		connections = append(connections, fmt.Sprintf("%d-%s-%d-%s-%d", TCP, entry.localIP, entry.localPort, entry.remoteIP, entry.remotePort))
+	}
+	for _, entry := range udp4Connections {
+		connections = append(connections, fmt.Sprintf("%d-%s-%d-%s-%d", UDP, entry.localIP, entry.localPort, entry.remoteIP, entry.remotePort))
+	}
+	for _, entry := range udp6Connections {
+		connections = append(connections, fmt.Sprintf("%d-%s-%d-%s-%d", UDP, entry.localIP, entry.localPort, entry.remoteIP, entry.remotePort))
+	}
+
+	return
+}