mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-09-30 07:46:41 -04:00
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package packetaddr
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/v2fly/v2ray-core/v4/common/buf"
|
|
"github.com/v2fly/v2ray-core/v4/common/net"
|
|
"github.com/v2fly/v2ray-core/v4/common/protocol"
|
|
gonet "net"
|
|
)
|
|
|
|
var addrParser = protocol.NewAddressParser(
|
|
protocol.AddressFamilyByte(0x01, net.AddressFamilyIPv4),
|
|
protocol.AddressFamilyByte(0x02, net.AddressFamilyIPv6),
|
|
)
|
|
|
|
func AttachAddressToPacket(data []byte, address gonet.Addr) []byte {
|
|
packetBuf := buf.StackNew()
|
|
udpaddr := address.(*gonet.UDPAddr)
|
|
port, err := net.PortFromInt(uint32(udpaddr.Port))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
err = addrParser.WriteAddressPort(&packetBuf, net.IPAddress(udpaddr.IP), port)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
data = append(packetBuf.Bytes(), data...)
|
|
packetBuf.Release()
|
|
return data
|
|
}
|
|
|
|
func ExtractAddressFromPacket(data []byte) ([]byte, gonet.Addr) {
|
|
packetBuf := buf.StackNew()
|
|
address, port, err := addrParser.ReadAddressPort(&packetBuf, bytes.NewReader(data))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
var addr = &gonet.UDPAddr{
|
|
IP: address.IP(),
|
|
Port: int(port.Value()),
|
|
Zone: "",
|
|
}
|
|
payload := data[int(packetBuf.Len()):]
|
|
packetBuf.Release()
|
|
return payload, addr
|
|
}
|