From 3a247bc72066ac3180cc60af18119bca3c7279d0 Mon Sep 17 00:00:00 2001
From: ZhouYixun <291028775@qq.com>
Date: Wed, 12 Oct 2022 00:23:00 +0800
Subject: [PATCH] feat: init ip2

---
 cmd/ip.go | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/cmd/ip.go b/cmd/ip.go
index 0eb201b..44fcdf9 100644
--- a/cmd/ip.go
+++ b/cmd/ip.go
@@ -51,6 +51,7 @@ type NetworkInfo struct {
 	IPv6 string
 }
 
+// GetNetworkIP @link https://github.com/danielpaulus/go-ios/blob/main/ios/pcap/ipfinder.go
 func GetNetworkIP(device giDevice.Device) error {
 	mac, err := device.GetValue("", "WiFiAddress")
 	if err != nil {
@@ -79,26 +80,24 @@ func GetNetworkIP(device giDevice.Device) error {
 
 func findIP(p []byte, info *NetworkInfo) error {
 	packet := gopacket.NewPacket(p, layers.LayerTypeEthernet, gopacket.Default)
-	// Get the TCP layer from this packet
 	if tcpLayer := packet.Layer(layers.LayerTypeEthernet); tcpLayer != nil {
 		tcp, _ := tcpLayer.(*layers.Ethernet)
 		if tcp.SrcMAC.String() == info.Mac {
 			for _, layer := range packet.Layers() {
-				log.Printf("layer:%s", layer.LayerType().String())
-
+				log.Printf("layer: %s", layer.LayerType().String())
 			}
 			if ipv4Layer := packet.Layer(layers.LayerTypeIPv4); ipv4Layer != nil {
 				ipv4, ok := ipv4Layer.(*layers.IPv4)
 				if ok {
 					info.IPv4 = ipv4.SrcIP.String()
-					log.Printf("ip4 found:%s", info.IPv4)
+					log.Printf("ip4 found: %s", info.IPv4)
 				}
 			}
 			if ipv6Layer := packet.Layer(layers.LayerTypeIPv6); ipv6Layer != nil {
 				ipv6, ok := ipv6Layer.(*layers.IPv6)
 				if ok {
 					info.IPv6 = ipv6.SrcIP.String()
-					log.Printf("ip6 found:%s", info.IPv6)
+					log.Printf("ip6 found: %s", info.IPv6)
 				}
 			}
 		}