safing-portmaster/network/iphelper/get.go

// +build windows

package iphelper

import (
	"sync"

	"github.com/safing/portmaster/network/socket"
)

var (
	ipHelper *IPHelper

	// lock locks access to the whole DLL.
	// TODO: It's unproven if we can access the iphlpapi.dll concurrently, especially as we might be encountering various versions of the DLL. In the future, we could possibly investigate and improve performance here.
	lock sync.RWMutex
)

// GetTCP4Table returns the system table for IPv4 TCP activity.
func GetTCP4Table() (connections []*socket.ConnectionInfo, listeners []*socket.BindInfo, err error) {
	lock.Lock()
	defer lock.Unlock()

	err = checkIPHelper()
	if err != nil {
		return nil, nil, err
	}

	return ipHelper.getTable(IPv4, TCP)
}

// GetTCP6Table returns the system table for IPv6 TCP activity.
func GetTCP6Table() (connections []*socket.ConnectionInfo, listeners []*socket.BindInfo, err error) {
	lock.Lock()
	defer lock.Unlock()

	err = checkIPHelper()
	if err != nil {
		return nil, nil, err
	}

	return ipHelper.getTable(IPv6, TCP)
}

// GetUDP4Table returns the system table for IPv4 UDP activity.
func GetUDP4Table() (binds []*socket.BindInfo, err error) {
	lock.Lock()
	defer lock.Unlock()

	err = checkIPHelper()
	if err != nil {
		return nil, err
	}

	_, binds, err = ipHelper.getTable(IPv4, UDP)
	return
}

// GetUDP6Table returns the system table for IPv6 UDP activity.
func GetUDP6Table() (binds []*socket.BindInfo, err error) {
	lock.Lock()
	defer lock.Unlock()

	err = checkIPHelper()
	if err != nil {
		return nil, err
	}

	_, binds, err = ipHelper.getTable(IPv6, UDP)
	return
}