Golang DHCP 中继器：使用 gopacket 库转发 DHCP 请求和 Offer

这是一个使用 Golang 和 gopacket 库实现的 DHCP 中继器程序，可以将捕获的 DHCP 请求包转发给指定的 DHCP 服务器，并将收到的 DHCP Offer 响应发送给指定的网络接口。

package main

import (
	"fmt"
	"log"
	"net"

	"github.com/google/gopacket"
	"github.com/google/gopacket/layers"
	"github.com/google/gopacket/pcap"
)

const (
	iface   = "eth0"              // 网络接口名
	filter  = "udp and port 67"   // 过滤器，只捕获 UDP 67 端口的数据包
	snaplen = int32(65535)        // 指定捕获数据包的最大长度
	timeout = pcap.BlockForever   // 捕获数据包的超时时间
	dhcpIP  = "192.168.199.1"     // DHCP 服务器 IP
)

func main() {
	// 打开网络接口
	handle, err := pcap.OpenLive(iface, snaplen, true, timeout)
	if err != nil {
		log.Fatal(err)
	}
	defer handle.Close()

	// 设置过滤器
	if err := handle.SetBPFFilter(filter); err != nil {
		log.Fatal(err)
	}

	// 开始捕获数据包
	packetSource := gopacket.NewPacketSource(handle, handle.LinkType())
	for packet := range packetSource.Packets() {
		// 从数据包中解析出 UDP 层和 DHCP 层
		udpLayer := packet.Layer(layers.LayerTypeUDP)
		if udpLayer == nil {
			continue
		}
		udp, _ := udpLayer.(*layers.UDP)
		dhcpLayer := packet.Layer(layers.LayerTypeDHCPv4)
		if dhcpLayer == nil {
			continue
		}
		dhcp, _ := dhcpLayer.(*layers.DHCPv4)

		// 判断是否为 DHCP 请求包
		if dhcp.OpCode != layers.DHCPOpRequest {
			continue
		}

		// 构造 DHCP 消息
		msg := layers.DHCPv4{
			OpCode:      layers.DHCPOpRequest,
			HardwareLen: dhcp.HardwareLen,
			HardwareType: dhcp.HardwareType,
			ClientHWAddr: dhcp.ClientHWAddr,
			TransactionID: dhcp.TransactionID,
			Options: dhcp.Options,
		}

		// 发送 DHCP 请求包
		if err := sendPacket(msg, net.IPv4bcast); err != nil {
			log.Println(err)
			continue
		}

		// 等待 DHCP Offer 响应，超时时间为 5 秒
		packet, err := handle.NextPacketTimeout(5e9)
		if err != nil {
			log.Println(err)
			continue
		}

		// 解析 DHCP Offer 响应
		udpLayer = packet.Layer(layers.LayerTypeUDP)
		if udpLayer == nil {
			continue
		}
		udp, _ = udpLayer.(*layers.UDP)
		dhcpLayer = packet.Layer(layers.LayerTypeDHCPv4)
		if dhcpLayer == nil {
			continue
		}
		dhcp, _ = dhcpLayer.(*layers.DHCPv4)

		// 判断是否为 DHCP Offer 响应包
		if dhcp.OpCode != layers.DHCPOpReply || dhcp.MessageType != layers.DHCPMsgTypeOffer {
			continue
		}

		// 发送 DHCP Offer 响应
		if err := sendPacket(*dhcp, udp.SrcIP); err != nil {
			log.Println(err)
			continue
		}
	}
}

// 发送数据包
func sendPacket(dhcp layers.DHCPv4, dstIP net.IP) error {
	// 构造 UDP 层
	udp := layers.UDP{
		SrcPort: layers.UDPPort(68),
		DstPort: layers.UDPPort(67),
	}

	// 构造 IPv4 层
	ip := layers.IPv4{
		Version:    4,
		IHL:        5,
		TOS:        0,
		Id:         0,
		Flags:      0,
		FragOffset: 0,
		TTL:        64,
		Protocol:   layers.IPProtocolUDP,
		SrcIP:      net.IPv4zero,
		DstIP:      dstIP,
	}

	// 构造以太网层
	eth := layers.Ethernet{
		SrcMAC:       net.HardwareAddr{0xff, 0xff, 0xff, 0xff, 0xff, 0xff},
		DstMAC:       net.HardwareAddr{0xff, 0xff, 0xff, 0xff, 0xff, 0xff},
		EthernetType: layers.EthernetTypeIPv4,
	}

	// 构造数据包
	buffer := gopacket.NewSerializeBuffer()
	if err := gopacket.SerializeLayers(buffer,
		gopacket.SerializeOptions{},
		&eth,
		&ip,
		&udp,
		&dhcp); err != nil {
		return fmt.Errorf("failed to serialize layers: %v", err)
	}

	// 发送数据包
	if err := pcap.SendPacketBuffer(iface, buffer.Bytes()); err != nil {
		return fmt.Errorf("failed to send packet: %v", err)
	}

	return nil
}

该程序首先打开指定的网络接口，然后设置过滤器以捕获 UDP 67 端口的数据包。接下来，程序会循环捕获数据包，如果捕获到的数据包是 DHCP 请求包，则会构造 DHCP 消息并发送给 DHCP 服务器。然后程序会等待 DHCP Offer 响应，如果超时则继续捕获下一个数据包。如果捕获到的数据包是 DHCP Offer 响应包，则会将其发送给 eth1 接口。注意，在发送数据包之前需要构造以太网、IPv4 和 UDP 层。

使用方法:

1. 确保已安装 gopacket 库: go get github.com/google/gopacket
2. 修改程序中的 iface、dhcpIP 和 eth1 参数以匹配你的网络环境。
3. 运行程序: go run main.go

注意:

• 该程序假设你的网络环境中只有一个 DHCP 服务器。
• 该程序只转发 DHCP 请求和 Offer 包，不会转发其他类型的 DHCP 包。
• 该程序可能会受到网络延迟的影响，导致 DHCP 响应超时。
• 该程序仅用于学习和测试目的，不建议在生产环境中使用。

代码说明:

• iface：指定捕获和发送数据包的网络接口名。
• filter：指定过滤器，只捕获 UDP 67 端口的数据包。
• snaplen：指定捕获数据包的最大长度。
• timeout：指定捕获数据包的超时时间。
• dhcpIP：指定 DHCP 服务器的 IP 地址。
• sendPacket 函数：构造并发送数据包。
• main 函数：打开网络接口，设置过滤器，捕获和转发数据包。

希望以上内容对你有所帮助。如有任何问题，请随时提出。