extern/netbird
1
0
Fork 0
mirror of https://github.com/netbirdio/netbird.git synced 2025-08-09 07:15:15 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use buf pool

Browse Source 
- eliminate reader function generation
- fix write to closed channel panic
This commit is contained in:
Zoltan Papp
2024-06-09 20:27:40 +02:00
parent 8c70b7d7ff
commit 5e93d117cf
3 changed files with 78 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
relay/messages/message.go
View File

@ -110,12 +110,13 @@ func MarshalTransportMsg(peerID []byte, payload []byte) []byte {
return msg
}
func UnmarshalTransportPayload(buf []byte) ([]byte, error) {
func UnmarshalTransportMsg(buf []byte) ([]byte, []byte, error) {
headerSize := 1 + IDSize
if len(buf) < headerSize {
return nil, ErrInvalidMessageLength
return nil, nil, ErrInvalidMessageLength
}
return buf[headerSize:], nil
return buf[1:headerSize], buf[headerSize:], nil
}
func UnmarshalTransportID(buf []byte) ([]byte, error) {