void/amneziawg-go
2
0
Fork 0
mirror of https://github.com/amnezia-vpn/amneziawg-go.git synced 2025-07-30 16:42:50 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

improve msgType lookup

Browse source
This commit is contained in:
Mark Puha 2023-09-16 07:45:48 +02:00
parent 336282f299
commit df87fbfcd6
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
device/receive.go
View file

@ -137,10 +137,16 @@ func (device *Device) RoutineReceiveIncoming(
packet := bufsArrs[i][:size]
var msgType uint32
if device.isAdvancedSecurityOn() {
var junkSize int
if mapMsgType, ok := packetSizeToMsgType[size]; ok {
junkSize = msgTypeToJunkSize[mapMsgType]
msgType = mapMsgType
if assumedMsgType, ok := packetSizeToMsgType[size]; ok {
junkSize := msgTypeToJunkSize[assumedMsgType]
// transport size can align with other header types;
// making sure we have the right msgType
msgType = binary.LittleEndian.Uint32(packet[junkSize:4])
if msgType == assumedMsgType {
packet = packet[junkSize:]
} else {
msgType = binary.LittleEndian.Uint32(packet[:4])
}
} else {
msgType = binary.LittleEndian.Uint32(packet[:4])
if msgType != MessageTransportType {
@ -148,8 +154,6 @@ func (device *Device) RoutineReceiveIncoming(
continue
}
}
// shift junk
packet = packet[junkSize:]
} else {
msgType = binary.LittleEndian.Uint32(packet[:4])
}