netutil.go

//go:build linux || darwin
// +build linux darwin

// Package netutil provides methods for working with network
package netutil

// ////////////////////////////////////////////////////////////////////////////////// //
//                                                                                    //
//                         Copyright (c) 2022 ESSENTIAL KAOS                          //
//      Apache License, Version 2.0 <https://www.apache.org/licenses/LICENSE-2.0>     //
//                                                                                    //
// ////////////////////////////////////////////////////////////////////////////////// //

import (
	"net"
	"strings"
)

// ////////////////////////////////////////////////////////////////////////////////// //

// GetIP returns main IPv4 address
func GetIP() string {
	return getMainIP(false)
}

// GetIP6 returns main IPv6 address
func GetIP6() string {
	return getMainIP(true)
}

// ////////////////////////////////////////////////////////////////////////////////// //

func getMainIP(ipv6 bool) string {
	interfaces, err := net.Interfaces()

	if err != nil {
		return ""
	}

	defaultInterface := getDefaultRouteInterface()

	for i := len(interfaces) - 1; i >= 0; i-- {
		if defaultInterface != "" && interfaces[i].Name != defaultInterface {
			continue
		}

		// Ignore TUN/TAP interfaces
		switch {
		case strings.Contains(interfaces[i].Name, "tun"),
			strings.Contains(interfaces[i].Name, "tap"):
			continue
		}

		addr, err := interfaces[i].Addrs()

		if err != nil || len(addr) == 0 {
			continue
		}

		for _, a := range addr {
			ipnet, ok := a.(*net.IPNet)

			if ipnet.IP.IsLoopback() {
				continue
			}

			if ok && strings.Contains(ipnet.IP.String(), "::") == ipv6 {
				return ipnet.IP.String()
			}
		}
	}

	return ""
}