scalebox/golang/misc/net.go

84 lines
1.8 KiB
Go

package misc
import (
"fmt"
"net"
"os"
"os/exec"
"regexp"
"runtime"
"strings"
"github.com/sirupsen/logrus"
)
// GetLocalIP ...
// hostname -i / -I
func GetLocalIP() string {
// 0=1, 2, 3, .. ,n
localIPIndexStr := os.Getenv("LOCAL_IP_INDEX")
var cmd string
if localIPIndexStr == "" {
cmd = "hostname -i"
} else if runtime.GOOS == "darwin" { // macos
// 172.*.*.* are used for out-of-band Network/container network
cmd = `ifconfig|grep "inet "|grep -v 127.0.0.1|grep -v "inet 172."|head -1|cut -d ' ' -f 2`
} else { // linux
cmd = fmt.Sprintf("hostname -I | awk '{print $%s}'", localIPIndexStr)
}
out, _ := exec.Command("/bin/bash", "-c", cmd).Output()
localIP := strings.Split(string(out), " ")[0]
// remove '\n'
localIP = strings.Replace(localIP, "\n", "", -1)
reIPv4 := regexp.MustCompile("^([0-9]+\\.){3}[0-9]+$")
if !reIPv4.MatchString(localIP) {
logrus.Warnf("error get_local_ip, localIP=%s\n", localIP)
}
return localIP
}
// GetLocalIPv4Addr ...
// @Deprecated
func getLocalIPv4Addr() string {
addrs, err := net.InterfaceAddrs()
if err != nil {
fmt.Println(err)
}
var ip string
for _, address := range addrs {
if ipnet, ok := address.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
if ipnet.IP.To4() != nil {
ip = ipnet.IP.String()
}
}
}
return ip
}
// GetLocalIPv4AddrByInterface ...
// @Deprecated
func getLocalIPv4AddrByInterface(interfaceName string) string {
interfaces, err := net.Interfaces()
if err != nil {
panic(err)
}
var ip string
for _, interf := range interfaces {
if interf.Name == "en7" {
addrs, err := interf.Addrs()
if err != nil {
panic(err)
}
for _, add := range addrs {
if ipnet, ok := add.(*net.IPNet); ok {
if ipnet.IP.To4() != nil {
ip = ipnet.IP.String()
}
}
}
}
}
return ip
}