yggdrasil-network/water
1
0
Fork 0
mirror of https://github.com/yggdrasil-network/water.git synced 2025-05-19 16:35:10 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

add port accessors and modifiers for TCP/UDP

Browse source
This commit is contained in:
Sean Purser-Haskell 2016-05-09 16:17:22 +08:00
parent 6a76f249f6
commit 735b3a96b7
1 changed files with 31 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

31
waterutil/tun_ipv4.go
View file

@ -28,11 +28,42 @@ func IPv4Source(packet []byte) net.IP {
return net.IPv4(packet[12], packet[13], packet[14], packet[15]) return net.IPv4(packet[12], packet[13], packet[14], packet[15])
} }
func SetIPv4Source(packet []byte, source net.IP) {
copy(packet[12:16], source)
}
func IPv4Destination(packet []byte) net.IP { func IPv4Destination(packet []byte) net.IP {
return net.IPv4(packet[16], packet[17], packet[18], packet[19]) return net.IPv4(packet[16], packet[17], packet[18], packet[19])
} }
func SetIPv4Destination(packet []byte, dest net.IP) {
copy(packet[16:20], dest)
}
func IPv4Payload(packet []byte) []byte { func IPv4Payload(packet []byte) []byte {
ihl := packet[0] & 0x0F ihl := packet[0] & 0x0F
return packet[ihl*4:] return packet[ihl*4:]
} }
// For TCP/UDP
func IPv4SourcePort(packet []byte) uint16 {
payload := IPv4Payload(packet)
return (uint16(payload[0]) << 8) | uint16(payload[1])
}
func IPv4DestinationPort(packet []byte) uint16 {
payload := IPv4Payload(packet)
return (uint16(payload[2]) << 8) | uint16(payload[3])
}
func SetIPv4SourcePort(packet []byte, port uint16) {
payload := IPv4Payload(packet)
payload[0] = byte(port >> 8)
payload[1] = byte(port & 0xFF)
}
func SetIPv4DestinationPort(packet []byte, port uint16) {
payload := IPv4Payload(packet)
payload[2] = byte(port >> 8)
payload[3] = byte(port & 0xFF)
}