mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-30 07:46:41 -04:00
36 lines
881 B
Go
36 lines
881 B
Go
package packetaddr
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
sysnet "net"
|
|
"testing"
|
|
)
|
|
|
|
func TestPacketEncodingIPv4(t *testing.T) {
|
|
packetAddress := &sysnet.UDPAddr{
|
|
IP: sysnet.IPv4(1, 2, 3, 4).To4(),
|
|
Port: 1234,
|
|
}
|
|
var packetData [256]byte
|
|
wrapped := AttachAddressToPacket(packetData[:], packetAddress)
|
|
|
|
packetPayload, decodedAddress := ExtractAddressFromPacket(wrapped)
|
|
|
|
assert.Equal(t, packetPayload, packetData[:])
|
|
assert.Equal(t, packetAddress, decodedAddress)
|
|
}
|
|
|
|
func TestPacketEncodingIPv6(t *testing.T) {
|
|
packetAddress := &sysnet.UDPAddr{
|
|
IP: sysnet.IPv6linklocalallrouters,
|
|
Port: 1234,
|
|
}
|
|
var packetData [256]byte
|
|
wrapped := AttachAddressToPacket(packetData[:], packetAddress)
|
|
|
|
packetPayload, decodedAddress := ExtractAddressFromPacket(wrapped)
|
|
|
|
assert.Equal(t, packetPayload, packetData[:])
|
|
assert.Equal(t, packetAddress, decodedAddress)
|
|
}
|