gofmt+fix ipv4 bare literal

jiphex · Aug 17, 2016 · cb4aeb9 · cb4aeb9
1 parent 4546675
commit cb4aeb9
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/ipgrep.go b/ipgrep.go
@@ -18,14 +18,18 @@ import (
 )
 
 func str2net(s string) (ipm *net.IPNet) {
-  if !strings.Contains(s, "/") {
-    s += "/32"
-  }
-  _,ipm,err := net.ParseCIDR(s)
-  if err != nil {
-    ipm = nil
-  }
-  return
+	if !strings.Contains(s, "/") {
+		if !strings.Contains(":") {
+			s += "/128"
+		} else {
+			s += "/32"
+		}
+	}
+	_, ipm, err := net.ParseCIDR(s)
+	if err != nil {
+		ipm = nil
+	}
+	return
 }
 
 type linelast struct {