fix: iface panic

#1130
MetaCubeX · Mar 24, 2024 · d2ae94f · d2ae94f
1 parent d56a439
commit d2ae94f
Showing 1 changed file with 17 additions and 8 deletions.
diff --git a/component/iface/iface.go b/component/iface/iface.go
@@ -40,16 +40,25 @@ func ResolveInterface(name string) (*Interface, error) {
 
 			ipNets := make([]netip.Prefix, 0, len(addrs))
 			for _, addr := range addrs {
-				ipNet := addr.(*net.IPNet)
-				ip, _ := netip.AddrFromSlice(ipNet.IP)
-
-				ones, bits := ipNet.Mask.Size()
-				if bits == 32 {
+				var pf netip.Prefix
+				switch addr.(type) {
+				case *net.IPNet:
+					ipNet := addr.(*net.IPNet)
+					ip, _ := netip.AddrFromSlice(ipNet.IP)
+					ones, bits := ipNet.Mask.Size()
+					if bits == 32 {
+						ip = ip.Unmap()
+					}
+					pf = netip.PrefixFrom(ip, ones)
+				case *net.IPAddr:
+					ipNet := addr.(*net.IPAddr)
+					ip, _ := netip.AddrFromSlice(ipNet.IP)
 					ip = ip.Unmap()
+					pf = netip.PrefixFrom(ip, ip.BitLen())
+				}
+				if pf.IsValid() {
+					ipNets = append(ipNets, pf)
 				}
-
-				pf := netip.PrefixFrom(ip, ones)
-				ipNets = append(ipNets, pf)
 			}
 
 			r[iface.Name] = &Interface{