Rework IP set

Graviton-One · Oct 8, 2020 · 483c4c1 · 483c4c1
1 parent 5c5d6d4
commit 483c4c1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/cmd/gravity/commands/ledger.go b/cmd/gravity/commands/ledger.go
@@ -185,26 +185,32 @@ var (
 		},
 	}
 )
-
 func getPublicIP() (string, error) {
 	ifaces, _ := net.Interfaces()
 
 	for _, i := range ifaces {
 		addrs, _ := i.Addrs()
 		for _, addr := range addrs {
 
+			var ip net.IP
 			switch v := addr.(type) {
+			case *net.IPNet:
+				ip = v.IP
 			case *net.IPAddr:
-				if strings.Contains(fmt.Sprintf("%v", v), "/24") {
-					return fmt.Sprintf("%v", v), nil
-				}
+				ip = v.IP
+			}
+
+			if strings.Contains(fmt.Sprintf("%v", addr), "/24") {
+				return fmt.Sprintf("%v", ip), nil
 			}
+
 		}
 	}
 
 	return "", fmt.Errorf("not found valid ip")
 }
 
+
 func initLedgerConfig(ctx *cli.Context) error {
 	var err error
 

diff --git a/cmd/gravity/gravity b/cmd/gravity/gravity