mirror of
https://github.com/v2fly/v2ray-core.git
synced 2025-07-26 11:44:22 -04:00
22 lines
487 B
Go
22 lines
487 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)
|
||
|
|
}
|