126 lines
2.9 KiB
Go
126 lines
2.9 KiB
Go
/*
|
|
Copyright 2019 Stellar Project
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
this software and associated documentation files (the "Software"), to deal in the
|
|
Software without restriction, including without limitation the rights to use, copy,
|
|
modify, merge, publish, distribute, sublicense, and/or sell copies of the Software,
|
|
and to permit persons to whom the Software is furnished to do so, subject to the
|
|
following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all copies
|
|
or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
|
|
INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR
|
|
PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE
|
|
FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
|
|
TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
|
|
USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
*/
|
|
|
|
package heimdall
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"sort"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// NodeID returns a unique local node ID
|
|
func NodeID() string {
|
|
ifaces, err := net.Interfaces()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// sort
|
|
sort.SliceStable(ifaces, func(i, j int) bool { return ifaces[i].Name < ifaces[j].Name })
|
|
|
|
var i net.Interface
|
|
for _, iface := range ifaces {
|
|
if iface.Flags&net.FlagLoopback != 0 || iface.Flags&net.FlagPointToPoint != 0 {
|
|
continue
|
|
}
|
|
|
|
i = iface
|
|
break
|
|
}
|
|
|
|
s := uuid.NewSHA1(uuid.Nil, i.HardwareAddr)
|
|
return s.String()
|
|
}
|
|
|
|
// GetIP returns the first non-local IP address for the system
|
|
func GetIP() string {
|
|
ip := "127.0.0.1"
|
|
ifaces, err := net.Interfaces()
|
|
if err != nil {
|
|
logrus.Warnf("unable to detect network interfaces")
|
|
return ip
|
|
}
|
|
for _, i := range ifaces {
|
|
a := getInterfaceIP(i)
|
|
if a != "" {
|
|
return a
|
|
}
|
|
}
|
|
|
|
return ip
|
|
}
|
|
|
|
func getInterfaceIP(iface net.Interface) string {
|
|
addrs, err := iface.Addrs()
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
for _, addr := range addrs {
|
|
var ip net.IP
|
|
switch v := addr.(type) {
|
|
case *net.IPNet:
|
|
ip = v.IP
|
|
case *net.IPAddr:
|
|
ip = v.IP
|
|
}
|
|
// skip loopback
|
|
if ip.IsLoopback() {
|
|
return ""
|
|
}
|
|
if ip.To4() == nil {
|
|
return ""
|
|
}
|
|
return ip.To4().String()
|
|
}
|
|
|
|
return ""
|
|
}
|
|
|
|
// HashData returns a sha256 sum of the specified data
|
|
func HashData(data []byte) string {
|
|
h := sha256.New()
|
|
h.Write(data)
|
|
return fmt.Sprintf("%x", h.Sum(nil))
|
|
}
|
|
|
|
// HashConfig returns the sha256 hash of the specified path
|
|
func HashConfig(cfgPath string) (string, error) {
|
|
if _, err := os.Stat(cfgPath); err != nil {
|
|
if os.IsNotExist(err) {
|
|
return "", nil
|
|
}
|
|
return "", err
|
|
}
|
|
peerData, err := ioutil.ReadFile(cfgPath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return HashData(peerData), nil
|
|
}
|