To4 bug修复

2 files changed, 3 insertions, 3 deletions

diff --git a/src/ucr.edu/fakedns6/attack.go b/src/ucr.edu/fakedns6/attack.go
index e55d1a2..49d1c49 100644
--- a/src/ucr.edu/fakedns6/attack.go
+++ b/src/ucr.edu/fakedns6/attack.go
@@ -530,7 +530,7 @@ func Main(ifaceName string, authIPArg net.IP, resolverIPArg net.IP, resolverBack
 
 	//figure out MAC address
 	//test if it's in LAN first
-	dstMac, err := GetGatewayAddr(iface, handle, backendResolvers[0].resolverBackendIP.To4())
+	dstMac, err := GetGatewayAddr(iface, handle, backendResolvers[0].resolverBackendIP.To16())
 	if err == nil {
 		ethernetLayer = &layers.Ethernet{
 			SrcMAC: iface.HardwareAddr,
diff --git a/src/ucr.edu/fakedns6/library.go b/src/ucr.edu/fakedns6/library.go
index 4697e12..edc4548 100644
--- a/src/ucr.edu/fakedns6/library.go
+++ b/src/ucr.edu/fakedns6/library.go
@@ -64,8 +64,8 @@ func GetIfaceAddr(iface *net.Interface) (net.IP, error) {
 	var srcIP net.IP
 	for _, address := range addrs {
 		if ipnet, ok := address.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
-			if ipnet.IP.To4() != nil {
-				srcIP = ipnet.IP.To4()
+			if ipnet.IP.To16() != nil {
+				srcIP = ipnet.IP.To16()
 				break
 			}
 		}