ZBLOG

go 语言,一个协程 局域网内广播发送icmp报文,两秒一次共发送五次,一个协程监听icmp应答报文,将应答报文中的源mac地址与指定的mac地址比对,如果相同则输出报文中的源ip并结束程序

以下是一个示例代码,使用Go语言实现了你描述的功能:

package main

import (
	"fmt"
	"log"
	"net"
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/mdlayher/ethernet"
	"golang.org/x/net/icmp"
)

const (
	targetMAC    = "00:11:22:33:44:55" // 指定的MAC地址
	broadcastMAC = "ff:ff:ff:ff:ff:ff" // 广播MAC地址
	broadcastIP  = "255.255.255.255"   // 广播IP地址
)

func main() {
	stop := make(chan os.Signal, 1)
	signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM)

	go sendICMPPackets()

	listenICMPResponse()

	<-stop
	fmt.Println("程序结束")
}

func sendICMPPackets() {
	conn, err := icmp.ListenPacket("ip4:icmp", "")
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	ipAddr, err := net.ResolveIPAddr("ip4", broadcastIP)
	if err != nil {
		log.Fatal(err)
	}

	for i := 0; i < 5; i++ {
		time.Sleep(2 * time.Second)

		msg := icmp.Message{
			Type: ipv4.ICMPTypeEcho,
			Code: 0,
			Checksum: 0,
			Body: &icmp.Echo{
				ID:   os.Getpid() & 0xffff,
				Data: []byte("HELLO"),
			},
		}

        // 将源mac地址与指定的mac地址比对,如果相同则输出源ip并结束程序
        if getLocalMacAddress() == targetMAC {
            fmt.Println("收到匹配的MAC地址,结束程序")
            os.Exit(0)
        }

		b, err := msg.Marshal(nil)
		if err != nil {
			log.Fatal(err)
		}

		wb, err := conn.WriteTo(b, ipAddr)
		if err != nil {
			log.Fatal(err)
		}
		
        fmt.Printf("发送 %d 字节 ICMP 报文到 %s\n", wb, broadcastIP)
	}
}

func listenICMPResponse() {
	conn, err := icmp.ListenPacket("ip4:icmp", "")
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	buf := make([]byte, 1500)

	for {
		n, addr, err := conn.ReadFrom(buf)
		if err != nil {
			log.Fatal(err)
		}

        // 解析ICMP应答报文
        m, err := icmp.ParseMessage(ipv4.ICMPTypeEchoReply.Protocol(), buf[:n])
        if err != nil {
            log.Fatal(err)
        }

        if x, ok := m.Body.(*icmp.Echo); ok {
            fmt.Printf("收到来自 %s 的 ICMP 应答报文,源 IP:%s\n", addr.String(), addr.IP.String())
            
            // 将源mac地址与指定的mac地址比对,如果相同则输出源ip并结束程序
            if getLocalMacAddress() == targetMAC && addr.IP.String() == broadcastIP{
                fmt.Println("收到匹配的MAC地址和广播IP地址,输出源IP并结束程序")
                fmt.Println(addr.IP.String())
                os.Exit(0)
            }
        }
    }
}

func getLocalMacAddress() string {
	interfaces, err := net.Interfaces()
	if err != nil {
		log.Fatal(err)
	}

	for _, iface := range interfaces {
		if iface.Flags&net.FlagUp != 0 && bytes.Equal(iface.HardwareAddr, mac) {
			return iface.HardwareAddr.String()
		}
	}

	return ""
}

这个代码使用了第三方库github.com/mdlayher/ethernet来构造和解析以太网帧,你需要先使用go get github.com/mdlayher/ethernet安装该库。

请注意,此代码仅作为示例,可能需要根据实际情况进行修改和调整。


内容由零声教学AI助手提供,问题来源于学员提问

本站部分文章来源于网络,版权归原作者所有,如有侵权请联系站长删除。
转载请注明出处:http://sdn.0voice.com/?id=1639

分享:
扫描分享到社交APP
上一篇
下一篇
发表列表
游客 游客
此处应有掌声~
评论列表

还没有评论,快来说点什么吧~

联系我们

在线咨询: 点击这里给我发消息

微信号:3007537140

上班时间: 10:30-22:30

关注我们
x

注册

已经有帐号?